HIDControllerV2/CustomHID.cpp at main · VictorBorchenko/HIDControllerV2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#include "CustomHID.h"

CustomHID_::CustomHID_()
    : PluggableUSBModule(2, 1, epType)
{
  epType[0] = EP_TYPE_INTERRUPT_IN;
  epType[1] = EP_TYPE_INTERRUPT_OUT;
  PluggableUSB().plug(this);
  outCallback = nullptr;
  memset(featureReportBufferIn, 0, sizeof(featureReportBufferIn));
  memset(featureReportBufferOut, 0, sizeof(featureReportBufferOut));
}

int CustomHID_::getInterface(uint8_t *interfaceCount)
{
  *interfaceCount += 1;
  // Interface descriptor (9 bytes)
  uint8_t interfaceDesc[] = {
      9, 4, pluggedInterface, 0x00, 1, 0x03, 0x00, 0x00, 0};
  // HID descriptor (9 bytes)
  uint8_t hidDesc[] = {
      9, 0x21, 0x11, 0x01, 0x00, 0x01, 0x22,
      (uint8_t)(sizeof(customHIDReportDescriptor)),
      (uint8_t)(sizeof(customHIDReportDescriptor) >> 8)};
  // Endpoint IN (7 bytes)
  uint8_t epInDesc[] = {
      7, 5, (uint8_t)(0x80 | pluggedEndpoint), 0x03, USB_EP_SIZE, 0, 0x01};

  USB_SendControl(0, interfaceDesc, sizeof(interfaceDesc));
  USB_SendControl(0, hidDesc, sizeof(hidDesc));
  USB_SendControl(0, epInDesc, sizeof(epInDesc));
  return sizeof(interfaceDesc) + sizeof(hidDesc) + sizeof(epInDesc);
}

int CustomHID_::getDescriptor(USBSetup &setup)
{
  if (setup.wValueH == HID_REPORT_DESCRIPTOR_TYPE)
  {
    return USB_SendControl(TRANSFER_PGM, customHIDReportDescriptor, sizeof(customHIDReportDescriptor));
  }
  return 0;
}

bool CustomHID_::setup(USBSetup &setup)
{
  if (pluggedInterface != setup.wIndex)
    return false;

  if (setup.bmRequestType == REQUEST_DEVICETOHOST_STANDARD_INTERFACE && setup.bRequest == HID_GET_REPORT)
  {
    return USB_SendControl(0, reportBuffer, sizeof(reportBuffer)) >= 0;
  }

  if (setup.bmRequestType == REQUEST_HOSTTODEVICE_CLASS_INTERFACE && setup.bRequest == HID_SET_REPORT)
  {
    if (setup.wValueH == 3)
    { // Feature report
      uint8_t buf[16];
      if (USB_RecvControl(buf, sizeof(buf)) == sizeof(buf))
      {
        memcpy(featureReportBufferIn, buf, sizeof(buf));
        if (featureCallback)
          featureCallback(buf, 16);
      }
      return true;
    }
    else
    {
      uint8_t outBuf[4];
      if (USB_RecvControl(outBuf, 4) == 4 && outCallback)
      {
        uint32_t value =
            (uint32_t)outBuf[0] | ((uint32_t)outBuf[1] << 8) |
            ((uint32_t)outBuf[2] << 16) | ((uint32_t)outBuf[3] << 24);
        outCallback(value);
      }
      return true;
    }
  }

  if (setup.bmRequestType == REQUEST_DEVICETOHOST_CLASS_INTERFACE && setup.bRequest == HID_GET_REPORT && setup.wValueH == 3)
  {
    return USB_SendControl(0, featureReportBufferOut, sizeof(featureReportBufferOut)) >= 0;
  }

  return false;
}

void CustomHID_::sendReport(uint8_t b1, uint8_t b2, uint16_t bools)
{
  reportBuffer[0] = b1;
  reportBuffer[1] = b2;
  reportBuffer[2] = bools & 0xFF;
  reportBuffer[3] = (bools >> 8) & 0xFF;
  USB_Send(pluggedEndpoint | TRANSFER_RELEASE, reportBuffer, sizeof(reportBuffer));
}

void CustomHID_::setOutCallback(void (*cb)(uint32_t))
{
  outCallback = cb;
}

void CustomHID_::setFeatureCallback(void (*cb)(const uint8_t *data, size_t len))
{
  featureCallback = cb;
}

void CustomHID_::setFeatureReport(const uint8_t *data, size_t len)
{
  memcpy(featureReportBufferOut, data, min(len, sizeof(featureReportBufferOut)));
}

void CustomHID_::getFeatureReport(uint8_t *data, size_t len)
{
  memcpy(data, featureReportBufferIn, min(len, sizeof(featureReportBufferIn)));
}