mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 02:15:10 +00:00 
			
		
		
		
	 2116b12da5
			
		
	
	
		2116b12da5
		
	
	
	
	
		
			
			robust at shutdown. If needs to release multiple resources, they are released even if errors are occured.
		
			
				
	
	
		
			670 lines
		
	
	
	
		
			22 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			670 lines
		
	
	
	
		
			22 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| r"""TELNET client class.
 | |
| 
 | |
| Based on RFC 854: TELNET Protocol Specification, by J. Postel and
 | |
| J. Reynolds
 | |
| 
 | |
| Example:
 | |
| 
 | |
| >>> from telnetlib import Telnet
 | |
| >>> tn = Telnet('www.python.org', 79)   # connect to finger port
 | |
| >>> tn.write(b'guido\r\n')
 | |
| >>> print(tn.read_all())
 | |
| Login       Name               TTY         Idle    When    Where
 | |
| guido    Guido van Rossum      pts/2        <Dec  2 11:10> snag.cnri.reston..
 | |
| 
 | |
| >>>
 | |
| 
 | |
| Note that read_all() won't read until eof -- it just reads some data
 | |
| -- but it guarantees to read at least one byte unless EOF is hit.
 | |
| 
 | |
| It is possible to pass a Telnet object to a selector in order to wait until
 | |
| more data is available.  Note that in this case, read_eager() may return b''
 | |
| even if there was data on the socket, because the protocol negotiation may have
 | |
| eaten the data.  This is why EOFError is needed in some cases to distinguish
 | |
| between "no data" and "connection closed" (since the socket also appears ready
 | |
| for reading when it is closed).
 | |
| 
 | |
| To do:
 | |
| - option negotiation
 | |
| - timeout should be intrinsic to the connection object instead of an
 | |
|   option on one of the read calls only
 | |
| 
 | |
| """
 | |
| 
 | |
| 
 | |
| # Imported modules
 | |
| import sys
 | |
| import socket
 | |
| import selectors
 | |
| from time import monotonic as _time
 | |
| 
 | |
| __all__ = ["Telnet"]
 | |
| 
 | |
| # Tunable parameters
 | |
| DEBUGLEVEL = 0
 | |
| 
 | |
| # Telnet protocol defaults
 | |
| TELNET_PORT = 23
 | |
| 
 | |
| # Telnet protocol characters (don't change)
 | |
| IAC  = bytes([255]) # "Interpret As Command"
 | |
| DONT = bytes([254])
 | |
| DO   = bytes([253])
 | |
| WONT = bytes([252])
 | |
| WILL = bytes([251])
 | |
| theNULL = bytes([0])
 | |
| 
 | |
| SE  = bytes([240])  # Subnegotiation End
 | |
| NOP = bytes([241])  # No Operation
 | |
| DM  = bytes([242])  # Data Mark
 | |
| BRK = bytes([243])  # Break
 | |
| IP  = bytes([244])  # Interrupt process
 | |
| AO  = bytes([245])  # Abort output
 | |
| AYT = bytes([246])  # Are You There
 | |
| EC  = bytes([247])  # Erase Character
 | |
| EL  = bytes([248])  # Erase Line
 | |
| GA  = bytes([249])  # Go Ahead
 | |
| SB =  bytes([250])  # Subnegotiation Begin
 | |
| 
 | |
| 
 | |
| # Telnet protocol options code (don't change)
 | |
| # These ones all come from arpa/telnet.h
 | |
| BINARY = bytes([0]) # 8-bit data path
 | |
| ECHO = bytes([1]) # echo
 | |
| RCP = bytes([2]) # prepare to reconnect
 | |
| SGA = bytes([3]) # suppress go ahead
 | |
| NAMS = bytes([4]) # approximate message size
 | |
| STATUS = bytes([5]) # give status
 | |
| TM = bytes([6]) # timing mark
 | |
| RCTE = bytes([7]) # remote controlled transmission and echo
 | |
| NAOL = bytes([8]) # negotiate about output line width
 | |
| NAOP = bytes([9]) # negotiate about output page size
 | |
| NAOCRD = bytes([10]) # negotiate about CR disposition
 | |
| NAOHTS = bytes([11]) # negotiate about horizontal tabstops
 | |
