]> git.meshlink.io Git - catta/blob - dns-spin-test.c
d93e508bbf4bf58bca2a697d56d4bf5e0623c286
[catta] / dns-spin-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 /* Regression test for Avahi bug #84.
23  * This program tests whether the avahi_dns_packet_consume_name function
24  * returns (rather than spinning forever). For a function as simple as
25  * avahi_dns_packet_consume_name, we assume that 1 second of CPU time ≈ forever
26  */
27
28 #ifdef HAVE_CONFIG_H
29 # include <config.h>
30 #endif
31
32 #include <string.h>
33 #include <stdint.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <signal.h>
37 #include <sys/time.h>
38
39 #include "dns.h"
40
41 #define MAX_CPU_SECONDS 1
42
43 #define TEST_NAME "dns-spin-test"
44
45 static void fail(const char *fmt, ...) __attribute__((format(printf, 1, 2), noreturn));
46 static void unresolved(const char *fmt, ...) __attribute__((format(printf, 1, 2), noreturn));
47 static void stdlib_fail(const char *msg) __attribute__((noreturn));
48 static void handle(int sig) __attribute__((noreturn));
49
50 void stdlib_fail(const char *msg) {
51     perror(msg);
52
53     printf("UNRESOLVED: " TEST_NAME " (stdlib failure)\n");
54
55     exit(77);
56 }
57
58 void unresolved(const char *fmt, ...) {
59     va_list ap;
60
61     printf("UNRESOLVED: " TEST_NAME ": ");
62     va_start(ap, fmt);
63     vprintf(fmt, ap);
64     va_end(ap);
65     printf("\n");
66
67     exit(77);
68 }
69
70 void fail(const char *fmt, ...) {
71     va_list ap;
72
73     va_start(ap, fmt);
74     vprintf(fmt, ap);
75     va_end(ap);
76     printf("\n");
77
78     exit(EXIT_FAILURE);
79 }
80
81 void handle(AVAHI_GCC_UNUSED int sig) {
82     fail("Interrupted after %d second of CPU time", MAX_CPU_SECONDS);
83 }
84
85 #define TRY_EXCEPT(cmd, badresult) \
86     do { \
87         if ((cmd) == (badresult)) \
88             unresolved("%s returned %s", #cmd, #badresult); \
89     } while (0)
90
91 int main(AVAHI_GCC_UNUSED int argc, AVAHI_GCC_UNUSED char *argv[]) {
92     struct itimerval itval;
93     AvahiDnsPacket *packet;
94     char name[512];
95     int ret;
96     uint8_t badrr[] = {
97         0xC0, AVAHI_DNS_PACKET_HEADER_SIZE, /* self-referential QNAME pointer */
98         0, 1, /* QTYPE A (host addr) */
99         0, 1, /* QCLASS IN (internet/ipv4) */
100     };
101
102     if (signal(SIGVTALRM, handle) == SIG_ERR)
103         stdlib_fail("signal(SIGVTALRM)");
104
105     memset(&itval, 0, sizeof(itval));
106     itval.it_value.tv_sec = MAX_CPU_SECONDS;
107
108     if (setitimer(ITIMER_VIRTUAL, &itval, NULL) == -1)
109         stdlib_fail("setitimer()");
110
111     TRY_EXCEPT(packet = avahi_dns_packet_new_query(512), NULL);
112     TRY_EXCEPT(avahi_dns_packet_append_bytes(packet, badrr, sizeof(badrr)), NULL);
113
114     /* This is expected to fail (if it returns) */
115     ret = avahi_dns_packet_consume_name(packet, name, sizeof(name));
116
117     if (ret != -1)
118         fail("avahi_dns_packet_consume_name() returned %d; -1 was expected", ret);
119
120     return EXIT_SUCCESS;
121 }
122
123 /* vim:ts=4:sw=4:et
124  */