Skip to content
Snippets Groups Projects
Commit 8ad5e44c authored by Martine Lenders's avatar Martine Lenders
Browse files

sock_dns: remove some magic numbers

parent 2840b382
No related branches found
No related tags found
No related merge requests found
/*
* Copyright (C) 2019 Freie Universität Berlin
*
* This file is subject to the terms and conditions of the GNU Lesser
* General Public License v2.1. See the file LICENSE in the top level
* directory for more details.
*/
/**
* @defgroup net_dns DNS defines
* @ingroup net
* @brief Generic DNS values
* @{
*
* @file
* @brief Generic DNS values
*
* @author Martine Lenders <m.lenders@fu-berlin.de>
*/
#ifndef NET_DNS_H
#define NET_DNS_H
#ifdef __cplusplus
extern "C" {
#endif
/**
* @name Field lengths
* @{
*/
#define RR_TYPE_LENGTH (2U)
#define RR_CLASS_LENGTH (2U)
#define RR_TTL_LENGTH (4U)
#define RR_RDLENGTH_LENGTH (2U)
/** @} */
#ifdef __cplusplus
}
#endif
#endif /* NET_DNS_H */
/** @} */
...@@ -19,6 +19,7 @@ ...@@ -19,6 +19,7 @@
#include <string.h> #include <string.h>
#include <stdio.h> #include <stdio.h>
#include "net/dns.h"
#include "net/sock/udp.h" #include "net/sock/udp.h"
#include "net/sock/dns.h" #include "net/sock/dns.h"
...@@ -114,7 +115,8 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family ...@@ -114,7 +115,8 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family
return tmp; return tmp;
} }
bufpos += tmp; bufpos += tmp;
bufpos += 4; /* skip type and class of query */ /* skip type and class of query */
bufpos += (RR_TYPE_LENGTH + RR_CLASS_LENGTH);
} }
for (unsigned n = 0; n < ntohs(hdr->ancount); n++) { for (unsigned n = 0; n < ntohs(hdr->ancount); n++) {
...@@ -123,14 +125,14 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family ...@@ -123,14 +125,14 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family
return tmp; return tmp;
} }
bufpos += tmp; bufpos += tmp;
if ((bufpos + 2 + 2 + 4) >= buflim) { if ((bufpos + RR_TYPE_LENGTH + RR_CLASS_LENGTH + RR_TTL_LENGTH) >= buflim) {
return -EBADMSG; return -EBADMSG;
} }
uint16_t _type = ntohs(_get_short(bufpos)); uint16_t _type = ntohs(_get_short(bufpos));
bufpos += 2; bufpos += RR_TYPE_LENGTH;
uint16_t class = ntohs(_get_short(bufpos)); uint16_t class = ntohs(_get_short(bufpos));
bufpos += 2; bufpos += RR_CLASS_LENGTH;
bufpos += 4; /* skip ttl */ bufpos += RR_TTL_LENGTH; /* skip ttl */
unsigned addrlen = ntohs(_get_short(bufpos)); unsigned addrlen = ntohs(_get_short(bufpos));
/* skip unwanted answers */ /* skip unwanted answers */
...@@ -154,7 +156,7 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family ...@@ -154,7 +156,7 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family
(family == AF_UNSPEC))) { (family == AF_UNSPEC))) {
return -EBADMSG; return -EBADMSG;
} }
bufpos += 2; bufpos += RR_RDLENGTH_LENGTH;
if ((bufpos + addrlen) >= buflim) { if ((bufpos + addrlen) >= buflim) {
return -EBADMSG; return -EBADMSG;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment