File: //usr/lib/python3.6/site-packages/sshuttle/methods/__pycache__/__init__.cpython-36.pyc
3
�m�Z� � @ s| d dl Z d dlZd dlZd dlZd dlZd dlmZmZ dd� ZG dd� de �Z
G dd� de �Z d d
� Z d d � Z
d
d� ZdS )� N)�Fatal�debug3c C s� ydd}d}| j tj||�}tjd|d d� �\}}}}}} tj|�tjksNt�d|||| f }
|
|fS tjk
r� } z | j d t
j kr�| j � S � W Y d d } ~ X nX d S )N�P � z!HHBBBB� z %d.%d.%d.%dr )
Z
getsockopt�socketZSOL_IP�struct�unpackZhtonsZAF_INET�AssertionError�error�args�errnoZ ENOPROTOOPTZ getsockname) �sockZSO_ORIGINAL_DSTZ SOCKADDR_MINZ sockaddr_in�proto�port�a�b�c�dZip�e� r � /usr/lib/python3.6/__init__.py� original_dst s r c @ s e Zd ZdS )�FeaturesN)�__name__�
__module__� __qualname__r r r r r s r c @ s| e Zd Zdd� Zdd� Zedd� �Zedd� �Zed d
� �Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zedd� �ZdS )�
BaseMethodc C s d | _ || _d S )N)�firewall�name)�selfr r r r �__init__ s zBaseMethod.__init__c C s
|| _ d S )N)r )r r r r r � set_firewall# s zBaseMethod.set_firewallc C s" t � } d| _d| _d| _d| _| S )NFT)r �ipv6�udp�dns�user)�resultr r r �get_supported_features& s z!BaseMethod.get_supported_featuresc C s t | �S )N)r )r r r r �
get_tcp_dstip/ s zBaseMethod.get_tcp_dstipc C s t d� | j|�\}}|d |fS )NzAccept UDP using recvfrom.
)r Zrecvfrom)� udp_listener�bufsize�data�srcipr r r �recv_udp3 s zBaseMethod.recv_udpc C s* |d k rt d| j|f � |j||� d S )Nz7Method %s send_udp does not support setting srcip to %r)r r Zsendto)r r r- Zdstipr, r r r �send_udp9 s zBaseMethod.send_udpc C s d S )Nr )r Z tcp_listenerr r r �setup_tcp_listener? s z BaseMethod.setup_tcp_listenerc C s d S )Nr )r r* r r r �setup_udp_listenerB s z BaseMethod.setup_udp_listenerc C sB | j � }x4dD ],}t||�rt||� rtd|| jf ��qW d S )Nr$ r% r# r& z)Feature %s not supported with method %s.
)r$ r% r# r& )r( �getattrr r )r ZfeaturesZavail�keyr r r �assert_featuresE s
zBaseMethod.assert_featuresc C s
t � �d S )N)�NotImplementedError)r r ZdnsportZnslist�familyZsubnetsr$ r& r r r �setup_firewallM s zBaseMethod.setup_firewallc C s
t � �d S )N)r5 )r r r6 r$ r& r r r �restore_firewallQ s zBaseMethod.restore_firewallc C s dS )NFr )�liner r r �firewall_commandT s zBaseMethod.firewall_commandN)r r r r! r" � staticmethodr( r) r. r/ r0 r1 r4 r7 r8 r: r r r r r s r c C s^ t jd�pt jjt j�}x@|D ]8}d|| f }t jj|�r t jj|� oTt j|t j �S q W d S )N�PATHz%s/%s)
�os�getenv�defpath�split�pathsep�path�exists�isdir�access�X_OK)r �paths�p�fnr r r �_program_existsY s
rJ c C s t jd| �}|j| �S )Nzsshuttle.methods.%s)� importlib�
import_moduleZMethod)� method_name�moduler r r �
get_methoda s rO c C sH t d�rd} n2t d�r d} n$t d�r*d} nt d�r8d} ntd��t| �S )NZiptablesZnatZnftZpfctlZpfZipfwz9can't find either iptables, nft or pfctl; check your PATH)rJ r rO )rM r r r �get_auto_methodf s rP )r= rK r r r
Zsshuttle.helpersr r r �objectr r rJ rO rP r r r r �<module> s ;