1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/*
* answer.c -- manipulating query answers and encoding them.
*
* Copyright (c) 2001-2006, NLnet Labs. All rights reserved.
*
* See LICENSE for the license.
*
*/
#include <config.h>
#include <string.h>
#include "answer.h"
#include "packet.h"
#include "query.h"
void
answer_init(answer_type *answer)
{
answer->rrset_count = 0;
}
int
answer_add_rrset(answer_type *answer, rr_section_type section,
domain_type *domain, rrset_type *rrset)
{
size_t i;
assert(section >= ANSWER_SECTION && section < RR_SECTION_COUNT);
assert(domain);
assert(rrset);
/* Don't add an RRset multiple times. */
for (i = 0; i < answer->rrset_count; ++i) {
if (answer->rrsets[i] == rrset) {
if (section < answer->section[i]) {
answer->section[i] = section;
return 1;
} else {
return 0;
}
}
}
if (answer->rrset_count >= MAXRRSPP) {
/* XXX: Generate warning/error? */
return 0;
}
answer->section[answer->rrset_count] = section;
answer->domains[answer->rrset_count] = domain;
answer->rrsets[answer->rrset_count] = rrset;
++answer->rrset_count;
return 1;
}
void
encode_answer(query_type *q, const answer_type *answer)
{
uint16_t counts[RR_SECTION_COUNT];
rr_section_type section;
size_t i;
for (section = ANSWER_SECTION; section < RR_SECTION_COUNT; ++section) {
counts[section] = 0;
}
for (section = ANSWER_SECTION;
!TC(q->packet) && section < RR_SECTION_COUNT;
++section)
{
for (i = 0; !TC(q->packet) && i < answer->rrset_count; ++i) {
if (answer->section[i] == section) {
int truncate
= (section == ANSWER_SECTION
|| section == AUTHORITY_SECTION);
counts[section] += packet_encode_rrset(
q,
answer->domains[i],
answer->rrsets[i],
truncate);
}
}
}
ANCOUNT_SET(q->packet, counts[ANSWER_SECTION]);
NSCOUNT_SET(q->packet, counts[AUTHORITY_SECTION]);
ARCOUNT_SET(q->packet,
counts[ADDITIONAL_A_SECTION]
+ counts[ADDITIONAL_AAAA_SECTION]
+ counts[ADDITIONAL_OTHER_SECTION]);
}
|
