刘河 6 years ago
parent
commit
6c3c09428e
2 changed files with 2 additions and 2 deletions
  1. 1 1
      utils/conn.go
  2. 1 1
      utils/util.go

+ 1 - 1
utils/conn.go

@@ -113,7 +113,7 @@ func (s *SnappyConn) Read(b []byte) (n int, err error) {
 			n = 0
 		}
 	}()
-	if n, err = s.r.Read(b); err != nil {
+	if n, err = s.r.Read(b); err != nil || err == io.EOF {
 		return
 	}
 	if s.crypt {

+ 1 - 1
utils/util.go

@@ -153,7 +153,7 @@ func GetIntNoerrByStr(str string) int {
 
 var bufPool = sync.Pool{
 	New: func() interface{} {
-		return make([]byte, 65535)
+		return make([]byte, 65536)
 	},
 }
 // io.copy的优化版,读取buffer长度原为32*1024,与snappy不同,导致读取出的内容存在差异,不利于解密,特此修改