net: dns: Check compression flag properly (92602700) · Commits · Wenxi XU / zephyr

subsys/net/lib/dns/dns_pack.c

+5 −2

Changes for subsys/net/lib/dns/dns_pack.c: 5 added lines, 2 removed lines.
Original line number	Diff line number	Diff line
		@@ -380,7 +380,7 @@ int dns_copy_qname(uint8_t buf, uint16_t len, uint16_t size,
		lb_size = msg[pos];

		/* pointer */
		if (lb_size > DNS_LABEL_MAX_SIZE) {
		if ((lb_size & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
		uint8_t mask = DNS_LABEL_MAX_SIZE;

		if (pos + 1 >= msg_size) {
		@@ -403,6 +403,9 @@ int dns_copy_qname(uint8_t buf, uint16_t len, uint16_t size,
		}

		continue;
		} else if (lb_size & NS_CMPRSFLGS) {
		rc = -EINVAL;
		break;
		}

		/* validate that the label (i.e. size + elements),
		@@ -491,7 +494,7 @@ static int dns_unpack_name(const uint8_t msg, int maxlen, const uint8_t src,
		}

		while ((val = *curr_src++)) {
		if (val & NS_CMPRSFLGS) {
		if ((val & NS_CMPRSFLGS) == NS_CMPRSFLGS) {
		/* Follow pointer */
		int pos;