This source file includes following definitions.
- regress_get_socket_port
- regress_get_dnsserver
- regress_clean_dnsserver
- regress_dns_server_cb
- regress_dnsserver
- regress_get_listener_addr
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 #ifdef WIN32
28 #include <winsock2.h>
29 #include <windows.h>
30 #include <ws2tcpip.h>
31 #endif
32
33 #include "event2/event-config.h"
34
35 #include <sys/types.h>
36 #include <sys/stat.h>
37 #ifdef _EVENT_HAVE_SYS_TIME_H
38 #include <sys/time.h>
39 #endif
40 #include <sys/queue.h>
41 #ifndef WIN32
42 #include <sys/socket.h>
43 #include <signal.h>
44 #include <netinet/in.h>
45 #include <arpa/inet.h>
46 #include <unistd.h>
47 #endif
48 #ifdef _EVENT_HAVE_NETINET_IN6_H
49 #include <netinet/in6.h>
50 #endif
51 #ifdef HAVE_NETDB_H
52 #include <netdb.h>
53 #endif
54 #include <fcntl.h>
55 #include <stdlib.h>
56 #include <stdio.h>
57 #include <string.h>
58 #include <errno.h>
59
60 #include "event2/dns.h"
61 #include "event2/dns_struct.h"
62 #include "event2/event.h"
63 #include "event2/event_compat.h"
64 #include "event2/util.h"
65 #include "event2/listener.h"
66 #include "event2/bufferevent.h"
67 #include "log-internal.h"
68 #include "regress.h"
69 #include "regress_testutils.h"
70
71 #include "../util-internal.h"
72
73
74 static struct evdns_server_port *dns_port;
75 evutil_socket_t dns_sock = -1;
76
77
78 int
79 regress_get_socket_port(evutil_socket_t fd)
80 {
81 struct sockaddr_storage ss;
82 ev_socklen_t socklen = sizeof(ss);
83 if (getsockname(fd, (struct sockaddr*)&ss, &socklen) != 0)
84 return -1;
85 if (ss.ss_family == AF_INET)
86 return ntohs( ((struct sockaddr_in*)&ss)->sin_port);
87 else if (ss.ss_family == AF_INET6)
88 return ntohs( ((struct sockaddr_in6*)&ss)->sin6_port);
89 else
90 return -1;
91 }
92
93 struct evdns_server_port *
94 regress_get_dnsserver(struct event_base *base,
95 ev_uint16_t *portnum,
96 evutil_socket_t *psock,
97 evdns_request_callback_fn_type cb,
98 void *arg)
99 {
100 struct evdns_server_port *port = NULL;
101 evutil_socket_t sock;
102 struct sockaddr_in my_addr;
103
104 sock = socket(AF_INET, SOCK_DGRAM, 0);
105 if (sock < 0) {
106 tt_abort_perror("socket");
107 }
108
109 evutil_make_socket_nonblocking(sock);
110
111 memset(&my_addr, 0, sizeof(my_addr));
112 my_addr.sin_family = AF_INET;
113 my_addr.sin_port = htons(*portnum);
114 my_addr.sin_addr.s_addr = htonl(0x7f000001UL);
115 if (bind(sock, (struct sockaddr*)&my_addr, sizeof(my_addr)) < 0) {
116 evutil_closesocket(sock);
117 tt_abort_perror("bind");
118 }
119 port = evdns_add_server_port_with_base(base, sock, 0, cb, arg);
120 if (!*portnum)
121 *portnum = regress_get_socket_port(sock);
122 if (psock)
123 *psock = sock;
124
125 return port;
126 end:
127 return NULL;
128 }
129
130 void
131 regress_clean_dnsserver(void)
132 {
133 if (dns_port)
134 evdns_close_server_port(dns_port);
135 if (dns_sock >= 0)
136 evutil_closesocket(dns_sock);
137 }
138
139 void
140 regress_dns_server_cb(struct evdns_server_request *req, void *data)
141 {
142 struct regress_dns_server_table *tab = data;
143 const char *question;
144
145 if (req->nquestions != 1)
146 TT_DIE(("Only handling one question at a time; got %d",
147 req->nquestions));
148
149 question = req->questions[0]->name;
150
151 while (tab->q && evutil_ascii_strcasecmp(question, tab->q) &&
152 strcmp("*", tab->q))
153 ++tab;
154 if (tab->q == NULL)
155 TT_DIE(("Unexpected question: '%s'", question));
156
157 ++tab->seen;
158
159 if (!strcmp(tab->anstype, "err")) {
160 int err = atoi(tab->ans);
161 tt_assert(! evdns_server_request_respond(req, err));
162 return;
163 } else if (!strcmp(tab->anstype, "errsoa")) {
164 int err = atoi(tab->ans);
165 char soa_record[] =
166 "\x04" "dns1" "\x05" "icann" "\x03" "org" "\0"
167 "\x0a" "hostmaster" "\x05" "icann" "\x03" "org" "\0"
168 "\x77\xde\x5e\xba"
169 "\x00\x00\x1c\x20"
170 "\x00\x00\x0e\x10"
171 "\x00\x12\x75\x00"
172 "\x00\x00\x0e\x10"
173 ;
174 evdns_server_request_add_reply(
175 req, EVDNS_AUTHORITY_SECTION,
176 "example.com", EVDNS_TYPE_SOA, EVDNS_CLASS_INET,
177 42, sizeof(soa_record) - 1, 0, soa_record);
178 tt_assert(! evdns_server_request_respond(req, err));
179 return;
180 } else if (!strcmp(tab->anstype, "A")) {
181 struct in_addr in;
182 if (!evutil_inet_pton(AF_INET, tab->ans, &in)) {
183 TT_DIE(("Bad A value %s in table", tab->ans));
184 }
185 evdns_server_request_add_a_reply(req, question, 1, &in.s_addr,
186 100);
187 } else if (!strcmp(tab->anstype, "AAAA")) {
188 struct in6_addr in6;
189 if (!evutil_inet_pton(AF_INET6, tab->ans, &in6)) {
190 TT_DIE(("Bad AAAA value %s in table", tab->ans));
191 }
192 evdns_server_request_add_aaaa_reply(req,
193 question, 1, &in6.s6_addr, 100);
194 } else {
195 TT_DIE(("Weird table entry with type '%s'", tab->anstype));
196 }
197 tt_assert(! evdns_server_request_respond(req, 0))
198 return;
199 end:
200 tt_want(! evdns_server_request_drop(req));
201 }
202
203 int
204 regress_dnsserver(struct event_base *base, ev_uint16_t *port,
205 struct regress_dns_server_table *search_table)
206 {
207 dns_port = regress_get_dnsserver(base, port, &dns_sock,
208 regress_dns_server_cb, search_table);
209 return dns_port != NULL;
210 }
211
212 int
213 regress_get_listener_addr(struct evconnlistener *lev,
214 struct sockaddr *sa, ev_socklen_t *socklen)
215 {
216 evutil_socket_t s = evconnlistener_get_fd(lev);
217 if (s <= 0)
218 return -1;
219 return getsockname(s, sa, socklen);
220 }