3
\�me|& �
@ s d dl Z d dlZd dlZd dlmZmZmZ d dlmZm Z m
Z
ej e �Z
dZdZdZdZeegZdgZd Zd
Zd d d
dddddddddd� Zeeej� �dd� d�ZG dd� de jd �Z d d � Z d d � Z d!d"� Z d#d$� Z d%d&� Z!d'd(� Z"d)d*� Z#d+d,� Z$e
fd-d.�Z%d1d/d0�Z&dS )2� N)�importer�
type_utils�util)� FREQUENCIES�
PER_ALWAYS� PER_INSTANCEztext/x-not-multipartzapplication/octet-streamZ__end__Z __begin__ztext/part-handlerzpart-handler-%03dz part-%03dztext/x-include-urlztext/x-include-once-urlztext/x-shellscriptztext/cloud-configztext/cloud-boothookztext/cloud-config-archiveztext/cloud-config-jsonpz text/jinja2ztext/x-shellscript-per-bootz text/x-shellscript-per-instanceztext/x-shellscript-per-once) z#includez
#include-oncez#!z
#cloud-configz
#part-handlerz#cloud-boothookz#cloud-config-archivez#cloud-config-jsonpz## template: jinjaztext/x-shellscript-per-bootz text/x-shellscript-per-instanceztext/x-shellscript-per-oncec C s dt | � S )Nr )�len)�e� r
� /usr/lib/python3.6/__init__.py�<lambda>> s r )�keyc @ s4 e Zd Zd dd�Zdd� Zdd� Zejdd � �Zd
S ) �Handler� c C s || _ || _d S )N)�handler_version� frequency)�selfr �versionr
r
r �__init__C s zHandler.__init__c C s dt j| �| j� f S )Nz%s: [%s])r Zobj_name�
list_types)r r
r
r �__repr__G s zHandler.__repr__c C s, t | d�std��ndd� t| d�D �S d S )N�prefixesz#Missing prefixes subclass attributec S s g | ] }t | �qS r
)�INCLUSION_TYPES_MAP)�.0�prefixr
r
r �
<listcomp>P s z&Handler.list_types.<locals>.<listcomp>)�hasattr�NotImplementedError�getattr)r r
r
r r J s
zHandler.list_typesc O s
t � �d S )N)r )r �args�kwargsr
r
r � handle_partT s zHandler.handle_partN)r ) �__name__�
__module__� __qualname__r r r �abc�abstractmethodr! r
r
r
r r B s
r )� metaclassc C s | j }|tkp |tko |tks"d S y| j}t|�}W n tttfk
rR d}Y nX |d }yxtj d| ||||� |dkr�| j
||||||� nB|dkr�| j
|||||� n&|dkr�| j
||||� n td| ��W n, t k
�r t j
td| ||||� Y nX d S )N� z Content-Typez1Calling handler %s (%s, %s, %s) with frequency %s� r zUnknown module version %sz8Failed calling handler %s (%s, %s, %s) with frequency %s)r r r r �int� TypeError�
ValueError�AttributeError�LOG�debugr! � Exceptionr �logexc) �mod�data�filename�payloadr �headersZmod_freqZmod_ver� content_typer
r
r �run_partY sF
r8 c C s dt i}t| |d d ||� d S )Nz Content-Type)�
CONTENT_STARTr8 )r2 r3 r r6 r
r
r �
call_begin� s r: c C s dt i}t| |d d ||� d S )Nz Content-Type)� CONTENT_ENDr8 )r2 r3 r r6 r
r
r �call_end� s r< c
C s� | d }t | }| d }tjj| d d| �}|jd�s@d| }tj||d� | d }y<ttj |��} t
| | d |� |j | d
d � |d | d< W n$ t k
r� tj
td
||� Y nX d S )NZ handlercountr Z
handlerdirz%sz.pyz%s.pyi� �handlersr3 T)Z initializedr( z7Failed at registering python file: %s (part handler %s))�PART_HANDLER_FN_TMPL�os�path�join�endswithr Z
write_file�
fixup_handlerr �
import_moduler: �registerr0 r1 r. )
ZpdataZ_ctypeZ _filenamer5 Zcurcount�modnamer Zmodfnamer= r2 r
r
r �walker_handle_handler� s&
rG c C s� y\t | t�r | jdd�d }n$| d d| � jdd�jdd�d }t|�|krZ|d |� }W n tk
r| | d|� }Y nX |S )N�
r( r � �ignore)�errors)�
isinstance�str�split�decoder �UnicodeDecodeError)Zblob�size�startr
r
r �_extract_first_or_bytes� s
$ rS c C sZ y
| j d�S ttfk
rB y
| j d�S tk
r< Y nX Y n tk
rT Y nX | S )NZ
string_escapeZunicode_escape)�encode� LookupErrorr+ r- )�textr
r
r �_escape_string� s
rW c C s� |d }|| j d�kr&tjd|� d S |tkr@t| |||� d S | d }||krpt|| | d ||| d |� nN|r�t|d�}dt|� }|tkr�tj d ||� q�tj d
||� n tjd |� d S ) Nz Content-TypeZexcludedz content_type "%s" is excludedr= r3 r � z'%s...'z)Unhandled non-multipart (%s) userdata: %sz0Unhandled unknown content-type (%s) userdata: %szEmpty payload of type %s)
�getr. r/ �PART_CONTENT_TYPESrG r8 rS rW �NOT_MULTIPART_TYPE�warning)r3 r4 r5 r6 r7 r= rR Zdetailsr
r
r �walker_callback� s:
r] c C s� d}x�| j � D ]t}|j� dkr q|j� }|d kr4t}|j� }|sHt| }t|�}tj|� ||d< t j
|�}|||||� |d }qW d S )Nr Z multipartz Content-Typer( ) �walkZget_content_maintypeZget_content_type�
OCTET_TYPE� get_filename� PART_FN_TPL�dictr. r/ r Zfully_decoded_payload) �msg�callbackr3 Zpartnum�partZctyper4 r6 r5 r
r
r r^ s
r^ c C sR t | d�st| dd� t | d�s.t| d|� n | j}|rN|tkrNtjd| |� | S )Nr r( r z&Handler %s has an unknown frequency %s)r �setattrr r r. r\ )r2 Zdef_freqZfreqr
r
r rC s
rC c C sT yt j| �j� }W n tk
r& |S X |j� }x tD ]}|j|�r6t| S q6W |S )N)r Z
decode_binary�lowerrP �lstrip�INCLUSION_SRCH�
startswithr )r5 �defaultZ
payload_lcrV r
r
r �type_from_starts_with# s
rl )N)'r% Zloggingr? Z cloudinitr r r Zcloudinit.settingsr r r Z getLoggerr" r. r[ r_ r; r9 ZCONTENT_SIGNALSrZ r> ra r �sorted�list�keysri �ABCMetar r8 r: r< rG rS rW r] r^ rC rl r
r
r
r �<module> sL
0)