system/tkey/keydelete.c

/*
 * Copyright (C) 2001, 2004, 2005, 2007, 2009-2011, 2014-2016  Internet Systems Consortium, Inc. ("ISC")
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/* $Id: keydelete.c,v 1.18 2011/01/11 23:47:13 tbox Exp $ */

#include <config.h>

#include <stdlib.h>
#include <string.h>

#include <isc/app.h>
#include <isc/base64.h>
#include <isc/entropy.h>
#include <isc/hash.h>
#include <isc/log.h>
#include <isc/mem.h>
#include <isc/print.h>
#include <isc/sockaddr.h>
#include <isc/socket.h>
#include <isc/task.h>
#include <isc/timer.h>
#include <isc/util.h>

#include <pk11/site.h>

#include <dns/dispatch.h>
#include <dns/fixedname.h>
#include <dns/keyvalues.h>
#include <dns/message.h>
#include <dns/name.h>
#include <dns/request.h>
#include <dns/result.h>
#include <dns/tkey.h>
#include <dns/tsig.h>
#include <dns/view.h>

#include <dst/result.h>

#define CHECK(str, x) { \
    if ((x) != ISC_R_SUCCESS) { \
        fprintf(stderr, "I:%s: %s\n", (str), isc_result_totext(x)); \
        exit(-1); \
    } \
}

#define RUNCHECK(x) RUNTIME_CHECK((x) == ISC_R_SUCCESS)

#define PORT 5300
#define TIMEOUT 30

static isc_mem_t *mctx;
static dns_tsigkey_t *tsigkey;
static dns_tsig_keyring_t *ring;
static dns_requestmgr_t *requestmgr;

static void
recvquery(isc_task_t *task, isc_event_t *event) {
    dns_requestevent_t *reqev = (dns_requestevent_t *)event;
    isc_result_t result;
    dns_message_t *query, *response;

    UNUSED(task);

    REQUIRE(reqev != NULL);

    if (reqev->result != ISC_R_SUCCESS) {
        fprintf(stderr, "I:request event result: %s\n",
            isc_result_totext(reqev->result));
        exit(-1);
    }

    query = reqev->ev_arg;

    response = NULL;
    result = dns_message_create(mctx, DNS_MESSAGE_INTENTPARSE, &response);
    CHECK("dns_message_create", result);

    result = dns_request_getresponse(reqev->request, response,
                     DNS_MESSAGEPARSE_PRESERVEORDER);
    CHECK("dns_request_getresponse", result);

    if (response->rcode != dns_rcode_noerror) {
        result = ISC_RESULTCLASS_DNSRCODE + response->rcode;
        fprintf(stderr, "I:response rcode: %s\n",
            isc_result_totext(result));
            exit(-1);
    }

    result = dns_tkey_processdeleteresponse(query, response, ring);
    CHECK("dns_tkey_processdhresponse", result);

    dns_message_destroy(&query);
    dns_message_destroy(&response);
    dns_request_destroy(&reqev->request);
    isc_event_free(&event);
    isc_app_shutdown();
    return;
}

static void
sendquery(isc_task_t *task, isc_event_t *event) {
    struct in_addr inaddr;
    isc_sockaddr_t address;
    isc_result_t result;
    dns_message_t *query;
    dns_request_t *request;

    isc_event_free(&event);

    result = ISC_R_FAILURE;
    if (inet_pton(AF_INET, "10.53.0.1", &inaddr) != 1)
        CHECK("inet_pton", result);
    isc_sockaddr_fromin(&address, &inaddr, PORT);

    query = NULL;
    result = dns_message_create(mctx, DNS_MESSAGE_INTENTRENDER, &query);
    CHECK("dns_message_create", result);

    result = dns_tkey_builddeletequery(query, tsigkey);
    CHECK("dns_tkey_builddeletequery", result);

    request = NULL;
    result = dns_request_create(requestmgr, query, &address,
                    DNS_REQUESTOPT_TCP, tsigkey, TIMEOUT,
                    task, recvquery, query, &request);
    CHECK("dns_request_create", result);
}

