U Vh@sddlZddlZddlmZddlmZmZddlmZm Z m Z m Z Gddde ej ZGdd d e ej ZGd d d ejZGd d d ejZGdddejZGdddZdS)N) StrOrBytes) ProtocolError SOCKSError) AddressTypedecode_addressencode_address#get_address_port_tuple_from_addressc@s eZdZdZdZdZdZdZdS)SOCKS4ReplyCodez"Enumeration of SOCKS4 reply codes.Z[\]N)__name__ __module__ __qualname____doc__REQUEST_GRANTEDREQUEST_REJECTED_OR_FAILEDCONNECTION_FAILEDAUTHENTICATION_FAILEDrrP/var/www/html/arya_register/venvv2/lib/python3.8/site-packages/socksio/socks4.pyr s r c@seZdZdZdZdZdS) SOCKS4Commandz$Enumeration of SOCKS4 command codes.N)rrrrCONNECTBINDrrrrrsrc@seZdZUdZeed<eed<eed<dZe j eed<e d ee j e e je effe j edddd Zde j eed d d ZdS) SOCKS4RequestajEncapsulates a request to the SOCKS4 proxy server Args: command: The command to request. port: The port number to connect to on the target host. addr: IP address of the target host. user_id: Optional user ID to be included in the request, if not supplied the user *must* provide one in the packing operation. commandportaddrNuser_idraddressr"returncCs:t|\}}t|\}}|tjkr*td|||||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. user_id: Optional user ID. Returns: A SOCKS4Request instance. Raises: SOCKSError: If a domain name or IPv6 address was supplied. z;IPv6 addresses and domain names are not supported by SOCKS4rr!r r")r rrIPV4rclsrr$r"r ZatypeZ encoded_addrrrr from_address.s   zSOCKS4Request.from_addressr"r%cCs@|p|j}|dkrtddd|j|jjddd|j|dgS) Packs the instance into a raw binary in the appropriate form. Args: user_id: Optional user ID as an override, if not provided the instance's will be used, if none was provided at initialization an error is raised. Returns: The packed request. Raises: SOCKSError: If no user was specified in this call or on initialization. N-SOCKS4 requires a user_id, none was specifiedbig byteorderr"rjoinrr to_bytesr!selfr"rrrdumpsKs zSOCKS4Request.dumps)N)Nrrrrr__annotations__intbytesr"typingOptional classmethodUnionrTupler*r:rrrrrs  rc@seZdZUdZeed<eed<eed<dZe j eed<e d ee j e e je effe j edddd Zde j eed d d ZdS)SOCKS4ARequestakEncapsulates a request to the SOCKS4A proxy server Args: command: The command to request. port: The port number to connect to on the target host. addr: IP address of the target host. user_id: Optional user ID to be included in the request, if not supplied the user *must* provide one in the packing operation. rr r!Nr"r#cCs(t|\}}t|\}}|||||dS)acConvenience 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. user_id: Optional user ID. Returns: A SOCKS4ARequest instance. r&)r rr(rrrr*xs  zSOCKS4ARequest.from_addressr+c CsD|p|j}|dkrtddd|j|jjdddd|d |jd gS) r,Nr-r.r/r0r1r2sr4r5r8rrrr:s zSOCKS4ARequest.dumps)N)Nr;rrrrrDhs  rDc@sDeZdZUdZeed<eed<eje ed<e e ddddZ dS) SOCKS4ReplyzEncapsulates a reply from the SOCKS4 proxy server Args: reply_code: The code representing the type of reply. port: The port number returned. addr: Optional IP address returned. reply_coder r!datar%c Cst|dks|dddkr$tdz>|t|ddtj|dddd ttj|ddd 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. rrr4zMalformed replyr0r1r2)rFr r!N) lenrr r= from_bytesrrr' ValueError)r)rHexcrrrloadss zSOCKS4Reply.loadsN) rrrrr r<r=r?r@strrAr>rOrrrrrEs rEc@sVeZdZdZedddZejee fddddZ ee d d d Z ed d dZ dS)SOCKS4ConnectionzEncapsulates a SOCKS4 and SOCKS4A connection. Packs request objects into data suitable to be send and unpacks reply data into their appropriate reply objects. Args: user_id: The user ID to be sent as part of the requests. r"cCs||_t|_t|_dS)N)r" bytearray _data_to_send_received_datar8rrr__init__szSOCKS4Connection.__init__N)requestr%cCs&|jp |j}|j|j|d7_dS)zPacks a request object and adds it to the send data buffer. Args: request: The request instance to be packed. rRN)r"rTr:)r9rWr"rrrsends zSOCKS4Connection.sendrGcCs|j|7_tt|jS)zUnpacks response data into a reply object. Args: data: The raw response data from the proxy server. Returns: The appropriate reply object. )rUrErOr>r9rHrrr receive_datas zSOCKS4Connection.receive_data)r%cCst|j}t|_|S)zqReturns the data to be sent via the I/O library of choice. Also clears the connection's buffer. )r>rTrSrYrrr data_to_sends zSOCKS4Connection.data_to_send)rrrrr>rVr?rBrrDrXrErZr[rrrrrQs    rQ)enumr?_typesr exceptionsrrutilsrrrr r>Enumr r NamedTuplerrDrErQrrrrs  JE$