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.
*/
#include <aws/auth/credentials.h>
#include <aws/auth/private/credentials_utils.h>
#include <aws/common/environment.h>
#include <aws/common/string.h>
AWS_STATIC_STRING_FROM_LITERAL(s_access_key_id_env_var, "AWS_ACCESS_KEY_ID");
AWS_STATIC_STRING_FROM_LITERAL(s_secret_access_key_env_var, "AWS_SECRET_ACCESS_KEY");
AWS_STATIC_STRING_FROM_LITERAL(s_session_token_env_var, "AWS_SESSION_TOKEN");
static int s_credentials_provider_environment_get_credentials_async(
struct aws_credentials_provider *provider,
aws_on_get_credentials_callback_fn callback,
void *user_data) {
struct aws_allocator *allocator = provider->allocator;
struct aws_string *access_key_id = NULL;
struct aws_string *secret_access_key = NULL;
struct aws_string *session_token = NULL;
struct aws_credentials *credentials = NULL;
int error_code = AWS_ERROR_SUCCESS;
aws_get_environment_value(allocator, s_access_key_id_env_var, &access_key_id);
aws_get_environment_value(allocator, s_secret_access_key_env_var, &secret_access_key);
aws_get_environment_value(allocator, s_session_token_env_var, &session_token);
if (access_key_id != NULL && secret_access_key != NULL) {
credentials =
aws_credentials_new_from_string(allocator, access_key_id, secret_access_key, session_token, UINT64_MAX);
if (credentials == NULL) {
error_code = aws_last_error();
}
} else {
error_code = AWS_AUTH_CREDENTIALS_PROVIDER_INVALID_ENVIRONMENT;
}
callback(credentials, error_code, user_data);
aws_credentials_release(credentials);
aws_string_destroy(session_token);
aws_string_destroy(secret_access_key);
aws_string_destroy(access_key_id);
return AWS_OP_SUCCESS;
}
static void s_credentials_provider_environment_destroy(struct aws_credentials_provider *provider) {
aws_credentials_provider_invoke_shutdown_callback(provider);
aws_mem_release(provider->allocator, provider);
}
static struct aws_credentials_provider_vtable s_aws_credentials_provider_environment_vtable = {
.get_credentials = s_credentials_provider_environment_get_credentials_async,
.destroy = s_credentials_provider_environment_destroy,
};
struct aws_credentials_provider *aws_credentials_provider_new_environment(
struct aws_allocator *allocator,
const struct aws_credentials_provider_environment_options *options) {
struct aws_credentials_provider *provider = aws_mem_acquire(allocator, sizeof(struct aws_credentials_provider));
if (provider == NULL) {
return NULL;
}
AWS_ZERO_STRUCT(*provider);
aws_credentials_provider_init_base(provider, allocator, &s_aws_credentials_provider_environment_vtable, NULL);
provider->shutdown_options = options->shutdown_options;
return provider;
}
|
