blob: 754242eced15772b5054db049cf6c687e4041706 [file] [log] [blame]
cliechti8099bed2009-08-01 23:59:18 +00001#! python
2#
3# Python Serial Port Extension for Win32, Linux, BSD, Jython
4# see __init__.py
5#
6# This module implements a RFC2217 compatible client. RF2217 descibes a
7# protocol to access serial ports over TCP/IP and allows setting the baud rate,
8# modem control lines etc.
9#
Chris Liechti01587b12015-08-05 02:39:32 +020010# (C) 2001-2015 Chris Liechti <cliechti@gmx.net>
Chris Liechtifbdd8a02015-08-09 02:37:45 +020011#
12# SPDX-License-Identifier: BSD-3-Clause
cliechti8099bed2009-08-01 23:59:18 +000013
14# TODO:
15# - setting control line -> answer is not checked (had problems with one of the
16# severs). consider implementing a compatibility mode flag to make check
17# conditional
18# - write timeout not implemented at all
cliechti8099bed2009-08-01 23:59:18 +000019
cliechti81c54762009-08-03 23:53:27 +000020##############################################################################
21# observations and issues with servers
22#=============================================================================
23# sredird V2.2.1
24# - http://www.ibiblio.org/pub/Linux/system/serial/ sredird-2.2.2.tar.gz
25# - does not acknowledge SET_CONTROL (RTS/DTR) correctly, always responding
26# [105 1] instead of the actual value.
27# - SET_BAUDRATE answer contains 4 extra null bytes -> probably for larger
28# numbers than 2**32?
29# - To get the signature [COM_PORT_OPTION 0] has to be sent.
30# - run a server: while true; do nc -l -p 7000 -c "sredird debug /dev/ttyUSB0 /var/lock/sredir"; done
31#=============================================================================
32# telnetcpcd (untested)
33# - http://ftp.wayne.edu/kermit/sredird/telnetcpcd-1.09.tar.gz
34# - To get the signature [COM_PORT_OPTION] w/o data has to be sent.
35#=============================================================================
36# ser2net
37# - does not negotiate BINARY or COM_PORT_OPTION for his side but at least
38# acknowledges that the client activates these options
39# - The configuration may be that the server prints a banner. As this client
40# implementation does a flushInput on connect, this banner is hidden from
41# the user application.
42# - NOTIFY_MODEMSTATE: the poll interval of the server seems to be one
43# second.
44# - To get the signature [COM_PORT_OPTION 0] has to be sent.
45# - run a server: run ser2net daemon, in /etc/ser2net.conf:
46# 2000:telnet:0:/dev/ttyS0:9600 remctl banner
47##############################################################################
48
49# How to identify ports? pySerial might want to support other protocols in the
50# future, so lets use an URL scheme.
51# for RFC2217 compliant servers we will use this:
Chris Liechti142ae562015-08-23 01:11:06 +020052# rfc2217://<host>:<port>[?option[&option...]]
cliechti81c54762009-08-03 23:53:27 +000053#
54# options:
Chris Liechti01587b12015-08-05 02:39:32 +020055# - "logging" set log level print diagnostic messages (e.g. "logging=debug")
cliechti81c54762009-08-03 23:53:27 +000056# - "ign_set_control": do not look at the answers to SET_CONTROL
cliechti7cb78e82009-08-05 15:47:57 +000057# - "poll_modem": issue NOTIFY_MODEMSTATE requests when CTS/DTR/RI/CD is read.
58# Without this option it expects that the server sends notifications
59# automatically on change (which most servers do and is according to the
60# RFC).
cliechti81c54762009-08-03 23:53:27 +000061# the order of the options is not relevant
62
cliechti5cc3eb12009-08-11 23:04:30 +000063import logging
Chris Liechtic4bca9e2015-08-07 14:40:41 +020064import socket
65import struct
66import threading
67import time
68try:
69 import urlparse
70except ImportError:
71 import urllib.parse as urlparse
Chris Liechtid2146002015-08-04 16:57:16 +020072try:
73 import Queue
74except ImportError:
75 import queue as Queue
76
Chris Liechtib4cda3a2015-08-08 17:12:08 +020077from serial.serialutil import *
78
cliechti8099bed2009-08-01 23:59:18 +000079# port string is expected to be something like this:
80# rfc2217://host:port
81# host may be an IP or including domain, whatever.
82# port is 0...65535
83
cliechti86844e82009-08-12 00:05:33 +000084# map log level names to constants. used in fromURL()
cliechti5cc3eb12009-08-11 23:04:30 +000085LOGGER_LEVELS = {
Chris Liechtic4bca9e2015-08-07 14:40:41 +020086 'debug': logging.DEBUG,
87 'info': logging.INFO,
88 'warning': logging.WARNING,
89 'error': logging.ERROR,
90 }
cliechti5cc3eb12009-08-11 23:04:30 +000091
92
cliechti8099bed2009-08-01 23:59:18 +000093# telnet protocol characters
Chris Liechtib4cda3a2015-08-08 17:12:08 +020094SE = b'\xf0' # Subnegotiation End
95NOP = b'\xf1' # No Operation
96DM = b'\xf2' # Data Mark
97BRK = b'\xf3' # Break
98IP = b'\xf4' # Interrupt process
99AO = b'\xf5' # Abort output
100AYT = b'\xf6' # Are You There
101EC = b'\xf7' # Erase Character
102EL = b'\xf8' # Erase Line
103GA = b'\xf9' # Go Ahead
104SB = b'\xfa' # Subnegotiation Begin
105WILL = b'\xfb'
106WONT = b'\xfc'
107DO = b'\xfd'
108DONT = b'\xfe'
109IAC = b'\xff' # Interpret As Command
110IAC_DOUBLED = b'\xff\xff'
cliechti8099bed2009-08-01 23:59:18 +0000111
112# selected telnet options
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200113BINARY = b'\x00' # 8-bit data path
114ECHO = b'\x01' # echo
115SGA = b'\x03' # suppress go ahead
cliechti8099bed2009-08-01 23:59:18 +0000116
117# RFC2217
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200118COM_PORT_OPTION = b'\x2c'
cliechti8099bed2009-08-01 23:59:18 +0000119
120# Client to Access Server
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200121SET_BAUDRATE = b'\x01'
122SET_DATASIZE = b'\x02'
123SET_PARITY = b'\x03'
124SET_STOPSIZE = b'\x04'
125SET_CONTROL = b'\x05'
126NOTIFY_LINESTATE = b'\x06'
127NOTIFY_MODEMSTATE = b'\x07'
128FLOWCONTROL_SUSPEND = b'\x08'
129FLOWCONTROL_RESUME = b'\x09'
130SET_LINESTATE_MASK = b'\x0a'
131SET_MODEMSTATE_MASK = b'\x0b'
132PURGE_DATA = b'\x0c'
cliechti8099bed2009-08-01 23:59:18 +0000133
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200134SERVER_SET_BAUDRATE = b'\x65'
135SERVER_SET_DATASIZE = b'\x66'
136SERVER_SET_PARITY = b'\x67'
137SERVER_SET_STOPSIZE = b'\x68'
138SERVER_SET_CONTROL = b'\x69'
139SERVER_NOTIFY_LINESTATE = b'\x6a'
140SERVER_NOTIFY_MODEMSTATE = b'\x6b'
141SERVER_FLOWCONTROL_SUSPEND = b'\x6c'
142SERVER_FLOWCONTROL_RESUME = b'\x6d'
143SERVER_SET_LINESTATE_MASK = b'\x6e'
144SERVER_SET_MODEMSTATE_MASK = b'\x6f'
145SERVER_PURGE_DATA = b'\x70'
cliechti8099bed2009-08-01 23:59:18 +0000146
147RFC2217_ANSWER_MAP = {
148 SET_BAUDRATE: SERVER_SET_BAUDRATE,
149 SET_DATASIZE: SERVER_SET_DATASIZE,
150 SET_PARITY: SERVER_SET_PARITY,
151 SET_STOPSIZE: SERVER_SET_STOPSIZE,
152 SET_CONTROL: SERVER_SET_CONTROL,
153 NOTIFY_LINESTATE: SERVER_NOTIFY_LINESTATE,
154 NOTIFY_MODEMSTATE: SERVER_NOTIFY_MODEMSTATE,
155 FLOWCONTROL_SUSPEND: SERVER_FLOWCONTROL_SUSPEND,
156 FLOWCONTROL_RESUME: SERVER_FLOWCONTROL_RESUME,
157 SET_LINESTATE_MASK: SERVER_SET_LINESTATE_MASK,
158 SET_MODEMSTATE_MASK: SERVER_SET_MODEMSTATE_MASK,
159 PURGE_DATA: SERVER_PURGE_DATA,
160}
161
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200162SET_CONTROL_REQ_FLOW_SETTING = b'\x00' # Request Com Port Flow Control Setting (outbound/both)
163SET_CONTROL_USE_NO_FLOW_CONTROL = b'\x01' # Use No Flow Control (outbound/both)
164SET_CONTROL_USE_SW_FLOW_CONTROL = b'\x02' # Use XON/XOFF Flow Control (outbound/both)
165SET_CONTROL_USE_HW_FLOW_CONTROL = b'\x03' # Use HARDWARE Flow Control (outbound/both)
166SET_CONTROL_REQ_BREAK_STATE = b'\x04' # Request BREAK State
167SET_CONTROL_BREAK_ON = b'\x05' # Set BREAK State ON
168SET_CONTROL_BREAK_OFF = b'\x06' # Set BREAK State OFF
169SET_CONTROL_REQ_DTR = b'\x07' # Request DTR Signal State
170SET_CONTROL_DTR_ON = b'\x08' # Set DTR Signal State ON
171SET_CONTROL_DTR_OFF = b'\x09' # Set DTR Signal State OFF
172SET_CONTROL_REQ_RTS = b'\x0a' # Request RTS Signal State
173SET_CONTROL_RTS_ON = b'\x0b' # Set RTS Signal State ON
174SET_CONTROL_RTS_OFF = b'\x0c' # Set RTS Signal State OFF
175SET_CONTROL_REQ_FLOW_SETTING_IN = b'\x0d' # Request Com Port Flow Control Setting (inbound)
176SET_CONTROL_USE_NO_FLOW_CONTROL_IN = b'\x0e' # Use No Flow Control (inbound)
177SET_CONTROL_USE_SW_FLOW_CONTOL_IN = b'\x0f' # Use XON/XOFF Flow Control (inbound)
178SET_CONTROL_USE_HW_FLOW_CONTOL_IN = b'\x10' # Use HARDWARE Flow Control (inbound)
179SET_CONTROL_USE_DCD_FLOW_CONTROL = b'\x11' # Use DCD Flow Control (outbound/both)
180SET_CONTROL_USE_DTR_FLOW_CONTROL = b'\x12' # Use DTR Flow Control (inbound)
181SET_CONTROL_USE_DSR_FLOW_CONTROL = b'\x13' # Use DSR Flow Control (outbound/both)
cliechti8099bed2009-08-01 23:59:18 +0000182
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200183LINESTATE_MASK_TIMEOUT = 128 # Time-out Error
184LINESTATE_MASK_SHIFTREG_EMPTY = 64 # Transfer Shift Register Empty
185LINESTATE_MASK_TRANSREG_EMPTY = 32 # Transfer Holding Register Empty
186LINESTATE_MASK_BREAK_DETECT = 16 # Break-detect Error
187LINESTATE_MASK_FRAMING_ERROR = 8 # Framing Error
188LINESTATE_MASK_PARTIY_ERROR = 4 # Parity Error
189LINESTATE_MASK_OVERRUN_ERROR = 2 # Overrun Error
190LINESTATE_MASK_DATA_READY = 1 # Data Ready
cliechti8099bed2009-08-01 23:59:18 +0000191
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200192MODEMSTATE_MASK_CD = 128 # Receive Line Signal Detect (also known as Carrier Detect)
193MODEMSTATE_MASK_RI = 64 # Ring Indicator
194MODEMSTATE_MASK_DSR = 32 # Data-Set-Ready Signal State
195MODEMSTATE_MASK_CTS = 16 # Clear-To-Send Signal State
196MODEMSTATE_MASK_CD_CHANGE = 8 # Delta Receive Line Signal Detect
197MODEMSTATE_MASK_RI_CHANGE = 4 # Trailing-edge Ring Detector
198MODEMSTATE_MASK_DSR_CHANGE = 2 # Delta Data-Set-Ready
199MODEMSTATE_MASK_CTS_CHANGE = 1 # Delta Clear-To-Send
cliechti8099bed2009-08-01 23:59:18 +0000200
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200201PURGE_RECEIVE_BUFFER = b'\x01' # Purge access server receive data buffer
202PURGE_TRANSMIT_BUFFER = b'\x02' # Purge access server transmit data buffer
203PURGE_BOTH_BUFFERS = b'\x03' # Purge both the access server receive data buffer and the access server transmit data buffer
cliechti8099bed2009-08-01 23:59:18 +0000204
205
206RFC2217_PARITY_MAP = {
207 PARITY_NONE: 1,
208 PARITY_ODD: 2,
209 PARITY_EVEN: 3,
210 PARITY_MARK: 4,
211 PARITY_SPACE: 5,
212}
cliechti130d1f02009-08-04 02:10:58 +0000213RFC2217_REVERSE_PARITY_MAP = dict((v,k) for k,v in RFC2217_PARITY_MAP.items())
cliechti8099bed2009-08-01 23:59:18 +0000214
215RFC2217_STOPBIT_MAP = {
216 STOPBITS_ONE: 1,
217 STOPBITS_ONE_POINT_FIVE: 3,
218 STOPBITS_TWO: 2,
219}
cliechti130d1f02009-08-04 02:10:58 +0000220RFC2217_REVERSE_STOPBIT_MAP = dict((v,k) for k,v in RFC2217_STOPBIT_MAP.items())
cliechti8099bed2009-08-01 23:59:18 +0000221
cliechti130d1f02009-08-04 02:10:58 +0000222# Telnet filter states
223M_NORMAL = 0
224M_IAC_SEEN = 1
225M_NEGOTIATE = 2
cliechti8099bed2009-08-01 23:59:18 +0000226
cliechti130d1f02009-08-04 02:10:58 +0000227# TelnetOption and TelnetSubnegotiation states
cliechtiac205322009-08-02 20:40:21 +0000228REQUESTED = 'REQUESTED'
229ACTIVE = 'ACTIVE'
230INACTIVE = 'INACTIVE'
231REALLY_INACTIVE = 'REALLY_INACTIVE'
232
233class TelnetOption(object):
cliechti1ef7e3e2009-08-03 02:38:43 +0000234 """Manage a single telnet option, keeps track of DO/DONT WILL/WONT."""
235
cliechti86b593e2009-08-05 16:28:12 +0000236 def __init__(self, connection, name, option, send_yes, send_no, ack_yes, ack_no, initial_state, activation_callback=None):
cliechtieada4fd2013-07-31 16:26:07 +0000237 """\
238 Initialize option.
cliechti1ef7e3e2009-08-03 02:38:43 +0000239 :param connection: connection used to transmit answers
240 :param name: a readable name for debug outputs
241 :param send_yes: what to send when option is to be enabled.
242 :param send_no: what to send when option is to be disabled.
243 :param ack_yes: what to expect when remote agrees on option.
244 :param ack_no: what to expect when remote disagrees on option.
245 :param initial_state: options initialized with REQUESTED are tried to
246 be enabled on startup. use INACTIVE for all others.
247 """
cliechti2b929b72009-08-02 23:49:02 +0000248 self.connection = connection
cliechtiac205322009-08-02 20:40:21 +0000249 self.name = name
250 self.option = option
251 self.send_yes = send_yes
252 self.send_no = send_no
253 self.ack_yes = ack_yes
254 self.ack_no = ack_no
255 self.state = initial_state
256 self.active = False
cliechti86b593e2009-08-05 16:28:12 +0000257 self.activation_callback = activation_callback
cliechtiac205322009-08-02 20:40:21 +0000258
259 def __repr__(self):
cliechti1ef7e3e2009-08-03 02:38:43 +0000260 """String for debug outputs"""
cliechtiac205322009-08-02 20:40:21 +0000261 return "%s:%s(%s)" % (self.name, self.active, self.state)
262
cliechti2b929b72009-08-02 23:49:02 +0000263 def process_incoming(self, command):
cliechti7d448562014-08-03 21:57:45 +0000264 """\
265 A DO/DONT/WILL/WONT was received for this option, update state and
266 answer when needed.
267 """
cliechtiac205322009-08-02 20:40:21 +0000268 if command == self.ack_yes:
269 if self.state is REQUESTED:
270 self.state = ACTIVE
271 self.active = True
cliechti86b593e2009-08-05 16:28:12 +0000272 if self.activation_callback is not None:
273 self.activation_callback()
cliechtiac205322009-08-02 20:40:21 +0000274 elif self.state is ACTIVE:
275 pass
276 elif self.state is INACTIVE:
277 self.state = ACTIVE
cliechti1ef7e3e2009-08-03 02:38:43 +0000278 self.connection.telnetSendOption(self.send_yes, self.option)
cliechtiac205322009-08-02 20:40:21 +0000279 self.active = True
cliechti86b593e2009-08-05 16:28:12 +0000280 if self.activation_callback is not None:
281 self.activation_callback()
cliechtiac205322009-08-02 20:40:21 +0000282 elif self.state is REALLY_INACTIVE:
cliechti1ef7e3e2009-08-03 02:38:43 +0000283 self.connection.telnetSendOption(self.send_no, self.option)
cliechtiac205322009-08-02 20:40:21 +0000284 else:
285 raise ValueError('option in illegal state %r' % self)
286 elif command == self.ack_no:
287 if self.state is REQUESTED:
288 self.state = INACTIVE
289 self.active = False
290 elif self.state is ACTIVE:
291 self.state = INACTIVE
cliechti1ef7e3e2009-08-03 02:38:43 +0000292 self.connection.telnetSendOption(self.send_no, self.option)
cliechtiac205322009-08-02 20:40:21 +0000293 self.active = False
294 elif self.state is INACTIVE:
295 pass
296 elif self.state is REALLY_INACTIVE:
297 pass
298 else:
299 raise ValueError('option in illegal state %r' % self)
300
301
cliechti2b929b72009-08-02 23:49:02 +0000302class TelnetSubnegotiation(object):
cliechtieada4fd2013-07-31 16:26:07 +0000303 """\
304 A object to handle subnegotiation of options. In this case actually
305 sub-sub options for RFC 2217. It is used to track com port options.
306 """
cliechti2b929b72009-08-02 23:49:02 +0000307
308 def __init__(self, connection, name, option, ack_option=None):
309 if ack_option is None: ack_option = option
310 self.connection = connection
311 self.name = name
312 self.option = option
313 self.value = None
314 self.ack_option = ack_option
315 self.state = INACTIVE
316
317 def __repr__(self):
cliechti044d8662009-08-11 21:40:31 +0000318 """String for debug outputs."""
cliechti2b929b72009-08-02 23:49:02 +0000319 return "%s:%s" % (self.name, self.state)
320
321 def set(self, value):
cliechtieada4fd2013-07-31 16:26:07 +0000322 """\
cliechti7d448562014-08-03 21:57:45 +0000323 Request a change of the value. a request is sent to the server. if
cliechti2b929b72009-08-02 23:49:02 +0000324 the client needs to know if the change is performed he has to check the
cliechtieada4fd2013-07-31 16:26:07 +0000325 state of this object.
326 """
cliechti2b929b72009-08-02 23:49:02 +0000327 self.value = value
328 self.state = REQUESTED
cliechti1ef7e3e2009-08-03 02:38:43 +0000329 self.connection.rfc2217SendSubnegotiation(self.option, self.value)
cliechti6a300772009-08-12 02:28:56 +0000330 if self.connection.logger:
331 self.connection.logger.debug("SB Requesting %s -> %r" % (self.name, self.value))
cliechti2b929b72009-08-02 23:49:02 +0000332
333 def isReady(self):
cliechtieada4fd2013-07-31 16:26:07 +0000334 """\
cliechti7d448562014-08-03 21:57:45 +0000335 Check if answer from server has been received. when server rejects
cliechtieada4fd2013-07-31 16:26:07 +0000336 the change, raise a ValueError.
337 """
cliechti2b929b72009-08-02 23:49:02 +0000338 if self.state == REALLY_INACTIVE:
339 raise ValueError("remote rejected value for option %r" % (self.name))
340 return self.state == ACTIVE
cliechti1ef7e3e2009-08-03 02:38:43 +0000341 # add property to have a similar interface as TelnetOption
cliechti2b929b72009-08-02 23:49:02 +0000342 active = property(isReady)
343
cliechti044d8662009-08-11 21:40:31 +0000344 def wait(self, timeout=3):
cliechtieada4fd2013-07-31 16:26:07 +0000345 """\
cliechti7d448562014-08-03 21:57:45 +0000346 Wait until the subnegotiation has been acknowledged or timeout. It
cliechti1ef7e3e2009-08-03 02:38:43 +0000347 can also throw a value error when the answer from the server does not
cliechtieada4fd2013-07-31 16:26:07 +0000348 match the value sent.
349 """
cliechti044d8662009-08-11 21:40:31 +0000350 timeout_time = time.time() + timeout
cliechti2b929b72009-08-02 23:49:02 +0000351 while time.time() < timeout_time:
352 time.sleep(0.05) # prevent 100% CPU load
353 if self.isReady():
354 break
355 else:
356 raise SerialException("timeout while waiting for option %r" % (self.name))
357
358 def checkAnswer(self, suboption):
cliechtieada4fd2013-07-31 16:26:07 +0000359 """\
cliechti7d448562014-08-03 21:57:45 +0000360 Check an incoming subnegotiation block. The parameter already has
cliechtieada4fd2013-07-31 16:26:07 +0000361 cut off the header like sub option number and com port option value.
362 """
cliechti2b929b72009-08-02 23:49:02 +0000363 if self.value == suboption[:len(self.value)]:
364 self.state = ACTIVE
365 else:
366 # error propagation done in isReady
367 self.state = REALLY_INACTIVE
cliechti6a300772009-08-12 02:28:56 +0000368 if self.connection.logger:
369 self.connection.logger.debug("SB Answer %s -> %r -> %s" % (self.name, suboption, self.state))
cliechti2b929b72009-08-02 23:49:02 +0000370
371
Chris Liechtief6b7b42015-08-06 22:19:26 +0200372class Serial(SerialBase):
cliechti044d8662009-08-11 21:40:31 +0000373 """Serial port implementation for RFC 2217 remote serial ports."""
cliechti8099bed2009-08-01 23:59:18 +0000374
375 BAUDRATES = (50, 75, 110, 134, 150, 200, 300, 600, 1200, 1800, 2400, 4800,
376 9600, 19200, 38400, 57600, 115200)
377
378 def open(self):
cliechtieada4fd2013-07-31 16:26:07 +0000379 """\
380 Open port with current settings. This may throw a SerialException
381 if the port cannot be opened.
382 """
cliechti6a300772009-08-12 02:28:56 +0000383 self.logger = None
cliechti81c54762009-08-03 23:53:27 +0000384 self._ignore_set_control_answer = False
cliechti7cb78e82009-08-05 15:47:57 +0000385 self._poll_modem_state = False
cliechtidfe2d272009-08-10 22:19:41 +0000386 self._network_timeout = 3
cliechti8099bed2009-08-01 23:59:18 +0000387 if self._port is None:
388 raise SerialException("Port must be configured before it can be used.")
cliechti8f69e702011-03-19 00:22:32 +0000389 if self._isOpen:
390 raise SerialException("Port is already open.")
cliechti8099bed2009-08-01 23:59:18 +0000391 try:
Chris Liechtic4bca9e2015-08-07 14:40:41 +0200392 self._socket = socket.create_connection(self.fromURL(self.portstr))
cliechti6a300772009-08-12 02:28:56 +0000393 self._socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Chris Liechti68340d72015-08-03 14:15:48 +0200394 except Exception as msg:
cliechti8099bed2009-08-01 23:59:18 +0000395 self._socket = None
396 raise SerialException("Could not open port %s: %s" % (self.portstr, msg))
397
cliechti1ef7e3e2009-08-03 02:38:43 +0000398 self._socket.settimeout(5) # XXX good value?
cliechti8099bed2009-08-01 23:59:18 +0000399
cliechti1ef7e3e2009-08-03 02:38:43 +0000400 # use a thread save queue as buffer. it also simplifies implementing
401 # the read timeout
cliechti8099bed2009-08-01 23:59:18 +0000402 self._read_buffer = Queue.Queue()
cliechti81c54762009-08-03 23:53:27 +0000403 # to ensure that user writes does not interfere with internal
404 # telnet/rfc2217 options establish a lock
405 self._write_lock = threading.Lock()
cliechtiac205322009-08-02 20:40:21 +0000406 # name the following separately so that, below, a check can be easily done
407 mandadory_options = [
cliechti2b929b72009-08-02 23:49:02 +0000408 TelnetOption(self, 'we-BINARY', BINARY, WILL, WONT, DO, DONT, INACTIVE),
cliechti2b929b72009-08-02 23:49:02 +0000409 TelnetOption(self, 'we-RFC2217', COM_PORT_OPTION, WILL, WONT, DO, DONT, REQUESTED),
cliechtiac205322009-08-02 20:40:21 +0000410 ]
411 # all supported telnet options
412 self._telnet_options = [
cliechtia29275e2009-08-03 00:08:04 +0000413 TelnetOption(self, 'ECHO', ECHO, DO, DONT, WILL, WONT, REQUESTED),
cliechti2b929b72009-08-02 23:49:02 +0000414 TelnetOption(self, 'we-SGA', SGA, WILL, WONT, DO, DONT, REQUESTED),
415 TelnetOption(self, 'they-SGA', SGA, DO, DONT, WILL, WONT, REQUESTED),
cliechti81c54762009-08-03 23:53:27 +0000416 TelnetOption(self, 'they-BINARY', BINARY, DO, DONT, WILL, WONT, INACTIVE),
417 TelnetOption(self, 'they-RFC2217', COM_PORT_OPTION, DO, DONT, WILL, WONT, REQUESTED),
cliechtiac205322009-08-02 20:40:21 +0000418 ] + mandadory_options
cliechti044d8662009-08-11 21:40:31 +0000419 # RFC 2217 specific states
cliechti2b929b72009-08-02 23:49:02 +0000420 # COM port settings
421 self._rfc2217_port_settings = {
422 'baudrate': TelnetSubnegotiation(self, 'baudrate', SET_BAUDRATE, SERVER_SET_BAUDRATE),
423 'datasize': TelnetSubnegotiation(self, 'datasize', SET_DATASIZE, SERVER_SET_DATASIZE),
424 'parity': TelnetSubnegotiation(self, 'parity', SET_PARITY, SERVER_SET_PARITY),
425 'stopsize': TelnetSubnegotiation(self, 'stopsize', SET_STOPSIZE, SERVER_SET_STOPSIZE),
426 }
cliechticb20a4f2011-04-25 02:25:54 +0000427 # There are more subnegotiation objects, combine all in one dictionary
cliechti2b929b72009-08-02 23:49:02 +0000428 # for easy access
429 self._rfc2217_options = {
430 'purge': TelnetSubnegotiation(self, 'purge', PURGE_DATA, SERVER_PURGE_DATA),
cliechti81c54762009-08-03 23:53:27 +0000431 'control': TelnetSubnegotiation(self, 'control', SET_CONTROL, SERVER_SET_CONTROL),
cliechti2b929b72009-08-02 23:49:02 +0000432 }
433 self._rfc2217_options.update(self._rfc2217_port_settings)
434 # cache for line and modem states that the server sends to us
cliechti8099bed2009-08-01 23:59:18 +0000435 self._linestate = 0
cliechti7cb78e82009-08-05 15:47:57 +0000436 self._modemstate = None
437 self._modemstate_expires = 0
cliechti044d8662009-08-11 21:40:31 +0000438 # RFC 2217 flow control between server and client
cliechti672d0292009-08-03 02:01:57 +0000439 self._remote_suspend_flow = False
cliechti8099bed2009-08-01 23:59:18 +0000440
cliechti1ef7e3e2009-08-03 02:38:43 +0000441 self._thread = threading.Thread(target=self._telnetReadLoop)
cliechti8099bed2009-08-01 23:59:18 +0000442 self._thread.setDaemon(True)
cliechti5cc3eb12009-08-11 23:04:30 +0000443 self._thread.setName('pySerial RFC 2217 reader thread for %s' % (self._port,))
cliechti8099bed2009-08-01 23:59:18 +0000444 self._thread.start()
445
cliechti044d8662009-08-11 21:40:31 +0000446 # negotiate Telnet/RFC 2217 -> send initial requests
cliechtiac205322009-08-02 20:40:21 +0000447 for option in self._telnet_options:
448 if option.state is REQUESTED:
cliechti1ef7e3e2009-08-03 02:38:43 +0000449 self.telnetSendOption(option.send_yes, option.option)
cliechtiac205322009-08-02 20:40:21 +0000450 # now wait until important options are negotiated
cliechtidfe2d272009-08-10 22:19:41 +0000451 timeout_time = time.time() + self._network_timeout
cliechtiac205322009-08-02 20:40:21 +0000452 while time.time() < timeout_time:
cliechtiac205322009-08-02 20:40:21 +0000453 time.sleep(0.05) # prevent 100% CPU load
cliechti7c213a92014-07-31 15:29:34 +0000454 if sum(o.active for o in mandadory_options) == sum(o.state != INACTIVE for o in mandadory_options):
cliechti2b929b72009-08-02 23:49:02 +0000455 break
456 else:
457 raise SerialException("Remote does not seem to support RFC2217 or BINARY mode %r" % mandadory_options)
cliechti6a300772009-08-12 02:28:56 +0000458 if self.logger:
459 self.logger.info("Negotiated options: %s" % self._telnet_options)
cliechti8099bed2009-08-01 23:59:18 +0000460
cliechti044d8662009-08-11 21:40:31 +0000461 # fine, go on, set RFC 2271 specific things
cliechti8099bed2009-08-01 23:59:18 +0000462 self._reconfigurePort()
cliechti2b929b72009-08-02 23:49:02 +0000463 # all things set up get, now a clean start
cliechti8099bed2009-08-01 23:59:18 +0000464 self._isOpen = True
Chris Liechtief1fe252015-08-27 23:25:21 +0200465 if not self.dsrdtr:
466 self._update_dtr_state()
cliechti8099bed2009-08-01 23:59:18 +0000467 if not self._rtscts:
Chris Liechtief1fe252015-08-27 23:25:21 +0200468 self._update_rts_state()
469 self.reset_input_buffer()
470 self.reset_output_buffer()
cliechti8099bed2009-08-01 23:59:18 +0000471
472 def _reconfigurePort(self):
473 """Set communication parameters on opened port."""
474 if self._socket is None:
475 raise SerialException("Can only operate on open ports")
476
cliechti8099bed2009-08-01 23:59:18 +0000477 # if self._timeout != 0 and self._interCharTimeout is not None:
cliechti8099bed2009-08-01 23:59:18 +0000478 # XXX
479
480 if self._writeTimeout is not None:
481 raise NotImplementedError('writeTimeout is currently not supported')
cliechti2b929b72009-08-02 23:49:02 +0000482 # XXX
cliechti8099bed2009-08-01 23:59:18 +0000483
cliechti2b929b72009-08-02 23:49:02 +0000484 # Setup the connection
cliechti1ef7e3e2009-08-03 02:38:43 +0000485 # to get good performance, all parameter changes are sent first...
Chris Liechti01587b12015-08-05 02:39:32 +0200486 if not 0 < self._baudrate < 2**32:
cliechti81c54762009-08-03 23:53:27 +0000487 raise ValueError("invalid baudrate: %r" % (self._baudrate))
Chris Liechti01587b12015-08-05 02:39:32 +0200488 self._rfc2217_port_settings['baudrate'].set(struct.pack(b'!I', self._baudrate))
489 self._rfc2217_port_settings['datasize'].set(struct.pack(b'!B', self._bytesize))
490 self._rfc2217_port_settings['parity'].set(struct.pack(b'!B', RFC2217_PARITY_MAP[self._parity]))
491 self._rfc2217_port_settings['stopsize'].set(struct.pack(b'!B', RFC2217_STOPBIT_MAP[self._stopbits]))
cliechti8099bed2009-08-01 23:59:18 +0000492
cliechti2b929b72009-08-02 23:49:02 +0000493 # and now wait until parameters are active
494 items = self._rfc2217_port_settings.values()
cliechti6a300772009-08-12 02:28:56 +0000495 if self.logger:
496 self.logger.debug("Negotiating settings: %s" % (items,))
cliechtidfe2d272009-08-10 22:19:41 +0000497 timeout_time = time.time() + self._network_timeout
cliechti2b929b72009-08-02 23:49:02 +0000498 while time.time() < timeout_time:
499 time.sleep(0.05) # prevent 100% CPU load
500 if sum(o.active for o in items) == len(items):
501 break
502 else:
503 raise SerialException("Remote does not accept parameter change (RFC2217): %r" % items)
cliechti6a300772009-08-12 02:28:56 +0000504 if self.logger:
505 self.logger.info("Negotiated settings: %s" % (items,))
cliechti8099bed2009-08-01 23:59:18 +0000506
507 if self._rtscts and self._xonxoff:
cliechti1ef7e3e2009-08-03 02:38:43 +0000508 raise ValueError('xonxoff and rtscts together are not supported')
cliechti8099bed2009-08-01 23:59:18 +0000509 elif self._rtscts:
cliechti1ef7e3e2009-08-03 02:38:43 +0000510 self.rfc2217SetControl(SET_CONTROL_USE_HW_FLOW_CONTROL)
cliechti8099bed2009-08-01 23:59:18 +0000511 elif self._xonxoff:
cliechti1ef7e3e2009-08-03 02:38:43 +0000512 self.rfc2217SetControl(SET_CONTROL_USE_SW_FLOW_CONTROL)
cliechti8099bed2009-08-01 23:59:18 +0000513 else:
cliechti1ef7e3e2009-08-03 02:38:43 +0000514 self.rfc2217SetControl(SET_CONTROL_USE_NO_FLOW_CONTROL)
cliechti8099bed2009-08-01 23:59:18 +0000515
516 def close(self):
517 """Close port"""
518 if self._isOpen:
519 if self._socket:
520 try:
521 self._socket.shutdown(socket.SHUT_RDWR)
522 self._socket.close()
523 except:
524 # ignore errors.
525 pass
526 self._socket = None
527 if self._thread:
528 self._thread.join()
529 self._isOpen = False
530 # in case of quick reconnects, give the server some time
531 time.sleep(0.3)
532
533 def makeDeviceName(self, port):
534 raise SerialException("there is no sensible way to turn numbers into URLs")
535
536 def fromURL(self, url):
cliechti1ef7e3e2009-08-03 02:38:43 +0000537 """extract host and port from an URL string"""
Chris Liechtia4222112015-08-07 01:03:12 +0200538 parts = urlparse.urlsplit(url)
Chris Liechtid14b1ab2015-08-21 00:28:53 +0200539 if parts.scheme != "rfc2217":
540 raise SerialException('expected a string in the form "rfc2217://<host>:<port>[?option[&option...]]": not starting with rfc2217:// (%r)' % (parts.scheme,))
cliechti8099bed2009-08-01 23:59:18 +0000541 try:
Chris Liechtia4222112015-08-07 01:03:12 +0200542 # process options now, directly altering self
Chris Liechtid14b1ab2015-08-21 00:28:53 +0200543 for option, values in urlparse.parse_qs(parts.query, True).items():
544 if option == 'logging':
Chris Liechtia4222112015-08-07 01:03:12 +0200545 logging.basicConfig() # XXX is that good to call it here?
546 self.logger = logging.getLogger('pySerial.rfc2217')
Chris Liechtid14b1ab2015-08-21 00:28:53 +0200547 self.logger.setLevel(LOGGER_LEVELS[values[0]])
Chris Liechtia4222112015-08-07 01:03:12 +0200548 self.logger.debug('enabled logging')
549 elif option == 'ign_set_control':
550 self._ignore_set_control_answer = True
551 elif option == 'poll_modem':
552 self._poll_modem_state = True
553 elif option == 'timeout':
Chris Liechtid14b1ab2015-08-21 00:28:53 +0200554 self._network_timeout = float(values[0])
Chris Liechtia4222112015-08-07 01:03:12 +0200555 else:
556 raise ValueError('unknown option: %r' % (option,))
cliechti81c54762009-08-03 23:53:27 +0000557 # get host and port
Chris Liechtia4222112015-08-07 01:03:12 +0200558 host, port = parts.hostname, parts.port
cliechti8099bed2009-08-01 23:59:18 +0000559 if not 0 <= port < 65536: raise ValueError("port not in range 0...65535")
Chris Liechti68340d72015-08-03 14:15:48 +0200560 except ValueError as e:
Chris Liechtid14b1ab2015-08-21 00:28:53 +0200561 raise SerialException('expected a string in the form "rfc2217://<host>:<port>[?option[&option...]]": %s' % e)
cliechti8099bed2009-08-01 23:59:18 +0000562 return (host, port)
563
564 # - - - - - - - - - - - - - - - - - - - - - - - -
565
Chris Liechtief1fe252015-08-27 23:25:21 +0200566 @property
567 def in_waiting(self):
cliechti8099bed2009-08-01 23:59:18 +0000568 """Return the number of characters currently in the input buffer."""
569 if not self._isOpen: raise portNotOpenError
570 return self._read_buffer.qsize()
571
572 def read(self, size=1):
cliechtieada4fd2013-07-31 16:26:07 +0000573 """\
574 Read size bytes from the serial port. If a timeout is set it may
cliechti8099bed2009-08-01 23:59:18 +0000575 return less characters as requested. With no timeout it will block
cliechtieada4fd2013-07-31 16:26:07 +0000576 until the requested number of bytes is read.
577 """
cliechti8099bed2009-08-01 23:59:18 +0000578 if not self._isOpen: raise portNotOpenError
579 data = bytearray()
580 try:
581 while len(data) < size:
cliechti81c54762009-08-03 23:53:27 +0000582 if self._thread is None:
583 raise SerialException('connection failed (reader thread died)')
Chris Liechti01587b12015-08-05 02:39:32 +0200584 data += self._read_buffer.get(True, self._timeout)
cliechti8099bed2009-08-01 23:59:18 +0000585 except Queue.Empty: # -> timeout
586 pass
587 return bytes(data)
588
589 def write(self, data):
cliechtieada4fd2013-07-31 16:26:07 +0000590 """\
591 Output the given string over the serial port. Can block if the
cliechti8099bed2009-08-01 23:59:18 +0000592 connection is blocked. May raise SerialException if the connection is
cliechtieada4fd2013-07-31 16:26:07 +0000593 closed.
594 """
cliechti8099bed2009-08-01 23:59:18 +0000595 if not self._isOpen: raise portNotOpenError
Chris Liechti01587b12015-08-05 02:39:32 +0200596 with self._write_lock:
cliechti81c54762009-08-03 23:53:27 +0000597 try:
cliechti38077122013-10-16 02:57:27 +0000598 self._socket.sendall(to_bytes(data).replace(IAC, IAC_DOUBLED))
Chris Liechti68340d72015-08-03 14:15:48 +0200599 except socket.error as e:
Chris Liechti142ae562015-08-23 01:11:06 +0200600 raise SerialException("connection failed (socket error): %s" % (e,))
cliechti8099bed2009-08-01 23:59:18 +0000601 return len(data)
602
Chris Liechtief1fe252015-08-27 23:25:21 +0200603 def reset_input_buffer(self):
cliechti8099bed2009-08-01 23:59:18 +0000604 """Clear input buffer, discarding all that is in the buffer."""
605 if not self._isOpen: raise portNotOpenError
cliechti1ef7e3e2009-08-03 02:38:43 +0000606 self.rfc2217SendPurge(PURGE_RECEIVE_BUFFER)
cliechti8099bed2009-08-01 23:59:18 +0000607 # empty read buffer
608 while self._read_buffer.qsize():
609 self._read_buffer.get(False)
610
Chris Liechtief1fe252015-08-27 23:25:21 +0200611 def reset_output_buffer(self):
cliechtieada4fd2013-07-31 16:26:07 +0000612 """\
613 Clear output buffer, aborting the current output and
614 discarding all that is in the buffer.
615 """
cliechti8099bed2009-08-01 23:59:18 +0000616 if not self._isOpen: raise portNotOpenError
cliechti1ef7e3e2009-08-03 02:38:43 +0000617 self.rfc2217SendPurge(PURGE_TRANSMIT_BUFFER)
cliechti8099bed2009-08-01 23:59:18 +0000618
Chris Liechtief1fe252015-08-27 23:25:21 +0200619 def _update_break_state(self):
cliechtieada4fd2013-07-31 16:26:07 +0000620 """\
621 Set break: Controls TXD. When active, to transmitting is
622 possible.
623 """
cliechti8099bed2009-08-01 23:59:18 +0000624 if not self._isOpen: raise portNotOpenError
cliechti6a300772009-08-12 02:28:56 +0000625 if self.logger:
Chris Liechtief1fe252015-08-27 23:25:21 +0200626 self.logger.info('set BREAK to %s' % ('active' if self._break_state else 'inactive'))
627 if self._break_state:
cliechti1ef7e3e2009-08-03 02:38:43 +0000628 self.rfc2217SetControl(SET_CONTROL_BREAK_ON)
cliechti8099bed2009-08-01 23:59:18 +0000629 else:
cliechti1ef7e3e2009-08-03 02:38:43 +0000630 self.rfc2217SetControl(SET_CONTROL_BREAK_OFF)
cliechti8099bed2009-08-01 23:59:18 +0000631
Chris Liechtief1fe252015-08-27 23:25:21 +0200632 def _update_rts_state(self):
cliechti044d8662009-08-11 21:40:31 +0000633 """Set terminal status line: Request To Send."""
cliechti8099bed2009-08-01 23:59:18 +0000634 if not self._isOpen: raise portNotOpenError
cliechti6a300772009-08-12 02:28:56 +0000635 if self.logger:
Chris Liechtief1fe252015-08-27 23:25:21 +0200636 self.logger.info('set RTS to %s' % ('active' if self._rts_state else 'inactive'))
637 if self._rts_state:
cliechti1ef7e3e2009-08-03 02:38:43 +0000638 self.rfc2217SetControl(SET_CONTROL_RTS_ON)
cliechti8099bed2009-08-01 23:59:18 +0000639 else:
cliechti1ef7e3e2009-08-03 02:38:43 +0000640 self.rfc2217SetControl(SET_CONTROL_RTS_OFF)
cliechti8099bed2009-08-01 23:59:18 +0000641
Chris Liechtief1fe252015-08-27 23:25:21 +0200642 def _update_dtr_state(self, level=True):
cliechti044d8662009-08-11 21:40:31 +0000643 """Set terminal status line: Data Terminal Ready."""
cliechti8099bed2009-08-01 23:59:18 +0000644 if not self._isOpen: raise portNotOpenError
cliechti6a300772009-08-12 02:28:56 +0000645 if self.logger:
Chris Liechtief1fe252015-08-27 23:25:21 +0200646 self.logger.info('set DTR to %s' % ('active' if self._dtr_state else 'inactive'))
647 if self._dtr_state:
cliechti1ef7e3e2009-08-03 02:38:43 +0000648 self.rfc2217SetControl(SET_CONTROL_DTR_ON)
cliechti8099bed2009-08-01 23:59:18 +0000649 else:
cliechti1ef7e3e2009-08-03 02:38:43 +0000650 self.rfc2217SetControl(SET_CONTROL_DTR_OFF)
cliechti8099bed2009-08-01 23:59:18 +0000651
Chris Liechtief1fe252015-08-27 23:25:21 +0200652 @property
653 def cts(self):
cliechti044d8662009-08-11 21:40:31 +0000654 """Read terminal status line: Clear To Send."""
cliechti8099bed2009-08-01 23:59:18 +0000655 if not self._isOpen: raise portNotOpenError
cliechti7cb78e82009-08-05 15:47:57 +0000656 return bool(self.getModemState() & MODEMSTATE_MASK_CTS)
cliechti8099bed2009-08-01 23:59:18 +0000657
Chris Liechtief1fe252015-08-27 23:25:21 +0200658 @property
659 def dsr(self):
cliechti044d8662009-08-11 21:40:31 +0000660 """Read terminal status line: Data Set Ready."""
cliechti8099bed2009-08-01 23:59:18 +0000661 if not self._isOpen: raise portNotOpenError
cliechti7cb78e82009-08-05 15:47:57 +0000662 return bool(self.getModemState() & MODEMSTATE_MASK_DSR)
cliechti8099bed2009-08-01 23:59:18 +0000663
Chris Liechtief1fe252015-08-27 23:25:21 +0200664 @property
665 def ri(self):
cliechti044d8662009-08-11 21:40:31 +0000666 """Read terminal status line: Ring Indicator."""
cliechti8099bed2009-08-01 23:59:18 +0000667 if not self._isOpen: raise portNotOpenError
cliechti7cb78e82009-08-05 15:47:57 +0000668 return bool(self.getModemState() & MODEMSTATE_MASK_RI)
cliechti8099bed2009-08-01 23:59:18 +0000669
Chris Liechtief1fe252015-08-27 23:25:21 +0200670 @property
671 def cd(self):
cliechti044d8662009-08-11 21:40:31 +0000672 """Read terminal status line: Carrier Detect."""
cliechti8099bed2009-08-01 23:59:18 +0000673 if not self._isOpen: raise portNotOpenError
cliechti7cb78e82009-08-05 15:47:57 +0000674 return bool(self.getModemState() & MODEMSTATE_MASK_CD)
cliechti8099bed2009-08-01 23:59:18 +0000675
676 # - - - platform specific - - -
677 # None so far
678
679 # - - - RFC2217 specific - - -
680
cliechti1ef7e3e2009-08-03 02:38:43 +0000681 def _telnetReadLoop(self):
cliechti7d448562014-08-03 21:57:45 +0000682 """Read loop for the socket."""
cliechti8099bed2009-08-01 23:59:18 +0000683 mode = M_NORMAL
684 suboption = None
cliechti81c54762009-08-03 23:53:27 +0000685 try:
686 while self._socket is not None:
687 try:
688 data = self._socket.recv(1024)
689 except socket.timeout:
690 # just need to get out of recv form time to time to check if
691 # still alive
692 continue
Chris Liechti68340d72015-08-03 14:15:48 +0200693 except socket.error as e:
cliechti81c54762009-08-03 23:53:27 +0000694 # connection fails -> terminate loop
cliechticb20a4f2011-04-25 02:25:54 +0000695 if self.logger:
696 self.logger.debug("socket error in reader thread: %s" % (e,))
cliechti81c54762009-08-03 23:53:27 +0000697 break
cliechticb20a4f2011-04-25 02:25:54 +0000698 if not data: break # lost connection
Chris Liechtif99cd5c2015-08-13 22:54:16 +0200699 for byte in iterbytes(data):
cliechti81c54762009-08-03 23:53:27 +0000700 if mode == M_NORMAL:
701 # interpret as command or as data
702 if byte == IAC:
703 mode = M_IAC_SEEN
cliechti8099bed2009-08-01 23:59:18 +0000704 else:
cliechti81c54762009-08-03 23:53:27 +0000705 # store data in read buffer or sub option buffer
706 # depending on state
707 if suboption is not None:
Chris Liechti01587b12015-08-05 02:39:32 +0200708 suboption += byte
cliechti81c54762009-08-03 23:53:27 +0000709 else:
710 self._read_buffer.put(byte)
711 elif mode == M_IAC_SEEN:
712 if byte == IAC:
713 # interpret as command doubled -> insert character
714 # itself
cliechtif325c032009-12-25 16:09:49 +0000715 if suboption is not None:
Chris Liechti01587b12015-08-05 02:39:32 +0200716 suboption += IAC
cliechtif325c032009-12-25 16:09:49 +0000717 else:
718 self._read_buffer.put(IAC)
cliechti81c54762009-08-03 23:53:27 +0000719 mode = M_NORMAL
720 elif byte == SB:
721 # sub option start
722 suboption = bytearray()
723 mode = M_NORMAL
724 elif byte == SE:
725 # sub option end -> process it now
726 self._telnetProcessSubnegotiation(bytes(suboption))
727 suboption = None
728 mode = M_NORMAL
729 elif byte in (DO, DONT, WILL, WONT):
730 # negotiation
731 telnet_command = byte
732 mode = M_NEGOTIATE
733 else:
734 # other telnet commands
735 self._telnetProcessCommand(byte)
736 mode = M_NORMAL
737 elif mode == M_NEGOTIATE: # DO, DONT, WILL, WONT was received, option now following
738 self._telnetNegotiateOption(telnet_command, byte)
cliechti8099bed2009-08-01 23:59:18 +0000739 mode = M_NORMAL
cliechti81c54762009-08-03 23:53:27 +0000740 finally:
741 self._thread = None
cliechti6a300772009-08-12 02:28:56 +0000742 if self.logger:
743 self.logger.debug("read thread terminated")
cliechti8099bed2009-08-01 23:59:18 +0000744
745 # - incoming telnet commands and options
746
cliechti1ef7e3e2009-08-03 02:38:43 +0000747 def _telnetProcessCommand(self, command):
cliechti044d8662009-08-11 21:40:31 +0000748 """Process commands other than DO, DONT, WILL, WONT."""
cliechti1ef7e3e2009-08-03 02:38:43 +0000749 # Currently none. RFC2217 only uses negotiation and subnegotiation.
cliechti6a300772009-08-12 02:28:56 +0000750 if self.logger:
751 self.logger.warning("ignoring Telnet command: %r" % (command,))
cliechti8099bed2009-08-01 23:59:18 +0000752
cliechti1ef7e3e2009-08-03 02:38:43 +0000753 def _telnetNegotiateOption(self, command, option):
cliechti044d8662009-08-11 21:40:31 +0000754 """Process incoming DO, DONT, WILL, WONT."""
cliechti2b929b72009-08-02 23:49:02 +0000755 # check our registered telnet options and forward command to them
756 # they know themselves if they have to answer or not
cliechtiac205322009-08-02 20:40:21 +0000757 known = False
758 for item in self._telnet_options:
cliechti2b929b72009-08-02 23:49:02 +0000759 # can have more than one match! as some options are duplicated for
760 # 'us' and 'them'
cliechtiac205322009-08-02 20:40:21 +0000761 if item.option == option:
cliechti2b929b72009-08-02 23:49:02 +0000762 item.process_incoming(command)
cliechtiac205322009-08-02 20:40:21 +0000763 known = True
764 if not known:
765 # handle unknown options
766 # only answer to positive requests and deny them
767 if command == WILL or command == DO:
Chris Liechti142ae562015-08-23 01:11:06 +0200768 self.telnetSendOption((DONT if command == WILL else WONT), option)
cliechti6a300772009-08-12 02:28:56 +0000769 if self.logger:
770 self.logger.warning("rejected Telnet option: %r" % (option,))
cliechtiac205322009-08-02 20:40:21 +0000771
cliechti8099bed2009-08-01 23:59:18 +0000772
cliechti1ef7e3e2009-08-03 02:38:43 +0000773 def _telnetProcessSubnegotiation(self, suboption):
cliechti044d8662009-08-11 21:40:31 +0000774 """Process subnegotiation, the data between IAC SB and IAC SE."""
cliechti8099bed2009-08-01 23:59:18 +0000775 if suboption[0:1] == COM_PORT_OPTION:
776 if suboption[1:2] == SERVER_NOTIFY_LINESTATE and len(suboption) >= 3:
cliechti672d0292009-08-03 02:01:57 +0000777 self._linestate = ord(suboption[2:3]) # ensure it is a number
cliechti6a300772009-08-12 02:28:56 +0000778 if self.logger:
779 self.logger.info("NOTIFY_LINESTATE: %s" % self._linestate)
cliechti8099bed2009-08-01 23:59:18 +0000780 elif suboption[1:2] == SERVER_NOTIFY_MODEMSTATE and len(suboption) >= 3:
cliechti672d0292009-08-03 02:01:57 +0000781 self._modemstate = ord(suboption[2:3]) # ensure it is a number
cliechti6a300772009-08-12 02:28:56 +0000782 if self.logger:
783 self.logger.info("NOTIFY_MODEMSTATE: %s" % self._modemstate)
cliechti7cb78e82009-08-05 15:47:57 +0000784 # update time when we think that a poll would make sense
785 self._modemstate_expires = time.time() + 0.3
cliechti672d0292009-08-03 02:01:57 +0000786 elif suboption[1:2] == FLOWCONTROL_SUSPEND:
787 self._remote_suspend_flow = True
788 elif suboption[1:2] == FLOWCONTROL_RESUME:
789 self._remote_suspend_flow = False
cliechti8099bed2009-08-01 23:59:18 +0000790 else:
cliechti2b929b72009-08-02 23:49:02 +0000791 for item in self._rfc2217_options.values():
792 if item.ack_option == suboption[1:2]:
cliechti81c54762009-08-03 23:53:27 +0000793 #~ print "processing COM_PORT_OPTION: %r" % list(suboption[1:])
cliechti2b929b72009-08-02 23:49:02 +0000794 item.checkAnswer(bytes(suboption[2:]))
795 break
796 else:
cliechti6a300772009-08-12 02:28:56 +0000797 if self.logger:
798 self.logger.warning("ignoring COM_PORT_OPTION: %r" % (suboption,))
cliechti8099bed2009-08-01 23:59:18 +0000799 else:
cliechti6a300772009-08-12 02:28:56 +0000800 if self.logger:
801 self.logger.warning("ignoring subnegotiation: %r" % (suboption,))
cliechti8099bed2009-08-01 23:59:18 +0000802
803 # - outgoing telnet commands and options
804
cliechti81c54762009-08-03 23:53:27 +0000805 def _internal_raw_write(self, data):
cliechti044d8662009-08-11 21:40:31 +0000806 """internal socket write with no data escaping. used to send telnet stuff."""
Chris Liechti01587b12015-08-05 02:39:32 +0200807 with self._write_lock:
cliechti81c54762009-08-03 23:53:27 +0000808 self._socket.sendall(data)
cliechti81c54762009-08-03 23:53:27 +0000809
cliechti1ef7e3e2009-08-03 02:38:43 +0000810 def telnetSendOption(self, action, option):
cliechti044d8662009-08-11 21:40:31 +0000811 """Send DO, DONT, WILL, WONT."""
cliechti81c54762009-08-03 23:53:27 +0000812 self._internal_raw_write(to_bytes([IAC, action, option]))
cliechti8099bed2009-08-01 23:59:18 +0000813
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200814 def rfc2217SendSubnegotiation(self, option, value=b''):
cliechti044d8662009-08-11 21:40:31 +0000815 """Subnegotiation of RFC2217 parameters."""
cliechtif325c032009-12-25 16:09:49 +0000816 value = value.replace(IAC, IAC_DOUBLED)
cliechti81c54762009-08-03 23:53:27 +0000817 self._internal_raw_write(to_bytes([IAC, SB, COM_PORT_OPTION, option] + list(value) + [IAC, SE]))
cliechti2b929b72009-08-02 23:49:02 +0000818
cliechti1ef7e3e2009-08-03 02:38:43 +0000819 def rfc2217SendPurge(self, value):
cliechti2b929b72009-08-02 23:49:02 +0000820 item = self._rfc2217_options['purge']
cliechti672d0292009-08-03 02:01:57 +0000821 item.set(value) # transmit desired purge type
cliechtidfe2d272009-08-10 22:19:41 +0000822 item.wait(self._network_timeout) # wait for acknowledge from the server
cliechti2b929b72009-08-02 23:49:02 +0000823
cliechti1ef7e3e2009-08-03 02:38:43 +0000824 def rfc2217SetControl(self, value):
cliechti81c54762009-08-03 23:53:27 +0000825 item = self._rfc2217_options['control']
cliechticb20a4f2011-04-25 02:25:54 +0000826 item.set(value) # transmit desired control type
cliechti81c54762009-08-03 23:53:27 +0000827 if self._ignore_set_control_answer:
828 # answers are ignored when option is set. compatibility mode for
cliechticb20a4f2011-04-25 02:25:54 +0000829 # servers that answer, but not the expected one... (or no answer
cliechti81c54762009-08-03 23:53:27 +0000830 # at all) i.e. sredird
831 time.sleep(0.1) # this helps getting the unit tests passed
832 else:
cliechtidfe2d272009-08-10 22:19:41 +0000833 item.wait(self._network_timeout) # wait for acknowledge from the server
cliechti8099bed2009-08-01 23:59:18 +0000834
cliechti1ef7e3e2009-08-03 02:38:43 +0000835 def rfc2217FlowServerReady(self):
cliechtieada4fd2013-07-31 16:26:07 +0000836 """\
837 check if server is ready to receive data. block for some time when
838 not.
839 """
cliechti672d0292009-08-03 02:01:57 +0000840 #~ if self._remote_suspend_flow:
841 #~ wait---
842
cliechti7cb78e82009-08-05 15:47:57 +0000843 def getModemState(self):
cliechtieada4fd2013-07-31 16:26:07 +0000844 """\
cliechti7d448562014-08-03 21:57:45 +0000845 get last modem state (cached value. If value is "old", request a new
846 one. This cache helps that we don't issue to many requests when e.g. all
847 status lines, one after the other is queried by the user (getCTS, getDSR
cliechtieada4fd2013-07-31 16:26:07 +0000848 etc.)
849 """
cliechti7cb78e82009-08-05 15:47:57 +0000850 # active modem state polling enabled? is the value fresh enough?
851 if self._poll_modem_state and self._modemstate_expires < time.time():
cliechti6a300772009-08-12 02:28:56 +0000852 if self.logger:
853 self.logger.debug('polling modem state')
cliechti7cb78e82009-08-05 15:47:57 +0000854 # when it is older, request an update
855 self.rfc2217SendSubnegotiation(NOTIFY_MODEMSTATE)
cliechtidfe2d272009-08-10 22:19:41 +0000856 timeout_time = time.time() + self._network_timeout
cliechti7cb78e82009-08-05 15:47:57 +0000857 while time.time() < timeout_time:
858 time.sleep(0.05) # prevent 100% CPU load
859 # when expiration time is updated, it means that there is a new
860 # value
861 if self._modemstate_expires > time.time():
862 break
Chris Liechti01587b12015-08-05 02:39:32 +0200863 else:
864 if self.logger:
865 self.logger.warning('poll for modem state failed')
cliechti7cb78e82009-08-05 15:47:57 +0000866 # even when there is a timeout, do not generate an error just
867 # return the last known value. this way we can support buggy
868 # servers that do not respond to polls, but send automatic
869 # updates.
870 if self._modemstate is not None:
cliechti6a300772009-08-12 02:28:56 +0000871 if self.logger:
872 self.logger.debug('using cached modem state')
cliechti7cb78e82009-08-05 15:47:57 +0000873 return self._modemstate
874 else:
875 # never received a notification from the server
cliechti8fb119c2009-08-05 23:39:45 +0000876 raise SerialException("remote sends no NOTIFY_MODEMSTATE")
cliechti8099bed2009-08-01 23:59:18 +0000877
cliechti5cc3eb12009-08-11 23:04:30 +0000878
cliechti595ed5b2009-08-10 01:43:32 +0000879#############################################################################
cliechti5cc3eb12009-08-11 23:04:30 +0000880# The following is code that helps implementing an RFC 2217 server.
cliechti8099bed2009-08-01 23:59:18 +0000881
cliechti8ccc2ff2009-08-05 12:44:46 +0000882class PortManager(object):
cliechtieada4fd2013-07-31 16:26:07 +0000883 """\
884 This class manages the state of Telnet and RFC 2217. It needs a serial
cliechticb20a4f2011-04-25 02:25:54 +0000885 instance and a connection to work with. Connection is expected to implement
cliechtieada4fd2013-07-31 16:26:07 +0000886 a (thread safe) write function, that writes the string to the network.
887 """
cliechti130d1f02009-08-04 02:10:58 +0000888
cliechti6a300772009-08-12 02:28:56 +0000889 def __init__(self, serial_port, connection, logger=None):
cliechti130d1f02009-08-04 02:10:58 +0000890 self.serial = serial_port
891 self.connection = connection
cliechti6a300772009-08-12 02:28:56 +0000892 self.logger = logger
cliechti86b593e2009-08-05 16:28:12 +0000893 self._client_is_rfc2217 = False
cliechti130d1f02009-08-04 02:10:58 +0000894
895 # filter state machine
896 self.mode = M_NORMAL
897 self.suboption = None
898 self.telnet_command = None
899
900 # states for modem/line control events
901 self.modemstate_mask = 255
902 self.last_modemstate = None
903 self.linstate_mask = 0
904
905 # all supported telnet options
906 self._telnet_options = [
907 TelnetOption(self, 'ECHO', ECHO, WILL, WONT, DO, DONT, REQUESTED),
908 TelnetOption(self, 'we-SGA', SGA, WILL, WONT, DO, DONT, REQUESTED),
909 TelnetOption(self, 'they-SGA', SGA, DO, DONT, WILL, WONT, INACTIVE),
910 TelnetOption(self, 'we-BINARY', BINARY, WILL, WONT, DO, DONT, INACTIVE),
911 TelnetOption(self, 'they-BINARY', BINARY, DO, DONT, WILL, WONT, REQUESTED),
cliechti86b593e2009-08-05 16:28:12 +0000912 TelnetOption(self, 'we-RFC2217', COM_PORT_OPTION, WILL, WONT, DO, DONT, REQUESTED, self._client_ok),
913 TelnetOption(self, 'they-RFC2217', COM_PORT_OPTION, DO, DONT, WILL, WONT, INACTIVE, self._client_ok),
cliechti130d1f02009-08-04 02:10:58 +0000914 ]
915
916 # negotiate Telnet/RFC2217 -> send initial requests
cliechti6a300772009-08-12 02:28:56 +0000917 if self.logger:
918 self.logger.debug("requesting initial Telnet/RFC 2217 options")
cliechti130d1f02009-08-04 02:10:58 +0000919 for option in self._telnet_options:
920 if option.state is REQUESTED:
921 self.telnetSendOption(option.send_yes, option.option)
922 # issue 1st modem state notification
cliechti86b593e2009-08-05 16:28:12 +0000923
924 def _client_ok(self):
cliechtieada4fd2013-07-31 16:26:07 +0000925 """\
cliechti7d448562014-08-03 21:57:45 +0000926 callback of telnet option. It gets called when option is activated.
927 This one here is used to detect when the client agrees on RFC 2217. A
cliechti86b593e2009-08-05 16:28:12 +0000928 flag is set so that other functions like check_modem_lines know if the
cliechti7d448562014-08-03 21:57:45 +0000929 client is OK.
cliechtieada4fd2013-07-31 16:26:07 +0000930 """
cliechti86b593e2009-08-05 16:28:12 +0000931 # The callback is used for we and they so if one party agrees, we're
932 # already happy. it seems not all servers do the negotiation correctly
933 # and i guess there are incorrect clients too.. so be happy if client
934 # answers one or the other positively.
935 self._client_is_rfc2217 = True
cliechti6a300772009-08-12 02:28:56 +0000936 if self.logger:
937 self.logger.info("client accepts RFC 2217")
cliechti8fb119c2009-08-05 23:39:45 +0000938 # this is to ensure that the client gets a notification, even if there
939 # was no change
940 self.check_modem_lines(force_notification=True)
cliechti130d1f02009-08-04 02:10:58 +0000941
942 # - outgoing telnet commands and options
943
944 def telnetSendOption(self, action, option):
cliechti044d8662009-08-11 21:40:31 +0000945 """Send DO, DONT, WILL, WONT."""
cliechti130d1f02009-08-04 02:10:58 +0000946 self.connection.write(to_bytes([IAC, action, option]))
947
Chris Liechtib4cda3a2015-08-08 17:12:08 +0200948 def rfc2217SendSubnegotiation(self, option, value=b''):
cliechti044d8662009-08-11 21:40:31 +0000949 """Subnegotiation of RFC 2217 parameters."""
cliechtif325c032009-12-25 16:09:49 +0000950 value = value.replace(IAC, IAC_DOUBLED)
cliechti130d1f02009-08-04 02:10:58 +0000951 self.connection.write(to_bytes([IAC, SB, COM_PORT_OPTION, option] + list(value) + [IAC, SE]))
952
953 # - check modem lines, needs to be called periodically from user to
954 # establish polling
955
cliechti7cb78e82009-08-05 15:47:57 +0000956 def check_modem_lines(self, force_notification=False):
cliechti130d1f02009-08-04 02:10:58 +0000957 modemstate = (
Chris Liechti142ae562015-08-23 01:11:06 +0200958 (self.serial.getCTS() and MODEMSTATE_MASK_CTS) |
959 (self.serial.getDSR() and MODEMSTATE_MASK_DSR) |
960 (self.serial.getRI() and MODEMSTATE_MASK_RI) |
961 (self.serial.getCD() and MODEMSTATE_MASK_CD))
cliechti7cb78e82009-08-05 15:47:57 +0000962 # check what has changed
963 deltas = modemstate ^ (self.last_modemstate or 0) # when last is None -> 0
964 if deltas & MODEMSTATE_MASK_CTS:
965 modemstate |= MODEMSTATE_MASK_CTS_CHANGE
966 if deltas & MODEMSTATE_MASK_DSR:
967 modemstate |= MODEMSTATE_MASK_DSR_CHANGE
968 if deltas & MODEMSTATE_MASK_RI:
969 modemstate |= MODEMSTATE_MASK_RI_CHANGE
970 if deltas & MODEMSTATE_MASK_CD:
971 modemstate |= MODEMSTATE_MASK_CD_CHANGE
972 # if new state is different and the mask allows this change, send
cliechti86b593e2009-08-05 16:28:12 +0000973 # notification. suppress notifications when client is not rfc2217
cliechti7cb78e82009-08-05 15:47:57 +0000974 if modemstate != self.last_modemstate or force_notification:
cliechti8fb119c2009-08-05 23:39:45 +0000975 if (self._client_is_rfc2217 and (modemstate & self.modemstate_mask)) or force_notification:
cliechti7cb78e82009-08-05 15:47:57 +0000976 self.rfc2217SendSubnegotiation(
977 SERVER_NOTIFY_MODEMSTATE,
978 to_bytes([modemstate & self.modemstate_mask])
979 )
cliechti6a300772009-08-12 02:28:56 +0000980 if self.logger:
981 self.logger.info("NOTIFY_MODEMSTATE: %s" % (modemstate,))
cliechti7cb78e82009-08-05 15:47:57 +0000982 # save last state, but forget about deltas.
983 # otherwise it would also notify about changing deltas which is
984 # probably not very useful
985 self.last_modemstate = modemstate & 0xf0
cliechti130d1f02009-08-04 02:10:58 +0000986
cliechti32c10332009-08-05 13:23:43 +0000987 # - outgoing data escaping
988
989 def escape(self, data):
cliechtieada4fd2013-07-31 16:26:07 +0000990 """\
cliechti7d448562014-08-03 21:57:45 +0000991 This generator function is for the user. All outgoing data has to be
cliechticb20a4f2011-04-25 02:25:54 +0000992 properly escaped, so that no IAC character in the data stream messes up
993 the Telnet state machine in the server.
cliechti32c10332009-08-05 13:23:43 +0000994
995 socket.sendall(escape(data))
996 """
997 for byte in data:
998 if byte == IAC:
999 yield IAC
1000 yield IAC
1001 else:
1002 yield byte
1003
cliechti130d1f02009-08-04 02:10:58 +00001004 # - incoming data filter
1005
1006 def filter(self, data):
cliechtieada4fd2013-07-31 16:26:07 +00001007 """\
cliechti7d448562014-08-03 21:57:45 +00001008 Handle a bunch of incoming bytes. This is a generator. It will yield
cliechti044d8662009-08-11 21:40:31 +00001009 all characters not of interest for Telnet/RFC 2217.
cliechti130d1f02009-08-04 02:10:58 +00001010
1011 The idea is that the reader thread pushes data from the socket through
1012 this filter:
1013
1014 for byte in filter(socket.recv(1024)):
1015 # do things like CR/LF conversion/whatever
1016 # and write data to the serial port
1017 serial.write(byte)
1018
1019 (socket error handling code left as exercise for the reader)
1020 """
Chris Liechtif99cd5c2015-08-13 22:54:16 +02001021 for byte in iterbytes(data):
cliechti130d1f02009-08-04 02:10:58 +00001022 if self.mode == M_NORMAL:
1023 # interpret as command or as data
1024 if byte == IAC:
1025 self.mode = M_IAC_SEEN
1026 else:
1027 # store data in sub option buffer or pass it to our
1028 # consumer depending on state
1029 if self.suboption is not None:
Chris Liechti01587b12015-08-05 02:39:32 +02001030 self.suboption += byte
cliechti130d1f02009-08-04 02:10:58 +00001031 else:
1032 yield byte
1033 elif self.mode == M_IAC_SEEN:
1034 if byte == IAC:
1035 # interpret as command doubled -> insert character
1036 # itself
cliechtif325c032009-12-25 16:09:49 +00001037 if self.suboption is not None:
Chris Liechti01587b12015-08-05 02:39:32 +02001038 self.suboption += byte
cliechtif325c032009-12-25 16:09:49 +00001039 else:
1040 yield byte
cliechti130d1f02009-08-04 02:10:58 +00001041 self.mode = M_NORMAL
1042 elif byte == SB:
1043 # sub option start
1044 self.suboption = bytearray()
1045 self.mode = M_NORMAL
1046 elif byte == SE:
1047 # sub option end -> process it now
1048 self._telnetProcessSubnegotiation(bytes(self.suboption))
1049 self.suboption = None
1050 self.mode = M_NORMAL
1051 elif byte in (DO, DONT, WILL, WONT):
1052 # negotiation
1053 self.telnet_command = byte
1054 self.mode = M_NEGOTIATE
1055 else:
1056 # other telnet commands
1057 self._telnetProcessCommand(byte)
1058 self.mode = M_NORMAL
1059 elif self.mode == M_NEGOTIATE: # DO, DONT, WILL, WONT was received, option now following
1060 self._telnetNegotiateOption(self.telnet_command, byte)
1061 self.mode = M_NORMAL
1062
1063 # - incoming telnet commands and options
1064
1065 def _telnetProcessCommand(self, command):
cliechti044d8662009-08-11 21:40:31 +00001066 """Process commands other than DO, DONT, WILL, WONT."""
cliechti130d1f02009-08-04 02:10:58 +00001067 # Currently none. RFC2217 only uses negotiation and subnegotiation.
cliechti6a300772009-08-12 02:28:56 +00001068 if self.logger:
1069 self.logger.warning("ignoring Telnet command: %r" % (command,))
cliechti130d1f02009-08-04 02:10:58 +00001070
1071 def _telnetNegotiateOption(self, command, option):
cliechti044d8662009-08-11 21:40:31 +00001072 """Process incoming DO, DONT, WILL, WONT."""
cliechti130d1f02009-08-04 02:10:58 +00001073 # check our registered telnet options and forward command to them
1074 # they know themselves if they have to answer or not
1075 known = False
1076 for item in self._telnet_options:
1077 # can have more than one match! as some options are duplicated for
1078 # 'us' and 'them'
1079 if item.option == option:
1080 item.process_incoming(command)
1081 known = True
1082 if not known:
1083 # handle unknown options
1084 # only answer to positive requests and deny them
1085 if command == WILL or command == DO:
Chris Liechti142ae562015-08-23 01:11:06 +02001086 self.telnetSendOption((DONT if command == WILL else WONT), option)
cliechti6a300772009-08-12 02:28:56 +00001087 if self.logger:
1088 self.logger.warning("rejected Telnet option: %r" % (option,))
cliechti130d1f02009-08-04 02:10:58 +00001089
1090
1091 def _telnetProcessSubnegotiation(self, suboption):
cliechti044d8662009-08-11 21:40:31 +00001092 """Process subnegotiation, the data between IAC SB and IAC SE."""
cliechti130d1f02009-08-04 02:10:58 +00001093 if suboption[0:1] == COM_PORT_OPTION:
cliechti6a300772009-08-12 02:28:56 +00001094 if self.logger:
1095 self.logger.debug('received COM_PORT_OPTION: %r' % (suboption,))
cliechti130d1f02009-08-04 02:10:58 +00001096 if suboption[1:2] == SET_BAUDRATE:
1097 backup = self.serial.baudrate
1098 try:
Chris Liechti01587b12015-08-05 02:39:32 +02001099 (baudrate,) = struct.unpack(b"!I", suboption[2:6])
cliechtieada4fd2013-07-31 16:26:07 +00001100 if baudrate != 0:
1101 self.serial.baudrate = baudrate
Chris Liechtid2146002015-08-04 16:57:16 +02001102 except ValueError as e:
cliechti6a300772009-08-12 02:28:56 +00001103 if self.logger:
1104 self.logger.error("failed to set baud rate: %s" % (e,))
cliechti130d1f02009-08-04 02:10:58 +00001105 self.serial.baudrate = backup
cliechti5cc3eb12009-08-11 23:04:30 +00001106 else:
cliechti6a300772009-08-12 02:28:56 +00001107 if self.logger:
Chris Liechti142ae562015-08-23 01:11:06 +02001108 self.logger.info("%s baud rate: %s" % ('set' if baudrate else 'get', self.serial.baudrate))
Chris Liechti01587b12015-08-05 02:39:32 +02001109 self.rfc2217SendSubnegotiation(SERVER_SET_BAUDRATE, struct.pack(b"!I", self.serial.baudrate))
cliechti130d1f02009-08-04 02:10:58 +00001110 elif suboption[1:2] == SET_DATASIZE:
1111 backup = self.serial.bytesize
1112 try:
Chris Liechti142ae562015-08-23 01:11:06 +02001113 (datasize,) = struct.unpack(b"!B", suboption[2:3])
cliechtieada4fd2013-07-31 16:26:07 +00001114 if datasize != 0:
1115 self.serial.bytesize = datasize
Chris Liechtid2146002015-08-04 16:57:16 +02001116 except ValueError as e:
cliechti6a300772009-08-12 02:28:56 +00001117 if self.logger:
1118 self.logger.error("failed to set data size: %s" % (e,))
cliechti130d1f02009-08-04 02:10:58 +00001119 self.serial.bytesize = backup
cliechti5cc3eb12009-08-11 23:04:30 +00001120 else:
cliechti6a300772009-08-12 02:28:56 +00001121 if self.logger:
Chris Liechti142ae562015-08-23 01:11:06 +02001122 self.logger.info("%s data size: %s" % ('set' if datasize else 'get', self.serial.bytesize))
Chris Liechti01587b12015-08-05 02:39:32 +02001123 self.rfc2217SendSubnegotiation(SERVER_SET_DATASIZE, struct.pack(b"!B", self.serial.bytesize))
cliechti130d1f02009-08-04 02:10:58 +00001124 elif suboption[1:2] == SET_PARITY:
1125 backup = self.serial.parity
1126 try:
Chris Liechti01587b12015-08-05 02:39:32 +02001127 parity = struct.unpack(b"!B", suboption[2:3])[0]
cliechtieada4fd2013-07-31 16:26:07 +00001128 if parity != 0:
1129 self.serial.parity = RFC2217_REVERSE_PARITY_MAP[parity]
Chris Liechtid2146002015-08-04 16:57:16 +02001130 except ValueError as e:
cliechti6a300772009-08-12 02:28:56 +00001131 if self.logger:
1132 self.logger.error("failed to set parity: %s" % (e,))
cliechti130d1f02009-08-04 02:10:58 +00001133 self.serial.parity = backup
cliechti5cc3eb12009-08-11 23:04:30 +00001134 else:
cliechti6a300772009-08-12 02:28:56 +00001135 if self.logger:
Chris Liechti142ae562015-08-23 01:11:06 +02001136 self.logger.info("%s parity: %s" % ('set' if parity else 'get', self.serial.parity))
cliechti130d1f02009-08-04 02:10:58 +00001137 self.rfc2217SendSubnegotiation(
1138 SERVER_SET_PARITY,
Chris Liechtib4cda3a2015-08-08 17:12:08 +02001139 struct.pack(b"!B", RFC2217_PARITY_MAP[self.serial.parity])
cliechti130d1f02009-08-04 02:10:58 +00001140 )
1141 elif suboption[1:2] == SET_STOPSIZE:
1142 backup = self.serial.stopbits
1143 try:
Chris Liechti01587b12015-08-05 02:39:32 +02001144 stopbits = struct.unpack(b"!B", suboption[2:3])[0]
cliechtieada4fd2013-07-31 16:26:07 +00001145 if stopbits != 0:
1146 self.serial.stopbits = RFC2217_REVERSE_STOPBIT_MAP[stopbits]
Chris Liechtid2146002015-08-04 16:57:16 +02001147 except ValueError as e:
cliechti6a300772009-08-12 02:28:56 +00001148 if self.logger:
1149 self.logger.error("failed to set stop bits: %s" % (e,))
cliechti130d1f02009-08-04 02:10:58 +00001150 self.serial.stopbits = backup
cliechti5cc3eb12009-08-11 23:04:30 +00001151 else:
cliechti6a300772009-08-12 02:28:56 +00001152 if self.logger:
Chris Liechti142ae562015-08-23 01:11:06 +02001153 self.logger.info("%s stop bits: %s" % ('set' if stopbits else 'get', self.serial.stopbits))
cliechti130d1f02009-08-04 02:10:58 +00001154 self.rfc2217SendSubnegotiation(
1155 SERVER_SET_STOPSIZE,
Chris Liechti01587b12015-08-05 02:39:32 +02001156 struct.pack(b"!B", RFC2217_STOPBIT_MAP[self.serial.stopbits])
cliechti130d1f02009-08-04 02:10:58 +00001157 )
1158 elif suboption[1:2] == SET_CONTROL:
1159 if suboption[2:3] == SET_CONTROL_REQ_FLOW_SETTING:
1160 if self.serial.xonxoff:
1161 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_SW_FLOW_CONTROL)
1162 elif self.serial.rtscts:
1163 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_HW_FLOW_CONTROL)
1164 else:
1165 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_NO_FLOW_CONTROL)
1166 elif suboption[2:3] == SET_CONTROL_USE_NO_FLOW_CONTROL:
1167 self.serial.xonxoff = False
1168 self.serial.rtscts = False
cliechti6a300772009-08-12 02:28:56 +00001169 if self.logger:
1170 self.logger.info("changed flow control to None")
cliechti130d1f02009-08-04 02:10:58 +00001171 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_NO_FLOW_CONTROL)
1172 elif suboption[2:3] == SET_CONTROL_USE_SW_FLOW_CONTROL:
1173 self.serial.xonxoff = True
cliechti6a300772009-08-12 02:28:56 +00001174 if self.logger:
1175 self.logger.info("changed flow control to XON/XOFF")
cliechti130d1f02009-08-04 02:10:58 +00001176 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_SW_FLOW_CONTROL)
1177 elif suboption[2:3] == SET_CONTROL_USE_HW_FLOW_CONTROL:
1178 self.serial.rtscts = True
cliechti6a300772009-08-12 02:28:56 +00001179 if self.logger:
1180 self.logger.info("changed flow control to RTS/CTS")
cliechti130d1f02009-08-04 02:10:58 +00001181 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_USE_HW_FLOW_CONTROL)
1182 elif suboption[2:3] == SET_CONTROL_REQ_BREAK_STATE:
cliechti6a300772009-08-12 02:28:56 +00001183 if self.logger:
1184 self.logger.warning("requested break state - not implemented")
cliechti130d1f02009-08-04 02:10:58 +00001185 pass # XXX needs cached value
1186 elif suboption[2:3] == SET_CONTROL_BREAK_ON:
1187 self.serial.setBreak(True)
cliechti6a300772009-08-12 02:28:56 +00001188 if self.logger:
1189 self.logger.info("changed BREAK to active")
cliechti130d1f02009-08-04 02:10:58 +00001190 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_BREAK_ON)
1191 elif suboption[2:3] == SET_CONTROL_BREAK_OFF:
1192 self.serial.setBreak(False)
cliechti6a300772009-08-12 02:28:56 +00001193 if self.logger:
1194 self.logger.info("changed BREAK to inactive")
cliechti130d1f02009-08-04 02:10:58 +00001195 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_BREAK_OFF)
1196 elif suboption[2:3] == SET_CONTROL_REQ_DTR:
cliechti6a300772009-08-12 02:28:56 +00001197 if self.logger:
1198 self.logger.warning("requested DTR state - not implemented")
cliechti130d1f02009-08-04 02:10:58 +00001199 pass # XXX needs cached value
1200 elif suboption[2:3] == SET_CONTROL_DTR_ON:
1201 self.serial.setDTR(True)
cliechti6a300772009-08-12 02:28:56 +00001202 if self.logger:
1203 self.logger.info("changed DTR to active")
cliechti130d1f02009-08-04 02:10:58 +00001204 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_DTR_ON)
1205 elif suboption[2:3] == SET_CONTROL_DTR_OFF:
1206 self.serial.setDTR(False)
cliechti6a300772009-08-12 02:28:56 +00001207 if self.logger:
1208 self.logger.info("changed DTR to inactive")
cliechti130d1f02009-08-04 02:10:58 +00001209 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_DTR_OFF)
1210 elif suboption[2:3] == SET_CONTROL_REQ_RTS:
cliechti6a300772009-08-12 02:28:56 +00001211 if self.logger:
1212 self.logger.warning("requested RTS state - not implemented")
cliechti130d1f02009-08-04 02:10:58 +00001213 pass # XXX needs cached value
1214 #~ self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_RTS_ON)
1215 elif suboption[2:3] == SET_CONTROL_RTS_ON:
1216 self.serial.setRTS(True)
cliechti6a300772009-08-12 02:28:56 +00001217 if self.logger:
1218 self.logger.info("changed RTS to active")
cliechti130d1f02009-08-04 02:10:58 +00001219 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_RTS_ON)
1220 elif suboption[2:3] == SET_CONTROL_RTS_OFF:
1221 self.serial.setRTS(False)
cliechti6a300772009-08-12 02:28:56 +00001222 if self.logger:
1223 self.logger.info("changed RTS to inactive")
cliechti130d1f02009-08-04 02:10:58 +00001224 self.rfc2217SendSubnegotiation(SERVER_SET_CONTROL, SET_CONTROL_RTS_OFF)
1225 #~ elif suboption[2:3] == SET_CONTROL_REQ_FLOW_SETTING_IN:
1226 #~ elif suboption[2:3] == SET_CONTROL_USE_NO_FLOW_CONTROL_IN:
1227 #~ elif suboption[2:3] == SET_CONTROL_USE_SW_FLOW_CONTOL_IN:
1228 #~ elif suboption[2:3] == SET_CONTROL_USE_HW_FLOW_CONTOL_IN:
1229 #~ elif suboption[2:3] == SET_CONTROL_USE_DCD_FLOW_CONTROL:
1230 #~ elif suboption[2:3] == SET_CONTROL_USE_DTR_FLOW_CONTROL:
1231 #~ elif suboption[2:3] == SET_CONTROL_USE_DSR_FLOW_CONTROL:
1232 elif suboption[1:2] == NOTIFY_LINESTATE:
cliechti7cb78e82009-08-05 15:47:57 +00001233 # client polls for current state
1234 self.rfc2217SendSubnegotiation(
Chris Liechti142ae562015-08-23 01:11:06 +02001235 SERVER_NOTIFY_LINESTATE,
1236 to_bytes([0])) # sorry, nothing like that implemented
cliechti130d1f02009-08-04 02:10:58 +00001237 elif suboption[1:2] == NOTIFY_MODEMSTATE:
cliechti6a300772009-08-12 02:28:56 +00001238 if self.logger:
1239 self.logger.info("request for modem state")
cliechti7cb78e82009-08-05 15:47:57 +00001240 # client polls for current state
1241 self.check_modem_lines(force_notification=True)
cliechti130d1f02009-08-04 02:10:58 +00001242 elif suboption[1:2] == FLOWCONTROL_SUSPEND:
cliechti6a300772009-08-12 02:28:56 +00001243 if self.logger:
1244 self.logger.info("suspend")
cliechti130d1f02009-08-04 02:10:58 +00001245 self._remote_suspend_flow = True
1246 elif suboption[1:2] == FLOWCONTROL_RESUME:
cliechti6a300772009-08-12 02:28:56 +00001247 if self.logger:
1248 self.logger.info("resume")
cliechti130d1f02009-08-04 02:10:58 +00001249 self._remote_suspend_flow = False
1250 elif suboption[1:2] == SET_LINESTATE_MASK:
1251 self.linstate_mask = ord(suboption[2:3]) # ensure it is a number
cliechti6a300772009-08-12 02:28:56 +00001252 if self.logger:
cliechtif325c032009-12-25 16:09:49 +00001253 self.logger.info("line state mask: 0x%02x" % (self.linstate_mask,))
cliechti130d1f02009-08-04 02:10:58 +00001254 elif suboption[1:2] == SET_MODEMSTATE_MASK:
1255 self.modemstate_mask = ord(suboption[2:3]) # ensure it is a number
cliechti6a300772009-08-12 02:28:56 +00001256 if self.logger:
cliechtif325c032009-12-25 16:09:49 +00001257 self.logger.info("modem state mask: 0x%02x" % (self.modemstate_mask,))
cliechti130d1f02009-08-04 02:10:58 +00001258 elif suboption[1:2] == PURGE_DATA:
1259 if suboption[2:3] == PURGE_RECEIVE_BUFFER:
1260 self.serial.flushInput()
cliechti6a300772009-08-12 02:28:56 +00001261 if self.logger:
1262 self.logger.info("purge in")
cliechti130d1f02009-08-04 02:10:58 +00001263 self.rfc2217SendSubnegotiation(SERVER_PURGE_DATA, PURGE_RECEIVE_BUFFER)
1264 elif suboption[2:3] == PURGE_TRANSMIT_BUFFER:
1265 self.serial.flushOutput()
cliechti6a300772009-08-12 02:28:56 +00001266 if self.logger:
1267 self.logger.info("purge out")
cliechti130d1f02009-08-04 02:10:58 +00001268 self.rfc2217SendSubnegotiation(SERVER_PURGE_DATA, PURGE_TRANSMIT_BUFFER)
1269 elif suboption[2:3] == PURGE_BOTH_BUFFERS:
1270 self.serial.flushInput()
1271 self.serial.flushOutput()
cliechti6a300772009-08-12 02:28:56 +00001272 if self.logger:
1273 self.logger.info("purge both")
cliechti130d1f02009-08-04 02:10:58 +00001274 self.rfc2217SendSubnegotiation(SERVER_PURGE_DATA, PURGE_BOTH_BUFFERS)
1275 else:
cliechti6a300772009-08-12 02:28:56 +00001276 if self.logger:
1277 self.logger.error("undefined PURGE_DATA: %r" % list(suboption[2:]))
cliechti130d1f02009-08-04 02:10:58 +00001278 else:
cliechti6a300772009-08-12 02:28:56 +00001279 if self.logger:
1280 self.logger.error("undefined COM_PORT_OPTION: %r" % list(suboption[1:]))
cliechti130d1f02009-08-04 02:10:58 +00001281 else:
cliechti6a300772009-08-12 02:28:56 +00001282 if self.logger:
1283 self.logger.warning("unknown subnegotiation: %r" % (suboption,))
cliechti130d1f02009-08-04 02:10:58 +00001284
1285
1286# simple client test
cliechti8099bed2009-08-01 23:59:18 +00001287if __name__ == '__main__':
1288 import sys
1289 s = Serial('rfc2217://localhost:7000', 115200)
1290 sys.stdout.write('%s\n' % s)
1291
cliechti2b929b72009-08-02 23:49:02 +00001292 #~ s.baudrate = 1898
1293
cliechti8099bed2009-08-01 23:59:18 +00001294 sys.stdout.write("write...\n")
Chris Liechtib4cda3a2015-08-08 17:12:08 +02001295 s.write(b"hello\n")
cliechti8099bed2009-08-01 23:59:18 +00001296 s.flush()
cliechti8099bed2009-08-01 23:59:18 +00001297 sys.stdout.write("read: %s\n" % s.read(5))
1298
1299 #~ s.baudrate = 19200
1300 #~ s.databits = 7
1301 s.close()