Coverage Report

Created: 2025-08-26 06:35

/src/unbound/util/data/msgparse.c
Line
Count
Source (jump to first uncovered line)
1
/* 
2
 * util/data/msgparse.c - parse wireformat DNS messages.
3
 * 
4
 * Copyright (c) 2007, NLnet Labs. All rights reserved.
5
 * 
6
 * This software is open source.
7
 * 
8
 * Redistribution and use in source and binary forms, with or without
9
 * modification, are permitted provided that the following conditions
10
 * are met:
11
 * 
12
 * Redistributions of source code must retain the above copyright notice,
13
 * this list of conditions and the following disclaimer.
14
 * 
15
 * Redistributions in binary form must reproduce the above copyright notice,
16
 * this list of conditions and the following disclaimer in the documentation
17
 * and/or other materials provided with the distribution.
18
 * 
19
 * Neither the name of the NLNET LABS nor the names of its contributors may
20
 * be used to endorse or promote products derived from this software without
21
 * specific prior written permission.
22
 * 
23
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27
 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34
 */
35
/**
36
 * \file
37
 * Routines for message parsing a packet buffer to a descriptive structure.
38
 */
39
#include "config.h"
40
#include "util/config_file.h"
41
#include "util/data/msgparse.h"
42
#include "util/data/msgreply.h"
43
#include "util/data/dname.h"
44
#include "util/data/packed_rrset.h"
45
#include "util/netevent.h"
46
#include "util/storage/lookup3.h"
47
#include "util/regional.h"
48
#include "util/rfc_1982.h"
49
#include "util/edns.h"
50
#include "util/net_help.h"
51
#include "sldns/rrdef.h"
52
#include "sldns/sbuffer.h"
53
#include "sldns/parseutil.h"
54
#include "sldns/wire2str.h"
55
56
/** smart comparison of (compressed, valid) dnames from packet */
57
static int
58
smart_compare(sldns_buffer* pkt, uint8_t* dnow, 
59
  uint8_t* dprfirst, uint8_t* dprlast)
60
1.18M
{
61
1.18M
  if(LABEL_IS_PTR(*dnow)) {
62
    /* ptr points to a previous dname */
63
32.0k
    uint8_t* p;
64
32.0k
    if((size_t)PTR_OFFSET(dnow[0], dnow[1])
65
32.0k
      >= sldns_buffer_limit(pkt))
66
0
      return -1;
67
32.0k
    p = sldns_buffer_at(pkt, PTR_OFFSET(dnow[0], dnow[1]));
68
32.0k
    if( p == dprfirst || p == dprlast )
69
2.58k
      return 0;
70
    /* prev dname is also a ptr, both ptrs are the same. */
71
29.4k
    if(LABEL_IS_PTR(*dprlast) &&
72
29.4k
      dprlast[0] == dnow[0] && dprlast[1] == dnow[1])
73
11.6k
      return 0;
74
29.4k
  }
75
1.16M
  return dname_pkt_compare(pkt, dnow, dprlast);
76
1.18M
}
77
78
/**
79
 * Allocate new rrset in region, fill with data.
80
 */
81
static struct rrset_parse* 
82
new_rrset(struct msg_parse* msg, uint8_t* dname, size_t dnamelen, 
83
  uint16_t type, uint16_t dclass, hashvalue_type hash, 
84
  uint32_t rrset_flags, sldns_pkt_section section, 
85
  struct regional* region)
86
172k
{
87
172k
  struct rrset_parse* p = regional_alloc(region, sizeof(*p));
88
172k
  if(!p) return NULL;
89
172k
  p->rrset_bucket_next = msg->hashtable[hash & (PARSE_TABLE_SIZE-1)];
90
172k
  msg->hashtable[hash & (PARSE_TABLE_SIZE-1)] = p;
91
172k
  p->rrset_all_next = 0;
92
172k
  if(msg->rrset_last)
93
161k
    msg->rrset_last->rrset_all_next = p;
94
10.5k
  else  msg->rrset_first = p;
95
172k
  msg->rrset_last = p;
96
172k
  p->hash = hash;
97
172k
  p->section = section;
98
172k
  p->dname = dname;
99
172k
  p->dname_len = dnamelen;
100
172k
  p->type = type;
101
172k
  p->rrset_class = dclass;
102
172k
  p->flags = rrset_flags;
103
172k
  p->rr_count = 0;
104
172k
  p->size = 0;
105
172k
  p->rr_first = 0;
106
172k
  p->rr_last = 0;
107
172k
  p->rrsig_count = 0;
108
172k
  p->rrsig_first = 0;
109
172k
  p->rrsig_last = 0;
110
172k
  return p;
111
172k
}
112
113
/** See if next rrset is nsec at zone apex */
114
static int
115
nsec_at_apex(sldns_buffer* pkt)
116
57.3k
{
117
  /* we are at ttl position in packet. */
118
57.3k
  size_t pos = sldns_buffer_position(pkt);
119
57.3k
  uint16_t rdatalen;
120
57.3k
  if(sldns_buffer_remaining(pkt) < 7) /* ttl+len+root */
121
187
    return 0; /* eek! */
122
57.2k
  sldns_buffer_skip(pkt, 4); /* ttl */;
123
57.2k
  rdatalen = sldns_buffer_read_u16(pkt);
124
57.2k
  if(sldns_buffer_remaining(pkt) < rdatalen) {
125
133
    sldns_buffer_set_position(pkt, pos);
126
133
    return 0; /* parse error happens later */
127
133
  }
128
  /* must validate the nsec next domain name format */
129
57.0k
  if(pkt_dname_len(pkt) == 0) {
130
5.37k
    sldns_buffer_set_position(pkt, pos);
131
5.37k
    return 0; /* parse error */
132
5.37k
  }
133
134
  /* see if SOA bit is set. */
135
51.6k
  if(sldns_buffer_position(pkt) < pos+4+rdatalen) {
136
    /* nsec type bitmap contains items */
137
22.9k
    uint8_t win, blen, bits;
138
    /* need: windownum, bitmap len, firstbyte */
139
22.9k
    if(sldns_buffer_position(pkt)+3 > pos+4+rdatalen) {
140
2.00k
      sldns_buffer_set_position(pkt, pos);
141
2.00k
      return 0; /* malformed nsec */
142
2.00k
    }
143
20.9k
    win = sldns_buffer_read_u8(pkt);
144
20.9k
    blen = sldns_buffer_read_u8(pkt);
145
20.9k
    bits = sldns_buffer_read_u8(pkt);
146
    /* 0window always first window. bitlen >=1 or parse
147
       error really. bit 0x2 is SOA. */
148
20.9k
    if(win == 0 && blen >= 1 && (bits & 0x02)) {
149
2.59k
      sldns_buffer_set_position(pkt, pos);
150
2.59k
      return 1;
151
2.59k
    }
152
20.9k
  }
153
154
47.1k
  sldns_buffer_set_position(pkt, pos);
155
47.1k
  return 0;
156
51.6k
}
157
158
/** Calculate rrset flags */
159
static uint32_t
160
pkt_rrset_flags(sldns_buffer* pkt, uint16_t type, sldns_pkt_section sec)
161
646k
{
162
646k
  uint32_t f = 0;
163
646k
  if(type == LDNS_RR_TYPE_NSEC && nsec_at_apex(pkt)) {
164
2.59k
    f |= PACKED_RRSET_NSEC_AT_APEX;
165
643k
  } else if(type == LDNS_RR_TYPE_SOA && sec == LDNS_SECTION_AUTHORITY) {
166
1.90k
    f |= PACKED_RRSET_SOA_NEG;
167
1.90k
  }
168
646k
  return f;
169
646k
}
170
171
hashvalue_type
172
pkt_hash_rrset(sldns_buffer* pkt, uint8_t* dname, uint16_t type, 
173
  uint16_t dclass, uint32_t rrset_flags)
174
28.3k
{
175
  /* note this MUST be identical to rrset_key_hash in packed_rrset.c */
176
  /* this routine handles compressed names */
177
28.3k
  hashvalue_type h = 0xab;
178
28.3k
  h = dname_pkt_hash(pkt, dname, h);
179
28.3k
  h = hashlittle(&type, sizeof(type), h);   /* host order */
180
28.3k
  h = hashlittle(&dclass, sizeof(dclass), h); /* netw order */
181
28.3k
  h = hashlittle(&rrset_flags, sizeof(uint32_t), h);
182
28.3k
  return h;
183
28.3k
}
184
185
/** create partial dname hash for rrset hash */
186
static hashvalue_type
187
pkt_hash_rrset_first(sldns_buffer* pkt, uint8_t* dname)
188
1.79M
{
189
  /* works together with pkt_hash_rrset_rest */
190
  /* note this MUST be identical to rrset_key_hash in packed_rrset.c */
191
  /* this routine handles compressed names */
192
1.79M
  hashvalue_type h = 0xab;
193
1.79M
  h = dname_pkt_hash(pkt, dname, h);
194
1.79M
  return h;
195
1.79M
}
196
197
/** create a rrset hash from a partial dname hash */
198
static hashvalue_type
199
pkt_hash_rrset_rest(hashvalue_type dname_h, uint16_t type, uint16_t dclass, 
200
  uint32_t rrset_flags)
