[linux-2.6-block.git] / drivers / hid / hid-roccat-arvo.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Roccat Arvo driver for Linux
 *
 * Copyright (c) 2011 Stefan Achatz <erazor_de@users.sourceforge.net>
 */

/*
 */

/*
 * Roccat Arvo is a gamer keyboard with 5 macro keys that can be configured in
 * 5 profiles.
 */

#include <linux/device.h>
#include <linux/input.h>
#include <linux/hid.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/hid-roccat.h>
#include "hid-ids.h"
#include "hid-roccat-common.h"
#include "hid-roccat-arvo.h"

static struct class *arvo_class;

static ssize_t arvo_sysfs_show_mode_key(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	struct usb_device *usb_dev =
			interface_to_usbdev(to_usb_interface(dev->parent->parent));
	struct arvo_mode_key temp_buf;
	int retval;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_MODE_KEY,
			&temp_buf, sizeof(struct arvo_mode_key));
	mutex_unlock(&arvo->arvo_lock);
	if (retval)
		return retval;

	return snprintf(buf, PAGE_SIZE, "%d\n", temp_buf.state);
}

static ssize_t arvo_sysfs_set_mode_key(struct device *dev,
		struct device_attribute *attr, char const *buf, size_t size)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	struct usb_device *usb_dev =
			interface_to_usbdev(to_usb_interface(dev->parent->parent));
	struct arvo_mode_key temp_buf;
	unsigned long state;
	int retval;

	retval = kstrtoul(buf, 10, &state);
	if (retval)
		return retval;

	temp_buf.command = ARVO_COMMAND_MODE_KEY;
	temp_buf.state = state;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_MODE_KEY,
			&temp_buf, sizeof(struct arvo_mode_key));
	mutex_unlock(&arvo->arvo_lock);
	if (retval)
		return retval;

	return size;
}
static DEVICE_ATTR(mode_key, 0660,
		   arvo_sysfs_show_mode_key, arvo_sysfs_set_mode_key);

static ssize_t arvo_sysfs_show_key_mask(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	struct usb_device *usb_dev =
			interface_to_usbdev(to_usb_interface(dev->parent->parent));
	struct arvo_key_mask temp_buf;
	int retval;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_KEY_MASK,
			&temp_buf, sizeof(struct arvo_key_mask));
	mutex_unlock(&arvo->arvo_lock);
	if (retval)
		return retval;

	return snprintf(buf, PAGE_SIZE, "%d\n", temp_buf.key_mask);
}

static ssize_t arvo_sysfs_set_key_mask(struct device *dev,
		struct device_attribute *attr, char const *buf, size_t size)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	struct usb_device *usb_dev =
			interface_to_usbdev(to_usb_interface(dev->parent->parent));
	struct arvo_key_mask temp_buf;
	unsigned long key_mask;
	int retval;

	retval = kstrtoul(buf, 10, &key_mask);
	if (retval)
		return retval;

	temp_buf.command = ARVO_COMMAND_KEY_MASK;
	temp_buf.key_mask = key_mask;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_KEY_MASK,
			&temp_buf, sizeof(struct arvo_key_mask));
	mutex_unlock(&arvo->arvo_lock);
	if (retval)
		return retval;

	return size;
}
static DEVICE_ATTR(key_mask, 0660,
		   arvo_sysfs_show_key_mask, arvo_sysfs_set_key_mask);

/* retval is 1-5 on success, < 0 on error */
static int arvo_get_actual_profile(struct usb_device *usb_dev)
{
	struct arvo_actual_profile temp_buf;
	int retval;

	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_ACTUAL_PROFILE,
			&temp_buf, sizeof(struct arvo_actual_profile));

	if (retval)
		return retval;

	return temp_buf.actual_profile;
}

static ssize_t arvo_sysfs_show_actual_profile(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));

	return snprintf(buf, PAGE_SIZE, "%d\n", arvo->actual_profile);
}

