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     s, status;
 740         struct altos_file *file;
 741
 742         file = calloc(1, sizeof (struct altos_file));
 743         if (!file)
 744                 goto no_file;
 745         file->fd = socket(AF_BLUETOOTH, SOCK_STREAM, BTPROTO_RFCOMM);
 746         if (file->fd < 0) {
 747                 altos_set_last_posix_error();
 748                 goto no_sock;
 749         }
 750
 751         addr.rc_family = AF_BLUETOOTH;
 752         addr.rc_channel = 1;
 753         str2ba(device->addr, &addr.rc_bdaddr);
 754
 755         status = connect(file->fd,
 756                          (struct sockaddr *)&addr,
 757                          sizeof(addr));
 758         if (status < 0) {
 759                 altos_set_last_posix_error();
 760                 goto no_link;
 761         }
 762         sleep(1);
 763
 764 #ifdef USE_POLL
 765         pipe(file->pipe);
 766 #else
 767         file->out_fd = dup(file->fd);
 768 #endif
 769         return file;
 770 no_link:
 771         close(s);
 772 no_sock:
 773         free(file);
 774 no_file:
 775         return NULL;
 776 }
 777
 778 #endif
 779
 780 #ifdef DARWIN
 781
 782 #include <IOKitLib.h>
 783 #include <IOKit/usb/USBspec.h>
 784 #include <sys/param.h>
 785 #include <paths.h>
 786 #include <CFNumber.h>
 787 #include <IOBSD.h>
 788 #include <string.h>
 789 #include <stdio.h>
 790 #include <stdlib.h>
 791
 792 struct altos_list {
 793         io_iterator_t iterator;
 794         int ftdi;
 795 };
 796
 797 static int
 798 get_string(io_object_t object, CFStringRef entry, char *result, int result_len)
 799 {
 800         CFTypeRef entry_as_string;
 801         Boolean got_string;
 802
 803         entry_as_string = IORegistryEntrySearchCFProperty (object,
 804                                                            kIOServicePlane,
 805                                                            entry,
 806                                                            kCFAllocatorDefault,
 807                                                            kIORegistryIterateRecursively);
 808         if (entry_as_string) {
 809                 got_string = CFStringGetCString(entry_as_string,
 810                                                 result, result_len,
 811                                                 kCFStringEncodingASCII);
 812
 813                 CFRelease(entry_as_string);
 814                 if (got_string)
 815                         return 1;
 816         }
 817         return 0;
 818 }
 819
 820 static int
 821 get_number(io_object_t object, CFStringRef entry, int *result)
 822 {
 823         CFTypeRef entry_as_number;
 824         Boolean got_number;
 825
 826         entry_as_number = IORegistryEntrySearchCFProperty (object,
 827                                                            kIOServicePlane,
 828                                                            entry,
 829                                                            kCFAllocatorDefault,
 830                                                            kIORegistryIterateRecursively);
 831         if (entry_as_number) {
 832                 got_number = CFNumberGetValue(entry_as_number,
 833                                               kCFNumberIntType,
 834                                               result);
 835                 if (got_number)
 836                         return 1;
 837         }
 838         return 0;
 839 }
 840
 841 PUBLIC struct altos_list *
 842 altos_list_start(void)
 843 {
 844         struct altos_list *list = calloc (sizeof (struct altos_list), 1);
 845         CFMutableDictionaryRef matching_dictionary = IOServiceMatching("IOUSBDevice");
 846         io_iterator_t tdIterator;
 847         io_object_t tdObject;
 848         kern_return_t ret;
 849         int i;
 850
 851         ret = IOServiceGetMatchingServices(kIOMasterPortDefault, matching_dictionary, &list->iterator);
 852         if (ret != kIOReturnSuccess) {
 853                 free(list);
 854                 return NULL;
 855         }
 856         list->ftdi = 0;
 857         return list;
 858 }
 859
 860 PUBLIC struct altos_list *
 861 altos_ftdi_list_start(void)
 862 {
 863         struct altos_list *list = altos_list_start();
 864
 865         if (list)
 866                 list->ftdi = 1;
 867         return list;
 868 }
 869
 870 PUBLIC int
 871 altos_list_next(struct altos_list *list, struct altos_device *device)
 872 {
 873         io_object_t object;
 874         char serial_string[128];
 875
 876         for (;;) {
 877                 object = IOIteratorNext(list->iterator);
 878                 if (!object)
 879                         return 0;
 880
 881                 if (!get_number (object, CFSTR(kUSBVendorID), &device->vendor) ||
 882                     !get_number (object, CFSTR(kUSBProductID), &device->product))
 883                         continue;
 884                 if (list->ftdi) {
 885                         if (device->vendor != 0x0403)
 886                                 continue;
 887                 } else {
 888                         if (device->vendor != 0xfffe)
 889                                 continue;
 890                         if (device->product < 0x000a || 0x0013 < device->product)
 891                                 continue;
 892                 }
 893                 if (get_string (object, CFSTR("IOCalloutDevice"), device->path, sizeof (device->path)) &&
 894                     get_string (object, CFSTR("USB Product Name"), device->name, sizeof (device->name)) &&
 895                     get_string (object, CFSTR("USB Serial Number"), serial_string, sizeof (serial_string))) {
 896                         device->serial = atoi(serial_string);
 897                         return 1;
 898                 }
 899         }
 900 }
 901
 902 PUBLIC void
 903 altos_list_finish(struct altos_list *list)
 904 {
 905         IOObjectRelease (list->iterator);
 906         free(list);
 907 }
 908
 909 struct altos_bt_list {
 910         int             sock;
 911         int             dev_id;
 912         int             rsp;
 913         int             num_rsp;
 914 };
 915
 916 #define INQUIRY_MAX_RSP 255
 917
 918 struct altos_bt_list *
 919 altos_bt_list_start(int inquiry_time)
 920 {
 921         return NULL;
 922 }
 923
 924 int
 925 altos_bt_list_next(struct altos_bt_list *bt_list,
 926                    struct altos_bt_device *device)
 927 {
 928         return 0;
 929 }
 930
 931 void
 932 altos_bt_list_finish(struct altos_bt_list *bt_list)
 933 {
 934 }
 935
 936 void
 937 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
 938 {
 939         strncpy(device->name, name, sizeof (device->name));
 940         device->name[sizeof(device->name)-1] = '\0';
 941         strncpy(device->addr, addr, sizeof (device->addr));
 942         device->addr[sizeof(device->addr)-1] = '\0';
 943 }
 944
 945 struct altos_file *
 946 altos_bt_open(struct altos_bt_device *device)
 947 {
 948         return NULL;
 949 }
 950
 951 #endif
 952
 953
 954 #ifdef WINDOWS
 955
 956 #include <stdlib.h>
 957 #include <windows.h>
 958 #include <setupapi.h>
 959
 960 struct altos_list {
 961         HDEVINFO        dev_info;
 962         int             index;
 963         int             ftdi;
 964 };
 965
 966 #define USB_BUF_SIZE    64
 967
 968 struct altos_file {
 969         HANDLE                          handle;
 970         unsigned char                   out_data[USB_BUF_SIZE];
 971         int                             out_used;
 972         unsigned char                   in_data[USB_BUF_SIZE];
 973         int                             in_used;
 974         int                             in_read;
 975         OVERLAPPED                      ov_read;
 976         BOOL                            pend_read;
 977         OVERLAPPED                      ov_write;
 978 };
 979
 980 static void
 981 _altos_set_last_windows_error(char *file, int line)
 982 {
 983         DWORD   error = GetLastError();
 984         TCHAR   message[1024];
 985         FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
 986                       0,
 987                       error,
 988                       0,
 989                       message,
 990                       sizeof (message) / sizeof (TCHAR),
 991                       NULL);
 992         if (error != ERROR_SUCCESS)
 993                 printf ("%s:%d %s\n", file, line, message);
 994         altos_set_last_error(error, message);
 995 }
 996
 997 #define altos_set_last_windows_error() _altos_set_last_windows_error(__FILE__, __LINE__)
 998
 999 PUBLIC struct altos_list *
