← Back
Editing: __init__.cpython-38.pyc
U t5Y^�/ � @ s� d dl Z d dlZd dlZzd dlmZ W n ek r@ eZY nX d dlZd dlZd dl Z d dl Z d dl Z d dl Z d dl Z d dlZzd dlmZ W n ek r� d dlmZ Y nX ddlmZ dZejdejd� e jd d d �Zejd d ddd� ejdddddd� ejdddddd � ejd d d d d!� de_g Zd"d#� Z d$d%� Z d&d'� Z dEd)d*�Z d+d,� Z d-d.� Z!d/d0� Z"dFd1d2�Z#d3d4� Z$d5d6� Z%d7d8� Z&d9d:� Z'd;d<� Z(dGd=d>�Z)d?d@� Z*dAdB� Z+dCdD� Z,dS )H� N)�JSONDecodeError)� quote_plus� )�VERSION�lpz%%(asctime)s %(levelname)s %(message)s)�format�levelz9Authorize SSH public keys from trusted online identities.z ssh-import-id)Z description�progz-oz--outputZFILEz5Write output to file (default ~/.ssh/authorized_keys))�metavar�helpz-rz--remove� store_trueFz&Remove a key from authorized keys file)�action�defaultr z-uz --useragentZ USERAGENT� z$Append to the http user agent string)r r r �userids�+ZUSERIDzUser IDs to import)�nargsr r c C s$ t D ]} tj�| �rt�| � qdS )z Cleanup tempfiles N)� TEMPFILES�os�path�exists�unlink)�f� r �8/usr/lib/python3/dist-packages/ssh_import_id/__init__.py�cleanupD s r c C s t �| � t� t�d� dS )z. The only thing in Perl worth keeping r N)�logging�errorr �sys�exit)�msgr r r �dieM s r! c C s� | sdS t | �dk rdS tjddd�\}}t�|� t�|d�� }|�d�| �� |�d� W 5 Q R X t j d d d |gt j d �}|� d�\}}|j r�dS t�|� |�� }|r�t |�d k r�dS g }|�� D ] } |�t| �d��� �� q�|S )z^ Get the fingerprint for an SSH public key Returns None if not valid key material N� zssh-auth-key-checkz.pub)�prefix�suffix�w� � z ssh-keygenz-lz-f)�stdout� zutf-8)�len�tempfileZmkstempr �appendr �fdopen�write�join� subprocess�Popen�PIPEZ communicate� returncoder �split�str�decode�strip) �fieldsZtempfdZtempnameZtempfZ keygen_procZ keygen_out�_Z keygen_fields�out�kr r r �key_fingerprintV s6 � � r< �a+c C s | dkr dS t | |�S )zI Open output for writing, supporting either stdout or a filename �-F)�open)�name�moder r r � open_outputv s rB c C sv | dkr dS t j�| �r&t j�| �}nd}t j�|�sVt �d�}t �|d� t �|� t j�|�rfdS td| � dS )z9 Ensure that the keyfile parent directory exists r> T�.�? i� z*Parent directory not found for output [%s]F)r r �dirnamer �umask�makedirs�isdirr! )�keyfileZ parent_dirrF r r r �assert_parent_dir s rJ c C sn t tjj�} | dks tj�| �s&g }nDz"t| d��}|�� }W 5 Q R X W n t k rh t d| � Y nX |S )zI Locate key file, read the current state, return lines in a list r> �rz'Could not read authorized key file [%s]) � get_keyfile�parser�options�outputr r r r? � readlines�OSErrorr! )rI �lines�fpr r r � read_keyfile� s rT c C s� t tjj�}|dkrF| D ] }|rtj�|� tj�d� qtj�� nDt|�r�t ||��,}| D ] }|� � r^|�|� |�d� q^W 5 Q R X dS )z, Locate key file, write lines to it r> z N) rL rM rN rO r r( r. �flushrJ r? r7 )� keyfile_linesrA Z output_file�liner r r r � write_keyfile� s rX c C sD | s@t j�d�r t jd }nt j�dt�� �}t j�|dd�} | S )zEReturn 'path' if true, else a path to current user's authorized_keys.�HOME�~z.sshZauthorized_keys)r �environ�getr � expanduser�getpassZgetuserr/ )r �homer r r rL � s rL c C s d� | d | d | d g�S )z7 Build a string that uniquely identifies a key r& r r ���)r/ )rS r r r �fp_tuple� s ra c C sB g }| D ]"}t |�� �}|r|�t|�� qt�dd�|�� |S )z3 Return a list of uniquely identified keys z"Already have SSH public keys: [%s]r& )r< r4 r, ra r �debugr/ )rV �keysrW �ssh_fpr r r �key_list� s re c C s4 | dkrt ||�S | dkr$t||�S td| � dS )zP Call out to a subcommand to handle the specified protocol and username r Zghz>ssh-import-id protocol handler %s: not found or cannot executeN)� fetch_keys_lp� fetch_keys_ghr! )�proto�username� useragentr r r � fetch_keys� s �rk c C s� t t� �}g }g }d| |f }t| ||��d�D ]�}|�� }|�� }|�|� t|�} | r0t| �|kr�t� d| dd� | dd� � |�|� q0|�d� |�� |�|� t� d| dd� | dd� � q0t |d � |S ) zZ Import keys from service at 'proto' for 'username', appending to output file z# ssh-import-id %s:%sr' zAlready authorized %sNr" r` r& zAuthorized key %sr= ) re rT rk r4 r7 r, r< ra r �infor/ rX ) rh ri rj Z local_keys�resultrV �comment_stringrW r8 rd r r r � import_keys� s* � " ro c C sz d| |f }g }g }t � D ]P}|�|�r`t|�� �}t�d|dd� |dd� � |�|� q|�|� qt|d� |S )zN Remove keys from the output file, if they were inserted by this tool z# ssh-import-id %s:%s zRemoved labeled key %sNr" r` r% )rT �endswithr< r4 r rl r, rX )rh ri rn Z update_linesZremovedrW rd r r r � remove_keys s rq c C sh dt }dtjjtjjtjjf }d�t�� �}dt � � d t � � d t � � d f }d||||| f S ) z/" Construct a useful user agent string zssh-import-id/%szpython/%d.%d.%d�/z%s/%s/%sr r) � z%s %s %s %s %s) r r � version_info�major�minor�micror/ �distroZlinux_distributionr �uname)ZextraZ ssh_import_id�pythonZ linux_distry r r r � user_agent s �&r{ c C s\ d}�z t �dd �}|d kr�t j�|�r�zt|��� }W n tk rX td| ��Y nX zt� |�}W n t k r� td| ��Y nX |� dd �t | � }n|d k r�|t | � }|d kr�dt | � }dt |�i}tj |d|d�}|jd k�rd }|jd k�rd }t|d |j| f � t|j�} W n0 tk �rV } ztt| �� W 5 d } ~ X Y nX | S )Nz/etc/ssh/ssh_import_idZURLzFailed to read %sz File %s did not have valid JSON.z"https://launchpad.net/~%s/+sshkeys� User-AgentT)�verify�headers�� z!Requesting Launchpad keys failed.� zLaunchpad user not found.� status_code=%d user=%s)r �getenvr r r? �readrQ � Exception�json�loadsr r\ r r{ �requests� status_coder! r5 �text) Zlpidrj Z conf_file�url�contents�confr~ Zresponser rc �er r r rf ) s< � rf c C s� d}d}d}z�dt | � }dt|�i}tj||dd�}|j}t�|�} |jdkr�d } |jd krhd | } n|j�|�d kr�d | } t | d|j| f � | D ] } |d| d | | d f 7 }q�W n. t k r� } zt t | �� W 5 d } ~ X Y nX |S )Nzx-ratelimit-remainingz.https://developer.github.com/v3/#rate-limitingr z$https://api.github.com/users/%s/keysr| T)r~ r} r z Requesting GitHub keys failed.r� z&Username "%s" not found at GitHub API.�0z6GitHub REST API rate-limited this IP address. See %s .r� z%s %s@github/%s �key�id) r r{ r� r\ r� r� r� r� r~ r! r� r5 ) Zghidrj Zx_ratelimit_remainingZhelp_urlrc r� r~ Zrespr� �datar Zkeyobjr� r r r rg N s. � rg c C s0 g } z�t �d� t�� t_g }tjjD ]�}|�d�}t|�dkrJ|\}}n$t|�dkrbt| }}n t d| � tjj r�t ||�}|� |� d}n t ||tjj�}|� |� d}|s&| �|� q&t�dt|�|� W n0 tk �r } zt t|�� W 5 d }~X Y nX t� | �r"t d d �| � � t�d � d S ) N� �:r) r zInvalid user ID: [%s]ZRemovedZ Authorizedz[%d] SSH keys [%s]z No matching keys found for [%s]�,r )r rF rM � parse_argsrN r r4 r* � DEFAULT_PROTOr! �removerq �extendro rj r, r rl r� r5 r r/ r r ) �errorsrc ZuseridZ user_piecesrh ri Zchangesr r� r r r �mainh s@ � r� )r= )N)r )-�argparser^ r� Z json.decoderr � ImportError� ValueErrorr r rx �statr0 r r+ r� Z urllib.parser Zurllib�versionr r� Z basicConfig�INFO�ArgumentParserrM � add_argumentrN r r r! r< rB rJ rT rX rL ra re rk ro rq r{ rf rg r� r r r r �<module> s� �� � � � � %
Save File
Cancel