j+/ZddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z Gdde ejZGd d e ejZGd d e ejZGd de ejZGddejZGddejZGddejZGddejZGddejZGddejZGddejZGddejZejeefZGdd ZdS)!N) StrOrBytes)singledispatchmethod) ProtocolError) AddressTypedecode_addressencode_address#get_address_port_tuple_from_addressc"eZdZdZdZdZdZdZdS)SOCKS5AuthMethodz-Enumeration of SOCKS5 authentication methods.N)__name__ __module__ __qualname____doc__NO_AUTH_REQUIREDGSSAPIUSERNAME_PASSWORDNO_ACCEPTABLE_METHODS>C:\Projects\Muse2API\.venv\Lib\site-packages\socksio/socks5.pyr r s-77 F#rr ceZdZdZdZdZdZdS) SOCKS5CommandzEnumeration of SOCKS5 commands.rrN)rrrrCONNECTBIND UDP_ASSOCIATErrrrrs#))G DMMMrrc>eZdZdZdZdZdZededdfdZ dS) SOCKS5ATypez$Enumeration of SOCKS5 address types.rratypereturnc|tjkr tjS|tjkr tjS|tjkr tjSt|N) rIPV4r# IPV4_ADDRESSDN DOMAIN_NAMEIPV6 IPV6_ADDRESS ValueError)clsr%s r from_atypezSOCKS5AType.from_atype'sT K$ $ $+ + kn $ $* * k& & &+ +rN) rrrrr*r,r. classmethodrr1rrrr#r# sU..LKL { }   [   rr#c6eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) SOCKS5ReplyCodez"Enumeration of SOCKS5 reply codes.r rrrr$N) rrrr SUCCEEDEDGENERAL_SERVER_FAILURE!CONNECTION_NOT_ALLOWED_BY_RULESETNETWORK_UNREACHABLEHOST_UNREACHABLECONNECTION_REFUSED TTL_EXPIREDCOMMAND_NOT_SUPPORTEDADDRESS_TYPE_NOT_SUPPORTEDrrrr4r42sJ,,I$(/%! K#!(rr4c@eZdZUdZejeed<defdZ dS)SOCKS5AuthMethodsRequestzEncapsulates a request to the proxy for available authentication methods. Args: methods: A list of acceptable authentication methods. methodsr&cddt|jddd|jgS)z=Packs the instance into a raw binary in the appropriate form.rr5rbig byteorder)joinlenrDto_bytesselfs rdumpszSOCKS5AuthMethodsRequest.dumpsIsSxxDL!!**1*>>&&    rN) rrrrtypingListr __annotations__bytesrNrrrrCrC@sQ [) ****  u       rrCc>eZdZUdZeed<ededdfdZdS)SOCKS5AuthReplyzEncapsulates a reply from the proxy with the authentication method to be used. Args: method: The authentication method to be used. Raises: ProtocolError: If the data does not conform with the expected structure. methoddatar&ct|dkrtd |t|ddS#t$r}td|d}~wwxYw)zUnpacks the authentication reply data into an instance. Returns: The unpacked authentication reply instance. Raises: ProtocolError: If the data does not match the spec. Malformed replyr)rUN)rJrr r/)r0rVexcs rloadszSOCKS5AuthReply.loadsas{ t99>> 122 2 <3.tAaCy99::: : < < < 122 ; eZdZUdZeed<ededdfdZdS)SOCKS5UsernamePasswordReplyzLEncapsulates a username/password authentication reply from the proxy server.successrVr&c"||dkS)zUnpacks the reply authentication data into an instance. Returns: The unpacked authentication reply instance. s)rdrr0rVs rr[z!SOCKS5UsernamePasswordReply.loadsss4;.////rN) rrrrboolrQr2rRr[rrrrcrcsRVV MMM00#@000[000rrcc eZdZUdZeed<eed<eed<eed<e dede j e e j e effddfdZdefd Zedefd Zd S) SOCKS5CommandRequesta Encapsulates a command request to the proxy server. Args: command: The command to request. atype: The address type of the addr field. addr: Address of the target host. port: The port number to connect to on the target host. commandr%addrportaddressr&ct|\}}t|\}}||t|||S)aConvenience class method to build an instance from command and address. Args: command: The command to request. address: A string in the form 'HOST:PORT' or a tuple of ip address string and port number. The address type will be inferred. Returns: A SOCKS5CommandRequest instance. Raises: SOCKSError: If a domain name or IPv6 address was supplied. )rjr%rkrl)r r r#r1)r0rjrmrlr% encoded_addrs r from_addressz!SOCKS5CommandRequest.from_addresssZ&>Q&&&&9  Z;3 3 3ty>>R''''9 ^^F??1?66B BrN)rrrrrrQr#rRintr2rOUnionrTuplerprNpropertyrrrrrriris  KKK III  j&,z3*GGH     [ 6 u    " CU C C CX C C Crric\eZdZUdZeed<eed<eed<eed<e de ddfdZ d S) SOCKS5ReplyaEncapsulates a reply from the SOCKS5 proxy server Args: reply_code: The code representing the type of reply. atype: The address type of the addr field. addr: Optional IP address returned. port: The port number returned. reply_coder%rkrlrVr&c |dddkrtd t|dd}|t|dd|tt j||ddt |dd d  S#t$r}td|d }~wwxYw) zUnpacks the reply data into an instance. Returns: The unpacked reply instance. Raises: ProtocolError: If the data does not match the spec. rrr5rYrtrXNrFrG)r}r%rkrl) rr#r4rrfrom_socks5_atyperw from_bytesr/)r0rVr%rZs rr[zSOCKS5Reply.loadss !9   122 2 <QqS **E3*4!955#K$A%$H$H$qQSt*UU^^DI^??    < < < 122 ; >CN) rrrrr4rQr#strrwr2rRr[rrrr|r|sx   III III<<=<<<[<<s {rrequestct)zPacks a request object and adds it to the send data buffer. Also progresses the protocol state of the connection. Args: request: The request instance to be packed. rrMrs rsendzSOCKS5Connection.sendCs"###rcl|xj|z c_tj|_dSr()rrNrrrrs r _auth_methodszSOCKS5Connection._auth_methodsNs, gmmoo-!3 rc|jtjkrtdtj|_|xj|z c_dS)Nz/Not currently waiting for username and password)rrrrrrrNrs r_auth_username_passwordz(SOCKS5Connection._auth_username_passwordSsM ;+J J J QRR R!A  gmmoo-rc|jtjkrtd|xj|z c_dS)NzASOCKS5 connections must be authenticated before sending a request)rrrrrrNrs r_commandzSOCKS5Connection._commandZsK ;9 9 9S  gmmoo-rrVc|jtjkrit|}|jt jkrtj|_n&|jt j krtj |_|S|jtj krFt|}|j rtj |_ntj|_|S|jtj krTt|}|jt"jkrtj|_ntj|_|St))zUnpacks response data into a reply object. Args: data: The raw response data from the proxy server. Returns: A reply instance corresponding to the connection state and reply data. )rrrrTr[rUr rrrrrrcrdrr|r}r4r9rr)rMrV auth_replyusername_password_replyreplys r receive_datazSOCKS5Connection.receive_databs  ;+7 7 7(..t44J $4$FFF)N "&6&GGG)>   ;+E E E&A&G&G&M&M #&. 5)> )4 * * ;+: : :%%d++E?#<<<)6 )4 L!###rcTt|j}t|_|S)zqReturns the data to be sent via the I/O library of choice. Also clears the connection's buffer. )rRrr)rMrVs r data_to_sendzSOCKS5Connection.data_to_sends% T'((&[[ r)r&N)rrrrrrzrrrSOCKS5RequestTyperregisterrCrr]rrirrRrOrxrTr|rcrrrrrrr2s 7777 {X$-$$$$$$ ]]+,,4%=4$444-,4 ]]011./L.QU...21.  ]]'((. 8.T...)(.$$$$ o{4OO P$$$$$$$$Lerr) enumrO_typesrcompatr exceptionsrutilsrrr r rREnumr rr#r4 NamedTuplerCrTr]rcrir|rIntEnumrrxrrrrrrs  ((((((%%%%%%$$$$$udi$$$E49     %   $ ) ) ) ) )eTY ) ) )     v0   *<<<<     F$5   . 0 0 0 0 0&"3 0 0 0GCGCGCGCGC6,GCGCGCT&<&<&<&<&<&#&<&<&