Documentation addresses and check buffer limit.

git-svn-id: file:///svn/unbound/trunk@273 be551aaa-1e26-0410-a405-d3ace91eadb9
This commit is contained in:
Wouter Wijngaards 2007-05-01 15:11:53 +00:00
parent f31e3e5024
commit 182855d233
5 changed files with 48 additions and 8 deletions

View file

@ -3,6 +3,9 @@
But only for answers from other servers, not for plain queries. But only for answers from other servers, not for plain queries.
- compression and decompression test cases. - compression and decompression test cases.
- some stats added. - some stats added.
- example.conf interface: line is changed from 127.0.0.1 which leads
to problems if used (restricting communication to the localhost),
to a documentation and test address.
27 April 2007: Wouter 27 April 2007: Wouter
- removed iov usage, it is not good for dns message encoding. - removed iov usage, it is not good for dns message encoding.

View file

@ -20,7 +20,10 @@ server:
# specify the interfaces to answer queries from by ip-address. # specify the interfaces to answer queries from by ip-address.
# If you give none the default (all) interface is used. # If you give none the default (all) interface is used.
# interface: 192.168.0.153 # specify every interface on a new 'interface:' labelled line.
# interface: 192.0.2.153
# interface: 192.0.2.154
# interface: 2001:DB8::5
# port to answer queries from # port to answer queries from
# port: 53 # port: 53

View file

@ -452,17 +452,22 @@ dname_lab_cmp(uint8_t* d1, int labs1, uint8_t* d2, int labs2, int* mlabs)
return lastdiff; return lastdiff;
} }
void int
dname_buffer_write(ldns_buffer* pkt, uint8_t* dname) dname_buffer_write(ldns_buffer* pkt, uint8_t* dname)
{ {
uint8_t lablen; uint8_t lablen;
if(ldns_buffer_remaining(pkt) < 1)
return 0;
lablen = *dname++; lablen = *dname++;
ldns_buffer_write_u8(pkt, lablen); ldns_buffer_write_u8(pkt, lablen);
while(lablen) { while(lablen) {
if(ldns_buffer_remaining(pkt) < (size_t)lablen+1)
return 0;
ldns_buffer_write(pkt, dname, lablen); ldns_buffer_write(pkt, dname, lablen);
dname += lablen; dname += lablen;
lablen = *dname++; lablen = *dname++;
ldns_buffer_write_u8(pkt, lablen); ldns_buffer_write_u8(pkt, lablen);
} }
return 1;
} }

View file

