From 0a8e5f5b728fd1053a84b3e6522fd1cd6fa882e4 Mon Sep 17 00:00:00 2001
From: David Fifield <david@bamsoftware.com>
Date: Sat, 28 Mar 2020 09:57:55 -0600
Subject: [PATCH] Packet framing downstream.

---
 dnstt-server/main.go | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/dnstt-server/main.go b/dnstt-server/main.go
index 0251055..e73cab3 100644
--- a/dnstt-server/main.go
+++ b/dnstt-server/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"bytes"
 	"encoding/base32"
+	"encoding/binary"
 	"flag"
 	"fmt"
 	"io"
@@ -313,16 +314,23 @@ func sendLoop(dnsConn net.PacketConn, ttConn *turbotunnel.QueuePacketConn, ch <-
 			// If it's a non-error response, we can fill the Answer
 			// section with downstream packets.
 			// TODO: can bundle multiple packets here.
+			var buf bytes.Buffer
 			select {
 			case p := <-ttConn.OutgoingQueue(rec.ClientID):
-				rec.Resp.Answer = append(rec.Resp.Answer, dns.RR{
-					Name: rec.Resp.Question[0].Name,
-					Type: dns.RRTypeTXT,
-					TTL:  responseTTL,
-					Data: dns.EncodeRDataTXT(p),
-				})
+				n := uint16(len(p))
+				if int(n) != len(p) {
+					panic(len(p))
+				}
+				binary.Write(&buf, binary.BigEndian, n)
+				buf.Write(p)
 			default:
 			}
+			rec.Resp.Answer = append(rec.Resp.Answer, dns.RR{
+				Name: rec.Resp.Question[0].Name,
+				Type: dns.RRTypeTXT,
+				TTL:  responseTTL,
+				Data: dns.EncodeRDataTXT(buf.Bytes()),
+			})
 		}
 		buf, err := rec.Resp.WireFormat()
 		if err != nil {