Quelle  hid-razer.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0+
/*
 *  HID driver for gaming keys on Razer Blackwidow gaming keyboards
 *  Macro Key Keycodes: M1 = 191, M2 = 192, M3 = 193, M4 = 194, M5 = 195
 *
 *  Copyright (c) 2021 Jelle van der Waa <jvanderwaa@redhat.com>
 */

#include <linux/device.h>
#include <linux/hid.h>
#include <linux/module.h>
#include <linux/random.h>
#include <linux/sched.h>
#include <linux/usb.h>
#include <linux/wait.h>

#include "hid-ids.h"

#define map_key_clear(c) hid_map_usage_clear(hi, usage, bit, max, EV_KEY, (c))

#define RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE 91

static bool macro_key_remapping = 1;
module_param(macro_key_remapping, bool, 0644);
MODULE_PARM_DESC(macro_key_remapping, " on (Y) off (N)");


static unsigned char blackwidow_init[RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE] = {
 0x00,
 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x04,
 0x02, 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, 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, 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, 0x00, 0x00, 0x00, 0x00,
 0x04, 0x00
};

static int razer_input_mapping(struct hid_device *hdev,
  struct hid_input *hi, struct hid_field *field,
  struct hid_usage *usage, unsigned long **bit, int *max)
{

 if (!macro_key_remapping)
  return 0;

 if ((usage->hid & HID_UP_KEYBOARD) != HID_UP_KEYBOARD)
  return 0;

 switch (usage->hid & ~HID_UP_KEYBOARD) {
 case 0x68:
  map_key_clear(KEY_MACRO1);
  return 1;
 case 0x69:
  map_key_clear(KEY_MACRO2);
  return 1;
 case 0x6a:
  map_key_clear(KEY_MACRO3);
  return 1;
 case 0x6b:
  map_key_clear(KEY_MACRO4);
  return 1;
 case 0x6c:
  map_key_clear(KEY_MACRO5);
  return 1;
 }

 return 0;
}

static int razer_probe(struct hid_device *hdev, const struct hid_device_id *id)
{
 char *buf;
 int ret = 0;

 ret = hid_parse(hdev);
 if (ret)
  return ret;

 /*
  * Only send the enable macro keys command for the third device
  * identified as mouse input.
 */
 if (hdev->type == HID_TYPE_USBMOUSE) {
  buf = kmemdup(blackwidow_init, RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE, GFP_KERNEL);
  if (buf == NULL)
   return -ENOMEM;

  ret = hid_hw_raw_request(hdev, 0, buf, RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE,
    HID_FEATURE_REPORT, HID_REQ_SET_REPORT);
  if (ret != RAZER_BLACKWIDOW_TRANSFER_BUF_SIZE)
   hid_err(hdev, "failed to enable macro keys: %d\n", ret);

  kfree(buf);
 }

 return hid_hw_start(hdev, HID_CONNECT_DEFAULT);
}

static const struct hid_device_id razer_devices[] = {
 { HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
  USB_DEVICE_ID_RAZER_BLACKWIDOW) },
 { HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
  USB_DEVICE_ID_RAZER_BLACKWIDOW_CLASSIC) },
 { HID_USB_DEVICE(USB_VENDOR_ID_RAZER,
  USB_DEVICE_ID_RAZER_BLACKWIDOW_ULTIMATE) },
 { }
};
MODULE_DEVICE_TABLE(hid, razer_devices);

static struct hid_driver razer_driver = {
 .name = "razer",
 .id_table = razer_devices,
 .input_mapping = razer_input_mapping,
 .probe = razer_probe,
};
module_hid_driver(razer_driver);

MODULE_AUTHOR("Jelle van der Waa <jvanderwaa@redhat.com>");
MODULE_DESCRIPTION("HID driver for gaming keys on Razer Blackwidow gaming keyboards");
MODULE_LICENSE("GPL");