From 9131b0004e7c640cc028179e1d049a4c62210d94 Mon Sep 17 00:00:00 2001
From: James O'Doherty <james@theodohertyfamily.com>
Date: Fri, 22 May 2026 10:46:02 -0400
Subject: Security hardening: prevent shell injection and null-byte crashes,
 implement 8-bit clean argument fuzzing and portable E2E binary discovery

---
 tests/e2e/arg_integrity_test.go | 45 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 tests/e2e/arg_integrity_test.go

(limited to 'tests/e2e/arg_integrity_test.go')

diff --git a/tests/e2e/arg_integrity_test.go b/tests/e2e/arg_integrity_test.go
new file mode 100644
index 0000000..7121c2b
--- /dev/null
+++ b/tests/e2e/arg_integrity_test.go
@@ -0,0 +1,45 @@
+package e2e
+
+import (
+	"fmt"
+	"os/exec"
+	"strings"
+	"testing"
+)
+
+func TestArgumentIntegrity(t *testing.T) {
+	payloads := []string{
+		"$(whoami)",
+		"; rm -rf /",
+		"`id`",
+		"| wall 'hacked'",
+		"\"'\"'\"", // Complex quoting
+		"   spaced   argument   ",
+		"$\nnewline",
+	}
+
+	for _, payload := range payloads {
+		t.Run(fmt.Sprintf("Payload_%s", payload), func(t *testing.T) {
+			binaryPath := GetBinaryPath()
+			cmd := exec.Command(binaryPath, "test-args", payload)
+			out, err := cmd.CombinedOutput()
+			if err != nil {
+				t.Fatalf("wg-wrap test-args failed for payload %s: %v\nOutput: %s", payload, err, string(out))
+			}
+
+			lines := strings.Split(strings.TrimSpace(string(out)), "\n")
+			if len(lines) < 3 {
+				t.Fatalf("Unexpected output format for payload %s\nOutput: %s", payload, string(out))
+			}
+
+			parts := strings.Split(lines[len(lines)-1], ":")
+			if len(parts) < 2 {
+				t.Fatalf("Malformed hex line for payload %s: %s", payload, lines[len(lines)-1])
+			}
+
+			if parts[1] != fmt.Sprintf("%x", payload) {
+				t.Errorf("8-bit mismatch!\nSent Hex: %s\nRecv Hex: %s\nPayload: %q", fmt.Sprintf("%x", payload), parts[1], payload)
+			}
+		})
+	}
+}
-- 
cgit v1.2.3