@ -123,8 +123,9 @@ void dname_pkt_copy(ldns_buffer* pkt, uint8_t* to, uint8_t* dname);
* Copy over a valid dname to a packet. * Copy over a valid dname to a packet.
* @param pkt: packet to copy to. * @param pkt: packet to copy to.
* @param dname: dname to copy. * @param dname: dname to copy.
* @return: 0 if not enough space in buffer.
*/ */
void dname_buffer_write(ldns_buffer* pkt, uint8_t* dname); int dname_buffer_write(ldns_buffer* pkt, uint8_t* dname);
/** debug helper. Print wireformat dname to output. /** debug helper. Print wireformat dname to output.
* @param out: like stdout or a file. * @param out: like stdout or a file.

View file

@ -705,12 +705,17 @@ compress_owner(struct ub_packed_rrset_key* key, ldns_buffer* pkt,
* not capable of decoding ptr after a ptr. */ * not capable of decoding ptr after a ptr. */
*owner_ptr = htons((uint16_t)(0xc000 | *owner_ptr = htons((uint16_t)(0xc000 |
p->offset)); p->offset));
/* check if ptr + tc+4 ttl + rdatalen is available */
if(ldns_buffer_remaining(pkt) < 2+4+4+2)
return 0;
write_compressed_dname(pkt, key->rk.dname, write_compressed_dname(pkt, key->rk.dname,
owner_labs, p); owner_labs, p);
ldns_buffer_write(pkt, &key->rk.dname[ ldns_buffer_write(pkt, &key->rk.dname[
key->rk.dname_len], 4); key->rk.dname_len], 4);
} else { } else {
/* no compress */ /* no compress */
if(ldns_buffer_remaining(pkt) < key->rk.dname_len+4+4+2)
return 0;
ldns_buffer_write(pkt, key->rk.dname, ldns_buffer_write(pkt, key->rk.dname,
key->rk.dname_len+4); key->rk.dname_len+4);
if(owner_pos <= PTR_MAX_OFFSET) if(owner_pos <= PTR_MAX_OFFSET)
@ -722,6 +727,8 @@ compress_owner(struct ub_packed_rrset_key* key, ldns_buffer* pkt,
return 0; return 0;
} else { } else {
/* always compress 2nd-further RRs in RRset */ /* always compress 2nd-further RRs in RRset */
if(ldns_buffer_remaining(pkt) < 2+4+4+2)
return 0;
ldns_buffer_write(pkt, owner_ptr, 2); ldns_buffer_write(pkt, owner_ptr, 2);
ldns_buffer_write(pkt, &key->rk.dname[key->rk.dname_len], 4); ldns_buffer_write(pkt, &key->rk.dname[key->rk.dname_len], 4);
} }
@ -736,6 +743,8 @@ compress_any_dname(uint8_t* dname, ldns_buffer* pkt, int labs,
struct compress_tree_node* p; struct compress_tree_node* p;
size_t pos = ldns_buffer_position(pkt); size_t pos = ldns_buffer_position(pkt);
if((p = compress_tree_lookup(*tree, dname, labs))) { if((p = compress_tree_lookup(*tree, dname, labs))) {
if(ldns_buffer_remaining(pkt) < 2)
return 0;
write_compressed_dname(pkt, dname, labs, p); write_compressed_dname(pkt, dname, labs, p);
} else { } else {
dname_buffer_write(pkt, dname); dname_buffer_write(pkt, dname);
@ -767,6 +776,7 @@ compress_rdata(ldns_buffer* pkt, uint8_t* rdata, size_t todolen,
uint8_t count = desc->_dname_count; uint8_t count = desc->_dname_count;
ldns_buffer_skip(pkt, 2); /* rdata len fill in later */ ldns_buffer_skip(pkt, 2); /* rdata len fill in later */
/* space for rdatalen checked for already */
rdata += 2; rdata += 2;
todolen -= 2; todolen -= 2;
while(todolen > 0 && count) { while(todolen > 0 && count) {
@ -788,6 +798,8 @@ compress_rdata(ldns_buffer* pkt, uint8_t* rdata, size_t todolen,
} }
if(len) { if(len) {
/* copy over */ /* copy over */
if(ldns_buffer_remaining(pkt) < len)
return 0;
ldns_buffer_write(pkt, rdata, len); ldns_buffer_write(pkt, rdata, len);
todolen -= len; todolen -= len;
rdata += len; rdata += len;
@ -796,6 +808,8 @@ compress_rdata(ldns_buffer* pkt, uint8_t* rdata, size_t todolen,
} }
/* copy remainder */ /* copy remainder */
if(todolen > 0) { if(todolen > 0) {
if(ldns_buffer_remaining(pkt) < todolen)
return 0;
ldns_buffer_write(pkt, rdata, todolen); ldns_buffer_write(pkt, rdata, todolen);
} }
@ -829,6 +843,8 @@ packed_rrset_iov(struct ub_packed_rrset_key* key, ldns_buffer* pkt,
return 0; return 0;
} else { } else {
/* no compression */ /* no compression */
if(ldns_buffer_remaining(pkt) <
key->rk.dname_len + 4+4+2) return 0;
ldns_buffer_write(pkt, key->rk.dname, ldns_buffer_write(pkt, key->rk.dname,
key->rk.dname_len + 4); key->rk.dname_len + 4);
} }
@ -837,8 +853,12 @@ packed_rrset_iov(struct ub_packed_rrset_key* key, ldns_buffer* pkt,
if(!compress_rdata(pkt, data->rr_data[i], if(!compress_rdata(pkt, data->rr_data[i],
data->rr_len[i], region, tree, c)) data->rr_len[i], region, tree, c))
return 0; return 0;
} else ldns_buffer_write(pkt, data->rr_data[i], } else {
data->rr_len[i]); if(ldns_buffer_remaining(pkt) < data->rr_len[i])
return 0;
ldns_buffer_write(pkt, data->rr_data[i],
data->rr_len[i]);
}
} }
} }
/* insert rrsigs */ /* insert rrsigs */
@ -847,15 +867,21 @@ packed_rrset_iov(struct ub_packed_rrset_key* key, ldns_buffer* pkt,
*num_rrs += data->rrsig_count; *num_rrs += data->rrsig_count;
for(i=data->count; i<total; i++) { for(i=data->count; i<total; i++) {
if(1) { /* compression */ if(1) { /* compression */
if(owner_ptr) if(owner_ptr) {
if(ldns_buffer_remaining(pkt) <
2+4+4+data->rr_len[i])
return 0;
ldns_buffer_write(pkt, &owner_ptr, 2); ldns_buffer_write(pkt, &owner_ptr, 2);
else { } else {
if(!compress_any_dname(key->rk.dname, if(!compress_any_dname(key->rk.dname,
pkt, owner_labs, region, tree)) pkt, owner_labs, region, tree))
return 0; return 0;
} }
} else { } else {
/* no compression */ /* no compression */
if(ldns_buffer_remaining(pkt) <
key->rk.dname_len+4+4+data->rr_len[i])
return 0;
ldns_buffer_write(pkt, key->rk.dname, ldns_buffer_write(pkt, key->rk.dname,
key->rk.dname_len); key->rk.dname_len);
} }
@ -885,8 +911,10 @@ insert_section(struct reply_info* rep, size_t num_rrsets, uint16_t* num_rrs,
for(i=0; i<num_rrsets; i++) for(i=0; i<num_rrsets; i++)
if(!packed_rrset_iov(rep->rrsets[rrsets_before+i], pkt, if(!packed_rrset_iov(rep->rrsets[rrsets_before+i], pkt,
num_rrs, timenow, region, 1, 1, tree)) num_rrs, timenow, region, 1, 1, tree))
/* Bad, but if due to size must set TC bit */
return 0; return 0;
} else { } else {
/* might be OKAY to discard rrsets at end */
for(i=0; i<num_rrsets; i++) for(i=0; i<num_rrsets; i++)
if(!packed_rrset_iov(rep->rrsets[rrsets_before+i], pkt, if(!packed_rrset_iov(rep->rrsets[rrsets_before+i], pkt,
num_rrs, timenow, region, 1, 0, tree)) num_rrs, timenow, region, 1, 0, tree))
@ -907,7 +935,7 @@ int reply_info_encode(struct query_info* qinfo, struct reply_info* rep,
struct compress_tree_node* tree = 0; struct compress_tree_node* tree = 0;
ldns_buffer_clear(buffer); ldns_buffer_clear(buffer);
if(ldns_buffer_capacity(buffer) < LDNS_HEADER_SIZE) if(ldns_buffer_remaining(buffer) < LDNS_HEADER_SIZE)
return 0; return 0;
ldns_buffer_write(buffer, &id, sizeof(uint16_t)); ldns_buffer_write(buffer, &id, sizeof(uint16_t));