fix: check wry size before read header

zu1k · Oct 20, 2022 · fc9df8d · fc9df8d
1 parent 2da7512
commit fc9df8d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/qqwry/qqwry.go b/pkg/qqwry/qqwry.go
@@ -45,6 +45,10 @@ func NewQQwry(filePath string) (*QQwry, error) {
 		}
 	}
 
+	if len(fileData) < 8 {
+		log.Fatalln("纯真 IP 库存在错误，请重新下载")
+	}
+
 	header := fileData[0:8]
 	start := binary.LittleEndian.Uint32(header[:4])
 	end := binary.LittleEndian.Uint32(header[4:])

diff --git a/pkg/zxipv6wry/zxipv6wry.go b/pkg/zxipv6wry/zxipv6wry.go
@@ -39,6 +39,10 @@ func NewZXwry(filePath string) (*ZXwry, error) {
 		}
 	}
 
+	if len(fileData) < 24 {
+		log.Fatalln("ZX IPv6数据库存在错误，请重新下载")
+	}
+
 	header := fileData[:24]
 	offLen := header[6]
 	ipLen := header[7]