0x1949 Team - FAZEMRX - MANAGER
Edit File: DataSourceSmartOS.cpython-310.pyc
o ���f�� � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl m Z m Z mZmZmZ d dlmZmZ e�e�Zdddddd d d d d dddd� Zdddd�ZdZdZdZdegZ g d�Z dZ d Z d Z e e e e g d d d id!�Z!d d"d d d#�id d$d d%�gd&�Z"d'Z#d(Z$G d)d*� d*ej%�Z&G d+d,� d,e'�Z(G d-d.� d.e(�Z)G d/d0� d0�Z*G d1d2� d2e*�Z+G d3d4� d4e*�Z,G d5d6� d6e,�Z-de e e dfd7d8�Z.d9d:� Z/ ;dOd<d=�Z0dPd>d?�Z1 dQd@dA�Z2e&ej3ffgZ4dBdC� Z5edDk�rQd dl6Z6e.� Z7e7du �re8dE� e6�9dF� e:e6j;�dFk�r&e<e�=� �e<e�=� � dGg Z=ne6j;dFd� Z=dHdI� Z>i Z?e@eAdJ< e=D ] ZBe>e7eBe?dK� �q9e8ejCe?dFdLdMdN�� dS dS )R� N)� atomic_helper�dmi�sources�subp�util)� EventScope� EventType)zsdc:uuidT)�hostnameT)�root_authorized_keysT)� user-scriptF)� user-dataF)zcloud-init:user-dataF)�iptables_disableT)� motd_sys_infoT)�sdc:datacenter_nameT)zsdc:vendor-dataF)zsdc:operator-scriptF)z sdc:hostnameT)zsdc:dns_domainT) � instance-id�local-hostnamez public-keysr �legacy-user-datar r r �availability_zone� vendor-data�operator-scriptr � dns_domainzsdc:nicsz sdc:resolversz sdc:routes)� network-data� dns_servers�routeszlx-brand�kvm�SmartOS� datasource)r r r r zsdc:uuiduser-datar z"/native/.zonecontrol/metadata.sockz /dev/ttyS1�< F� ephemeral0z/dev/vdb)� serial_device�serial_timeout�metadata_sockfile�no_base64_decode� base64_keys� base64_all� disk_aliases�mbr)� table_type�layout� overwrite�ext4)�label� filesystem�device)� disk_setup�fs_setupaU #cloud-boothook #!/bin/sh fname="%(per_boot_d)s/01_smartos_vendor_data.sh" mkdir -p "${fname%%/*}" cat > "$fname" <<"END_SCRIPT" #!/bin/sh ## # This file is written as part of the default vendor data for SmartOS. # The SmartOS datasource writes the listed file from the listed metadata key # sdc:operator-script -> %(operator_script)s # user-script -> %(user_script)s # # You can view content with 'mdata-get <key>' # for script in "%(operator_script)s" "%(user_script)s"; do [ -x "$script" ] || continue echo "executing '$script'" 1>&2 "$script" done END_SCRIPT chmod +x "$fname" z/var/dbc @ s� e Zd ZdZejZejZej e j e j e j hiZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zedd� �ZdS )�DataSourceSmartOS�Joyentc C sj t j�| |||� t�| jt�|ti �tg�| _i | _ d | _ d | _ d | _ t j�| j�d��| _| �� d S )N�scripts)r � DataSource�__init__r � mergemanydict�ds_cfg�get_cfg_by_path� DS_CFG_PATH�BUILTIN_DS_CONFIG�metadata� network_data�_network_config� routes_data�os�path�join�paths� get_cpath� script_base_d�_init)�self�sys_cfg�distrorA � rH �E/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceSmartOS.pyr4 � s �� zDataSourceSmartOS.__init__c C s t j�| �}d|| jf S )Nz%s [client=%s])r r3 �__str__� md_client)rE �rootrH rH rI rJ � s zDataSourceSmartOS.__str__c C s^ | j tjkrt� | _ | j d u rd | _| jtjkr-t| j | jd | jd | jd d�| _d S d S )Nr! r r )� smartos_typer! r r )rM r �UNSET�get_smartos_environrK �jmc_client_factoryr6 �rE rH rH rI rD � s ��zDataSourceSmartOS._initc C sL t �d� d}tj�d�|dg��r$t�d�|dg�d�|dg�� dS dS )a� Mark the instance provisioning state as successful. When run in a zone, the host OS will look for /var/svc/provisioning to be renamed as /var/svc/provision_success. This should be done after meta-data is successfully retrieved and from this point the host considers the provision of the zone to be a success and keeps the zone running. z-Instance provisioning state set as successfulz/var/svc�/� provisioning�provision_successN)�LOG�debugr> r? �existsr@ �rename)rE �svc_pathrH rH rI �_set_provisioned� s ��z"DataSourceSmartOS._set_provisionedc C s� | � � i }d}| jst�d� dS | j�� s t�d| j� dS | j�� t�� D ]\}}|\}}| jj ||d�||< q)t �� D ] \}}| j� |�||< q@| j� � t j�| j�� d|d d�}t j�|d �}d t } t|� d �|| d d d � t j�|d�} t|� d�| dd d� |� d�} dt } t| | � |d s�|d r�|d |d< n|d |d< d }|d r�|d }|d s�t|| t j�| j�d�d�d� |d< t�|| jg�| _|| _|d | _|d | _|d | _| � � d S )N� zNot running on smartosFz4No metadata device '%r' found for SmartOS datasource��strip� instancesr �datar z%s/user-scriptTi� )� content_f�link�shebang�moder )r` rb rc r z%s/mdata-user-datar r r r r2 zper-boot)� user_script�operator_script� per_boot_dr r ) rD rM rU rV rK rW �open_transport�SMARTOS_ATTRIB_MAP�items�get�SMARTOS_ATTRIB_JSON�get_json�close_transportr> r? r@ rA rB � LEGACY_USER_D�write_boot_content�BUILTIN_VENDOR_DATAr r5 r: � userdata_raw�vendordata_rawr; r= rZ ) rE �md�ud�ci_noun� attribute� smartos_nounr] �data_drd � u_script_lre �u_data�u_data_frH rH rI � _get_data� s~ � ��� � � zDataSourceSmartOS._get_datac C s dt S )Nz serial (%s))� SERIAL_DEVICErQ rH rH rI �_get_subplatformN � z"DataSourceSmartOS._get_subplatformc C s | j d �|�S )Nr% )r6 rj )rE �namerH rH rI �device_name_to_deviceQ � z'DataSourceSmartOS.device_name_to_devicec C s | j tkrtS i S �N)rM �SMARTOS_ENV_KVM�BUILTIN_CLOUD_CONFIGrQ rH rH rI �get_config_objT s z DataSourceSmartOS.get_config_objc C s | j d S )Nr )r: rQ rH rH rI �get_instance_idY s z!DataSourceSmartOS.get_instance_idc C sN | j tjkr d | _ | j d u r$| jd ur$t| j| jd | jd | jd�| _ | j S )Nr r �r; r r r )r< r rN r; � convert_smartos_network_datar: r= rQ rH rH rI �network_config\ s �z DataSourceSmartOS.network_configN)�__name__� __module__� __qualname__�dsnamer rN rM rK r �NETWORKr �BOOT_NEW_INSTANCE�BOOT� BOOT_LEGACY�default_update_eventsr4 rJ rD rZ r| r~ r� r� r� �propertyr� rH rH rH rI r0 � s* ��cr0 c @ � e Zd ZdS )� JoyentMetadataFetchExceptionN�r� r� r� rH rH rH rI r� m � r� c @ r� )� JoyentMetadataTimeoutExceptionNr� rH rH rH rI r� q r� r� c @ s� e Zd ZdZe�d�Zd#dd�Zdd� Zdd � Z d d � Z d d � Z dd� Z d$dd�Z d%dd�Zd$dd�Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )&�JoyentMetadataClientz� A client implementing v2 of the Joyent Metadata Protocol Specification. The full specification can be found at http://eng.joyent.com/mdata/protocol.html z~V2 (?P<length>\d+) (?P<checksum>[0-9a-f]+) (?P<body>(?P<request_id>[0-9a-f]+) (?P<status>SUCCESS|NOTFOUND)( (?P<payload>.+))?)Nc C s |d u rt � }|| _|| _d S r� )rO rM �fp)rE rM r� rH rH rI r4 � s z JoyentMetadataClient.__init__c C s d� t�|�d��d@ �S )N�{0:08x}zutf-8� �� )�format�binascii�crc32�encode)rE �bodyrH rH rI � _checksum� s �z JoyentMetadataClient._checksumc C s� | j �|��� }t|d �t|d �kr"td�|d t|d ����| �|d �}|d |kr9td�||d ���|d |krItd�||d ���|�dd �sVt � d � d S t � |d �}t � d |� |S ) N�lengthr� z*Incorrect frame length given ({0} != {1}).�checksumz*Invalid checksum (expected: {0}; got {1}).� request_idz-Request ID mismatch (expected: {0}; got {1}).�payloadzNo value found.zValue "%s" found.) � line_regex�match� groupdict�int�lenr� r� r� rj rU rV r �b64d)rE �expected_request_id�frame� frame_data�expected_checksum�valuerH rH rI �_get_value_from_frame� s4 �� �� �� z*JoyentMetadataClient._get_value_from_framec s� g � � fdd�}d} z"| j �d�}t|�dkr t||� ��|dkr'|� W S � �|� W n tyH } z|jtjkrCt||� �|�� d}~ww q ) z� Reads a line a byte at a time until \n is encountered. Returns an ascii string with the trailing newline removed. If a timeout (per-byte) is set and it expires, a JoyentMetadataFetchException will be thrown. c s d� � ��d�S )N� �ascii)r@ �decoderH ��responserH rI �as_ascii� r� z0JoyentMetadataClient._readline.<locals>.as_asciizPartial response: '%s'T� r � N)r� �readr� r� �append�OSError�errno�EAGAIN)rE r� �msg�byte�excrH r� rI � _readline� s, �����z JoyentMetadataClient._readlinec C s | j �|�d�� | j �� d S )Nr� )r� �writer� �flush)rE r� rH rH rI �_write� s zJoyentMetadataClient._writec C sJ t �d� | �d� | �� }t �d|� |dkr td| ��t �d� d S )NzNegotiating protocol V2z NEGOTIATE V2 z read "%s"�V2_OKz'Invalid response "%s" to "NEGOTIATE V2"zNegotiation complete)rU rV r� r� r� )rE r� rH rH rI � _negotiate� s �z JoyentMetadataClient._negotiatec C s� d� t�dd��}d�||f�}|r |dt�|�� ��� 7 }d� t|�| � |�|�}t � d|� d}| j s<| � � d}| �|� | �� }|rK| �� t � d |� d |vrWd S | �||�}|S ) Nr� r r� � zV2 {0} {1} {2} z#Writing "%s" to metadata transport.FTz"Read "%s" from metadata transport.�SUCCESS)r� �random�randintr@ �base64� b64encoder� r� r� r� rU rV r� rg r� r� rm r� ) rE �rtype�paramr� � message_bodyr� � need_closer� r� rH rH rI �request� s2 ��� z JoyentMetadataClient.requestFc C s. | j d|d�}|d u r |S |r|r|�� }|S )N�GET�r� r� )r� r] )rE �key�defaultr] �resultrH rH rI rj � s zJoyentMetadataClient.getc C s$ | j ||d�}|d u r |S t�|�S )N)r� )rj �json�loads)rE r� r� r� rH rH rI rl � s z JoyentMetadataClient.get_jsonc C s | j dd�}|s g S |�d�S )N�KEYS)r� � )r� �split)rE r� rH rH rI �list s zJoyentMetadataClient.listc C s* d� dd� ||fD ���� }| jd|d�S )N� c S s g | ] }t �|�� ��qS rH )r� r� r� )�.0�irH rH rI � <listcomp> s z,JoyentMetadataClient.put.<locals>.<listcomp>�PUTr� )r@ r� r� )rE r� �valr� rH rH rI �put s ��zJoyentMetadataClient.putc C s | j r | j �� d | _ d S d S r� )r� �closerQ rH rH rI rm s �z$JoyentMetadataClient.close_transportc C s | j r| S | �� | S r� )r� rg rQ rH rH rI � __enter__ s z JoyentMetadataClient.__enter__c C s | � � d S r� )rm )rE �exc_type� exc_value� tracebackrH rH rI �__exit__ s z JoyentMetadataClient.__exit__c C s t �r� )�NotImplementedErrorrQ rH rH rI rg s z#JoyentMetadataClient.open_transport�NNr� �NF)r� r� r� �__doc__�re�compiler� r4 r� r� r� r� r� r� rj rl r� r� rm r� r� rg rH rH rH rI r� u s( � ! r� c s8 e Zd Zef� fdd� Zdd� Zdd� Zdd� Z� ZS ) �JoyentMetadataSocketClientc s t t| ��|� || _d S r� )�superr� r4 � socketpath)rE r� rM �� __class__rH rI r4 % s z#JoyentMetadataSocketClient.__init__c C s4 t � t jt j�}|�| j� |�d�| _| �� d S )N�rwb)�socket�AF_UNIX� SOCK_STREAM�connectr� �makefiler� r� )rE �sockrH rH rI rg ) s z)JoyentMetadataSocketClient.open_transportc C � t j�| j�S r� )r> r? rW r� rQ rH rH rI rW / � z!JoyentMetadataSocketClient.existsc C s d| j j| jf S )Nz%s(socketpath=%s))r� r� r� rQ rH rH rI �__repr__2 s z#JoyentMetadataSocketClient.__repr__) r� r� r� �SMARTOS_ENV_LX_BRANDr4 rg rW r � __classcell__rH rH r� rI r� $ s r� c sD e Zd Zdedf� fdd� Zdd� Zdd� Zd d � Zd d � Z� Z S ) �JoyentMetadataSerialClient� Nc s" t t| ��||� || _|| _d S r� )r� r r4 r- �timeout)rE r- r rM r� r� rH rI r4 7 s z#JoyentMetadataSerialClient.__init__c C r� r� )r> r? rW r- rQ rH rH rI rW > r z!JoyentMetadataSerialClient.existsc C sZ | j d u r#tj| j| jd�}|�� std| j ��|| _ t�|tj � | � � | � � d S )N)r zUnable to open %s) r� �serial�Serialr- r �isOpen� SystemError�fcntl�lockf�LOCK_EX�_flushr� )rE �serrH rH rI rg A s z)JoyentMetadataSerialClient.open_transportc C s� t �d� | jj}d| j_ z| �� W n ty Y nw qt �d� |dkr-d| j_n|| j_ t �d� | �d� z | �� }|dkrGW n"|d krRt �d � W q1t �d |� W n tyg t �d � Y nw q2t �d � || j_d S )NzFlushing inputg�������?Tz Input empty� z,Writing newline, expecting "invalid command"r� zinvalid command�FAILUREzGot "FAILURE". Retrying.z%Unexpected response "%s" during flushzQTimeout while initializing metadata client. Is the host metadata service running?z'Got "invalid command". Flush complete.)rU rV r� r r� r� r� �warning)rE r r� rH rH rI r K sB �� ��� z!JoyentMetadataSerialClient._flushc C s d| j j| j| jf S )Nz%s(device=%s, timeout=%s))r� r� r- r rQ rH rH rI r r s �z#JoyentMetadataSerialClient.__repr__) r� r� r� r� r4 rW rg r r r rH rH r� rI r 6 s � 'r c sN e Zd ZdZd� fdd� Zddd�Zd� fd d � Zdd d �Zdd d�Z� Z S )� JoyentMetadataLegacySerialClienta� V1 of the protocol was not safe for all values. Thus, we allowed the user to pass values in as base64 encoded. Users may still reasonably expect to be able to send base64 data and have it transparently decoded. So even though the V2 format is now used, and is safe (using base64 itself), we keep legacy support. The way for a user to do this was: a.) specify 'base64_keys' key whose value is a comma delimited list of keys that were base64 encoded. b.) base64_all: string interpreted as a boolean that indicates if all keys are base64 encoded. c.) set a key named b64-<keyname> with a boolean indicating that <keyname> is base64 encoded.r Nc s( t t| �}|�|||� d | _d | _d S r� )r� r r4 r# r$ )rE r- r rM �sr� rH rI r4 � s z)JoyentMetadataLegacySerialClient.__init__Fc C s� |rd | _ d | _d }| jd u r$| �� }d|v r!t�| �d��| _nd| _| jr)d S | j d u rm|d u r6| �� }t� }d|v rGt| �d��d��}dd� |D �D ]}t�| �|��r^|�|� qN||v rg|� |� qN|| _ d S d S )Nr$ Fr# �,c S s" g | ] }|� d �r|dd� �qS )zb64-� N)� startswith)r� �krH rH rI r� � s " zFJoyentMetadataLegacySerialClient._init_base64_keys.<locals>.<listcomp>) r# r$ r� r �is_true�_get�setr� �add�remove)rE �reset�keys�b64_keysr� rH rH rI �_init_base64_keys� s2 � �z2JoyentMetadataLegacySerialClient._init_base64_keysc s t t| �j|||d�S )N)r� r] )r� r rj )rE r� r� r] r� rH rI r � s �z%JoyentMetadataLegacySerialClient._getc C s, |t v rdS | j|d� | jrdS || jv S )NF)r T)�NO_BASE64_DECODEr! r$ r# )rE r� r rH rH rI �is_b64_encoded� s z/JoyentMetadataLegacySerialClient.is_b64_encodedc C sv t � }| j|d|d�}||u r|S | �|�r3z t�|�� ��� }W n tjy2 t � d||� Y nw |r9|� � }|S )NF)r] r� z#Failed base64 decoding key '%s': %s) �objectr r# r� � b64decoder� r� r� �ErrorrU r r] )rE r� r� r] �mdefaultr� rH rH rI rj � s �z$JoyentMetadataLegacySerialClient.get)r N)Fr� ) r� r� r� r� r4 r! r r# rj r rH rH r� rI r z s " r c C sR | d u rt |�} | d u rd S | tkrt||| d�S | tkr#t|| d�S td| ��)N)r- r rM )r� rM z"Unknown value for smartos_type: %s)rO r� r r r� � ValueError)rM r! r r � uname_versionrH rH rI rP � s �� rP c C s� ddd| g}d }zt � |�\}}t�d| |� W n t jy5 } zt�td| |f� W Y d }~nd }~ww |d u r<d S |�� S )N�filez--briefz --mime-typezscript %s mime type is %sz%Failed to identify script type for %s)r rU rV �ProcessExecutionErrorr �logexcr] )r` �cmd�f_type�_err�erH rH rI � identify_file� s ���r1 � c C s$ | s t j�|�r t �|� |rt j�|�rt �|� | s dS tj|| |d� |rH| �d�sHt|�}|dkrHtj|d� d| g�|d� t � d|� |r�z+t j�|�rVt �|� | rpt j�|�rst� t j� |�� t �||� W dS W dS W dS ty� } zt�t d|� W Y d}~dS d}~ww dS ) a Write the content to content_f. Under the following rules: 1. If no content, remove the file 2. Write the content 3. If executable and no file magic, add it 4. If there is a link, create it @param content: what to write @param content_f: the file name @param backup_d: the directory to save the backup at @param link: if defined, location to create a symlink to @param shebang: if no file magic, set shebang @param mode: file mode Because of the way that Cloud-init executes scripts (no shell), a script will fail to execute if does not have a magic bit (shebang) set for the file. If shebang=True, then the script will be checked for a magic bit and to the SmartOS default of assuming that bash. N)rc z#!z text/plainr� z #!/bin/bashzadded shebang to file %sz$failed establishing content link: %s)r> r? rW �unlink�islinkr � write_filer r1 r@ rU rV � ensure_dir�dirname�symlink�IOErrorr, )�contentr` ra rb rc r. r0 rH rH rI ro � s6 � � ���ro c C sR t �� }| d u r |d } | dkrtS |d u r t�d�}n|}|r'|�d�r'tS d S )Nr zBrandZ virtual linuxzsystem-product-name�SmartDC)r> �unamer r � read_dmi_datar r� )r) � product_namer< � system_typerH rH rI rO - s rO c s: g d�g d�ddgd��|rt |ttf�s|g}ng }|r(t |ttf�s'|g}ng }|s.g }dd� }dd � }|d d �|d d �d �� g }| D ]�}t�fd d�|�� D ��}|�d|d d�� d|v rk|�d|d i� g } |�dg �D ]�} | dkr~ddi} n{g } t�fdd�|�� D ��} | �d| d�� || �r�dnd �d |v r�|�d d �r҈ � d s҇ �fd d!�|�d"g �D �} t| �r�| d# � � d < | �d� � d i� |D ]$}t�fd$d�|�� D ��}|�d|d% i� | �|� | �d&| i� q�| �| � qs|�d'| i� |�|� qE|�r|�d(||d)�� d*|d+�S ),a Return a dictionary of network_config by parsing provided SMARTOS sdc:nics configuration data sdc:nics data is a dictionary of properties of a nic and the ip configuration desired. Additional nic dictionaries are appended to the list. Converting the format is straightforward though it does include duplicate information as well as data which appears to be relevant to the hostOS rather than the guest. For each entry in the nics list returned from query sdc:nics, we create a type: physical entry, and extract the interface properties: 'mac' -> 'mac_address', 'mtu', 'interface' -> 'name'. The remaining keys are related to ip configuration. For each ip in the 'ips' list we create a subnet entry under 'subnets' pairing the ip to a one in the 'gateways' list. Each route in sdc:routes is mapped to a route on each interface. The sdc:routes properties 'dst' and 'gateway' map to 'network' and 'gateway'. The 'linklocal' sdc:routes property is ignored. )� mac_address�mtur� �params�subnets�type) �address� broadcast�dns_nameservers� dns_search�metric� pointopointr �scoperD �network�gateway)�physical�subnet�routec S � d| v S )N�.rH ��addrrH rH rI � is_valid_ipv4� r z3convert_smartos_network_data.<locals>.is_valid_ipv4c S rQ )N�:rH rS rH rH rI � is_valid_ipv6� r z3convert_smartos_network_data.<locals>.is_valid_ipv6N)r� �gw)�ipv4�ipv6c 3 �( � | ]\}}|� d v r||fV qdS )rN NrH �r� r �v�� valid_keysrH rI � <genexpr>� � � �z/convert_smartos_network_data.<locals>.<genexpr>rN � interface)rD r� �macr@ �ips�dhcprD �dhcp4c 3 r[ )rO NrH r\ r^ rH rI r` � ra �static)rD rE rY rZ �primaryFrX c s g | ] }� � d |�r|�qS )r� rH )r� rX )�pgws�protorH rI r� � s ��z0convert_smartos_network_data.<locals>.<listcomp>�gatewaysr c 3 r[ )rP NrH r\ r^ rH rI r` � s � ���dstr rC � nameserver)rD rE �searchr� )�version�config) � isinstancer� �tuple�dictri �updaterj r� r� )r; r r r rU rW rp �nic�cfgrC �iprO � routeentsrk rP �rcfgrH )ri rj r_ rI r� C s� ����� � ��� � � �� r� c C s t �| t�S r� )r �list_from_depends� datasources)�dependsrH rH rI �get_datasource_list� s r} �__main__z Do not appear to be on smartos.r� r� c C s� ||v r|| S |t v rt | }| �|�||< || S |dkr@dD ]}t| ||� q!t|d |d |d |d d�||< || S |tv rKt| \}}n|d}}| j||d �||< || S ) Nr� )r r r r r r r r r� Fr\ )rk rl �load_keyr� rh rj )�clientr� r_ �keyname�depkeyr] rH rH rI r � s* � � � r r_ )r� r� r_ T)r z: )�indent� sort_keys� separators)NFr2 r� )NNNN)Dr� r� r� r r� �loggingr> r� r� r� r � cloudinitr r r r r �cloudinit.eventr r � getLoggerr� rU rh rk r r� �DS_NAMEr8 r" �METADATA_SOCKFILEr} �SERIAL_TIMEOUTr9 r� rp rn r3 r0 � Exceptionr� r� r� r� r r rP r1 ro rO r� �DEP_FILESYSTEMr{ r} �sys�jmc�print�exitr� �argvr� r r r_ rs �__annotations__r� �dumpsrH rH rH rI �<module> s� �� � �� �� H 0DY � � 3 � !� ��� �