o -j@sUddlZddlZddlZddlZddlmZmZmZmZddl Z e e Z e j e je jdZeeefed<GdddeZGdddZdS) N)TupleDict TypedDictOptional)SOCKS4SOCKS5HTTPproxy_type_by_schemec@s>eZdZUeed<eed<eed<eeed<eeed<dS)ProxyschemehostnameportusernamepasswordN)__name__ __module__ __qualname__str__annotations__intrrrb/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/pyrogram/connection/transport/tcp/tcp.pyr %s  r c@seZdZdZdededdfddZdeee fddfd d Z deee fddfd d Z deee fddfd dZ deee fddfddZ dddZdeddfddZdde deefddZdS)TCP ipv6proxyreturnNcCs0||_||_d|_d|_t|_t|_dSN) rrreaderwriterasyncioLocklockget_event_looploop)selfrrrrr__init__0s  z TCP.__init__ destinationc s|jd}|durtdt|}|dur!td||jd}|jd}|jd}|jd}zt|}Wn tyKd} Ynwt|tj} | rWt j nt j } t | } | j|||||d | tj|jj| |d IdH| dtj| d IdH\|_|_dS) Nr zNo scheme specifiedzUnknown proxy type r r rrF) proxy_typeaddrr rr)sockaddress)r*)rget ValueErrorr upper ipaddress ip_address isinstance IPv6AddresssocketAF_INET6AF_INETsocks socksocket set_proxy settimeoutrTIMEOUTr$ sock_connect setblockingr open_connectionrr) r%r'r r(r r rrr0 is_proxy_ipv6 proxy_familyr*rrr_connect_via_proxy:sF          zTCP._connect_via_proxycs>|\}}|jr tjntj}tj|||dIdH\|_|_dS)N)hostr family)rr3r4r5r r=rr)r%r'rAr rBrrr_connect_via_directiszTCP._connect_via_directcs0|jr||IdHdS||IdHdSr)rr@rC)r%r'rrr_connectusz TCP._connectr+cs<zt||tjIdHWdStjytdw)NzConnection timed out)r wait_forrDrr: TimeoutError)r%r+rrrconnect{s  z TCP.connectc sv|jdurdSz|jt|jtjIdHWdSty:}zt dt |j |WYd}~dSd}~ww)NzClose exception: %s %s) rcloser rE wait_closedrr: Exceptionloginfotyper)r%errrrHs   "z TCP.closedatac s|jdurdS|j4IdH7z|j||jIdHWnty:}ztdt|j|t |d}~wwWdIdHdS1IdHsLwYdS)NzSend exception: %s %s) rr"writedrainrJrKrLrMrOSError)r%rOrNrrrsends  .zTCP.sendrlengthc sxd}t||kr:zt|j|t|tjIdH}Wn ttjfy*YdSw|r2||7}ndSt||ks |S)N) lenr rErreadrr:rRrF)r%rTrOchunkrrrrecvs    zTCP.recv)rN)r)rrrr:boolr r&rrrr@rCrDrGrHbytesrSrrYrrrrr-s$  /      r)r r/loggingr3typingrrrrr6 getLoggerrrKrrrr rrrr rrrrrs