fix

vcv88 · Sep 6, 2023 · 20dc35c · 20dc35c
1 parent 44e6ebc
commit 20dc35c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gorcon.go b/gorcon.go
@@ -98,6 +98,9 @@ func (client *RCONClient) Login(password string) error {
 	switch packet.Type {
 	case PacketCommand:
 		if packet.RequestID != client.RequestID {
+			if err = client.Close(); err != nil {
+				return err
+			}
 			return InvalidPassword
 		}
 	}
@@ -119,6 +122,9 @@ func (client *RCONClient) SendCommand(command string) (string, error) {
 		if err != nil {
 			return res, err
 		}
+		if packet.Type != PacketResponse {
+			continue
+		}
 		if len(packet.Data) < PacketChunkLength {
 			break
 		}