static ssize_t arvo_sysfs_set_actual_profile(struct device *dev,
		struct device_attribute *attr, char const *buf, size_t size)
{
	struct arvo_device *arvo =
			hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	struct usb_device *usb_dev =
			interface_to_usbdev(to_usb_interface(dev->parent->parent));
	struct arvo_actual_profile temp_buf;
	unsigned long profile;
	int retval;

	retval = kstrtoul(buf, 10, &profile);
	if (retval)
		return retval;

	if (profile < 1 || profile > 5)
		return -EINVAL;

	temp_buf.command = ARVO_COMMAND_ACTUAL_PROFILE;
	temp_buf.actual_profile = profile;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_ACTUAL_PROFILE,
			&temp_buf, sizeof(struct arvo_actual_profile));
	if (!retval) {
		arvo->actual_profile = profile;
		retval = size;
	}
	mutex_unlock(&arvo->arvo_lock);
	return retval;
}
static DEVICE_ATTR(actual_profile, 0660,
		   arvo_sysfs_show_actual_profile,
		   arvo_sysfs_set_actual_profile);

static ssize_t arvo_sysfs_write(struct file *fp,
		struct kobject *kobj, void const *buf,
		loff_t off, size_t count, size_t real_size, uint command)
{
	struct device *dev = kobj_to_dev(kobj)->parent->parent;
	struct arvo_device *arvo = hid_get_drvdata(dev_get_drvdata(dev));
	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
	int retval;

	if (off != 0 || count != real_size)
		return -EINVAL;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_send(usb_dev, command, buf, real_size);
	mutex_unlock(&arvo->arvo_lock);

	return (retval ? retval : real_size);
}

static ssize_t arvo_sysfs_read(struct file *fp,
		struct kobject *kobj, void *buf, loff_t off,
		size_t count, size_t real_size, uint command)
{
	struct device *dev = kobj_to_dev(kobj)->parent->parent;
	struct arvo_device *arvo = hid_get_drvdata(dev_get_drvdata(dev));
	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
	int retval;

	if (off >= real_size)
		return 0;

	if (off != 0 || count != real_size)
		return -EINVAL;

	mutex_lock(&arvo->arvo_lock);
	retval = roccat_common2_receive(usb_dev, command, buf, real_size);
	mutex_unlock(&arvo->arvo_lock);

	return (retval ? retval : real_size);
}

static ssize_t arvo_sysfs_write_button(struct file *fp,
		struct kobject *kobj, struct bin_attribute *attr, char *buf,
		loff_t off, size_t count)
{
	return arvo_sysfs_write(fp, kobj, buf, off, count,
			sizeof(struct arvo_button), ARVO_COMMAND_BUTTON);
}
static BIN_ATTR(button, 0220, NULL, arvo_sysfs_write_button,
		sizeof(struct arvo_button));

static ssize_t arvo_sysfs_read_info(struct file *fp,
		struct kobject *kobj, struct bin_attribute *attr, char *buf,
		loff_t off, size_t count)
{
	return arvo_sysfs_read(fp, kobj, buf, off, count,
			sizeof(struct arvo_info), ARVO_COMMAND_INFO);
}
static BIN_ATTR(info, 0440, arvo_sysfs_read_info, NULL,
		sizeof(struct arvo_info));

static struct attribute *arvo_attrs[] = {
	&dev_attr_mode_key.attr,
	&dev_attr_key_mask.attr,
	&dev_attr_actual_profile.attr,
	NULL,
};

static struct bin_attribute *arvo_bin_attributes[] = {
	&bin_attr_button,
	&bin_attr_info,
	NULL,
};

static const struct attribute_group arvo_group = {
	.attrs = arvo_attrs,
	.bin_attrs = arvo_bin_attributes,
};

static const struct attribute_group *arvo_groups[] = {
	&arvo_group,
	NULL,
};

static int arvo_init_arvo_device_struct(struct usb_device *usb_dev,
		struct arvo_device *arvo)
{
	int retval;

	mutex_init(&arvo->arvo_lock);

	retval = arvo_get_actual_profile(usb_dev);
	if (retval < 0)
		return retval;
	arvo->actual_profile = retval;

	return 0;
}

