customizations.go 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. package s3
  2. import (
  3. "github.com/aws/aws-sdk-go/aws/client"
  4. "github.com/aws/aws-sdk-go/aws/request"
  5. "github.com/aws/aws-sdk-go/internal/s3err"
  6. )
  7. func init() {
  8. initClient = defaultInitClientFn
  9. initRequest = defaultInitRequestFn
  10. }
  11. func defaultInitClientFn(c *client.Client) {
  12. // Support building custom endpoints based on config
  13. c.Handlers.Build.PushFront(updateEndpointForS3Config)
  14. // Require SSL when using SSE keys
  15. c.Handlers.Validate.PushBack(validateSSERequiresSSL)
  16. c.Handlers.Build.PushBack(computeSSEKeys)
  17. // S3 uses custom error unmarshaling logic
  18. c.Handlers.UnmarshalError.Clear()
  19. c.Handlers.UnmarshalError.PushBack(unmarshalError)
  20. c.Handlers.UnmarshalError.PushBackNamed(s3err.RequestFailureWrapperHandler())
  21. }
  22. func defaultInitRequestFn(r *request.Request) {
  23. // Add reuest handlers for specific platforms.
  24. // e.g. 100-continue support for PUT requests using Go 1.6
  25. platformRequestHandlers(r)
  26. switch r.Operation.Name {
  27. case opPutBucketCors, opPutBucketLifecycle, opPutBucketPolicy,
  28. opPutBucketTagging, opDeleteObjects, opPutBucketLifecycleConfiguration,
  29. opPutObjectLegalHold, opPutObjectRetention, opPutObjectLockConfiguration,
  30. opPutBucketReplication:
  31. // These S3 operations require Content-MD5 to be set
  32. r.Handlers.Build.PushBack(contentMD5)
  33. case opGetBucketLocation:
  34. // GetBucketLocation has custom parsing logic
  35. r.Handlers.Unmarshal.PushFront(buildGetBucketLocation)
  36. case opCreateBucket:
  37. // Auto-populate LocationConstraint with current region
  38. r.Handlers.Validate.PushFront(populateLocationConstraint)
  39. case opCopyObject, opUploadPartCopy, opCompleteMultipartUpload:
  40. r.Handlers.Unmarshal.PushFront(copyMultipartStatusOKUnmarhsalError)
  41. r.Handlers.Unmarshal.PushBackNamed(s3err.RequestFailureWrapperHandler())
  42. case opPutObject, opUploadPart:
  43. r.Handlers.Build.PushBack(computeBodyHashes)
  44. // Disabled until #1837 root issue is resolved.
  45. // case opGetObject:
  46. // r.Handlers.Build.PushBack(askForTxEncodingAppendMD5)
  47. // r.Handlers.Unmarshal.PushBack(useMD5ValidationReader)
  48. }
  49. }
  50. // bucketGetter is an accessor interface to grab the "Bucket" field from
  51. // an S3 type.
  52. type bucketGetter interface {
  53. getBucket() string
  54. }
  55. // sseCustomerKeyGetter is an accessor interface to grab the "SSECustomerKey"
  56. // field from an S3 type.
  57. type sseCustomerKeyGetter interface {
  58. getSSECustomerKey() string
  59. }
  60. // copySourceSSECustomerKeyGetter is an accessor interface to grab the
  61. // "CopySourceSSECustomerKey" field from an S3 type.
  62. type copySourceSSECustomerKeyGetter interface {
  63. getCopySourceSSECustomerKey() string
  64. }