201
1.23M
{
202
  /* works together with pkt_hash_rrset_first */
203
  /* note this MUST be identical to rrset_key_hash in packed_rrset.c */
204
1.23M
  hashvalue_type h;
205
1.23M
  h = hashlittle(&type, sizeof(type), dname_h); /* host order */
206
1.23M
  h = hashlittle(&dclass, sizeof(dclass), h); /* netw order */
207
1.23M
  h = hashlittle(&rrset_flags, sizeof(uint32_t), h);
208
1.23M
  return h;
209
1.23M
}
210
211
/** compare rrset_parse with data */
212
static int
213
rrset_parse_equals(struct rrset_parse* p, sldns_buffer* pkt, hashvalue_type h, 
214
  uint32_t rrset_flags, uint8_t* dname, size_t dnamelen, 
215
  uint16_t type, uint16_t dclass)
216
7.37M
{
217
7.37M
  if(p->hash == h && p->dname_len == dnamelen && p->type == type &&
218
7.37M
    p->rrset_class == dclass && p->flags == rrset_flags &&
219
7.37M
    dname_pkt_compare(pkt, dname, p->dname) == 0)
220
628k
    return 1;
221
6.74M
  return 0;
222
7.37M
}
223
224
225
struct rrset_parse*
226
msgparse_hashtable_lookup(struct msg_parse* msg, sldns_buffer* pkt, 
227
  hashvalue_type h, uint32_t rrset_flags, uint8_t* dname,
228
  size_t dnamelen, uint16_t type, uint16_t dclass)
229
1.24M
{
230
1.24M
  struct rrset_parse* p = msg->hashtable[h & (PARSE_TABLE_SIZE-1)];
231
7.98M
  while(p) {
232
7.37M
    if(rrset_parse_equals(p, pkt, h, rrset_flags, dname, dnamelen,
233
7.37M
      type, dclass))
234
628k
      return p;
235
6.74M
    p = p->rrset_bucket_next;
236
6.74M
  }
237
614k
  return NULL;
238
1.24M
}
239
240
/** return type networkformat that rrsig in packet covers */
241
static int
242
pkt_rrsig_covered(sldns_buffer* pkt, uint8_t* here, uint16_t* type)
243
3.04M
{
244
3.04M
  size_t pos = sldns_buffer_position(pkt);
245
3.04M
  sldns_buffer_set_position(pkt, (size_t)(here-sldns_buffer_begin(pkt)));
246
  /* ttl + len + size of small rrsig(rootlabel, no signature) */
247
3.04M
  if(sldns_buffer_remaining(pkt) < 4+2+19)
248
4.67k
    return 0;
249
3.03M
  sldns_buffer_skip(pkt, 4); /* ttl */
250
3.03M
  if(sldns_buffer_read_u16(pkt) < 19) /* too short */ {
251
1.81M
    sldns_buffer_set_position(pkt, pos);
252
1.81M
    return 0;
253
1.81M
  }
254
1.22M
  *type = sldns_buffer_read_u16(pkt);
255
1.22M
  sldns_buffer_set_position(pkt, pos);
256
1.22M
  return 1;
257
3.03M
}
258
259
/** true if covered type equals prevtype */
260
static int
261
pkt_rrsig_covered_equals(sldns_buffer* pkt, uint8_t* here, uint16_t type)
262
2.96M
{
263
2.96M
  uint16_t t;
264
2.96M
  if(pkt_rrsig_covered(pkt, here, &t) && t == type)
265
539k
    return 1;
266
2.42M
  return 0;
267
2.96M
}
268
269
void
270
msgparse_bucket_remove(struct msg_parse* msg, struct rrset_parse* rrset)
271
73.8k
{
272
73.8k
  struct rrset_parse** p;
273
73.8k
  p = &msg->hashtable[ rrset->hash & (PARSE_TABLE_SIZE-1) ];
274
441k
  while(*p) {
275
441k
    if(*p == rrset) {
276
73.8k
      *p = rrset->rrset_bucket_next;
277
73.8k
      return;
278
73.8k
    }
279
367k
    p = &( (*p)->rrset_bucket_next );
280
367k
  }
281
73.8k
}
282
283
/** change section of rrset from previous to current section */
284
static void
285
change_section(struct msg_parse* msg, struct rrset_parse* rrset,
286
  sldns_pkt_section section)
287
1.37k
{
288
1.37k
  struct rrset_parse *p, *prev;
289
  /* remove from list */
290
1.37k
  if(section == rrset->section)
291
1.09k
    return;
292
282
  p = msg->rrset_first;
293
282
  prev = 0;
294
1.42k
  while(p) {
295
1.42k
    if(p == rrset) {
296
282
      if(prev) prev->rrset_all_next = p->rrset_all_next;
297
131
      else  msg->rrset_first = p->rrset_all_next;
298
282
      if(msg->rrset_last == rrset)
299
68
        msg->rrset_last = prev;
300
282
      break;
301
282
    }
302
1.14k
    prev = p;
303
1.14k
    p = p->rrset_all_next;
304
1.14k
  }
305
  /* remove from count */
306
282
  switch(rrset->section) {
307
174
    case LDNS_SECTION_ANSWER: msg->an_rrsets--; break;
308
108
    case LDNS_SECTION_AUTHORITY: msg->ns_rrsets--; break;
309
0
    case LDNS_SECTION_ADDITIONAL: msg->ar_rrsets--; break;
310
0
    default: log_assert(0);
311
282
  }
312
  /* insert at end of list */
313
282
  rrset->rrset_all_next = 0;
314
282
  if(msg->rrset_last)
315
242
    msg->rrset_last->rrset_all_next = rrset;
316
40
  else  msg->rrset_first = rrset;
317
282
  msg->rrset_last = rrset;
318
  /* up count of new section */
319
282
  switch(section) {
320
156
    case LDNS_SECTION_AUTHORITY: msg->ns_rrsets++; break;
321
126
    case LDNS_SECTION_ADDITIONAL: msg->ar_rrsets++; break;
322
0
    default: log_assert(0);
323
282
  }
324
282
  rrset->section = section;
325
282
}
326
327
/** see if rrset of type RRSIG contains sig over given type */
328
static int
329
rrset_has_sigover(sldns_buffer* pkt, struct rrset_parse* rrset, uint16_t type,
330
  int* hasother)
331
152k
{
332
152k
  int res = 0;
333
152k
  struct rr_parse* rr = rrset->rr_first;
334
152k
  log_assert( rrset->type == LDNS_RR_TYPE_RRSIG );
335
2.57M
  while(rr) {
336
2.42M
    if(pkt_rrsig_covered_equals(pkt, rr->ttl_data, type))
337
258k
      res = 1;
338
2.16M
    else  *hasother = 1;
339
2.42M
    rr = rr->next;
340
2.42M
  }
341
152k
  return res;
342
152k
}
343
344
/** move rrsigs from sigset to dataset */
345
static int
346
moveover_rrsigs(sldns_buffer* pkt, struct regional* region, 
347
  struct rrset_parse* sigset, struct rrset_parse* dataset, int duplicate)