static int arvo_init_specials(struct hid_device *hdev)
{
	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
	struct usb_device *usb_dev = interface_to_usbdev(intf);
	struct arvo_device *arvo;
	int retval;

	if (intf->cur_altsetting->desc.bInterfaceProtocol
			== USB_INTERFACE_PROTOCOL_KEYBOARD) {
		hid_set_drvdata(hdev, NULL);
		return 0;
	}

	arvo = kzalloc(sizeof(*arvo), GFP_KERNEL);
	if (!arvo) {
		hid_err(hdev, "can't alloc device descriptor\n");
		return -ENOMEM;
	}
	hid_set_drvdata(hdev, arvo);

	retval = arvo_init_arvo_device_struct(usb_dev, arvo);
	if (retval) {
		hid_err(hdev, "couldn't init struct arvo_device\n");
		goto exit_free;
	}

	retval = roccat_connect(arvo_class, hdev,
			sizeof(struct arvo_roccat_report));
	if (retval < 0) {
		hid_err(hdev, "couldn't init char dev\n");
	} else {
		arvo->chrdev_minor = retval;
		arvo->roccat_claimed = 1;
	}

	return 0;
exit_free:
	kfree(arvo);
	return retval;
}

static void arvo_remove_specials(struct hid_device *hdev)
{
	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
	struct arvo_device *arvo;

	if (intf->cur_altsetting->desc.bInterfaceProtocol
			== USB_INTERFACE_PROTOCOL_KEYBOARD)
		return;

	arvo = hid_get_drvdata(hdev);
	if (arvo->roccat_claimed)
		roccat_disconnect(arvo->chrdev_minor);
	kfree(arvo);
}

static int arvo_probe(struct hid_device *hdev,
		const struct hid_device_id *id)
{
	int retval;

	retval = hid_parse(hdev);
	if (retval) {
		hid_err(hdev, "parse failed\n");
		goto exit;
	}

	retval = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
	if (retval) {
		hid_err(hdev, "hw start failed\n");
		goto exit;
	}

	retval = arvo_init_specials(hdev);
	if (retval) {
		hid_err(hdev, "couldn't install keyboard\n");
		goto exit_stop;
	}

	return 0;

exit_stop:
	hid_hw_stop(hdev);
exit:
	return retval;
}

static void arvo_remove(struct hid_device *hdev)
{
	arvo_remove_specials(hdev);
	hid_hw_stop(hdev);
}

static void arvo_report_to_chrdev(struct arvo_device const *arvo,
		u8 const *data)
{
	struct arvo_special_report const *special_report;
	struct arvo_roccat_report roccat_report;

	special_report = (struct arvo_special_report const *)data;

	roccat_report.profile = arvo->actual_profile;
	roccat_report.button = special_report->event &
			ARVO_SPECIAL_REPORT_EVENT_MASK_BUTTON;
	if ((special_report->event & ARVO_SPECIAL_REPORT_EVENT_MASK_ACTION) ==
			ARVO_SPECIAL_REPORT_EVENT_ACTION_PRESS)
		roccat_report.action = ARVO_ROCCAT_REPORT_ACTION_PRESS;
	else
		roccat_report.action = ARVO_ROCCAT_REPORT_ACTION_RELEASE;

	roccat_report_event(arvo->chrdev_minor,
			(uint8_t const *)&roccat_report);
}

static int arvo_raw_event(struct hid_device *hdev,
		struct hid_report *report, u8 *data, int size)
{
	struct arvo_device *arvo = hid_get_drvdata(hdev);

	if (size != 3)
		return 0;

	if (arvo && arvo->roccat_claimed)
		arvo_report_to_chrdev(arvo, data);

	return 0;
}

static const struct hid_device_id arvo_devices[] = {
	{ HID_USB_DEVICE(USB_VENDOR_ID_ROCCAT, USB_DEVICE_ID_ROCCAT_ARVO) },
	{ }
};

MODULE_DEVICE_TABLE(hid, arvo_devices);

static struct hid_driver arvo_driver = {
	.name = "arvo",
	.id_table = arvo_devices,
	.probe = arvo_probe,
	.remove = arvo_remove,
	.raw_event = arvo_raw_event
};

