diff --git a/net/rcon.go b/net/rcon.go index f0960c7..651b7e8 100644 --- a/net/rcon.go +++ b/net/rcon.go @@ -177,12 +177,14 @@ func (r *RCONConn) RespCmd(resp string) error { type RCONClientConn interface { Cmd(cmd string) error Resp() (resp string, err error) + Close() error } type RCONServerConn interface { AcceptLogin(password string) error AcceptCmd() (cmd string, err error) RespCmd(resp string) error + Close() error } func ListenRCON(addr string) (*RCONListener, error) { diff --git a/net/rcon_test.go b/net/rcon_test.go index a93a0e3..e2a45bc 100644 --- a/net/rcon_test.go +++ b/net/rcon_test.go @@ -17,6 +17,8 @@ func server(t *testing.T, prepare chan<- int) { if err != nil { t.Fatal(err) } + defer l.Close() + prepare <- 1 for { @@ -55,6 +57,7 @@ func client(t *testing.T) { if err != nil { t.Fatal(err) } + defer conn.Close() err = conn.Cmd("TEST COMMAND") if err != nil {