1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include "question.h"
#include "buffer.h"
#include "record.h"
void create_question(const char* domain, RecordType qtype, Question* question) {
uint8_t len = strlen(domain);
question->domain = malloc(len + 1);
question->qtype = qtype;
question->domain[0] = len;
question->cls = 1;
memcpy(question->domain + 1, domain, len);
}
bool read_question(PacketBuffer* buffer, Question* question) {
buffer_read_qname(buffer, &question->domain);
uint16_t qtype_num = buffer_read_short(buffer);
record_from_id(qtype_num, &question->qtype);
question->cls = buffer_read_short(buffer);
if (question->qtype == UNKOWN) {
free(question->domain);
return false;
}
return true;
}
void write_question(PacketBuffer* buffer, Question* question) {
buffer_write_qname(buffer, question->domain);
uint16_t id = record_to_id(question->qtype);
buffer_write_short(buffer, id);
buffer_write_short(buffer, question->cls);
}
void free_question(Question* question) {
free(question->domain);
}
static const char* class_to_str(Question* question) {
switch(question->cls) {
case 1: return "IN";
case 3: return "CH";
case 4: return "HS";
default: return "??";
}
}
static const char* qtype_to_str(Question* question) {
switch(question->qtype) {
case A: return "A";
case NS: return "NS";
case CNAME: return "CNAME";
case SOA: return "SOA";
case PTR: return "PTR";
case MX: return "MX";
case TXT: return "TXT";
case AAAA: return "AAAA";
case SRV: return "SRV";
case CAA: return "CAA";
default: return "UNKOWN";
}
}
void print_question(Question* question) {
printf("%.*s.\t%s %s\n",
question->domain[0],
question->domain + 1,
class_to_str(question),
qtype_to_str(question)
);
}
|
