U Vh @sddlZddlZddlZddlZddlZddlmZejrFddlm Z e dZ Gdddej Z ejdd eeje efd d d Zejdd e eed ddZeejeefdddZejeejeeffejeefdddZdS)N) StrOrBytes SOCKS5ATypez'^\[(?P
[^\]]+)\]:(?P\d+)$c@s,eZdZdZdZdZedddddZdS) AddressTypeIPV4IPV6DNr) socks5atypereturncCsHddlm}||jkrtjS||jkr,tjS||jkrt| SdS)z&Decodes the address from a SOCKS replyN) rrr inet_ntoprrr r AssertionErrorr)r#r$rrrdecode_address/s   r')addressr cCst|tr|n|}tt|}|r>|d|d}}n|d\}}}z|t|fWSt k rzt ddYnXdS)zReturns a tuple (address: str, port: int) from an address string with a port i.e. '127.0.0.1:8080', '[0:0:0:0:0:0:0:1]:3080' or 'localhost:8080'. Note no validation is done on the domain or IP itself. r(port:zInvalid address + port. Please supply a valid domain name, IPV4 or IPV6 address with the port as a suffix, i.e. `127.0.0.1:3080`, `[0:0:0:0:0:0:0:1]:3080` or `localhost:3080`N) rrrrematchIP_V6_WITH_PORT_REGEXgroup partitionintr )r(r,Zstr_port_rrrsplit_address_port_from_string;s r2cCsPt|tr<|\}}t|tr$|}t|ttfrHt|}n t|\}}||fS)z@Returns an (address, port) from an address string-like or tuple.)rtuplerrstrr0r2)r(r)rrr#get_address_port_tuple_from_addressRs    r5)enum functoolsr+rtyping_typesr TYPE_CHECKINGr rcompiler-Enumr lru_cacheTuplerr"r4r'r0r2Unionr5rrrrs"