// 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";