mod: 默认不输出到文件，需要指定-o filename

cmd/go-portScan.go

@@ -37,6 +37,7 @@ var (
 	netLive     bool
 	maxOpenPort int
 	oCsv        string
+	oFile       string
 )
 
 func parseFlag(c *cli.Context) {
@@ -56,14 +57,15 @@ func parseFlag(c *cli.Context) {
 	netLive = c.Bool("netLive")
 	maxOpenPort = c.Int("maxOpenPort")
 	oCsv = c.String("oCsv")
+	oFile = c.String("oFile")
 }
 
 func run(c *cli.Context) error {
-	myLog := util.NewLogger("output.log", true)
 	if c.NumFlags() == 0 {
 		cli.ShowAppHelpAndExit(c, 0)
 	}
 	parseFlag(c)
+	myLog := util.NewLogger(oFile, true)
 	if devices {
 		if r, err := syn.GetAllDevs(); err != nil {
 			myLog.Fatal(err.Error())
@@ -430,6 +432,12 @@ func main() {
 				Usage:   "output csv file",
 				Value:   "",
 			},
+			&cli.StringFlag{
+				Name:    "oFile",
+				Aliases: []string{"o"},
+				Usage:   "output to file",
+				Value:   "",
+			},
 		},
 	}
 

util/log.go

@@ -8,11 +8,15 @@ import (
 
 func NewLogger(filename string, std bool) *log.Logger {
 	var out io.Writer
-	outFile, _ := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
-	if std {
-		out = io.MultiWriter(os.Stdout, outFile)
+	if filename == "" {
+		out = os.Stdout
 	} else {
-		out = outFile
+		outFile, _ := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
+		if std {
+			out = io.MultiWriter(os.Stdout, outFile)
+		} else {
+			out = outFile
+		}
 	}
 	logger := log.New(out, "", 0)
 	return logger