h4dZddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZdd lmZmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5dgZ6d!dZ7d"dZ8GddZ9GddZ:d#dZ;Gd dZ %%chht|&<&<=== E E E N?!C D D D D D D D D E s9< A'A""A'cd|_dSrT)rErVs r0closez_ConnectionStdout.closeks  r2c|jSr/)rArVs r0r=z_ConnectionStdout.encodingns ~r2c|jSr/)rCrVs r0rKz_ConnectionStdout.errorsrs |r2N)r3r4r=r>r,r5)rHr>r,r5)r,rRr,r5)r,r>) __name__ __module__ __qualname____doc__rGrQrWrOr^propertyr=rKrUr2r0r<r<Ms  XXr2r<cVeZdZdZ d$d%dZd&dZd'dZd&dZd(dZd(dZ d)d!Z d&d"Z d#S)*TelnetConnectionz6 Class that represents one Telnet connection. Tconnr4addrtuple[str, int]interact-Callable[[TelnetConnection], Awaitable[None]]serverr)r=r>styleBaseStyle | None vt100_inputr enable_cprrRr,r5c  |_|_|_|_|_|_d_tj_ |_ _ d_ tdd_t|dfd t!t"t%||_dfd } dfd } d fd } t)| | | _d_dS)NF(Orowscolumnsr,rcjSr/)sizerVsr0get_sizez+TelnetConnection.__init__..get_sizes 9 r2)r=rHr-r5c<j|dS)z-TelnetProtocolParser 'data_received' callbackN)rq send_bytesrHrFs r0 data_receivedz0TelnetConnection.__init__..data_receiveds   ' ' - - - - -r2rwr+rxct||_j$jrjddSdSdS)z-TelnetProtocolParser 'size_received' callbackrvNcBtSr/)r _on_resizerUr2r0zBTelnetConnection.__init__..size_received..s)=)=)?)?r2)rrz vt100_outputcontextrun)rwrxrFs r0 size_receivedz0TelnetConnection.__init__..size_receivedsU$888DI ,,   !?!?@@@@@-,,,r2ttyper>cttj|_jdS)z.TelnetProtocolParser 'ttype_received' callback)termrrN)rstdoutr_readyset)rrrr{rFs r0ttype_receivedz1TelnetConnection.__init__..ttype_receiveds? , XEj!!!D  KOO     r2)r,rrHr-r,r5)rwr+rxr+r,r5)rr>r,r5)rirjrlrnr=rorEasyncioEventrrqrrrrrzr:r r r<rr(parserr) rFrirjrlrnr=rorqrrrrrr{s ` ` @r0rGzTelnetConnection.__init__|sK         moo &$15b"---  4         6#4TH#M#M#MNN  . . . . . . A A A A A A         +=-XX 37 r2cKdfd }t}|j| jd{Vt jj5tj _ d{Vdddn #1swxYwY dS# wxYw)z" Run application. r,r5cjd}|r|dStjdjzdS)Niz"Connection closed by client. %r %r)rirecvfeedrr7rjr^r~s r0handle_incoming_dataz>TelnetConnection.run_application..handle_incoming_datas\9>>$''D  $ @49LMMM r2N)inputoutputra) r add_readerrirwaitrrqr contextvars copy_contextrrlr^)rFrloops` r0run_applicationz TelnetConnection.run_applications;        !!  #7888 +""$$ $ $ $ $ $ $ $#$*:4CTUUU * **799 mmD))))))))) * * * * * * * * * * * * * * * JJLLLLLDJJLLLLs/:C.4B." C.B22C5B26CC%rHr-c:|j|dS)zF Handler for incoming data. (Called by TelnetServer.) N)rrrPs r0rzTelnetConnection.feeds  r2c|jszd|_|jt|j|j|jdSdS)z# Closed by client. TN)rErqr^r remove_readerrirrVs r0r^zTelnetConnection.closes|| DL   " " $ $ $    , ,TY 7 7 7 IOO    K          r2formatted_textrc|jdSt|}t|j||jp t dS)z* Send text to the client. N)rrrrorrFrs r0r8zTelnetConnection.sendsP   $ F*>::  ~tz/IZ\\     r2cVtfdS)z\ Send text to the client. This is asynchronous, returns a `Future`. c.Sr/)r8)rrFsr0rz4TelnetConnection.send_above_prompt..sTYY~-F-Fr2)r_run_in_terminalrs``r0send_above_promptz"TelnetConnection.send_above_prompts4 +>::$$%F%F%F%F%FGGGr2funcCallable[[], None]cr|jr"|jt|dStd)Nz2Called _run_in_terminal outside `run_application`.)rrr RuntimeError)rFrs r0rz!TelnetConnection._run_in_terminals< < U L  _d 3 3 3 3 3STT Tr2c|jdS|j|jdd|jdS)zB Erase the screen and move the cursor to the top. Nr)r erase_screen cursor_gotorOrVs r0rzTelnetConnection.erase_screens[   $ F &&((( %%a+++ !!!!!r2N)T)rir4rjrkrlrmrnr)r=r>rorprqrrrrRr,r5rar)rrr,r5)rrr,r5) rbrcrdrerGrrr^r8rrrrUr2r0rhrhws 5858585858n6          HHHHUUUU""""""r2rhc KdSr/rUr9s r0_dummy_interactrs Dr2cbeZdZdZddedddfddZeddZd d!dZd"dZ d"dZ d#dZ dS)$r)a Telnet server implementation. Example:: async def interact(connection): connection.send("Welcome") session = PromptSession() result = await session.prompt_async(message="Say something: ") connection.send(f"You said: {result} ") async def main(): server = TelnetServer(interact=interact, port=2323) await server.run() z 127.0.0.1zutf-8NThostr>portr+rlrmr=rorprrrRr,r5c||_||_||_||_||_||_d|_g|_t|_ dSr/) rrrlr=rorr _run_task_application_tasksr connections)rFrrrlr=rorrs r0rGzTelnetServer.__init__sO       $48<>25%%r2r4ctjtjtj}|tjtjd|||f|d|S)Nr)socketAF_INET SOCK_STREAM setsockopt SOL_SOCKET SO_REUSEADDRbindlisten)clsrrss r0_create_socketzTelnetServer._create_socket0s` M&.&*< = = V&(;Q??? d|  r2ready_cbCallable[[], None] | Nonec~Kjjtjdjjt fd|r | tjd{Vt j j D]}| tj dkr.tjj dtjd{VdSdS#t j j D]}| tj dkr-tjj dtjd{VwwxYw)z Run the telnet server, until this gets cancelled. :param ready_cb: Callback that will be called at the point that we're actually listening. z.Listening for telnet connections on %s port %rc.Sr/)_accept)rFrsr0rz"TelnetServer.run..Fsdll66J6Jr2Nr)timeout return_when)rrrrr7rrrFuturerr^rcancellenr ALL_COMPLETED)rFrtrs` @r0rzTelnetServer.run:s$$TY :: % F)++77 CCr2cK|j?|j |jd{VdS#tj$rYdSwxYwdS)z Deprecated: Use `.run()` instead. Stop a telnet server that was started using `.start()` and wait for the cancellation to complete. N)rrrCancelledErrorrVs r0stopzTelnetServer.stopmsv > % N ! ! # # # n$$$$$$$$$)     & %s 3AA listen_socketc|\tjdgRdfd }t|jdS)z1 Accept new incoming connection. zNew connection %r %rr,r5c K t5}tjjj|j}j|tj dgR | d{Vj |tj dgRn1#j |tj dgRwxYw dddn #1swxYwYn#t$rtj dYnrt$rtj dYnSt$rG}tdt!|jz|ddl}|Yd}~nd}~wwxYwj dS#j wxYw)N)r=rorqrrzStarting interaction %r %rzStopping interaction %r %rz)Unhandled EOFError in telnet application.z2Unhandled KeyboardInterrupt in telnet application.zGot %sr)rrhrlr=rorrraddrr7rremoveEOFErrorKeyboardInterrupt BaseExceptionprinttyperb traceback print_excr)rqr3r\rrjrirFtasks r0rz!TelnetServer._accept..runs|# 5&((IK!1 !%"j$/#'? " " "J$((444K <DtDDDDI(88:::::::::(// ;;; $@H4HHHHH(// ;;; $@H4HHHHHH%IIIIIIIIIIIIIII& I I I  GHHHHH$ R R R PQQQQQ  & & &ha!111555    ##%%%%%%%%  & '..t44444'..t4444sC=AC1,B3-C13.C!!C1% C=1C55C=8C59C=<F+=F F+F :F+< F =FF+F  F++GNra)acceptrr7rrrrM)rFrrrjrirs` @@@r0rzTelnetServer._accept{s#))++ d *2T2222$ 5$ 5$ 5$ 5$ 5$ 5$ 5$ 5$ 5L !!--ccee44 &&t,,,,,r2)rr>rr+rlrmr=r>rorprrrRr,r5)rr>rr+r,r4r/)rrr,r5ra)rr4r,r5) rbrcrdrerrG classmethodrrrrrrUr2r0r)r) s$ BQ"&88888*[%%%%%N D D D D    /-/-/-/-/-/-r2)r*r+r,r-)r3r4r,r5)r3rhr,r5)=re __future__rrrrsysrtypingrrrrr r r r r "prompt_toolkit.application.currentrr*prompt_toolkit.application.run_in_terminalrprompt_toolkit.data_structuresrprompt_toolkit.formatted_textrrprompt_toolkit.inputrrprompt_toolkit.output.vt100rprompt_toolkit.rendererrprompt_toolkit.stylesrrlogrprotocolrrrrr r!r"r#r$r%r&r'r(__all__r1r:r<rhrr)rUr2r0rs#""""" $$$$$$UUUUUUUUUUUUUUUUUUUUUUJJJJJJJJFFFFFF//////MMMMMMMM========444444PPPPPP77777777"  8888:''''''''TL"L"L"L"L"L"L"L"^    `-`-`-`-`-`-`-`-`-`-r2