348
18.1k
{
349
18.1k
  struct rr_parse* sig = sigset->rr_first;
350
18.1k
  struct rr_parse* prev = NULL;
351
18.1k
  struct rr_parse* insert;
352
18.1k
  struct rr_parse* nextsig;
353
487k
  while(sig) {
354
468k
    nextsig = sig->next;
355
468k
    if(pkt_rrsig_covered_equals(pkt, sig->ttl_data, 
356
468k
      dataset->type)) {
357
256k
      if(duplicate) {
358
        /* new */
359
252k
        insert = (struct rr_parse*)regional_alloc(
360
252k
          region, sizeof(struct rr_parse));
361
252k
        if(!insert) return 0;
362
252k
        insert->outside_packet = 0;
363
252k
        insert->ttl_data = sig->ttl_data;
364
252k
        insert->size = sig->size;
365
        /* prev not used */
366
252k
      } else {
367
        /* remove from sigset */
368
3.57k
        if(prev) prev->next = sig->next;
369
1.01k
        else  sigset->rr_first = sig->next;
370
3.57k
        if(sigset->rr_last == sig)
371
365
          sigset->rr_last = prev;
372
3.57k
        sigset->rr_count--;
373
3.57k
        sigset->size -= sig->size;
374
3.57k
        insert = sig;
375
        /* prev not changed */
376
3.57k
      }
377
      /* add to dataset */
378
256k
      dataset->rrsig_count++;
379
256k
      insert->next = 0;
380
256k
      if(dataset->rrsig_last) 
381
237k
        dataset->rrsig_last->next = insert;
382
18.1k
      else  dataset->rrsig_first = insert;
383
256k
      dataset->rrsig_last = insert;
384
256k
      dataset->size += insert->size;
385
256k
    } else  {
386
212k
      prev = sig;
387
212k
    }
388
468k
    sig = nextsig;
389
468k
  }
390
18.1k
  return 1;
391
18.1k
}
392
393
/** change an rrsig rrset for use as data rrset */
394
static struct rrset_parse*
395
change_rrsig_rrset(struct rrset_parse* sigset, struct msg_parse* msg, 
396
  sldns_buffer* pkt, uint16_t datatype, uint32_t rrset_flags,
397
  int hasother, sldns_pkt_section section, struct regional* region)
398
19.5k
{
399
19.5k
  struct rrset_parse* dataset = sigset;
400
19.5k
  hashvalue_type hash = pkt_hash_rrset(pkt, sigset->dname, datatype, 
401
19.5k
    sigset->rrset_class, rrset_flags);
402
19.5k
  log_assert( sigset->type == LDNS_RR_TYPE_RRSIG );
403
19.5k
  log_assert( datatype != LDNS_RR_TYPE_RRSIG );
404
19.5k
  if(hasother) {
405
    /* need to make new rrset to hold data type */
406
18.1k
    dataset = new_rrset(msg, sigset->dname, sigset->dname_len, 
407
18.1k
      datatype, sigset->rrset_class, hash, rrset_flags, 
408
18.1k
      section, region);
409
18.1k
    if(!dataset) 
410
0
      return NULL;
411
18.1k
    switch(section) {
412
5.56k
      case LDNS_SECTION_ANSWER: msg->an_rrsets++; break;
413
5.70k
      case LDNS_SECTION_AUTHORITY: msg->ns_rrsets++; break;
414
6.89k
      case LDNS_SECTION_ADDITIONAL: msg->ar_rrsets++; break;
415
0
      default: log_assert(0);
416
18.1k
    }
417
18.1k
    if(!moveover_rrsigs(pkt, region, sigset, dataset, 
418
18.1k
      msg->qtype == LDNS_RR_TYPE_RRSIG ||
419
18.1k
      (msg->qtype == LDNS_RR_TYPE_ANY &&
420
4.00k
      section != LDNS_SECTION_ANSWER) ))
421
0
      return NULL;
422
18.1k
    return dataset;
423
18.1k
  }
424
  /* changeover the type of the rrset to data set */
425
1.37k
  msgparse_bucket_remove(msg, dataset);
426
  /* insert into new hash bucket */
427
1.37k
  dataset->rrset_bucket_next = msg->hashtable[hash&(PARSE_TABLE_SIZE-1)];
428
1.37k
  msg->hashtable[hash&(PARSE_TABLE_SIZE-1)] = dataset;
429
1.37k
  dataset->hash = hash;
430
  /* use section of data item for result */
431
1.37k
  change_section(msg, dataset, section);
432
1.37k
  dataset->type = datatype;
433
1.37k
  dataset->flags = rrset_flags;
434
1.37k
  dataset->rrsig_count += dataset->rr_count;
435
1.37k
  dataset->rr_count = 0;
436
  /* move sigs to end of siglist */
437
1.37k
  if(dataset->rrsig_last)
438
0
    dataset->rrsig_last->next = dataset->rr_first;
439
1.37k
  else  dataset->rrsig_first = dataset->rr_first;
440
1.37k
  dataset->rrsig_last = dataset->rr_last;
441
1.37k
  dataset->rr_first = 0;
442
1.37k
  dataset->rr_last = 0;
443
1.37k
  return dataset;
444
19.5k
}
445
446
/** Find rrset. If equal to previous it is fast. hash if not so.
447
 * @param msg: the message with hash table.
448
 * @param pkt: the packet in wireformat (needed for compression ptrs).
449
 * @param dname: pointer to start of dname (compressed) in packet.
450
 * @param dnamelen: uncompressed wirefmt length of dname.
451
 * @param type: type of current rr.
452
 * @param dclass: class of current rr.
453
 * @param hash: hash value is returned if the rrset could not be found.
454
 * @param rrset_flags: is returned if the rrset could not be found.
455
 * @param prev_dname_first: dname of last seen RR. First seen dname.
456
 * @param prev_dname_last: dname of last seen RR. Last seen dname.
457
 * @param prev_dnamelen: dname len of last seen RR.
458
 * @param prev_type: type of last seen RR.
459
 * @param prev_dclass: class of last seen RR.
460
 * @param rrset_prev: last seen RRset.
461
 * @param section: the current section in the packet.
462
 * @param region: used to allocate temporary parsing data.
463
 * @return 0 on out of memory.
464
 */
465
static int
466
find_rrset(struct msg_parse* msg, sldns_buffer* pkt, uint8_t* dname, 
467
  size_t dnamelen, uint16_t type, uint16_t dclass, hashvalue_type* hash, 
468
  uint32_t* rrset_flags,
469
  uint8_t** prev_dname_first, uint8_t** prev_dname_last,
470
  size_t* prev_dnamelen, uint16_t* prev_type,
471
  uint16_t* prev_dclass, struct rrset_parse** rrset_prev,
472
  sldns_pkt_section section, struct regional* region)