int
main(int argc, char **argv) {
    char *keyname;
    isc_taskmgr_t *taskmgr;
    isc_timermgr_t *timermgr;
    isc_socketmgr_t *socketmgr;
    isc_socket_t *sock;
    unsigned int attrs, attrmask;
    isc_sockaddr_t bind_any;
    dns_dispatchmgr_t *dispatchmgr;
    dns_dispatch_t *dispatchv4;
    dns_view_t *view;
    isc_entropy_t *ectx;
    dns_tkeyctx_t *tctx;
    dst_key_t *dstkey;
    isc_log_t *log;
    isc_logconfig_t *logconfig;
    isc_task_t *task;
    isc_result_t result;
    int type;

    RUNCHECK(isc_app_start());

    if (argc < 2) {
        fprintf(stderr, "I:no key to delete\n");
        exit(-1);
    }
    keyname = argv[1];

    dns_result_register();

    mctx = NULL;
    RUNCHECK(isc_mem_create(0, 0, &mctx));

    ectx = NULL;
    RUNCHECK(isc_entropy_create(mctx, &ectx));
    RUNCHECK(isc_entropy_createfilesource(ectx, "../random.data"));
    RUNCHECK(isc_hash_create(mctx, ectx, DNS_NAME_MAXWIRE));

    log = NULL;
    logconfig = NULL;
    RUNCHECK(isc_log_create(mctx, &log, &logconfig));

    RUNCHECK(dst_lib_init(mctx, ectx, ISC_ENTROPY_GOODONLY));

    taskmgr = NULL;
    RUNCHECK(isc_taskmgr_create(mctx, 1, 0, &taskmgr));
    task = NULL;
    RUNCHECK(isc_task_create(taskmgr, 0, &task));
    timermgr = NULL;
    RUNCHECK(isc_timermgr_create(mctx, &timermgr));
    socketmgr = NULL;
    RUNCHECK(isc_socketmgr_create(mctx, &socketmgr));
    dispatchmgr = NULL;
    RUNCHECK(dns_dispatchmgr_create(mctx, NULL, &dispatchmgr));
    isc_sockaddr_any(&bind_any);
    attrs = DNS_DISPATCHATTR_UDP |
        DNS_DISPATCHATTR_MAKEQUERY |
        DNS_DISPATCHATTR_IPV4;
    attrmask = DNS_DISPATCHATTR_UDP |
           DNS_DISPATCHATTR_TCP |
           DNS_DISPATCHATTR_IPV4 |
           DNS_DISPATCHATTR_IPV6;
    dispatchv4 = NULL;
    RUNCHECK(dns_dispatch_getudp(dispatchmgr, socketmgr, taskmgr,
                      &bind_any, 4096, 4, 2, 3, 5,
                      attrs, attrmask, &dispatchv4));
    requestmgr = NULL;
    RUNCHECK(dns_requestmgr_create(mctx, timermgr, socketmgr, taskmgr,
                        dispatchmgr, dispatchv4, NULL,
                        &requestmgr));

    ring = NULL;
    RUNCHECK(dns_tsigkeyring_create(mctx, &ring));
    tctx = NULL;
    RUNCHECK(dns_tkeyctx_create(mctx, ectx, &tctx));

    view = NULL;
    RUNCHECK(dns_view_create(mctx, 0, "_test", &view));
    dns_view_setkeyring(view, ring);

    sock = NULL;
    RUNCHECK(isc_socket_create(socketmgr, PF_INET, isc_sockettype_udp,
                   &sock));

    RUNCHECK(isc_app_onrun(mctx, task, sendquery, NULL));

    dstkey = NULL;
    type = DST_TYPE_PUBLIC | DST_TYPE_PRIVATE | DST_TYPE_KEY;
    result = dst_key_fromnamedfile(keyname, NULL, type, mctx, &dstkey);
    CHECK("dst_key_fromnamedfile", result);
#ifndef PK11_MD5_DISABLE
    result = dns_tsigkey_createfromkey(dst_key_name(dstkey),
                       DNS_TSIG_HMACMD5_NAME,
                       dstkey, ISC_TRUE, NULL, 0, 0,
                       mctx, ring, &tsigkey);
    dst_key_free(&dstkey);
    CHECK("dns_tsigkey_createfromkey", result);
#else
    dst_key_free(&dstkey);
    CHECK("MD5 was disabled", ISC_R_NOTIMPLEMENTED);
#endif

    (void)isc_app_run();

    dns_requestmgr_shutdown(requestmgr);
    dns_requestmgr_detach(&requestmgr);
    dns_dispatch_detach(&dispatchv4);
    dns_dispatchmgr_destroy(&dispatchmgr);
    isc_task_shutdown(task);
    isc_task_detach(&task);
    isc_taskmgr_destroy(&taskmgr);
    isc_socket_detach(&sock);
    isc_socketmgr_destroy(&socketmgr);
    isc_timermgr_destroy(&timermgr);

    dns_tsigkeyring_detach(&ring);

    dns_tsigkey_detach(&tsigkey);

    dns_tkeyctx_destroy(&tctx);

    dns_view_detach(&view);

    isc_log_destroy(&log);

    dst_lib_destroy();
    isc_hash_destroy();
    isc_entropy_detach(&ectx);

    isc_mem_destroy(&mctx);

    isc_app_finish();

    return (0);
}