static int __init arvo_init(void)
{
	int retval;

	arvo_class = class_create(THIS_MODULE, "arvo");
	if (IS_ERR(arvo_class))
		return PTR_ERR(arvo_class);
	arvo_class->dev_groups = arvo_groups;

	retval = hid_register_driver(&arvo_driver);
	if (retval)
		class_destroy(arvo_class);
	return retval;
}

static void __exit arvo_exit(void)
{
	hid_unregister_driver(&arvo_driver);
	class_destroy(arvo_class);
}

module_init(arvo_init);
module_exit(arvo_exit);

MODULE_AUTHOR("Stefan Achatz");
MODULE_DESCRIPTION("USB Roccat Arvo driver");
MODULE_LICENSE("GPL v2");
Commit	Line	Data
2874c5fd	1	// SPDX-License-Identifier: GPL-2.0-or-later
e68cc603 SA	2	/*
	3	* Roccat Arvo driver for Linux
	4	*
	5	* Copyright (c) 2011 Stefan Achatz <erazor_de@users.sourceforge.net>
	6	*/
	7
	8	/*
e68cc603 SA	9	*/
	10
	11	/*
	12	* Roccat Arvo is a gamer keyboard with 5 macro keys that can be configured in
	13	* 5 profiles.
	14	*/
	15
	16	#include <linux/device.h>
	17	#include <linux/input.h>
	18	#include <linux/hid.h>
e68cc603 SA	19	#include <linux/module.h>
e68cc603 SA	20	#include <linux/slab.h>
5dc0c983	21	#include <linux/hid-roccat.h>
e68cc603	22	#include "hid-ids.h"
5772f636	23	#include "hid-roccat-common.h"
e68cc603 SA	24	#include "hid-roccat-arvo.h"
	25
	26	static struct class *arvo_class;
	27