1000 altos_list_start(void)
1001 {
1002         struct altos_list       *list = calloc(1, sizeof (struct altos_list));
1003
1004         if (!list)
1005                 return NULL;
1006         list->dev_info = SetupDiGetClassDevs(NULL, "USB", NULL,
1007                                              DIGCF_ALLCLASSES|DIGCF_PRESENT);
1008         if (list->dev_info == INVALID_HANDLE_VALUE) {
1009                 altos_set_last_windows_error();
1010                 free(list);
1011                 return NULL;
1012         }
1013         list->index = 0;
1014         list->ftdi = 0;
1015         return list;
1016 }
1017
1018 PUBLIC struct altos_list *
1019 altos_ftdi_list_start(void)
1020 {
1021         struct altos_list       *list = calloc(1, sizeof (struct altos_list));
1022
1023         if (!list)
1024                 return NULL;
1025         list->dev_info = SetupDiGetClassDevs(NULL, "FTDIBUS", NULL,
1026                                              DIGCF_ALLCLASSES|DIGCF_PRESENT);
1027         if (list->dev_info == INVALID_HANDLE_VALUE) {
1028                 altos_set_last_windows_error();
1029                 free(list);
1030                 return NULL;
1031         }
1032         list->index = 0;
1033         list->ftdi = 1;
1034         return list;
1035 }
1036
1037 PUBLIC int
1038 altos_list_next(struct altos_list *list, struct altos_device *device)
1039 {
1040         SP_DEVINFO_DATA dev_info_data;
1041         BYTE            port[128];
1042         DWORD           port_len;
1043         char            friendlyname[256];
1044         BYTE            symbolic[256];
1045         DWORD           symbolic_len;
1046         HKEY            dev_key;
1047         unsigned int    vid, pid;
1048         int             serial;
1049         HRESULT         result;
1050         DWORD           friendlyname_type;
1051         DWORD           friendlyname_len;
1052
1053         dev_info_data.cbSize = sizeof (SP_DEVINFO_DATA);
1054         while(SetupDiEnumDeviceInfo(list->dev_info, list->index,
1055                                     &dev_info_data))
1056         {
1057                 list->index++;
1058
1059                 dev_key = SetupDiOpenDevRegKey(list->dev_info, &dev_info_data,
1060                                                DICS_FLAG_GLOBAL, 0, DIREG_DEV,
1061                                                KEY_READ);
1062                 if (dev_key == INVALID_HANDLE_VALUE) {
1063                         altos_set_last_windows_error();
1064                         printf("cannot open device registry key\n");
1065                         continue;
1066                 }
1067
1068                 if (list->ftdi) {
1069                         vid = 0x0403;
1070                         pid = 0x6015;
1071                         serial = 0;
1072                 } else {
1073                         /* Fetch symbolic name for this device and parse out
1074                          * the vid/pid/serial info */
1075                         symbolic_len = sizeof(symbolic);
1076                         result = RegQueryValueEx(dev_key, "SymbolicName", NULL, NULL,
1077                                                  symbolic, &symbolic_len);
1078                         if (result != 0) {
1079                                 altos_set_last_windows_error();
1080                                 printf("cannot find SymbolicName value\n");
1081                                 RegCloseKey(dev_key);
1082                                 continue;
1083                         }
1084                         vid = pid = serial = 0;
1085                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_") - 1,
1086                                "%04X", &vid);
1087                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_") - 1,
1088                                "%04X", &pid);
1089                         sscanf((char *) symbolic + sizeof("\\??\\USB#VID_XXXX&PID_XXXX#") - 1,
1090                                "%d", &serial);
1091                 }
1092
1093                 /* Fetch the com port name */
1094                 port_len = sizeof (port);
1095                 result = RegQueryValueEx(dev_key, "PortName", NULL, NULL,
1096                                          port, &port_len);
1097                 RegCloseKey(dev_key);
1098                 if (result != 0) {
1099                         altos_set_last_windows_error();
1100                         printf("failed to get PortName\n");
1101                         continue;
1102                 }
1103
1104                 /* Fetch the device description which is the device name,
1105                  * with firmware that has unique USB ids */
1106                 friendlyname_len = sizeof (friendlyname);
1107                 if(!SetupDiGetDeviceRegistryProperty(list->dev_info,
1108                                                      &dev_info_data,
1109                                                      SPDRP_FRIENDLYNAME,
1110                                                      &friendlyname_type,
1111                                                      (BYTE *)friendlyname,
1112                                                      sizeof(friendlyname),
1113                                                      &friendlyname_len))
1114                 {
1115                         altos_set_last_windows_error();
1116                         printf("Failed to get friendlyname\n");
1117                         continue;
1118                 }
1119                 device->vendor = vid;
1120                 device->product = pid;
1121                 device->serial = serial;
1122                 strcpy(device->name, friendlyname);
1123
1124                 strcpy(device->path, (char *) port);
1125                 return 1;
1126         }
1127         result = GetLastError();
1128         if (result != ERROR_NO_MORE_ITEMS) {
1129                 altos_set_last_windows_error();
1130                 printf ("SetupDiEnumDeviceInfo failed error %d\n", (int) result);
1131         }
1132         return 0;
1133 }
1134
1135 PUBLIC void
1136 altos_list_finish(struct altos_list *list)
1137 {
1138         SetupDiDestroyDeviceInfoList(list->dev_info);
1139         free(list);
1140 }
1141
1142 static int
1143 altos_queue_read(struct altos_file *file)
1144 {
1145         DWORD   got;
1146         if (file->pend_read)
1147                 return LIBALTOS_SUCCESS;
1148
1149         if (!ReadFile(file->handle, file->in_data, USB_BUF_SIZE, &got, &file->ov_read)) {
1150                 if (GetLastError() != ERROR_IO_PENDING) {
1151                         altos_set_last_windows_error();
1152                         return LIBALTOS_ERROR;
1153                 }
1154                 file->pend_read = TRUE;
1155         } else {
1156                 file->pend_read = FALSE;
1157                 file->in_read = 0;
1158                 file->in_used = got;
1159         }
1160         return LIBALTOS_SUCCESS;
1161 }
1162
1163 static int
1164 altos_wait_read(struct altos_file *file, int timeout)
1165 {
1166         DWORD   ret;
1167         DWORD   got;
1168
1169         if (!file->pend_read)
1170                 return LIBALTOS_SUCCESS;
1171
1172         if (!timeout)
1173                 timeout = INFINITE;
1174
1175         ret = WaitForSingleObject(file->ov_read.hEvent, timeout);
1176         switch (ret) {
1177         case WAIT_OBJECT_0:
1178                 if (!GetOverlappedResult(file->handle, &file->ov_read, &got, FALSE)) {
1179                         altos_set_last_windows_error();
1180                         return LIBALTOS_ERROR;
1181                 }
1182                 file->pend_read = FALSE;
1183                 file->in_read = 0;
1184                 file->in_used = got;
1185                 break;
1186         case WAIT_TIMEOUT:
1187                 return LIBALTOS_TIMEOUT;
1188                 break;
1189         default:
1190                 return LIBALTOS_ERROR;
1191         }
1192         return LIBALTOS_SUCCESS;
1193 }
1194
1195 static int
1196 altos_fill(struct altos_file *file, int timeout)
1197 {
1198         int     ret;
1199
1200         if (file->in_read < file->in_used)
1201                 return LIBALTOS_SUCCESS;
1202
1203         file->in_read = file->in_used = 0;
1204
1205         ret = altos_queue_read(file);
1206         if (ret)
1207                 return ret;
1208         ret = altos_wait_read(file, timeout);
1209         if (ret)
1210                 return ret;
1211
1212         return LIBALTOS_SUCCESS;
1213 }
1214
1215 PUBLIC int
1216 altos_flush(struct altos_file *file)
1217 {
1218         DWORD           put;
1219         unsigned char   *data = file->out_data;
1220         int             used = file->out_used;
1221         DWORD           ret;
1222
1223         while (used) {
1224                 if (!WriteFile(file->handle, data, used, &put, &file->ov_write)) {
1225                         if (GetLastError() != ERROR_IO_PENDING) {
1226                                 altos_set_last_windows_error();
1227                                 printf ("\tflush write error\n");
1228                                 return LIBALTOS_ERROR;
1229                         }
1230                         ret = WaitForSingleObject(file->ov_write.hEvent, INFINITE);
1231                         switch (ret) {
1232                         case WAIT_OBJECT_0:
1233                                 if (!GetOverlappedResult(file->handle, &file->ov_write, &put, FALSE)) {
1234                                         altos_set_last_windows_error();
1235                                         printf ("\tflush result error\n");
1236                                         return LIBALTOS_ERROR;
1237                                 }
1238                                 break;
1239                         default:
1240                                 altos_set_last_windows_error();
1241                                 printf ("\tflush wait error\n");
1242                                 return LIBALTOS_ERROR;
1243                         }
1244                 }
1245                 data += put;
1246                 used -= put;
1247         }
1248         file->out_used = 0;
1249         return LIBALTOS_SUCCESS;
1250 }
1251
1252 PUBLIC struct altos_file *
1253 altos_open(struct altos_device *device)
1254 {
1255         struct altos_file       *file = calloc (1, sizeof (struct altos_file));
1256         char    full_name[64];
1257         COMMTIMEOUTS timeouts;
1258         DCB          dcb;
1259
1260         if (!file)
1261                 return NULL;
1262
1263         strcpy(full_name, "\\\\.\\");
1264         strcat(full_name, device->path);
1265         file->handle = CreateFile(full_name, GENERIC_READ|GENERIC_WRITE,
1266                                   0, NULL, OPEN_EXISTING,
1267                                   FILE_FLAG_OVERLAPPED, NULL);
1268         if (file->handle == INVALID_HANDLE_VALUE) {
1269                 altos_set_last_windows_error();
1270                 printf ("cannot open %s\n", full_name);
1271                 free(file);
1272                 return NULL;
1273         }
1274         file->ov_read.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1275         file->ov_write.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1276
1277         timeouts.ReadIntervalTimeout = MAXDWORD;
1278         timeouts.ReadTotalTimeoutMultiplier = MAXDWORD;
1279         timeouts.ReadTotalTimeoutConstant = 1 << 30;    /* almost forever */
1280         timeouts.WriteTotalTimeoutMultiplier = 0;
1281         timeouts.WriteTotalTimeoutConstant = 0;
1282         SetCommTimeouts(file->handle, &timeouts);
1283
1284         if (GetCommState(file->handle, &dcb)) {
1285                 dcb.BaudRate = CBR_9600;
1286                 (void) SetCommState(file->handle, &dcb);
1287         }
1288
1289         return file;
1290 }
1291
1292 PUBLIC void
1293 altos_close(struct altos_file *file)
1294 {
1295         if (file->handle != INVALID_HANDLE_VALUE) {
1296                 CloseHandle(file->handle);
1297                 file->handle = INVALID_HANDLE_VALUE;
1298                 SetEvent(file->ov_read.hEvent);
1299                 SetEvent(file->ov_write.hEvent);
1300                 CloseHandle(file->ov_read.hEvent);
1301                 CloseHandle(file->ov_write.hEvent);
1302         }
1303 }
1304
1305 PUBLIC void
1306 altos_free(struct altos_file *file)
1307 {
1308         altos_close(file);
1309         free(file);
1310 }
1311
1312 PUBLIC int
1313 altos_putchar(struct altos_file *file, char c)
1314 {
1315         int     ret;
1316
1317         if (file->out_used == USB_BUF_SIZE) {
1318                 ret = altos_flush(file);
1319                 if (ret)
1320                         return ret;
1321         }
1322         file->out_data[file->out_used++] = c;
1323         if (file->out_used == USB_BUF_SIZE)
1324                 return altos_flush(file);
1325         return LIBALTOS_SUCCESS;
1326 }
1327
1328 PUBLIC int
1329 altos_getchar(struct altos_file *file, int timeout)
1330 {
1331         int     ret;
1332         while (file->in_read == file->in_used) {
1333                 if (file->handle == INVALID_HANDLE_VALUE)
1334                         return LIBALTOS_ERROR;
1335                 ret = altos_fill(file, timeout);
1336                 if (ret)
1337                         return ret;
1338         }
1339         return file->in_data[file->in_read++];
1340 }
1341
1342 struct altos_bt_list *
1343 altos_bt_list_start(int inquiry_time)
1344 {
1345         return NULL;
1346 }
1347
1348 int
1349 altos_bt_list_next(struct altos_bt_list *bt_list,
1350                    struct altos_bt_device *device)
1351 {
1352         return 0;
1353 }
1354
1355 void
1356 altos_bt_list_finish(struct altos_bt_list *bt_list)
1357 {
1358         free(bt_list);
1359 }
1360
1361 void
1362 altos_bt_fill_in(char *name, char *addr, struct altos_bt_device *device)
1363 {
1364         strncpy(device->name, name, sizeof (device->name));
1365         device->name[sizeof(device->name)-1] = '\0';
1366         strncpy(device->addr, addr, sizeof (device->addr));
1367         device->addr[sizeof(device->addr)-1] = '\0';
1368 }
1369
1370 struct altos_file *
1371 altos_bt_open(struct altos_bt_device *device)
1372 {
1373         return NULL;
1374 }
1375
1376 #endif