473
1.79M
{
474
1.79M
  hashvalue_type dname_h = pkt_hash_rrset_first(pkt, dname);
475
1.79M
  uint16_t covtype;
476
1.79M
  if(*rrset_prev) {
477
    /* check if equal to previous item */
478
1.77M
    if(type == *prev_type && dclass == *prev_dclass &&
479
1.77M
      dnamelen == *prev_dnamelen &&
480
1.77M
      smart_compare(pkt, dname, *prev_dname_first, 
481
1.15M
        *prev_dname_last) == 0 &&
482
1.77M
      type != LDNS_RR_TYPE_RRSIG) {
483
      /* same as previous */
484
1.12M
      *prev_dname_last = dname;
485
1.12M
      return 1;
486
1.12M
    }
487
    /* check if rrsig over previous item */
488
653k
    if(type == LDNS_RR_TYPE_RRSIG && dclass == *prev_dclass &&
489
653k
      pkt_rrsig_covered_equals(pkt, sldns_buffer_current(pkt),
490
71.8k
        *prev_type) &&
491
653k
      smart_compare(pkt, dname, *prev_dname_first,
492
25.6k
        *prev_dname_last) == 0) {
493
      /* covers previous */
494
20.9k
      *prev_dname_last = dname;
495
20.9k
      return 1;
496
20.9k
    }
497
653k
  }
498
  /* find by hashing and lookup in hashtable */
499
646k
  *rrset_flags = pkt_rrset_flags(pkt, type, section);
500
  
501
  /* if rrsig - try to lookup matching data set first */
502
646k
  if(type == LDNS_RR_TYPE_RRSIG && pkt_rrsig_covered(pkt, 
503
74.3k
    sldns_buffer_current(pkt), &covtype)) {
504
41.4k
    *hash = pkt_hash_rrset_rest(dname_h, covtype, dclass, 
505
41.4k
      *rrset_flags);
506
41.4k
    *rrset_prev = msgparse_hashtable_lookup(msg, pkt, *hash, 
507
41.4k
      *rrset_flags, dname, dnamelen, covtype, dclass);
508
41.4k
    if(!*rrset_prev && covtype == LDNS_RR_TYPE_NSEC) {
509
      /* if NSEC try with NSEC apex bit twiddled */
510
4.53k
      *rrset_flags ^= PACKED_RRSET_NSEC_AT_APEX;
511
4.53k
      *hash = pkt_hash_rrset_rest(dname_h, covtype, dclass, 
512
4.53k
        *rrset_flags);
513
4.53k
      *rrset_prev = msgparse_hashtable_lookup(msg, pkt, 
514
4.53k
        *hash, *rrset_flags, dname, dnamelen, covtype, 
515
4.53k
        dclass);
516
4.53k
      if(!*rrset_prev) /* untwiddle if not found */
517
3.56k
        *rrset_flags ^= PACKED_RRSET_NSEC_AT_APEX;
518
4.53k
    }
519
41.4k
    if(!*rrset_prev && covtype == LDNS_RR_TYPE_SOA) {
520
      /* if SOA try with SOA neg flag twiddled */
521
5.92k
      *rrset_flags ^= PACKED_RRSET_SOA_NEG;
522
5.92k
      *hash = pkt_hash_rrset_rest(dname_h, covtype, dclass, 
523
5.92k
        *rrset_flags);
524
5.92k
      *rrset_prev = msgparse_hashtable_lookup(msg, pkt, 
525
5.92k
        *hash, *rrset_flags, dname, dnamelen, covtype, 
526
5.92k
        dclass);
527
5.92k
      if(!*rrset_prev) /* untwiddle if not found */
528
4.75k
        *rrset_flags ^= PACKED_RRSET_SOA_NEG;
529
5.92k
    }
530
41.4k
    if(*rrset_prev) {
531
16.8k
      *prev_dname_first = (*rrset_prev)->dname;
532
16.8k
      *prev_dname_last = dname;
533
16.8k
      *prev_dnamelen = dnamelen;
534
16.8k
      *prev_type = covtype;
535
16.8k
      *prev_dclass = dclass;
536
16.8k
      return 1;
537
16.8k
    }
538
41.4k
  }
539
629k
  if(type != LDNS_RR_TYPE_RRSIG) {
540
571k
    int hasother = 0;
541
    /* find matching rrsig */
542
571k
    *hash = pkt_hash_rrset_rest(dname_h, LDNS_RR_TYPE_RRSIG, 
543
571k
      dclass, 0);
544
571k
    *rrset_prev = msgparse_hashtable_lookup(msg, pkt, *hash, 
545
571k
      0, dname, dnamelen, LDNS_RR_TYPE_RRSIG, 
546
571k
      dclass);
547
571k
    if(*rrset_prev && rrset_has_sigover(pkt, *rrset_prev, type,
548
152k
      &hasother)) {
549
      /* yes! */
550
19.5k
      *prev_dname_first = (*rrset_prev)->dname;
551
19.5k
      *prev_dname_last = dname;
552
19.5k
      *prev_dnamelen = dnamelen;
553
19.5k
      *prev_type = type;
554
19.5k
      *prev_dclass = dclass;
555
19.5k
      *rrset_prev = change_rrsig_rrset(*rrset_prev, msg, 
556
19.5k
        pkt, type, *rrset_flags, hasother, section, 
557
19.5k
        region);
558
19.5k
      if(!*rrset_prev) return 0;
559
19.5k
      return 1;
560
19.5k
    }
561
571k
  }
562
563
609k
  *hash = pkt_hash_rrset_rest(dname_h, type, dclass, *rrset_flags);
564
609k
  *rrset_prev = msgparse_hashtable_lookup(msg, pkt, *hash, *rrset_flags, 
565
609k
    dname, dnamelen, type, dclass);
566
609k
  if(*rrset_prev)
567
455k
    *prev_dname_first = (*rrset_prev)->dname;
568
153k
  else  *prev_dname_first = dname;
569
609k
  *prev_dname_last = dname;
570
609k
  *prev_dnamelen = dnamelen;
571
609k
  *prev_type = type;
572
609k
  *prev_dclass = dclass;
573
609k
  return 1;
574
629k
}
575
576
/**
577
 * Parse query section. 
578
 * @param pkt: packet, position at call must be at start of query section.
579
 *  at end position is after query section.
580
 * @param msg: store results here.
581
 * @return: 0 if OK, or rcode on error.
582
 */
583
static int
584
parse_query_section(sldns_buffer* pkt, struct msg_parse* msg)
585
11.1k
{
586
11.1k
  if(msg->qdcount == 0)
587
7.30k
    return 0;
588
3.86k
  if(msg->qdcount > 1)
589
0
    return LDNS_RCODE_FORMERR;
590
3.86k
  log_assert(msg->qdcount == 1);
591
3.86k
  if(sldns_buffer_remaining(pkt) <= 0)
592
3
    return LDNS_RCODE_FORMERR;
593
3.85k
  msg->qname = sldns_buffer_current(pkt);
594
3.85k
  if((msg->qname_len = pkt_dname_len(pkt)) == 0)
595
32
    return LDNS_RCODE_FORMERR;
596
3.82k
  if(sldns_buffer_remaining(pkt) < sizeof(uint16_t)*2)
597
50
    return LDNS_RCODE_FORMERR;
598
3.77k
  msg->qtype = sldns_buffer_read_u16(pkt);
599
3.77k
  msg->qclass = sldns_buffer_read_u16(pkt);
600
3.77k
  return 0;
601
3.82k
}
602
603
size_t
604
get_rdf_size(sldns_rdf_type rdf)
605
404k
{
606
404k
  switch(rdf) {
607
0
    case LDNS_RDF_TYPE_CLASS:
608
58.4k
    case LDNS_RDF_TYPE_ALG:
609
116k
    case LDNS_RDF_TYPE_INT8:
610
116k
      return 1;
611
0
      break;
612
59.2k
    case LDNS_RDF_TYPE_INT16:
613
118k
    case LDNS_RDF_TYPE_TYPE:
614
118k
    case LDNS_RDF_TYPE_CERT_ALG:
615
118k
      return 2;
616
0
      break;
617
57.0k
    case LDNS_RDF_TYPE_INT32:
618
170k
    case LDNS_RDF_TYPE_TIME:
619
170k
    case LDNS_RDF_TYPE_A:
620
170k
    case LDNS_RDF_TYPE_PERIOD:
621
170k
      return 4;
622
0
      break;
623
0
    case LDNS_RDF_TYPE_TSIGTIME:
624
0
      return 6;
625
0
      break;
626
0
    case LDNS_RDF_TYPE_AAAA:
627
0
      return 16;
628
0
      break;
629
0
    default:
630
0
      log_assert(0); /* add type above */
631
      /* only types that appear before a domain  *
632
       * name are needed. rest is simply copied. */
633
404k
  }
634
0
  return 0;
635
404k
}
636
637
/** calculate the size of one rr */
638
static int
639
calc_size(sldns_buffer* pkt, uint16_t type, struct rr_parse* rr)
640
1.36M
{
641
1.36M
  const sldns_rr_descriptor* desc;
642
1.36M
  uint16_t pkt_len; /* length of rr inside the packet */
643
1.36M
  rr->size = sizeof(uint16_t); /* the rdatalen */
644
1.36M
  sldns_buffer_skip(pkt, 4); /* skip ttl */
645
1.36M
  pkt_len = sldns_buffer_read_u16(pkt);
646
1.36M
  if(sldns_buffer_remaining(pkt) < pkt_len)
647
3.08k
    return 0;
648
1.36M
  desc = sldns_rr_descript(type);
649
1.36M
  if(pkt_len > 0 && desc && desc->_dname_count > 0) {
650
101k
    int count = (int)desc->_dname_count;
651
101k
    int rdf = 0;
652
101k
    size_t len;
653
101k
    size_t oldpos;
654
    /* skip first part. */
655
551k
    while(pkt_len > 0 && count) {
656
450k
      switch(desc->_wireformat[rdf]) {
657
97.3k
      case LDNS_RDF_TYPE_DNAME:
658
        /* decompress every domain name */
659
97.3k
        oldpos = sldns_buffer_position(pkt);
660
97.3k
        if((len = pkt_dname_len(pkt)) == 0)
661
482
          return 0; /* malformed dname */
662
96.8k
        if(sldns_buffer_position(pkt)-oldpos > pkt_len)
663
57
          return 0; /* dname exceeds rdata */
664
96.7k
        pkt_len -= sldns_buffer_position(pkt)-oldpos;
665
96.7k
        rr->size += len;
666
96.7k
        count--;
667
96.7k
        len = 0;
668
96.7k
        break;
669
1.86k
      case LDNS_RDF_TYPE_STR:
670
1.86k
        if(pkt_len < 1) {
671
          /* NOTREACHED, due to 'while(>0)' */
672
0
          return 0; /* len byte exceeds rdata */
673
0
        }
674
1.86k
        len = sldns_buffer_current(pkt)[0] + 1;
675
1.86k
        break;
676
351k
      default:
677
351k
        len = get_rdf_size(desc->_wireformat[rdf]);
678
450k
      }
679
450k
      if(len) {
680
353k
        if(pkt_len < len)
681
127
          return 0; /* exceeds rdata */
682
353k
        pkt_len -= len;
683
353k
        sldns_buffer_skip(pkt, (ssize_t)len);
684
353k
        rr->size += len;
685
353k
      }
686
450k
      rdf++;
687
450k
    }
688
101k
  }
689
  /* remaining rdata */
690
1.36M
  rr->size += pkt_len;
691
1.36M
  sldns_buffer_skip(pkt, (ssize_t)pkt_len);
692
1.36M
  return 1;
693
1.36M
}
694
695
/** skip rr ttl and rdata */
696
static int
697
skip_ttl_rdata(sldns_buffer* pkt) 
698
426k
{
699
426k
  uint16_t rdatalen;
700
426k
  if(sldns_buffer_remaining(pkt) < 6) /* ttl + rdatalen */
701
0
    return 0;
702
426k
  sldns_buffer_skip(pkt, 4); /* ttl */
703
426k
  rdatalen = sldns_buffer_read_u16(pkt);
704
426k
  if(sldns_buffer_remaining(pkt) < rdatalen)
705
333
    return 0;
706
426k
  sldns_buffer_skip(pkt, (ssize_t)rdatalen);
707
426k
  return 1;
708
426k
}
709
710
/** see if RRSIG is a duplicate of another */
711
static int
712
sig_is_double(sldns_buffer* pkt, struct rrset_parse* rrset, uint8_t* ttldata)
713
455k
{
714
455k
  uint16_t rlen, siglen;
715
455k
  size_t pos = sldns_buffer_position(pkt);
716
455k
  struct rr_parse* sig;
717
455k
  if(sldns_buffer_remaining(pkt) < 6) 
718
0
    return 0;
719
455k
  sldns_buffer_skip(pkt, 4); /* ttl */
720
455k
  rlen = sldns_buffer_read_u16(pkt);
721
455k
  if(sldns_buffer_remaining(pkt) < rlen) {
722
579
    sldns_buffer_set_position(pkt, pos);
723
579
    return 0;
724
579
  }
725
455k
  sldns_buffer_set_position(pkt, pos);
726
727
455k
  sig = rrset->rrsig_first;
728
1.77M
  while(sig) {
729
    /* check if rdatalen is same */
730
1.33M
    memmove(&siglen, sig->ttl_data+4, sizeof(siglen));
731
1.33M
    siglen = ntohs(siglen);
732
    /* checks if data in packet is exactly the same, this means
733
     * also dname in rdata is the same, but rrsig is not allowed
734
     * to have compressed dnames anyway. If it is compressed anyway
735
     * it will lead to duplicate rrs for qtype=RRSIG. (or ANY).
736
     *
737
     * Cannot use sig->size because size of the other one is not 
738
     * calculated yet.
739
     */
740
1.33M
    if(siglen == rlen) {
741
136k
      if(siglen>0 && memcmp(sig->ttl_data+6, ttldata+6, 
742
136k
        siglen) == 0) {
743
        /* same! */
744
13.6k
        return 1;
745
13.6k
      }
746
136k
    }
747
1.32M
    sig = sig->next;
748
1.32M
  }
749
441k
  return 0;
750
455k
}
751
752
/** Add rr (from packet here) to rrset, skips rr */
753
static int
754
add_rr_to_rrset(struct rrset_parse* rrset, sldns_buffer* pkt, 
755
  struct msg_parse* msg, struct regional* region, 
756
  sldns_pkt_section section, uint16_t type)
