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
92
93
94
|
#include <dnswire/writer.h>
#include <assert.h>
int main(void)
{
uint8_t buf[1], buf2[1];
size_t s;
struct dnswire_writer w;
assert(dnswire_writer_init(&w) == dnswire_ok);
// set bidirectional
assert(dnswire_writer_set_bidirectional(&w, true) == dnswire_ok);
assert(dnswire_writer_set_bidirectional(&w, false) == dnswire_ok);
// set bufsize
w.left = 2;
assert(dnswire_writer_set_bufsize(&w, 1) == dnswire_error);
w.left = 1;
w.at = 1;
assert(dnswire_writer_set_bufsize(&w, 1) == dnswire_ok);
w.left = 0;
assert(dnswire_writer_set_bufsize(&w, DNSWIRE_MAXIMUM_BUF_SIZE + 1) == dnswire_error);
assert(dnswire_writer_set_bufsize(&w, DNSWIRE_DEFAULT_BUF_SIZE) == dnswire_ok);
// set bufinc
assert(dnswire_writer_set_bufinc(&w, DNSWIRE_DEFAULT_BUF_SIZE) == dnswire_ok);
// set bufmax
assert(dnswire_writer_set_bufmax(&w, DNSWIRE_DEFAULT_BUF_SIZE - 1) == dnswire_error);
assert(dnswire_writer_set_bufmax(&w, DNSWIRE_MAXIMUM_BUF_SIZE) == dnswire_ok);
// writer pop
w.state = dnswire_writer_encoding_ready;
s = sizeof(buf2);
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_again);
w.state = dnswire_writer_reading_accept;
s = 0;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_need_more);
s = sizeof(buf2);
w.read_left = 1;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_need_more);
w.state = dnswire_writer_stopping;
w.read_left = 1;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_again);
w.read_left = 0;
w.state = dnswire_writer_reading_finish;
s = 0;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_need_more);
s = sizeof(buf2);
w.read_left = 1;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_need_more);
w.decoder.state = dnswire_decoder_done;
w.state = dnswire_writer_decoding_accept;
w.read_left = 1;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_error);
w.state = dnswire_writer_decoding_finish;
w.read_left = 1;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_error);
w.state = dnswire_writer_done;
assert(dnswire_writer_pop(&w, buf, 1, buf2, &s) == dnswire_error);
// reset writer
dnswire_writer_destroy(w);
assert(dnswire_writer_init(&w) == dnswire_ok);
assert(dnswire_writer_set_bidirectional(&w, true) == dnswire_ok);
// writer write
w.state = dnswire_writer_writing_ready;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_reading_accept;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_writing;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_stopping;
w.left = 1;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_writing_stop;
w.left = 1;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.left = 0;
w.state = dnswire_writer_reading_finish;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.read_left = 1;
w.decoder.state = dnswire_decoder_done;
w.state = dnswire_writer_decoding_accept;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_decoding_finish;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
w.state = dnswire_writer_done;
assert(dnswire_writer_write(&w, -1) == dnswire_error);
return 0;
}
|
