rdata/in_1/wks_11.c

	wks_11.c revision 9f139761ca06977d1db8051842efc620c15b8199
/*
 * Copyright (C) 1999, 2000  Internet Software Consortium.
 *
 * Permission to use, copy, modify, and 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 INTERNET SOFTWARE CONSORTIUM
 * DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
 * INTERNET SOFTWARE CONSORTIUM 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: wks_11.c,v 1.35 2000/08/24 21:41:44 gson Exp $ */

/* Reviewed: Fri Mar 17 15:01:49 PST 2000 by explorer */

#ifndef RDATA_IN_1_WKS_11_C
#define RDATA_IN_1_WKS_11_C

#include <limits.h>
#include <stdlib.h>

#include <isc/net.h>
#include <isc/netdb.h>

#define RRTYPE_WKS_ATTRIBUTES (0)

static inline isc_result_t
fromtext_in_wks(ARGS_FROMTEXT) {
    isc_token_t token;
    isc_region_t region;
    struct in_addr addr;
    struct protoent *pe;
    struct servent *se;
    char *e;
    long proto;
    unsigned char bm[8*1024]; /* 64k bits */
    long port;
    long maxport = -1;
    const char *ps = NULL;
    unsigned int n;
    char service[32];
    int i;

    UNUSED(origin);
    UNUSED(downcase);

    REQUIRE(type == 11);
    REQUIRE(rdclass == 1);

    /*
     * IPv4 dotted quad.
     */
    RETERR(gettoken(lexer, &token, isc_tokentype_string, ISC_FALSE));

    isc_buffer_availableregion(target, &region);
    if (inet_aton(token.value.as_pointer, &addr) != 1)
        return (DNS_R_BADDOTTEDQUAD);
    if (region.length < 4)
        return (ISC_R_NOSPACE);
    memcpy(region.base, &addr, 4);
    isc_buffer_add(target, 4);

    /*
     * Protocol.
     */
    RETERR(gettoken(lexer, &token, isc_tokentype_string, ISC_FALSE));

    proto = strtol(token.value.as_pointer, &e, 10);
    if (*e == 0)
        ;
    else if ((pe = getprotobyname(token.value.as_pointer)) != NULL)
        proto = pe->p_proto;
    else
        return (ISC_R_UNEXPECTED);
    if (proto < 0 || proto > 0xff)
        return (ISC_R_RANGE);

    if (proto == IPPROTO_TCP)
        ps = "tcp";
    else if (proto == IPPROTO_UDP)
        ps = "udp";

    RETERR(uint8_tobuffer(proto, target));

    memset(bm, 0, sizeof bm);
    do {
        RETERR(gettoken(lexer, &token, isc_tokentype_string,
                  ISC_TRUE));
        if (token.type != isc_tokentype_string)
            break;

        /*
         * Lowercase the service string as some getservbyname() are
         * case sensitive and the database is usually in lowercase.
         */
        strncpy(service, token.value.as_pointer, sizeof(service));
        service[sizeof(service)-1] = '\0';
        for (i = strlen(service) - 1; i >= 0; i--)
            if (isupper(service[i]&0xff))
                service[i] = tolower(service[i]);

        port = strtol(token.value.as_pointer, &e, 10);
        if (*e == 0)
            ;
        else if ((se = getservbyname(service, ps)) != NULL)
            port = ntohs(se->s_port);
        else if ((se = getservbyname(token.value.as_pointer, ps))
              != NULL)
            port = ntohs(se->s_port);
        else
            return (ISC_R_UNEXPECTED);
        if (port < 0 || port > 0xffff)
            return (ISC_R_RANGE);
        if (port > maxport)
            maxport = port;
        bm[port / 8] |= (0x80 >> (port % 8));
    } while (1);

    /*
     * Let upper layer handle eol/eof.
     */
    isc_lex_ungettoken(lexer, &token);

    n = (maxport + 8) / 8;
    return (mem_tobuffer(target, bm, n));
}

static inline isc_result_t
totext_in_wks(ARGS_TOTEXT) {
    isc_region_t sr;
    isc_region_t tr;
    unsigned short proto;
    char buf[sizeof "65535"];
    unsigned int i, j;

    UNUSED(tctx);

    REQUIRE(rdata->type == 11);
    REQUIRE(rdata->rdclass == 1);

    dns_rdata_toregion(rdata, &sr);
    isc_buffer_availableregion(target, &tr);
    if (inet_ntop(AF_INET, sr.base, (char *)tr.base, tr.length) == NULL)
        return (ISC_R_NOSPACE);
    isc_buffer_add(target, strlen((char *)tr.base));
    isc_region_consume(&sr, 4);

    proto = uint8_fromregion(&sr);
    sprintf(buf, "%u", proto);
    RETERR(str_totext(" ", target));
    RETERR(str_totext(buf, target));
    isc_region_consume(&sr, 1);

    for (i = 0 ; i < sr.length ; i++) {
        if (sr.base[i] != 0)
            for (j = 0 ; j < 8 ; j++)
                if ((sr.base[i] & (0x80 >> j)) != 0) {
                    sprintf(buf, "%u", i * 8 + j);
                    RETERR(str_totext(" ", target));
                    RETERR(str_totext(buf, target));
                }
    }

    return (ISC_R_SUCCESS);
}

