/*
 * testcode/signit.c - debug tool to sign rrsets with given keys.
 *
 * Copyright (c) 2007, NLnet Labs. All rights reserved.
 *
 * This software is open source.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * Neither the name of the NLNET LABS nor the names of its contributors may
 * be used to endorse or promote products derived from this software without
 * specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/**
 * \file
 *
 * This program signs rrsets with the given keys. It can be used to 
 * construct input to test the validator with.
 */
#include "config.h"
#include <ldns/ldns.h>
#include <assert.h>

#define DNSKEY_BIT_ZSK 0x0100

/**
 * Key settings
 */
struct keysets {
        /** signature inception */
        uint32_t incep;
        /** signature expiration */
        uint32_t expi;
        /** owner name */
        char* owner;
        /** keytag */
        uint16_t keytag;
        /** DNSKEY flags */
        uint16_t flags;
};

/** print usage and exit */
static void
usage(void)
{
        printf("usage:  signit expi ince keytag owner keyfile\n");
        printf("present rrset data on stdin.\n");
        printf("signed data is printed to stdout.\n");
        printf("\n");
        printf("Or use: signit NSEC3PARAM hash flags iter salt\n");
        printf("present names on stdin, hashed names are printed to stdout.\n");
        exit(1);
}

static time_t 
convert_timeval(const char* str)
{
        time_t t;
        struct tm tm;
        memset(&tm, 0, sizeof(tm));
        if(strlen(str) < 14)
                return 0;
        if(sscanf(str, "%4d%2d%2d%2d%2d%2d", &tm.tm_year, &tm.tm_mon, 
                &tm.tm_mday, &tm.tm_hour, &tm.tm_min, &tm.tm_sec) != 6)
                return 0;
        tm.tm_year -= 1900;
        tm.tm_mon--;
        /* Check values */
        if (tm.tm_year < 70)    return 0;
        if (tm.tm_mon < 0 || tm.tm_mon > 11)    return 0;
        if (tm.tm_mday < 1 || tm.tm_mday > 31)  return 0;
        if (tm.tm_hour < 0 || tm.tm_hour > 23)  return 0;
        if (tm.tm_min < 0 || tm.tm_min > 59)    return 0;
        if (tm.tm_sec < 0 || tm.tm_sec > 59)    return 0;
        /* call ldns conversion function */
        t = ldns_mktime_from_utc(&tm);
        return t;
}

static void fatal_exit(const char* format, ...)
{
        va_list args;
        va_start(args, format);
        printf("fatal exit: ");
        vprintf(format, args);
        va_end(args);
        exit(1);
}

/** read expi ince keytag owner from cmdline */
static void
parse_cmdline(char *argv[], struct keysets* s)
{
        s->expi = convert_timeval(argv[1]);
        s->incep = convert_timeval(argv[2]);
        s->keytag = (uint16_t)atoi(argv[3]);
        s->owner = argv[4];
        s->flags = DNSKEY_BIT_ZSK; /* to enforce signing */
}

/** read all key files, exit on error */
static ldns_key_list*
read_keys(int num, char* names[], struct keysets* set)
{
        int i;
        ldns_key_list* keys = ldns_key_list_new();
        ldns_key* k;
        ldns_rdf* rdf;
        ldns_status s;
        int b;
        FILE* in;

        if(!keys) fatal_exit("alloc failure");
        for(i=0; i<num; i++) {
                printf("read keyfile %s\n", names[i]);
                in = fopen(names[i], "r");
                if(!in) fatal_exit("could not open %s: %s", names[i],
                                strerror(errno));
                s = ldns_key_new_frm_fp(&k, in);
                fclose(in);
                if(s != LDNS_STATUS_OK)
                        fatal_exit("bad keyfile %s: %s", names[i],
                                ldns_get_errorstr_by_id(s));
                ldns_key_set_expiration(k, set->expi);
                ldns_key_set_inception(k, set->incep);
                s = ldns_str2rdf_dname(&rdf, set->owner);
                if(s != LDNS_STATUS_OK)
                        fatal_exit("bad owner name %s: %s", set->owner,
                                ldns_get_errorstr_by_id(s));
                ldns_key_set_pubkey_owner(k, rdf);
                ldns_key_set_flags(k, set->flags);
                ldns_key_set_keytag(k, set->keytag);
                b = ldns_key_list_push_key(keys, k);
                assert(b);
        }
        return keys;
}

