mdns: Handle header parsing errors

videolabs · Mar 19, 2020 · 4fb1828 · 4fb1828
1 parent 80860fa
commit 4fb1828
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/mdns.c b/src/mdns.c
@@ -541,6 +541,8 @@ mdns_recv(const struct mdns_conn* conn, struct mdns_hdr *hdr, struct rr_entry **
 
         n = (size_t)length;
         const uint8_t *ptr = mdns_read_header(buf, &n, hdr);
+        if (ptr == NULL)
+                return (MDNS_ERROR);
 
         num_entry = hdr->num_qn + hdr->num_ans_rr + hdr->num_add_rr;
         for (size_t i = 0; i < num_entry; ++i) {