2
0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-27 12:38:24 +00:00
bind/lib/dns/rdata/generic/naptr_35.c
Tony Finch 1d807d84f1 Shrink decompression contexts
It's wasteful to use 20 bytes and a pointer indirection to represent
two bits of information, so turn the struct into an enum. And change
the names of the enumeration constants to make the intent more clear.

This change introduces some inline functions into another header,
which confuses `gcovr` when it is trying to collect code coverage
statistics. So, in the CI job, copy more header files into a directory
where `gcovr` looks for them.
2022-06-01 13:00:40 +01:00

718 lines
14 KiB
C

/*
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
*
* SPDX-License-Identifier: MPL-2.0
*
* 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 https://mozilla.org/MPL/2.0/.
*
* See the COPYRIGHT file distributed with this work for additional
* information regarding copyright ownership.
*/
/* RFC2915 */
#ifndef RDATA_GENERIC_NAPTR_35_C
#define RDATA_GENERIC_NAPTR_35_C
#define RRTYPE_NAPTR_ATTRIBUTES (0)
#include <isc/regex.h>
/*
* Check the wire format of the Regexp field.
* Don't allow embedded NUL's.
*/
static isc_result_t
txt_valid_regex(const unsigned char *txt) {
unsigned int nsub = 0;
char regex[256];
char *cp;
bool flags = false;
bool replace = false;
unsigned char c;
unsigned char delim;
unsigned int len;
int n;
len = *txt++;
if (len == 0U) {
return (ISC_R_SUCCESS);
}
delim = *txt++;
len--;
/*
* Digits, backslash and flags can't be delimiters.
*/
switch (delim) {
case '0':
case '1':
case '2':
case '3':
case '4':
case '5':
case '6':
case '7':
case '8':
case '9':
case '\\':
case 'i':
case 0:
return (DNS_R_SYNTAX);
}
cp = regex;
while (len-- > 0) {
c = *txt++;
if (c == 0) {
return (DNS_R_SYNTAX);
}
if (c == delim && !replace) {
replace = true;
continue;
} else if (c == delim && !flags) {
flags = true;
continue;
} else if (c == delim) {
return (DNS_R_SYNTAX);
}
/*
* Flags are not escaped.
*/
if (flags) {
switch (c) {
case 'i':
continue;
default:
return (DNS_R_SYNTAX);
}
}
if (!replace) {
*cp++ = c;
}
if (c == '\\') {
if (len == 0) {
return (DNS_R_SYNTAX);
}
c = *txt++;
if (c == 0) {
return (DNS_R_SYNTAX);
}
len--;
if (replace) {
switch (c) {
case '0':
return (DNS_R_SYNTAX);
case '1':
if (nsub < 1) {
nsub = 1;
}
break;
case '2':
if (nsub < 2) {
nsub = 2;
}
break;
case '3':
if (nsub < 3) {
nsub = 3;
}
break;
case '4':
if (nsub < 4) {
nsub = 4;
}
break;
case '5':
if (nsub < 5) {
nsub = 5;
}
break;
case '6':
if (nsub < 6) {
nsub = 6;
}
break;
case '7':
if (nsub < 7) {
nsub = 7;
}
break;
case '8':
if (nsub < 8) {
nsub = 8;
}
break;
case '9':
if (nsub < 9) {
nsub = 9;
}
break;
}
}
if (!replace) {
*cp++ = c;
}
}
}
if (!flags) {
return (DNS_R_SYNTAX);
}
*cp = '\0';
n = isc_regex_validate(regex);
if (n < 0 || nsub > (unsigned int)n) {
return (DNS_R_SYNTAX);
}
return (ISC_R_SUCCESS);
}
static isc_result_t
fromtext_naptr(ARGS_FROMTEXT) {
isc_token_t token;
dns_name_t name;
isc_buffer_t buffer;
unsigned char *regex;
REQUIRE(type == dns_rdatatype_naptr);
UNUSED(type);
UNUSED(rdclass);
UNUSED(callbacks);
/*
* Order.
*/
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_number,
false));
if (token.value.as_ulong > 0xffffU) {
RETTOK(ISC_R_RANGE);
}
RETERR(uint16_tobuffer(token.value.as_ulong, target));
/*
* Preference.
*/
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_number,
false));
if (token.value.as_ulong > 0xffffU) {
RETTOK(ISC_R_RANGE);
}
RETERR(uint16_tobuffer(token.value.as_ulong, target));
/*
* Flags.
*/
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_qstring,
false));
RETTOK(txt_fromtext(&token.value.as_textregion, target));
/*
* Service.
*/
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_qstring,
false));
RETTOK(txt_fromtext(&token.value.as_textregion, target));
/*
* Regexp.
*/
regex = isc_buffer_used(target);
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_qstring,
false));
RETTOK(txt_fromtext(&token.value.as_textregion, target));
RETTOK(txt_valid_regex(regex));
/*
* Replacement.
*/
RETERR(isc_lex_getmastertoken(lexer, &token, isc_tokentype_string,
false));
dns_name_init(&name, NULL);
buffer_fromregion(&buffer, &token.value.as_region);
if (origin == NULL) {
origin = dns_rootname;
}
RETTOK(dns_name_fromtext(&name, &buffer, origin, options, target));
return (ISC_R_SUCCESS);
}
static isc_result_t
totext_naptr(ARGS_TOTEXT) {
isc_region_t region;
dns_name_t name;
dns_name_t prefix;
bool sub;
char buf[sizeof("64000")];
unsigned short num;
REQUIRE(rdata->type == dns_rdatatype_naptr);
REQUIRE(rdata->length != 0);
dns_name_init(&name, NULL);
dns_name_init(&prefix, NULL);
dns_rdata_toregion(rdata, &region);
/*
* Order.
*/
num = uint16_fromregion(&region);
isc_region_consume(&region, 2);
snprintf(buf, sizeof(buf), "%u", num);
RETERR(str_totext(buf, target));
RETERR(str_totext(" ", target));
/*
* Preference.
*/
num = uint16_fromregion(&region);
isc_region_consume(&region, 2);
snprintf(buf, sizeof(buf), "%u", num);
RETERR(str_totext(buf, target));
RETERR(str_totext(" ", target));
/*
* Flags.
*/
RETERR(txt_totext(&region, true, target));
RETERR(str_totext(" ", target));
/*
* Service.
*/
RETERR(txt_totext(&region, true, target));
RETERR(str_totext(" ", target));
/*
* Regexp.
*/
RETERR(txt_totext(&region, true, target));
RETERR(str_totext(" ", target));
/*
* Replacement.
*/
dns_name_fromregion(&name, &region);
sub = name_prefix(&name, tctx->origin, &prefix);
return (dns_name_totext(&prefix, sub, target));
}
static isc_result_t
fromwire_naptr(ARGS_FROMWIRE) {
dns_name_t name;
isc_region_t sr;
unsigned char *regex;
REQUIRE(type == dns_rdatatype_naptr);
UNUSED(type);
UNUSED(rdclass);
dctx = dns_decompress_setpermitted(dctx, false);
dns_name_init(&name, NULL);
/*
* Order, preference.
*/
isc_buffer_activeregion(source, &sr);
if (sr.length < 4) {
return (ISC_R_UNEXPECTEDEND);
}
RETERR(mem_tobuffer(target, sr.base, 4));
isc_buffer_forward(source, 4);
/*
* Flags.
*/
RETERR(txt_fromwire(source, target));
/*
* Service.
*/
RETERR(txt_fromwire(source, target));
/*
* Regexp.
*/
regex = isc_buffer_used(target);
RETERR(txt_fromwire(source, target));
RETERR(txt_valid_regex(regex));
/*
* Replacement.
*/
return (dns_name_fromwire(&name, source, dctx, options, target));
}
static isc_result_t
towire_naptr(ARGS_TOWIRE) {
dns_name_t name;
dns_offsets_t offsets;
isc_region_t sr;
REQUIRE(rdata->type == dns_rdatatype_naptr);
REQUIRE(rdata->length != 0);
dns_compress_setpermitted(cctx, false);
/*
* Order, preference.
*/
dns_rdata_toregion(rdata, &sr);
RETERR(mem_tobuffer(target, sr.base, 4));
isc_region_consume(&sr, 4);
/*
* Flags.
*/
RETERR(mem_tobuffer(target, sr.base, sr.base[0] + 1));
isc_region_consume(&sr, sr.base[0] + 1);
/*
* Service.
*/
RETERR(mem_tobuffer(target, sr.base, sr.base[0] + 1));
isc_region_consume(&sr, sr.base[0] + 1);
/*
* Regexp.
*/
RETERR(mem_tobuffer(target, sr.base, sr.base[0] + 1));
isc_region_consume(&sr, sr.base[0] + 1);
/*
* Replacement.
*/
dns_name_init(&name, offsets);
dns_name_fromregion(&name, &sr);
return (dns_name_towire(&name, cctx, target));
}
static int
compare_naptr(ARGS_COMPARE) {
dns_name_t name1;
dns_name_t name2;
isc_region_t region1;
isc_region_t region2;
int order, len;
REQUIRE(rdata1->type == rdata2->type);
REQUIRE(rdata1->rdclass == rdata2->rdclass);
REQUIRE(rdata1->type == dns_rdatatype_naptr);
REQUIRE(rdata1->length != 0);
REQUIRE(rdata2->length != 0);
dns_rdata_toregion(rdata1, &region1);
dns_rdata_toregion(rdata2, &region2);
/*
* Order, preference.
*/
order = memcmp(region1.base, region2.base, 4);
if (order != 0) {
return (order < 0 ? -1 : 1);
}
isc_region_consume(&region1, 4);
isc_region_consume(&region2, 4);
/*
* Flags.
*/
len = ISC_MIN(region1.base[0], region2.base[0]);
order = memcmp(region1.base, region2.base, len + 1);
if (order != 0) {
return (order < 0 ? -1 : 1);
}
isc_region_consume(&region1, region1.base[0] + 1);
isc_region_consume(&region2, region2.base[0] + 1);
/*
* Service.
*/
len = ISC_MIN(region1.base[0], region2.base[0]);
order = memcmp(region1.base, region2.base, len + 1);
if (order != 0) {
return (order < 0 ? -1 : 1);
}
isc_region_consume(&region1, region1.base[0] + 1);
isc_region_consume(&region2, region2.base[0] + 1);
/*
* Regexp.
*/
len = ISC_MIN(region1.base[0], region2.base[0]);
order = memcmp(region1.base, region2.base, len + 1);
if (order != 0) {
return (order < 0 ? -1 : 1);
}
isc_region_consume(&region1, region1.base[0] + 1);
isc_region_consume(&region2, region2.base[0] + 1);
/*
* Replacement.
*/
dns_name_init(&name1, NULL);
dns_name_init(&name2, NULL);
dns_name_fromregion(&name1, &region1);
dns_name_fromregion(&name2, &region2);
return (dns_name_rdatacompare(&name1, &name2));
}
static isc_result_t
fromstruct_naptr(ARGS_FROMSTRUCT) {
dns_rdata_naptr_t *naptr = source;
isc_region_t region;
REQUIRE(type == dns_rdatatype_naptr);
REQUIRE(naptr != NULL);
REQUIRE(naptr->common.rdtype == type);
REQUIRE(naptr->common.rdclass == rdclass);
REQUIRE(naptr->flags != NULL || naptr->flags_len == 0);
REQUIRE(naptr->service != NULL || naptr->service_len == 0);
REQUIRE(naptr->regexp != NULL || naptr->regexp_len == 0);
UNUSED(type);
UNUSED(rdclass);
RETERR(uint16_tobuffer(naptr->order, target));
RETERR(uint16_tobuffer(naptr->preference, target));
RETERR(uint8_tobuffer(naptr->flags_len, target));
RETERR(mem_tobuffer(target, naptr->flags, naptr->flags_len));
RETERR(uint8_tobuffer(naptr->service_len, target));
RETERR(mem_tobuffer(target, naptr->service, naptr->service_len));
RETERR(uint8_tobuffer(naptr->regexp_len, target));
RETERR(mem_tobuffer(target, naptr->regexp, naptr->regexp_len));
dns_name_toregion(&naptr->replacement, &region);
return (isc_buffer_copyregion(target, &region));
}
static isc_result_t
tostruct_naptr(ARGS_TOSTRUCT) {
dns_rdata_naptr_t *naptr = target;
isc_region_t r;
dns_name_t name;
REQUIRE(rdata->type == dns_rdatatype_naptr);
REQUIRE(naptr != NULL);
REQUIRE(rdata->length != 0);
naptr->common.rdclass = rdata->rdclass;
naptr->common.rdtype = rdata->type;
ISC_LINK_INIT(&naptr->common, link);
naptr->flags = NULL;
naptr->service = NULL;
naptr->regexp = NULL;
dns_rdata_toregion(rdata, &r);
naptr->order = uint16_fromregion(&r);
isc_region_consume(&r, 2);
naptr->preference = uint16_fromregion(&r);
isc_region_consume(&r, 2);
naptr->flags_len = uint8_fromregion(&r);
isc_region_consume(&r, 1);
INSIST(naptr->flags_len <= r.length);
naptr->flags = mem_maybedup(mctx, r.base, naptr->flags_len);
isc_region_consume(&r, naptr->flags_len);
naptr->service_len = uint8_fromregion(&r);
isc_region_consume(&r, 1);
INSIST(naptr->service_len <= r.length);
naptr->service = mem_maybedup(mctx, r.base, naptr->service_len);
isc_region_consume(&r, naptr->service_len);
naptr->regexp_len = uint8_fromregion(&r);
isc_region_consume(&r, 1);
INSIST(naptr->regexp_len <= r.length);
naptr->regexp = mem_maybedup(mctx, r.base, naptr->regexp_len);
isc_region_consume(&r, naptr->regexp_len);
dns_name_init(&name, NULL);
dns_name_fromregion(&name, &r);
dns_name_init(&naptr->replacement, NULL);
name_duporclone(&name, mctx, &naptr->replacement);
naptr->mctx = mctx;
return (ISC_R_SUCCESS);
}
static void
freestruct_naptr(ARGS_FREESTRUCT) {
dns_rdata_naptr_t *naptr = source;
REQUIRE(naptr != NULL);
REQUIRE(naptr->common.rdtype == dns_rdatatype_naptr);
if (naptr->mctx == NULL) {
return;
}
if (naptr->flags != NULL) {
isc_mem_free(naptr->mctx, naptr->flags);
}
if (naptr->service != NULL) {
isc_mem_free(naptr->mctx, naptr->service);
}
if (naptr->regexp != NULL) {
isc_mem_free(naptr->mctx, naptr->regexp);
}
dns_name_free(&naptr->replacement, naptr->mctx);
naptr->mctx = NULL;
}
static isc_result_t
additionaldata_naptr(ARGS_ADDLDATA) {
dns_name_t name;
dns_offsets_t offsets;
isc_region_t sr;
dns_rdatatype_t atype;
unsigned int i, flagslen;
char *cp;
REQUIRE(rdata->type == dns_rdatatype_naptr);
UNUSED(owner);
/*
* Order, preference.
*/
dns_rdata_toregion(rdata, &sr);
isc_region_consume(&sr, 4);
/*
* Flags.
*/
atype = 0;
flagslen = sr.base[0];
cp = (char *)&sr.base[1];
for (i = 0; i < flagslen; i++, cp++) {
if (*cp == 'S' || *cp == 's') {
atype = dns_rdatatype_srv;
break;
}
if (*cp == 'A' || *cp == 'a') {
atype = dns_rdatatype_a;
break;
}
}
isc_region_consume(&sr, flagslen + 1);
/*
* Service.
*/
isc_region_consume(&sr, sr.base[0] + 1);
/*
* Regexp.
*/
isc_region_consume(&sr, sr.base[0] + 1);
/*
* Replacement.
*/
dns_name_init(&name, offsets);
dns_name_fromregion(&name, &sr);
if (atype != 0) {
return ((add)(arg, &name, atype, NULL));
}
return (ISC_R_SUCCESS);
}
static isc_result_t
digest_naptr(ARGS_DIGEST) {
isc_region_t r1, r2;
unsigned int length, n;
isc_result_t result;
dns_name_t name;
REQUIRE(rdata->type == dns_rdatatype_naptr);
dns_rdata_toregion(rdata, &r1);
r2 = r1;
length = 0;
/*
* Order, preference.
*/
length += 4;
isc_region_consume(&r2, 4);
/*
* Flags.
*/
n = r2.base[0] + 1;
length += n;
isc_region_consume(&r2, n);
/*
* Service.
*/
n = r2.base[0] + 1;
length += n;
isc_region_consume(&r2, n);
/*
* Regexp.
*/
n = r2.base[0] + 1;
length += n;
isc_region_consume(&r2, n);
/*
* Digest the RR up to the replacement name.
*/
r1.length = length;
result = (digest)(arg, &r1);
if (result != ISC_R_SUCCESS) {
return (result);
}
/*
* Replacement.
*/
dns_name_init(&name, NULL);
dns_name_fromregion(&name, &r2);
return (dns_name_digest(&name, digest, arg));
}
static bool
checkowner_naptr(ARGS_CHECKOWNER) {
REQUIRE(type == dns_rdatatype_naptr);
UNUSED(name);
UNUSED(type);
UNUSED(rdclass);
UNUSED(wildcard);
return (true);
}
static bool
checknames_naptr(ARGS_CHECKNAMES) {
REQUIRE(rdata->type == dns_rdatatype_naptr);
UNUSED(rdata);
UNUSED(owner);
UNUSED(bad);
return (true);
}
static int
casecompare_naptr(ARGS_COMPARE) {
return (compare_naptr(rdata1, rdata2));
}
#endif /* RDATA_GENERIC_NAPTR_35_C */