blob: 67a3fd4872a1b2cf431b4fc2deac367767c52a3c [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 Liechtie13d0f62016-04-25 23:30:44 +02006# (C) 2001-2016 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 Liechtie13d0f62016-04-25 23:30:44 +020015# Collection of port names (was previously used by number_to_device which was
16# removed.
17# - Linux /dev/ttyS%d (confirmed)
18# - cygwin/win32 /dev/com%d (confirmed)
19# - openbsd (OpenBSD) /dev/cua%02d
20# - bsd*, freebsd* /dev/cuad%d
21# - darwin (OS X) /dev/cuad%d
22# - netbsd /dev/dty%02d (NetBSD 1.6 testing by Erk)
23# - irix (IRIX) /dev/ttyf%d (partially tested) names depending on flow control
24# - hp (HP-UX) /dev/tty%dp0 (not tested)
25# - sunos (Solaris/SunOS) /dev/tty%c (letters, 'a'..'z') (confirmed)
26# - aix (AIX) /dev/tty%d
27
28
Chris Liechti9eaa40c2016-02-12 23:32:59 +010029# pylint: disable=abstract-method
Chris Liechti33f0ec52015-08-06 16:37:21 +020030import errno
31import fcntl
Chris Liechti33f0ec52015-08-06 16:37:21 +020032import os
33import select
34import struct
35import sys
36import termios
37import time
Chris Liechti033f17c2015-08-30 21:28:04 +020038
39import serial
40from serial.serialutil import SerialBase, SerialException, to_bytes, portNotOpenError, writeTimeoutError
cliechti89b4af12002-02-12 23:24:41 +000041
cliechti89b4af12002-02-12 23:24:41 +000042
Chris Liechtid6847af2015-08-06 17:54:30 +020043class PlatformSpecificBase(object):
44 BAUDRATE_CONSTANTS = {}
cliechti89b4af12002-02-12 23:24:41 +000045
Chris Liechtid6847af2015-08-06 17:54:30 +020046 def _set_special_baudrate(self, baudrate):
47 raise NotImplementedError('non-standard baudrates are not supported on this platform')
48
49 def _set_rs485_mode(self, rs485_settings):
50 raise NotImplementedError('RS485 not supported on this platform')
51
52# try to detect the OS so that a device can be selected...
53# this code block should supply a device() and set_special_baudrate() function
54# for the platform
55plat = sys.platform.lower()
56
Chris Liechtiba45c522016-02-06 23:53:23 +010057if plat[:5] == 'linux': # Linux (confirmed) # noqa
Chris Liechtid6847af2015-08-06 17:54:30 +020058 import array
59
60 # baudrate ioctls
61 TCGETS2 = 0x802C542A
62 TCSETS2 = 0x402C542B
63 BOTHER = 0o010000
64
65 # RS485 ioctls
66 TIOCGRS485 = 0x542E
67 TIOCSRS485 = 0x542F
Chris Liechti033f17c2015-08-30 21:28:04 +020068 SER_RS485_ENABLED = 0b00000001
69 SER_RS485_RTS_ON_SEND = 0b00000010
Chris Liechtid6847af2015-08-06 17:54:30 +020070 SER_RS485_RTS_AFTER_SEND = 0b00000100
Chris Liechti033f17c2015-08-30 21:28:04 +020071 SER_RS485_RX_DURING_TX = 0b00010000
Chris Liechtid6847af2015-08-06 17:54:30 +020072
73 class PlatformSpecific(PlatformSpecificBase):
74 BAUDRATE_CONSTANTS = {
75 0: 0o000000, # hang up
76 50: 0o000001,
77 75: 0o000002,
78 110: 0o000003,
79 134: 0o000004,
80 150: 0o000005,
81 200: 0o000006,
82 300: 0o000007,
83 600: 0o000010,
84 1200: 0o000011,
85 1800: 0o000012,
86 2400: 0o000013,
87 4800: 0o000014,
88 9600: 0o000015,
89 19200: 0o000016,
90 38400: 0o000017,
91 57600: 0o010001,
92 115200: 0o010002,
93 230400: 0o010003,
94 460800: 0o010004,
95 500000: 0o010005,
96 576000: 0o010006,
97 921600: 0o010007,
98 1000000: 0o010010,
99 1152000: 0o010011,
100 1500000: 0o010012,
101 2000000: 0o010013,
102 2500000: 0o010014,
103 3000000: 0o010015,
104 3500000: 0o010016,
105 4000000: 0o010017
106 }
107
Chris Liechtid6847af2015-08-06 17:54:30 +0200108 def _set_special_baudrate(self, baudrate):
109 # right size is 44 on x86_64, allow for some growth
110 buf = array.array('i', [0] * 64)
111 try:
112 # get serial_struct
113 fcntl.ioctl(self.fd, TCGETS2, buf)
114 # set custom speed
115 buf[2] &= ~termios.CBAUD
116 buf[2] |= BOTHER
117 buf[9] = buf[10] = baudrate
118
119 # set serial_struct
Chris Liechti033f17c2015-08-30 21:28:04 +0200120 fcntl.ioctl(self.fd, TCSETS2, buf)
Chris Liechtid6847af2015-08-06 17:54:30 +0200121 except IOError as e:
Chris Liechti984c5c52016-02-15 23:48:45 +0100122 raise ValueError('Failed to set custom baud rate ({}): {}'.format(baudrate, e))
Chris Liechtid6847af2015-08-06 17:54:30 +0200123
124 def _set_rs485_mode(self, rs485_settings):
Chris Liechti033f17c2015-08-30 21:28:04 +0200125 buf = array.array('i', [0] * 8) # flags, delaytx, delayrx, padding
Chris Liechtid6847af2015-08-06 17:54:30 +0200126 try:
127 fcntl.ioctl(self.fd, TIOCGRS485, buf)
Chris Liechti7a554462016-03-24 21:17:22 +0100128 buf[0] |= SER_RS485_ENABLED
Chris Liechtid6847af2015-08-06 17:54:30 +0200129 if rs485_settings is not None:
130 if rs485_settings.loopback:
131 buf[0] |= SER_RS485_RX_DURING_TX
132 else:
133 buf[0] &= ~SER_RS485_RX_DURING_TX
134 if rs485_settings.rts_level_for_tx:
135 buf[0] |= SER_RS485_RTS_ON_SEND
136 else:
137 buf[0] &= ~SER_RS485_RTS_ON_SEND
138 if rs485_settings.rts_level_for_rx:
139 buf[0] |= SER_RS485_RTS_AFTER_SEND
140 else:
141 buf[0] &= ~SER_RS485_RTS_AFTER_SEND
Chris Liechti01df8892016-03-17 23:01:42 +0100142 buf[1] = int(rs485_settings.delay_before_tx * 1000)
143 buf[2] = int(rs485_settings.delay_before_rx * 1000)
Chris Liechtid6847af2015-08-06 17:54:30 +0200144 else:
145 buf[0] = 0 # clear SER_RS485_ENABLED
Chris Liechti033f17c2015-08-30 21:28:04 +0200146 fcntl.ioctl(self.fd, TIOCSRS485, buf)
Chris Liechtid6847af2015-08-06 17:54:30 +0200147 except IOError as e:
Chris Liechti984c5c52016-02-15 23:48:45 +0100148 raise ValueError('Failed to set RS485 mode: {}'.format(e))
Chris Liechtid6847af2015-08-06 17:54:30 +0200149
150
151elif plat == 'cygwin': # cygwin/win32 (confirmed)
152
153 class PlatformSpecific(PlatformSpecificBase):
154 BAUDRATE_CONSTANTS = {
155 128000: 0x01003,
156 256000: 0x01005,
157 500000: 0x01007,
158 576000: 0x01008,
159 921600: 0x01009,
160 1000000: 0x0100a,
161 1152000: 0x0100b,
162 1500000: 0x0100c,
163 2000000: 0x0100d,
164 2500000: 0x0100e,
165 3000000: 0x0100f
166 }
167
Chris Liechtid6847af2015-08-06 17:54:30 +0200168
169elif plat[:6] == 'darwin': # OS X
170 import array
Chris Liechti033f17c2015-08-30 21:28:04 +0200171 IOSSIOSPEED = 0x80045402 # _IOW('T', 2, speed_t)
Chris Liechtid6847af2015-08-06 17:54:30 +0200172
173 class PlatformSpecific(PlatformSpecificBase):
Chris Liechtid6847af2015-08-06 17:54:30 +0200174 osx_version = os.uname()[2].split('.')
175 # Tiger or above can support arbitrary serial speeds
176 if int(osx_version[0]) >= 8:
177 def _set_special_baudrate(self, baudrate):
178 # use IOKit-specific call to set up high speeds
179 buf = array.array('i', [baudrate])
180 fcntl.ioctl(self.fd, IOSSIOSPEED, buf, 1)
181
Chris Liechtid6847af2015-08-06 17:54:30 +0200182else:
183 class PlatformSpecific(PlatformSpecificBase):
184 pass
cliechti89b4af12002-02-12 23:24:41 +0000185
cliechti89b4af12002-02-12 23:24:41 +0000186
cliechti58b481c2009-02-16 20:42:32 +0000187# load some constants for later use.
Chris Liechti11465c82015-08-04 15:55:22 +0200188# try to use values from termios, use defaults from linux otherwise
Chris Liechti033f17c2015-08-30 21:28:04 +0200189TIOCMGET = getattr(termios, 'TIOCMGET', 0x5415)
190TIOCMBIS = getattr(termios, 'TIOCMBIS', 0x5416)
191TIOCMBIC = getattr(termios, 'TIOCMBIC', 0x5417)
192TIOCMSET = getattr(termios, 'TIOCMSET', 0x5418)
cliechti89b4af12002-02-12 23:24:41 +0000193
Chris Liechti033f17c2015-08-30 21:28:04 +0200194# TIOCM_LE = getattr(termios, 'TIOCM_LE', 0x001)
Chris Liechtid6847af2015-08-06 17:54:30 +0200195TIOCM_DTR = getattr(termios, 'TIOCM_DTR', 0x002)
196TIOCM_RTS = getattr(termios, 'TIOCM_RTS', 0x004)
Chris Liechti033f17c2015-08-30 21:28:04 +0200197# TIOCM_ST = getattr(termios, 'TIOCM_ST', 0x008)
198# TIOCM_SR = getattr(termios, 'TIOCM_SR', 0x010)
cliechti89b4af12002-02-12 23:24:41 +0000199
Chris Liechtid6847af2015-08-06 17:54:30 +0200200TIOCM_CTS = getattr(termios, 'TIOCM_CTS', 0x020)
201TIOCM_CAR = getattr(termios, 'TIOCM_CAR', 0x040)
202TIOCM_RNG = getattr(termios, 'TIOCM_RNG', 0x080)
203TIOCM_DSR = getattr(termios, 'TIOCM_DSR', 0x100)
Chris Liechti033f17c2015-08-30 21:28:04 +0200204TIOCM_CD = getattr(termios, 'TIOCM_CD', TIOCM_CAR)
205TIOCM_RI = getattr(termios, 'TIOCM_RI', TIOCM_RNG)
206# TIOCM_OUT1 = getattr(termios, 'TIOCM_OUT1', 0x2000)
207# TIOCM_OUT2 = getattr(termios, 'TIOCM_OUT2', 0x4000)
Chris Liechti11465c82015-08-04 15:55:22 +0200208if hasattr(termios, 'TIOCINQ'):
209 TIOCINQ = termios.TIOCINQ
cliechti28b8fd02011-12-28 21:39:42 +0000210else:
Chris Liechtid6847af2015-08-06 17:54:30 +0200211 TIOCINQ = getattr(termios, 'FIONREAD', 0x541B)
Chris Liechti033f17c2015-08-30 21:28:04 +0200212TIOCOUTQ = getattr(termios, 'TIOCOUTQ', 0x5411)
cliechti89b4af12002-02-12 23:24:41 +0000213
214TIOCM_zero_str = struct.pack('I', 0)
215TIOCM_RTS_str = struct.pack('I', TIOCM_RTS)
216TIOCM_DTR_str = struct.pack('I', TIOCM_DTR)
217
Chris Liechti033f17c2015-08-30 21:28:04 +0200218TIOCSBRK = getattr(termios, 'TIOCSBRK', 0x5427)
219TIOCCBRK = getattr(termios, 'TIOCCBRK', 0x5428)
cliechti997b63c2008-06-21 00:09:31 +0000220
Chris Liechti033f17c2015-08-30 21:28:04 +0200221CMSPAR = 0o10000000000 # Use "stick" (mark/space) parity
cliechtiaec27ab2014-07-31 22:21:24 +0000222
cliechti89b4af12002-02-12 23:24:41 +0000223
Chris Liechtief6b7b42015-08-06 22:19:26 +0200224class Serial(SerialBase, PlatformSpecific):
cliechti7d448562014-08-03 21:57:45 +0000225 """\
Chris Liechti033f17c2015-08-30 21:28:04 +0200226 Serial port class POSIX implementation. Serial port configuration is
cliechtid6bf52c2003-10-01 02:28:12 +0000227 done with termios and fcntl. Runs on Linux and many other Un*x like
cliechtif0a81d42014-08-04 14:03:53 +0000228 systems.
229 """
cliechtid6bf52c2003-10-01 02:28:12 +0000230
231 def open(self):
cliechti7d448562014-08-03 21:57:45 +0000232 """\
233 Open port with current settings. This may throw a SerialException
234 if the port cannot be opened."""
cliechtid6bf52c2003-10-01 02:28:12 +0000235 if self._port is None:
236 raise SerialException("Port must be configured before it can be used.")
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200237 if self.is_open:
cliechti02ef43a2011-03-24 23:33:12 +0000238 raise SerialException("Port is already open.")
239 self.fd = None
cliechti58b481c2009-02-16 20:42:32 +0000240 # open
cliechti4616bf12002-04-08 23:13:14 +0000241 try:
Chris Liechti033f17c2015-08-30 21:28:04 +0200242 self.fd = os.open(self.portstr, os.O_RDWR | os.O_NOCTTY | os.O_NONBLOCK)
Chris Liechti68340d72015-08-03 14:15:48 +0200243 except OSError as msg:
cliechti4616bf12002-04-08 23:13:14 +0000244 self.fd = None
Chris Liechti984c5c52016-02-15 23:48:45 +0100245 raise SerialException(msg.errno, "could not open port {}: {}".format(self._port, msg))
Chris Liechti11465c82015-08-04 15:55:22 +0200246 #~ fcntl.fcntl(self.fd, fcntl.F_SETFL, 0) # set blocking
cliechti58b481c2009-02-16 20:42:32 +0000247
cliechtib2f5fc82006-10-20 00:09:07 +0000248 try:
Chris Liechti94284702015-11-15 01:21:48 +0100249 self._reconfigure_port(force_update=True)
cliechtib2f5fc82006-10-20 00:09:07 +0000250 except:
cliechti2750b832009-07-28 00:13:52 +0000251 try:
252 os.close(self.fd)
253 except:
254 # ignore any exception when closing the port
255 # also to keep original exception that happened when setting up
256 pass
cliechtib2f5fc82006-10-20 00:09:07 +0000257 self.fd = None
cliechtif0a4f0f2009-07-21 21:12:37 +0000258 raise
cliechtib2f5fc82006-10-20 00:09:07 +0000259 else:
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200260 self.is_open = True
261 if not self._dsrdtr:
Chris Liechtief1fe252015-08-27 23:25:21 +0200262 self._update_dtr_state()
Chris Liechtidc2beec2015-09-18 21:25:28 +0200263 if not self._rtscts:
Chris Liechtief1fe252015-08-27 23:25:21 +0200264 self._update_rts_state()
265 self.reset_input_buffer()
cliechti58b481c2009-02-16 20:42:32 +0000266
Chris Liechti94284702015-11-15 01:21:48 +0100267 def _reconfigure_port(self, force_update=False):
cliechtib2f5fc82006-10-20 00:09:07 +0000268 """Set communication parameters on opened port."""
cliechtic6178262004-03-22 22:04:52 +0000269 if self.fd is None:
cliechtia9a093e2010-01-02 03:05:08 +0000270 raise SerialException("Can only operate on a valid file descriptor")
cliechtie8c45422008-06-20 23:23:14 +0000271 custom_baud = None
cliechti58b481c2009-02-16 20:42:32 +0000272
cliechti2750b832009-07-28 00:13:52 +0000273 vmin = vtime = 0 # timeout is done via select
Chris Liechti518b0d32015-08-30 02:20:39 +0200274 if self._inter_byte_timeout is not None:
cliechti679bfa62008-06-20 23:58:15 +0000275 vmin = 1
Chris Liechti518b0d32015-08-30 02:20:39 +0200276 vtime = int(self._inter_byte_timeout * 10)
cliechti6ce7ab12002-11-07 02:15:00 +0000277 try:
cliechti4d0af5e2011-08-05 02:18:16 +0000278 orig_attr = termios.tcgetattr(self.fd)
279 iflag, oflag, cflag, lflag, ispeed, ospeed, cc = orig_attr
Chris Liechti68340d72015-08-03 14:15:48 +0200280 except termios.error as msg: # if a port is nonexistent but has a /dev file, it'll fail here
Chris Liechti984c5c52016-02-15 23:48:45 +0100281 raise SerialException("Could not configure port: {}".format(msg))
cliechti58b481c2009-02-16 20:42:32 +0000282 # set up raw mode / no echo / binary
Chris Liechti033f17c2015-08-30 21:28:04 +0200283 cflag |= (termios.CLOCAL | termios.CREAD)
284 lflag &= ~(termios.ICANON | termios.ECHO | termios.ECHOE |
285 termios.ECHOK | termios.ECHONL |
286 termios.ISIG | termios.IEXTEN) # |termios.ECHOPRT
287 for flag in ('ECHOCTL', 'ECHOKE'): # netbsd workaround for Erk
Chris Liechti11465c82015-08-04 15:55:22 +0200288 if hasattr(termios, flag):
289 lflag &= ~getattr(termios, flag)
cliechti58b481c2009-02-16 20:42:32 +0000290
Chris Liechti033f17c2015-08-30 21:28:04 +0200291 oflag &= ~(termios.OPOST | termios.ONLCR | termios.OCRNL)
292 iflag &= ~(termios.INLCR | termios.IGNCR | termios.ICRNL | termios.IGNBRK)
Chris Liechti11465c82015-08-04 15:55:22 +0200293 if hasattr(termios, 'IUCLC'):
294 iflag &= ~termios.IUCLC
295 if hasattr(termios, 'PARMRK'):
296 iflag &= ~termios.PARMRK
cliechti58b481c2009-02-16 20:42:32 +0000297
cliechtif0a4f0f2009-07-21 21:12:37 +0000298 # setup baud rate
cliechti89b4af12002-02-12 23:24:41 +0000299 try:
Chris Liechti984c5c52016-02-15 23:48:45 +0100300 ispeed = ospeed = getattr(termios, 'B{}'.format(self._baudrate))
cliechti895e8302004-04-20 02:40:28 +0000301 except AttributeError:
cliechtif1559d02007-11-08 23:43:58 +0000302 try:
Chris Liechtid6847af2015-08-06 17:54:30 +0200303 ispeed = ospeed = self.BAUDRATE_CONSTANTS[self._baudrate]
cliechtif1559d02007-11-08 23:43:58 +0000304 except KeyError:
cliechtie8c45422008-06-20 23:23:14 +0000305 #~ raise ValueError('Invalid baud rate: %r' % self._baudrate)
cliechtif0a4f0f2009-07-21 21:12:37 +0000306 # may need custom baud rate, it isn't in our list.
Chris Liechti11465c82015-08-04 15:55:22 +0200307 ispeed = ospeed = getattr(termios, 'B38400')
cliechtif0a4f0f2009-07-21 21:12:37 +0000308 try:
Chris Liechti033f17c2015-08-30 21:28:04 +0200309 custom_baud = int(self._baudrate) # store for later
cliechtif0a4f0f2009-07-21 21:12:37 +0000310 except ValueError:
Chris Liechti984c5c52016-02-15 23:48:45 +0100311 raise ValueError('Invalid baud rate: {!r}'.format(self._baudrate))
cliechtif0a4f0f2009-07-21 21:12:37 +0000312 else:
313 if custom_baud < 0:
Chris Liechti984c5c52016-02-15 23:48:45 +0100314 raise ValueError('Invalid baud rate: {!r}'.format(self._baudrate))
cliechti58b481c2009-02-16 20:42:32 +0000315
316 # setup char len
Chris Liechti11465c82015-08-04 15:55:22 +0200317 cflag &= ~termios.CSIZE
cliechtid6bf52c2003-10-01 02:28:12 +0000318 if self._bytesize == 8:
Chris Liechti11465c82015-08-04 15:55:22 +0200319 cflag |= termios.CS8
cliechtid6bf52c2003-10-01 02:28:12 +0000320 elif self._bytesize == 7:
Chris Liechti11465c82015-08-04 15:55:22 +0200321 cflag |= termios.CS7
cliechtid6bf52c2003-10-01 02:28:12 +0000322 elif self._bytesize == 6:
Chris Liechti11465c82015-08-04 15:55:22 +0200323 cflag |= termios.CS6
cliechtid6bf52c2003-10-01 02:28:12 +0000324 elif self._bytesize == 5:
Chris Liechti11465c82015-08-04 15:55:22 +0200325 cflag |= termios.CS5
cliechti89b4af12002-02-12 23:24:41 +0000326 else:
Chris Liechti984c5c52016-02-15 23:48:45 +0100327 raise ValueError('Invalid char len: {!r}'.format(self._bytesize))
cliechtif0a81d42014-08-04 14:03:53 +0000328 # setup stop bits
Chris Liechti033f17c2015-08-30 21:28:04 +0200329 if self._stopbits == serial.STOPBITS_ONE:
Chris Liechti11465c82015-08-04 15:55:22 +0200330 cflag &= ~(termios.CSTOPB)
Chris Liechti033f17c2015-08-30 21:28:04 +0200331 elif self._stopbits == serial.STOPBITS_ONE_POINT_FIVE:
332 cflag |= (termios.CSTOPB) # XXX same as TWO.. there is no POSIX support for 1.5
333 elif self._stopbits == serial.STOPBITS_TWO:
334 cflag |= (termios.CSTOPB)
cliechti89b4af12002-02-12 23:24:41 +0000335 else:
Chris Liechti984c5c52016-02-15 23:48:45 +0100336 raise ValueError('Invalid stop bit specification: {!r}'.format(self._stopbits))
cliechti58b481c2009-02-16 20:42:32 +0000337 # setup parity
Chris Liechti033f17c2015-08-30 21:28:04 +0200338 iflag &= ~(termios.INPCK | termios.ISTRIP)
339 if self._parity == serial.PARITY_NONE:
340 cflag &= ~(termios.PARENB | termios.PARODD)
341 elif self._parity == serial.PARITY_EVEN:
Chris Liechti11465c82015-08-04 15:55:22 +0200342 cflag &= ~(termios.PARODD)
Chris Liechti033f17c2015-08-30 21:28:04 +0200343 cflag |= (termios.PARENB)
344 elif self._parity == serial.PARITY_ODD:
345 cflag |= (termios.PARENB | termios.PARODD)
346 elif self._parity == serial.PARITY_MARK and plat[:5] == 'linux':
347 cflag |= (termios.PARENB | CMSPAR | termios.PARODD)
348 elif self._parity == serial.PARITY_SPACE and plat[:5] == 'linux':
349 cflag |= (termios.PARENB | CMSPAR)
Chris Liechti11465c82015-08-04 15:55:22 +0200350 cflag &= ~(termios.PARODD)
cliechti89b4af12002-02-12 23:24:41 +0000351 else:
Chris Liechti984c5c52016-02-15 23:48:45 +0100352 raise ValueError('Invalid parity: {!r}'.format(self._parity))
cliechti58b481c2009-02-16 20:42:32 +0000353 # setup flow control
354 # xonxoff
Chris Liechti11465c82015-08-04 15:55:22 +0200355 if hasattr(termios, 'IXANY'):
cliechtid6bf52c2003-10-01 02:28:12 +0000356 if self._xonxoff:
Chris Liechti033f17c2015-08-30 21:28:04 +0200357 iflag |= (termios.IXON | termios.IXOFF) # |termios.IXANY)
cliechti89b4af12002-02-12 23:24:41 +0000358 else:
Chris Liechti033f17c2015-08-30 21:28:04 +0200359 iflag &= ~(termios.IXON | termios.IXOFF | termios.IXANY)
cliechti89b4af12002-02-12 23:24:41 +0000360 else:
cliechtid6bf52c2003-10-01 02:28:12 +0000361 if self._xonxoff:
Chris Liechti033f17c2015-08-30 21:28:04 +0200362 iflag |= (termios.IXON | termios.IXOFF)
cliechti89b4af12002-02-12 23:24:41 +0000363 else:
Chris Liechti033f17c2015-08-30 21:28:04 +0200364 iflag &= ~(termios.IXON | termios.IXOFF)
cliechti58b481c2009-02-16 20:42:32 +0000365 # rtscts
Chris Liechti11465c82015-08-04 15:55:22 +0200366 if hasattr(termios, 'CRTSCTS'):
cliechtid6bf52c2003-10-01 02:28:12 +0000367 if self._rtscts:
Chris Liechti033f17c2015-08-30 21:28:04 +0200368 cflag |= (termios.CRTSCTS)
cliechti89b4af12002-02-12 23:24:41 +0000369 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200370 cflag &= ~(termios.CRTSCTS)
371 elif hasattr(termios, 'CNEW_RTSCTS'): # try it with alternate constant name
cliechtid6bf52c2003-10-01 02:28:12 +0000372 if self._rtscts:
Chris Liechti033f17c2015-08-30 21:28:04 +0200373 cflag |= (termios.CNEW_RTSCTS)
cliechtid4743692002-04-08 22:39:53 +0000374 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200375 cflag &= ~(termios.CNEW_RTSCTS)
cliechti2750b832009-07-28 00:13:52 +0000376 # XXX should there be a warning if setting up rtscts (and xonxoff etc) fails??
cliechti58b481c2009-02-16 20:42:32 +0000377
378 # buffer
cliechtif0a81d42014-08-04 14:03:53 +0000379 # vmin "minimal number of characters to be read. 0 for non blocking"
cliechtid6bf52c2003-10-01 02:28:12 +0000380 if vmin < 0 or vmin > 255:
Chris Liechti984c5c52016-02-15 23:48:45 +0100381 raise ValueError('Invalid vmin: {!r}'.format(vmin))
Chris Liechti11465c82015-08-04 15:55:22 +0200382 cc[termios.VMIN] = vmin
cliechti58b481c2009-02-16 20:42:32 +0000383 # vtime
cliechtid6bf52c2003-10-01 02:28:12 +0000384 if vtime < 0 or vtime > 255:
Chris Liechti984c5c52016-02-15 23:48:45 +0100385 raise ValueError('Invalid vtime: {!r}'.format(vtime))
Chris Liechti11465c82015-08-04 15:55:22 +0200386 cc[termios.VTIME] = vtime
cliechti58b481c2009-02-16 20:42:32 +0000387 # activate settings
Chris Liechti94284702015-11-15 01:21:48 +0100388 if force_update or [iflag, oflag, cflag, lflag, ispeed, ospeed, cc] != orig_attr:
Chris Liechti033f17c2015-08-30 21:28:04 +0200389 termios.tcsetattr(
Chris Liechti9eaa40c2016-02-12 23:32:59 +0100390 self.fd,
391 termios.TCSANOW,
392 [iflag, oflag, cflag, lflag, ispeed, ospeed, cc])
cliechti58b481c2009-02-16 20:42:32 +0000393
cliechtie8c45422008-06-20 23:23:14 +0000394 # apply custom baud rate, if any
395 if custom_baud is not None:
Chris Liechtid6847af2015-08-06 17:54:30 +0200396 self._set_special_baudrate(custom_baud)
397
398 if self._rs485_mode is not None:
399 self._set_rs485_mode(self._rs485_mode)
cliechti89b4af12002-02-12 23:24:41 +0000400
401 def close(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000402 """Close port"""
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200403 if self.is_open:
cliechtic6178262004-03-22 22:04:52 +0000404 if self.fd is not None:
cliechtid6bf52c2003-10-01 02:28:12 +0000405 os.close(self.fd)
406 self.fd = None
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200407 self.is_open = False
cliechtid6bf52c2003-10-01 02:28:12 +0000408
409 # - - - - - - - - - - - - - - - - - - - - - - - -
cliechti95c62212002-03-04 22:17:53 +0000410
Chris Liechtief1fe252015-08-27 23:25:21 +0200411 @property
412 def in_waiting(self):
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200413 """Return the number of bytes currently in the input buffer."""
Chris Liechti11465c82015-08-04 15:55:22 +0200414 #~ s = fcntl.ioctl(self.fd, termios.FIONREAD, TIOCM_zero_str)
cliechtif5831e02002-12-05 23:15:27 +0000415 s = fcntl.ioctl(self.fd, TIOCINQ, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200416 return struct.unpack('I', s)[0]
cliechti89b4af12002-02-12 23:24:41 +0000417
cliechtia9a093e2010-01-02 03:05:08 +0000418 # select based implementation, proved to work on many systems
419 def read(self, size=1):
cliechti7d448562014-08-03 21:57:45 +0000420 """\
421 Read size bytes from the serial port. If a timeout is set it may
422 return less characters as requested. With no timeout it will block
423 until the requested number of bytes is read.
424 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200425 if not self.is_open:
426 raise portNotOpenError
cliechtia9a093e2010-01-02 03:05:08 +0000427 read = bytearray()
Cristiano De Altic30622f2015-12-12 11:00:01 +0100428 timeout = self._timeout
cliechtia9a093e2010-01-02 03:05:08 +0000429 while len(read) < size:
cliechti8d744de2013-10-11 14:31:13 +0000430 try:
Cristiano De Altic30622f2015-12-12 11:00:01 +0100431 start_time = time.time()
432 ready, _, _ = select.select([self.fd], [], [], timeout)
cliechti8d744de2013-10-11 14:31:13 +0000433 # If select was used with a timeout, and the timeout occurs, it
434 # returns with empty lists -> thus abort read operation.
Chris Liechti033f17c2015-08-30 21:28:04 +0200435 # For timeout == 0 (non-blocking operation) also abort when
436 # there is nothing to read.
cliechti8d744de2013-10-11 14:31:13 +0000437 if not ready:
438 break # timeout
Chris Liechti033f17c2015-08-30 21:28:04 +0200439 buf = os.read(self.fd, size - len(read))
cliechti8d744de2013-10-11 14:31:13 +0000440 # read should always return some data as select reported it was
441 # ready to read when we get to this point.
442 if not buf:
443 # Disconnected devices, at least on Linux, show the
444 # behavior that they are always ready to read immediately
445 # but reading returns nothing.
Chris Liechti92df95a2016-02-09 23:30:37 +0100446 raise SerialException(
447 'device reports readiness to read but returned no data '
448 '(device disconnected or multiple access on port?)')
cliechti8d744de2013-10-11 14:31:13 +0000449 read.extend(buf)
Cristiano De Altic30622f2015-12-12 11:00:01 +0100450 if timeout is not None:
451 timeout -= time.time() - start_time
452 if timeout <= 0:
453 break
Chris Liechti68340d72015-08-03 14:15:48 +0200454 except OSError as e:
Chris Liechti033f17c2015-08-30 21:28:04 +0200455 # this is for Python 3.x where select.error is a subclass of
456 # OSError ignore EAGAIN errors. all other errors are shown
cliechtic7cd7212014-08-03 21:34:38 +0000457 if e.errno != errno.EAGAIN:
Chris Liechti984c5c52016-02-15 23:48:45 +0100458 raise SerialException('read failed: {}'.format(e))
Chris Liechti68340d72015-08-03 14:15:48 +0200459 except select.error as e:
cliechtic7cd7212014-08-03 21:34:38 +0000460 # this is for Python 2.x
cliechti8d744de2013-10-11 14:31:13 +0000461 # ignore EAGAIN errors. all other errors are shown
462 # see also http://www.python.org/dev/peps/pep-3151/#select
463 if e[0] != errno.EAGAIN:
Chris Liechti984c5c52016-02-15 23:48:45 +0100464 raise SerialException('read failed: {}'.format(e))
cliechtia9a093e2010-01-02 03:05:08 +0000465 return bytes(read)
cliechti89b4af12002-02-12 23:24:41 +0000466
cliechti4a567a02009-07-27 22:09:31 +0000467 def write(self, data):
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200468 """Output the given byte string over the serial port."""
Chris Liechti033f17c2015-08-30 21:28:04 +0200469 if not self.is_open:
470 raise portNotOpenError
cliechti38077122013-10-16 02:57:27 +0000471 d = to_bytes(data)
472 tx_len = len(d)
Robert Smallshire325a7382016-03-25 21:18:38 +0100473 timeout = self._write_timeout
474 if timeout and timeout > 0: # Avoid comparing None with zero
475 timeout += time.time()
cliechti9f7c2352013-10-11 01:13:46 +0000476 while tx_len > 0:
cliechti5d4d0bd2004-11-13 03:27:39 +0000477 try:
cliechti5d4d0bd2004-11-13 03:27:39 +0000478 n = os.write(self.fd, d)
Robert Smallshire325a7382016-03-25 21:18:38 +0100479 if timeout == 0:
480 # Zero timeout indicates non-blocking - simply return the
481 # number of bytes of data actually written
482 return n
483 elif timeout and timeout > 0: # Avoid comparing None with zero
cliechti3cf46d62009-08-07 00:19:57 +0000484 # when timeout is set, use select to wait for being ready
485 # with the time left as timeout
486 timeleft = timeout - time.time()
487 if timeleft < 0:
488 raise writeTimeoutError
489 _, ready, _ = select.select([], [self.fd], [], timeleft)
cliechti5d4d0bd2004-11-13 03:27:39 +0000490 if not ready:
491 raise writeTimeoutError
cliechti88c62442013-10-12 04:03:16 +0000492 else:
Robert Smallshire325a7382016-03-25 21:18:38 +0100493 assert timeout is None
cliechti88c62442013-10-12 04:03:16 +0000494 # wait for write operation
495 _, ready, _ = select.select([], [self.fd], [], None)
496 if not ready:
497 raise SerialException('write failed (select)')
cliechti5d4d0bd2004-11-13 03:27:39 +0000498 d = d[n:]
cliechti9f7c2352013-10-11 01:13:46 +0000499 tx_len -= n
Chris Liechti675f7e12015-08-03 15:48:41 +0200500 except SerialException:
501 raise
Chris Liechti68340d72015-08-03 14:15:48 +0200502 except OSError as v:
cliechti5d4d0bd2004-11-13 03:27:39 +0000503 if v.errno != errno.EAGAIN:
Chris Liechti984c5c52016-02-15 23:48:45 +0100504 raise SerialException('write failed: {}'.format(v))
Chris Liechtic6362db2015-12-13 23:44:35 +0100505 # still calculate and check timeout
506 if timeout and timeout - time.time() < 0:
507 raise writeTimeoutError
cliechtif81362e2009-07-25 03:44:33 +0000508 return len(data)
cliechtid6bf52c2003-10-01 02:28:12 +0000509
cliechtia30a8a02003-10-05 12:28:13 +0000510 def flush(self):
cliechti7d448562014-08-03 21:57:45 +0000511 """\
512 Flush of file like objects. In this case, wait until all data
513 is written.
514 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200515 if not self.is_open:
516 raise portNotOpenError
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200517 termios.tcdrain(self.fd)
cliechtia30a8a02003-10-05 12:28:13 +0000518
Chris Liechtief1fe252015-08-27 23:25:21 +0200519 def reset_input_buffer(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000520 """Clear input buffer, discarding all that is in the buffer."""
Chris Liechti033f17c2015-08-30 21:28:04 +0200521 if not self.is_open:
522 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200523 termios.tcflush(self.fd, termios.TCIFLUSH)
cliechti89b4af12002-02-12 23:24:41 +0000524
Chris Liechtief1fe252015-08-27 23:25:21 +0200525 def reset_output_buffer(self):
cliechti7d448562014-08-03 21:57:45 +0000526 """\
527 Clear output buffer, aborting the current output and discarding all
528 that is in the buffer.
529 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200530 if not self.is_open:
531 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200532 termios.tcflush(self.fd, termios.TCOFLUSH)
cliechti89b4af12002-02-12 23:24:41 +0000533
Chris Liechtief1fe252015-08-27 23:25:21 +0200534 def send_break(self, duration=0.25):
cliechti7d448562014-08-03 21:57:45 +0000535 """\
536 Send break condition. Timed, returns to idle state after given
537 duration.
538 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200539 if not self.is_open:
540 raise portNotOpenError
541 termios.tcsendbreak(self.fd, int(duration / 0.25))
cliechti89b4af12002-02-12 23:24:41 +0000542
Chris Liechtief1fe252015-08-27 23:25:21 +0200543 def _update_break_state(self):
cliechti7d448562014-08-03 21:57:45 +0000544 """\
545 Set break: Controls TXD. When active, no transmitting is possible.
546 """
Chris Liechtief1fe252015-08-27 23:25:21 +0200547 if self._break_state:
cliechti997b63c2008-06-21 00:09:31 +0000548 fcntl.ioctl(self.fd, TIOCSBRK)
549 else:
550 fcntl.ioctl(self.fd, TIOCCBRK)
551
Chris Liechtief1fe252015-08-27 23:25:21 +0200552 def _update_rts_state(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000553 """Set terminal status line: Request To Send"""
Chris Liechtif7534c82016-05-07 23:35:54 +0200554 if self._rts_state:
555 fcntl.ioctl(self.fd, TIOCMBIS, TIOCM_RTS_str)
556 else:
557 fcntl.ioctl(self.fd, TIOCMBIC, TIOCM_RTS_str)
cliechtid6bf52c2003-10-01 02:28:12 +0000558
Chris Liechtief1fe252015-08-27 23:25:21 +0200559 def _update_dtr_state(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000560 """Set terminal status line: Data Terminal Ready"""
Chris Liechtif7534c82016-05-07 23:35:54 +0200561 if self._dtr_state:
562 fcntl.ioctl(self.fd, TIOCMBIS, TIOCM_DTR_str)
563 else:
564 fcntl.ioctl(self.fd, TIOCMBIC, TIOCM_DTR_str)
cliechtid6bf52c2003-10-01 02:28:12 +0000565
Chris Liechtief1fe252015-08-27 23:25:21 +0200566 @property
567 def cts(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000568 """Read terminal status line: Clear To Send"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200569 if not self.is_open:
570 raise portNotOpenError
cliechtid6bf52c2003-10-01 02:28:12 +0000571 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200572 return struct.unpack('I', s)[0] & TIOCM_CTS != 0
cliechtid6bf52c2003-10-01 02:28:12 +0000573
Chris Liechtief1fe252015-08-27 23:25:21 +0200574 @property
575 def dsr(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000576 """Read terminal status line: Data Set Ready"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200577 if not self.is_open:
578 raise portNotOpenError
cliechtid6bf52c2003-10-01 02:28:12 +0000579 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200580 return struct.unpack('I', s)[0] & TIOCM_DSR != 0
cliechtid6bf52c2003-10-01 02:28:12 +0000581
Chris Liechtief1fe252015-08-27 23:25:21 +0200582 @property
583 def ri(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000584 """Read terminal status line: Ring Indicator"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200585 if not self.is_open:
586 raise portNotOpenError
cliechti89b4af12002-02-12 23:24:41 +0000587 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200588 return struct.unpack('I', s)[0] & TIOCM_RI != 0
cliechti89b4af12002-02-12 23:24:41 +0000589
Chris Liechtief1fe252015-08-27 23:25:21 +0200590 @property
591 def cd(self):
cliechtid6bf52c2003-10-01 02:28:12 +0000592 """Read terminal status line: Carrier Detect"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200593 if not self.is_open:
594 raise portNotOpenError
cliechti89b4af12002-02-12 23:24:41 +0000595 s = fcntl.ioctl(self.fd, TIOCMGET, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200596 return struct.unpack('I', s)[0] & TIOCM_CD != 0
cliechti89b4af12002-02-12 23:24:41 +0000597
cliechtia30a8a02003-10-05 12:28:13 +0000598 # - - platform specific - - - -
599
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200600 @property
601 def out_waiting(self):
602 """Return the number of bytes currently in the output buffer."""
Chris Liechti11465c82015-08-04 15:55:22 +0200603 #~ s = fcntl.ioctl(self.fd, termios.FIONREAD, TIOCM_zero_str)
cliechti28b8fd02011-12-28 21:39:42 +0000604 s = fcntl.ioctl(self.fd, TIOCOUTQ, TIOCM_zero_str)
Chris Liechti033f17c2015-08-30 21:28:04 +0200605 return struct.unpack('I', s)[0]
cliechti28b8fd02011-12-28 21:39:42 +0000606
cliechtia30a8a02003-10-05 12:28:13 +0000607 def nonblocking(self):
608 """internal - not portable!"""
Chris Liechti033f17c2015-08-30 21:28:04 +0200609 if not self.is_open:
610 raise portNotOpenError
Chris Liechti11465c82015-08-04 15:55:22 +0200611 fcntl.fcntl(self.fd, fcntl.F_SETFL, os.O_NONBLOCK)
cliechtia30a8a02003-10-05 12:28:13 +0000612
cliechti8753bbc2005-01-15 20:32:51 +0000613 def fileno(self):
cliechti2f0f8a32011-12-28 22:10:00 +0000614 """\
615 For easier use of the serial port instance with select.
616 WARNING: this function is not portable to different platforms!
617 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200618 if not self.is_open:
619 raise portNotOpenError
cliechti8753bbc2005-01-15 20:32:51 +0000620 return self.fd
cliechti89b4af12002-02-12 23:24:41 +0000621
Chris Liechti518b0d32015-08-30 02:20:39 +0200622 def set_input_flow_control(self, enable=True):
cliechti2f0f8a32011-12-28 22:10:00 +0000623 """\
624 Manually control flow - when software flow control is enabled.
Chris Liechti3ad62fb2015-08-29 21:53:32 +0200625 This will send XON (true) or XOFF (false) to the other device.
cliechti2f0f8a32011-12-28 22:10:00 +0000626 WARNING: this function is not portable to different platforms!
627 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200628 if not self.is_open:
629 raise portNotOpenError
cliechti4a601342011-12-29 02:22:17 +0000630 if enable:
Chris Liechti11465c82015-08-04 15:55:22 +0200631 termios.tcflow(self.fd, termios.TCION)
cliechti57e48a62009-08-03 22:29:58 +0000632 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200633 termios.tcflow(self.fd, termios.TCIOFF)
cliechti57e48a62009-08-03 22:29:58 +0000634
Chris Liechti518b0d32015-08-30 02:20:39 +0200635 def set_output_flow_control(self, enable=True):
cliechti2f0f8a32011-12-28 22:10:00 +0000636 """\
637 Manually control flow of outgoing data - when hardware or software flow
638 control is enabled.
639 WARNING: this function is not portable to different platforms!
640 """
Chris Liechti033f17c2015-08-30 21:28:04 +0200641 if not self.is_open:
642 raise portNotOpenError
cliechti2f0f8a32011-12-28 22:10:00 +0000643 if enable:
Chris Liechti11465c82015-08-04 15:55:22 +0200644 termios.tcflow(self.fd, termios.TCOON)
cliechti2f0f8a32011-12-28 22:10:00 +0000645 else:
Chris Liechti11465c82015-08-04 15:55:22 +0200646 termios.tcflow(self.fd, termios.TCOOFF)
cliechti2f0f8a32011-12-28 22:10:00 +0000647
cliechtif81362e2009-07-25 03:44:33 +0000648
cliechtia9a093e2010-01-02 03:05:08 +0000649class PosixPollSerial(Serial):
cliechti7d448562014-08-03 21:57:45 +0000650 """\
cliechtif0a81d42014-08-04 14:03:53 +0000651 Poll based read implementation. Not all systems support poll properly.
652 However this one has better handling of errors, such as a device
cliechti7d448562014-08-03 21:57:45 +0000653 disconnecting while it's in use (e.g. USB-serial unplugged).
654 """
cliechtia9a093e2010-01-02 03:05:08 +0000655
656 def read(self, size=1):
cliechti7d448562014-08-03 21:57:45 +0000657 """\
658 Read size bytes from the serial port. If a timeout is set it may
659 return less characters as requested. With no timeout it will block
660 until the requested number of bytes is read.
661 """
Chris Liechtiacac2362016-03-29 22:37:48 +0200662 if not self.is_open:
Chris Liechti033f17c2015-08-30 21:28:04 +0200663 raise portNotOpenError
cliechtia9a093e2010-01-02 03:05:08 +0000664 read = bytearray()
665 poll = select.poll()
Chris Liechti033f17c2015-08-30 21:28:04 +0200666 poll.register(self.fd, select.POLLIN | select.POLLERR | select.POLLHUP | select.POLLNVAL)
cliechtia9a093e2010-01-02 03:05:08 +0000667 if size > 0:
668 while len(read) < size:
669 # print "\tread(): size",size, "have", len(read) #debug
670 # wait until device becomes ready to read (or something fails)
Chris Liechti033f17c2015-08-30 21:28:04 +0200671 for fd, event in poll.poll(self._timeout * 1000):
672 if event & (select.POLLERR | select.POLLHUP | select.POLLNVAL):
cliechtia9a093e2010-01-02 03:05:08 +0000673 raise SerialException('device reports error (poll)')
674 # we don't care if it is select.POLLIN or timeout, that's
675 # handled below
676 buf = os.read(self.fd, size - len(read))
677 read.extend(buf)
Chris Liechti518b0d32015-08-30 02:20:39 +0200678 if ((self._timeout is not None and self._timeout >= 0) or
Chris Liechti033f17c2015-08-30 21:28:04 +0200679 (self._inter_byte_timeout is not None and self._inter_byte_timeout > 0)) and not buf:
cliechtia9a093e2010-01-02 03:05:08 +0000680 break # early abort on timeout
681 return bytes(read)
682
cliechtif81362e2009-07-25 03:44:33 +0000683
Chris Liechti4cf54702015-10-18 00:21:56 +0200684class VTIMESerial(Serial):
685 """\
686 Implement timeout using vtime of tty device instead of using select.
687 This means that no inter character timeout can be specified and that
688 the error handling is degraded.
689
690 Overall timeout is disabled when inter-character timeout is used.
691 """
692
Chris Liechti94284702015-11-15 01:21:48 +0100693 def _reconfigure_port(self, force_update=True):
Chris Liechti4cf54702015-10-18 00:21:56 +0200694 """Set communication parameters on opened port."""
695 super(VTIMESerial, self)._reconfigure_port()
Chris Liechtid6bcaaf2016-02-01 22:55:26 +0100696 fcntl.fcntl(self.fd, fcntl.F_SETFL, 0) # clear O_NONBLOCK
Chris Liechti4cf54702015-10-18 00:21:56 +0200697
698 if self._inter_byte_timeout is not None:
699 vmin = 1
700 vtime = int(self._inter_byte_timeout * 10)
701 else:
702 vmin = 0
703 vtime = int(self._timeout * 10)
704 try:
705 orig_attr = termios.tcgetattr(self.fd)
706 iflag, oflag, cflag, lflag, ispeed, ospeed, cc = orig_attr
707 except termios.error as msg: # if a port is nonexistent but has a /dev file, it'll fail here
Chris Liechti984c5c52016-02-15 23:48:45 +0100708 raise serial.SerialException("Could not configure port: {}".format(msg))
Chris Liechti4cf54702015-10-18 00:21:56 +0200709
710 if vtime < 0 or vtime > 255:
Chris Liechti984c5c52016-02-15 23:48:45 +0100711 raise ValueError('Invalid vtime: {!r}'.format(vtime))
Chris Liechti4cf54702015-10-18 00:21:56 +0200712 cc[termios.VTIME] = vtime
713 cc[termios.VMIN] = vmin
714
715 termios.tcsetattr(
716 self.fd,
717 termios.TCSANOW,
718 [iflag, oflag, cflag, lflag, ispeed, ospeed, cc])
719
Chris Liechti4cf54702015-10-18 00:21:56 +0200720 def read(self, size=1):
721 """\
722 Read size bytes from the serial port. If a timeout is set it may
723 return less characters as requested. With no timeout it will block
724 until the requested number of bytes is read.
725 """
726 if not self.is_open:
727 raise portNotOpenError
728 read = bytearray()
729 while len(read) < size:
730 buf = os.read(self.fd, size - len(read))
731 if not buf:
732 break
733 read.extend(buf)
734 return bytes(read)