parent
d99d72c583
commit
2840b3825e
@ -689,6 +689,7 @@ endif
|
|||||||
|
|
||||||
ifneq (,$(filter sock_dns,$(USEMODULE)))
|
ifneq (,$(filter sock_dns,$(USEMODULE)))
|
||||||
USEMODULE += sock_util
|
USEMODULE += sock_util
|
||||||
|
USEMODULE += posix
|
||||||
endif
|
endif
|
||||||
|
|
||||||
ifneq (,$(filter sock_util,$(USEMODULE)))
|
ifneq (,$(filter sock_util,$(USEMODULE)))
|
||||||
|
|||||||
@ -15,6 +15,7 @@
|
|||||||
* @}
|
* @}
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#include <arpa/inet.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
@ -73,34 +74,58 @@ static unsigned _get_short(uint8_t *buf)
|
|||||||
return _tmp;
|
return _tmp;
|
||||||
}
|
}
|
||||||
|
|
||||||
static size_t _skip_hostname(uint8_t *buf)
|
static ssize_t _skip_hostname(const uint8_t *buf, size_t len, uint8_t *bufpos)
|
||||||
{
|
{
|
||||||
uint8_t *bufpos = buf;
|
const uint8_t *buflim = buf + len;
|
||||||
|
unsigned res = 0;
|
||||||
|
|
||||||
|
if (bufpos >= buflim) {
|
||||||
|
/* out-of-bound */
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
/* handle DNS Message Compression */
|
/* handle DNS Message Compression */
|
||||||
if (*bufpos >= 192) {
|
if (*bufpos >= 192) {
|
||||||
|
if ((bufpos + 2) >= buflim) {
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
return 2;
|
return 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
while(*bufpos) {
|
while (bufpos[res]) {
|
||||||
bufpos += *bufpos + 1;
|
res += bufpos[res] + 1;
|
||||||
|
if ((&bufpos[res]) >= buflim) {
|
||||||
|
/* out-of-bound */
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return (bufpos - buf + 1);
|
return res + 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family)
|
static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family)
|
||||||
{
|
{
|
||||||
|
const uint8_t *buflim = buf + len;
|
||||||
sock_dns_hdr_t *hdr = (sock_dns_hdr_t*) buf;
|
sock_dns_hdr_t *hdr = (sock_dns_hdr_t*) buf;
|
||||||
uint8_t *bufpos = buf + sizeof(*hdr);
|
uint8_t *bufpos = buf + sizeof(*hdr);
|
||||||
|
|
||||||
/* skip all queries that are part of the reply */
|
/* skip all queries that are part of the reply */
|
||||||
for (unsigned n = 0; n < ntohs(hdr->qdcount); n++) {
|
for (unsigned n = 0; n < ntohs(hdr->qdcount); n++) {
|
||||||
bufpos += _skip_hostname(bufpos);
|
ssize_t tmp = _skip_hostname(buf, len, bufpos);
|
||||||
|
if (tmp < 0) {
|
||||||
|
return tmp;
|
||||||
|
}
|
||||||
|
bufpos += tmp;
|
||||||
bufpos += 4; /* skip type and class of query */
|
bufpos += 4; /* skip type and class of query */
|
||||||
}
|
}
|
||||||
|
|
||||||
for (unsigned n = 0; n < ntohs(hdr->ancount); n++) {
|
for (unsigned n = 0; n < ntohs(hdr->ancount); n++) {
|
||||||
bufpos += _skip_hostname(bufpos);
|
ssize_t tmp = _skip_hostname(buf, len, bufpos);
|
||||||
|
if (tmp < 0) {
|
||||||
|
return tmp;
|
||||||
|
}
|
||||||
|
bufpos += tmp;
|
||||||
|
if ((bufpos + 2 + 2 + 4) >= buflim) {
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
uint16_t _type = ntohs(_get_short(bufpos));
|
uint16_t _type = ntohs(_get_short(bufpos));
|
||||||
bufpos += 2;
|
bufpos += 2;
|
||||||
uint16_t class = ntohs(_get_short(bufpos));
|
uint16_t class = ntohs(_get_short(bufpos));
|
||||||
@ -108,20 +133,31 @@ static int _parse_dns_reply(uint8_t *buf, size_t len, void* addr_out, int family
|
|||||||
bufpos += 4; /* skip ttl */
|
bufpos += 4; /* skip ttl */
|
||||||
|
|
||||||
unsigned addrlen = ntohs(_get_short(bufpos));
|
unsigned addrlen = ntohs(_get_short(bufpos));
|
||||||
bufpos += 2;
|
|
||||||
if ((bufpos + addrlen) > (buf + len)) {
|
|
||||||
return -EBADMSG;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* skip unwanted answers */
|
/* skip unwanted answers */
|
||||||
if ((class != DNS_CLASS_IN) ||
|
if ((class != DNS_CLASS_IN) ||
|
||||||
((_type == DNS_TYPE_A) && (family == AF_INET6)) ||
|
((_type == DNS_TYPE_A) && (family == AF_INET6)) ||
|
||||||
((_type == DNS_TYPE_AAAA) && (family == AF_INET)) ||
|
((_type == DNS_TYPE_AAAA) && (family == AF_INET)) ||
|
||||||
! ((_type == DNS_TYPE_A) || ((_type == DNS_TYPE_AAAA))
|
! ((_type == DNS_TYPE_A) || ((_type == DNS_TYPE_AAAA))
|
||||||
)) {
|
)) {
|
||||||
|
if (addrlen > len) {
|
||||||
|
/* buffer wraps around memory space */
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
bufpos += addrlen;
|
bufpos += addrlen;
|
||||||
|
/* other out-of-bound is checked in `_skip_hostname()` at start of
|
||||||
|
* loop */
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
if (((addrlen != INADDRSZ) && (family == AF_INET)) ||
|
||||||
|
((addrlen != IN6ADDRSZ) && (family == AF_INET6)) ||
|
||||||
|
((addrlen != IN6ADDRSZ) && (addrlen != INADDRSZ) &&
|
||||||
|
(family == AF_UNSPEC))) {
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
|
bufpos += 2;
|
||||||
|
if ((bufpos + addrlen) >= buflim) {
|
||||||
|
return -EBADMSG;
|
||||||
|
}
|
||||||
|
|
||||||
memcpy(addr_out, bufpos, addrlen);
|
memcpy(addr_out, bufpos, addrlen);
|
||||||
return addrlen;
|
return addrlen;
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user