757
1.79M
{
758
1.79M
  struct rr_parse* rr;
759
  /* check section of rrset. */
760
1.79M
  if(rrset->section != section && type != LDNS_RR_TYPE_RRSIG &&
761
1.79M
    rrset->type != LDNS_RR_TYPE_RRSIG) {
762
    /* silently drop it - we drop the last part, since
763
     * trust in rr data depends on the section it is in. 
764
     * the less trustworthy part is discarded. 
765
     * also the last part is more likely to be incomplete.
766
     * RFC 2181: must put RRset only once in response. */
767
    /*
768
    verbose(VERB_QUERY, "Packet contains rrset data in "
769
      "multiple sections, dropped last part.");
770
    log_buf(VERB_QUERY, "packet was", pkt);
771
    */
772
    /* forwards */
773
413k
    if(!skip_ttl_rdata(pkt))
774
333
      return LDNS_RCODE_FORMERR;
775
413k
    return 0;
776
413k
  } 
777
778
1.37M
  if( (msg->qtype == LDNS_RR_TYPE_RRSIG ||
779
1.37M
       msg->qtype == LDNS_RR_TYPE_ANY) 
780
1.37M
      && sig_is_double(pkt, rrset, sldns_buffer_current(pkt))) {
781
13.6k
    if(!skip_ttl_rdata(pkt))
782
0
      return LDNS_RCODE_FORMERR;
783
13.6k
    return 0;
784
13.6k
  }
785
  
786
  /* create rr */
787
1.36M
  if(!(rr = (struct rr_parse*)regional_alloc(region, sizeof(*rr))))
788
0
    return LDNS_RCODE_SERVFAIL;
789
1.36M
  rr->outside_packet = 0;
790
1.36M
  rr->ttl_data = sldns_buffer_current(pkt);
791
1.36M
  rr->next = 0;
792
1.36M
  if(type == LDNS_RR_TYPE_RRSIG && rrset->type != LDNS_RR_TYPE_RRSIG) {
793
20.8k
    if(rrset->rrsig_last) 
794
19.6k
      rrset->rrsig_last->next = rr;
795
1.16k
    else  rrset->rrsig_first = rr;
796
20.8k
    rrset->rrsig_last = rr;
797
20.8k
    rrset->rrsig_count++;
798
1.34M
  } else {
799
1.34M
    if(rrset->rr_last)
800
1.17M
      rrset->rr_last->next = rr;
801
173k
    else  rrset->rr_first = rr;
802
1.34M
    rrset->rr_last = rr;
803
1.34M
    rrset->rr_count++;
804
1.34M
  }
805
806
  /* calc decompressed size */
807
1.36M
  if(!calc_size(pkt, type, rr))
808
3.75k
    return LDNS_RCODE_FORMERR;
809
1.36M
  rrset->size += rr->size;
810
811
1.36M
  return 0;
812
1.36M
}
813
814
/**
815
 * Parse packet RR section, for answer, authority and additional sections. 
816
 * @param pkt: packet, position at call must be at start of section.
817
 *  at end position is after section.
818
 * @param msg: store results here.
819
 * @param region: how to alloc results.
820
 * @param section: section enum.
821
 * @param num_rrs: how many rrs are in the section.
822
 * @param num_rrsets: returns number of rrsets in the section.
823
 * @return: 0 if OK, or rcode on error.
824
 */
825
static int
826
parse_section(sldns_buffer* pkt, struct msg_parse* msg, 
827
  struct regional* region, sldns_pkt_section section, 
828
  uint16_t num_rrs, size_t* num_rrsets)
