/
usr
/
libexec
/
oracle-cloud-agent
/
plugins
/
osms
/
OpenSSL
/
Upload File
HOME
a L��f u � @ s| d dl Z d dlZd dlmZ d dlmZmZ d dlmZm Z d dl m Z d dl m Z d dlmZmZmZmZmZmZmZ m Z d dl m Z m!Z!m"Z"m#Z#m$Z$m%Z% g d �Z&ej'Z'ej( Z(Z)ej* Z*Z+ej, Z,Z-ej. Z.Z/ej0 Z0Z1ej2Z3ej4Z5d Z6d Z7d Z8d Z9dZ:dZ;dZ<dZ=dZ>dZ?z"ej@Z@ejAZAejBZBejCZCejDZDW n( eE�yZ dZ@dZAdZBdZCdZDY n0 ejFZGejHZIejJZKejLZMejNZOz ejPZQW n eE�y� Y n0 ejRZSejTZUejVZWejXZYejZZ[ej\Z]ej^Z_ej`ZaejbZcejdZeejfZgejhZiejjZkejlZmejnZoejpZqejrZsejtZuejvZwejxZyejzZ{ej|Z}ej~Zej�Z�z ej�Z�W n eE�yH Y n0 z ej�Z�W n eE�yh Y n0 ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�ej�Z�g d�Z�dgZ�dZ�d Z�G d d � d e��Z�eee��Z�ee��Z�G d d � d e��Z�G d!d"� d"e��Z�G d#d$� d$e��Z�G d%d&� d&e��Z�G d'd(� d(e��Z�G d)d*� d*�Z�G d+d,� d,e��Z�e�� Z�G d-d.� d.e��Z�G d/d0� d0e��Z�G d1d2� d2e��Z�G d3d4� d4e��Z�G d5d6� d6e��Z�d7d8� Z�d9d:� Z�e�Z�d;d<� Z�e�ej�d=�Z�e�e�ed>d�d?�Z�G d@dA� dA�Z�G dBdC� dC�Z�G dDdE� dE�Z�dS )F� N)� errorcode)�partial�wraps)�chain�count)�platform)�WeakValueDictionary)� UNSPECIFIED�exception_from_error_queue�ffi�lib� make_assert�no_zero_allocator� path_bytes�text_to_bytes_and_warn)� FILETYPE_PEM�PKey�X509�X509Name� X509Store�_PassphraseHelper)d�OPENSSL_VERSION_NUMBER�SSLEAY_VERSION� SSLEAY_CFLAGS�SSLEAY_PLATFORM� SSLEAY_DIR�SSLEAY_BUILT_ON�OPENSSL_VERSION�OPENSSL_CFLAGS�OPENSSL_PLATFORM� OPENSSL_DIR�OPENSSL_BUILT_ON� SENT_SHUTDOWN�RECEIVED_SHUTDOWN� SSLv23_METHOD� TLSv1_METHOD�TLSv1_1_METHOD�TLSv1_2_METHOD� TLS_METHOD�TLS_SERVER_METHOD�TLS_CLIENT_METHOD� DTLS_METHOD�DTLS_SERVER_METHOD�DTLS_CLIENT_METHOD� SSL3_VERSION� TLS1_VERSION�TLS1_1_VERSION�TLS1_2_VERSION�TLS1_3_VERSION� OP_NO_SSLv2� OP_NO_SSLv3� OP_NO_TLSv1� OP_NO_TLSv1_1� OP_NO_TLSv1_2� OP_NO_TLSv1_3�MODE_RELEASE_BUFFERS�OP_SINGLE_DH_USE�OP_SINGLE_ECDH_USE�OP_EPHEMERAL_RSA�OP_MICROSOFT_SESS_ID_BUG�OP_NETSCAPE_CHALLENGE_BUG�#OP_NETSCAPE_REUSE_CIPHER_CHANGE_BUG� OP_SSLREF2_REUSE_CERT_TYPE_BUG� OP_MICROSOFT_BIG_SSLV3_BUFFER�OP_MSIE_SSLV2_RSA_PADDING�OP_SSLEAY_080_CLIENT_DH_BUG� OP_TLS_D5_BUG�OP_TLS_BLOCK_PADDING_BUG� OP_DONT_INSERT_EMPTY_FRAGMENTS�OP_CIPHER_SERVER_PREFERENCE�OP_TLS_ROLLBACK_BUG�OP_PKCS1_CHECK_1�OP_PKCS1_CHECK_2�OP_NETSCAPE_CA_DN_BUG�"OP_NETSCAPE_DEMO_CIPHER_CHANGE_BUG�OP_NO_COMPRESSION�OP_NO_QUERY_MTU�OP_COOKIE_EXCHANGE� OP_NO_TICKET�OP_NO_RENEGOTIATION�OP_IGNORE_UNEXPECTED_EOF�OP_ALL� VERIFY_PEER�VERIFY_FAIL_IF_NO_PEER_CERT�VERIFY_CLIENT_ONCE� VERIFY_NONE�SESS_CACHE_OFF�SESS_CACHE_CLIENT�SESS_CACHE_SERVER�SESS_CACHE_BOTH�SESS_CACHE_NO_AUTO_CLEAR� SESS_CACHE_NO_INTERNAL_LOOKUP� SESS_CACHE_NO_INTERNAL_STORE�SESS_CACHE_NO_INTERNAL�SSL_ST_CONNECT� SSL_ST_ACCEPT� SSL_ST_MASK� SSL_CB_LOOP� SSL_CB_EXIT� SSL_CB_READ� SSL_CB_WRITE� SSL_CB_ALERT�SSL_CB_READ_ALERT�SSL_CB_WRITE_ALERT�SSL_CB_ACCEPT_LOOP�SSL_CB_ACCEPT_EXIT�SSL_CB_CONNECT_LOOP�SSL_CB_CONNECT_EXIT�SSL_CB_HANDSHAKE_START�SSL_CB_HANDSHAKE_DONE�Error� WantReadError�WantWriteError�WantX509LookupError�ZeroReturnError� SysCallError�NO_OVERLAPPING_PROTOCOLS�SSLeay_version�Session�Context� Connection� � � � � � � � � � i i i i i )z"/etc/ssl/certs/ca-certificates.crtz /etc/pki/tls/certs/ca-bundle.crtz/etc/ssl/ca-bundle.pemz/etc/pki/tls/cacert.pemz1/etc/pki/ca-trust/extracted/pem/tls-ca-bundle.pemz/etc/ssl/certss$ /opt/pyca/cryptography/openssl/certss' /opt/pyca/cryptography/openssl/cert.pemc @ s e Zd ZdZdS )rp z4 An error occurred in an `OpenSSL.SSL` API. N��__name__� __module__� __qualname__�__doc__� r� r� �o/sparta/input/_build_configuration/image_build+validate/lib/bmcenv/lib64/python3.9/site-packages/OpenSSL/SSL.pyrp s rp c @ s e Zd ZdS )rq N�r� r� r� r� r� r� r� rq s rq c @ s e Zd ZdS )rr Nr� r� r� r� r� rr s rr c @ s e Zd ZdS )rs Nr� r� r� r� r� rs " s rs c @ s e Zd ZdS )rt Nr� r� r� r� r� rt & s rt c @ s e Zd ZdS )ru Nr� r� r� r� r� ru * s ru c @ s e Zd ZdZdd� Zdd� ZdS )�_CallbackExceptionHelpera� A base class for wrapper classes that allow for intelligent exception handling in OpenSSL callbacks. :ivar list _problems: Any exceptions that occurred while executing in a context where they could not be raised in the normal way. Typically this is because OpenSSL has called into some Python code and requires a return value. The exceptions are saved to be raised later when it is possible to do so. c C s g | _ d S �N)� _problems��selfr� r� r� �__init__: s z!_CallbackExceptionHelper.__init__c C s4 | j r0z t� W n ty" Y n0 | j �d��dS )z� Raise an exception from the OpenSSL error queue or that was previously captured whe running a callback. r N)r� �_raise_current_errorrp �popr� r� r� r� �raise_if_problem= s z)_CallbackExceptionHelper.raise_if_problemN)r� r� r� r� r� r� r� r� r� r� r� . s r� c @ s e Zd ZdZdd� ZdS )� _VerifyHelperz^ Wrap a callback such that it can be used as a certificate verification callback. c s2 t ��� t� �� �fdd��}t�d|��_d S )Nc s� t �|�}t �|� t�|�}t �|�}t �|�}t �� }t �||�}t j | }z� ||||| �} W n2 t y� } z�j � | � W Y d } ~ dS d } ~ 0 0 | r�t �|t j� dS dS d S �Nr � )�_lib� X509_STORE_CTX_get_current_cert� X509_up_refr �_from_raw_x509_ptr�X509_STORE_CTX_get_error� X509_STORE_CTX_get_error_depth�"SSL_get_ex_data_X509_STORE_CTX_idx�X509_STORE_CTX_get_ex_datarz �_reverse_mapping� Exceptionr� �append�X509_STORE_CTX_set_error� X509_V_OK) �ok� store_ctx�x509�cert� error_number� error_depth�index�ssl� connection�result�e��callbackr� r� r� �wrapperS s& � z'_VerifyHelper.__init__.<locals>.wrapperz int (*)(int, X509_STORE_CTX *)�r� r� r �_ffir� �r� r� r� r� r� r� r� P s �z_VerifyHelper.__init__N�r� r� r� r� r� r� r� r� r� r� J s r� c @ s e Zd ZdZdd� ZdS )�_ALPNSelectHelperzQ Wrap a callback such that it can be used as an ALPN selection callback. c s2 t ��� t� �� �fdd��}t�d|��_d S )Nc s z�t j| }t�||�d d � }g }|r\|d } |d| d � } |�| � || d d � }q$� ||�} d} | tu r|d} d} nt| t�s�td��t� dt | ��t� d| �g|_ |j d d |d< |j d |d< | s�t j W S t jW S t�y } z �j�| � t jW Y d } ~ S d } ~ 0 0 d S ) Nr r� T� Fz^ALPN callback must return a bytestring or the special NO_OVERLAPPING_PROTOCOLS sentinel value.zunsigned char *�unsigned char[])rz r� r� �bufferr� rv � isinstance�bytes� TypeError�new�len�_alpn_select_callback_argsr� �SSL_TLSEXT_ERR_NOACK�SSL_TLSEXT_ERR_OKr� r� �SSL_TLSEXT_ERR_ALERT_FATAL)r� �out�outlen�in_�inlen�arg�conn�instr� protolist� encoded_len�proto�outbytes� any_acceptedr� r� r� r� r� } s: � � z+_ALPNSelectHelper.__init__.<locals>.wrapperz^int (*)(SSL *, unsigned char **, unsigned char *, const unsigned char *, unsigned int, void *)r� r� r� r� r� r� z s +�z_ALPNSelectHelper.__init__Nr� r� r� r� r� r� u s r� c @ s e Zd ZdZdd� ZdS )�_OCSPServerCallbackHelpera� Wrap a callback such that it can be used as an OCSP callback for the server side. Annoyingly, OpenSSL defines one OCSP callback but uses it in two different ways. For servers, that callback is expected to retrieve some OCSP data and hand it to OpenSSL, and may return only SSL_TLSEXT_ERR_OK, SSL_TLSEXT_ERR_FATAL, and SSL_TLSEXT_ERR_NOACK. For clients, that callback is expected to check the OCSP data, and returns a negative value on error, 0 if the response is not acceptable, or positive if it is. These are mutually exclusive return code behaviours, and they mean that we need two helpers so that we always return an appropriate error code if the user's code throws an exception. Given that we have to have two helpers anyway, these helpers are a bit more helpery than most: specifically, they hide a few more of the OpenSSL functions so that the user has an easier time writing these callbacks. This helper implements the server side. c s2 t ��� t� �� �fdd��}t�d|��_d S )Nc s� z�t j| }|tjkr"t�|�}nd }� ||�}t|t�sBtd��|sLW dS t|�}t � |�}|t� ||�d d �< t � | ||� W dS t y� } z�j�|� W Y d }~dS d }~0 0 d S )Nz'OCSP callback must return a bytestring.r{ r � )rz r� r� �NULL� from_handler� r� r� r� r� �OPENSSL_mallocr� � SSL_set_tlsext_status_ocsp_respr� r� r� )r� �cdatar� �data� ocsp_data�ocsp_data_length�data_ptrr� r� r� r� r� � s( � z3_OCSPServerCallbackHelper.__init__.<locals>.wrapper�int (*)(SSL *, void *)r� r� r� r� r� r� � s &z"_OCSPServerCallbackHelper.__init__Nr� r� r� r� r� r� � s r� c @ s e Zd ZdZdd� ZdS )�_OCSPClientCallbackHelpera� Wrap a callback such that it can be used as an OCSP callback for the client side. Annoyingly, OpenSSL defines one OCSP callback but uses it in two different ways. For servers, that callback is expected to retrieve some OCSP data and hand it to OpenSSL, and may return only SSL_TLSEXT_ERR_OK, SSL_TLSEXT_ERR_FATAL, and SSL_TLSEXT_ERR_NOACK. For clients, that callback is expected to check the OCSP data, and returns a negative value on error, 0 if the response is not acceptable, or positive if it is. These are mutually exclusive return code behaviours, and they mean that we need two helpers so that we always return an appropriate error code if the user's code throws an exception. Given that we have to have two helpers anyway, these helpers are a bit more helpery than most: specifically, they hide a few more of the OpenSSL functions so that the user has an easier time writing these callbacks. This helper implements the client side. c s2 t ��� t� �� �fdd��}t�d|��_d S )Nc s� zzt j| }|tjkr"t�|�}nd }t�d�}t�| |�}|dk rJd}nt�|d |�d d � }� |||�}t t |��W S t y� } z�j � |� W Y d }~dS d }~0 0 d S )N�unsigned char **r r� ���)rz r� r� r� r� r� r� � SSL_get_tlsext_status_ocsp_respr� �int�boolr� r� r� ) r� r� r� r� �ocsp_ptr�ocsp_lenr� �validr� r� r� r� r� s z3_OCSPClientCallbackHelper.__init__.<locals>.wrapperr� r� r� r� r� r� r� s z"_OCSPClientCallbackHelper.__init__Nr� r� r� r� r� r� � s r� c @ s e Zd Zdd� ZdS )� _CookieGenerateCallbackHelperc s2 t ��� t� �� �fdd��}t�d|��_d S )Nc sl z4t j| }� |�}||dt|��< t|�|d< W dS tyf } z�j�|� W Y d }~dS d }~0 0 d S r� )rz r� r� r� r� r� )r� r� r� r� �cookier� r� r� r� r� 4 s z7_CookieGenerateCallbackHelper.__init__.<locals>.wrapperz/int (*)(SSL *, unsigned char *, unsigned int *)r� r� r� r� r� r� 1 s �z&_CookieGenerateCallbackHelper.__init__N�r� r� r� r� r� r� r� r� r� 0 s r� c @ s e Zd Zdd� ZdS )�_CookieVerifyCallbackHelperc s2 t ��� t� �� �fdd��}t�d|��_d S )Nc sZ z"t j| }� |t|d|� ��W S tyT } z�j�|� W Y d }~dS d }~0 0 d S �Nr )rz r� r� r� r� r� )r� �c_cookie� cookie_lenr� r� r� r� r� r� K s z5_CookieVerifyCallbackHelper.__init__.<locals>.wrapperz-int (*)(SSL *, unsigned char *, unsigned int)r� r� r� r� r� r� H s �z$_CookieVerifyCallbackHelper.__init__Nr� r� r� r� r� r� G s r� c C sd d }t | t�s(t| dd �}|d ur(|� } t | t�r6| }t |t�sJtd��n|dk r`td|f ��|S )N�filenoz3argument must be an int, or have a fileno() method.r z1file descriptor cannot be a negative integer (%i))r� r� �getattrr� � ValueError)�obj�fd�methr� r� r� �_asFileDescriptorZ s �r� c C s t �t�| ��S )z� Return a string describing the version of OpenSSL in use. :param type: One of the :const:`OPENSSL_` constants defined in this module. )r� �stringr� �OpenSSL_version)�typer� r� r� r� n s r� c s � �fdd�}|S )a� Builds a decorator that ensures that functions that rely on OpenSSL functions that are not present in this build raise NotImplementedError, rather than AttributeError coming out of cryptography. :param flag: A cryptography flag that guards the functions, e.g. ``Cryptography_HAS_NEXTPROTONEG``. :param error: The string to be used in the exception if the flag is false. c s$ �s t | �� fdd��}|S | S d S )Nc s t � ��d S r� ��NotImplementedError)�args�kwargs)�errorr� r� �explode� s z<_make_requires.<locals>._requires_decorator.<locals>.explode)r )�funcr� �r� �flagr� r� �_requires_decorator� s z+_make_requires.<locals>._requires_decoratorr� )r� r� r r� r� r� �_make_requiresz s r zALPN not available�Cryptography_HAS_KEYLOGzKey logging not availablec @ s e Zd ZdZdS )rx z� A class representing an SSL session. A session defines certain connection parameters which may be re-used to speed up the setup of subsequent connections. .. versionadded:: 0.14 Nr� r� r� r� r� rx � s rx c @ s e Zd ZdZeejdfeejefe eje fe eje fe ejdfeejdfeejdfeejdfeejdfeejdfi Zdd� Zdd� Zdd� Zdcd d �Z d d � Z ddd d�Z dd� Z dd� Z dd� Z!dd� Z"e#fdd�Z$dd� Z%dd � Z&d d � Z'e(fd d �Z)d!d"� Z*d#d$� Z+d%d&� Z,d'd(� Z-d)d*� Z.d+d,� Z/ded-d.�Z0d/d0� Z1d1d2� Z2d3d4� Z3d5d6� Z4d7d8� Z5d9d:� Z6d;d<� Z7d=d>� Z8d?d@� Z9dAdB� Z:dCdD� Z;e<dEdF� �Z=dGdH� Z>dIdJ� Z?dKdL� Z@dMdN� ZAdOdP� ZBdQdR� ZCdSdT� ZDeEdUdV� �ZFeEdWdX� �ZGdYdZ� ZHdfd[d\�ZIdgd]d^�ZJd_d`� ZKdadb� ZLdS )hry au :class:`OpenSSL.SSL.Context` instances define the parameters for setting up new SSL connections. :param method: One of TLS_METHOD, TLS_CLIENT_METHOD, TLS_SERVER_METHOD, DTLS_METHOD, DTLS_CLIENT_METHOD, or DTLS_SERVER_METHOD. SSLv23_METHOD, TLSv1_METHOD, etc. are deprecated and should not be used. Nc C s t |t�std��z| j| \}}W n ty> td��Y n0 |� }t|tjk� t � |�}t|tjk� t� |t j �}|| _ d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _ d | _ | � t j � |d u�r | � |� | �!|� d S )Nzmethod must be an integerzNo such protocol)"r� r� r� �_methods�KeyErrorr� �_openssl_assertr� r� r� � SSL_CTX_new�gc� SSL_CTX_free�_context�_passphrase_helper�_passphrase_callback�_passphrase_userdata�_verify_helper�_verify_callback�_info_callback�_keylog_callback�_tlsext_servername_callback� _app_data�_alpn_select_helper�_alpn_select_callback� _ocsp_helper�_ocsp_callback� _ocsp_data�_cookie_generate_helper�_cookie_verify_helper�set_mode� SSL_MODE_ENABLE_PARTIAL_WRITE�set_min_proto_version�set_max_proto_version)r� �method� method_func�version� method_obj�contextr� r� r� r� � s@ zContext.__init__c C s t t�| j|�dk� dS )aD Set the minimum supported protocol version. Setting the minimum version to 0 will enable protocol versions down to the lowest version supported by the library. If the underlying OpenSSL build is missing support for the selected version, this method will raise an exception. r� N)r r� � SSL_CTX_set_min_proto_versionr �r� r r� r� r� r � s �z Context.set_min_proto_versionc C s t t�| j|�dk� dS )aC Set the maximum supported protocol version. Setting the maximum version to 0 will enable protocol versions up to the highest version supported by the library. If the underlying OpenSSL build is missing support for the selected version, this method will raise an exception. r� N)r r� � SSL_CTX_set_max_proto_versionr r$ r� r� r� r � s �z Context.set_max_proto_versionc C sN |du rt j}nt|�}|du r(t j}nt|�}t�| j||�}|sJt� dS )aU Let SSL know where we can find trusted certificates for the certificate chain. Note that the certificates have to be in PEM format. If capath is passed, it must be a directory prepared using the ``c_rehash`` tool included with OpenSSL. Either, but not both, of *pemfile* or *capath* may be :data:`None`. :param cafile: In which file we can find the certificates (``bytes`` or ``unicode``). :param capath: In which directory we can find the certificates (``bytes`` or ``unicode``). :return: None N)r� r� � _path_bytesr� � SSL_CTX_load_verify_locationsr r� )r� �cafile�capath� load_resultr� r� r� �load_verify_locations s �z Context.load_verify_locationsc s& t � �� �fdd��}tt|ddd�S )Nc s � | |�j �S r� )r )�size�verify�userdatar� r� r� r� # s z'Context._wrap_callback.<locals>.wrapperT)� more_args�truncate)r r r r� r� r� r� �_wrap_callback" s �zContext._wrap_callbackc C s@ t |�std��| �|�| _| jj| _t�| j| j� || _ dS )a� Set the passphrase callback. This function will be called when a private key with a passphrase is loaded. :param callback: The Python callback to use. This must accept three positional arguments. First, an integer giving the maximum length of the passphrase it may return. If the returned passphrase is longer than this, it will be truncated. Second, a boolean value which will be true if the user should be prompted for the passphrase twice and the callback should verify that the two values supplied are equal. Third, the value given as the *userdata* parameter to :meth:`set_passwd_cb`. The *callback* must return a byte string. If an error occurs, *callback* should return a false value (e.g. an empty string). :param userdata: (optional) A Python object which will be given as argument to the callback :return: None �callback must be callableN) �callabler� r1 r r� r r� � SSL_CTX_set_default_passwd_cbr r )r� r� r. r� r� r� � set_passwd_cb+ s �zContext.set_passwd_cbc C s� t �| j�}t|dk� t�t �� ��d�}t�t �� ��d�}| � ||�s�t�t � � �}t�t � � �}|t kr�|t kr�| �tt� dS )a� Specify that the platform provided CA certificates are to be used for verification purposes. This method has some caveats related to the binary wheels that cryptography (pyOpenSSL's primary dependency) ships: * macOS will only load certificates using this method if the user has the ``openssl@1.1`` `Homebrew <https://brew.sh>`_ formula installed in the default location. * Windows will not work. * manylinux1 cryptography wheels will work on most common Linux distributions in pyOpenSSL 17.1.0 and above. pyOpenSSL detects the manylinux1 wheel and attempts to load roots via a fallback path. :return: None r� �asciiN)r� � SSL_CTX_set_default_verify_pathsr r r� r� � X509_get_default_cert_dir_env�decode� X509_get_default_cert_file_env�_check_env_vars_set�X509_get_default_cert_dir�X509_get_default_cert_file� _CRYPTOGRAPHY_MANYLINUX_CA_DIR� _CRYPTOGRAPHY_MANYLINUX_CA_FILE� _fallback_default_verify_paths�_CERTIFICATE_FILE_LOCATIONS�_CERTIFICATE_PATH_LOCATIONS)r� � set_result� dir_env_var� file_env_var� default_dir� default_filer� r� r� �set_default_verify_pathsH s( ��� ���z Context.set_default_verify_pathsc C s t j�|�dup t j�|�duS )zp Check to see if the default cert dir/file environment vars are present. :return: bool N)�os�environ�get)r� rD rE r� r� r� r; x s �zContext._check_env_vars_setc C sN |D ] }t j�|�r| �|� q$q|D ] }t j�|�r(| �d|� qJq(dS )aW Default verify paths are based on the compiled version of OpenSSL. However, when pyca/cryptography is compiled as a manylinux1 wheel that compiled location can potentially be wrong. So, like Go, we will try a predefined set of paths and attempt to load roots from there. :return: None N)rI �path�isfiler+ �isdir)r� � file_path�dir_pathr( r) r� r� r� r@ � s z&Context._fallback_default_verify_pathsc C s$ t |�}t�| j|�}|s t� dS )z� Load a certificate chain from a file. :param certfile: The name of the certificate chain file (``bytes`` or ``unicode``). Must be PEM encoded. :return: None N)r& r� �"SSL_CTX_use_certificate_chain_filer r� )r� �certfiler� r� r� r� �use_certificate_chain_file� s �z"Context.use_certificate_chain_filec C s8 t |�}t|t�std��t�| j||�}|s4t� dS )ah Load a certificate from a file :param certfile: The name of the certificate file (``bytes`` or ``unicode``). :param filetype: (optional) The encoding of the file, which is either :const:`FILETYPE_PEM` or :const:`FILETYPE_ASN1`. The default is :const:`FILETYPE_PEM`. :return: None �filetype must be an integerN)r& r� r� r� r� � SSL_CTX_use_certificate_filer r� )r� rR �filetype� use_resultr� r� r� �use_certificate_file� s �z Context.use_certificate_filec C s0 t |t�std��t�| j|j�}|s,t� dS �zs Load a certificate from a X509 object :param cert: The X509 object :return: None z cert must be an X509 instanceN)r� r r� r� �SSL_CTX_use_certificater �_x509r� �r� r� rW r� r� r� �use_certificate� s zContext.use_certificatec C sD t |t�std��t�|j�}t�| j|�}|s@t�|� t � dS )z� Add certificate to chain :param certobj: The X509 certificate object to add to the chain :return: None z certobj must be an X509 instanceN) r� r r� r� �X509_dupr[ � SSL_CTX_add_extra_chain_certr � X509_freer� )r� �certobj�copy� add_resultr� r� r� �add_extra_chain_cert� s z Context.add_extra_chain_certc C s | j d ur| j �t� t� d S r� )r r� rp r� r� r� r� r� �_raise_passphrase_exception� s z#Context._raise_passphrase_exceptionc C sH t |�}|tu rt}nt|t�s(td��t�| j||�}|sD| � � dS )aR Load a private key from a file :param keyfile: The name of the key file (``bytes`` or ``unicode``) :param filetype: (optional) The encoding of the file, which is either :const:`FILETYPE_PEM` or :const:`FILETYPE_ASN1`. The default is :const:`FILETYPE_PEM`. :return: None rT N) r& � _UNSPECIFIEDr r� r� r� r� �SSL_CTX_use_PrivateKey_filer re )r� �keyfilerV rW r� r� r� �use_privatekey_file� s �zContext.use_privatekey_filec C s2 t |t�std��t�| j|j�}|s.| �� dS �zs Load a private key from a PKey object :param pkey: The PKey object :return: None z pkey must be a PKey instanceN)r� r r� r� �SSL_CTX_use_PrivateKeyr �_pkeyre �r� �pkeyrW r� r� r� �use_privatekey� s zContext.use_privatekeyc C s t �| j�st� dS )z� Check if the private key (loaded with :meth:`use_privatekey`) matches the certificate (loaded with :meth:`use_certificate`) :return: :data:`None` (raises :exc:`Error` if something's wrong) N)r� �SSL_CTX_check_private_keyr r� r� r� r� r� �check_privatekey s zContext.check_privatekeyc C s0 t �td|��}t|tjk� t �| j|� dS )a% Load the trusted certificates that will be sent to the client. Does not actually imply any of the certificates are trusted; that must be configured separately. :param bytes cafile: The path to a certificates file in PEM format. :return: None r( N)r� �SSL_load_client_CA_file�_text_to_bytes_and_warnr r� r� �SSL_CTX_set_client_CA_listr )r� r( �ca_listr� r� r� �load_client_ca s �zContext.load_client_cac C s* t d|�}tt�| j|t|��dk� dS )aV Set the session id to *buf* within which a session can be reused for this Context object. This is needed when doing session resumption, because there is no way for a stored session to know which Context object it is associated with. :param bytes buf: The session id. :returns: None �bufr� N)rs r r� � SSL_CTX_set_session_id_contextr r� )r� rw r� r� r� �set_session_id$ s ��zContext.set_session_idc C s t |t�std��t�| j|�S )a� Set the behavior of the session cache used by all connections using this Context. The previously set mode is returned. See :const:`SESS_CACHE_*` for details about particular modes. :param mode: One or more of the SESS_CACHE_* flags (combine using bitwise or) :returns: The previously set caching mode. .. versionadded:: 0.14 �mode must be an integer)r� r� r� r� � SSL_CTX_set_session_cache_moder �r� �moder� r� r� �set_session_cache_mode5 s z Context.set_session_cache_modec C s t �| j�S )z� Get the current session cache mode. :returns: The currently used cache mode. .. versionadded:: 0.14 )r� � SSL_CTX_get_session_cache_moder r� r� r� r� �get_session_cache_modeF s z Context.get_session_cache_modec C st t |t�std��|du r:d| _d| _t�| j|tj � n6t |�sJtd��t |�| _| jj | _t�| j|| j� dS )a� Set the verification flags for this Context object to *mode* and specify that *callback* should be used for verification callbacks. :param mode: The verify mode, this should be one of :const:`VERIFY_NONE` and :const:`VERIFY_PEER`. If :const:`VERIFY_PEER` is used, *mode* can be OR:ed with :const:`VERIFY_FAIL_IF_NO_PEER_CERT` and :const:`VERIFY_CLIENT_ONCE` to further control the behaviour. :param callback: The optional Python verification callback to use. This should take five arguments: A Connection object, an X509 object, and three integer variables, which are in turn potential error number, error depth and return code. *callback* should return True if verification passes and False otherwise. If omitted, OpenSSL's default verification is used. :return: None See SSL_CTX_set_verify(3SSL) for further details. rz Nr2 ) r� r� r� r r r� �SSL_CTX_set_verifyr r� r� r3 r� r� �r� r} r� r� r� r� � set_verifyP s zContext.set_verifyc C s$ t |t�std��t�| j|� dS )z� Set the maximum depth for the certificate chain verification that shall be allowed for this Context object. :param depth: An integer specifying the verify depth :return: None zdepth must be an integerN)r� r� r� r� �SSL_CTX_set_verify_depthr )r� �depthr� r� r� �set_verify_depths s zContext.set_verify_depthc C s t �| j�S )z� Retrieve the Context object's verify mode, as set by :meth:`set_verify`. :return: The verify mode )r� �SSL_CTX_get_verify_moder r� r� r� r� �get_verify_mode� s zContext.get_verify_modec C s t �| j�S )z� Retrieve the Context object's verify depth, as set by :meth:`set_verify_depth`. :return: The verify depth )r� �SSL_CTX_get_verify_depthr r� r� r� r� �get_verify_depth� s zContext.get_verify_depthc C st t |�}t�|d�}|tjkr$t� t�|tj�}t�|tjtjtj�}t�|tj �}t� | j |�}t |dk� dS )z� Load parameters for Ephemeral Diffie-Hellman :param dhfile: The file to load EDH parameters from (``bytes`` or ``unicode``). :return: None � rr� N) r& r� � BIO_new_filer� r� r� r �BIO_free�PEM_read_bio_DHparams�DH_free�SSL_CTX_set_tmp_dhr r )r� �dhfile�bio�dh�resr� r� r� � load_tmp_dh� s zContext.load_tmp_dhc C s t �| j|�� � dS )a Select a curve to use for ECDHE key exchange. :param curve: A curve object to use as returned by either :meth:`OpenSSL.crypto.get_elliptic_curve` or :meth:`OpenSSL.crypto.get_elliptic_curves`. :return: None N)r� �SSL_CTX_set_tmp_ecdhr � _to_EC_KEY)r� �curver� r� r� � set_tmp_ecdh� s zContext.set_tmp_ecdhc C sZ t d|�}t|t�s td��tt�| j|�dk� t| d�}|� � g d�krVt dg��dS )z� Set the list of ciphers to be used in this context. See the OpenSSL manual for more information (e.g. :manpage:`ciphers(1)`). :param bytes cipher_list: An OpenSSL cipher string. :return: None � cipher_listz"cipher_list must be a byte string.r� N)�TLS_AES_256_GCM_SHA384� TLS_CHACHA20_POLY1305_SHA256�TLS_AES_128_GCM_SHA256)z SSL routines�SSL_CTX_set_cipher_listzno cipher match) rs r� r� r� r r� r� r rz �get_cipher_listrp )r� r� �tmpconnr� r� r� �set_cipher_list� s � ��zContext.set_cipher_listc C s� t �� }t|tjk� zf|D ]\}t|t�s>tdt|�j f ��t � |j �}t|tjk� t � ||�}|s t � |� t� q W n ty� t �|� � Y n0 t �| j|� dS )a_ Set the list of preferred client certificate signers for this server context. This list of certificate authorities will be sent to the client when the server requests a client certificate. :param certificate_authorities: a sequence of X509Names. :return: None .. versionadded:: 0.10 z3client CAs must be X509Name objects, not %s objectsN)r� �sk_X509_NAME_new_nullr r� r� r� r r� r� r� � X509_NAME_dup�_name�sk_X509_NAME_push�X509_NAME_freer� r� �sk_X509_NAME_freert r )r� �certificate_authorities� name_stack�ca_namerb � push_resultr� r� r� �set_client_ca_list� s( �� zContext.set_client_ca_listc C s2 t |t�std��t�| j|j�}t|dk� dS )ai Add the CA certificate to the list of preferred signers for this context. The list of certificate authorities will be sent to the client when the server requests a client certificate. :param certificate_authority: certificate authority's X509 certificate. :return: None .. versionadded:: 0.10 z.certificate_authority must be an X509 instancer� N)r� r r� r� �SSL_CTX_add_client_CAr r[ r )r� �certificate_authorityrc r� r� r� � add_client_ca� s �zContext.add_client_cac C s t |t�std��t�| j|�S )aQ Set the timeout for newly created sessions for this Context object to *timeout*. The default value is 300 seconds. See the OpenSSL manual for more information (e.g. :manpage:`SSL_CTX_set_timeout(3)`). :param timeout: The timeout in (whole) seconds :return: The previous session timeout ztimeout must be an integer)r� r� r� r� �SSL_CTX_set_timeoutr )r� �timeoutr� r� r� � set_timeout s zContext.set_timeoutc C s t �| j�S )z� Retrieve session timeout, as set by :meth:`set_timeout`. The default is 300 seconds. :return: The session timeout )r� �SSL_CTX_get_timeoutr r� r� r� r� � get_timeout s zContext.get_timeoutc s6 t � �� fdd��}t�d|�| _t�| j| j� dS )a� Set the information callback to *callback*. This function will be called from time to time during SSL handshakes. :param callback: The Python callback to use. This should take three arguments: a Connection object and two integers. The first integer specifies where in the SSL handshake the function was called, and the other the return code from a (possibly failed) internal function call. :return: None c s � t j| ||� d S r� �rz r� )r� �where� return_code�r� r� r� r� 6 s z*Context.set_info_callback.<locals>.wrapperz void (*)(const SSL *, int, int)N)r r� r� r r� �SSL_CTX_set_info_callbackr r� r� r� r� �set_info_callback) s �zContext.set_info_callbackc s6 t � �� fdd��}t�d|�| _t�| j| j� dS )a Set the TLS key logging callback to *callback*. This function will be called whenever TLS key material is generated or received, in order to allow applications to store this keying material for debugging purposes. :param callback: The Python callback to use. This should take two arguments: a Connection object and a bytestring that contains the key material in the format used by NSS for its SSLKEYLOGFILE debugging output. :return: None c s t �|�}� tj| |� d S r� )r� r� rz r� )r� �liner� r� r� r� N s z,Context.set_keylog_callback.<locals>.wrapperz#void (*)(const SSL *, const char *)N)r r� r� r r� �SSL_CTX_set_keylog_callbackr r� r� r� r� �set_keylog_callback? s �zContext.set_keylog_callbackc C s | j S )zw Get the application data (supplied via :meth:`set_app_data()`) :return: The application data �r r� r� r� r� � get_app_dataX s zContext.get_app_datac C s || _ dS )z� Set the application data (will be returned from get_app_data()) :param data: Any Python object :return: None Nr� �r� r� r� r� r� � set_app_data` s zContext.set_app_datac C s. t �| j�}|tjkrdS t�t�}||_|S )z� Get the certificate store for the context. This can be used to add "trusted" certificates without using the :meth:`load_verify_locations` method. :return: A X509Store object or None if it does not have one. N)r� �SSL_CTX_get_cert_storer r� r� r �__new__�_store)r� �store�pystorer� r� r� �get_cert_storei s zContext.get_cert_storec C s t |t�std��t�| j|�S )z� Add options. Options set before are not cleared! This method should be used with the :const:`OP_*` constants. :param options: The options to add. :return: The new option bitmask. zoptions must be an integer)r� r� r� r� �SSL_CTX_set_optionsr )r� �optionsr� r� r� � set_optionsz s zContext.set_optionsc C s t |t�std��t�| j|�S )z� Add modes via bitmask. Modes set before are not cleared! This method should be used with the :const:`MODE_*` constants. :param mode: The mode to add. :return: The new mode bitmask. rz )r� r� r� r� �SSL_CTX_set_moder r| r� r� r� r � s zContext.set_modec s6 t � �� fdd��}t�d|�| _t�| j| j� dS )a Specify a callback function to be called when clients specify a server name. :param callback: The callback function. It will be invoked with one argument, the Connection instance. .. versionadded:: 0.13 c s � t j| � dS r� r� )r� �alertr� r� r� r� r� � s z7Context.set_tlsext_servername_callback.<locals>.wrapperz int (*)(SSL *, int *, void *)N)r r� r� r r� �&SSL_CTX_set_tlsext_servername_callbackr r� r� r� r� � set_tlsext_servername_callback� s ��z&Context.set_tlsext_servername_callbackc C s, t |t�std��tt�| j|�dk� dS )z� Enable support for negotiating SRTP keying material. :param bytes profiles: A colon delimited list of protection profile names, like ``b'SRTP_AES128_CM_SHA1_80:SRTP_AES128_CM_SHA1_32'``. :return: None z profiles must be a byte string.r N)r� r� r� r r� �SSL_CTX_set_tlsext_use_srtpr )r� �profilesr� r� r� �set_tlsext_use_srtp� s �zContext.set_tlsext_use_srtpc C sR |s t d��d�t�dd� |D ���}t�d|�}tt�| j |t |��dk� dS )a� Specify the protocols that the client is prepared to speak after the TLS connection has been negotiated using Application Layer Protocol Negotiation. :param protos: A list of the protocols to be offered to the server. This list should be a Python list of bytestrings representing the protocols to offer, e.g. ``[b'http/1.1', b'spdy/2']``. �'at least one protocol must be specifiedr� c s s | ]}t t|�f�|fV qd S r� �r� r� ��.0�pr� r� r� � <genexpr>� r� z*Context.set_alpn_protos.<locals>.<genexpr>r� r N) r� �joinr � from_iterabler� r� r r� �SSL_CTX_set_alpn_protosr r� �r� �protos�protostr� input_strr� r� r� �set_alpn_protos� s � ���zContext.set_alpn_protosc C s, t |�| _| jj| _t�| j| jtj� dS )a� Specify a callback function that will be called on the server when a client offers protocols using ALPN. :param callback: The callback function. It will be invoked with two arguments: the Connection, and a list of offered protocols as bytestrings, e.g ``[b'http/1.1', b'spdy/2']``. It can return one of those bytestrings to indicate the chosen protocol, the empty bytestring to terminate the TLS connection, or the :py:obj:`NO_OVERLAPPING_PROTOCOLS` to indicate that no offered protocol was selected, but that the connection should not be aborted. N) r� r r� r r� �SSL_CTX_set_alpn_select_cbr r� r� �r� r� r� r� r� �set_alpn_select_callback� s �z Context.set_alpn_select_callbackc C sh || _ |j| _|du r tj| _n t�|�| _t�| j | j�}t |dk� t� | j | j�}t |dk� dS )z� This internal helper does the common work for ``set_ocsp_server_callback`` and ``set_ocsp_client_callback``, which is almost all of it. Nr� ) r r� r r� r� r � new_handler� � SSL_CTX_set_tlsext_status_cbr r � SSL_CTX_set_tlsext_status_arg)r� �helperr� �rcr� r� r� �_set_ocsp_callback� s � zContext._set_ocsp_callbackc C s t |�}| �||� dS )a� Set a callback to provide OCSP data to be stapled to the TLS handshake on the server side. :param callback: The callback function. It will be invoked with two arguments: the Connection, and the optional arbitrary data you have provided. The callback must return a bytestring that contains the OCSP data to staple to the handshake. If no OCSP data is available for this connection, return the empty bytestring. :param data: Some opaque data that will be passed into the callback function when called. This can be used to avoid needing to do complex data lookups or to keep track of what context is being used. This parameter is optional. N)r� r� �r� r� r� r� r� r� r� �set_ocsp_server_callback s z Context.set_ocsp_server_callbackc C s t |�}| �||� dS )a� Set a callback to validate OCSP data stapled to the TLS handshake on the client side. :param callback: The callback function. It will be invoked with three arguments: the Connection, a bytestring containing the stapled OCSP assertion, and the optional arbitrary data you have provided. The callback must return a boolean that indicates the result of validating the OCSP data: ``True`` if the OCSP data is valid and the certificate can be trusted, or ``False`` if either the OCSP data is invalid or the certificate has been revoked. :param data: Some opaque data that will be passed into the callback function when called. This can be used to avoid needing to do complex data lookups or to keep track of what context is being used. This parameter is optional. N)r� r� r� r� r� r� �set_ocsp_client_callback s z Context.set_ocsp_client_callbackc C s t |�| _t�| j| jj� d S r� )r� r r� � SSL_CTX_set_cookie_generate_cbr r� r� r� r� r� � set_cookie_generate_callback/ s �z$Context.set_cookie_generate_callbackc C s t |�| _t�| j| jj� d S r� )r� r r� � SSL_CTX_set_cookie_verify_cbr r� r� r� r� r� �set_cookie_verify_callback6 s �z"Context.set_cookie_verify_callback)N)N)N)N)N)Mr� r� r� r� r$ r� � TLS_methodr% r/ r& r0 r' r1 r( r) �TLS_server_methodr* �TLS_client_methodr+ � DTLS_methodr, �DTLS_server_methodr- �DTLS_client_methodr r� r r r+ r1 r5 rH r; r@ rS r rX r] rd re rf ri ro rq rv ry r~ r� r� r� r� r� r� r� r� r� r� r� r� r� �_requires_keylogr� r� r� r� r� r r� r� �_requires_alpnr� r� r� r� r� r� r� r� r� r� r� ry � s~ � ' 0 # '# % ry c @ sD e Zd Ze� Zd�dd�Zdd� Zdd� Zdd � Zd d � Z d d � Z d�dd�Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd�d d �ZeZd�d!d"�Zd�d#d$�ZeZd�d%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'dEdF� Z(dGdH� Z)dIdJ� Z*dKdL� Z+dMdN� Z,dOdP� Z-dQdR� Z.dSdT� Z/dUdV� Z0d�dWdX�Z1dYdZ� Z2d[d\� Z3d]d^� Z4e5d_d`� �Z6dadb� Z7dcdd� Z8dedf� Z9dgdh� Z:didj� Z;dkdl� Z<dmdn� Z=dodp� Z>dqdr� Z?dsdt� Z@dudv� ZAdwdx� ZBdydz� ZCd{d|� ZDd}d~� ZEdd�� ZFeGd�d�� �ZHeGd�d�� �ZId�d�� ZJdS )�rz Nc C s t |t�std��t�|j�}t�|tj�| _ t� | j tj � || _d| _ d| _ |j| _|j| _|j| _|j| _| | j| j < |du r�d| _t�t�� �| _t| jtjk� t�t�� �| _t| jtjk� t�| j | j| j� n2d| _d| _|| _t�| j t | j��}t|dk� dS )z� Create a new Connection object, using the given OpenSSL.SSL.Context instance and socket. :param context: An SSL Context to use for this connection :param socket: The socket to use for transport layer �"context must be a Context instanceNr� ) r� ry r� r� �SSL_newr r� r �SSL_free�_ssl� SSL_set_mode�SSL_MODE_AUTO_RETRYr r� r r r r r� �_socket�BIO_new� BIO_s_mem� _into_sslr r� � _from_ssl� SSL_set_bio� SSL_set_fdr� )r� r" �socketr� rC r� r� r� r� A s6 �zConnection.__init__c C s0 | j du r td| jj|f ��n t| j |�S dS )zy Look up attributes on the wrapped socket object if they are not found on the Connection object. Nz!'%s' object has no attribute '%s')r� �AttributeError� __class__r� r� �r� �namer� r� r� � __getattr__{ s ��zConnection.__getattr__c C s| | j jd ur| j j�� | j jd ur0| j j�� | j jd urH| j j�� t�||�}|tjkrht� ��n|tj krzt � �n�|tj kr�t � �n�|tj kr�t� �n�|tjk�rt�� dkr�|dk r�tdkr�t�� d }ntj}|dkr�t|t�|���tdd��nt� nr|tjk�rdt�� dk�rdt�� }t�|�}tj�r\t |tj k� t� � tdd��nt� n|tj k�rrnt� d S )Nr �win32r� zUnexpected EOF) r r r� r r r� � SSL_get_error�SSL_ERROR_WANT_READrq �SSL_ERROR_WANT_WRITErr �SSL_ERROR_ZERO_RETURNrt �SSL_ERROR_WANT_X509_LOOKUPrs �SSL_ERROR_SYSCALL�ERR_peek_errorr r� � getwinerror�errnoru r rK r� � SSL_ERROR_SSL�ERR_GET_REASON�-Cryptography_HAS_UNEXPECTED_EOF_WHILE_READINGr �"SSL_R_UNEXPECTED_EOF_WHILE_READING�ERR_clear_error�SSL_ERROR_NONE)r� r� r� r� r � peeked_error�reasonr� r� r� �_raise_ssl_error� sL � zConnection._raise_ssl_errorc C s | j S )zh Retrieve the :class:`Context` object associated with this :class:`Connection`. )r r� r� r� r� � get_context� s zConnection.get_contextc C s, t |t�std��t�| j|j� || _dS )z� Switch this connection to a new session context. :param context: A :class:`Context` instance giving the new session context to use. r� N)r� ry r� r� �SSL_set_SSL_CTXr� r )r� r" r� r� r� � set_context� s zConnection.set_contextc C s( t �| jt j�}|tjkr dS t�|�S )z� Retrieve the servername extension value if provided in the client hello message, or None if there wasn't one. :return: A byte string giving the server name or :data:`None`. .. versionadded:: 0.13 N)r� �SSL_get_servernamer� �TLSEXT_NAMETYPE_host_namer� r� r� r r� r� r� �get_servername� s � zConnection.get_servernamec C st t |t�std��|du r:d| _d| _t�| j|tj � n6t |�sJtd��t |�| _| jj | _t�| j|| j� dS )z� Override the Context object's verification flags for this specific connection. See :py:meth:`Context.set_verify` for details. rz Nr2 ) r� r� r� r r r� �SSL_set_verifyr� r� r� r3 r� r� r� r� r� r� r� � s zConnection.set_verifyc C s t �| j�S )z� Retrieve the Connection object's verify mode, as set by :meth:`set_verify`. :return: The verify mode )r� �SSL_get_verify_moder� r� r� r� r� r� � s zConnection.get_verify_modec C s0 t |t�std��t�| j|j�}|s,t� dS rY )r� r r� r� �SSL_use_certificater� r[ r� r\ r� r� r� r] s zConnection.use_certificatec C s4 t |t�std��t�| j|j�}|s0| j�� dS rj ) r� r r� r� �SSL_use_PrivateKeyr� rl r re rm r� r� r� ro s zConnection.use_privatekeyc C s t �| j|� dS )aK For DTLS, set the maximum UDP payload size (*not* including IP/UDP overhead). Note that you might have to set :data:`OP_NO_QUERY_MTU` to prevent OpenSSL from spontaneously clearing this. :param mtu: An integer giving the maximum transmission unit. .. versionadded:: 21.1 N)r� � SSL_set_mtur� )r� �mtur� r� r� �set_ciphertext_mtu s z Connection.set_ciphertext_mtuc C s t td�std��t�| j�S )a For DTLS, get the maximum size of unencrypted data you can pass to :meth:`write` without exceeding the MTU (as passed to :meth:`set_ciphertext_mtu`). :return: The effective MTU as an integer. .. versionadded:: 21.1 �DTLS_get_data_mtuz requires OpenSSL 1.1.1 or better)�hasattrr� r� r+ r� r� r� r� r� �get_cleartext_mtu- s z Connection.get_cleartext_mtuc C s6 t |t�std��nd|v r$td��t�| j|� dS )z� Set the value of the servername extension to send in the client hello. :param name: A byte string giving the name. .. versionadded:: 0.13 zname must be a byte string� z name must not contain NUL byteN)r� r� r� r� �SSL_set_tlsext_host_namer� r r� r� r� �set_tlsext_host_name< s z Connection.set_tlsext_host_namec C s t �| j�S )z� Get the number of bytes that can be safely read from the SSL buffer (**not** the underlying transport buffer). :return: The number of bytes available in the receive buffer. )r� � SSL_pendingr� r� r� r� r� �pendingL s zConnection.pendingr c C sr t d|�}t�|��J}t|�dkr*td��t�| j|t|��}| �| j|� |W d � S 1 sd0 Y dS )a� Send data on the connection. NOTE: If you get one of the WantRead, WantWrite or WantX509Lookup exceptions on this, you have to call the method again with the SAME buffer. :param buf: The string, buffer or memoryview to send :param flags: (optional) Included for compatibility with the socket API, the value is ignored :return: The number of bytes written rw ��z,Cannot send more than 2**31-1 bytes at once.N) rs r� � from_bufferr� r� r� � SSL_writer� r )r� rw �flagsr� r� r� r� r� �sendU s �zConnection.sendc C s� t d|�}t�|��^}t|�}d}|r`t�| j|| t|d��}| �| j|� ||7 }||8 }q"|W d � S 1 sx0 Y dS )a� Send "all" data on the connection. This calls send() repeatedly until all data is sent. If an error occurs, it's impossible to tell how much data has been sent. :param buf: The string, buffer or memoryview to send :param flags: (optional) Included for compatibility with the socket API, the value is ignored :return: The number of bytes written rw r r3 N) rs r� r4 r� r� r5 r� �minr )r� rw r6 r� � left_to_send� total_sentr� r� r� r� �sendallq s � zConnection.sendallc C s` t d|�}|dur.|tj@ r.t�| j||�}nt�| j||�}| �| j|� t� ||�dd� S )a Receive data on the connection. :param bufsiz: The maximum number of bytes to read :param flags: (optional) The only supported flag is ``MSG_PEEK``, all other flags are ignored. :return: The string read from the Connection �char[]N) �_no_zero_allocatorr �MSG_PEEKr� �SSL_peekr� �SSL_readr r� r� )r� �bufsizr6 rw r� r� r� r� �recv� s zConnection.recvc C s� |du rt |�}nt|t |��}td|�}|durN|tj@ rNt�| j||�}nt�| j||�}| � | j|� t t � ||��|d|�< |S )ae Receive data on the connection and copy it directly into the provided buffer, rather than creating a new string. :param buffer: The buffer to copy into. :param nbytes: (optional) The maximum number of bytes to read into the buffer. If not present, defaults to the size of the buffer. If larger than the size of the buffer, is reduced to the size of the buffer. :param flags: (optional) The only supported flag is ``MSG_PEEK``, all other flags are ignored. :return: The number of bytes read into the buffer. Nr< ) r� r8 r= r r> r� r? r� r@ r � memoryviewr� r� )r� r� �nbytesr6 rw r� r� r� r� � recv_into� s zConnection.recv_intoc C sV t �|�rLt �|�r t� �qRt �|�r.t� �qRt �|�rBtd��qRtd��nt� d S )N�BIO_should_io_specialzunknown bio failure) r� �BIO_should_retry�BIO_should_readrq �BIO_should_writerr rF r� r� )r� r� r� r� r� r� �_handle_bio_errors� s z Connection._handle_bio_errorsc C sh | j du rtd��t|t�s$td��td|�}t�| j ||�}|dkrT| �| j |� t� ||�dd� S )a� If the Connection was created with a memory BIO, this method can be used to read bytes from the write end of that memory BIO. Many Connection methods will add bytes which must be read in this manner or the buffer will eventually fill up and the Connection will be able to take no further actions. :param bufsiz: The maximum number of bytes to read :return: The string read. N� Connection sock was not Nonezbufsiz must be an integerr< r ) r r� r� r� r= r� �BIO_readrJ r� r� )r� rA rw r� r� r� r� �bio_read� s zConnection.bio_readc C sx t d|�}| jdu r td��t�|��>}t�| j|t|��}|dkrR| �| j|� |W d � S 1 sj0 Y dS )aj If the Connection was created with a memory BIO, this method can be used to add bytes to the read end of that memory BIO. The Connection can then read the bytes (for example, in response to a call to :meth:`recv`). :param buf: The string to put into the memory BIO. :return: The number of bytes written rw NrK r ) rs r r� r� r4 r� � BIO_writer� rJ )r� rw r� r� r� r� r� � bio_write� s zConnection.bio_writec C s$ | � � s tt�| j�dk� dS dS )z� Renegotiate the session. :return: True if the renegotiation can be started, False otherwise :rtype: bool r� TF)�renegotiate_pendingr r� �SSL_renegotiater� r� r� r� r� � renegotiate s zConnection.renegotiatec C s t �| j�}| �| j|� dS )a Perform an SSL handshake (usually called after :meth:`renegotiate` or one of :meth:`set_accept_state` or :meth:`set_connect_state`). This can raise the same exceptions as :meth:`send` and :meth:`recv`. :return: None. N)r� �SSL_do_handshaker� r �r� r� r� r� r� � do_handshake s zConnection.do_handshakec C s t �| j�dkS )z� Check if there's a renegotiation in progress, it will return False once a renegotiation is finished. :return: Whether there's a renegotiation in progress :rtype: bool r� )r� �SSL_renegotiate_pendingr� r� r� r� r� rP s z Connection.renegotiate_pendingc C s t �| j�S )z� Find out the total number of renegotiations. :return: The number of renegotiations. :rtype: int )r� �SSL_total_renegotiationsr� r� r� r� r� �total_renegotiations' s z Connection.total_renegotiationsc C s t �| j� | j�|�S )a4 Call the :meth:`connect` method of the underlying socket and set up SSL on the socket, using the :class:`Context` object supplied to this :class:`Connection` object at creation. :param addr: A remote address :return: What the socket's connect method returns )r� �SSL_set_connect_stater� r� �connect)r� �addrr� r� r� rZ 0 s zConnection.connectc C s | j j}| �� ||�S )a� Call the :meth:`connect_ex` method of the underlying socket and set up SSL on the socket, using the Context object supplied to this Connection object at creation. Note that if the :meth:`connect_ex` method of the socket doesn't return 0, SSL won't be initialized. :param addr: A remove address :return: What the socket's connect_ex method returns )r� � connect_ex�set_connect_state)r� r[ r\ r� r� r� r\ < s zConnection.connect_exc C s* | j �� \}}t| j|�}|�� ||fS )a� Call the :meth:`accept` method of the underlying socket and set up SSL on the returned socket, using the Context object supplied to this :class:`Connection` object at creation. :return: A *(conn, addr)* pair where *conn* is the new :class:`Connection` object created, and *address* is as returned by the socket's :meth:`accept`. )r� �acceptrz r �set_accept_state)r� �clientr[ r� r� r� r� r^ J s zConnection.acceptc C s� t �� }z t �| j|�}W t �|� n t �|� 0 | jdurF| j�� | jdurZ| j�� |dkrht� �|dk r~| � | j|� dS )z� Call the OpenSSL function DTLSv1_listen on this connection. See the OpenSSL manual for more details. :return: None Nr ) r� � BIO_ADDR_new� DTLSv1_listenr� � BIO_ADDR_freer r� r rq r )r� �bio_addrr� r� r� r� rb Y s zConnection.DTLSv1_listenc C s$ | j du rtd��t�| jd� dS )z� If the Connection was created with a memory BIO, this method can be used to indicate that *end of file* has been reached on the read end of that memory BIO. :return: None NrK r )r r� r� �BIO_set_mem_eof_returnr r� r� r� r� � bio_shutdowns s zConnection.bio_shutdownc C s8 t �| j�}|dk r$| �| j|� n|dkr0dS dS dS )aQ Send the shutdown message to the Connection. :return: True if the shutdown completed successfully (i.e. both sides have sent closure alerts), False otherwise (in which case you call :meth:`recv` or :meth:`send` when the connection becomes readable/writeable). r TFN)r� � SSL_shutdownr� r rT r� r� r� �shutdown� s zConnection.shutdownc C sF g }t � D ]6}t�| j|�}|tjkr* qB|�t�|��d�� q |S )z� Retrieve the list of ciphers used by the Connection object. :return: A list of native cipher strings. �utf-8) r r� �SSL_get_cipher_listr� r� r� r� r� r9 )r� �ciphers�ir� r� r� r� r� � s zConnection.get_cipher_listc C s~ t �| j�}|tjkrg S g }tt �|��D ]L}t �||�}t �|�}t |tjk� t � t �}t� |t j �|_|�|� q,|S )a� Get CAs whose certificates are suggested for client authentication. :return: If this is a server connection, the list of certificate authorities that will be sent or has been sent to the client, as controlled by this :class:`Connection`'s :class:`Context`. If this is a client connection, the list will be empty until the connection with the server is established. .. versionadded:: 0.10 )r� �SSL_get_client_CA_listr� r� r� �range�sk_X509_NAME_num�sk_X509_NAME_valuer� r r r� r r� r� r� )r� �ca_namesr� rl r rb �pynamer� r� r� �get_client_ca_list� s z Connection.get_client_ca_listc O s t d��dS )z� The makefile() method is not implemented, since there is no dup semantics for SSL connections :raise: NotImplementedError z1Cannot make file object of OpenSSL.SSL.ConnectionNr� �r� r� r� r� r� r� �makefile� s �zConnection.makefilec C s | j S )zr Retrieve application data as set by :meth:`set_app_data`. :return: The application data r� r� r� r� r� r� � s zConnection.get_app_datac C s || _ dS )zg Set application data :param data: The application data :return: None Nr� r� r� r� r� r� � s zConnection.set_app_datac C s t �| j�S )z� Get the shutdown state of the Connection. :return: The shutdown state, a bitvector of SENT_SHUTDOWN, RECEIVED_SHUTDOWN. )r� �SSL_get_shutdownr� r� r� r� r� � get_shutdown� s zConnection.get_shutdownc C s$ t |t�std��t�| j|� dS )z� Set the shutdown state of the Connection. :param state: bitvector of SENT_SHUTDOWN, RECEIVED_SHUTDOWN. :return: None zstate must be an integerN)r� r� r� r� �SSL_set_shutdownr� )r� �stater� r� r� � set_shutdown� s zConnection.set_shutdownc C s t �t�| j��S )z� Retrieve a verbose string detailing the state of the Connection. :return: A string representing the state :rtype: bytes )r� r� r� �SSL_state_string_longr� r� r� r� r� �get_state_string� s zConnection.get_state_stringc C sf t �| j�}|tjkrdS t �| jtjd�}t|dk� td|�}t �| j||� t�||�dd� S )z� Retrieve the random value used with the server hello message. :return: A string representing the state Nr r� ) r� �SSL_get_sessionr� r� r� �SSL_get_server_randomr r= r� �r� �session�length�outpr� r� r� � server_random� s zConnection.server_randomc C sf t �| j�}|tjkrdS t �| jtjd�}t|dk� td|�}t �| j||� t�||�dd� S )z� Retrieve the random value used with the client hello message. :return: A string representing the state Nr r� ) r� r} r� r� r� �SSL_get_client_randomr r= r� r r� r� r� � client_random s zConnection.client_randomc C sb t �| j�}|tjkrdS t �|tjd�}t|dk� td|�}t �|||� t�||�dd� S )zz Retrieve the value of the master key for this session. :return: A string representing the state Nr r� ) r� r} r� r� r� �SSL_SESSION_get_master_keyr r= r� r r� r� r� � master_key s zConnection.master_keyc C sn t d|�}tj}d}d}|dur0|}t|�}d}t�| j|||t|�|||�}t|dk� t�||�dd� S )aH Obtain keying material for application use. :param: label - a disambiguating label string as described in RFC 5705 :param: olen - the length of the exported key material in bytes :param: context - a per-association context value :return: the exported key material bytes or None r� r Nr� ) r= r� r� r� r� �SSL_export_keying_materialr� r r� ) r� �label�olenr" r� � context_buf� context_len� use_context�successr� r� r� �export_keying_material% s( � z!Connection.export_keying_materialc O s | j j|i |��S )z� Call the :meth:`shutdown` method of the underlying socket. See :manpage:`shutdown(2)`. :return: What the socket's shutdown() method returns )r� rh rt r� r� r� � sock_shutdownC s zConnection.sock_shutdownc C s. t �| j�}|tjkr*t �|� t�|�S dS )za Retrieve the local certificate (if any) :return: The local certificate N)r� �SSL_get_certificater� r� r� r� r r� �r� r� r� r� r� �get_certificateL s zConnection.get_certificatec C s$ t �| j�}|tjkr t�|�S dS )zi Retrieve the other side's certificate (if any) :return: The peer's certificate N)r� �SSL_get_peer_certificater� r� r� r r� r� r� r� r� �get_peer_certificateX s z Connection.get_peer_certificatec C s` g }t t�| ��D ]H}t�| |�}t|tjk� t�|�}t|dk� t� |�}|� |� q|S )zb Internal helper to convert a STACK_OF(X509) to a list of X509 instances. r� ) rn r� � sk_X509_num� sk_X509_valuer r� r� r� r r� r� )� cert_stackr� rl r� r� �pycertr� r� r� �_cert_stack_to_listc s z Connection._cert_stack_to_listc C s$ t �| j�}|tjkrdS | �|�S )z� Retrieve the other side's certificate (if any) :return: A list of X509 instances giving the peer's certificate chain, or None if it does not have one. N)r� �SSL_get_peer_cert_chainr� r� r� r� �r� r� r� r� r� �get_peer_cert_chains s z Connection.get_peer_cert_chainc C s$ t �| j�}|tjkrdS | �|�S )a� Retrieve the verified certificate chain of the peer including the peer's end entity certificate. It must be called after a session has been successfully established. If peer verification was not successful the chain may be incomplete, invalid, or None. :return: A list of X509 instances giving the peer's verified certificate chain, or None if it does not have one. .. versionadded:: 20.0 N)r� �SSL_get0_verified_chainr� r� r� r� r� r� r� r� �get_verified_chain� s z Connection.get_verified_chainc C s t �| j�S )z� Checks if more data has to be read from the transport layer to complete an operation. :return: True iff more data has to be read )r� � SSL_want_readr� r� r� r� r� � want_read� s zConnection.want_readc C s t �| j�S )z� Checks if there is data to write to the transport layer to complete an operation. :return: True iff there is data to write )r� �SSL_want_writer� r� r� r� r� � want_write� s zConnection.want_writec C s t �| j� dS )z� Set the connection to work in server mode. The handshake will be handled automatically by read/write. :return: None N)r� �SSL_set_accept_stater� r� r� r� r� r_ � s zConnection.set_accept_statec C s t �| j� dS )z� Set the connection to work in client mode. The handshake will be handled automatically by read/write. :return: None N)r� rY r� r� r� r� r� r] � s z Connection.set_connect_statec C s8 t �| j�}|tjkrdS t�t�}t�|t j�|_ |S )z� Returns the Session currently used. :return: An instance of :class:`OpenSSL.SSL.Session` or :obj:`None` if no session exists. .. versionadded:: 0.14 N) r� �SSL_get1_sessionr� r� r� rx r� r �SSL_SESSION_free�_session)r� r� � pysessionr� r� r� � get_session� s zConnection.get_sessionc C s2 t |t�std��t�| j|j�}t|dk� dS )z� Set the session to be used when the TLS/SSL connection is established. :param session: A Session instance representing the session to use. :returns: None .. versionadded:: 0.14 z"session must be a Session instancer� N)r� rx r� r� �SSL_set_sessionr� r� r )r� r� r� r� r� r� � set_session� s zConnection.set_sessionc C sR t �dd�}|| j|d�}|dkr&dS td|�}|| j||� t �||�dd� S )a� Helper to implement :meth:`get_finished` and :meth:`get_peer_finished`. :param function: Either :data:`SSL_get_finished`: or :data:`SSL_get_peer_finished`. :return: :data:`None` if the desired message has not yet been received, otherwise the contents of the message. :rtype: :class:`bytes` or :class:`NoneType` r< r N)r� r� r� r= r� )r� �function�emptyr, rw r� r� r� �_get_finished_message� s z Connection._get_finished_messagec C s | � tj�S )a Obtain the latest TLS Finished message that we sent. :return: The contents of the message or :obj:`None` if the TLS handshake has not yet completed. :rtype: :class:`bytes` or :class:`NoneType` .. versionadded:: 0.15 )r� r� �SSL_get_finishedr� r� r� r� � get_finished� s zConnection.get_finishedc C s | � tj�S )a! Obtain the latest TLS Finished message that we received from the peer. :return: The contents of the message or :obj:`None` if the TLS handshake has not yet completed. :rtype: :class:`bytes` or :class:`NoneType` .. versionadded:: 0.15 )r� r� �SSL_get_peer_finishedr� r� r� r� �get_peer_finished s z Connection.get_peer_finishedc C s8 t �| j�}|tjkrdS t�t �|��}|�d�S dS )a Obtain the name of the currently used cipher. :returns: The name of the currently used cipher or :obj:`None` if no connection has been established. :rtype: :class:`unicode` or :class:`NoneType` .. versionadded:: 0.15 Nri )r� �SSL_get_current_cipherr� r� r� r� �SSL_CIPHER_get_namer9 )r� �cipherr r� r� r� �get_cipher_name s zConnection.get_cipher_namec C s, t �| j�}|tjkrdS t �|tj�S dS )a. Obtain the number of secret bits of the currently used cipher. :returns: The number of secret bits of the currently used cipher or :obj:`None` if no connection has been established. :rtype: :class:`int` or :class:`NoneType` .. versionadded:: 0.15 N)r� r� r� r� r� �SSL_CIPHER_get_bits)r� r� r� r� r� �get_cipher_bits" s zConnection.get_cipher_bitsc C s8 t �| j�}|tjkrdS t�t �|��}|�d�S dS )a% Obtain the protocol version of the currently used cipher. :returns: The protocol name of the currently used cipher or :obj:`None` if no connection has been established. :rtype: :class:`unicode` or :class:`NoneType` .. versionadded:: 0.15 Nri )r� r� r� r� r� r� �SSL_CIPHER_get_versionr9 )r� r� r r� r� r� �get_cipher_version2 s z Connection.get_cipher_versionc C s t �t�| j��}|�d�S )a> Retrieve the protocol version of the current connection. :returns: The TLS version of the current connection, for example the value for TLS 1.2 would be ``TLSv1.2``or ``Unknown`` for connections that were not successfully established. :rtype: :class:`unicode` ri )r� r� r� �SSL_get_versionr� r9 r$ r� r� r� �get_protocol_version_nameC s z$Connection.get_protocol_version_namec C s t �| j�}|S )a Retrieve the SSL or TLS protocol version of the current connection. :returns: The TLS version of the current connection. For example, it will return ``0x769`` for connections made over TLS version 1. :rtype: :class:`int` )r� � SSL_versionr� r$ r� r� r� �get_protocol_versionO s z Connection.get_protocol_versionc C sR |s t d��d�t�dd� |D ���}t�d|�}tt�| j |t |��dk� dS )ah Specify the client's ALPN protocol list. These protocols are offered to the server during protocol negotiation. :param protos: A list of the protocols to be offered to the server. This list should be a Python list of bytestrings representing the protocols to offer, e.g. ``[b'http/1.1', b'spdy/2']``. r� r� c s s | ]}t t|�f�|fV qd S r� r� r� r� r� r� r� n r� z-Connection.set_alpn_protos.<locals>.<genexpr>r� r N) r� r� r r� r� r� r r� �SSL_set_alpn_protosr� r� r� r� r� r� r� Z s � �zConnection.set_alpn_protosc C sH t �d�}t �d�}t�| j||� |s,dS t �|d |d �dd� S )z� Get the protocol that was negotiated by ALPN. :returns: A bytestring of the protocol name. If no protocol has been negotiated yet, returns an empty bytestring. r� zunsigned int *r� r N)r� r� r� �SSL_get0_alpn_selectedr� r� )r� r� �data_lenr� r� r� �get_alpn_proto_negotiated} s z$Connection.get_alpn_proto_negotiatedc C s t �| jt j�}t|dk� dS )a Called to request that the server sends stapled OCSP data, if available. If this is not called on the client side then the server will not send OCSP data. Should be used in conjunction with :meth:`Context.set_ocsp_client_callback`. r� N)r� �SSL_set_tlsext_status_typer� �TLSEXT_STATUSTYPE_ocspr )r� r� r� r� r� � request_ocsp� s �zConnection.request_ocsp)N)N)r )r )N)NN)N)Kr� r� r� r r� r� r r r r r# r� r� r] ro r* r- r0 r2 r7 �writer; rB �readrE rJ rM rO rR rU rP rX rZ r\ r^ rb rf rh r� rs ru r� r� rw rz r| r� r� r� r� r� r� r� � staticmethodr� r� r� r� r� r_ r] r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rz > s� : 7 % " " rz )�rI r r r � functoolsr r � itertoolsr r �sysr �weakrefr � OpenSSL._utilr rf r �_exception_from_error_queuer r� r r� r � _make_assertr r= r r& r rs �OpenSSL.cryptor r r r r r �__all__r r r r r r r r r r! r �SSL_SENT_SHUTDOWNr" �SSL_RECEIVED_SHUTDOWNr# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r �SSL_OP_NO_SSLv2r3 �SSL_OP_NO_SSLv3r4 �SSL_OP_NO_TLSv1r5 �SSL_OP_NO_TLSv1_1r6 �SSL_OP_NO_TLSv1_2r7 �SSL_OP_NO_TLSv1_3r8 �SSL_MODE_RELEASE_BUFFERSr9 �SSL_OP_SINGLE_DH_USEr: �SSL_OP_SINGLE_ECDH_USEr; �SSL_OP_EPHEMERAL_RSAr< � SSL_OP_MICROSOFT_SESS_ID_BUGr= � SSL_OP_NETSCAPE_CHALLENGE_BUGr> �'SSL_OP_NETSCAPE_REUSE_CIPHER_CHANGE_BUGr? �"SSL_OP_SSLREF2_REUSE_CERT_TYPE_BUGr@ �!SSL_OP_MICROSOFT_BIG_SSLV3_BUFFERrA � SSL_OP_MSIE_SSLV2_RSA_PADDINGrB � SSL_OP_SSLEAY_080_CLIENT_DH_BUGrC �SSL_OP_TLS_D5_BUGrD � SSL_OP_TLS_BLOCK_PADDING_BUGrE �"SSL_OP_DONT_INSERT_EMPTY_FRAGMENTSrF � SSL_OP_CIPHER_SERVER_PREFERENCErG �SSL_OP_TLS_ROLLBACK_BUGrH �SSL_OP_PKCS1_CHECK_1rI �SSL_OP_PKCS1_CHECK_2rJ �SSL_OP_NETSCAPE_CA_DN_BUGrK �&SSL_OP_NETSCAPE_DEMO_CIPHER_CHANGE_BUGrL �SSL_OP_NO_COMPRESSIONrM �SSL_OP_NO_QUERY_MTUrN �SSL_OP_COOKIE_EXCHANGErO �SSL_OP_NO_TICKETrP �SSL_OP_NO_RENEGOTIATIONrQ � SSL_OP_IGNORE_UNEXPECTED_EOFrR � SSL_OP_ALLrS �SSL_VERIFY_PEERrT � SSL_VERIFY_FAIL_IF_NO_PEER_CERTrU �SSL_VERIFY_CLIENT_ONCErV �SSL_VERIFY_NONErW �SSL_SESS_CACHE_OFFrX �SSL_SESS_CACHE_CLIENTrY �SSL_SESS_CACHE_SERVERrZ �SSL_SESS_CACHE_BOTHr[ � SSL_SESS_CACHE_NO_AUTO_CLEARr\ �!SSL_SESS_CACHE_NO_INTERNAL_LOOKUPr] � SSL_SESS_CACHE_NO_INTERNAL_STOREr^ �SSL_SESS_CACHE_NO_INTERNALr_ r` ra rb rc rd re rf rg rh ri rj rk rl rm rn ro rA rB r> r? r� rp r� r rq rr rs rt ru r� r� �objectrv r� r� r� r� r� r� r� rw r �Cryptography_HAS_ALPNr� r� r� rx ry rz r� r� r� r� �<module> s0 ( h �� � (=C; � �