puck_hid.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309
  1. #include "puck_hid.h"
  2. #include "bonds.h"
  3. #include "config.h"
  4. #include "identity.h"
  5. #include "haptics.h"
  6. #include "rf_link.h"
  7. #include "triton.h"
  8. #include "mode_lizard.h"
  9. #include <Adafruit_TinyUSB.h>
  10. #include <Arduino.h>
  11. #include <string.h>
  12. uint8_t g_fwdNewOnly = 1;
  13. SteamPuckController g_steamPuck;
  14. // ---- cloned puck HID report descriptor (verbatim): mouse(0x40)+keyboard(0x41)+vendor(FF00) with the 63-byte
  15. // FEATURE command reports on report id 1/2. Each of the 4 interfaces uses this. ----
  16. static const uint8_t PUCK_HID_DESC[] = {
  17. 0x05,0x01,0x09,0x02,0xA1,0x01,0x85,0x40,0x09,0x01,0xA1,0x00,0x05,0x09,0x19,0x01,
  18. 0x29,0x02,0x15,0x00,0x25,0x01,0x75,0x01,0x95,0x02,0x81,0x02,0x75,0x06,0x95,0x01,
  19. 0x81,0x01,0x05,0x01,0x09,0x30,0x09,0x31,0x15,0x81,0x25,0x7F,0x75,0x08,0x95,0x02,
  20. 0x81,0x06,0x95,0x01,0x09,0x38,0x81,0x06,0x05,0x0C,0x0A,0x38,0x02,0x95,0x01,0x81,
  21. 0x06,0xC0,0xC0,0x05,0x01,0x09,0x06,0xA1,0x01,0x85,0x41,0x05,0x07,0x19,0xE0,0x29,
  22. 0xE7,0x15,0x00,0x25,0x01,0x75,0x01,0x95,0x08,0x81,0x02,0x81,0x01,0x19,0x00,0x29,
  23. 0x65,0x15,0x00,0x25,0x65,0x75,0x08,0x95,0x06,0x81,0x00,0xC0,0x06,0x00,0xFF,0x09,
  24. 0x01,0xA1,0x01,0x85,0x42,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x35,0x09,0x42,
  25. 0x81,0x02,0x85,0x44,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x05,0x09,0x44,0x81,
  26. 0x02,0x85,0x79,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x01,0x09,0x79,0x81,0x02,
  27. 0x85,0x43,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x0E,0x09,0x43,0x81,0x02,0x85,
  28. 0x7B,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x0C,0x09,0x7B,0x81,0x02,0x85,0x45,
  29. 0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x2D,0x09,0x45,0x81,0x02,0x85,0x80,0x15,
  30. 0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x09,0x09,0x80,0x91,0x02,0x85,0x81,0x15,0x00,
  31. 0x26,0xFF,0x00,0x75,0x08,0x95,0x07,0x09,0x81,0x91,0x02,0x85,0x82,0x15,0x00,0x26,
  32. 0xFF,0x00,0x75,0x08,0x95,0x03,0x09,0x82,0x91,0x02,0x85,0x83,0x15,0x00,0x26,0xFF,
  33. 0x00,0x75,0x08,0x95,0x09,0x09,0x83,0x91,0x02,0x85,0x84,0x15,0x00,0x26,0xFF,0x00,
  34. 0x75,0x08,0x95,0x08,0x09,0x84,0x91,0x02,0x85,0x85,0x15,0x00,0x26,0xFF,0x00,0x75,
  35. 0x08,0x95,0x03,0x09,0x85,0x91,0x02,0x85,0x86,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,
  36. 0x95,0x03,0x09,0x86,0x91,0x02,0x85,0x87,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,
  37. 0x3F,0x09,0x87,0x91,0x02,0x85,0x89,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x3F,
  38. 0x09,0x89,0x91,0x02,0x85,0x88,0x15,0x00,0x26,0xFF,0x00,0x75,0x08,0x95,0x3F,0x09,
  39. 0x88,0x91,0x02,0x85,0x01,0x95,0x3F,0x09,0x01,0xB1,0x02,0x85,0x02,0x95,0x3F,0x09,
  40. 0x01,0xB1,0x02,0xC0,
  41. 0x05,0x0C,0x09,0x01,0xA1,0x01,0x85,0x03,0x15,0x00,0x25,0x01,0x09,0xE9,0x09,0xEA,
  42. 0x75,0x01,0x95,0x02,0x81,0x02,0x75,0x06,0x95,0x01,0x81,0x01,0xC0
  43. };
  44. static Adafruit_USBD_HID hid[NSLOT];
  45. // ===================== seamless LIZARD decision =====================
  46. // Steam, while running, re-sends settings report 0x87 (lizard-off) every ~3s as a heartbeat (captured on HW),
  47. // and ANY OUTPUT report likewise stamps g_steamAliveMs. When the heartbeat stops we fall back to lizard, so the
  48. // controller drives desktop keyboard+mouse whenever Steam isn't running. MODE_LIZARD forces lizard always.
  49. static unsigned long g_steamAliveMs = 0; // millis of last Steam OUTPUT/settings write; 0 at boot => lizard until Steam appears
  50. // Fall back to lizard this long after Steam's ~3s settings heartbeat stops. Keep >2x the cadence (7s): the
  51. // haptic relay is gated by !lizardActive(), so a shorter window lets a single jittered/delayed heartbeat flip
  52. // lizard mid-session while Steam is still running -> a haptic that arrives in that window gets gated out
  53. // (dropped/missing haptic). 7s tolerates one missed beat. (This matches main; do not shorten without a lizard
  54. // trigger that doesn't depend on the heartbeat timeout.)
  55. #define LIZARD_WD_MS 7000u
  56. static bool g_autoLizard = true; // master switch; false => Steam mode always forwards 0x45
  57. // Single source of truth, shared by the USB input path AND the haptic relay gate: if we ever relay a 0x82 to
  58. // the controller while presenting lizard (Steam isn't reading 0x45 back), Steam loops the same haptic -> buzz.
  59. static inline bool steamDrivingGamepad(){ return g_steamAliveMs && (millis()-g_steamAliveMs < LIZARD_WD_MS); }
  60. static inline bool lizardActive(){ return modeIsPuck(g_usbMode) && (g_usbMode==MODE_LIZARD || (g_autoLizard && !steamDrivingGamepad())); }
  61. // Right after the host resumes from suspend, MUTE input forwarding briefly. Otherwise the controller's input
  62. // in that instant (a trackpad click/trigger, or residual button state from the wake gesture) gets forwarded
  63. // as a real click/keypress into the just-woken desktop -- which was activating the highlighted Start tile
  64. // (Edge) and closing the Start menu on every wake. Set when task() sees the suspended->active transition.
  65. static unsigned long g_resumeMs = 0;
  66. #define POST_RESUME_MUTE_MS 1500u
  67. // ===================== puck feature command channel =====================
  68. // `slot` is the interface index (interface N == bond slot N).
  69. static void handleSet(int slot, uint8_t rid, hid_report_type_t type, uint8_t const *b, uint16_t n) {
  70. if (type == HID_REPORT_TYPE_OUTPUT) { // Steam OUTPUT reports 0x80-0x89. The haptic/actuator reports (0x80-0x86)
  71. // are relayed to the controller, and ONLY when they arrive on the CONNECTED slot's interface. We have one
  72. // controller but expose 4 puck slots, and a report aimed at a DIFFERENT slot made the controller buzz at
  73. // random -> that is what the slot gate below fixes. (This used to also clamp to 0x82-ONLY, which silently
  74. // dropped the ping / grip / test haptics: those ride other report IDs such as 0x85/0x86, so they never
  75. // reached the controller.) The 63-byte settings/config reports 0x87/0x88/0x89 are NOT haptics and are not
  76. // pushed here -- 0x87 (lizard-off/settings) reaches the controller through the feature-0x01 passthrough path.
  77. if (rid >= 0x80 && rid <= 0x89) {
  78. hapLogAdd((uint8_t)slot, rid, b, n); // capture ALL OUTPUT reports (even un-relayed) for the 'H' dump
  79. g_steamAliveMs = millis(); // ANY Steam OUTPUT report (not just the 0x87 heartbeat) means Steam is present and
  80. // driving -> leave lizard for gamepad NOW, so a haptic that arrives before the
  81. // first 0x87 doesn't get relayed while we're still presenting lizard (-> buzz loop).
  82. }
  83. // Post-resume mute also gates haptics: while onReport45 is muted Steam reads NO 0x45 back, which is the
  84. // exact condition under which Steam loops the same haptic command (-> connect/wake buzz loop). Same logic
  85. // as the lizard gate; the mute is short (1.5s), so a dropped wake chime is the cheap end of the trade.
  86. bool muted = g_resumeMs && millis()-g_resumeMs < POST_RESUME_MUTE_MS;
  87. if (rid >= 0x80 && rid <= 0x86 && n >= 1 && hapticRelaySlotOk(slot) && !lizardActive() && !muted) { // wrap as a SET sub-TLV like the report-01 path
  88. if (!haptic82Blocked()) {
  89. relayEnqueue(rid, b, (uint8_t)(n > RELAY_MAXP ? RELAY_MAXP : n));
  90. // Track on/off from what was actually RELAYED (= the controller's believed state), not from every host
  91. // report: a BLOCKED stop must leave "on" set (the controller may be latched -> reconnect stop-burst),
  92. // and a blocked ON must not set it (nothing reached the controller -> no spurious burst clicks).
  93. if (rid == 0x82) haptic82HostReport(b, n);
  94. }
  95. }
  96. if (Serial.availableForWrite() > 80) { // log so we can see what Steam actually sends (e.g. glide haptics)
  97. Serial.printf("# OUT if%d rid=%02X n=%u:", slot, rid, n);
  98. for (uint16_t i = 0; i < n && i < 14; i++) Serial.printf(" %02X", b[i]);
  99. Serial.println();
  100. }
  101. return;
  102. }
  103. if (type != HID_REPORT_TYPE_FEATURE || n < 1) return;
  104. Slot &S = g_slot[slot];
  105. uint8_t cmd = b[0], len = (n > 1) ? b[1] : 0;
  106. const uint8_t *pl = b + 2; uint16_t pln = (n >= 2) ? n - 2 : 0;
  107. if (cmd >= 0x80 && cmd <= 0x89) g_steamAliveMs = millis(); // any Steam settings/haptic/LED report (incl. the 0x87 lizard-off heartbeat) -> Steam present, forward gamepad, suppress auto-lizard
  108. // Controller power-off: Steam's "turn off controller" is feature-0x01 frame 9F 04 6F 66 66 21 ("off!"),
  109. // confirmed from a real puck capture. The feature-0x01 relay below forwards it once; hapticSendShutdown()
  110. // bursts it for NO-ACK reliability (the single hook the test button + host-suspend also drive).
  111. if (rid == 1 && cmd == 0x9F) hapticSendShutdown();
  112. if (rid == 1 && n >= 2) { // report 0x01 = raw passthrough -> queue for RF relay to the controller
  113. hapLogAdd((uint8_t)slot, cmd, b, n); // capture EVERY relayed feature-1 command for the WebUSB capture view
  114. // (haptics, LED SET_LED_COLOR, 0x87 settings, 0x9F power-off). The log
  115. // shows cmd as "rid"; bytes start [cmd][len]... Was gated to 0x80-0x89,
  116. // which hid exactly the LED/power-off frames we now need to see.
  117. bool haptic82 = (cmd == 0x82 && len <= pln);
  118. bool muted = g_resumeMs && millis()-g_resumeMs < POST_RESUME_MUTE_MS; // see the OUTPUT path: no haptics while Steam can't read 0x45 back
  119. bool relayOk = hapticRelaySlotOk(slot) && !(haptic82 && (lizardActive() || muted)); // never push haptics to the controller while presenting lizard (Steam isn't reading 0x45 -> would buzz-loop)
  120. if (relayOk && (!haptic82 || !haptic82Blocked())) {
  121. // Relay the DECLARED length (up to the 60B RF frame ceiling), not a truncation: Steam's multi-register
  122. // 0x87 settings blocks (LED brightness) and calibration writes are longer than the old 18B cap, and the
  123. // chopped frames were why those settings never landed on the controller.
  124. uint8_t rl = (len <= pln) ? len : (uint8_t)pln;
  125. if (len > RELAY_MAXP && Serial.availableForWrite() > 60)
  126. Serial.printf("# RELAY TRUNC cmd=%02X len=%u>%u\n", cmd, len, (unsigned)RELAY_MAXP);
  127. relayEnqueue(cmd, pl, rl);
  128. if (haptic82) haptic82HostReport(pl, len); // track from RELAYED frames only (see the OUTPUT path)
  129. }
  130. }
  131. if (Serial.availableForWrite() > 80) { // log host feature writes (non-blocking)
  132. Serial.printf("# SET if%d rid=%02X cmd=%02X len=%u:", slot, rid, cmd, len);
  133. for (uint16_t i = 0; i < n && i < 14; i++) Serial.printf(" %02X", b[i]);
  134. Serial.println();
  135. }
  136. memset(S.resp, 0, sizeof S.resp); S.resp_len = 0;
  137. switch (cmd) {
  138. case 0x83:
  139. S.resp[0] = 0x83; S.resp[1] = sizeof ATTR83; memcpy(S.resp + 2, ATTR83, sizeof ATTR83); S.resp_len = 63; break;
  140. case 0xAE: {
  141. uint8_t idx = pln > 0 ? pl[0] : 1; const char *s = (idx == 0) ? g_board : (idx == 1) ? g_unit : "NA";
  142. 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; }
  143. case 0xB4: // connection/version state per slot: value 0x02 = controller connected, 0x01 = not
  144. S.resp[0] = 0xB4; S.resp[1] = 0x01;
  145. S.resp[2] = (slot == g_connSlot && !g_xbox && (millis() - g_connReplyMs < 500)) ? 0x02 : 0x01;
  146. S.resp_len = 63; break;
  147. case 0xAD:
  148. g_pairing = (pln > 0 && pl[0] != 0); Serial.printf("# pairing %s\n", g_pairing ? "ON" : "off");
  149. S.resp[0] = 0xAD; S.resp[1] = 0; S.resp_len = 63; break;
  150. case 0xA2: // write/clear THIS interface's slot
  151. if (len >= 24 && pln >= 24) {
  152. if (recEmpty(pl)) { S.used = false; memset(S.rec, 0, 24); }
  153. else { memcpy(S.rec, pl, 24); S.used = true; }
  154. g_dirty = true; Serial.printf("# slot %d %s\n", slot, recEmpty(pl) ? "cleared" : "bonded");
  155. }
  156. S.resp[0] = 0xA2; S.resp[1] = 0; S.resp_len = 63; break;
  157. case 0xA3: // read THIS interface's slot
  158. S.resp[0] = 0xA3; S.resp[1] = 0x18; memset(S.resp + 2, 0, 24);
  159. if (S.used) memcpy(S.resp + 2, S.rec, 24); S.resp_len = 63; break;
  160. default:
  161. S.resp[0] = cmd; S.resp[1] = len; if (pln) memcpy(S.resp + 2, pl, pln > 60 ? 60 : pln); S.resp_len = 63; break;
  162. }
  163. }
  164. static uint16_t handleGet(int slot, uint8_t rid, hid_report_type_t type, uint8_t *buf, uint16_t reqlen) {
  165. (void)rid;
  166. if (type != HID_REPORT_TYPE_FEATURE) return 0;
  167. Slot &S = g_slot[slot];
  168. uint16_t n = S.resp_len ? S.resp_len : 63; if (n > reqlen) n = reqlen;
  169. memcpy(buf, S.resp, n); return n;
  170. }
  171. // one callback pair per interface (the Adafruit core routes by interface to the matching object)
  172. #define SLOTCB(N) \
  173. 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); } \
  174. 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); }
  175. SLOTCB(0) SLOTCB(1) SLOTCB(2) SLOTCB(3)
  176. typedef uint16_t (*getcb_t)(uint8_t, hid_report_type_t, uint8_t *, uint16_t);
  177. typedef void (*setcb_t)(uint8_t, hid_report_type_t, uint8_t const *, uint16_t);
  178. static getcb_t GETCB[NSLOT] = { getcb0, getcb1, getcb2, getcb3 };
  179. static setcb_t SETCB[NSLOT] = { setcb0, setcb1, setcb2, setcb3 };
  180. // ===================== IController =====================
  181. void SteamPuckController::begin(){
  182. USBDevice.setID(0x28DE, 0x1304);
  183. // Distinct bcdDevice so Windows keys a FRESH usbflags entry (cache is VID:PID:bcdDevice) and actually runs
  184. // MS OS 2.0 / WinUSB binding for the WebUSB vendor interface -- instead of reusing a stale "no WinUSB" entry
  185. // tied to the real Steam Controller (28DE:1304), which has no WebUSB interface. The normal (wake-mouse) and
  186. // one-shot debug (CDC) boots present DIFFERENT interface sets, so they use DIFFERENT bcdDevice values --
  187. // otherwise Windows would serve one's cached descriptor for the other when you reboot between them.
  188. USBDevice.setDeviceVersion(g_debugCdcThisBoot ? 0x0212 : 0x0211);
  189. USBDevice.setManufacturerDescriptor("Valve Software");
  190. USBDevice.setProductDescriptor("Steam Controller Puck");
  191. for (int i = 0; i < NSLOT; i++) {
  192. hid[i].setReportDescriptor(PUCK_HID_DESC, sizeof PUCK_HID_DESC);
  193. hid[i].setReportCallback(GETCB[i], SETCB[i]);
  194. hid[i].setPollInterval(1); // 1ms USB poll (was default 10ms = 100/s cap -> choppy)
  195. hid[i].begin();
  196. }
  197. }
  198. // Forward the controller's report 0x45 to Steam, or drive lizard kb/mouse when Steam is closed. This is a
  199. // PURELY USB-SIDE decision -- it changes nothing about the RF poll or the host->controller relay.
  200. void SteamPuckController::onReport45(const uint8_t* rep, bool fresh, uint8_t bodyTlen){
  201. // Host asleep -> forward NOTHING. While suspended, every sendReport attempt (a lizard-mouse report from a
  202. // trackpad graze, a 0x45 forward) can translate into a host wake -- which made the PC wake on any controller
  203. // movement and nearly impossible to keep asleep. Waking is an explicit gesture only (Steam-button short press
  204. // / controller connect), handled in rf_link.cpp via the device-level USB resume signal.
  205. if (USBDevice.suspended()) return;
  206. // Just woke? Hold off forwarding for a beat so the wake gesture's residual controller input doesn't click /
  207. // type into the freshly-woken desktop (it was launching Edge from the Start menu on every wake).
  208. if (g_resumeMs && millis()-g_resumeMs < POST_RESUME_MUTE_MS) return;
  209. if (g_connSlot < 0 || g_connSlot >= NSLOT) return;
  210. if (lizardActive()){
  211. rfLizard(rep, &hid[g_connSlot], &hid[g_connSlot], 0x40, 0x41);
  212. } else {
  213. uint8_t blen = bodyTlen - 1; if(blen>45)blen=45; // body after the 0x45 id byte
  214. // forward the puck's raw pad coords untouched -- Steam does its own interpolation/smoothing. Forward only
  215. // FRESH reports (the real puck dedupes -> Steam gets a clean unique stream; sending stale repeats makes
  216. // Steam's velocity/smoothing stair-step). g_fwdNewOnly toggles for A/B.
  217. if((fresh || !g_fwdNewOnly) && g_slot[g_connSlot].used && hid[g_connSlot].ready())
  218. hid[g_connSlot].sendReport(0x45, rep+1, blen); // Steam/SDL: input report -> "connected"
  219. }
  220. }
  221. // Forward the controller's NON-input status reports (0x43 power/battery, 0x44) to Steam verbatim -- the real
  222. // puck does this and it's how Steam reads battery. OpenPuck used to drop everything but 0x45, so Steam never got
  223. // the power report (-> default/unknown battery). Same host-asleep / post-resume gating as 0x45; no lizard path
  224. // (status reports aren't input, so they forward regardless of the lizard decision).
  225. void SteamPuckController::onAuxReport(uint8_t rid, const uint8_t* data, uint8_t n){
  226. if (USBDevice.suspended()) return;
  227. if (g_resumeMs && millis()-g_resumeMs < POST_RESUME_MUTE_MS) return;
  228. if (g_connSlot < 0 || g_connSlot >= NSLOT) return;
  229. if (g_slot[g_connSlot].used && hid[g_connSlot].ready())
  230. hid[g_connSlot].sendReport(rid, data, n);
  231. }
  232. // ---- wake nudge: a bare USB resume signal is NOT enough to wake some hosts (Windows in particular) -- they
  233. // only wake when actual mouse/keyboard input follows. So on a deliberate wake gesture we play a HARMLESS mouse
  234. // JIGGLE (move a few px right, then back -- NET ZERO cursor, NO button): real mouse activity wakes the host,
  235. // but it clicks nothing and doesn't close/activate anything (an open Start menu stays open). Earlier versions
  236. // also sent a click / space / Ctrl; those either activated the focused item or weren't needed. Queued by
  237. // wakeEvent() (rf_link, on a Steam short press / controller connect while suspended); reports can't cross a
  238. // suspended bus, so it's delivered once the bus has resumed.
  239. static uint8_t g_nudgeStep = 0; // 0=idle; 1=jiggle+, 2=jiggle-
  240. static unsigned long g_nudgeMs = 0;
  241. #define NUDGE_JIGGLE_PX 10
  242. void SteamPuckController::wakeEvent(){
  243. g_nudgeStep = 1; g_nudgeMs = millis();
  244. }
  245. static void wakeNudgeTask(){
  246. if(!g_nudgeStep) return;
  247. if(millis()-g_nudgeMs > 5000){ g_nudgeStep=0; return; } // bus never resumed -> drop the nudge
  248. if(USBDevice.suspended()) return; // wait for resume; reports can't cross a suspended bus
  249. if(g_connSlot<0 || g_connSlot>=NSLOT || !hid[g_connSlot].ready()) return;
  250. static unsigned long stepMs=0;
  251. if(millis()-stepMs < 15) return; // pace the edges
  252. stepMs=millis();
  253. hid_mouse_report_t m; m.buttons=0; m.x=(g_nudgeStep==1)?NUDGE_JIGGLE_PX:-NUDGE_JIGGLE_PX; m.y=0; m.wheel=0; m.pan=0;
  254. hid[g_connSlot].sendReport(0x40,&m,sizeof m); // jiggle right, then back
  255. g_nudgeStep = (g_nudgeStep>=2) ? 0 : (uint8_t)(g_nudgeStep+1);
  256. }
  257. // USB connection presentation (like the real dongle): report 0x79 = connection state (01=disc, 02=conn),
  258. // edge-triggered, + periodic 0x7B status. Live-captured: this is what Steam reads to mark the controller
  259. // connected. Without it Steam shows disconnected even though 0x45 input is streaming.
  260. void SteamPuckController::task(){
  261. wakeNudgeTask();
  262. { static bool wasSusp=false; bool susp=USBDevice.suspended(); // stamp the suspended->active edge for the post-resume mute
  263. if(wasSusp && !susp) g_resumeMs=millis();
  264. wasSusp=susp; }
  265. if (USBDevice.suspended()) return; // no periodic 0x79/0x7B while the host sleeps -- those sends can wake it too
  266. static bool usbConn=false; static unsigned long last79=0, last7B=0, connEdgeMs=0;
  267. bool conn = (g_connSlot>=0 && millis()-g_connReplyMs < 300);
  268. if (g_connSlot>=0 && g_connSlot<NSLOT && hid[g_connSlot].ready()){
  269. // 0x79 connection state: on edge, then repeated every 750ms ONLY until Steam reacts (its first OUTPUT/
  270. // settings write after the edge -- g_steamAliveMs). The real puck sends 0x79 ONCE, edge-triggered; the old
  271. // unconditional forever-resend could re-trigger Steam's connect handling (connect chime) every 750ms
  272. // before Steam started consuming 0x45 -> the connect-time loop of haptic buzzes that only stopped on the
  273. // first real input (the Steam button). Resending until acked still covers "Steam missed the edge".
  274. bool steamAcked = g_steamAliveMs && (int32_t)(g_steamAliveMs - connEdgeMs) >= 0;
  275. if (conn!=usbConn || (conn && !steamAcked && millis()-last79>=750)){
  276. if (conn && !usbConn) connEdgeMs = millis();
  277. uint8_t st=conn?0x02:0x01; hid[g_connSlot].sendReport(0x79,&st,1); usbConn=conn; last79=millis();
  278. } else if (conn && millis()-last7B>=2000){
  279. // 0x7B status, live-captured template. Byte 8 is the controller->puck signal strength as signed dBm
  280. // (capture showed 0xDD = -35; replaying it verbatim is why Steam pinned -35dBm forever) -- patch in the
  281. // smoothed RSSI the radio now samples on each controller reply (rf_link). 0 = no sample yet -> keep the
  282. // capture value rather than report garbage.
  283. //
  284. // CALIBRATION: our raw RSSISAMPLE reads ~RSSI_DBM_OFFSET dB lower than the real Valve puck at the same
  285. // distance (the Pro Micro's PCB-trace antenna vs Valve's tuned front-end). The 2Mbit ESB link has ~55dB
  286. // of margin, so a -75dBm reading still works across a house -- but Steam's bar maps raw dBm to "weak"
  287. // long before that. Adding the offset lines our close-range value up with the puck's captured -35 so the
  288. // bar tracks usable range instead of antenna gain. Clamp keeps it in a sane window (no wrap/garbage).
  289. uint8_t s7b[12]={0xF7,0x01,0x89,0x00,0x00,0x00,0x03,0x00,0xDD,0x00,0x3A,0x02};
  290. if (g_linkRssi){
  291. int mag = (int)g_linkRssi - RSSI_DBM_OFFSET;
  292. if (mag < 25) mag = 25; else if (mag > 95) mag = 95;
  293. s7b[8]=(uint8_t)(0u-(uint8_t)mag);
  294. }
  295. hid[g_connSlot].sendReport(0x7B,s7b,12); last7B=millis();
  296. }
  297. } else if(!conn) usbConn=false;
  298. }