o US`cny @s|ddlmZzddlmZWneyddlmZYnwddlmZmZm Z ddl Z ddl m Z ddl Z ddlZddlmZddlZddlZddlZdZejdkrjejd krjzddlZWn eyied we eZd ZZd ZZd ZZeeedZ e!e"e #e $Z%ejZ&Z'ddZ(Gddde)Z*Gddde*Z+Gddde*Z,Gddde*Z-Gddde*Z.Gddde*Z/Gddde*Z0dd d!d"Z1d#d$d%d&d'd(d)d*d+Z2ed,ed,ed-iZ3 . d@d/d0Z4d1d2Z5d3d4Z6e6Z7d5d6Z8e8Z9   .  dAd7d8Z:Gd9d:d:ejZ;d;d<Ze?e>es2e;j@AeeBe;eed?d?e;ZCdS)B) b64encode)Callable) EOPNOTSUPPEINVALEAGAINN)BytesIO)SEEK_CURz1.7.1nt)rz8To run PySocks on Windows you must install win_inet_ptonr )SOCKS4SOCKS5HTTPcstfdd}|S)Nc s||d}z.z |}|dkr|d|i|WW|dkr%|dSSty2}zd}~ww|dkr=|dww)NrTF) gettimeout setblocking Exception)argskwargsselfZ _is_blockingefunction0/usr/local/lib/python3.10/dist-packages/socks.pywrapper(s    z"set_self_blocking..wrapper) functoolswraps)rrrrrset_self_blocking&s rc@s"eZdZdZdddZddZdS) ProxyErrorz4Socket_err contains original socket.error exception.NcCs,||_||_|r|jd|7_dSdS)Nz: {})msg socket_errformat)rr r!rrr__init__;s zProxyError.__init__cC|jSN)r rrrr__str__BzProxyError.__str__r%)__name__ __module__ __qualname____doc__r#r'rrrrr9s  rc@ eZdZdS)GeneralProxyErrorNr)r*r+rrrrr.Fr.c@r-)ProxyConnectionErrorNr/rrrrr1Jr0r1c@r-)SOCKS5AuthErrorNr/rrrrr2Nr0r2c@r-) SOCKS5ErrorNr/rrrrr3Rr0r3c@r-) SOCKS4ErrorNr/rrrrr4Vr0r4c@r-) HTTPErrorNr/rrrrr5Zr0r5zRequest rejected or failedzLRequest rejected because SOCKS server cannot connect to identd on the clientzPRequest rejected because the client program and identd report different user-ids)[\]zGeneral SOCKS server failurez!Connection not allowed by rulesetzNetwork unreachablezHost unreachablezConnection refusedz TTL expiredz(Command not supported, or protocol errorzAddress type not supported)r r r i8iTcCs.|||||r |nd|r|ndft_dS)zSets a default proxy. All further socksocket objects will use the default unless explicitly changed. All parameters are as for socket.set_proxy().N)encode socksocket default_proxy) proxy_typeaddrportrdnsusernamepasswordrrrset_default_proxyss rGcOs$d|vr |d|d<t|i|SNZ proxytyperA)poprG)rrrrrsetdefaultproxy~srJcCstjS)z4Returns the default proxy, set by set_default_proxy.)r?r@rrrrget_default_proxyrKcCstjr t|j_dStd)aAttempts to replace a module's socket library with a SOCKS socket. Must set a default proxy using set_default_proxy(...) first. This will only work on modules that import socket directly into the namespace; most of the Python Standard Library falls into this category.zNo default proxy specifiedN)r?r@socketr.)modulerrr wrap_modules rOc  Cs,|\} } | dr| d} |r|dr|d}d} t||dtjD]g} | \}}}}}d}z=t|||}| rC| D]}|j|q;t|tt frO| ||r[| |||||||rb| || | | f|WStjtfy}z|} |r|d}WYd}~q%d}~ww| r| td)acreate_connection(dest_pair, *[, timeout], **proxy_args) -> socket object Like socket.create_connection(), but connects to proxy before returning the socket object. dest_pair - 2-tuple of (IP/hostname, port). **proxy_args - Same args passed to socksocket.set_proxy() if present. timeout - Optional socket timeout value, in seconds. source_address - tuple (host, port) for the socket to bind to as its source address before connecting (only for compatibility) [z[]Nrzgai returned empty list.) startswithstriprM getaddrinfo SOCK_STREAMr? setsockopt isinstanceintfloat settimeout set_proxybindconnecterrorrclose) dest_pairtimeoutsource_addressrA proxy_addr proxy_portZ proxy_rdnsZproxy_usernameZproxy_passwordsocket_optionsZ remote_hostZ remote_porterrrfamily socket_typeproto canonnamesasockoptrrrrcreate_connectionsD          rnc@seZdZdZddZeZdS) _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.cOsJtj|g|Ri|t|_|jD]}t|||j|<t||qdSr%) _orig_socketr#dict _savedmethods _savenamesgetattrdelattr)rposkwnamerrrr#s   z_BaseSocket.__init__N)r)r*r+r,r#listrsrrrrros rocs fddS)Ncs|j|i|Sr%)rr)rrvrwrxrrsz_makemethod..rrzrrzr _makemethods r|)sendtosendrecvfromrecvcs@eZdZdZdZejejdffdd ZddZ fdd Z d d Z d d Z  d7ddZ ddZfddZfddZd8fdd Zd8fdd ZddZfddZdd ZeZd!d"ZeZd#d$ZeZd%d&Zfd'd(Zd)d*Zd+d,Zd-d.Zd/d0Z e!ee"ee#e iZ$e%d9fd1d2 Z&e%d3d4Z'd5d6Z(Z)S):r?a2socksocket([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 and proto=0. The "type" argument must be either SOCK_STREAM or SOCK_DGRAM. Nrcsv|tjtjfvrd}t||tt|j|||g|Ri|d|_|j r-|j |_ nd|_ d|_ d|_ d|_ dS)Nz0Socket type must be stream or datagram, not {!r})NNNNNN)rMrT SOCK_DGRAM ValueErrorr"superr?r# _proxyconnr@proxyproxy_socknameproxy_peername_timeout)rrgtyperirrr  __class__rrr#s"  zsocksocket.__init__cCsFd}t||kr!||t|}|std||7}t||ks|S)zReceive EXACTLY the number of bytes requested from the file object. Blocks until the required number of bytes have been received.Connection closed unexpectedly)lenreadr.)rfilecountdatadrrr_readalls  zsocksocket._readallcs>||_z|}tt||jWdStjyYdSwr%)rget_proxy_peernamerr?rYrMr])rr`ZpeerrrrrYszsocksocket.settimeoutcCr$r%)rr&rrrr#r(zsocksocket.gettimeoutcCs |r |ddS|ddS)Ng)rY)rvrrrr&szsocksocket.setblockingTcCs.|||||r |nd|r|ndf|_dS)a Sets the proxy to be used. proxy_type - 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 performed 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)rrArBrCrDrErFrrrrZ,s zsocksocket.set_proxycOs&d|vr |d|d<|j|i|SrH)rIrZ)rrrrrrsetproxyAszsocksocket.setproxycs|j\}}}}}}|r|jtjkrtj|g|Ri|S|jr&ttd|t kr2d} tt | t t |j|i|| \} } d| f} t|_|} |j| d}||j|| \} }| \}} |\} } t t ||| ft t ||jd|_dS)zVImplements proxy connection for UDP sockets. Happens during the bind() phase.z"Socket already bound to an addressz'UDP only supported by SOCKS5 proxy type0)0.0.0.0rN)rrrMrrpr[rr]rrrrr? getsockname _proxy_addrr\_SOCKS5_requestrYrr)rrvrwrArbrcrDrErFr _rCdstrZ UDP_ASSOCIATEZrelayhostrrrr[Fs.      zsocksocket.bindc s|jtjkrtt|j|g|Ri|S|js|d|d}|dd}t}d}| |d}| || ||tt|j | |g|Ri|} | | S)Nrs)rrMrrr?r}rr[rwrite_write_SOCKS5_addressr~getvaluetell) rbytesrraddressflagsheaderZRSVZ STANDALONEsentrrrr}ks       zsocksocket.sendtoc s>|jtjkr|j|||jfi|Stt|j||fi|Sr%)rrMrr}rrr?r~)rrrrrrrr~s zsocksocket.sendc s|jtjkrtt|||S|js|dttt| |d|}| dt | d}t |r7td||\}}|jrV|j\}}||ksP|d|fvrVttd| |||ffS)Nrir r zReceived UDP packet fragmentrzPacket filtered)rrMrrr?rrr[rrseekrrordNotImplementedError_read_SOCKS5_addressrr]r) rbufsizerbuffragZfromhostZfromportZpeerhostZpeerportrrrrs      zsocksocket.recvfromcOs|j|i|\}}|Sr%)r)rrvrwrrrrrrszsocksocket.recvcs|jr|jtt|Sr%)rr^rr?r&rrrr^s zsocksocket.closecCr$)z:Returns the bound IP address and port number at the proxy.)rr&rrrget_proxy_socknamerLzsocksocket.get_proxy_socknamecCs|S)z> Returns the IP and port number of the proxy. ) getpeernamer&rrrrszsocksocket.get_proxy_peernamecCr$)zwReturns the IP address and port number of the destination machine. Note: get_proxy_peername returns the proxy.)rr&rrr get_peernameszsocksocket.get_peernamecGsd}||||\|_|_dS)z7Negotiates a stream connection through a SOCKS5 server.N)rrr)r dest_addrZCONNECTrrr_negotiate_SOCKS5szsocksocket._negotiate_SOCKS5cs|j\}}}}}} |d} |dd} z|r| r| dn| d| || d} | dddkr:td | ddd kr|rF| sJtd | d tt| |tt|  | | || d} | ddd krytd | ddd krtdn| ddd kr| dddkrtdtd | d|d | || }| || d}|dddkrtd t |dd}|dkrt |d}td|||| }tt||j||fW| | S| | w)z Send SOCKS5 request with given command (CMD field) and address (DST field). Returns resolved DST address that was used. wbrbrssr r %SOCKS5 proxy server sent invalid datazPNo username/password supplied. Server requested username/password authenticationrrzSOCKS5 authentication failedz7All offered SOCKS5 authentication methods were rejectedr Unknown error {:#04x}: {})rmakefilerflushrr.r2chrrr>rr SOCKS5_ERRORSgetr3r"rrr?rYrr^)rconncmdrrArBrCrDrErFwriterreaderZ chosen_authZ auth_statusresolvedrespstatusr]Zbndrrrrsv            zsocksocket._SOCKS5_requestc CsD|\}}|j\}}}}}} tjdtjdi} tjtjfD]1} z%t| |} || | | t| | }|td|||fWStj yLYqw|rd| d} |dt t |  | n1t ||tjtjtjtj}|d}|d} |dd}t| |} || | | t| | }|td|||fS)z~ Return the host and port packed for the SOCKS5 protocol, and the resolved address as a tuple object. r>Hidnarrr9)rrMAF_INETAF_INET6 inet_ptonr inet_ntopstructpackr]r>rrrS AF_UNSPECrT IPPROTO_TCP AI_ADDRCONFIG)rrBrrrCrArrDrErFZfamily_to_byterg addr_bytesZ host_bytes addressesZ target_addrrrrr s:        z socksocket._write_SOCKS5_addresscCs||d}|dkrt||d}n'|dkr'||d}||t|}n|dkr7ttj||d}ntdtd||d d }||fS) Nr rr9rrrrr r) rrM inet_ntoarrrr.runpack)rrZatyprBlengthrCrrrrMs  zsocksocket._read_SOCKS5_addresscCs|j\}}}}}}|d} |dd} zd} zt|} Wntjy7|r-d} d} ntt|} Ynw| tddd || | |rO| || d | r`| | d d | | | d } | dd d krvt d t | d d}|dkrt|d}td||t| ddtd| dddf|_| rt| |f|_n||f|_W| | dSW| | dS| | w)z0Negotiates a connection through a SOCKS4 server.rrrFsTz>BBHr9r rrr=z%SOCKS4 proxy server sent invalid datar ZrrNr)rrrM inet_atonr] gethostbynamerrrr>rrr.r SOCKS4_ERRORSrr4r"rrrrr^)rr dest_portrArBrCrDrErFrrZremote_resolverrrr]rrr_negotiate_SOCKS4\s\           zsocksocket._negotiate_SOCKS4cCsX|j\}}}}}}|r |nt|}d|ddt|dd|dg} |r;|r;| dt|d|| d|d| | } | } | | sZt dz | d d \} } }Wn typt d w| d szt d zt| } Wn tytdw| dkrd| |}| dvr|d7}t|d|_||f|_dS)zwNegotiates a connection through an HTTP server. NOTE: This currently only supports HTTP CONNECT-style proxies.sCONNECT r:s HTTP/1.1sHost: sProxy-Authorization: basic s r r z'HTTP proxy server sent invalid responsezHTTP/z0Proxy server does not appear to be an HTTP proxyz4HTTP proxy server did not return a valid HTTP statusz{}: {})iiiza [*] Note: The HTTP proxy server may not be supported by PySocks (must be a CONNECT tunnel proxy))s0.0.0.0rN)rrMrr>strappendrsendalljoinrreadliner^r.splitrrQrWr5r"rr)rrrrArBrCrDrErFZ http_headersZfobjZ status_lineri status_codeZ status_msgr]rrr_negotiate_HTTPsV        zsocksocket._negotiate_HTTPc st|dks |ddrtdt||\}}|jtjkr?|js(|dt |}|dkr8|s8d|_ dS||f|_ dS|j \}}}}} } t |t tfr\t|dks\|r\t |ts`tdtt||j|dur||_ tt||jtt|||fdS|}z tt||Wn5tjy} z(||s|\}}d ||} t|} d | | }td || t|| | d} ~ wwz|j|}||||WdStjy} z |s|td | | d} ~ wty|w) z Connects to the specified destination through a proxy. Uses the same API as socket's connect(). To select the proxy server, use set_proxy(). dest_pair - 2-tuple of (IP/hostname, port). r rrPz PySocks doesn't support IPv6: %srrNz0Invalid destination-connection (host, port) pairz{}:{}zError connecting to {} proxy {}z %s due to: %sz Socket error)rrQrMr]rrrrr[rrrrVrytuplerWr.rr?rYrr\rr^r"PRINTABLE_PROXY_TYPESlogdebugr1_proxy_negotiatorsr)rr_ catch_errorsrrrArbrcrDrErFr]Z proxy_serverZprintable_typer Z negotiaterrrr\s~             zsocksocket.connectc CsHz |j|ddWdSty#}z |jr|jWYd}~Sd}~ww)a& https://docs.python.org/3/library/socket.html#socket.socket.connect_ex Like connect(address), but return an error indicator instead of raising an exception for errors returned by the C-level connect() call (other problems, such as "host not found" can still raise exceptions). T)rrN)r\OSErrorerrno)rr_rrrr connect_ex6szsocksocket.connect_excCs4|j\}}}}}}|pt|}|std||fS)zD Return proxy address to connect to as tuple object zInvalid proxy type)r DEFAULT_PORTSrr.)rrArbrcrDrErFrrrrFs zsocksocket._proxy_addrNNNTNN)rr%)*r)r*r+r,r@rMrrTr#rrYrrrZrr[r}r~rrr^rZgetproxysocknamerZgetproxypeernamerrrrrrrrr rrrrr\rr __classcell__rrrrr?sR     %  b-<<\ r?r) NNNNNTNNN)Dbase64rcollections.abcr ImportError collectionsrrrrriorloggingosrrMrsys __version__rx version_infoZ win_inet_pton getLoggerr)rZPROXY_TYPE_SOCKS4r ZPROXY_TYPE_SOCKS5rZPROXY_TYPE_HTTPrZ PROXY_TYPESrqzipvalueskeysrZ _orgsocketrprIOErrorrr.r1r2r3r4r5rrrrGrJrKZgetdefaultproxyrOZ wrapmodulernror|rtmethodrVrsrsetattrr?rrrrs                =