diff --git a/cmd/smoketest/main.go b/cmd/smoketest/main.go index 3b86a2e..10f924b 100644 --- a/cmd/smoketest/main.go +++ b/cmd/smoketest/main.go @@ -2,19 +2,21 @@ package main import ( "fmt" + "io" + "log" "net" "os" ) const ( connPort = "3030" - connType = "tcp" + connType = "tcp4" ) func main() { l, err := net.Listen(connType, ":"+connPort) if err != nil { - fmt.Printf("net.Listen fail: %v\n", err) + log.Fatalln(err) os.Exit(1) } defer l.Close() @@ -24,19 +26,15 @@ func main() { for { conn, err := l.Accept() if err != nil { - fmt.Printf("Error accepting: %v\n", err) + log.Fatalln(err) } go handleRequest(conn) } } func handleRequest(conn net.Conn) { - buf := make([]byte, 1024) - _, err := conn.Read(buf) - fmt.Printf("read in %s.\nSending it back\n", string(buf)) - if err != nil { - fmt.Printf("Error reading: %v\n", err) + defer conn.Close() + if _, err := io.Copy(conn, conn); err != nil { + log.Fatalln(err) } - conn.Write(buf) - conn.Close() }