829
22.3k
{
830
22.3k
  uint16_t i;
831
22.3k
  uint8_t* dname, *prev_dname_f = NULL, *prev_dname_l = NULL;
832
22.3k
  size_t dnamelen, prev_dnamelen = 0;
833
22.3k
  uint16_t type, prev_type = 0;
834
22.3k
  uint16_t dclass, prev_dclass = 0;
835
22.3k
  uint32_t rrset_flags = 0;
836
22.3k
  hashvalue_type hash = 0;
837
22.3k
  struct rrset_parse* rrset = NULL;
838
22.3k
  int r;
839
840
22.3k
  if(num_rrs == 0)
841
8.47k
    return 0;
842
13.8k
  if(sldns_buffer_remaining(pkt) <= 0)
843
123
    return LDNS_RCODE_FORMERR;
844
1.80M
  for(i=0; i<num_rrs; i++) {
845
    /* parse this RR. */
846
1.79M
    dname = sldns_buffer_current(pkt);
847
1.79M
    if((dnamelen = pkt_dname_len(pkt)) == 0)
848
2.90k
      return LDNS_RCODE_FORMERR;
849
1.79M
    if(sldns_buffer_remaining(pkt) < 10) /* type, class, ttl, len */
850
684
      return LDNS_RCODE_FORMERR;
851
1.79M
    type = sldns_buffer_read_u16(pkt);
852
1.79M
    sldns_buffer_read(pkt, &dclass, sizeof(dclass));
853
854
1.79M
    if(0) { /* debug show what is being parsed. */
855
0
      if(type == LDNS_RR_TYPE_RRSIG) {
856
0
        uint16_t t;
857
0
        if(pkt_rrsig_covered(pkt, 
858
0
          sldns_buffer_current(pkt), &t))
859
0
          fprintf(stderr, "parse of %s(%d) [%s(%d)]",
860
0
          sldns_rr_descript(type)?
861
0
          sldns_rr_descript(type)->_name: "??",
862
0
          (int)type,
863
0
          sldns_rr_descript(t)?
864
0
          sldns_rr_descript(t)->_name: "??",
865
0
          (int)t);
866
0
      } else
867
0
        fprintf(stderr, "parse of %s(%d)",
868
0
        sldns_rr_descript(type)?
869
0
        sldns_rr_descript(type)->_name: "??",
870
0
        (int)type);
871
0
      fprintf(stderr, " %s(%d) ",
872
0
        sldns_lookup_by_id(sldns_rr_classes, 
873
0
        (int)ntohs(dclass))?sldns_lookup_by_id(
874
0
        sldns_rr_classes, (int)ntohs(dclass))->name: 
875
0
        "??", (int)ntohs(dclass));
876
0
      dname_print(stderr, pkt, dname);
877
0
      fprintf(stderr, "\n");
878
0
    }
879
880
    /* see if it is part of an existing RR set */
881
1.79M
    if(!find_rrset(msg, pkt, dname, dnamelen, type, dclass, &hash, 
882
1.79M
      &rrset_flags, &prev_dname_f, &prev_dname_l, 
883
1.79M
      &prev_dnamelen, &prev_type, &prev_dclass, &rrset, 
884
1.79M
      section, region))
885
0
      return LDNS_RCODE_SERVFAIL;
886
1.79M
    if(!rrset) {
887
      /* it is a new RR set. hash&flags already calculated.*/
888
153k
      (*num_rrsets)++;
889
153k
      rrset = new_rrset(msg, dname, dnamelen, type, dclass,
890
153k
        hash, rrset_flags, section, region);
891
153k
      if(!rrset) 
892
0
        return LDNS_RCODE_SERVFAIL;
893
153k
    }
894
1.63M
    else if(0)  { 
895
0
      fprintf(stderr, "is part of existing: ");
896
0
      dname_print(stderr, pkt, rrset->dname);
897
0
      fprintf(stderr, " type %s(%d)\n",
898
0
        sldns_rr_descript(rrset->type)?
899
0
        sldns_rr_descript(rrset->type)->_name: "??",
900
0
        (int)rrset->type);
901
0
    }
902
    /* add to rrset. */
903
1.79M
    if((r=add_rr_to_rrset(rrset, pkt, msg, region, section, 
904
1.79M
      type)) != 0)
905
4.08k
      return r;
906
1.79M
  }
907
6.03k
  return 0;
908
13.7k
}
909
910
int
911
parse_packet(sldns_buffer* pkt, struct msg_parse* msg, struct regional* region)
912
11.2k
{
913
11.2k
  int ret;
914
11.2k
  if(sldns_buffer_remaining(pkt) < LDNS_HEADER_SIZE)
915
30
    return LDNS_RCODE_FORMERR;
916
  /* read the header */
917
11.2k
  sldns_buffer_read(pkt, &msg->id, sizeof(uint16_t));
918
11.2k
  msg->flags = sldns_buffer_read_u16(pkt);
919
11.2k
  msg->qdcount = sldns_buffer_read_u16(pkt);
920
11.2k
  msg->ancount = sldns_buffer_read_u16(pkt);
921
11.2k
  msg->nscount = sldns_buffer_read_u16(pkt);
922
11.2k
  msg->arcount = sldns_buffer_read_u16(pkt);
923
11.2k
  if(msg->qdcount > 1)
924
57
    return LDNS_RCODE_FORMERR;
925
11.1k
  if((ret = parse_query_section(pkt, msg)) != 0)
926
85
    return ret;
927
11.0k
  if((ret = parse_section(pkt, msg, region, LDNS_SECTION_ANSWER,
928
11.0k
    msg->ancount, &msg->an_rrsets)) != 0)
929
4.17k
    return ret;
930
6.90k
  if((ret = parse_section(pkt, msg, region, LDNS_SECTION_AUTHORITY,
931
6.90k
    msg->nscount, &msg->ns_rrsets)) != 0)
932
2.12k
    return ret;
933
4.78k
  if(sldns_buffer_remaining(pkt) == 0 && msg->arcount == 1) {
934
    /* BIND accepts leniently that an EDNS record is missing.
935
     * so, we do too. */
936
4.31k
  } else if((ret = parse_section(pkt, msg, region,
937
4.31k
    LDNS_SECTION_ADDITIONAL, msg->arcount, &msg->ar_rrsets)) != 0)
938
1.49k
    return ret;
939
  /* if(sldns_buffer_remaining(pkt) > 0) { */
940
    /* there is spurious data at end of packet. ignore */
941
  /* } */
942
3.28k
  msg->rrset_count = msg->an_rrsets + msg->ns_rrsets + msg->ar_rrsets;
943
3.28k
  return 0;
944
4.78k
}
945
946
/** parse EDNS options from EDNS wireformat rdata */
947
static int
948
parse_edns_options_from_query(uint8_t* rdata_ptr, size_t rdata_len,
949
  struct edns_data* edns, struct config_file* cfg, struct comm_point* c,
950
  struct comm_reply* repinfo, uint32_t now, struct regional* region,
951
  struct cookie_secrets* cookie_secrets)
