Merge pull request projectdiscovery#210 from projectdiscovery/198-fea…

…ture-on-result-callback

Adding on result callback function

v2/pkg/runner/options.go

@@ -46,8 +46,12 @@ type Options struct {
 	ScanAllIPS        bool   // Scan all the ips
 	ScanType          string // Scan Type
 	config            *ConfigFile
+	OnResult          OnResultCallback // OnResult callback
 }
 
+// OnResultCallback (hostname, ip, ports)
+type OnResultCallback func(string, string, []int)
+
 // ParseOptions parses the command line flags provided by a user
 func ParseOptions() *Options {
 	options := &Options{}

v2/pkg/runner/runner.go

@@ -381,6 +381,10 @@ func (r *Runner) handleOutput() {
 					gologger.Error().Msgf("Could not write results to file %s for %s: %s\n", output, host, err)
 				}
 			}
+
+			if r.options.OnResult != nil {
+				r.options.OnResult(host, hostIP, mapKeysToSliceInt(ports))
+			}
 		}
 	}
 }

v2/pkg/runner/util.go

@@ -44,3 +44,10 @@ func (r *Runner) host2ips(target string) (targetIPs []string, err error) {
 func isOSSupported() bool {
 	return runtime.GOOS == "linux" || runtime.GOOS == "darwin"
 }
+
+func mapKeysToSliceInt(m map[int]struct{}) (s []int) {
+	for k := range m {
+		s = append(s, k)
+	}
+	return
+}