diff --git a/send-command/main.go b/send-command/main.go index 90965b9..c8628bb 100644 --- a/send-command/main.go +++ b/send-command/main.go @@ -3,6 +3,7 @@ package main import ( "errors" "fmt" + "io" "io/fs" "log" "net" @@ -37,7 +38,9 @@ func main() { for { buffer := make([]byte, 1024) n, err := os.Stdin.Read(buffer) - if err != nil { + if err == io.EOF { + break + } else if err != nil { log.Fatal(err) } ln.Write(buffer[:n]) @@ -47,7 +50,9 @@ func main() { for { buffer := make([]byte, 1024) n, err := ln.Read(buffer) - if err != nil { + if err == io.EOF { + break + } else if err != nil { log.Fatal(err) } os.Stdout.Write(buffer[:n])