mirror of
https://github.com/torvalds/linux.git
synced 2024-11-25 05:32:00 +00:00
e342d6f6f7
This is mostly mechanical: attach_prog is dropped, and the SEC are converted into struct_ops. Link: https://lore.kernel.org/r/20240608-hid_bpf_struct_ops-v3-5-6ac6ade58329@kernel.org Signed-off-by: Benjamin Tissoires <bentiss@kernel.org>
139 lines
2.9 KiB
C
139 lines
2.9 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/* Copyright (c) 2022 Benjamin Tissoires
|
|
*
|
|
* This is a pure HID-BPF example, and should be considered as such:
|
|
* on the Etekcity Scroll 6E, the X and Y axes will be swapped and
|
|
* inverted. On any other device... Not sure what this will do.
|
|
*
|
|
* This C main file is generic though. To adapt the code and test, users
|
|
* must amend only the .bpf.c file, which this program will load any
|
|
* eBPF program it finds.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <libgen.h>
|
|
#include <signal.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/resource.h>
|
|
#include <unistd.h>
|
|
|
|
#include <linux/bpf.h>
|
|
#include <linux/errno.h>
|
|
|
|
#include <bpf/bpf.h>
|
|
#include <bpf/libbpf.h>
|
|
|
|
#include "hid_mouse.skel.h"
|
|
|
|
static bool running = true;
|
|
|
|
static void int_exit(int sig)
|
|
{
|
|
running = false;
|
|
exit(0);
|
|
}
|
|
|
|
static void usage(const char *prog)
|
|
{
|
|
fprintf(stderr,
|
|
"%s: %s /sys/bus/hid/devices/0BUS:0VID:0PID:00ID\n\n",
|
|
__func__, prog);
|
|
fprintf(stderr,
|
|
"This program will upload and attach a HID-BPF program to the given device.\n"
|
|
"On the Etekcity Scroll 6E, the X and Y axis will be inverted, but on any other\n"
|
|
"device, chances are high that the device will not be working anymore\n\n"
|
|
"consider this as a demo and adapt the eBPF program to your needs\n"
|
|
"Hit Ctrl-C to unbind the program and reset the device\n");
|
|
}
|
|
|
|
static int get_hid_id(const char *path)
|
|
{
|
|
const char *str_id, *dir;
|
|
char uevent[1024];
|
|
int fd;
|
|
|
|
memset(uevent, 0, sizeof(uevent));
|
|
snprintf(uevent, sizeof(uevent) - 1, "%s/uevent", path);
|
|
|
|
fd = open(uevent, O_RDONLY | O_NONBLOCK);
|
|
if (fd < 0)
|
|
return -ENOENT;
|
|
|
|
close(fd);
|
|
|
|
dir = basename((char *)path);
|
|
|
|
str_id = dir + sizeof("0003:0001:0A37.");
|
|
return (int)strtol(str_id, NULL, 16);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
struct hid_mouse *skel;
|
|
struct bpf_link *link;
|
|
int err;
|
|
const char *optstr = "";
|
|
const char *sysfs_path;
|
|
int opt, hid_id;
|
|
|
|
while ((opt = getopt(argc, argv, optstr)) != -1) {
|
|
switch (opt) {
|
|
default:
|
|
usage(basename(argv[0]));
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if (optind == argc) {
|
|
usage(basename(argv[0]));
|
|
return 1;
|
|
}
|
|
|
|
sysfs_path = argv[optind];
|
|
if (!sysfs_path) {
|
|
perror("sysfs");
|
|
return 1;
|
|
}
|
|
|
|
skel = hid_mouse__open();
|
|
if (!skel) {
|
|
fprintf(stderr, "%s %s:%d", __func__, __FILE__, __LINE__);
|
|
return -1;
|
|
}
|
|
|
|
hid_id = get_hid_id(sysfs_path);
|
|
|
|
if (hid_id < 0) {
|
|
fprintf(stderr, "can not open HID device: %m\n");
|
|
return 1;
|
|
}
|
|
skel->struct_ops.mouse_invert->hid_id = hid_id;
|
|
|
|
err = hid_mouse__load(skel);
|
|
if (err < 0) {
|
|
fprintf(stderr, "can not load HID-BPF program: %m\n");
|
|
return 1;
|
|
}
|
|
|
|
link = bpf_map__attach_struct_ops(skel->maps.mouse_invert);
|
|
if (!link) {
|
|
fprintf(stderr, "can not attach HID-BPF program: %m\n");
|
|
return 1;
|
|
}
|
|
|
|
signal(SIGINT, int_exit);
|
|
signal(SIGTERM, int_exit);
|
|
|
|
while (running)
|
|
sleep(1);
|
|
|
|
hid_mouse__destroy(skel);
|
|
|
|
return 0;
|
|
}
|