mirror of
https://github.com/ndeadly/MissionControl
synced 2024-12-13 06:12:43 +00:00
254 lines
12 KiB
C++
254 lines
12 KiB
C++
/*
|
|
* Copyright (c) 2020-2021 ndeadly
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms and conditions of the GNU General Public License,
|
|
* version 2, as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "emulated_switch_controller.hpp"
|
|
#include <memory>
|
|
|
|
namespace ams::controller {
|
|
|
|
bluetooth::HidReport EmulatedSwitchController::s_input_report;
|
|
bluetooth::HidReport EmulatedSwitchController::s_output_report;
|
|
|
|
EmulatedSwitchController::EmulatedSwitchController(const bluetooth::Address *address)
|
|
: SwitchController(address)
|
|
, m_charging(false)
|
|
, m_battery(BATTERY_MAX) {
|
|
this->ClearControllerState();
|
|
|
|
m_colours.body = {0x32, 0x32, 0x32};
|
|
m_colours.buttons = {0xe6, 0xe6, 0xe6};
|
|
m_colours.left_grip = {0x46, 0x46, 0x46};
|
|
m_colours.right_grip = {0x46, 0x46, 0x46};
|
|
};
|
|
|
|
void EmulatedSwitchController::ClearControllerState(void) {
|
|
std::memset(&m_buttons, 0, sizeof(m_buttons));
|
|
m_left_stick = this->PackStickData(STICK_ZERO, STICK_ZERO);
|
|
m_right_stick = this->PackStickData(STICK_ZERO, STICK_ZERO);
|
|
std::memset(&m_motion_data, 0, sizeof(m_motion_data));
|
|
}
|
|
|
|
void EmulatedSwitchController::ApplyButtonCombos(SwitchButtonData *buttons) {
|
|
// Home combo = MINUS + DPAD_DOWN
|
|
if (buttons->minus && buttons->dpad_down) {
|
|
buttons->home = 1;
|
|
buttons->minus = 0;
|
|
buttons->dpad_down = 0;
|
|
}
|
|
|
|
// Capture combo = MINUS + DPAD_UP
|
|
if (buttons->minus && buttons->dpad_up) {
|
|
buttons->capture = 1;
|
|
buttons->minus = 0;
|
|
buttons->dpad_up = 0;
|
|
}
|
|
}
|
|
|
|
Result EmulatedSwitchController::HandleIncomingReport(const bluetooth::HidReport *report) {
|
|
this->UpdateControllerState(report);
|
|
|
|
// Prepare Switch report
|
|
s_input_report.size = sizeof(SwitchInputReport0x30) + 1;
|
|
auto switch_report = reinterpret_cast<SwitchReportData *>(s_input_report.data);
|
|
switch_report->id = 0x30;
|
|
switch_report->input0x30.conn_info = 0;
|
|
switch_report->input0x30.battery = m_battery | m_charging;
|
|
switch_report->input0x30.buttons = m_buttons;
|
|
switch_report->input0x30.left_stick = m_left_stick;
|
|
switch_report->input0x30.right_stick = m_right_stick;
|
|
std::memcpy(&switch_report->input0x30.motion, &m_motion_data, sizeof(m_motion_data));
|
|
|
|
this->ApplyButtonCombos(&switch_report->input0x30.buttons);
|
|
|
|
switch_report->input0x30.timer = os::ConvertToTimeSpan(os::GetSystemTick()).GetMilliSeconds() & 0xff;
|
|
return bluetooth::hid::report::WriteHidReportBuffer(&m_address, &s_input_report);
|
|
}
|
|
|
|
Result EmulatedSwitchController::HandleOutgoingReport(const bluetooth::HidReport *report) {
|
|
uint8_t cmdId = report->data[0];
|
|
switch (cmdId) {
|
|
case 0x01: // Subcmd
|
|
R_TRY(this->HandleSubCmdReport(report));
|
|
break;
|
|
case 0x10: // Rumble
|
|
// Todo: add rumble support
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return ams::ResultSuccess();
|
|
}
|
|
|
|
Result EmulatedSwitchController::HandleSubCmdReport(const bluetooth::HidReport *report) {
|
|
const uint8_t *subcmd = &report->data[10];
|
|
auto subcmd_id = static_cast<SubCmdType>(subcmd[0]);
|
|
|
|
switch (subcmd_id) {
|
|
case SubCmd_RequestDeviceInfo:
|
|
R_TRY(this->SubCmdRequestDeviceInfo(report));
|
|
break;
|
|
case SubCmd_SpiFlashRead:
|
|
R_TRY(this->SubCmdSpiFlashRead(report));
|
|
break;
|
|
case SubCmd_SpiFlashWrite:
|
|
R_TRY(this->SubCmdSpiFlashWrite(report));
|
|
break;
|
|
case SubCmd_SpiSectorErase:
|
|
R_TRY(this->SubCmdSpiSectorErase(report));
|
|
break;
|
|
case SubCmd_SetInputReportMode:
|
|
R_TRY(this->SubCmdSetInputReportMode(report));
|
|
break;
|
|
case SubCmd_TriggersElapsedTime:
|
|
R_TRY(this->SubCmdTriggersElapsedTime(report));
|
|
break;
|
|
case SubCmd_SetShipPowerState:
|
|
R_TRY(this->SubCmdSetShipPowerState(report));
|
|
break;
|
|
case SubCmd_SetMcuConfig:
|
|
R_TRY(this->SubCmdSetMcuConfig(report));
|
|
break;
|
|
case SubCmd_SetMcuState:
|
|
R_TRY(this->SubCmdSetMcuState(report));
|
|
break;
|
|
case SubCmd_SetPlayerLeds:
|
|
R_TRY(this->SubCmdSetPlayerLeds(report));
|
|
break;
|
|
case SubCmd_SetHomeLed:
|
|
R_TRY(this->SubCmdSetHomeLed(report));
|
|
break;
|
|
case SubCmd_EnableImu:
|
|
R_TRY(this->SubCmdEnableImu(report));
|
|
break;
|
|
case SubCmd_EnableVibration:
|
|
R_TRY(this->SubCmdEnableVibration(report));
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return ams::ResultSuccess();
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdRequestDeviceInfo(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x82, 0x02, 0x03, 0x48, 0x03, 0x02, m_address.address[0], m_address.address[1], m_address.address[2], m_address.address[3], m_address.address[4], m_address.address[5], 0x01, 0x02};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSpiFlashRead(const bluetooth::HidReport *report) {
|
|
// These are read from official Pro Controller
|
|
// @ 0x00006000: ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff <= Serial
|
|
// @ 0x00006050: 32 32 32 ff ff ff ff ff ff ff ff ff <= RGB colours (body, buttons, left grip, right grip)
|
|
// @ 0x00006080: 50 fd 00 00 c6 0f 0f 30 61 ae 90 d9 d4 14 54 41 15 54 c7 79 9c 33 36 63 <= Factory Sensor and Stick device parameters
|
|
// @ 0x00006098: 0f 30 61 ae 90 d9 d4 14 54 41 15 54 c7 79 9c 33 36 63 <= Stick device parameters 2. Normally the same with 1, even in Pro Contr.
|
|
// @ 0x00008010: ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff ff <= User Analog sticks calibration
|
|
// @ 0x0000603d: e6 a5 67 1a 58 78 50 56 60 1a f8 7f 20 c6 63 d5 15 5e ff 32 32 32 ff ff ff <= Left analog stick calibration
|
|
// @ 0x00006020: 64 ff 33 00 b8 01 00 40 00 40 00 40 17 00 d7 ff bd ff 3b 34 3b 34 3b 34 <= 6-Axis motion sensor Factory calibration
|
|
|
|
uint32_t read_addr = *(uint32_t *)(&report->data[11]);
|
|
uint8_t read_size = report->data[15];
|
|
|
|
const uint8_t prefix[] = {0x90, SubCmd_SpiFlashRead, report->data[11], report->data[12], report->data[13], report->data[14], report->data[15]};
|
|
|
|
int response_size = read_size + sizeof(prefix);
|
|
auto response = std::make_unique<uint8_t[]>(response_size);
|
|
std::memcpy(response.get(), prefix, sizeof(prefix));
|
|
std::memset(response.get() + sizeof(prefix), 0xff, read_size); // Console doesn't seem to mind if response is uninitialised data (0xff)
|
|
|
|
// Set controller colours
|
|
if (read_addr == 0x6050) {
|
|
std::memcpy(response.get() + sizeof(prefix), &m_colours, sizeof(m_colours));
|
|
}
|
|
|
|
return this->FakeSubCmdResponse(response.get(), response_size);
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSpiFlashWrite(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SpiFlashWrite, 0x01};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSpiSectorErase(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SpiSectorErase, 0x01};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetInputReportMode(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SetInputReportMode};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdTriggersElapsedTime(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x83, SubCmd_TriggersElapsedTime};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetShipPowerState(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SetShipPowerState, 0x00};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetMcuConfig(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0xa0, SubCmd_SetMcuConfig, 0x01, 0x00, 0xff, 0x00, 0x03, 0x00, 0x05, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x5c};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetMcuState(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SetMcuState};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetPlayerLeds(const bluetooth::HidReport *report) {
|
|
const uint8_t *subCmd = &report->data[10];
|
|
uint8_t led_mask = subCmd[1];
|
|
R_TRY(this->SetPlayerLed(led_mask));
|
|
|
|
const uint8_t response[] = {0x80, SubCmd_SetPlayerLeds};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdSetHomeLed(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_SetHomeLed};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdEnableImu(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_EnableImu};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::SubCmdEnableVibration(const bluetooth::HidReport *report) {
|
|
const uint8_t response[] = {0x80, SubCmd_EnableVibration};
|
|
return this->FakeSubCmdResponse(response, sizeof(response));
|
|
}
|
|
|
|
Result EmulatedSwitchController::FakeSubCmdResponse(const uint8_t response[], size_t size) {
|
|
s_input_report.size = sizeof(SwitchInputReport0x21) + 1;
|
|
auto report_data = reinterpret_cast<SwitchReportData *>(s_input_report.data);
|
|
report_data->id = 0x21;
|
|
report_data->input0x21.conn_info = 0;
|
|
report_data->input0x21.battery = m_battery | m_charging;
|
|
report_data->input0x21.buttons = m_buttons;
|
|
report_data->input0x21.left_stick = m_left_stick;
|
|
report_data->input0x21.right_stick = m_right_stick;
|
|
report_data->input0x21.vibrator = 0;
|
|
std::memcpy(&report_data->input0x21.response, response, size);
|
|
report_data->input0x21.timer = os::ConvertToTimeSpan(os::GetSystemTick()).GetMilliSeconds() & 0xff;
|
|
|
|
//Write a fake response into the report buffer
|
|
return bluetooth::hid::report::WriteHidReportBuffer(&m_address, &s_input_report);
|
|
}
|
|
|
|
}
|