daffainfo
/
subfinder
mirror of https://github.com/daffainfo/subfinder.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #212 from projectdiscovery/add-version-command 

Added version flag to show current version
master
Ice3man 2020-01-10 03:47:05 -08:00 committed by GitHub
parent 83bed26d74 45a70618dc
commit 803e6517bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/runner/banners.go
View File

@ -13,6 +13,9 @@ ____ _| |__ / _(_)_ _ __| |___ _ _
/__/\_,_|_.__/_| |_|_||_\__,_\___|_| v2 /__/\_,_|_.__/_| |_|_||_\__,_\___|_| v2
` `
// Version is the current version of subfinder
const Version = `2.2.5`
// showBanner is used to show the banner to the user // showBanner is used to show the banner to the user
func showBanner() { func showBanner() {
log.Printf("%s\n", banner) log.Printf("%s\n", banner)

7
pkg/runner/options.go
View File

@ -30,6 +30,7 @@ type Options struct {
RemoveWildcard bool // RemoveWildcard specifies whether to remove potential wildcard or dead subdomains from the results. RemoveWildcard bool // RemoveWildcard specifies whether to remove potential wildcard or dead subdomains from the results.
ConfigFile string // ConfigFile contains the location of the config file ConfigFile string // ConfigFile contains the location of the config file
Stdin bool // Stdin specifies whether stdin input was given to the process Stdin bool // Stdin specifies whether stdin input was given to the process
Version bool // Version specifies if we should just show version and exit
YAMLConfig ConfigFile // YAMLConfig contains the unmarshalled yaml config file YAMLConfig ConfigFile // YAMLConfig contains the unmarshalled yaml config file
} }
@ -62,6 +63,7 @@ func ParseOptions() *Options {
flag.StringVar(&options.ResolverList, "rL", "", "Text file containing list of resolvers to use") flag.StringVar(&options.ResolverList, "rL", "", "Text file containing list of resolvers to use")
flag.BoolVar(&options.RemoveWildcard, "nW", false, "Remove Wildcard & Dead Subdomains from output") flag.BoolVar(&options.RemoveWildcard, "nW", false, "Remove Wildcard & Dead Subdomains from output")
flag.StringVar(&options.ConfigFile, "config", path.Join(config, "config.yaml"), "Configuration file for API Keys, etc") flag.StringVar(&options.ConfigFile, "config", path.Join(config, "config.yaml"), "Configuration file for API Keys, etc")
flag.BoolVar(&options.Version, "version", false, "Show version of subfinder")
flag.Parse() flag.Parse()
// Check if stdin pipe was given // Check if stdin pipe was given
@ -73,6 +75,11 @@ func ParseOptions() *Options {
// Show the user the banner // Show the user the banner
showBanner() showBanner()
if options.Version {
log.Infof("Current Version: %s\n", Version)
os.Exit(0)
}
// Check if the config file exists. If not, it means this is the // Check if the config file exists. If not, it means this is the
// first run of the program. Show the first run notices and initialize the config file. // first run of the program. Show the first run notices and initialize the config file.
// Else show the normal banners and read the yaml fiile to the config // Else show the normal banners and read the yaml fiile to the config