Gecko [ theprintave.ca ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-09-18 19:49
azure	[ DIR ]	drwxr-xr-x	2024-09-18 19:49
helpers	[ DIR ]	drwxr-xr-x	2024-09-18 19:49
DataSourceAkamai.py	12.67 KB	-rw-r--r--	2024-07-03 01:59
DataSourceAliYun.py	3.26 KB	-rw-r--r--	2024-07-03 01:59
DataSourceAltCloud.py	8.42 KB	-rw-r--r--	2024-07-03 01:59
DataSourceAzure.py	73.13 KB	-rw-r--r--	2024-07-03 01:59
DataSourceBigstep.py	1.9 KB	-rw-r--r--	2024-07-03 01:59
DataSourceCloudSigma.py	3.86 KB	-rw-r--r--	2024-07-03 01:59
DataSourceCloudStack.py	11.21 KB	-rw-r--r--	2024-07-03 01:59
DataSourceConfigDrive.py	10.88 KB	-rw-r--r--	2024-07-03 01:59
DataSourceDigitalOcean.py	4.18 KB	-rw-r--r--	2024-07-03 01:59
DataSourceEc2.py	42.44 KB	-rw-r--r--	2024-07-11 21:27
DataSourceExoscale.py	8.79 KB	-rw-r--r--	2024-07-03 01:59
DataSourceGCE.py	13.76 KB	-rw-r--r--	2024-07-03 01:59
DataSourceHetzner.py	5.34 KB	-rw-r--r--	2024-07-03 01:59
DataSourceIBMCloud.py	13.94 KB	-rw-r--r--	2024-07-03 01:59
DataSourceLXD.py	17.28 KB	-rw-r--r--	2024-07-03 01:59
DataSourceMAAS.py	14.85 KB	-rw-r--r--	2024-07-03 01:59
DataSourceNWCS.py	4.36 KB	-rw-r--r--	2024-07-03 01:59
DataSourceNoCloud.py	14.18 KB	-rw-r--r--	2024-07-03 01:59
DataSourceNone.py	1.27 KB	-rw-r--r--	2024-07-03 01:59
DataSourceOVF.py	12.7 KB	-rw-r--r--	2024-07-03 01:59
DataSourceOpenNebula.py	15.67 KB	-rw-r--r--	2024-07-03 01:59
DataSourceOpenStack.py	10.44 KB	-rw-r--r--	2024-07-03 01:59
DataSourceOracle.py	16.72 KB	-rw-r--r--	2024-07-03 01:59
DataSourceRbxCloud.py	7.85 KB	-rw-r--r--	2024-07-03 01:59
DataSourceScaleway.py	15.72 KB	-rw-r--r--	2024-07-03 01:59
DataSourceSmartOS.py	34 KB	-rw-r--r--	2024-07-03 01:59
DataSourceUpCloud.py	5.51 KB	-rw-r--r--	2024-07-03 01:59
DataSourceVMware.py	35.3 KB	-rw-r--r--	2024-07-03 01:59
DataSourceVultr.py	4.5 KB	-rw-r--r--	2024-07-03 01:59
DataSourceWSL.py	11.79 KB	-rw-r--r--	2024-07-03 01:59
__init__.py	43.56 KB	-rw-r--r--	2024-07-03 01:59

Rename

# 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_url_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,)),
]