| NAOHTD = bytes([12]) # negotiate about horizontal tab disposition
 | |
| NAOFFD = bytes([13]) # negotiate about formfeed disposition
 | |
| NAOVTS = bytes([14]) # negotiate about vertical tab stops
 | |
| NAOVTD = bytes([15]) # negotiate about vertical tab disposition
 | |
| NAOLFD = bytes([16]) # negotiate about output LF disposition
 | |
| XASCII = bytes([17]) # extended ascii character set
 | |
| LOGOUT = bytes([18]) # force logout
 | |
| BM = bytes([19]) # byte macro
 | |
| DET = bytes([20]) # data entry terminal
 | |
| SUPDUP = bytes([21]) # supdup protocol
 | |
| SUPDUPOUTPUT = bytes([22]) # supdup output
 | |
| SNDLOC = bytes([23]) # send location
 | |
| TTYPE = bytes([24]) # terminal type
 | |
| EOR = bytes([25]) # end or record
 | |
| TUID = bytes([26]) # TACACS user identification
 | |
| OUTMRK = bytes([27]) # output marking
 | |
| TTYLOC = bytes([28]) # terminal location number
 | |
| VT3270REGIME = bytes([29]) # 3270 regime
 | |
| X3PAD = bytes([30]) # X.3 PAD
 | |
| NAWS = bytes([31]) # window size
 | |
| TSPEED = bytes([32]) # terminal speed
 | |
| LFLOW = bytes([33]) # remote flow control
 | |
| LINEMODE = bytes([34]) # Linemode option
 | |
| XDISPLOC = bytes([35]) # X Display Location
 | |
| OLD_ENVIRON = bytes([36]) # Old - Environment variables
 | |
| AUTHENTICATION = bytes([37]) # Authenticate
 | |
| ENCRYPT = bytes([38]) # Encryption option
 | |
| NEW_ENVIRON = bytes([39]) # New - Environment variables
 | |
| # the following ones come from
 | |
| # http://www.iana.org/assignments/telnet-options
 | |
| # Unfortunately, that document does not assign identifiers
 | |
| # to all of them, so we are making them up
 | |
| TN3270E = bytes([40]) # TN3270E
 | |
| XAUTH = bytes([41]) # XAUTH
 | |
| CHARSET = bytes([42]) # CHARSET
 | |
| RSP = bytes([43]) # Telnet Remote Serial Port
 | |
| COM_PORT_OPTION = bytes([44]) # Com Port Control Option
 | |
| SUPPRESS_LOCAL_ECHO = bytes([45]) # Telnet Suppress Local Echo
 | |
| TLS = bytes([46]) # Telnet Start TLS
 | |
| KERMIT = bytes([47]) # KERMIT
 | |
| SEND_URL = bytes([48]) # SEND-URL
 | |
| FORWARD_X = bytes([49]) # FORWARD_X
 | |
| PRAGMA_LOGON = bytes([138]) # TELOPT PRAGMA LOGON
 | |
| SSPI_LOGON = bytes([139]) # TELOPT SSPI LOGON
 | |
| PRAGMA_HEARTBEAT = bytes([140]) # TELOPT PRAGMA HEARTBEAT
 | |
| EXOPL = bytes([255]) # Extended-Options-List
 | |
| NOOPT = bytes([0])
 | |
| 
 | |
| 
 | |
| # poll/select have the advantage of not requiring any extra file descriptor,
 | |
| # contrarily to epoll/kqueue (also, they require a single syscall).
 | |
| if hasattr(selectors, 'PollSelector'):
 | |
|     _TelnetSelector = selectors.PollSelector
 | |
| else:
 | |
|     _TelnetSelector = selectors.SelectSelector
 | |
| 
 | |
| 
 | |
| class Telnet:
 | |
| 
 | |
