forked from awsdocs/aws-doc-sdk-examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
EncryptOnServerWithKms.go
79 lines (67 loc) · 2.39 KB
/
EncryptOnServerWithKms.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
// snippet-start:[s3.go.encrypt_on_server_with_kms]
package main
// snippet-start:[s3.go.encrypt_on_server_with_kms.imports]
import (
"flag"
"fmt"
"strings"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3"
)
// snippet-end:[s3.go.encrypt_on_server_with_kms.imports]
// AddObjectWithServerKms adds an object to a bucket with encryption based on a KMS key
// Inputs:
//
// sess is the current session, which provides configuration for the SDK's service clients
// bucket is the name of the bucket
// object is the name of the object to upload to the bucket
// kmsKey is the KMS key used to encrypt the object in the bucket
//
// Output:
//
// If success, nil
// Otherwise, an error from the call to PutObject
func AddObjectWithServerKms(sess *session.Session, bucket, object, kmsKey *string) error {
// snippet-start:[s3.go.encrypt_on_server_with_kms.call]
svc := s3.New(sess)
_, err := svc.PutObject(&s3.PutObjectInput{
Body: strings.NewReader(*object),
Bucket: bucket,
Key: object,
ServerSideEncryption: aws.String("aws:kms"),
SSEKMSKeyId: kmsKey,
})
// snippet-end:[s3.go.encrypt_on_server_with_kms.call]
if err != nil {
return err
}
return nil
}
func main() {
// snippet-start:[s3.go.encrypt_on_server_with_kms.args]
bucket := flag.String("b", "", "The bucket")
object := flag.String("o", "", "The file to upload")
kmsKey := flag.String("k", "", "The KMS key to encrypt with")
flag.Parse()
if *bucket == "" || *object == "" || *kmsKey == "" {
fmt.Println("You must supply a bucket (-b BUCKET), object to upload (-o OBJECT), and KMS key (-k KEY)")
return
}
// snippet-end:[s3.go.encrypt_on_server_with_kms.args]
// snippet-start:[s3.go.encrypt_on_server_with_kms.session]
sess := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
// snippet-end:[s3.go.encrypt_on_server_with_kms.session]
err := AddObjectWithServerKms(sess, bucket, object, kmsKey)
if err != nil {
fmt.Println("Got an error adding object to bucket:")
fmt.Println(err)
return
}
fmt.Println("Added object " + *object + " to bucket " + *bucket + " with AWS KMS encryption")
}
// snippet-end:[s3.go.encrypt_on_server_with_kms]