From 98ec4443538705291a55f86a9a05096bafac4648 Mon Sep 17 00:00:00 2001
From: Konstantin Pereiaslov <perk11@perk11.info>
Date: Tue, 6 Aug 2024 01:03:30 -0500
Subject: [PATCH] Add a helper package for future tests

---
 test/listen-send-pid-close-connection/main.go | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 test/listen-send-pid-close-connection/main.go

diff --git a/test/listen-send-pid-close-connection/main.go b/test/listen-send-pid-close-connection/main.go
new file mode 100644
index 0000000..3596c35
--- /dev/null
+++ b/test/listen-send-pid-close-connection/main.go
@@ -0,0 +1,52 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net"
+	"os"
+)
+
+func main() {
+	port := flag.String("p", "", "Port to listen on (required)")
+	flag.Parse()
+
+	if *port == "" {
+		fmt.Println("The -p parameter is required to specify the port")
+		flag.Usage() // Print usage information
+		os.Exit(1)
+	}
+
+	listener, err := net.Listen("tcp", ":"+*port)
+	if err != nil {
+		fmt.Println("Error listening:", err.Error())
+		os.Exit(1)
+	}
+	defer func(listener net.Listener) {
+		err := listener.Close()
+		if err != nil {
+			fmt.Println("Failed to stop listening: ", err.Error())
+		}
+	}(listener)
+	fmt.Printf("Listening on port %s\n", *port)
+
+	// Handle incoming connections
+	for {
+		conn, err := listener.Accept()
+		if err != nil {
+			fmt.Println("Error accepting: ", err.Error())
+			os.Exit(1)
+		}
+		fmt.Println("Connection received.")
+		pid := os.Getpid()
+		_, writeErr := conn.Write([]byte(fmt.Sprintf("%d", pid)))
+		if writeErr != nil {
+			fmt.Println("Error writing to connection: ", writeErr.Error())
+		}
+
+		connectionCloseErr := conn.Close()
+		if connectionCloseErr != nil {
+			fmt.Println("Error closing connection: ", connectionCloseErr.Error())
+		}
+	}
+}