serialport.c

   1 /*
   2  * This file is part of the libserialport project.
   3  *
   4  * Copyright (C) 2010-2012 Bert Vermeulen <bert@biot.com>
   5  * Copyright (C) 2010-2015 Uwe Hermann <uwe@hermann-uwe.de>
   6  * Copyright (C) 2013-2015 Martin Ling <martin-libserialport@earth.li>
   7  * Copyright (C) 2013 Matthias Heidbrink <m-sigrok@heidbrink.biz>
   8  * Copyright (C) 2014 Aurelien Jacobs <aurel@gnuage.org>
   9  *
  10  * This program is free software: you can redistribute it and/or modify
  11  * it under the terms of the GNU Lesser General Public License as
  12  * published by the Free Software Foundation, either version 3 of the
  13  * License, or (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public License
  21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  */
  23
  24 #include "libserialport_internal.h"
  25
  26 static const struct std_baudrate std_baudrates[] = {
  27 #ifdef _WIN32
  28         /*
  29          * The baudrates 50/75/134/150/200/1800/230400/460800 do not seem to
  30          * have documented CBR_* macros.
  31          */
  32         BAUD(110), BAUD(300), BAUD(600), BAUD(1200), BAUD(2400), BAUD(4800),
  33         BAUD(9600), BAUD(14400), BAUD(19200), BAUD(38400), BAUD(57600),
  34         BAUD(115200), BAUD(128000), BAUD(256000),
  35 #else
  36         BAUD(50), BAUD(75), BAUD(110), BAUD(134), BAUD(150), BAUD(200),
  37         BAUD(300), BAUD(600), BAUD(1200), BAUD(1800), BAUD(2400), BAUD(4800),
  38         BAUD(9600), BAUD(19200), BAUD(38400), BAUD(57600), BAUD(115200),
  39         BAUD(230400),
  40 #if !defined(__APPLE__) && !defined(__OpenBSD__)
  41         BAUD(460800),
  42 #endif
  43 #endif
  44 };
  45
  46 #define NUM_STD_BAUDRATES ARRAY_SIZE(std_baudrates)
  47
  48 void (*sp_debug_handler)(const char *format, ...) = sp_default_debug_handler;
  49
  50 static enum sp_return get_config(struct sp_port *port, struct port_data *data,
  51         struct sp_port_config *config);
  52
  53 static enum sp_return set_config(struct sp_port *port, struct port_data *data,
  54         const struct sp_port_config *config);
  55
  56 SP_API enum sp_return sp_get_port_by_name(const char *portname, struct sp_port **port_ptr)
  57 {
  58         struct sp_port *port;
  59 #ifndef NO_PORT_METADATA
  60         enum sp_return ret;
  61 #endif
  62         size_t len;
  63
  64         TRACE("%s, %p", portname, port_ptr);
  65
  66         if (!port_ptr)
  67                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
  68
  69         *port_ptr = NULL;
  70
  71         if (!portname)
  72                 RETURN_ERROR(SP_ERR_ARG, "Null port name");
  73
  74         DEBUG_FMT("Building structure for port %s", portname);
  75
  76 #if !defined(_WIN32) && defined(HAVE_REALPATH)
  77         /*
  78          * get_port_details() below tries to be too smart and figure out
  79          * some transport properties from the port name which breaks with
  80          * symlinks. Therefore we canonicalize the portname first.
  81          */
  82         char pathbuf[PATH_MAX + 1];
  83         char *res = realpath(portname, pathbuf);
  84         if (!res)
  85                 RETURN_ERROR(SP_ERR_ARG, "Could not retrieve realpath behind port name");
  86
  87         portname = pathbuf;
  88 #endif
  89
  90         if (!(port = malloc(sizeof(struct sp_port))))
  91                 RETURN_ERROR(SP_ERR_MEM, "Port structure malloc failed");
  92
  93         len = strlen(portname) + 1;
  94
  95         if (!(port->name = malloc(len))) {
  96                 free(port);
  97                 RETURN_ERROR(SP_ERR_MEM, "Port name malloc failed");
  98         }
  99
 100         memcpy(port->name, portname, len);
 101
 102 #ifdef _WIN32
 103         port->usb_path = NULL;
 104         port->hdl = INVALID_HANDLE_VALUE;
 105         port->write_buf = NULL;
 106         port->write_buf_size = 0;
 107 #else
 108         port->fd = -1;
 109 #endif
 110
 111         port->description = NULL;
 112         port->transport = SP_TRANSPORT_NATIVE;
 113         port->usb_bus = -1;
 114         port->usb_address = -1;
 115         port->usb_vid = -1;
 116         port->usb_pid = -1;
 117         port->usb_manufacturer = NULL;
 118         port->usb_product = NULL;
 119         port->usb_serial = NULL;
 120         port->bluetooth_address = NULL;
 121
 122 #ifndef NO_PORT_METADATA
 123         if ((ret = get_port_details(port)) != SP_OK) {
 124                 sp_free_port(port);
 125                 return ret;
 126         }
 127 #endif
 128
 129         *port_ptr = port;
 130
 131         RETURN_OK();
 132 }
 133
 134 SP_API char *sp_get_port_name(const struct sp_port *port)
 135 {
 136         TRACE("%p", port);
 137
 138         if (!port)
 139                 return NULL;
 140
 141         RETURN_STRING(port->name);
 142 }
 143
 144 SP_API char *sp_get_port_description(const struct sp_port *port)
 145 {
 146         TRACE("%p", port);
 147
 148         if (!port || !port->description)
 149                 return NULL;
 150
 151         RETURN_STRING(port->description);
 152 }
 153
 154 SP_API enum sp_transport sp_get_port_transport(const struct sp_port *port)
 155 {
 156         TRACE("%p", port);
 157
 158         if (!port)
 159                 RETURN_ERROR(SP_ERR_ARG, "Null port");
 160
 161         RETURN_INT(port->transport);
 162 }
 163
 164 SP_API enum sp_return sp_get_port_usb_bus_address(const struct sp_port *port,
 165                                                   int *usb_bus,int *usb_address)
 166 {
 167         TRACE("%p", port);
 168
 169         if (!port)
 170                 RETURN_ERROR(SP_ERR_ARG, "Null port");
 171         if (port->transport != SP_TRANSPORT_USB)
 172                 RETURN_ERROR(SP_ERR_ARG, "Port does not use USB transport");
 173         if (port->usb_bus < 0 || port->usb_address < 0)
 174                 RETURN_ERROR(SP_ERR_SUPP, "Bus and address values are not available");
 175
 176         if (usb_bus)
 177                 *usb_bus = port->usb_bus;
 178         if (usb_address)
 179                 *usb_address = port->usb_address;
 180
 181         RETURN_OK();
 182 }
 183
 184 SP_API enum sp_return sp_get_port_usb_vid_pid(const struct sp_port *port,
 185                                               int *usb_vid, int *usb_pid)
 186 {
 187         TRACE("%p", port);
 188
 189         if (!port)
 190                 RETURN_ERROR(SP_ERR_ARG, "Null port");
 191         if (port->transport != SP_TRANSPORT_USB)
 192                 RETURN_ERROR(SP_ERR_ARG, "Port does not use USB transport");
 193         if (port->usb_vid < 0 || port->usb_pid < 0)
 194                 RETURN_ERROR(SP_ERR_SUPP, "VID:PID values are not available");
 195
 196         if (usb_vid)
 197                 *usb_vid = port->usb_vid;
 198         if (usb_pid)
 199                 *usb_pid = port->usb_pid;
 200
 201         RETURN_OK();
 202 }
 203
 204 SP_API char *sp_get_port_usb_manufacturer(const struct sp_port *port)
 205 {
 206         TRACE("%p", port);
 207
 208         if (!port || port->transport != SP_TRANSPORT_USB || !port->usb_manufacturer)
 209                 return NULL;
 210
 211         RETURN_STRING(port->usb_manufacturer);
 212 }
 213
 214 SP_API char *sp_get_port_usb_product(const struct sp_port *port)
 215 {
 216         TRACE("%p", port);
 217
 218         if (!port || port->transport != SP_TRANSPORT_USB || !port->usb_product)
 219                 return NULL;
 220
 221         RETURN_STRING(port->usb_product);
 222 }
 223
 224 SP_API char *sp_get_port_usb_serial(const struct sp_port *port)
 225 {
 226         TRACE("%p", port);
 227
 228         if (!port || port->transport != SP_TRANSPORT_USB || !port->usb_serial)
 229                 return NULL;
 230
 231         RETURN_STRING(port->usb_serial);
 232 }
 233
 234 SP_API char *sp_get_port_bluetooth_address(const struct sp_port *port)
 235 {
 236         TRACE("%p", port);
 237
 238         if (!port || port->transport != SP_TRANSPORT_BLUETOOTH
 239             || !port->bluetooth_address)
 240                 return NULL;
 241
 242         RETURN_STRING(port->bluetooth_address);
 243 }
 244
 245 SP_API enum sp_return sp_get_port_handle(const struct sp_port *port,
 246                                          void *result_ptr)
 247 {
 248         TRACE("%p, %p", port, result_ptr);
 249
 250         if (!port)
 251                 RETURN_ERROR(SP_ERR_ARG, "Null port");
 252         if (!result_ptr)
 253                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
 254
 255 #ifdef _WIN32
 256         HANDLE *handle_ptr = result_ptr;
 257         *handle_ptr = port->hdl;
 258 #else
 259         int *fd_ptr = result_ptr;
 260         *fd_ptr = port->fd;
 261 #endif
 262
 263         RETURN_OK();
 264 }
 265
 266 SP_API enum sp_return sp_copy_port(const struct sp_port *port,
 267                                    struct sp_port **copy_ptr)
 268 {
 269         TRACE("%p, %p", port, copy_ptr);
 270
 271         if (!copy_ptr)
 272                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
 273
 274         *copy_ptr = NULL;
 275
 276         if (!port)
 277                 RETURN_ERROR(SP_ERR_ARG, "Null port");
 278
 279         if (!port->name)
 280                 RETURN_ERROR(SP_ERR_ARG, "Null port name");
 281
 282         DEBUG("Copying port structure");
 283
 284         RETURN_INT(sp_get_port_by_name(port->name, copy_ptr));
 285 }
 286
 287 SP_API void sp_free_port(struct sp_port *port)
 288 {
 289         TRACE("%p", port);
 290
 291         if (!port) {
 292                 DEBUG("Null port");
 293                 RETURN();
 294         }
 295
 296         DEBUG("Freeing port structure");
 297
 298         if (port->name)
 299                 free(port->name);
 300         if (port->description)
 301                 free(port->description);
 302         if (port->usb_manufacturer)
 303                 free(port->usb_manufacturer);
 304         if (port->usb_product)
 305                 free(port->usb_product);
 306         if (port->usb_serial)
 307                 free(port->usb_serial);
 308         if (port->bluetooth_address)
 309                 free(port->bluetooth_address);
 310 #ifdef _WIN32
 311         if (port->usb_path)
 312                 free(port->usb_path);
 313         if (port->write_buf)
 314                 free(port->write_buf);
 315 #endif
 316
 317         free(port);
 318
 319         RETURN();
 320 }
 321
 322 SP_PRIV struct sp_port **list_append(struct sp_port **list,
 323                                      const char *portname)
 324 {
 325         void *tmp;
 326         size_t count;
 327
 328         for (count = 0; list[count]; count++)
 329                 ;
 330         if (!(tmp = realloc(list, sizeof(struct sp_port *) * (count + 2))))
 331                 goto fail;
 332         list = tmp;
 333         if (sp_get_port_by_name(portname, &list[count]) != SP_OK)
 334                 goto fail;
 335         list[count + 1] = NULL;
 336         return list;
 337
 338 fail:
 339         sp_free_port_list(list);
 340         return NULL;
 341 }
 342
 343 SP_API enum sp_return sp_list_ports(struct sp_port ***list_ptr)
 344 {
 345 #ifndef NO_ENUMERATION
 346         struct sp_port **list;
 347         int ret;
 348 #endif
 349
 350         TRACE("%p", list_ptr);
 351
 352         if (!list_ptr)
 353                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
 354
 355         *list_ptr = NULL;
 356
 357 #ifdef NO_ENUMERATION
 358         RETURN_ERROR(SP_ERR_SUPP, "Enumeration not supported on this platform");
 359 #else
 360         DEBUG("Enumerating ports");
 361
 362         if (!(list = malloc(sizeof(struct sp_port *))))
 363                 RETURN_ERROR(SP_ERR_MEM, "Port list malloc failed");
 364
 365         list[0] = NULL;
 366
 367         ret = list_ports(&list);
 368
 369         if (ret == SP_OK) {
 370                 *list_ptr = list;
 371         } else {
 372                 sp_free_port_list(list);
 373                 *list_ptr = NULL;
 374         }
 375
 376         RETURN_CODEVAL(ret);
 377 #endif
 378 }
 379
 380 SP_API void sp_free_port_list(struct sp_port **list)
 381 {
 382         unsigned int i;
 383
 384         TRACE("%p", list);
 385
 386         if (!list) {
 387                 DEBUG("Null list");
 388                 RETURN();
 389         }
 390
 391         DEBUG("Freeing port list");
 392
 393         for (i = 0; list[i]; i++)
 394                 sp_free_port(list[i]);
 395         free(list);
 396
 397         RETURN();
 398 }
 399
 400 #define CHECK_PORT() do { \
 401         if (!port) \
 402                 RETURN_ERROR(SP_ERR_ARG, "Null port"); \
 403         if (!port->name) \
 404                 RETURN_ERROR(SP_ERR_ARG, "Null port name"); \
 405 } while (0)
 406 #ifdef _WIN32
 407 #define CHECK_PORT_HANDLE() do { \
 408         if (port->hdl == INVALID_HANDLE_VALUE) \
 409                 RETURN_ERROR(SP_ERR_ARG, "Port not open"); \
 410 } while (0)
 411 #else
 412 #define CHECK_PORT_HANDLE() do { \
 413         if (port->fd < 0) \
 414                 RETURN_ERROR(SP_ERR_ARG, "Port not open"); \
 415 } while (0)
 416 #endif
 417 #define CHECK_OPEN_PORT() do { \
 418         CHECK_PORT(); \
 419         CHECK_PORT_HANDLE(); \
 420 } while (0)
 421
 422 #ifdef WIN32
 423 /** To be called after port receive buffer is emptied. */
 424 static enum sp_return restart_wait(struct sp_port *port)
 425 {
 426         DWORD wait_result;
 427
 428         if (port->wait_running) {
 429                 /* Check status of running wait operation. */
 430                 if (GetOverlappedResult(port->hdl, &port->wait_ovl,
 431                                 &wait_result, FALSE)) {
 432                         DEBUG("Previous wait completed");
 433                         port->wait_running = FALSE;
 434                 } else if (GetLastError() == ERROR_IO_INCOMPLETE) {
 435                         DEBUG("Previous wait still running");
 436                         RETURN_OK();
 437                 } else {
 438                         RETURN_FAIL("GetOverlappedResult() failed");
 439                 }
 440         }
 441
 442         if (!port->wait_running) {
 443                 /* Start new wait operation. */
 444                 if (WaitCommEvent(port->hdl, &port->events,
 445                                 &port->wait_ovl)) {
 446                         DEBUG("New wait returned, events already pending");
 447                 } else if (GetLastError() == ERROR_IO_PENDING) {
 448                         DEBUG("New wait running in background");
 449                         port->wait_running = TRUE;
 450                 } else {
 451                         RETURN_FAIL("WaitCommEvent() failed");
 452                 }
 453         }
 454
 455         RETURN_OK();
 456 }
 457 #endif
 458
 459 SP_API enum sp_return sp_open(struct sp_port *port, enum sp_mode flags)
 460 {
 461         struct port_data data;
 462         struct sp_port_config config;
 463         enum sp_return ret;
 464
 465         TRACE("%p, 0x%x", port, flags);
 466
 467         CHECK_PORT();
 468
 469         if (flags > SP_MODE_READ_WRITE)
 470                 RETURN_ERROR(SP_ERR_ARG, "Invalid flags");
 471
 472         DEBUG_FMT("Opening port %s", port->name);
 473
 474 #ifdef _WIN32
 475         DWORD desired_access = 0, flags_and_attributes = 0, errors;
 476         char *escaped_port_name;
 477         COMSTAT status;
 478
 479         /* Prefix port name with '\\.\' to work with ports above COM9. */
 480         if (!(escaped_port_name = malloc(strlen(port->name) + 5)))
 481                 RETURN_ERROR(SP_ERR_MEM, "Escaped port name malloc failed");
 482         sprintf(escaped_port_name, "\\\\.\\%s", port->name);
 483
 484         /* Map 'flags' to the OS-specific settings. */
 485         flags_and_attributes = FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED;
 486         if (flags & SP_MODE_READ)
 487                 desired_access |= GENERIC_READ;
 488         if (flags & SP_MODE_WRITE)
 489                 desired_access |= GENERIC_WRITE;
 490
 491         port->hdl = CreateFileA(escaped_port_name, desired_access, 0, 0,
 492                          OPEN_EXISTING, flags_and_attributes, 0);
 493
 494         free(escaped_port_name);
 495
 496         if (port->hdl == INVALID_HANDLE_VALUE)
 497                 RETURN_FAIL("Port CreateFile() failed");
 498
 499         /* All timeouts initially disabled. */
 500         port->timeouts.ReadIntervalTimeout = 0;
 501         port->timeouts.ReadTotalTimeoutMultiplier = 0;
 502         port->timeouts.ReadTotalTimeoutConstant = 0;
 503         port->timeouts.WriteTotalTimeoutMultiplier = 0;
 504         port->timeouts.WriteTotalTimeoutConstant = 0;
 505
 506         if (SetCommTimeouts(port->hdl, &port->timeouts) == 0) {
 507                 sp_close(port);
 508                 RETURN_FAIL("SetCommTimeouts() failed");
 509         }
 510
 511         /* Prepare OVERLAPPED structures. */
 512 #define INIT_OVERLAPPED(ovl) do { \
 513         memset(&port->ovl, 0, sizeof(port->ovl)); \
 514         port->ovl.hEvent = INVALID_HANDLE_VALUE; \
 515         if ((port->ovl.hEvent = CreateEvent(NULL, TRUE, TRUE, NULL)) \
 516                         == INVALID_HANDLE_VALUE) { \
 517                 sp_close(port); \
 518                 RETURN_FAIL(#ovl "CreateEvent() failed"); \
 519         } \
 520 } while (0)
 521
 522         INIT_OVERLAPPED(read_ovl);
 523         INIT_OVERLAPPED(write_ovl);
 524         INIT_OVERLAPPED(wait_ovl);
 525
 526         /* Set event mask for RX and error events. */
 527         if (SetCommMask(port->hdl, EV_RXCHAR | EV_ERR) == 0) {
 528                 sp_close(port);
 529                 RETURN_FAIL("SetCommMask() failed");
 530         }
 531
 532         port->writing = FALSE;
 533         port->wait_running = FALSE;
 534
 535         ret = restart_wait(port);
 536
 537         if (ret < 0) {
 538                 sp_close(port);
 539                 RETURN_CODEVAL(ret);
 540         }
 541 #else
 542         int flags_local = O_NONBLOCK | O_NOCTTY | O_CLOEXEC;
 543
 544         /* Map 'flags' to the OS-specific settings. */
 545         if ((flags & SP_MODE_READ_WRITE) == SP_MODE_READ_WRITE)
 546                 flags_local |= O_RDWR;
 547         else if (flags & SP_MODE_READ)
 548                 flags_local |= O_RDONLY;
 549         else if (flags & SP_MODE_WRITE)
 550                 flags_local |= O_WRONLY;
 551
 552         if ((port->fd = open(port->name, flags_local)) < 0)
 553                 RETURN_FAIL("open() failed");
 554 #endif
 555
 556         ret = get_config(port, &data, &config);
 557
 558         if (ret < 0) {
 559                 sp_close(port);
 560                 RETURN_CODEVAL(ret);
 561         }
 562
 563         /* Set sane port settings. */
 564 #ifdef _WIN32
 565         data.dcb.fBinary = TRUE;
 566         data.dcb.fDsrSensitivity = FALSE;
 567         data.dcb.fErrorChar = FALSE;
 568         data.dcb.fNull = FALSE;
 569         data.dcb.fAbortOnError = FALSE;
 570 #else
 571         /* Turn off all fancy termios tricks, give us a raw channel. */
 572         data.term.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP | INLCR | IGNCR | ICRNL | IMAXBEL);
 573 #ifdef IUCLC
 574         data.term.c_iflag &= ~IUCLC;
 575 #endif
 576         data.term.c_oflag &= ~(OPOST | ONLCR | OCRNL | ONOCR | ONLRET);
 577 #ifdef OLCUC
 578         data.term.c_oflag &= ~OLCUC;
 579 #endif
 580 #ifdef NLDLY
 581         data.term.c_oflag &= ~NLDLY;
 582 #endif
 583 #ifdef CRDLY
 584         data.term.c_oflag &= ~CRDLY;
 585 #endif
 586 #ifdef TABDLY
 587         data.term.c_oflag &= ~TABDLY;
 588 #endif
 589 #ifdef BSDLY
 590         data.term.c_oflag &= ~BSDLY;
 591 #endif
 592 #ifdef VTDLY
 593         data.term.c_oflag &= ~VTDLY;
 594 #endif
 595 #ifdef FFDLY
 596         data.term.c_oflag &= ~FFDLY;
 597 #endif
 598 #ifdef OFILL
 599         data.term.c_oflag &= ~OFILL;
 600 #endif
 601         data.term.c_lflag &= ~(ISIG | ICANON | ECHO | IEXTEN);
 602         data.term.c_cc[VMIN] = 0;
 603         data.term.c_cc[VTIME] = 0;
 604
 605         /* Ignore modem status lines; enable receiver; leave control lines alone on close. */
 606         data.term.c_cflag |= (CLOCAL | CREAD | HUPCL);
 607 #endif
 608
 609 #ifdef _WIN32
 610         if (ClearCommError(port->hdl, &errors, &status) == 0)
 611                 RETURN_FAIL("ClearCommError() failed");
 612 #endif
 613
 614         ret = set_config(port, &data, &config);
 615
 616         if (ret < 0) {
 617                 sp_close(port);
 618                 RETURN_CODEVAL(ret);
 619         }
 620
 621         RETURN_OK();
 622 }
 623
 624 SP_API enum sp_return sp_close(struct sp_port *port)
 625 {
 626         TRACE("%p", port);
 627
 628         CHECK_OPEN_PORT();
 629
 630         DEBUG_FMT("Closing port %s", port->name);
 631
 632 #ifdef _WIN32
 633         /* Returns non-zero upon success, 0 upon failure. */
 634         if (CloseHandle(port->hdl) == 0)
 635                 RETURN_FAIL("Port CloseHandle() failed");
 636         port->hdl = INVALID_HANDLE_VALUE;
 637
 638         /* Close event handles for overlapped structures. */
 639 #define CLOSE_OVERLAPPED(ovl) do { \
 640         if (port->ovl.hEvent != INVALID_HANDLE_VALUE && \
 641                 CloseHandle(port->ovl.hEvent) == 0) \
 642                 RETURN_FAIL(# ovl "event CloseHandle() failed"); \
 643 } while (0)
 644         CLOSE_OVERLAPPED(read_ovl);
 645         CLOSE_OVERLAPPED(write_ovl);
 646         CLOSE_OVERLAPPED(wait_ovl);
 647
 648         if (port->write_buf) {
 649                 free(port->write_buf);
 650                 port->write_buf = NULL;
 651         }
 652 #else
 653         /* Returns 0 upon success, -1 upon failure. */
 654         if (close(port->fd) == -1)
 655                 RETURN_FAIL("close() failed");
 656         port->fd = -1;
 657 #endif
 658
 659         RETURN_OK();
 660 }
 661
 662 SP_API enum sp_return sp_flush(struct sp_port *port, enum sp_buffer buffers)
 663 {
 664         TRACE("%p, 0x%x", port, buffers);
 665
 666         CHECK_OPEN_PORT();
 667
 668         if (buffers > SP_BUF_BOTH)
 669                 RETURN_ERROR(SP_ERR_ARG, "Invalid buffer selection");
 670
 671         const char *buffer_names[] = {"no", "input", "output", "both"};
 672
 673         DEBUG_FMT("Flushing %s buffers on port %s",
 674                 buffer_names[buffers], port->name);
 675
 676 #ifdef _WIN32
 677         DWORD flags = 0;
 678         if (buffers & SP_BUF_INPUT)
 679                 flags |= PURGE_RXCLEAR;
 680         if (buffers & SP_BUF_OUTPUT)
 681                 flags |= PURGE_TXCLEAR;
 682
 683         /* Returns non-zero upon success, 0 upon failure. */
 684         if (PurgeComm(port->hdl, flags) == 0)
 685                 RETURN_FAIL("PurgeComm() failed");
 686
 687         if (buffers & SP_BUF_INPUT)
 688                 TRY(restart_wait(port));
 689 #else
 690         int flags = 0;
 691         if (buffers == SP_BUF_BOTH)
 692                 flags = TCIOFLUSH;
 693         else if (buffers == SP_BUF_INPUT)
 694                 flags = TCIFLUSH;
 695         else if (buffers == SP_BUF_OUTPUT)
 696                 flags = TCOFLUSH;
 697
 698         /* Returns 0 upon success, -1 upon failure. */
 699         if (tcflush(port->fd, flags) < 0)
 700                 RETURN_FAIL("tcflush() failed");
 701 #endif
 702         RETURN_OK();
 703 }
 704
 705 SP_API enum sp_return sp_drain(struct sp_port *port)
 706 {
 707         TRACE("%p", port);
 708
 709         CHECK_OPEN_PORT();
 710
 711         DEBUG_FMT("Draining port %s", port->name);
 712
 713 #ifdef _WIN32
 714         /* Returns non-zero upon success, 0 upon failure. */
 715         if (FlushFileBuffers(port->hdl) == 0)
 716                 RETURN_FAIL("FlushFileBuffers() failed");
 717         RETURN_OK();
 718 #else
 719         int result;
 720         while (1) {
 721 #if defined(__ANDROID__) && (__ANDROID_API__ < 21)
 722                 /* Android only has tcdrain from platform 21 onwards.
 723                  * On previous API versions, use the ioctl directly. */
 724                 int arg = 1;
 725                 result = ioctl(port->fd, TCSBRK, &arg);
 726 #else
 727                 result = tcdrain(port->fd);
 728 #endif
 729                 if (result < 0) {
 730                         if (errno == EINTR) {
 731                                 DEBUG("tcdrain() was interrupted");
 732                                 continue;
 733                         } else {
 734                                 RETURN_FAIL("tcdrain() failed");
 735                         }
 736                 } else {
 737                         RETURN_OK();
 738                 }
 739         }
 740 #endif
 741 }
 742
 743 #ifdef _WIN32
 744 static enum sp_return await_write_completion(struct sp_port *port)
 745 {
 746         TRACE("%p", port);
 747         DWORD bytes_written;
 748         BOOL result;
 749
 750         /* Wait for previous non-blocking write to complete, if any. */
 751         if (port->writing) {
 752                 DEBUG("Waiting for previous write to complete");
 753                 result = GetOverlappedResult(port->hdl, &port->write_ovl, &bytes_written, TRUE);
 754                 port->writing = 0;
 755                 if (!result)
 756                         RETURN_FAIL("Previous write failed to complete");
 757                 DEBUG("Previous write completed");
 758         }
 759
 760         RETURN_OK();
 761 }
 762 #endif
 763
 764 SP_API enum sp_return sp_blocking_write(struct sp_port *port, const void *buf,
 765                                         size_t count, unsigned int timeout_ms)
 766 {
 767         TRACE("%p, %p, %d, %d", port, buf, count, timeout_ms);
 768
 769         CHECK_OPEN_PORT();
 770
 771         if (!buf)
 772                 RETURN_ERROR(SP_ERR_ARG, "Null buffer");
 773
 774         if (timeout_ms)
 775                 DEBUG_FMT("Writing %d bytes to port %s, timeout %d ms",
 776                         count, port->name, timeout_ms);
 777         else
 778                 DEBUG_FMT("Writing %d bytes to port %s, no timeout",
 779                         count, port->name);
 780
 781         if (count == 0)
 782                 RETURN_INT(0);
 783
 784 #ifdef _WIN32
 785         DWORD remaining_ms, write_size, bytes_written;
 786         size_t remaining_bytes, total_bytes_written = 0;
 787         const uint8_t *write_ptr = (uint8_t *) buf;
 788         bool result;
 789         struct timeout timeout;
 790
 791         timeout_start(&timeout, timeout_ms);
 792
 793         TRY(await_write_completion(port));
 794
 795         while (total_bytes_written < count) {
 796
 797                 if (timeout_check(&timeout))
 798                         break;
 799
 800                 remaining_ms = timeout_remaining_ms(&timeout);
 801
 802                 if (port->timeouts.WriteTotalTimeoutConstant != remaining_ms) {
 803                         port->timeouts.WriteTotalTimeoutConstant = remaining_ms;
 804                         if (SetCommTimeouts(port->hdl, &port->timeouts) == 0)
 805                                 RETURN_FAIL("SetCommTimeouts() failed");
 806                 }
 807
 808                 /* Reduce write size if it exceeds the WriteFile limit. */
 809                 remaining_bytes = count - total_bytes_written;
 810                 if (remaining_bytes > WRITEFILE_MAX_SIZE)
 811                         write_size = WRITEFILE_MAX_SIZE;
 812                 else
 813                         write_size = (DWORD) remaining_bytes;
 814
 815                 /* Start write. */
 816
 817                 result = WriteFile(port->hdl, write_ptr, write_size, NULL, &port->write_ovl);
 818
 819                 timeout_update(&timeout);
 820
 821                 if (result) {
 822                         DEBUG("Write completed immediately");
 823                         bytes_written = write_size;
 824                 } else if (GetLastError() == ERROR_IO_PENDING) {
 825                         DEBUG("Waiting for write to complete");
 826                         if (GetOverlappedResult(port->hdl, &port->write_ovl, &bytes_written, TRUE) == 0) {
 827                                 if (GetLastError() == ERROR_SEM_TIMEOUT) {
 828                                         DEBUG("Write timed out");
 829                                         break;
 830                                 } else {
 831                                         RETURN_FAIL("GetOverlappedResult() failed");
 832                                 }
 833                         }
 834                         DEBUG_FMT("Write completed, %d/%d bytes written", bytes_written, write_size);
 835                 } else {
 836                         RETURN_FAIL("WriteFile() failed");
 837                 }
 838
 839                 write_ptr += bytes_written;
 840                 total_bytes_written += bytes_written;
 841         }
 842
 843         RETURN_INT((int) total_bytes_written);
 844 #else
 845         size_t bytes_written = 0;
 846         unsigned char *ptr = (unsigned char *) buf;
 847         struct timeout timeout;
 848         fd_set fds;
 849         int result;
 850
 851         timeout_start(&timeout, timeout_ms);
 852
 853         FD_ZERO(&fds);
 854         FD_SET(port->fd, &fds);
 855
 856         /* Loop until we have written the requested number of bytes. */
 857         while (bytes_written < count) {
 858
 859                 if (timeout_check(&timeout))
 860                         break;
 861
 862                 result = select(port->fd + 1, NULL, &fds, NULL, timeout_timeval(&timeout));
 863
 864                 timeout_update(&timeout);
 865
 866                 if (result < 0) {
 867                         if (errno == EINTR) {
 868                                 DEBUG("select() call was interrupted, repeating");
 869                                 continue;
 870                         } else {
 871                                 RETURN_FAIL("select() failed");
 872                         }
 873                 } else if (result == 0) {
 874                         /* Timeout has expired. */
 875                         break;
 876                 }
 877
 878                 /* Do write. */
 879                 result = write(port->fd, ptr, count - bytes_written);
 880
 881                 if (result < 0) {
 882                         if (errno == EAGAIN)
 883                                 /* This shouldn't happen because we did a select() first, but handle anyway. */
 884                                 continue;
 885                         else
 886                                 /* This is an actual failure. */
 887                                 RETURN_FAIL("write() failed");
 888                 }
 889
 890                 bytes_written += result;
 891                 ptr += result;
 892         }
 893
 894         if (bytes_written < count)
 895                 DEBUG("Write timed out");
 896
 897         RETURN_INT(bytes_written);
 898 #endif
 899 }
 900
 901 SP_API enum sp_return sp_nonblocking_write(struct sp_port *port,
 902                                            const void *buf, size_t count)
 903 {
 904         TRACE("%p, %p, %d", port, buf, count);
 905
 906         CHECK_OPEN_PORT();
 907
 908         if (!buf)
 909                 RETURN_ERROR(SP_ERR_ARG, "Null buffer");
 910
 911         DEBUG_FMT("Writing up to %d bytes to port %s", count, port->name);
 912
 913         if (count == 0)
 914                 RETURN_INT(0);
 915
 916 #ifdef _WIN32
 917         size_t buf_bytes;
 918
 919         /* Check whether previous write is complete. */
 920         if (port->writing) {
 921                 if (HasOverlappedIoCompleted(&port->write_ovl)) {
 922                         DEBUG("Previous write completed");
 923                         port->writing = 0;
 924                 } else {
 925                         DEBUG("Previous write not complete");
 926                         /* Can't take a new write until the previous one finishes. */
 927                         RETURN_INT(0);
 928                 }
 929         }
 930
 931         /* Set timeout. */
 932         if (port->timeouts.WriteTotalTimeoutConstant != 0) {
 933                 port->timeouts.WriteTotalTimeoutConstant = 0;
 934                 if (SetCommTimeouts(port->hdl, &port->timeouts) == 0)
 935                         RETURN_FAIL("SetCommTimeouts() failed");
 936         }
 937
 938         /* Reduce count if it exceeds the WriteFile limit. */
 939         if (count > WRITEFILE_MAX_SIZE)
 940                 count = WRITEFILE_MAX_SIZE;
 941
 942         /* Copy data to our write buffer. */
 943         buf_bytes = min(port->write_buf_size, count);
 944         memcpy(port->write_buf, buf, buf_bytes);
 945
 946         /* Start asynchronous write. */
 947         if (WriteFile(port->hdl, port->write_buf, (DWORD) buf_bytes, NULL, &port->write_ovl) == 0) {
 948                 if (GetLastError() == ERROR_IO_PENDING) {
 949                         if ((port->writing = !HasOverlappedIoCompleted(&port->write_ovl)))
 950                                 DEBUG("Asynchronous write completed immediately");
 951                         else
 952                                 DEBUG("Asynchronous write running");
 953                 } else {
 954                         /* Actual failure of some kind. */
 955                         RETURN_FAIL("WriteFile() failed");
 956                 }
 957         }
 958
 959         DEBUG("All bytes written immediately");
 960
 961         RETURN_INT((int) buf_bytes);
 962 #else
 963         /* Returns the number of bytes written, or -1 upon failure. */
 964         ssize_t written = write(port->fd, buf, count);
 965
 966         if (written < 0) {
 967                 if (errno == EAGAIN)
 968                         // Buffer is full, no bytes written.
 969                         RETURN_INT(0);
 970                 else
 971                         RETURN_FAIL("write() failed");
 972         } else {
 973                 RETURN_INT(written);
 974         }
 975 #endif
 976 }
 977
 978 #ifdef _WIN32
 979 /* Restart wait operation if buffer was emptied. */
 980 static enum sp_return restart_wait_if_needed(struct sp_port *port, unsigned int bytes_read)
 981 {
 982         DWORD errors;
 983         COMSTAT comstat;
 984
 985         if (bytes_read == 0)
 986                 RETURN_OK();
 987
 988         if (ClearCommError(port->hdl, &errors, &comstat) == 0)
 989                 RETURN_FAIL("ClearCommError() failed");
 990
 991         if (comstat.cbInQue == 0)
 992                 TRY(restart_wait(port));
 993
 994         RETURN_OK();
 995 }
 996 #endif
 997
 998 SP_API enum sp_return sp_blocking_read(struct sp_port *port, void *buf,
 999                                        size_t count, unsigned int timeout_ms)
