X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=avahi-core%2Fdns.c;h=c545c00ea33c19f4301974a3debc7b32d52a9bac;hb=59f3e8bef545762113c041a1ef5868e08b353045;hp=63ea8a0e4d65c14e38d4d67803e774dc90a4bf6a;hpb=943a6b93697fc0a7bcd0cc574744d7de609c189c;p=catta diff --git a/avahi-core/dns.c b/avahi-core/dns.c index 63ea8a0..c545c00 100644 --- a/avahi-core/dns.c +++ b/avahi-core/dns.c @@ -23,13 +23,14 @@ #include #endif -#include - #include #include #include #include +#include + +#include #include #include @@ -321,7 +322,6 @@ int avahi_dns_packet_check_valid_multicast(AvahiDnsPacket *p) { return 0; } - int avahi_dns_packet_is_query(AvahiDnsPacket *p) { assert(p); @@ -544,7 +544,8 @@ static int parse_rdata(AvahiDnsPacket *p, AvahiRecord *r, uint16_t rdlength) { case AVAHI_DNS_TYPE_TXT: if (rdlength > 0) { - r->data.txt.string_list = avahi_string_list_parse(avahi_dns_packet_get_rptr(p), rdlength); + if (avahi_string_list_parse(avahi_dns_packet_get_rptr(p), rdlength, &r->data.txt.string_list) < 0) + return -1; if (avahi_dns_packet_skip(p, rdlength) < 0) return -1; @@ -619,6 +620,9 @@ AvahiRecord* avahi_dns_packet_consume_record(AvahiDnsPacket *p, int *ret_cache_f if (parse_rdata(p, r, rdlength) < 0) goto fail; + + if (!avahi_record_is_valid(r)) + goto fail; return r; @@ -632,6 +636,7 @@ fail: AvahiKey* avahi_dns_packet_consume_key(AvahiDnsPacket *p, int *ret_unicast_response) { char name[256]; uint16_t type, class; + AvahiKey *k; assert(p); @@ -645,7 +650,15 @@ AvahiKey* avahi_dns_packet_consume_key(AvahiDnsPacket *p, int *ret_unicast_respo class &= ~AVAHI_DNS_UNICAST_RESPONSE; - return avahi_key_new(name, class, type); + if (!(k = avahi_key_new(name, class, type))) + return NULL; + + if (!avahi_key_is_valid(k)) { + avahi_key_unref(k); + return NULL; + } + + return k; } uint8_t* avahi_dns_packet_append_key(AvahiDnsPacket *p, AvahiKey *k, int unicast_response) {