U Vh+/@sLddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z Gddde ejZGd d d e ejZGd d d e ejZGd dde ejZGdddejZGdddejZGdddejZGdddejZGdddejZGdddejZGdddejZGdddejZejeefZGdd d ZdS)!N) StrOrBytes)singledispatchmethod) ProtocolError) AddressTypedecode_addressencode_address#get_address_port_tuple_from_addressc@s eZdZdZdZdZdZdZdS)SOCKS5AuthMethodz-Enumeration of SOCKS5 authentication methods.N)__name__ __module__ __qualname____doc__NO_AUTH_REQUIREDGSSAPIUSERNAME_PASSWORDZNO_ACCEPTABLE_METHODSrrP/var/www/html/arya_register/venvv2/lib/python3.8/site-packages/socksio/socks5.pyr s r c@seZdZdZdZdZdZdS) SOCKS5CommandzEnumeration of SOCKS5 commands.r r N)rrrrCONNECTBIND UDP_ASSOCIATErrrrrsrc@s0eZdZdZdZdZdZeeddddZ dS) SOCKS5ATypez$Enumeration of SOCKS5 address types.r r)atypereturncCs<|tjkrtjS|tjkr tjS|tjkr0tjSt|dSN) rIPV4r IPV4_ADDRESSDN DOMAIN_NAMEIPV6 IPV6_ADDRESS ValueError)clsrrrr from_atype's   zSOCKS5AType.from_atypeN) rrrrr#r%r' classmethodrr*rrrrr s rc@s4eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) SOCKS5ReplyCodez"Enumeration of SOCKS5 reply codes.r r r rrN) rrrr SUCCEEDEDZGENERAL_SERVER_FAILUREZ!CONNECTION_NOT_ALLOWED_BY_RULESETNETWORK_UNREACHABLEHOST_UNREACHABLECONNECTION_REFUSED TTL_EXPIREDCOMMAND_NOT_SUPPORTEDADDRESS_TYPE_NOT_SUPPORTEDrrrrr,2sr,c@s.eZdZUdZejeed<edddZ dS)SOCKS5AuthMethodsRequestzEncapsulates a request to the proxy for available authentication methods. Args: methods: A list of acceptable authentication methods. methodsr cCs(ddt|jjdddd|jgS)z=Packs the instance into a raw binary in the appropriate form.r-rbig byteorder)joinlenr9to_bytesselfrrrdumpsIs  zSOCKS5AuthMethodsRequest.dumpsN) rrrrtypingListr __annotations__bytesrDrrrrr8@s r8c@s.eZdZUdZeed<eeddddZdS)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 c Cs^t|dkrtdz|t|dddWStk rX}ztd|W5d}~XYnXdS)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)rJN)r@rr r()r)rLexcrrrloadsas zSOCKS5AuthReply.loadsN) rrrrr rGr+rHrPrrrrrIUs  rIc@s0eZdZUdZeed<eed<edddZdS)SOCKS5UsernamePasswordRequestzLEncapsulates a username/password authentication request to the proxy server.usernamepasswordr:c Cs8ddt|jjddd|jt|jjddd|jgS)xPacks the instance into a raw binary in the appropriate form. Returns: The packed request. r;r rr<r=)r?r@rRrArSrBrrrrDzsz#SOCKS5UsernamePasswordRequest.dumpsN)rrrrrHrGrDrrrrrQts rQc@s.eZdZUdZeed<eeddddZdS)SOCKS5UsernamePasswordReplyzLEncapsulates a username/password authentication reply from the proxy server.successrKcCs||dkdS)zUnpacks the reply authentication data into an instance. Returns: The unpacked authentication reply instance. s)rVrr)rLrrrrPsz!SOCKS5UsernamePasswordReply.loadsN) rrrrboolrGr+rHrPrrrrrUs rUc@s|eZdZUdZeed<eed<eed<eed<e ee j e e j e effddddZed d d Zeed d d ZdS)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. commandraddrport)rZaddressr cCs.t|\}}t|\}}||t|||dS)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. )rZrr[r\)r rrr*)r)rZr]r\r encoded_addrrrr from_addresss  z!SOCKS5CommandRequest.from_addressr:c Cs(dd|jd|j|j|jjdddgS)rTr;r-r rMr<r=)r?rZr packed_addrr\rArBrrrrDszSOCKS5CommandRequest.dumpscCsj|jtjkr$t|jdkst|jS|jtjkrHt|jdksBt|jSt|j}|jddd|jSdS)zGProperty returning the packed address in the correct form for its type.rr<r=N)rrr#r@r[AssertionErrorr'rA)rClengthrrrr`s   z SOCKS5CommandRequest.packed_addrN)rrrrrrGrrHintr+rEUnionrTupler_rDpropertyr`rrrrrYs  rYc@sFeZdZUdZeed<eed<eed<eed<e e ddddZ 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_coderr[r\rKc Cs|dddkrtdzTt|dd}|t|dd|tt||ddtj|dd d d d WStk r}ztd|W5d }~XYnXd S) zUnpacks the reply data into an instance. Returns: The unpacked reply instance. Raises: ProtocolError: If the data does not match the spec. rrr-rNrarMNr<r=)rjrr[r\) rrr,rrfrom_socks5_atypere from_bytesr()r)rLrrOrrrrPs zSOCKS5Reply.loadsN) rrrrr,rGrstrrer+rHrPrrrrris  ric@sdeZdZUdZeed<eed<eed<eed<eed<eed<e eddd d Z ed d d Z dS)SOCKS5DatagramzXEncapsulates a SOCKS5 datagram for UDP connections. Currently not implemented. rr[r\rLfragmentZ last_fragmentrKcCs tdSr!NotImplementedErrorrWrrrrPszSOCKS5Datagram.loadsr:cCs tdSr!rrrBrrrrDszSOCKS5Datagram.dumpsN) rrrrrrGrHrerXr+rPrDrrrrrp s rpc@s,eZdZdZdZdZdZdZdZdZ dZ d S) SOCKS5Statez&Enumeration of SOCKS5 protocol states.rrMrkraN) rrrrCLIENT_AUTH_REQUIREDSERVER_AUTH_REPLYCLIENT_AUTHENTICATED TUNNEL_READY$CLIENT_WAITING_FOR_USERNAME_PASSWORDSERVER_VERIFY_USERNAME_PASSWORD MUST_CLOSErrrrrt#srtc@seZdZdZddddZeedddZee ddd d Z e e e ddd d Z e eeddd dZe ee ddddZeejeeefdddZedddZdS)SOCKS5ConnectionzEncapsulates a SOCKS5 connection. Packs request objects into data suitable to be send and unpacks reply data into their appropriate reply objects. Nr:cCst|_t|_tj|_dSr!) bytearray _data_to_send_received_datartrx_staterBrrr__init__9szSOCKS5Connection.__init__cCs|jS)z*Returns the current state of the protocol.)rrBrrrstate>szSOCKS5Connection.state)requestr cCs tdS)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. NrrrCrrrrsendCs zSOCKS5Connection.sendcCs|j|7_tj|_dSr!)rrDrtryrrrrr _auth_methodsNszSOCKS5Connection._auth_methodscCs2|jtjkrtdtj|_|j|7_dS)Nz/Not currently waiting for username and password)rrtr|rr}rrDrrrr_auth_username_passwordSs z(SOCKS5Connection._auth_username_passwordcCs*|jtjkrtd|j|7_dS)NzASOCKS5 connections must be authenticated before sending a request)rrtrzrrrDrrrr_commandZs  zSOCKS5Connection._commandrKcCs|jtjkrDt|}|jtjkr,tj|_n|jtj kr@tj |_|S|jtj krvt |}|j rjtj |_ntj|_|S|jtj krt|}|jtjkrtj|_ntj|_|StdS)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. N)rrtryrIrPrJr rr|rrzr}rUrVr~rirjr,r1r{rs)rCrLZ auth_replyZusername_password_replyreplyrrr receive_databs(            zSOCKS5Connection.receive_datacCst|j}t|_|S)zqReturns the data to be sent via the I/O library of choice. Also clears the connection's buffer. )rHrr)rCrLrrr data_to_sends zSOCKS5Connection.data_to_send)rrrrrrhrtrrSOCKS5RequestTyperregisterr8rrQrrYrrHrErfrIrirUrrrrrrr2s   &r) enumrE_typesrcompatr exceptionsrutilsrrrr rHEnumr rrr, NamedTupler8rIrQrUrYrirpIntEnumrtrfrrrrrrs&    J)