git.gag.com Git - fw/altos/blob - libaltos/libaltos.c

   1 /*
   2  * Copyright © 2010 Keith Packard <keithp@keithp.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  16  */
  17
  18 #include "libaltos.h"
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22
  23 #define BLUETOOTH_PRODUCT_TELEBT        "TeleBT"
  24
  25 #define USE_POLL
  26
  27 PUBLIC int
  28 altos_init(void)
  29 {
  30         return LIBALTOS_SUCCESS;
  31 }
  32
  33 PUBLIC void
  34 altos_fini(void)
  35 {
  36 }
  37
  38 static struct altos_error last_error;
  39
  40 static void
  41 altos_set_last_error(int code, char *string)
  42 {
  43         last_error.code = code;
  44         strncpy(last_error.string, string, sizeof (last_error.string) -1);
  45         last_error.string[sizeof(last_error.string)-1] = '\0';
  46 }
  47
  48 PUBLIC void
  49 altos_get_last_error(struct altos_error *error)
  50 {
  51         *error = last_error;
  52 }
  53
  54 #ifdef DARWIN
  55
  56 #undef USE_POLL
  57
  58 /* Mac OS X don't have strndup even if _GNU_SOURCE is defined */
  59 static char *
  60 altos_strndup (const char *s, size_t n)
  61 {
  62     size_t len = strlen (s);
  63     char *ret;
  64
  65     if (len <= n)
  66        return strdup (s);
  67     ret = malloc(n + 1);
  68     strncpy(ret, s, n);
  69     ret[n] = '\0';
  70     return ret;
  71 }
  72
  73 #else
  74 #define altos_strndup strndup
  75 #endif
  76
  77 #ifdef POSIX_TTY
  78
  79 #include <stdio.h>
  80 #include <stdlib.h>
  81 #include <fcntl.h>
  82 #include <termios.h>
  83 #include <errno.h>
  84
  85 #define USB_BUF_SIZE    64
  86
  87 struct altos_file {
  88         int                             fd;
  89 #ifdef USE_POLL
  90         int                             pipe[2];
  91 #else
  92         int                             out_fd;
  93 #endif
  94         unsigned char                   out_data[USB_BUF_SIZE];
  95         int                             out_used;
  96         unsigned char                   in_data[USB_BUF_SIZE];
  97         int                             in_used;
  98         int                             in_read;
  99 };
 100
 101 static void
 102 altos_set_last_posix_error(void)
 103 {
 104         altos_set_last_error(errno, strerror(errno));
 105 }
 106
 107 PUBLIC struct altos_file *
 108 altos_open(struct altos_device *device)
 109 {
 110         struct altos_file       *file = calloc (sizeof (struct altos_file), 1);
 111         int                     ret;
 112         struct termios          term;
 113
 114         if (!file) {
 115                 altos_set_last_posix_error();
 116                 return NULL;
 117         }
 118
 119 //      altos_set_last_error(12, "yeah yeah, failed again");
 120 //      free(file);
 121 //      return NULL;
 122
 123         file->fd = open(device->path, O_RDWR | O_NOCTTY);
 124         if (file->fd < 0) {
 125                 altos_set_last_posix_error();
 126                 free(file);
 127                 return NULL;
 128         }
 129 #ifdef USE_POLL
 130         pipe(file->pipe);
 131 #else
 132         file->out_fd = open(device->path, O_RDWR | O_NOCTTY);
 133         if (file->out_fd < 0) {
 134                 altos_set_last_posix_error();
 135                 close(file->fd);
 136                 free(file);
 137                 return NULL;
 138         }
 139 #endif
 140         ret = tcgetattr(file->fd, &term);
 141         if (ret < 0) {
 142                 altos_set_last_posix_error();
 143                 close(file->fd);
 144 #ifndef USE_POLL
 145                 close(file->out_fd);
 146 #endif
 147                 free(file);
 148                 return NULL;
 149         }
 150         cfmakeraw(&term);
 151         cfsetospeed(&term, B9600);
 152         cfsetispeed(&term, B9600);
 153 #ifdef USE_POLL
 154         term.c_cc[VMIN] = 1;
 155         term.c_cc[VTIME] = 0;
 156 #else
 157         term.c_cc[VMIN] = 0;
 158         term.c_cc[VTIME] = 1;
 159 #endif
 160         ret = tcsetattr(file->fd, TCSAFLUSH, &term);
 161         if (ret < 0) {
 162                 altos_set_last_posix_error();
 163                 close(file->fd);
 164 #ifndef USE_POLL
 165                 close(file->out_fd);
 166 #endif
 167                 free(file);
 168                 return NULL;
 169         }
 170         return file;
 171 }
 172
 173 PUBLIC void
 174 altos_close(struct altos_file *file)
 175 {
 176         if (file->fd != -1) {
 177                 int     fd = file->fd;
 178                 file->fd = -1;
 179 #ifdef USE_POLL
 180                 write(file->pipe[1], "\r", 1);
 181 #else
 182                 close(file->out_fd);
 183                 file->out_fd = -1;
 184 #endif
 185                 close(fd);
 186         }
 187 }
 188
 189 PUBLIC void
 190 altos_free(struct altos_file *file)
 191 {
 192         altos_close(file);
 193         free(file);
 194 }
 195
 196 PUBLIC int
 197 altos_flush(struct altos_file *file)
 198 {
 199         if (file->out_used && 0) {
 200                 printf ("flush \"");
 201                 fwrite(file->out_data, 1, file->out_used, stdout);
 202                 printf ("\"\n");
 203         }
 204         while (file->out_used) {
 205                 int     ret;
 206
 207                 if (file->fd < 0)
 208                         return -EBADF;
 209 #ifdef USE_POLL
 210                 ret = write (file->fd, file->out_data, file->out_used);
 211 #else
 212                 ret = write (file->out_fd, file->out_data, file->out_used);
 213 #endif
 214                 if (ret < 0) {
 215                         altos_set_last_posix_error();
 216                         return -last_error.code;
 217                 }
 218                 if (ret) {
 219                         memmove(file->out_data, file->out_data + ret,
 220                                 file->out_used - ret);
 221                         file->out_used -= ret;
 222                 }
 223         }
 224         return 0;
 225 }
 226
 227 PUBLIC int
 228 altos_putchar(struct altos_file *file, char c)
 229 {
 230         int     ret;
 231
 232         if (file->out_used == USB_BUF_SIZE) {
 233                 ret = altos_flush(file);
 234                 if (ret) {
 235                         return ret;
 236                 }
 237         }
 238         file->out_data[file->out_used++] = c;
 239         ret = 0;
 240         if (file->out_used == USB_BUF_SIZE)
 241                 ret = altos_flush(file);
 242         return ret;
 243 }
 244
 245 #ifdef USE_POLL
 246 #include <poll.h>
 247 #endif
 248
 249 static int
 250 altos_fill(struct altos_file *file, int timeout)
 251 {
 252         int             ret;
 253 #ifdef USE_POLL
 254         struct pollfd   fd[2];
 255 #endif
 256
 257         if (timeout == 0)
 258                 timeout = -1;
 259         while (file->in_read == file->in_used) {
 260                 if (file->fd < 0)
 261                         return LIBALTOS_ERROR;
 262 #ifdef USE_POLL
 263                 fd[0].fd = file->fd;
 264                 fd[0].events = POLLIN|POLLERR|POLLHUP|POLLNVAL;
 265                 fd[1].fd = file->pipe[0];
 266                 fd[1].events = POLLIN;
 267                 ret = poll(fd, 2, timeout);
 268                 if (ret < 0) {
 269                         altos_set_last_posix_error();
 270                         return LIBALTOS_ERROR;
 271                 }
 272                 if (ret == 0)
 273                         return LIBALTOS_TIMEOUT;
 274
 275                 if (fd[0].revents & (POLLHUP|POLLERR|POLLNVAL))
 276                         return LIBALTOS_ERROR;
 277                 if (fd[0].revents & POLLIN)
 278 #endif
 279                 {
 280                         ret = read(file->fd, file->in_data, USB_BUF_SIZE);
 281                         if (ret < 0) {
 282                                 altos_set_last_posix_error();
 283                                 return LIBALTOS_ERROR;
 284                         }
 285                         file->in_read = 0;
 286                         file->in_used = ret;
 287 #ifndef USE_POLL
 288                         if (ret == 0 && timeout > 0)
 289                                 return LIBALTOS_TIMEOUT;
 290 #endif
 291                 }
 292         }
 293         if (file->in_used && 0) {
 294                 printf ("fill \"");
 295                 fwrite(file->in_data, 1, file->in_used, stdout);
 296                 printf ("\"\n");
 297         }
 298         return 0;
 299 }
 300
 301 PUBLIC int
 302 altos_getchar(struct altos_file *file, int timeout)
 303 {
 304         int     ret;
 305         while (file->in_read == file->in_used) {
 306                 if (file->fd < 0)
 307                         return LIBALTOS_ERROR;
 308                 ret = altos_fill(file, timeout);
 309                 if (ret)
 310                         return ret;
 311         }
 312         return file->in_data[file->in_read++];
 313 }
 314
 315 #endif /* POSIX_TTY */
 316
 317 /*
 318  * Scan for Altus Metrum devices by looking through /sys
 319  */
 320
 321 #ifdef LINUX
 322
 323 #define _GNU_SOURCE
 324 #include <ctype.h>
 325 #include <dirent.h>
 326 #include <stdio.h>
 327 #include <stdlib.h>
 328 #include <string.h>
 329 #include <bluetooth/bluetooth.h>
 330 #include <bluetooth/hci.h>
 331 #include <bluetooth/hci_lib.h>
 332 #include <bluetooth/rfcomm.h>
 333
 334 static char *
 335 cc_fullname (char *dir, char *file)
 336 {
 337         char    *new;
 338         int     dlen = strlen (dir);
 339         int     flen = strlen (file);
 340         int     slen = 0;
 341
 342         if (dir[dlen-1] != '/')
 343                 slen = 1;
 344         new = malloc (dlen + slen + flen + 1);
 345         if (!new)
 346                 return 0;
 347         strcpy(new, dir);
 348         if (slen)
 349                 strcat (new, "/");
 350         strcat(new, file);
 351         return new;
 352 }
 353
 354 static char *
 355 cc_basename(char *file)
 356 {
 357         char *b;
 358
 359         b = strrchr(file, '/');
 360         if (!b)
 361                 return file;
 362         return b + 1;
 363 }
 364
 365 static char *
 366 load_string(char *dir, char *file)
 367 {
 368         char    *full = cc_fullname(dir, file);
 369         char    line[4096];
 370         char    *r;
 371         FILE    *f;
 372         int     rlen;
 373
 374         f = fopen(full, "r");
 375         free(full);
 376         if (!f)
 377                 return NULL;
 378         r = fgets(line, sizeof (line), f);
 379         fclose(f);
 380         if (!r)
 381                 return NULL;
 382         rlen = strlen(r);
 383         if (r[rlen-1] == '\n')
 384                 r[rlen-1] = '\0';
 385         return strdup(r);
 386 }
 387
 388 static int
 389 load_hex(char *dir, char *file)
 390 {
 391         char    *line;
 392         char    *end;
 393         long    i;
 394
 395         line = load_string(dir, file);
 396         if (!line)
 397                 return -1;
 398         i = strtol(line, &end, 16);
 399         free(line);
 400         if (end == line)
 401                 return -1;
 402         return i;
 403 }
 404
 405 static int
 406 load_dec(char *dir, char *file)
 407 {
 408         char    *line;
 409         char    *end;
 410         long    i;
 411
 412         line = load_string(dir, file);
 413         if (!line)
 414                 return -1;
 415         i = strtol(line, &end, 10);
 416         free(line);
 417         if (end == line)
 418                 return -1;
 419         return i;
 420 }
 421
 422 static int
 423 dir_filter_tty_colon(const struct dirent *d)
 424 {
 425         return strncmp(d->d_name, "tty:", 4) == 0;
 426 }
 427
 428 static int
 429 dir_filter_tty(const struct dirent *d)
 430 {
 431         return strncmp(d->d_name, "tty", 3) == 0;
 432 }
 433
 434 struct altos_usbdev {
 435         char    *sys;
 436         char    *tty;
 437         char    *manufacturer;
 438         char    *product_name;
 439         int     serial; /* AltOS always uses simple integer serial numbers */
 440         int     idProduct;
 441         int     idVendor;
 442 };
 443
 444 static char *
 445 usb_tty(char *sys)
 446 {
 447         char *base;
 448         int num_configs;
 449         int config;
 450         struct dirent **namelist;
 451         int interface;
 452         int num_interfaces;
 453         char endpoint_base[20];
 454         char *endpoint_full;
 455         char *tty_dir;
 456         int ntty;
 457         char *tty;
 458
 459         base = cc_basename(sys);
 460         num_configs = load_hex(sys, "bNumConfigurations");
 461         num_interfaces = load_hex(sys, "bNumInterfaces");
 462         for (config = 1; config <= num_configs; config++) {
 463                 for (interface = 0; interface < num_interfaces; interface++) {
 464                         sprintf(endpoint_base, "%s:%d.%d",
 465                                 base, config, interface);
 466                         endpoint_full = cc_fullname(sys, endpoint_base);
 467
 468
 469                         /* Check for tty:ttyACMx style names
 470                          */
 471                         ntty = scandir(endpoint_full, &namelist,
 472                                        dir_filter_tty_colon,
 473                                        alphasort);
 474                         if (ntty > 0) {
 475                                 free(endpoint_full);
 476                                 tty = cc_fullname("/dev", namelist[0]->d_name + 4);
 477                                 free(namelist);
 478                                 return tty;
 479                         }
 480
 481                         /* Check for tty/ttyACMx style names
 482                          */
 483                         tty_dir = cc_fullname(endpoint_full, "tty");
 484                         ntty = scandir(tty_dir, &namelist,
 485                                        dir_filter_tty,
 486                                        alphasort);
 487                         free (tty_dir);
 488                         if (ntty > 0) {
 489                                 tty = cc_fullname("/dev", namelist[0]->d_name);
 490                                 free(endpoint_full);
 491                                 free(namelist);
 492                                 return tty;
 493                         }
 494
 495                         /* Check for ttyACMx style names
 496                          */
 497                         ntty = scandir(endpoint_full, &namelist,
 498                                        dir_filter_tty,
 499                                        alphasort);
 500                         free(endpoint_full);
 501                         if (ntty > 0) {
 502                                 tty = cc_fullname("/dev", namelist[0]->d_name);
 503                                 free(namelist);
 504                                 return tty;
 505                         }
 506
 507                 }
 508         }
 509         return NULL;
 510 }
 511
 512 static struct altos_usbdev *
 513 usb_scan_device(char *sys)
 514 {
 515         struct altos_usbdev *usbdev;
 516         char *tty;
 517
 518         tty = usb_tty(sys);
 519         if (!tty)
 520                 return NULL;
 521         usbdev = calloc(1, sizeof (struct altos_usbdev));
 522         if (!usbdev)
 523                 return NULL;
 524         usbdev->sys = strdup(sys);
 525         usbdev->manufacturer = load_string(sys, "manufacturer");
 526         usbdev->product_name = load_string(sys, "product");
 527         usbdev->serial = load_dec(sys, "serial");
 528         usbdev->idProduct = load_hex(sys, "idProduct");
 529         usbdev->idVendor = load_hex(sys, "idVendor");
 530         usbdev->tty = tty;
 531         return usbdev;
 532 }
 533
 534 static void
 535 usbdev_free(struct altos_usbdev *usbdev)
 536 {
 537         free(usbdev->sys);
 538         free(usbdev->manufacturer);
 539         free(usbdev->product_name);
 540         /* this can get used as a return value */
 541         if (usbdev->tty)
 542                 free(usbdev->tty);
 543         free(usbdev);
 544 }
 545
 546 #define USB_DEVICES     "/sys/bus/usb/devices"
 547
 548 static int
 549 dir_filter_dev(const struct dirent *d)
 550 {
 551         const char      *n = d->d_name;
 552         char    c;
 553
 554         while ((c = *n++)) {
 555                 if (isdigit(c))
 556                         continue;
 557                 if (c == '-')
 558                         continue;
 559                 if (c == '.' && n != d->d_name + 1)
 560                         continue;
 561                 return 0;
 562         }
 563         return 1;
 564 }
 565
 566 struct altos_list {
 567         struct altos_usbdev     **dev;
 568         int                     current;
 569         int                     ndev;
 570 };
 571
 572 struct altos_list *
 573 altos_list_start(void)
 574 {
 575         int                     e;
 576         struct dirent           **ents;
 577         char                    *dir;
 578         struct altos_usbdev     *dev;
 579         struct altos_list       *devs;
 580         int                     n;
 581
 582         devs = calloc(1, sizeof (struct altos_list));
 583         if (!devs)
 584                 return NULL;
 585
 586         n = scandir (USB_DEVICES, &ents,
 587                      dir_filter_dev,
 588                      alphasort);
 589         if (!n)
 590                 return 0;
 591         for (e = 0; e < n; e++) {
 592                 dir = cc_fullname(USB_DEVICES, ents[e]->d_name);
 593                 dev = usb_scan_device(dir);
 594                 if (!dev)
 595                         continue;
 596                 free(dir);
 597                 if (devs->dev)
 598                         devs->dev = realloc(devs->dev,
 599                                             (devs->ndev + 1) * sizeof (struct usbdev *));
 600                 else
 601                         devs->dev = malloc (sizeof (struct usbdev *));
 602                 devs->dev[devs->ndev++] = dev;
 603         }
 604         free(ents);
 605         devs->current = 0;
 606         return devs;
 607 }
 608
 609 PUBLIC struct altos_list *
 610 altos_ftdi_list_start(void)
 611 {
 612         return altos_list_start();
 613 }
 614
 615 int
 616 altos_list_next(struct altos_list *list, struct altos_device *device)
 617 {
 618         struct altos_usbdev *dev;
 619         if (list->current >= list->ndev) {
 620                 return 0;
 621         }
 622         dev = list->dev[list->current];
 623         strcpy(device->name, dev->product_name);
 624         device->vendor = dev->idVendor;
 625         device->product = dev->idProduct;
 626         strcpy(device->path, dev->tty);
 627         device->serial = dev->serial;
 628         list->current++;
 629         return 1;
 630 }
 631
 632 void
 633 altos_list_finish(struct altos_list *usbdevs)
 634 {
 635         int     i;
 636
 637         if (!usbdevs)
 638                 return;
 639         for (i = 0; i < usbdevs->ndev; i++)
 640                 usbdev_free(usbdevs->dev[i]);
 641         free(usbdevs);
 642 }
 643
 644 struct altos_bt_list {
 645         inquiry_info    *ii;
 646         int             sock;
 647         int             dev_id;
 648         int             rsp;
 649         int             num_rsp;
 650 };
 651
 652 #define INQUIRY_MAX_RSP 255
 653
 654 struct altos_bt_list *
 655 altos_bt_list_start(int inquiry_time)
 656 {
 657         struct altos_bt_list    *bt_list;
 658
 659         bt_list = calloc(1, sizeof (struct altos_bt_list));
 660         if (!bt_list)
 661                 goto no_bt_list;
 662
 663         bt_list->ii = calloc(INQUIRY_MAX_RSP, sizeof (inquiry_info));
 664         if (!bt_list->ii)
 665                 goto no_ii;
 666         bt_list->dev_id = hci_get_route(NULL);
 667         if (bt_list->dev_id < 0)
 668                 goto no_dev_id;
 669
 670         bt_list->sock = hci_open_dev(bt_list->dev_id);
 671         if (bt_list->sock < 0)
 672                 goto no_sock;
 673
 674         bt_list->num_rsp = hci_inquiry(bt_list->dev_id,
 675                                        inquiry_time,
 676                                        INQUIRY_MAX_RSP,
 677                                        NULL,
 678                                        &bt_list->ii,
 679                                        IREQ_CACHE_FLUSH);
 680         if (bt_list->num_rsp < 0)
 681                 goto no_rsp;
 682
 683         bt_list->rsp = 0;
 684         return bt_list;
 685
 686 no_rsp:
 687         close(bt_list->sock);
 688 no_sock:
 689 no_dev_id:
 690         free(bt_list->ii);
 691 no_ii:
 692         free(bt_list);
 693 no_bt_list:
 694         return NULL;
 695 }
 696
 697 int
 698 altos_bt_list_next(struct altos_bt_list *bt_list,
 699                    struct altos_bt_device *device)
 700 {
 701         inquiry_info    *ii;
 702
 703         if (bt_list->rsp >= bt_list->num_rsp)
 704                 return 0;
 705
 706         ii = &bt_list->ii[bt_list->rsp];
 707         ba2str(&ii->bdaddr, device->addr);
 708         memset(&device->name, '\0', sizeof (device->name));
 709         if (hci_read_remote_name(bt_list->sock, &ii->bdaddr,
 710                                  sizeof (device->name),
 711                                  device->name, 0) < 0) {
 712                 strcpy(device->name, "[unknown]");
 713         }
 714         bt_list->rsp++;
 715         return 1;
 716 }
 717
 718 void
 719 altos_bt_list_finish(struct altos_bt_list *bt_list)
 720 {
 721         close(bt_list->sock);
 722         free(bt_list->ii);
 723         free(bt_list);
 724 }
 725
 726 void
 727 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
 728 {
 729         strncpy(device->name, name, sizeof (device->name));
 730         device->name[sizeof(device->name)-1] = '\0';
 731         strncpy(device->addr, addr, sizeof (device->addr));
 732         device->addr[sizeof(device->addr)-1] = '\0';
 733 }
 734
 735 struct altos_file *
 736 altos_bt_open(struct altos_bt_device *device)
 737 {
 738         struct sockaddr_rc addr = { 0 };
 739         int     status, i;
 740         struct altos_file *file;
 741
 742         file = calloc(1, sizeof (struct altos_file));
 743         if (!file)
 744                 goto no_file;
 745         addr.rc_family = AF_BLUETOOTH;
 746         addr.rc_channel = 1;
 747         str2ba(device->addr, &addr.rc_bdaddr);
 748
 749         for (i = 0; i < 5; i++) {
 750                 file->fd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
 751                 if (file->fd < 0) {
 752                         altos_set_last_posix_error();
 753                         goto no_sock;
 754                 }
 755
 756                 status = connect(file->fd,
 757                                  (struct sockaddr *)&addr,
 758                                  sizeof(addr));
 759                 if (status >= 0 || errno != EBUSY)
 760                         break;
 761                 close(file->fd);
 762                 usleep(100 * 1000);
 763         }
 764         if (status < 0) {
 765                 altos_set_last_posix_error();
 766                 goto no_link;
 767         }
 768
 769 #ifdef USE_POLL
 770         pipe(file->pipe);
 771 #else
 772         file->out_fd = dup(file->fd);
 773 #endif
 774         return file;
 775 no_link:
 776         close(file->fd);
 777 no_sock:
 778         free(file);
 779 no_file:
 780         return NULL;
 781 }
 782
 783 #endif
 784
 785 #ifdef DARWIN
 786
 787 #include <IOKitLib.h>
 788 #include <IOKit/usb/USBspec.h>
 789 #include <sys/param.h>
 790 #include <paths.h>
 791 #include <CFNumber.h>
 792 #include <IOBSD.h>
 793 #include <string.h>
 794 #include <stdio.h>
 795 #include <stdlib.h>
 796
 797 struct altos_list {
 798         io_iterator_t iterator;
 799         int ftdi;
 800 };
 801
 802 static int
 803 get_string(io_object_t object, CFStringRef entry, char *result, int result_len)
 804 {
 805         CFTypeRef entry_as_string;
 806         Boolean got_string;
 807
 808         entry_as_string = IORegistryEntrySearchCFProperty (object,
 809                                                            kIOServicePlane,
 810                                                            entry,
 811                                                            kCFAllocatorDefault,
 812                                                            kIORegistryIterateRecursively);
 813         if (entry_as_string) {
 814                 got_string = CFStringGetCString(entry_as_string,
 815                                                 result, result_len,
 816                                                 kCFStringEncodingASCII);
 817
 818                 CFRelease(entry_as_string);
 819                 if (got_string)
 820                         return 1;
 821         }
 822         return 0;
 823 }
 824
 825 static int
 826 get_number(io_object_t object, CFStringRef entry, int *result)
 827 {
 828         CFTypeRef entry_as_number;
 829         Boolean got_number;
 830
 831         entry_as_number = IORegistryEntrySearchCFProperty (object,
 832                                                            kIOServicePlane,
 833                                                            entry,
 834                                                            kCFAllocatorDefault,
 835                                                            kIORegistryIterateRecursively);
 836         if (entry_as_number) {
 837                 got_number = CFNumberGetValue(entry_as_number,
 838                                               kCFNumberIntType,
 839                                               result);
 840                 if (got_number)
 841                         return 1;
 842         }
 843         return 0;
 844 }
 845
 846 PUBLIC struct altos_list *
 847 altos_list_start(void)
 848 {
 849         struct altos_list *list = calloc (sizeof (struct altos_list), 1);
 850         CFMutableDictionaryRef matching_dictionary = IOServiceMatching("IOUSBDevice");
 851         io_iterator_t tdIterator;
 852         io_object_t tdObject;
 853         kern_return_t ret;
 854         int i;
 855
 856         ret = IOServiceGetMatchingServices(kIOMasterPortDefault, matching_dictionary, &list->iterator);
 857         if (ret != kIOReturnSuccess) {
 858                 free(list);
 859                 return NULL;
 860         }
 861         list->ftdi = 0;
 862         return list;
 863 }
 864
 865 PUBLIC struct altos_list *
 866 altos_ftdi_list_start(void)
 867 {
 868         struct altos_list *list = altos_list_start();
 869
 870         if (list)
 871                 list->ftdi = 1;
 872         return list;
 873 }
 874
 875 PUBLIC int
 876 altos_list_next(struct altos_list *list, struct altos_device *device)
 877 {
 878         io_object_t object;
 879         char serial_string[128];
 880
 881         for (;;) {
 882                 object = IOIteratorNext(list->iterator);
 883                 if (!object)
 884                         return 0;
 885
 886                 if (!get_number (object, CFSTR(kUSBVendorID), &device->vendor) ||
 887                     !get_number (object, CFSTR(kUSBProductID), &device->product))
 888                         continue;
 889                 if (list->ftdi) {
 890                         if (device->vendor != 0x0403)
 891                                 continue;
 892                 } else {
 893                         if (device->vendor != 0xfffe)
 894                                 continue;
 895                         if (device->product < 0x000a || 0x0013 < device->product)
 896                                 continue;
 897                 }
 898                 if (get_string (object, CFSTR("IOCalloutDevice"), device->path, sizeof (device->path)) &&
 899                     get_string (object, CFSTR("USB Product Name"), device->name, sizeof (device->name)) &&
 900                     get_string (object, CFSTR("USB Serial Number"), serial_string, sizeof (serial_string))) {
 901                         device->serial = atoi(serial_string);
 902                         return 1;
 903                 }
 904         }
 905 }
 906
 907 PUBLIC void
 908 altos_list_finish(struct altos_list *list)
 909 {
 910         IOObjectRelease (list->iterator);
 911         free(list);
 912 }
 913
 914 struct altos_bt_list {
 915         int             sock;
 916         int             dev_id;
 917         int             rsp;
 918         int             num_rsp;
 919 };
 920
 921 #define INQUIRY_MAX_RSP 255
 922
 923 struct altos_bt_list *
 924 altos_bt_list_start(int inquiry_time)
 925 {
 926         return NULL;
 927 }
 928
 929 int
 930 altos_bt_list_next(struct altos_bt_list *bt_list,
 931                    struct altos_bt_device *device)
 932 {
 933         return 0;
 934 }
 935
 936 void
 937 altos_bt_list_finish(struct altos_bt_list *bt_list)
 938 {
 939 }
 940
 941 void
 942 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
 943 {
 944         strncpy(device->name, name, sizeof (device->name));
 945         device->name[sizeof(device->name)-1] = '\0';
 946         strncpy(device->addr, addr, sizeof (device->addr));
 947         device->addr[sizeof(device->addr)-1] = '\0';
 948 }
 949
 950 struct altos_file *
 951 altos_bt_open(struct altos_bt_device *device)
 952 {
 953         return NULL;
 954 }
 955
 956 #endif
 957
 958
 959 #ifdef WINDOWS
 960
 961 #include <stdlib.h>
 962 #include <windows.h>
 963 #include <setupapi.h>
 964
 965 struct altos_list {
 966         HDEVINFO        dev_info;
 967         int             index;
 968         int             ftdi;
 969 };
 970
 971 #define USB_BUF_SIZE    64
 972
 973 struct altos_file {
 974         HANDLE                          handle;
 975         unsigned char                   out_data[USB_BUF_SIZE];
 976         int                             out_used;
 977         unsigned char                   in_data[USB_BUF_SIZE];
 978         int                             in_used;
 979         int                             in_read;
 980         OVERLAPPED                      ov_read;
 981         BOOL                            pend_read;
 982         OVERLAPPED                      ov_write;
 983 };
 984
 985 #include <stdarg.h>
 986
 987 static void
 988 log_message(char *fmt, ...)
 989 {
 990         static  FILE *log = NULL;
 991         va_list a;
 992
 993         if (!log)
 994                 log = fopen("\\temp\\altos.txt", "w");
 995         if (log) {
 996                 va_start(a, fmt);
 997                 vfprintf(log, fmt, a);
 998                 va_end(a);
 999                 fflush(log);
1000         }
1001 }
1002
1003 static void
1004 _altos_set_last_windows_error(char *file, int line)
1005 {
1006         DWORD   error = GetLastError();
1007         TCHAR   message[1024];
1008         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
1009                       0,
1010                       error,
1011                       0,
1012                       message,
1013                       sizeof (message) / sizeof (TCHAR),
1014                       NULL);
1015         if (error != ERROR_SUCCESS)
1016                 log_message ("%s:%d %s\n", file, line, message);
1017         altos_set_last_error(error, message);
1018 }
1019
1020 #define altos_set_last_windows_error() _altos_set_last_windows_error(__FILE__, __LINE__)
1021
1022 PUBLIC struct altos_list *
1023 altos_list_start(void)
1024 {
1025         struct altos_list       *list = calloc(1, sizeof (struct altos_list));
1026
1027         if (!list)
1028                 return NULL;
1029         list->dev_info = SetupDiGetClassDevs(NULL, "USB", NULL,
1030                                              DIGCF_ALLCLASSES|DIGCF_PRESENT);
1031         if (list->dev_info == INVALID_HANDLE_VALUE) {
1032                 altos_set_last_windows_error();
1033                 free(list);
1034                 return NULL;
1035         }
1036         list->index = 0;
1037         list->ftdi = 0;
1038         return list;
1039 }
1040
1041 PUBLIC struct altos_list *
1042 altos_ftdi_list_start(void)
1043 {
1044         struct altos_list       *list = calloc(1, sizeof (struct altos_list));
1045
1046         if (!list)
1047                 return NULL;
1048         list->dev_info = SetupDiGetClassDevs(NULL, "FTDIBUS", NULL,
1049                                              DIGCF_ALLCLASSES|DIGCF_PRESENT);
1050         if (list->dev_info == INVALID_HANDLE_VALUE) {
1051                 altos_set_last_windows_error();
1052                 free(list);
1053                 return NULL;
1054         }
1055         list->index = 0;
1056         list->ftdi = 1;
1057         return list;
1058 }
1059
1060 PUBLIC int
1061 altos_list_next(struct altos_list *list, struct altos_device *device)
1062 {
1063         SP_DEVINFO_DATA dev_info_data;
1064         BYTE            port[128];
1065         DWORD           port_len;
1066         char            friendlyname[256];
1067         BYTE            symbolic[256];
1068         DWORD           symbolic_len;
1069         HKEY            dev_key;
1070         unsigned int    vid, pid;
1071         int             serial;
1072         HRESULT         result;
1073         DWORD           friendlyname_type;
1074         DWORD           friendlyname_len;
1075
1076         dev_info_data.cbSize = sizeof (SP_DEVINFO_DATA);
1077         while(SetupDiEnumDeviceInfo(list->dev_info, list->index,
1078                                     &dev_info_data))
1079         {
1080                 list->index++;
1081
1082                 dev_key = SetupDiOpenDevRegKey(list->dev_info, &dev_info_data,
1083                                                DICS_FLAG_GLOBAL, 0, DIREG_DEV,
1084                                                KEY_READ);
1085                 if (dev_key == INVALID_HANDLE_VALUE) {
1086                         altos_set_last_windows_error();
1087                         continue;
1088                 }
1089
1090                 if (list->ftdi) {
1091                         vid = 0x0403;
1092                         pid = 0x6015;
1093                         serial = 0;
1094                 } else {
1095                         /* Fetch symbolic name for this device and parse out
1096                          * the vid/pid/serial info */
1097                         symbolic_len = sizeof(symbolic);
1098                         result = RegQueryValueEx(dev_key, "SymbolicName", NULL, NULL,
1099                                                  symbolic, &symbolic_len);
1100                         if (result != 0) {
1101                                 altos_set_last_windows_error();
1102                                 RegCloseKey(dev_key);
1103                                 continue;
1104                         }
1105                         vid = pid = serial = 0;
1106                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_") - 1,
1107                                "%04X", &vid);
1108                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_") - 1,
1109                                "%04X", &pid);
1110                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_XXXX#") - 1,
1111                                "%d", &serial);
1112                 }
1113
1114                 /* Fetch the com port name */
1115                 port_len = sizeof (port);
1116                 result = RegQueryValueEx(dev_key, "PortName", NULL, NULL,
1117                                          port, &port_len);
1118                 RegCloseKey(dev_key);
1119                 if (result != 0) {
1120                         altos_set_last_windows_error();
1121                         continue;
1122                 }
1123
1124                 /* Fetch the device description which is the device name,
1125                  * with firmware that has unique USB ids */
1126                 friendlyname_len = sizeof (friendlyname);
1127                 if(!SetupDiGetDeviceRegistryProperty(list->dev_info,
1128                                                      &dev_info_data,
1129                                                      SPDRP_FRIENDLYNAME,
1130                                                      &friendlyname_type,
1131                                                      (BYTE *)friendlyname,
1132                                                      sizeof(friendlyname),
1133                                                      &friendlyname_len))
1134                 {
1135                         altos_set_last_windows_error();
1136                         continue;
1137                 }
1138                 device->vendor = vid;
1139                 device->product = pid;
1140                 device->serial = serial;
1141                 strcpy(device->name, friendlyname);
1142
1143                 strcpy(device->path, (char *) port);
1144                 return 1;
1145         }
1146         result = GetLastError();
1147         if (result != ERROR_NO_MORE_ITEMS)
1148                 altos_set_last_windows_error();
1149         return 0;
1150 }
1151
1152 PUBLIC void
1153 altos_list_finish(struct altos_list *list)
1154 {
1155         SetupDiDestroyDeviceInfoList(list->dev_info);
1156         free(list);
1157 }
1158
1159 static int
1160 altos_queue_read(struct altos_file *file)
1161 {
1162         DWORD   got;
1163         if (file->pend_read)
1164                 return LIBALTOS_SUCCESS;
1165
1166         if (!ReadFile(file->handle, file->in_data, USB_BUF_SIZE, &got, &file->ov_read)) {
1167                 if (GetLastError() != ERROR_IO_PENDING) {
1168                         altos_set_last_windows_error();
1169                         return LIBALTOS_ERROR;
1170                 }
1171                 file->pend_read = TRUE;
1172         } else {
1173                 file->pend_read = FALSE;
1174                 file->in_read = 0;
1175                 file->in_used = got;
1176         }
1177         return LIBALTOS_SUCCESS;
1178 }
1179
1180 static int
1181 altos_wait_read(struct altos_file *file, int timeout)
1182 {
1183         DWORD   ret;
1184         DWORD   got;
1185
1186         if (!file->pend_read)
1187                 return LIBALTOS_SUCCESS;
1188
1189         if (!timeout)
1190                 timeout = INFINITE;
1191
1192         ret = WaitForSingleObject(file->ov_read.hEvent, timeout);
1193         switch (ret) {
1194         case WAIT_OBJECT_0:
1195                 if (!GetOverlappedResult(file->handle, &file->ov_read, &got, FALSE)) {
1196                         altos_set_last_windows_error();
1197                         return LIBALTOS_ERROR;
1198                 }
1199                 file->pend_read = FALSE;
1200                 file->in_read = 0;
1201                 file->in_used = got;
1202                 break;
1203         case WAIT_TIMEOUT:
1204                 return LIBALTOS_TIMEOUT;
1205                 break;
1206         default:
1207                 altos_set_last_windows_error();
1208                 return LIBALTOS_ERROR;
1209         }
1210         return LIBALTOS_SUCCESS;
1211 }
1212
1213 static int
1214 altos_fill(struct altos_file *file, int timeout)
1215 {
1216         int     ret;
1217
1218         if (file->in_read < file->in_used)
1219                 return LIBALTOS_SUCCESS;
1220
1221         file->in_read = file->in_used = 0;
1222
1223         ret = altos_queue_read(file);
1224         if (ret)
1225                 return ret;
1226         ret = altos_wait_read(file, timeout);
1227         if (ret)
1228                 return ret;
1229
1230         return LIBALTOS_SUCCESS;
1231 }
1232
1233 PUBLIC int
1234 altos_flush(struct altos_file *file)
1235 {
1236         DWORD           put;
1237         unsigned char   *data = file->out_data;
1238         int             used = file->out_used;
1239         DWORD           ret;
1240
1241         while (used) {
1242                 if (!WriteFile(file->handle, data, used, &put, &file->ov_write)) {
1243                         if (GetLastError() != ERROR_IO_PENDING) {
1244                                 altos_set_last_windows_error();
1245                                 return LIBALTOS_ERROR;
1246                         }
1247                         ret = WaitForSingleObject(file->ov_write.hEvent, INFINITE);
1248                         switch (ret) {
1249                         case WAIT_OBJECT_0:
1250                                 if (!GetOverlappedResult(file->handle, &file->ov_write, &put, FALSE)) {
1251                                         altos_set_last_windows_error();
1252                                         return LIBALTOS_ERROR;
1253                                 }
1254                                 break;
1255                         default:
1256                                 altos_set_last_windows_error();
1257                                 return LIBALTOS_ERROR;
1258                         }
1259                 }
1260                 data += put;
1261                 used -= put;
1262         }
1263         file->out_used = 0;
1264         return LIBALTOS_SUCCESS;
1265 }
1266
1267 static HANDLE
1268 open_serial(char *full_name)
1269 {
1270         HANDLE  handle;
1271         DCB     dcb;
1272
1273         handle = CreateFile(full_name, GENERIC_READ|GENERIC_WRITE,
1274                             0, NULL, OPEN_EXISTING,
1275                             FILE_FLAG_OVERLAPPED, NULL);
1276
1277         if (handle == INVALID_HANDLE_VALUE) {
1278                 altos_set_last_windows_error();
1279                 return INVALID_HANDLE_VALUE;
1280         }
1281
1282         if (!GetCommState(handle, &dcb)) {
1283                 altos_set_last_windows_error();
1284                 CloseHandle(handle);
1285                 return INVALID_HANDLE_VALUE;
1286         }
1287         dcb.BaudRate = CBR_9600;
1288         dcb.fBinary = TRUE;
1289         dcb.fParity = FALSE;
1290         dcb.fOutxCtsFlow = FALSE;
1291         dcb.fOutxDsrFlow = FALSE;
1292         dcb.fDtrControl = DTR_CONTROL_ENABLE;
1293         dcb.fDsrSensitivity = FALSE;
1294         dcb.fTXContinueOnXoff = FALSE;
1295         dcb.fOutX = FALSE;
1296         dcb.fInX = FALSE;
1297         dcb.fErrorChar = FALSE;
1298         dcb.fNull = FALSE;
1299         dcb.fRtsControl = RTS_CONTROL_ENABLE;
1300         dcb.fAbortOnError = FALSE;
1301         dcb.XonLim = 10;
1302         dcb.XoffLim = 10;
1303         dcb.ByteSize = 8;
1304         dcb.Parity = NOPARITY;
1305         dcb.StopBits = ONESTOPBIT;
1306         dcb.XonChar = 17;
1307         dcb.XoffChar = 19;
1308 #if 0
1309         dcb.ErrorChar = 0;
1310         dcb.EofChar = 0;
1311         dcb.EvtChar = 0;
1312 #endif
1313         if (!SetCommState(handle, &dcb)) {
1314                 altos_set_last_windows_error();
1315                 CloseHandle(handle);
1316                 return INVALID_HANDLE_VALUE;
1317         }
1318         return handle;
1319 }
1320
1321 PUBLIC struct altos_file *
1322 altos_open(struct altos_device *device)
1323 {
1324         struct altos_file       *file = calloc (1, sizeof (struct altos_file));
1325         char    full_name[64];
1326         COMMTIMEOUTS timeouts;
1327         int i;
1328
1329         if (!file)
1330                 return NULL;
1331
1332         strcpy(full_name, "\\\\.\\");
1333         strcat(full_name, device->path);
1334
1335         file->handle = INVALID_HANDLE_VALUE;
1336
1337         for (i = 0; i < 5; i++) {
1338                 file->handle = open_serial(full_name);
1339                 if (file->handle != INVALID_HANDLE_VALUE)
1340                         break;
1341                 Sleep(100);
1342         }
1343
1344         if (file->handle == INVALID_HANDLE_VALUE) {
1345                 free(file);
1346                 return NULL;
1347         }
1348
1349         /* The FTDI driver doesn't appear to work right unless you open it twice */
1350         if (device->vendor == 0x0403) {
1351                 CloseHandle(file->handle);
1352                 file->handle = open_serial(full_name);
1353                 if (file->handle == INVALID_HANDLE_VALUE) {
1354                         free(file);
1355                         return NULL;
1356                 }
1357         }
1358
1359         timeouts.ReadIntervalTimeout = MAXDWORD;
1360         timeouts.ReadTotalTimeoutMultiplier = MAXDWORD;
1361         timeouts.ReadTotalTimeoutConstant = 1 << 30;    /* almost forever */
1362         timeouts.WriteTotalTimeoutMultiplier = 0;
1363         timeouts.WriteTotalTimeoutConstant = 0;
1364         SetCommTimeouts(file->handle, &timeouts);
1365
1366         file->ov_read.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1367         file->ov_write.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1368
1369         return file;
1370 }
1371
1372 PUBLIC void
1373 altos_close(struct altos_file *file)
1374 {
1375         if (file->handle != INVALID_HANDLE_VALUE) {
1376                 CloseHandle(file->handle);
1377                 file->handle = INVALID_HANDLE_VALUE;
1378                 SetEvent(file->ov_read.hEvent);
1379                 SetEvent(file->ov_write.hEvent);
1380                 CloseHandle(file->ov_read.hEvent);
1381                 CloseHandle(file->ov_write.hEvent);
1382         }
1383 }
1384
1385 PUBLIC void
1386 altos_free(struct altos_file *file)
1387 {
1388         altos_close(file);
1389         free(file);
1390 }
1391
1392 PUBLIC int
1393 altos_putchar(struct altos_file *file, char c)
1394 {
1395         int     ret;
1396
1397         if (file->out_used == USB_BUF_SIZE) {
1398                 ret = altos_flush(file);
1399                 if (ret)
1400                         return ret;
1401         }
1402         file->out_data[file->out_used++] = c;
1403         if (file->out_used == USB_BUF_SIZE)
1404                 return altos_flush(file);
1405         return LIBALTOS_SUCCESS;
1406 }
1407
1408 PUBLIC int
1409 altos_getchar(struct altos_file *file, int timeout)
1410 {
1411         int     ret;
1412         while (file->in_read == file->in_used) {
1413                 if (file->handle == INVALID_HANDLE_VALUE) {
1414                         altos_set_last_windows_error();
1415                         return LIBALTOS_ERROR;
1416                 }
1417                 ret = altos_fill(file, timeout);
1418                 if (ret)
1419                         return ret;
1420         }
1421         return file->in_data[file->in_read++];
1422 }
1423
1424 struct altos_bt_list *
1425 altos_bt_list_start(int inquiry_time)
1426 {
1427         return NULL;
1428 }
1429
1430 int
1431 altos_bt_list_next(struct altos_bt_list *bt_list,
1432                    struct altos_bt_device *device)
1433 {
1434         return 0;
1435 }
1436
1437 void
1438 altos_bt_list_finish(struct altos_bt_list *bt_list)
1439 {
1440         free(bt_list);
1441 }
1442
1443 void
1444 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
1445 {
1446         strncpy(device->name, name, sizeof (device->name));
1447         device->name[sizeof(device->name)-1] = '\0';
1448         strncpy(device->addr, addr, sizeof (device->addr));
1449         device->addr[sizeof(device->addr)-1] = '\0';
1450 }
1451
1452 struct altos_file *
1453 altos_bt_open(struct altos_bt_device *device)
1454 {
1455         return NULL;
1456 }
1457
1458 #endif