1000 {
1001         TRACE("%p, %p, %d, %d", port, buf, count, timeout_ms);
1002
1003         CHECK_OPEN_PORT();
1004
1005         if (!buf)
1006                 RETURN_ERROR(SP_ERR_ARG, "Null buffer");
1007
1008         if (timeout_ms)
1009                 DEBUG_FMT("Reading %d bytes from port %s, timeout %d ms",
1010                         count, port->name, timeout_ms);
1011         else
1012                 DEBUG_FMT("Reading %d bytes from port %s, no timeout",
1013                         count, port->name);
1014
1015         if (count == 0)
1016                 RETURN_INT(0);
1017
1018 #ifdef _WIN32
1019         DWORD bytes_read;
1020
1021         /* Set timeout. */
1022         if (port->timeouts.ReadIntervalTimeout != 0 ||
1023                         port->timeouts.ReadTotalTimeoutMultiplier != 0 ||
1024                         port->timeouts.ReadTotalTimeoutConstant != timeout_ms) {
1025                 port->timeouts.ReadIntervalTimeout = 0;
1026                 port->timeouts.ReadTotalTimeoutMultiplier = 0;
1027                 port->timeouts.ReadTotalTimeoutConstant = timeout_ms;
1028                 if (SetCommTimeouts(port->hdl, &port->timeouts) == 0)
1029                         RETURN_FAIL("SetCommTimeouts() failed");
1030         }
1031
1032         /* Start read. */
1033         if (ReadFile(port->hdl, buf, (DWORD) count, NULL, &port->read_ovl)) {
1034                 DEBUG("Read completed immediately");
1035                 bytes_read = (DWORD) count;
1036         } else if (GetLastError() == ERROR_IO_PENDING) {
1037                 DEBUG("Waiting for read to complete");
1038                 if (GetOverlappedResult(port->hdl, &port->read_ovl, &bytes_read, TRUE) == 0)
1039                         RETURN_FAIL("GetOverlappedResult() failed");
1040                 DEBUG_FMT("Read completed, %d/%d bytes read", bytes_read, count);
1041         } else {
1042                 RETURN_FAIL("ReadFile() failed");
1043         }
1044
1045         TRY(restart_wait_if_needed(port, bytes_read));
1046
1047         RETURN_INT((int) bytes_read);
1048
1049 #else
1050         size_t bytes_read = 0;
1051         unsigned char *ptr = (unsigned char *) buf;
1052         struct timeout timeout;
1053         fd_set fds;
1054         int result;
1055
1056         timeout_start(&timeout, timeout_ms);
1057
1058         FD_ZERO(&fds);
1059         FD_SET(port->fd, &fds);
1060
1061         /* Loop until we have the requested number of bytes. */
1062         while (bytes_read < count) {
1063
1064                 if (timeout_check(&timeout))
1065                         /* Timeout has expired. */
1066                         break;
1067
1068                 result = select(port->fd + 1, &fds, NULL, NULL, timeout_timeval(&timeout));
1069
1070                 timeout_update(&timeout);
1071
1072                 if (result < 0) {
1073                         if (errno == EINTR) {
1074                                 DEBUG("select() call was interrupted, repeating");
1075                                 continue;
1076                         } else {
1077                                 RETURN_FAIL("select() failed");
1078                         }
1079                 } else if (result == 0) {
1080                         /* Timeout has expired. */
1081                         break;
1082                 }
1083
1084                 /* Do read. */
1085                 result = read(port->fd, ptr, count - bytes_read);
1086
1087                 if (result < 0) {
1088                         if (errno == EAGAIN)
1089                                 /*
1090                                  * This shouldn't happen because we did a
1091                                  * select() first, but handle anyway.
1092                                  */
1093                                 continue;
1094                         else
1095                                 /* This is an actual failure. */
1096                                 RETURN_FAIL("read() failed");
1097                 }
1098
1099                 bytes_read += result;
1100                 ptr += result;
1101         }
1102
1103         if (bytes_read < count)
1104                 DEBUG("Read timed out");
1105
1106         RETURN_INT(bytes_read);
1107 #endif
1108 }
1109
1110 SP_API enum sp_return sp_blocking_read_next(struct sp_port *port, void *buf,
1111                                             size_t count, unsigned int timeout_ms)
1112 {
1113         TRACE("%p, %p, %d, %d", port, buf, count, timeout_ms);
1114
1115         CHECK_OPEN_PORT();
1116
1117         if (!buf)
1118                 RETURN_ERROR(SP_ERR_ARG, "Null buffer");
1119
1120         if (count == 0)
1121                 RETURN_ERROR(SP_ERR_ARG, "Zero count");
1122
1123         if (timeout_ms)
1124                 DEBUG_FMT("Reading next max %d bytes from port %s, timeout %d ms",
1125                         count, port->name, timeout_ms);
1126         else
1127                 DEBUG_FMT("Reading next max %d bytes from port %s, no timeout",
1128                         count, port->name);
1129
1130 #ifdef _WIN32
1131         DWORD bytes_read = 0;
1132
1133         /* If timeout_ms == 0, set maximum timeout. */
1134         DWORD timeout_val = (timeout_ms == 0 ? MAXDWORD - 1 : timeout_ms);
1135
1136         /* Set timeout. */
1137         if (port->timeouts.ReadIntervalTimeout != MAXDWORD ||
1138                         port->timeouts.ReadTotalTimeoutMultiplier != MAXDWORD ||
1139                         port->timeouts.ReadTotalTimeoutConstant != timeout_val) {
1140                 port->timeouts.ReadIntervalTimeout = MAXDWORD;
1141                 port->timeouts.ReadTotalTimeoutMultiplier = MAXDWORD;
1142                 port->timeouts.ReadTotalTimeoutConstant = timeout_val;
1143                 if (SetCommTimeouts(port->hdl, &port->timeouts) == 0)
1144                         RETURN_FAIL("SetCommTimeouts() failed");
1145         }
1146
1147         /* Loop until we have at least one byte, or timeout is reached. */
1148         while (bytes_read == 0) {
1149                 /* Start read. */
1150                 if (ReadFile(port->hdl, buf, (DWORD) count, &bytes_read, &port->read_ovl)) {
1151                         DEBUG("Read completed immediately");
1152                 } else if (GetLastError() == ERROR_IO_PENDING) {
1153                         DEBUG("Waiting for read to complete");
1154                         if (GetOverlappedResult(port->hdl, &port->read_ovl, &bytes_read, TRUE) == 0)
1155                                 RETURN_FAIL("GetOverlappedResult() failed");
1156                         if (bytes_read > 0) {
1157                                 DEBUG("Read completed");
1158                         } else if (timeout_ms > 0) {
1159                                 DEBUG("Read timed out");
1160                                 break;
1161                         } else {
1162                                 DEBUG("Restarting read");
1163                         }
1164                 } else {
1165                         RETURN_FAIL("ReadFile() failed");
1166                 }
1167         }
1168
1169         TRY(restart_wait_if_needed(port, bytes_read));
1170
1171         RETURN_INT(bytes_read);
1172
1173 #else
1174         size_t bytes_read = 0;
1175         struct timeout timeout;
1176         fd_set fds;
1177         int result;
1178
1179         timeout_start(&timeout, timeout_ms);
1180
1181         FD_ZERO(&fds);
1182         FD_SET(port->fd, &fds);
1183
1184         /* Loop until we have at least one byte, or timeout is reached. */
1185         while (bytes_read == 0) {
1186
1187                 if (timeout_check(&timeout))
1188                         /* Timeout has expired. */
1189                         break;
1190
1191                 result = select(port->fd + 1, &fds, NULL, NULL, timeout_timeval(&timeout));
1192
1193                 timeout_update(&timeout);
1194
1195                 if (result < 0) {
1196                         if (errno == EINTR) {
1197                                 DEBUG("select() call was interrupted, repeating");
1198                                 continue;
1199                         } else {
1200                                 RETURN_FAIL("select() failed");
1201                         }
1202                 } else if (result == 0) {
1203                         /* Timeout has expired. */
1204                         break;
1205                 }
1206
1207                 /* Do read. */
1208                 result = read(port->fd, buf, count);
1209
1210                 if (result < 0) {
1211                         if (errno == EAGAIN)
1212                                 /* This shouldn't happen because we did a select() first, but handle anyway. */
1213                                 continue;
1214                         else
1215                                 /* This is an actual failure. */
1216                                 RETURN_FAIL("read() failed");
1217                 }
1218
1219                 bytes_read = result;
1220         }
1221
1222         if (bytes_read == 0)
1223                 DEBUG("Read timed out");
1224
1225         RETURN_INT(bytes_read);
1226 #endif
1227 }
1228
1229 SP_API enum sp_return sp_nonblocking_read(struct sp_port *port, void *buf,
1230                                           size_t count)
1231 {
1232         TRACE("%p, %p, %d", port, buf, count);
1233
1234         CHECK_OPEN_PORT();
1235
1236         if (!buf)
1237                 RETURN_ERROR(SP_ERR_ARG, "Null buffer");
1238
1239         DEBUG_FMT("Reading up to %d bytes from port %s", count, port->name);
1240
1241 #ifdef _WIN32
1242         DWORD bytes_read;
1243
1244         /* Set timeout. */
1245         if (port->timeouts.ReadIntervalTimeout != MAXDWORD ||
1246                         port->timeouts.ReadTotalTimeoutMultiplier != 0 ||
1247                         port->timeouts.ReadTotalTimeoutConstant != 0) {
1248                 port->timeouts.ReadIntervalTimeout = MAXDWORD;
1249                 port->timeouts.ReadTotalTimeoutMultiplier = 0;
1250                 port->timeouts.ReadTotalTimeoutConstant = 0;
1251                 if (SetCommTimeouts(port->hdl, &port->timeouts) == 0)
1252                         RETURN_FAIL("SetCommTimeouts() failed");
1253         }
1254
1255         /* Do read. */
1256         if (ReadFile(port->hdl, buf, (DWORD) count, NULL, &port->read_ovl) == 0)
1257                 if (GetLastError() != ERROR_IO_PENDING)
1258                         RETURN_FAIL("ReadFile() failed");
1259
1260         /* Get number of bytes read. */
1261         if (GetOverlappedResult(port->hdl, &port->read_ovl, &bytes_read, FALSE) == 0)
1262                 RETURN_FAIL("GetOverlappedResult() failed");
1263
1264         TRY(restart_wait_if_needed(port, bytes_read));
1265
1266         RETURN_INT(bytes_read);
1267 #else
1268         ssize_t bytes_read;
1269
1270         /* Returns the number of bytes read, or -1 upon failure. */
1271         if ((bytes_read = read(port->fd, buf, count)) < 0) {
1272                 if (errno == EAGAIN)
1273                         /* No bytes available. */
1274                         bytes_read = 0;
1275                 else
1276                         /* This is an actual failure. */
1277                         RETURN_FAIL("read() failed");
1278         }
1279         RETURN_INT(bytes_read);
1280 #endif
1281 }
1282
1283 SP_API enum sp_return sp_input_waiting(struct sp_port *port)
1284 {
1285         TRACE("%p", port);
1286
1287         CHECK_OPEN_PORT();
1288
1289         DEBUG_FMT("Checking input bytes waiting on port %s", port->name);
1290
1291 #ifdef _WIN32
1292         DWORD errors;
1293         COMSTAT comstat;
1294
1295         if (ClearCommError(port->hdl, &errors, &comstat) == 0)
1296                 RETURN_FAIL("ClearCommError() failed");
1297         RETURN_INT(comstat.cbInQue);
1298 #else
1299         int bytes_waiting;
1300         if (ioctl(port->fd, TIOCINQ, &bytes_waiting) < 0)
1301                 RETURN_FAIL("TIOCINQ ioctl failed");
1302         RETURN_INT(bytes_waiting);
1303 #endif
1304 }
1305
1306 SP_API enum sp_return sp_output_waiting(struct sp_port *port)
1307 {
1308         TRACE("%p", port);
1309
1310 #ifdef __CYGWIN__
1311         /* TIOCOUTQ is not defined in Cygwin headers */
1312         RETURN_ERROR(SP_ERR_SUPP,
1313                         "Getting output bytes waiting is not supported on Cygwin");
1314 #else
1315         CHECK_OPEN_PORT();
1316
1317         DEBUG_FMT("Checking output bytes waiting on port %s", port->name);
1318
1319 #ifdef _WIN32
1320         DWORD errors;
1321         COMSTAT comstat;
1322
1323         if (ClearCommError(port->hdl, &errors, &comstat) == 0)
1324                 RETURN_FAIL("ClearCommError() failed");
1325         RETURN_INT(comstat.cbOutQue);
1326 #else
1327         int bytes_waiting;
1328         if (ioctl(port->fd, TIOCOUTQ, &bytes_waiting) < 0)
1329                 RETURN_FAIL("TIOCOUTQ ioctl failed");
1330         RETURN_INT(bytes_waiting);
1331 #endif
1332 #endif
1333 }
1334
1335 SP_API enum sp_return sp_new_event_set(struct sp_event_set **result_ptr)
1336 {
1337         struct sp_event_set *result;
1338
1339         TRACE("%p", result_ptr);
1340
1341         if (!result_ptr)
1342                 RETURN_ERROR(SP_ERR_ARG, "Null result");
1343
1344         *result_ptr = NULL;
1345
1346         if (!(result = malloc(sizeof(struct sp_event_set))))
1347                 RETURN_ERROR(SP_ERR_MEM, "sp_event_set malloc() failed");
1348
1349         memset(result, 0, sizeof(struct sp_event_set));
1350
1351         *result_ptr = result;
1352
1353         RETURN_OK();
1354 }
1355
1356 static enum sp_return add_handle(struct sp_event_set *event_set,
1357                 event_handle handle, enum sp_event mask)
1358 {
1359         void *new_handles;
1360         enum sp_event *new_masks;
1361
1362         TRACE("%p, %d, %d", event_set, handle, mask);
1363
1364         if (!(new_handles = realloc(event_set->handles,
1365                         sizeof(event_handle) * (event_set->count + 1))))
1366                 RETURN_ERROR(SP_ERR_MEM, "Handle array realloc() failed");
1367
1368         event_set->handles = new_handles;
1369
1370         if (!(new_masks = realloc(event_set->masks,
1371                         sizeof(enum sp_event) * (event_set->count + 1))))
1372                 RETURN_ERROR(SP_ERR_MEM, "Mask array realloc() failed");
1373
1374         event_set->masks = new_masks;
1375
1376         ((event_handle *) event_set->handles)[event_set->count] = handle;
1377         event_set->masks[event_set->count] = mask;
1378
1379         event_set->count++;
1380
1381         RETURN_OK();
1382 }
1383
1384 SP_API enum sp_return sp_add_port_events(struct sp_event_set *event_set,
1385         const struct sp_port *port, enum sp_event mask)
1386 {
1387         TRACE("%p, %p, %d", event_set, port, mask);
1388
1389         if (!event_set)
1390                 RETURN_ERROR(SP_ERR_ARG, "Null event set");
1391
1392         if (!port)
1393                 RETURN_ERROR(SP_ERR_ARG, "Null port");
1394
1395         if (mask > (SP_EVENT_RX_READY | SP_EVENT_TX_READY | SP_EVENT_ERROR))
1396                 RETURN_ERROR(SP_ERR_ARG, "Invalid event mask");
1397
1398         if (!mask)
1399                 RETURN_OK();
1400
1401 #ifdef _WIN32
1402         enum sp_event handle_mask;
1403         if ((handle_mask = mask & SP_EVENT_TX_READY))
1404                 TRY(add_handle(event_set, port->write_ovl.hEvent, handle_mask));
1405         if ((handle_mask = mask & (SP_EVENT_RX_READY | SP_EVENT_ERROR)))
1406                 TRY(add_handle(event_set, port->wait_ovl.hEvent, handle_mask));
1407 #else
1408         TRY(add_handle(event_set, port->fd, mask));
1409 #endif
1410
1411         RETURN_OK();
1412 }
1413
1414 SP_API void sp_free_event_set(struct sp_event_set *event_set)
1415 {
1416         TRACE("%p", event_set);
1417
1418         if (!event_set) {
1419                 DEBUG("Null event set");
1420                 RETURN();
1421         }
1422
1423         DEBUG("Freeing event set");
1424
1425         if (event_set->handles)
1426                 free(event_set->handles);
1427         if (event_set->masks)
1428                 free(event_set->masks);
1429
1430         free(event_set);
1431
1432         RETURN();
1433 }
1434
1435 SP_API enum sp_return sp_wait(struct sp_event_set *event_set,
1436                               unsigned int timeout_ms)
1437 {
1438         TRACE("%p, %d", event_set, timeout_ms);
1439
1440         if (!event_set)
1441                 RETURN_ERROR(SP_ERR_ARG, "Null event set");
1442
1443 #ifdef _WIN32
1444         if (WaitForMultipleObjects(event_set->count, event_set->handles, FALSE,
1445                         timeout_ms ? timeout_ms : INFINITE) == WAIT_FAILED)
1446                 RETURN_FAIL("WaitForMultipleObjects() failed");
1447
1448         RETURN_OK();
1449 #else
1450         struct timeout timeout;
1451         int poll_timeout;
1452         int result;
1453         struct pollfd *pollfds;
1454         unsigned int i;
1455
1456         if (!(pollfds = malloc(sizeof(struct pollfd) * event_set->count)))
1457                 RETURN_ERROR(SP_ERR_MEM, "pollfds malloc() failed");
1458
1459         for (i = 0; i < event_set->count; i++) {
1460                 pollfds[i].fd = ((int *)event_set->handles)[i];
1461                 pollfds[i].events = 0;
1462                 pollfds[i].revents = 0;
1463                 if (event_set->masks[i] & SP_EVENT_RX_READY)
1464                         pollfds[i].events |= POLLIN;
1465                 if (event_set->masks[i] & SP_EVENT_TX_READY)
1466                         pollfds[i].events |= POLLOUT;
1467                 if (event_set->masks[i] & SP_EVENT_ERROR)
1468                         pollfds[i].events |= POLLERR;
1469         }
1470
1471         timeout_start(&timeout, timeout_ms);
1472         timeout_limit(&timeout, INT_MAX);
1473
1474         /* Loop until an event occurs. */
1475         while (1) {
1476
1477                 if (timeout_check(&timeout)) {
1478                         DEBUG("Wait timed out");
1479                         break;
1480                 }
1481
1482                 poll_timeout = (int) timeout_remaining_ms(&timeout);
1483                 if (poll_timeout == 0)
1484                         poll_timeout = -1;
1485
1486                 result = poll(pollfds, event_set->count, poll_timeout);
1487
1488                 timeout_update(&timeout);
1489
1490                 if (result < 0) {
1491                         if (errno == EINTR) {
1492                                 DEBUG("poll() call was interrupted, repeating");
1493                                 continue;
1494                         } else {
1495                                 free(pollfds);
1496                                 RETURN_FAIL("poll() failed");
1497                         }
1498                 } else if (result == 0) {
1499                         DEBUG("poll() timed out");
1500                         if (!timeout.overflow)
1501                                 break;
1502                 } else {
1503                         DEBUG("poll() completed");
1504                         break;
1505                 }
1506         }
1507
1508         free(pollfds);
1509         RETURN_OK();
1510 #endif
1511 }
1512
1513 #ifdef USE_TERMIOS_SPEED
1514 static enum sp_return get_baudrate(int fd, int *baudrate)
1515 {
1516         void *data;
1517
1518         TRACE("%d, %p", fd, baudrate);
1519
1520         DEBUG("Getting baud rate");
1521
1522         if (!(data = malloc(get_termios_size())))
1523                 RETURN_ERROR(SP_ERR_MEM, "termios malloc failed");
1524
1525         if (ioctl(fd, get_termios_get_ioctl(), data) < 0) {
1526                 free(data);
1527                 RETURN_FAIL("Getting termios failed");
1528         }
1529
1530         *baudrate = get_termios_speed(data);
1531
1532         free(data);
1533
1534         RETURN_OK();
1535 }
1536
1537 static enum sp_return set_baudrate(int fd, int baudrate)
1538 {
1539         void *data;
1540
1541         TRACE("%d, %d", fd, baudrate);
1542
1543         DEBUG("Getting baud rate");
1544
1545         if (!(data = malloc(get_termios_size())))
1546                 RETURN_ERROR(SP_ERR_MEM, "termios malloc failed");
1547
1548         if (ioctl(fd, get_termios_get_ioctl(), data) < 0) {
1549                 free(data);
1550                 RETURN_FAIL("Getting termios failed");
1551         }
1552
1553         DEBUG("Setting baud rate");
1554
1555         set_termios_speed(data, baudrate);
1556
1557         if (ioctl(fd, get_termios_set_ioctl(), data) < 0) {
1558                 free(data);
1559                 RETURN_FAIL("Setting termios failed");
1560         }
1561
1562         free(data);
1563
1564         RETURN_OK();
1565 }
1566 #endif /* USE_TERMIOS_SPEED */
1567
1568 #ifdef USE_TERMIOX
1569 static enum sp_return get_flow(int fd, struct port_data *data)
1570 {
1571         void *termx;
1572
1573         TRACE("%d, %p", fd, data);
1574
1575         DEBUG("Getting advanced flow control");
1576
1577         if (!(termx = malloc(get_termiox_size())))
1578                 RETURN_ERROR(SP_ERR_MEM, "termiox malloc failed");
1579
1580         if (ioctl(fd, TCGETX, termx) < 0) {
1581                 free(termx);
1582                 RETURN_FAIL("Getting termiox failed");
1583         }
1584
1585         get_termiox_flow(termx, &data->rts_flow, &data->cts_flow,
1586                         &data->dtr_flow, &data->dsr_flow);
1587
1588         free(termx);
1589
1590         RETURN_OK();
1591 }
1592
1593 static enum sp_return set_flow(int fd, struct port_data *data)
1594 {
1595         void *termx;
1596
1597         TRACE("%d, %p", fd, data);
1598
1599         DEBUG("Getting advanced flow control");
1600
1601         if (!(termx = malloc(get_termiox_size())))
1602                 RETURN_ERROR(SP_ERR_MEM, "termiox malloc failed");
1603
1604         if (ioctl(fd, TCGETX, termx) < 0) {
1605                 free(termx);
1606                 RETURN_FAIL("Getting termiox failed");
1607         }
1608
1609         DEBUG("Setting advanced flow control");
1610
1611         set_termiox_flow(termx, data->rts_flow, data->cts_flow,
1612                         data->dtr_flow, data->dsr_flow);
1613
1614         if (ioctl(fd, TCSETX, termx) < 0) {
1615                 free(termx);
1616                 RETURN_FAIL("Setting termiox failed");
1617         }
1618
1619         free(termx);
1620
1621         RETURN_OK();
1622 }
1623 #endif /* USE_TERMIOX */
1624
1625 static enum sp_return get_config(struct sp_port *port, struct port_data *data,
1626         struct sp_port_config *config)
1627 {
1628         unsigned int i;
1629
1630         TRACE("%p, %p, %p", port, data, config);
1631
1632         DEBUG_FMT("Getting configuration for port %s", port->name);
1633
1634 #ifdef _WIN32
1635         if (!GetCommState(port->hdl, &data->dcb))
1636                 RETURN_FAIL("GetCommState() failed");
1637
1638         for (i = 0; i < NUM_STD_BAUDRATES; i++) {
1639                 if (data->dcb.BaudRate == std_baudrates[i].index) {
1640                         config->baudrate = std_baudrates[i].value;
1641                         break;
1642                 }
1643         }
1644
1645         if (i == NUM_STD_BAUDRATES)
1646                 /* BaudRate field can be either an index or a custom baud rate. */
1647                 config->baudrate = data->dcb.BaudRate;
1648
1649         config->bits = data->dcb.ByteSize;
1650
1651         if (data->dcb.fParity)
1652                 switch (data->dcb.Parity) {
1653                 case NOPARITY:
1654                         config->parity = SP_PARITY_NONE;
1655                         break;
1656                 case ODDPARITY:
1657                         config->parity = SP_PARITY_ODD;
1658                         break;
1659                 case EVENPARITY:
1660                         config->parity = SP_PARITY_EVEN;
1661                         break;
1662                 case MARKPARITY:
1663                         config->parity = SP_PARITY_MARK;
1664                         break;
1665                 case SPACEPARITY:
1666                         config->parity = SP_PARITY_SPACE;
1667                         break;
1668                 default:
1669                         config->parity = -1;
1670                 }
1671         else
1672                 config->parity = SP_PARITY_NONE;
1673
1674         switch (data->dcb.StopBits) {
1675         case ONESTOPBIT:
1676                 config->stopbits = 1;
1677                 break;
1678         case TWOSTOPBITS:
1679                 config->stopbits = 2;
1680                 break;
1681         default:
1682                 config->stopbits = -1;
1683         }
1684
1685         switch (data->dcb.fRtsControl) {
1686         case RTS_CONTROL_DISABLE:
1687                 config->rts = SP_RTS_OFF;
1688                 break;
1689         case RTS_CONTROL_ENABLE:
1690                 config->rts = SP_RTS_ON;
1691                 break;
1692         case RTS_CONTROL_HANDSHAKE:
1693                 config->rts = SP_RTS_FLOW_CONTROL;
1694                 break;
1695         default:
1696                 config->rts = -1;
1697         }
1698
1699         config->cts = data->dcb.fOutxCtsFlow ? SP_CTS_FLOW_CONTROL : SP_CTS_IGNORE;
1700
1701         switch (data->dcb.fDtrControl) {
1702         case DTR_CONTROL_DISABLE:
1703                 config->dtr = SP_DTR_OFF;
1704                 break;
1705         case DTR_CONTROL_ENABLE:
1706                 config->dtr = SP_DTR_ON;
1707                 break;
1708         case DTR_CONTROL_HANDSHAKE:
1709                 config->dtr = SP_DTR_FLOW_CONTROL;
1710                 break;
1711         default:
1712                 config->dtr = -1;
1713         }
1714
1715         config->dsr = data->dcb.fOutxDsrFlow ? SP_DSR_FLOW_CONTROL : SP_DSR_IGNORE;
1716
1717         if (data->dcb.fInX) {
1718                 if (data->dcb.fOutX)
1719                         config->xon_xoff = SP_XONXOFF_INOUT;
1720                 else
1721                         config->xon_xoff = SP_XONXOFF_IN;
1722         } else {
1723                 if (data->dcb.fOutX)
1724                         config->xon_xoff = SP_XONXOFF_OUT;
1725                 else
1726                         config->xon_xoff = SP_XONXOFF_DISABLED;
1727         }
1728
1729 #else // !_WIN32
1730
1731         if (tcgetattr(port->fd, &data->term) < 0)
1732                 RETURN_FAIL("tcgetattr() failed");
1733
1734         if (ioctl(port->fd, TIOCMGET, &data->controlbits) < 0)
1735                 RETURN_FAIL("TIOCMGET ioctl failed");
1736
1737 #ifdef USE_TERMIOX
1738         int ret = get_flow(port->fd, data);
1739
1740         if (ret == SP_ERR_FAIL && errno == EINVAL)
1741                 data->termiox_supported = 0;
1742         else if (ret < 0)
1743                 RETURN_CODEVAL(ret);
1744         else
1745                 data->termiox_supported = 1;
1746 #else
1747         data->termiox_supported = 0;
1748 #endif
1749
1750         for (i = 0; i < NUM_STD_BAUDRATES; i++) {
1751                 if (cfgetispeed(&data->term) == std_baudrates[i].index) {
1752                         config->baudrate = std_baudrates[i].value;
1753                         break;
1754                 }
1755         }
1756
1757         if (i == NUM_STD_BAUDRATES) {
1758 #ifdef __APPLE__
1759                 config->baudrate = (int)data->term.c_ispeed;
1760 #elif defined(USE_TERMIOS_SPEED)
1761                 TRY(get_baudrate(port->fd, &config->baudrate));
1762 #else
1763                 config->baudrate = -1;
1764 #endif
1765         }
1766
1767         switch (data->term.c_cflag & CSIZE) {
1768         case CS8:
1769                 config->bits = 8;
1770                 break;
1771         case CS7:
1772                 config->bits = 7;
1773                 break;
1774         case CS6:
1775                 config->bits = 6;
1776                 break;
1777         case CS5:
1778                 config->bits = 5;
1779                 break;
1780         default:
1781                 config->bits = -1;
1782         }
1783
1784         if (!(data->term.c_cflag & PARENB) && (data->term.c_iflag & IGNPAR))
1785                 config->parity = SP_PARITY_NONE;
1786         else if (!(data->term.c_cflag & PARENB) || (data->term.c_iflag & IGNPAR))
1787                 config->parity = -1;
1788 #ifdef CMSPAR
1789         else if (data->term.c_cflag & CMSPAR)
1790                 config->parity = (data->term.c_cflag & PARODD) ? SP_PARITY_MARK : SP_PARITY_SPACE;
1791 #endif
1792         else
1793                 config->parity = (data->term.c_cflag & PARODD) ? SP_PARITY_ODD : SP_PARITY_EVEN;
1794
1795         config->stopbits = (data->term.c_cflag & CSTOPB) ? 2 : 1;
1796
1797         if (data->term.c_cflag & CRTSCTS) {
1798                 config->rts = SP_RTS_FLOW_CONTROL;
1799                 config->cts = SP_CTS_FLOW_CONTROL;
1800         } else {
1801                 if (data->termiox_supported && data->rts_flow)
1802                         config->rts = SP_RTS_FLOW_CONTROL;
1803                 else
1804                         config->rts = (data->controlbits & TIOCM_RTS) ? SP_RTS_ON : SP_RTS_OFF;
1805
1806                 config->cts = (data->termiox_supported && data->cts_flow) ?
1807                         SP_CTS_FLOW_CONTROL : SP_CTS_IGNORE;
1808         }
1809
1810         if (data->termiox_supported && data->dtr_flow)
1811                 config->dtr = SP_DTR_FLOW_CONTROL;
1812         else
1813                 config->dtr = (data->controlbits & TIOCM_DTR) ? SP_DTR_ON : SP_DTR_OFF;
1814
1815         config->dsr = (data->termiox_supported && data->dsr_flow) ?
1816                 SP_DSR_FLOW_CONTROL : SP_DSR_IGNORE;
1817
1818         if (data->term.c_iflag & IXOFF) {
1819                 if (data->term.c_iflag & IXON)
1820                         config->xon_xoff = SP_XONXOFF_INOUT;
1821                 else
1822                         config->xon_xoff = SP_XONXOFF_IN;
1823         } else {
1824                 if (data->term.c_iflag & IXON)
1825                         config->xon_xoff = SP_XONXOFF_OUT;
1826                 else
1827                         config->xon_xoff = SP_XONXOFF_DISABLED;
1828         }
1829 #endif
1830
1831         RETURN_OK();
1832 }
1833
1834 static enum sp_return set_config(struct sp_port *port, struct port_data *data,
1835         const struct sp_port_config *config)
1836 {
1837         unsigned int i;
1838 #ifdef __APPLE__
1839         BAUD_TYPE baud_nonstd;
1840
1841         baud_nonstd = B0;
1842 #endif
1843 #ifdef USE_TERMIOS_SPEED
1844         int baud_nonstd = 0;
1845 #endif
1846
1847         TRACE("%p, %p, %p", port, data, config);
1848
1849         DEBUG_FMT("Setting configuration for port %s", port->name);
1850
1851 #ifdef _WIN32
1852         BYTE* new_buf;
1853
1854         TRY(await_write_completion(port));
1855
1856         if (config->baudrate >= 0) {
1857                 for (i = 0; i < NUM_STD_BAUDRATES; i++) {
1858                         if (config->baudrate == std_baudrates[i].value) {
1859                                 data->dcb.BaudRate = std_baudrates[i].index;
1860                                 break;
1861                         }
1862                 }
1863
1864                 if (i == NUM_STD_BAUDRATES)
1865                         data->dcb.BaudRate = config->baudrate;
1866
1867                 /* Allocate write buffer for 50ms of data at baud rate. */
1868                 port->write_buf_size = max(config->baudrate / (8 * 20), 1);
1869                 new_buf = realloc(port->write_buf, port->write_buf_size);
1870                 if (!new_buf)
1871                         RETURN_ERROR(SP_ERR_MEM, "Allocating write buffer failed");
1872                 port->write_buf = new_buf;
1873         }
1874
1875         if (config->bits >= 0)
1876                 data->dcb.ByteSize = config->bits;
1877
1878         if (config->parity >= 0) {
1879                 switch (config->parity) {
1880                 case SP_PARITY_NONE:
1881                         data->dcb.Parity = NOPARITY;
1882                         break;
1883                 case SP_PARITY_ODD:
1884                         data->dcb.Parity = ODDPARITY;
1885                         break;
1886                 case SP_PARITY_EVEN:
1887                         data->dcb.Parity = EVENPARITY;
1888                         break;
1889                 case SP_PARITY_MARK:
1890                         data->dcb.Parity = MARKPARITY;
1891                         break;
1892                 case SP_PARITY_SPACE:
1893                         data->dcb.Parity = SPACEPARITY;
1894                         break;
1895                 default:
1896                         RETURN_ERROR(SP_ERR_ARG, "Invalid parity setting");
1897                 }
1898         }
1899
1900         if (config->stopbits >= 0) {
1901                 switch (config->stopbits) {
1902                 /* Note: There's also ONE5STOPBITS == 1.5 (unneeded so far). */
1903                 case 1:
1904                         data->dcb.StopBits = ONESTOPBIT;
1905                         break;
1906                 case 2:
1907                         data->dcb.StopBits = TWOSTOPBITS;
1908                         break;
1909                 default:
1910                         RETURN_ERROR(SP_ERR_ARG, "Invalid stop bit setting");
1911                 }
1912         }
1913
1914         if (config->rts >= 0) {
1915                 switch (config->rts) {
1916                 case SP_RTS_OFF:
1917                         data->dcb.fRtsControl = RTS_CONTROL_DISABLE;
1918                         break;
1919                 case SP_RTS_ON:
1920                         data->dcb.fRtsControl = RTS_CONTROL_ENABLE;
1921                         break;
1922                 case SP_RTS_FLOW_CONTROL:
1923                         data->dcb.fRtsControl = RTS_CONTROL_HANDSHAKE;
1924                         break;
1925                 default:
1926                         RETURN_ERROR(SP_ERR_ARG, "Invalid RTS setting");
1927                 }
1928         }
1929
1930         if (config->cts >= 0) {
1931                 switch (config->cts) {
1932                 case SP_CTS_IGNORE:
1933                         data->dcb.fOutxCtsFlow = FALSE;
1934                         break;
1935                 case SP_CTS_FLOW_CONTROL:
1936                         data->dcb.fOutxCtsFlow = TRUE;
1937                         break;
1938                 default:
1939                         RETURN_ERROR(SP_ERR_ARG, "Invalid CTS setting");
1940                 }
1941         }
1942
1943         if (config->dtr >= 0) {
1944                 switch (config->dtr) {
1945                 case SP_DTR_OFF:
1946                         data->dcb.fDtrControl = DTR_CONTROL_DISABLE;
1947                         break;
1948                 case SP_DTR_ON:
1949                         data->dcb.fDtrControl = DTR_CONTROL_ENABLE;
1950                         break;
1951                 case SP_DTR_FLOW_CONTROL:
1952                         data->dcb.fDtrControl = DTR_CONTROL_HANDSHAKE;
1953                         break;
1954                 default:
1955                         RETURN_ERROR(SP_ERR_ARG, "Invalid DTR setting");
1956                 }
1957         }
1958
1959         if (config->dsr >= 0) {
1960                 switch (config->dsr) {
1961                 case SP_DSR_IGNORE:
1962                         data->dcb.fOutxDsrFlow = FALSE;
1963                         break;
1964                 case SP_DSR_FLOW_CONTROL:
1965                         data->dcb.fOutxDsrFlow = TRUE;
1966                         break;
1967                 default:
1968                         RETURN_ERROR(SP_ERR_ARG, "Invalid DSR setting");
1969                 }
1970         }
1971
1972         if (config->xon_xoff >= 0) {
1973                 switch (config->xon_xoff) {
1974                 case SP_XONXOFF_DISABLED:
1975                         data->dcb.fInX = FALSE;
1976                         data->dcb.fOutX = FALSE;
1977                         break;
1978                 case SP_XONXOFF_IN:
1979                         data->dcb.fInX = TRUE;
1980                         data->dcb.fOutX = FALSE;
1981                         break;
1982                 case SP_XONXOFF_OUT:
1983                         data->dcb.fInX = FALSE;
1984                         data->dcb.fOutX = TRUE;
1985                         break;
1986                 case SP_XONXOFF_INOUT:
1987                         data->dcb.fInX = TRUE;
1988                         data->dcb.fOutX = TRUE;
1989                         break;
1990                 default:
1991                         RETURN_ERROR(SP_ERR_ARG, "Invalid XON/XOFF setting");
1992                 }
1993         }
1994
1995         if (!SetCommState(port->hdl, &data->dcb))
1996                 RETURN_FAIL("SetCommState() failed");
1997
1998 #else /* !_WIN32 */
1999
2000         int controlbits;
2001
2002         if (config->baudrate >= 0) {
2003                 for (i = 0; i < NUM_STD_BAUDRATES; i++) {
2004                         if (config->baudrate == std_baudrates[i].value) {
2005                                 if (cfsetospeed(&data->term, std_baudrates[i].index) < 0)
2006                                         RETURN_FAIL("cfsetospeed() failed");
2007
2008                                 if (cfsetispeed(&data->term, std_baudrates[i].index) < 0)
2009                                         RETURN_FAIL("cfsetispeed() failed");
2010                                 break;
2011                         }
2012                 }
2013
2014                 /* Non-standard baud rate */
2015                 if (i == NUM_STD_BAUDRATES) {
2016 #ifdef __APPLE__
2017                         /* Set "dummy" baud rate. */
2018                         if (cfsetspeed(&data->term, B9600) < 0)
2019                                 RETURN_FAIL("cfsetspeed() failed");
2020                         baud_nonstd = config->baudrate;
2021 #elif defined(USE_TERMIOS_SPEED)
2022                         baud_nonstd = 1;
2023 #else
2024                         RETURN_ERROR(SP_ERR_SUPP, "Non-standard baudrate not supported");
2025 #endif
2026                 }
2027         }
2028
2029         if (config->bits >= 0) {
2030                 data->term.c_cflag &= ~CSIZE;
2031                 switch (config->bits) {
2032                 case 8:
2033                         data->term.c_cflag |= CS8;
2034                         break;
2035                 case 7:
2036                         data->term.c_cflag |= CS7;
2037                         break;
2038                 case 6:
2039                         data->term.c_cflag |= CS6;
2040                         break;
2041                 case 5:
2042                         data->term.c_cflag |= CS5;
2043                         break;
2044                 default:
2045                         RETURN_ERROR(SP_ERR_ARG, "Invalid data bits setting");
2046                 }
2047         }
2048
2049         if (config->parity >= 0) {
2050                 data->term.c_iflag &= ~IGNPAR;
2051                 data->term.c_cflag &= ~(PARENB | PARODD);
2052 #ifdef CMSPAR
2053                 data->term.c_cflag &= ~CMSPAR;
2054 #endif
2055                 switch (config->parity) {
2056                 case SP_PARITY_NONE:
2057                         data->term.c_iflag |= IGNPAR;
2058                         break;
2059                 case SP_PARITY_EVEN:
2060                         data->term.c_cflag |= PARENB;
2061                         break;
2062                 case SP_PARITY_ODD:
2063                         data->term.c_cflag |= PARENB | PARODD;
2064                         break;
2065 #ifdef CMSPAR
2066                 case SP_PARITY_MARK:
2067                         data->term.c_cflag |= PARENB | PARODD;
2068                         data->term.c_cflag |= CMSPAR;
2069                         break;
2070                 case SP_PARITY_SPACE:
2071                         data->term.c_cflag |= PARENB;
2072                         data->term.c_cflag |= CMSPAR;
2073                         break;
2074 #else
2075                 case SP_PARITY_MARK:
2076                 case SP_PARITY_SPACE:
2077                         RETURN_ERROR(SP_ERR_SUPP, "Mark/space parity not supported");
2078 #endif
2079                 default:
2080                         RETURN_ERROR(SP_ERR_ARG, "Invalid parity setting");
2081                 }
2082         }
2083
2084         if (config->stopbits >= 0) {
2085                 data->term.c_cflag &= ~CSTOPB;
2086                 switch (config->stopbits) {
2087                 case 1:
2088                         data->term.c_cflag &= ~CSTOPB;
2089                         break;
2090                 case 2:
2091                         data->term.c_cflag |= CSTOPB;
2092                         break;
2093                 default:
2094                         RETURN_ERROR(SP_ERR_ARG, "Invalid stop bits setting");
2095                 }
2096         }
2097
2098         if (config->rts >= 0 || config->cts >= 0) {
2099                 if (data->termiox_supported) {
2100                         data->rts_flow = data->cts_flow = 0;
2101                         switch (config->rts) {
2102                         case SP_RTS_OFF:
2103                         case SP_RTS_ON:
2104                                 controlbits = TIOCM_RTS;
2105                                 if (ioctl(port->fd, config->rts == SP_RTS_ON ? TIOCMBIS : TIOCMBIC, &controlbits) < 0)
2106                                         RETURN_FAIL("Setting RTS signal level failed");
2107                                 break;
2108                         case SP_RTS_FLOW_CONTROL:
2109                                 data->rts_flow = 1;
2110                                 break;
2111                         default:
2112                                 break;
2113                         }
2114                         if (config->cts == SP_CTS_FLOW_CONTROL)
2115                                 data->cts_flow = 1;
2116
2117                         if (data->rts_flow && data->cts_flow)
2118                                 data->term.c_iflag |= CRTSCTS;
2119                         else
2120                                 data->term.c_iflag &= ~CRTSCTS;
2121                 } else {
2122                         /* Asymmetric use of RTS/CTS not supported. */
2123                         if (data->term.c_iflag & CRTSCTS) {
2124                                 /* Flow control can only be disabled for both RTS & CTS together. */
2125                                 if (config->rts >= 0 && config->rts != SP_RTS_FLOW_CONTROL) {
2126                                         if (config->cts != SP_CTS_IGNORE)
2127                                                 RETURN_ERROR(SP_ERR_SUPP, "RTS & CTS flow control must be disabled together");
2128                                 }
2129                                 if (config->cts >= 0 && config->cts != SP_CTS_FLOW_CONTROL) {
2130                                         if (config->rts <= 0 || config->rts == SP_RTS_FLOW_CONTROL)
2131                                                 RETURN_ERROR(SP_ERR_SUPP, "RTS & CTS flow control must be disabled together");
2132                                 }
2133                         } else {
2134                                 /* Flow control can only be enabled for both RTS & CTS together. */
2135                                 if (((config->rts == SP_RTS_FLOW_CONTROL) && (config->cts != SP_CTS_FLOW_CONTROL)) ||
2136                                         ((config->cts == SP_CTS_FLOW_CONTROL) && (config->rts != SP_RTS_FLOW_CONTROL)))
2137                                         RETURN_ERROR(SP_ERR_SUPP, "RTS & CTS flow control must be enabled together");
2138                         }
2139
2140                         if (config->rts >= 0) {
2141                                 if (config->rts == SP_RTS_FLOW_CONTROL) {
2142                                         data->term.c_iflag |= CRTSCTS;
2143                                 } else {
2144                                         controlbits = TIOCM_RTS;
2145                                         if (ioctl(port->fd, config->rts == SP_RTS_ON ? TIOCMBIS : TIOCMBIC,
2146                                                         &controlbits) < 0)
2147                                                 RETURN_FAIL("Setting RTS signal level failed");
2148                                 }
2149                         }
2150                 }
2151         }
2152
2153         if (config->dtr >= 0 || config->dsr >= 0) {
2154                 if (data->termiox_supported) {
2155                         data->dtr_flow = data->dsr_flow = 0;
2156                         switch (config->dtr) {
2157                         case SP_DTR_OFF:
2158                         case SP_DTR_ON:
2159                                 controlbits = TIOCM_DTR;
2160                                 if (ioctl(port->fd, config->dtr == SP_DTR_ON ? TIOCMBIS : TIOCMBIC, &controlbits) < 0)
2161                                         RETURN_FAIL("Setting DTR signal level failed");
2162                                 break;
2163                         case SP_DTR_FLOW_CONTROL:
2164                                 data->dtr_flow = 1;
2165                                 break;
2166                         default:
2167                                 break;
2168                         }
2169                         if (config->dsr == SP_DSR_FLOW_CONTROL)
2170                                 data->dsr_flow = 1;
2171                 } else {
2172                         /* DTR/DSR flow control not supported. */
2173                         if (config->dtr == SP_DTR_FLOW_CONTROL || config->dsr == SP_DSR_FLOW_CONTROL)
2174                                 RETURN_ERROR(SP_ERR_SUPP, "DTR/DSR flow control not supported");
2175
2176                         if (config->dtr >= 0) {
2177                                 controlbits = TIOCM_DTR;
2178                                 if (ioctl(port->fd, config->dtr == SP_DTR_ON ? TIOCMBIS : TIOCMBIC,
2179                                                 &controlbits) < 0)
2180                                         RETURN_FAIL("Setting DTR signal level failed");
2181                         }
2182                 }
2183         }
2184
2185         if (config->xon_xoff >= 0) {
2186                 data->term.c_iflag &= ~(IXON | IXOFF | IXANY);
2187                 switch (config->xon_xoff) {
2188                 case SP_XONXOFF_DISABLED:
2189                         break;
2190                 case SP_XONXOFF_IN:
2191                         data->term.c_iflag |= IXOFF;
2192                         break;
2193                 case SP_XONXOFF_OUT:
2194                         data->term.c_iflag |= IXON | IXANY;
2195                         break;
2196                 case SP_XONXOFF_INOUT:
2197                         data->term.c_iflag |= IXON | IXOFF | IXANY;
2198                         break;
2199                 default:
2200                         RETURN_ERROR(SP_ERR_ARG, "Invalid XON/XOFF setting");
2201                 }
2202         }
2203
2204         if (tcsetattr(port->fd, TCSANOW, &data->term) < 0)
2205                 RETURN_FAIL("tcsetattr() failed");
2206
2207 #ifdef __APPLE__
2208         if (baud_nonstd != B0) {
2209                 if (ioctl(port->fd, IOSSIOSPEED, &baud_nonstd) == -1)
2210                         RETURN_FAIL("IOSSIOSPEED ioctl failed");
2211                 /*
2212                  * Set baud rates in data->term to correct, but incompatible
2213                  * with tcsetattr() value, same as delivered by tcgetattr().
2214                  */
2215                 if (cfsetspeed(&data->term, baud_nonstd) < 0)
2216                         RETURN_FAIL("cfsetspeed() failed");
2217         }
2218 #elif defined(__linux__)
2219 #ifdef USE_TERMIOS_SPEED
2220         if (baud_nonstd)
2221                 TRY(set_baudrate(port->fd, config->baudrate));
2222 #endif
2223 #ifdef USE_TERMIOX
2224         if (data->termiox_supported)
2225                 TRY(set_flow(port->fd, data));
2226 #endif
2227 #endif
2228
2229 #endif /* !_WIN32 */
2230
2231         RETURN_OK();
2232 }
2233
2234 SP_API enum sp_return sp_new_config(struct sp_port_config **config_ptr)
2235 {
2236         struct sp_port_config *config;
2237
2238         TRACE("%p", config_ptr);
2239
2240         if (!config_ptr)
2241                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
2242
2243         *config_ptr = NULL;
2244
2245         if (!(config = malloc(sizeof(struct sp_port_config))))
2246                 RETURN_ERROR(SP_ERR_MEM, "Config malloc failed");
2247
2248         config->baudrate = -1;
2249         config->bits = -1;
2250         config->parity = -1;
2251         config->stopbits = -1;
2252         config->rts = -1;
2253         config->cts = -1;
2254         config->dtr = -1;
2255         config->dsr = -1;
2256
2257         *config_ptr = config;
2258
2259         RETURN_OK();
2260 }
2261
2262 SP_API void sp_free_config(struct sp_port_config *config)
2263 {
2264         TRACE("%p", config);
2265
2266         if (!config)
2267                 DEBUG("Null config");
2268         else
2269                 free(config);
2270
2271         RETURN();
2272 }
2273
2274 SP_API enum sp_return sp_get_config(struct sp_port *port,
2275                                     struct sp_port_config *config)
2276 {
2277         struct port_data data;
2278
2279         TRACE("%p, %p", port, config);
2280
2281         CHECK_OPEN_PORT();
2282
2283         if (!config)
2284                 RETURN_ERROR(SP_ERR_ARG, "Null config");
2285
2286         TRY(get_config(port, &data, config));
2287
2288         RETURN_OK();
2289 }
2290
2291 SP_API enum sp_return sp_set_config(struct sp_port *port,
2292                                     const struct sp_port_config *config)
2293 {
2294         struct port_data data;
2295         struct sp_port_config prev_config;
2296
2297         TRACE("%p, %p", port, config);
2298
2299         CHECK_OPEN_PORT();
2300
2301         if (!config)
2302                 RETURN_ERROR(SP_ERR_ARG, "Null config");
2303
2304         TRY(get_config(port, &data, &prev_config));
2305         TRY(set_config(port, &data, config));
2306
2307         RETURN_OK();
2308 }
2309
2310 #define CREATE_ACCESSORS(x, type) \
2311 SP_API enum sp_return sp_set_##x(struct sp_port *port, type x) { \
2312         struct port_data data; \
2313         struct sp_port_config config; \
2314         TRACE("%p, %d", port, x); \
2315         CHECK_OPEN_PORT(); \
2316         TRY(get_config(port, &data, &config)); \
2317         config.x = x; \
2318         TRY(set_config(port, &data, &config)); \
2319         RETURN_OK(); \
2320 } \
2321 SP_API enum sp_return sp_get_config_##x(const struct sp_port_config *config, \
2322                                         type *x) { \
2323         TRACE("%p, %p", config, x); \
2324         if (!x) \
2325                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer"); \
2326         if (!config) \
2327                 RETURN_ERROR(SP_ERR_ARG, "Null config"); \
2328         *x = config->x; \
2329         RETURN_OK(); \
2330 } \
2331 SP_API enum sp_return sp_set_config_##x(struct sp_port_config *config, \
2332                                         type x) { \
2333         TRACE("%p, %d", config, x); \
2334         if (!config) \
2335                 RETURN_ERROR(SP_ERR_ARG, "Null config"); \
2336         config->x = x; \
2337         RETURN_OK(); \
2338 }
2339
2340 CREATE_ACCESSORS(baudrate, int)
2341 CREATE_ACCESSORS(bits, int)
2342 CREATE_ACCESSORS(parity, enum sp_parity)
2343 CREATE_ACCESSORS(stopbits, int)
2344 CREATE_ACCESSORS(rts, enum sp_rts)
2345 CREATE_ACCESSORS(cts, enum sp_cts)
2346 CREATE_ACCESSORS(dtr, enum sp_dtr)
2347 CREATE_ACCESSORS(dsr, enum sp_dsr)
2348 CREATE_ACCESSORS(xon_xoff, enum sp_xonxoff)
2349
2350 SP_API enum sp_return sp_set_config_flowcontrol(struct sp_port_config *config,
2351                                                 enum sp_flowcontrol flowcontrol)
2352 {
2353         if (!config)
2354                 RETURN_ERROR(SP_ERR_ARG, "Null configuration");
2355
2356         if (flowcontrol > SP_FLOWCONTROL_DTRDSR)
2357                 RETURN_ERROR(SP_ERR_ARG, "Invalid flow control setting");
2358
2359         if (flowcontrol == SP_FLOWCONTROL_XONXOFF)
2360                 config->xon_xoff = SP_XONXOFF_INOUT;
2361         else
2362                 config->xon_xoff = SP_XONXOFF_DISABLED;
2363
2364         if (flowcontrol == SP_FLOWCONTROL_RTSCTS) {
2365                 config->rts = SP_RTS_FLOW_CONTROL;
2366                 config->cts = SP_CTS_FLOW_CONTROL;
2367         } else {
2368                 if (config->rts == SP_RTS_FLOW_CONTROL)
2369                         config->rts = SP_RTS_ON;
2370                 config->cts = SP_CTS_IGNORE;
2371         }
2372
2373         if (flowcontrol == SP_FLOWCONTROL_DTRDSR) {
2374                 config->dtr = SP_DTR_FLOW_CONTROL;
2375                 config->dsr = SP_DSR_FLOW_CONTROL;
2376         } else {
2377                 if (config->dtr == SP_DTR_FLOW_CONTROL)
2378                         config->dtr = SP_DTR_ON;
2379                 config->dsr = SP_DSR_IGNORE;
2380         }
2381
2382         RETURN_OK();
2383 }
2384
2385 SP_API enum sp_return sp_set_flowcontrol(struct sp_port *port,
2386                                          enum sp_flowcontrol flowcontrol)
2387 {
2388         struct port_data data;
2389         struct sp_port_config config;
2390
2391         TRACE("%p, %d", port, flowcontrol);
2392
2393         CHECK_OPEN_PORT();
2394
2395         TRY(get_config(port, &data, &config));
2396
2397         TRY(sp_set_config_flowcontrol(&config, flowcontrol));
2398
2399         TRY(set_config(port, &data, &config));
2400
2401         RETURN_OK();
2402 }
2403
2404 SP_API enum sp_return sp_get_signals(struct sp_port *port,
2405                                      enum sp_signal *signals)
2406 {
2407         TRACE("%p, %p", port, signals);
2408
2409         CHECK_OPEN_PORT();
2410
2411         if (!signals)
2412                 RETURN_ERROR(SP_ERR_ARG, "Null result pointer");
2413
2414         DEBUG_FMT("Getting control signals for port %s", port->name);
2415
2416         *signals = 0;
2417 #ifdef _WIN32
2418         DWORD bits;
2419         if (GetCommModemStatus(port->hdl, &bits) == 0)
2420                 RETURN_FAIL("GetCommModemStatus() failed");
2421         if (bits & MS_CTS_ON)
2422                 *signals |= SP_SIG_CTS;
2423         if (bits & MS_DSR_ON)
2424                 *signals |= SP_SIG_DSR;
2425         if (bits & MS_RLSD_ON)
2426                 *signals |= SP_SIG_DCD;
2427         if (bits & MS_RING_ON)
2428                 *signals |= SP_SIG_RI;
2429 #else
2430         int bits;
2431         if (ioctl(port->fd, TIOCMGET, &bits) < 0)
2432                 RETURN_FAIL("TIOCMGET ioctl failed");
2433         if (bits & TIOCM_CTS)
2434                 *signals |= SP_SIG_CTS;
2435         if (bits & TIOCM_DSR)
2436                 *signals |= SP_SIG_DSR;
2437         if (bits & TIOCM_CAR)
2438                 *signals |= SP_SIG_DCD;
2439         if (bits & TIOCM_RNG)
2440                 *signals |= SP_SIG_RI;
2441 #endif
2442         RETURN_OK();
2443 }
2444
2445 SP_API enum sp_return sp_start_break(struct sp_port *port)
2446 {
2447         TRACE("%p", port);
2448
2449         CHECK_OPEN_PORT();
2450 #ifdef _WIN32
2451         if (SetCommBreak(port->hdl) == 0)
2452                 RETURN_FAIL("SetCommBreak() failed");
2453 #else
2454         if (ioctl(port->fd, TIOCSBRK, 1) < 0)
2455                 RETURN_FAIL("TIOCSBRK ioctl failed");
2456 #endif
2457
2458         RETURN_OK();
2459 }
2460
2461 SP_API enum sp_return sp_end_break(struct sp_port *port)
2462 {
2463         TRACE("%p", port);
2464
2465         CHECK_OPEN_PORT();
2466 #ifdef _WIN32
2467         if (ClearCommBreak(port->hdl) == 0)
2468                 RETURN_FAIL("ClearCommBreak() failed");
2469 #else
2470         if (ioctl(port->fd, TIOCCBRK, 1) < 0)
2471                 RETURN_FAIL("TIOCCBRK ioctl failed");
2472 #endif
2473
2474         RETURN_OK();
2475 }
2476
2477 SP_API int sp_last_error_code(void)
2478 {
2479         TRACE_VOID();
2480 #ifdef _WIN32
2481         RETURN_INT(GetLastError());
2482 #else
2483         RETURN_INT(errno);
2484 #endif
2485 }
2486
2487 SP_API char *sp_last_error_message(void)
2488 {
2489         TRACE_VOID();
2490
2491 #ifdef _WIN32
2492         char *message;
2493         DWORD error = GetLastError();
2494
2495         DWORD length = FormatMessageA(
2496                 FORMAT_MESSAGE_ALLOCATE_BUFFER |
2497                 FORMAT_MESSAGE_FROM_SYSTEM |
2498                 FORMAT_MESSAGE_IGNORE_INSERTS,
2499                 NULL,
2500                 error,
2501                 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
2502                 (LPSTR) &message,
2503                 0, NULL );
2504
2505         if (length >= 2 && message[length - 2] == '\r')
2506                 message[length - 2] = '\0';
2507
2508         RETURN_STRING(message);
2509 #else
2510         RETURN_STRING(strerror(errno));
2511 #endif
2512 }
2513
2514 SP_API void sp_free_error_message(char *message)
2515 {
2516         TRACE("%s", message);
2517
2518 #ifdef _WIN32
2519         LocalFree(message);
2520 #else
2521         (void)message;
2522 #endif
2523
2524         RETURN();
2525 }
2526
2527 SP_API void sp_set_debug_handler(void (*handler)(const char *format, ...))
2528 {
2529         TRACE("%p", handler);
2530
2531         sp_debug_handler = handler;
2532
2533         RETURN();
2534 }
2535
2536 SP_API void sp_default_debug_handler(const char *format, ...)
2537 {
2538         va_list args;
2539         va_start(args, format);
2540         if (getenv("LIBSERIALPORT_DEBUG")) {
2541                 fputs("sp: ", stderr);
2542                 vfprintf(stderr, format, args);
2543         }
2544         va_end(args);
2545 }
2546
2547 SP_API int sp_get_major_package_version(void)
2548 {
2549         return SP_PACKAGE_VERSION_MAJOR;
2550 }
2551
2552 SP_API int sp_get_minor_package_version(void)
2553 {
2554         return SP_PACKAGE_VERSION_MINOR;
2555 }
2556
2557 SP_API int sp_get_micro_package_version(void)
2558 {
2559         return SP_PACKAGE_VERSION_MICRO;
2560 }
2561
2562 SP_API const char *sp_get_package_version_string(void)
2563 {
2564         return SP_PACKAGE_VERSION_STRING;
2565 }
2566
2567 SP_API int sp_get_current_lib_version(void)
2568 {
2569         return SP_LIB_VERSION_CURRENT;
2570 }
2571
2572 SP_API int sp_get_revision_lib_version(void)
2573 {
2574         return SP_LIB_VERSION_REVISION;
2575 }
2576
2577 SP_API int sp_get_age_lib_version(void)
2578 {
2579         return SP_LIB_VERSION_AGE;
2580 }
2581
2582 SP_API const char *sp_get_lib_version_string(void)
2583 {
2584         return SP_LIB_VERSION_STRING;
2585 }
2586
2587 /** @} */