[+新增扫描进度跟踪功能，现在会定期输出扫描进度

Author: lcvvvv

kscan_test.go

@@ -98,7 +98,6 @@ func Context() {
 			if err := recover(); err != nil {
 			}
 		}()
-
 		time.Sleep(3 * time.Second)
 		resChan <- ""
 	}()

lib/gonmap

@@ -1 +1 @@
-Subproject commit 99d64e58dbc83919ac0648fcf4ed2f7af79416bc
+Subproject commit b41d2cda02b711ae0a7377185fe925871809cf2e

lib/misc/misc.go

@@ -291,3 +291,19 @@ func CloneIntMap(intMap map[int]string) map[int]string {
 	}
 	return newIntMap
 }
+
+func RandomString(i ...int) string {
+	var length int
+	var str string
+	if len(i) != 1 {
+		length = 32
+	} else {
+		length = i[0]
+	}
+	Char := "01234567890abcdef"
+	for range Xrange(length) {
+		j := rand.Intn(len(Char) - 1)
+		str += Char[j : j+1]
+	}
+	return str
+}

lib/pool/pool.go

@@ -3,7 +3,9 @@ package pool
 import (
 	"errors"
 	"fmt"
+	"kscan/lib/misc"
 	"kscan/lib/slog"
+	"kscan/lib/smap"
 	"sync"
 )
 
