OwlCyberSecurity - MANAGER
Edit File: disable_internet.cpython-38.pyc
U ��]� � @ sd d dl Z d dlZejZejZejjZejjZda da dd� Z d dd�Z d dd�Z e jd d d ��ZdS )� NFc s � fdd�}|S )z� Wraps ``original_function``, which in most cases is assumed to be a `socket.socket` method, to raise an `IOError` for any operations on non-local AF_INET sockets. c s t | d tj�rF| d jtjtjfkr0�| |�S | d d � d}d}n<t | d t�rdt| d �dksn�| |�S | d d � d}d}t�� }t�� }� ||fkr�d� � | | d f}| d |� |f | |d d � } t � fdd�|D ��r�| |�S t d ��d S ) Nr � )� localhost� 127.0.0.1z::1� )r r r c 3 s | ]}|� kV qd S )N� )�.0�h�Zhostr �E/usr/lib/python3/dist-packages/matplotlib/testing/disable_internet.py� <genexpr>? s z;check_internet_off.<locals>.new_function.<locals>.<genexpr>z[An attempt was made to connect to the internet by a test that was not marked `remote_data`.) � isinstance�socketZfamilyZAF_INETZAF_INET6�tuple�lenZ gethostnameZgetfqdn�any�IOError)�args�kwargsZaddr_argZ valid_hostsZhostnameZfqdnZnew_addr��original_functionr r � new_function" s( " z(check_internet_off.<locals>.new_functionr )r r r r r �check_internet_off s "r c C sx ddl }trdS dad}| r$td� |j�� a|j�i �}|j�|�}|j�|� tt �t _ tt �t j _ tt�t j _t S )a1 Disable internet access via python by preventing connections from being created using the socket module. Presumably this could be worked around by using some other means of accessing the internet, but all default python modules (urllib, requests, etc.) use socket [citation needed]. r NTzInternet access disabled)�urllib.request� INTERNET_OFF�print�requestZ build_opener� _orig_openerZ ProxyHandler�install_openerr �socket_create_connectionr �create_connection� socket_bind�bind�socket_connect�connect)�verbose�urllibZ__tracebackhide__Zno_proxy_handlerZopenerr r r �turn_off_internetG s r& c C sF ddl }tsdS da| r td� |j�t� tt_t tj_ t tj_ tS )zL Restore internet access. Not used, but kept in case it is needed. r NFzInternet access enabled) r r r r r r r r r r r! r" r# )r$ r% r r r �turn_on_internetk s r' c c s. t }t| d� z dV W 5 |s(t| d� X dS )a7 Context manager to temporarily disable internet access (if not already disabled). If it was already disabled before entering the context manager (i.e. `turn_off_internet` was called previously) then this is a no-op and leaves internet access disabled until a manual call to `turn_on_internet`. )r$ N)r r& r' )r$ Zalready_disabledr r r � no_internet� s r( )F)F)F)� contextlibr Zsocket_originalr r r! r r# r" r r r r& r' �contextmanagerr( r r r r �<module> s , $