Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.

před 7 roky
před 7 roky
před 7 roky
před 7 roky
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777
  1. /* USB EHCI Host for Teensy 3.6
  2. * Copyright 2017 Paul Stoffregen (paul@pjrc.com)
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a
  5. * copy of this software and associated documentation files (the
  6. * "Software"), to deal in the Software without restriction, including
  7. * without limitation the rights to use, copy, modify, merge, publish,
  8. * distribute, sublicense, and/or sell copies of the Software, and to
  9. * permit persons to whom the Software is furnished to do so, subject to
  10. * the following conditions:
  11. *
  12. * The above copyright notice and this permission notice shall be included
  13. * in all copies or substantial portions of the Software.
  14. *
  15. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  16. * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  17. * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
  18. * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
  19. * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  20. * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  21. * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  22. */
  23. #ifndef USB_HOST_TEENSY36_
  24. #define USB_HOST_TEENSY36_
  25. #include <stdint.h>
  26. #if !defined(__MK66FX1M0__) && !defined(__IMXRT1052__) && !defined(__IMXRT1062__)
  27. #error "USBHost_t36 only works with Teensy 3.6. Please select it in Tools > Boards"
  28. #endif
  29. #include "utility/imxrt_usbhs.h"
  30. // Dear inquisitive reader, USB is a complex protocol defined with
  31. // very specific terminology. To have any chance of understand this
  32. // source code, you absolutely must have solid knowledge of specific
  33. // USB terms such as host, device, endpoint, pipe, enumeration....
  34. // You really must also have at least a basic knowledge of the
  35. // different USB transfers: control, bulk, interrupt, isochronous.
  36. //
  37. // The USB 2.0 specification explains these in chapter 4 (pages 15
  38. // to 24), and provides more detail in the first part of chapter 5
  39. // (pages 25 to 55). The USB spec is published for free at
  40. // www.usb.org. Here is a convenient link to just the main PDF:
  41. //
  42. // https://www.pjrc.com/teensy/beta/usb20.pdf
  43. //
  44. // This is a huge file, but chapter 4 is short and easy to read.
  45. // If you're not familiar with the USB lingo, please do yourself
  46. // a favor by reading at least chapter 4 to get up to speed on the
  47. // meaning of these important USB concepts and terminology.
  48. //
  49. // If you wish to ask questions (which belong on the forum, not
  50. // github issues) or discuss development of this library, you
  51. // ABSOLUTELY MUST know the basic USB terminology from chapter 4.
  52. // Please repect other people's valuable time & effort by making
  53. // your best effort to read chapter 4 before asking USB questions!
  54. //#define USBHOST_PRINT_DEBUG
  55. #define USBHDBGSerial Serial1
  56. #ifndef USBHDBGSerial
  57. #define USBHDBGSerial Serial
  58. #endif
  59. /************************************************/
  60. /* Data Types */
  61. /************************************************/
  62. // These 6 types are the key to understanding how this USB Host
  63. // library really works.
  64. // USBHost is a static class controlling the hardware.
  65. // All common USB functionality is implemented here.
  66. class USBHost;
  67. // These 3 structures represent the actual USB entities
  68. // USBHost manipulates. One Device_t is created for
  69. // each active USB device. One Pipe_t is create for
  70. // each endpoint. Transfer_t structures are created
  71. // when any data transfer is added to the EHCI work
  72. // queues, and then returned to the free pool after the
  73. // data transfer completes and the driver has processed
  74. // the results.
  75. typedef struct Device_struct Device_t;
  76. typedef struct Pipe_struct Pipe_t;
  77. typedef struct Transfer_struct Transfer_t;
  78. typedef enum { CLAIM_NO=0, CLAIM_REPORT, CLAIM_INTERFACE} hidclaim_t;
  79. // All USB device drivers inherit use these classes.
  80. // Drivers build user-visible functionality on top
  81. // of these classes, which receive USB events from
  82. // USBHost.
  83. class USBDriver;
  84. class USBDriverTimer;
  85. /************************************************/
  86. /* Added Defines */
  87. /************************************************/
  88. // Keyboard special Keys
  89. #define KEYD_UP 0xDA
  90. #define KEYD_DOWN 0xD9
  91. #define KEYD_LEFT 0xD8
  92. #define KEYD_RIGHT 0xD7
  93. #define KEYD_INSERT 0xD1
  94. #define KEYD_DELETE 0xD4
  95. #define KEYD_PAGE_UP 0xD3
  96. #define KEYD_PAGE_DOWN 0xD6
  97. #define KEYD_HOME 0xD2
  98. #define KEYD_END 0xD5
  99. #define KEYD_F1 0xC2
  100. #define KEYD_F2 0xC3
  101. #define KEYD_F3 0xC4
  102. #define KEYD_F4 0xC5
  103. #define KEYD_F5 0xC6
  104. #define KEYD_F6 0xC7
  105. #define KEYD_F7 0xC8
  106. #define KEYD_F8 0xC9
  107. #define KEYD_F9 0xCA
  108. #define KEYD_F10 0xCB
  109. #define KEYD_F11 0xCC
  110. #define KEYD_F12 0xCD
  111. // USBSerial formats - Lets encode format into bits
  112. // Bits: 0-4 - Number of data bits
  113. // Bits: 5-7 - Parity (0=none, 1=odd, 2 = even)
  114. // bits: 8-9 - Stop bits. 0=1, 1=2
  115. #define USBHOST_SERIAL_7E1 0x047
  116. #define USBHOST_SERIAL_7O1 0x027
  117. #define USBHOST_SERIAL_8N1 0x08
  118. #define USBHOST_SERIAL_8N2 0x108
  119. #define USBHOST_SERIAL_8E1 0x048
  120. #define USBHOST_SERIAL_8O1 0x028
  121. /************************************************/
  122. /* Data Structure Definitions */
  123. /************************************************/
  124. // setup_t holds the 8 byte USB SETUP packet data.
  125. // These unions & structs allow convenient access to
  126. // the setup fields.
  127. typedef union {
  128. struct {
  129. union {
  130. struct {
  131. uint8_t bmRequestType;
  132. uint8_t bRequest;
  133. };
  134. uint16_t wRequestAndType;
  135. };
  136. uint16_t wValue;
  137. uint16_t wIndex;
  138. uint16_t wLength;
  139. };
  140. struct {
  141. uint32_t word1;
  142. uint32_t word2;
  143. };
  144. } setup_t;
  145. typedef struct {
  146. enum {STRING_BUF_SIZE=50};
  147. enum {STR_ID_MAN=0, STR_ID_PROD, STR_ID_SERIAL, STR_ID_CNT};
  148. uint8_t iStrings[STR_ID_CNT]; // Index into array for the three indexes
  149. uint8_t buffer[STRING_BUF_SIZE];
  150. } strbuf_t;
  151. #define DEVICE_STRUCT_STRING_BUF_SIZE 50
  152. // Device_t holds all the information about a USB device
  153. struct Device_struct {
  154. Pipe_t *control_pipe;
  155. Pipe_t *data_pipes;
  156. Device_t *next;
  157. USBDriver *drivers;
  158. strbuf_t *strbuf;
  159. uint8_t speed; // 0=12, 1=1.5, 2=480 Mbit/sec
  160. uint8_t address;
  161. uint8_t hub_address;
  162. uint8_t hub_port;
  163. uint8_t enum_state;
  164. uint8_t bDeviceClass;
  165. uint8_t bDeviceSubClass;
  166. uint8_t bDeviceProtocol;
  167. uint8_t bmAttributes;
  168. uint8_t bMaxPower;
  169. uint16_t idVendor;
  170. uint16_t idProduct;
  171. uint16_t LanguageID;
  172. };
  173. // Pipe_t holes all information about each USB endpoint/pipe
  174. // The first half is an EHCI QH structure for the pipe.
  175. struct Pipe_struct {
  176. // Queue Head (QH), EHCI page 46-50
  177. struct { // must be aligned to 32 byte boundary
  178. volatile uint32_t horizontal_link;
  179. volatile uint32_t capabilities[2];
  180. volatile uint32_t current;
  181. volatile uint32_t next;
  182. volatile uint32_t alt_next;
  183. volatile uint32_t token;
  184. volatile uint32_t buffer[5];
  185. } qh;
  186. Device_t *device;
  187. uint8_t type; // 0=control, 1=isochronous, 2=bulk, 3=interrupt
  188. uint8_t direction; // 0=out, 1=in (changes for control, others fixed)
  189. uint8_t start_mask;
  190. uint8_t complete_mask;
  191. Pipe_t *next;
  192. void (*callback_function)(const Transfer_t *);
  193. uint16_t periodic_interval;
  194. uint16_t periodic_offset;
  195. uint16_t bandwidth_interval;
  196. uint16_t bandwidth_offset;
  197. uint16_t bandwidth_shift;
  198. uint8_t bandwidth_stime;
  199. uint8_t bandwidth_ctime;
  200. uint32_t unused1;
  201. uint32_t unused2;
  202. uint32_t unused3;
  203. uint32_t unused4;
  204. uint32_t unused5;
  205. };
  206. // Transfer_t represents a single transaction on the USB bus.
  207. // The first portion is an EHCI qTD structure. Transfer_t are
  208. // allocated as-needed from a memory pool, loaded with pointers
  209. // to the actual data buffers, linked into a followup list,
  210. // and placed on ECHI Queue Heads. When the ECHI interrupt
  211. // occurs, the followup lists are used to find the Transfer_t
  212. // in memory. Callbacks are made, and then the Transfer_t are
  213. // returned to the memory pool.
  214. struct Transfer_struct {
  215. // Queue Element Transfer Descriptor (qTD), EHCI pg 40-45
  216. struct { // must be aligned to 32 byte boundary
  217. volatile uint32_t next;
  218. volatile uint32_t alt_next;
  219. volatile uint32_t token;
  220. volatile uint32_t buffer[5];
  221. } qtd;
  222. // Linked list of queued, not-yet-completed transfers
  223. Transfer_t *next_followup;
  224. Transfer_t *prev_followup;
  225. Pipe_t *pipe;
  226. // Data to be used by callback function. When a group
  227. // of Transfer_t are created, these fields and the
  228. // interrupt-on-complete bit in the qTD token are only
  229. // set in the last Transfer_t of the list.
  230. void *buffer;
  231. uint32_t length;
  232. setup_t setup;
  233. USBDriver *driver;
  234. };
  235. /************************************************/
  236. /* Main USB EHCI Controller */
  237. /************************************************/
  238. class USBHost {
  239. public:
  240. static void begin();
  241. static void Task();
  242. static void countFree(uint32_t &devices, uint32_t &pipes, uint32_t &trans, uint32_t &strs);
  243. protected:
  244. static Pipe_t * new_Pipe(Device_t *dev, uint32_t type, uint32_t endpoint,
  245. uint32_t direction, uint32_t maxlen, uint32_t interval=0);
  246. static bool queue_Control_Transfer(Device_t *dev, setup_t *setup,
  247. void *buf, USBDriver *driver);
  248. static bool queue_Data_Transfer(Pipe_t *pipe, void *buffer,
  249. uint32_t len, USBDriver *driver);
  250. static Device_t * new_Device(uint32_t speed, uint32_t hub_addr, uint32_t hub_port);
  251. static void disconnect_Device(Device_t *dev);
  252. static void enumeration(const Transfer_t *transfer);
  253. static void driver_ready_for_device(USBDriver *driver);
  254. static volatile bool enumeration_busy;
  255. public: // Maybe others may want/need to contribute memory example HID devices may want to add transfers.
  256. static void contribute_Devices(Device_t *devices, uint32_t num);
  257. static void contribute_Pipes(Pipe_t *pipes, uint32_t num);
  258. static void contribute_Transfers(Transfer_t *transfers, uint32_t num);
  259. static void contribute_String_Buffers(strbuf_t *strbuf, uint32_t num);
  260. private:
  261. static void isr();
  262. static void convertStringDescriptorToASCIIString(uint8_t string_index, Device_t *dev, const Transfer_t *transfer);
  263. static void claim_drivers(Device_t *dev);
  264. static uint32_t assign_address(void);
  265. static bool queue_Transfer(Pipe_t *pipe, Transfer_t *transfer);
  266. static void init_Device_Pipe_Transfer_memory(void);
  267. static Device_t * allocate_Device(void);
  268. static void delete_Pipe(Pipe_t *pipe);
  269. static void free_Device(Device_t *q);
  270. static Pipe_t * allocate_Pipe(void);
  271. static void free_Pipe(Pipe_t *q);
  272. static Transfer_t * allocate_Transfer(void);
  273. static void free_Transfer(Transfer_t *q);
  274. static strbuf_t * allocate_string_buffer(void);
  275. static void free_string_buffer(strbuf_t *strbuf);
  276. static bool allocate_interrupt_pipe_bandwidth(Pipe_t *pipe,
  277. uint32_t maxlen, uint32_t interval);
  278. static void add_qh_to_periodic_schedule(Pipe_t *pipe);
  279. static bool followup_Transfer(Transfer_t *transfer);
  280. static void followup_Error(void);
  281. protected:
  282. #ifdef USBHOST_PRINT_DEBUG
  283. static void print_(const Transfer_t *transfer);
  284. static void print_(const Transfer_t *first, const Transfer_t *last);
  285. static void print_token(uint32_t token);
  286. static void print_(const Pipe_t *pipe);
  287. static void print_driverlist(const char *name, const USBDriver *driver);
  288. static void print_qh_list(const Pipe_t *list);
  289. static void print_device_descriptor(const uint8_t *p);
  290. static void print_config_descriptor(const uint8_t *p, uint32_t maxlen);
  291. static void print_string_descriptor(const char *name, const uint8_t *p);
  292. static void print_hexbytes(const void *ptr, uint32_t len);
  293. static void print_(const char *s) { USBHDBGSerial.print(s); }
  294. static void print_(int n) { USBHDBGSerial.print(n); }
  295. static void print_(unsigned int n) { USBHDBGSerial.print(n); }
  296. static void print_(long n) { USBHDBGSerial.print(n); }
  297. static void print_(unsigned long n) { USBHDBGSerial.print(n); }
  298. static void println_(const char *s) { USBHDBGSerial.println(s); }
  299. static void println_(int n) { USBHDBGSerial.println(n); }
  300. static void println_(unsigned int n) { USBHDBGSerial.println(n); }
  301. static void println_(long n) { USBHDBGSerial.println(n); }
  302. static void println_(unsigned long n) { USBHDBGSerial.println(n); }
  303. static void println_() { USBHDBGSerial.println(); }
  304. static void print_(uint32_t n, uint8_t b) { USBHDBGSerial.print(n, b); }
  305. static void println_(uint32_t n, uint8_t b) { USBHDBGSerial.println(n, b); }
  306. static void print_(const char *s, int n, uint8_t b = DEC) {
  307. USBHDBGSerial.print(s); USBHDBGSerial.print(n, b); }
  308. static void print_(const char *s, unsigned int n, uint8_t b = DEC) {
  309. USBHDBGSerial.print(s); USBHDBGSerial.print(n, b); }
  310. static void print_(const char *s, long n, uint8_t b = DEC) {
  311. USBHDBGSerial.print(s); USBHDBGSerial.print(n, b); }
  312. static void print_(const char *s, unsigned long n, uint8_t b = DEC) {
  313. USBHDBGSerial.print(s); USBHDBGSerial.print(n, b); }
  314. static void println_(const char *s, int n, uint8_t b = DEC) {
  315. USBHDBGSerial.print(s); USBHDBGSerial.println(n, b); }
  316. static void println_(const char *s, unsigned int n, uint8_t b = DEC) {
  317. USBHDBGSerial.print(s); USBHDBGSerial.println(n, b); }
  318. static void println_(const char *s, long n, uint8_t b = DEC) {
  319. USBHDBGSerial.print(s); USBHDBGSerial.println(n, b); }
  320. static void println_(const char *s, unsigned long n, uint8_t b = DEC) {
  321. USBHDBGSerial.print(s); USBHDBGSerial.println(n, b); }
  322. friend class USBDriverTimer; // for access to print & println
  323. #else
  324. static void print_(const Transfer_t *transfer) {}
  325. static void print_(const Transfer_t *first, const Transfer_t *last) {}
  326. static void print_token(uint32_t token) {}
  327. static void print_(const Pipe_t *pipe) {}
  328. static void print_driverlist(const char *name, const USBDriver *driver) {}
  329. static void print_qh_list(const Pipe_t *list) {}
  330. static void print_device_descriptor(const uint8_t *p) {}
  331. static void print_config_descriptor(const uint8_t *p, uint32_t maxlen) {}
  332. static void print_string_descriptor(const char *name, const uint8_t *p) {}
  333. static void print_hexbytes(const void *ptr, uint32_t len) {}
  334. static void print_(const char *s) {}
  335. static void print_(int n) {}
  336. static void print_(unsigned int n) {}
  337. static void print_(long n) {}
  338. static void print_(unsigned long n) {}
  339. static void println_(const char *s) {}
  340. static void println_(int n) {}
  341. static void println_(unsigned int n) {}
  342. static void println_(long n) {}
  343. static void println_(unsigned long n) {}
  344. static void println_() {}
  345. static void print_(uint32_t n, uint8_t b) {}
  346. static void println_(uint32_t n, uint8_t b) {}
  347. static void print_(const char *s, int n, uint8_t b = DEC) {}
  348. static void print_(const char *s, unsigned int n, uint8_t b = DEC) {}
  349. static void print_(const char *s, long n, uint8_t b = DEC) {}
  350. static void print_(const char *s, unsigned long n, uint8_t b = DEC) {}
  351. static void println_(const char *s, int n, uint8_t b = DEC) {}
  352. static void println_(const char *s, unsigned int n, uint8_t b = DEC) {}
  353. static void println_(const char *s, long n, uint8_t b = DEC) {}
  354. static void println_(const char *s, unsigned long n, uint8_t b = DEC) {}
  355. #endif
  356. static void mk_setup(setup_t &s, uint32_t bmRequestType, uint32_t bRequest,
  357. uint32_t wValue, uint32_t wIndex, uint32_t wLength) {
  358. s.word1 = bmRequestType | (bRequest << 8) | (wValue << 16);
  359. s.word2 = wIndex | (wLength << 16);
  360. }
  361. };
  362. /************************************************/
  363. /* USB Device Driver Common Base Class */
  364. /************************************************/
  365. // All USB device drivers inherit from this base class.
  366. class USBDriver : public USBHost {
  367. public:
  368. operator bool() {
  369. Device_t *dev = *(Device_t * volatile *)&device;
  370. return dev != nullptr;
  371. }
  372. uint16_t idVendor() {
  373. Device_t *dev = *(Device_t * volatile *)&device;
  374. return (dev != nullptr) ? dev->idVendor : 0;
  375. }
  376. uint16_t idProduct() {
  377. Device_t *dev = *(Device_t * volatile *)&device;
  378. return (dev != nullptr) ? dev->idProduct : 0;
  379. }
  380. const uint8_t *manufacturer() {
  381. Device_t *dev = *(Device_t * volatile *)&device;
  382. if (dev == nullptr || dev->strbuf == nullptr) return nullptr;
  383. return &dev->strbuf->buffer[dev->strbuf->iStrings[strbuf_t::STR_ID_MAN]];
  384. }
  385. const uint8_t *product() {
  386. Device_t *dev = *(Device_t * volatile *)&device;
  387. if (dev == nullptr || dev->strbuf == nullptr) return nullptr;
  388. return &dev->strbuf->buffer[dev->strbuf->iStrings[strbuf_t::STR_ID_PROD]];
  389. }
  390. const uint8_t *serialNumber() {
  391. Device_t *dev = *(Device_t * volatile *)&device;
  392. if (dev == nullptr || dev->strbuf == nullptr) return nullptr;
  393. return &dev->strbuf->buffer[dev->strbuf->iStrings[strbuf_t::STR_ID_SERIAL]];
  394. }
  395. protected:
  396. USBDriver() : next(NULL), device(NULL) {}
  397. // Check if a driver wishes to claim a device or interface or group
  398. // of interfaces within a device. When this function returns true,
  399. // the driver is considered bound or loaded for that device. When
  400. // new devices are detected, enumeration.cpp calls this function on
  401. // all unbound driver objects, to give them an opportunity to bind
  402. // to the new device.
  403. // device has its vid&pid, class/subclass fields initialized
  404. // type is 0 for device level, 1 for interface level, 2 for IAD
  405. // descriptors points to the specific descriptor data
  406. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  407. // When an unknown (not chapter 9) control transfer completes, this
  408. // function is called for all drivers bound to the device. Return
  409. // true means this driver originated this control transfer, so no
  410. // more drivers need to be offered an opportunity to process it.
  411. // This function is optional, only needed if the driver uses control
  412. // transfers and wishes to be notified when they complete.
  413. virtual void control(const Transfer_t *transfer) { }
  414. // When any of the USBDriverTimer objects a driver creates generates
  415. // a timer event, this function is called.
  416. virtual void timer_event(USBDriverTimer *whichTimer) { }
  417. // When the user calls USBHost::Task, this Task function for all
  418. // active drivers is called, so they may update state and/or call
  419. // any attached user callback functions.
  420. virtual void Task() { }
  421. // When a device disconnects from the USB, this function is called.
  422. // The driver must free all resources it allocated and update any
  423. // internal state necessary to deal with the possibility of user
  424. // code continuing to call its API. However, pipes and transfers
  425. // are the handled by lower layers, so device drivers do not free
  426. // pipes they created or cancel transfers they had in progress.
  427. virtual void disconnect();
  428. // Drivers are managed by this single-linked list. All inactive
  429. // (not bound to any device) drivers are linked from
  430. // available_drivers in enumeration.cpp. When bound to a device,
  431. // drivers are linked from that Device_t drivers list.
  432. USBDriver *next;
  433. // The device this object instance is bound to. In words, this
  434. // is the specific device this driver is using. When not bound
  435. // to any device, this must be NULL. Drivers may set this to
  436. // any non-NULL value if they are in a state where they do not
  437. // wish to claim any device or interface (eg, if getting data
  438. // from the HID parser).
  439. Device_t *device;
  440. friend class USBHost;
  441. };
  442. // Device drivers may create these timer objects to schedule a timer call
  443. class USBDriverTimer {
  444. public:
  445. USBDriverTimer() { }
  446. USBDriverTimer(USBDriver *d) : driver(d) { }
  447. void init(USBDriver *d) { driver = d; };
  448. void start(uint32_t microseconds);
  449. void stop();
  450. void *pointer;
  451. uint32_t integer;
  452. uint32_t started_micros; // testing only
  453. private:
  454. USBDriver *driver;
  455. uint32_t usec;
  456. USBDriverTimer *next;
  457. USBDriverTimer *prev;
  458. friend class USBHost;
  459. };
  460. // Device drivers may inherit from this base class, if they wish to receive
  461. // HID input data fully decoded by the USBHIDParser driver
  462. class USBHIDParser;
  463. class USBHIDInput {
  464. public:
  465. operator bool() { return (mydevice != nullptr); }
  466. uint16_t idVendor() { return (mydevice != nullptr) ? mydevice->idVendor : 0; }
  467. uint16_t idProduct() { return (mydevice != nullptr) ? mydevice->idProduct : 0; }
  468. const uint8_t *manufacturer()
  469. { return ((mydevice == nullptr) || (mydevice->strbuf == nullptr)) ? nullptr : &mydevice->strbuf->buffer[mydevice->strbuf->iStrings[strbuf_t::STR_ID_MAN]]; }
  470. const uint8_t *product()
  471. { return ((mydevice == nullptr) || (mydevice->strbuf == nullptr)) ? nullptr : &mydevice->strbuf->buffer[mydevice->strbuf->iStrings[strbuf_t::STR_ID_PROD]]; }
  472. const uint8_t *serialNumber()
  473. { return ((mydevice == nullptr) || (mydevice->strbuf == nullptr)) ? nullptr : &mydevice->strbuf->buffer[mydevice->strbuf->iStrings[strbuf_t::STR_ID_SERIAL]]; }
  474. private:
  475. virtual hidclaim_t claim_collection(USBHIDParser *driver, Device_t *dev, uint32_t topusage);
  476. virtual bool hid_process_in_data(const Transfer_t *transfer) {return false;}
  477. virtual bool hid_process_out_data(const Transfer_t *transfer) {return false;}
  478. virtual void hid_input_begin(uint32_t topusage, uint32_t type, int lgmin, int lgmax);
  479. virtual void hid_input_data(uint32_t usage, int32_t value);
  480. virtual void hid_input_end();
  481. virtual void disconnect_collection(Device_t *dev);
  482. void add_to_list();
  483. USBHIDInput *next = NULL;
  484. friend class USBHIDParser;
  485. protected:
  486. Device_t *mydevice = NULL;
  487. };
  488. // Device drivers may inherit from this base class, if they wish to receive
  489. // HID input like data from Bluetooth HID device.
  490. class BluetoothController;
  491. class BTHIDInput {
  492. public:
  493. operator bool() { return (btdevice != nullptr); }
  494. uint16_t idVendor() { return (btdevice != nullptr) ? btdevice->idVendor : 0; }
  495. uint16_t idProduct() { return (btdevice != nullptr) ? btdevice->idProduct : 0; }
  496. const uint8_t *manufacturer()
  497. { return ((btdevice == nullptr) || (btdevice->strbuf == nullptr)) ? nullptr : &btdevice->strbuf->buffer[btdevice->strbuf->iStrings[strbuf_t::STR_ID_MAN]]; }
  498. const uint8_t *product()
  499. { return remote_name_; }
  500. const uint8_t *serialNumber()
  501. { return ((btdevice == nullptr) || (btdevice->strbuf == nullptr)) ? nullptr : &btdevice->strbuf->buffer[btdevice->strbuf->iStrings[strbuf_t::STR_ID_SERIAL]]; }
  502. private:
  503. virtual bool claim_bluetooth(BluetoothController *driver, uint32_t bluetooth_class, uint8_t *remoteName) {return false;}
  504. virtual bool process_bluetooth_HID_data(const uint8_t *data, uint16_t length) {return false;}
  505. virtual void release_bluetooth() {};
  506. virtual bool remoteNameComplete(const uint8_t *remoteName) {return true;}
  507. virtual void connectionComplete(void) {};
  508. void add_to_list();
  509. BTHIDInput *next = NULL;
  510. friend class BluetoothController;
  511. protected:
  512. enum {SP_NEED_CONNECT=0x1, SP_DONT_NEED_CONNECT=0x02, SP_PS3_IDS=0x4};
  513. enum {REMOTE_NAME_SIZE=32};
  514. uint8_t special_process_required = 0;
  515. Device_t *btdevice = NULL;
  516. uint8_t remote_name_[REMOTE_NAME_SIZE] = {0};
  517. };
  518. /************************************************/
  519. /* USB Device Drivers */
  520. /************************************************/
  521. class USBHub : public USBDriver {
  522. public:
  523. USBHub(USBHost &host) : debouncetimer(this), resettimer(this) { init(); }
  524. USBHub(USBHost *host) : debouncetimer(this), resettimer(this) { init(); }
  525. // Hubs with more more than 7 ports are built from two tiers of hubs
  526. // using 4 or 7 port hub chips. While the USB spec seems to allow
  527. // hubs to have up to 255 ports, in practice all hub chips on the
  528. // market are only 2, 3, 4 or 7 ports.
  529. enum { MAXPORTS = 7 };
  530. typedef uint8_t portbitmask_t;
  531. enum {
  532. PORT_OFF = 0,
  533. PORT_DISCONNECT = 1,
  534. PORT_DEBOUNCE1 = 2,
  535. PORT_DEBOUNCE2 = 3,
  536. PORT_DEBOUNCE3 = 4,
  537. PORT_DEBOUNCE4 = 5,
  538. PORT_DEBOUNCE5 = 6,
  539. PORT_RESET = 7,
  540. PORT_RECOVERY = 8,
  541. PORT_ACTIVE = 9
  542. };
  543. protected:
  544. virtual bool claim(Device_t *dev, int type, const uint8_t *descriptors, uint32_t len);
  545. virtual void control(const Transfer_t *transfer);
  546. virtual void timer_event(USBDriverTimer *whichTimer);
  547. virtual void disconnect();
  548. void init();
  549. bool can_send_control_now();
  550. void send_poweron(uint32_t port);
  551. void send_getstatus(uint32_t port);
  552. void send_clearstatus_connect(uint32_t port);
  553. void send_clearstatus_enable(uint32_t port);
  554. void send_clearstatus_suspend(uint32_t port);
  555. void send_clearstatus_overcurrent(uint32_t port);
  556. void send_clearstatus_reset(uint32_t port);
  557. void send_setreset(uint32_t port);
  558. void send_setinterface();
  559. static void callback(const Transfer_t *transfer);
  560. void status_change(const Transfer_t *transfer);
  561. void new_port_status(uint32_t port, uint32_t status);
  562. void start_debounce_timer(uint32_t port);
  563. void stop_debounce_timer(uint32_t port);
  564. private:
  565. Device_t mydevices[MAXPORTS];
  566. Pipe_t mypipes[2] __attribute__ ((aligned(32)));
  567. Transfer_t mytransfers[4] __attribute__ ((aligned(32)));
  568. strbuf_t mystring_bufs[1];
  569. USBDriverTimer debouncetimer;
  570. USBDriverTimer resettimer;
  571. setup_t setup;
  572. Pipe_t *changepipe;
  573. Device_t *devicelist[MAXPORTS];
  574. uint32_t changebits;
  575. uint32_t statusbits;
  576. uint8_t hub_desc[16];
  577. uint8_t interface_count;
  578. uint8_t interface_number;
  579. uint8_t altsetting;
  580. uint8_t protocol;
  581. uint8_t endpoint;
  582. uint8_t interval;
  583. uint8_t numports;
  584. uint8_t characteristics;
  585. uint8_t powertime;
  586. uint8_t sending_control_transfer;
  587. uint8_t port_doing_reset;
  588. uint8_t port_doing_reset_speed;
  589. uint8_t portstate[MAXPORTS];
  590. portbitmask_t send_pending_poweron;
  591. portbitmask_t send_pending_getstatus;
  592. portbitmask_t send_pending_clearstatus_connect;
  593. portbitmask_t send_pending_clearstatus_enable;
  594. portbitmask_t send_pending_clearstatus_suspend;
  595. portbitmask_t send_pending_clearstatus_overcurrent;
  596. portbitmask_t send_pending_clearstatus_reset;
  597. portbitmask_t send_pending_setreset;
  598. portbitmask_t debounce_in_use;
  599. static volatile bool reset_busy;
  600. };
  601. //--------------------------------------------------------------------------
  602. class USBHIDParser : public USBDriver {
  603. public:
  604. USBHIDParser(USBHost &host) { init(); }
  605. static void driver_ready_for_hid_collection(USBHIDInput *driver);
  606. bool sendPacket(const uint8_t *buffer, int cb=-1);
  607. void setTXBuffers(uint8_t *buffer1, uint8_t *buffer2, uint8_t cb);
  608. bool sendControlPacket(uint32_t bmRequestType, uint32_t bRequest,
  609. uint32_t wValue, uint32_t wIndex, uint32_t wLength, void *buf);
  610. protected:
  611. enum { TOPUSAGE_LIST_LEN = 4 };
  612. enum { USAGE_LIST_LEN = 24 };
  613. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  614. virtual void control(const Transfer_t *transfer);
  615. virtual void disconnect();
  616. static void in_callback(const Transfer_t *transfer);
  617. static void out_callback(const Transfer_t *transfer);
  618. void in_data(const Transfer_t *transfer);
  619. void out_data(const Transfer_t *transfer);
  620. bool check_if_using_report_id();
  621. void parse();
  622. USBHIDInput * find_driver(uint32_t topusage);
  623. void parse(uint16_t type_and_report_id, const uint8_t *data, uint32_t len);
  624. void init();
  625. // Atempt for RAWhid to take over processing of data
  626. //
  627. uint16_t inSize(void) {return in_size;}
  628. uint16_t outSize(void) {return out_size;}
  629. uint8_t activeSendMask(void) {return txstate;}
  630. private:
  631. Pipe_t *in_pipe;
  632. Pipe_t *out_pipe;
  633. static USBHIDInput *available_hid_drivers_list;
  634. //uint32_t topusage_list[TOPUSAGE_LIST_LEN];
  635. USBHIDInput *topusage_drivers[TOPUSAGE_LIST_LEN];
  636. uint16_t in_size;
  637. uint16_t out_size;
  638. setup_t setup;
  639. uint8_t descriptor[512];
  640. uint8_t report[64];
  641. uint16_t descsize;
  642. bool use_report_id;
  643. Pipe_t mypipes[3] __attribute__ ((aligned(32)));
  644. Transfer_t mytransfers[4] __attribute__ ((aligned(32)));
  645. strbuf_t mystring_bufs[1];
  646. uint8_t txstate = 0;
  647. uint8_t *tx1 = nullptr;
  648. uint8_t *tx2 = nullptr;
  649. bool hid_driver_claimed_control_ = false;
  650. };
  651. //--------------------------------------------------------------------------
  652. class KeyboardController : public USBDriver , public USBHIDInput, public BTHIDInput {
  653. public:
  654. typedef union {
  655. struct {
  656. uint8_t numLock : 1;
  657. uint8_t capsLock : 1;
  658. uint8_t scrollLock : 1;
  659. uint8_t compose : 1;
  660. uint8_t kana : 1;
  661. uint8_t reserved : 3;
  662. };
  663. uint8_t byte;
  664. } KBDLeds_t;
  665. public:
  666. KeyboardController(USBHost &host) { init(); }
  667. KeyboardController(USBHost *host) { init(); }
  668. // need their own versions as both USBDriver and USBHIDInput provide
  669. uint16_t idVendor();
  670. uint16_t idProduct();
  671. const uint8_t *manufacturer();
  672. const uint8_t *product();
  673. const uint8_t *serialNumber();
  674. operator bool() { return ((device != nullptr) || (btdevice != nullptr)); }
  675. // Main boot keyboard functions.
  676. uint16_t getKey() { return keyCode; }
  677. uint8_t getModifiers() { return modifiers; }
  678. uint8_t getOemKey() { return keyOEM; }
  679. void attachPress(void (*f)(int unicode)) { keyPressedFunction = f; }
  680. void attachRelease(void (*f)(int unicode)) { keyReleasedFunction = f; }
  681. void LEDS(uint8_t leds);
  682. uint8_t LEDS() {return leds_.byte;}
  683. void updateLEDS(void);
  684. bool numLock() {return leds_.numLock;}
  685. bool capsLock() {return leds_.capsLock;}
  686. bool scrollLock() {return leds_.scrollLock;}
  687. void numLock(bool f);
  688. void capsLock(bool f);
  689. void scrollLock(bool f);
  690. // Added for extras information.
  691. void attachExtrasPress(void (*f)(uint32_t top, uint16_t code)) { extrasKeyPressedFunction = f; }
  692. void attachExtrasRelease(void (*f)(uint32_t top, uint16_t code)) { extrasKeyReleasedFunction = f; }
  693. void forceBootProtocol();
  694. enum {MAX_KEYS_DOWN=4};
  695. protected:
  696. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  697. virtual void control(const Transfer_t *transfer);
  698. virtual void disconnect();
  699. static void callback(const Transfer_t *transfer);
  700. void new_data(const Transfer_t *transfer);
  701. void init();
  702. // Bluetooth data
  703. virtual bool claim_bluetooth(BluetoothController *driver, uint32_t bluetooth_class, uint8_t *remoteName);
  704. virtual bool process_bluetooth_HID_data(const uint8_t *data, uint16_t length);
  705. virtual bool remoteNameComplete(const uint8_t *remoteName);
  706. virtual void release_bluetooth();
  707. protected: // HID functions for extra keyboard data.
  708. virtual hidclaim_t claim_collection(USBHIDParser *driver, Device_t *dev, uint32_t topusage);
  709. virtual void hid_input_begin(uint32_t topusage, uint32_t type, int lgmin, int lgmax);
  710. virtual void hid_input_data(uint32_t usage, int32_t value);
  711. virtual void hid_input_end();
  712. virtual void disconnect_collection(Device_t *dev);
  713. private:
  714. void update();
  715. uint16_t convert_to_unicode(uint32_t mod, uint32_t key);
  716. void key_press(uint32_t mod, uint32_t key);
  717. void key_release(uint32_t mod, uint32_t key);
  718. void (*keyPressedFunction)(int unicode);
  719. void (*keyReleasedFunction)(int unicode);
  720. Pipe_t *datapipe;
  721. setup_t setup;
  722. uint8_t report[8];
  723. uint16_t keyCode;
  724. uint8_t modifiers;
  725. uint8_t keyOEM;
  726. uint8_t prev_report[8];
  727. KBDLeds_t leds_ = {0};
  728. Pipe_t mypipes[2] __attribute__ ((aligned(32)));
  729. Transfer_t mytransfers[4] __attribute__ ((aligned(32)));
  730. strbuf_t mystring_bufs[1];
  731. // Added to process secondary HID data.
  732. void (*extrasKeyPressedFunction)(uint32_t top, uint16_t code);
  733. void (*extrasKeyReleasedFunction)(uint32_t top, uint16_t code);
  734. uint32_t topusage_ = 0; // What top report am I processing?
  735. uint8_t collections_claimed_ = 0;
  736. volatile bool hid_input_begin_ = false;
  737. volatile bool hid_input_data_ = false; // did we receive any valid data with report?
  738. uint8_t count_keys_down_ = 0;
  739. uint16_t keys_down[MAX_KEYS_DOWN];
  740. bool force_boot_protocol; // User or VID/PID said force boot protocol?
  741. bool control_queued;
  742. };
  743. class MouseController : public USBHIDInput, public BTHIDInput {
  744. public:
  745. MouseController(USBHost &host) { init(); }
  746. bool available() { return mouseEvent; }
  747. void mouseDataClear();
  748. uint8_t getButtons() { return buttons; }
  749. int getMouseX() { return mouseX; }
  750. int getMouseY() { return mouseY; }
  751. int getWheel() { return wheel; }
  752. int getWheelH() { return wheelH; }
  753. protected:
  754. virtual hidclaim_t claim_collection(USBHIDParser *driver, Device_t *dev, uint32_t topusage);
  755. virtual void hid_input_begin(uint32_t topusage, uint32_t type, int lgmin, int lgmax);
  756. virtual void hid_input_data(uint32_t usage, int32_t value);
  757. virtual void hid_input_end();
  758. virtual void disconnect_collection(Device_t *dev);
  759. // Bluetooth data
  760. virtual bool claim_bluetooth(BluetoothController *driver, uint32_t bluetooth_class, uint8_t *remoteName);
  761. virtual bool process_bluetooth_HID_data(const uint8_t *data, uint16_t length);
  762. virtual void release_bluetooth();
  763. private:
  764. void init();
  765. BluetoothController *btdriver_ = nullptr;
  766. uint8_t collections_claimed = 0;
  767. volatile bool mouseEvent = false;
  768. volatile bool hid_input_begin_ = false;
  769. uint8_t buttons = 0;
  770. int mouseX = 0;
  771. int mouseY = 0;
  772. int wheel = 0;
  773. int wheelH = 0;
  774. };
  775. //--------------------------------------------------------------------------
  776. class JoystickController : public USBDriver, public USBHIDInput, public BTHIDInput {
  777. public:
  778. JoystickController(USBHost &host) { init(); }
  779. uint16_t idVendor();
  780. uint16_t idProduct();
  781. const uint8_t *manufacturer();
  782. const uint8_t *product();
  783. const uint8_t *serialNumber();
  784. operator bool() { return (((device != nullptr) || (mydevice != nullptr || (btdevice != nullptr))) && connected_); } // override as in both USBDriver and in USBHIDInput
  785. bool available() { return joystickEvent; }
  786. void joystickDataClear();
  787. uint32_t getButtons() { return buttons; }
  788. int getAxis(uint32_t index) { return (index < (sizeof(axis)/sizeof(axis[0]))) ? axis[index] : 0; }
  789. uint64_t axisMask() {return axis_mask_;}
  790. uint64_t axisChangedMask() { return axis_changed_mask_;}
  791. uint64_t axisChangeNotifyMask() {return axis_change_notify_mask_;}
  792. void axisChangeNotifyMask(uint64_t notify_mask) {axis_change_notify_mask_ = notify_mask;}
  793. // set functions functionality depends on underlying joystick.
  794. bool setRumble(uint8_t lValue, uint8_t rValue, uint8_t timeout=0xff);
  795. // setLEDs on PS4(RGB), PS3 simple LED setting (only uses lr)
  796. bool setLEDs(uint8_t lr, uint8_t lg=0, uint8_t lb=0); // sets Leds,
  797. enum { STANDARD_AXIS_COUNT = 10, ADDITIONAL_AXIS_COUNT = 54, TOTAL_AXIS_COUNT = (STANDARD_AXIS_COUNT+ADDITIONAL_AXIS_COUNT) };
  798. typedef enum { UNKNOWN=0, PS3, PS4, XBOXONE, XBOX360} joytype_t;
  799. joytype_t joystickType() {return joystickType_;}
  800. // PS3 pair function. hack, requires that it be connect4ed by USB and we have the address of the Bluetooth dongle...
  801. bool PS3Pair(uint8_t* bdaddr);
  802. protected:
  803. // From USBDriver
  804. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  805. virtual void control(const Transfer_t *transfer);
  806. virtual void disconnect();
  807. // From USBHIDInput
  808. virtual hidclaim_t claim_collection(USBHIDParser *driver, Device_t *dev, uint32_t topusage);
  809. virtual void hid_input_begin(uint32_t topusage, uint32_t type, int lgmin, int lgmax);
  810. virtual void hid_input_data(uint32_t usage, int32_t value);
  811. virtual void hid_input_end();
  812. virtual void disconnect_collection(Device_t *dev);
  813. virtual bool hid_process_out_data(const Transfer_t *transfer);
  814. // Bluetooth data
  815. virtual bool claim_bluetooth(BluetoothController *driver, uint32_t bluetooth_class, uint8_t *remoteName);
  816. virtual bool process_bluetooth_HID_data(const uint8_t *data, uint16_t length);
  817. virtual void release_bluetooth();
  818. virtual bool remoteNameComplete(const uint8_t *remoteName);
  819. virtual void connectionComplete(void);
  820. joytype_t joystickType_ = UNKNOWN;
  821. private:
  822. // Class specific
  823. void init();
  824. USBHIDParser *driver_ = nullptr;
  825. BluetoothController *btdriver_ = nullptr;
  826. joytype_t mapVIDPIDtoJoystickType(uint16_t idVendor, uint16_t idProduct, bool exclude_hid_devices);
  827. bool transmitPS4UserFeedbackMsg();
  828. bool transmitPS3UserFeedbackMsg();
  829. bool mapNameToJoystickType(const uint8_t *remoteName);
  830. bool anychange = false;
  831. volatile bool joystickEvent = false;
  832. uint32_t buttons = 0;
  833. int axis[TOTAL_AXIS_COUNT] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
  834. uint64_t axis_mask_ = 0; // which axis have valid data
  835. uint64_t axis_changed_mask_ = 0;
  836. uint64_t axis_change_notify_mask_ = 0x3ff; // assume the low 10 values only.
  837. uint16_t additional_axis_usage_page_ = 0;
  838. uint16_t additional_axis_usage_start_ = 0;
  839. uint16_t additional_axis_usage_count_ = 0;
  840. // State values to output to Joystick.
  841. uint8_t rumble_lValue_ = 0;
  842. uint8_t rumble_rValue_ = 0;
  843. uint8_t rumble_timeout_ = 0;
  844. uint8_t leds_[3] = {0,0,0};
  845. uint8_t connected_ = 0; // what type of device if any is connected xbox 360...
  846. // Used by HID code
  847. uint8_t collections_claimed = 0;
  848. // Used by USBDriver code
  849. static void rx_callback(const Transfer_t *transfer);
  850. static void tx_callback(const Transfer_t *transfer);
  851. void rx_data(const Transfer_t *transfer);
  852. void tx_data(const Transfer_t *transfer);
  853. Pipe_t mypipes[3] __attribute__ ((aligned(32)));
  854. Transfer_t mytransfers[7] __attribute__ ((aligned(32)));
  855. strbuf_t mystring_bufs[1];
  856. uint8_t rx_ep_ = 0; // remember which end point this object is...
  857. uint16_t rx_size_ = 0;
  858. uint16_t tx_size_ = 0;
  859. Pipe_t *rxpipe_;
  860. Pipe_t *txpipe_;
  861. uint8_t rxbuf_[64]; // receive circular buffer
  862. uint8_t txbuf_[64]; // buffer to use to send commands to joystick
  863. // Mapping table to say which devices we handle
  864. typedef struct {
  865. uint16_t idVendor;
  866. uint16_t idProduct;
  867. joytype_t joyType;
  868. bool hidDevice;
  869. } product_vendor_mapping_t;
  870. static product_vendor_mapping_t pid_vid_mapping[];
  871. };
  872. //--------------------------------------------------------------------------
  873. class MIDIDevice : public USBDriver {
  874. public:
  875. enum { SYSEX_MAX_LEN = 290 };
  876. // Message type names for compatibility with Arduino MIDI library 4.3.1
  877. enum MidiType {
  878. InvalidType = 0x00, // For notifying errors
  879. NoteOff = 0x80, // Note Off
  880. NoteOn = 0x90, // Note On
  881. AfterTouchPoly = 0xA0, // Polyphonic AfterTouch
  882. ControlChange = 0xB0, // Control Change / Channel Mode
  883. ProgramChange = 0xC0, // Program Change
  884. AfterTouchChannel = 0xD0, // Channel (monophonic) AfterTouch
  885. PitchBend = 0xE0, // Pitch Bend
  886. SystemExclusive = 0xF0, // System Exclusive
  887. TimeCodeQuarterFrame = 0xF1, // System Common - MIDI Time Code Quarter Frame
  888. SongPosition = 0xF2, // System Common - Song Position Pointer
  889. SongSelect = 0xF3, // System Common - Song Select
  890. TuneRequest = 0xF6, // System Common - Tune Request
  891. Clock = 0xF8, // System Real Time - Timing Clock
  892. Start = 0xFA, // System Real Time - Start
  893. Continue = 0xFB, // System Real Time - Continue
  894. Stop = 0xFC, // System Real Time - Stop
  895. ActiveSensing = 0xFE, // System Real Time - Active Sensing
  896. SystemReset = 0xFF, // System Real Time - System Reset
  897. };
  898. MIDIDevice(USBHost &host) { init(); }
  899. MIDIDevice(USBHost *host) { init(); }
  900. void sendNoteOff(uint8_t note, uint8_t velocity, uint8_t channel, uint8_t cable=0) {
  901. send(0x80, note, velocity, channel, cable);
  902. }
  903. void sendNoteOn(uint8_t note, uint8_t velocity, uint8_t channel, uint8_t cable=0) {
  904. send(0x90, note, velocity, channel, cable);
  905. }
  906. void sendPolyPressure(uint8_t note, uint8_t pressure, uint8_t channel, uint8_t cable=0) {
  907. send(0xA0, note, pressure, channel, cable);
  908. }
  909. void sendAfterTouchPoly(uint8_t note, uint8_t pressure, uint8_t channel, uint8_t cable=0) {
  910. send(0xA0, note, pressure, channel, cable);
  911. }
  912. void sendControlChange(uint8_t control, uint8_t value, uint8_t channel, uint8_t cable=0) {
  913. send(0xB0, control, value, channel, cable);
  914. }
  915. void sendProgramChange(uint8_t program, uint8_t channel, uint8_t cable=0) {
  916. send(0xC0, program, 0, channel, cable);
  917. }
  918. void sendAfterTouch(uint8_t pressure, uint8_t channel, uint8_t cable=0) {
  919. send(0xD0, pressure, 0, channel, cable);
  920. }
  921. void sendPitchBend(int value, uint8_t channel, uint8_t cable=0) {
  922. if (value < -8192) {
  923. value = -8192;
  924. } else if (value > 8191) {
  925. value = 8191;
  926. }
  927. value += 8192;
  928. send(0xE0, value, value >> 7, channel, cable);
  929. }
  930. void sendSysEx(uint32_t length, const uint8_t *data, bool hasTerm=false, uint8_t cable=0) {
  931. //if (cable >= MIDI_NUM_CABLES) return;
  932. if (hasTerm) {
  933. send_sysex_buffer_has_term(data, length, cable);
  934. } else {
  935. send_sysex_add_term_bytes(data, length, cable);
  936. }
  937. }
  938. void sendRealTime(uint8_t type, uint8_t cable=0) {
  939. switch (type) {
  940. case 0xF8: // Clock
  941. case 0xFA: // Start
  942. case 0xFB: // Continue
  943. case 0xFC: // Stop
  944. case 0xFE: // ActiveSensing
  945. case 0xFF: // SystemReset
  946. send(type, 0, 0, 0, cable);
  947. break;
  948. default: // Invalid Real Time marker
  949. break;
  950. }
  951. }
  952. void sendTimeCodeQuarterFrame(uint8_t type, uint8_t value, uint8_t cable=0) {
  953. send(0xF1, ((type & 0x07) << 4) | (value & 0x0F), 0, 0, cable);
  954. }
  955. void sendSongPosition(uint16_t beats, uint8_t cable=0) {
  956. send(0xF2, beats, beats >> 7, 0, cable);
  957. }
  958. void sendSongSelect(uint8_t song, uint8_t cable=0) {
  959. send(0xF3, song, 0, 0, cable);
  960. }
  961. void sendTuneRequest(uint8_t cable=0) {
  962. send(0xF6, 0, 0, 0, cable);
  963. }
  964. void beginRpn(uint16_t number, uint8_t channel, uint8_t cable=0) {
  965. sendControlChange(101, number >> 7, channel, cable);
  966. sendControlChange(100, number, channel, cable);
  967. }
  968. void sendRpnValue(uint16_t value, uint8_t channel, uint8_t cable=0) {
  969. sendControlChange(6, value >> 7, channel, cable);
  970. sendControlChange(38, value, channel, cable);
  971. }
  972. void sendRpnIncrement(uint8_t amount, uint8_t channel, uint8_t cable=0) {
  973. sendControlChange(96, amount, channel, cable);
  974. }
  975. void sendRpnDecrement(uint8_t amount, uint8_t channel, uint8_t cable=0) {
  976. sendControlChange(97, amount, channel, cable);
  977. }
  978. void endRpn(uint8_t channel, uint8_t cable=0) {
  979. sendControlChange(101, 0x7F, channel, cable);
  980. sendControlChange(100, 0x7F, channel, cable);
  981. }
  982. void beginNrpn(uint16_t number, uint8_t channel, uint8_t cable=0) {
  983. sendControlChange(99, number >> 7, channel, cable);
  984. sendControlChange(98, number, channel, cable);
  985. }
  986. void sendNrpnValue(uint16_t value, uint8_t channel, uint8_t cable=0) {
  987. sendControlChange(6, value >> 7, channel, cable);
  988. sendControlChange(38, value, channel, cable);
  989. }
  990. void sendNrpnIncrement(uint8_t amount, uint8_t channel, uint8_t cable=0) {
  991. sendControlChange(96, amount, channel, cable);
  992. }
  993. void sendNrpnDecrement(uint8_t amount, uint8_t channel, uint8_t cable=0) {
  994. sendControlChange(97, amount, channel, cable);
  995. }
  996. void endNrpn(uint8_t channel, uint8_t cable=0) {
  997. sendControlChange(99, 0x7F, channel, cable);
  998. sendControlChange(98, 0x7F, channel, cable);
  999. }
  1000. void send(uint8_t type, uint8_t data1, uint8_t data2, uint8_t channel, uint8_t cable=0) {
  1001. //if (cable >= MIDI_NUM_CABLES) return;
  1002. if (type < 0xF0) {
  1003. if (type < 0x80) return;
  1004. type &= 0xF0;
  1005. write_packed((type << 8) | (type >> 4) | ((cable & 0x0F) << 4)
  1006. | (((channel - 1) & 0x0F) << 8) | ((data1 & 0x7F) << 16)
  1007. | ((data2 & 0x7F) << 24));
  1008. } else if (type >= 0xF8 || type == 0xF6) {
  1009. write_packed((type << 8) | 0x0F | ((cable & 0x0F) << 4));
  1010. } else if (type == 0xF1 || type == 0xF3) {
  1011. write_packed((type << 8) | 0x02 | ((cable & 0x0F) << 4)
  1012. | ((data1 & 0x7F) << 16));
  1013. } else if (type == 0xF2) {
  1014. write_packed((type << 8) | 0x03 | ((cable & 0x0F) << 4)
  1015. | ((data1 & 0x7F) << 16) | ((data2 & 0x7F) << 24));
  1016. }
  1017. }
  1018. void send_now(void) __attribute__((always_inline)) {
  1019. }
  1020. bool read(uint8_t channel=0);
  1021. uint8_t getType(void) {
  1022. return msg_type;
  1023. };
  1024. uint8_t getCable(void) {
  1025. return msg_cable;
  1026. }
  1027. uint8_t getChannel(void) {
  1028. return msg_channel;
  1029. };
  1030. uint8_t getData1(void) {
  1031. return msg_data1;
  1032. };
  1033. uint8_t getData2(void) {
  1034. return msg_data2;
  1035. };
  1036. uint8_t * getSysExArray(void) {
  1037. return msg_sysex;
  1038. }
  1039. uint16_t getSysExArrayLength(void) {
  1040. return msg_data2 << 8 | msg_data1;
  1041. }
  1042. void setHandleNoteOff(void (*fptr)(uint8_t channel, uint8_t note, uint8_t velocity)) {
  1043. // type: 0x80 NoteOff
  1044. handleNoteOff = fptr;
  1045. }
  1046. void setHandleNoteOn(void (*fptr)(uint8_t channel, uint8_t note, uint8_t velocity)) {
  1047. // type: 0x90 NoteOn
  1048. handleNoteOn = fptr;
  1049. }
  1050. void setHandleVelocityChange(void (*fptr)(uint8_t channel, uint8_t note, uint8_t velocity)) {
  1051. // type: 0xA0 AfterTouchPoly
  1052. handleVelocityChange = fptr;
  1053. }
  1054. void setHandleAfterTouchPoly(void (*fptr)(uint8_t channel, uint8_t note, uint8_t pressure)) {
  1055. // type: 0xA0 AfterTouchPoly
  1056. handleVelocityChange = fptr;
  1057. }
  1058. void setHandleControlChange(void (*fptr)(uint8_t channel, uint8_t control, uint8_t value)) {
  1059. // type: 0xB0 ControlChange
  1060. handleControlChange = fptr;
  1061. }
  1062. void setHandleProgramChange(void (*fptr)(uint8_t channel, uint8_t program)) {
  1063. // type: 0xC0 ProgramChange
  1064. handleProgramChange = fptr;
  1065. }
  1066. void setHandleAfterTouch(void (*fptr)(uint8_t channel, uint8_t pressure)) {
  1067. // type: 0xD0 AfterTouchChannel
  1068. handleAfterTouch = fptr;
  1069. }
  1070. void setHandleAfterTouchChannel(void (*fptr)(uint8_t channel, uint8_t pressure)) {
  1071. // type: 0xD0 AfterTouchChannel
  1072. handleAfterTouch = fptr;
  1073. }
  1074. void setHandlePitchChange(void (*fptr)(uint8_t channel, int pitch)) {
  1075. // type: 0xE0 PitchBend
  1076. handlePitchChange = fptr;
  1077. }
  1078. void setHandleSysEx(void (*fptr)(const uint8_t *data, uint16_t length, bool complete)) {
  1079. // type: 0xF0 SystemExclusive - multiple calls for message bigger than buffer
  1080. handleSysExPartial = (void (*)(const uint8_t *, uint16_t, uint8_t))fptr;
  1081. }
  1082. void setHandleSystemExclusive(void (*fptr)(const uint8_t *data, uint16_t length, bool complete)) {
  1083. // type: 0xF0 SystemExclusive - multiple calls for message bigger than buffer
  1084. handleSysExPartial = (void (*)(const uint8_t *, uint16_t, uint8_t))fptr;
  1085. }
  1086. void setHandleSystemExclusive(void (*fptr)(uint8_t *data, unsigned int size)) {
  1087. // type: 0xF0 SystemExclusive - single call, message larger than buffer is truncated
  1088. handleSysExComplete = fptr;
  1089. }
  1090. void setHandleTimeCodeQuarterFrame(void (*fptr)(uint8_t data)) {
  1091. // type: 0xF1 TimeCodeQuarterFrame
  1092. handleTimeCodeQuarterFrame = fptr;
  1093. }
  1094. void setHandleSongPosition(void (*fptr)(uint16_t beats)) {
  1095. // type: 0xF2 SongPosition
  1096. handleSongPosition = fptr;
  1097. }
  1098. void setHandleSongSelect(void (*fptr)(uint8_t songnumber)) {
  1099. // type: 0xF3 SongSelect
  1100. handleSongSelect = fptr;
  1101. }
  1102. void setHandleTuneRequest(void (*fptr)(void)) {
  1103. // type: 0xF6 TuneRequest
  1104. handleTuneRequest = fptr;
  1105. }
  1106. void setHandleClock(void (*fptr)(void)) {
  1107. // type: 0xF8 Clock
  1108. handleClock = fptr;
  1109. }
  1110. void setHandleStart(void (*fptr)(void)) {
  1111. // type: 0xFA Start
  1112. handleStart = fptr;
  1113. }
  1114. void setHandleContinue(void (*fptr)(void)) {
  1115. // type: 0xFB Continue
  1116. handleContinue = fptr;
  1117. }
  1118. void setHandleStop(void (*fptr)(void)) {
  1119. // type: 0xFC Stop
  1120. handleStop = fptr;
  1121. }
  1122. void setHandleActiveSensing(void (*fptr)(void)) {
  1123. // type: 0xFE ActiveSensing
  1124. handleActiveSensing = fptr;
  1125. }
  1126. void setHandleSystemReset(void (*fptr)(void)) {
  1127. // type: 0xFF SystemReset
  1128. handleSystemReset = fptr;
  1129. }
  1130. void setHandleRealTimeSystem(void (*fptr)(uint8_t realtimebyte)) {
  1131. // type: 0xF8-0xFF - if more specific handler not configured
  1132. handleRealTimeSystem = fptr;
  1133. }
  1134. protected:
  1135. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  1136. virtual void disconnect();
  1137. static void rx_callback(const Transfer_t *transfer);
  1138. static void tx_callback(const Transfer_t *transfer);
  1139. void rx_data(const Transfer_t *transfer);
  1140. void tx_data(const Transfer_t *transfer);
  1141. void init();
  1142. void write_packed(uint32_t data);
  1143. void send_sysex_buffer_has_term(const uint8_t *data, uint32_t length, uint8_t cable);
  1144. void send_sysex_add_term_bytes(const uint8_t *data, uint32_t length, uint8_t cable);
  1145. void sysex_byte(uint8_t b);
  1146. private:
  1147. Pipe_t *rxpipe;
  1148. Pipe_t *txpipe;
  1149. enum { MAX_PACKET_SIZE = 64 };
  1150. enum { RX_QUEUE_SIZE = 80 }; // must be more than MAX_PACKET_SIZE/4
  1151. uint32_t rx_buffer[MAX_PACKET_SIZE/4];
  1152. uint32_t tx_buffer1[MAX_PACKET_SIZE/4];
  1153. uint32_t tx_buffer2[MAX_PACKET_SIZE/4];
  1154. uint16_t rx_size;
  1155. uint16_t tx_size;
  1156. uint32_t rx_queue[RX_QUEUE_SIZE];
  1157. bool rx_packet_queued;
  1158. uint16_t rx_head;
  1159. uint16_t rx_tail;
  1160. volatile uint8_t tx1_count;
  1161. volatile uint8_t tx2_count;
  1162. uint8_t rx_ep;
  1163. uint8_t tx_ep;
  1164. uint8_t rx_ep_type;
  1165. uint8_t tx_ep_type;
  1166. uint8_t msg_cable;
  1167. uint8_t msg_channel;
  1168. uint8_t msg_type;
  1169. uint8_t msg_data1;
  1170. uint8_t msg_data2;
  1171. uint8_t msg_sysex[SYSEX_MAX_LEN];
  1172. uint16_t msg_sysex_len;
  1173. void (*handleNoteOff)(uint8_t ch, uint8_t note, uint8_t vel);
  1174. void (*handleNoteOn)(uint8_t ch, uint8_t note, uint8_t vel);
  1175. void (*handleVelocityChange)(uint8_t ch, uint8_t note, uint8_t vel);
  1176. void (*handleControlChange)(uint8_t ch, uint8_t control, uint8_t value);
  1177. void (*handleProgramChange)(uint8_t ch, uint8_t program);
  1178. void (*handleAfterTouch)(uint8_t ch, uint8_t pressure);
  1179. void (*handlePitchChange)(uint8_t ch, int pitch);
  1180. void (*handleSysExPartial)(const uint8_t *data, uint16_t length, uint8_t complete);
  1181. void (*handleSysExComplete)(uint8_t *data, unsigned int size);
  1182. void (*handleTimeCodeQuarterFrame)(uint8_t data);
  1183. void (*handleSongPosition)(uint16_t beats);
  1184. void (*handleSongSelect)(uint8_t songnumber);
  1185. void (*handleTuneRequest)(void);
  1186. void (*handleClock)(void);
  1187. void (*handleStart)(void);
  1188. void (*handleContinue)(void);
  1189. void (*handleStop)(void);
  1190. void (*handleActiveSensing)(void);
  1191. void (*handleSystemReset)(void);
  1192. void (*handleRealTimeSystem)(uint8_t rtb);
  1193. Pipe_t mypipes[3] __attribute__ ((aligned(32)));
  1194. Transfer_t mytransfers[7] __attribute__ ((aligned(32)));
  1195. strbuf_t mystring_bufs[1];
  1196. };
  1197. //--------------------------------------------------------------------------
  1198. class USBSerial: public USBDriver, public Stream {
  1199. public:
  1200. // FIXME: need different USBSerial, with bigger buffers for 480 Mbit & faster speed
  1201. enum { BUFFER_SIZE = 648 }; // must hold at least 6 max size packets, plus 2 extra bytes
  1202. enum { DEFAULT_WRITE_TIMEOUT = 3500};
  1203. USBSerial(USBHost &host) : txtimer(this) { init(); }
  1204. void begin(uint32_t baud, uint32_t format=USBHOST_SERIAL_8N1);
  1205. void end(void);
  1206. uint32_t writeTimeout() {return write_timeout_;}
  1207. void writeTimeOut(uint32_t write_timeout) {write_timeout_ = write_timeout;} // Will not impact current ones.
  1208. virtual int available(void);
  1209. virtual int peek(void);
  1210. virtual int read(void);
  1211. virtual int availableForWrite();
  1212. virtual size_t write(uint8_t c);
  1213. virtual void flush(void);
  1214. using Print::write;
  1215. protected:
  1216. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  1217. virtual void control(const Transfer_t *transfer);
  1218. virtual void disconnect();
  1219. virtual void timer_event(USBDriverTimer *whichTimer);
  1220. private:
  1221. static void rx_callback(const Transfer_t *transfer);
  1222. static void tx_callback(const Transfer_t *transfer);
  1223. void rx_data(const Transfer_t *transfer);
  1224. void tx_data(const Transfer_t *transfer);
  1225. void rx_queue_packets(uint32_t head, uint32_t tail);
  1226. void init();
  1227. static bool check_rxtx_ep(uint32_t &rxep, uint32_t &txep);
  1228. bool init_buffers(uint32_t rsize, uint32_t tsize);
  1229. void ch341_setBaud(uint8_t byte_index);
  1230. private:
  1231. Pipe_t mypipes[3] __attribute__ ((aligned(32)));
  1232. Transfer_t mytransfers[7] __attribute__ ((aligned(32)));
  1233. strbuf_t mystring_bufs[1];
  1234. USBDriverTimer txtimer;
  1235. uint32_t bigbuffer[(BUFFER_SIZE+3)/4];
  1236. setup_t setup;
  1237. uint8_t setupdata[16]; //
  1238. uint32_t baudrate;
  1239. uint32_t format_;
  1240. uint32_t write_timeout_ = DEFAULT_WRITE_TIMEOUT;
  1241. Pipe_t *rxpipe;
  1242. Pipe_t *txpipe;
  1243. uint8_t *rx1; // location for first incoming packet
  1244. uint8_t *rx2; // location for second incoming packet
  1245. uint8_t *rxbuf; // receive circular buffer
  1246. uint8_t *tx1; // location for first outgoing packet
  1247. uint8_t *tx2; // location for second outgoing packet
  1248. uint8_t *txbuf;
  1249. volatile uint16_t rxhead;// receive head
  1250. volatile uint16_t rxtail;// receive tail
  1251. volatile uint16_t txhead;
  1252. volatile uint16_t txtail;
  1253. uint16_t rxsize;// size of receive circular buffer
  1254. uint16_t txsize;// size of transmit circular buffer
  1255. volatile uint8_t rxstate;// bitmask: which receive packets are queued
  1256. volatile uint8_t txstate;
  1257. uint8_t pending_control;
  1258. uint8_t setup_state; // PL2303 - has several steps... Could use pending control?
  1259. uint8_t pl2303_v1; // Which version do we have
  1260. uint8_t pl2303_v2;
  1261. uint8_t interface;
  1262. bool control_queued; // Is there already a queued control messaged
  1263. typedef enum { UNKNOWN=0, CDCACM, FTDI, PL2303, CH341, CP210X } sertype_t;
  1264. sertype_t sertype;
  1265. typedef struct {
  1266. uint16_t idVendor;
  1267. uint16_t idProduct;
  1268. sertype_t sertype;
  1269. } product_vendor_mapping_t;
  1270. static product_vendor_mapping_t pid_vid_mapping[];
  1271. };
  1272. //--------------------------------------------------------------------------
  1273. class AntPlus: public USBDriver {
  1274. // Please post any AntPlus feedback or contributions on this forum thread:
  1275. // https://forum.pjrc.com/threads/43110-Ant-libarary-and-USB-driver-for-Teensy-3-5-6
  1276. public:
  1277. AntPlus(USBHost &host) : /* txtimer(this),*/ updatetimer(this) { init(); }
  1278. void begin(const uint8_t key=0);
  1279. void onStatusChange(void (*function)(int channel, int status)) {
  1280. user_onStatusChange = function;
  1281. }
  1282. void onDeviceID(void (*function)(int channel, int devId, int devType, int transType)) {
  1283. user_onDeviceID = function;
  1284. }
  1285. void onHeartRateMonitor(void (*f)(int bpm, int msec, int seqNum), uint32_t devid=0) {
  1286. profileSetup_HRM(&ant.dcfg[PROFILE_HRM], devid);
  1287. memset(&hrm, 0, sizeof(hrm));
  1288. user_onHeartRateMonitor = f;
  1289. }
  1290. void onSpeedCadence(void (*f)(float speed, float distance, float rpm), uint32_t devid=0) {
  1291. profileSetup_SPDCAD(&ant.dcfg[PROFILE_SPDCAD], devid);
  1292. memset(&spdcad, 0, sizeof(spdcad));
  1293. user_onSpeedCadence = f;
  1294. }
  1295. void onSpeed(void (*f)(float speed, float distance), uint32_t devid=0) {
  1296. profileSetup_SPEED(&ant.dcfg[PROFILE_SPEED], devid);
  1297. memset(&spd, 0, sizeof(spd));
  1298. user_onSpeed = f;
  1299. }
  1300. void onCadence(void (*f)(float rpm), uint32_t devid=0) {
  1301. profileSetup_CADENCE(&ant.dcfg[PROFILE_CADENCE], devid);
  1302. memset(&cad, 0, sizeof(cad));
  1303. user_onCadence = f;
  1304. }
  1305. void setWheelCircumference(float meters) {
  1306. wheelCircumference = meters * 1000.0f;
  1307. }
  1308. protected:
  1309. virtual void Task();
  1310. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  1311. virtual void disconnect();
  1312. virtual void timer_event(USBDriverTimer *whichTimer);
  1313. private:
  1314. static void rx_callback(const Transfer_t *transfer);
  1315. static void tx_callback(const Transfer_t *transfer);
  1316. void rx_data(const Transfer_t *transfer);
  1317. void tx_data(const Transfer_t *transfer);
  1318. void init();
  1319. size_t write(const void *data, const size_t size);
  1320. int read(void *data, const size_t size);
  1321. void transmit();
  1322. private:
  1323. Pipe_t mypipes[2] __attribute__ ((aligned(32)));
  1324. Transfer_t mytransfers[3] __attribute__ ((aligned(32)));
  1325. strbuf_t mystring_bufs[1];
  1326. //USBDriverTimer txtimer;
  1327. USBDriverTimer updatetimer;
  1328. Pipe_t *rxpipe;
  1329. Pipe_t *txpipe;
  1330. bool first_update;
  1331. uint8_t txbuffer[240];
  1332. uint8_t rxpacket[64];
  1333. volatile uint16_t txhead;
  1334. volatile uint16_t txtail;
  1335. volatile bool txready;
  1336. volatile uint8_t rxlen;
  1337. volatile bool do_polling;
  1338. private:
  1339. enum _eventi {
  1340. EVENTI_MESSAGE = 0,
  1341. EVENTI_CHANNEL,
  1342. EVENTI_TOTAL
  1343. };
  1344. enum _profiles {
  1345. PROFILE_HRM = 0,
  1346. PROFILE_SPDCAD,
  1347. PROFILE_POWER,
  1348. PROFILE_STRIDE,
  1349. PROFILE_SPEED,
  1350. PROFILE_CADENCE,
  1351. PROFILE_TOTAL
  1352. };
  1353. typedef struct {
  1354. uint8_t channel;
  1355. uint8_t RFFreq;
  1356. uint8_t networkNumber;
  1357. uint8_t stub;
  1358. uint8_t searchTimeout;
  1359. uint8_t channelType;
  1360. uint8_t deviceType;
  1361. uint8_t transType;
  1362. uint16_t channelPeriod;
  1363. uint16_t searchWaveform;
  1364. uint32_t deviceNumber; // deviceId
  1365. struct {
  1366. uint8_t chanIdOnce;
  1367. uint8_t keyAccepted;
  1368. uint8_t profileValid;
  1369. uint8_t channelStatus;
  1370. uint8_t channelStatusOld;
  1371. } flags;
  1372. } TDCONFIG;
  1373. struct {
  1374. uint8_t initOnce;
  1375. uint8_t key; // key index
  1376. int iDevice; // index to the antplus we're interested in, if > one found
  1377. TDCONFIG dcfg[PROFILE_TOTAL]; // channel config, we're using one channel per device
  1378. } ant;
  1379. void (*user_onStatusChange)(int channel, int status);
  1380. void (*user_onDeviceID)(int channel, int devId, int devType, int transType);
  1381. void (*user_onHeartRateMonitor)(int beatsPerMinute, int milliseconds, int sequenceNumber);
  1382. void (*user_onSpeedCadence)(float speed, float distance, float cadence);
  1383. void (*user_onSpeed)(float speed, float distance);
  1384. void (*user_onCadence)(float cadence);
  1385. void dispatchPayload(TDCONFIG *cfg, const uint8_t *payload, const int len);
  1386. static const uint8_t *getAntKey(const uint8_t keyIdx);
  1387. static uint8_t calcMsgChecksum (const uint8_t *buffer, const uint8_t len);
  1388. static uint8_t * findStreamSync(uint8_t *stream, const size_t rlen, int *pos);
  1389. static int msgCheckIntegrity(uint8_t *stream, const int len);
  1390. static int msgGetLength(uint8_t *stream);
  1391. int handleMessages(uint8_t *buffer, int tBytes);
  1392. void sendMessageChannelStatus(TDCONFIG *cfg, const uint32_t channelStatus);
  1393. void message_channel(const int chan, const int eventId,
  1394. const uint8_t *payload, const size_t dataLength);
  1395. void message_response(const int chan, const int msgId,
  1396. const uint8_t *payload, const size_t dataLength);
  1397. void message_event(const int channel, const int msgId,
  1398. const uint8_t *payload, const size_t dataLength);
  1399. int ResetSystem();
  1400. int RequestMessage(const int channel, const int message);
  1401. int SetNetworkKey(const int netNumber, const uint8_t *key);
  1402. int SetChannelSearchTimeout(const int channel, const int searchTimeout);
  1403. int SetChannelPeriod(const int channel, const int period);
  1404. int SetChannelRFFreq(const int channel, const int freq);
  1405. int SetSearchWaveform(const int channel, const int wave);
  1406. int OpenChannel(const int channel);
  1407. int CloseChannel(const int channel);
  1408. int AssignChannel(const int channel, const int channelType, const int network);
  1409. int SetChannelId(const int channel, const int deviceNum, const int deviceType,
  1410. const int transmissionType);
  1411. int SendBurstTransferPacket(const int channelSeq, const uint8_t *data);
  1412. int SendBurstTransfer(const int channel, const uint8_t *data, const int nunPackets);
  1413. int SendBroadcastData(const int channel, const uint8_t *data);
  1414. int SendAcknowledgedData(const int channel, const uint8_t *data);
  1415. int SendExtAcknowledgedData(const int channel, const int devNum, const int devType,
  1416. const int TranType, const uint8_t *data);
  1417. int SendExtBroadcastData(const int channel, const int devNum, const int devType,
  1418. const int TranType, const uint8_t *data);
  1419. int SendExtBurstTransferPacket(const int chanSeq, const int devNum,
  1420. const int devType, const int TranType, const uint8_t *data);
  1421. int SendExtBurstTransfer(const int channel, const int devNum, const int devType,
  1422. const int tranType, const uint8_t *data, const int nunPackets);
  1423. static void profileSetup_HRM(TDCONFIG *cfg, const uint32_t deviceId);
  1424. static void profileSetup_SPDCAD(TDCONFIG *cfg, const uint32_t deviceId);
  1425. static void profileSetup_POWER(TDCONFIG *cfg, const uint32_t deviceId);
  1426. static void profileSetup_STRIDE(TDCONFIG *cfg, const uint32_t deviceId);
  1427. static void profileSetup_SPEED(TDCONFIG *cfg, const uint32_t deviceId);
  1428. static void profileSetup_CADENCE(TDCONFIG *cfg, const uint32_t deviceId);
  1429. struct {
  1430. struct {
  1431. uint8_t bpm;
  1432. uint8_t sequence;
  1433. uint16_t time;
  1434. } previous;
  1435. } hrm;
  1436. void payload_HRM(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1437. struct {
  1438. struct {
  1439. uint16_t cadenceTime;
  1440. uint16_t cadenceCt;
  1441. uint16_t speedTime;
  1442. uint16_t speedCt;
  1443. } previous;
  1444. float distance;
  1445. } spdcad;
  1446. void payload_SPDCAD(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1447. /* struct {
  1448. struct {
  1449. uint8_t sequence;
  1450. uint16_t pedalPowerContribution;
  1451. uint8_t pedalPower;
  1452. uint8_t instantCadence;
  1453. uint16_t sumPower;
  1454. uint16_t instantPower;
  1455. } current;
  1456. struct {
  1457. uint16_t stub;
  1458. } previous;
  1459. } pwr; */
  1460. void payload_POWER(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1461. /* struct {
  1462. struct {
  1463. uint16_t speed;
  1464. uint16_t cadence;
  1465. uint8_t strides;
  1466. } current;
  1467. struct {
  1468. uint8_t strides;
  1469. uint16_t speed;
  1470. uint16_t cadence;
  1471. } previous;
  1472. } stride; */
  1473. void payload_STRIDE(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1474. struct {
  1475. struct {
  1476. uint16_t speedTime;
  1477. uint16_t speedCt;
  1478. } previous;
  1479. float distance;
  1480. } spd;
  1481. void payload_SPEED(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1482. struct {
  1483. struct {
  1484. uint16_t cadenceTime;
  1485. uint16_t cadenceCt;
  1486. } previous;
  1487. } cad;
  1488. void payload_CADENCE(TDCONFIG *cfg, const uint8_t *data, const size_t dataLength);
  1489. uint16_t wheelCircumference; // default is WHEEL_CIRCUMFERENCE (2122cm)
  1490. };
  1491. //--------------------------------------------------------------------------
  1492. class RawHIDController : public USBHIDInput {
  1493. public:
  1494. RawHIDController(USBHost &host, uint32_t usage = 0) : fixed_usage_(usage) { init(); }
  1495. uint32_t usage(void) {return usage_;}
  1496. void attachReceive(bool (*f)(uint32_t usage, const uint8_t *data, uint32_t len)) {receiveCB = f;}
  1497. bool sendPacket(const uint8_t *buffer);
  1498. protected:
  1499. virtual hidclaim_t claim_collection(USBHIDParser *driver, Device_t *dev, uint32_t topusage);
  1500. virtual bool hid_process_in_data(const Transfer_t *transfer);
  1501. virtual bool hid_process_out_data(const Transfer_t *transfer);
  1502. virtual void hid_input_begin(uint32_t topusage, uint32_t type, int lgmin, int lgmax);
  1503. virtual void hid_input_data(uint32_t usage, int32_t value);
  1504. virtual void hid_input_end();
  1505. virtual void disconnect_collection(Device_t *dev);
  1506. private:
  1507. void init();
  1508. USBHIDParser *driver_;
  1509. enum { MAX_PACKET_SIZE = 64 };
  1510. bool (*receiveCB)(uint32_t usage, const uint8_t *data, uint32_t len) = nullptr;
  1511. uint8_t collections_claimed = 0;
  1512. //volatile bool hid_input_begin_ = false;
  1513. uint32_t fixed_usage_;
  1514. uint32_t usage_ = 0;
  1515. // See if we can contribute transfers
  1516. Transfer_t mytransfers[2] __attribute__ ((aligned(32)));
  1517. };
  1518. //--------------------------------------------------------------------------
  1519. class BluetoothController: public USBDriver {
  1520. public:
  1521. BluetoothController(USBHost &host, bool pair = false, const char *pin = "0000") : do_pair_device_(pair), pair_pincode_(pin), delayTimer_(this)
  1522. { init(); }
  1523. enum {MAX_ENDPOINTS=4, NUM_SERVICES=4, }; // Max number of Bluetooth services - if you need more than 4 simply increase this number
  1524. enum {BT_CLASS_DEVICE= 0x0804}; // Toy - Robot
  1525. static void driver_ready_for_bluetooth(BTHIDInput *driver);
  1526. const uint8_t* myBDAddr(void) {return my_bdaddr_;}
  1527. // BUGBUG version to allow some of the controlled objects to call?
  1528. enum {CONTROL_SCID=-1};
  1529. void sendL2CapCommand(uint8_t* data, uint8_t nbytes, int channel = (int)0x0001);
  1530. protected:
  1531. virtual bool claim(Device_t *device, int type, const uint8_t *descriptors, uint32_t len);
  1532. virtual void control(const Transfer_t *transfer);
  1533. virtual void disconnect();
  1534. virtual void timer_event(USBDriverTimer *whichTimer);
  1535. BTHIDInput * find_driver(uint32_t device_type, uint8_t *remoteName=nullptr);
  1536. // Hack to allow PS3 to maybe change values
  1537. uint16_t connection_rxid_ = 0;
  1538. uint16_t control_dcid_ = 0x70;
  1539. uint16_t interrupt_dcid_ = 0x71;
  1540. uint16_t interrupt_scid_;
  1541. uint16_t control_scid_;
  1542. private:
  1543. friend class BTHIDInput;
  1544. static void rx_callback(const Transfer_t *transfer);
  1545. static void rx2_callback(const Transfer_t *transfer);
  1546. static void tx_callback(const Transfer_t *transfer);
  1547. void rx_data(const Transfer_t *transfer);
  1548. void rx2_data(const Transfer_t *transfer);
  1549. void tx_data(const Transfer_t *transfer);
  1550. void init();
  1551. // HCI support functions...
  1552. void sendHCICommand(uint16_t hciCommand, uint16_t cParams, const uint8_t* data);
  1553. //void sendHCIReadLocalSupportedFeatures();
  1554. void inline sendHCI_INQUIRY();
  1555. void inline sendHCIInquiryCancel();
  1556. void inline sendHCICreateConnection();
  1557. void inline sendHCIAuthenticationRequested();
  1558. void inline sendHCIAcceptConnectionRequest();
  1559. void inline sendHCILinkKeyNegativeReply();
  1560. void inline sendHCIPinCodeReply();
  1561. void inline sendResetHCI();
  1562. void inline sendHDCWriteClassOfDev();
  1563. void inline sendHCIReadBDAddr();
  1564. void inline sendHCIReadLocalVersionInfo();
  1565. void inline sendHCIWriteScanEnable(uint8_t scan_op);
  1566. void inline sendHCIHCIWriteInquiryMode(uint8_t inquiry_mode);
  1567. void inline sendHCISetEventMask();
  1568. void inline sendHCIRemoteNameRequest();
  1569. void inline sendHCIRemoteVersionInfoRequest();
  1570. void handle_hci_command_complete();
  1571. void handle_hci_command_status();
  1572. void handle_hci_inquiry_result(bool fRSSI=false);
  1573. void handle_hci_extended_inquiry_result();
  1574. void handle_hci_inquiry_complete();
  1575. void handle_hci_incoming_connect();
  1576. void handle_hci_connection_complete();
  1577. void handle_hci_disconnect_complete();
  1578. void handle_hci_authentication_complete();
  1579. void handle_hci_remote_name_complete();
  1580. void handle_hci_remote_version_information_complete();
  1581. void handle_hci_pin_code_request();
  1582. void handle_hci_link_key_notification();
  1583. void handle_hci_link_key_request();
  1584. void queue_next_hci_command();
  1585. void sendl2cap_ConnectionResponse(uint16_t handle, uint8_t rxid, uint16_t dcid, uint16_t scid, uint8_t result);
  1586. void sendl2cap_ConnectionRequest(uint16_t handle, uint8_t rxid, uint16_t scid, uint16_t psm);
  1587. void sendl2cap_ConfigRequest(uint16_t handle, uint8_t rxid, uint16_t dcid);
  1588. void sendl2cap_ConfigResponse(uint16_t handle, uint8_t rxid, uint16_t scid);
  1589. void sendL2CapCommand(uint16_t handle, uint8_t* data, uint8_t nbytes, uint8_t channelLow = 0x01, uint8_t channelHigh = 0x00);
  1590. void process_l2cap_connection_request(uint8_t *data);
  1591. void process_l2cap_connection_response(uint8_t *data);
  1592. void process_l2cap_config_request(uint8_t *data);
  1593. void process_l2cap_config_response(uint8_t *data);
  1594. void process_l2cap_command_reject(uint8_t *data);
  1595. void process_l2cap_disconnect_request(uint8_t *data);
  1596. void setHIDProtocol(uint8_t protocol);
  1597. void handleHIDTHDRData(uint8_t *buffer); // Pass the whole buffer...
  1598. static BTHIDInput *available_bthid_drivers_list;
  1599. setup_t setup;
  1600. Pipe_t mypipes[4] __attribute__ ((aligned(32)));
  1601. Transfer_t mytransfers[7] __attribute__ ((aligned(32)));
  1602. strbuf_t mystring_bufs[2]; // 2 string buffers - one for our device - one for remote device...
  1603. uint16_t pending_control_ = 0;
  1604. uint16_t pending_control_tx_ = 0;
  1605. uint16_t rx_size_ = 0;
  1606. uint16_t rx2_size_ = 0;
  1607. uint16_t tx_size_ = 0;
  1608. Pipe_t *rxpipe_;
  1609. Pipe_t *rx2pipe_;
  1610. Pipe_t *txpipe_;
  1611. uint8_t rxbuf_[256]; // used to receive data from RX, which may come with several packets...
  1612. uint8_t rx_packet_data_remaining=0; // how much data remaining
  1613. uint8_t rx2buf_[64]; // receive buffer from Bulk end point
  1614. uint8_t txbuf_[256]; // buffer to use to send commands to bluetooth
  1615. uint8_t hciVersion; // what version of HCI do we have?
  1616. bool do_pair_device_; // Should we do a pair for a new device?
  1617. const char *pair_pincode_; // What pin code to use for the pairing
  1618. USBDriverTimer delayTimer_;
  1619. uint8_t my_bdaddr_[6]; // The bluetooth dongles Bluetooth address.
  1620. uint8_t features[8]; // remember our local features.
  1621. BTHIDInput * device_driver_ = nullptr;;
  1622. uint8_t device_bdaddr_[6];// remember devices address
  1623. uint8_t device_ps_repetion_mode_ ; // mode
  1624. uint8_t device_clock_offset_[2];
  1625. uint32_t device_class_; // class of device.
  1626. uint16_t device_connection_handle_; // handle to connection
  1627. uint8_t remote_ver_;
  1628. uint16_t remote_man_;
  1629. uint8_t remote_subv_;
  1630. uint8_t connection_complete_ = false; //
  1631. typedef struct {
  1632. uint16_t idVendor;
  1633. uint16_t idProduct;
  1634. } product_vendor_mapping_t;
  1635. static product_vendor_mapping_t pid_vid_mapping[];
  1636. };
  1637. #endif