1 #pragma once
2 
3 /*
4  * Copyright (C) 2017 The Android Open Source Project
5  *
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *      http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  */
18 
19 #include <linux/usb/functionfs.h>
20 
21 #include <liburing.h>
22 #include <atomic>
23 #include <condition_variable>
24 #include <memory>
25 #include <mutex>
26 #include <vector>
27 
28 #include <android-base/unique_fd.h>
29 #include <asyncio/AsyncIO.h>
30 
31 struct aio_block {
32     std::vector<struct iocb> iocb;
33     std::vector<struct iocb*> iocbs;
34     std::vector<struct io_event> events;
35     aio_context_t ctx;
36     int num_submitted;
37     int fd;
38 };
39 
40 int getMaxPacketSize(int ffs_fd);
41 
42 enum class AIOType { SYNC_IO, AIO, IO_URING };
43 
44 struct usb_handle {
45     std::condition_variable notify;
46     std::mutex lock;
47     bool open_new_connection = true;
48 
49     int (*write)(usb_handle* h, const void* data, int len);
50     int (*read)(usb_handle* h, void* data, int len, bool allow_partial);
51     void (*close)(usb_handle* h);
52 
53     // FunctionFS
54     android::base::unique_fd control;
55     android::base::unique_fd bulk_out;  // "out" from the host's perspective => source for adbd
56     android::base::unique_fd bulk_in;   // "in" from the host's perspective => sink for adbd
57 
58     // Access to these blocks is very not thread safe. Have one block for each of the
59     // read and write threads.
60     struct aio_block read_aiob;
61     struct aio_block write_aiob;
62 
63     io_uring ring;
64     size_t io_size;
65     AIOType aio_type;
66 };
67 
68 std::unique_ptr<usb_handle> create_usb_handle(unsigned num_bufs, unsigned io_size);
69