From 499f78cb23ffabd4c1e53201350847454e8b6668 Mon Sep 17 00:00:00 2001 From: ndeadly <24677491+ndeadly@users.noreply.github.com> Date: Sun, 6 Sep 2020 13:45:50 +0200 Subject: [PATCH] btdrv-mitm: add support for gembox controller --- .../controllers/controller_management.cpp | 6 ++ .../controllers/controller_management.hpp | 2 + .../source/controllers/gembox_controller.cpp | 100 ++++++++++++++++++ .../source/controllers/gembox_controller.hpp | 98 +++++++++++++++++ 4 files changed, 206 insertions(+) create mode 100644 btdrv-mitm/source/controllers/gembox_controller.cpp create mode 100644 btdrv-mitm/source/controllers/gembox_controller.hpp diff --git a/btdrv-mitm/source/controllers/controller_management.cpp b/btdrv-mitm/source/controllers/controller_management.cpp index b5985d9..b164110 100644 --- a/btdrv-mitm/source/controllers/controller_management.cpp +++ b/btdrv-mitm/source/controllers/controller_management.cpp @@ -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;; diff --git a/btdrv-mitm/source/controllers/controller_management.hpp b/btdrv-mitm/source/controllers/controller_management.hpp index 428c079..b324331 100644 --- a/btdrv-mitm/source/controllers/controller_management.hpp +++ b/btdrv-mitm/source/controllers/controller_management.hpp @@ -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, }; diff --git a/btdrv-mitm/source/controllers/gembox_controller.cpp b/btdrv-mitm/source/controllers/gembox_controller.cpp new file mode 100644 index 0000000..9afbfdf --- /dev/null +++ b/btdrv-mitm/source/controllers/gembox_controller.cpp @@ -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 . + */ +#include "gembox_controller.hpp" +#include + +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(&in_report->data); + auto switch_report = reinterpret_cast(&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(stick_scale_factor * static_cast(~src->input0x07.left_stick.x + 1) + 0x7ff) & 0xfff, + static_cast(stick_scale_factor * (UINT8_MAX - static_cast(~src->input0x07.left_stick.y + 1)) + 0x7ff) & 0xfff + ); + this->PackStickData(&dst->input0x30.right_stick, + static_cast(stick_scale_factor * static_cast(~src->input0x07.right_stick.x + 1) + 0x7ff) & 0xfff, + static_cast(stick_scale_factor * (UINT8_MAX - static_cast(~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; + } + +} diff --git a/btdrv-mitm/source/controllers/gembox_controller.hpp b/btdrv-mitm/source/controllers/gembox_controller.hpp new file mode 100644 index 0000000..827f31c --- /dev/null +++ b/btdrv-mitm/source/controllers/gembox_controller.hpp @@ -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 . + */ +#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); + + }; + +}