Blame view

Modif/epsilon-master/ion/src/device/usb/calculator.cpp 3.44 KB
6663b6c9   adorian   projet complet av...
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
  #include "calculator.h"
  #include <ion/usb.h>
  #include <ion/src/device/regs/regs.h>
  #include <ion/src/device/device.h>
  #include <ion/src/device/keyboard.h>
  
  namespace Ion {
  namespace USB {
  namespace Device {
  
  void Calculator::PollAndReset(bool exitWithKeyboard) {
    char serialNumber[Ion::Device::SerialNumberLength+1];
    Ion::Device::copySerialNumber(serialNumber);
    Calculator c(serialNumber);
  
    /* Leave DFU mode if the Back key is pressed, the calculator unplugged or the
     * USB core soft-disconnected. */
    Ion::Keyboard::Key exitKey = Ion::Keyboard::Key::A6;
    uint8_t exitKeyRow = Ion::Keyboard::Device::rowForKey(exitKey);
    uint8_t exitKeyColumn = Ion::Keyboard::Device::columnForKey(exitKey);
  
    Ion::Keyboard::Device::activateRow(exitKeyRow);
  
    while (!(exitWithKeyboard && Ion::Keyboard::Device::columnIsActive(exitKeyColumn)) &&
        Ion::USB::isPlugged() &&
        !c.isSoftDisconnected()) {
      c.poll();
    }
    if (!c.isSoftDisconnected()) {
      c.detach();
    }
    if (c.resetOnDisconnect()) {
      /* We don't perform a core reset because at this point in time the USB cable
       * is most likely plugged in. Doing a full core reset would be the clean
       * thing to do but would therefore result in the device entering the ROMed
       * DFU bootloader, which we want to avoid. By performing a jump-reset, we
       * will enter the newly flashed firmware. */
      Ion::Device::jumpReset();
    }
  }
  
  Descriptor * Calculator::descriptor(uint8_t type, uint8_t index) {
    /* Special case: Microsoft OS String Descriptor should be returned when
     * searching for string descriptor at index 0xEE. */
    if (type == m_microsoftOSStringDescriptor.type() && index == 0xEE) {
      return &m_microsoftOSStringDescriptor;
    }
    int typeCount = 0;
    for (size_t i=0; i<sizeof(m_descriptors)/sizeof(m_descriptors[0]); i++) {
      Descriptor * descriptor = m_descriptors[i];
      if (descriptor->type() != type) {
        continue;
      }
      if (typeCount == index) {
        return descriptor;
      } else {
        typeCount++;
      }
    }
    return nullptr;
  }
  
  bool Calculator::processSetupInRequest(SetupPacket * request, uint8_t * transferBuffer, uint16_t * transferBufferLength, uint16_t transferBufferMaxLength) {
    if (Device::processSetupInRequest(request, transferBuffer, transferBufferLength, transferBufferMaxLength)) {
      return true;
    }
    if (request->requestType() == SetupPacket::RequestType::Vendor) {
      if (request->bRequest() == k_webUSBVendorCode && request->wIndex() == 2) {
        // This is a WebUSB, GET_URL request
        assert(request->wValue() == k_webUSBLandingPageIndex);
        return getURLCommand(transferBuffer, transferBufferLength, transferBufferMaxLength);
      }
      if (request->bRequest() == k_microsoftOSVendorCode && request->wIndex() == 0x0004) {
        // This is a Microsoft OS descriptor, Extended Compat ID request
        assert(request->wValue() == 0);
        return getExtendedCompatIDCommand(transferBuffer, transferBufferLength, transferBufferMaxLength);
      }
    }
    return false;
  }
  
  bool Calculator::getURLCommand(uint8_t * transferBuffer, uint16_t * transferBufferLength, uint16_t transferBufferMaxLength) {
    *transferBufferLength = m_workshopURLDescriptor.copy(transferBuffer, transferBufferMaxLength);
    return true;
  }
  
  bool Calculator::getExtendedCompatIDCommand(uint8_t * transferBuffer, uint16_t * transferBufferLength, uint16_t transferBufferMaxLength) {
    *transferBufferLength = m_extendedCompatIdDescriptor.copy(transferBuffer, transferBufferMaxLength);
    return true;
  }
  
  }
  }
  }