参考nuclie实现self更新

Author: SummerSec

cmd/commons/core/banner.go

@@ -14,6 +14,9 @@ const banner = `
 																			
 `
 
+// TODO 修改版本号
+const version = "0.0.9"
+
 func ShowBanner(Version string) {
 	log.Println(banner)
 	log.Println("\t\t\tAuthor: SummerSec Version:", Version+" Github: https://Github.com/SummerSec\n")

cmd/commons/core/options.go

@@ -46,6 +46,8 @@ type Options struct {
 	Shell bool
 	// 强制开启HTTP 1.1
 	H1 bool
+	// 更新到最新版本
+	Update bool
 }
 
 func (o Options) toString() interface{} {
@@ -72,12 +74,13 @@ func ParseOptions() *Options {
 	flag.StringVar(&options.IP, "i", "", "ip segment example: -i=192.168.1.1/32")
 	flag.BoolVar(&options.Shell, "shell", false, "whether to enter the interactive shell")
 	flag.BoolVar(&options.H1, "h1", false, "force to use HTTP 1.1")
+	flag.BoolVar(&options.Update, "update", false, "update to the latest version")
 	flag.Parse()
 
 	// TODO 修改版本号
-	v := "0.0.9"
-
-	ShowBanner(v)
+	ShowBanner(version)
+	logs.SaveLogs(options.LogFile)
+	showVerbose(options)
 
 	if options.Version {
 		//ShowBanner(v)
@@ -97,8 +100,6 @@ func ParseOptions() *Options {
 		//ShowBanner(v)
 		flag.PrintDefaults()
 	}
-	showVerbose(options)
-	logs.SaveLogs(options.LogFile)
 
 	return options
 

cmd/commons/core/runner.go

@@ -40,6 +40,8 @@ func (r *Runner) Run() {
 		urls, _ = utils.ReadFile(r.options.File)
 	} else if r.options.IP != "" {
 		urls = utils.GetIPToUrlsLinks(r.options.IP, urls)
+	} else if r.options.Update {
+		selfUpdate()
 	} else {
 		log.Error("No file or url or ips specified")
 		return

cmd/commons/core/update.go

@@ -0,0 +1,66 @@
+package core
+
+import (
+	log "github.com/sirupsen/logrus"
+	"github.com/tj/go-update"
+	"github.com/tj/go-update/progress"
+	githubUpdateStore "github.com/tj/go-update/stores/github"
+	"runtime"
+)
+
+const (
+	Owner = "SummerSec"
+	Repo  = "SpringExploit"
+)
+
+func selfUpdate() {
+	var command string
+	switch runtime.GOOS {
+	case "windows":
+		command = "nuclei.exe"
+	default:
+		command = "nuclei"
+	}
+	m := &update.Manager{
+		Command: command,
+		Store: &githubUpdateStore.Store{
+			Owner:   Owner,
+			Repo:    Repo,
+			Version: version,
+		},
+	}
+
+	releases, err := m.LatestReleases()
+	if err != nil {
+		log.Error("Failed to get releases", err)
+		return
+	}
+	if len(releases) == 0 {
+		log.Info("No updates available")
+		return
+	}
+	latest := releases[0]
+	var currentOS string
+	switch runtime.GOOS {
+	case "darwin":
+		currentOS = "macOS"
+	default:
+		currentOS = runtime.GOOS
+	}
+	final := latest.FindZip(currentOS, runtime.GOARCH)
+	if final == nil {
+		log.Error("No update available for", currentOS, "and", runtime.GOARCH)
+	}
+	tarball, err := final.DownloadProxy(progress.Reader)
+	if err != nil {
+		log.Error("could not install latest release ", err)
+		return
+	}
+	if err := m.Install(tarball); err != nil {
+		log.Error("could not install latest release", err)
+		return
+	}
+
+	log.Infof("Successfully Updated to %s Version %s", Repo, latest.Version)
+
+}

cmd/commons/core/update_test.go

@@ -0,0 +1,16 @@
+package core
+
+import "testing"
+
+func Test_selfUpdate(t *testing.T) {
+	tests := []struct {
+		name string
+	}{
+		// TODO: Add test cases.
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			selfUpdate()
+		})
+	}
+}

cmd/commons/utils/os.go

@@ -0,0 +1,9 @@
+package utils
+
+import "runtime"
+
+// GetOSInfo 获取当前运行的操作系统信息
+func GetOSInfo() string {
+	// 获取操作系统信息
+	return runtime.GOOS + "_" + runtime.GOARCH
+}

cmd/test/emun.go

@@ -1,8 +1,25 @@
 package main
 
-import "github.com/SummerSec/SpringExploit/cmd/commons/attack"
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/imroc/req/v3"
+	"runtime"
+)
 
 func main() {
-	attack.CVE202222963()
+	u := "https://api.github.com/repos/SummerSec/SpringExploit/releases/latest"
+	fmt.Println(u)
+	resp, _ := req.R().Get(u)
+	body, _ := resp.ToString()
+	// body to map
+	fmt.Println(body)
+	var m map[string]interface{}
+	json.Unmarshal([]byte(body), &m)
+	// get release tag
+	tag := m["tag_name"]
+
+	fmt.Println(tag)
+	fmt.Println(runtime.GOOS + "_" + runtime.GOARCH)
 
 }

go.mod

@@ -13,10 +13,17 @@ require (
 )
 
 require (
+	github.com/apex/log v1.9.0 // indirect
 	github.com/c-bata/go-prompt v0.2.6 // indirect
+	github.com/c4milo/unpackit v0.1.0 // indirect
+	github.com/google/go-github v17.0.0+incompatible // indirect
+	github.com/google/go-querystring v1.1.0 // indirect
+	github.com/gosuri/uilive v0.0.4 // indirect
+	github.com/gosuri/uiprogress v0.0.1 // indirect
 	github.com/hashicorp/errwrap v1.0.0 // indirect
 	github.com/hashicorp/go-multierror v1.1.1 // indirect
 	github.com/projectdiscovery/blackrock v0.0.0-20210415162320-b38689ae3a2e // indirect
+	github.com/tj/go-update v2.2.5-0.20200519121640-62b4b798fd68+incompatible
 	golang.org/x/net v0.0.0-20220111093109-d55c255bac03 // indirect
 	golang.org/x/text v0.3.7 // indirect
 )