e68cc603 SA	28	static ssize_t arvo_sysfs_show_mode_key(struct device *dev,
	29	struct device_attribute attr, char buf)
	30	{
	31	struct arvo_device *arvo =
	32	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	33	struct usb_device *usb_dev =
	34	interface_to_usbdev(to_usb_interface(dev->parent->parent));
5772f636	35	struct arvo_mode_key temp_buf;
e68cc603 SA	36	int retval;
e68cc603 SA	37
e68cc603	38	mutex_lock(&arvo->arvo_lock);
7392d73b	39	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_MODE_KEY,
5772f636	40	&temp_buf, sizeof(struct arvo_mode_key));
e68cc603 SA	41	mutex_unlock(&arvo->arvo_lock);
e68cc603 SA	42	if (retval)
5772f636	43	return retval;
e68cc603	44
5772f636	45	return snprintf(buf, PAGE_SIZE, "%d\n", temp_buf.state);
e68cc603 SA	46	}
	47
	48	static ssize_t arvo_sysfs_set_mode_key(struct device *dev,
	49	struct device_attribute attr, char const buf, size_t size)
	50	{
	51	struct arvo_device *arvo =
	52	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	53	struct usb_device *usb_dev =
	54	interface_to_usbdev(to_usb_interface(dev->parent->parent));
5772f636	55	struct arvo_mode_key temp_buf;
e68cc603 SA	56	unsigned long state;
	57	int retval;
	58
dfc450b5	59	retval = kstrtoul(buf, 10, &state);
e68cc603	60	if (retval)
5772f636	61	return retval;
e68cc603	62
5772f636 SA	63	temp_buf.command = ARVO_COMMAND_MODE_KEY;
5772f636 SA	64	temp_buf.state = state;
e68cc603 SA	65
e68cc603 SA	66	mutex_lock(&arvo->arvo_lock);
7392d73b	67	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_MODE_KEY,
5772f636	68	&temp_buf, sizeof(struct arvo_mode_key));
e68cc603 SA	69	mutex_unlock(&arvo->arvo_lock);
e68cc603 SA	70	if (retval)
5772f636	71	return retval;
e68cc603	72
5772f636	73	return size;
e68cc603	74	}
46a58c44 GKH	75	static DEVICE_ATTR(mode_key, 0660,
46a58c44 GKH	76	arvo_sysfs_show_mode_key, arvo_sysfs_set_mode_key);
e68cc603 SA	77
	78	static ssize_t arvo_sysfs_show_key_mask(struct device *dev,
	79	struct device_attribute attr, char buf)
	80	{
	81	struct arvo_device *arvo =
	82	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	83	struct usb_device *usb_dev =
	84	interface_to_usbdev(to_usb_interface(dev->parent->parent));
5772f636	85	struct arvo_key_mask temp_buf;
e68cc603 SA	86	int retval;
e68cc603 SA	87
e68cc603	88	mutex_lock(&arvo->arvo_lock);
7392d73b	89	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_KEY_MASK,
5772f636	90	&temp_buf, sizeof(struct arvo_key_mask));
e68cc603 SA	91	mutex_unlock(&arvo->arvo_lock);
e68cc603 SA	92	if (retval)
5772f636	93	return retval;
e68cc603	94
5772f636	95	return snprintf(buf, PAGE_SIZE, "%d\n", temp_buf.key_mask);
e68cc603 SA	96	}
	97
	98	static ssize_t arvo_sysfs_set_key_mask(struct device *dev,
	99	struct device_attribute attr, char const buf, size_t size)
	100	{
	101	struct arvo_device *arvo =
	102	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	103	struct usb_device *usb_dev =
	104	interface_to_usbdev(to_usb_interface(dev->parent->parent));
5772f636	105	struct arvo_key_mask temp_buf;
e68cc603 SA	106	unsigned long key_mask;
	107	int retval;
	108
dfc450b5	109	retval = kstrtoul(buf, 10, &key_mask);
e68cc603	110	if (retval)
5772f636	111	return retval;
e68cc603	112
5772f636 SA	113	temp_buf.command = ARVO_COMMAND_KEY_MASK;
5772f636 SA	114	temp_buf.key_mask = key_mask;
e68cc603 SA	115
e68cc603 SA	116	mutex_lock(&arvo->arvo_lock);
7392d73b	117	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_KEY_MASK,
5772f636	118	&temp_buf, sizeof(struct arvo_key_mask));
e68cc603 SA	119	mutex_unlock(&arvo->arvo_lock);
e68cc603 SA	120	if (retval)
5772f636	121	return retval;
e68cc603	122
5772f636	123	return size;
e68cc603	124	}
46a58c44 GKH	125	static DEVICE_ATTR(key_mask, 0660,
46a58c44 GKH	126	arvo_sysfs_show_key_mask, arvo_sysfs_set_key_mask);
e68cc603 SA	127
	128	/* retval is 1-5 on success, < 0 on error */
	129	static int arvo_get_actual_profile(struct usb_device *usb_dev)
	130	{
5772f636	131	struct arvo_actual_profile temp_buf;
e68cc603 SA	132	int retval;
e68cc603 SA	133
7392d73b	134	retval = roccat_common2_receive(usb_dev, ARVO_COMMAND_ACTUAL_PROFILE,
5772f636	135	&temp_buf, sizeof(struct arvo_actual_profile));
e68cc603	136
5772f636 SA	137	if (retval)
5772f636 SA	138	return retval;
e68cc603	139
5772f636	140	return temp_buf.actual_profile;
e68cc603 SA	141	}
	142
	143	static ssize_t arvo_sysfs_show_actual_profile(struct device *dev,
	144	struct device_attribute attr, char buf)
	145	{
	146	struct arvo_device *arvo =
	147	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	148
	149	return snprintf(buf, PAGE_SIZE, "%d\n", arvo->actual_profile);
	150	}
	151
	152	static ssize_t arvo_sysfs_set_actual_profile(struct device *dev,
	153	struct device_attribute attr, char const buf, size_t size)
	154	{
	155	struct arvo_device *arvo =
	156	hid_get_drvdata(dev_get_drvdata(dev->parent->parent));
	157	struct usb_device *usb_dev =
	158	interface_to_usbdev(to_usb_interface(dev->parent->parent));
5772f636	159	struct arvo_actual_profile temp_buf;
e68cc603 SA	160	unsigned long profile;
	161	int retval;
	162
dfc450b5	163	retval = kstrtoul(buf, 10, &profile);
e68cc603	164	if (retval)
5772f636	165	return retval;
e68cc603	166
901e64db SA	167	if (profile < 1 \|\| profile > 5)
	168	return -EINVAL;
	169
5772f636 SA	170	temp_buf.command = ARVO_COMMAND_ACTUAL_PROFILE;
5772f636 SA	171	temp_buf.actual_profile = profile;
e68cc603 SA	172
e68cc603 SA	173	mutex_lock(&arvo->arvo_lock);
7392d73b	174	retval = roccat_common2_send(usb_dev, ARVO_COMMAND_ACTUAL_PROFILE,
5772f636	175	&temp_buf, sizeof(struct arvo_actual_profile));
e68cc603 SA	176	if (!retval) {
	177	arvo->actual_profile = profile;
	178	retval = size;
	179	}
	180	mutex_unlock(&arvo->arvo_lock);
e68cc603 SA	181	return retval;
e68cc603 SA	182	}
46a58c44 GKH	183	static DEVICE_ATTR(actual_profile, 0660,
	184	arvo_sysfs_show_actual_profile,
	185	arvo_sysfs_set_actual_profile);
