Commit | Line | Data |
---|---|---|
abbca718 LOK |
1 | /* |
2 | * boblight | |
3 | * Copyright (C) Bob 2009 | |
4 | * | |
5 | * boblight is free software: you can redistribute it and/or modify it | |
6 | * under the terms of the GNU General Public License as published by the | |
7 | * Free Software Foundation, either version 3 of the License, or | |
8 | * (at your option) any later version. | |
9 | * | |
10 | * boblight is distributed in the hope that it will be useful, but | |
11 | * WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. | |
13 | * See the GNU General Public License for more details. | |
14 | * | |
15 | * You should have received a copy of the GNU General Public License along | |
16 | * with this program. If not, see <http://www.gnu.org/licenses/>. | |
17 | */ | |
18 | ||
19 | #include <iostream>//debug | |
20 | ||
21 | #include <stdio.h> | |
22 | #include <unistd.h> | |
23 | #include <fcntl.h> | |
24 | ||
25 | #include "../serialport.h" | |
26 | #include "../baudrate.h" | |
27 | #include "../../util/misc.h" | |
28 | #include "../../util/timeutils.h" | |
29 | ||
30 | using namespace std; | |
31 | ||
32 | CSerialPort::CSerialPort() | |
33 | { | |
34 | m_fd = -1; | |
35 | } | |
36 | ||
37 | CSerialPort::~CSerialPort() | |
38 | { | |
39 | Close(); | |
40 | } | |
41 | ||
42 | int CSerialPort::Write(uint8_t* data, int len) | |
43 | { | |
44 | fd_set port; | |
45 | ||
46 | if (m_fd == -1) | |
47 | { | |
48 | m_error = "port closed"; | |
49 | return -1; | |
50 | } | |
51 | ||
52 | int byteswritten = 0; | |
53 | ||
54 | while (byteswritten < len) | |
55 | { | |
56 | FD_ZERO(&port); | |
57 | FD_SET(m_fd, &port); | |
58 | int returnv = select(m_fd + 1, NULL, &port, NULL, NULL); | |
59 | if (returnv == -1) | |
60 | { | |
61 | m_error = GetErrno(); | |
62 | return -1; | |
63 | } | |
64 | ||
65 | returnv = write(m_fd, data + byteswritten, len - byteswritten); | |
66 | if (returnv == -1) | |
67 | { | |
68 | m_error = GetErrno(); | |
69 | return -1; | |
70 | } | |
71 | byteswritten += returnv; | |
72 | } | |
73 | ||
74 | //print what's written to stdout for debugging | |
75 | // if (m_tostdout) | |
76 | // { | |
77 | // printf("%s write:", m_name.c_str()); | |
78 | // for (int i = 0; i < byteswritten; i++) | |
79 | // printf(" %02x", (unsigned int)data[i]); | |
80 | ||
81 | // printf("\n"); | |
82 | // } | |
83 | ||
84 | return byteswritten; | |
85 | } | |
86 | ||
87 | int CSerialPort::Read(uint8_t* data, int len, int iTimeoutMs /*= -1*/) | |
88 | { | |
89 | fd_set port; | |
90 | struct timeval timeout, *tv; | |
91 | int64_t now, target; | |
92 | int bytesread = 0; | |
93 | ||
94 | if (m_fd == -1) | |
95 | { | |
96 | m_error = "port closed"; | |
97 | return -1; | |
98 | } | |
99 | ||
100 | if (iTimeoutMs >= 0) | |
101 | { | |
102 | now = GetTimeMs(); | |
103 | target = now + (int64_t) iTimeoutMs; | |
104 | } | |
105 | ||
106 | while (bytesread < len && (iTimeoutMs < 0 || target > now)) | |
107 | { | |
108 | if (iTimeoutMs < 0) | |
109 | { | |
110 | tv = NULL; | |
111 | } | |
112 | else | |
113 | { | |
114 | timeout.tv_sec = ((long int)target - (long int)now) / (long int)1000.; | |
115 | timeout.tv_usec = ((long int)target - (long int)now) % (long int)1000.; | |
116 | tv = &timeout; | |
117 | } | |
118 | ||
119 | FD_ZERO(&port); | |
120 | FD_SET(m_fd, &port); | |
121 | int returnv = select(m_fd + 1, &port, NULL, NULL, tv); | |
122 | ||
123 | if (returnv == -1) | |
124 | { | |
125 | m_error = GetErrno(); | |
126 | return -1; | |
127 | } | |
128 | else if (returnv == 0) | |
129 | { | |
130 | break; //nothing to read | |
131 | } | |
132 | ||
133 | returnv = read(m_fd, data + bytesread, len - bytesread); | |
134 | if (returnv == -1) | |
135 | { | |
136 | m_error = GetErrno(); | |
137 | return -1; | |
138 | } | |
139 | ||
140 | bytesread += returnv; | |
141 | ||
142 | if (iTimeoutMs > 0) | |
143 | now = GetTimeMs(); | |
144 | } | |
145 | ||
146 | //print what's read to stdout for debugging | |
147 | // if (m_tostdout && bytesread > 0) | |
148 | // { | |
149 | // printf("%s read:", m_name.c_str()); | |
150 | // for (int i = 0; i < bytesread; i++) | |
151 | // printf(" %02x", (unsigned int)data[i]); | |
152 | // | |
153 | // printf("\n"); | |
154 | // } | |
155 | ||
156 | return bytesread; | |
157 | } | |
158 | ||
159 | //setting all this stuff up is a pain in the ass | |
160 | bool CSerialPort::Open(string name, int baudrate, int databits/* = 8*/, int stopbits/* = 1*/, int parity/* = PAR_NONE*/) | |
161 | { | |
162 | m_name = name; | |
163 | m_error = GetErrno(); | |
164 | ||
165 | if (databits < 5 || databits > 8) | |
166 | { | |
167 | m_error = "Databits has to be between 5 and 8"; | |
168 | return false; | |
169 | } | |
170 | ||
171 | if (stopbits != 1 && stopbits != 2) | |
172 | { | |
173 | m_error = "Stopbits has to be 1 or 2"; | |
174 | return false; | |
175 | } | |
176 | ||
177 | if (parity != PAR_NONE && parity != PAR_EVEN && parity != PAR_ODD) | |
178 | { | |
179 | m_error = "Parity has to be none, even or odd"; | |
180 | return false; | |
181 | } | |
182 | ||
183 | m_fd = open(name.c_str(), O_RDWR | O_NOCTTY | O_NDELAY); | |
184 | ||
185 | if (m_fd == -1) | |
186 | { | |
187 | m_error = GetErrno(); | |
188 | return false; | |
189 | } | |
190 | ||
191 | fcntl(m_fd, F_SETFL, 0); | |
192 | ||
193 | if (!SetBaudRate(baudrate)) | |
194 | { | |
195 | return false; | |
196 | } | |
197 | ||
198 | m_options.c_cflag |= (CLOCAL | CREAD); | |
199 | m_options.c_cflag &= ~HUPCL; | |
200 | ||
201 | m_options.c_cflag &= ~CSIZE; | |
202 | if (databits == 5) m_options.c_cflag |= CS5; | |
203 | if (databits == 6) m_options.c_cflag |= CS6; | |
204 | if (databits == 7) m_options.c_cflag |= CS7; | |
205 | if (databits == 8) m_options.c_cflag |= CS8; | |
206 | ||
207 | m_options.c_cflag &= ~PARENB; | |
208 | if (parity == PAR_EVEN || parity == PAR_ODD) | |
209 | m_options.c_cflag |= PARENB; | |
210 | if (parity == PAR_ODD) | |
211 | m_options.c_cflag |= PARODD; | |
212 | ||
213 | #ifdef CRTSCTS | |
214 | m_options.c_cflag &= ~CRTSCTS; | |
215 | #elif defined(CNEW_RTSCTS) | |
216 | m_options.c_cflag &= ~CNEW_RTSCTS; | |
217 | #endif | |
218 | ||
219 | if (stopbits == 1) m_options.c_cflag &= ~CSTOPB; | |
220 | else m_options.c_cflag |= CSTOPB; | |
221 | ||
222 | //I guessed a little here | |
223 | m_options.c_lflag &= ~(ICANON | ECHO | ECHOE | ISIG | XCASE | ECHOK | ECHONL | ECHOCTL | ECHOPRT | ECHOKE | TOSTOP); | |
224 | ||
225 | if (parity == PAR_NONE) | |
226 | { | |
227 | m_options.c_iflag &= ~INPCK; | |
228 | } | |
229 | else | |
230 | { | |
231 | m_options.c_iflag |= INPCK | ISTRIP; | |
232 | } | |
233 | ||
234 | m_options.c_iflag &= ~(IXON | IXOFF | IXANY | BRKINT | INLCR | IGNCR | ICRNL | IUCLC | IMAXBEL); | |
235 | m_options.c_oflag &= ~(OPOST | ONLCR | OCRNL); | |
236 | ||
237 | if (tcsetattr(m_fd, TCSANOW, &m_options) != 0) | |
238 | { | |
239 | m_error = GetErrno(); | |
240 | return false; | |
241 | } | |
242 | ||
243 | //non-blocking port | |
244 | fcntl(m_fd, F_SETFL, FNDELAY); | |
245 | ||
246 | return true; | |
247 | } | |
248 | ||
249 | void CSerialPort::Close() | |
250 | { | |
251 | if (m_fd != -1) | |
252 | { | |
253 | close(m_fd); | |
254 | m_fd = -1; | |
255 | m_name = ""; | |
256 | m_error = ""; | |
257 | } | |
258 | } | |
259 | ||
260 | bool CSerialPort::SetBaudRate(int baudrate) | |
261 | { | |
262 | int rate = IntToRate(baudrate); | |
263 | if (rate == -1) | |
264 | { | |
265 | char buff[255]; | |
266 | sprintf(buff, "%i is not a valid baudrate", baudrate); | |
267 | m_error = buff; | |
268 | return false; | |
269 | } | |
270 | ||
271 | //get the current port attributes | |
272 | if (tcgetattr(m_fd, &m_options) != 0) | |
273 | { | |
274 | m_error = GetErrno(); | |
275 | return false; | |
276 | } | |
277 | ||
278 | if (cfsetispeed(&m_options, rate) != 0) | |
279 | { | |
280 | m_error = GetErrno(); | |
281 | return false; | |
282 | } | |
283 | ||
284 | if (cfsetospeed(&m_options, rate) != 0) | |
285 | { | |
286 | m_error = GetErrno(); | |
287 | return false; | |
288 | } | |
289 | ||
290 | return true; | |
291 | } |