1/*
2 * Copyright (C) 2004, 2005, 2007, 2009-2011  Internet Systems Consortium, Inc. ("ISC")
3 * Copyright (C) 2001  Internet Software Consortium.
4 *
5 * Permission to use, copy, modify, and/or distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
10 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
11 * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
12 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
14 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
15 * PERFORMANCE OF THIS SOFTWARE.
16 */
17
18/* $Id: keydelete.c,v 1.18 2011/01/11 23:47:13 tbox Exp $ */
19
20#include <config.h>
21
22#include <stdlib.h>
23#include <string.h>
24
25#include <isc/app.h>
26#include <isc/base64.h>
27#include <isc/entropy.h>
28#include <isc/hash.h>
29#include <isc/log.h>
30#include <isc/mem.h>
31#include <isc/sockaddr.h>
32#include <isc/socket.h>
33#include <isc/task.h>
34#include <isc/timer.h>
35#include <isc/util.h>
36
37#include <dns/dispatch.h>
38#include <dns/fixedname.h>
39#include <dns/keyvalues.h>
40#include <dns/message.h>
41#include <dns/name.h>
42#include <dns/request.h>
43#include <dns/result.h>
44#include <dns/tkey.h>
45#include <dns/tsig.h>
46#include <dns/view.h>
47
48#include <dst/result.h>
49
50#define CHECK(str, x) { \
51	if ((x) != ISC_R_SUCCESS) { \
52		fprintf(stderr, "I:%s: %s\n", (str), isc_result_totext(x)); \
53		exit(-1); \
54	} \
55}
56
57#define RUNCHECK(x) RUNTIME_CHECK((x) == ISC_R_SUCCESS)
58
59#define PORT 5300
60#define TIMEOUT 30
61
62static isc_mem_t *mctx;
63static dns_tsigkey_t *tsigkey;
64static dns_tsig_keyring_t *ring;
65static dns_requestmgr_t *requestmgr;
66
67static void
68recvquery(isc_task_t *task, isc_event_t *event) {
69	dns_requestevent_t *reqev = (dns_requestevent_t *)event;
70	isc_result_t result;
71	dns_message_t *query, *response;
72
73	UNUSED(task);
74
75	REQUIRE(reqev != NULL);
76
77	if (reqev->result != ISC_R_SUCCESS) {
78		fprintf(stderr, "I:request event result: %s\n",
79			isc_result_totext(reqev->result));
80		exit(-1);
81	}
82
83	query = reqev->ev_arg;
84
85	response = NULL;
86	result = dns_message_create(mctx, DNS_MESSAGE_INTENTPARSE, &response);
87	CHECK("dns_message_create", result);
88
89	result = dns_request_getresponse(reqev->request, response,
90					 DNS_MESSAGEPARSE_PRESERVEORDER);
91	CHECK("dns_request_getresponse", result);
92
93	if (response->rcode != dns_rcode_noerror) {
94		result = ISC_RESULTCLASS_DNSRCODE + response->rcode;
95		fprintf(stderr, "I:response rcode: %s\n",
96			isc_result_totext(result));
97			exit(-1);
98	}
99
100	result = dns_tkey_processdeleteresponse(query, response, ring);
101	CHECK("dns_tkey_processdhresponse", result);
102
103	dns_message_destroy(&query);
104	dns_message_destroy(&response);
105	dns_request_destroy(&reqev->request);
106	isc_event_free(&event);
107	isc_app_shutdown();
108	return;
109}
110
111static void
112sendquery(isc_task_t *task, isc_event_t *event) {
113	struct in_addr inaddr;
114	isc_sockaddr_t address;
115	isc_result_t result;
116	dns_message_t *query;
117	dns_request_t *request;
118
119	isc_event_free(&event);
120
121	result = ISC_R_FAILURE;
122	if (inet_pton(AF_INET, "10.53.0.1", &inaddr) != 1)
123		CHECK("inet_pton", result);
124	isc_sockaddr_fromin(&address, &inaddr, PORT);
125
126	query = NULL;
127	result = dns_message_create(mctx, DNS_MESSAGE_INTENTRENDER, &query);
128	CHECK("dns_message_create", result);
129
130	result = dns_tkey_builddeletequery(query, tsigkey);
131	CHECK("dns_tkey_builddeletequery", result);
132
133	request = NULL;
134	result = dns_request_create(requestmgr, query, &address,
135				    0, tsigkey, TIMEOUT, task,
136				    recvquery, query, &request);
137	CHECK("dns_request_create", result);
138}
139
140int
141main(int argc, char **argv) {
142	char *keyname;
143	isc_taskmgr_t *taskmgr;
144	isc_timermgr_t *timermgr;
145	isc_socketmgr_t *socketmgr;
146	isc_socket_t *sock;
147	unsigned int attrs, attrmask;
148	isc_sockaddr_t bind_any;
149	dns_dispatchmgr_t *dispatchmgr;
150	dns_dispatch_t *dispatchv4;
151	dns_view_t *view;
152	isc_entropy_t *ectx;
153	dns_tkeyctx_t *tctx;
154	dst_key_t *dstkey;
155	isc_log_t *log;
156	isc_logconfig_t *logconfig;
157	isc_task_t *task;
158	isc_result_t result;
159	int type;
160
161	RUNCHECK(isc_app_start());
162
163	if (argc < 2) {
164		fprintf(stderr, "I:no key to delete\n");
165		exit(-1);
166	}
167	keyname = argv[1];
168
169	dns_result_register();
170
171	mctx = NULL;
172	RUNCHECK(isc_mem_create(0, 0, &mctx));
173
174	ectx = NULL;
175	RUNCHECK(isc_entropy_create(mctx, &ectx));
176	RUNCHECK(isc_entropy_createfilesource(ectx, "random.data"));
177	RUNCHECK(isc_hash_create(mctx, ectx, DNS_NAME_MAXWIRE));
178
179	log = NULL;
180	logconfig = NULL;
181	RUNCHECK(isc_log_create(mctx, &log, &logconfig));
182
183	RUNCHECK(dst_lib_init(mctx, ectx, ISC_ENTROPY_GOODONLY));
184
185	taskmgr = NULL;
186	RUNCHECK(isc_taskmgr_create(mctx, 1, 0, &taskmgr));
187	task = NULL;
188	RUNCHECK(isc_task_create(taskmgr, 0, &task));
189	timermgr = NULL;
190	RUNCHECK(isc_timermgr_create(mctx, &timermgr));
191	socketmgr = NULL;
192	RUNCHECK(isc_socketmgr_create(mctx, &socketmgr));
193	dispatchmgr = NULL;
194	RUNCHECK(dns_dispatchmgr_create(mctx, NULL, &dispatchmgr));
195	isc_sockaddr_any(&bind_any);
196	attrs = DNS_DISPATCHATTR_UDP |
197		DNS_DISPATCHATTR_MAKEQUERY |
198		DNS_DISPATCHATTR_IPV4;
199	attrmask = DNS_DISPATCHATTR_UDP |
200		   DNS_DISPATCHATTR_TCP |
201		   DNS_DISPATCHATTR_IPV4 |
202		   DNS_DISPATCHATTR_IPV6;
203	dispatchv4 = NULL;
204	RUNCHECK(dns_dispatch_getudp(dispatchmgr, socketmgr, taskmgr,
205					  &bind_any, 4096, 4, 2, 3, 5,
206					  attrs, attrmask, &dispatchv4));
207	requestmgr = NULL;
208	RUNCHECK(dns_requestmgr_create(mctx, timermgr, socketmgr, taskmgr,
209					    dispatchmgr, dispatchv4, NULL,
210					    &requestmgr));
211
212	ring = NULL;
213	RUNCHECK(dns_tsigkeyring_create(mctx, &ring));
214	tctx = NULL;
215	RUNCHECK(dns_tkeyctx_create(mctx, ectx, &tctx));
216
217	view = NULL;
218	RUNCHECK(dns_view_create(mctx, 0, "_test", &view));
219	dns_view_setkeyring(view, ring);
220
221	sock = NULL;
222	RUNCHECK(isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp,
223				   &sock));
224
225	RUNCHECK(isc_app_onrun(mctx, task, sendquery, NULL));
226
227	dstkey = NULL;
228	type = DST_TYPE_PUBLIC | DST_TYPE_PRIVATE | DST_TYPE_KEY;
229	result = dst_key_fromnamedfile(keyname, NULL, type, mctx, &dstkey);
230	CHECK("dst_key_fromnamedfile", result);
231	result = dns_tsigkey_createfromkey(dst_key_name(dstkey),
232					   DNS_TSIG_HMACMD5_NAME,
233					   dstkey, ISC_TRUE, NULL, 0, 0,
234					   mctx, ring, &tsigkey);
235	dst_key_free(&dstkey);
236	CHECK("dns_tsigkey_createfromkey", result);
237
238	(void)isc_app_run();
239
240	dns_requestmgr_shutdown(requestmgr);
241	dns_requestmgr_detach(&requestmgr);
242	dns_dispatch_detach(&dispatchv4);
243	dns_dispatchmgr_destroy(&dispatchmgr);
244	isc_task_shutdown(task);
245	isc_task_detach(&task);
246	isc_taskmgr_destroy(&taskmgr);
247	isc_socket_detach(&sock);
248	isc_socketmgr_destroy(&socketmgr);
249	isc_timermgr_destroy(&timermgr);
250
251	dns_tsigkeyring_detach(&ring);
252
253	dns_tsigkey_detach(&tsigkey);
254
255	dns_tkeyctx_destroy(&tctx);
256
257	dns_view_detach(&view);
258
259	isc_log_destroy(&log);
260
261	dst_lib_destroy();
262	isc_hash_destroy();
263	isc_entropy_detach(&ectx);
264
265	isc_mem_destroy(&mctx);
266
267	isc_app_finish();
268
269	return (0);
270}
271