/*
 * Copyright (C) Internet Systems Consortium, Inc. ("ISC")
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

/* $Id: key_25.c,v 1.13 2020/02/26 18:47:59 florian Exp $ */

/*
 * Reviewed: Wed Mar 15 16:47:10 PST 2000 by halley.
 */

/* RFC2535 */

#ifndef RDATA_GENERIC_KEY_25_C
#define RDATA_GENERIC_KEY_25_C

#include <dst/dst.h>

static inline isc_result_t
generic_totext_key(ARGS_TOTEXT) {
        isc_region_t sr;
        char buf[sizeof("[key id = 64000]")];
        unsigned int flags;
        unsigned char algorithm;
        char algbuf[DNS_NAME_FORMATSIZE];
        const char *keyinfo;
        isc_region_t tmpr;

        REQUIRE(rdata->length != 0);

        dns_rdata_toregion(rdata, &sr);

        /* flags */
        flags = uint16_fromregion(&sr);
        isc_region_consume(&sr, 2);
        snprintf(buf, sizeof(buf), "%u", flags);
        RETERR(isc_str_tobuffer(buf, target));
        RETERR(isc_str_tobuffer(" ", target));
        if ((flags & DNS_KEYFLAG_KSK) != 0) {
                if (flags & DNS_KEYFLAG_REVOKE)
                        keyinfo = "revoked KSK";
                else
                        keyinfo = "KSK";
        } else
                keyinfo = "ZSK";

        /* protocol */
        snprintf(buf, sizeof(buf), "%u", sr.base[0]);
        isc_region_consume(&sr, 1);
        RETERR(isc_str_tobuffer(buf, target));
        RETERR(isc_str_tobuffer(" ", target));

        /* algorithm */
        algorithm = sr.base[0];
        snprintf(buf, sizeof(buf), "%u", algorithm);
        isc_region_consume(&sr, 1);
        RETERR(isc_str_tobuffer(buf, target));

        /* No Key? */
        if ((flags & 0xc000) == 0xc000)
                return (ISC_R_SUCCESS);

        if ((tctx->flags & DNS_STYLEFLAG_RRCOMMENT) != 0 &&
             algorithm == DNS_KEYALG_PRIVATEDNS) {
                dns_name_t name;
                dns_name_init(&name, NULL);
                dns_name_fromregion(&name, &sr);
                dns_name_format(&name, algbuf, sizeof(algbuf));
        } else {
                dns_secalg_format((dns_secalg_t) algorithm, algbuf,
                                  sizeof(algbuf));
        }

        /* key */
        if ((tctx->flags & DNS_STYLEFLAG_MULTILINE) != 0)
                RETERR(isc_str_tobuffer(" (", target));
        RETERR(isc_str_tobuffer(tctx->linebreak, target));

        if ((tctx->flags & DNS_STYLEFLAG_NOCRYPTO) == 0) {
                if (tctx->width == 0)   /* No splitting */
                        RETERR(isc_base64_totext(&sr, 60, "", target));
                else
                        RETERR(isc_base64_totext(&sr, tctx->width - 2,
                                                 tctx->linebreak, target));
        } else {
                dns_rdata_toregion(rdata, &tmpr);
                snprintf(buf, sizeof(buf), "[key id = %u]",
                         dst_region_computeid(&tmpr, algorithm));
                RETERR(isc_str_tobuffer(buf, target));
        }

        if ((tctx->flags & DNS_STYLEFLAG_RRCOMMENT) != 0)
                RETERR(isc_str_tobuffer(tctx->linebreak, target));
        else if ((tctx->flags & DNS_STYLEFLAG_MULTILINE) != 0)
                RETERR(isc_str_tobuffer(" ", target));

        if ((tctx->flags & DNS_STYLEFLAG_MULTILINE) != 0)
                RETERR(isc_str_tobuffer(")", target));

        if ((tctx->flags & DNS_STYLEFLAG_RRCOMMENT) != 0) {

                if (rdata->type == dns_rdatatype_dnskey ||
                    rdata->type == dns_rdatatype_cdnskey) {
                        RETERR(isc_str_tobuffer(" ; ", target));
                        RETERR(isc_str_tobuffer(keyinfo, target));
                }
                RETERR(isc_str_tobuffer("; alg = ", target));
                RETERR(isc_str_tobuffer(algbuf, target));
                RETERR(isc_str_tobuffer(" ; key id = ", target));
                dns_rdata_toregion(rdata, &tmpr);
                snprintf(buf, sizeof(buf), "%u",
                         dst_region_computeid(&tmpr, algorithm));
                RETERR(isc_str_tobuffer(buf, target));
        }
        return (ISC_R_SUCCESS);
}

static inline isc_result_t
generic_fromwire_key(ARGS_FROMWIRE) {
        unsigned char algorithm;
        isc_region_t sr;

        UNUSED(type);
        UNUSED(rdclass);
        UNUSED(dctx);
        UNUSED(options);

        isc_buffer_activeregion(source, &sr);
        if (sr.length < 4)
                return (ISC_R_UNEXPECTEDEND);

        algorithm = sr.base[3];
        RETERR(isc_mem_tobuffer(target, sr.base, 4));
        isc_region_consume(&sr, 4);
        isc_buffer_forward(source, 4);

        if (algorithm == DNS_KEYALG_PRIVATEDNS) {
                dns_name_t name;
                dns_decompress_setmethods(dctx, DNS_COMPRESS_NONE);
                dns_name_init(&name, NULL);
                RETERR(dns_name_fromwire(&name, source, dctx, options, target));
        }

        /*
         * RSAMD5 computes key ID differently from other
         * algorithms: we need to ensure there's enough data
         * present for the computation
         */
        if (algorithm == DST_ALG_RSAMD5 && sr.length < 3)
                return (ISC_R_UNEXPECTEDEND);

        isc_buffer_activeregion(source, &sr);
        isc_buffer_forward(source, sr.length);
        return (isc_mem_tobuffer(target, sr.base, sr.length));
}

static inline isc_result_t
totext_key(ARGS_TOTEXT) {

        REQUIRE(rdata != NULL);
        REQUIRE(rdata->type == dns_rdatatype_key);

        return (generic_totext_key(rdata, tctx, target));
}

static inline isc_result_t
fromwire_key(ARGS_FROMWIRE) {

        REQUIRE(type == dns_rdatatype_key);

        return (generic_fromwire_key(rdclass, type, source, dctx,
                                     options, target));
}

static inline isc_result_t
towire_key(ARGS_TOWIRE) {
        isc_region_t sr;

        REQUIRE(rdata != NULL);
        REQUIRE(rdata->type == dns_rdatatype_key);
        REQUIRE(rdata->length != 0);

        UNUSED(cctx);

        dns_rdata_toregion(rdata, &sr);
        return (isc_mem_tobuffer(target, sr.base, sr.length));
}

#endif  /* RDATA_GENERIC_KEY_25_C */