blob: c920cbe56a6b9be76e89bac7db09def3f990ac45 [file] [log] [blame]
cliechti89b4af12002-02-12 23:24:41 +00001#!/usr/bin/env python
cliechti58b481c2009-02-16 20:42:32 +00002#
Chris Liechti3e02f702015-12-16 23:06:04 +01003# backend for serial IO for POSIX compatible systems, like Linux, OSX
cliechti89b4af12002-02-12 23:24:41 +00004#
Chris Liechti3e02f702015-12-16 23:06:04 +01005# This file is part of pySerial. https://github.com/pyserial/pyserial
Chris Liechti68340d72015-08-03 14:15:48 +02006# (C) 2001-2015 Chris Liechti <cliechti@gmx.net>
Chris Liechtifbdd8a02015-08-09 02:37:45 +02007#
8# SPDX-License-Identifier: BSD-3-Clause
cliechti89b4af12002-02-12 23:24:41 +00009#
cliechtic54b2c82008-06-21 01:59:08 +000010# parts based on code from Grant B. Edwards <grante@visi.com>:
cliechti89b4af12002-02-12 23:24:41 +000011# ftp://ftp.visi.com/users/grante/python/PosixSerial.py
cliechti53c9fd42009-07-23 23:51:51 +000012#
cliechti89b4af12002-02-12 23:24:41 +000013# references: http://www.easysw.com/~mike/serial/serial.html
14
Chris Liechti9eaa40c2016-02-12 23:32:59 +010015# pylint: disable=abstract-method
Chris Liechti33f0ec52015-08-06 16:37:21 +020016import errno
17import fcntl
Chris Liechti33f0ec52015-08-06 16:37:21 +020018import os
19import select
20import struct
21import sys
22import termios
23import time
Chris Liechti033f17c2015-08-30 21:28:04 +020024
25import serial
26from serial.serialutil import SerialBase, SerialException, to_bytes, portNotOpenError, writeTimeoutError
cliechti89b4af12002-02-12 23:24:41 +000027
cliechti89b4af12002-02-12 23:24:41 +000028
Chris Liechtid6847af2015-08-06 17:54:30 +020029class PlatformSpecificBase(object):
30 BAUDRATE_CONSTANTS = {}
cliechti89b4af12002-02-12 23:24:41 +000031
Chris Liechtid6847af2015-08-06 17:54:30 +020032 def number_to_device(self, port_number):
33 sys.stderr.write("""\
cliechti7aaead32009-07-23 14:02:41 +000034don't know how to number ttys on this system.
cliechti895e8302004-04-20 02:40:28 +000035! Use an explicit path (eg /dev/ttyS1) or send this information to
36! the author of this module:
cliechti89b4af12002-02-12 23:24:41 +000037
cliechti895e8302004-04-20 02:40:28 +000038sys.platform = %r
39os.name = %r
40serialposix.py version = %s
cliechti89b4af12002-02-12 23:24:41 +000041
42also add the device name of the serial port and where the
43counting starts for the first serial port.
44e.g. 'first serial port: /dev/ttyS0'
45and with a bit luck you can get this module running...
Chris Liechti033f17c2015-08-30 21:28:04 +020046""" % (sys.platform, os.name, serial.VERSION))
Chris Liechtid6847af2015-08-06 17:54:30 +020047 raise NotImplementedError('no number-to-device mapping defined on this platform')
48
49 def _set_special_baudrate(self, baudrate):
50 raise NotImplementedError('non-standard baudrates are not supported on this platform')
51
52 def _set_rs485_mode(self, rs485_settings):
53 raise NotImplementedError('RS485 not supported on this platform')
54
55# try to detect the OS so that a device can be selected...
56# this code block should supply a device() and set_special_baudrate() function
57# for the platform
58plat = sys.platform.lower()
59
Chris Liechtiba45c522016-02-06 23:53:23 +010060if plat[:5] == 'linux': # Linux (confirmed) # noqa
Chris Liechtid6847af2015-08-06 17:54:30 +020061 import array
62
63 # baudrate ioctls
64 TCGETS2 = 0x802C542A
65 TCSETS2 = 0x402C542B
66 BOTHER = 0o010000
67
68 # RS485 ioctls
69 TIOCGRS485 = 0x542E
70 TIOCSRS485 = 0x542F
Chris Liechti033f17c2015-08-30 21:28:04 +020071 SER_RS485_ENABLED = 0b00000001
72 SER_RS485_RTS_ON_SEND = 0b00000010
Chris Liechtid6847af2015-08-06 17:54:30 +020073 SER_RS485_RTS_AFTER_SEND = 0b00000100
Chris Liechti033f17c2015-08-30 21:28:04 +020074 SER_RS485_RX_DURING_TX = 0b00010000
Chris Liechtid6847af2015-08-06 17:54:30 +020075
76 class PlatformSpecific(PlatformSpecificBase):
77 BAUDRATE_CONSTANTS = {
78 0: 0o000000, # hang up
79 50: 0o000001,
80 75: 0o000002,
81 110: 0o000003,
82 134: 0o000004,
83 150: 0o000005,
84 200: 0o000006,
85 300: 0o000007,
86 600: 0o000010,
87 1200: 0o000011,
88 1800: 0o000012,
89 2400: 0o000013,
90 4800: 0o000014,
91 9600: 0o000015,
92 19200: 0o000016,
93 38400: 0o000017,
94 57600: 0o010001,
95 115200: 0o010002,
96 230400: 0o010003,
97 460800: 0o010004,
98 500000: 0o010005,
99 576000: 0o010006,
100 921600: 0o010007,
101 1000000: 0o010010,
102 1152000: 0o010011,
103 1500000: 0o010012,
104 2000000: 0o010013,
105 2500000: 0o010014,
106 3000000: 0o010015,
107 3500000: 0o010016,
108 4000000: 0o010017
109 }
110
111 def number_to_device(self, port_number):
112 return '/dev/ttyS%d' % (port_number,)
113
114 def _set_special_baudrate(self, baudrate):
115 # right size is 44 on x86_64, allow for some growth
116 buf = array.array('i', [0] * 64)
117 try:
118 # get serial_struct
119 fcntl.ioctl(self.fd, TCGETS2, buf)
120 # set custom speed
121 buf[2] &= ~termios.CBAUD
122 buf[2] |= BOTHER
123 buf[9] = buf[10] = baudrate
124
125 # set serial_struct
Chris Liechti033f17c2015-08-30 21:28:04 +0200126 fcntl.ioctl(self.fd, TCSETS2, buf)
Chris Liechtid6847af2015-08-06 17:54:30 +0200127 except IOError as e:
128 raise ValueError('Failed to set custom baud rate (%s): %s' % (baudrate, e))
129
130 def _set_rs485_mode(self, rs485_settings):
Chris Liechti033f17c2015-08-30 21:28:04 +0200131 buf = array.array('i', [0] * 8) # flags, delaytx, delayrx, padding
Chris Liechtid6847af2015-08-06 17:54:30 +0200132 try:
133 fcntl.ioctl(self.fd, TIOCGRS485, buf)
134 if rs485_settings is not None:
135 if rs485_settings.loopback:
136 buf[0] |= SER_RS485_RX_DURING_TX
137 else:
138 buf[0] &= ~SER_RS485_RX_DURING_TX
139 if rs485_settings.rts_level_for_tx:
140 buf[0] |= SER_RS485_RTS_ON_SEND
141 else:
142 buf[0] &= ~SER_RS485_RTS_ON_SEND
143 if rs485_settings.rts_level_for_rx:
144 buf[0] |= SER_RS485_RTS_AFTER_SEND
145 else:
146 buf[0] &= ~SER_RS485_RTS_AFTER_SEND
147 buf[1] = int(rs485_settings.delay_rts_before_send * 1000)
148 buf[2] = int(rs485_settings.delay_rts_after_send * 1000)
149 else:
150 buf[0] = 0 # clear SER_RS485_ENABLED
Chris Liechti033f17c2015-08-30 21:28:04 +0200151 fcntl.ioctl(self.fd, TIOCSRS485, buf)
Chris Liechtid6847af2015-08-06 17:54:30 +0200152 except IOError as e:
153 raise ValueError('Failed to set RS485 mode: %s' % (e,))
154
155
156elif plat == 'cygwin': # cygwin/win32 (confirmed)
157
158 class PlatformSpecific(PlatformSpecificBase):
159 BAUDRATE_CONSTANTS = {
160 128000: 0x01003,
161 256000: 0x01005,
162 500000: 0x01007,
163 576000: 0x01008,
164 921600: 0x01009,
165 1000000: 0x0100a,
166 1152000: 0x0100b,
167 1500000: 0x0100c,
168 2000000: 0x0100d,
169 2500000: 0x0100e,
170 3000000: 0x0100f
171 }
172
173 def number_to_device(self, port_number):
174 return '/dev/com%d' % (port_number + 1,)
175
176
177elif plat[:7] == 'openbsd': # OpenBSD
178 class PlatformSpecific(PlatformSpecificBase):
179 def number_to_device(self, port_number):
180 return '/dev/cua%02d' % (port_number,)
181
182elif plat[:3] == 'bsd' or plat[:7] == 'freebsd':
183 class PlatformSpecific(PlatformSpecificBase):
184 def number_to_device(self, port_number):
185 return '/dev/cuad%d' % (port_number,)
186
187elif plat[:6] == 'darwin': # OS X
188 import array
Chris Liechti033f17c2015-08-30 21:28:04 +0200189 IOSSIOSPEED = 0x80045402 # _IOW('T', 2, speed_t)
Chris Liechtid6847af2015-08-06 17:54:30 +0200190
191 class PlatformSpecific(PlatformSpecificBase):
192 def number_to_device(self, port_number):
193 return '/dev/cuad%d' % (port_number,)
194
195 osx_version = os.uname()[2].split('.')
196 # Tiger or above can support arbitrary serial speeds
197 if int(osx_version[0]) >= 8:
198 def _set_special_baudrate(self, baudrate):
199 # use IOKit-specific call to set up high speeds
200 buf = array.array('i', [baudrate])
201 fcntl.ioctl(self.fd, IOSSIOSPEED, buf, 1)
202
203
204elif plat[:6] == 'netbsd': # NetBSD 1.6 testing by Erk
205 class PlatformSpecific(PlatformSpecificBase):
206 def number_to_device(self, port_number):
207 return '/dev/dty%02d' % (port_number,)
208
209elif plat[:4] == 'irix': # IRIX (partially tested)
210 class PlatformSpecific(PlatformSpecificBase):
211 def number_to_device(self, port_number):
Chris Liechti033f17c2015-08-30 21:28:04 +0200212 return '/dev/ttyf%d' % (port_number + 1,) # XXX different device names depending on flow control
Chris Liechtid6847af2015-08-06 17:54:30 +0200213
214elif plat[:2] == 'hp': # HP-UX (not tested)
215 class PlatformSpecific(PlatformSpecificBase):
216 def number_to_device(self, port_number):
217 return '/dev/tty%dp0' % (port_number + 1,)
218
219elif plat[:5] == 'sunos': # Solaris/SunOS (confirmed)
220 class PlatformSpecific(PlatformSpecificBase):
221 def number_to_device(self, port_number):
222 return '/dev/tty%c' % (ord('a') + port_number,)
223
224elif plat[:3] == 'aix': # AIX
225 class PlatformSpecific(PlatformSpecificBase):
226 def number_to_device(self, port_number):
227 return '/dev/tty%d' % (port_number,)
228
229else:
230 class PlatformSpecific(PlatformSpecificBase):
231 pass
cliechti89b4af12002-02-12 23:24:41 +0000232
cliechti58b481c2009-02-16 20:42:32 +0000233# whats up with "aix", "beos", ....
234# they should work, just need to know the device names.
cliechti89b4af12002-02-12 23:24:41 +0000235
236
cliechti58b481c2009-02-16 20:42:32 +0000237# load some constants for later use.
Chris Liechti11465c82015-08-04 15:55:22 +0200238# try to use values from termios, use defaults from linux otherwise
Chris Liechti033f17c2015-08-30 21:28:04 +0200239TIOCMGET = getattr(termios, 'TIOCMGET', 0x5415)
240TIOCMBIS = getattr(termios, 'TIOCMBIS', 0x5416)
241TIOCMBIC = getattr(termios, 'TIOCMBIC', 0x5417)
242TIOCMSET = getattr(termios, 'TIOCMSET', 0x5418)
cliechti89b4af12002-02-12 23:24:41 +0000243
Chris Liechti033f17c2015-08-30 21:28:04 +0200244# TIOCM_LE = getattr(termios, 'TIOCM_LE', 0x001)
Chris Liechtid6847af2015-08-06 17:54:30 +0200245TIOCM_DTR = getattr(termios, 'TIOCM_DTR', 0x002)
246TIOCM_RTS = getattr(termios, 'TIOCM_RTS', 0x004)
Chris Liechti033f17c2015-08-30 21:28:04 +0200247# TIOCM_ST = getattr(termios, 'TIOCM_ST', 0x008)
248# TIOCM_SR = getattr(termios, 'TIOCM_SR', 0x010)
cliechti89b4af12002-02-12 23:24:41 +0000249
Chris Liechtid6847af2015-08-06 17:54:30 +0200250TIOCM_CTS = getattr(termios, 'TIOCM_CTS', 0x020)
251TIOCM_CAR = getattr(termios, 'TIOCM_CAR', 0x040)
252TIOCM_RNG = getattr(termios, 'TIOCM_RNG', 0x080)
253TIOCM_DSR = getattr(termios, 'TIOCM_DSR', 0x100)
Chris Liechti033f17c2015-08-30 21:28:04 +0200254TIOCM_CD = getattr(termios, 'TIOCM_CD', TIOCM_CAR)
255TIOCM_RI = getattr(termios, 'TIOCM_RI', TIOCM_RNG)
256# TIOCM_OUT1 = getattr(termios, 'TIOCM_OUT1', 0x2000)
257# TIOCM_OUT2 = getattr(termios, 'TIOCM_OUT2', 0x4000)
Chris Liechti11465c82015-08-04 15:55:22 +0200258if hasattr(termios, 'TIOCINQ'):
259 TIOCINQ = termios.TIOCINQ
cliechti28b8fd02011-12-28 21:39:42 +0000260else:
Chris Liechtid6847af2015-08-06 17:54:30 +0200261 TIOCINQ = getattr(termios, 'FIONREAD', 0x541B)
Chris Liechti033f17c2015-08-30 21:28:04 +0200262TIOCOUTQ = getattr(termios, 'TIOCOUTQ', 0x5411)
cliechti89b4af12002-02-12 23:24:41 +0000263
264TIOCM_zero_str = struct.pack('I', 0)
265TIOCM_RTS_str = struct.pack('I', TIOCM_RTS)
266TIOCM_DTR_str = struct.pack('I', TIOCM_DTR)
267
Chris Liechti033f17c2015-08-30 21:28:04 +0200268TIOCSBRK = getattr(termios, 'TIOCSBRK', 0x5427)
269TIOCCBRK = getattr(termios, 'TIOCCBRK', 0x5428)
cliechti997b63c2008-06-21 00:09:31 +0000270
Chris Liechti033f17c2015-08-30 21:28:04 +0200271CMSPAR = 0o10000000000 # Use "stick" (mark/space) parity
cliechtiaec27ab2014-07-31 22:21:24 +0000272
cliechti89b4af12002-02-12 23:24:41 +0000273
Chris Liechtief6b7b42015-08-06 22:19:26 +0200274class Serial(SerialBase, PlatformSpecific):
cliechti7d448562014-08-03 21:57:45 +0000275 """\
Chris Liechti033f17c2015-08-30 21:28:04 +0200276 Serial port class POSIX implementation. Serial port configuration is
cliechtid6bf52c2003-10-01 02:28:12 +0000277 done with termios and fcntl. Runs on Linux and many other Un*x like
cliechtif0a81d42014-08-04 14:03:53 +0000278 systems.
279 """
cliechtid6bf52c2003-10-01 02:28:12 +0000280
281 def open(self):
cliechti7d448562014-08-03 21:57:45 +0000282 """\
283 Open port with current settings. This may throw a SerialException
284 if the port cannot be opened."""
cliechtid6bf52c2003-10-01 02:28:12 +0000285 if self._port is None:
286 raise SerialException("Port must be configured before it can be used.")
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200287 if self.is_open:
cliechti02ef43a2011-03-24 23:33:12 +0000288 raise SerialException("Port is already open.")
289 self.fd = None
cliechti58b481c2009-02-16 20:42:32 +0000290 # open
cliechti4616bf12002-04-08 23:13:14 +0000291 try:
Chris Liechti033f17c2015-08-30 21:28:04 +0200292 self.fd = os.open(self.portstr, os.O_RDWR | os.O_NOCTTY | os.O_NONBLOCK)
Chris Liechti68340d72015-08-03 14:15:48 +0200293 except OSError as msg:
cliechti4616bf12002-04-08 23:13:14 +0000294 self.fd = None
cliechtiaf84daa2013-10-10 23:57:00 +0000295 raise SerialException(msg.errno, "could not open port %s: %s" % (self._port, msg))
Chris Liechti11465c82015-08-04 15:55:22 +0200296 #~ fcntl.fcntl(self.fd, fcntl.F_SETFL, 0) # set blocking
cliechti58b481c2009-02-16 20:42:32 +0000297
cliechtib2f5fc82006-10-20 00:09:07 +0000298 try:
Chris Liechti94284702015-11-15 01:21:48 +0100299 self._reconfigure_port(force_update=True)
cliechtib2f5fc82006-10-20 00:09:07 +0000300 except:
cliechti2750b832009-07-28 00:13:52 +0000301 try:
302 os.close(self.fd)
303 except:
304 # ignore any exception when closing the port
305 # also to keep original exception that happened when setting up
306 pass
cliechtib2f5fc82006-10-20 00:09:07 +0000307 self.fd = None
cliechtif0a4f0f2009-07-21 21:12:37 +0000308 raise
cliechtib2f5fc82006-10-20 00:09:07 +0000309 else:
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200310 self.is_open = True
311 if not self._dsrdtr:
Chris Liechtief1fe252015-08-27 23:25:21 +0200312 self._update_dtr_state()
Chris Liechtidc2beec2015-09-18 21:25:28 +0200313 if not self._rtscts:
Chris Liechtief1fe252015-08-27 23:25:21 +0200314 self._update_rts_state()
315 self.reset_input_buffer()
cliechti58b481c2009-02-16 20:42:32 +0000316
Chris Liechti94284702015-11-15 01:21:48 +0100317 def _reconfigure_port(self, force_update=False):
cliechtib2f5fc82006-10-20 00:09:07 +0000318 """Set communication parameters on opened port."""
cliechtic6178262004-03-22 22:04:52 +0000319 if self.fd is None:
cliechtia9a093e2010-01-02 03:05:08 +0000320 raise SerialException("Can only operate on a valid file descriptor")
cliechtie8c45422008-06-20 23:23:14 +0000321 custom_baud = None
cliechti58b481c2009-02-16 20:42:32 +0000322
cliechti2750b832009-07-28 00:13:52 +0000323 vmin = vtime = 0 # timeout is done via select
Chris Liechti518b0d32015-08-30 02:20:39 +0200324 if self._inter_byte_timeout is not None:
cliechti679bfa62008-06-20 23:58:15 +0000325 vmin = 1
Chris Liechti518b0d32015-08-30 02:20:39 +0200326 vtime = int(self._inter_byte_timeout * 10)
cliechti6ce7ab12002-11-07 02:15:00 +0000327 try:
cliechti4d0af5e2011-08-05 02:18:16 +0000328 orig_attr = termios.tcgetattr(self.fd)
329 iflag, oflag, cflag, lflag, ispeed, ospeed, cc = orig_attr
Chris Liechti68340d72015-08-03 14:15:48 +0200330 except termios.error as msg: # if a port is nonexistent but has a /dev file, it'll fail here
cliechtid6bf52c2003-10-01 02:28:12 +0000331 raise SerialException("Could not configure port: %s" % msg)
cliechti58b481c2009-02-16 20:42:32 +0000332 # set up raw mode / no echo / binary
Chris Liechti033f17c2015-08-30 21:28:04 +0200333 cflag |= (termios.CLOCAL | termios.CREAD)
334 lflag &= ~(termios.ICANON | termios.ECHO | termios.ECHOE |
335 termios.ECHOK | termios.ECHONL |
336 termios.ISIG | termios.IEXTEN) # |termios.ECHOPRT
337 for flag in ('ECHOCTL', 'ECHOKE'): # netbsd workaround for Erk
Chris Liechti11465c82015-08-04 15:55:22 +0200338 if hasattr(termios, flag):
339 lflag &= ~getattr(termios, flag)
cliechti58b481c2009-02-16 20:42:32 +0000340
Chris Liechti033f17c2015-08-30 21:28:04 +0200341 oflag &= ~(termios.OPOST | termios.ONLCR | termios.OCRNL)
342 iflag &= ~(termios.INLCR | termios.IGNCR | termios.ICRNL | termios.IGNBRK)
Chris Liechti11465c82015-08-04 15:55:22 +0200343 if hasattr(termios, 'IUCLC'):
344 iflag &= ~termios.IUCLC
345 if hasattr(termios, 'PARMRK'):
346 iflag &= ~termios.PARMRK
cliechti58b481c2009-02-16 20:42:32 +0000347
cliechtif0a4f0f2009-07-21 21:12:37 +0000348 # setup baud rate
cliechti89b4af12002-02-12 23:24:41 +0000349 try:
Chris Liechti11465c82015-08-04 15:55:22 +0200350 ispeed = ospeed = getattr(termios, 'B%s' % (self._baudrate))
cliechti895e8302004-04-20 02:40:28 +0000351 except AttributeError:
cliechtif1559d02007-11-08 23:43:58 +0000352 try:
Chris Liechtid6847af2015-08-06 17:54:30 +0200353 ispeed = ospeed = self.BAUDRATE_CONSTANTS[self._baudrate]
cliechtif1559d02007-11-08 23:43:58 +0000354 except KeyError:
cliechtie8c45422008-06-20 23:23:14 +0000355 #~ raise ValueError('Invalid baud rate: %r' % self._baudrate)
cliechtif0a4f0f2009-07-21 21:12:37 +0000356 # may need custom baud rate, it isn't in our list.
Chris Liechti11465c82015-08-04 15:55:22 +0200357 ispeed = ospeed = getattr(termios, 'B38400')
cliechtif0a4f0f2009-07-21 21:12:37 +0000358 try:
Chris Liechti033f17c2015-08-30 21:28:04 +0200359 custom_baud = int(self._baudrate) # store for later
cliechtif0a4f0f2009-07-21 21:12:37 +0000360 except ValueError:
361 raise ValueError('Invalid baud rate: %r' % self._baudrate)
362 else:
363 if custom_baud < 0:
364 raise ValueError('Invalid baud rate: %r' % self._baudrate)
cliechti58b481c2009-02-16 20:42:32 +0000365
366 # setup char len
Chris Liechti11465c82015-08-04 15:55:22 +0200367 cflag &= ~termios.CSIZE
cliechtid6bf52c2003-10-01 02:28:12 +0000368 if self._bytesize == 8:
Chris Liechti11465c82015-08-04 15:55:22 +0200369 cflag |= termios.CS8
cliechtid6bf52c2003-10-01 02:28:12 +0000370 elif self._bytesize == 7:
Chris Liechti11465c82015-08-04 15:55:22 +0200371 cflag |= termios.CS7
cliechtid6bf52c2003-10-01 02:28:12 +0000372 elif self._bytesize == 6:
Chris Liechti11465c82015-08-04 15:55:22 +0200373 cflag |= termios.CS6
cliechtid6bf52c2003-10-01 02:28:12 +0000374 elif self._bytesize == 5:
Chris Liechti11465c82015-08-04 15:55:22 +0200375 cflag |= termios.CS5
cliechti89b4af12002-02-12 23:24:41 +0000376 else:
cliechtid6bf52c2003-10-01 02:28:12 +0000377 raise ValueError('Invalid char len: %r' % self._bytesize)
cliechtif0a81d42014-08-04 14:03:53 +0000378 # setup stop bits
Chris Liechti033f17c2015-08-30 21:28:04 +0200379 if self._stopbits == serial.STOPBITS_ONE:
Chris Liechti11465c82015-08-04 15:55:22 +0200380 cflag &= ~(termios.CSTOPB)
Chris Liechti033f17c2015-08-30 21:28:04 +0200381 elif self._stopbits == serial.STOPBITS_ONE_POINT_FIVE:
382 cflag |= (termios.CSTOPB) # XXX same as TWO.. there is no POSIX support for 1.5
383 elif self._stopbits == serial.STOPBITS_TWO:
384 cflag |= (termios.CSTOPB)
cliechti89b4af12002-02-12 23:24:41 +0000385 else:
cliechti3172d3d2009-07-21 22:33:40 +0000386 raise ValueError('Invalid stop bit specification: %r' % self._stopbits)
cliechti58b481c2009-02-16 20:42:32 +0000387 # setup parity
Chris Liechti033f17c2015-08-30 21:28:04 +0200388 iflag &= ~(termios.INPCK | termios.ISTRIP)
389 if self._parity == serial.PARITY_NONE:
390 cflag &= ~(termios.PARENB | termios.PARODD)
391 elif self._parity == serial.PARITY_EVEN:
Chris Liechti11465c82015-08-04 15:55:22 +0200392 cflag &= ~(termios.PARODD)
Chris Liechti033f17c2015-08-30 21:28:04 +0200393 cflag |= (termios.PARENB)
394 elif self._parity == serial.PARITY_ODD:
395 cflag |= (termios.PARENB | termios.PARODD)
396 elif self._parity == serial.PARITY_MARK and plat[:5] == 'linux':
397 cflag |= (termios.PARENB | CMSPAR | termios.PARODD)
398 elif self._parity == serial.PARITY_SPACE and plat[:5] == 'linux':
399 cflag |= (termios.PARENB | CMSPAR)
Chris Liechti11465c82015-08-04 15:55:22 +0200400 cflag &= ~(termios.PARODD)
cliechti89b4af12002-02-12 23:24:41 +0000401 else:
cliechtid6bf52c2003-10-01 02:28:12 +0000402 raise ValueError('Invalid parity: %r' % self._parity)
cliechti58b481c2009-02-16 20:42:32 +0000403 # setup flow control
404 # xonxoff
Chris Liechti11465c82015-08-04 15:55:22 +0200405 if hasattr(termios, 'IXANY'):
cliechtid6bf52c2003-10-01 02:28:12 +0000406 if self._xonxoff:
Chris Liechti033f17c2015-08-30 21:28:04 +0200407 iflag |= (termios.IXON | termios.IXOFF) # |termios.IXANY)
cliechti89b4af12002-02-12 23:24:41 +0000408 else:
Chris Liechti033f17c2015-08-30 21:28:04 +0200409 iflag &= ~(termios.IXON | termios.IXOFF | termios.IXANY)
cliechti89b4af12002-02-12 23:24:41 +0000410 else:
cliechtid6bf52c2003-10-01 02:28:12 +0000411 if self._xonxoff:
Chris Liechti033f17c2015-08-30 21:28:04 +0200412 iflag |= (termios.IXON | termios.IXOFF)
cliechti89b4af12002-02-12 23:24:41 +0000413 else:
Chris Liechti033f17c2015-08-30 21:28:04 +0200414 iflag &= ~(termios.IXON | termios.IXOFF)
cliechti58b481c2009-02-16 20:42:32 +0000415 # rtscts
Chris Liechti11465c82015-08-04 15:55:22 +0200416 if hasattr(termios, 'CRTSCTS'):
cliechtid6bf52c2003-10-01 02:28:12 +0000417 if self._rtscts:
Chris Liechti033f17c2015-08-30 21:28:04 +0200418 cflag |= (termios.CRTSCTS)
cliechti89b4af12002-02-12 23:24:41 +0000419 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200420 cflag &= ~(termios.CRTSCTS)
421 elif hasattr(termios, 'CNEW_RTSCTS'): # try it with alternate constant name
cliechtid6bf52c2003-10-01 02:28:12 +0000422 if self._rtscts:
Chris Liechti033f17c2015-08-30 21:28:04 +0200423 cflag |= (termios.CNEW_RTSCTS)
cliechtid4743692002-04-08 22:39:53 +0000424 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200425 cflag &= ~(termios.CNEW_RTSCTS)
cliechti2750b832009-07-28 00:13:52 +0000426 # XXX should there be a warning if setting up rtscts (and xonxoff etc) fails??
cliechti58b481c2009-02-16 20:42:32 +0000427
428 # buffer
cliechtif0a81d42014-08-04 14:03:53 +0000429 # vmin "minimal number of characters to be read. 0 for non blocking"
cliechtid6bf52c2003-10-01 02:28:12 +0000430 if vmin < 0 or vmin > 255:
431 raise ValueError('Invalid vmin: %r ' % vmin)
Chris Liechti11465c82015-08-04 15:55:22 +0200432 cc[termios.VMIN] = vmin
cliechti58b481c2009-02-16 20:42:32 +0000433 # vtime
cliechtid6bf52c2003-10-01 02:28:12 +0000434 if vtime < 0 or vtime > 255:
435 raise ValueError('Invalid vtime: %r' % vtime)
Chris Liechti11465c82015-08-04 15:55:22 +0200436 cc[termios.VTIME] = vtime
cliechti58b481c2009-02-16 20:42:32 +0000437 # activate settings
Chris Liechti94284702015-11-15 01:21:48 +0100438 if force_update or [iflag, oflag, cflag, lflag, ispeed, ospeed, cc] != orig_attr:
Chris Liechti033f17c2015-08-30 21:28:04 +0200439 termios.tcsetattr(
Chris Liechti9eaa40c2016-02-12 23:32:59 +0100440 self.fd,
441 termios.TCSANOW,
442 [iflag, oflag, cflag, lflag, ispeed, ospeed, cc])
cliechti58b481c2009-02-16 20:42:32 +0000443
cliechtie8c45422008-06-20 23:23:14 +0000444 # apply custom baud rate, if any
445 if custom_baud is not None:
Chris Liechtid6847af2015-08-06 17:54:30 +0200446 self._set_special_baudrate(custom_baud)
447
448 if self._rs485_mode is not None:
449 self._set_rs485_mode(self._rs485_mode)
cliechti89b4af12002-02-12 23:24:41 +0000450
451 def close(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000452 """Close port"""
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200453 if self.is_open:
cliechtic6178262004-03-22 22:04:52 +0000454 if self.fd is not None:
cliechtid6bf52c2003-10-01 02:28:12 +0000455 os.close(self.fd)
456 self.fd = None
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200457 self.is_open = False
cliechtid6bf52c2003-10-01 02:28:12 +0000458
459 # - - - - - - - - - - - - - - - - - - - - - - - -
cliechti95c62212002-03-04 22:17:53 +0000460
Chris Liechtief1fe252015-08-27 23:25:21 +0200461 @property
462 def in_waiting(self):
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200463 """Return the number of bytes currently in the input buffer."""
Chris Liechti11465c82015-08-04 15:55:22 +0200464 #~ s = fcntl.ioctl(self.fd, termios.FIONREAD, TIOCM_zero_str)
cliechtif5831e02002-12-05 23:15:27 +0000465 s = fcntl.ioctl(self.fd, TIOCINQ, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200466 return struct.unpack('I', s)[0]
cliechti89b4af12002-02-12 23:24:41 +0000467
cliechtia9a093e2010-01-02 03:05:08 +0000468 # select based implementation, proved to work on many systems
469 def read(self, size=1):
cliechti7d448562014-08-03 21:57:45 +0000470 """\
471 Read size bytes from the serial port. If a timeout is set it may
472 return less characters as requested. With no timeout it will block
473 until the requested number of bytes is read.
474 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200475 if not self.is_open:
476 raise portNotOpenError
cliechtia9a093e2010-01-02 03:05:08 +0000477 read = bytearray()
Cristiano De Altic30622f2015-12-12 11:00:01 +0100478 timeout = self._timeout
cliechtia9a093e2010-01-02 03:05:08 +0000479 while len(read) < size:
cliechti8d744de2013-10-11 14:31:13 +0000480 try:
Cristiano De Altic30622f2015-12-12 11:00:01 +0100481 start_time = time.time()
482 ready, _, _ = select.select([self.fd], [], [], timeout)
cliechti8d744de2013-10-11 14:31:13 +0000483 # If select was used with a timeout, and the timeout occurs, it
484 # returns with empty lists -> thus abort read operation.
Chris Liechti033f17c2015-08-30 21:28:04 +0200485 # For timeout == 0 (non-blocking operation) also abort when
486 # there is nothing to read.
cliechti8d744de2013-10-11 14:31:13 +0000487 if not ready:
488 break # timeout
Chris Liechti033f17c2015-08-30 21:28:04 +0200489 buf = os.read(self.fd, size - len(read))
cliechti8d744de2013-10-11 14:31:13 +0000490 # read should always return some data as select reported it was
491 # ready to read when we get to this point.
492 if not buf:
493 # Disconnected devices, at least on Linux, show the
494 # behavior that they are always ready to read immediately
495 # but reading returns nothing.
Chris Liechti92df95a2016-02-09 23:30:37 +0100496 raise SerialException(
497 'device reports readiness to read but returned no data '
498 '(device disconnected or multiple access on port?)')
cliechti8d744de2013-10-11 14:31:13 +0000499 read.extend(buf)
Cristiano De Altic30622f2015-12-12 11:00:01 +0100500 if timeout is not None:
501 timeout -= time.time() - start_time
502 if timeout <= 0:
503 break
Chris Liechti68340d72015-08-03 14:15:48 +0200504 except OSError as e:
Chris Liechti033f17c2015-08-30 21:28:04 +0200505 # this is for Python 3.x where select.error is a subclass of
506 # OSError ignore EAGAIN errors. all other errors are shown
cliechtic7cd7212014-08-03 21:34:38 +0000507 if e.errno != errno.EAGAIN:
508 raise SerialException('read failed: %s' % (e,))
Chris Liechti68340d72015-08-03 14:15:48 +0200509 except select.error as e:
cliechtic7cd7212014-08-03 21:34:38 +0000510 # this is for Python 2.x
cliechti8d744de2013-10-11 14:31:13 +0000511 # ignore EAGAIN errors. all other errors are shown
512 # see also http://www.python.org/dev/peps/pep-3151/#select
513 if e[0] != errno.EAGAIN:
514 raise SerialException('read failed: %s' % (e,))
cliechtia9a093e2010-01-02 03:05:08 +0000515 return bytes(read)
cliechti89b4af12002-02-12 23:24:41 +0000516
cliechti4a567a02009-07-27 22:09:31 +0000517 def write(self, data):
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200518 """Output the given byte string over the serial port."""
Chris Liechti033f17c2015-08-30 21:28:04 +0200519 if not self.is_open:
520 raise portNotOpenError
cliechti38077122013-10-16 02:57:27 +0000521 d = to_bytes(data)
522 tx_len = len(d)
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200523 if self._write_timeout is not None and self._write_timeout > 0:
524 timeout = time.time() + self._write_timeout
cliechti3cf46d62009-08-07 00:19:57 +0000525 else:
526 timeout = None
cliechti9f7c2352013-10-11 01:13:46 +0000527 while tx_len > 0:
cliechti5d4d0bd2004-11-13 03:27:39 +0000528 try:
cliechti5d4d0bd2004-11-13 03:27:39 +0000529 n = os.write(self.fd, d)
cliechti3cf46d62009-08-07 00:19:57 +0000530 if timeout:
531 # when timeout is set, use select to wait for being ready
532 # with the time left as timeout
533 timeleft = timeout - time.time()
534 if timeleft < 0:
535 raise writeTimeoutError
536 _, ready, _ = select.select([], [self.fd], [], timeleft)
cliechti5d4d0bd2004-11-13 03:27:39 +0000537 if not ready:
538 raise writeTimeoutError
cliechti88c62442013-10-12 04:03:16 +0000539 else:
540 # wait for write operation
541 _, ready, _ = select.select([], [self.fd], [], None)
542 if not ready:
543 raise SerialException('write failed (select)')
cliechti5d4d0bd2004-11-13 03:27:39 +0000544 d = d[n:]
cliechti9f7c2352013-10-11 01:13:46 +0000545 tx_len -= n
Chris Liechti675f7e12015-08-03 15:48:41 +0200546 except SerialException:
547 raise
Chris Liechti68340d72015-08-03 14:15:48 +0200548 except OSError as v:
cliechti5d4d0bd2004-11-13 03:27:39 +0000549 if v.errno != errno.EAGAIN:
cliechti65722c92009-08-07 00:48:53 +0000550 raise SerialException('write failed: %s' % (v,))
Chris Liechtic6362db2015-12-13 23:44:35 +0100551 # still calculate and check timeout
552 if timeout and timeout - time.time() < 0:
553 raise writeTimeoutError
cliechtif81362e2009-07-25 03:44:33 +0000554 return len(data)
cliechtid6bf52c2003-10-01 02:28:12 +0000555
cliechtia30a8a02003-10-05 12:28:13 +0000556 def flush(self):
cliechti7d448562014-08-03 21:57:45 +0000557 """\
558 Flush of file like objects. In this case, wait until all data
559 is written.
560 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200561 if not self.is_open:
562 raise portNotOpenError
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200563 termios.tcdrain(self.fd)
cliechtia30a8a02003-10-05 12:28:13 +0000564
Chris Liechtief1fe252015-08-27 23:25:21 +0200565 def reset_input_buffer(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000566 """Clear input buffer, discarding all that is in the buffer."""
Chris Liechti033f17c2015-08-30 21:28:04 +0200567 if not self.is_open:
568 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200569 termios.tcflush(self.fd, termios.TCIFLUSH)
cliechti89b4af12002-02-12 23:24:41 +0000570
Chris Liechtief1fe252015-08-27 23:25:21 +0200571 def reset_output_buffer(self):
cliechti7d448562014-08-03 21:57:45 +0000572 """\
573 Clear output buffer, aborting the current output and discarding all
574 that is in the buffer.
575 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200576 if not self.is_open:
577 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200578 termios.tcflush(self.fd, termios.TCOFLUSH)
cliechti89b4af12002-02-12 23:24:41 +0000579
Chris Liechtief1fe252015-08-27 23:25:21 +0200580 def send_break(self, duration=0.25):
cliechti7d448562014-08-03 21:57:45 +0000581 """\
582 Send break condition. Timed, returns to idle state after given
583 duration.
584 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200585 if not self.is_open:
586 raise portNotOpenError
587 termios.tcsendbreak(self.fd, int(duration / 0.25))
cliechti89b4af12002-02-12 23:24:41 +0000588
Chris Liechtief1fe252015-08-27 23:25:21 +0200589 def _update_break_state(self):
cliechti7d448562014-08-03 21:57:45 +0000590 """\
591 Set break: Controls TXD. When active, no transmitting is possible.
592 """
Chris Liechtief1fe252015-08-27 23:25:21 +0200593 if self._break_state:
cliechti997b63c2008-06-21 00:09:31 +0000594 fcntl.ioctl(self.fd, TIOCSBRK)
595 else:
596 fcntl.ioctl(self.fd, TIOCCBRK)
597
Chris Liechtief1fe252015-08-27 23:25:21 +0200598 def _update_rts_state(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000599 """Set terminal status line: Request To Send"""
Chris Liechti4cf65392016-01-21 23:57:02 +0100600 if self._rts_state is not None:
601 if self._rts_state:
602 fcntl.ioctl(self.fd, TIOCMBIS, TIOCM_RTS_str)
603 else:
604 fcntl.ioctl(self.fd, TIOCMBIC, TIOCM_RTS_str)
cliechtid6bf52c2003-10-01 02:28:12 +0000605
Chris Liechtief1fe252015-08-27 23:25:21 +0200606 def _update_dtr_state(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000607 """Set terminal status line: Data Terminal Ready"""
Chris Liechti4cf65392016-01-21 23:57:02 +0100608 if self._dtr_state is not None:
609 if self._dtr_state:
610 fcntl.ioctl(self.fd, TIOCMBIS, TIOCM_DTR_str)
611 else:
612 fcntl.ioctl(self.fd, TIOCMBIC, TIOCM_DTR_str)
cliechtid6bf52c2003-10-01 02:28:12 +0000613
Chris Liechtief1fe252015-08-27 23:25:21 +0200614 @property
615 def cts(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000616 """Read terminal status line: Clear To Send"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200617 if not self.is_open:
618 raise portNotOpenError
cliechtid6bf52c2003-10-01 02:28:12 +0000619 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200620 return struct.unpack('I', s)[0] & TIOCM_CTS != 0
cliechtid6bf52c2003-10-01 02:28:12 +0000621
Chris Liechtief1fe252015-08-27 23:25:21 +0200622 @property
623 def dsr(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000624 """Read terminal status line: Data Set Ready"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200625 if not self.is_open:
626 raise portNotOpenError
cliechtid6bf52c2003-10-01 02:28:12 +0000627 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200628 return struct.unpack('I', s)[0] & TIOCM_DSR != 0
cliechtid6bf52c2003-10-01 02:28:12 +0000629
Chris Liechtief1fe252015-08-27 23:25:21 +0200630 @property
631 def ri(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000632 """Read terminal status line: Ring Indicator"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200633 if not self.is_open:
634 raise portNotOpenError
cliechti89b4af12002-02-12 23:24:41 +0000635 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200636 return struct.unpack('I', s)[0] & TIOCM_RI != 0
cliechti89b4af12002-02-12 23:24:41 +0000637
Chris Liechtief1fe252015-08-27 23:25:21 +0200638 @property
639 def cd(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000640 """Read terminal status line: Carrier Detect"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200641 if not self.is_open:
642 raise portNotOpenError
cliechti89b4af12002-02-12 23:24:41 +0000643 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200644 return struct.unpack('I', s)[0] & TIOCM_CD != 0
cliechti89b4af12002-02-12 23:24:41 +0000645
cliechtia30a8a02003-10-05 12:28:13 +0000646 # - - platform specific - - - -
647
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200648 @property
649 def out_waiting(self):
650 """Return the number of bytes currently in the output buffer."""
Chris Liechti11465c82015-08-04 15:55:22 +0200651 #~ s = fcntl.ioctl(self.fd, termios.FIONREAD, TIOCM_zero_str)
cliechti28b8fd02011-12-28 21:39:42 +0000652 s = fcntl.ioctl(self.fd, TIOCOUTQ, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200653 return struct.unpack('I', s)[0]
cliechti28b8fd02011-12-28 21:39:42 +0000654
cliechtia30a8a02003-10-05 12:28:13 +0000655 def nonblocking(self):
656 """internal - not portable!"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200657 if not self.is_open:
658 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200659 fcntl.fcntl(self.fd, fcntl.F_SETFL, os.O_NONBLOCK)
cliechtia30a8a02003-10-05 12:28:13 +0000660
cliechti8753bbc2005-01-15 20:32:51 +0000661 def fileno(self):
cliechti2f0f8a32011-12-28 22:10:00 +0000662 """\
663 For easier use of the serial port instance with select.
664 WARNING: this function is not portable to different platforms!
665 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200666 if not self.is_open:
667 raise portNotOpenError
cliechti8753bbc2005-01-15 20:32:51 +0000668 return self.fd
cliechti89b4af12002-02-12 23:24:41 +0000669
Chris Liechti518b0d32015-08-30 02:20:39 +0200670 def set_input_flow_control(self, enable=True):
cliechti2f0f8a32011-12-28 22:10:00 +0000671 """\
672 Manually control flow - when software flow control is enabled.
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200673 This will send XON (true) or XOFF (false) to the other device.
cliechti2f0f8a32011-12-28 22:10:00 +0000674 WARNING: this function is not portable to different platforms!
675 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200676 if not self.is_open:
677 raise portNotOpenError
cliechti4a601342011-12-29 02:22:17 +0000678 if enable:
Chris Liechti11465c82015-08-04 15:55:22 +0200679 termios.tcflow(self.fd, termios.TCION)
cliechti57e48a62009-08-03 22:29:58 +0000680 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200681 termios.tcflow(self.fd, termios.TCIOFF)
cliechti57e48a62009-08-03 22:29:58 +0000682
Chris Liechti518b0d32015-08-30 02:20:39 +0200683 def set_output_flow_control(self, enable=True):
cliechti2f0f8a32011-12-28 22:10:00 +0000684 """\
685 Manually control flow of outgoing data - when hardware or software flow
686 control is enabled.
687 WARNING: this function is not portable to different platforms!
688 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200689 if not self.is_open:
690 raise portNotOpenError
cliechti2f0f8a32011-12-28 22:10:00 +0000691 if enable:
Chris Liechti11465c82015-08-04 15:55:22 +0200692 termios.tcflow(self.fd, termios.TCOON)
cliechti2f0f8a32011-12-28 22:10:00 +0000693 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200694 termios.tcflow(self.fd, termios.TCOOFF)
cliechti2f0f8a32011-12-28 22:10:00 +0000695
cliechtif81362e2009-07-25 03:44:33 +0000696
cliechtia9a093e2010-01-02 03:05:08 +0000697class PosixPollSerial(Serial):
cliechti7d448562014-08-03 21:57:45 +0000698 """\
cliechtif0a81d42014-08-04 14:03:53 +0000699 Poll based read implementation. Not all systems support poll properly.
700 However this one has better handling of errors, such as a device
cliechti7d448562014-08-03 21:57:45 +0000701 disconnecting while it's in use (e.g. USB-serial unplugged).
702 """
cliechtia9a093e2010-01-02 03:05:08 +0000703
704 def read(self, size=1):
cliechti7d448562014-08-03 21:57:45 +0000705 """\
706 Read size bytes from the serial port. If a timeout is set it may
707 return less characters as requested. With no timeout it will block
708 until the requested number of bytes is read.
709 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200710 if self.fd is None:
711 raise portNotOpenError
cliechtia9a093e2010-01-02 03:05:08 +0000712 read = bytearray()
713 poll = select.poll()
Chris Liechti033f17c2015-08-30 21:28:04 +0200714 poll.register(self.fd, select.POLLIN | select.POLLERR | select.POLLHUP | select.POLLNVAL)
cliechtia9a093e2010-01-02 03:05:08 +0000715 if size > 0:
716 while len(read) < size:
717 # print "\tread(): size",size, "have", len(read) #debug
718 # wait until device becomes ready to read (or something fails)
Chris Liechti033f17c2015-08-30 21:28:04 +0200719 for fd, event in poll.poll(self._timeout * 1000):
720 if event & (select.POLLERR | select.POLLHUP | select.POLLNVAL):
cliechtia9a093e2010-01-02 03:05:08 +0000721 raise SerialException('device reports error (poll)')
722 # we don't care if it is select.POLLIN or timeout, that's
723 # handled below
724 buf = os.read(self.fd, size - len(read))
725 read.extend(buf)
Chris Liechti518b0d32015-08-30 02:20:39 +0200726 if ((self._timeout is not None and self._timeout >= 0) or
Chris Liechti033f17c2015-08-30 21:28:04 +0200727 (self._inter_byte_timeout is not None and self._inter_byte_timeout > 0)) and not buf:
cliechtia9a093e2010-01-02 03:05:08 +0000728 break # early abort on timeout
729 return bytes(read)
730
cliechtif81362e2009-07-25 03:44:33 +0000731
Chris Liechti4cf54702015-10-18 00:21:56 +0200732class VTIMESerial(Serial):
733 """\
734 Implement timeout using vtime of tty device instead of using select.
735 This means that no inter character timeout can be specified and that
736 the error handling is degraded.
737
738 Overall timeout is disabled when inter-character timeout is used.
739 """
740
Chris Liechti94284702015-11-15 01:21:48 +0100741 def _reconfigure_port(self, force_update=True):
Chris Liechti4cf54702015-10-18 00:21:56 +0200742 """Set communication parameters on opened port."""
743 super(VTIMESerial, self)._reconfigure_port()
Chris Liechtid6bcaaf2016-02-01 22:55:26 +0100744 fcntl.fcntl(self.fd, fcntl.F_SETFL, 0) # clear O_NONBLOCK
Chris Liechti4cf54702015-10-18 00:21:56 +0200745
746 if self._inter_byte_timeout is not None:
747 vmin = 1
748 vtime = int(self._inter_byte_timeout * 10)
749 else:
750 vmin = 0
751 vtime = int(self._timeout * 10)
752 try:
753 orig_attr = termios.tcgetattr(self.fd)
754 iflag, oflag, cflag, lflag, ispeed, ospeed, cc = orig_attr
755 except termios.error as msg: # if a port is nonexistent but has a /dev file, it'll fail here
756 raise serial.SerialException("Could not configure port: %s" % msg)
757
758 if vtime < 0 or vtime > 255:
759 raise ValueError('Invalid vtime: %r' % vtime)
760 cc[termios.VTIME] = vtime
761 cc[termios.VMIN] = vmin
762
763 termios.tcsetattr(
764 self.fd,
765 termios.TCSANOW,
766 [iflag, oflag, cflag, lflag, ispeed, ospeed, cc])
767
Chris Liechti4cf54702015-10-18 00:21:56 +0200768 def read(self, size=1):
769 """\
770 Read size bytes from the serial port. If a timeout is set it may
771 return less characters as requested. With no timeout it will block
772 until the requested number of bytes is read.
773 """
774 if not self.is_open:
775 raise portNotOpenError
776 read = bytearray()
777 while len(read) < size:
778 buf = os.read(self.fd, size - len(read))
779 if not buf:
780 break
781 read.extend(buf)
782 return bytes(read)