Invalid AWN Request Fixer

pkg/ambient/ambienthttp/ambienthttp.go

@@ -32,8 +32,9 @@ type LFStrippingListener struct {
 }
 
 type LFStrippingConn struct {
-	net.Conn
+	ctx    context.Context
 	reader io.Reader
+	net.Conn
 }
 
 func (l *LFStrippingListener) WrapConn(conn net.Conn) net.Conn {
@@ -49,34 +50,47 @@ func (l *LFStrippingListener) WrapConn(conn net.Conn) net.Conn {
 		// Only restore newline if not a bad request
 		if !badReqURI.Match(line) {
 			newData = append(newData, '\n')
+		} else {
+			zerolog.Ctx(l.ctx).Trace().Bytes("line", line).
+				Msg("malformed request found, stripped 0x0a")
+		}
+
+		if len(line) == 0 {
+			break
 		}
 	}
 	if scanner.Err() != nil {
 		zerolog.Ctx(l.ctx).Err(scanner.Err()).Send()
 	}
 
+	zerolog.Ctx(l.ctx).Trace().
+		Int("numBytes", len(newData)).
+		Bytes("request", newData).
+		Msg("stripping conn complete")
+
 	// Use a multi-reader to prepend the modified request
 	finalReader := io.MultiReader(bytes.NewReader(newData), conn)
 
 	return &LFStrippingConn{
 		Conn:   conn,
+		ctx:    l.ctx,
 		reader: finalReader,
 	}
 }
 
-func NewLFStrippingListener(ctx context.Context, listen string) net.Listener {
-	// Create the underlying TCP listener
+func NewAWNMutatingListener(ctx context.Context, listen string) net.Listener {
 	rawListener, err := net.Listen("tcp", listen)
 	if err != nil {
 		panic(err)
 	}
+
+	// Encapsulate the raw listener with ours
 	return &LFStrippingListener{
 		Listener: rawListener,
 		ctx:      ctx,
 	}
 }
 
-// Accept waits for and returns the next connection to the listener.
 func (l *LFStrippingListener) Accept() (net.Conn, error) {
 	conn, err := l.Listener.Accept()
 	if err != nil {
@@ -85,13 +99,10 @@ func (l *LFStrippingListener) Accept() (net.Conn, error) {
 	return l.WrapConn(conn), nil
 }
 
-// Close closes the listener.
-// Any blocked Accept operations will be unblocked and return errors.
 func (l *LFStrippingListener) Close() error {
 	return l.Listener.Close()
 }
 
-// Addr returns the listener's network address.
 func (l *LFStrippingListener) Addr() net.Addr {
 	return l.Listener.Addr()
 }