mirror of
https://github.com/OJ/gobuster.git
synced 2024-05-04 22:46:07 +02:00
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/OJ/gobuster/v3/cli"
|
|
"github.com/OJ/gobuster/v3/gobusters3"
|
|
"github.com/OJ/gobuster/v3/libgobuster"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// nolint:gochecknoglobals
|
|
var cmdS3 *cobra.Command
|
|
|
|
func runS3(cmd *cobra.Command, args []string) error {
|
|
globalopts, pluginopts, err := parseS3Options()
|
|
if err != nil {
|
|
return fmt.Errorf("error on parsing arguments: %w", err)
|
|
}
|
|
|
|
plugin, err := gobusters3.NewGobusterS3(globalopts, pluginopts)
|
|
if err != nil {
|
|
return fmt.Errorf("error on creating gobusters3: %w", err)
|
|
}
|
|
|
|
if err := cli.Gobuster(mainContext, globalopts, plugin); err != nil {
|
|
return fmt.Errorf("error on running gobuster: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func parseS3Options() (*libgobuster.Options, *gobusters3.OptionsS3, error) {
|
|
globalopts, err := parseGlobalOptions()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
pluginOpts := gobusters3.NewOptionsS3()
|
|
|
|
httpOpts, err := parseBasicHTTPOptions(cmdS3)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
pluginOpts.UserAgent = httpOpts.UserAgent
|
|
pluginOpts.Proxy = httpOpts.Proxy
|
|
pluginOpts.Timeout = httpOpts.Timeout
|
|
pluginOpts.NoTLSValidation = httpOpts.NoTLSValidation
|
|
pluginOpts.RetryOnTimeout = httpOpts.RetryOnTimeout
|
|
pluginOpts.RetryAttempts = httpOpts.RetryAttempts
|
|
pluginOpts.TLSCertificate = httpOpts.TLSCertificate
|
|
|
|
pluginOpts.MaxFilesToList, err = cmdS3.Flags().GetInt("maxfiles")
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("invalid value for maxfiles: %w", err)
|
|
}
|
|
|
|
return globalopts, pluginOpts, nil
|
|
}
|
|
|
|
// nolint:gochecknoinits
|
|
func init() {
|
|
cmdS3 = &cobra.Command{
|
|
Use: "s3",
|
|
Short: "Uses aws bucket enumeration mode",
|
|
RunE: runS3,
|
|
}
|
|
|
|
addBasicHTTPOptions(cmdS3)
|
|
cmdS3.Flags().IntP("maxfiles", "m", 5, "max files to list when listing buckets (only shown in verbose mode)")
|
|
|
|
cmdS3.PersistentPreRun = func(cmd *cobra.Command, args []string) {
|
|
configureGlobalOptions()
|
|
}
|
|
|
|
rootCmd.AddCommand(cmdS3)
|
|
}
|