/
opt
/
gsutil
/
gslib
/
utils
/
__pycache__
/
up file
home
a +(WgU � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlZddlm Z ddl m Z dd l m Z dd l mZ G d d � d e�Zd d� Zdd� Zdd� Zdd� ZdS )z=Helper functions for Split Trust Encryption Tool integration.� )�absolute_import)�print_function)�division)�unicode_literalsN)� storage_url)�execution_util)�temporary_file_util)�configc @ s e Zd ZdZdZdZdS )�StetSubcommandNamez*Enum class for available STET subcommands.�encrypt�decryptN)�__name__� __module__� __qualname__�__doc__�ENCRYPT�DECRYPT� r r �$/opt/gsutil/gslib/utils/stet_util.pyr s r c C s@ t �d��t jj�D ]&} t j�| d�}t j�|�r| S qdS )zBRetrieves STET binary from path if available. Python 2 compatible.�PATH�stetN)�os�getenv�split�path�pathsep�join�exists)Zpath_directory� binary_pathr r r �_get_stet_binary_from_path& s r c C s� t �ddt� �}|s td��tj�|�| g}t �ddd�}|rT|�dtj�|� � |�d| ||g� t � |�\}} |� | � dS )a Runs a STET transform on a file. Encrypts for uploads. Decrypts for downloads. Automatically populates flags for the STET binary. Args: subcommand (StetSubcommandName): Subcommand to call on STET binary. blob_id (str): Cloud URL that binary uses for validation. in_file_path (str): File to be transformed source. out_file_path (str): Where to write result of transform. logger (logging.Logger): For logging STET binary output. Raises: KeyError: STET binary or config could not be found. �GSUtil�stet_binary_pathz2Could not find STET binary in boto config or PATH.�stet_config_pathNz--config-file=z --blob-id=) r �getr �KeyErrorr r � expanduser�append�extendr �ExecuteExternalCommand�debug) � subcommand�blob_idZ in_file_pathZ out_file_path�loggerr Z command_argsZ config_path�_�stderrr r r �_stet_transform. s �r/ c C s2 | j }t�| �}|j}ttj||||� t�|�S )a) Encrypts a file with STET binary before upload. Args: source_url (StorageUrl): Copy source. destination_url (StorageUrl): Copy destination. logger (logging.Logger): For logging STET binary output. Returns: stet_temporary_file_url (StorageUrl): Path to STET-encrypted file. ) � object_namer �GetStetTempFileName� url_stringr/ r r r �StorageUrlFromString)� source_url�destination_urlr, �in_file�out_filer+ r r r �encrypt_uploadM s �r8 c C s6 |}t �|�}| j}ttj||||� t�||� dS )a STET-decrypts downloaded file. Args: source_url (StorageUrl): Copy source. destination_url (StorageUrl): Copy destination. temporary_file_name (str): Path to temporary file used for download. logger (logging.Logger): For logging STET binary output. N)r r1 r2 r/ r r �shutil�move)r4 r5 �temporary_file_namer, r6 r7 r+ r r r �decrypt_downloada s �r< )r � __future__r r r r r r9 �gslibr � gslib.utilsr r �botor �objectr r r/ r8 r< r r r r �<module> s