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
80
81
82
83
84
85
86
87
88
89
90
91
|
// RUN: %clangxx %s -o %t && %run %t %p
#include <assert.h>
#include <resolv.h>
#include <string.h>
#include "sanitizer_common/sanitizer_specific.h"
void testWrite() {
char unsigned input[] = {0xff, 0xc5, 0xf7, 0xff, 0x00, 0x00, 0xff, 0x0a, 0x00,
0x00, 0x00, 0x01, 0x00, 0x00, 0x02, 0x00, 0x01, 0x00,
0x10, 0x01, 0x05, 0x00, 0x01, 0x0a, 0x67, 0x6f, 0x6f,
0x67, 0x6c, 0x65, 0x2e, 0x63, 0x6f, 0x6d, 0x00};
char output[1024];
int res = dn_expand(input, input + sizeof(input), input + 23, output,
sizeof(output));
assert(res == 12);
assert(strcmp(output, "google\\.com") == 0);
check_mem_is_good(output, strlen(output) + 1);
}
void testWriteZeroLength() {
char unsigned input[] = {
0xff, 0xc5, 0xf7, 0xff, 0x00, 0x00, 0xff, 0x0a, 0x00, 0x00, 0x00, 0x01,
0x00, 0x00, 0x02, 0x00, 0x01, 0x00, 0x10, 0x01, 0x05, 0x00, 0x01, 0x00,
};
char output[1024];
int res = dn_expand(input, input + sizeof(input), input + 23, output,
sizeof(output));
assert(res == 1);
assert(strcmp(output, "") == 0);
check_mem_is_good(output, strlen(output) + 1);
}
void testComp() {
char unsigned msg[1024];
char unsigned *mb = msg;
char unsigned *me = msg + sizeof(msg);
char unsigned **pb = (char unsigned **)mb;
pb[0] = msg;
pb[1] = nullptr;
mb += 64;
char unsigned **pe = (char unsigned **)mb;
char unsigned *n1 = mb;
int res = dn_comp("llvm.org", mb, me - mb, pb, pe);
assert(res == 10);
check_mem_is_good(mb, res);
// pb is [msg, llvm.org, nullptr]
check_mem_is_good(pb, sizeof(*pb) * 3);
mb += res;
char unsigned *n2 = mb;
res = dn_comp("lab.llvm.org", mb, me - mb, pb, pe);
assert(res == 6);
check_mem_is_good(mb, res);
// pb is [msg, llvm.org, lab.llvm.org, nullptr]
check_mem_is_good(pb, sizeof(*pb) * 4);
mb += res;
{
char output[1024];
res = dn_expand(msg, msg + sizeof(msg), n1, output, sizeof(output));
fprintf(stderr, "%d\n", res);
assert(res == 10);
assert(strcmp(output, "llvm.org") == 0);
check_mem_is_good(output, strlen(output) + 1);
}
{
char output[1024];
res = dn_expand(msg, msg + sizeof(msg), n2, output, sizeof(output));
assert(res == 6);
assert(strcmp(output, "lab.llvm.org") == 0);
check_mem_is_good(output, strlen(output) + 1);
}
}
int main(int iArgc, const char *szArgv[]) {
testWrite();
testWriteZeroLength();
testComp();
return 0;
}
|
