External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-22 18:06:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding support for aws file credential

Browse Source
This commit is contained in:
mzack 2021-11-18 21:44:58 +01:00
parent 38ff8f75b1
commit ddb07ebb75
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
v2/pkg/protocols/http/aws_utils.go
View File

@ -49,6 +49,14 @@ func NewAwsSignerFromEnv() (*AwsSigner, error) {
return &AwsSigner{creds: creds}, nil return &AwsSigner{creds: creds}, nil
} }
func NewAwsSignerFromFile() (*AwsSigner, error) {
creds := credentials.NewSharedCredentials("", "")
if creds == nil {
return nil, errors.New("couldn't create the credentials structure")
}
return &AwsSigner{creds: creds}, nil
}
func (awsSigner *AwsSigner) SignHTTP(request *http.Request, args SignArguments) error { func (awsSigner *AwsSigner) SignHTTP(request *http.Request, args SignArguments) error {
awsSigner.prepareRequest(request) awsSigner.prepareRequest(request)
var body *bytes.Reader var body *bytes.Reader

8
v2/pkg/protocols/http/request.go
View File

@ -359,7 +359,15 @@ func (request *Request) executeRequest(reqURL string, generatedRequest *generate
if awsAccessKeyId != "" && awsSecretAccessKey != "" { if awsAccessKeyId != "" && awsSecretAccessKey != "" {
awsSigner, err = NewAwsSigner(awsAccessKeyId, awsSecretAccessKey) awsSigner, err = NewAwsSigner(awsAccessKeyId, awsSecretAccessKey)
} else { } else {
// env variables
awsSigner, err = NewAwsSignerFromEnv() awsSigner, err = NewAwsSignerFromEnv()
if err != nil {
// $HOME/.aws/credentials
awsSigner, err = NewAwsSignerFromFile()
if err != nil {
return err
}
}
} }
if err != nil { if err != nil {
return err return err