优化：nmap未自动加.exe的bug；优化windows下缓存文件未优化体积的bug

Author: x51pwn

config/config.json

@@ -1,5 +1,6 @@
 {
   "CacheName": ".DbCache",
+  "autoRmCache": "true",
   "ssh_username": "pkg/hydra/dicts/ssh_user.txt",
   "ssh_pswd": "pkg/hydra/dicts/ssh_pswd.txt",
   "ssh_default": "pkg/hydra/dicts/ssh_default.txt",

main.go

@@ -5,10 +5,17 @@ import (
 	"github.com/hktalent/scan4all/pkg/hydra"
 	naaburunner "github.com/hktalent/scan4all/pkg/naabu/v2/pkg/runner"
 	"github.com/projectdiscovery/gologger"
+	"os"
 	"runtime"
 )
 
 func main() {
+	defer func() {
+		pkg.Cache1.Close()
+		if "true" == pkg.GetVal("autoRmCache") {
+			os.RemoveAll(pkg.GetVal(pkg.CacheName))
+		}
+	}()
 	options := naaburunner.ParseOptions()
 	pkg.G_Options = options
 	if runtime.GOOS == "windows" {

pkg/kvDb.go

@@ -34,6 +34,9 @@ func (r *KvDbOp) SetExpiresAt(ExpiresAt uint64) {
 
 func (r *KvDbOp) Init(szDb string) error {
 	opts := badger.DefaultOptions(szDb)
+	opts.CompactL0OnClose = true
+	opts.LevelOneSize = 256 << 10
+	opts.LevelSizeMultiplier = 20
 	db, err := badger.Open(opts)
 	if nil != err {
 		log.Println("Init k-v db 不能多个进程同时开启", err)

pkg/naabu/v2/pkg/runner/options.go

@@ -3,6 +3,7 @@ package runner
 import (
 	"github.com/hktalent/scan4all/pkg"
 	"os"
+	"runtime"
 	"strings"
 
 	"github.com/projectdiscovery/fileutil"
@@ -109,6 +110,9 @@ func ParseOptions() *Options {
 	// 读取结果
 	szNmap := pkg.GetVal("nmap")
 	if "" != szNmap {
+		if runtime.GOOS == "windows" {
+			szNmap = strings.Replace(szNmap, "nmap", "nmap.exe", -1)
+		}
 		tempInput := pkg.GetTempFile(pkg.Naabu)
 		if tempInput != nil {
 			szNmap = strings.ReplaceAll(szNmap, "{filename}", tempInput.Name())