Merge pull request #7 from tarihub/wildcard

fix: file overwriting problem
Esonhugh · Mar 22, 2024 · b9622bb · b9622bb
2 parents 0514447 + 3b67cfc
commit b9622bb
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/cmd/all/all.go b/cmd/all/all.go
@@ -74,7 +74,7 @@ func RunBatch(net *net.IPNet) {
 
 func printResult(records define.Records) {
 	if command.Opts.OutputFile != "" {
-		f, err := os.OpenFile(command.Opts.OutputFile, os.O_CREATE|os.O_WRONLY, 0644)
+		f, err := os.OpenFile(command.Opts.OutputFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 		if err != nil {
 			log.Warnf("OpenFile failed: %v", err)
 		}

diff --git a/cmd/axfr/axfr.go b/cmd/axfr/axfr.go
@@ -42,7 +42,7 @@ var AxfrCmd = &cobra.Command{
 			return
 		}
 		if command.Opts.OutputFile != "" {
-			f, err := os.OpenFile(command.Opts.OutputFile, os.O_CREATE|os.O_WRONLY, 0644)
+			f, err := os.OpenFile(command.Opts.OutputFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 			if err != nil {
 				log.Warnf("OpenFile failed: %v", err)
 			}

diff --git a/cmd/service/service.go b/cmd/service/service.go
@@ -29,7 +29,7 @@ var ServiceCmd = &cobra.Command{
 		}
 		records = scanner.ScanSvcForPorts(records)
 		if command.Opts.OutputFile != "" {
-			f, err := os.OpenFile(command.Opts.OutputFile, os.O_CREATE|os.O_WRONLY, 0644)
+			f, err := os.OpenFile(command.Opts.OutputFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 			if err != nil {
 				log.Warnf("OpenFile failed: %v", err)
 			}

diff --git a/cmd/subnet/subnet.go b/cmd/subnet/subnet.go
@@ -62,7 +62,7 @@ func BatchRun(net *net.IPNet) {
 
 func printResult(records define.Records) {
 	if command.Opts.OutputFile != "" {
-		f, err := os.OpenFile(command.Opts.OutputFile, os.O_CREATE|os.O_WRONLY, 0644)
+		f, err := os.OpenFile(command.Opts.OutputFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 		if err != nil {
 			log.Warnf("OpenFile failed: %v", err)
 		}

diff --git a/cmd/wildcard/wildcard.go b/cmd/wildcard/wildcard.go
@@ -28,7 +28,7 @@ var WildCardCmd = &cobra.Command{
 
 func printResult(records define.Records) {
 	if command.Opts.OutputFile != "" {
-		f, err := os.OpenFile(command.Opts.OutputFile, os.O_CREATE|os.O_WRONLY, 0644)
+		f, err := os.OpenFile(command.Opts.OutputFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 		if err != nil {
 			log.Warnf("OpenFile failed: %v", err)
 		}