rndc.c revision 1.7 1 1.5 christos /* $NetBSD: rndc.c,v 1.7 2021/08/19 11:50:15 christos Exp $ */
2 1.1 christos
3 1.1 christos /*
4 1.1 christos * Copyright (C) Internet Systems Consortium, Inc. ("ISC")
5 1.1 christos *
6 1.1 christos * This Source Code Form is subject to the terms of the Mozilla Public
7 1.1 christos * License, v. 2.0. If a copy of the MPL was not distributed with this
8 1.6 christos * file, you can obtain one at https://mozilla.org/MPL/2.0/.
9 1.1 christos *
10 1.1 christos * See the COPYRIGHT file distributed with this work for additional
11 1.1 christos * information regarding copyright ownership.
12 1.1 christos */
13 1.1 christos
14 1.1 christos /*! \file */
15 1.1 christos
16 1.3 christos #include <inttypes.h>
17 1.3 christos #include <stdbool.h>
18 1.1 christos #include <stdlib.h>
19 1.1 christos
20 1.1 christos #include <isc/app.h>
21 1.4 christos #include <isc/atomic.h>
22 1.1 christos #include <isc/buffer.h>
23 1.1 christos #include <isc/commandline.h>
24 1.1 christos #include <isc/file.h>
25 1.1 christos #include <isc/log.h>
26 1.7 christos #include <isc/managers.h>
27 1.4 christos #include <isc/mem.h>
28 1.1 christos #include <isc/net.h>
29 1.1 christos #include <isc/print.h>
30 1.1 christos #include <isc/random.h>
31 1.4 christos #include <isc/refcount.h>
32 1.1 christos #include <isc/socket.h>
33 1.1 christos #include <isc/stdtime.h>
34 1.1 christos #include <isc/string.h>
35 1.1 christos #include <isc/task.h>
36 1.1 christos #include <isc/thread.h>
37 1.1 christos #include <isc/util.h>
38 1.1 christos
39 1.1 christos #include <pk11/site.h>
40 1.1 christos
41 1.4 christos #include <dns/name.h>
42 1.1 christos
43 1.1 christos #include <isccc/alist.h>
44 1.1 christos #include <isccc/base64.h>
45 1.1 christos #include <isccc/cc.h>
46 1.1 christos #include <isccc/ccmsg.h>
47 1.1 christos #include <isccc/result.h>
48 1.1 christos #include <isccc/sexpr.h>
49 1.1 christos #include <isccc/types.h>
50 1.1 christos #include <isccc/util.h>
51 1.1 christos
52 1.4 christos #include <isccfg/namedconf.h>
53 1.1 christos
54 1.1 christos #include <bind9/getaddresses.h>
55 1.1 christos
56 1.1 christos #include "util.h"
57 1.1 christos
58 1.1 christos #define SERVERADDRS 10
59 1.1 christos
60 1.1 christos const char *progname;
61 1.3 christos bool verbose;
62 1.1 christos
63 1.1 christos static const char *admin_conffile;
64 1.1 christos static const char *admin_keyfile;
65 1.1 christos static const char *version = VERSION;
66 1.1 christos static const char *servername = NULL;
67 1.1 christos static isc_sockaddr_t serveraddrs[SERVERADDRS];
68 1.1 christos static isc_sockaddr_t local4, local6;
69 1.3 christos static bool local4set = false, local6set = false;
70 1.1 christos static int nserveraddrs;
71 1.1 christos static int currentaddr = 0;
72 1.1 christos static unsigned int remoteport = 0;
73 1.1 christos static isc_socketmgr_t *socketmgr = NULL;
74 1.1 christos static isc_buffer_t *databuf;
75 1.1 christos static isccc_ccmsg_t ccmsg;
76 1.3 christos static uint32_t algorithm;
77 1.1 christos static isccc_region_t secret;
78 1.3 christos static bool failed = false;
79 1.3 christos static bool c_flag = false;
80 1.1 christos static isc_mem_t *rndc_mctx;
81 1.4 christos static atomic_uint_fast32_t sends = ATOMIC_VAR_INIT(0);
82 1.4 christos static atomic_uint_fast32_t recvs = ATOMIC_VAR_INIT(0);
83 1.4 christos static atomic_uint_fast32_t connects = ATOMIC_VAR_INIT(0);
84 1.1 christos static char *command;
85 1.1 christos static char *args;
86 1.1 christos static char program[256];
87 1.1 christos static isc_socket_t *sock = NULL;
88 1.3 christos static uint32_t serial;
89 1.3 christos static bool quiet = false;
90 1.3 christos static bool showresult = false;
91 1.1 christos
92 1.4 christos static void
93 1.4 christos rndc_startconnect(isc_sockaddr_t *addr, isc_task_t *task);
94 1.1 christos
95 1.1 christos ISC_PLATFORM_NORETURN_PRE static void
96 1.1 christos usage(int status) ISC_PLATFORM_NORETURN_POST;
97 1.1 christos
98 1.1 christos static void
99 1.1 christos usage(int status) {
100 1.1 christos fprintf(stderr, "\
101 1.1 christos Usage: %s [-b address] [-c config] [-s server] [-p port]\n\
102 1.1 christos [-k key-file ] [-y key] [-r] [-V] [-4 | -6] command\n\
103 1.1 christos \n\
104 1.1 christos command is one of the following:\n\
105 1.1 christos \n\
106 1.1 christos addzone zone [class [view]] { zone-options }\n\
107 1.1 christos Add zone to given view. Requires allow-new-zones option.\n\
108 1.1 christos delzone [-clean] zone [class [view]]\n\
109 1.1 christos Removes zone from given view.\n\
110 1.6 christos dnssec -checkds [-key id [-alg algorithm]] [-when time] (published|withdrawn) zone [class [view]]\n\
111 1.6 christos Mark the DS record for the KSK of the given zone as seen\n\
112 1.6 christos in the parent. If the zone has multiple KSKs, select a\n\
113 1.6 christos specific key by providing the keytag with -key id and\n\
114 1.6 christos optionally the key's algorithm with -alg algorithm.\n\
115 1.6 christos Requires the zone to have a dnssec-policy.\n\
116 1.6 christos dnssec -rollover -key id [-alg algorithm] [-when time] zone [class [view]]\n\
117 1.6 christos Rollover key with id of the given zone. Requires the zone\n\
118 1.6 christos to have a dnssec-policy.\n\
119 1.5 christos dnssec -status zone [class [view]]\n\
120 1.5 christos Show the DNSSEC signing state for the specified zone.\n\
121 1.5 christos Requires the zone to have a dnssec-policy.\n\
122 1.1 christos dnstap -reopen\n\
123 1.1 christos Close, truncate and re-open the DNSTAP output file.\n\
124 1.1 christos dnstap -roll count\n\
125 1.1 christos Close, rename and re-open the DNSTAP output file(s).\n\
126 1.6 christos dumpdb [-all|-cache|-zones|-adb|-bad|-expired|-fail] [view ...]\n\
127 1.1 christos Dump cache(s) to the dump file (named_dump.db).\n\
128 1.1 christos flush Flushes all of the server's caches.\n\
129 1.1 christos flush [view] Flushes the server's cache for a view.\n\
130 1.1 christos flushname name [view]\n\
131 1.1 christos Flush the given name from the server's cache(s)\n\
132 1.1 christos flushtree name [view]\n\
133 1.1 christos Flush all names under the given name from the server's cache(s)\n\
134 1.1 christos freeze Suspend updates to all dynamic zones.\n\
135 1.1 christos freeze zone [class [view]]\n\
136 1.1 christos Suspend updates to a dynamic zone.\n\
137 1.1 christos halt Stop the server without saving pending updates.\n\
138 1.1 christos halt -p Stop the server without saving pending updates reporting\n\
139 1.1 christos process id.\n\
140 1.1 christos loadkeys zone [class [view]]\n\
141 1.1 christos Update keys without signing immediately.\n\
142 1.1 christos managed-keys refresh [class [view]]\n\
143 1.1 christos Check trust anchor for RFC 5011 key changes\n\
144 1.1 christos managed-keys status [class [view]]\n\
145 1.1 christos Display RFC 5011 managed keys information\n\
146 1.1 christos managed-keys sync [class [view]]\n\
147 1.1 christos Write RFC 5011 managed keys to disk\n\
148 1.1 christos modzone zone [class [view]] { zone-options }\n\
149 1.1 christos Modify a zone's configuration.\n\
150 1.1 christos Requires allow-new-zones option.\n\
151 1.1 christos notify zone [class [view]]\n\
152 1.1 christos Resend NOTIFY messages for the zone.\n\
153 1.1 christos notrace Set debugging level to 0.\n\
154 1.1 christos nta -dump\n\
155 1.1 christos List all negative trust anchors.\n\
156 1.1 christos nta [-lifetime duration] [-force] domain [view]\n\
157 1.1 christos Set a negative trust anchor, disabling DNSSEC validation\n\
158 1.1 christos for the given domain.\n\
159 1.1 christos Using -lifetime specifies the duration of the NTA, up\n\
160 1.1 christos to one week.\n\
161 1.1 christos Using -force prevents the NTA from expiring before its\n\
162 1.1 christos full lifetime, even if the domain can validate sooner.\n\
163 1.1 christos nta -remove domain [view]\n\
164 1.1 christos Remove a negative trust anchor, re-enabling validation\n\
165 1.1 christos for the given domain.\n\
166 1.1 christos querylog [ on | off ]\n\
167 1.1 christos Enable / disable query logging.\n\
168 1.1 christos reconfig Reload configuration file and new zones only.\n\
169 1.1 christos recursing Dump the queries that are currently recursing (named.recursing)\n\
170 1.1 christos refresh zone [class [view]]\n\
171 1.1 christos Schedule immediate maintenance for a zone.\n\
172 1.1 christos reload Reload configuration file and zones.\n\
173 1.1 christos reload zone [class [view]]\n\
174 1.1 christos Reload a single zone.\n\
175 1.1 christos retransfer zone [class [view]]\n\
176 1.1 christos Retransfer a single zone without checking serial number.\n\
177 1.1 christos scan Scan available network interfaces for changes.\n\
178 1.1 christos secroots [view ...]\n\
179 1.1 christos Write security roots to the secroots file.\n\
180 1.4 christos serve-stale [ on | off | reset | status ] [class [view]]\n\
181 1.1 christos Control whether stale answers are returned\n\
182 1.1 christos showzone zone [class [view]]\n\
183 1.1 christos Print a zone's configuration.\n\
184 1.1 christos sign zone [class [view]]\n\
185 1.1 christos Update zone keys, and sign as needed.\n\
186 1.1 christos signing -clear all zone [class [view]]\n\
187 1.1 christos Remove the private records for all keys that have\n\
188 1.1 christos finished signing the given zone.\n\
189 1.1 christos signing -clear <keyid>/<algorithm> zone [class [view]]\n\
190 1.1 christos Remove the private record that indicating the given key\n\
191 1.1 christos has finished signing the given zone.\n\
192 1.1 christos signing -list zone [class [view]]\n\
193 1.1 christos List the private records showing the state of DNSSEC\n\
194 1.1 christos signing in the given zone.\n\
195 1.1 christos signing -nsec3param hash flags iterations salt zone [class [view]]\n\
196 1.1 christos Add NSEC3 chain to zone if already signed.\n\
197 1.1 christos Prime zone with NSEC3 chain if not yet signed.\n\
198 1.1 christos signing -nsec3param none zone [class [view]]\n\
199 1.1 christos Remove NSEC3 chains from zone.\n\
200 1.1 christos signing -serial <value> zone [class [view]]\n\
201 1.1 christos Set the zones's serial to <value>.\n\
202 1.1 christos stats Write server statistics to the statistics file.\n\
203 1.1 christos status Display status of the server.\n\
204 1.1 christos stop Save pending updates to master files and stop the server.\n\
205 1.1 christos stop -p Save pending updates to master files and stop the server\n\
206 1.1 christos reporting process id.\n\
207 1.1 christos sync [-clean] Dump changes to all dynamic zones to disk, and optionally\n\
208 1.1 christos remove their journal files.\n\
209 1.1 christos sync [-clean] zone [class [view]]\n\
210 1.1 christos Dump a single zone's changes to disk, and optionally\n\
211 1.1 christos remove its journal file.\n\
212 1.1 christos tcp-timeouts Display the tcp-*-timeout option values\n\
213 1.1 christos tcp-timeouts initial idle keepalive advertised\n\
214 1.1 christos Update the tcp-*-timeout option values\n\
215 1.1 christos thaw Enable updates to all dynamic zones and reload them.\n\
216 1.1 christos thaw zone [class [view]]\n\
217 1.1 christos Enable updates to a frozen dynamic zone and reload it.\n\
218 1.1 christos trace Increment debugging level by one.\n\
219 1.1 christos trace level Change the debugging level.\n\
220 1.1 christos tsig-delete keyname [view]\n\
221 1.1 christos Delete a TKEY-negotiated TSIG key.\n\
222 1.1 christos tsig-list List all currently active TSIG keys, including both statically\n\
223 1.1 christos configured and TKEY-negotiated keys.\n\
224 1.4 christos validation [ on | off | status ] [view]\n\
225 1.1 christos Enable / disable DNSSEC validation.\n\
226 1.1 christos zonestatus zone [class [view]]\n\
227 1.1 christos Display the current status of a zone.\n\
228 1.1 christos \n\
229 1.1 christos Version: %s\n",
230 1.1 christos progname, version);
231 1.1 christos
232 1.1 christos exit(status);
233 1.1 christos }
234 1.1 christos
235 1.1 christos #define CMDLINE_FLAGS "46b:c:hk:Mmp:qrs:Vy:"
236 1.1 christos
237 1.1 christos static void
238 1.1 christos preparse_args(int argc, char **argv) {
239 1.3 christos bool ipv4only = false, ipv6only = false;
240 1.1 christos int ch;
241 1.1 christos
242 1.1 christos while ((ch = isc_commandline_parse(argc, argv, CMDLINE_FLAGS)) != -1) {
243 1.1 christos switch (ch) {
244 1.1 christos case '4':
245 1.1 christos if (ipv6only) {
246 1.1 christos fatal("only one of -4 and -6 allowed");
247 1.1 christos }
248 1.3 christos ipv4only = true;
249 1.1 christos break;
250 1.1 christos case '6':
251 1.1 christos if (ipv4only) {
252 1.1 christos fatal("only one of -4 and -6 allowed");
253 1.1 christos }
254 1.3 christos ipv6only = true;
255 1.1 christos break;
256 1.1 christos default:
257 1.1 christos break;
258 1.1 christos }
259 1.1 christos }
260 1.1 christos
261 1.3 christos isc_commandline_reset = true;
262 1.1 christos isc_commandline_index = 1;
263 1.1 christos }
264 1.1 christos
265 1.1 christos static void
266 1.1 christos get_addresses(const char *host, in_port_t port) {
267 1.1 christos isc_result_t result;
268 1.1 christos int found = 0, count;
269 1.1 christos
270 1.4 christos REQUIRE(host != NULL);
271 1.4 christos
272 1.1 christos if (*host == '/') {
273 1.1 christos result = isc_sockaddr_frompath(&serveraddrs[nserveraddrs],
274 1.1 christos host);
275 1.4 christos if (result == ISC_R_SUCCESS) {
276 1.1 christos nserveraddrs++;
277 1.4 christos }
278 1.1 christos } else {
279 1.1 christos count = SERVERADDRS - nserveraddrs;
280 1.4 christos result = bind9_getaddresses(
281 1.4 christos host, port, &serveraddrs[nserveraddrs], count, &found);
282 1.1 christos nserveraddrs += found;
283 1.1 christos }
284 1.4 christos if (result != ISC_R_SUCCESS) {
285 1.4 christos fatal("couldn't get address for '%s': %s", host,
286 1.4 christos isc_result_totext(result));
287 1.4 christos }
288 1.1 christos INSIST(nserveraddrs > 0);
289 1.1 christos }
290 1.1 christos
291 1.1 christos static void
292 1.1 christos rndc_senddone(isc_task_t *task, isc_event_t *event) {
293 1.1 christos isc_socketevent_t *sevent = (isc_socketevent_t *)event;
294 1.1 christos
295 1.1 christos UNUSED(task);
296 1.1 christos
297 1.4 christos if (sevent->result != ISC_R_SUCCESS) {
298 1.1 christos fatal("send failed: %s", isc_result_totext(sevent->result));
299 1.4 christos }
300 1.1 christos isc_event_free(&event);
301 1.4 christos if (atomic_fetch_sub_release(&sends, 1) == 1 &&
302 1.4 christos atomic_load_acquire(&recvs) == 0)
303 1.4 christos {
304 1.1 christos isc_socket_detach(&sock);
305 1.1 christos isc_task_shutdown(task);
306 1.4 christos isc_app_shutdown();
307 1.1 christos }
308 1.1 christos }
309 1.1 christos
310 1.1 christos static void
311 1.1 christos rndc_recvdone(isc_task_t *task, isc_event_t *event) {
312 1.1 christos isccc_sexpr_t *response = NULL;
313 1.1 christos isccc_sexpr_t *data;
314 1.1 christos isccc_region_t source;
315 1.1 christos char *errormsg = NULL;
316 1.1 christos char *textmsg = NULL;
317 1.1 christos isc_result_t result;
318 1.1 christos
319 1.4 christos atomic_fetch_sub_release(&recvs, 1);
320 1.1 christos
321 1.4 christos if (ccmsg.result == ISC_R_EOF) {
322 1.1 christos fatal("connection to remote host closed\n"
323 1.1 christos "This may indicate that\n"
324 1.1 christos "* the remote server is using an older version of"
325 1.1 christos " the command protocol,\n"
326 1.1 christos "* this host is not authorized to connect,\n"
327 1.1 christos "* the clocks are not synchronized, or\n"
328 1.1 christos "* the key is invalid.");
329 1.4 christos }
330 1.1 christos
331 1.4 christos if (ccmsg.result != ISC_R_SUCCESS) {
332 1.1 christos fatal("recv failed: %s", isc_result_totext(ccmsg.result));
333 1.4 christos }
334 1.1 christos
335 1.1 christos source.rstart = isc_buffer_base(&ccmsg.buffer);
336 1.1 christos source.rend = isc_buffer_used(&ccmsg.buffer);
337 1.1 christos
338 1.1 christos DO("parse message",
339 1.1 christos isccc_cc_fromwire(&source, &response, algorithm, &secret));
340 1.1 christos
341 1.1 christos data = isccc_alist_lookup(response, "_data");
342 1.4 christos if (!isccc_alist_alistp(data)) {
343 1.1 christos fatal("bad or missing data section in response");
344 1.4 christos }
345 1.1 christos result = isccc_cc_lookupstring(data, "err", &errormsg);
346 1.1 christos if (result == ISC_R_SUCCESS) {
347 1.3 christos failed = true;
348 1.4 christos fprintf(stderr, "%s: '%s' failed: %s\n", progname, command,
349 1.4 christos errormsg);
350 1.4 christos } else if (result != ISC_R_NOTFOUND) {
351 1.4 christos fprintf(stderr, "%s: parsing response failed: %s\n", progname,
352 1.4 christos isc_result_totext(result));
353 1.1 christos }
354 1.1 christos
355 1.1 christos result = isccc_cc_lookupstring(data, "text", &textmsg);
356 1.1 christos if (result == ISC_R_SUCCESS) {
357 1.4 christos if ((!quiet || failed) && strlen(textmsg) != 0U) {
358 1.1 christos fprintf(failed ? stderr : stdout, "%s\n", textmsg);
359 1.4 christos }
360 1.4 christos } else if (result != ISC_R_NOTFOUND) {
361 1.4 christos fprintf(stderr, "%s: parsing response failed: %s\n", progname,
362 1.4 christos isc_result_totext(result));
363 1.4 christos }
364 1.1 christos
365 1.1 christos if (showresult) {
366 1.1 christos isc_result_t eresult;
367 1.1 christos
368 1.1 christos result = isccc_cc_lookupuint32(data, "result", &eresult);
369 1.4 christos if (result == ISC_R_SUCCESS) {
370 1.1 christos printf("%s %u\n", isc_result_toid(eresult), eresult);
371 1.4 christos } else {
372 1.1 christos printf("NONE -1\n");
373 1.4 christos }
374 1.1 christos }
375 1.1 christos
376 1.1 christos isc_event_free(&event);
377 1.1 christos isccc_sexpr_free(&response);
378 1.4 christos if (atomic_load_acquire(&sends) == 0 &&
379 1.4 christos atomic_load_acquire(&recvs) == 0) {
380 1.1 christos isc_socket_detach(&sock);
381 1.1 christos isc_task_shutdown(task);
382 1.4 christos isc_app_shutdown();
383 1.1 christos }
384 1.1 christos }
385 1.1 christos
386 1.1 christos static void
387 1.1 christos rndc_recvnonce(isc_task_t *task, isc_event_t *event) {
388 1.1 christos isccc_sexpr_t *response = NULL;
389 1.1 christos isccc_sexpr_t *_ctrl;
390 1.1 christos isccc_region_t source;
391 1.1 christos isc_result_t result;
392 1.3 christos uint32_t nonce;
393 1.1 christos isccc_sexpr_t *request = NULL;
394 1.1 christos isccc_time_t now;
395 1.1 christos isc_region_t r;
396 1.1 christos isccc_sexpr_t *data;
397 1.1 christos isc_buffer_t b;
398 1.1 christos
399 1.4 christos atomic_fetch_sub_release(&recvs, 1);
400 1.1 christos
401 1.4 christos if (ccmsg.result == ISC_R_EOF) {
402 1.1 christos fatal("connection to remote host closed\n"
403 1.1 christos "This may indicate that\n"
404 1.1 christos "* the remote server is using an older version of"
405 1.1 christos " the command protocol,\n"
406 1.1 christos "* this host is not authorized to connect,\n"
407 1.1 christos "* the clocks are not synchronized,\n"
408 1.1 christos "* the key signing algorithm is incorrect, or\n"
409 1.1 christos "* the key is invalid.");
410 1.4 christos }
411 1.1 christos
412 1.4 christos if (ccmsg.result != ISC_R_SUCCESS) {
413 1.1 christos fatal("recv failed: %s", isc_result_totext(ccmsg.result));
414 1.4 christos }
415 1.1 christos
416 1.1 christos source.rstart = isc_buffer_base(&ccmsg.buffer);
417 1.1 christos source.rend = isc_buffer_used(&ccmsg.buffer);
418 1.1 christos
419 1.1 christos DO("parse message",
420 1.1 christos isccc_cc_fromwire(&source, &response, algorithm, &secret));
421 1.1 christos
422 1.1 christos _ctrl = isccc_alist_lookup(response, "_ctrl");
423 1.4 christos if (!isccc_alist_alistp(_ctrl)) {
424 1.1 christos fatal("bad or missing ctrl section in response");
425 1.4 christos }
426 1.1 christos nonce = 0;
427 1.4 christos if (isccc_cc_lookupuint32(_ctrl, "_nonce", &nonce) != ISC_R_SUCCESS) {
428 1.1 christos nonce = 0;
429 1.4 christos }
430 1.1 christos
431 1.1 christos isc_stdtime_get(&now);
432 1.1 christos
433 1.1 christos DO("create message", isccc_cc_createmessage(1, NULL, NULL, ++serial,
434 1.1 christos now, now + 60, &request));
435 1.1 christos data = isccc_alist_lookup(request, "_data");
436 1.4 christos if (data == NULL) {
437 1.1 christos fatal("_data section missing");
438 1.4 christos }
439 1.4 christos if (isccc_cc_definestring(data, "type", args) == NULL) {
440 1.1 christos fatal("out of memory");
441 1.4 christos }
442 1.1 christos if (nonce != 0) {
443 1.1 christos _ctrl = isccc_alist_lookup(request, "_ctrl");
444 1.4 christos if (_ctrl == NULL) {
445 1.1 christos fatal("_ctrl section missing");
446 1.4 christos }
447 1.4 christos if (isccc_cc_defineuint32(_ctrl, "_nonce", nonce) == NULL) {
448 1.1 christos fatal("out of memory");
449 1.4 christos }
450 1.1 christos }
451 1.1 christos
452 1.1 christos isc_buffer_clear(databuf);
453 1.1 christos /* Skip the length field (4 bytes) */
454 1.1 christos isc_buffer_add(databuf, 4);
455 1.1 christos
456 1.1 christos DO("render message",
457 1.1 christos isccc_cc_towire(request, &databuf, algorithm, &secret));
458 1.1 christos
459 1.1 christos isc_buffer_init(&b, databuf->base, 4);
460 1.1 christos isc_buffer_putuint32(&b, databuf->used - 4);
461 1.1 christos
462 1.1 christos r.base = databuf->base;
463 1.1 christos r.length = databuf->used;
464 1.1 christos
465 1.1 christos isccc_ccmsg_cancelread(&ccmsg);
466 1.4 christos DO("schedule recv",
467 1.4 christos isccc_ccmsg_readmessage(&ccmsg, task, rndc_recvdone, NULL));
468 1.4 christos atomic_fetch_add_relaxed(&recvs, 1);
469 1.4 christos DO("send message",
470 1.4 christos isc_socket_send(sock, &r, task, rndc_senddone, NULL));
471 1.4 christos atomic_fetch_add_relaxed(&sends, 1);
472 1.1 christos
473 1.1 christos isc_event_free(&event);
474 1.1 christos isccc_sexpr_free(&response);
475 1.1 christos isccc_sexpr_free(&request);
476 1.1 christos return;
477 1.1 christos }
478 1.1 christos
479 1.1 christos static void
480 1.1 christos rndc_connected(isc_task_t *task, isc_event_t *event) {
481 1.1 christos char socktext[ISC_SOCKADDR_FORMATSIZE];
482 1.1 christos isc_socketevent_t *sevent = (isc_socketevent_t *)event;
483 1.1 christos isccc_sexpr_t *request = NULL;
484 1.1 christos isccc_sexpr_t *data;
485 1.1 christos isccc_time_t now;
486 1.1 christos isc_region_t r;
487 1.1 christos isc_buffer_t b;
488 1.1 christos isc_result_t result;
489 1.1 christos
490 1.4 christos atomic_fetch_sub_release(&connects, 1);
491 1.1 christos
492 1.1 christos if (sevent->result != ISC_R_SUCCESS) {
493 1.1 christos isc_sockaddr_format(&serveraddrs[currentaddr], socktext,
494 1.1 christos sizeof(socktext));
495 1.1 christos if (sevent->result != ISC_R_CANCELED &&
496 1.4 christos ++currentaddr < nserveraddrs) {
497 1.1 christos notify("connection failed: %s: %s", socktext,
498 1.1 christos isc_result_totext(sevent->result));
499 1.1 christos isc_socket_detach(&sock);
500 1.1 christos isc_event_free(&event);
501 1.1 christos rndc_startconnect(&serveraddrs[currentaddr], task);
502 1.1 christos return;
503 1.4 christos } else {
504 1.1 christos fatal("connect failed: %s: %s", socktext,
505 1.1 christos isc_result_totext(sevent->result));
506 1.4 christos }
507 1.1 christos }
508 1.1 christos
509 1.1 christos isc_stdtime_get(&now);
510 1.1 christos DO("create message", isccc_cc_createmessage(1, NULL, NULL, ++serial,
511 1.1 christos now, now + 60, &request));
512 1.1 christos data = isccc_alist_lookup(request, "_data");
513 1.4 christos if (data == NULL) {
514 1.1 christos fatal("_data section missing");
515 1.4 christos }
516 1.4 christos if (isccc_cc_definestring(data, "type", "null") == NULL) {
517 1.1 christos fatal("out of memory");
518 1.4 christos }
519 1.1 christos
520 1.1 christos isc_buffer_clear(databuf);
521 1.1 christos /* Skip the length field (4 bytes) */
522 1.1 christos isc_buffer_add(databuf, 4);
523 1.1 christos
524 1.1 christos DO("render message",
525 1.1 christos isccc_cc_towire(request, &databuf, algorithm, &secret));
526 1.1 christos
527 1.1 christos isc_buffer_init(&b, databuf->base, 4);
528 1.1 christos isc_buffer_putuint32(&b, databuf->used - 4);
529 1.1 christos
530 1.1 christos r.base = databuf->base;
531 1.1 christos r.length = databuf->used;
532 1.1 christos
533 1.1 christos isccc_ccmsg_init(rndc_mctx, sock, &ccmsg);
534 1.1 christos isccc_ccmsg_setmaxsize(&ccmsg, 1024 * 1024);
535 1.1 christos
536 1.4 christos DO("schedule recv",
537 1.4 christos isccc_ccmsg_readmessage(&ccmsg, task, rndc_recvnonce, NULL));
538 1.4 christos atomic_fetch_add_relaxed(&recvs, 1);
539 1.4 christos DO("send message",
540 1.4 christos isc_socket_send(sock, &r, task, rndc_senddone, NULL));
541 1.4 christos atomic_fetch_add_relaxed(&sends, 1);
542 1.1 christos isc_event_free(&event);
543 1.1 christos isccc_sexpr_free(&request);
544 1.1 christos }
545 1.1 christos
546 1.1 christos static void
547 1.1 christos rndc_startconnect(isc_sockaddr_t *addr, isc_task_t *task) {
548 1.1 christos isc_result_t result;
549 1.1 christos int pf;
550 1.1 christos isc_sockettype_t type;
551 1.1 christos
552 1.1 christos char socktext[ISC_SOCKADDR_FORMATSIZE];
553 1.1 christos
554 1.1 christos isc_sockaddr_format(addr, socktext, sizeof(socktext));
555 1.1 christos
556 1.1 christos notify("using server %s (%s)", servername, socktext);
557 1.1 christos
558 1.1 christos pf = isc_sockaddr_pf(addr);
559 1.4 christos if (pf == AF_INET || pf == AF_INET6) {
560 1.1 christos type = isc_sockettype_tcp;
561 1.4 christos } else {
562 1.1 christos type = isc_sockettype_unix;
563 1.4 christos }
564 1.1 christos DO("create socket", isc_socket_create(socketmgr, pf, type, &sock));
565 1.1 christos switch (isc_sockaddr_pf(addr)) {
566 1.1 christos case AF_INET:
567 1.1 christos DO("bind socket", isc_socket_bind(sock, &local4, 0));
568 1.1 christos break;
569 1.1 christos case AF_INET6:
570 1.1 christos DO("bind socket", isc_socket_bind(sock, &local6, 0));
571 1.1 christos break;
572 1.1 christos default:
573 1.1 christos break;
574 1.1 christos }
575 1.4 christos DO("connect",
576 1.4 christos isc_socket_connect(sock, addr, task, rndc_connected, NULL));
577 1.4 christos atomic_fetch_add_relaxed(&connects, 1);
578 1.1 christos }
579 1.1 christos
580 1.1 christos static void
581 1.1 christos rndc_start(isc_task_t *task, isc_event_t *event) {
582 1.1 christos isc_event_free(&event);
583 1.1 christos
584 1.1 christos currentaddr = 0;
585 1.1 christos rndc_startconnect(&serveraddrs[currentaddr], task);
586 1.1 christos }
587 1.1 christos
588 1.1 christos static void
589 1.1 christos parse_config(isc_mem_t *mctx, isc_log_t *log, const char *keyname,
590 1.4 christos cfg_parser_t **pctxp, cfg_obj_t **configp) {
591 1.1 christos isc_result_t result;
592 1.1 christos const char *conffile = admin_conffile;
593 1.1 christos const cfg_obj_t *addresses = NULL;
594 1.1 christos const cfg_obj_t *defkey = NULL;
595 1.1 christos const cfg_obj_t *options = NULL;
596 1.1 christos const cfg_obj_t *servers = NULL;
597 1.1 christos const cfg_obj_t *server = NULL;
598 1.1 christos const cfg_obj_t *keys = NULL;
599 1.1 christos const cfg_obj_t *key = NULL;
600 1.1 christos const cfg_obj_t *defport = NULL;
601 1.1 christos const cfg_obj_t *secretobj = NULL;
602 1.1 christos const cfg_obj_t *algorithmobj = NULL;
603 1.1 christos cfg_obj_t *config = NULL;
604 1.1 christos const cfg_obj_t *address = NULL;
605 1.1 christos const cfg_listelt_t *elt;
606 1.1 christos const char *secretstr;
607 1.1 christos const char *algorithmstr;
608 1.1 christos static char secretarray[1024];
609 1.1 christos const cfg_type_t *conftype = &cfg_type_rndcconf;
610 1.3 christos bool key_only = false;
611 1.1 christos const cfg_listelt_t *element;
612 1.1 christos
613 1.4 christos if (!isc_file_exists(conffile)) {
614 1.1 christos conffile = admin_keyfile;
615 1.1 christos conftype = &cfg_type_rndckey;
616 1.1 christos
617 1.4 christos if (c_flag) {
618 1.1 christos fatal("%s does not exist", admin_conffile);
619 1.4 christos }
620 1.1 christos
621 1.4 christos if (!isc_file_exists(conffile)) {
622 1.4 christos fatal("neither %s nor %s was found", admin_conffile,
623 1.4 christos admin_keyfile);
624 1.4 christos }
625 1.3 christos key_only = true;
626 1.4 christos } else if (!c_flag && isc_file_exists(admin_keyfile)) {
627 1.4 christos fprintf(stderr,
628 1.4 christos "WARNING: key file (%s) exists, but using "
629 1.1 christos "default configuration file (%s)\n",
630 1.1 christos admin_keyfile, admin_conffile);
631 1.1 christos }
632 1.1 christos
633 1.1 christos DO("create parser", cfg_parser_create(mctx, log, pctxp));
634 1.1 christos
635 1.1 christos /*
636 1.1 christos * The parser will output its own errors, so DO() is not used.
637 1.1 christos */
638 1.1 christos result = cfg_parse_file(*pctxp, conffile, conftype, &config);
639 1.4 christos if (result != ISC_R_SUCCESS) {
640 1.1 christos fatal("could not load rndc configuration");
641 1.4 christos }
642 1.1 christos
643 1.4 christos if (!key_only) {
644 1.1 christos (void)cfg_map_get(config, "options", &options);
645 1.4 christos }
646 1.1 christos
647 1.4 christos if (key_only && servername == NULL) {
648 1.1 christos servername = "127.0.0.1";
649 1.4 christos } else if (servername == NULL && options != NULL) {
650 1.1 christos const cfg_obj_t *defserverobj = NULL;
651 1.1 christos (void)cfg_map_get(options, "default-server", &defserverobj);
652 1.4 christos if (defserverobj != NULL) {
653 1.1 christos servername = cfg_obj_asstring(defserverobj);
654 1.4 christos }
655 1.1 christos }
656 1.1 christos
657 1.4 christos if (servername == NULL) {
658 1.1 christos fatal("no server specified and no default");
659 1.4 christos }
660 1.1 christos
661 1.1 christos if (!key_only) {
662 1.1 christos (void)cfg_map_get(config, "server", &servers);
663 1.1 christos if (servers != NULL) {
664 1.4 christos for (elt = cfg_list_first(servers); elt != NULL;
665 1.4 christos elt = cfg_list_next(elt)) {
666 1.1 christos const char *name;
667 1.1 christos server = cfg_listelt_value(elt);
668 1.4 christos name = cfg_obj_asstring(
669 1.4 christos cfg_map_getname(server));
670 1.4 christos if (strcasecmp(name, servername) == 0) {
671 1.1 christos break;
672 1.4 christos }
673 1.1 christos server = NULL;
674 1.1 christos }
675 1.1 christos }
676 1.1 christos }
677 1.1 christos
678 1.1 christos /*
679 1.1 christos * Look for the name of the key to use.
680 1.1 christos */
681 1.4 christos if (keyname != NULL) {
682 1.4 christos /* Was set on command line, do nothing. */
683 1.4 christos } else if (server != NULL) {
684 1.1 christos DO("get key for server", cfg_map_get(server, "key", &defkey));
685 1.1 christos keyname = cfg_obj_asstring(defkey);
686 1.1 christos } else if (options != NULL) {
687 1.4 christos DO("get default key",
688 1.4 christos cfg_map_get(options, "default-key", &defkey));
689 1.1 christos keyname = cfg_obj_asstring(defkey);
690 1.4 christos } else if (!key_only) {
691 1.1 christos fatal("no key for server and no default");
692 1.4 christos }
693 1.1 christos
694 1.1 christos /*
695 1.1 christos * Get the key's definition.
696 1.1 christos */
697 1.4 christos if (key_only) {
698 1.1 christos DO("get key", cfg_map_get(config, "key", &key));
699 1.4 christos } else {
700 1.1 christos DO("get config key list", cfg_map_get(config, "key", &keys));
701 1.4 christos for (elt = cfg_list_first(keys); elt != NULL;
702 1.4 christos elt = cfg_list_next(elt)) {
703 1.1 christos key = cfg_listelt_value(elt);
704 1.1 christos if (strcasecmp(cfg_obj_asstring(cfg_map_getname(key)),
705 1.4 christos keyname) == 0) {
706 1.1 christos break;
707 1.4 christos }
708 1.1 christos }
709 1.4 christos if (elt == NULL) {
710 1.1 christos fatal("no key definition for name %s", keyname);
711 1.4 christos }
712 1.1 christos }
713 1.1 christos (void)cfg_map_get(key, "secret", &secretobj);
714 1.1 christos (void)cfg_map_get(key, "algorithm", &algorithmobj);
715 1.4 christos if (secretobj == NULL || algorithmobj == NULL) {
716 1.1 christos fatal("key must have algorithm and secret");
717 1.4 christos }
718 1.1 christos
719 1.1 christos secretstr = cfg_obj_asstring(secretobj);
720 1.1 christos algorithmstr = cfg_obj_asstring(algorithmobj);
721 1.1 christos
722 1.3 christos if (strcasecmp(algorithmstr, "hmac-md5") == 0) {
723 1.1 christos algorithm = ISCCC_ALG_HMACMD5;
724 1.3 christos } else if (strcasecmp(algorithmstr, "hmac-sha1") == 0) {
725 1.1 christos algorithm = ISCCC_ALG_HMACSHA1;
726 1.3 christos } else if (strcasecmp(algorithmstr, "hmac-sha224") == 0) {
727 1.1 christos algorithm = ISCCC_ALG_HMACSHA224;
728 1.3 christos } else if (strcasecmp(algorithmstr, "hmac-sha256") == 0) {
729 1.1 christos algorithm = ISCCC_ALG_HMACSHA256;
730 1.3 christos } else if (strcasecmp(algorithmstr, "hmac-sha384") == 0) {
731 1.1 christos algorithm = ISCCC_ALG_HMACSHA384;
732 1.3 christos } else if (strcasecmp(algorithmstr, "hmac-sha512") == 0) {
733 1.1 christos algorithm = ISCCC_ALG_HMACSHA512;
734 1.3 christos } else {
735 1.1 christos fatal("unsupported algorithm: %s", algorithmstr);
736 1.3 christos }
737 1.1 christos
738 1.1 christos secret.rstart = (unsigned char *)secretarray;
739 1.1 christos secret.rend = (unsigned char *)secretarray + sizeof(secretarray);
740 1.1 christos DO("decode base64 secret", isccc_base64_decode(secretstr, &secret));
741 1.1 christos secret.rend = secret.rstart;
742 1.1 christos secret.rstart = (unsigned char *)secretarray;
743 1.1 christos
744 1.1 christos /*
745 1.1 christos * Find the port to connect to.
746 1.1 christos */
747 1.4 christos if (remoteport != 0) {
748 1.4 christos /* Was set on command line, do nothing. */
749 1.4 christos } else {
750 1.4 christos if (server != NULL) {
751 1.1 christos (void)cfg_map_get(server, "port", &defport);
752 1.4 christos }
753 1.4 christos if (defport == NULL && options != NULL) {
754 1.1 christos (void)cfg_map_get(options, "default-port", &defport);
755 1.4 christos }
756 1.1 christos }
757 1.1 christos if (defport != NULL) {
758 1.1 christos remoteport = cfg_obj_asuint32(defport);
759 1.4 christos if (remoteport > 65535 || remoteport == 0) {
760 1.1 christos fatal("port %u out of range", remoteport);
761 1.4 christos }
762 1.4 christos } else if (remoteport == 0) {
763 1.1 christos remoteport = NS_CONTROL_PORT;
764 1.4 christos }
765 1.1 christos
766 1.4 christos if (server != NULL) {
767 1.1 christos result = cfg_map_get(server, "addresses", &addresses);
768 1.4 christos } else {
769 1.1 christos result = ISC_R_NOTFOUND;
770 1.4 christos }
771 1.1 christos if (result == ISC_R_SUCCESS) {
772 1.4 christos for (element = cfg_list_first(addresses); element != NULL;
773 1.1 christos element = cfg_list_next(element))
774 1.1 christos {
775 1.1 christos isc_sockaddr_t sa;
776 1.1 christos
777 1.1 christos address = cfg_listelt_value(element);
778 1.1 christos if (!cfg_obj_issockaddr(address)) {
779 1.1 christos unsigned int myport;
780 1.1 christos const char *name;
781 1.1 christos const cfg_obj_t *obj;
782 1.1 christos
783 1.1 christos obj = cfg_tuple_get(address, "name");
784 1.1 christos name = cfg_obj_asstring(obj);
785 1.1 christos obj = cfg_tuple_get(address, "port");
786 1.1 christos if (cfg_obj_isuint32(obj)) {
787 1.1 christos myport = cfg_obj_asuint32(obj);
788 1.4 christos if (myport > UINT16_MAX || myport == 0)
789 1.4 christos {
790 1.1 christos fatal("port %u out of range",
791 1.1 christos myport);
792 1.4 christos }
793 1.4 christos } else {
794 1.1 christos myport = remoteport;
795 1.4 christos }
796 1.4 christos if (nserveraddrs < SERVERADDRS) {
797 1.4 christos get_addresses(name, (in_port_t)myport);
798 1.4 christos } else {
799 1.4 christos fprintf(stderr,
800 1.4 christos "too many address: "
801 1.4 christos "%s: dropped\n",
802 1.4 christos name);
803 1.4 christos }
804 1.1 christos continue;
805 1.1 christos }
806 1.1 christos sa = *cfg_obj_assockaddr(address);
807 1.4 christos if (isc_sockaddr_getport(&sa) == 0) {
808 1.1 christos isc_sockaddr_setport(&sa, remoteport);
809 1.4 christos }
810 1.4 christos if (nserveraddrs < SERVERADDRS) {
811 1.1 christos serveraddrs[nserveraddrs++] = sa;
812 1.4 christos } else {
813 1.1 christos char socktext[ISC_SOCKADDR_FORMATSIZE];
814 1.1 christos
815 1.1 christos isc_sockaddr_format(&sa, socktext,
816 1.1 christos sizeof(socktext));
817 1.1 christos fprintf(stderr,
818 1.1 christos "too many address: %s: dropped\n",
819 1.1 christos socktext);
820 1.1 christos }
821 1.1 christos }
822 1.1 christos }
823 1.1 christos
824 1.1 christos if (!local4set && server != NULL) {
825 1.1 christos address = NULL;
826 1.1 christos cfg_map_get(server, "source-address", &address);
827 1.1 christos if (address != NULL) {
828 1.1 christos local4 = *cfg_obj_assockaddr(address);
829 1.3 christos local4set = true;
830 1.1 christos }
831 1.1 christos }
832 1.1 christos if (!local4set && options != NULL) {
833 1.1 christos address = NULL;
834 1.1 christos cfg_map_get(options, "default-source-address", &address);
835 1.1 christos if (address != NULL) {
836 1.1 christos local4 = *cfg_obj_assockaddr(address);
837 1.3 christos local4set = true;
838 1.1 christos }
839 1.1 christos }
840 1.1 christos
841 1.1 christos if (!local6set && server != NULL) {
842 1.1 christos address = NULL;
843 1.1 christos cfg_map_get(server, "source-address-v6", &address);
844 1.1 christos if (address != NULL) {
845 1.1 christos local6 = *cfg_obj_assockaddr(address);
846 1.3 christos local6set = true;
847 1.1 christos }
848 1.1 christos }
849 1.1 christos if (!local6set && options != NULL) {
850 1.1 christos address = NULL;
851 1.1 christos cfg_map_get(options, "default-source-address-v6", &address);
852 1.1 christos if (address != NULL) {
853 1.1 christos local6 = *cfg_obj_assockaddr(address);
854 1.3 christos local6set = true;
855 1.1 christos }
856 1.1 christos }
857 1.1 christos
858 1.1 christos *configp = config;
859 1.1 christos }
860 1.1 christos
861 1.1 christos int
862 1.1 christos main(int argc, char **argv) {
863 1.1 christos isc_result_t result = ISC_R_SUCCESS;
864 1.3 christos bool show_final_mem = false;
865 1.7 christos isc_nm_t *netmgr = NULL;
866 1.1 christos isc_taskmgr_t *taskmgr = NULL;
867 1.1 christos isc_task_t *task = NULL;
868 1.1 christos isc_log_t *log = NULL;
869 1.1 christos isc_logconfig_t *logconfig = NULL;
870 1.1 christos isc_logdestination_t logdest;
871 1.1 christos cfg_parser_t *pctx = NULL;
872 1.1 christos cfg_obj_t *config = NULL;
873 1.1 christos const char *keyname = NULL;
874 1.1 christos struct in_addr in;
875 1.1 christos struct in6_addr in6;
876 1.1 christos char *p;
877 1.1 christos size_t argslen;
878 1.1 christos int ch;
879 1.1 christos int i;
880 1.1 christos
881 1.1 christos result = isc_file_progname(*argv, program, sizeof(program));
882 1.4 christos if (result != ISC_R_SUCCESS) {
883 1.1 christos memmove(program, "rndc", 5);
884 1.4 christos }
885 1.1 christos progname = program;
886 1.1 christos
887 1.1 christos admin_conffile = RNDC_CONFFILE;
888 1.1 christos admin_keyfile = RNDC_KEYFILE;
889 1.1 christos
890 1.1 christos isc_sockaddr_any(&local4);
891 1.1 christos isc_sockaddr_any6(&local6);
892 1.1 christos
893 1.1 christos result = isc_app_start();
894 1.4 christos if (result != ISC_R_SUCCESS) {
895 1.1 christos fatal("isc_app_start() failed: %s", isc_result_totext(result));
896 1.4 christos }
897 1.1 christos
898 1.3 christos isc_commandline_errprint = false;
899 1.1 christos
900 1.1 christos preparse_args(argc, argv);
901 1.1 christos
902 1.1 christos while ((ch = isc_commandline_parse(argc, argv, CMDLINE_FLAGS)) != -1) {
903 1.1 christos switch (ch) {
904 1.1 christos case '4':
905 1.1 christos if (isc_net_probeipv4() != ISC_R_SUCCESS) {
906 1.1 christos fatal("can't find IPv4 networking");
907 1.1 christos }
908 1.1 christos isc_net_disableipv6();
909 1.1 christos break;
910 1.1 christos case '6':
911 1.1 christos if (isc_net_probeipv6() != ISC_R_SUCCESS) {
912 1.1 christos fatal("can't find IPv6 networking");
913 1.1 christos }
914 1.1 christos isc_net_disableipv4();
915 1.1 christos break;
916 1.1 christos case 'b':
917 1.4 christos if (inet_pton(AF_INET, isc_commandline_argument, &in) ==
918 1.4 christos 1) {
919 1.1 christos isc_sockaddr_fromin(&local4, &in, 0);
920 1.3 christos local4set = true;
921 1.1 christos } else if (inet_pton(AF_INET6, isc_commandline_argument,
922 1.1 christos &in6) == 1) {
923 1.1 christos isc_sockaddr_fromin6(&local6, &in6, 0);
924 1.3 christos local6set = true;
925 1.1 christos }
926 1.1 christos break;
927 1.1 christos
928 1.1 christos case 'c':
929 1.1 christos admin_conffile = isc_commandline_argument;
930 1.3 christos c_flag = true;
931 1.1 christos break;
932 1.1 christos
933 1.1 christos case 'k':
934 1.1 christos admin_keyfile = isc_commandline_argument;
935 1.1 christos break;
936 1.1 christos
937 1.1 christos case 'M':
938 1.1 christos isc_mem_debugging = ISC_MEM_DEBUGTRACE;
939 1.1 christos break;
940 1.1 christos
941 1.1 christos case 'm':
942 1.3 christos show_final_mem = true;
943 1.1 christos break;
944 1.1 christos
945 1.1 christos case 'p':
946 1.1 christos remoteport = atoi(isc_commandline_argument);
947 1.4 christos if (remoteport > 65535 || remoteport == 0) {
948 1.1 christos fatal("port '%s' out of range",
949 1.1 christos isc_commandline_argument);
950 1.4 christos }
951 1.1 christos break;
952 1.1 christos
953 1.1 christos case 'q':
954 1.3 christos quiet = true;
955 1.1 christos break;
956 1.1 christos
957 1.1 christos case 'r':
958 1.3 christos showresult = true;
959 1.1 christos break;
960 1.1 christos
961 1.1 christos case 's':
962 1.1 christos servername = isc_commandline_argument;
963 1.1 christos break;
964 1.1 christos
965 1.1 christos case 'V':
966 1.3 christos verbose = true;
967 1.1 christos break;
968 1.1 christos
969 1.1 christos case 'y':
970 1.1 christos keyname = isc_commandline_argument;
971 1.1 christos break;
972 1.1 christos
973 1.1 christos case '?':
974 1.1 christos if (isc_commandline_option != '?') {
975 1.1 christos fprintf(stderr, "%s: invalid argument -%c\n",
976 1.1 christos program, isc_commandline_option);
977 1.1 christos usage(1);
978 1.1 christos }
979 1.4 christos /* FALLTHROUGH */
980 1.1 christos case 'h':
981 1.1 christos usage(0);
982 1.1 christos break;
983 1.1 christos default:
984 1.4 christos fprintf(stderr, "%s: unhandled option -%c\n", program,
985 1.4 christos isc_commandline_option);
986 1.1 christos exit(1);
987 1.1 christos }
988 1.1 christos }
989 1.1 christos
990 1.1 christos argc -= isc_commandline_index;
991 1.1 christos argv += isc_commandline_index;
992 1.1 christos
993 1.4 christos if (argc < 1) {
994 1.1 christos usage(1);
995 1.4 christos }
996 1.1 christos
997 1.3 christos serial = isc_random32();
998 1.1 christos
999 1.4 christos isc_mem_create(&rndc_mctx);
1000 1.4 christos DO("create socket manager",
1001 1.4 christos isc_socketmgr_create(rndc_mctx, &socketmgr));
1002 1.4 christos DO("create task manager",
1003 1.7 christos isc_managers_create(rndc_mctx, 1, 0, &netmgr, &taskmgr));
1004 1.1 christos DO("create task", isc_task_create(taskmgr, 0, &task));
1005 1.1 christos
1006 1.4 christos isc_log_create(rndc_mctx, &log, &logconfig);
1007 1.1 christos isc_log_setcontext(log);
1008 1.4 christos isc_log_settag(logconfig, progname);
1009 1.1 christos logdest.file.stream = stderr;
1010 1.1 christos logdest.file.name = NULL;
1011 1.1 christos logdest.file.versions = ISC_LOG_ROLLNEVER;
1012 1.1 christos logdest.file.maximum_size = 0;
1013 1.4 christos isc_log_createchannel(logconfig, "stderr", ISC_LOG_TOFILEDESC,
1014 1.4 christos ISC_LOG_INFO, &logdest,
1015 1.4 christos ISC_LOG_PRINTTAG | ISC_LOG_PRINTLEVEL);
1016 1.4 christos DO("enabling log channel",
1017 1.4 christos isc_log_usechannel(logconfig, "stderr", NULL, NULL));
1018 1.1 christos
1019 1.1 christos parse_config(rndc_mctx, log, keyname, &pctx, &config);
1020 1.1 christos
1021 1.1 christos isccc_result_register();
1022 1.1 christos
1023 1.1 christos command = *argv;
1024 1.1 christos
1025 1.4 christos isc_buffer_allocate(rndc_mctx, &databuf, 2048);
1026 1.1 christos
1027 1.1 christos /*
1028 1.1 christos * Convert argc/argv into a space-delimited command string
1029 1.1 christos * similar to what the user might enter in interactive mode
1030 1.1 christos * (if that were implemented).
1031 1.1 christos */
1032 1.1 christos argslen = 0;
1033 1.4 christos for (i = 0; i < argc; i++) {
1034 1.1 christos argslen += strlen(argv[i]) + 1;
1035 1.4 christos }
1036 1.1 christos
1037 1.1 christos args = isc_mem_get(rndc_mctx, argslen);
1038 1.1 christos
1039 1.1 christos p = args;
1040 1.1 christos for (i = 0; i < argc; i++) {
1041 1.1 christos size_t len = strlen(argv[i]);
1042 1.1 christos memmove(p, argv[i], len);
1043 1.1 christos p += len;
1044 1.1 christos *p++ = ' ';
1045 1.1 christos }
1046 1.1 christos
1047 1.1 christos p--;
1048 1.1 christos *p++ = '\0';
1049 1.1 christos INSIST(p == args + argslen);
1050 1.1 christos
1051 1.1 christos notify("%s", command);
1052 1.1 christos
1053 1.4 christos if (strcmp(command, "restart") == 0) {
1054 1.1 christos fatal("'%s' is not implemented", command);
1055 1.4 christos }
1056 1.1 christos
1057 1.4 christos if (nserveraddrs == 0 && servername != NULL) {
1058 1.4 christos get_addresses(servername, (in_port_t)remoteport);
1059 1.4 christos }
1060 1.1 christos
1061 1.1 christos DO("post event", isc_app_onrun(rndc_mctx, task, rndc_start, NULL));
1062 1.1 christos
1063 1.1 christos result = isc_app_run();
1064 1.4 christos if (result != ISC_R_SUCCESS) {
1065 1.1 christos fatal("isc_app_run() failed: %s", isc_result_totext(result));
1066 1.4 christos }
1067 1.1 christos
1068 1.4 christos if (atomic_load_acquire(&connects) > 0 ||
1069 1.4 christos atomic_load_acquire(&sends) > 0 || atomic_load_acquire(&recvs) > 0)
1070 1.4 christos {
1071 1.1 christos isc_socket_cancel(sock, task, ISC_SOCKCANCEL_ALL);
1072 1.4 christos }
1073 1.1 christos
1074 1.1 christos isc_task_detach(&task);
1075 1.7 christos isc_managers_destroy(&netmgr, &taskmgr);
1076 1.1 christos isc_socketmgr_destroy(&socketmgr);
1077 1.1 christos isc_log_destroy(&log);
1078 1.1 christos isc_log_setcontext(NULL);
1079 1.1 christos
1080 1.1 christos cfg_obj_destroy(pctx, &config);
1081 1.1 christos cfg_parser_destroy(&pctx);
1082 1.1 christos
1083 1.1 christos isc_mem_put(rndc_mctx, args, argslen);
1084 1.1 christos isccc_ccmsg_invalidate(&ccmsg);
1085 1.1 christos
1086 1.1 christos isc_buffer_free(&databuf);
1087 1.1 christos
1088 1.4 christos if (show_final_mem) {
1089 1.1 christos isc_mem_stats(rndc_mctx, stderr);
1090 1.4 christos }
1091 1.1 christos
1092 1.1 christos isc_mem_destroy(&rndc_mctx);
1093 1.1 christos
1094 1.4 christos if (failed) {
1095 1.1 christos return (1);
1096 1.4 christos }
1097 1.1 christos
1098 1.1 christos return (0);
1099 1.1 christos }
1100