Fix panic

Reported by Tomás Aparicio. Closed #15
2012-05-02 22:54:57 +02:00 · 2012-05-02 22:54:57 +02:00 · ac713d3128
parent 3b4800cbbf
commit ac713d3128
1 changed files with 2 additions and 2 deletions
--- a/server.go
+++ b/server.go
@ -231,7 +231,7 @@ forever:
 		}
 		m := make([]byte, int(length))
 		n, err = rw.Read(m[:int(length)])
-		if err != nil {
+		if err != nil || n == 0 {
 			continue
 		}
 		i := n
@ -267,7 +267,7 @@ func (srv *Server) ServeUDP(l *net.UDPConn) error {
 	for {
 		m := make([]byte, srv.UDPSize)
 		n, a, e := l.ReadFromUDP(m)
-		if e != nil {
+		if e != nil || n == 0 {
 			return e
 		}
 		m = m[:n]