static inline isc_result_t
fromwire_in_wks(ARGS_FROMWIRE) {
    isc_region_t sr;
    isc_region_t tr;

    UNUSED(dctx);
    UNUSED(downcase);

    REQUIRE(type == 11);
    REQUIRE(rdclass == 1);

    isc_buffer_activeregion(source, &sr);
    isc_buffer_availableregion(target, &tr);

    if (sr.length < 5)
        return (ISC_R_UNEXPECTEDEND);
    if (sr.length > 8 * 1024 + 5)
        return (DNS_R_EXTRADATA);
    if (tr.length < sr.length)
        return (ISC_R_NOSPACE);

    memcpy(tr.base, sr.base, sr.length);
    isc_buffer_add(target, sr.length);
    isc_buffer_forward(source, sr.length);

    return (ISC_R_SUCCESS);
}

static inline isc_result_t
towire_in_wks(ARGS_TOWIRE) {
    isc_region_t sr;

    UNUSED(cctx);

    REQUIRE(rdata->type == 11);
    REQUIRE(rdata->rdclass == 1);

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

static inline int
compare_in_wks(ARGS_COMPARE) {
    isc_region_t r1;
    isc_region_t r2;

    REQUIRE(rdata1->type == rdata2->type);
    REQUIRE(rdata1->rdclass == rdata2->rdclass);
    REQUIRE(rdata1->type == 11);
    REQUIRE(rdata1->rdclass == 1);

    dns_rdata_toregion(rdata1, &r1);
    dns_rdata_toregion(rdata2, &r2);
    return (compare_region(&r1, &r2));
}

static inline isc_result_t
fromstruct_in_wks(ARGS_FROMSTRUCT) {
    dns_rdata_in_wks_t *wks = source;
    isc_uint32_t a;

    REQUIRE(type == 11);
    REQUIRE(rdclass == 1);
    REQUIRE(source != NULL);
    REQUIRE(wks->common.rdtype == type);
    REQUIRE(wks->common.rdclass == rdclass);

    a = ntohl(wks->in_addr.s_addr);
    RETERR(uint32_tobuffer(a, target));
    RETERR(uint16_tobuffer(wks->protocol, target));
    return (mem_tobuffer(target, wks->map, wks->map_len));
}

static inline isc_result_t
tostruct_in_wks(ARGS_TOSTRUCT) {
    dns_rdata_in_wks_t *wks = target;
    isc_uint32_t n;
    isc_region_t region;

    REQUIRE(rdata->type == 11);
    REQUIRE(rdata->rdclass == 1);

    wks->common.rdclass = rdata->rdclass;
    wks->common.rdtype = rdata->type;
    ISC_LINK_INIT(&wks->common, link);

    dns_rdata_toregion(rdata, &region);
    n = uint32_fromregion(&region);
    wks->in_addr.s_addr = htonl(n);
    isc_region_consume(&region, 4);
    wks->protocol = uint16_fromregion(&region);
    isc_region_consume(&region, 2);
    wks->map_len = region.length;
    if (wks->map_len > 0) {
        wks->map = mem_maybedup(mctx, region.base, region.length);
        if (wks->map == NULL)
            return (ISC_R_NOMEMORY);
    } else
        wks->map = NULL;
    wks->mctx = mctx;
    return (ISC_R_SUCCESS);
}

static inline void
freestruct_in_wks(ARGS_FREESTRUCT) {
    dns_rdata_in_wks_t *wks = source;

    REQUIRE(source != NULL);
    REQUIRE(wks->common.rdtype == 11);
    REQUIRE(wks->common.rdclass == 1);

    if (wks->mctx == NULL)
        return;

    if (wks->map != NULL)
        isc_mem_free(wks->mctx, wks->map);
    wks->mctx = NULL;
}

static inline isc_result_t
additionaldata_in_wks(ARGS_ADDLDATA) {
    UNUSED(rdata);
    UNUSED(add);
    UNUSED(arg);

    REQUIRE(rdata->type == 11);
    REQUIRE(rdata->rdclass == 1);

    return (ISC_R_SUCCESS);
}

static inline isc_result_t
digest_in_wks(ARGS_DIGEST) {
    isc_region_t r;

    REQUIRE(rdata->type == 11);
    REQUIRE(rdata->rdclass == 1);

    dns_rdata_toregion(rdata, &r);

    return ((digest)(arg, &r));
}

#endif  /* RDATA_IN_1_WKS_11_C */