1/* ----------------------------------------------------------------------------
2 * "THE BEER-WARE LICENSE" (Revision 42) (by Poul-Henning Kamp):
3 * <joerg@FreeBSD.ORG> wrote this file.  As long as you retain this notice you
4 * can do whatever you want with this stuff. If we meet some day, and you think
5 * this stuff is worth it, you can buy me a beer in return.        Joerg Wunsch
6 * ----------------------------------------------------------------------------
7 *
8 * $FreeBSD$
9 */
10
11/*
12 * Simple demo program to illustrate the handling of FreeBSD's
13 * libusb20.
14 *
15 * Issues a bulk output, and then requests a bulk input.
16 */
17
18/*
19 * Examples:
20 * Just list all VID:PID pairs
21 * ./bulk
22 *
23 * Say "hello" to an Atmel JTAGICEmkII.
24 * ./bulk -o 2 -i 0x82 -v 0x03eb -p 0x2103 0x1b 0 0 1 0 0 0 0x0e 1 0xf3 0x97
25 *
26 * Return the INQUIRY data of an USB mass storage device.
27 * (It's best to have the umass(4) driver unloaded while doing such
28 * experiments, and perform a "usbconfig reset" for the device if it
29 * gets stuck.)
30 * ./bulk -v 0x5e3 -p 0x723 -i 0x81 -o 2 0x55 0x53 0x42 0x43 1 2 3 4 31 12 0x80 0x24 0 0 0 0x12 0 0 0 36 0 0 0 0 0 0 0 0 0 0
31 */
32
33
34#include <limits.h>
35#include <stdio.h>
36#include <stdint.h>
37#include <stdlib.h>
38#include <sysexits.h>
39#include <unistd.h>
40
41#include <libusb20.h>
42#include <libusb20_desc.h>
43
44#include "util.h"
45
46/*
47 * If you want to see the details of the internal datastructures
48 * in the debugger, unifdef the following.
49 */
50#ifdef DEBUG
51#  include <sys/queue.h>
52#  include "/usr/src/lib/libusb/libusb20_int.h"
53#endif
54
55#define BUFLEN 64
56
57#define TIMEOUT 5000 		/* 5 s */
58
59int in_ep, out_ep;		/* endpoints */
60uint8_t out_buf[BUFLEN];
61uint16_t out_len;
62
63static void
64doit(struct libusb20_device *dev)
65{
66  int rv;
67
68  /*
69   * Open the device, allocating memory for two possible (bulk or
70   * interrupt) transfers.
71   *
72   * If only control transfers are intended (via
73   * libusb20_dev_request_sync()), transfer_max can be given as 0.
74   */
75  if ((rv = libusb20_dev_open(dev, 2)) != 0)
76    {
77      fprintf(stderr, "libusb20_dev_open: %s\n", libusb20_strerror(rv));
78      return;
79    }
80
81  /*
82   * If the device has more than one configuration, select the desired
83   * one here.
84   */
85  if ((rv = libusb20_dev_set_config_index(dev, 0)) != 0)
86    {
87      fprintf(stderr, "libusb20_dev_set_config_index: %s\n", libusb20_strerror(rv));
88      return;
89    }
90
91  /*
92   * Two transfers have been requested in libusb20_dev_open() above;
93   * obtain the corresponding transfer struct pointers.
94   */
95  struct libusb20_transfer *xfr_out = libusb20_tr_get_pointer(dev, 0);
96  struct libusb20_transfer *xfr_in = libusb20_tr_get_pointer(dev, 1);
97
98  if (xfr_in == NULL || xfr_out == NULL)
99    {
100      fprintf(stderr, "libusb20_tr_get_pointer: %s\n", libusb20_strerror(rv));
101      return;
102    }
103
104  /*
105   * Open both transfers, the "out" one for the write endpoint, the
106   * "in" one for the read endpoint (ep | 0x80).
107   */
108  if ((rv = libusb20_tr_open(xfr_out, 0, 1, out_ep)) != 0)
109    {
110      fprintf(stderr, "libusb20_tr_open: %s\n", libusb20_strerror(rv));
111      return;
112    }
113  if ((rv = libusb20_tr_open(xfr_in, 0, 1, in_ep)) != 0)
114    {
115      fprintf(stderr, "libusb20_tr_open: %s\n", libusb20_strerror(rv));
116      return;
117    }
118
119  uint8_t in_buf[BUFLEN];
120  uint32_t rlen;
121
122  if (out_len > 0)
123    {
124      if ((rv = libusb20_tr_bulk_intr_sync(xfr_out, out_buf, out_len, &rlen, TIMEOUT))
125	  != 0)
126	{
127	  fprintf(stderr, "libusb20_tr_bulk_intr_sync (OUT): %s\n", libusb20_strerror(rv));
128	}
129      printf("sent %d bytes\n", rlen);
130    }
131
132  if ((rv = libusb20_tr_bulk_intr_sync(xfr_in, in_buf, BUFLEN, &rlen, TIMEOUT))
133      != 0)
134    {
135      fprintf(stderr, "libusb20_tr_bulk_intr_sync: %s\n", libusb20_strerror(rv));
136    }
137      printf("received %d bytes\n", rlen);
138      if (rlen > 0)
139	print_formatted(in_buf, rlen);
140
141  libusb20_tr_close(xfr_out);
142  libusb20_tr_close(xfr_in);
143
144  libusb20_dev_close(dev);
145}
146
147static void
148usage(void)
149{
150  fprintf(stderr,
151	  "Usage ./usb -i <IN_EP> -o <OUT_EP> -v <VID> -p <PID> [<outdata> ...\n]");
152  exit(EX_USAGE);
153}
154
155int
156main(int argc, char **argv)
157{
158  unsigned int vid = UINT_MAX, pid = UINT_MAX; /* impossible VID:PID */
159  int c;
160
161  while ((c = getopt(argc, argv, "i:o:p:v:")) != -1)
162    switch (c)
163      {
164      case 'i':
165	in_ep = strtol(optarg, NULL, 0);
166	break;
167
168      case 'o':
169	out_ep = strtol(optarg, NULL, 0);
170	break;
171
172      case 'p':
173	pid = strtol(optarg, NULL, 0);
174	break;
175
176      case 'v':
177	vid = strtol(optarg, NULL, 0);
178	break;
179
180      default:
181	usage();
182	break;
183      }
184  argc -= optind;
185  argv += optind;
186
187  if (vid != UINT_MAX || pid != UINT_MAX)
188    {
189      if (in_ep == 0 || out_ep == 0)
190	{
191	  usage();
192	}
193      if ((in_ep & 0x80) == 0)
194	{
195	  fprintf(stderr, "IN_EP must have bit 7 set\n");
196	  return (EX_USAGE);
197	}
198
199      if (argc > 0)
200	{
201	  for (out_len = 0; argc > 0 && out_len < BUFLEN; out_len++, argc--)
202	    {
203	      unsigned n = strtoul(argv[out_len], 0, 0);
204	      if (n > 255)
205		fprintf(stderr,
206			"Warning: data #%d 0x%0x > 0xff, truncating\n",
207			out_len, n);
208	      out_buf[out_len] = (uint8_t)n;
209	    }
210	  out_len++;
211	  if (argc > 0)
212	    fprintf(stderr,
213		    "Data count exceeds maximum of %d, ignoring %d elements\n",
214		    BUFLEN, optind);
215	}
216    }
217
218  struct libusb20_backend *be;
219  struct libusb20_device *dev;
220
221  if ((be = libusb20_be_alloc_default()) == NULL)
222    {
223      perror("libusb20_be_alloc()");
224      return 1;
225    }
226
227  dev = NULL;
228  while ((dev = libusb20_be_device_foreach(be, dev)) != NULL)
229    {
230      struct LIBUSB20_DEVICE_DESC_DECODED *ddp =
231      libusb20_dev_get_device_desc(dev);
232
233      printf("Found device %s (VID:PID = 0x%04x:0x%04x)\n",
234	     libusb20_dev_get_desc(dev),
235	     ddp->idVendor, ddp->idProduct);
236
237      if (ddp->idVendor == vid && ddp->idProduct == pid)
238	doit(dev);
239    }
240
241  libusb20_be_free(be);
242  return 0;
243}
244