/** read list of rrs from the file */
static ldns_rr_list*
read_rrs(FILE* in)
{
        uint32_t my_ttl = 3600;
        ldns_rdf *my_origin = NULL;
        ldns_rdf *my_prev = NULL;
        ldns_status s;
        int line_nr = 1;
        int b;

        ldns_rr_list* list;
        ldns_rr *rr;

        list = ldns_rr_list_new();
        if(!list) fatal_exit("alloc error");

        while(!feof(in)) {
                s = ldns_rr_new_frm_fp_l(&rr, in, &my_ttl, &my_origin,
                        &my_prev, &line_nr);
                if(s == LDNS_STATUS_SYNTAX_TTL || 
                        s == LDNS_STATUS_SYNTAX_ORIGIN ||
                        s == LDNS_STATUS_SYNTAX_EMPTY)
                        continue;
                else if(s != LDNS_STATUS_OK)
                        fatal_exit("parse error in line %d: %s", line_nr,
                                ldns_get_errorstr_by_id(s));
                b = ldns_rr_list_push_rr(list, rr);
                assert(b);
        }
        printf("read %d lines\n", line_nr);

        return list;
}

/** sign the rrs with the keys */
static void
signit(ldns_rr_list* rrs, ldns_key_list* keys)
{
        ldns_rr_list* rrset;
        ldns_rr_list* sigs;
        
        while(ldns_rr_list_rr_count(rrs) > 0) {
                rrset = ldns_rr_list_pop_rrset(rrs);
                if(!rrset) fatal_exit("copy alloc failure");
                sigs = ldns_sign_public(rrset, keys);
                if(!sigs) fatal_exit("failed to sign");
                ldns_rr_list_print(stdout, rrset);
                ldns_rr_list_print(stdout, sigs);
                printf("\n");
                ldns_rr_list_free(rrset);
                ldns_rr_list_free(sigs);
        }
}

/** process keys and signit */
static void
process_keys(int argc, char* argv[])
{
        ldns_rr_list* rrs;
        ldns_key_list* keys;
        struct keysets settings;
        assert(argc == 6);

        parse_cmdline(argv, &settings);
        keys = read_keys(1, argv+5, &settings);
        rrs = read_rrs(stdin);
        signit(rrs, keys);

        ldns_rr_list_deep_free(rrs);
        ldns_key_list_free(keys);
}

/** process nsec3 params and perform hashing */
static void
process_nsec3(int argc, char* argv[])
{
        char line[10240];
        ldns_rdf* salt;
        ldns_rdf* in, *out;
        ldns_status status;
        status = ldns_str2rdf_nsec3_salt(&salt, argv[5]);
        if(status != LDNS_STATUS_OK)
                fatal_exit("Could not parse salt %s: %s", argv[5],
                        ldns_get_errorstr_by_id(status));
        assert(argc == 6);
        while(fgets(line, (int)sizeof(line), stdin)) {
                if(strlen(line) > 0)
                        line[strlen(line)-1] = 0; /* remove trailing newline */
                if(line[0]==0)
                        continue;
                status = ldns_str2rdf_dname(&in, line);
                if(status != LDNS_STATUS_OK)
                        fatal_exit("Could not parse name %s: %s", line,
                                ldns_get_errorstr_by_id(status));
                ldns_rdf_print(stdout, in);
                printf(" -> ");
                /* arg 3 is flags, unused */
                out = ldns_nsec3_hash_name(in, (uint8_t)atoi(argv[2]), 
                        (uint16_t)atoi(argv[4]),
                        ldns_rdf_data(salt)[0], ldns_rdf_data(salt)+1);
                if(!out)
                        fatal_exit("Could not hash %s", line);
                ldns_rdf_print(stdout, out);
                printf("\n");
                ldns_rdf_deep_free(in);
                ldns_rdf_deep_free(out);
        }
        ldns_rdf_deep_free(salt);
}

/** main program */
int main(int argc, char* argv[])
{
        if(argc != 6) {
                usage();
        }
        if(strcmp(argv[1], "NSEC3PARAM") == 0) {
                process_nsec3(argc, argv);
                return 0;
        }
        process_keys(argc, argv);
        return 0;
}