dns

commit a8959a402c8c0f2557c2a365ab5b3d83c2cfa11d
parent 88787149cf89ee12e50156d2be193b2194e4052f
Author: Oliver Lowe <o@olowe.co>
Date:   Wed, 17 Nov 2021 18:50:47 +1100

oops

Diffstat:
M
dns.go
 | 
6
++++--

1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/dns.go b/dns.go
@@ -40,15 +40,17 @@ func send(msg dnsmessage.Message, conn net.Conn) (dnsmessage.Message, error) {
 	if err != nil {
 		return dnsmessage.Message{}, err
 	}
+	var b []byte
 	if _, ok := conn.(net.PacketConn); ok {
-		b, err := dnsPacketExchange(packed, conn)
+		b, err = dnsPacketExchange(packed, conn)
 		if err != nil {
 			return dnsmessage.Message{}, fmt.Errorf("exchange DNS packet: %v", err)
 		}
 	} else {
-		b, err := dnsStreamExchange(packed, conn)
+		b, err = dnsStreamExchange(packed, conn)
 		if err != nil {
 			return dnsmessage.Message{}, fmt.Errorf("exchange DNS TCP stream: %v", err)
+		}
 	}
 	var rmsg dnsmessage.Message
 	if err := rmsg.Unpack(b); err != nil {