|     """Telnet interface class.
 | |
| 
 | |
|     An instance of this class represents a connection to a telnet
 | |
|     server.  The instance is initially not connected; the open()
 | |
|     method must be used to establish a connection.  Alternatively, the
 | |
|     host name and optional port number can be passed to the
 | |
|     constructor, too.
 | |
| 
 | |
|     Don't try to reopen an already connected instance.
 | |
| 
 | |
|     This class has many read_*() methods.  Note that some of them
 | |
|     raise EOFError when the end of the connection is read, because
 | |
|     they can return an empty string for other reasons.  See the
 | |
|     individual doc strings.
 | |
| 
 | |
|     read_until(expected, [timeout])
 | |
|         Read until the expected string has been seen, or a timeout is
 | |
|         hit (default is no timeout); may block.
 | |
| 
 | |
|     read_all()
 | |
|         Read all data until EOF; may block.
 | |
| 
 | |
|     read_some()
 | |
|         Read at least one byte or EOF; may block.
 | |
| 
 | |
|     read_very_eager()
 | |
|         Read all data available already queued or on the socket,
 | |
|         without blocking.
 | |
| 
 | |
|     read_eager()
 | |
|         Read either data already queued or some data available on the
 | |
|         socket, without blocking.
 | |
| 
 | |
|     read_lazy()
 | |
|         Read all data in the raw queue (processing it first), without
 | |
|         doing any socket I/O.
 | |
| 
 | |
|     read_very_lazy()
 | |
|         Reads all data in the cooked queue, without doing any socket
 | |
|         I/O.
 | |
| 
 | |
|     read_sb_data()
 | |
|         Reads available data between SB ... SE sequence. Don't block.
 | |
| 
 | |
|     set_option_negotiation_callback(callback)
 | |
|         Each time a telnet option is read on the input flow, this callback
 | |
|         (if set) is called with the following parameters :
 | |
|         callback(telnet socket, command, option)
 | |
|             option will be chr(0) when there is no option.
 | |
|         No other action is done afterwards by telnetlib.
 | |
| 
 | |
|     """
 | |
| 
 | |
