Added list for dns servers and hostnames

Joel Grunbaum

2020-10-19 415e3b82ed70c48f61b51b7a93576335af2da7cb

 main.c

@@ -4,51 +4,83 @@
#include <string.h>
#include <time.h>
#include "dns.h"
#include "servers.h"
#include "slist.h"

#define NUM_TESTS 10

struct timespec test_server(char *dns_addr, struct hosts_list *hosts);

int main(int argc, char** argv)
{
    int option, type = T_A;
    char* server = "1.1.1.1", *hostname = "google.com";
    while((option = getopt(argc, argv, "s:h:t:")) != -1) {
        switch (option) {
        case 's': //server to use
            server = optarg;
            break;
        case 'h': //hostname to search
            hostname = optarg;
            break;
        case 't':
            if (!strcmp(optarg, "A")) {
               type = T_A;
            } else if (!strcmp(optarg, "AAAA")) {
               type = T_AAAA;
            } else if (!strcmp(optarg, "NS")) {
               type = T_NS;
            } else if (!strcmp(optarg, "CNAME")) {
               type = T_CNAME;
            } else if (!strcmp(optarg, "SOA")) {
               type = T_SOA;
            } else if (!strcmp(optarg, "PTR")) {
               type = T_PTR;
            } else if (!strcmp(optarg, "MX")) {
               type = T_MX;
            } else if (!strcmp(optarg, "TXT")) {
               type = T_MX;
            } else {
               printf("Error: %s is not a valid DNS record type\n", optarg);
               exit(1);
            }
            break;
        case '?':
        default:
            printf("Error: invalid option -%c\n", optopt);
            exit(1);
        }
    }
    unsigned char buf[65536];
    struct timespec time;
    time = resolve(buf, hostname, server, type);
    printf("Request took %ld.%09lds\n", time.tv_sec, time.tv_nsec);
    print_packet(buf);
    return 0;
   int option, type = T_A;
   char* server = "1.1.1.1", *hostname = "google.com";
   struct hosts_list *servers_hosts = NULL;
   struct dns_list *servers_dns = NULL;
   while((option = getopt(argc, argv, "s:h:t:")) != -1) {
      switch (option) {
      case 's': //server to use
         add_dns_server(&servers_dns, optarg);
         break;
      case 'h': //hostname to search
         add_hosts_server(&servers_hosts, optarg);
         break;
      case 't':
         if (!strcmp(optarg, "A")) {
            type = T_A;
         } else if (!strcmp(optarg, "AAAA")) {
            type = T_AAAA;
         } else if (!strcmp(optarg, "NS")) {
            type = T_NS;
         } else if (!strcmp(optarg, "CNAME")) {
            type = T_CNAME;
         } else if (!strcmp(optarg, "SOA")) {
            type = T_SOA;
         } else if (!strcmp(optarg, "PTR")) {
            type = T_PTR;
         } else if (!strcmp(optarg, "MX")) {
            type = T_MX;
         } else if (!strcmp(optarg, "TXT")) {
            type = T_TXT;
         } else {
            printf("Error: %s is not a valid DNS record type\n", optarg);
            free_hosts_list(&servers_hosts);
            free_dns_list(&servers_dns);
            exit(1);
         }
         break;
      case '?':
      default:
         printf("Error: invalid option -%c\n", optopt);
         exit(1);
      }
   }
   struct timespec total;
   total = test_server(server, servers_hosts);
   printf("Test took %ld.%09lds\n", total.tv_sec, total.tv_nsec);
   free_dns_list(&servers_dns);
   free_hosts_list(&servers_hosts);
   return 0;
}

struct timespec test_server(char* dns_addr, struct hosts_list *hosts)
{
   struct timespec total;
   total.tv_sec = 0; total.tv_nsec = 0;
   for (int i = 0; i < NUM_TESTS; i++) {
      struct hosts_list *curr = hosts;
      while (curr) {
         struct timespec run;
         unsigned char buf[65536];
         run = resolve(buf, curr->server, dns_addr, T_A);
         total.tv_sec += run.tv_sec;
         total.tv_nsec += run.tv_nsec;
         if (total.tv_nsec >= 1000000000) {
            total.tv_sec += 1;
            total.tv_nsec -= 1000000000;
         }
         curr = curr->next;
      }
   }
   return total;
}