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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
# Author: NWCS.sh <foss@nwcs.sh>
#
# This file is part of cloud-init. See LICENSE file for license information.
import logging
from requests import exceptions
from cloudinit import dmi, net, sources, subp, url_helper, util
from cloudinit.net.dhcp import NoDHCPLeaseError
from cloudinit.net.ephemeral import EphemeralDHCPv4
LOG = logging.getLogger(__name__)
BASE_URL_V1 = "http://169.254.169.254/api/v1"
BUILTIN_DS_CONFIG = {
"metadata_url": BASE_URL_V1 + "/metadata",
}
MD_RETRIES = 30
MD_TIMEOUT = 5
MD_WAIT_RETRY = 5
class DataSourceNWCS(sources.DataSource):
dsname = "NWCS"
def __init__(self, sys_cfg, distro, paths):
sources.DataSource.__init__(self, sys_cfg, distro, paths)
self.distro = distro
self.metadata = dict()
self.ds_cfg = util.mergemanydict(
[
util.get_cfg_by_path(sys_cfg, ["datasource", "NWCS"], {}),
BUILTIN_DS_CONFIG,
]
)
self.metadata_address = self.ds_cfg["metadata_url"]
self.retries = self.ds_cfg.get("retries", MD_RETRIES)
self.timeout = self.ds_cfg.get("timeout", MD_TIMEOUT)
self.wait_retry = self.ds_cfg.get("wait_retry", MD_WAIT_RETRY)
self._network_config = sources.UNSET
self.dsmode = sources.DSMODE_NETWORK
self.metadata_full = None
def _unpickle(self, ci_pkl_version: int) -> None:
super()._unpickle(ci_pkl_version)
if not self._network_config:
self._network_config = sources.UNSET
def _get_data(self):
md = self.get_metadata()
if md is None:
raise RuntimeError("failed to get metadata")
self.metadata_full = md
self.metadata["instance-id"] = md["instance-id"]
self.metadata["public-keys"] = md["public-keys"]
self.metadata["network"] = md["network"]
self.metadata["local-hostname"] = md["hostname"]
self.userdata_raw = md.get("userdata", None)
self.vendordata_raw = md.get("vendordata", None)
return True
def get_metadata(self):
try:
LOG.info("Attempting to get metadata via DHCP")
with EphemeralDHCPv4(
self.distro,
iface=net.find_fallback_nic(),
connectivity_urls_data=[
{
"url": BASE_URL_V1 + "/metadata/instance-id",
}
],
):
return read_metadata(
self.metadata_address,
timeout=self.timeout,
sec_between=self.wait_retry,
retries=self.retries,
)
except (
NoDHCPLeaseError,
subp.ProcessExecutionError,
RuntimeError,
exceptions.RequestException,
) as e:
LOG.error("DHCP failure: %s", e)
raise
@property
def network_config(self):
LOG.debug("Attempting network configuration")
if self._network_config != sources.UNSET:
return self._network_config
if not self.metadata["network"]["config"]:
raise RuntimeError("Unable to get metadata from server")
# metadata sends interface names, but we dont want to use them
for i in self.metadata["network"]["config"]:
iface_name = get_interface_name(i["mac_address"])
if iface_name:
LOG.info("Overriding %s with %s", i["name"], iface_name)
i["name"] = iface_name
self._network_config = self.metadata["network"]
return self._network_config
@staticmethod
def ds_detect():
return "NWCS" == dmi.read_dmi_data("system-manufacturer")
def get_interface_name(mac):
macs_to_nic = net.get_interfaces_by_mac()
if mac not in macs_to_nic:
return None
return macs_to_nic.get(mac)
# Return a list of data sources that match this set of dependencies
def get_datasource_list(depends):
return sources.list_from_depends(depends, datasources)
def read_metadata(url, timeout=2, sec_between=2, retries=30):
response = url_helper.readurl(
url, timeout=timeout, sec_between=sec_between, retries=retries
)
if not response.ok():
raise RuntimeError("unable to read metadata at %s" % url)
return util.load_json(response.contents.decode())
# Used to match classes to dependencies
datasources = [
(DataSourceNWCS, (sources.DEP_FILESYSTEM,)),
]
|
