增加判断是否最新版本

Author: SummerSec

cmd/commons/core/banner.go

@@ -1,7 +1,7 @@
 package core
 
 import (
-	log "github.com/sirupsen/logrus"
+	"fmt"
 )
 
 const banner = `
@@ -18,7 +18,7 @@ const banner = `
 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")
+	fmt.Println(banner)
+	fmt.Println("\t\t\tAuthor: SummerSec Version:", Version+" Github: https://Github.com/SummerSec\n")
 
 }

cmd/commons/core/options.go

@@ -78,8 +78,8 @@ func ParseOptions() *Options {
 	flag.Parse()
 
 	// TODO 修改版本号
-	ShowBanner(version)
 	logs.SaveLogs(options.LogFile)
+	ShowBanner(version)
 	showVerbose(options)
 
 	if options.Version {

cmd/commons/core/runner.go

@@ -19,16 +19,16 @@ func NewRunner(options *Options) (*Runner, error) {
 
 	mops := structs.Map(&r.options)
 	data, _ := json.Marshal(mops)
-	log.Info("Runner created")
+	log.Debug("Runner created")
 	log.Debug(mops)
 	log.Debug("Runner options: ", string(data))
 	return &r, nil
 
 }
 
 func (r *Runner) Run() {
-	log.Debugln("github.com/SummerSec/SpringExploit/cmd/commons/core/runner.go: Run()")
-	log.Info("Runner Running")
+	log.Info("Starting SpringExploit")
+	log.Debug("Runner Running")
 	//ip := r.options.IP
 	var urls []string
 	// TODO: check if options are valid
@@ -42,6 +42,9 @@ func (r *Runner) Run() {
 		urls = utils.GetIPToUrlsLinks(r.options.IP, urls)
 	} else if r.options.Update {
 		selfUpdate()
+	} else if r.options.Version {
+		getLatestVersion()
+		return
 	} else {
 		log.Error("No file or url or ips specified")
 		return

cmd/commons/core/update.go

@@ -64,3 +64,28 @@ func selfUpdate() {
 	log.Infof("Successfully Updated to %s Version %s", Repo, latest.Version)
 
 }
+
+// 获取最新版本
+func getLatestVersion() {
+	log.Info("Crrunent Version: ", version)
+	log.Infof("Latest Version: %s", getLatestVersionFromGithub())
+
+}
+
+// 获取最新版本从github
+func getLatestVersionFromGithub() string {
+	m := &update.Manager{
+		Store: &githubUpdateStore.Store{
+			Owner:   Owner,
+			Repo:    Repo,
+			Version: version,
+		},
+	}
+	releases, err := m.LatestReleases()
+	if err != nil {
+		log.Error("Failed to get releases", err)
+		return ""
+	}
+	latest := releases[0]
+	return latest.Version
+}

cmd/main.go

@@ -8,17 +8,17 @@ import (
 func main() {
 	options := core.ParseOptions()
 
-	if options.Url != "" {
+	//if options.Url != "" {
+	//
+	//} else if options.File != "" {
+	//} else if options.IP != "" {
+	//} else if options.SP {
+	//	return
+	//} else {
+	//	log.Info("No url, file or ip specified")
+	//	return
+	//}
 
-	} else if options.File != "" {
-	} else if options.IP != "" {
-	} else if options.SP {
-		return
-	} else {
-		log.Info("No url, file or ip specified")
-		return
-	}
-	log.Info("Starting SpringExploit")
 	r, _ := core.NewRunner(options)
 	//log.Errorf("Error: %s", err)