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 ttyACMx style names
 482                          */
 483                         ntty = scandir(endpoint_full, &namelist,
 484                                        dir_filter_tty,
 485                                        alphasort);
 486                         if (ntty > 0) {
 487                                 free(endpoint_full);
 488                                 tty = cc_fullname("/dev", namelist[0]->d_name);
 489                                 free(namelist);
 490                                 return tty;
 491                         }
 492
 493                         /* Check for tty/ttyACMx style names
 494                          */
 495                         tty_dir = cc_fullname(endpoint_full, "tty");
 496                         free(endpoint_full);
 497                         ntty = scandir(tty_dir, &namelist,
 498                                        dir_filter_tty,
 499                                        alphasort);
 500                         free (tty_dir);
 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 int
 610 altos_list_next(struct altos_list *list, struct altos_device *device)
 611 {
 612         struct altos_usbdev *dev;
 613         if (list->current >= list->ndev) {
 614                 return 0;
 615         }
 616         dev = list->dev[list->current];
 617         strcpy(device->name, dev->product_name);
 618         device->vendor = dev->idVendor;
 619         device->product = dev->idProduct;
 620         strcpy(device->path, dev->tty);
 621         device->serial = dev->serial;
 622         list->current++;
 623         return 1;
 624 }
 625
 626 void
 627 altos_list_finish(struct altos_list *usbdevs)
 628 {
 629         int     i;
 630
 631         if (!usbdevs)
 632                 return;
 633         for (i = 0; i < usbdevs->ndev; i++)
 634                 usbdev_free(usbdevs->dev[i]);
 635         free(usbdevs);
 636 }
 637
 638 struct altos_bt_list {
 639         inquiry_info    *ii;
 640         int             sock;
 641         int             dev_id;
 642         int             rsp;
 643         int             num_rsp;
 644 };
 645
 646 #define INQUIRY_MAX_RSP 255
 647
 648 struct altos_bt_list *
 649 altos_bt_list_start(int inquiry_time)
 650 {
 651         struct altos_bt_list    *bt_list;
 652
 653         bt_list = calloc(1, sizeof (struct altos_bt_list));
 654         if (!bt_list)
 655                 goto no_bt_list;
 656
 657         bt_list->ii = calloc(INQUIRY_MAX_RSP, sizeof (inquiry_info));
 658         if (!bt_list->ii)
 659                 goto no_ii;
 660         bt_list->dev_id = hci_get_route(NULL);
 661         if (bt_list->dev_id < 0)
 662                 goto no_dev_id;
 663
 664         bt_list->sock = hci_open_dev(bt_list->dev_id);
 665         if (bt_list->sock < 0)
 666                 goto no_sock;
 667
 668         bt_list->num_rsp = hci_inquiry(bt_list->dev_id,
 669                                        inquiry_time,
 670                                        INQUIRY_MAX_RSP,
 671                                        NULL,
 672                                        &bt_list->ii,
 673                                        IREQ_CACHE_FLUSH);
 674         if (bt_list->num_rsp < 0)
 675                 goto no_rsp;
 676
 677         bt_list->rsp = 0;
 678         return bt_list;
 679
 680 no_rsp:
 681         close(bt_list->sock);
 682 no_sock:
 683 no_dev_id:
 684         free(bt_list->ii);
 685 no_ii:
 686         free(bt_list);
 687 no_bt_list:
 688         return NULL;
 689 }
 690
 691 int
 692 altos_bt_list_next(struct altos_bt_list *bt_list,
 693                    struct altos_bt_device *device)
 694 {
 695         inquiry_info    *ii;
 696
 697         if (bt_list->rsp >= bt_list->num_rsp)
 698                 return 0;
 699
 700         ii = &bt_list->ii[bt_list->rsp];
 701         ba2str(&ii->bdaddr, device->addr);
 702         memset(&device->name, '\0', sizeof (device->name));
 703         if (hci_read_remote_name(bt_list->sock, &ii->bdaddr,
 704                                  sizeof (device->name),
 705                                  device->name, 0) < 0) {
 706                 strcpy(device->name, "[unknown]");
 707         }
 708         bt_list->rsp++;
 709         return 1;
 710 }
 711
 712 void
 713 altos_bt_list_finish(struct altos_bt_list *bt_list)
 714 {
 715         close(bt_list->sock);
 716         free(bt_list->ii);
 717         free(bt_list);
 718 }
 719
 720 void
 721 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
 722 {
 723         strncpy(device->name, name, sizeof (device->name));
 724         device->name[sizeof(device->name)-1] = '\0';
 725         strncpy(device->addr, addr, sizeof (device->addr));
 726         device->addr[sizeof(device->addr)-1] = '\0';
 727 }
 728
 729 struct altos_file *
 730 altos_bt_open(struct altos_bt_device *device)
 731 {
 732         struct sockaddr_rc addr = { 0 };
 733         int     s, status;
 734         struct altos_file *file;
 735
 736         file = calloc(1, sizeof (struct altos_file));
 737         if (!file)
 738                 goto no_file;
 739         file->fd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
 740         if (file->fd < 0) {
 741                 altos_set_last_posix_error();
 742                 goto no_sock;
 743         }
 744
 745         addr.rc_family = AF_BLUETOOTH;
 746         addr.rc_channel = 1;
 747         str2ba(device->addr, &addr.rc_bdaddr);
 748
 749         status = connect(file->fd,
 750                          (struct sockaddr *)&addr,
 751                          sizeof(addr));
 752         if (status < 0) {
 753                 altos_set_last_posix_error();
 754                 goto no_link;
 755         }
 756         sleep(1);
 757
 758 #ifdef USE_POLL
 759         pipe(file->pipe);
 760 #else
 761         file->out_fd = dup(file->fd);
 762 #endif
 763         return file;
 764 no_link:
 765         close(s);
 766 no_sock:
 767         free(file);
 768 no_file:
 769         return NULL;
 770 }
 771
 772 #endif
 773
 774 #ifdef DARWIN
 775
 776 #include <IOKitLib.h>
 777 #include <IOKit/usb/USBspec.h>
 778 #include <sys/param.h>
 779 #include <paths.h>
 780 #include <CFNumber.h>
 781 #include <IOBSD.h>
 782 #include <string.h>
 783 #include <stdio.h>
 784 #include <stdlib.h>
 785
 786 struct altos_list {
 787         io_iterator_t iterator;
 788 };
 789
 790 static int
 791 get_string(io_object_t object, CFStringRef entry, char *result, int result_len)
 792 {
 793         CFTypeRef entry_as_string;
 794         Boolean got_string;
 795
 796         entry_as_string = IORegistryEntrySearchCFProperty (object,
 797                                                            kIOServicePlane,
 798                                                            entry,
 799                                                            kCFAllocatorDefault,
 800                                                            kIORegistryIterateRecursively);
 801         if (entry_as_string) {
 802                 got_string = CFStringGetCString(entry_as_string,
 803                                                 result, result_len,
 804                                                 kCFStringEncodingASCII);
 805
 806                 CFRelease(entry_as_string);
 807                 if (got_string)
 808                         return 1;
 809         }
 810         return 0;
 811 }
 812
 813 static int
 814 get_number(io_object_t object, CFStringRef entry, int *result)
 815 {
 816         CFTypeRef entry_as_number;
 817         Boolean got_number;
 818
 819         entry_as_number = IORegistryEntrySearchCFProperty (object,
 820                                                            kIOServicePlane,
 821                                                            entry,
 822                                                            kCFAllocatorDefault,
 823                                                            kIORegistryIterateRecursively);
 824         if (entry_as_number) {
 825                 got_number = CFNumberGetValue(entry_as_number,
 826                                               kCFNumberIntType,
 827                                               result);
 828                 if (got_number)
 829                         return 1;
 830         }
 831         return 0;
 832 }
 833
 834 PUBLIC struct altos_list *
 835 altos_list_start(void)
 836 {
 837         struct altos_list *list = calloc (sizeof (struct altos_list), 1);
 838         CFMutableDictionaryRef matching_dictionary = IOServiceMatching("IOUSBDevice");
 839         io_iterator_t tdIterator;
 840         io_object_t tdObject;
 841         kern_return_t ret;
 842         int i;
 843
 844         ret = IOServiceGetMatchingServices(kIOMasterPortDefault, matching_dictionary, &list->iterator);
 845         if (ret != kIOReturnSuccess)
 846                 return NULL;
 847         return list;
 848 }
 849
 850 PUBLIC int
 851 altos_list_next(struct altos_list *list, struct altos_device *device)
 852 {
 853         io_object_t object;
 854         char serial_string[128];
 855
 856         for (;;) {
 857                 object = IOIteratorNext(list->iterator);
 858                 if (!object)
 859                         return 0;
 860
 861                 if (!get_number (object, CFSTR(kUSBVendorID), &device->vendor) ||
 862                     !get_number (object, CFSTR(kUSBProductID), &device->product))
 863                         continue;
 864                 if (device->vendor != 0xfffe)
 865                         continue;
 866                 if (device->product < 0x000a || 0x0013 < device->product)
 867                         continue;
 868                 if (get_string (object, CFSTR("IOCalloutDevice"), device->path, sizeof (device->path)) &&
 869                     get_string (object, CFSTR("USB Product Name"), device->name, sizeof (device->name)) &&
 870                     get_string (object, CFSTR("USB Serial Number"), serial_string, sizeof (serial_string))) {
 871                         device->serial = atoi(serial_string);
 872                         return 1;
 873                 }
 874         }
 875 }
 876
 877 PUBLIC void
 878 altos_list_finish(struct altos_list *list)
 879 {
 880         IOObjectRelease (list->iterator);
 881         free(list);
 882 }
 883
 884 struct altos_bt_list {
 885         int             sock;
 886         int             dev_id;
 887         int             rsp;
 888         int             num_rsp;
 889 };
 890
 891 #define INQUIRY_MAX_RSP 255
 892
 893 struct altos_bt_list *
 894 altos_bt_list_start(int inquiry_time)
 895 {
 896         return NULL;
 897 }
 898
 899 int
 900 altos_bt_list_next(struct altos_bt_list *bt_list,
 901                    struct altos_bt_device *device)
 902 {
 903         return 0;
 904 }
 905
 906 void
 907 altos_bt_list_finish(struct altos_bt_list *bt_list)
 908 {
 909 }
 910
 911 void
 912 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
 913 {
 914         strncpy(device->name, name, sizeof (device->name));
 915         device->name[sizeof(device->name)-1] = '\0';
 916         strncpy(device->addr, addr, sizeof (device->addr));
 917         device->addr[sizeof(device->addr)-1] = '\0';
 918 }
 919
 920 struct altos_file *
 921 altos_bt_open(struct altos_bt_device *device)
 922 {
 923         return NULL;
 924 }
 925
 926 #endif
 927
 928
 929 #ifdef WINDOWS
 930
 931 #include <stdlib.h>
 932 #include <windows.h>
 933 #include <setupapi.h>
 934
 935 struct altos_list {
 936         HDEVINFO        dev_info;
 937         int             index;
 938 };
 939
 940 #define USB_BUF_SIZE    64
 941
 942 struct altos_file {
 943         HANDLE                          handle;
 944         unsigned char                   out_data[USB_BUF_SIZE];
 945         int                             out_used;
 946         unsigned char                   in_data[USB_BUF_SIZE];
 947         int                             in_used;
 948         int                             in_read;
 949         OVERLAPPED                      ov_read;
 950         BOOL                            pend_read;
 951         OVERLAPPED                      ov_write;
 952 };
 953
 954 static void
 955 _altos_set_last_windows_error(char *file, int line)
 956 {
 957         DWORD   error = GetLastError();
 958         TCHAR   message[1024];
 959         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
 960                       0,
 961                       error,
 962                       0,
 963                       message,
 964                       sizeof (message) / sizeof (TCHAR),
 965                       NULL);
 966         if (error != ERROR_SUCCESS)
 967                 printf ("%s:%d %s\n", file, line, message);
 968         altos_set_last_error(error, message);
 969 }
 970
 971 #define altos_set_last_windows_error() _altos_set_last_windows_error(__FILE__, __LINE__)
 972
 973 PUBLIC struct altos_list *
 974 altos_list_start(void)
 975 {
 976         struct altos_list       *list = calloc(1, sizeof (struct altos_list));
 977
 978         if (!list)
 979                 return NULL;
 980         list->dev_info = SetupDiGetClassDevs(NULL, "USB", NULL,
 981                                              DIGCF_ALLCLASSES|DIGCF_PRESENT);
 982         if (list->dev_info == INVALID_HANDLE_VALUE) {
 983                 altos_set_last_windows_error();
 984                 free(list);
 985                 return NULL;
 986         }
 987         list->index = 0;
 988         return list;
 989 }
 990
 991 PUBLIC int
 992 altos_list_next(struct altos_list *list, struct altos_device *device)
 993 {
 994         SP_DEVINFO_DATA dev_info_data;
 995         BYTE            port[128];
 996         DWORD           port_len;
 997         char            friendlyname[256];
 998         BYTE            symbolic[256];
 999         DWORD           symbolic_len;
1000         HKEY            dev_key;
1001         unsigned int    vid, pid;
1002         int             serial;
1003         HRESULT         result;
1004         DWORD           friendlyname_type;
1005         DWORD           friendlyname_len;
1006
1007         dev_info_data.cbSize = sizeof (SP_DEVINFO_DATA);
1008         while(SetupDiEnumDeviceInfo(list->dev_info, list->index,
1009                                     &dev_info_data))
1010         {
1011                 list->index++;
1012
1013                 dev_key = SetupDiOpenDevRegKey(list->dev_info, &dev_info_data,
1014                                                DICS_FLAG_GLOBAL, 0, DIREG_DEV,
1015                                                KEY_READ);
1016                 if (dev_key == INVALID_HANDLE_VALUE) {
1017                         altos_set_last_windows_error();
1018                         printf("cannot open device registry key\n");
1019                         continue;
1020                 }
1021
1022                 /* Fetch symbolic name for this device and parse out
1023                  * the vid/pid/serial info */
1024                 symbolic_len = sizeof(symbolic);
1025                 result = RegQueryValueEx(dev_key, "SymbolicName", NULL, NULL,
1026                                          symbolic, &symbolic_len);
1027                 if (result != 0) {
1028                         altos_set_last_windows_error();
1029                         printf("cannot find SymbolicName value\n");
1030                         RegCloseKey(dev_key);
1031                         continue;
1032                 }
1033                 vid = pid = serial = 0;
1034                 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_") - 1,
1035                        "%04X", &vid);
1036                 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_") - 1,
1037                        "%04X", &pid);
1038                 sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_XXXX#") - 1,
1039                        "%d", &serial);
1040
1041                 /* Fetch the com port name */
1042                 port_len = sizeof (port);
1043                 result = RegQueryValueEx(dev_key, "PortName", NULL, NULL,
1044                                          port, &port_len);
1045                 RegCloseKey(dev_key);
1046                 if (result != 0) {
1047                         altos_set_last_windows_error();
1048                         printf("failed to get PortName\n");
1049                         continue;
1050                 }
1051
1052                 /* Fetch the device description which is the device name,
1053                  * with firmware that has unique USB ids */
1054                 friendlyname_len = sizeof (friendlyname);
1055                 if(!SetupDiGetDeviceRegistryProperty(list->dev_info,
1056                                                      &dev_info_data,
1057                                                      SPDRP_FRIENDLYNAME,
1058                                                      &friendlyname_type,
1059                                                      (BYTE *)friendlyname,
1060                                                      sizeof(friendlyname),
1061                                                      &friendlyname_len))
1062                 {
1063                         altos_set_last_windows_error();
1064                         printf("Failed to get friendlyname\n");
1065                         continue;
1066                 }
1067                 device->vendor = vid;
1068                 device->product = pid;
1069                 device->serial = serial;
1070                 strcpy(device->name, friendlyname);
1071
1072                 strcpy(device->path, (char *) port);
1073                 return 1;
1074         }
1075         result = GetLastError();
1076         if (result != ERROR_NO_MORE_ITEMS) {
1077                 altos_set_last_windows_error();
1078                 printf ("SetupDiEnumDeviceInfo failed error %d\n", (int) result);
1079         }
1080         return 0;
1081 }
1082
1083 PUBLIC void
1084 altos_list_finish(struct altos_list *list)
1085 {
1086         SetupDiDestroyDeviceInfoList(list->dev_info);
1087         free(list);
1088 }
1089
1090 static int
1091 altos_queue_read(struct altos_file *file)
1092 {
1093         DWORD   got;
1094         if (file->pend_read)
1095                 return LIBALTOS_SUCCESS;
1096
1097         if (!ReadFile(file->handle, file->in_data, USB_BUF_SIZE, &got, &file->ov_read)) {
1098                 if (GetLastError() != ERROR_IO_PENDING) {
1099                         altos_set_last_windows_error();
1100                         return LIBALTOS_ERROR;
1101                 }
1102                 file->pend_read = TRUE;
1103         } else {
1104                 file->pend_read = FALSE;
1105                 file->in_read = 0;
1106                 file->in_used = got;
1107         }
1108         return LIBALTOS_SUCCESS;
1109 }
1110
1111 static int
1112 altos_wait_read(struct altos_file *file, int timeout)
1113 {
1114         DWORD   ret;
1115         DWORD   got;
1116
1117         if (!file->pend_read)
1118                 return LIBALTOS_SUCCESS;
1119
1120         if (!timeout)
1121                 timeout = INFINITE;
1122
1123         ret = WaitForSingleObject(file->ov_read.hEvent, timeout);
1124         switch (ret) {
1125         case WAIT_OBJECT_0:
1126                 if (!GetOverlappedResult(file->handle, &file->ov_read, &got, FALSE)) {
1127                         altos_set_last_windows_error();
1128                         return LIBALTOS_ERROR;
1129                 }
1130                 file->pend_read = FALSE;
1131                 file->in_read = 0;
1132                 file->in_used = got;
1133                 break;
1134         case WAIT_TIMEOUT:
1135                 return LIBALTOS_TIMEOUT;
1136                 break;
1137         default:
1138                 return LIBALTOS_ERROR;
1139         }
1140         return LIBALTOS_SUCCESS;
1141 }
1142
1143 static int
1144 altos_fill(struct altos_file *file, int timeout)
1145 {
1146         int     ret;
1147
1148         if (file->in_read < file->in_used)
1149                 return LIBALTOS_SUCCESS;
1150
1151         file->in_read = file->in_used = 0;
1152
1153         ret = altos_queue_read(file);
1154         if (ret)
1155                 return ret;
1156         ret = altos_wait_read(file, timeout);
1157         if (ret)
1158                 return ret;
1159
1160         return LIBALTOS_SUCCESS;
1161 }
1162
1163 PUBLIC int
1164 altos_flush(struct altos_file *file)
1165 {
1166         DWORD           put;
1167         unsigned char   *data = file->out_data;
1168         int             used = file->out_used;
1169         DWORD           ret;
1170
1171         while (used) {
1172                 if (!WriteFile(file->handle, data, used, &put, &file->ov_write)) {
1173                         if (GetLastError() != ERROR_IO_PENDING) {
1174                                 altos_set_last_windows_error();
1175                                 printf ("\tflush write error\n");
1176                                 return LIBALTOS_ERROR;
1177                         }
1178                         ret = WaitForSingleObject(file->ov_write.hEvent, INFINITE);
1179                         switch (ret) {
1180                         case WAIT_OBJECT_0:
1181                                 if (!GetOverlappedResult(file->handle, &file->ov_write, &put, FALSE)) {
1182                                         altos_set_last_windows_error();
1183                                         printf ("\tflush result error\n");
1184                                         return LIBALTOS_ERROR;
1185                                 }
1186                                 break;
1187                         default:
1188                                 altos_set_last_windows_error();
1189                                 printf ("\tflush wait error\n");
1190                                 return LIBALTOS_ERROR;
1191                         }
1192                 }
1193                 data += put;
1194                 used -= put;
1195         }
1196         file->out_used = 0;
1197         return LIBALTOS_SUCCESS;
1198 }
1199
1200 PUBLIC struct altos_file *
1201 altos_open(struct altos_device *device)
1202 {
1203         struct altos_file       *file = calloc (1, sizeof (struct altos_file));
1204         char    full_name[64];
1205         COMMTIMEOUTS timeouts;
1206
1207         if (!file)
1208                 return NULL;
1209
1210         strcpy(full_name, "\\\\.\\");
1211         strcat(full_name, device->path);
1212         file->handle = CreateFile(full_name, GENERIC_READ|GENERIC_WRITE,
1213                                   0, NULL, OPEN_EXISTING,
1214                                   FILE_FLAG_OVERLAPPED, NULL);
1215         if (file->handle == INVALID_HANDLE_VALUE) {
1216                 altos_set_last_windows_error();
1217                 printf ("cannot open %s\n", full_name);
1218                 free(file);
1219                 return NULL;
1220         }
1221         file->ov_read.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1222         file->ov_write.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1223
1224         timeouts.ReadIntervalTimeout = MAXDWORD;
1225         timeouts.ReadTotalTimeoutMultiplier = MAXDWORD;
1226         timeouts.ReadTotalTimeoutConstant = 1 << 30;    /* almost forever */
1227         timeouts.WriteTotalTimeoutMultiplier = 0;
1228         timeouts.WriteTotalTimeoutConstant = 0;
1229         SetCommTimeouts(file->handle, &timeouts);
1230
1231         return file;
1232 }
1233
1234 PUBLIC void
1235 altos_close(struct altos_file *file)
1236 {
1237         if (file->handle != INVALID_HANDLE_VALUE) {
1238                 CloseHandle(file->handle);
1239                 file->handle = INVALID_HANDLE_VALUE;
1240                 SetEvent(file->ov_read.hEvent);
1241                 SetEvent(file->ov_write.hEvent);
1242                 CloseHandle(file->ov_read.hEvent);
1243                 CloseHandle(file->ov_write.hEvent);
1244         }
1245 }
1246
1247 PUBLIC void
1248 altos_free(struct altos_file *file)
1249 {
1250         altos_close(file);
1251         free(file);
1252 }
1253
1254 PUBLIC int
1255 altos_putchar(struct altos_file *file, char c)
1256 {
1257         int     ret;
1258
1259         if (file->out_used == USB_BUF_SIZE) {
1260                 ret = altos_flush(file);
1261                 if (ret)
1262                         return ret;
1263         }
1264         file->out_data[file->out_used++] = c;
1265         if (file->out_used == USB_BUF_SIZE)
1266                 return altos_flush(file);
1267         return LIBALTOS_SUCCESS;
1268 }
1269
1270 PUBLIC int
1271 altos_getchar(struct altos_file *file, int timeout)
1272 {
1273         int     ret;
1274         while (file->in_read == file->in_used) {
1275                 if (file->handle == INVALID_HANDLE_VALUE)
1276                         return LIBALTOS_ERROR;
1277                 ret = altos_fill(file, timeout);
1278                 if (ret)
1279                         return ret;
1280         }
1281         return file->in_data[file->in_read++];
1282 }
1283
1284 struct altos_bt_list *
1285 altos_bt_list_start(int inquiry_time)
1286 {
1287         return NULL;
1288 }
1289
1290 int
1291 altos_bt_list_next(struct altos_bt_list *bt_list,
1292                    struct altos_bt_device *device)
1293 {
1294         return 0;
1295 }
1296
1297 void
1298 altos_bt_list_finish(struct altos_bt_list *bt_list)
1299 {
1300         free(bt_list);
1301 }
1302
1303 void
1304 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
1305 {
1306         strncpy(device->name, name, sizeof (device->name));
1307         device->name[sizeof(device->name)-1] = '\0';
1308         strncpy(device->addr, addr, sizeof (device->addr));
1309         device->addr[sizeof(device->addr)-1] = '\0';
1310 }
1311
1312 struct altos_file *
1313 altos_bt_open(struct altos_bt_device *device)
1314 {
1315         return NULL;
1316 }
1317
1318 #endif