12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- package ec2
- import (
- "time"
- "github.com/aws/aws-sdk-go/aws"
- "github.com/aws/aws-sdk-go/aws/awsutil"
- "github.com/aws/aws-sdk-go/aws/endpoints"
- "github.com/aws/aws-sdk-go/aws/request"
- )
- func init() {
- initRequest = func(r *request.Request) {
- if r.Operation.Name == opCopySnapshot { // fill the PresignedURL parameter
- r.Handlers.Build.PushFront(fillPresignedURL)
- }
- }
- }
- func fillPresignedURL(r *request.Request) {
- if !r.ParamsFilled() {
- return
- }
- origParams := r.Params.(*CopySnapshotInput)
- // Stop if PresignedURL/DestinationRegion is set
- if origParams.PresignedUrl != nil || origParams.DestinationRegion != nil {
- return
- }
- origParams.DestinationRegion = r.Config.Region
- newParams := awsutil.CopyOf(r.Params).(*CopySnapshotInput)
- // Create a new request based on the existing request. We will use this to
- // presign the CopySnapshot request against the source region.
- cfg := r.Config.Copy(aws.NewConfig().
- WithEndpoint("").
- WithRegion(aws.StringValue(origParams.SourceRegion)))
- clientInfo := r.ClientInfo
- resolved, err := r.Config.EndpointResolver.EndpointFor(
- clientInfo.ServiceName, aws.StringValue(cfg.Region),
- func(opt *endpoints.Options) {
- opt.DisableSSL = aws.BoolValue(cfg.DisableSSL)
- opt.UseDualStack = aws.BoolValue(cfg.UseDualStack)
- },
- )
- if err != nil {
- r.Error = err
- return
- }
- clientInfo.Endpoint = resolved.URL
- clientInfo.SigningRegion = resolved.SigningRegion
- // Presign a CopySnapshot request with modified params
- req := request.New(*cfg, clientInfo, r.Handlers, r.Retryer, r.Operation, newParams, r.Data)
- url, err := req.Presign(5 * time.Minute) // 5 minutes should be enough.
- if err != nil { // bubble error back up to original request
- r.Error = err
- return
- }
- // We have our URL, set it on params
- origParams.PresignedUrl = &url
- }
|