btdrv-mitm: add support for gembox controller

This commit is contained in:
ndeadly 2020-09-06 13:45:50 +02:00
parent a7c22b451c
commit 499f78cb23
4 changed files with 206 additions and 0 deletions

View file

@ -75,6 +75,12 @@ namespace ams::controller {
}
}
for (auto hwId : GemboxController::hardware_ids) {
if ( (device->vid == hwId.vid) && (device->pid == hwId.pid) ) {
return ControllerType_Gembox;
}
}
// Handle the case where joycons have been assigned random hardware ids when paired via rails
if (IsJoyCon(device->name)) {
return ControllerType_Switch;;

View file

@ -22,6 +22,7 @@
#include "xbox_one_controller.hpp"
#include "ouya_controller.hpp"
#include "gamestick_controller.hpp"
#include "gembox_controller.hpp"
namespace ams::controller {
@ -35,6 +36,7 @@ namespace ams::controller {
ControllerType_XboxOne,
ControllerType_Ouya,
ControllerType_Gamestick,
ControllerType_Gembox,
ControllerType_Unknown,
};

View file

@ -0,0 +1,100 @@
/*
* Copyright (C) 2020 ndeadly
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that 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 "gembox_controller.hpp"
#include <stratosphere.hpp>
namespace ams::controller {
namespace {
const constexpr float stick_scale_factor = float(UINT12_MAX) / UINT8_MAX;
}
void GemboxController::ConvertReportFormat(const bluetooth::HidReport *in_report, bluetooth::HidReport *out_report) {
auto gembox_report = reinterpret_cast<const GemboxReportData *>(&in_report->data);
auto switch_report = reinterpret_cast<SwitchReportData *>(&out_report->data);
switch(gembox_report->id) {
case 0x03:
this->HandleInputReport0x02(gembox_report, switch_report);
break;
case 0x07:
this->HandleInputReport0x07(gembox_report, switch_report);
break;
default:
break;
}
out_report->size = sizeof(SwitchInputReport0x30) + 1;
switch_report->id = 0x30;
switch_report->input0x30.conn_info = 0x0;
switch_report->input0x30.battery = m_battery | m_charging;
std::memset(switch_report->input0x30.motion, 0, sizeof(switch_report->input0x30.motion));
switch_report->input0x30.timer = os::ConvertToTimeSpan(os::GetSystemTick()).GetMilliSeconds() & 0xff;
}
void GemboxController::HandleInputReport0x02(const GemboxReportData *src, SwitchReportData *dst) {
dst->input0x30.buttons.minus = src->input0x02.back;
this->PackStickData(&dst->input0x30.left_stick, STICK_ZERO, STICK_ZERO);
this->PackStickData(&dst->input0x30.right_stick, STICK_ZERO, STICK_ZERO);
}
void GemboxController::HandleInputReport0x07(const GemboxReportData *src, SwitchReportData *dst) {
this->PackStickData(&dst->input0x30.left_stick,
static_cast<uint16_t>(stick_scale_factor * static_cast<int8_t>(~src->input0x07.left_stick.x + 1) + 0x7ff) & 0xfff,
static_cast<uint16_t>(stick_scale_factor * (UINT8_MAX - static_cast<int8_t>(~src->input0x07.left_stick.y + 1)) + 0x7ff) & 0xfff
);
this->PackStickData(&dst->input0x30.right_stick,
static_cast<uint16_t>(stick_scale_factor * static_cast<int8_t>(~src->input0x07.right_stick.x + 1) + 0x7ff) & 0xfff,
static_cast<uint16_t>(stick_scale_factor * (UINT8_MAX - static_cast<int8_t>(~src->input0x07.right_stick.y + 1)) + 0x7ff) & 0xfff
);
dst->input0x30.buttons.dpad_down = (src->input0x07.dpad == GemboxDPad_S) ||
(src->input0x07.dpad == GemboxDPad_SE) ||
(src->input0x07.dpad == GemboxDPad_SW);
dst->input0x30.buttons.dpad_up = (src->input0x07.dpad == GemboxDPad_N) ||
(src->input0x07.dpad == GemboxDPad_NE) ||
(src->input0x07.dpad == GemboxDPad_NW);
dst->input0x30.buttons.dpad_right = (src->input0x07.dpad == GemboxDPad_E) ||
(src->input0x07.dpad == GemboxDPad_NE) ||
(src->input0x07.dpad == GemboxDPad_SE);
dst->input0x30.buttons.dpad_left = (src->input0x07.dpad == GemboxDPad_W) ||
(src->input0x07.dpad == GemboxDPad_NW) ||
(src->input0x07.dpad == GemboxDPad_SW);
dst->input0x30.buttons.A = src->input0x07.buttons.B;
dst->input0x30.buttons.B = src->input0x07.buttons.A;
dst->input0x30.buttons.X = src->input0x07.buttons.Y;
dst->input0x30.buttons.Y = src->input0x07.buttons.X;
dst->input0x30.buttons.R = src->input0x07.buttons.RB;
dst->input0x30.buttons.ZR = src->input0x07.right_trigger > 0;
dst->input0x30.buttons.L = src->input0x07.buttons.LB;
dst->input0x30.buttons.ZL = src->input0x07.left_trigger > 0;
dst->input0x30.buttons.plus = src->input0x07.buttons.start;
dst->input0x30.buttons.lstick_press = src->input0x07.buttons.L3;
dst->input0x30.buttons.rstick_press = src->input0x07.buttons.R3;
dst->input0x30.buttons.capture = 0;
dst->input0x30.buttons.home = 0;
}
}

View file

@ -0,0 +1,98 @@
/*
* Copyright (C) 2020 ndeadly
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that 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/>.
*/
#pragma once
#include "emulated_switch_controller.hpp"
namespace ams::controller {
enum GemboxDPadDirection {
GemboxDPad_N,
GemboxDPad_NE,
GemboxDPad_E,
GemboxDPad_SE,
GemboxDPad_S,
GemboxDPad_SW,
GemboxDPad_W,
GemboxDPad_NW,
GemboxDPad_Released = 0x0f
};
struct GemboxStickData {
uint8_t x;
uint8_t y;
} __attribute__ ((__packed__));
struct GemboxButtonData {
uint8_t A : 1;
uint8_t B : 1;
uint8_t : 1;
uint8_t X : 1;
uint8_t Y : 1;
uint8_t : 1;
uint8_t LB : 1;
uint8_t RB : 1;
uint8_t : 3;
uint8_t start : 1;
uint8_t : 1;
uint8_t L3 : 1;
uint8_t R3 : 1;
uint8_t : 0;
} __attribute__ ((__packed__));
struct GemboxInputReport0x02 {
uint8_t : 2;
uint8_t back : 1;
uint8_t : 0;
} __attribute__((packed));
struct GemboxInputReport0x07 {
uint8_t dpad;
GemboxStickData left_stick;
GemboxStickData right_stick;
uint8_t left_trigger;
uint8_t right_trigger;
GemboxButtonData buttons;
} __attribute__((packed));
struct GemboxReportData {
uint8_t id;
union {
GemboxInputReport0x02 input0x02;
GemboxInputReport0x07 input0x07;
};
} __attribute__((packed));
class GemboxController : public EmulatedSwitchController {
public:
static constexpr const HardwareID hardware_ids[] = {
{0x1d79, 0x0009}
};
GemboxController(const bluetooth::Address *address)
: EmulatedSwitchController(address) { };
void ConvertReportFormat(const bluetooth::HidReport *in_report, bluetooth::HidReport *out_report);
private:
void HandleInputReport0x02(const GemboxReportData *src, SwitchReportData *dst);
void HandleInputReport0x07(const GemboxReportData *src, SwitchReportData *dst);
};
}