mirror/iodine
1
0
Fork 0
mirror of https://github.com/yarrick/iodine.git synced 2024-11-25 14:36:05 +03:00
Code Issues Projects Releases Wiki Activity

Make sure replies with errors get the name parsed

Browse Source
This commit is contained in:
Erik Ekman 2009-09-21 20:35:18 +00:00
parent 3614a96417
commit a130abbfd9
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/dns.c
View File

@ -302,14 +302,15 @@ dns_decode(char *buf, size_t buflen, struct query *q, qr_t qr, char *packet, siz
switch (qr) { switch (qr) {
case QR_ANSWER: case QR_ANSWER:
if(qdcount < 1 || ancount < 1) { if(qdcount < 1) {
/* We may get both CNAME and A, then ancount=2 */ /* We need a question */
return -1; return -1;
} }
if (q != NULL) if (q != NULL)
q->id = id; q->id = id;
/* Read name even if no answer, to give better error message */
readname(packet, packetlen, &data, name, sizeof(name)); readname(packet, packetlen, &data, name, sizeof(name));
CHECKLEN(4); CHECKLEN(4);
readshort(packet, &data, &type); readshort(packet, &data, &type);
@ -320,6 +321,11 @@ dns_decode(char *buf, size_t buflen, struct query *q, qr_t qr, char *packet, siz
/* We only need the first char to check it */ /* We only need the first char to check it */
q->name[0] = name[0]; q->name[0] = name[0];
q->name[1] = '\0'; q->name[1] = '\0';
}
if (ancount < 1) {
/* We may get both CNAME and A, then ancount=2 */
return -1;
} }
/* Assume that first answer is NULL/CNAME that we wanted */ /* Assume that first answer is NULL/CNAME that we wanted */