]> git.meshlink.io Git - catta/blob - avahi-common/strlst-test.c
Add validity checking to TXT data parsing, this fixes a remotely exploitable vulnerab...
[catta] / avahi-common / strlst-test.c
1 /* $Id$ */
2
3 /***
4   This file is part of avahi.
5  
6   avahi is free software; you can redistribute it and/or modify it
7   under the terms of the GNU Lesser General Public License as
8   published by the Free Software Foundation; either version 2.1 of the
9   License, or (at your option) any later version.
10  
11   avahi is distributed in the hope that it will be useful, but WITHOUT
12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
14   Public License for more details.
15  
16   You should have received a copy of the GNU Lesser General Public
17   License along with avahi; if not, write to the Free Software
18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19   USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdio.h>
27 #include <assert.h>
28
29 #include "strlst.h"
30 #include "malloc.h"
31
32 int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char *argv[]) {
33     char *t, *v;
34     uint8_t data[1024];
35     AvahiStringList *a = NULL, *b, *p;
36     size_t size, n;
37     int r;
38
39     a = avahi_string_list_new("prefix", "a", "b", NULL);
40     
41     a = avahi_string_list_add(a, "start");
42     a = avahi_string_list_add(a, "foo=99");
43     a = avahi_string_list_add(a, "bar");
44     a = avahi_string_list_add(a, "quux");
45     a = avahi_string_list_add_arbitrary(a, (const uint8_t*) "null\0null", 9);
46     a = avahi_string_list_add_printf(a, "seven=%i %c", 7, 'x');
47     a = avahi_string_list_add_pair(a, "blubb", "blaa");
48     a = avahi_string_list_add_pair(a, "uxknurz", NULL);
49     a = avahi_string_list_add_pair_arbitrary(a, "uxknurz2", (const uint8_t*) "blafasel\0oerks", 14);
50     
51     a = avahi_string_list_add(a, "end");
52
53     t = avahi_string_list_to_string(a);
54     printf("--%s--\n", t);
55     avahi_free(t);
56     
57     n = avahi_string_list_serialize(a, NULL, 0);
58     size = avahi_string_list_serialize(a, data, sizeof(data));
59     assert(size == n);
60
61     printf("%u\n", size);
62
63     for (t = (char*) data, n = 0; n < size; n++, t++) {
64         if (*t <= 32)
65             printf("(%u)", *t);
66         else
67             printf("%c", *t);
68     }
69
70     printf("\n");
71     
72     assert(avahi_string_list_parse(data, size, &b) == 0);
73
74     assert(avahi_string_list_equal(a, b));
75     
76     t = avahi_string_list_to_string(b);
77     printf("--%s--\n", t);
78     avahi_free(t);
79
80     avahi_string_list_free(b);
81
82     b = avahi_string_list_copy(a);
83
84     assert(avahi_string_list_equal(a, b));
85
86     t = avahi_string_list_to_string(b);
87     printf("--%s--\n", t);
88     avahi_free(t);
89
90     p = avahi_string_list_find(a, "seven");
91     assert(p);
92     
93     r = avahi_string_list_get_pair(p, &t, &v, NULL);  
94     assert(r >= 0);
95     assert(t);
96     assert(v);
97     
98     printf("<%s>=<%s>\n", t, v);
99     avahi_free(t);
100     avahi_free(v);
101
102     p = avahi_string_list_find(a, "quux");
103     assert(p);
104
105     r = avahi_string_list_get_pair(p, &t, &v, NULL);
106     assert(r >= 0);
107     assert(t);
108     assert(!v);
109
110     printf("<%s>=<%s>\n", t, v);
111     avahi_free(t);
112     avahi_free(v);
113     
114     avahi_string_list_free(a);
115     avahi_string_list_free(b);
116
117     n = avahi_string_list_serialize(NULL, NULL, 0);
118     size = avahi_string_list_serialize(NULL, data, sizeof(data));
119     assert(size == 1);
120     assert(size == n);
121
122     assert(avahi_string_list_parse(data, size, &a) == 0);
123     assert(!a);
124     
125     return 0;
126 }