From 2edf2937403e6448f155b31403a4c43d5fe277d4 Mon Sep 17 00:00:00 2001 From: Joel Grunbaum <joelgrun@gmail.com> Date: Sun, 18 Oct 2020 13:03:16 +0000 Subject: [PATCH] added command line arguments --- dns.c | 19 ++++++++----------- 1 files changed, 8 insertions(+), 11 deletions(-) diff --git a/dns.c b/dns.c index 042902b..068753f 100644 --- a/dns.c +++ b/dns.c @@ -9,6 +9,9 @@ #include <unistd.h> #include "dns.h" +void change_to_DNS_name_format(unsigned char* dns, unsigned char* host); +char* read_name(unsigned char* reader, unsigned char* buffer, int* count); + // DNS code copied from // https://gist.github.com/fffaraz/9d9170b57791c28ccda9255b48315168 @@ -26,7 +29,7 @@ unsigned char ad :1; //authenticated data unsigned char z :1; //reserved for future use unsigned char ra :1; //recursion available - unsigned short q_count; //number of question entries + unsigned short q_count; //number of question entrise unsigned short ans_count; //number of answer entries unsigned short auth_count; //number of authority entries unsigned short add_count; //number of resource entries @@ -65,13 +68,13 @@ unsigned char* rdata; }; -void resolve(char* hostname, char* dns_ip, int query_type, int read) +void resolve(unsigned char* buf, char* hostname, char* dns_ip, int query_type) { int s, i, j; struct sockaddr_in dest, a; - unsigned char buf[65536], *qname, *reader; - struct DNS_HEADER* dns = NULL; - struct QUESTION* qinfo = NULL; + unsigned char *qname, *reader; + struct DNS_HEADER* dns = (struct DNS_HEADER*)buf; + struct QUESTION* qinfo; s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); dest.sin_family = AF_INET; @@ -79,7 +82,6 @@ dest.sin_addr.s_addr = inet_addr(dns_ip); //dns packet header - dns = (struct DNS_HEADER*)&buf; dns->id = (unsigned short) htons(getpid()); dns->qr = 0; //make query dns->opcode = 0; //standard query @@ -111,11 +113,6 @@ //negative return is a fail i = sizeof(dest); recvfrom(s, (char*)buf, 65536, 0, (struct sockaddr*)&dest, (socklen_t*)&i); - - //read response - if (read) { - print_packet(buf); - } return; } -- Gitblit v1.10.0