// SPDX-License-Identifier: GPL-2.0-only
/* Copyright (c) 2024 Benjamin Tissoires
 */

#include "vmlinux.h"
#include "hid_bpf.h"
#include "hid_bpf_helpers.h"
#include <bpf/bpf_tracing.h>

#define VID_MICROSOFT 0x045e
#define PID_XBOX_ELITE_2 0x0b22

HID_BPF_CONFIG(
 HID_DEVICE(BUS_BLUETOOTH, HID_GROUP_GENERIC, VID_MICROSOFT, PID_XBOX_ELITE_2)
);

/*
 * When using the Xbox Wireless Controller Elite 2 over Bluetooth,
 * the device exports the paddles on the back of the device as a single
 * bitfield value of usage "Assign Selection".
 *
 * The kernel doesn't process the paddles usage properly and reports KEY_UNKNOWN.
 *
 * SDL doesn't know how to interpret KEY_UNKNOWN and thus ignores the paddles.
 *
 * Given that over USB the kernel uses BTN_TRIGGER_HAPPY[5-8], we
 * can tweak the report descriptor to make the kernel interpret it properly:
 * - We need an application collection of gamepad (so we have to close the current
 *   Consumer Control one)
 * - We need to change the usage to be buttons from 0x15 to 0x18
 */

#define OFFSET_ASSIGN_SELECTION  211
#define ORIGINAL_RDESC_SIZE  464

const __u8 rdesc_assign_selection[] = {
 0x0a, 0x99, 0x00,              //   Usage (Media Select Security)     211
 0x15, 0x00,                    //   Logical Minimum (0)               214
 0x26, 0xff, 0x00,              //   Logical Maximum (255)             216
 0x95, 0x01,                    //   Report Count (1)                  219
 0x75, 0x04,                    //   Report Size (4)                   221
 0x81, 0x02,                    //   Input (Data,Var,Abs)              223
 0x15, 0x00,                    //   Logical Minimum (0)               225
 0x25, 0x00,                    //   Logical Maximum (0)               227
 0x95, 0x01,                    //   Report Count (1)                  229
 0x75, 0x04,                    //   Report Size (4)                   231
 0x81, 0x03,                    //   Input (Cnst,Var,Abs)              233
 0x0a, 0x81, 0x00,              //   Usage (Assign Selection)          235
 0x15, 0x00,                    //   Logical Minimum (0)               238
 0x26, 0xff, 0x00,              //   Logical Maximum (255)             240
 0x95, 0x01,                    //   Report Count (1)                  243
 0x75, 0x04,                    //   Report Size (4)                   245
 0x81, 0x02,                    //   Input (Data,Var,Abs)              247
};

/*
 * we replace the above report descriptor extract
 * with the one below.
 * To make things equal in size, we take out a larger
 * portion than just the "Assign Selection" range, because
 * we need to insert a new application collection to force
 * the kernel to use BTN_TRIGGER_HAPPY[4-7].
 */
const __u8 fixed_rdesc_assign_selection[] = {
 0x0a, 0x99, 0x00,              //   Usage (Media Select Security)     211
 0x15, 0x00,                    //   Logical Minimum (0)               214
 0x26, 0xff, 0x00,              //   Logical Maximum (255)             216
 0x95, 0x01,                    //   Report Count (1)                  219
 0x75, 0x04,                    //   Report Size (4)                   221
 0x81, 0x02,                    //   Input (Data,Var,Abs)              223
 /* 0x15, 0x00, */              //   Logical Minimum (0)               ignored
 0x25, 0x01,                    //   Logical Maximum (1)               225
 0x95, 0x04,                    //   Report Count (4)                  227
 0x75, 0x01,                    //   Report Size (1)                   229
 0x81, 0x03,                    //   Input (Cnst,Var,Abs)              231
 0xc0,                          //  End Collection                     233
 0x05, 0x01,                    //  Usage Page (Generic Desktop)       234
 0x0a, 0x05, 0x00,              //  Usage (Game Pad)                   236
 0xa1, 0x01,                    //  Collection (Application)           239
 0x05, 0x09,                    //   Usage Page (Button)               241
 0x19, 0x15,                    //   Usage Minimum (21)                243
 0x29, 0x18,                    //   Usage Maximum (24)                245
 /* 0x15, 0x00, */              //  Logical Minimum (0)                ignored
 /* 0x25, 0x01, */              //  Logical Maximum (1)                ignored
 /* 0x95, 0x01, */              //  Report Size (1)                    ignored
 /* 0x75, 0x04, */              //  Report Count (4)                   ignored
 0x81, 0x02,                    //   Input (Data,Var,Abs)              247
};

_Static_assert(sizeof(rdesc_assign_selection) == sizeof(fixed_rdesc_assign_selection),
        "Rdesc and fixed rdesc of different size");
_Static_assert(sizeof(rdesc_assign_selection) + OFFSET_ASSIGN_SELECTION < ORIGINAL_RDESC_SIZE,
        "Rdesc at given offset is too big");

SEC(HID_BPF_RDESC_FIXUP)
int BPF_PROG(hid_fix_rdesc, struct hid_bpf_ctx *hctx)
{
 __u8 *data = hid_bpf_get_data(hctx, 0 /* offset */, 4096 /* size */);

 if (!data)
  return 0; /* EPERM check */

 /* Check that the device is compatible */
 if (__builtin_memcmp(data + OFFSET_ASSIGN_SELECTION,
        rdesc_assign_selection,
        sizeof(rdesc_assign_selection)))
  return 0;

 __builtin_memcpy(data + OFFSET_ASSIGN_SELECTION,
    fixed_rdesc_assign_selection,
    sizeof(fixed_rdesc_assign_selection));

 return 0;
}

HID_BPF_OPS(xbox_elite_2) = {
 .hid_rdesc_fixup = (void *)hid_fix_rdesc,
};

SEC("syscall")
int probe(struct hid_bpf_probe_args *ctx)
{
 /* only bind to the keyboard interface */
 ctx->retval = ctx->rdesc_size != ORIGINAL_RDESC_SIZE;
 if (ctx->retval)
  ctx->retval = -EINVAL;

 if (__builtin_memcmp(ctx->rdesc + OFFSET_ASSIGN_SELECTION,
        rdesc_assign_selection,
        sizeof(rdesc_assign_selection)))
  ctx->retval = -EINVAL;

 return 0;
}

char _license[] SEC("license") = "GPL";