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
|
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include <aws/common/environment.h>
#include <aws/common/string.h>
#include <stdlib.h>
int aws_get_environment_value(
struct aws_allocator *allocator,
const struct aws_string *variable_name,
struct aws_string **value_out) {
#ifdef _MSC_VER
# pragma warning(push)
# pragma warning(disable : 4996)
#endif
const char *value = getenv(aws_string_c_str(variable_name));
#ifdef _MSC_VER
# pragma warning(pop)
#endif
if (value == NULL) {
*value_out = NULL;
return AWS_OP_SUCCESS;
}
*value_out = aws_string_new_from_c_str(allocator, value);
if (*value_out == NULL) {
return aws_raise_error(AWS_ERROR_ENVIRONMENT_GET);
}
return AWS_OP_SUCCESS;
}
int aws_set_environment_value(const struct aws_string *variable_name, const struct aws_string *value) {
if (_putenv_s(aws_string_c_str(variable_name), aws_string_c_str(value)) != 0) {
return aws_raise_error(AWS_ERROR_ENVIRONMENT_SET);
}
return AWS_OP_SUCCESS;
}
AWS_STATIC_STRING_FROM_LITERAL(s_empty_string, "");
int aws_unset_environment_value(const struct aws_string *variable_name) {
if (_putenv_s(aws_string_c_str(variable_name), aws_string_c_str(s_empty_string)) != 0) {
return aws_raise_error(AWS_ERROR_ENVIRONMENT_UNSET);
}
return AWS_OP_SUCCESS;
}
|
