| Current File : //lib/python2.7/site-packages/httplib2/socks.pyo |
�
�F Vc
@ sa d Z d d l Z d d l Z d d l Z d d l Z e e d d � d k r] e d � � n d Z d Z d Z
d Z d a e j Z
d e f d
� � YZ d e f d � � YZ d
e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d1 Z d2 Z d3 Z d4 Z d d d e d d d- � Z d. � Z d/ e j f d0 � � YZ d S(5 s SocksiPy - Python SOCKS module.
Version 1.00
Copyright 2006 Dan-Haim. All rights reserved.
Redistribution and use in source and binary forms, with or without modification,
are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright notice, this
list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
3. Neither the name of Dan Haim nor the names of his contributors may be used
to endorse or promote products derived from this software without specific
prior written permission.
THIS SOFTWARE IS PROVIDED BY DAN HAIM "AS IS" AND ANY EXPRESS OR IMPLIED
WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
EVENT SHALL DAN HAIM OR HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA
OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMANGE.
This module provides a standard socket-like interface for Python
for tunneling connections through SOCKS proxies.
i����Nt sockets- socket.socket missing, proxy support unusablei i i i t
ProxyErrorc B s e Z RS( ( t __name__t
__module__( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR ; s t GeneralProxyErrorc B s e Z RS( ( R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR < s t Socks5AuthErrorc B s e Z RS( ( R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR = s t Socks5Errorc B s e Z RS( ( R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR > s t Socks4Errorc B s e Z RS( ( R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR ? s t HTTPErrorc B s e Z RS( ( R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR @ s t successs invalid datas
not connecteds
not availables bad proxy types bad inputt succeededs general SOCKS server failures! connection not allowed by rulesets Network unreachables Host unreachables Connection refuseds TTL expireds Command not supporteds Address type not supporteds
Unknown errors authentication is requireds0 all offered authentication methods were rejecteds$ unknown username or invalid passwords
unknown errors request granteds request rejected or failedsL request rejected because SOCKS server cannot connect to identd on the clientsP request rejected because the client program and identd report different user-idsc C s | | | | | | f a d S( s� setdefaultproxy(proxytype, addr[, port[, rdns[, username[, password]]]])
Sets a default proxy which all further socksocket objects will use,
unless explicitly changed.
N( t
_defaultproxy( t proxytypet addrt portt rdnst usernamet password( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt setdefaultproxy` s c C s+ t d k r t | j _ n t d � � d S( s2 wrapmodule(module)
Attempts to replace a module's socket library with a SOCKS socket. Must set
a default proxy using setdefaultproxy(...) first.
This will only work on modules that import socket directly into the namespace;
most of the Python Standard Library falls into this category.
i s no proxy specifiedN( i s no proxy specified( R t Nonet
socksocketR R ( t module( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt
wrapmoduleh s R c B s� e Z d Z e j e j d d d � Z d � Z d � Z d � Z
d � Z d d d e d d d � Z
d � Z d � Z d
� Z d � Z d � Z d
� Z d � Z RS( s socksocket([family[, type[, proto]]]) -> socket object
Open a SOCKS enabled socket. The parameters are the same as
those of the standard socket init. In order for SOCKS to work,
you must specify family=AF_INET, type=SOCK_STREAM and proto=0.
i c C sY t j | | | | | � t d k r1 t | _ n d | _ d | _ d | _ t | _ d S( N( NNNNNN( t
_orgsockett __init__R R t _socksocket__proxyt _socksocket__proxysocknamet _socksocket__proxypeernamet Truet _socksocket__httptunnel( t selft familyt typet protot _sock( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR { s c C sd | j | � } xN t | � | k r_ | j | t | � � } | sR t d � � n | | } q W| S( s� __recvall(count) -> data
Receive EXACTLY the number of bytes requested from the socket.
Blocks until the required number of bytes have been received.
i s connection closed unexpectedly( i s connection closed unexpectedly( t recvt lenR ( R t countt datat d( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __recvall� s c G s4 | j s | j | � } n t t | � j | | � S( sq override socket.socket.sendall method to rewrite the header
for non-tunneling proxies if needed
( R t _socksocket__rewriteproxyt superR t sendall( R t contentt args( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR+ � s c
C sT d \ } } | j d � } x_ | D]W } | j � j d � rF | } q" | j � j d � sp | j � j d � r" | } q" q" W| rG| rG| j | � | j | � | j d � d } | j d � } | j d d
k r| j d d
k r| j d | j � � n | j d d
| � | j d d | d | | d | d f � n d j | � S( s� rewrite HTTP request headers to support non-tunneling proxies
(i.e. those which do not support the CONNECT method).
This only works for HTTP (not HTTPS) since HTTPS requires tunneling.
s
s host:t gett postt i i i i s Host: %ss %s http://%s%s %si N( NN( R t splitt lowert
startswitht removeR t insertt _socksocket__getauthheadert join( R t headert hostt endptt hdrst hdr( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __rewriteproxy� s"
*
&/c C s- | j d d | j d } d t j | � S( Ni t :i s Proxy-Authorization: Basic ( R t base64t b64encode( R t auth( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __getauthheader� s c C s | | | | | | f | _ d S( s� setproxy(proxytype, addr[, port[, rdns[, username[, password]]]])
Sets the proxy to be used.
proxytype - The type of the proxy to be used. Three types
are supported: PROXY_TYPE_SOCKS4 (including socks4a),
PROXY_TYPE_SOCKS5 and PROXY_TYPE_HTTP
addr - The address of the server (IP or DNS).
port - The port of the server. Defaults to 1080 for SOCKS
servers and 8080 for HTTP proxy servers.
rdns - Should DNS queries be preformed on the remote side
(rather than the local side). The default is True.
Note: This has no effect with SOCKS4 servers.
username - Username to authenticate with to the server.
The default is no authentication.
password - Password to authenticate with to the server.
Only relevant when username is also provided.
N( R ( R R R
R R R R ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt setproxy� s c
C s$ | j d d
k rK | j d d
k rK | j t j d d d d d � � n | j t j d d d d � � | j d � } | d d !t d � j � k r� | j � t d t
d f � � n | d d !t d � j � k r� ng| d d !t d � j � k r�| j t d � j � t t | j d � � | j d t t | j d � � | j d � | j d � } | d d !t d � j � k r�| j � t d t
d f � � n | d d !t d � j � k rD| j � t d t
d f � � qDnU | j � | d t d � j � k r.t d t
d f � � n t d t
d f � � t j d d d d � } y- t j | � } | t d � j � | } Wn� t j k
r| j d r�d
} | t d � j � t t | � � j � | } qt j t j | � � } | t d � j � | } n X| t j d
| � } | j | � | j d � } | d d !t d � j � k r�| j � t d t
d f � � n0| d d !t d � j � k r| j � t | d d !� d k rt t | d d !� t t | d d !� f � � q�t d t d f � � n� | d d !t d � j � k rK| j d � } nq | d d !t d � j � k r�| | j d � } | j t | d d !� � } n | j � t d t
d f � � t j d
| j d � � d } | | f | _ | d
k rt j | � | f | _ n | | f | _ d
S( sk __negotiatesocks5(self,destaddr,destport)
Negotiates a connection through a SOCKS5 server.
i i t BBBBi i t BBBi i i� s >Hi i N( R R R+ t structt packt _socksocket__recvallt chrt encodet closeR t _generalerrorsR$ R t _socks5autherrorsR t inet_atont errort
gethostbynamet ordR t
_socks5errorsR# t unpackR t inet_ntoaR (
R t destaddrt destportt
chosenautht authstatt reqt ipaddrt respt boundaddrt boundport( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __negotiatesocks5� sn &%
]
3
3
c C s | j S( si getsockname() -> address info
Returns the bound IP address and port number at the proxy.
( R ( R ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt getproxysockname# s c C s
t j | � S( s` getproxypeername() -> address info
Returns the IP and port number of the proxy.
( R t getpeername( R ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt getproxypeername) s c C s | j S( s� getpeername() -> address info
Returns the IP address and port number of the destination
machine (note: getproxypeername returns the proxy)
( R ( R ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR` / s c C s\ t } y t j | � } Wn] t j k
rx | j d r] t j d d d d d � } t } qy t j t j | � � } n Xt j d d d | � | } | j d d k r� | | j d } n | t
d � j � } | r� | | t
d � j � } n | j | � | j
d � } | d d !t
d � j � k rS| j � t d t d f � � n | d d !t
d � j � k r�| j � t | d d !� d k r�| j � t t | d d !� t t | d d !� d f � � q�t d
t d f � � n t j | d � t j d | d d !� d f | _ | d k rIt j | � | f | _ n | | f | _ d S( sk __negotiatesocks4(self,destaddr,destport)
Negotiates a connection through a SOCKS4 server.
i RD i i s >BBHi i i iZ i[ i\ i] i^ s >HN( i[ i\ i] ( t FalseR RN RO R RF RG R RP R RI RJ R+ RH RK R RL RQ R t
_socks4errorsRT RS R R ( R RU RV t rmtrslvRZ RY R[ ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __negotiatesocks46 s:
73 c C s� | j d s t j | � } n | } d | d t | � d g } | d | d g 7} | j d d k r� | j d d k r� | | j � d g 7} n | j d � | j d j | � j � � | j
d
� } x2 | j d j � � d k r� | | j
d
� } q� W| j � d
j
d j � d � } | d
d j � d j � f k rf| j � t d
t d
f � � n y t | d
� } Wn1 t k
r�| j � t d
t d
f � � n X| d k r�| j � t | | d f � � n d | _ | | f | _ d S( sh __negotiatehttp(self,destaddr,destport)
Negotiates a connection through an HTTP server.
i s CONNECT R> s HTTP/1.1
s Host: s
i i t i s
i����i R0 i s HTTP/1.0s HTTP/1.1i� s 0.0.0.0N( s 0.0.0.0i ( R R RP t strR R6 t appendR+ R7 RJ R# t findt
splitlinesR1 RK R RL t intt
ValueErrorR R R ( R RU RV R
t headersR[ t
statuslinet
statuscode( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt __negotiatehttpf s4
&
""
c C s� t | � t t f k sT t | � d k sT t | d t � sT t | d � t k rm t d t d f � � n | j d t
k r� | j d d k r� | j d } n d } t j
| | j d | f � | j | d | d � n�| j d t k rU| j d d k r| j d } n d } t j
| | j d | f � | j | d | d � nN| j d t k r�| j d d k r�| j d } n d } t j
| | j d | f � | j | d | d � n� | j d t k rY| j d d k r�| j d } n d } t j
| | j d | f � | d d k rM| j | d | d � q�t | _ nJ | j d d k r�t j
| | d | d f � n t d t d f � � d S(
s� connect(self, despair)
Connects to the specified destination through a proxy.
destpar - A tuple of the IP/DNS address and the port number.
(identical to socket's connect).
To select the proxy server use setproxy().
i i i i i8 i� i� i N( R t listt tupleR$ t
isinstancet
basestringRk R RL R t PROXY_TYPE_SOCKS5R R t connectt _socksocket__negotiatesocks5t PROXY_TYPE_SOCKS4t _socksocket__negotiatesocks4t PROXY_TYPE_HTTPt _socksocket__negotiatehttpt PROXY_TYPE_HTTP_NO_TUNNELRb R ( R t destpairt portnum( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyRv � s> T !N( R R t __doc__R t AF_INETt SOCK_STREAMR R RH R+ R) R6 R RC Rw R_ Ra R` Ry R{ Rv ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyR t s
\ 0 $( s successs invalid datas
not connecteds
not availables bad proxy types bad input(
R
s general SOCKS server failures! connection not allowed by rulesets Network unreachables Host unreachables Connection refuseds TTL expireds Command not supporteds Address type not supporteds
Unknown error( R
s authentication is requireds0 all offered authentication methods were rejecteds$ unknown username or invalid passwords
unknown error( s request granteds request rejected or failedsL request rejected because SOCKS server cannot connect to identd on the clientsP request rejected because the client program and identd report different user-idss
unknown error( R R? R RF t syst getattrR t ImportErrorRx Ru Rz R| R R t ExceptionR R R R R R RL RR RM Rc R R R R ( ( ( s2 /usr/lib/python2.7/site-packages/httplib2/socks.pyt <module> s^