@@ -43,6 +45,8 @@ type Pool struct {
 	Out chan interface{}
 	//size用来表明池的大小，不能超发。
 	threads int
+	//正在执行的任务清单
+	JobsList *smap.SMap
 	//jobs表示执行任务的通道用于作为队列，我们将任务从切片当中取出来，然后存放到通道当中，再从通道当中取出任务并执行。
 	Jobs chan *Worker
 	//用于阻塞
@@ -55,6 +59,7 @@ type Pool struct {
 func NewPool(threads int) *Pool {
 	return &Pool{
 		threads:  threads,
+		JobsList: smap.New(),
 		wg:       &sync.WaitGroup{},
 		Out:      make(chan interface{}),
 		In:       make(chan interface{}),
@@ -66,13 +71,23 @@ func NewPool(threads int) *Pool {
 //从jobs当中取出任务并执行。
 func (p *Pool) work() {
 	//减少waitGroup计数器的值
-	defer func() { p.wg.Done() }()
+	defer func() {
+		p.wg.Done()
+	}()
 	for param := range p.In {
 		if p.Done {
 			return
 		}
+		//获取任务唯一票据
+		Tick := p.NewTick()
+		//压入工作任务到工作清单
+		p.JobsList.Set(Tick, param)
+		//开始工作
 		f := NewWorker(p.Function)
+		//开始工作，输出工作结果
 		out, err := f.Run(param)
+		//工作结束，删除工作清单
+		p.JobsList.Delete(Tick)
 		if err == nil && out != nil {
 			p.Out <- out
 		}
@@ -108,3 +123,8 @@ func (p *Pool) OutDone() {
 func (p *Pool) Stop() {
 	p.Done = true
 }
+
+//生成工作票据
+func (p *Pool) NewTick() string {
+	return misc.RandomString()
+}

lib/smap/smap.go

@@ -0,0 +1,64 @@
+package smap
+
+import (
+	"sync"
+)
+
+type SMap struct {
+	value sync.Map
+}
+
+func New() *SMap {
+	return &SMap{sync.Map{}}
+}
+
+func (s *SMap) Set(key interface{}, value interface{}) {
+	s.value.Store(key, value)
+}
+
+func (s *SMap) Get(key interface{}) (value interface{}, ok bool) {
+	return s.value.Load(key)
+}
+
+func (s *SMap) Delete(key interface{}) {
+	s.value.Delete(key)
+}
+
+func (s *SMap) Length() int {
+	var i int
+	f := func(key, value interface{}) bool {
+		i = i + 1
+		return true
+	}
+	s.value.Range(f)
+	return i
+}
+
+func (s *SMap) Exist(key interface{}) bool {
+	if _, ok := s.value.Load(key); ok {
+		return true
+	}
+	return false
+}
+
+func (s *SMap) Peek() interface{} {
+	var i interface{}
+	f := func(key, value interface{}) bool {
+		i = value
+		return false
+	}
+	s.value.Range(f)
+	return i
+}
+
+func (s *SMap) Range(f func(key interface{}, value interface{}) bool) {
+	s.value.Range(f)
+}
+
+//func handler(key, value interface{}) bool {
+//	fmt.Printf("Name :%s %s\n", key, value)
+//	return true
+//}
+
+//遍历，传入一个函数，遍历的时候函数返回false则停止遍历
+//s.value.Range(handler)

lib/smap/smap_test.go

@@ -0,0 +1,19 @@
+package smap
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestSMap_Length(t *testing.T) {
+	s := New()
+	s.Set("aaaa", "bbbb")
+	s.Set("aaaa", "bbbb")
+	s.Set("aaaa", "bbbb")
+	s.Set("aaaa", "bbbb")
+	s.Set("aaaa", "bbbb")
+	s.Set("BBBB", "bbbb")
+	s.Delete("BBBB")
+	s.Delete("cccc")
+	fmt.Println(s.Peek())
+}

run/type-kscan.go

@@ -16,6 +16,7 @@ import (
 	"path"
 	"strings"
 	"sync"
+	"time"
 )
 
 type kscan struct {
@@ -29,9 +30,10 @@ type kscan struct {
 		appBanner *pool.Pool
 	}
 	watchDog struct {
-		output chan interface{}
-		hydra  chan interface{}
-		wg     *sync.WaitGroup
+		output  chan interface{}
+		hydra   chan interface{}
+		wg      *sync.WaitGroup
+		trigger bool
 	}
 	hydra struct {
 		pool  *pool.Pool
@@ -61,6 +63,7 @@ func New(config app.Config) *kscan {
 	k.watchDog.hydra = make(chan interface{})
 	k.watchDog.output = make(chan interface{})
 	k.watchDog.wg = &sync.WaitGroup{}
+	k.watchDog.trigger = false
 
 	k.hydra.pool = pool.NewPool(10)
 	k.hydra.queue = queue.New()
@@ -258,6 +261,9 @@ func (k *kscan) Output() {
 		}
 		var disp string
 		var write string
+		//打开触发器,若长时间无输出，触发器会输出进度
+		k.watchDog.trigger = true
+		//输出结果
 		switch out.(type) {
 		case *gonmap.AppBanner:
 			banner := out.(*gonmap.AppBanner)
@@ -299,11 +305,61 @@ func (k *kscan) Output() {
 func (k *kscan) WatchDog() {
 
 	k.watchDog.wg.Add(1)
-
+	//触发器校准，每隔60秒会将触发器关闭
+	go func() {
+		for true {
+			time.Sleep(60 * time.Second)
+			k.watchDog.trigger = false
+		}
+	}()
+	//轮询触发器，每隔一段时间会检测触发器是否打开
+	go func() {
+		for true {
+			time.Sleep(59 * time.Second)
+			if k.watchDog.trigger == false {
+				if num := k.pool.host.JobsList.Length(); num > 0 {
+					i := k.pool.host.JobsList.Peek()
+					info := i.(string)
+					slog.Warningf("当前主机存活性检测任务未完成，其并发协程数为：%d，具体其中的一个协程信息为：%s", num, info)
+					continue
+				}
+				if num := k.pool.port.JobsList.Length(); num > 0 {
+					i := k.pool.port.JobsList.Peek()
+					info := i.(string)
+					slog.Warningf("当前端口存活性检测任务未完成，其并发协程数为：%d，具体其中的一个协程信息为：%s", num, info)
+					continue
+				}
+				if num := k.pool.tcpBanner.JobsList.Length(); num > 0 {
+					i := k.pool.tcpBanner.JobsList.Peek()
+					info := i.(string)
+					slog.Warningf("当前TCP层指纹识别任务未完成，其并发协程数为：%d，具体其中的一个协程信息为：%s", num, info)
+					continue
+				}
+				if num := k.pool.appBanner.JobsList.Length(); num > 0 {
+					i := k.pool.appBanner.JobsList.Peek()
+					var info string
+					switch i.(type) {
+					case string:
+						info = i.(string)
+					case *gonmap.TcpBanner:
+						tcpBanner := i.(*gonmap.TcpBanner)
+						if tcpBanner == nil {
+							continue
+						}
+						info = tcpBanner.Target.URI()
+					}
+					slog.Warningf("当前应用层指纹检测并发协程数为：%d，具体其中的一个协程信息为：%s", num, info)
+					continue
+				}
+			}
+		}
+	}()
+	//Hydra模块
 	if app.Setting.Hydra {
 		slog.Info("hydra模块已开启，开始监听暴力破解任务")
 		k.watchDog.wg.Add(1)
 	}
+
 	for out := range k.pool.appBanner.Out {
 		k.watchDog.output <- out
 		if app.Setting.Hydra {