summaryrefslogtreecommitdiff
path: root/starbit.c
blob: 44d5ae3cb7c29a35cf2484274e291b59aa2ef95a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#include "linux/input.h"
#include "linux/kern_levels.h"
#include "linux/printk.h"
#include <linux/usb.h>
#include <linux/module.h>
#include <linux/kernel.h> 
#include <linux/init.h>
#include <linux/device.h>
#include <linux/hid.h>
#include "startbit.h"

struct wiimouse_dev {
    struct hid_device* hdev;
    struct input_dev* in_dev;
};


static const struct hid_device_id wiimouse_ids[] = {
	{HID_BLUETOOTH_DEVICE(NINTENDO_WIIMOTE_VENDOR_ID, NINTENDO_WIIMOTE_PRODUCT_ID)},
	{}
};



/*
 * wiimote_init
 * Initialize certain aspects of the wiimote such as the LEDs and sensors by
 * sending appropriate HID reports to the device. Returns 0 on success, non-zero
 * on error
 * Details about the report details can be found in https://wiibrew.org/wiki/Wiimote
 */
static int wiimote_init(struct hid_device* hdev) {
    int ret = 0;
    __u8 rumble[2] = { HID_REPORT_ID_WIIMOTE_RUMBLE_ENABLE, HID_REPORT_DATA_WIIMOTE_RUMBLE_ENABLE };
    if ((ret = hid_hw_output_report(hdev, (__u8*)rumble, 2)) < 0) return ret;
    __u8 led_p1[2] = { HID_REPORT_ID_WIIMOTE_LED_ENABLE, HID_REPORT_DATA_WIIMOTE_LED_ENABLE_P1 };
    if ((ret = hid_hw_output_report(hdev, (__u8*)led_p1, 2)) < 0) return ret;

    __u8 enable_ir1[2] = { HID_REPORT_ID_WIIMOTE_IR1_ENABLE, HID_REPORT_DATA_WIIMOTE_IR1_ENABLE };
    if ((ret = hid_hw_output_report(hdev, (__u8*)enable_ir1, 2)) < 0) return ret;
    __u8 enable_ir2[2] = { HID_REPORT_ID_WIIMOTE_IR2_ENABLE, HID_REPORT_DATA_WIIMOTE_IR2_ENABLE };
    if ((ret = hid_hw_output_report(hdev, (__u8*)enable_ir2, 2)) < 0) return ret;

    return ret;
}


static int wiimote_event(struct hid_device* hdev, struct hid_report* report, u8 *data, int size) {
    printk(KERN_DEBUG "Report ID: %02x | Buttons: %02x %02x %02x\n", report->id, data[0], data[1], data[2]);


    if (data[2] & 0x08) {
        printk(KERN_DEBUG "A button was pressed\n");
    }
    return 0;
}

MODULE_DEVICE_TABLE(hid, wiimouse_ids);
static int wiimouse_probe(struct hid_device *hdev, const struct hid_device_id *id)
{
    int ret = 0;
    struct wiimouse_dev* wdev;
    wdev = devm_kzalloc(&hdev->dev, sizeof(struct wiimouse_dev), GFP_KERNEL);
    wdev->hdev = hdev;
    struct input_dev* in_dev = input_allocate_device();
    if(in_dev == NULL) {
        return -1;
    }


    printk(KERN_DEBUG "wiimouse-mouse: Starting parse");
    /* Init hid device, setup input subsystem etc. */
    ret = hid_parse(hdev);
    if (ret) return ret;
    ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
    if (ret) return ret;

    wiimote_init(hdev);
    /* Init done, device ready to use */
    printk(KERN_DEBUG "Hello I'm a wiimouse!");
    return 0;
}


static void wiimote_report(struct hid_device *hdev, struct hid_report *report) {
    printk(KERN_DEBUG "Report ID: %02x\n", report->id);
}

static struct hid_driver wiimouse_driver = {
	.name = "wiimouse-mouse",
	.id_table = wiimouse_ids,
	.probe = wiimouse_probe, 
        .raw_event = wiimote_event,
        .report = wiimote_report,
};

module_hid_driver(wiimouse_driver);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Aaditya Dhruv");
MODULE_DESCRIPTION("Use your wiimote as a simple mouse");