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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
package config
import (
"fmt"
"strings"
)
// ClientEnableState provides an enumeration if the client is enabled,
// disabled, or default behavior.
type ClientEnableState uint
// Enumeration values for ClientEnableState
const (
ClientDefaultEnableState ClientEnableState = iota
ClientDisabled
ClientEnabled
)
// EndpointModeState is the EC2 IMDS Endpoint Configuration Mode
type EndpointModeState uint
// Enumeration values for ClientEnableState
const (
EndpointModeStateUnset EndpointModeState = iota
EndpointModeStateIPv4
EndpointModeStateIPv6
)
// SetFromString sets the EndpointModeState based on the provided string value. Unknown values will default to EndpointModeStateUnset
func (e *EndpointModeState) SetFromString(v string) error {
v = strings.TrimSpace(v)
switch {
case len(v) == 0:
*e = EndpointModeStateUnset
case strings.EqualFold(v, "IPv6"):
*e = EndpointModeStateIPv6
case strings.EqualFold(v, "IPv4"):
*e = EndpointModeStateIPv4
default:
return fmt.Errorf("unknown EC2 IMDS endpoint mode, must be either IPv6 or IPv4")
}
return nil
}
// ClientEnableStateResolver is a config resolver interface for retrieving whether the IMDS client is disabled.
type ClientEnableStateResolver interface {
GetEC2IMDSClientEnableState() (ClientEnableState, bool, error)
}
// EndpointModeResolver is a config resolver interface for retrieving the EndpointModeState configuration.
type EndpointModeResolver interface {
GetEC2IMDSEndpointMode() (EndpointModeState, bool, error)
}
// EndpointResolver is a config resolver interface for retrieving the endpoint.
type EndpointResolver interface {
GetEC2IMDSEndpoint() (string, bool, error)
}
// ResolveClientEnableState resolves the ClientEnableState from a list of configuration sources.
func ResolveClientEnableState(sources []interface{}) (value ClientEnableState, found bool, err error) {
for _, source := range sources {
if resolver, ok := source.(ClientEnableStateResolver); ok {
value, found, err = resolver.GetEC2IMDSClientEnableState()
if err != nil || found {
return value, found, err
}
}
}
return value, found, err
}
// ResolveEndpointModeConfig resolves the EndpointModeState from a list of configuration sources.
func ResolveEndpointModeConfig(sources []interface{}) (value EndpointModeState, found bool, err error) {
for _, source := range sources {
if resolver, ok := source.(EndpointModeResolver); ok {
value, found, err = resolver.GetEC2IMDSEndpointMode()
if err != nil || found {
return value, found, err
}
}
}
return value, found, err
}
// ResolveEndpointConfig resolves the endpoint from a list of configuration sources.
func ResolveEndpointConfig(sources []interface{}) (value string, found bool, err error) {
for _, source := range sources {
if resolver, ok := source.(EndpointResolver); ok {
value, found, err = resolver.GetEC2IMDSEndpoint()
if err != nil || found {
return value, found, err
}
}
}
return value, found, err
}
|
