Fix unlocked flying_transfers list access
[platform/upstream/libusb.git] / libusb / io.c
1 /*
2  * I/O functions for libusb
3  * Copyright (C) 2007-2008 Daniel Drake <dsd@gentoo.org>
4  * Copyright (c) 2001 Johannes Erdfelt <johannes@erdfelt.com>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <config.h>
22 #include <errno.h>
23 #include <poll.h>
24 #include <pthread.h>
25 #include <signal.h>
26 #include <stdint.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <sys/select.h>
30 #include <sys/time.h>
31 #include <time.h>
32 #include <unistd.h>
33
34 #include "libusbi.h"
35
36 /* this is a list of in-flight transfer handles, sorted by timeout expiration.
37  * URBs to timeout the soonest are placed at the beginning of the list, URBs
38  * that will time out later are placed after, and urbs with infinite timeout
39  * are always placed at the very end. */
40 static struct list_head flying_transfers;
41 static pthread_mutex_t flying_transfers_lock = PTHREAD_MUTEX_INITIALIZER;
42
43 /* list of poll fd's */
44 static struct list_head pollfds;
45 static pthread_mutex_t pollfds_lock = PTHREAD_MUTEX_INITIALIZER;
46
47 /* user callbacks for pollfd changes */
48 static libusb_pollfd_added_cb fd_added_cb = NULL;
49 static libusb_pollfd_removed_cb fd_removed_cb = NULL;
50
51 /**
52  * \page io Synchronous and asynchronous device I/O
53  *
54  * \section intro Introduction
55  *
56  * If you're using libusb in your application, you're probably wanting to
57  * perform I/O with devices - you want to perform USB data transfers.
58  *
59  * libusb offers two separate interfaces for device I/O. This page aims to
60  * introduce the two in order to help you decide which one is more suitable
61  * for your application. You can also choose to use both interfaces in your
62  * application by considering each transfer on a case-by-case basis.
63  *
64  * Once you have read through the following discussion, you should consult the
65  * detailed API documentation pages for the details:
66  * - \ref syncio
67  * - \ref asyncio
68  *
69  * \section theory Transfers at a logical level
70  *
71  * At a logical level, USB transfers typically happen in two parts. For
72  * example, when reading data from a endpoint:
73  * -# A request for data is sent to the device
74  * -# Some time later, the incoming data is received by the host
75  *
76  * or when writing data to an endpoint:
77  *
78  * -# The data is sent to the device
79  * -# Some time later, the host receives acknowledgement from the device that
80  *    the data has been transferred.
81  *
82  * There may be an indefinite delay between the two steps. Consider a
83  * fictional USB input device with a button that the user can press. In order
84  * to determine when the button is pressed, you would likely submit a request
85  * to read data on a bulk or interrupt endpoint and wait for data to arrive.
86  * Data will arrive when the button is pressed by the user, which is
87  * potentially hours later.
88  *
89  * libusb offers both a synchronous and an asynchronous interface to performing
90  * USB transfers. The main difference is that the synchronous interface
91  * combines both steps indicated above into a single function call, whereas
92  * the asynchronous interface separates them.
93  *
94  * \section sync The synchronous interface
95  *
96  * The synchronous I/O interface allows you to perform a USB transfer with
97  * a single function call. When the function call returns, the transfer has
98  * completed and you can parse the results.
99  *
100  * If you have used the libusb-0.1 before, this I/O style will seem familar to
101  * you. libusb-0.1 only offered a synchronous interface.
102  *
103  * In our input device example, to read button presses you might write code
104  * in the following style:
105 \code
106 unsigned char data[4];
107 int actual_length,
108 int r = libusb_bulk_transfer(handle, EP_IN, data, sizeof(data), &actual_length, 0);
109 if (r == 0 && actual_length == sizeof(data)) {
110         // results of the transaction can now be found in the data buffer
111         // parse them here and report button press
112 } else {
113         error();
114 }
115 \endcode
116  *
117  * The main advantage of this model is simplicity: you did everything with
118  * a single simple function call.
119  *
120  * However, this interface has its limitations. Your application will sleep
121  * inside libusb_bulk_transfer() until the transaction has completed. If it
122  * takes the user 3 hours to press the button, your application will be
123  * sleeping for that long. Execution will be tied up inside the library -
124  * the entire thread will be useless for that duration.
125  *
126  * Another issue is that by tieing up the thread with that single transaction
127  * there is no possibility of performing I/O with multiple endpoints and/or
128  * multiple devices simultaneously, unless you resort to creating one thread
129  * per transaction.
130  *
131  * Additionally, there is no opportunity to cancel the transfer after the
132  * request has been submitted.
133  *
134  * For details on how to use the synchronous API, see the
135  * \ref syncio "synchronous I/O API documentation" pages.
136  * 
137  * \section async The asynchronous interface
138  *
139  * Asynchronous I/O is the most significant new feature in libusb-1.0.
140  * Although it is a more complex interface, it solves all the issues detailed
141  * above.
142  *
143  * Instead of providing which functions that block until the I/O has complete,
144  * libusb's asynchronous interface presents non-blocking functions which
145  * begin a transfer and then return immediately. Your application passes a
146  * callback function pointer to this non-blocking function, which libusb will
147  * call with the results of the transaction when it has completed.
148  *
149  * Transfers which have been submitted through the non-blocking functions
150  * can be cancelled with a separate function call.
151  *
152  * The non-blocking nature of this interface allows you to be simultaneously
153  * performing I/O to multiple endpoints on multiple devices, without having
154  * to use threads.
155  *
156  * This added flexibility does come with some complications though:
157  * - In the interest of being a lightweight library, libusb does not create
158  * threads and can only operate when your application is calling into it. Your
159  * application must call into libusb from it's main loop when events are ready
160  * to be handled, or you must use some other scheme to allow libusb to
161  * undertake whatever work needs to be done.
162  * - libusb also needs to be called into at certain fixed points in time in
163  * order to accurately handle transfer timeouts.
164  * - Memory handling becomes more complex. You cannot use stack memory unless
165  * the function with that stack is guaranteed not to return until the transfer
166  * callback has finished executing.
167  * - You generally lose some linearity from your code flow because submitting
168  * the transfer request is done in a separate function from where the transfer
169  * results are handled. This becomes particularly obvious when you want to
170  * submit a second transfer based on the results of an earlier transfer.
171  *
172  * Internally, libusb's synchronous interface is expressed in terms of function
173  * calls to the asynchronous interface.
174  *
175  * For details on how to use the asynchronous API, see the
176  * \ref asyncio "asynchronous I/O API" documentation pages.
177  */
178
179 /**
180  * @defgroup asyncio Asynchronous device I/O
181  *
182  * This page details libusb's asynchronous (non-blocking) API for USB device
183  * I/O. This interface is very powerful but is also quite complex - you will
184  * need to read this page carefully to understand the necessary considerations
185  * and issues surrounding use of this interface. Simplistic applications
186  * may wish to consider the \ref syncio "synchronous I/O API" instead.
187  *
188  * The asynchronous interface is built around the idea of separating transfer
189  * submission and handling of transfer completion (the synchronous model
190  * combines both of these into one). There may be a long delay between
191  * submission and completion, however the asynchronous submission function
192  * is non-blocking so will return control to your application during that
193  * potentially long delay.
194  *
195  * \section asyncabstraction Transfer abstraction
196  *
197  * For the asynchronous I/O, libusb implements the concept of a generic
198  * transfer entity for all types of I/O (control, bulk, interrupt,
199  * isochronous). The generic transfer object must be treated slightly
200  * differently depending on which type of I/O you are performing with it.
201  *
202  * This is represented by the public libusb_transfer structure type.
203  *
204  * \section asynctrf Asynchronous transfers
205  *
206  * We can view asynchronous I/O as a 5 step process:
207  * -# Allocation
208  * -# Filling
209  * -# Submission
210  * -# Completion handling
211  * -# Deallocation
212  *
213  * \subsection asyncalloc Allocation
214  *
215  * This step involves allocating memory for a USB transfer. This is the
216  * generic transfer object mentioned above. At this stage, the transfer
217  * is "blank" with no details about what type of I/O it will be used for.
218  *
219  * Allocation is done with the libusb_alloc_transfer() function. You must use
220  * this function rather than allocating your own transfers.
221  *
222  * \subsection asyncfill Filling
223  *
224  * This step is where you take a previously allocated transfer and fill it
225  * with information to determine the message type and direction, data buffer,
226  * callback function, etc.
227  *
228  * You can either fill the required fields yourself or you can use the
229  * helper functions: libusb_fill_control_transfer(), libusb_fill_bulk_transfer()
230  * and libusb_fill_interrupt_transfer().
231  *
232  * \subsection asyncsubmit Submission
233  *
234  * When you have allocated a transfer and filled it, you can submit it using
235  * libusb_submit_transfer(). This function returns immediately but can be
236  * regarded as firing off the I/O request in the background.
237  *
238  * \subsection asynccomplete Completion handling
239  *
240  * After a transfer has been submitted, one of four things can happen to it:
241  *
242  * - The transfer completes (i.e. some data was transferred)
243  * - The transfer has a timeout and the timeout expires before all data is
244  * transferred
245  * - The transfer fails due to an error
246  * - The transfer is cancelled
247  *
248  * Each of these will cause the user-specified transfer callback function to
249  * be invoked. It is up to the callback function to determine which of the
250  * above actually happened and to act accordingly.
251  *
252  * \subsection Deallocation
253  *
254  * When a transfer has completed (i.e. the callback function has been invoked),
255  * you are advised to free the transfer (unless you wish to resubmit it, see
256  * below). Transfers are deallocated with libusb_free_transfer().
257  *
258  * It is undefined behaviour to free a transfer which has not completed.
259  *
260  * \section asyncresubmit Resubmission
261  *
262  * You may be wondering why allocation, filling, and submission are all
263  * separated above where they could reasonably be combined into a single
264  * operation.
265  *
266  * The reason for separation is to allow you to resubmit transfers without
267  * having to allocate new ones every time. This is especially useful for
268  * common situations dealing with interrupt endpoints - you allocate one
269  * transfer, fill and submit it, and when it returns with results you just
270  * resubmit it for the next interrupt.
271  *
272  * \section asynccancel Cancellation
273  *
274  * Another advantage of using the asynchronous interface is that you have
275  * the ability to cancel transfers which have not yet completed. This is
276  * done by calling the libusb_cancel_transfer() function.
277  *
278  * libusb_cancel_transfer() is asynchronous/non-blocking in itself. When the
279  * cancellation actually completes, the transfer's callback function will
280  * be invoked, and the callback function should check the transfer status to
281  * determine that it was cancelled.
282  *
283  * Freeing the transfer after it has been cancelled but before cancellation
284  * has completed will result in undefined behaviour.
285  *
286  * \section asyncctrl Considerations for control transfers
287  *
288  * The <tt>libusb_transfer</tt> structure is generic and hence does not
289  * include specific fields for the control-specific setup packet structure.
290  *
291  * In order to perform a control transfer, you must place the 8-byte setup
292  * packet at the start of the data buffer. To simplify this, you could
293  * cast the buffer pointer to type struct libusb_control_setup, or you can
294  * use the helper function libusb_fill_control_setup().
295  *
296  * The wLength field placed in the setup packet must be the length you would
297  * expect to be sent in the setup packet: the length of the payload that
298  * follows (or the expected maximum number of bytes to receive). However,
299  * the length field of the libusb_transfer object must be the length of
300  * the data buffer - i.e. it should be wLength <em>plus</em> the size of
301  * the setup packet (LIBUSB_CONTROL_SETUP_SIZE).
302  *
303  * If you use the helper functions, this is simplified for you:
304  * -# Allocate a buffer of size LIBUSB_CONTROL_SETUP_SIZE plus the size of the
305  * data you are sending/requesting.
306  * -# Call libusb_fill_control_setup() on the data buffer, using the transfer
307  * request size as the wLength value (i.e. do not include the extra space you
308  * allocated for the control setup).
309  * -# If this is a host-to-device transfer, place the data to be transferred
310  * in the data buffer, starting at offset LIBUSB_CONTROL_SETUP_SIZE.
311  * -# Call libusb_fill_control_transfer() to associate the data buffer with
312  * the transfer (and to set the remaining details such as callback and timeout).
313  *   - Note that there is no parameter to set the length field of the transfer.
314  *     The length is automatically inferred from the wLength field of the setup
315  *     packet.
316  * -# Submit the transfer.
317  *
318  * Further considerations are needed when handling transfer completion in
319  * your callback function:
320  * - As you might expect, the setup packet will still be sitting at the start
321  * of the data buffer.
322  * - If this was a device-to-host transfer, the received data will be sitting
323  * at offset LIBUSB_CONTROL_SETUP_SIZE into the buffer.
324  * - The actual_length field of the transfer structure is relative to the
325  * wLength of the setup packet, rather than the size of the data buffer. So,
326  * if your wLength was 4, your transfer's <tt>length</tt> was 12, then you
327  * should expect an <tt>actual_length</tt> of 4 to indicate that the data was
328  * transferred in entirity.
329  *
330  * To simplify parsing of setup packets and obtaining the data from the
331  * correct offset, you may wish to use the libusb_control_transfer_get_data()
332  * and libusb_control_transfer_get_setup() functions within your transfer
333  * callback.
334  *
335  * \section asyncintr Considerations for interrupt transfers
336  * 
337  * All interrupt transfers are performed using the polling interval presented
338  * by the bInterval value of the endpoint descriptor.
339  *
340  * \section asynciso Considerations for isochronous transfers
341  *
342  * As isochronous transfers are more complicated than transfers to
343  * non-isochronous endpoints.
344  *
345  * To perform I/O to an isochronous endpoint, allocate the transfer by calling
346  * libusb_alloc_transfer() with an appropriate number of isochronous packets.
347  *
348  * During filling, set \ref libusb_transfer::endpoint_type "endpoint_type" to
349  * \ref libusb_endpoint_type::LIBUSB_ENDPOINT_TYPE_ISOCHRONOUS
350  * "LIBUSB_ENDPOINT_TYPE_ISOCHRONOUS", and set
351  * \ref libusb_transfer::num_iso_packets "num_iso_packets" to a value less than
352  * or equal to the number of packets you requested during allocation.
353  * libusb_alloc_transfer() does not set either of these fields for you, given
354  * that you might not even use the transfer on an isochronous endpoint.
355  *
356  * Next, populate the length field for the first num_iso_packets entries in
357  * the \ref libusb_transfer::iso_packet_desc "iso_packet_desc" array. Section
358  * 5.6.3 of the USB2 specifications describe how the maximum isochronous
359  * packet length is determined by the endpoint descriptor. FIXME need a helper
360  * function to find this.
361  * FIXME, write a helper function to set the length for all iso packets in an
362  * array
363  *
364  * For outgoing transfers, you'll obviously fill the buffer and populate the
365  * packet descriptors in hope that all the data gets transferred. For incoming
366  * transfers, you must ensure the buffer has sufficient capacity for
367  * the situation where all packets transfer the full amount of requested data.
368  *
369  * Completion handling requires some extra consideration. The
370  * \ref libusb_transfer::actual_length "actual_length" field of the transfer
371  * is meaningless and should not be examined; instead you must refer to the
372  * \ref libusb_iso_packet_descriptor::actual_length "actual_length" field of
373  * each individual packet.
374  *
375  * The \ref libusb_transfer::status "status" field of the transfer is also a
376  * little misleading:
377  *  - If the packets were submitted and the isochronous data microframes
378  *    completed normally, status will have value
379  *    \ref libusb_transfer_status::LIBUSB_TRANSFER_COMPLETED
380  *    "LIBUSB_TRANSFER_COMPLETED". Note that bus errors and software-incurred
381  *    delays are not counted as transfer errors; the transfer.status field may
382  *    indicate COMPLETED even if some or all of the packets failed. Refer to
383  *    the \ref libusb_iso_packet_descriptor::status "status" field of each
384  *    individual packet to determine packet failures.
385  *  - The status field will have value
386  *    \ref libusb_transfer_status::LIBUSB_TRANSFER_ERROR
387  *    "LIBUSB_TRANSFER_ERROR" only when serious errors were encountered.
388  *  - Other transfer status codes occur with normal behaviour.
389  *
390  * The data for each packet will be found at an offset into the buffer that
391  * can be calculated as if each prior packet completed in full. FIXME write
392  * a helper function to determine this, and flesh this description out a bit
393  * more.
394  *
395  * \section asyncmem Memory caveats
396  *
397  * In most circumstances, it is not safe to use stack memory for transfer
398  * buffers. This is because the function that fired off the asynchronous
399  * transfer may return before libusb has finished using the buffer, and when
400  * the function returns it's stack gets destroyed. This is true for both
401  * host-to-device and device-to-host transfers.
402  *
403  * The only case in which it is safe to use stack memory is where you can
404  * guarantee that the function owning the stack space for the buffer does not
405  * return until after the transfer's callback function has completed. In every
406  * other case, you need to use heap memory instead.
407  *
408  * \section asyncflags Fine control
409  *
410  * Through using this asynchronous interface, you may find yourself repeating
411  * a few simple operations many times. You can apply a bitwise OR of certain
412  * flags to a transfer to simplify certain things:
413  * - \ref libusb_transfer_flags::LIBUSB_TRANSFER_SHORT_NOT_OK
414  *   "LIBUSB_TRANSFER_SHORT_NOT_OK" results in transfers which transferred
415  *   less than the requested amount of data being marked with status
416  *   \ref libusb_transfer_status::LIBUSB_TRANSFER_ERROR "LIBUSB_TRANSFER_ERROR"
417  *   (they would normally be regarded as COMPLETED)
418  * - \ref libusb_transfer_flags::LIBUSB_TRANSFER_FREE_BUFFER
419  *   "LIBUSB_TRANSFER_FREE_BUFFER" allows you to ask libusb to free the transfer
420  *   buffer when freeing the transfer.
421  * - \ref libusb_transfer_flags::LIBUSB_TRANSFER_FREE_TRANSFER
422  *   "LIBUSB_TRANSFER_FREE_TRANSFER" causes libusb to automatically free the
423  *   transfer after the transfer callback returns.
424  *
425  * \section asyncevent Event handling
426  *
427  * In accordance of the aim of being a lightweight library, libusb does not
428  * create threads internally. This means that libusb code does not execute
429  * at any time other than when your application is calling a libusb function.
430  * However, an asynchronous model requires that libusb perform work at various
431  * points in time - namely processing the results of previously-submitted
432  * transfers and invoking the user-supplied callback function.
433  *
434  * This gives rise to the libusb_handle_events() function which your
435  * application must call into when libusb has work do to. This gives libusb
436  * the opportunity to reap pending transfers, invoke callbacks, etc.
437  *
438  * The first issue to discuss here is how your application can figure out
439  * when libusb has work to do. In fact, there are two naive options which
440  * do not actually require your application to know this:
441  * -# Periodically call libusb_handle_events() in non-blocking mode at fixed
442  *    short intervals from your main loop
443  * -# Repeatedly call libusb_handle_events() in blocking mode from a dedicated
444  *    thread.
445  *
446  * The first option is plainly not very nice, and will cause unnecessary 
447  * CPU wakeups leading to increased power usage and decreased battery life.
448  * The second option is not very nice either, but may be the nicest option
449  * available to you if the "proper" approach can not be applied to your
450  * application (read on...).
451  * 
452  * The recommended option is to integrate libusb with your application main
453  * event loop. libusb exposes a set of file descriptors which allow you to do
454  * this. Your main loop is probably already calling poll() or select() or a
455  * variant on a set of file descriptors for other event sources (e.g. keyboard
456  * button presses, mouse movements, network sockets, etc). You then add
457  * libusb's file descriptors to your poll()/select() calls, and when activity
458  * is detected on such descriptors you know it is time to call
459  * libusb_handle_events().
460  *
461  * There is one final event handling complication. libusb supports
462  * asynchronous transfers which time out after a specified time period, and
463  * this requires that libusb is called into at or after the timeout so that
464  * the timeout can be handled. So, in addition to considering libusb's file
465  * descriptors in your main event loop, you must also consider that libusb
466  * sometimes needs to be called into at fixed points in time even when there
467  * is no file descriptor activity.
468  *
469  * For the details on retrieving the set of file descriptors and determining
470  * the next timeout, see the \ref poll "polling and timing" API documentation.
471  */
472
473 /**
474  * @defgroup poll Polling and timing
475  *
476  * This page documents libusb's functions for polling events and timing.
477  * These functions are only necessary for users of the
478  * \ref asyncio "asynchronous API". If you are only using the simpler
479  * \ref syncio "synchronous API" then you do not need to ever call these
480  * functions.
481  *
482  * The justification for the functionality described here has already been
483  * discussed in the \ref asyncevent "event handling" section of the
484  * asynchronous API documentation. In summary, libusb does not create internal
485  * threads for event processing and hence relies on your application calling
486  * into libusb at certain points in time so that pending events can be handled.
487  * In order to know precisely when libusb needs to be called into, libusb
488  * offers you a set of pollable file descriptors and information about when
489  * the next timeout expires.
490  *
491  * If you are using the asynchronous I/O API, you must take one of the two
492  * following options, otherwise your I/O will not complete.
493  *
494  * \section pollsimple The simple option
495  *
496  * If your application revolves solely around libusb and does not need to
497  * handle other event sources, you can have a program structure as follows:
498 \code
499 // initialize libusb
500 // find and open device
501 // maybe fire off some initial async I/O
502
503 while (user_has_not_requested_exit)
504         libusb_handle_events();
505
506 // clean up and exit
507 \endcode
508  *
509  * With such a simple main loop, you do not have to worry about managing
510  * sets of file descriptors or handling timeouts. libusb_handle_events() will
511  * handle those details internally.
512  *
513  * \section pollmain The more advanced option
514  *
515  * In more advanced applications, you will already have a main loop which
516  * is monitoring other event sources: network sockets, X11 events, mouse
517  * movements, etc. Through exposing a set of file descriptors, libusb is
518  * designed to cleanly integrate into such main loops.
519  *
520  * In addition to polling file descriptors for the other event sources, you
521  * take a set of file descriptors from libusb and monitor those too. When you
522  * detect activity on libusb's file descriptors, you call
523  * libusb_handle_events_timeout() in non-blocking mode.
524  *
525  * You must also consider the fact that libusb sometimes has to handle events
526  * at certain known times which do not generate activity on file descriptors.
527  * Your main loop must also consider these times, modify it's poll()/select()
528  * timeout accordingly, and track time so that libusb_handle_events_timeout()
529  * is called in non-blocking mode when timeouts expire.
530  *
531  * In pseudo-code, you want something that looks like:
532 \code
533 // initialise libusb
534
535 libusb_get_pollfds()
536 while (user has not requested application exit) {
537         libusb_get_next_timeout();
538         select(on libusb file descriptors plus any other event sources of interest,
539                 using a timeout no larger than the value libusb just suggested)
540         if (select() indicated activity on libusb file descriptors)
541                 libusb_handle_events_timeout(0);
542         if (time has elapsed to or beyond the libusb timeout)
543                 libusb_handle_events_timeout(0);
544 }
545
546 // clean up and exit
547 \endcode
548  *
549  * The set of file descriptors that libusb uses as event sources may change
550  * during the life of your application. Rather than having to repeatedly
551  * call libusb_get_pollfds(), you can set up notification functions for when
552  * the file descriptor set changes using libusb_set_pollfd_notifiers().
553  *
554  */
555
556 void usbi_io_init()
557 {
558         list_init(&flying_transfers);
559         list_init(&pollfds);
560         fd_added_cb = NULL;
561         fd_removed_cb = NULL;
562 }
563
564 static int calculate_timeout(struct usbi_transfer *transfer)
565 {
566         int r;
567         struct timespec current_time;
568         unsigned int timeout =
569                 __USBI_TRANSFER_TO_LIBUSB_TRANSFER(transfer)->timeout;
570
571         if (!timeout)
572                 return 0;
573
574         r = clock_gettime(CLOCK_MONOTONIC, &current_time);
575         if (r < 0) {
576                 usbi_err("failed to read monotonic clock, errno=%d", errno);
577                 return r;
578         }
579
580         current_time.tv_sec += timeout / 1000;
581         current_time.tv_nsec += (timeout % 1000) * 1000000;
582
583         if (current_time.tv_nsec > 1000000000) {
584                 current_time.tv_nsec -= 1000000000;
585                 current_time.tv_sec++;
586         }
587
588         TIMESPEC_TO_TIMEVAL(&transfer->timeout, &current_time);
589         return 0;
590 }
591
592 static void add_to_flying_list(struct usbi_transfer *transfer)
593 {
594         struct usbi_transfer *cur;
595         struct timeval *timeout = &transfer->timeout;
596         
597         pthread_mutex_lock(&flying_transfers_lock);
598
599         /* if we have no other flying transfers, start the list with this one */
600         if (list_empty(&flying_transfers)) {
601                 list_add(&transfer->list, &flying_transfers);
602                 goto out;
603         }
604
605         /* if we have infinite timeout, append to end of list */
606         if (!timerisset(timeout)) {
607                 list_add_tail(&transfer->list, &flying_transfers);
608                 goto out;
609         }
610
611         /* otherwise, find appropriate place in list */
612         list_for_each_entry(cur, &flying_transfers, list) {
613                 /* find first timeout that occurs after the transfer in question */
614                 struct timeval *cur_tv = &cur->timeout;
615
616                 if (!timerisset(cur_tv) || (cur_tv->tv_sec > timeout->tv_sec) ||
617                                 (cur_tv->tv_sec == timeout->tv_sec &&
618                                         cur_tv->tv_usec > timeout->tv_usec)) {
619                         list_add_tail(&transfer->list, &cur->list);
620                         goto out;
621                 }
622         }
623
624         /* otherwise we need to be inserted at the end */
625         list_add_tail(&transfer->list, &flying_transfers);
626 out:
627         pthread_mutex_unlock(&flying_transfers_lock);
628 }
629
630 static int submit_transfer(struct usbi_transfer *itransfer)
631 {
632         int r;
633         
634         add_to_flying_list(itransfer);
635         r = usbi_backend->submit_transfer(itransfer);
636         if (r < 0) {
637                 pthread_mutex_lock(&flying_transfers_lock);
638                 list_del(&itransfer->list);
639                 pthread_mutex_unlock(&flying_transfers_lock);
640         }
641
642         return r;
643 }
644
645 /** \ingroup asyncio
646  * Allocate a libusb transfer with a specified number of isochronous packet
647  * descriptors. The returned transfer is pre-initialized for you. When the new
648  * transfer is no longer needed, it should be freed with
649  * libusb_free_transfer().
650  *
651  * Transfers intended for non-isochronous endpoints (e.g. control, bulk,
652  * interrupt) should specify an iso_packets count of zero.
653  *
654  * For transfers intended for isochronous endpoints, specify an appropriate
655  * number of packet descriptors to be allocated as part of the transfer.
656  * The returned transfer is not specially initialized for isochronous I/O;
657  * you are still required to set the
658  * \ref libusb_transfer::num_iso_packets "num_iso_packets" and
659  * \ref libusb_transfer::endpoint_type "endpoint_type" fields accordingly.
660  *
661  * It is safe to allocate a transfer with some isochronous packets and then
662  * use it on a non-isochronous endpoint. If you do this, ensure that at time
663  * of submission, num_iso_packets is 0 and that endpoint_type is set
664  * appropriately.
665  *
666  * \param iso_packets number of isochronous packet descriptors to allocate
667  * \returns a newly allocated transfer, or NULL on error
668  */
669 API_EXPORTED struct libusb_transfer *libusb_alloc_transfer(int iso_packets)
670 {
671         size_t os_alloc_size = usbi_backend->transfer_priv_size
672                 + (usbi_backend->add_iso_packet_size * iso_packets);
673         int alloc_size = sizeof(struct usbi_transfer)
674                 + sizeof(struct libusb_transfer)
675                 + (sizeof(struct libusb_iso_packet_descriptor) * iso_packets)
676                 + os_alloc_size;
677         struct usbi_transfer *itransfer = malloc(alloc_size);
678         if (!itransfer)
679                 return NULL;
680
681         memset(itransfer, 0, alloc_size);
682         itransfer->num_iso_packets = iso_packets;
683         return __USBI_TRANSFER_TO_LIBUSB_TRANSFER(itransfer);
684 }
685
686 /** \ingroup asyncio
687  * Free a transfer structure. This should be called for all transfers
688  * allocated with libusb_alloc_transfer().
689  *
690  * If the \ref libusb_transfer_flags::LIBUSB_TRANSFER_FREE_BUFFER
691  * "LIBUSB_TRANSFER_FREE_BUFFER" flag is set and the transfer buffer is
692  * non-NULL, this function will also free the transfer buffer using the
693  * standard system memory allocator (e.g. free()).
694  *
695  * It is legal to call this function with a NULL transfer. In this case,
696  * the function will simply return safely.
697  *
698  * \param transfer the transfer to free
699  */
700 API_EXPORTED void libusb_free_transfer(struct libusb_transfer *transfer)
701 {
702         struct usbi_transfer *itransfer;
703         if (!transfer)
704                 return;
705
706         if (transfer->flags & LIBUSB_TRANSFER_FREE_BUFFER && transfer->buffer)
707                 free(transfer->buffer);
708
709         itransfer = __LIBUSB_TRANSFER_TO_USBI_TRANSFER(transfer);
710         free(itransfer);
711 }
712
713 /** \ingroup asyncio
714  * Submit a transfer. This function will fire off the USB transfer and then
715  * return immediately.
716  *
717  * It is undefined behaviour to submit a transfer that has already been
718  * submitted but has not yet completed.
719  *
720  * \param transfer the transfer to submit
721  * \returns 0 on success
722  * \returns negative on error
723  */
724 API_EXPORTED int libusb_submit_transfer(struct libusb_transfer *transfer)
725 {
726         struct usbi_transfer *itransfer =
727                 __LIBUSB_TRANSFER_TO_USBI_TRANSFER(transfer);
728         int r;
729
730         itransfer->transferred = 0;
731         r = calculate_timeout(itransfer);
732         if (r < 0)
733                 return r;
734
735         if (transfer->endpoint_type == LIBUSB_ENDPOINT_TYPE_CONTROL) {
736                 struct libusb_control_setup *setup =
737                         (struct libusb_control_setup *) transfer->buffer;
738         
739                 usbi_dbg("RQT=%02x RQ=%02x VAL=%04x IDX=%04x length=%d",
740                         setup->bmRequestType, setup->bRequest, setup->wValue, setup->wIndex,
741                         setup->wLength);
742
743                 setup->wValue = cpu_to_le16(setup->wValue);
744                 setup->wIndex = cpu_to_le16(setup->wIndex);
745                 setup->wLength = cpu_to_le16(setup->wLength);
746         }
747
748         return submit_transfer(itransfer);
749 }
750
751 /** \ingroup asyncio
752  * Asynchronously cancel a previously submitted transfer.
753  * It is undefined behaviour to call this function on a transfer that is
754  * already being cancelled or has already completed.
755  * This function returns immediately, but this does not indicate cancellation
756  * is complete. Your callback function will be invoked at some later time
757  * with a transfer status of
758  * \ref libusb_transfer_status::LIBUSB_TRANSFER_CANCELLED
759  * "LIBUSB_TRANSFER_CANCELLED."
760  *
761  * \param transfer the transfer to cancel
762  * \returns 0 on success
763  * \returns non-zero on error
764  */
765 API_EXPORTED int libusb_cancel_transfer(struct libusb_transfer *transfer)
766 {
767         struct usbi_transfer *itransfer =
768                 __LIBUSB_TRANSFER_TO_USBI_TRANSFER(transfer);
769         int r;
770
771         usbi_dbg("");
772         r = usbi_backend->cancel_transfer(itransfer);
773         if (r < 0)
774                 usbi_err("cancel transfer failed error %d", r);
775         return r;
776 }
777
778 void usbi_handle_transfer_completion(struct usbi_transfer *itransfer,
779         enum libusb_transfer_status status)
780 {
781         struct libusb_transfer *transfer =
782                 __USBI_TRANSFER_TO_LIBUSB_TRANSFER(itransfer);
783         uint8_t flags;
784
785         pthread_mutex_lock(&flying_transfers_lock);
786         list_del(&itransfer->list);
787         pthread_mutex_unlock(&flying_transfers_lock);
788
789         if (status == LIBUSB_TRANSFER_COMPLETED
790                         && transfer->flags & LIBUSB_TRANSFER_SHORT_NOT_OK) {
791                 int rqlen = transfer->length;
792                 if (transfer->endpoint_type == LIBUSB_ENDPOINT_TYPE_CONTROL)
793                         rqlen -= LIBUSB_CONTROL_SETUP_SIZE;
794                 if (rqlen != itransfer->transferred) {
795                         usbi_dbg("interpreting short transfer as error");
796                         status = LIBUSB_TRANSFER_ERROR;
797                 }
798         }
799
800         flags = transfer->flags;
801         transfer->status = status;
802         transfer->actual_length = itransfer->transferred;
803         if (transfer->callback)
804                 transfer->callback(transfer);
805         /* transfer might have been freed by the above call, do not use from
806          * this point. */
807         if (flags & LIBUSB_TRANSFER_FREE_TRANSFER)
808                 libusb_free_transfer(transfer);
809 }
810
811 void usbi_handle_transfer_cancellation(struct usbi_transfer *transfer)
812 {
813         /* if the URB was cancelled due to timeout, report timeout to the user */
814         if (transfer->flags & USBI_TRANSFER_TIMED_OUT) {
815                 usbi_dbg("detected timeout cancellation");
816                 usbi_handle_transfer_completion(transfer, LIBUSB_TRANSFER_TIMED_OUT);
817                 return;
818         }
819
820         /* otherwise its a normal async cancel */
821         usbi_handle_transfer_completion(transfer, LIBUSB_TRANSFER_CANCELLED);
822 }
823
824 static void handle_timeout(struct usbi_transfer *itransfer)
825 {
826         struct libusb_transfer *transfer =
827                 __USBI_TRANSFER_TO_LIBUSB_TRANSFER(itransfer);
828         int r;
829
830         itransfer->flags |= USBI_TRANSFER_TIMED_OUT;
831         r = libusb_cancel_transfer(transfer);
832         if (r < 0)
833                 usbi_warn("async cancel failed %d errno=%d", r, errno);
834 }
835
836 static int handle_timeouts(void)
837 {
838         struct timespec systime_ts;
839         struct timeval systime;
840         struct usbi_transfer *transfer;
841         int r = 0;
842
843         pthread_mutex_lock(&flying_transfers_lock);
844         if (list_empty(&flying_transfers))
845                 goto out;
846
847         /* get current time */
848         r = clock_gettime(CLOCK_MONOTONIC, &systime_ts);
849         if (r < 0)
850                 goto out;
851
852         TIMESPEC_TO_TIMEVAL(&systime, &systime_ts);
853
854         /* iterate through flying transfers list, finding all transfers that
855          * have expired timeouts */
856         list_for_each_entry(transfer, &flying_transfers, list) {
857                 struct timeval *cur_tv = &transfer->timeout;
858
859                 /* if we've reached transfers of infinite timeout, we're all done */
860                 if (!timerisset(cur_tv))
861                         goto out;
862
863                 /* ignore timeouts we've already handled */
864                 if (transfer->flags & USBI_TRANSFER_TIMED_OUT)
865                         continue;
866
867                 /* if transfer has non-expired timeout, nothing more to do */
868                 if ((cur_tv->tv_sec > systime.tv_sec) ||
869                                 (cur_tv->tv_sec == systime.tv_sec &&
870                                         cur_tv->tv_usec > systime.tv_usec))
871                         goto out;
872         
873                 /* otherwise, we've got an expired timeout to handle */
874                 handle_timeout(transfer);
875         }
876
877 out:
878         pthread_mutex_unlock(&flying_transfers_lock);
879         return r;
880 }
881
882 static int handle_events(struct timeval *tv)
883 {
884         int r;
885         int maxfd = 0;
886         fd_set readfds, writefds;
887         fd_set *_readfds = NULL;
888         fd_set *_writefds = NULL;
889         struct usbi_pollfd *ipollfd;
890         int have_readfds = 0;
891         int have_writefds = 0;
892         struct timeval select_timeout;
893         struct timeval timeout;
894
895         r = libusb_get_next_timeout(&timeout);
896         if (r) {
897                 /* timeout already expired? */
898                 if (!timerisset(&timeout))
899                         return handle_timeouts();
900
901                 /* choose the smallest of next URB timeout or user specified timeout */
902                 if (timercmp(&timeout, tv, <))
903                         select_timeout = timeout;
904                 else
905                         select_timeout = *tv;
906         } else {
907                 select_timeout = *tv;
908         }
909
910         FD_ZERO(&readfds);
911         FD_ZERO(&writefds);
912         pthread_mutex_lock(&pollfds_lock);
913         list_for_each_entry(ipollfd, &pollfds, list) {
914                 struct libusb_pollfd *pollfd = &ipollfd->pollfd;
915                 int fd = pollfd->fd;
916                 if (pollfd->events & POLLIN) {
917                         have_readfds = 1;
918                         FD_SET(fd, &readfds);
919                 }
920                 if (pollfd->events & POLLOUT) {
921                         have_writefds = 1;
922                         FD_SET(fd, &writefds);
923                 }
924                 if (fd > maxfd)
925                         maxfd = fd;
926         }
927         pthread_mutex_unlock(&pollfds_lock);
928
929         if (have_readfds)
930                 _readfds = &readfds;
931         if (have_writefds)
932                 _writefds = &writefds;
933
934         usbi_dbg("select() with timeout in %d.%06ds", select_timeout.tv_sec,
935                 select_timeout.tv_usec);
936         r = select(maxfd + 1, _readfds, _writefds, NULL, &select_timeout);
937         usbi_dbg("select() returned %d with %d.%06ds remaining",
938                 r, select_timeout.tv_sec, select_timeout.tv_usec);
939         if (r == 0) {
940                 *tv = select_timeout;
941                 return handle_timeouts();
942         } else if (r == -1 && errno == EINTR) {
943                 return 0;
944         } else if (r < 0) {
945                 usbi_err("select failed %d err=%d\n", r, errno);
946                 return r;
947         }
948
949         r = usbi_backend->handle_events(_readfds, _writefds);
950         if (r)
951                 usbi_err("backend handle_events failed with error %d", r);
952
953         return r;
954 }
955
956 /** \ingroup poll
957  * Handle any pending events.
958  *
959  * libusb determines "pending events" by checking if any timeouts have expired
960  * and by checking the set of file descriptors for activity.
961  *
962  * If a zero timeval is passed, this function will handle any already-pending
963  * events and then immediately return in non-blocking style.
964  *
965  * If a non-zero timeval is passed and no events are currently pending, this
966  * function will block waiting for events to handle up until the specified
967  * timeout. If an event arrives or a signal is raised, this function will
968  * return early.
969  *
970  * \param tv the maximum time to block waiting for events, or zero for
971  * non-blocking mode
972  * \returns 0 on success
973  * \returns non-zero on error
974  */
975 API_EXPORTED int libusb_handle_events_timeout(struct timeval *tv)
976 {
977         return handle_events(tv);
978 }
979
980 /** \ingroup poll
981  * Handle any pending events in blocking mode with a sensible timeout. This
982  * timeout is currently hardcoded at 2 seconds but we may change this if we
983  * decide other values are more sensible. For finer control over whether this
984  * function is blocking or non-blocking, or the maximum timeout, use
985  * libusb_handle_events_timeout() instead.
986  *
987  * \returns 0 on success
988  * \returns non-zero on error
989  */
990 API_EXPORTED int libusb_handle_events(void)
991 {
992         struct timeval tv;
993         tv.tv_sec = 2;
994         tv.tv_usec = 0;
995         return handle_events(&tv);
996 }
997
998 /** \ingroup poll
999  * Determine the next internal timeout that libusb needs to handle. You only
1000  * need to use this function if you are calling poll() or select() or similar
1001  * on libusb's file descriptors yourself - you do not need to use it if you
1002  * are calling libusb_handle_events() or a variant directly.
1003  * 
1004  * You should call this function in your main loop in order to determine how
1005  * long to wait for select() or poll() to return results. libusb needs to be
1006  * called into at this timeout, so you should use it as an upper bound on
1007  * your select() or poll() call.
1008  *
1009  * When the timeout has expired, call into libusb_handle_events_timeout()
1010  * (perhaps in non-blocking mode) so that libusb can handle the timeout.
1011  *
1012  * This function may return 0 (success) and an all-zero timeval. If this is
1013  * the case, it indicates that libusb has a timeout that has already expired
1014  * so you should call libusb_handle_events_timeout() or similar immediately.
1015  *
1016  * \param tv output location for a relative time against the current
1017  * clock in which libusb must be called into in order to process timeout events
1018  * \returns 0 on success
1019  * \returns non-zero on error
1020  */
1021 API_EXPORTED int libusb_get_next_timeout(struct timeval *tv)
1022 {
1023         struct usbi_transfer *transfer;
1024         struct timespec cur_ts;
1025         struct timeval cur_tv;
1026         struct timeval *next_timeout;
1027         int r;
1028         int found = 0;
1029
1030         pthread_mutex_lock(&flying_transfers_lock);
1031         if (list_empty(&flying_transfers)) {
1032                 pthread_mutex_unlock(&flying_transfers_lock);
1033                 usbi_dbg("no URBs, no timeout!");
1034                 return 0;
1035         }
1036
1037         /* find next transfer which hasn't already been processed as timed out */
1038         list_for_each_entry(transfer, &flying_transfers, list) {
1039                 if (!(transfer->flags & USBI_TRANSFER_TIMED_OUT)) {
1040                         found = 1;
1041                         break;
1042                 }
1043         }
1044         pthread_mutex_unlock(&flying_transfers_lock);
1045
1046         if (!found) {
1047                 usbi_dbg("all URBs have already been processed for timeouts");
1048                 return 0;
1049         }
1050
1051         next_timeout = &transfer->timeout;
1052
1053         /* no timeout for next transfer */
1054         if (!timerisset(next_timeout)) {
1055                 usbi_dbg("no URBs with timeouts, no timeout!");
1056                 return 0;
1057         }
1058
1059         r = clock_gettime(CLOCK_MONOTONIC, &cur_ts);
1060         if (r < 0) {
1061                 usbi_err("failed to read monotonic clock, errno=%d", errno);
1062                 return r;
1063         }
1064         TIMESPEC_TO_TIMEVAL(&cur_tv, &cur_ts);
1065
1066         if (timercmp(&cur_tv, next_timeout, >=)) {
1067                 usbi_dbg("first timeout already expired");
1068                 timerclear(tv);
1069         } else {
1070                 timersub(next_timeout, &cur_tv, tv);
1071                 usbi_dbg("next timeout in %d.%06ds", tv->tv_sec, tv->tv_usec);
1072         }
1073
1074         return 1;
1075 }
1076
1077 /** \ingroup poll
1078  * Register notification functions for file descriptor additions/removals.
1079  * These functions will be invoked for every new or removed file descriptor
1080  * that libusb uses as an event source.
1081  *
1082  * To remove notifiers, pass NULL values for the function pointers.
1083  *
1084  * \param added_cb pointer to function for addition notifications
1085  * \param removed_cb pointer to function for removal notifications
1086  */
1087 API_EXPORTED void libusb_set_pollfd_notifiers(libusb_pollfd_added_cb added_cb,
1088         libusb_pollfd_removed_cb removed_cb)
1089 {
1090         fd_added_cb = added_cb;
1091         fd_removed_cb = removed_cb;
1092 }
1093
1094 int usbi_add_pollfd(int fd, short events)
1095 {
1096         struct usbi_pollfd *ipollfd = malloc(sizeof(*ipollfd));
1097         if (!ipollfd)
1098                 return -ENOMEM;
1099
1100         usbi_dbg("add fd %d events %d", fd, events);
1101         ipollfd->pollfd.fd = fd;
1102         ipollfd->pollfd.events = events;
1103         pthread_mutex_lock(&pollfds_lock);
1104         list_add(&ipollfd->list, &pollfds);
1105         pthread_mutex_unlock(&pollfds_lock);
1106
1107         if (fd_added_cb)
1108                 fd_added_cb(fd, events);
1109         return 0;
1110 }
1111
1112 void usbi_remove_pollfd(int fd)
1113 {
1114         struct usbi_pollfd *ipollfd;
1115         int found = 0;
1116
1117         usbi_dbg("remove fd %d", fd);
1118         pthread_mutex_lock(&pollfds_lock);
1119         list_for_each_entry(ipollfd, &pollfds, list)
1120                 if (ipollfd->pollfd.fd == fd) {
1121                         found = 1;
1122                         break;
1123                 }
1124
1125         if (!found) {
1126                 usbi_err("couldn't find fd %d to remove", fd);
1127                 pthread_mutex_unlock(&pollfds_lock);
1128                 return;
1129         }
1130
1131         list_del(&ipollfd->list);
1132         pthread_mutex_unlock(&pollfds_lock);
1133         free(ipollfd);
1134         if (fd_removed_cb)
1135                 fd_removed_cb(fd);
1136 }
1137
1138 /** \ingroup poll
1139  * Retrieve a list of file descriptors that should be polled by your main loop
1140  * as libusb event sources.
1141  *
1142  * The returned list is NULL-terminated and should be freed with free() when
1143  * done. The actual list contents must not be touched.
1144  *
1145  * \returns a NULL-terminated list of libusb_pollfd structures, or NULL on
1146  * error
1147  */
1148 API_EXPORTED const struct libusb_pollfd **libusb_get_pollfds(void)
1149 {
1150         struct libusb_pollfd **ret = NULL;
1151         struct usbi_pollfd *ipollfd;
1152         size_t i = 0;
1153         size_t cnt = 0;
1154
1155         pthread_mutex_lock(&pollfds_lock);
1156         list_for_each_entry(ipollfd, &pollfds, list)
1157                 cnt++;
1158
1159         ret = calloc(cnt + 1, sizeof(struct libusb_pollfd *));
1160         if (!ret)
1161                 goto out;
1162
1163         list_for_each_entry(ipollfd, &pollfds, list)
1164                 ret[i++] = (struct libusb_pollfd *) ipollfd;
1165         ret[cnt] = NULL;
1166
1167 out:
1168         pthread_mutex_unlock(&pollfds_lock);
1169         return (const struct libusb_pollfd **) ret;
1170 }
1171