OpenPuck.ino 82 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182
  1. // OpenPuck.ino -- Steam Controller 2 ("Triton") puck reimplementation for an nRF52840.
  2. //
  3. // This sketch impersonates the Valve puck over USB, maintains puck-style bond slots, speaks the
  4. // reverse-engineered RF protocol to the controller, and can re-enumerate into Steam, Xbox, Switch,
  5. // or lizard modes. The real puck exposes four separate HID control interfaces; interface N owns
  6. // bond slot N, so this sketch does the same. Build with -DCFG_TUD_HID=4.
  7. #include <Adafruit_TinyUSB.h>
  8. #include <Adafruit_LittleFS.h>
  9. #include <InternalFileSystem.h>
  10. // custom class-driver API (usbd_class_driver_t, usbd_edpt_*) for the XInput interface
  11. extern "C" {
  12. #include "device/usbd_pvt.h"
  13. }
  14. // Bounded radio-disable wait: NEVER spin forever. A wedged RADIO peripheral would otherwise hang the whole
  15. // main loop (and USB stops being serviced -> device "dies" until replug). On timeout we bail and continue;
  16. // the next rfConfig re-inits the radio.
  17. #define RWAIT_DISABLED() do{ uint32_t _w=micros(); while(!NRF_RADIO->EVENTS_DISABLED && (uint32_t)(micros()-_w)<3000){} }while(0)
  18. using namespace Adafruit_LittleFS_Namespace;
  19. #if CFG_TUD_HID < 4
  20. #error "build with -DCFG_TUD_HID=4 (extra_flags): up to 4 HID interfaces per mode"
  21. #endif
  22. // ---- cloned puck HID report descriptor (verbatim): mouse(0x40)+keyboard(0x41)+vendor(FF00) with
  23. // the 63-byte FEATURE command reports on report id 1/2. Each of the 4 interfaces uses this. ----
  24. static const uint8_t PUCK_HID_DESC[] = {
  25. 0x05,0x01,0x09,0x02,0xA1,0x01,0x85,0x40,0x09,0x01,0xA1,0x00,0x05,0x09,0x19,0x01,
  26. 0x29,0x02,0x15,0x00,0x25,0x01,0x75,0x01,0x95,0x02,0x81,0x02,0x75,0x06,0x95,0x01,
  27. 0x81,0x01,0x05,0x01,0x09,0x30,0x09,0x31,0x15,0x81,0x25,0x7F,0x75,0x08,0x95,0x02,
  28. 0x81,0x06,0x95,0x01,0x09,0x38,0x81,0x06,0x05,0x0C,0x0A,0x38,0x02,0x95,0x01,0x81,
  29. 0x06,0xC0,0xC0,0x05,0x01,0x09,0x06,0xA1,0x01,0x85,0x41,0x05,0x07,0x19,0xE0,0x29,
  30. 0xE7,0x15,0x00,0x25,0x01,0x75,0x01,0x95,0x08,0x81,0x02,0x81,0x01,0x19,0x00,0x29,
  31. 0x65,0x15,0x00,0x25,0x65,0x75,0x08,0x95,0x06,0x81,0x00,0xC0,0x06,0x00,0xFF,0x09,
  32. 0x01,0xA1,0x01,0x85,0x42,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x35,0x09,0x42,
  33. 0x81,0x02,0x85,0x44,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x05,0x09,0x44,0x81,
  34. 0x02,0x85,0x79,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x01,0x09,0x79,0x81,0x02,
  35. 0x85,0x43,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x0E,0x09,0x43,0x81,0x02,0x85,
  36. 0x7B,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x0C,0x09,0x7B,0x81,0x02,0x85,0x45,
  37. 0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x2D,0x09,0x45,0x81,0x02,0x85,0x80,0x15,
  38. 0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x09,0x09,0x80,0x91,0x02,0x85,0x81,0x15,0x00,
  39. 0x26,0xFF,0x00,0x75,0x08,0x95,0x07,0x09,0x81,0x91,0x02,0x85,0x82,0x15,0x00,0x26,
  40. 0xFF,0x00,0x75,0x08,0x95,0x03,0x09,0x82,0x91,0x02,0x85,0x83,0x15,0x00,0x26,0xFF,
  41. 0x00,0x75,0x08,0x95,0x09,0x09,0x83,0x91,0x02,0x85,0x84,0x15,0x00,0x26,0xFF,0x00,
  42. 0x75,0x08,0x95,0x08,0x09,0x84,0x91,0x02,0x85,0x85,0x15,0x00,0x26,0xFF,0x00,0x75,
  43. 0x08,0x95,0x03,0x09,0x85,0x91,0x02,0x85,0x86,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,
  44. 0x95,0x03,0x09,0x86,0x91,0x02,0x85,0x87,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,
  45. 0x3F,0x09,0x87,0x91,0x02,0x85,0x89,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x3F,
  46. 0x09,0x89,0x91,0x02,0x85,0x88,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x3F,0x09,
  47. 0x88,0x91,0x02,0x85,0x01,0x95,0x3F,0x09,0x01,0xB1,0x02,0x85,0x02,0x95,0x3F,0x09,
  48. 0x01,0xB1,0x02,0xC0
  49. };
  50. #define NSLOT 4
  51. Adafruit_USBD_HID hid[NSLOT];
  52. // XBOX MODE = a MUTUALLY-EXCLUSIVE USB presentation: the device enumerates as EITHER the 4 puck slot
  53. // interfaces (Steam mode) OR a standard gamepad + a mouse (Xbox mode) — never both. The mode is persisted to
  54. // flash and a button combo (back4+X -> xbox, back4+A -> steam) saves it and reboots, so each boot enumerates
  55. // the right interface set. RF (beacon/poll/bonds) is identical in both modes; only the USB side differs.
  56. static const uint8_t MOUSE_HID_DESC[] = { TUD_HID_REPORT_DESC_MOUSE() };
  57. Adafruit_USBD_HID g_mouse; // Xbox/Lizard-mode mouse interface (right trackpad)
  58. static const uint8_t KEYBOARD_HID_DESC[] = { TUD_HID_REPORT_DESC_KEYBOARD() };
  59. Adafruit_USBD_HID g_keyboard; // Lizard-mode keyboard interface (buttons -> keys)
  60. static uint8_t g_usbMode = 0; // 0=STEAM(puck) 1=XBOX(xinput) 2=SWITCH 3=LIZARD(kbd+mouse); loaded from flash at boot
  61. static bool g_xbox = false; // == (g_usbMode != 0); kept for the existing puck-vs-other checks (true for xbox/switch/lizard)
  62. // Mode persistence policy: by DEFAULT every fresh power-on/reconnect lands in STEAM mode (0). An explicit
  63. // mode switch still works for the session via a ONE-SHOT bootMode (honored once, then cleared, so the next
  64. // cold boot reverts to Steam). The WebUI "persist last mode" toggle (g_persistMode) instead remembers the
  65. // last selected mode across reboots.
  66. static bool g_persistMode = false; // false (default) = always boot Steam; true = boot into last mode
  67. static uint8_t g_bootMode = 0xFF; // one-shot: boot into this mode once then clear (used when !persistMode + explicit switch)
  68. volatile uint8_t g_rumble = 0; // Xbox-mode rumble strength from the host's XInput OUT packet (0=off); drives the haptic relay
  69. volatile unsigned long g_rumbleMs = 0; // millis of last rumble OUT packet (declared here — used by xi_xfer above the watchdog block); stuck-rumble watchdog
  70. // persisted, runtime-tunable config (Cfg struct + load/saveCfg below; CDC M/F/D/W/B + WebUSB set these):
  71. static int g_mDiv = 64, g_mFric = 94, g_padSmooth = 35; // xbox mouse sens/friction%, steam pad smoothing%
  72. static int g_lizDiv = 320; // LIZARD right-pad mouse sensitivity divisor (higher=slower; desktop cursor wants way less than Xbox aim). CDC 'n'.
  73. static int g_lizFric = 88; // LIZARD right-pad mouse friction% (lower=less glide/crisper, higher=more coast). Tuned by feel (divisor 320 / friction 88). CDC 'm'.
  74. static uint8_t g_abSwap = 0; // 1 = swap A/B and X/Y (Nintendo face-button layout)
  75. static uint8_t g_back[4] = {5,6,7,8}; // back paddles L4,R4,L5,R5 -> button codes (see codeToXB; 5=LB 6=RB 7=L3 8=R3)
  76. static uint32_t g_pollUs = 800; // RF poll cadence (us). LOCKED default: 800us is the measured peak (~706 polls/s -> ~400 new/s with PID cycling; below 800 = RX-turnaround-limited, above 800 declines). NOT persisted — WebUSB/CDC 'U' tune it for the session only, reverts to 800 on reboot.
  77. // but too-aggressive over-polls -> reply collapse during trackpad. Live-tunable.
  78. static uint32_t g_rxWin = 1200; // poll RX-window (us): max wait for the reply. This (not pollUs) caps poll rate
  79. // (~1e6/rxWin). Shorter=more polls/s but may miss the controller's DELAYED replies
  80. // during trackpad. Tunable 'r<us>'. The reply normally arrives fast; 1200 is the worst-case wait.
  81. static uint8_t g_fwdNewOnly = 1; // Steam mode: forward report 0x45 only when seq advanced (dedupe like the real puck).
  82. // Sending stale repeats makes Steam's trackpad smoothing stair-step. Toggle to A/B.
  83. // QoS adaptive channel hopping: track link badness (crcfail+noRx) on the current channel; when it degrades,
  84. // hop to the next clean candidate (controller follows via rfHopTo). Conservative hysteresis to avoid blip-storms.
  85. static uint8_t g_qos = 0; // 0=off (static g_sessCh), 1=auto-hop on degradation
  86. static const uint8_t g_hopCand[] = {18,46,76,22,68}; // clean, spread channels (from the puck's RSSI/PER scan)
  87. static uint8_t g_hopIdx = 0;
  88. static volatile uint16_t g_qosBad = 0; // crcfail+noRx accumulated since last QoS check
  89. static unsigned long g_qosCheckMs = 0, g_qosLastHopMs = 0;
  90. // WebUSB config channel — a dedicated vendor interface present in every mode so the browser panel
  91. // can read/set the tunables above. Protocol + poll in webusbPoll() below.
  92. // NOTE: no setLandingPage() — that would make Chrome pop a "device detected, open <url>" notification
  93. // every plug-in. The interface works without it; the user just opens the published WebUSB page themselves.
  94. Adafruit_USBD_WebUSB usb_web;
  95. // ===================== XInput (Xbox 360 wired) device — Xbox-mode gamepad =====================
  96. // Real Xbox 360 pads are NOT HID: vendor interface class 0xFF/sub 0x5D/proto 0x01 + 20-byte XInput report.
  97. // Custom TinyUSB class driver (registered via usbd_app_driver_get_cb) + an Adafruit_USBD_Interface subclass
  98. // that emits the interface+0x21-blob+2 endpoints. Steam/SDL bind it as "Xbox 360 Controller" by VID/PID +
  99. // the FF/5D/01 interface. (Verified against Adafruit nRF52 1.7.0 / TinyUSB 0.18.) XInput button bits:
  100. enum { XB_DUP=0x0001,XB_DDOWN=0x0002,XB_DLEFT=0x0004,XB_DRIGHT=0x0008, XB_START=0x0010,XB_BACK=0x0020,
  101. XB_L3=0x0040,XB_R3=0x0080, XB_LB=0x0100,XB_RB=0x0200,XB_GUIDE=0x0400, XB_A=0x1000,XB_B=0x2000,XB_X=0x4000,XB_Y=0x8000 };
  102. #define XINPUT_DESC_LEN (9 + 17 + 7 + 7) // interface(9)+vendor0x21(17)+IN ep(7)+OUT ep(7) = 40
  103. static uint8_t g_xiItf=0xFF, g_xiEpIn=0, g_xiEpOut=0;
  104. static uint8_t g_xiInBuf[32], g_xiOutBuf[32];
  105. static void xi_init(void){}
  106. static bool xi_deinit(void){return true;}
  107. static void xi_reset(uint8_t rhport){(void)rhport; g_xiEpIn=g_xiEpOut=0;}
  108. static uint16_t xi_open(uint8_t rhport, tusb_desc_interface_t const* itf, uint16_t max_len){
  109. if(!(itf->bInterfaceClass==0xFF && itf->bInterfaceSubClass==0x5D && itf->bInterfaceProtocol==0x01)) return 0;
  110. g_xiItf=itf->bInterfaceNumber;
  111. uint8_t const* p=(uint8_t const*)itf; uint8_t const* end=p+max_len; uint16_t used=itf->bLength; p+=itf->bLength;
  112. uint8_t opened=0;
  113. while(p<end && opened<itf->bNumEndpoints){
  114. uint8_t blen=p[0], btype=p[1];
  115. if(btype==TUSB_DESC_ENDPOINT){ tusb_desc_endpoint_t const* ep=(tusb_desc_endpoint_t const*)p;
  116. usbd_edpt_open(rhport,ep);
  117. if(tu_edpt_dir(ep->bEndpointAddress)==TUSB_DIR_IN) g_xiEpIn=ep->bEndpointAddress; else g_xiEpOut=ep->bEndpointAddress;
  118. opened++; }
  119. used+=blen; p+=blen;
  120. }
  121. if(g_xiEpOut) usbd_edpt_xfer(rhport,g_xiEpOut,g_xiOutBuf,sizeof g_xiOutBuf); // arm OUT (rumble/LED)
  122. return used;
  123. }
  124. static bool xi_ctrl(uint8_t rhport,uint8_t stage,tusb_control_request_t const* req){(void)rhport;(void)req; return stage!=CONTROL_STAGE_SETUP;}
  125. static bool xi_xfer(uint8_t rhport,uint8_t ep,xfer_result_t res,uint32_t n){(void)res;
  126. if(ep==g_xiEpOut){
  127. // XInput rumble packet: [00][08][00][bigMotor][smallMotor][00][00][00]; LED pkt is [01][03][led]
  128. if(n>=5 && g_xiOutBuf[0]==0x00 && g_xiOutBuf[1]==0x08){
  129. uint8_t big=g_xiOutBuf[3], sml=g_xiOutBuf[4]; g_rumble = big>sml?big:sml; g_rumbleMs = millis(); // stamp for the stuck-rumble watchdog
  130. }
  131. usbd_edpt_xfer(rhport,g_xiEpOut,g_xiOutBuf,sizeof g_xiOutBuf); // re-arm OUT
  132. }
  133. return true; }
  134. static const usbd_class_driver_t g_xiDriver = {
  135. #if CFG_TUSB_DEBUG>=2
  136. .name="XINPUT",
  137. #endif
  138. .init=xi_init, .deinit=xi_deinit, .reset=xi_reset, .open=xi_open,
  139. .control_xfer_cb=xi_ctrl, .xfer_cb=xi_xfer, .sof=NULL };
  140. extern "C" const usbd_class_driver_t* usbd_app_driver_get_cb(uint8_t* count){ *count=1; return &g_xiDriver; }
  141. class Adafruit_USBD_XInput : public Adafruit_USBD_Interface {
  142. public:
  143. uint16_t getInterfaceDescriptor(uint8_t, uint8_t* buf, uint16_t bufsize) override {
  144. if(!buf) return XINPUT_DESC_LEN; if(bufsize<XINPUT_DESC_LEN) return 0;
  145. uint8_t itfnum=TinyUSBDevice.allocInterface(1);
  146. uint8_t epin=TinyUSBDevice.allocEndpoint(TUSB_DIR_IN), epout=TinyUSBDevice.allocEndpoint(TUSB_DIR_OUT);
  147. const uint8_t t[XINPUT_DESC_LEN]={
  148. 9,TUSB_DESC_INTERFACE,itfnum,0x00,0x02,0xFF,0x5D,0x01,_strid,
  149. 0x11,0x21,0x00,0x01,0x01,0x25,epin,0x14,0x00,0x00,0x00,0x00,0x13,epout,0x08,0x00,0x00,
  150. 7,TUSB_DESC_ENDPOINT,epin, TUSB_XFER_INTERRUPT,U16_TO_U8S_LE(0x20),1, // bInterval 1ms (1000Hz) so the RF rate is the only limit
  151. 7,TUSB_DESC_ENDPOINT,epout,TUSB_XFER_INTERRUPT,U16_TO_U8S_LE(0x20),8 };
  152. memcpy(buf,t,XINPUT_DESC_LEN); return XINPUT_DESC_LEN;
  153. }
  154. bool begin(){ return TinyUSBDevice.addInterface(*this); }
  155. };
  156. Adafruit_USBD_XInput g_xinput;
  157. static void xinputSend(uint16_t buttons,uint8_t lt,uint8_t rt,int16_t lx,int16_t ly,int16_t rx,int16_t ry){
  158. if(!tud_mounted() || g_xiEpIn==0 || usbd_edpt_busy(0,g_xiEpIn)) return;
  159. uint8_t* r=g_xiInBuf;
  160. r[0]=0x00; r[1]=0x14; r[2]=buttons&0xFF; r[3]=buttons>>8; r[4]=lt; r[5]=rt;
  161. r[6]=lx&0xFF; r[7]=lx>>8; r[8]=ly&0xFF; r[9]=ly>>8; r[10]=rx&0xFF; r[11]=rx>>8; r[12]=ry&0xFF; r[13]=ry>>8;
  162. memset(r+14,0,6);
  163. if(usbd_edpt_claim(0,g_xiEpIn)){ if(!usbd_edpt_xfer(0,g_xiEpIn,r,20)) usbd_edpt_release(0,g_xiEpIn); }
  164. }
  165. // ---- identity: a UNIQUE serial from the FICR DEVICEID (never clashes with a real puck) ----
  166. static char g_unit[16]; // "FXB99602xxxxx"
  167. static char g_board[16]; // "MXB99602xxxxx"
  168. static const uint8_t ATTR83[] = { // 0x83 attributes (product 0x1304 = puck)
  169. 0x01,0x04,0x13,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x0A,0xF2,0xF9,0xD2,0x68,
  170. 0x04,0x53,0xD0,0x18,0x6A,0x09,0x47,0x00,0x00,0x00 };
  171. // ---- per-slot state (record = [8 uuid][16 serial]) + per-interface response staging ----
  172. struct Slot { uint8_t rec[24]; bool used; uint8_t resp[63]; uint16_t resp_len; };
  173. static Slot g_slot[NSLOT];
  174. // connection tracking (defined here so the USB feature handler can report B4 connection state)
  175. static int g_connSlot = -1; // bonded slot being polled (== the USB-connected slot)
  176. static unsigned long g_connReplyMs = 0; // millis of last RF reply (link-alive timestamp)
  177. // host->controller relay: Steam writes feature report id 0x01 (raw passthrough) carrying a controller
  178. // feature report [report-id][len][payload...] (e.g. 87 03 09 00 00 = settings: lizard-mode off). The real
  179. // dongle forwards it to the controller as a SET sub-TLV in the E3 poll; OpenPuck does the same.
  180. static volatile bool g_relayPend = false;
  181. static uint8_t g_relayBuf[24];
  182. static volatile uint8_t g_relayN = 0;
  183. static uint8_t g_relayOp = 0xE3; // relay frame opcode (E3 poll)
  184. static uint8_t g_relaySub = 0x05; // relay sub-TLV type byte = SET (confirmed on hardware: E3 + [len][05][rid][payload] makes the controller act on it — haptics buzz + lizard-off lands). GET is 01.
  185. static volatile uint8_t g_testHaptic = 0; // 't<n>' injects n test haptics (output 0x82 [01 01 F7]) for the buzz hunt
  186. // Haptic watchdog — stops the "stuck high-freq buzz" when the host's stop command is lost over RF.
  187. // Xbox rumble is a persistent LEVEL with no keepalive: a dropped stop packet leaves g_rumble set and the
  188. // relay re-buzzes at 40Hz forever (until a host event like the Steam button clears it). Steam glide haptics
  189. // can likewise latch on the controller if Steam's stop relay is lost. We auto-release on host silence.
  190. static unsigned long g_haptic82Ms = 0; // millis of last 0x82 haptic OUTPUT relayed (Steam mode)
  191. static bool g_haptic82On = false; // a non-zero 0x82 haptic is currently active (awaiting host stop)
  192. #define RUMBLE_STUCK_MS 2500u // Xbox: release a held rumble if no OUT packet refreshes it for this long (covers a lost stop without cutting normal short rumbles)
  193. #define HAPTIC_QUIET_MS 300u // Steam: if 0x82 haptics go silent this long after being active, send an explicit off (glide stream is ~25ms-spaced, so 300ms silence = really stopped)
  194. // Auto-lizard: Steam, while running, re-sends settings report 0x87 id 0x09 (lizard-off) every ~3s as a
  195. // heartbeat (captured on hardware). While that heartbeat is alive we forward the gamepad report 0x45 to
  196. // Steam; when it stops (Steam closed) we fall back to lizard — mouse(0x40)+keyboard(0x41) on the SAME puck
  197. // interface, exactly like the real controller. g_steamAliveMs=0 at boot => lizard active until Steam appears.
  198. static unsigned long g_steamAliveMs = 0; // millis of last 0x87 settings write from Steam (its lizard-off heartbeat)
  199. #define LIZARD_WD_MS 7000u // Steam-mode lizard kicks in this long after the 0x87 heartbeat stops (>2x the 3s cadence, so a single missed beat won't trip it mid-session)
  200. #define BOND_FILE "/bonds.bin"
  201. static volatile bool g_dirty = false;
  202. static bool g_pairing = false;
  203. static bool recEmpty(const uint8_t *r) { for (int i = 0; i < 24; i++) if (r[i]) return false; return true; }
  204. static void genSerial() {
  205. uint32_t id = NRF_FICR->DEVICEID[0] ^ NRF_FICR->DEVICEID[1];
  206. snprintf(g_unit, sizeof g_unit, "FXB99602%05lX", (unsigned long)(id & 0xFFFFF));
  207. snprintf(g_board, sizeof g_board, "MXB99602%05lX", (unsigned long)(id & 0xFFFFF));
  208. }
  209. static void saveBonds() {
  210. InternalFS.remove(BOND_FILE);
  211. File f(InternalFS);
  212. if (f.open(BOND_FILE, FILE_O_WRITE)) {
  213. for (int i = 0; i < NSLOT; i++) { uint8_t u = g_slot[i].used ? 1 : 0; f.write(&u, 1); f.write(g_slot[i].rec, 24); }
  214. f.close();
  215. }
  216. }
  217. #define CFG_FILE "/cfg.bin"
  218. #define CFG_MAGIC 0xC6 // bumped (added persistMode+bootMode): old cfg ignored -> clean defaults on first boot
  219. struct Cfg { uint8_t magic, mode, mDiv, mFric, padSmooth, abSwap, back[4], pollU100, persistMode, bootMode; };
  220. static void saveCfg(){
  221. Cfg c={CFG_MAGIC,g_usbMode,(uint8_t)g_mDiv,(uint8_t)g_mFric,(uint8_t)g_padSmooth,g_abSwap,
  222. {g_back[0],g_back[1],g_back[2],g_back[3]},(uint8_t)(g_pollUs/100),(uint8_t)(g_persistMode?1:0),g_bootMode};
  223. InternalFS.remove(CFG_FILE); File f(InternalFS);
  224. if(f.open(CFG_FILE,FILE_O_WRITE)){ f.write((uint8_t*)&c,sizeof c); f.close(); }
  225. }
  226. static void loadCfg(){
  227. Cfg c; File f(InternalFS); bool consume=false;
  228. if(f.open(CFG_FILE,FILE_O_READ)){
  229. if(f.read((uint8_t*)&c,sizeof c)==(int)sizeof c && c.magic==CFG_MAGIC){
  230. g_mDiv=c.mDiv?c.mDiv:64; g_mFric=c.mFric; g_padSmooth=c.padSmooth?c.padSmooth:35;
  231. g_abSwap=c.abSwap; for(int i=0;i<4;i++) g_back[i]=c.back[i];
  232. g_persistMode = c.persistMode?true:false;
  233. // boot-mode policy: a one-shot bootMode (set by an explicit switch when !persist) wins once and is then
  234. // cleared; otherwise persist->last mode, else->Steam. (poll rate stays the locked 800us default — never restored.)
  235. if(c.bootMode!=0xFF){ g_usbMode=(c.bootMode<=3)?c.bootMode:0; consume=true; }
  236. else g_usbMode = g_persistMode ? ((c.mode<=3)?c.mode:0) : 0;
  237. }
  238. f.close();
  239. }
  240. if(consume){ g_bootMode=0xFF; saveCfg(); } // clear the one-shot so the NEXT cold boot reverts to the default/persist policy
  241. }
  242. // Mode switch (chord / WebUI): persist mode if the toggle is on, else arm a one-shot so this reboot lands in
  243. // the new mode but the next cold boot returns to Steam. Either way saveCfg + caller reboots.
  244. static void saveMode(uint8_t m){
  245. if(g_persistMode){ g_usbMode=m; g_bootMode=0xFF; }
  246. else { g_bootMode=m; }
  247. saveCfg();
  248. }
  249. static void loadBonds() {
  250. File f(InternalFS);
  251. if (f.open(BOND_FILE, FILE_O_READ))
  252. for (int i = 0; i < NSLOT; i++) { uint8_t u = 0; if (f.read(&u, 1) == 1) { g_slot[i].used = u; f.read(g_slot[i].rec, 24); } }
  253. f.close();
  254. }
  255. // ---- command channel; `slot` is the interface index (interface N == bond slot N) ----
  256. static void handleSet(int slot, uint8_t rid, hid_report_type_t type, uint8_t const *b, uint16_t n) {
  257. if (type == HID_REPORT_TYPE_OUTPUT) { // Steam haptic/LED OUTPUT reports 0x80-0x89 -> relay to controller
  258. if (rid >= 0x80 && rid <= 0x89 && n >= 1) { // wrap as a SET sub-TLV like the report-01 path
  259. uint8_t m = n > (uint16_t)(sizeof g_relayBuf - 2) ? (sizeof g_relayBuf - 2) : n;
  260. g_relayBuf[0] = rid; g_relayBuf[1] = m; memcpy(g_relayBuf + 2, b, m);
  261. g_relayN = m + 2; g_relayPend = true;
  262. if (rid == 0x82) { g_haptic82Ms = millis(); g_haptic82On = (n >= 3 && b[2] != 0); } // track on/off for the stuck-haptic watchdog
  263. }
  264. if (Serial.availableForWrite() > 80) { // log so we can see what Steam actually sends (e.g. glide haptics)
  265. Serial.printf("# OUT if%d rid=%02X n=%u:", slot, rid, n);
  266. for (uint16_t i = 0; i < n && i < 14; i++) Serial.printf(" %02X", b[i]);
  267. Serial.println();
  268. }
  269. return;
  270. }
  271. if (type != HID_REPORT_TYPE_FEATURE || n < 1) return;
  272. Slot &S = g_slot[slot];
  273. uint8_t cmd = b[0], len = (n > 1) ? b[1] : 0;
  274. const uint8_t *pl = b + 2; uint16_t pln = (n >= 2) ? n - 2 : 0;
  275. if (cmd == 0x87) g_steamAliveMs = millis(); // Steam's lizard-off heartbeat (settings 0x87, ~every 3s) -> keep gamepad mode, suppress auto-lizard
  276. if (rid == 1 && n >= 2) { // report 0x01 = raw passthrough -> queue for RF relay to the controller
  277. uint16_t m = n > sizeof g_relayBuf ? sizeof g_relayBuf : n;
  278. memcpy(g_relayBuf, b, m); g_relayN = m; g_relayPend = true;
  279. }
  280. if (Serial.availableForWrite() > 80) { // log host feature writes (non-blocking)
  281. Serial.printf("# SET if%d rid=%02X cmd=%02X len=%u:", slot, rid, cmd, len);
  282. for (uint16_t i = 0; i < n && i < 14; i++) Serial.printf(" %02X", b[i]);
  283. Serial.println();
  284. }
  285. memset(S.resp, 0, sizeof S.resp); S.resp_len = 0;
  286. switch (cmd) {
  287. case 0x83:
  288. S.resp[0] = 0x83; S.resp[1] = sizeof ATTR83; memcpy(S.resp + 2, ATTR83, sizeof ATTR83); S.resp_len = 63; break;
  289. case 0xAE: {
  290. uint8_t idx = pln > 0 ? pl[0] : 1; const char *s = (idx == 0) ? g_board : (idx == 1) ? g_unit : "NA";
  291. S.resp[0] = 0xAE; S.resp[1] = 0x14; S.resp[2] = idx; memset(S.resp + 3, 0, 60); memcpy(S.resp + 3, s, strlen(s)); S.resp_len = 63; break; }
  292. case 0xB4: // connection/version state per slot: value 0x02 = controller connected, 0x01 = not
  293. S.resp[0] = 0xB4; S.resp[1] = 0x01;
  294. S.resp[2] = (slot == g_connSlot && !g_xbox && (millis() - g_connReplyMs < 500)) ? 0x02 : 0x01;
  295. S.resp_len = 63; break;
  296. case 0xAD:
  297. g_pairing = (pln > 0 && pl[0] != 0); Serial.printf("# pairing %s\n", g_pairing ? "ON" : "off");
  298. S.resp[0] = 0xAD; S.resp[1] = 0; S.resp_len = 63; break;
  299. case 0xA2: // write/clear THIS interface's slot
  300. if (len >= 24 && pln >= 24) {
  301. if (recEmpty(pl)) { S.used = false; memset(S.rec, 0, 24); }
  302. else { memcpy(S.rec, pl, 24); S.used = true; }
  303. g_dirty = true; Serial.printf("# slot %d %s\n", slot, recEmpty(pl) ? "cleared" : "bonded");
  304. }
  305. S.resp[0] = 0xA2; S.resp[1] = 0; S.resp_len = 63; break;
  306. case 0xA3: // read THIS interface's slot
  307. S.resp[0] = 0xA3; S.resp[1] = 0x18; memset(S.resp + 2, 0, 24);
  308. if (S.used) memcpy(S.resp + 2, S.rec, 24); S.resp_len = 63; break;
  309. default:
  310. S.resp[0] = cmd; S.resp[1] = len; if (pln) memcpy(S.resp + 2, pl, pln > 60 ? 60 : pln); S.resp_len = 63; break;
  311. }
  312. }
  313. static uint16_t handleGet(int slot, uint8_t rid, hid_report_type_t type, uint8_t *buf, uint16_t reqlen) {
  314. if (type != HID_REPORT_TYPE_FEATURE) return 0;
  315. Slot &S = g_slot[slot];
  316. uint16_t n = S.resp_len ? S.resp_len : 63; if (n > reqlen) n = reqlen;
  317. memcpy(buf, S.resp, n); return n;
  318. }
  319. // one callback pair per interface (the Adafruit core routes by interface to the matching object)
  320. #define SLOTCB(N) \
  321. static void setcb##N(uint8_t r, hid_report_type_t t, uint8_t const *b, uint16_t n) { handleSet(N, r, t, b, n); } \
  322. static uint16_t getcb##N(uint8_t r, hid_report_type_t t, uint8_t *bf, uint16_t rl) { return handleGet(N, r, t, bf, rl); }
  323. SLOTCB(0) SLOTCB(1) SLOTCB(2) SLOTCB(3)
  324. typedef uint16_t (*getcb_t)(uint8_t, hid_report_type_t, uint8_t *, uint16_t);
  325. typedef void (*setcb_t)(uint8_t, hid_report_type_t, uint8_t const *, uint16_t);
  326. static getcb_t GETCB[NSLOT] = { getcb0, getcb1, getcb2, getcb3 };
  327. static setcb_t SETCB[NSLOT] = { setcb0, setcb1, setcb2, setcb3 };
  328. // ===================== RF link (firmware-derived) =====================
  329. // Register-confirmed ESB config (RESOLVED.md): 2Mbit, 5-byte addr (BALEN=4), dynamic length
  330. // (LFLEN=8,S1LEN=3), 16-bit preamble, big-endian, whitening off, CRC16 poly 0x11021 init 0xFFFF
  331. // (address included). On-air = bitrev8(stored); RADIO BASE = byteswap32(bitrev8(stored)).
  332. // Pairing rendezvous: ch2 (2402MHz), base 91 A2 A7 93 (FUN_0000b8b8 / PTR_DAT_0000ba30).
  333. // Dongle protocol (protocol/RF_CONNECTED.md): beacon [01][seq][..][E2@5]; handshake 0xF0/0xE0/0xF4.
  334. // Bare-metal RADIO (no SoftDevice — never start BLE); coexists with TinyUSB. Stage A: listen+log.
  335. // Discovery/reconnect address (IBEX FUN_00019000, firmware-derived, definitive):
  336. // base = "ibex" (69 62 65 78), prefix = 0x10, channel = 2.
  337. // (The 0x91A2A793 bytes in IBEX rodata precede the key string but are a *different* key,
  338. // NOT the discovery address — confirmed via PTR_s_ibexesb_common=0x56f99 -> "ibex...".)
  339. static const uint8_t PAIR_BASE[4] = {0x69, 0x62, 0x65, 0x78}; // "ibex"
  340. static uint8_t g_rfPrefix = 0x10; // discovery prefix (rodata byte @0x56f98)
  341. static uint8_t g_rfCh = 2;
  342. static uint8_t g_sessCh = 18; // connected-session channel: a CLEAN data channel, off the congested BLE-adv
  343. // ch2 (=adv37). ch18 measured clean (RSSI 95/90, PER 0) on the REAL PUCK's own
  344. // channel scan and is in its active hop set {18,2,80}; ch52 (old) was mediocre.
  345. // Trackpad data bursts collide on bad channels -> reply-rate crash. Tunable 'C'.
  346. static bool g_rfListen = false;
  347. static uint8_t g_rfBase[4] = {0x69, 0x62, 0x65, 0x78}; // "ibex"
  348. static uint8_t rfrx[80], rftx[80];
  349. static uint32_t g_rfRxCount = 0;
  350. static uint8_t rfBitrev8(uint8_t x){ uint8_t y=0; for(int i=0;i<8;i++){y=(y<<1)|(x&1);x>>=1;} return y; }
  351. // HYBRID puck-link config (firmware-derived, definitive):
  352. // PHY = Ble_1Mbit (IBEX FUN_0000f108 MODE=3 is the only RADIO MODE write; ESB inherits it)
  353. // PCNF0 = 0x00100108 (S0LEN=1 [buf[0]=0x12 IS the S0 byte], LFLEN=8, S1INCL=1, 8-bit preamble)
  354. // PCNF1 = 0x02030040 (WHITEEN ON, BALEN=3 [4-byte addr], little-endian, MAXLEN=64)
  355. // CRC = CRC16 poly 0x11021, init 0xFFFF, CRCCNF=2 (address INCLUDED) -- ESB override
  356. // (FUN_00037000); NOT the Bluetooth radio's CRC24 0x65b/CRCCNF=0x103.
  357. // whiten = ON, DATAWHITEIV=37 (RADIO FREQUENCY=2 = 2402MHz = BLE adv channel 37)
  358. // addr = ESB addr_conv (FUN_00027a78 bitrev8 each byte): "ibex" -> BASE0=0x9646A61E.
  359. // All tunable via CDC (M/0/2/i/w + I=crcinit) for hardware iteration.
  360. // DECODED from real-puck capture (definitive): puck TX = Ble_2Mbit, ENDIAN=Big (MSB-first), NO whitening,
  361. // addr "ibex", CRC16 0x11021/0xFFFF. Frame = FUN_00027e9a (static length). My Little-endian capture saw
  362. // it bit-reversed; decode (shift3+bitrev8) recovered 0xE1 + proteus_uuid_LE + ibex_uuid_LE for both bonds.
  363. static uint32_t g_crcinit = 0xFFFF;
  364. static uint8_t g_whiteiv = 37;
  365. static uint8_t g_mode = RADIO_MODE_MODE_Ble_2Mbit; // <-- was the bug (1Mbit). Real puck = 2Mbit.
  366. static uint32_t g_pcnf0 = 0x00030008UL; // S0LEN0, LFLEN8, S1LEN3 (ESB DPL) - CRC-VALIDATED
  367. static uint8_t g_statlen = 0x20; // (static-mode payload length; tunable 'T')
  368. // PCNF1: ENDIAN=Big (puck transmits MSB-first), WHITEEN=0, BALEN3, MAXLEN64.
  369. static uint32_t g_pcnf1 = 0x01040040; // ENDIAN=Big, BALEN4, MAXLEN64 - CRC-VALIDATED
  370. static uint32_t g_crcpoly = 0x11021UL;
  371. static uint16_t g_crccnf = 0x2; // CRC16, address included
  372. static uint8_t g_pid = 0;
  373. // Connected-mode poll (Phase 2b): puck (PTX) sends 0xE7 protocol-version then 0xE3 input polls on the
  374. // session addr/ch; controller responds in the ACK/reply with its input report (byte5==0xF1 + TLV,
  375. // parsed by IBEX FUN_0000ba38). Tunable for on-hardware iteration (j=type, Q=len, k=enable).
  376. static bool g_connOn = true; // auto-start poll on boot (so a USB replug resumes acting as a puck)
  377. static uint8_t g_connType = 0xE7; // start with protocol-version handshake, then 0xE3
  378. static uint8_t g_e7b = 0; // E7 payload B-byte: 0=current(slow/awake), 1=test protocol-version-1 (ODR agent: B!=0 sets controller 'v1 fast' latch -> ~100Hz?). 'V<n>' to toggle. RISK: original RE called [00][01] "suspended" — revert to 0 if input stops.
  379. static uint8_t g_connLen = 0x08;
  380. static uint32_t g_stPoll=0, g_stF1=0, g_stF3=0; static unsigned long g_stMs=0; // per-second poll/F1/F3 stat counters
  381. static uint8_t g_connF3v=0xFF; // last protocol version the controller reported in an F3 reply (0xFF=none seen)
  382. static uint16_t g_f1ps=0; // last completed second's F1 rate (for the WebUSB status readout)
  383. static uint8_t g_lastSeq=0; static uint32_t g_stNew=0; static uint16_t g_newps=0; // genuine new-report rate (report 0x45 seq byte changes)
  384. static uint32_t g_stCrc=0, g_stNoRx=0; // diagnostics: replies w/ bad CRC, and polls with no reply at all (timing vs RF-quality)
  385. static uint32_t g_lastPollUs=0; // (g_pollUs declared in the config block near the top; 'U' or WebUSB to tune)
  386. static unsigned long g_lastSessBeacon=0; // session keepalive beacon timer
  387. static uint32_t g_connRx = 0;
  388. static uint8_t g_balen = 4; // ESB 5-byte addr
  389. // Access address (IBEX FUN_00037530 + func_0x00049e5a): BASE0 = bitrev8 each base byte, packed
  390. // big-endian (= byteswap(bitrev8(uint32))); PREFIX0 = raw prefix (func_0x0003756c copies it). BALEN=3
  391. // uses BASE0[31:8] + PREFIX0. For "ibex" 69 62 65 78 -> BASE0=0x9646A61E, PREFIX0=0x10.
  392. static bool g_prefixRaw = false; // false = bitrev8 prefix (ESB addr_conv); true = raw prefix byte
  393. static void rfSetAddr(const uint8_t b4[4], uint8_t prefix){
  394. uint8_t b[4]; for(int i=0;i<4;i++) b[i]=rfBitrev8(b4[i]);
  395. NRF_RADIO->BASE0 = ((uint32_t)b[0]<<24)|((uint32_t)b[1]<<16)|((uint32_t)b[2]<<8)|b[3];
  396. NRF_RADIO->PREFIX0 = g_prefixRaw ? prefix : rfBitrev8(prefix);
  397. NRF_RADIO->TXADDRESS = 0; NRF_RADIO->RXADDRESSES = 1u<<0;
  398. }
  399. static void rfConfig(uint8_t ch){
  400. NRF_RADIO->TASKS_DISABLE = 1; RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED = 0;
  401. NRF_RADIO->MODE = ((uint32_t)g_mode << RADIO_MODE_MODE_Pos); // Ble_1Mbit (tunable 'M')
  402. NRF_RADIO->FREQUENCY = ch;
  403. NRF_RADIO->TXPOWER = (RADIO_TXPOWER_TXPOWER_0dBm << RADIO_TXPOWER_TXPOWER_Pos);
  404. #if defined(RADIO_MODECNF0_RU_Fast)
  405. NRF_RADIO->MODECNF0 = (RADIO_MODECNF0_RU_Fast<<RADIO_MODECNF0_RU_Pos);
  406. #endif
  407. NRF_RADIO->PCNF0 = g_pcnf0; // static length (tunable '0'); g_pcnf1=0 -> build from g_statlen/g_balen
  408. NRF_RADIO->PCNF1 = g_pcnf1 ? g_pcnf1 :
  409. ((1u<<RADIO_PCNF1_ENDIAN_Pos) // ENDIAN=Big (puck transmits MSB-first)
  410. |((uint32_t)g_balen<<RADIO_PCNF1_BALEN_Pos)
  411. |((uint32_t)g_statlen<<RADIO_PCNF1_STATLEN_Pos)
  412. |((uint32_t)g_statlen<<RADIO_PCNF1_MAXLEN_Pos)); // WHITEEN=0
  413. NRF_RADIO->CRCCNF = g_crccnf; // CRC16, address included
  414. NRF_RADIO->CRCPOLY = g_crcpoly; NRF_RADIO->CRCINIT = g_crcinit;
  415. NRF_RADIO->DATAWHITEIV = g_whiteiv;
  416. NRF_RADIO->PACKETPTR = (uint32_t)rfrx;
  417. }
  418. static void rfListenStart(){
  419. rfConfig(g_rfCh); rfSetAddr(g_rfBase, g_rfPrefix);
  420. NRF_RADIO->PACKETPTR = (uint32_t)rfrx;
  421. NRF_RADIO->SHORTS = RADIO_SHORTS_READY_START_Msk | RADIO_SHORTS_END_START_Msk;
  422. NRF_RADIO->EVENTS_END = 0; NRF_RADIO->TASKS_RXEN = 1;
  423. g_rfListen = true;
  424. Serial.printf("# RF listen ch%u base %02X%02X%02X%02X prefix %02X\n",
  425. g_rfCh, g_rfBase[0],g_rfBase[1],g_rfBase[2],g_rfBase[3], g_rfPrefix);
  426. }
  427. static void rfPoll(){
  428. if (!g_rfListen) return;
  429. if (NRF_RADIO->EVENTS_END){
  430. NRF_RADIO->EVENTS_END = 0;
  431. bool crcok = NRF_RADIO->CRCSTATUS & 1;
  432. uint8_t len = rfrx[0];
  433. if (len && len < 64){
  434. g_rfRxCount++;
  435. Serial.printf("RF#%lu ch%u crc%d len%u: ", (unsigned long)g_rfRxCount, g_rfCh, crcok, len);
  436. for (uint8_t i=0;i<=len && i<40;i++) Serial.printf("%02X ", rfrx[i]);
  437. Serial.println();
  438. }
  439. }
  440. }
  441. // Dongle is PTX: TX a beacon [01][seq][..][E2@5] then RX the controller's response (0xF0...).
  442. // RAM layout / payload length are byte-level unknowns to pin live (S1INCL, plen tweakable).
  443. static bool g_rfBeacon = false;
  444. static uint8_t g_seq = 0, g_plen = 0x18, g_s1incl = 0;
  445. static unsigned long g_lastBeacon = 0;
  446. static unsigned long g_lastDisc = 0; // last discovery beacon (ch2) time
  447. static bool g_rfHost = true; // auto-start host beacon on boot (resumes puck role after a USB replug)
  448. static unsigned long g_lastHop = 0;
  449. // HOST FRAME the bonded controller waits for (IBEX FUN_00019000 verify: b[0]=0x12,b[5]=0xE1,
  450. // b[6..10]=proteus_uuid, b[10..14]=ibex_uuid). Built like PROTEUS FUN_00027e9a. Sent on the shared
  451. // rendezvous addr; the controller filters by the uuids in the payload, then connects.
  452. static void rfHostFrameOnce(int slot){
  453. if (slot<0||slot>=NSLOT||!g_slot[slot].used) return;
  454. uint8_t *rec = g_slot[slot].rec; // [proteus_uuid 4][ibex_uuid 4][serial 16]
  455. // CRC-VALIDATED frame (decoded from real puck): ESB-DPL RAM = [LENGTH][S1=PID][payload(18)].
  456. // payload: [0]=0xE1, [1..5]=proteus_uuid LE, [5..9]=ibex_uuid LE, [9]=session channel,
  457. // [10..13]=0, [13..17]=session base, [17]=session prefix. Radio auto-appends CRC16 0x11021.
  458. memset(rftx,0,sizeof rftx);
  459. rftx[0]=0x12; // LENGTH = 18 (controller's buf[0]==0x12 check validates this)
  460. rftx[1]=(uint8_t)((g_pid++&3)<<1); // S1 = PID<<1 | noack0 (matches real puck 00/02/04/06)
  461. rftx[2]=0xE1; // payload[0] marker
  462. memcpy(rftx+3, rec+0, 4); // payload[1..5] proteus_uuid (LE, as bonded)
  463. memcpy(rftx+7, rec+4, 4); // payload[5..9] ibex_uuid
  464. rftx[11]=g_sessCh; // payload[9] session channel: tell the controller to run the session on
  465. // the clean channel (it adopts buf[0xe]); discovery beacon still TXes on ch2
  466. memcpy(rftx+15, g_rfBase, 4); // payload[13..17] session base
  467. rftx[19]=g_rfPrefix; // payload[17] session prefix
  468. rfConfig(g_rfCh); rfSetAddr(g_rfBase,g_rfPrefix);
  469. NRF_RADIO->PACKETPTR=(uint32_t)rftx;
  470. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_DISABLE_Msk;
  471. NRF_RADIO->EVENTS_DISABLED=0; NRF_RADIO->TASKS_TXEN=1;
  472. RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  473. NRF_RADIO->PACKETPTR=(uint32_t)rfrx; rfrx[0]=0;
  474. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk;
  475. NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1;
  476. uint32_t t0=micros(); while(!NRF_RADIO->EVENTS_END && (micros()-t0)<800){}
  477. if (NRF_RADIO->EVENTS_END){ NRF_RADIO->EVENTS_END=0; // ANY reception = controller answered our frame
  478. g_rfRxCount++; bool crcok=NRF_RADIO->CRCSTATUS&1; uint8_t len=rfrx[0];
  479. if (Serial.availableForWrite()>90){ // non-blocking: don't stall the loop on CDC backpressure
  480. Serial.printf("*** RESP#%lu ch%u crc%d rxmatch%lu len%u: ",(unsigned long)g_rfRxCount,
  481. g_rfCh,crcok,(unsigned long)NRF_RADIO->RXMATCH,len);
  482. for(uint8_t i=0;i<(len<40?len+2:40);i++)Serial.printf("%02X ",rfrx[i]); Serial.println(); } }
  483. NRF_RADIO->TASKS_DISABLE=1; RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  484. }
  485. // Mid-session channel hop: advertise newCh in the host frame, transmitted on the CURRENT session channel
  486. // (where the controller is listening), a few times so it reliably adopts it; then move our poll to newCh.
  487. // This is the foundational test for channel-hopping/QoS — does the controller follow a live channel change?
  488. static void rfHopTo(uint8_t newCh){
  489. if(g_connSlot<0 || newCh==g_sessCh) return;
  490. uint8_t cur=g_sessCh, savedRfCh=g_rfCh;
  491. g_sessCh=newCh; g_rfCh=cur; // host frame now advertises newCh but is TXed on cur
  492. for(int k=0;k<6;k++){ rfHostFrameOnce(g_connSlot); delayMicroseconds(700); }
  493. g_rfCh=savedRfCh; // poll + session beacon now run on g_sessCh=newCh
  494. }
  495. // CONNECTED MODE: OpenPuck is the PTX/master. It already chose the session base/prefix/channel
  496. // (advertised in the host frame the controller connected on), so no discovery/re-sync is needed. After
  497. // connect: send E7 (proto-version 00 01), send E4 (channel map {ch,2,80}), then loop E3 input-poll (+E2
  498. // keepalive periodically) hopping the map; RX the controller's reply each cycle (0xF1 = full input report
  499. // w/ TLVs, 0xF3 = short status). Captured PUCK->controller bytes [LEN][S1][payload], payload[0]=type:
  500. // E7 00 01 (S1=01) | E4 <ch> 02 50 (S1=05) | E2 (S1=07) | E3 (S1=07). See FUN_0000ba38 / FUN_0001b960.
  501. // One TX->RX cycle: TX the connected packet, SHORTS straight into a short RX window to catch the reply.
  502. // ---- XBOX-mode output: map report 0x45 -> standard gamepad (g_pad) + right-trackpad -> mouse (report 0x40) ----
  503. // report 0x45 layout: [0]=0x45 [1]=seq [2..5]=buttons u32; analog offsets are from buttons-low-byte (rep[2]).
  504. static int s16off(const uint8_t* r,int off){ int v=r[2+off]|(r[2+off+1]<<8); return (v&0x8000)?v-0x10000:v; }
  505. static int u16off(const uint8_t* r,int off){ return r[2+off]|(r[2+off+1]<<8); }
  506. static uint32_t btnsOf(const uint8_t* r){ return (uint32_t)r[2]|((uint32_t)r[3]<<8)|((uint32_t)r[4]<<16)|((uint32_t)r[5]<<24); }
  507. // Triton button masks
  508. #define TB_A 0x1u
  509. #define TB_B 0x2u
  510. #define TB_X 0x4u
  511. #define TB_Y 0x8u
  512. #define TB_R3 0x20u
  513. #define TB_VIEW 0x40u
  514. #define TB_R4 0x80u
  515. #define TB_R5 0x100u
  516. #define TB_RB 0x200u
  517. #define TB_DDN 0x400u
  518. #define TB_DRT 0x800u
  519. #define TB_DLF 0x1000u
  520. #define TB_DUP 0x2000u
  521. #define TB_MENU 0x4000u
  522. #define TB_L3 0x8000u
  523. #define TB_STEAM 0x10000u
  524. #define TB_L4 0x20000u
  525. #define TB_L5 0x40000u
  526. #define TB_LB 0x80000u
  527. #define TB_RPADT 0x200000u
  528. #define TB_RPADC 0x400000u
  529. #define TB_LPADT 0x2000000u
  530. #define TB_LPADC 0x4000000u
  531. // Trackpad coords genuinely refresh only ~46Hz (capsense floor) but reports ship ~300Hz -> stale-repeats.
  532. // Forwarding those raw to Steam defeats its interpolation (it sees no-motion then a jump). Smooth the pad
  533. // X/Y in the forwarded report 0x45 (exponential LERP toward the latest genuine value each report) so Steam
  534. // sees continuous motion. g_padSmooth = alpha% (100 = off, lower = smoother/laggier). 'D<pct>' tunes it.
  535. // (g_padSmooth declared at top with the persisted config)
  536. static void smoothPad(uint8_t* rep){ // rep = report 0x45; smooth LPad(@18/20) + RPad(@24/26) in place
  537. if(g_padSmooth>=100) return;
  538. float a=g_padSmooth/100.0f; uint32_t b=btnsOf(rep);
  539. static float lsx=0,lsy=0,rsx=0,rsy=0; static bool lpt=false,rpt=false;
  540. bool lt=b&TB_LPADT; int16_t lx=rep[18]|(rep[19]<<8), ly=rep[20]|(rep[21]<<8);
  541. if(lt){ if(!lpt){lsx=lx;lsy=ly;} else {lsx+=(lx-lsx)*a; lsy+=(ly-lsy)*a;}
  542. int16_t ox=(int16_t)lsx,oy=(int16_t)lsy; rep[18]=ox;rep[19]=ox>>8;rep[20]=oy;rep[21]=oy>>8; }
  543. lpt=lt;
  544. bool rt=b&TB_RPADT; int16_t rx=rep[24]|(rep[25]<<8), ry=rep[26]|(rep[27]<<8);
  545. if(rt){ if(!rpt){rsx=rx;rsy=ry;} else {rsx+=(rx-rsx)*a; rsy+=(ry-rsy)*a;}
  546. int16_t ox=(int16_t)rsx,oy=(int16_t)rsy; rep[24]=ox;rep[25]=ox>>8;rep[26]=oy;rep[27]=oy>>8; }
  547. rpt=rt;
  548. }
  549. // button code (g_back[], g_abSwap targets) -> XInput bit. 0=none 1=A 2=B 3=X 4=Y 5=LB 6=RB 7=L3 8=R3 9=Back 10=Start 11=Guide
  550. static uint16_t codeToXB(uint8_t c){
  551. switch(c){ case 1:return XB_A; case 2:return XB_B; case 3:return XB_X; case 4:return XB_Y;
  552. case 5:return XB_LB; case 6:return XB_RB; case 7:return XB_L3; case 8:return XB_R3;
  553. case 9:return XB_BACK; case 10:return XB_START; case 11:return XB_GUIDE; default:return 0; }
  554. }
  555. static void rfXboxGamepad(const uint8_t* r){
  556. uint32_t b=btnsOf(r);
  557. uint16_t btn=0; // map Triton buttons -> XInput button word
  558. if(b&TB_DUP)btn|=XB_DUP; if(b&TB_DDN)btn|=XB_DDOWN; if(b&TB_DLF)btn|=XB_DLEFT; if(b&TB_DRT)btn|=XB_DRIGHT;
  559. if(b&TB_MENU)btn|=XB_START; if(b&TB_VIEW)btn|=XB_BACK; if(b&TB_STEAM)btn|=XB_GUIDE;
  560. if(b&TB_LB)btn|=XB_LB; if(b&TB_RB)btn|=XB_RB;
  561. if(b&TB_L3)btn|=XB_L3; if(b&TB_R3)btn|=XB_R3;
  562. // face buttons, with optional A/B + X/Y swap (Nintendo layout)
  563. uint16_t fA=g_abSwap?XB_B:XB_A, fB=g_abSwap?XB_A:XB_B, fX=g_abSwap?XB_Y:XB_X, fY=g_abSwap?XB_X:XB_Y;
  564. if(b&TB_A)btn|=fA; if(b&TB_B)btn|=fB; if(b&TB_X)btn|=fX; if(b&TB_Y)btn|=fY;
  565. // back paddles -> configurable mapping (default L4->LB, R4->RB, L5->L3, R5->R3)
  566. if(b&TB_L4)btn|=codeToXB(g_back[0]); if(b&TB_R4)btn|=codeToXB(g_back[1]);
  567. if(b&TB_L5)btn|=codeToXB(g_back[2]); if(b&TB_R5)btn|=codeToXB(g_back[3]);
  568. uint8_t lt=u16off(r,4)>>8, rt=u16off(r,6)>>8; // triggers u16 -> u8
  569. xinputSend(btn, lt, rt, (int16_t)s16off(r,8), (int16_t)s16off(r,10), // L stick X/Y
  570. (int16_t)s16off(r,12), (int16_t)s16off(r,14)); // R stick X/Y
  571. }
  572. // Right-pad -> mouse, modeled on the real puck's lizard mouse: pad deltas feed a VELOCITY that is emitted
  573. // each report and decays with friction. This (a) smooths the stale-repeat bursts, (b) gives momentum/flick
  574. // (a fast swipe injects high velocity -> cursor keeps sliding after lift), (c) sub-pixel remainder carry so
  575. // slow drags still move. Tunable live: M<div>=sensitivity (lower=faster), F<pct>=friction% (higher=more glide).
  576. // (g_mDiv, g_mFric declared at top with the persisted config)
  577. static void rfXboxMouse(const uint8_t* r){
  578. uint32_t b=btnsOf(r);
  579. static int px=0,py=0; static bool pt=false; static float vx=0,vy=0,rmx=0,rmy=0; static uint8_t pmb=0;
  580. bool touch=b&TB_RPADT; int rx=s16off(r,22), ry=s16off(r,24);
  581. if(touch){ if(pt){ vx+=(rx-px); vy+=(ry-py); } px=rx; py=ry; } // inject genuine pad move into velocity
  582. pt=touch;
  583. float mxf = vx/(float)g_mDiv + rmx, myf = -(vy/(float)g_mDiv) + rmy; // Y inverted for screen coords
  584. int dx=(int)mxf, dy=(int)myf; rmx=mxf-dx; rmy=myf-dy; // fractional carry (sub-pixel)
  585. if(dx>127)dx=127; if(dx<-127)dx=-127; if(dy>127)dy=127; if(dy<-127)dy=-127;
  586. float f=g_mFric/100.0f; vx*=f; vy*=f; if(vx>-1&&vx<1)vx=0; if(vy>-1&&vy<1)vy=0; // friction (glide/decay)
  587. uint8_t mb=((b&TB_RPADC)?1:0)|((b&TB_LPADC)?2:0); // RPad click=left, LPad click=right
  588. if(dx||dy||mb!=pmb){ pmb=mb;
  589. hid_mouse_report_t m; m.buttons=mb; m.x=(int8_t)dx; m.y=(int8_t)dy; m.wheel=0; m.pan=0;
  590. if(g_mouse.ready()) g_mouse.sendReport(0,&m,sizeof m);
  591. }
  592. }
  593. // ===================== LIZARD mode (mode 3) — desktop keyboard + mouse =====================
  594. // The controller's native "lizard" desktop behavior, reproduced on the OpenPuck side so the controller is a
  595. // driverless keyboard+mouse when Steam isn't running. Canonical Valve Steam Controller firmware map (SC1,
  596. // the documented reference; SC2 presumed-similar): right pad -> mouse (trackball glide), left pad -> scroll
  597. // wheel + middle-click on press, R-trigger / R-pad-click -> left mouse, L-trigger -> right mouse;
  598. // A=Enter B=Esc X=PgUp Y=PgDn, D-pad & left-stick -> arrow keys, LB=LeftCtrl RB=LeftAlt, Back/View=Tab,
  599. // Start/Menu=Esc, Steam=none. The mouse reuses the SAME velocity+friction+sub-pixel GLIDE model as Xbox mode
  600. // (g_mDiv = sensitivity divisor, g_mFric = friction% — both live-tunable), which is the "custom smoothing
  601. // and gliding" the real lizard mouse has (cursor coasts after lift, smooth sub-pixel slow drags).
  602. static void rfLizard(const uint8_t* r, Adafruit_USBD_HID* mdev, Adafruit_USBD_HID* kdev, uint8_t mrid, uint8_t krid){
  603. uint32_t b=btnsOf(r);
  604. // --- right pad -> mouse motion with glide (mirrors rfXboxMouse) ---
  605. static int prx=0,pry=0; static bool prt=false; static float vx=0,vy=0,rmx=0,rmy=0;
  606. bool rtouch=b&TB_RPADT; int rx=s16off(r,22), ry=s16off(r,24);
  607. if(rtouch){ if(prt){ vx+=(rx-prx); vy+=(ry-pry); } prx=rx; pry=ry; } prt=rtouch;
  608. float mxf=vx/(float)g_lizDiv+rmx, myf=-(vy/(float)g_lizDiv)+rmy; // Y inverted for screen coords; g_lizDiv = lizard sensitivity (separate from Xbox aim)
  609. int dx=(int)mxf, dy=(int)myf; rmx=mxf-dx; rmy=myf-dy; // sub-pixel carry
  610. if(dx>127)dx=127; if(dx<-127)dx=-127; if(dy>127)dy=127; if(dy<-127)dy=-127;
  611. float f=g_lizFric/100.0f; vx*=f; vy*=f; if(vx>-1&&vx<1)vx=0; if(vy>-1&&vy<1)vy=0; // friction = glide/decay (lizard has its own, crisper than Xbox)
  612. // --- left pad -> vertical scroll wheel (no momentum; only while touching, coarse) ---
  613. static int ply=0; static bool plt=false; static float sacc=0;
  614. bool ltouch=b&TB_LPADT; int ly=s16off(r,18);
  615. if(ltouch){ if(plt){ sacc += (ly-ply)/(float)(g_lizDiv*8); } ply=ly; } else sacc=0; plt=ltouch;
  616. int dw=(int)sacc; sacc-=dw; if(dw>15)dw=15; if(dw<-15)dw=-15; // finger up = wheel up (positive)
  617. // --- mouse buttons: left=R-pad-click|R-trigger, right=L-trigger, middle=L-pad-click ---
  618. uint8_t rtrig=u16off(r,6)>>8, ltrig=u16off(r,4)>>8;
  619. uint8_t mbtn=0;
  620. if((b&TB_RPADC)||rtrig>180) mbtn|=1;
  621. if(ltrig>180) mbtn|=2;
  622. if(b&TB_LPADC) mbtn|=4;
  623. static uint8_t pmbtn=0;
  624. if(dx||dy||dw||mbtn!=pmbtn){ pmbtn=mbtn;
  625. hid_mouse_report_t m; m.buttons=mbtn; m.x=(int8_t)dx; m.y=(int8_t)dy; m.wheel=(int8_t)dw; m.pan=0;
  626. if(mdev->ready()) mdev->sendReport(mrid,&m,sizeof m);
  627. }
  628. // --- keyboard: modifiers + up to 6 keycodes ---
  629. uint8_t mod=0, kc[6]={0,0,0,0,0,0}, nk=0;
  630. if(b&TB_LB) mod|=KEYBOARD_MODIFIER_LEFTCTRL;
  631. if(b&TB_RB) mod|=KEYBOARD_MODIFIER_LEFTALT;
  632. #define LZK(cond,code) do{ if((cond)&&nk<6) kc[nk++]=(code); }while(0)
  633. LZK(b&TB_A, HID_KEY_ENTER);
  634. LZK(b&TB_B, HID_KEY_ESCAPE);
  635. LZK(b&TB_X, HID_KEY_PAGE_UP);
  636. LZK(b&TB_Y, HID_KEY_PAGE_DOWN);
  637. LZK(b&TB_VIEW, HID_KEY_TAB);
  638. LZK(b&TB_MENU, HID_KEY_ESCAPE);
  639. int sx=s16off(r,8), sy=s16off(r,10); // left stick (XInput sign: +Y = up); deflect ~37% acts as a d-pad
  640. LZK((b&TB_DUP)||sy> 12000, HID_KEY_ARROW_UP);
  641. LZK((b&TB_DDN)||sy<-12000, HID_KEY_ARROW_DOWN);
  642. LZK((b&TB_DLF)||sx<-12000, HID_KEY_ARROW_LEFT);
  643. LZK((b&TB_DRT)||sx> 12000, HID_KEY_ARROW_RIGHT);
  644. #undef LZK
  645. static uint8_t pmod=0, pkc[6]={0,0,0,0,0,0};
  646. bool chg=(mod!=pmod); for(int i=0;i<6;i++) if(kc[i]!=pkc[i]) chg=true;
  647. if(chg){ pmod=mod; for(int i=0;i<6;i++) pkc[i]=kc[i];
  648. uint8_t krep[8]={mod,0,kc[0],kc[1],kc[2],kc[3],kc[4],kc[5]};
  649. if(kdev->ready()) kdev->sendReport(krid,krep,8);
  650. }
  651. }
  652. // ===================== SWITCH mode (mode 2) — HORIPAD / Pokkén Tournament Pro Pad =====================
  653. // Targets a REAL Switch console: a plain HID gamepad (HORI 0F0D:0092) the console accepts on plug-in with
  654. // NO handshake / SPI calibration / encryption — unlike the genuine Pro Controller (057E:2009), whose USB
  655. // init the console handles fussily. This is the descriptor every DIY Switch controller copies (the LUFA
  656. // "Splatoon printer" descriptor). Also recognized by SDL/Steam on PC. 8-byte input report streamed at
  657. // ~125Hz: [btn_lo][btn_hi][hat][LX][LY][RX][RY][vendor], sticks uint8 center 0x80.
  658. static const uint8_t SWITCH_HID_DESC[]={
  659. 0x05,0x01, 0x09,0x05, 0xA1,0x01, // Usage Page (Generic Desktop), Usage (Game Pad), Collection (App)
  660. 0x15,0x00, 0x25,0x01, 0x35,0x00, 0x45,0x01, 0x75,0x01, 0x95,0x10, // 16 x 1-bit buttons
  661. 0x05,0x09, 0x19,0x01, 0x29,0x10, 0x81,0x02,
  662. 0x05,0x01, 0x25,0x07, 0x46,0x3B,0x01, 0x75,0x04, 0x95,0x01, 0x65,0x14, 0x09,0x39, 0x81,0x42, // hat switch (4 bits)
  663. 0x65,0x00, 0x95,0x01, 0x81,0x01, // 4-bit padding to a byte
  664. 0x26,0xFF,0x00, 0x46,0xFF,0x00, 0x09,0x30, 0x09,0x31, 0x09,0x32, 0x09,0x35, 0x75,0x08, 0x95,0x04, 0x81,0x02, // X,Y,Z,Rz
  665. 0x06,0x00,0xFF, 0x09,0x20, 0x95,0x01, 0x81,0x02, // vendor byte (input)
  666. 0x0A,0x21,0x26, 0x95,0x08, 0x91,0x02, // vendor (output, 8 bytes; unused)
  667. 0xC0
  668. };
  669. static uint32_t g_swBtns=0; static int16_t g_swLX=0,g_swLY=0,g_swRX=0,g_swRY=0; // latest decoded RF frame
  670. static unsigned long g_swLastMs=0;
  671. Adafruit_USBD_HID g_switch; // HORIPAD gamepad HID interface (mode 2)
  672. // HORIPAD/Switch button bits: Y=1 B=2 A=4 X=8 L=10 R=20 ZL=40 ZR=80 Minus=100 Plus=200 LClick=400 RClick=800 Home=1000 Capture=2000
  673. static inline uint8_t swStick(int16_t v,bool invert){ // int16 (center 0) -> uint8 (center 0x80)
  674. int32_t a = 0x80 + (invert ? -((int32_t)v>>8) : ((int32_t)v>>8));
  675. if(a<0)a=0; if(a>255)a=255; return (uint8_t)a;
  676. }
  677. static void switchBuildHoripad(uint8_t out[8]){
  678. uint32_t b=g_swBtns; uint16_t btn=0;
  679. // face buttons with optional A/B + X/Y swap (Nintendo physical-vs-label layout)
  680. uint16_t fY = g_abSwap?0x08:0x01, fB = g_abSwap?0x04:0x02, fA = g_abSwap?0x02:0x04, fX = g_abSwap?0x01:0x08;
  681. if(b&TB_Y)btn|=fY; if(b&TB_B)btn|=fB; if(b&TB_A)btn|=fA; if(b&TB_X)btn|=fX;
  682. if(b&TB_LB)btn|=0x10; if(b&TB_RB)btn|=0x20; // L, R
  683. if(b&0x8000000u)btn|=0x40; if(b&0x800000u)btn|=0x80; // ZL(LTrigClick), ZR(RTrigClick)
  684. if(b&TB_VIEW)btn|=0x100; if(b&TB_MENU)btn|=0x200; // Minus, Plus
  685. if(b&TB_L3)btn|=0x400; if(b&TB_R3)btn|=0x800; // LClick, RClick
  686. if(b&TB_STEAM)btn|=0x1000; // Home
  687. bool u=b&TB_DUP,d=b&TB_DDN,l=b&TB_DLF,r=b&TB_DRT; // hat: 0=N..7=NW, 8=neutral
  688. uint8_t hat=8;
  689. if(u&&r)hat=1; else if(r&&d)hat=3; else if(d&&l)hat=5; else if(l&&u)hat=7;
  690. else if(u)hat=0; else if(r)hat=2; else if(d)hat=4; else if(l)hat=6;
  691. out[0]=btn&0xFF; out[1]=btn>>8; out[2]=hat;
  692. out[3]=swStick(g_swLX,false); out[4]=swStick(g_swLY,true); // HID Y is down-positive -> invert
  693. out[5]=swStick(g_swRX,false); out[6]=swStick(g_swRY,true);
  694. out[7]=0;
  695. }
  696. static void swStream(){ // call from loop(); stream the 8-byte HORIPAD report at ~125Hz (no handshake needed)
  697. if(g_usbMode!=2 || !g_switch.ready()) return;
  698. if(millis()-g_swLastMs < 8) return; g_swLastMs=millis();
  699. uint8_t p[8]; switchBuildHoripad(p);
  700. g_switch.sendReport(0, p, sizeof p); // report-id-less descriptor
  701. }
  702. // RECIPE (IBEX disasm @0x18d80): controller streams 0xF1 input report ONLY when (a) connstate==3 [host
  703. // frame E1 establishes it], (b) we mark host-awake via 0xE7 payload [00][00] (vs [00][01]=suspended,
  704. // handler @0x193aa), and (c) EACH 0xE3 poll carries a GET-report-0x45 sub-TLV [len=02][subtype=01][id=45]
  705. // [param] (parser @0x1948c; F1 builder @0x20ca4 arms a one-shot latch @0x2000d36e cleared after each F1 ->
  706. // no free-run stream, 1 F1 per GET). F1 reply = type 0xF1 @rfrx[2] then TLV type4(analog/buttons)+
  707. // type6(per-module gyro/accel/sticks).
  708. static uint8_t g_connSt = 0; // 0=announce awake, 1=poll loop
  709. static uint8_t g_connStep = 0; // repeat counter within a state
  710. static uint16_t g_connPoll = 0; // poll counter (re-assert awake every 64th)
  711. static uint32_t g_connF1 = 0; // count of 0xF1 input reports seen
  712. static uint8_t g_getParam = 0x00; // GET report 0x45 param byte (try 0x00, fall back 0x2D) - 'q' cmd
  713. static uint8_t g_e3mode = 1, g_e3pid = 0; // E3-poll PID/S1 mode: 0=fixed 0x07, 1=cycle PID(noack1), 2=cycle PID(noack0). DEFAULT 1: cycling the ESB PID drains the controller's report queue (~400 new/s vs ~60 with a fixed PID) — THE rate fix. 'e<n>' to A/B.
  714. static bool g_connVerbose = false; // true=full multi-line debug dump; false=compact "I45 <hex>" stream
  715. static unsigned long g_lastStream = 0; // throttle for compact UI stream
  716. static unsigned long g_connCooldown = 0; // set on 0xF2 disconnect; pauses beacon+poll so a powering-off controller can sleep
  717. // (g_connReplyMs, g_connSlot declared near g_slot so the USB feature handler can read them)
  718. // TX one connected packet [LEN][S1][payload] on channel ch, then RX the reply into rfrx. Decodes 0xF1.
  719. static uint8_t rfConnTx(uint8_t ch, uint8_t s1, const uint8_t* payload, uint8_t plen){
  720. memset(rftx,0,sizeof rftx);
  721. rftx[0]=plen; // LENGTH = payload byte count
  722. rftx[1]=s1; // S1 (type-specific)
  723. memcpy(rftx+2, payload, plen); // payload[0]=type byte, then data/TLVs
  724. rfConfig(ch); rfSetAddr(g_rfBase,g_rfPrefix);
  725. NRF_RADIO->PACKETPTR=(uint32_t)rftx;
  726. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_DISABLE_Msk;
  727. NRF_RADIO->EVENTS_DISABLED=0; NRF_RADIO->TASKS_TXEN=1;
  728. RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  729. NRF_RADIO->PACKETPTR=(uint32_t)rfrx; rfrx[0]=0;
  730. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk; // RXEN->READY->START; catch the reply
  731. NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1;
  732. g_stPoll++;
  733. uint32_t t0=micros(); while(!NRF_RADIO->EVENTS_END && (micros()-t0)<g_rxWin){} // RX window (tunable 'r'): caps poll rate ~1e6/rxWin
  734. uint8_t rxlen=0;
  735. if (NRF_RADIO->EVENTS_END){ NRF_RADIO->EVENTS_END=0;
  736. bool crcok=NRF_RADIO->CRCSTATUS&1; rxlen=rfrx[0];
  737. if(!crcok){ g_stCrc++; g_qosBad++; } // reply arrived but CRC failed -> RF quality (channel/interference)
  738. if (crcok && rxlen && rxlen<=64){ // F1 report ~46B, so allow up to MAXLEN
  739. g_connRx++; g_connReplyMs=millis(); // link alive -> loop() suppresses the redundant E1 beacon
  740. if(rfrx[2]==0xF1) g_stF1++;
  741. uint8_t rtype=rfrx[2]; // reply type byte (proven offset from captures)
  742. if(rtype==0xF2) g_connCooldown=millis(); // controller disconnecting/powering off -> back off 2.5s
  743. if(rtype==0xF3){ // F3 = controller status/version reply (ODR agent: reply to E7 version handshake, byte[6]=version)
  744. g_stF3++; g_connF3v=rfrx[6];
  745. if(g_connVerbose && Serial.availableForWrite()>40){ Serial.print(" F3 "); for(uint8_t i=0;i<(rxlen+2<32?rxlen+2:32);i++)Serial.printf("%02X",rfrx[i]); Serial.println(); }
  746. }
  747. bool isF1=(rtype==0xF1);
  748. if(isF1){
  749. g_connF1++;
  750. uint8_t idx=3, end=rxlen+2; // walk ALL type6 TLVs (= HID report 0x45)
  751. const uint8_t* lastRep=nullptr; uint8_t lastTlen=0; // the controller PACKS several queued reports per F1;
  752. while(idx+1<end){ // draining all of them (not just the first) is what keeps
  753. uint8_t tlen=rfrx[idx], ttype=rfrx[idx+1]; // pace with the real puck — taking only [0] halved our rate.
  754. if(tlen==0) break;
  755. if(ttype==6 && tlen>=2 && rfrx[idx+2]==0x45){
  756. const uint8_t* rep=&rfrx[idx+2]; // report 0x45: [0x45][seq][buttons u32]...
  757. bool fresh=(rep[1]!=g_lastSeq); if(fresh){ g_stNew++; g_lastSeq=rep[1]; } // genuine new report vs stale poll-repeat
  758. uint32_t bb=btnsOf(rep);
  759. // cache the latest decoded frame for the Switch streamer + Xbox/Steam paths
  760. g_swBtns=bb; g_swLX=(int16_t)s16off(rep,8); g_swLY=(int16_t)s16off(rep,10);
  761. g_swRX=(int16_t)s16off(rep,12); g_swRY=(int16_t)s16off(rep,14);
  762. if(g_usbMode==2){ // SWITCH: streamed from loop() (swStream); nothing to send here
  763. } else if(g_usbMode==3){ // LIZARD: translate report 0x45 -> keyboard + mouse (forced)
  764. rfLizard(rep, &g_mouse, &g_keyboard, 0, 0);
  765. } else if(g_xbox){ // XBOX: standard gamepad + right-pad mouse
  766. rfXboxGamepad(rep); rfXboxMouse(rep);
  767. } else if(g_connSlot>=0 && g_connSlot<NSLOT){ // STEAM (puck interface)
  768. // Auto-lizard: while Steam's 0x87 heartbeat is alive -> forward gamepad report 0x45; once it
  769. // stops (Steam closed) -> emit lizard mouse(0x40)+keyboard(0x41) on the same interface, like the puck.
  770. bool steamAlive = g_steamAliveMs && (millis()-g_steamAliveMs < LIZARD_WD_MS);
  771. if(!steamAlive){
  772. rfLizard(rep, &hid[g_connSlot], &hid[g_connSlot], 0x40, 0x41);
  773. } else {
  774. uint8_t blen=tlen-1; if(blen>45)blen=45; // body after the 0x45 id byte
  775. smoothPad((uint8_t*)rep); // interpolate the 46Hz pad coords -> continuous motion for Steam
  776. // forward only FRESH reports (the real puck dedupes -> Steam gets a clean unique stream; sending
  777. // stale repeats makes Steam's velocity/smoothing stair-step). g_fwdNewOnly toggles for A/B.
  778. if((fresh || !g_fwdNewOnly) && g_slot[g_connSlot].used && hid[g_connSlot].ready())
  779. hid[g_connSlot].sendReport(0x45, rep+1, blen); // Steam/SDL: input report -> "connected"
  780. }
  781. }
  782. lastRep=rep; lastTlen=tlen;
  783. }
  784. idx+=tlen+2;
  785. }
  786. // mode-switch chord (back4 R4|L4|R5|L5 + A/X/Y), checked ONCE per frame + DEBOUNCED: a single
  787. // garbled/misaligned report must not trigger saveMode+reset (that looked like a crash under heavy input).
  788. { static uint8_t chWant=0xFF, chCnt=0;
  789. const uint32_t BACK4=TB_R4|TB_L4|TB_R5|TB_L5; uint8_t want=0xFF;
  790. if((g_swBtns&BACK4)==BACK4){ if(g_swBtns&TB_A)want=0; else if(g_swBtns&TB_X)want=1; else if(g_swBtns&TB_Y)want=2; else if(g_swBtns&TB_B)want=3; } // back4 + A=Steam X=Xbox Y=Switch B=Lizard
  791. if(want!=0xFF && want==chWant){ if(++chCnt>=12 && want!=g_usbMode){ saveMode(want); delay(40); NVIC_SystemReset(); } }
  792. else { chWant=want; chCnt=(want!=0xFF)?1:0; }
  793. }
  794. // compact stream for rf_controller_ui.py — NON-BLOCKING: skip if CDC TX is backed up (a blocking
  795. // Serial.print stalls the RF+USB loop -> jaggy input). One line/frame using the last record.
  796. if(lastRep && !g_connVerbose && Serial.availableForWrite()>110 && millis()-g_lastStream>=4){
  797. g_lastStream=millis(); Serial.print("I45 ");
  798. for(uint8_t i=0;i<lastTlen;i++)Serial.printf("%02X",lastRep[i]); Serial.println();
  799. }
  800. }
  801. if(g_connVerbose){
  802. Serial.printf("%s CRX#%lu txtype%02X ch%u len%u: ",isF1?"<<<F1":(rtype==0xF3?" F3":" rx"),
  803. (unsigned long)g_connRx,payload[0],ch,rxlen);
  804. for(uint8_t i=0;i<(rxlen+2<=66?rxlen+2:66);i++)Serial.printf("%02X",rfrx[i]); Serial.println();
  805. }
  806. } else rxlen=0;
  807. } else { g_stNoRx++; g_qosBad++; } // RX window expired with no packet at all -> turnaround/timing or controller silent
  808. NRF_RADIO->TASKS_DISABLE=1; RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  809. return rxlen;
  810. }
  811. // Drive the connected-mode sequence one step per call (from loop() while g_connOn). Camps on g_rfCh (the
  812. // host-frame channel the controller connected on); the E1 beacon (cmd 'H') runs in parallel as keepalive.
  813. static void rfConnStep(){
  814. g_connSlot=-1; for(int s=0;s<NSLOT;s++) if(g_slot[s].used){g_connSlot=s;break;}
  815. if(g_connSlot<0) return; // need a bonded slot (session established by host frame)
  816. if(g_connSt==0){ // announce HOST AWAKE: E7 00 00, a few times
  817. uint8_t p[3]={0xE7,0x00,g_e7b}; rfConnTx(g_sessCh,0x01,p,3);
  818. if(++g_connStep>=4){ g_connSt=1; g_connStep=0; Serial.println("# CONN: awake announced -> polling GET report 0x45"); }
  819. } else { // poll loop: E3 + GET report 0x45 every poll; re-assert awake periodically
  820. if((uint32_t)(micros()-g_lastPollUs) < g_pollUs) return; // CONTROLLED CADENCE: poll ~every g_pollUs,
  821. g_lastPollUs=micros(); // not as-fast-as-possible (over-polling starves replies)
  822. uint8_t ch=g_sessCh; // controller stays on the session channel (no hop)
  823. if((g_connPoll & 0x1F)==0){ uint8_t pa[3]={0xE7,0x00,g_e7b}; rfConnTx(ch,0x01,pa,3); } // re-assert awake/version
  824. if(!g_relayPend && g_testHaptic){ // buzz-hunt: synthesize a haptic relay (output 0x82 [01 01 F7])
  825. g_relayBuf[0]=0x82; g_relayBuf[1]=3; g_relayBuf[2]=0x01; g_relayBuf[3]=0x01; g_relayBuf[4]=0xF7; g_relayPend=true; g_testHaptic--;
  826. }
  827. if(g_relayPend){ // relay host cmd: [op][len][sub][rid][payload] (op/sub tunable for the buzz hunt)
  828. uint8_t rid=g_relayBuf[0], rl=g_relayBuf[1]; if(rl>18)rl=18;
  829. uint8_t p[24]; p[0]=g_relayOp; p[1]=(uint8_t)(1+rl); p[2]=g_relaySub; p[3]=rid; // len = sub-id + payload bytes
  830. for(uint8_t i=0;i<rl;i++) p[4+i]=g_relayBuf[2+i];
  831. rfConnTx(ch,0x07,p,(uint8_t)(4+rl));
  832. g_relayPend=false;
  833. } else {
  834. uint8_t p[5]={0xE3,0x02,0x01,0x45,g_getParam}; // E3 + TLV [len=02][subtype=01 GET][id=0x45][param]
  835. uint8_t s1 = (g_e3mode==1) ? (uint8_t)((((g_e3pid++)&3)<<1)|1) // cycle PID (S1 1,3,5,7), NO_ACK=1
  836. : (g_e3mode==2) ? (uint8_t)(((g_e3pid++)&3)<<1) // cycle PID (S1 0,2,4,6), NO_ACK=0
  837. : 0x07; // fixed (current default; matches captured puck poll)
  838. rfConnTx(ch,s1,p,5);
  839. }
  840. g_connPoll++;
  841. }
  842. }
  843. // SCAN-THEN-RESPOND (dongle role): listen on ch2/91A2A793 for the controller's advertisement,
  844. // and the instant one arrives, TX the host frame (its ack). Matches IBEX 'scan-before-initiating'.
  845. static uint8_t g_hostTx[48]; static int g_hostLen=0; static bool g_rfRespond=false; static uint32_t g_advCount=0;
  846. static void buildHostTx(){
  847. int slot=-1; for(int i=0;i<NSLOT;i++) if(g_slot[i].used){slot=i;break;}
  848. memset(g_hostTx,0,sizeof g_hostTx); g_hostLen=0; if(slot<0) return;
  849. // S0LEN=1 RAM layout: [0]=S0=0x12, [1]=LENGTH, raw RAM offsets validated by controller.
  850. memset(g_hostTx,0,sizeof g_hostTx);
  851. g_hostTx[0]=0x12; g_hostTx[1]=0x20; g_hostTx[3]=1; g_hostTx[5]=0xE1;
  852. memcpy(g_hostTx+6,g_slot[slot].rec,4); memcpy(g_hostTx+10,g_slot[slot].rec+4,4);
  853. g_hostTx[0xe]=2; memcpy(g_hostTx+0x12,g_rfBase,4); g_hostTx[0x16]=g_rfPrefix;
  854. g_hostLen=0x22;
  855. }
  856. static void rfRespondStart(){
  857. buildHostTx();
  858. rfConfig(2); rfSetAddr(g_rfBase,g_rfPrefix); // RX ch2 / 91A2A793
  859. NRF_RADIO->PACKETPTR=(uint32_t)rfrx;
  860. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_START_Msk;
  861. NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1; g_rfRespond=true;
  862. Serial.printf("# RESPOND mode: RX ch2/91A2A793, reply host frame (len%d)\n", g_hostLen);
  863. }
  864. static void rfRespondPoll(){
  865. if(!g_rfRespond) return;
  866. if(NRF_RADIO->EVENTS_END){
  867. NRF_RADIO->EVENTS_END=0; uint8_t len=rfrx[0];
  868. if(len && len<64){
  869. g_advCount++;
  870. // reply: TX the host frame immediately (ack)
  871. NRF_RADIO->TASKS_DISABLE=1; RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  872. NRF_RADIO->PACKETPTR=(uint32_t)g_hostTx;
  873. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_DISABLE_Msk;
  874. NRF_RADIO->EVENTS_DISABLED=0; NRF_RADIO->TASKS_TXEN=1;
  875. RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  876. Serial.printf("ADV#%lu len%u: ",(unsigned long)g_advCount,len);
  877. for(uint8_t i=0;i<=len&&i<32;i++)Serial.printf("%02X",rfrx[i]); Serial.println();
  878. // re-arm RX
  879. NRF_RADIO->PACKETPTR=(uint32_t)rfrx;
  880. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_START_Msk;
  881. NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1;
  882. }
  883. }
  884. }
  885. static void rfBeaconOnce(){
  886. uint8_t pl[48]; memset(pl,0,sizeof pl);
  887. pl[0]=0x01; pl[1]=g_seq++; pl[5]=0xE2; // dongle beacon (FUN_00027ed8 format)
  888. rfConfig(g_rfCh); rfSetAddr(g_rfBase,g_rfPrefix);
  889. if (g_s1incl) NRF_RADIO->PCNF0 |= (1u<<RADIO_PCNF0_S1INCL_Pos);
  890. // RAM: [LENGTH][ (S1 if S1INCL) ][payload]
  891. rftx[0]=0; rftx[1]=g_plen; memcpy(rftx+2,pl,g_plen);
  892. NRF_RADIO->PACKETPTR=(uint32_t)rftx;
  893. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_DISABLE_Msk;
  894. NRF_RADIO->EVENTS_DISABLED=0; NRF_RADIO->TASKS_TXEN=1;
  895. RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  896. // RX window for the controller's response
  897. NRF_RADIO->PACKETPTR=(uint32_t)rfrx; rfrx[0]=0;
  898. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk;
  899. NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1;
  900. uint32_t t0=micros(); while(!NRF_RADIO->EVENTS_END && (micros()-t0)<500){}
  901. if (NRF_RADIO->EVENTS_END){ NRF_RADIO->EVENTS_END=0;
  902. bool crcok=NRF_RADIO->CRCSTATUS&1; uint8_t len=rfrx[0];
  903. if (len && len<64){ g_rfRxCount++;
  904. Serial.printf("RESP#%lu crc%d len%u: ",(unsigned long)g_rfRxCount,crcok,len);
  905. for(uint8_t i=0;i<=len&&i<40;i++)Serial.printf("%02X ",rfrx[i]); Serial.println();
  906. }
  907. }
  908. NRF_RADIO->TASKS_DISABLE=1; RWAIT_DISABLED(); NRF_RADIO->EVENTS_DISABLED=0;
  909. }
  910. // ===================== WebUSB config channel =====================
  911. // Browser panel <-> firmware, binary framed.
  912. // host->dev: 0x01 GET -> reply status blob
  913. // 0x02 <field> <value> SET one byte field (1=mDiv 2=mFric 3=padSmooth 4=abSwap 5..8=back[0..3] 9=pollU100 10=e7b 11=relayOp 12=relaySub 13=testHaptic)
  914. // 0x03 <mode> switch USB mode (0=steam 1=xbox 2=switch): persist + reboot
  915. // dev->host: 0xA5 <len> <payload> payload = [ver=1][mode][mDiv][mFric][padSmooth][abSwap]
  916. // [back0..3][connSlot(0xFF=none)][linkUp][f1ps_lo][f1ps_hi][pollU100][newps_lo][newps_hi][e7b][relayOp][relaySub]
  917. #define WB_PAYLEN 23
  918. static void webusbSendBlob(){
  919. if(!usb_web.connected()) return;
  920. bool up = (g_connSlot>=0 && (millis()-g_connReplyMs) < 300);
  921. uint8_t p[2+WB_PAYLEN];
  922. p[0]=0xA5; p[1]=WB_PAYLEN;
  923. p[2]=1; // protocol version
  924. p[3]=g_usbMode; p[4]=(uint8_t)g_mDiv; p[5]=(uint8_t)g_mFric; p[6]=(uint8_t)g_padSmooth; p[7]=g_abSwap;
  925. p[8]=g_back[0]; p[9]=g_back[1]; p[10]=g_back[2]; p[11]=g_back[3];
  926. p[12]=(g_connSlot>=0)?(uint8_t)g_connSlot:0xFF;
  927. p[13]=up?1:0;
  928. p[14]=(uint8_t)g_f1ps; p[15]=(uint8_t)(g_f1ps>>8);
  929. p[16]=(uint8_t)(g_pollUs/100);
  930. p[17]=(uint8_t)g_newps; p[18]=(uint8_t)(g_newps>>8);
  931. p[19]=g_e7b;
  932. p[20]=g_relayOp; p[21]=g_relaySub; p[22]=g_fwdNewOnly; p[23]=g_qos; p[24]=g_persistMode?1:0;
  933. usb_web.write(p,sizeof p); usb_web.flush();
  934. }
  935. static void webusbPoll(){
  936. static uint8_t buf[16]; static uint8_t n=0;
  937. while(usb_web.available()){
  938. int c=usb_web.read(); if(c<0) break;
  939. if(n<sizeof buf) buf[n++]=(uint8_t)c;
  940. // process complete commands from the front of buf
  941. for(;;){
  942. if(n==0) break;
  943. uint8_t op=buf[0]; uint8_t need = (op==0x02)?3 : (op==0x03)?2 : (op==0x01)?1 : 1;
  944. if(op!=0x01 && op!=0x02 && op!=0x03){ // resync: drop one byte
  945. memmove(buf,buf+1,--n); continue;
  946. }
  947. if(n<need) break; // wait for more bytes
  948. if(op==0x01){ webusbSendBlob(); }
  949. else if(op==0x02){
  950. uint8_t f=buf[1], v=buf[2];
  951. bool persist=true; // most fields persist; the rate knobs are session-only (locked default on reboot)
  952. switch(f){
  953. case 1: g_mDiv = v<4?4:v; break;
  954. case 2: g_mFric = v>99?99:v; break;
  955. case 3: g_padSmooth = v<5?5:(v>100?100:v); break;
  956. case 4: g_abSwap = v?1:0; break;
  957. case 5: case 6: case 7: case 8: g_back[f-5]=v; break;
  958. case 9: g_pollUs = (uint32_t)(v<1?1:v)*100; persist=false; break; // pollU100 -> us. SESSION-ONLY: poll rate is locked; this tunes it live but reverts to 800 on reboot.
  959. case 10: g_e7b = v?1:0; break; // E7 protocol-version B-byte (experimental v1 fast)
  960. case 11: g_relayOp = v; break; // haptic-relay opcode (buzz hunt)
  961. case 12: g_relaySub = v; break; // haptic-relay sub-type (buzz hunt)
  962. case 13: g_testHaptic = v?v:40; break; // inject v test haptics (0->40)
  963. case 14: g_fwdNewOnly = v?1:0; break; // Steam: forward only fresh reports (dedupe)
  964. case 15: g_qos = v?1:0; g_hopIdx=0; g_qosBad=0; g_qosCheckMs=millis(); break; // QoS adaptive channel hopping
  965. case 16: g_persistMode = v?true:false; break; // persist last mode across reboots (else always boot Steam)
  966. }
  967. if(persist) saveCfg();
  968. webusbSendBlob();
  969. } else if(op==0x03){
  970. uint8_t m=buf[1]; if(m<=3){ webusbSendBlob(); usb_web.flush(); saveMode(m); delay(40); NVIC_SystemReset(); }
  971. }
  972. memmove(buf,buf+need,n-need); n-=need;
  973. }
  974. }
  975. }
  976. // CDC commands: operational controls only.
  977. static void rfSerialPoll(){
  978. static char line[24]; static uint8_t li=0;
  979. while (Serial.available()){
  980. char c=Serial.read();
  981. if (c=='\n'||c=='\r'){ line[li]=0;
  982. if (line[0]=='l') rfListenStart();
  983. else if (line[0]=='B'){ g_rfBeacon=!g_rfBeacon; g_rfListen=false; Serial.printf("# beacon %s ch%u\n",g_rfBeacon?"ON":"off",g_rfCh); }
  984. else if (line[0]=='L'){ g_plen=strtol(line+1,0,16); Serial.printf("# plen=%02X\n",g_plen); }
  985. else if (line[0]=='1'){ g_s1incl=!g_s1incl; Serial.printf("# s1incl=%u\n",g_s1incl); }
  986. else if (line[0]=='H'){ g_rfHost=!g_rfHost; g_rfBeacon=false; g_rfListen=false; g_rfCh=2;
  987. Serial.printf("# HOSTFRAME beacon %s base %02X%02X%02X%02X (slot0 uuids %02X%02X%02X%02X/%02X%02X%02X%02X)\n",
  988. g_rfHost?"ON":"off",g_rfBase[0],g_rfBase[1],g_rfBase[2],g_rfBase[3],
  989. g_slot[0].rec[0],g_slot[0].rec[1],g_slot[0].rec[2],g_slot[0].rec[3],
  990. g_slot[0].rec[4],g_slot[0].rec[5],g_slot[0].rec[6],g_slot[0].rec[7]); }
  991. else if (line[0]=='C'){ g_rfBeacon=g_rfListen=g_rfHost=false; rfRespondStart(); }
  992. else if (line[0]=='s'){ g_rfListen=false; g_rfBeacon=false; g_rfHost=false; g_rfRespond=false; NRF_RADIO->TASKS_DISABLE=1; Serial.println("# RF off"); }
  993. else if (line[0]=='c'){ g_rfCh=atoi(line+1); Serial.printf("# ch=%u\n",g_rfCh); if(g_rfListen) rfListenStart(); }
  994. else if (line[0]=='p'){ g_rfPrefix=strtol(line+1,0,16); Serial.printf("# prefix=%02X\n",g_rfPrefix); if(g_rfListen) rfListenStart(); }
  995. else if (line[0]=='i'){ g_crcinit=strtoul(line+1,0,16); Serial.printf("# crcinit=%06lX\n",(unsigned long)g_crcinit); }
  996. else if (line[0]=='w'){ g_whiteiv=atoi(line+1); Serial.printf("# whiteiv=%u\n",g_whiteiv); }
  997. else if (line[0]=='0'){ g_pcnf0=strtoul(line+1,0,16); Serial.printf("# pcnf0=%08lX\n",(unsigned long)g_pcnf0); }
  998. else if (line[0]=='2'){ g_pcnf1=strtoul(line+1,0,16); Serial.printf("# pcnf1=%08lX\n",(unsigned long)g_pcnf1); }
  999. else if (line[0]=='M'){ g_mode=strtoul(line+1,0,16); Serial.printf("# mode=%u (3=Ble1M,4=Ble2M,0/2=Nrf)\n",g_mode); }
  1000. else if (line[0]=='I'){ g_crcinit=strtoul(line+1,0,16); Serial.printf("# crcinit=%08lX\n",(unsigned long)g_crcinit); }
  1001. else if (line[0]=='P'){ g_crcpoly=strtoul(line+1,0,16); Serial.printf("# crcpoly=%lX\n",(unsigned long)g_crcpoly); }
  1002. else if (line[0]=='N'){ g_crccnf=strtoul(line+1,0,16); Serial.printf("# crccnf=%X\n",g_crccnf); }
  1003. else if (line[0]=='T'){ g_statlen=strtoul(line+1,0,16); g_pcnf1=0; Serial.printf("# statlen=%02X\n",g_statlen); }
  1004. else if (line[0]=='k'){ g_connOn=!g_connOn; if(g_connOn){ g_connSt=0; g_connStep=0; g_connPoll=0; g_connF1=0; } Serial.printf("# CONN mode %s: E7-awake[00 00] -> E3+GET-report-0x45 poll on ch%u (param=%02X). F1 seen=%lu\n",g_connOn?"ON":"off",g_rfCh,g_getParam,(unsigned long)g_connF1); }
  1005. else if (line[0]=='q'){ g_getParam = g_getParam ? 0x00 : 0x2D; Serial.printf("# GET-0x45 param=%02X\n",g_getParam); }
  1006. else if (line[0]=='v'){ g_connVerbose=!g_connVerbose; Serial.printf("# conn verbose %s (F1 seen=%lu)\n",g_connVerbose?"ON":"off",(unsigned long)g_connF1); }
  1007. else if (line[0]=='C'){ g_sessCh=strtoul(line+1,0,10); Serial.printf("# session channel=%u (re-pair/reconnect to apply)\n",g_sessCh); }
  1008. else if (line[0]=='U'){ g_pollUs=strtoul(line+1,0,10); if(g_pollUs<50)g_pollUs=50; Serial.printf("# poll interval=%lu us (~%lu/s) SESSION-ONLY (locked default 800us on reboot)\n",(unsigned long)g_pollUs,(unsigned long)(1000000/g_pollUs)); }
  1009. else if (line[0]=='E'){ g_mDiv=strtoul(line+1,0,10); if(g_mDiv<4)g_mDiv=4; saveCfg(); Serial.printf("# xbox-mouse sensitivity divisor=%d (lower=faster)\n",g_mDiv); }
  1010. else if (line[0]=='n'){ g_lizDiv=strtoul(line+1,0,10); if(g_lizDiv<8)g_lizDiv=8; Serial.printf("# lizard-mouse sensitivity divisor=%d (higher=slower)\n",g_lizDiv); }
  1011. else if (line[0]=='m'){ g_lizFric=strtoul(line+1,0,10); if(g_lizFric>99)g_lizFric=99; Serial.printf("# lizard-mouse friction=%d%% (lower=less glide/crisper)\n",g_lizFric); }
  1012. else if (line[0]=='F'){ g_mFric=strtoul(line+1,0,10); if(g_mFric>99)g_mFric=99; saveCfg(); Serial.printf("# xbox-mouse friction=%d%% (higher=more glide/momentum)\n",g_mFric); }
  1013. else if (line[0]=='D'){ g_padSmooth=strtoul(line+1,0,10); if(g_padSmooth<5)g_padSmooth=5; if(g_padSmooth>100)g_padSmooth=100; saveCfg(); Serial.printf("# steam pad smoothing alpha=%d%% (100=off, lower=smoother/laggier)\n",g_padSmooth); }
  1014. else if (line[0]=='W'){ g_abSwap=!g_abSwap; saveCfg(); Serial.printf("# A/B + X/Y swap %s (Nintendo layout)\n",g_abSwap?"ON":"off"); }
  1015. else if (line[0]=='K'){ int i=line[1]-'0'; uint8_t code=strtoul(line+2,0,10); if(i>=0&&i<4){ g_back[i]=code; saveCfg(); Serial.printf("# back[%d] (%s) -> code %u [0=none 1=A 2=B 3=X 4=Y 5=LB 6=RB 7=L3 8=R3 9=Back 10=Start 11=Guide]\n",i,(const char*[]){"L4","R4","L5","R5"}[i],code); } else Serial.println("# usage: K<0-3> <code> (0=L4 1=R4 2=L5 3=R5)"); }
  1016. else if (line[0]=='J'){ char* sp=0; uint8_t id=strtoul(line+1,&sp,0); uint16_t val=sp?strtoul(sp,0,0):0; // inject SET-SETTINGS to controller: report 0x87 [id][val u16 LE]
  1017. g_relayBuf[0]=0x87; g_relayBuf[1]=3; g_relayBuf[2]=id; g_relayBuf[3]=val&0xFF; g_relayBuf[4]=val>>8; g_relayN=5; g_relayPend=true;
  1018. Serial.printf("# queued SET-SETTINGS id=0x%02X val=%u (relay 0x87) — watch new=/s\n",id,val); }
  1019. else if (line[0]=='V'){ g_e7b=strtoul(line+1,0,10); Serial.printf("# E7 protocol-version B-byte=%u (0=awake/slow, 1=try v1-fast; watch new=/s; revert if input stops)\n",g_e7b); }
  1020. else if (line[0]=='O'){ g_relayOp=strtoul(line+1,0,16); Serial.printf("# relay opcode=%02X\n",g_relayOp); }
  1021. else if (line[0]=='o'){ g_relaySub=strtoul(line+1,0,16); Serial.printf("# relay sub-type=%02X\n",g_relaySub); }
  1022. else if (line[0]=='h'){ uint8_t nc=strtoul(line+1,0,10); Serial.printf("# HOP %u->%u (advertise on current ch, then poll new). Watch F1=/s\n",g_sessCh,nc); rfHopTo(nc); }
  1023. else if (line[0]=='z'){ g_qos=!g_qos; g_hopIdx=0; g_qosBad=0; g_qosCheckMs=millis(); Serial.printf("# QoS adaptive channel hopping %s (candidates 18,46,76,22,68)\n",g_qos?"ON":"off"); }
  1024. else if (line[0]=='r'){ g_rxWin=strtoul(line+1,0,10); if(g_rxWin<150)g_rxWin=150; Serial.printf("# poll RX-window=%lu us (poll rate caps ~%lu/s)\n",(unsigned long)g_rxWin,(unsigned long)(1000000/g_rxWin)); }
  1025. else if (line[0]=='e'){ g_e3mode=strtoul(line+1,0,10); Serial.printf("# E3 poll PID mode=%u (0=fixed07, 1=cyclePID+noack1, 2=cyclePID+noack0) - watch new=/s\n",g_e3mode); }
  1026. else if (line[0]=='t'){ uint8_t n=line[1]?strtoul(line+1,0,10):40; // inject n test haptics (output 0x82 [01 01 F7]) over the relay
  1027. g_testHaptic=n; Serial.printf("# test-haptic burst x%u queued (relay 0x82 01 01 F7 via op=%02X sub=%02X)\n",n,g_relayOp,g_relaySub); }
  1028. else if (line[0]=='j'){ g_connType=strtoul(line+1,0,16); Serial.printf("# connType=%02X\n",g_connType); }
  1029. else if (line[0]=='Q'){ g_connLen=strtoul(line+1,0,16); Serial.printf("# connLen=%02X\n",g_connLen); }
  1030. else if (line[0]=='A'){ g_balen=strtoul(line+1,0,16); g_pcnf1=0; Serial.printf("# balen=%u\n",g_balen); }
  1031. else if (line[0]=='a'){ uint32_t v=strtoul(line+1,0,16); g_rfBase[0]=v>>24;g_rfBase[1]=v>>16;g_rfBase[2]=v>>8;g_rfBase[3]=v; Serial.printf("# base=%08lX\n",(unsigned long)v); if(g_rfListen) rfListenStart(); }
  1032. else if (line[0]=='b'){ for(int i=0;i<NSLOT;i++) if(g_slot[i].used){Serial.printf("slot%d ",i); for(int j=0;j<24;j++)Serial.printf("%02X",g_slot[i].rec[j]); Serial.println();} }
  1033. li=0;
  1034. } else if (li<sizeof line-1) line[li++]=c;
  1035. }
  1036. }
  1037. void setup() {
  1038. genSerial();
  1039. InternalFS.begin();
  1040. loadCfg(); g_xbox = (g_usbMode != 0); // load persisted config + decide USB presentation BEFORE registering interfaces
  1041. USBDevice.setSerialDescriptor(g_unit);
  1042. if (g_usbMode == 1) { // XBOX presentation: XInput gamepad + mouse only (no puck)
  1043. USBDevice.setID(0x045E, 0x028E); // Microsoft Xbox 360 wired controller (so Steam/SDL bind it as Xbox 360)
  1044. USBDevice.setVersion(0x0200); USBDevice.setDeviceVersion(0x0114);
  1045. USBDevice.setManufacturerDescriptor("Microsoft");
  1046. USBDevice.setProductDescriptor("Controller");
  1047. g_xinput.setStringDescriptor("Controller"); g_xinput.begin(); // XInput vendor interface (FF/5D/01)
  1048. g_mouse.setReportDescriptor(MOUSE_HID_DESC, sizeof MOUSE_HID_DESC); g_mouse.setPollInterval(1); g_mouse.begin();
  1049. } else if (g_usbMode == 2) { // SWITCH presentation: HORIPAD/Pokkén pad (console-compatible)
  1050. USBDevice.setID(0x0F0D, 0x0092); // HORI Pokkén Tournament Pro Pad (accepted by a real Switch console)
  1051. USBDevice.setVersion(0x0200); USBDevice.setDeviceVersion(0x0200);
  1052. USBDevice.setManufacturerDescriptor("HORI CO.,LTD.");
  1053. USBDevice.setProductDescriptor("POKKEN CONTROLLER");
  1054. g_switch.setReportDescriptor(SWITCH_HID_DESC, sizeof SWITCH_HID_DESC);
  1055. g_switch.setPollInterval(8); // 8ms (~125Hz); we stream the 8-byte report (no handshake needed)
  1056. g_switch.begin();
  1057. } else if (g_usbMode == 3) { // LIZARD presentation: plain desktop keyboard + mouse (driverless)
  1058. USBDevice.setID(0x28DE, 0x1102); // Valve VID, distinct PID; KB+mouse are boot-protocol so the IDs don't matter functionally
  1059. USBDevice.setManufacturerDescriptor("Valve Software");
  1060. USBDevice.setProductDescriptor("Steam Controller (Desktop)");
  1061. g_mouse.setReportDescriptor(MOUSE_HID_DESC, sizeof MOUSE_HID_DESC); g_mouse.setPollInterval(1); g_mouse.begin();
  1062. g_keyboard.setReportDescriptor(KEYBOARD_HID_DESC, sizeof KEYBOARD_HID_DESC); g_keyboard.setPollInterval(1); g_keyboard.begin();
  1063. } else { // STEAM presentation: the 4 puck slot interfaces
  1064. USBDevice.setID(0x28DE, 0x1304);
  1065. USBDevice.setManufacturerDescriptor("Valve Software");
  1066. USBDevice.setProductDescriptor("Steam Controller Puck");
  1067. for (int i = 0; i < NSLOT; i++) {
  1068. hid[i].setReportDescriptor(PUCK_HID_DESC, sizeof PUCK_HID_DESC);
  1069. hid[i].setReportCallback(GETCB[i], SETCB[i]);
  1070. hid[i].setPollInterval(1); // 1ms USB poll (was default 10ms = 100/s cap -> choppy)
  1071. hid[i].begin();
  1072. }
  1073. }
  1074. usb_web.begin(); // config panel (present in every mode); no landing page -> no Chrome auto-notification
  1075. Serial.begin(115200);
  1076. while (!USBDevice.mounted()) delay(10);
  1077. loadBonds();
  1078. Serial.printf("# openpuck up: unit=%s board=%s, mode=%s\n", g_unit, g_board, g_usbMode==1?"XBOX(gamepad+mouse)":g_usbMode==2?"SWITCH(pro controller)":g_usbMode==3?"LIZARD(keyboard+mouse)":"STEAM(puck)");
  1079. }
  1080. void loop() {
  1081. if (g_dirty) { g_dirty = false; saveBonds(); }
  1082. webusbPoll();
  1083. swStream(); // SWITCH mode: stream 0x30 input reports once the host enabled mode 0x30
  1084. rfSerialPoll();
  1085. rfPoll();
  1086. rfRespondPoll();
  1087. if (g_rfBeacon && millis() - g_lastBeacon >= 5) { g_lastBeacon = millis(); rfBeaconOnce(); }
  1088. // Host-frame beacon: sent continuously, INCLUDING while connected. The controller uses the periodic E1
  1089. // (the real puck's per-hop-cycle announce) to stay synced and keep answering polls at full rate;
  1090. // suppressing it drops the reply rate from ~210/s to ~38/s. Paused only during the post-disconnect
  1091. // cooldown so a controller that's powering off isn't immediately re-woken/reconnected.
  1092. if (g_rfHost && millis()-g_connCooldown > 2500) {
  1093. bool connNow = (g_connSlot>=0 && millis()-g_connReplyMs < 300);
  1094. // session keepalive on the clean channel: every loop while connecting (fast), every 25ms once connected
  1095. // (every-loop beaconing also hammers ch52 and steals reply slots from the poll)
  1096. if (millis()-g_lastSessBeacon >= (connNow ? 25u : 0u)) { g_lastSessBeacon=millis(); g_rfCh=g_sessCh; for (int s=0;s<NSLOT;s++) rfHostFrameOnce(s); }
  1097. // discovery beacon on ch2 (where a searching controller looks): every loop when down, occasionally when up
  1098. if (millis()-g_lastDisc >= (connNow ? 200u : 0u)) { g_lastDisc=millis(); g_rfCh=2; for (int s=0;s<NSLOT;s++) rfHostFrameOnce(s); }
  1099. }
  1100. if (g_connOn && millis()-g_connCooldown > 2500) { rfConnStep(); } // connected-mode: poll controller, read input
  1101. // Xbox-mode rumble -> relay the haptic (0x82 [01 01 gain]) to the controller while the host commands it,
  1102. // re-queued ~40/s like Steam's glide haptic so it sustains. gain scaled by the XInput motor strength.
  1103. if (g_usbMode==1 && g_rumble && millis()-g_rumbleMs > RUMBLE_STUCK_MS) g_rumble=0; // watchdog: lost stop packet -> release the held rumble instead of buzzing forever
  1104. if (g_usbMode==1 && g_rumble && !g_relayPend && g_connSlot>=0) {
  1105. static unsigned long lastRumble=0;
  1106. if (millis()-lastRumble>=25) { lastRumble=millis();
  1107. uint8_t gain = g_rumble<0x30?0x30:g_rumble; // floor so low rumble is still feelable
  1108. g_relayBuf[0]=0x82; g_relayBuf[1]=3; g_relayBuf[2]=0x01; g_relayBuf[3]=0x01; g_relayBuf[4]=gain; g_relayPend=true;
  1109. }
  1110. }
  1111. // Steam-mode haptic watchdog: glide haptics are one-shot relays, but if Steam's stop is lost the controller
  1112. // can latch buzzing. If a 0x82 haptic was active and the host's stream has gone quiet, send one explicit off.
  1113. if (!g_xbox && g_haptic82On && millis()-g_haptic82Ms > HAPTIC_QUIET_MS && !g_relayPend && g_connSlot>=0) {
  1114. g_haptic82On=false;
  1115. g_relayBuf[0]=0x82; g_relayBuf[1]=3; g_relayBuf[2]=0x01; g_relayBuf[3]=0x01; g_relayBuf[4]=0x00; g_relayN=5; g_relayPend=true; // 0x82 [01 01 00] = haptic off
  1116. }
  1117. // QoS: if the current channel is degrading (crcfail+noRx), hop to the next clean candidate (conservative).
  1118. if (g_qos && g_connSlot>=0 && millis()-g_qosCheckMs>=600) {
  1119. uint16_t bad=g_qosBad; g_qosBad=0; g_qosCheckMs=millis();
  1120. if (bad>20 && millis()-g_qosLastHopMs>2000) {
  1121. for(int k=0;k<(int)sizeof g_hopCand;k++){ g_hopIdx=(g_hopIdx+1)%(sizeof g_hopCand); if(g_hopCand[g_hopIdx]!=g_sessCh) break; }
  1122. if(Serial.availableForWrite()>60) Serial.printf("# QoS: ch%u bad=%u -> hop ch%u\n",g_sessCh,bad,g_hopCand[g_hopIdx]);
  1123. rfHopTo(g_hopCand[g_hopIdx]); g_qosLastHopMs=millis();
  1124. }
  1125. }
  1126. // USB connection presentation (like the real dongle): report 0x79 = connection state (01=disc, 02=conn),
  1127. // edge-triggered, + periodic 0x7B status. Live-captured: this is what Steam reads to mark the controller
  1128. // connected. Without it Steam shows disconnected even though 0x45 input is streaming.
  1129. if(!g_xbox){ static bool usbConn=false; static unsigned long last79=0, last7B=0; // puck-only USB presentation
  1130. bool conn = (g_connSlot>=0 && millis()-g_connReplyMs < 300);
  1131. if (g_connSlot>=0 && g_connSlot<NSLOT && hid[g_connSlot].ready()){
  1132. // 0x79 connection state: on edge AND repeated every 750ms while connected (Steam can miss the edge,
  1133. // and the 0x45 stream can starve the endpoint at the transition instant -> resend until it sticks).
  1134. if (conn!=usbConn || (conn && millis()-last79>=750)){
  1135. uint8_t st=conn?0x02:0x01; hid[g_connSlot].sendReport(0x79,&st,1); usbConn=conn; last79=millis();
  1136. } else if (conn && millis()-last7B>=2000){
  1137. static const uint8_t s7b[12]={0xF7,0x01,0x89,0x00,0x00,0x00,0x03,0x00,0xDD,0x00,0x3A,0x02};
  1138. hid[g_connSlot].sendReport(0x7B,s7b,12); last7B=millis();
  1139. }
  1140. } else if(!conn) usbConn=false;
  1141. }
  1142. if (g_connOn && millis()-g_stMs>=1000){ g_f1ps=g_stF1; g_newps=g_stNew; if(Serial.availableForWrite()>70) Serial.printf("# stat polls=%lu/s F1=%lu/s new=%lu/s F3=%lu/s(v%d) e7b=%u crcfail=%lu noRx=%lu slot=%d\n",(unsigned long)g_stPoll,(unsigned long)g_stF1,(unsigned long)g_stNew,(unsigned long)g_stF3,(int8_t)g_connF3v,g_e7b,(unsigned long)g_stCrc,(unsigned long)g_stNoRx,g_connSlot); g_stPoll=0; g_stF1=0; g_stNew=0; g_stF3=0; g_stCrc=0; g_stNoRx=0; g_stMs=millis(); }
  1143. if (g_rfRespond && millis() - g_lastHop >= 25) { // scan the 3 BLE adv channels w/ matched whitening
  1144. g_lastHop = millis();
  1145. static const uint8_t advf[3]={2,26,80}, advw[3]={37,38,39}; static uint8_t ai=0; ai=(ai+1)%3;
  1146. g_rfCh=advf[ai]; g_whiteiv=advw[ai];
  1147. rfConfig(g_rfCh); rfSetAddr(g_rfBase,g_rfPrefix); NRF_RADIO->PACKETPTR=(uint32_t)rfrx;
  1148. NRF_RADIO->SHORTS=RADIO_SHORTS_READY_START_Msk|RADIO_SHORTS_END_START_Msk; NRF_RADIO->EVENTS_END=0; NRF_RADIO->TASKS_RXEN=1;
  1149. }
  1150. }