mirror of
https://github.com/OJ/gobuster.git
synced 2024-05-04 22:46:07 +02:00
99 lines
3.2 KiB
Go
99 lines
3.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/OJ/gobuster/v3/cli"
|
|
"github.com/OJ/gobuster/v3/gobustervhost"
|
|
"github.com/OJ/gobuster/v3/libgobuster"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// nolint:gochecknoglobals
|
|
var cmdVhost *cobra.Command
|
|
|
|
func runVhost(cmd *cobra.Command, args []string) error {
|
|
globalopts, pluginopts, err := parseVhostOptions()
|
|
if err != nil {
|
|
return fmt.Errorf("error on parsing arguments: %w", err)
|
|
}
|
|
|
|
plugin, err := gobustervhost.NewGobusterVhost(globalopts, pluginopts)
|
|
if err != nil {
|
|
return fmt.Errorf("error on creating gobustervhost: %w", err)
|
|
}
|
|
|
|
if err := cli.Gobuster(mainContext, globalopts, plugin); err != nil {
|
|
return fmt.Errorf("error on running gobuster: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func parseVhostOptions() (*libgobuster.Options, *gobustervhost.OptionsVhost, error) {
|
|
globalopts, err := parseGlobalOptions()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
pluginOpts := gobustervhost.NewOptionsVhost()
|
|
|
|
httpOpts, err := parseCommonHTTPOptions(cmdVhost)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
pluginOpts.Password = httpOpts.Password
|
|
pluginOpts.URL = httpOpts.URL
|
|
pluginOpts.UserAgent = httpOpts.UserAgent
|
|
pluginOpts.Username = httpOpts.Username
|
|
pluginOpts.Proxy = httpOpts.Proxy
|
|
pluginOpts.Cookies = httpOpts.Cookies
|
|
pluginOpts.Timeout = httpOpts.Timeout
|
|
pluginOpts.FollowRedirect = httpOpts.FollowRedirect
|
|
pluginOpts.NoTLSValidation = httpOpts.NoTLSValidation
|
|
pluginOpts.Headers = httpOpts.Headers
|
|
pluginOpts.Method = httpOpts.Method
|
|
pluginOpts.RetryOnTimeout = httpOpts.RetryOnTimeout
|
|
pluginOpts.RetryAttempts = httpOpts.RetryAttempts
|
|
pluginOpts.TLSCertificate = httpOpts.TLSCertificate
|
|
pluginOpts.NoCanonicalizeHeaders = httpOpts.NoCanonicalizeHeaders
|
|
|
|
pluginOpts.AppendDomain, err = cmdVhost.Flags().GetBool("append-domain")
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("invalid value for append-domain: %w", err)
|
|
}
|
|
|
|
pluginOpts.ExcludeLength, err = cmdVhost.Flags().GetIntSlice("exclude-length")
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("invalid value for excludelength: %w", err)
|
|
}
|
|
|
|
pluginOpts.Domain, err = cmdVhost.Flags().GetString("domain")
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("invalid value for domain: %w", err)
|
|
}
|
|
|
|
return globalopts, pluginOpts, nil
|
|
}
|
|
|
|
// nolint:gochecknoinits
|
|
func init() {
|
|
cmdVhost = &cobra.Command{
|
|
Use: "vhost",
|
|
Short: "Uses VHOST enumeration mode (you most probably want to use the IP address as the URL parameter)",
|
|
RunE: runVhost,
|
|
}
|
|
if err := addCommonHTTPOptions(cmdVhost); err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
cmdVhost.Flags().BoolP("append-domain", "", false, "Append main domain from URL to words from wordlist. Otherwise the fully qualified domains need to be specified in the wordlist.")
|
|
cmdVhost.Flags().IntSlice("exclude-length", []int{}, "exclude the following content length (completely ignores the status). Supply multiple times to exclude multiple sizes.")
|
|
cmdVhost.Flags().String("domain", "", "the domain to append when using an IP address as URL. If left empty and you specify a domain based URL the hostname from the URL is extracted")
|
|
|
|
cmdVhost.PersistentPreRun = func(cmd *cobra.Command, args []string) {
|
|
configureGlobalOptions()
|
|
}
|
|
|
|
rootCmd.AddCommand(cmdVhost)
|
|
}
|