1*00b67f09SDavid van Moolenbroek /* $NetBSD: isctest.c,v 1.1.1.7 2014/12/10 03:34:44 christos Exp $ */
2*00b67f09SDavid van Moolenbroek
3*00b67f09SDavid van Moolenbroek /*
4*00b67f09SDavid van Moolenbroek * Copyright (C) 2011-2014 Internet Systems Consortium, Inc. ("ISC")
5*00b67f09SDavid van Moolenbroek *
6*00b67f09SDavid van Moolenbroek * Permission to use, copy, modify, and/or distribute this software for any
7*00b67f09SDavid van Moolenbroek * purpose with or without fee is hereby granted, provided that the above
8*00b67f09SDavid van Moolenbroek * copyright notice and this permission notice appear in all copies.
9*00b67f09SDavid van Moolenbroek *
10*00b67f09SDavid van Moolenbroek * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
11*00b67f09SDavid van Moolenbroek * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
12*00b67f09SDavid van Moolenbroek * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
13*00b67f09SDavid van Moolenbroek * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
14*00b67f09SDavid van Moolenbroek * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
15*00b67f09SDavid van Moolenbroek * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
16*00b67f09SDavid van Moolenbroek * PERFORMANCE OF THIS SOFTWARE.
17*00b67f09SDavid van Moolenbroek */
18*00b67f09SDavid van Moolenbroek
19*00b67f09SDavid van Moolenbroek /* Id */
20*00b67f09SDavid van Moolenbroek
21*00b67f09SDavid van Moolenbroek /*! \file */
22*00b67f09SDavid van Moolenbroek
23*00b67f09SDavid van Moolenbroek #include <config.h>
24*00b67f09SDavid van Moolenbroek
25*00b67f09SDavid van Moolenbroek #include <time.h>
26*00b67f09SDavid van Moolenbroek
27*00b67f09SDavid van Moolenbroek #include <isc/app.h>
28*00b67f09SDavid van Moolenbroek #include <isc/buffer.h>
29*00b67f09SDavid van Moolenbroek #include <isc/entropy.h>
30*00b67f09SDavid van Moolenbroek #include <isc/hash.h>
31*00b67f09SDavid van Moolenbroek #include <isc/mem.h>
32*00b67f09SDavid van Moolenbroek #include <isc/os.h>
33*00b67f09SDavid van Moolenbroek #include <isc/socket.h>
34*00b67f09SDavid van Moolenbroek #include <isc/string.h>
35*00b67f09SDavid van Moolenbroek #include <isc/task.h>
36*00b67f09SDavid van Moolenbroek #include <isc/timer.h>
37*00b67f09SDavid van Moolenbroek #include <isc/util.h>
38*00b67f09SDavid van Moolenbroek
39*00b67f09SDavid van Moolenbroek #include "isctest.h"
40*00b67f09SDavid van Moolenbroek
41*00b67f09SDavid van Moolenbroek isc_mem_t *mctx = NULL;
42*00b67f09SDavid van Moolenbroek isc_entropy_t *ectx = NULL;
43*00b67f09SDavid van Moolenbroek isc_log_t *lctx = NULL;
44*00b67f09SDavid van Moolenbroek isc_taskmgr_t *taskmgr = NULL;
45*00b67f09SDavid van Moolenbroek isc_timermgr_t *timermgr = NULL;
46*00b67f09SDavid van Moolenbroek isc_socketmgr_t *socketmgr = NULL;
47*00b67f09SDavid van Moolenbroek isc_task_t *maintask = NULL;
48*00b67f09SDavid van Moolenbroek int ncpus;
49*00b67f09SDavid van Moolenbroek
50*00b67f09SDavid van Moolenbroek static isc_boolean_t hash_active = ISC_FALSE;
51*00b67f09SDavid van Moolenbroek
52*00b67f09SDavid van Moolenbroek /*
53*00b67f09SDavid van Moolenbroek * Logging categories: this needs to match the list in bin/named/log.c.
54*00b67f09SDavid van Moolenbroek */
55*00b67f09SDavid van Moolenbroek static isc_logcategory_t categories[] = {
56*00b67f09SDavid van Moolenbroek { "", 0 },
57*00b67f09SDavid van Moolenbroek { "client", 0 },
58*00b67f09SDavid van Moolenbroek { "network", 0 },
59*00b67f09SDavid van Moolenbroek { "update", 0 },
60*00b67f09SDavid van Moolenbroek { "queries", 0 },
61*00b67f09SDavid van Moolenbroek { "unmatched", 0 },
62*00b67f09SDavid van Moolenbroek { "update-security", 0 },
63*00b67f09SDavid van Moolenbroek { "query-errors", 0 },
64*00b67f09SDavid van Moolenbroek { NULL, 0 }
65*00b67f09SDavid van Moolenbroek };
66*00b67f09SDavid van Moolenbroek
67*00b67f09SDavid van Moolenbroek static void
cleanup_managers(void)68*00b67f09SDavid van Moolenbroek cleanup_managers(void) {
69*00b67f09SDavid van Moolenbroek if (maintask != NULL)
70*00b67f09SDavid van Moolenbroek isc_task_destroy(&maintask);
71*00b67f09SDavid van Moolenbroek if (socketmgr != NULL)
72*00b67f09SDavid van Moolenbroek isc_socketmgr_destroy(&socketmgr);
73*00b67f09SDavid van Moolenbroek if (taskmgr != NULL)
74*00b67f09SDavid van Moolenbroek isc_taskmgr_destroy(&taskmgr);
75*00b67f09SDavid van Moolenbroek if (timermgr != NULL)
76*00b67f09SDavid van Moolenbroek isc_timermgr_destroy(&timermgr);
77*00b67f09SDavid van Moolenbroek }
78*00b67f09SDavid van Moolenbroek
79*00b67f09SDavid van Moolenbroek static isc_result_t
create_managers(void)80*00b67f09SDavid van Moolenbroek create_managers(void) {
81*00b67f09SDavid van Moolenbroek isc_result_t result;
82*00b67f09SDavid van Moolenbroek #ifdef ISC_PLATFORM_USETHREADS
83*00b67f09SDavid van Moolenbroek ncpus = isc_os_ncpus();
84*00b67f09SDavid van Moolenbroek #else
85*00b67f09SDavid van Moolenbroek ncpus = 1;
86*00b67f09SDavid van Moolenbroek #endif
87*00b67f09SDavid van Moolenbroek
88*00b67f09SDavid van Moolenbroek CHECK(isc_taskmgr_create(mctx, ncpus, 0, &taskmgr));
89*00b67f09SDavid van Moolenbroek CHECK(isc_task_create(taskmgr, 0, &maintask));
90*00b67f09SDavid van Moolenbroek isc_taskmgr_setexcltask(taskmgr, maintask);
91*00b67f09SDavid van Moolenbroek
92*00b67f09SDavid van Moolenbroek CHECK(isc_timermgr_create(mctx, &timermgr));
93*00b67f09SDavid van Moolenbroek CHECK(isc_socketmgr_create(mctx, &socketmgr));
94*00b67f09SDavid van Moolenbroek return (ISC_R_SUCCESS);
95*00b67f09SDavid van Moolenbroek
96*00b67f09SDavid van Moolenbroek cleanup:
97*00b67f09SDavid van Moolenbroek cleanup_managers();
98*00b67f09SDavid van Moolenbroek return (result);
99*00b67f09SDavid van Moolenbroek }
100*00b67f09SDavid van Moolenbroek
101*00b67f09SDavid van Moolenbroek isc_result_t
isc_test_begin(FILE * logfile,isc_boolean_t start_managers)102*00b67f09SDavid van Moolenbroek isc_test_begin(FILE *logfile, isc_boolean_t start_managers) {
103*00b67f09SDavid van Moolenbroek isc_result_t result;
104*00b67f09SDavid van Moolenbroek
105*00b67f09SDavid van Moolenbroek isc_mem_debugging |= ISC_MEM_DEBUGRECORD;
106*00b67f09SDavid van Moolenbroek CHECK(isc_mem_create(0, 0, &mctx));
107*00b67f09SDavid van Moolenbroek CHECK(isc_entropy_create(mctx, &ectx));
108*00b67f09SDavid van Moolenbroek
109*00b67f09SDavid van Moolenbroek CHECK(isc_hash_create(mctx, ectx, 255));
110*00b67f09SDavid van Moolenbroek hash_active = ISC_TRUE;
111*00b67f09SDavid van Moolenbroek
112*00b67f09SDavid van Moolenbroek if (logfile != NULL) {
113*00b67f09SDavid van Moolenbroek isc_logdestination_t destination;
114*00b67f09SDavid van Moolenbroek isc_logconfig_t *logconfig = NULL;
115*00b67f09SDavid van Moolenbroek
116*00b67f09SDavid van Moolenbroek CHECK(isc_log_create(mctx, &lctx, &logconfig));
117*00b67f09SDavid van Moolenbroek isc_log_registercategories(lctx, categories);
118*00b67f09SDavid van Moolenbroek isc_log_setcontext(lctx);
119*00b67f09SDavid van Moolenbroek
120*00b67f09SDavid van Moolenbroek destination.file.stream = logfile;
121*00b67f09SDavid van Moolenbroek destination.file.name = NULL;
122*00b67f09SDavid van Moolenbroek destination.file.versions = ISC_LOG_ROLLNEVER;
123*00b67f09SDavid van Moolenbroek destination.file.maximum_size = 0;
124*00b67f09SDavid van Moolenbroek CHECK(isc_log_createchannel(logconfig, "stderr",
125*00b67f09SDavid van Moolenbroek ISC_LOG_TOFILEDESC,
126*00b67f09SDavid van Moolenbroek ISC_LOG_DYNAMIC,
127*00b67f09SDavid van Moolenbroek &destination, 0));
128*00b67f09SDavid van Moolenbroek CHECK(isc_log_usechannel(logconfig, "stderr", NULL, NULL));
129*00b67f09SDavid van Moolenbroek }
130*00b67f09SDavid van Moolenbroek
131*00b67f09SDavid van Moolenbroek #ifdef ISC_PLATFORM_USETHREADS
132*00b67f09SDavid van Moolenbroek ncpus = isc_os_ncpus();
133*00b67f09SDavid van Moolenbroek #else
134*00b67f09SDavid van Moolenbroek ncpus = 1;
135*00b67f09SDavid van Moolenbroek #endif
136*00b67f09SDavid van Moolenbroek
137*00b67f09SDavid van Moolenbroek if (start_managers)
138*00b67f09SDavid van Moolenbroek CHECK(create_managers());
139*00b67f09SDavid van Moolenbroek
140*00b67f09SDavid van Moolenbroek return (ISC_R_SUCCESS);
141*00b67f09SDavid van Moolenbroek
142*00b67f09SDavid van Moolenbroek cleanup:
143*00b67f09SDavid van Moolenbroek isc_test_end();
144*00b67f09SDavid van Moolenbroek return (result);
145*00b67f09SDavid van Moolenbroek }
146*00b67f09SDavid van Moolenbroek
147*00b67f09SDavid van Moolenbroek void
isc_test_end(void)148*00b67f09SDavid van Moolenbroek isc_test_end(void) {
149*00b67f09SDavid van Moolenbroek if (maintask != NULL)
150*00b67f09SDavid van Moolenbroek isc_task_detach(&maintask);
151*00b67f09SDavid van Moolenbroek if (taskmgr != NULL)
152*00b67f09SDavid van Moolenbroek isc_taskmgr_destroy(&taskmgr);
153*00b67f09SDavid van Moolenbroek if (lctx != NULL)
154*00b67f09SDavid van Moolenbroek isc_log_destroy(&lctx);
155*00b67f09SDavid van Moolenbroek if (hash_active) {
156*00b67f09SDavid van Moolenbroek isc_hash_destroy();
157*00b67f09SDavid van Moolenbroek hash_active = ISC_FALSE;
158*00b67f09SDavid van Moolenbroek }
159*00b67f09SDavid van Moolenbroek if (ectx != NULL)
160*00b67f09SDavid van Moolenbroek isc_entropy_detach(&ectx);
161*00b67f09SDavid van Moolenbroek
162*00b67f09SDavid van Moolenbroek cleanup_managers();
163*00b67f09SDavid van Moolenbroek
164*00b67f09SDavid van Moolenbroek if (mctx != NULL)
165*00b67f09SDavid van Moolenbroek isc_mem_destroy(&mctx);
166*00b67f09SDavid van Moolenbroek }
167*00b67f09SDavid van Moolenbroek
168*00b67f09SDavid van Moolenbroek /*
169*00b67f09SDavid van Moolenbroek * Sleep for 'usec' microseconds.
170*00b67f09SDavid van Moolenbroek */
171*00b67f09SDavid van Moolenbroek void
isc_test_nap(isc_uint32_t usec)172*00b67f09SDavid van Moolenbroek isc_test_nap(isc_uint32_t usec) {
173*00b67f09SDavid van Moolenbroek #ifdef HAVE_NANOSLEEP
174*00b67f09SDavid van Moolenbroek struct timespec ts;
175*00b67f09SDavid van Moolenbroek
176*00b67f09SDavid van Moolenbroek ts.tv_sec = usec / 1000000;
177*00b67f09SDavid van Moolenbroek ts.tv_nsec = (usec % 1000000) * 1000;
178*00b67f09SDavid van Moolenbroek nanosleep(&ts, NULL);
179*00b67f09SDavid van Moolenbroek #elif HAVE_USLEEP
180*00b67f09SDavid van Moolenbroek usleep(usec);
181*00b67f09SDavid van Moolenbroek #else
182*00b67f09SDavid van Moolenbroek /*
183*00b67f09SDavid van Moolenbroek * No fractional-second sleep function is available, so we
184*00b67f09SDavid van Moolenbroek * round up to the nearest second and sleep instead
185*00b67f09SDavid van Moolenbroek */
186*00b67f09SDavid van Moolenbroek sleep((usec / 1000000) + 1);
187*00b67f09SDavid van Moolenbroek #endif
188*00b67f09SDavid van Moolenbroek }
189