From e24f3960390b3ee8c0559dc60c4aca169e0139bb Mon Sep 17 00:00:00 2001
From: Erik Ekman <yarrick@kryo.se>
Date: Mon, 5 Jun 2006 23:56:32 +0000
Subject: [PATCH] Dont forward the ping packet

---
 dns.c  | 17 ++++++++++-------
 dnsd.c |  3 ++-
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/dns.c b/dns.c
index 53545ac..687f3ce 100644
--- a/dns.c
+++ b/dns.c
@@ -185,7 +185,7 @@ dns_ping(int dns_fd)
 		printf("No reply on chunk, resending\n");
 		dns_send_chunk(dns_fd);
 	} else {
-		dns_write(dns_fd, dns_fd, "", 0);
+		dns_write(dns_fd, chunkid++, "", 0);
 	}
 }
 
@@ -258,19 +258,22 @@ dns_write(int fd, int id, char *buf, int len)
 
 	avail = MIN(avail, len); // do not use more bytes than is available;
 	final = (avail == len);	// is this the last block?
+	bzero(data, sizeof(data));
 	d = data;
 
 	// First byte is 0 for middle packet and 1 for last packet
 	*d = '0' + final;
 	d++;
 
-	for (i = 0; i < avail; i++) {
-		if (i > 0 && i % 31 == 0) {
-			*d = '.';
-			d++;
+	if (len > 0) {
+		for (i = 0; i < avail; i++) {
+			if (i > 0 && i % 31 == 0) {
+				*d = '.';
+				d++;
+			}
+			put_hex(d, buf[i]);
+			d += 2;
 		}
-		put_hex(d, buf[i]);
-		d += 2;
 	}
 	if (*d != '.') {
 		*d++ = '.';
diff --git a/dnsd.c b/dnsd.c
index 21d51fd..562831f 100644
--- a/dnsd.c
+++ b/dnsd.c
@@ -290,8 +290,9 @@ dnsd_read(int fd, char *buf, int buflen)
 					packetlen++;
 				}
 				dnsd_respond(fd, id, from);
-				if (lastblock && packetlen == 0) {
+				if (lastblock && packetlen < 2) {
 					// Skipping ping packet
+					packetlen = 0;
 					return 0;
 				}
 				if (lastblock) {