docs(readme): update README (#21)

README.md

@@ -143,8 +143,8 @@ example.
 
 #### Multi-client
 
-RevSuit supports multiple clients, and each client in the connected state receives a push of `flag`, so distributed
-scanning can be supported.
+As shown above, RevSuit supports multiple clients, and each client in the connected state receives a push of `flag`, so
+distributed scanning can be supported.
 
 #### Temporary storage queue
 
@@ -213,7 +213,7 @@ the port scan.
 
 The running effect is as follows:
 
-![ftp-scan](./images/ftp-scan.gif)
+![ftp-scan](images/ftp_scan.gif)
 
 It because the FTP connection will be crashed if `Passive Address` is not accessible, we can determine whether the port
 is open or not based on whether the connection exits normally.For this example, we successfully detected that port 8005
@@ -236,7 +236,7 @@ Submit an issue or contact me through Weixin: `Li4n06`
 
 ### Specifically
 
-Thanks to my friend [@E99plant](https://github.com/wuhan005/) for all the help and advice I received during the
+Thanks to my friend [@E99p1ant](https://github.com/wuhan005/) for all the help and advice I received during the
 development of this project.
 
 ## License

pkg/dns/record.go

@@ -47,16 +47,17 @@ func ListRecords(c *gin.Context) {
 		res       []Record
 		count     int64
 		order     = c.Query("order")
-		pageSize  int
+		pageSize  = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
+
 	if err := c.ShouldBind(&dnsRecord); err != nil {
 		c.JSON(400, gin.H{
 			"status": "failed",

pkg/dns/rule.go

@@ -80,14 +80,14 @@ func ListRules(c *gin.Context) {
 		res      []Rule
 		count    int64
 		order    = c.Query("order")
-		pageSize int
+		pageSize = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/ftp/record.go

@@ -59,14 +59,14 @@ func ListRecords(c *gin.Context) {
 		res       []Record
 		count     int64
 		order     = c.Query("order")
-		pageSize  int
+		pageSize  = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/ftp/rule.go

@@ -75,14 +75,14 @@ func ListRules(c *gin.Context) {
 		res      []Rule
 		count    int64
 		order    = c.Query("order")
-		pageSize int
+		pageSize = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/mysql/record.go

@@ -57,14 +57,14 @@ func ListRecords(c *gin.Context) {
 		res         []Record
 		count       int64
 		order       = c.Query("order")
-		pageSize    int
+		pageSize    = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/mysql/rule.go

@@ -63,14 +63,14 @@ func ListRules(c *gin.Context) {
 		res       []Rule
 		count     int64
 		order     = c.Query("order")
-		pageSize  int
+		pageSize  = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/rhttp/record.go

@@ -50,15 +50,14 @@ func ListRecords(c *gin.Context) {
 		res        []Record
 		count      int64
 		order      = c.Query("order")
-		pageSize   int
+		pageSize   = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		pageSize = n
-		if pageSize <= 0 || pageSize > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/rhttp/rule.go

@@ -81,14 +81,14 @@ func ListRules(c *gin.Context) {
 		res      []Rule
 		count    int64
 		order    = c.Query("order")
-		pageSize int
+		pageSize = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/rmi/record.go

@@ -46,14 +46,14 @@ func ListRecords(c *gin.Context) {
 		res       []Record
 		count     int64
 		order     = c.Query("order")
-		pageSize  int
+		pageSize  = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}
 

pkg/rmi/rmi.go

@@ -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
 	}

pkg/rmi/rule.go

@@ -72,14 +72,14 @@ func ListRules(c *gin.Context) {
 		res      []Rule
 		count    int64
 		order    = c.Query("order")
-		pageSize int
+		pageSize = 10
 	)
 
-	if c.Query("pageSize") == "" {
-		pageSize = 10
-	} else if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
-		if n <= 0 || n > 100 {
-			pageSize = 10
+	if c.Query("pageSize") != "" {
+		if n, err := strconv.Atoi(c.Query("pageSize")); err == nil {
+			if n > 0 && n < 100 {
+				pageSize = n
+			}
 		}
 	}