1 /*
2  * Copyright (c) 2010-2012 Niels Provos and Nick Mathewson
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  * 3. The name of the author may not be used to endorse or promote products
13  *    derived from this software without specific prior written permission.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
27 #ifndef REGRESS_TESTUTILS_H_INCLUDED_
28 #define REGRESS_TESTUTILS_H_INCLUDED_
29 
30 #include "event2/dns.h"
31 
32 struct regress_dns_server_table {
33 	const char *q;
34 	const char *anstype;
35 	const char *ans;
36 	int seen;
37 	int lower;
38 };
39 
40 struct evdns_server_port *
41 regress_get_dnsserver(struct event_base *base,
42     ev_uint16_t *portnum,
43     evutil_socket_t *psock,
44     evdns_request_callback_fn_type cb,
45     void *arg);
46 
47 /* Helper: return the port that a socket is bound on, in host order. */
48 int regress_get_socket_port(evutil_socket_t fd);
49 
50 /* used to look up pre-canned responses in a search table */
51 void regress_dns_server_cb(
52 	struct evdns_server_request *req, void *data);
53 
54 /* globally allocates a dns server that serves from a search table */
55 int regress_dnsserver(struct event_base *base, ev_uint16_t *port,
56     struct regress_dns_server_table *seach_table);
57 
58 /* clean up the global dns server resources */
59 void regress_clean_dnsserver(void);
60 
61 struct evconnlistener;
62 struct sockaddr;
63 int regress_get_listener_addr(struct evconnlistener *lev,
64     struct sockaddr *sa, ev_socklen_t *socklen);
65 
66 #endif /* REGRESS_TESTUTILS_H_INCLUDED_ */
67 
68