File: //lib/python3.9/site-packages/sos/upload/targets/__pycache__/redhat.cpython-39.pyc
a
�:�h�( � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z
zd dlZdZW n eyv dZY n0 G d d
� d
e�ZdS ) � N)� UploadTarget)�
convert_bytes�TIMEOUT_DEFAULT)�DeviceAuthorizationClass)�
RHELPolicy)�_sosTFc s� e Zd ZdZdZdZdZd'� fdd� ZdZd Z e Z
d
Z dZ d Z
d d
� Z� fdd�Zdd� Zdd� Zdd� Zdd� Zd(dd�Zdd � Zd d � Zd d � Zd)� fd!d"� Zd#d$� Z� fd%d&�Z� ZS )*�RHELUploadTargetzVhttps://sso.redhat.com/auth/realms/redhat-external/protocol/openid-connect/auth/devicezPhttps://sso.redhat.com/auth/realms/redhat-external/protocol/openid-connect/tokenzRed Hat Upload TargetZredhatNc s t � j|||d� d S )N)�parser�args�cmdline)�super�__init__)�selfr r
r �� __class__� �=/usr/lib/python3.9/site-packages/sos/upload/targets/redhat.pyr
! s zRHELUploadTarget.__init__z https://api.access.redhat.comz sftp://sftp.access.redhat.com�posti @c C s t | jd t�S )z.Return true if we are running in a RHEL systemZpolicy)�
isinstance�commonsr �r r r r �check_distribution- s z#RHELUploadTarget.check_distributionc s t � �|� | jd j| _d S )N� cmdlineopts)r �pre_workr �upload_directory)r Z hook_commonsr r r r 1 s zRHELUploadTarget.pre_workc C s@ | j d jr | j�td�� | j d js<| j| _| j�d� d S )Nr zVThe option --upload-user has been deprecated in favour of device authorization in RHELz&No case id provided, uploading to SFTP)r Z upload_user�ui_log�info�_�case_id� RH_SFTP_HOST�
upload_urlr r r r �prompt_for_upload_user7 s � z'RHELUploadTarget.prompt_for_upload_userc C s | j d jr | j�td�� d S )Nr zVThe option --upload-pass has been deprecated in favour of device authorization in RHEL)r Z upload_passr r r r r r r �prompt_for_upload_passwordB s �z+RHELUploadTarget.prompt_for_upload_passwordc
C s� d| j d j� d�}zX| jr$| jW S | j d jr>| j d jW S | j d jdkrV| jW S | j d jsj| jW S W n6 ty� } z | j�d|� �� W Y d }~n
d }~0 0 | j� |� �S )Nz/support/v1/cases/r z /attachments�sftpz=There was a problem while setting the remote upload target: ) r r r Zupload_protocolr � Exceptionr r � RH_API_HOST)r Z rh_case_api�er r r �get_upload_urlJ s&
� ��z RHELUploadTarget.get_upload_urlc C s d| j � �}d|iS )NzBearer Z
Authorization)�
_device_token)r Zstr_authr r r �_get_upload_https_auth^ s z'RHELUploadTarget._get_upload_https_authTc C sh d|j �d�d || �� fi}t| j| j�}|�� | _| j� d| �
� � �� t j | �
� || �� |td�S )z�If upload_https() needs to use requests.post(), use this method.
Policies should override this method instead of the base upload_https()
:param archive: The open archive file object
�file�/����/Device authorized correctly. Uploading file to )�files�headers�verify�timeout)�name�split�_get_upload_headersr �client_identifier_url�token_endpoint�get_access_tokenr( r r �get_upload_url_string�requestsr r' r) r )r �archiver0 r. �RHELAuthr r r �_upload_https_postb s ���
� �z#RHELUploadTarget._upload_https_postc C s | � � �| j�rddd�S i S )NZfalsezno-cache)Z isPrivatez
cache-control)r' �
startswithr% r r r r r4 { s
z$RHELUploadTarget._get_upload_headersc C s4 | � � �| j�rdS | � � �| j�r(dS | �| j�S )NzRed Hat Customer PortalzRed Hat Secure FTP)r' r= r% r Z_get_obfuscated_upload_urlr r r r r r8 � s
z&RHELUploadTarget.get_upload_url_stringc C sL | j �d�d }| jd jr2| jd j� d|� �}| jrHtj�| j|�}|S )z�The RH SFTP server will only automatically connect file uploads to
cases if the filename _starts_ with the case number
r+ r, r r )Zupload_archive_namer3 r r r �os�path�join)r �fnamer r r �_get_sftp_upload_name� s z&RHELUploadTarget._get_sftp_upload_namec
s� | j �d�d | �� vr"t� �� S ts.td��d}d}| js�zt| j | j
�}W n< ty� } z$dt |�v rv| j �
d� W Y d}~nd}~0 0 |�� | _| jr�| j �
d| �� � �� | jd }d}| j�r:| �� } tj|| d d
�}|jd k�rt�|j�d }t�|j�d
}n*| j �d|j� d|�� � �� | j �d� n~ddi}
tj|t�|
�d d�} | jd k�r�t�| j�} | d }| d
}| j �
td|� d��� n | j �d| j� d| �� � �� |�r�|�r�t� j||d�S td��dS )z�Override the base upload_sftp to allow for setting an on-demand
generated anonymous login for the RH SFTP server if a username and
password are not given
z//� zPpython3-requests is not installed and is required for obtaining SFTP auth token.Nzend user deniedz:Device token authorization has been cancelled by the user.r- z/support/v2/sftp/token�
)r/ r1 �� �username�tokenz$DEBUG: auth attempt failed (status: z): zUUnable to retrieve Red Hat auth token using provided credentials. Will try anonymous.Z isAnonymousT)�datar1 zUser z^ used for anonymous upload. Please inform your support engineer so they may retrieve the data.z)DEBUG: anonymous request failed (status: )�user�passwordz1Could not retrieve valid or anonymous credentials) r r3 r' r � upload_sftp�REQUESTS_LOADEDr$ r( r r5 r6 �strr r r7 r8 r% r) r9 r Z status_code�json�loads�text�debug�error�dumpsr )
r rI rJ Z_token�_userr; r& �url�retr/ ZadataZanonZrespr r r rK � sr
� �
�
��� ���� z RHELUploadTarget.upload_sftpc C s@ t j�|�}|| jkr<| j�tdt| j�� d��� | j| _ d S )NzLSize of archive is bigger than Red Hat Customer Portal limit for uploads of z via sos http upload.
)
r> r? �getsize�_max_size_requestr Zwarningr r r r )r r: �sizer r r �check_file_too_big� s
��z#RHELUploadTarget.check_file_too_bigc
s� z*| � � �| j�r | �|� t� �|�}W nj ty� } zRd}| j�| j�sN� | j� t
d|� d| j � ��� | j | _t� �|�}W Y d}~n
d}~0 0 |S )zOverride the base upload_archive to provide for automatic failover
from RHCP failures to the public RH dropbox
Fz0Upload to Red Hat Customer Portal failed due to z . Trying N) r' r= r% rZ r �upload_archiver$ r r rR r r )r r: Zuploadedr& r r r r[ � s$
���"z RHELUploadTarget.upload_archive)NNN)T)NN) �__name__�
__module__� __qualname__r5 r6 Zupload_target_nameZupload_target_idr
r% r Z _upload_urlZ_upload_methodr( rX r r r! r" r' r) r< r4 r8 rB rK rZ r[ �
__classcell__r r r r r s0
Kr )r> rN Zsos.upload.targetsr Z
sos.utilitiesr r Zsos.policies.authr Zsos.policies.distros.redhatr Zsosr r r9 rL � ImportErrorr r r r r �<module> s