952
0
{
953
  /* To respond with a Keepalive option, the client connection must have
954
   * received one message with a TCP Keepalive EDNS option, and that
955
   * option must have 0 length data. Subsequent messages sent on that
956
   * connection will have a TCP Keepalive option.
957
   *
958
   * In the if-statement below, the option is added unsolicited. This
959
   * means that the client has sent an KEEPALIVE option earlier. We know
960
   * here this is true, because c->tcp_keepalive is set.
961
   */
962
0
  if (cfg && cfg->do_tcp_keepalive && c && c->type != comm_udp && c->tcp_keepalive) {
963
0
    if(!edns_opt_list_append_keepalive(&edns->opt_list_out,
964
0
          c->tcp_timeout_msec / 100, region)) {
965
0
      log_err("out of memory");
966
0
      return LDNS_RCODE_SERVFAIL;
967
0
    }
968
0
  }
969
970
  /* while still more options, and have code+len to read */
971
  /* ignores partial content (i.e. rdata len 3) */
972
0
  while(rdata_len >= 4) {
973
0
    uint16_t opt_code = sldns_read_uint16(rdata_ptr);
974
0
    uint16_t opt_len = sldns_read_uint16(rdata_ptr+2);
975
0
    uint8_t server_cookie[40];
976
0
    enum edns_cookie_val_status cookie_val_status;
977
0
    int cookie_is_v4 = 1;
978
979
0
    rdata_ptr += 4;
980
0
    rdata_len -= 4;
981
0
    if(opt_len > rdata_len)
982
0
      break; /* option code partial */
983
984
    /* handle parse time edns options here */
985
0
    switch(opt_code) {
986
0
    case LDNS_EDNS_NSID:
987
0
      if (!cfg || !cfg->nsid)
988
0
        break;
989
0
      if(!edns_opt_list_append(&edns->opt_list_out,
990
0
            LDNS_EDNS_NSID, cfg->nsid_len,
991
0
            cfg->nsid, region)) {
992
0
        log_err("out of memory");
993
0
        return LDNS_RCODE_SERVFAIL;
994
0
      }
995
0
      break;
996
997
0
    case LDNS_EDNS_KEEPALIVE:
998
      /* To respond with a Keepalive option, the client
999
       * connection must have received one message with a TCP
1000
       * Keepalive EDNS option, and that option must have 0
1001
       * length data. Subsequent messages sent on that
1002
       * connection will have a TCP Keepalive option.
1003
       *
1004
       * This should be the first time the client sends this
1005
       * option, so c->tcp_keepalive is not set.
1006
       * Besides adding the reply KEEPALIVE option, 
1007
       * c->tcp_keepalive will be set so that the
1008
       * option will be added unsolicited in subsequent
1009
       * responses (see the comment above the if-statement
1010
       * at the start of this function).
1011
       */
1012
0
      if (!cfg || !cfg->do_tcp_keepalive || !c ||
1013
0
          c->type == comm_udp || c->tcp_keepalive)
1014
0
        break;
1015
0
      if(opt_len) {
1016
0
        verbose(VERB_ALGO, "query with bad edns keepalive.");
1017
0
        return LDNS_RCODE_FORMERR;
1018
0
      }
1019
0
      if(!edns_opt_list_append_keepalive(&edns->opt_list_out,
1020
0
            c->tcp_timeout_msec / 100,
1021
0
            region)) {
1022
0
        log_err("out of memory");
1023
0
        return LDNS_RCODE_SERVFAIL;
1024
0
      }
1025
0
      c->tcp_keepalive = 1;
1026
0
      break;
1027
1028
0
    case LDNS_EDNS_PADDING:
1029
0
      if(!cfg || !cfg->pad_responses ||
1030
0
          !c || c->type != comm_tcp ||!c->ssl)
1031
0
        break;
1032
0
      if(!edns_opt_list_append(&edns->opt_list_out,
1033
0
            LDNS_EDNS_PADDING,
1034
0
            0, NULL, region)) {
1035
0
        log_err("out of memory");
1036
0
        return LDNS_RCODE_SERVFAIL;
1037
0
      }
1038
0
      edns->padding_block_size = cfg->pad_responses_block_size;
1039
0
      break;
1040
1041
0
    case LDNS_EDNS_COOKIE:
1042
0
      if(!cfg || !cfg->do_answer_cookie || !repinfo)
1043
0
        break;
1044
0
      if(opt_len != 8 && (opt_len < 16 || opt_len > 40)) {
1045
0
        verbose(VERB_ALGO, "worker request: "
1046
0
          "badly formatted cookie");
1047
0
        return LDNS_RCODE_FORMERR;
1048
0
      }
1049
0
      edns->cookie_present = 1;
1050
1051
      /* Copy client cookie, version and timestamp for
1052
       * validation and creation purposes.
1053
       */
1054
0
      if(opt_len >= 16) {
1055
0
        memmove(server_cookie, rdata_ptr, 16);
1056
0
      } else {
1057
0
        memset(server_cookie, 0, 16);
1058
0
        memmove(server_cookie, rdata_ptr, opt_len);
1059
0
      }
1060
1061
      /* Copy client ip for validation and creation
1062
       * purposes. It will be overwritten if (re)creation
1063
       * is needed.
1064
       */
1065
0
      if(repinfo->remote_addr.ss_family == AF_INET) {
1066
0
        memcpy(server_cookie + 16,
1067
0
          &((struct sockaddr_in*)&repinfo->remote_addr)->sin_addr, 4);
1068
0
      } else {
1069
0
        cookie_is_v4 = 0;
1070
0
        memcpy(server_cookie + 16,
1071
0
          &((struct sockaddr_in6*)&repinfo->remote_addr)->sin6_addr, 16);
1072
0
      }
1073
1074
0
      if(cfg->cookie_secret_file &&
1075
0
        cfg->cookie_secret_file[0]) {
1076
        /* Loop over the active and staging cookies. */
1077
0
        cookie_val_status =
1078
0
          cookie_secrets_server_validate(
1079
0
          rdata_ptr, opt_len, cookie_secrets,
1080
0
          cookie_is_v4, server_cookie, now);
1081
0
      } else {
1082
        /* Use the cookie option value to validate. */
1083
0
        cookie_val_status = edns_cookie_server_validate(
1084
0
          rdata_ptr, opt_len, cfg->cookie_secret,
1085
0
          cfg->cookie_secret_len, cookie_is_v4,
1086
0
          server_cookie, now);
1087
0
      }
1088
0
      if(cookie_val_status == COOKIE_STATUS_VALID_RENEW)
1089
0
        edns->cookie_valid = 1;
1090
0
      switch(cookie_val_status) {
1091
0
      case COOKIE_STATUS_VALID:
1092
0
        edns->cookie_valid = 1;
1093
        /* Reuse cookie */
1094
0
        if(!edns_opt_list_append(
1095
0
          &edns->opt_list_out, LDNS_EDNS_COOKIE,
1096
0
          opt_len, rdata_ptr, region)) {
1097
0
          log_err("out of memory");
1098
0
          return LDNS_RCODE_SERVFAIL;
1099
0
        }
1100
        /* Cookie to be reused added to outgoing
1101
         * options. Done!
1102
         */
1103
0
        break;
1104
0
      case COOKIE_STATUS_CLIENT_ONLY:
1105
0
        edns->cookie_client = 1;
1106
0
        ATTR_FALLTHROUGH
1107
        /* fallthrough */
1108
0
      case COOKIE_STATUS_VALID_RENEW:
1109
0
      case COOKIE_STATUS_FUTURE:
1110
0
      case COOKIE_STATUS_EXPIRED:
1111
0
      case COOKIE_STATUS_INVALID:
1112
0
      default:
1113
0
        if(cfg->cookie_secret_file &&
1114
0
          cfg->cookie_secret_file[0]) {
1115
0
          if(!cookie_secrets)
1116
0
            break;
1117
0
          lock_basic_lock(&cookie_secrets->lock);
1118
0
          if(cookie_secrets->cookie_count < 1) {
1119
0
            lock_basic_unlock(&cookie_secrets->lock);
1120
0
            break;
1121
0
          }
1122
0
          edns_cookie_server_write(server_cookie,
1123
0
            cookie_secrets->cookie_secrets[0].cookie_secret,
1124
0
            cookie_is_v4, now);
1125
0
          lock_basic_unlock(&cookie_secrets->lock);
1126
0
        } else {
1127
0
          edns_cookie_server_write(server_cookie,
1128
0
            cfg->cookie_secret, cookie_is_v4, now);
1129
0
        }
1130
0
        if(!edns_opt_list_append(&edns->opt_list_out,
1131
0
          LDNS_EDNS_COOKIE, 24, server_cookie,
1132
0
          region)) {
1133
0
          log_err("out of memory");
1134
0
          return LDNS_RCODE_SERVFAIL;
1135
0
        }
1136
0
        break;
1137
0
      }
1138
0
      break;
1139
0
    default:
1140
0
      break;
1141
0
    }
1142
0
    if(!edns_opt_list_append(&edns->opt_list_in,
1143
0
        opt_code, opt_len, rdata_ptr, region)) {
1144
0
      log_err("out of memory");
1145
0
      return LDNS_RCODE_SERVFAIL;
1146
0
    }
1147
0
    rdata_ptr += opt_len;
1148
0
    rdata_len -= opt_len;
1149
0
  }
1150
0
  return LDNS_RCODE_NOERROR;
1151
0
}
1152
1153
int 
1154
parse_extract_edns_from_response_msg(struct msg_parse* msg,
1155
  struct edns_data* edns, struct regional* region)
