fix ip/hosts file line calc

Author: cckuailong

README.md

@@ -147,6 +147,9 @@ Simple Nginx Web Page.
 
 ## ChangeLog
 
+v0.2.3
+- Fix the bug of wrong calculation of file line number
+
 v0.2.2
 - The -i option supports IP range scanning, such as 1.2.3.4/24
 - The -p option supports custom scan ports, such as 80,8000-8009

README_zh.md

@@ -151,6 +151,9 @@ server {
 
 ## ChangeLog
 
+v0.2.3
+- 修复文件行数计算错误的BUG
+
 v0.2.2
 - -i选项支持IP段扫描，1.2.3.4/24
 - -p选项支持自定义扫描端口，如 80,8000-8009

utils/file.go

@@ -13,18 +13,24 @@ func LineCounter(filepath string) (int, error) {
 		return 0, err
 	}
 	defer f.Close()
-	buf := make([]byte, 32*1024)
 	count := 0
 	lineSep := []byte{'\n'}
+	var last_buf []byte
 
 	for {
+		buf := make([]byte, 32*1024)
 		switch c, err := f.Read(buf[:]); true {
 		case c < 0:
 			return 0, err
 		case c == 0: // EOF
+			last_buf = bytes.Trim(last_buf, "\x00")
+			if last_buf[len(last_buf)-1] != lineSep[0]{
+				count += 1
+			}
 			return count, nil
 		case c > 0:
 			count += bytes.Count(buf[:c], lineSep)
+			last_buf = buf
 		}
 	}