|     def __init__(self, host=None, port=0,
 | |
|                  timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
 | |
|         """Constructor.
 | |
| 
 | |
|         When called without arguments, create an unconnected instance.
 | |
|         With a hostname argument, it connects the instance; port number
 | |
|         and timeout are optional.
 | |
|         """
 | |
|         self.debuglevel = DEBUGLEVEL
 | |
|         self.host = host
 | |
|         self.port = port
 | |
|         self.timeout = timeout
 | |
|         self.sock = None
 | |
|         self.rawq = b''
 | |
|         self.irawq = 0
 | |
|         self.cookedq = b''
 | |
|         self.eof = 0
 | |
|         self.iacseq = b'' # Buffer for IAC sequence.
 | |
|         self.sb = 0 # flag for SB and SE sequence.
 | |
|         self.sbdataq = b''
 | |
|         self.option_callback = None
 | |
|         if host is not None:
 | |
|             self.open(host, port, timeout)
 | |
| 
 | |
|     def open(self, host, port=0, timeout=socket._GLOBAL_DEFAULT_TIMEOUT):
 | |
|         """Connect to a host.
 | |
| 
 | |
|         The optional second argument is the port number, which
 | |
|         defaults to the standard telnet port (23).
 | |
| 
 | |
|         Don't try to reopen an already connected instance.
 | |
|         """
 | |
|         self.eof = 0
 | |
|         if not port:
 | |
|             port = TELNET_PORT
 | |
|         self.host = host
 | |
|         self.port = port
 | |
|         self.timeout = timeout
 | |
|         self.sock = socket.create_connection((host, port), timeout)
 | |
| 
 | |
|     def __del__(self):
 | |
|         """Destructor -- close the connection."""
 | |
|         self.close()
 | |
| 
 | |
|     def msg(self, msg, *args):
 | |
|         """Print a debug message, when the debug level is > 0.
 | |
| 
 | |
|         If extra arguments are present, they are substituted in the
 | |
|         message using the standard string formatting operator.
 | |
| 
 | |
|         """
 | |
|         if self.debuglevel > 0:
 | |
|             print('Telnet(%s,%s):' % (self.host, self.port), end=' ')
 | |
|             if args:
 | |
|                 print(msg % args)
 | |
|             else:
 | |
|                 print(msg)
 | |
| 
 | |
|     def set_debuglevel(self, debuglevel):
 | |
|         """Set the debug level.
 | |
| 
 | |
|         The higher it is, the more debug output you get (on sys.stdout).
 | |
| 
 | |
|         """
 | |
|         self.debuglevel = debuglevel
 | |
| 
 | |
|     def close(self):
 | |
|         """Close the connection."""
 | |
|         sock = self.sock
 | |
|         self.sock = None
 | |
|         self.eof = True
 | |
|         self.iacseq = b''
 | |
|         self.sb = 0
 | |
|         if sock:
 | |
|             sock.close()
 | |
| 
 | |
|     def get_socket(self):
 | |
|         """Return the socket object used internally."""
 | |
|         return self.sock
 | |
| 
 | |
|     def fileno(self):
 | |
|         """Return the fileno() of the socket object used internally."""
 | |
|         return self.sock.fileno()
 | |
| 
 | |
|     def write(self, buffer):
 | |
|         """Write a string to the socket, doubling any IAC characters.
 | |
| 
 | |
|         Can block if the connection is blocked.  May raise
 | |
|         OSError if the connection is closed.
 | |
| 
 | |
|         """
 | |
|         if IAC in buffer:
 | |
|             buffer = buffer.replace(IAC, IAC+IAC)
 | |
|         self.msg("send %r", buffer)
 | |
|         self.sock.sendall(buffer)
 | |
| 
 | |
|     def read_until(self, match, timeout=None):
 | |
|         """Read until a given string is encountered or until timeout.
 | |
| 
 | |
|         When no match is found, return whatever is available instead,
 | |
|         possibly the empty string.  Raise EOFError if the connection
 | |
|         is closed and no cooked data is available.
 | |
| 
 | |
|         """
 | |
|         n = len(match)
 | |
|         self.process_rawq()
 | |
|         i = self.cookedq.find(match)
 | |
|         if i >= 0:
 | |
|             i = i+n
 | |
|             buf = self.cookedq[:i]
 | |
|             self.cookedq = self.cookedq[i:]
 | |
|             return buf
 | |
|         if timeout is not None:
 | |
|             deadline = _time() + timeout
 | |
|         with _TelnetSelector() as selector:
 | |
|             selector.register(self, selectors.EVENT_READ)
 | |
|             while not self.eof:
 | |
|                 if selector.select(timeout):
 | |
|                     i = max(0, len(self.cookedq)-n)
 | |
|                     self.fill_rawq()
 | |
|                     self.process_rawq()
 | |
|                     i = self.cookedq.find(match, i)
 | |
|                     if i >= 0:
 | |
|                         i = i+n
 | |
|                         buf = self.cookedq[:i]
 | |
|                         self.cookedq = self.cookedq[i:]
 | |
|                         return buf
 | |
|                 if timeout is not None:
 | |
|                     timeout = deadline - _time()
 | |
|                     if timeout < 0:
 | |
|                         break
 | |
|         return self.read_very_lazy()
 | |
| 
 | |
|     def read_all(self):
 | |
|         """Read all data until EOF; block until connection closed."""
 | |
|         self.process_rawq()
 | |
|         while not self.eof:
 | |
|             self.fill_rawq()
 | |
|             self.process_rawq()
 | |
|         buf = self.cookedq
 | |
|         self.cookedq = b''
 | |
|         return buf
 | |
| 
 | |
|     def read_some(self):
 | |
|         """Read at least one byte of cooked data unless EOF is hit.
 | |
| 
 | |
|         Return b'' if EOF is hit.  Block if no data is immediately
 | |
|         available.
 | |
| 
 | |
|         """
 | |
|         self.process_rawq()
 | |
|         while not self.cookedq and not self.eof:
 | |
|             self.fill_rawq()
 | |
|             self.process_rawq()
 | |
|         buf = self.cookedq
 | |
|         self.cookedq = b''
 | |
|         return buf
 | |
| 
 | |
|     def read_very_eager(self):
 | |
|         """Read everything that's possible without blocking in I/O (eager).
 | |
| 
 | |
|         Raise EOFError if connection closed and no cooked data
 | |
|         available.  Return b'' if no cooked data available otherwise.
 | |
|         Don't block unless in the midst of an IAC sequence.
 | |
| 
 | |
|         """
 | |
|         self.process_rawq()
 | |
|         while not self.eof and self.sock_avail():
 | |
|             self.fill_rawq()
 | |
|             self.process_rawq()
 | |
|         return self.read_very_lazy()
 | |
| 
 | |
|     def read_eager(self):
 | |
|         """Read readily available data.
 | |
| 
 | |
|         Raise EOFError if connection closed and no cooked data
 | |
|         available.  Return b'' if no cooked data available otherwise.
 | |
|         Don't block unless in the midst of an IAC sequence.
 | |
| 
 | |
|         """
 | |
|         self.process_rawq()
 | |
|         while not self.cookedq and not self.eof and self.sock_avail():
 | |
|             self.fill_rawq()
 | |
|             self.process_rawq()
 | |
|         return self.read_very_lazy()
 | |
| 
 | |
|     def read_lazy(self):
 | |
|         """Process and return data that's already in the queues (lazy).
 | |
| 
 | |
|         Raise EOFError if connection closed and no data available.
 | |
|         Return b'' if no cooked data available otherwise.  Don't block
 | |
|         unless in the midst of an IAC sequence.
 | |
| 
 | |
|         """
 | |
|         self.process_rawq()
 | |
|         return self.read_very_lazy()
 | |
| 
 | |
|     def read_very_lazy(self):
 | |
|         """Return any data available in the cooked queue (very lazy).
 | |
| 
 | |
|         Raise EOFError if connection closed and no data available.
 | |
|         Return b'' if no cooked data available otherwise.  Don't block.
 | |
| 
 | |
|         """
 | |
|         buf = self.cookedq
 | |
|         self.cookedq = b''
 | |
|         if not buf and self.eof and not self.rawq:
 | |
|             raise EOFError('telnet connection closed')
 | |
|         return buf
 | |
| 
 | |
|     def read_sb_data(self):
 | |
|         """Return any data available in the SB ... SE queue.
 | |
| 
 | |
|         Return b'' if no SB ... SE available. Should only be called
 | |
|         after seeing a SB or SE command. When a new SB command is
 | |
|         found, old unread SB data will be discarded. Don't block.
 | |
| 
 | |
|         """
 | |
|         buf = self.sbdataq
 | |
|         self.sbdataq = b''
 | |
|         return buf
 | |
| 
 | |
|     def set_option_negotiation_callback(self, callback):
 | |
|         """Provide a callback function called after each receipt of a telnet option."""
 | |
|         self.option_callback = callback
 | |
| 
 | |
|     def process_rawq(self):
 | |
|         """Transfer from raw queue to cooked queue.
 | |
| 
 | |
|         Set self.eof when connection is closed.  Don't block unless in
 | |
|         the midst of an IAC sequence.
 | |
| 
 | |
|         """
 | |
|         buf = [b'', b'']
 | |
|         try:
 | |
|             while self.rawq:
 | |
|                 c = self.rawq_getchar()
 | |
|                 if not self.iacseq:
 | |
|                     if c == theNULL:
 | |
|                         continue
 | |
|                     if c == b"\021":
 | |
|                         continue
 | |
|                     if c != IAC:
 | |
|                         buf[self.sb] = buf[self.sb] + c
 | |
|                         continue
 | |
|                     else:
 | |
|                         self.iacseq += c
 | |
|                 elif len(self.iacseq) == 1:
 | |
|                     # 'IAC: IAC CMD [OPTION only for WILL/WONT/DO/DONT]'
 | |
|                     if c in (DO, DONT, WILL, WONT):
 | |
|                         self.iacseq += c
 | |
|                         continue
 | |
| 
 | |
|                     self.iacseq = b''
 | |
|                     if c == IAC:
 | |
|                         buf[self.sb] = buf[self.sb] + c
 | |
|                     else:
 | |
|                         if c == SB: # SB ... SE start.
 | |
|                             self.sb = 1
 | |
|                             self.sbdataq = b''
 | |
|                         elif c == SE:
 | |
|                             self.sb = 0
 | |
|                             self.sbdataq = self.sbdataq + buf[1]
 | |
|                             buf[1] = b''
 | |
|                         if self.option_callback:
 | |
|                             # Callback is supposed to look into
 | |
|                             # the sbdataq
 | |
|                             self.option_callback(self.sock, c, NOOPT)
 | |
|                         else:
 | |
|                             # We can't offer automatic processing of
 | |
|                             # suboptions. Alas, we should not get any
 | |
|                             # unless we did a WILL/DO before.
 | |
|                             self.msg('IAC %d not recognized' % ord(c))
 | |
|                 elif len(self.iacseq) == 2:
 | |
|                     cmd = self.iacseq[1:2]
 | |
|                     self.iacseq = b''
 | |
|                     opt = c
 | |
|                     if cmd in (DO, DONT):
 | |
|                         self.msg('IAC %s %d',
 | |
|                             cmd == DO and 'DO' or 'DONT', ord(opt))
 | |
|                         if self.option_callback:
 | |
|                             self.option_callback(self.sock, cmd, opt)
 | |
|                         else:
 | |
|                             self.sock.sendall(IAC + WONT + opt)
 | |
|                     elif cmd in (WILL, WONT):
 | |
|                         self.msg('IAC %s %d',
 | |
|                             cmd == WILL and 'WILL' or 'WONT', ord(opt))
 | |
|                         if self.option_callback:
 | |
|                             self.option_callback(self.sock, cmd, opt)
 | |
|                         else:
 | |
|                             self.sock.sendall(IAC + DONT + opt)
 | |
|         except EOFError: # raised by self.rawq_getchar()
 | |
|             self.iacseq = b'' # Reset on EOF
 | |
|             self.sb = 0
 | |
|             pass
 | |
|         self.cookedq = self.cookedq + buf[0]
 | |
|         self.sbdataq = self.sbdataq + buf[1]
 | |
| 
 | |
|     def rawq_getchar(self):
 | |
|         """Get next char from raw queue.
 | |
| 
 | |
|         Block if no data is immediately available.  Raise EOFError
 | |
|         when connection is closed.
 | |
| 
 | |
|         """
 | |
|         if not self.rawq:
 | |
|             self.fill_rawq()
 | |
|             if self.eof:
 | |
|                 raise EOFError
 | |
|         c = self.rawq[self.irawq:self.irawq+1]
 | |
|         self.irawq = self.irawq + 1
 | |
|         if self.irawq >= len(self.rawq):
 | |
|             self.rawq = b''
 | |
|             self.irawq = 0
 | |
|         return c
 | |
| 
 | |
|     def fill_rawq(self):
 | |
|         """Fill raw queue from exactly one recv() system call.
 | |
| 
 | |
|         Block if no data is immediately available.  Set self.eof when
 | |
|         connection is closed.
 | |
| 
 | |
|         """
 | |
|         if self.irawq >= len(self.rawq):
 | |
|             self.rawq = b''
 | |
|             self.irawq = 0
 | |
|         # The buffer size should be fairly small so as to avoid quadratic
 | |
|         # behavior in process_rawq() above
 | |
|         buf = self.sock.recv(50)
 | |
|         self.msg("recv %r", buf)
 | |
|         self.eof = (not buf)
 | |
|         self.rawq = self.rawq + buf
 | |
| 
 | |
|     def sock_avail(self):
 | |
|         """Test whether data is available on the socket."""
 | |
|         with _TelnetSelector() as selector:
 | |
|             selector.register(self, selectors.EVENT_READ)
 | |
|             return bool(selector.select(0))
 | |
| 
 | |
|     def interact(self):
 | |
|         """Interaction function, emulates a very dumb telnet client."""
 | |
|         if sys.platform == "win32":
 | |
|             self.mt_interact()
 | |
|             return
 | |
|         with _TelnetSelector() as selector:
 | |
|             selector.register(self, selectors.EVENT_READ)
 | |
|             selector.register(sys.stdin, selectors.EVENT_READ)
 | |
| 
 | |
|             while True:
 | |
|                 for key, events in selector.select():
 | |
|                     if key.fileobj is self:
 | |
|                         try:
 | |
|                             text = self.read_eager()
 | |
|                         except EOFError:
 | |
|                             print('*** Connection closed by remote host ***')
 | |
|                             return
 | |
|                         if text:
 | |
|                             sys.stdout.write(text.decode('ascii'))
 | |
|                             sys.stdout.flush()
 | |
|                     elif key.fileobj is sys.stdin:
 | |
|                         line = sys.stdin.readline().encode('ascii')
 | |
|                         if not line:
 | |
|                             return
 | |
|                         self.write(line)
 | |
| 
 | |
|     def mt_interact(self):
 | |
|         """Multithreaded version of interact()."""
 | |
|         import _thread
 | |
|         _thread.start_new_thread(self.listener, ())
 | |
|         while 1:
 | |
|             line = sys.stdin.readline()
 | |
|             if not line:
 | |
|                 break
 | |
|             self.write(line.encode('ascii'))
 | |
| 
 | |
|     def listener(self):
 | |
|         """Helper for mt_interact() -- this executes in the other thread."""
 | |
|         while 1:
 | |
|             try:
 | |
|                 data = self.read_eager()
 | |
|             except EOFError:
 | |
|                 print('*** Connection closed by remote host ***')
 | |
|                 return
 | |
|             if data:
 | |
|                 sys.stdout.write(data.decode('ascii'))
 | |
|             else:
 | |
|                 sys.stdout.flush()
 | |
| 
 | |
|     def expect(self, list, timeout=None):
 | |
|         """Read until one from a list of a regular expressions matches.
 | |
| 
 | |
|         The first argument is a list of regular expressions, either
 | |
|         compiled (re.RegexObject instances) or uncompiled (strings).
 | |
|         The optional second argument is a timeout, in seconds; default
 | |
|         is no timeout.
 | |
| 
 | |
|         Return a tuple of three items: the index in the list of the
 | |
|         first regular expression that matches; the match object
 | |
|         returned; and the text read up till and including the match.
 | |
| 
 | |
|         If EOF is read and no text was read, raise EOFError.
 | |
|         Otherwise, when nothing matches, return (-1, None, text) where
 | |
|         text is the text received so far (may be the empty string if a
 | |
|         timeout happened).
 | |
| 
 | |
|         If a regular expression ends with a greedy match (e.g. '.*')
 | |
|         or if more than one expression can match the same input, the
 | |
|         results are undeterministic, and may depend on the I/O timing.
 | |
| 
 | |
|         """
 | |
|         re = None
 | |
|         list = list[:]
 | |
|         indices = range(len(list))
 | |
|         for i in indices:
 | |
|             if not hasattr(list[i], "search"):
 | |
|                 if not re: import re
 | |
|                 list[i] = re.compile(list[i])
 | |
|         if timeout is not None:
 | |
|             deadline = _time() + timeout
 | |
|         with _TelnetSelector() as selector:
 | |
|             selector.register(self, selectors.EVENT_READ)
 | |
|             while not self.eof:
 | |
|                 self.process_rawq()
 | |
|                 for i in indices:
 | |
|                     m = list[i].search(self.cookedq)
 | |
|                     if m:
 | |
|                         e = m.end()
 | |
|                         text = self.cookedq[:e]
 | |
|                         self.cookedq = self.cookedq[e:]
 | |
|                         return (i, m, text)
 | |
|                 if timeout is not None:
 | |
|                     ready = selector.select(timeout)
 | |
|                     timeout = deadline - _time()
 | |
|                     if not ready:
 | |
|                         if timeout < 0:
 | |
|                             break
 | |
|                         else:
 | |
|                             continue
 | |
|                 self.fill_rawq()
 | |
|         text = self.read_very_lazy()
 | |
|         if not text and self.eof:
 | |
|             raise EOFError
 | |
|         return (-1, None, text)
 | |
| 
 | |
| 
 | |
| def test():
 | |
|     """Test program for telnetlib.
 | |
| 
 | |
|     Usage: python telnetlib.py [-d] ... [host [port]]
 | |
| 
 | |
|     Default host is localhost; default port is 23.
 | |
| 
 | |
|     """
 | |
|     debuglevel = 0
 | |
|     while sys.argv[1:] and sys.argv[1] == '-d':
 | |
|         debuglevel = debuglevel+1
 | |
|         del sys.argv[1]
 | |
|     host = 'localhost'
 | |
|     if sys.argv[1:]:
 | |
|         host = sys.argv[1]
 | |
|     port = 0
 | |
|     if sys.argv[2:]:
 | |
|         portstr = sys.argv[2]
 | |
|         try:
 | |
|             port = int(portstr)
 | |
|         except ValueError:
 | |
|             port = socket.getservbyname(portstr, 'tcp')
 | |
|     tn = Telnet()
 | |
|     tn.set_debuglevel(debuglevel)
 | |
|     tn.open(host, port, timeout=0.5)
 | |
|     tn.interact()
 | |
|     tn.close()
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     test()
 |