Your IP : 216.73.217.13
U
�Ad�$ � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZmZ dd lmZ dd
lmZmZmZmZ d d gZd
ZG dd� de �Zedd�edd�edd�gZedd�edd�edd�ed d �ed d �ed d!�ed"d#�ed$d%�ed&d'�ed(d)�ed*d)�ed+d,�g Z e d-�d.d/�Z e d-�d0d1�Z dFd2d3�Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%dGe&d>�d?d@�Z'dAdB� Z(dCdD� Z)e*dEk�r�e�+e)� � dS )HzGDefine 'collect-logs' utility and handler to include in cloud-init cmd.� N)�datetime)�Path)�
NamedTuple)�read_cfg_paths)�Paths)�ProcessExecutionError�subp)�tempdir)�chdir�copy�
ensure_dir�
write_filez/var/log/cloud-init.logz /var/log/cloud-init-output.logz/run/cloud-initc @ s e Zd ZU eed< eed< dS )�
ApportFile�pathZlabelN)�__name__�
__module__� __qualname__�str�__annotations__� r r �:/usr/lib/python3/dist-packages/cloudinit/cmd/devel/logs.pyr s
r z(/var/log/installer/autoinstall-user-dataZAutoInstallUserDataz/autoinstall.yamlZAutoInstallYAMLz'/etc/cloud/cloud.cfg.d/99-installer.cfgZInstallerCloudCfgz//var/log/installer/ubuntu_desktop_installer.logZUdiLogz-/var/log/installer/subiquity-server-debug.logZSubiquityServerDebugz-/var/log/installer/subiquity-client-debug.logZSubiquityClientDebugz%/var/log/installer/curtin-install.logZ CurtinLogz0/var/log/installer/subiquity-curtin-install.confZCurtinInstallConfigz8/var/log/installer/curtin-install/subiquity-initial.confZCurtinConfigInitialz:/var/log/installer/curtin-install/subiquity-curthooks.confZCurtinConfigCurtHooksz8/var/log/installer/curtin-install/subiquity-extract.confZCurtinConfigExtractz=/var/log/installer/curtin-install/subiquity-partitioning.confZCurtinConfigPartitioningz(/var/log/installer/curtin-error-logs.tarZ CurtinErrorz$/var/log/installer/curtin-errors.tarz(/var/log/installer/block/probe-data.jsonZ ProbeData)�returnc C s t � } | �d�S )NZ userdata_raw)r Z
get_ipath_cur��pathsr r r �_get_user_data_fileM s r c C s t � } | �d�S )N�data)r Z get_cpathr r r r �_get_cloud_data_pathR s r c C s` | st jddd�} | jdddddd d
� | jd d d
dd� t� }| jdddddd�|�d� | S )a2 Build or extend and arg parser for collect-logs utility.
@param parser: Optional existing ArgumentParser instance representing the
collect-logs subcommand which will be extended to support the args of
this utility.
@returns: ArgumentParser with proper argument configuration.
� collect-logsz)Collect and tar all cloud-init debug info)�progZ descriptionz --verbosez-v�countr � verbosityzBe more verbose.)�action�default�dest�helpz --tarfilez-tzcloud-init.tar.gzzOThe tarfile to create containing all collected logs. Default: cloud-init.tar.gz)r" r$ z--include-userdataz-uF�
store_true�userdatazPOptionally include user-data from {0} which could contain sensitive information.)r" r! r# r$ )�argparse�ArgumentParser� add_argumentr �format)�parser�user_data_filer r r �
get_parserW s> ��� �� r- c C s* dg}t �� dkr&|�ti �jd � |S )z>Return a list of files to ignore for /run/cloud-init directoryzhook-hotplug-cmdr Zinstance_data_sensitive)�os�getuid�appendr Zlookups)�curdir�filesZ
ignored_filesr r r � _copytree_rundir_ignore_files� s
� r3 c
C st zt | �\}}W n@ tk
rP } z"t|t|�� td| d|� W 5 d}~X Y n X t||� td| d|� |S dS )zCHelper which runs a command and writes output or error to filename.zcollecting %s failed.
� Nz
collected %s
)r r r
r �_debug)�cmd�filename�msgr �out�_�er r r � _write_command_output_to_file� s "
r< c C s ||krt j�| � d S )N)�sys�stderr�write)r8 �levelr r r r r5 � s r5 c C s< t j�| �r(t| |� td| d|� ntd| d|� d S )Nzcollected file: %s
r4 zfile %s did not exist
� )r. r �isfiler r5 )r Zout_dirr r r r �
_collect_file� s
rC c C s� t D ]:}t| |j �j}|�� s,tt|�� t|jt|�|� q|r�tD ]:}t| |j �j}|�� sptt|�� t|jt|�|� qHdS )z'Obtain subiquity logs and config files.N) �INSTALLER_APPORT_FILESr r �parent�existsr r rC � INSTALLER_APPORT_SENSITIVE_FILES)�log_dir�include_userdatar Zsrc_fileZdestination_dirr r r �collect_installer_logs� s rJ )rI c C sl |r t �� dkr tj�d� dS t j�| �} t�� � � �
d�}t dd���}t j� ||�}t
ddgt j� |d �d
|�}t
d d d
dgt j� |d�d|�}|s�|r�|nd}td| d|� t
dgt j� |d�d|� t
ddddgt j� |d�d|� tD ]}t|||� q�|�r t� }t|||� t|||� t j� |d�} t| � t j�t��r�z tjtt j� | d�td � W nF tjk
�r� }
z$tj�d � tj�t|
�d � W 5 d }
~
X Y nX td t d|� ntd!t d|� t j�t j� td"���r tt � �} | �� �r t�t| �t|t| � �� t |��" t d#d$| |� |d% d&�g� W 5 Q R X W 5 Q R X tj�d'| � dS )(z�Collect all cloud-init logs and tar them up into the provided tarfile.
@param tarfile: The path of the tar-gzipped file to create.
@param include_userdata: Boolean, true means include user-data.
r zMTo include userdata, root user is required. Try sudo cloud-init collect-logs
r4 zcloud-init-logs-%Y-%m-%dz/tmp)�dirz
cloud-initz --version�versionzcloud-init --versionz
dpkg-queryz--showz-f=${Version}
z dpkg-versionz dpkg versionz
not-availablez!collected cloud-init version: %s
Zdmesgz dmesg.txtz dmesg outputZ
journalctlz--boot=0z-oz
short-precisez journal.txtz systemd journal of current boot�run)�ignorez(Failed collecting file(s) due to error:
�
Nzcollected dir %s
z directory '%s' did not exist
ZdisabledZtarZczvf�/� z Wrote %s
) r. r/ r= r> r? r �abspathr Zutcnow�date�strftimer �joinr< r5 �CLOUDINIT_LOGSrC r rJ r rF �CLOUDINIT_RUN_DIR�shutilZcopytreer3 �Errorr r r r
r �replace) �tarfilerI r rH Ztmp_dirrL Zdpkg_ver�logr, Zrun_dirr; Zcloud_data_dirr r r � collect_logs� s� � �
� �
� �
&�
�
0r] c C s t |j|j|j�S )z:Handle calls to 'cloud-init collect-logs' as a subcommand.)r] r[ r& r )�name�argsr r r �handle_collect_logs_args
s r` c C s t � } td| �� �S )z4Tool to collect and tar all cloud-init related logs.r )r- r` �
parse_args)r+ r r r �main s rb �__main__)N)r ),�__doc__r' r. rX r= r �pathlibr �typingr Zcloudinit.cmd.develr Zcloudinit.helpersr Zcloudinit.subpr r Zcloudinit.temp_utilsr Zcloudinit.utilr
r r r
rV rW r rG rD r r r r- r3 r< r5 rC rJ �boolr] r` rb r �exitr r r r �<module> s� �� � �������%
.
T