Current Dir: /snap/core22/2955/usr/lib/python3/dist-packages/cloudinit/reporting/__pycache__/
[FILE] events.cpython-310.pyc [ edit | delete | rename ]
[FILE] handlers.cpython-310.pyc [ edit | delete | rename ]
[FILE] __init__.cpython-310.pyc [ edit | delete | rename ]
Viewing: /snap/core22/2955/usr/lib/python3/dist-packages/cloudinit/reporting/__pycache__/handlers.cpython-310.pyc
o
� �i@ � @ sV d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m
Z
m Z d dlm Z d dl
mZ d dlmZmZmZmZ d dlmZ e�e�ZG dd� de�ZG d d
� d
e jd �Z G d d
� d
e �Z G dd� de �Z G dd� de �Z G dd� de �Z ee e e e e f Z!e� Z"e"�#de � e"�#de � e"�#de � e"�#de � dS )� N)�datetime�timezone)�Event)�Union)�dmi� performance�
url_helper�util)� DictRegistryc @ s e Zd ZdS )�ReportExceptionN)�__name__�
__module__� __qualname__� r r �>/usr/lib/python3/dist-packages/cloudinit/reporting/handlers.pyr s r c @ s& e Zd ZdZejdd� �Zdd� ZdS )�ReportingHandlerz�Base class for report handlers.
Implement :meth:`~publish_event` for controlling what
the handler does with an event.
c C � dS )zPublish an event.Nr ��self�eventr r r �
publish_event" � z ReportingHandler.publish_eventc C r )z0Ensure ReportingHandler has published all eventsNr �r r r r �flush& r zReportingHandler.flushN)r r
r �__doc__�abc�abstractmethodr r r r r r r s
r )� metaclassc s* e Zd ZdZd� fdd� Zdd� Z� ZS )�
LogHandlerzBPublishes events to the cloud-init log at the ``DEBUG`` log level.�DEBUGc s` t t| ��� t|t�r
n |}z tt|�� �}W n ty* t �
d|� tj }Y nw || _ d S )Nz invalid level '%s', using WARN)
�superr �__init__�
isinstance�int�getattr�logging�upper� Exception�LOG�warning�WARN�level)r r+ � input_level�� __class__r r r! - s
�
zLogHandler.__init__c C s2 t �d�dd|j|jg��}|�| j|�� � d S )N�.� cloudinit� reporting)r% � getLogger�join�
event_type�name�logr+ � as_string)r r �loggerr r r r : s �zLogHandler.publish_event)r )r r
r r r! r �
__classcell__r r r- r r * s
r c @ s e Zd ZdZdd� ZdS )� PrintHandlerz Print the event as a string.c C s t |�� � d S �N)�printr7 r r r r r D s zPrintHandler.publish_eventN)r r
r r r r r r r r: A s r: c sB e Zd Z d
� fdd� Zdd� Zdd� Zdd � Z� ZS ) �WebHookHandlerNc s� t t| ��� t||||g�r tj||||d�}|j| _ntj| _|| _|| _|| _ t
� � | _ t
� | _t�� | _tj| jd�| _d| j_| j�� d S )N)� consumer_key� token_key� token_secret�consumer_secret��targetT)r r= r! �anyr �OauthUrlHelper�readurl�endpoint�timeout�retriesr �fetch_ssl_details� ssl_detailsr �flush_requested�queue�Queue� threading�Thread�process_requests�event_processor�daemon�start) r rG r> r? r@ rA rH rI � oauth_helperr- r r r! I s&
�
zWebHookHandler.__init__c
C s� d} | j �� r'|dkr't�d� | j�� s%| j�� | j�� | j�� rd}| jjdd�}z@z| j |d |d |d |d |d d d
� d}W n t
yg } zt�d |d |� |d7 }W Y d }~nd }~ww W | j�� n| j�� w q) Nr T� zNMultiple consecutive failures in WebHookHandler. Cancelling all queued events.��block� � � F)�datarH rI rK � log_req_respz0Failed posting event: %s. This was caused by: %s) rL �is_setr( r) rM �empty�
get_nowait� task_done�getrF r' )r �consecutive_failed�args�er r r rQ j sD �
�������z WebHookHandler.process_requestsc C s@ |� � }t�d| j|� | j�| jt�|�| j| j | j
f� d S )Nz Queuing POST to %s, data: %s) �as_dictr( �debugrG rM �put�json�dumpsrH rI rK )r r �
event_datar r r r � s ���z WebHookHandler.publish_eventc C s, | j �� t�d� | j�� | j �� d S )Nz(WebHookHandler flushing remaining events)rL �setr( rg rM r3 �clearr r r r r � s
zWebHookHandler.flush)NNNNNN)r r
r r! rQ r r r9 r r r- r r= H s �!&r= c s� e Zd ZdZdZdZdZee ZdZe e
j dd��Z dZ
d Zd
Zd Zd Zd
Zedf� fdd� Zedd� �Zdd� Zedd� �Zejdd� �Zdd� Zdd� Zd d � Z d d � Z d d!� Z d"d#� Z d$e d%e d&dfd'd(�Z d)d*� Z!d+d,� Z"d-d.� Z#d/d0� Z$� Z%S )1�HyperVKvpReportingHandlera,
Reports events to a Hyper-V host using Key-Value-Pair exchange protocol
and can be used to obtain high level diagnostic information from the host.
To use this facility, the KVP user-space daemon (hv_kvp_daemon) has to be
running. It reads the kvp_file when the host requests the guest to
enumerate the KVP's.
This reporter collates all events for a module (origin|name) in a single
json string in the dictionary.
For more information, see
https://technet.microsoft.com/en-us/library/dn798287.aspx#Linux%20guests
i i i �
CLOUD_INITr )r# �msg�result�msg_i)�,�:z/var/lib/hyperv/.kvp_pool_1FNc s� t t| ��� || _t�| j� || _t�� | _| � � | _
d� | j | j
�| _
| j| _t�� | _tj| jd�| _d| j_| j�� d S )Nz{0}|{1}rB T)r rn r! �_kvp_file_path�_truncate_guest_pool_file� _event_typesrM rN �q�_get_incarnation_no�incarnation_no�format� EVENT_PREFIX�event_key_prefix� ZERO_GUID�_vm_idrO �Lock� _vm_id_lockrP �_publish_event_routine�publish_threadrS rT )r �
kvp_file_path� event_typesr- r r r! � s$ �
�
�z"HyperVKvpReportingHandler.__init__c
C s� | j rdS t�� tt�� � }zFz tj�|�|k r.t|d�� W d � n1 s)w Y W n t t
fyI } z t � d|� W Y d}~n d}~ww W d| _ dS W d| _ dS d| _ w )a
Truncate the pool file if it has not been truncated since boot.
This should be done exactly once for the file indicated by
KVP_POOL_FILE_GUEST constant above. This method takes a filename
so that we can use an arbitrary file during unit testing.
Since KVP is a best-effort telemetry channel we only attempt to
truncate the file once and only if the file has not been modified
since boot. Additional truncation can lead to loss of existing
KVPs.
N�wz$failed to truncate kvp pool file, %sT)
� _already_truncated_pool_file�time�floatr �uptime�os�path�getmtime�open�OSError�IOErrorr( r) )�cls�kvp_file� boot_timere r r r rv � s" �����
�z3HyperVKvpReportingHandler._truncate_guest_pool_filec C s@ t �� }z tt�� t|� �W S ty t�d|� Y dS w )z�
use the time passed as the incarnation number.
the incarnation number is the number which are used to
distinguish the old data stored in kvp and the new data.
z"uptime '%s' not in correct format.r )r r� r# r� r� �
ValueErrorr( r) )r �
uptime_strr r r ry � s �z-HyperVKvpReportingHandler._get_incarnation_noc C s� | j �8 | j| jkr1zt�d�}|r|�� | _W n ty0 } z t�d|� W Y d}~nd}~ww | jW d � S 1 s>w Y dS )z�Get vm_id, with DMI system-uuid fallback.
If vm_id hasn't been set explicitly, falls back to reading
the system-uuid from DMI data.
z system-uuidz(Failed reading system-uuid for vm_id: %sN) r� r r~ r �
read_dmi_data�lowerr' r( rg )r � system_uuidre r r r �vm_id s
���$�z HyperVKvpReportingHandler.vm_idc C s2 | j � || _W d � dS 1 sw Y dS )zSet vm_id directly.N)r� r )r �valuer r r r� s "�c c s� � t | jd��>}t�|tj� |�|� |�| j�}t|�| jkr6| � |�}|V |�| j�}t|�| jks!t�|tj
� W d � dS 1 sHw Y dS )z-iterate the kvp file from the current offset.�rbN) r� ru �fcntl�flock�LOCK_EX�seek�read�HV_KVP_RECORD_SIZE�len�_decode_kvp_item�LOCK_UN)r �offset�f� record_data�kvp_itemr r r �
_iterate_kvps s �
�"�z'HyperVKvpReportingHandler._iterate_kvpsc C s d� | j|j|j| jt�� �S )z�
the event key format is:
CLOUD_INIT|<incarnation number>|<event_type>|<event_name>|<vm_id>
|<uuid>[|subevent_index]
z{0}|{1}|{2}|{3}|{4})r{ r} r4 r5 r� �uuid�uuid4r r r r �
_event_key' s �z$HyperVKvpReportingHandler._event_keyc C s* t �d| j| jf |�d�|�d��}|S )Nz%ds%ds�utf-8)�struct�pack� HV_KVP_EXCHANGE_MAX_KEY_SIZE� HV_KVP_EXCHANGE_MAX_VALUE_SIZE�encode�r �keyr� r\ r r r �_encode_kvp_item5 s ��� z*HyperVKvpReportingHandler._encode_kvp_itemc C sd t |�}|| jkrtd�|| j���|d| j� �d��d�}|| j| j� �d��d�}||d�S )Nz$record_data len not correct {0} {1}.r r� � )r� r� )r� r� r r{ r� �decode�strip)r r� �record_data_len�k�vr r r r� A s$
�� �
��
z*HyperVKvpReportingHandler._decode_kvp_itemc C s� t �d| j� ���C t| jd��$}t�|tj� |D ]}|�|� q|�� t�|tj � W d � n1 s7w Y W d � d S W d � d S 1 sOw Y d S )Nz
Appending �ab)
r �Timedru r� r� r� r� �writer r� )r r� r� r\ r r r �_append_kvp_itemX s � P�z*HyperVKvpReportingHandler._append_kvp_itemc C s� || j = t�|�}|dt|�d � }d}g }d| j d } ||| j< d|| j < tj|| jd�}| jt|� d } |�|d j| j |d | � d
��}
d �||�} |� | �
| |
�� |d7 }|| d � }|sh |S q ) NrY r �"z":""T� ��
separators� z"{key}":"{desc}")r� �descz{}|{}) �MSG_KEYri rj r� � DESC_IDX_KEY�JSON_SEPARATORS�HV_KVP_AZURE_MAX_VALUE_SIZE�replacer{ �appendr� ) r r� � meta_data� description� des_in_json�i� result_array�message_place_holder�data_without_desc�
room_for_descr� �subkeyr r r � _break_downb s8
���� �z%HyperVKvpReportingHandler._break_downr� r� �returnc C sf t |�| jkr|d| jd � }| �||�g}z| �|� W dS ttfy2 t�d||� Y dS w )zJWrite KVP key-value.
Values will be truncated as needed.
r rY z failed posting kvp=%s value=%sN)r� r� r� r� r� r� r( r) r� r r r � write_key� s �z#HyperVKvpReportingHandler.write_keyc C s� | � |�}|j|jt�|jtj��� d�}t || j
�r!|j || j
< t |d�r+|j |d< |j
|| j< tj|| jd�}t|�| jkrH| �|||j
�S | �||�}|gS )z�
encode the event into kvp data bytes.
if the event content reaches the maximum length of kvp value.
then it would be cut to multiple slices.
)r5 �type�ts�durationr� )r� r5 r4 r �
fromtimestamp� timestampr �utc� isoformat�hasattr�
RESULT_KEYrq r� r� r� ri rj r� r� r� r� r� )r r r� r� r� r\ r r r �
_encode_event� s$
��
z'HyperVKvpReportingHandler._encode_eventc
C s
d}zv| j jdd�}|d7 }g }|d ur:|| �|�7 }z
| j jdd�}|d7 }W n tjy5 d }Y nw |d usz0z| �|� W n ttfy\ } z t� d|� W Y d }~nd }~ww W t
|�D ]}| j � � qbn
t
|�D ]}| j � � qow W n
t y� Y d S w q)NTr rW rY Fz failed posting events to kvp, %s)
rx rb r� rM �Emptyr� r� r� r( r) �rangera �EOFError)r �items_from_queuer � encoded_datare �_r r r r� � s> ����� � � ��z0HyperVKvpReportingHandler._publish_event_routinec C s& | j r |j| j v r| j�|� d S d S r; )rw r4 rx rh r r r r r � s �z'HyperVKvpReportingHandler.publish_eventc C s t �d� | j�� d S )Nz0HyperVReportingHandler flushing remaining events)r( rg rx r3 r r r r r � s
z HyperVKvpReportingHandler.flush)&r r
r r r� r� r� r� r| �strr� �UUIDr~ r� r� r� r� �KVP_POOL_FILE_GUESTr� r! � classmethodrv ry �propertyr� �setterr� r� r� r� r� r� r� r� r� r r r9 r r r- r rn � sD �
rn r6 r< �webhook�hyperv)$r r� ri r% r� rM r� rO r� r� r r r �typingr r0 r r r r �cloudinit.registryr
r2 r r( r' r �ABCMetar r r: r= rn � HandlerType�available_handlers�
register_itemr r r r �<module> sJ
` 1��