Added list for dns servers and hostnames

Joel Grunbaum

2020-10-19 415e3b82ed70c48f61b51b7a93576335af2da7cb

 main.c

@@ -4,18 +4,26 @@
#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";
   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
            server = optarg;
         add_dns_server(&servers_dns, optarg);
            break;
        case 'h': //hostname to search
            hostname = optarg;
         add_hosts_server(&servers_hosts, optarg);
            break;
        case 't':
            if (!strcmp(optarg, "A")) {
@@ -33,9 +41,11 @@
            } else if (!strcmp(optarg, "MX")) {
               type = T_MX;
            } else if (!strcmp(optarg, "TXT")) {
               type = T_MX;
            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;
@@ -45,10 +55,32 @@
            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);
   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;
}