error handle and optimize

Author: shadowabi

main/main.go

@@ -18,6 +18,7 @@ func init() {
 	config.SpecificInit(configFile)
 	if config.C.ServerUrl == "http://" || config.C.ServerUrl == "" || config.C.PortList == "" {
 		Error.HandleFatal(errors.New("请配置config.json"))
+		return
 	}
 }
 

pkg/data.go

@@ -2,12 +2,10 @@ package pkg
 
 import (
 	"bufio"
-	"fmt"
 	"github.com/shadowabi/Serverless_PortScan_rebuild/config"
 	"github.com/shadowabi/Serverless_PortScan_rebuild/utils/Error"
 	"os"
 	"regexp"
-	"sort"
 	"strings"
 )
 
@@ -62,34 +60,3 @@ func RemoveDuplicates(reqList []string) []string {
 	}
 	return result
 }
-
-func GetResult(respData ...ResponseData) (writeResultList []string) {
-	if len(respData) != 0 {
-		for _, data := range respData {
-			dataBody := strings.Trim(data.body, "\"")
-			ports := strings.Split(dataBody, ",")
-			for _, port := range ports {
-				fmt.Println(fmt.Sprintf("[+] %s:%s TCP OPEN.", data.host, port))
-				writeResult := strings.Join([]string{data.host, ":", port}, "")
-				writeResultList = append(writeResultList, writeResult)
-			}
-		}
-	}
-	return writeResultList
-}
-
-func WriteToFile(writeResultList []string, output string) {
-	file, err := os.OpenFile(output, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
-	Error.HandlePanic(err)
-	defer file.Close()
-
-	writer := bufio.NewWriter(file)
-	defer writer.Flush()
-
-	sort.Strings(writeResultList)
-	if len(writeResultList) != 0 {
-		for _, i := range writeResultList {
-			fmt.Fprintln(writer, i)
-		}
-	}
-}

pkg/output.go

@@ -0,0 +1,41 @@
+package pkg
+
+import (
+	"bufio"
+	"fmt"
+	"github.com/shadowabi/Serverless_PortScan_rebuild/utils/Error"
+	"os"
+	"sort"
+	"strings"
+)
+
+func GetResult(respData ...ResponseData) (writeResultList []string) {
+	if len(respData) != 0 {
+		for _, data := range respData {
+			dataBody := strings.Trim(data.body, "\"")
+			ports := strings.Split(dataBody, ",")
+			for _, port := range ports {
+				fmt.Println(fmt.Sprintf("[+] %s:%s TCP OPEN.", data.host, port))
+				writeResult := strings.Join([]string{data.host, ":", port}, "")
+				writeResultList = append(writeResultList, writeResult)
+			}
+		}
+	}
+	return writeResultList
+}
+
+func WriteToFile(writeResultList []string, output string) {
+	file, err := os.OpenFile(output, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
+	Error.HandlePanic(err)
+	defer file.Close()
+
+	writer := bufio.NewWriter(file)
+	defer writer.Flush()
+
+	sort.Strings(writeResultList)
+	if len(writeResultList) != 0 {
+		for _, i := range writeResultList {
+			fmt.Fprintln(writer, i)
+		}
+	}
+}