1156
3.27k
{
1157
3.27k
  struct rrset_parse* rrset = msg->rrset_first;
1158
3.27k
  struct rrset_parse* prev = 0;
1159
3.27k
  struct rrset_parse* found = 0;
1160
3.27k
  struct rrset_parse* found_prev = 0;
1161
3.27k
  size_t rdata_len;
1162
3.27k
  uint8_t* rdata_ptr;
1163
  /* since the class encodes the UDP size, we cannot use hash table to
1164
   * find the EDNS OPT record. Scan the packet. */
1165
81.2k
  while(rrset) {
1166
77.9k
    if(rrset->type == LDNS_RR_TYPE_OPT) {
1167
      /* only one OPT RR allowed. */
1168
306
      if(found) return LDNS_RCODE_FORMERR;
1169
      /* found it! */
1170
303
      found_prev = prev;
1171
303
      found = rrset;
1172
303
    }
1173
77.9k
    prev = rrset;
1174
77.9k
    rrset = rrset->rrset_all_next;
1175
77.9k
  }
1176
3.26k
  if(!found) {
1177
2.96k
    memset(edns, 0, sizeof(*edns));
1178
2.96k
    edns->udp_size = 512;
1179
2.96k
    return 0;
1180
2.96k
  }
1181
  /* check the found RRset */
1182
  /* most lenient check possible. ignore dname, use last opt */
1183
300
  if(found->section != LDNS_SECTION_ADDITIONAL)
1184
3
    return LDNS_RCODE_FORMERR; 
1185
297
  if(found->rr_count == 0)
1186
0
    return LDNS_RCODE_FORMERR;
1187
297
  if(0) { /* strict checking of dname and RRcount */
1188
0
    if(found->dname_len != 1 || !found->dname 
1189
0
      || found->dname[0] != 0) return LDNS_RCODE_FORMERR; 
1190
0
    if(found->rr_count != 1) return LDNS_RCODE_FORMERR; 
1191
0
  }
1192
297
  log_assert(found->rr_first && found->rr_last);
1193
1194
  /* remove from packet */
1195
297
  if(found_prev)  found_prev->rrset_all_next = found->rrset_all_next;
1196
240
  else  msg->rrset_first = found->rrset_all_next;
1197
297
  if(found == msg->rrset_last)
1198
252
    msg->rrset_last = found_prev;
1199
297
  msg->arcount --;
1200
297
  msg->ar_rrsets --;
1201
297
  msg->rrset_count --;
1202
  
1203
  /* take the data ! */
1204
297
  edns->edns_present = 1;
1205
297
  edns->ext_rcode = found->rr_last->ttl_data[0];
1206
297
  edns->edns_version = found->rr_last->ttl_data[1];
1207
297
  edns->bits = sldns_read_uint16(&found->rr_last->ttl_data[2]);
1208
297
  edns->udp_size = ntohs(found->rrset_class);
1209
297
  edns->opt_list_in = NULL;
1210
297
  edns->opt_list_out = NULL;
1211
297
  edns->opt_list_inplace_cb_out = NULL;
1212
297
  edns->padding_block_size = 0;
1213
297
  edns->cookie_present = 0;
1214
297
  edns->cookie_valid = 0;
1215
1216
  /* take the options */
1217
297
  rdata_len = found->rr_first->size-2;
1218
297
  rdata_ptr = found->rr_first->ttl_data+6;
1219
1220
  /* while still more options, and have code+len to read */
1221
  /* ignores partial content (i.e. rdata len 3) */
1222
18.0k
  while(rdata_len >= 4) {
1223
17.9k
    uint16_t opt_code = sldns_read_uint16(rdata_ptr);
1224
17.9k
    uint16_t opt_len = sldns_read_uint16(rdata_ptr+2);
1225
17.9k
    rdata_ptr += 4;
1226
17.9k
    rdata_len -= 4;
1227
17.9k
    if(opt_len > rdata_len)
1228
139
      break; /* option code partial */
1229
1230
17.7k
    if(!edns_opt_list_append(&edns->opt_list_in,
1231
17.7k
        opt_code, opt_len, rdata_ptr, region)) {
1232
0
      log_err("out of memory");
1233
0
      break;
1234
0
    }
1235
17.7k
    rdata_ptr += opt_len;
1236
17.7k
    rdata_len -= opt_len;
1237
17.7k
  }
1238
  /* ignore rrsigs */
1239
297
  return LDNS_RCODE_NOERROR;
1240
297
}
1241
1242
/** skip RR in packet */
1243
static int
1244
skip_pkt_rr(sldns_buffer* pkt)
1245
0
{
1246
0
  if(sldns_buffer_remaining(pkt) < 1) return 0;
1247
0
  if(!pkt_dname_len(pkt))
1248
0
    return 0;
1249
0
  if(sldns_buffer_remaining(pkt) < 4) return 0;
1250
0
  sldns_buffer_skip(pkt, 4); /* type and class */
1251
0
  if(!skip_ttl_rdata(pkt))
1252
0
    return 0;
1253
0
  return 1;
1254
0
}
1255
1256
/** skip RRs from packet */
1257
int
1258
skip_pkt_rrs(sldns_buffer* pkt, int num)
1259
0
{
1260
0
  int i;
1261
0
  for(i=0; i<num; i++) {
1262
0
    if(!skip_pkt_rr(pkt))
1263
0
      return 0;
1264
0
  }
1265
0
  return 1;
1266
0
}
1267
1268
int 
1269
parse_edns_from_query_pkt(sldns_buffer* pkt, struct edns_data* edns,
1270
  struct config_file* cfg, struct comm_point* c,
1271
  struct comm_reply* repinfo, time_t now, struct regional* region,
1272
  struct cookie_secrets* cookie_secrets)
1273
0
{
1274
0
  size_t rdata_len;
1275
0
  uint8_t* rdata_ptr;
1276
0
  log_assert(LDNS_QDCOUNT(sldns_buffer_begin(pkt)) == 1);
1277
0
  memset(edns, 0, sizeof(*edns));
1278
0
  if(LDNS_ANCOUNT(sldns_buffer_begin(pkt)) != 0 ||
1279
0
    LDNS_NSCOUNT(sldns_buffer_begin(pkt)) != 0) {
1280
0
    if(!skip_pkt_rrs(pkt, ((int)LDNS_ANCOUNT(sldns_buffer_begin(pkt)))+
1281
0
      ((int)LDNS_NSCOUNT(sldns_buffer_begin(pkt)))))
1282
0
      return LDNS_RCODE_FORMERR;
1283
0
  }
1284
  /* check edns section is present */
1285
0
  if(LDNS_ARCOUNT(sldns_buffer_begin(pkt)) > 1) {
1286
0
    return LDNS_RCODE_FORMERR;
1287
0
  }
1288
0
  if(LDNS_ARCOUNT(sldns_buffer_begin(pkt)) == 0) {
1289
0
    edns->udp_size = 512;
1290
0
    return 0;
1291
0
  }
1292
  /* domain name must be the root of length 1. */
1293
0
  if(pkt_dname_len(pkt) != 1)
1294
0
    return LDNS_RCODE_FORMERR;
1295
0
  if(sldns_buffer_remaining(pkt) < 10) /* type, class, ttl, rdatalen */
1296
0
    return LDNS_RCODE_FORMERR;
1297
0
  if(sldns_buffer_read_u16(pkt) != LDNS_RR_TYPE_OPT)
1298
0
    return LDNS_RCODE_FORMERR;
1299
0
  edns->edns_present = 1;
1300
0
  edns->udp_size = sldns_buffer_read_u16(pkt); /* class is udp size */
1301
0
  edns->ext_rcode = sldns_buffer_read_u8(pkt); /* ttl used for bits */
1302
0
  edns->edns_version = sldns_buffer_read_u8(pkt);
1303
0
  edns->bits = sldns_buffer_read_u16(pkt);
1304
0
  edns->opt_list_in = NULL;
1305
0
  edns->opt_list_out = NULL;
1306
0
  edns->opt_list_inplace_cb_out = NULL;
1307
0
  edns->padding_block_size = 0;
1308
0
  edns->cookie_present = 0;
1309
0
  edns->cookie_valid = 0;
1310
1311
  /* take the options */
1312
0
  rdata_len = sldns_buffer_read_u16(pkt);
1313
0
  if(sldns_buffer_remaining(pkt) < rdata_len)
1314
0
    return LDNS_RCODE_FORMERR;
1315
0
  rdata_ptr = sldns_buffer_current(pkt);
1316
  /* ignore rrsigs */
1317
0
  return parse_edns_options_from_query(rdata_ptr, rdata_len, edns, cfg,
1318
0
    c, repinfo, now, region, cookie_secrets);
1319
0
}
1320
1321
void
1322
log_edns_opt_list(enum verbosity_value level, const char* info_str,
1323
  struct edns_option* list)
1324
0
{
1325
0
  if(verbosity >= level && list) {
1326
0
    char str[128], *s;
1327
0
    size_t slen;
1328
0
    verbose(level, "%s", info_str);
1329
0
    while(list) {
1330
0
      s = str;
1331
0
      slen = sizeof(str);
1332
0
      (void)sldns_wire2str_edns_option_print(&s, &slen, list->opt_code,
1333
0
        list->opt_data, list->opt_len);
1334
0
      verbose(level, "  %s", str);
1335
0
      list = list->next;
1336
0
    }
1337
0
  }
1338
0
}
1339
1340
/** remove RR from msgparse RRset, return true if rrset is entirely bad */
1341
int
1342
msgparse_rrset_remove_rr(const char* str, sldns_buffer* pkt, struct rrset_parse* rrset,
1343
  struct rr_parse* prev, struct rr_parse* rr, struct sockaddr_storage* addr, socklen_t addrlen)
1344
155k
{
1345
155k
  if(verbosity >= VERB_QUERY && rrset->dname_len <= LDNS_MAX_DOMAINLEN && str) {
1346
0
    uint8_t buf[LDNS_MAX_DOMAINLEN+1];
1347
0
    dname_pkt_copy(pkt, buf, rrset->dname);
1348
0
    if(addr)
1349
0
      log_name_addr(VERB_QUERY, str, buf, addr, addrlen);
1350
0
    else  log_nametypeclass(VERB_QUERY, str, buf,
1351
0
        rrset->type, ntohs(rrset->rrset_class));
1352
0
  }
1353
155k
  if(prev)
1354
26.3k
    prev->next = rr->next;
1355
129k
  else  rrset->rr_first = rr->next;
1356
155k
  if(rrset->rr_last == rr)
1357
5.21k
    rrset->rr_last = prev;
1358
155k
  rrset->rr_count --;
1359
155k
  rrset->size -= rr->size;
1360
  /* rr struct still exists, but is unlinked, so that in the for loop
1361
   * the rr->next works fine to continue. */
1362
155k
  return rrset->rr_count == 0;
1363
155k
}