OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	05/09/2024 07:15:07 AM	rwxr-xr-x
📄 DataSourceAliYun.py	2.17 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceAltCloud.py	8.29 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceAzure.py	68.68 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceBigstep.py	1.92 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceCloudSigma.py	4 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceCloudStack.py	9.85 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceConfigDrive.py	11.01 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceDigitalOcean.py	3.81 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceEc2.py	35.81 KB	04/21/2023 07:44:30 PM	rw-r--r--
📄 DataSourceExoscale.py	8.89 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceGCE.py	11.96 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceHetzner.py	5.41 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceIBMCloud.py	14.05 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceLXD.py	17.19 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceMAAS.py	14.39 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceNWCS.py	4.66 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceNoCloud.py	12.38 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceNone.py	1.36 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceOVF.py	10.47 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceOpenNebula.py	15.98 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceOpenStack.py	10.46 KB	04/21/2023 07:44:30 PM	rw-r--r--
📄 DataSourceOracle.py	14.72 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceRbxCloud.py	7.83 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceScaleway.py	9.47 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceSmartOS.py	34.09 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceUpCloud.py	5.55 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceVMware.py	32.7 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 DataSourceVultr.py	4.18 KB	04/20/2023 09:31:09 PM	rw-r--r--
📄 __init__.py	40.79 KB	04/20/2023 09:31:09 PM	rw-r--r--
📁 __pycache__	-	05/09/2024 07:15:08 AM	rwxr-xr-x
📁 azure	-	05/09/2024 07:15:08 AM	rwxr-xr-x
📁 helpers	-	05/09/2024 07:15:08 AM	rwxr-xr-x

Editing: DataSourceDigitalOcean.py

# Author: Neal Shrader <neal@digitalocean.com>
# Author: Ben Howard  <bh@digitalocean.com>
#
# This file is part of cloud-init. See LICENSE file for license information.

# DigitalOcean Droplet API:
# https://developers.digitalocean.com/documentation/metadata/

import cloudinit.sources.helpers.digitalocean as do_helper
from cloudinit import log as logging
from cloudinit import sources, util

LOG = logging.getLogger(__name__)

BUILTIN_DS_CONFIG = {
    "metadata_url": "http://169.254.169.254/metadata/v1.json",
}

# Wait for a up to a minute, retrying the meta-data server
# every 2 seconds.
MD_RETRIES = 30
MD_TIMEOUT = 2
MD_WAIT_RETRY = 2
MD_USE_IPV4LL = True

class DataSourceDigitalOcean(sources.DataSource):

dsname = "DigitalOcean"

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", "DigitalOcean"], {}
                ),
                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.use_ip4LL = self.ds_cfg.get("use_ip4LL", MD_USE_IPV4LL)
        self.wait_retry = self.ds_cfg.get("wait_retry", MD_WAIT_RETRY)
        self._network_config = None

def _get_sysinfo(self):
        return do_helper.read_sysinfo()

def _get_data(self):
        (is_do, droplet_id) = self._get_sysinfo()

# only proceed if we know we are on DigitalOcean
        if not is_do:
            return False

LOG.info("Running on DigitalOcean. droplet_id=%s", droplet_id)

ipv4LL_nic = None
        if self.use_ip4LL:
            ipv4LL_nic = do_helper.assign_ipv4_link_local(self.distro)

md = do_helper.read_metadata(
            self.metadata_address,
            timeout=self.timeout,
            sec_between=self.wait_retry,
            retries=self.retries,
        )

self.metadata_full = md
        self.metadata["instance-id"] = md.get("droplet_id", droplet_id)
        self.metadata["local-hostname"] = md.get("hostname", droplet_id)
        self.metadata["interfaces"] = md.get("interfaces")
        self.metadata["public-keys"] = md.get("public_keys")
        self.metadata["availability_zone"] = md.get("region", "default")
        self.vendordata_raw = md.get("vendor_data", None)
        self.userdata_raw = md.get("user_data", None)

if ipv4LL_nic:
            do_helper.del_ipv4_link_local(ipv4LL_nic)

return True

def check_instance_id(self, sys_cfg):
        return sources.instance_id_matches_system_uuid(
            self.get_instance_id(), "system-serial-number"
        )

@property
    def network_config(self):
        """Configure the networking. This needs to be done each boot, since
        the IP information may have changed due to snapshot and/or
        migration.
        """

if self._network_config:
            return self._network_config

interfaces = self.metadata.get("interfaces")
        LOG.debug(interfaces)
        if not interfaces:
            raise Exception("Unable to get meta-data from server....")

nameservers = self.metadata_full["dns"]["nameservers"]
        self._network_config = do_helper.convert_network_configuration(
            interfaces, nameservers
        )
        return self._network_config

# Used to match classes to dependencies
datasources = [
    (DataSourceDigitalOcean, (sources.DEP_FILESYSTEM,)),
]

# Return a list of data sources that match this set of dependencies
def get_datasource_list(depends):
    return sources.list_from_depends(depends, datasources)

# vi: ts=4 expandtab