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