_public_
uint16_t knot_pkt_type(const knot_pkt_t *pkt)
{
if (pkt == NULL) {
return 0;
}
bool is_query = (knot_wire_get_qr(pkt->wire) == 0);
uint16_t ret = KNOT_QUERY_INVALID;
uint8_t opcode = knot_wire_get_opcode(pkt->wire);
uint8_t query_type = knot_pkt_qtype(pkt);
switch (opcode) {
case KNOT_OPCODE_QUERY:
switch (query_type) {
case 0 /* RESERVED */: /* INVALID */ break;
case KNOT_RRTYPE_AXFR: ret = KNOT_QUERY_AXFR; break;
case KNOT_RRTYPE_IXFR: ret = KNOT_QUERY_IXFR; break;
default: ret = KNOT_QUERY_NORMAL; break;
}
break;
case KNOT_OPCODE_NOTIFY: ret = KNOT_QUERY_NOTIFY; break;
case KNOT_OPCODE_UPDATE: ret = KNOT_QUERY_UPDATE; break;
default: break;
}
if (!is_query) {
ret = ret|KNOT_RESPONSE;
}
return ret;
}
/* Basic response check (4 TAP tests). */
static void answer_sanity_check(const uint8_t *query,
const uint8_t *answer, uint16_t answer_len,
uint8_t expected_rcode, const char *name)
{
ok(answer_len >= KNOT_WIRE_HEADER_SIZE, "ns: len(%s answer) >= DNS header", name);
if (answer_len >= KNOT_WIRE_HEADER_SIZE) {
ok(knot_wire_get_qr(answer), "ns: %s answer has QR=1", name);
is_int(expected_rcode, knot_wire_get_rcode(answer), "ns: %s answer RCODE=%d", name, expected_rcode);
is_int(knot_wire_get_id(query), knot_wire_get_id(answer), "ns: %s MSGID match", name);
} else {
skip_block(3, "ns: can't check DNS header");
}
}
static void print_header(const knot_pkt_t *packet, const style_t *style,
const uint16_t ext_rcode)
{
char flags[64] = "";
uint8_t opcode_id;
const char *rcode_str = "Unknown";
const char *opcode_str = "Unknown";
lookup_table_t *rcode, *opcode;
// Get RCODE from Header and check for Extended RCODE from OPT RR.
rcode = lookup_by_id(knot_rcode_names, ext_rcode);
if (rcode != NULL) {
rcode_str = rcode->name;
}
// Get OPCODE.
opcode_id = knot_wire_get_opcode(packet->wire);
opcode = lookup_by_id(knot_opcode_names, opcode_id);
if (opcode != NULL) {
opcode_str = opcode->name;
}
// Get flags.
size_t flags_rest = sizeof(flags);
const size_t flag_len = 4;
if (knot_wire_get_qr(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " qr", flags_rest);
}
if (knot_wire_get_aa(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " aa", flags_rest);
}
if (knot_wire_get_tc(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " tc", flags_rest);
}
if (knot_wire_get_rd(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " rd", flags_rest);
}
if (knot_wire_get_ra(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " ra", flags_rest);
}
if (knot_wire_get_z(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " z", flags_rest);
}
if (knot_wire_get_ad(packet->wire) != 0 && flags_rest > flag_len) {
flags_rest -= strlcat(flags, " ad", flags_rest);
}
if (knot_wire_get_cd(packet->wire) != 0 && flags_rest > flag_len) {
strlcat(flags, " cd", flags_rest);
}
uint16_t id = knot_wire_get_id(packet->wire);
uint16_t qdcount = knot_wire_get_qdcount(packet->wire);
uint16_t ancount = knot_wire_get_ancount(packet->wire);
uint16_t nscount = knot_wire_get_nscount(packet->wire);
uint16_t arcount = knot_wire_get_arcount(packet->wire);
if (knot_pkt_has_tsig(packet)) {
arcount++;
}
// Print formatted info.
switch (style->format) {
case FORMAT_NSUPDATE:
printf(";; ->>HEADER<<- opcode: %s; status: %s; id: %u\n"
";; Flags:%1s; "
"ZONE: %u; PREREQ: %u; UPDATE: %u; ADDITIONAL: %u\n",
opcode_str, rcode_str, id, flags, qdcount, ancount,
nscount, arcount);
break;
default:
printf(";; ->>HEADER<<- opcode: %s; status: %s; id: %u\n"
";; Flags:%1s; "
"QUERY: %u; ANSWER: %u; AUTHORITY: %u; ADDITIONAL: %u\n",
opcode_str, rcode_str, id, flags, qdcount, ancount,
nscount, arcount);
break;
}
}