e68cc603 SA	186
	187	static ssize_t arvo_sysfs_write(struct file *fp,
	188	struct kobject kobj, void const buf,
	189	loff_t off, size_t count, size_t real_size, uint command)
	190	{
2cf83833	191	struct device *dev = kobj_to_dev(kobj)->parent->parent;
e68cc603 SA	192	struct arvo_device *arvo = hid_get_drvdata(dev_get_drvdata(dev));
	193	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
	194	int retval;
	195
	196	if (off != 0 \|\| count != real_size)
	197	return -EINVAL;
	198
	199	mutex_lock(&arvo->arvo_lock);
7392d73b	200	retval = roccat_common2_send(usb_dev, command, buf, real_size);
e68cc603 SA	201	mutex_unlock(&arvo->arvo_lock);
	202
	203	return (retval ? retval : real_size);
	204	}
	205
	206	static ssize_t arvo_sysfs_read(struct file *fp,
	207	struct kobject kobj, void buf, loff_t off,
	208	size_t count, size_t real_size, uint command)
	209	{
2cf83833	210	struct device *dev = kobj_to_dev(kobj)->parent->parent;
e68cc603 SA	211	struct arvo_device *arvo = hid_get_drvdata(dev_get_drvdata(dev));
	212	struct usb_device *usb_dev = interface_to_usbdev(to_usb_interface(dev));
	213	int retval;
	214
	215	if (off >= real_size)
	216	return 0;
	217
	218	if (off != 0 \|\| count != real_size)
	219	return -EINVAL;
	220
	221	mutex_lock(&arvo->arvo_lock);
7392d73b	222	retval = roccat_common2_receive(usb_dev, command, buf, real_size);
e68cc603 SA	223	mutex_unlock(&arvo->arvo_lock);
	224
	225	return (retval ? retval : real_size);
	226	}
	227
	228	static ssize_t arvo_sysfs_write_button(struct file *fp,
	229	struct kobject kobj, struct bin_attribute attr, char *buf,
	230	loff_t off, size_t count)
	231	{
	232	return arvo_sysfs_write(fp, kobj, buf, off, count,
1edd5b42	233	sizeof(struct arvo_button), ARVO_COMMAND_BUTTON);
e68cc603	234	}
e0a00d86 GKH	235	static BIN_ATTR(button, 0220, NULL, arvo_sysfs_write_button,
e0a00d86 GKH	236	sizeof(struct arvo_button));
e68cc603 SA	237
	238	static ssize_t arvo_sysfs_read_info(struct file *fp,
	239	struct kobject kobj, struct bin_attribute attr, char *buf,
	240	loff_t off, size_t count)
	241	{
	242	return arvo_sysfs_read(fp, kobj, buf, off, count,
1edd5b42	243	sizeof(struct arvo_info), ARVO_COMMAND_INFO);
e68cc603	244	}
e0a00d86 GKH	245	static BIN_ATTR(info, 0440, arvo_sysfs_read_info, NULL,
e0a00d86 GKH	246	sizeof(struct arvo_info));
e68cc603	247
46a58c44 GKH	248	static struct attribute *arvo_attrs[] = {
	249	&dev_attr_mode_key.attr,
	250	&dev_attr_key_mask.attr,
	251	&dev_attr_actual_profile.attr,
	252	NULL,
e68cc603	253	};
e0a00d86 GKH	254
	255	static struct bin_attribute *arvo_bin_attributes[] = {
	256	&bin_attr_button,
	257	&bin_attr_info,
	258	NULL,
	259	};
	260
	261	static const struct attribute_group arvo_group = {
	262	.attrs = arvo_attrs,
	263	.bin_attrs = arvo_bin_attributes,
	264	};
	265
	266	static const struct attribute_group *arvo_groups[] = {
	267	&arvo_group,
	268	NULL,
e68cc603 SA	269	};
	270
	271	static int arvo_init_arvo_device_struct(struct usb_device *usb_dev,
	272	struct arvo_device *arvo)
	273	{
	274	int retval;
	275
	276	mutex_init(&arvo->arvo_lock);
	277
	278	retval = arvo_get_actual_profile(usb_dev);
	279	if (retval < 0)
	280	return retval;
	281	arvo->actual_profile = retval;
	282
	283	return 0;
	284	}
	285
	286	static int arvo_init_specials(struct hid_device *hdev)
	287	{
	288	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
	289	struct usb_device *usb_dev = interface_to_usbdev(intf);
	290	struct arvo_device *arvo;
	291	int retval;
	292
	293	if (intf->cur_altsetting->desc.bInterfaceProtocol
	294	== USB_INTERFACE_PROTOCOL_KEYBOARD) {
	295	hid_set_drvdata(hdev, NULL);
	296	return 0;
	297	}
	298
	299	arvo = kzalloc(sizeof(*arvo), GFP_KERNEL);
	300	if (!arvo) {
a28764ef	301	hid_err(hdev, "can't alloc device descriptor\n");
e68cc603 SA	302	return -ENOMEM;
	303	}
	304	hid_set_drvdata(hdev, arvo);
	305
	306	retval = arvo_init_arvo_device_struct(usb_dev, arvo);
	307	if (retval) {
a28764ef	308	hid_err(hdev, "couldn't init struct arvo_device\n");
e68cc603 SA	309	goto exit_free;
	310	}
	311
8211e460 SA	312	retval = roccat_connect(arvo_class, hdev,
8211e460 SA	313	sizeof(struct arvo_roccat_report));
e68cc603	314	if (retval < 0) {
a28764ef	315	hid_err(hdev, "couldn't init char dev\n");
e68cc603 SA	316	} else {
	317	arvo->chrdev_minor = retval;
	318	arvo->roccat_claimed = 1;
	319	}
	320
	321	return 0;
	322	exit_free:
	323	kfree(arvo);
	324	return retval;
	325	}
	326
	327	static void arvo_remove_specials(struct hid_device *hdev)
	328	{
	329	struct usb_interface *intf = to_usb_interface(hdev->dev.parent);
	330	struct arvo_device *arvo;
	331
	332	if (intf->cur_altsetting->desc.bInterfaceProtocol
	333	== USB_INTERFACE_PROTOCOL_KEYBOARD)
	334	return;
	335
	336	arvo = hid_get_drvdata(hdev);
	337	if (arvo->roccat_claimed)
	338	roccat_disconnect(arvo->chrdev_minor);
	339	kfree(arvo);
	340	}
	341
	342	static int arvo_probe(struct hid_device *hdev,
	343	const struct hid_device_id *id)
	344	{
	345	int retval;
	346
	347	retval = hid_parse(hdev);
	348	if (retval) {
a28764ef	349	hid_err(hdev, "parse failed\n");
e68cc603 SA	350	goto exit;
	351	}
	352
	353	retval = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
	354	if (retval) {
a28764ef	355	hid_err(hdev, "hw start failed\n");
e68cc603 SA	356	goto exit;
	357	}
	358
	359	retval = arvo_init_specials(hdev);
	360	if (retval) {
a28764ef	361	hid_err(hdev, "couldn't install keyboard\n");
e68cc603 SA	362	goto exit_stop;
	363	}
	364
	365	return 0;
	366
	367	exit_stop:
	368	hid_hw_stop(hdev);
	369	exit:
	370	return retval;
	371	}
	372
	373	static void arvo_remove(struct hid_device *hdev)
	374	{
	375	arvo_remove_specials(hdev);
	376	hid_hw_stop(hdev);
	377	}
	378
	379	static void arvo_report_to_chrdev(struct arvo_device const *arvo,
	380	u8 const *data)
	381	{
	382	struct arvo_special_report const *special_report;
	383	struct arvo_roccat_report roccat_report;
	384
	385	special_report = (struct arvo_special_report const *)data;
	386
	387	roccat_report.profile = arvo->actual_profile;
	388	roccat_report.button = special_report->event &
	389	ARVO_SPECIAL_REPORT_EVENT_MASK_BUTTON;
	390	if ((special_report->event & ARVO_SPECIAL_REPORT_EVENT_MASK_ACTION) ==
	391	ARVO_SPECIAL_REPORT_EVENT_ACTION_PRESS)
	392	roccat_report.action = ARVO_ROCCAT_REPORT_ACTION_PRESS;
	393	else
	394	roccat_report.action = ARVO_ROCCAT_REPORT_ACTION_RELEASE;
	395
8211e460 SA	396	roccat_report_event(arvo->chrdev_minor,
8211e460 SA	397	(uint8_t const *)&roccat_report);
e68cc603 SA	398	}
	399
	400	static int arvo_raw_event(struct hid_device *hdev,
	401	struct hid_report report, u8 data, int size)
	402	{
	403	struct arvo_device *arvo = hid_get_drvdata(hdev);
	404
	405	if (size != 3)
	406	return 0;
	407
901e64db	408	if (arvo && arvo->roccat_claimed)
e68cc603 SA	409	arvo_report_to_chrdev(arvo, data);
	410
	411	return 0;
	412	}
	413
	414	static const struct hid_device_id arvo_devices[] = {
	415	{ HID_USB_DEVICE(USB_VENDOR_ID_ROCCAT, USB_DEVICE_ID_ROCCAT_ARVO) },
	416	{ }
	417	};
	418
	419	MODULE_DEVICE_TABLE(hid, arvo_devices);
	420
	421	static struct hid_driver arvo_driver = {
	422	.name = "arvo",
	423	.id_table = arvo_devices,
	424	.probe = arvo_probe,
	425	.remove = arvo_remove,
	426	.raw_event = arvo_raw_event
	427	};
	428
	429	static int __init arvo_init(void)
	430	{
	431	int retval;
	432
	433	arvo_class = class_create(THIS_MODULE, "arvo");
	434	if (IS_ERR(arvo_class))
	435	return PTR_ERR(arvo_class);
46a58c44	436	arvo_class->dev_groups = arvo_groups;
e68cc603 SA	437
	438	retval = hid_register_driver(&arvo_driver);
	439	if (retval)
	440	class_destroy(arvo_class);
	441	return retval;
	442	}
	443
	444	static void __exit arvo_exit(void)
	445	{
e68cc603	446	hid_unregister_driver(&arvo_driver);
74b643da	447	class_destroy(arvo_class);
e68cc603 SA	448	}
	449
	450	module_init(arvo_init);
	451	module_exit(arvo_exit);
	452
	453	MODULE_AUTHOR("Stefan Achatz");
	454	MODULE_DESCRIPTION("USB Roccat Arvo driver");
	455	MODULE_LICENSE("GPL v2");