archive/revsuit
1
0
Fork 0
mirror of https://github.com/Li4n0/revsuit.git synced 2026-09-26 00:21:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix the bug of page limit (#20)

Browse Source
This commit is contained in:
Li4n0
2021-05-26 00:09:57 +08:00
committed by GitHub
parent cbbf3ba0ba
commit b44e4891a6
17 changed files with 67 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/rmi/rmi.go
+4
View File
@@ -58,6 +58,7 @@ func (s *Server) handleConnection(conn net.Conn) {
if err := conn.SetDeadline(time.Now().Add(time.Second * 30)); err != nil {
log.Warn("RMI set connection deadline error:%v", err)
return
}
ip, port, _ := net.SplitHostPort(conn.RemoteAddr().String())
@@ -66,6 +67,7 @@ func (s *Server) handleConnection(conn net.Conn) {
_, err := conn.Read(buf)
if err != nil {
log.Warn("RMI read connection error:%v", err)
return
}
if !bytes.Contains(buf, []byte{0x4a, 0x52, 0x4d, 0x49}) {
@@ -86,6 +88,7 @@ func (s *Server) handleConnection(conn net.Conn) {
_, err = conn.Write(send)
if err != nil {
log.Warn("RMI write connection error: %v", err)
return
}
data := make([]byte, 512)
@@ -94,6 +97,7 @@ func (s *Server) handleConnection(conn net.Conn) {
n, err := conn.Read(data)
if err != nil {
log.Warn("RMI read connection error: %v", err)
return
}
length += n
}