From 27c7a7fd2b659666c082948a83997899834df96b Mon Sep 17 00:00:00 2001 From: qiwentaidi <104491549+qiwentaidi@users.noreply.github.com> Date: Thu, 2 Jan 2025 00:28:10 +0800 Subject: [PATCH] update 2.0.0 --- core/isic/bing.go | 22 +-- core/portscan/socket.go | 82 -------- core/webscan/infoscan.go | 3 + frontend/src/App.vue | 2 +- frontend/src/components/About.vue | 7 +- frontend/src/components/Exp/Hikvision.vue | 3 +- frontend/src/components/Sidebar.vue | 2 +- frontend/src/components/Titlebar.vue | 2 +- frontend/src/components/Update.vue | 2 +- frontend/src/config.ts | 5 +- frontend/src/i18n/zh_CN.ts | 2 +- frontend/src/linkage/index.ts | 2 +- frontend/src/main.ts | 2 +- frontend/src/{global => stores}/index.ts | 2 +- frontend/src/stores/interface.ts | 12 -- frontend/src/stores/style.ts | 2 +- frontend/src/style/dark.css | 4 + frontend/src/util.ts | 8 +- frontend/src/views/AppStarter.vue | 2 +- frontend/src/views/Asset/Company.vue | 2 +- frontend/src/views/Asset/ISICollection.vue | 2 +- frontend/src/views/Asset/Subdomain.vue | 4 +- frontend/src/views/Permeation/Dirsearch.vue | 2 +- frontend/src/views/Permeation/Jsfinder.vue | 6 +- frontend/src/views/Permeation/Webscan.vue | 66 +++---- frontend/src/views/PocManagement.vue | 2 +- frontend/src/views/Settings.vue | 5 +- frontend/src/views/SpaceEngine/AgentPool.vue | 2 +- frontend/src/views/SpaceEngine/FOFA.vue | 6 +- frontend/src/views/SpaceEngine/Hunter.vue | 6 +- .../src/views/SpaceEngine/Polymerization.vue | 2 +- frontend/src/views/SpaceEngine/Quake.vue | 6 +- frontend/src/views/Tools/DataHanding.vue | 176 +++++++++++++----- frontend/src/views/Tools/ExtractDbInfo.vue | 2 +- frontend/src/views/Tools/Memo.vue | 2 +- frontend/wailsjs/go/services/Database.d.ts | 2 + frontend/wailsjs/go/services/Database.js | 4 + services/database.go | 36 ++++ 38 files changed, 263 insertions(+), 236 deletions(-) delete mode 100644 core/portscan/socket.go rename frontend/src/{global => stores}/index.ts (98%) diff --git a/core/isic/bing.go b/core/isic/bing.go index af55fdb7..d8aaaf21 100644 --- a/core/isic/bing.go +++ b/core/isic/bing.go @@ -2,36 +2,20 @@ package isic import ( "bytes" - "net/http" + "fmt" "slack-wails/lib/clients" "strings" "github.com/PuerkitoBio/goquery" ) -// 根据网络环境,选择www.bing.com 还是cn.bing.com - -func chooseBingEnvironment() (string, error) { - resp, _, err := clients.NewSimpleGetRequest("https://www.bing.com", clients.NotFollowClient()) - if err != nil || resp == nil { - return "", err - } - if resp.StatusCode == http.StatusFound { - return "https://cn.bing.com", nil - } - return "https://www.bing.com", nil -} - func BingSearch(query string) (string, error) { - bingUrl, err := chooseBingEnvironment() - if err != nil { - return "", err - } - link := bingUrl + "/search?q=site%3a+" + strings.TrimSpace(query) + link := "https://www.bing.com/search?q=site%3a+" + strings.TrimSpace(query) _, body, err := clients.NewSimpleGetRequest(link, clients.DefaultClient()) if err != nil { return "请求失败", err } + fmt.Printf("string(body): %v\n", string(body)) doc, err := goquery.NewDocumentFromReader(bytes.NewReader(body)) if err != nil { return "解析网站失败", err diff --git a/core/portscan/socket.go b/core/portscan/socket.go deleted file mode 100644 index c5863c60..00000000 --- a/core/portscan/socket.go +++ /dev/null @@ -1,82 +0,0 @@ -// https://github.com/chainreactors/gogo/blob/master/v2/pkg/socket.go -package portscan - -import ( - "net" - "time" -) - -func NewSocket(network, target string, delay int) (*Socket, error) { - s := &Socket{ - Timeout: time.Duration(delay) * time.Second, - } - var conn net.Conn - var err error - if ProxyDialTimeout != nil { - conn, err = ProxyDialTimeout(network, target, s.Timeout) - } else { - conn, err = net.DialTimeout(network, target, s.Timeout) - } - if err != nil { - return nil, err - } - - s.Conn = conn - return s, nil -} - -type Socket struct { - Conn net.Conn - Count int - Timeout time.Duration -} - -func (s *Socket) Read(timeout int) ([]byte, error) { - buf := make([]byte, 16384) - s.Conn.SetReadDeadline(time.Now().Add(time.Duration(timeout) * time.Second)) - n, err := s.Conn.Read(buf) - if err != nil { - return nil, err - } - return buf[:n], nil -} - -func (s *Socket) Request(data []byte, max int) ([]byte, error) { - _ = s.Conn.SetDeadline(time.Now().Add(s.Timeout)) - var err error - _, err = s.Conn.Write(data) - if err != nil { - return []byte{}, err - } - s.Count++ - buf := make([]byte, max) - time.Sleep(time.Duration(500) * time.Millisecond) - n, err := s.Conn.Read(buf) - if err != nil { - return []byte{}, err - } - return buf[:n], err -} - -func (s *Socket) QuickRequest(data []byte, max int) ([]byte, error) { - // read small data, without wait for 500ms - _ = s.Conn.SetDeadline(time.Now().Add(s.Timeout)) - var err error - _, err = s.Conn.Write(data) - if err != nil { - return []byte{}, err - } - s.Count++ - buf := make([]byte, max) - n, err := s.Conn.Read(buf) - if err != nil { - return []byte{}, err - } - return buf[:n], err -} - -func (s *Socket) Close() error { - return s.Conn.Close() -} - -var ProxyDialTimeout func(network, address string, timeout time.Duration) (net.Conn, error) diff --git a/core/webscan/infoscan.go b/core/webscan/infoscan.go index 9d5c7ece..a6f29641 100644 --- a/core/webscan/infoscan.go +++ b/core/webscan/infoscan.go @@ -320,6 +320,9 @@ func (s *FingerScanner) NewActiveFingerScan(rootPath bool) { Title: ti.Title, Fingerprints: []string{fp.Fpe[0].ProductName}, Detect: "Active", + Port: ti.StatusCode, + Scheme: fp.URL.Scheme, + Host: fp.URL.Host, } } }) diff --git a/frontend/src/App.vue b/frontend/src/App.vue index 423b7b07..20a8144e 100644 --- a/frontend/src/App.vue +++ b/frontend/src/App.vue @@ -1,6 +1,6 @@