xp_pen_hack/xp_pen_hack.cpp at main · rhighs/xp_pen_hack · GitHub

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
102
103
104
105
106
107
108
109
110
111
112
#include <fstream>
#include <iostream>
#include <linux/input.h>
#include <regex>
#include <time.h>
#include <optional>
#include <unistd.h>
#include <vector>
#include <memory>

static bool verbosity = false;
static int magic_evt_code = 331;
static const char* device_file_path = "/proc/bus/input/devices";

typedef struct device {
    std::string name;
    std::string evt_n;
} device_t;

auto pack_devices(std::vector<device_t*>& devices) -> void {
    std::ifstream device_file { std::string(device_file_path), std::ios::in };
    if (!device_file) {
        std::cerr << "Could not open device file!" << std::endl;
        exit(1);
    }

    std::string number;
    std::smatch matches;
    std::string name_key = "Name=\"";
    std::regex event_nr_rx("event([0-9]+)");

    auto get_names = [&]() -> bool {
        device_t* to_add = new device_t();
        for (std::string line; getline(device_file, line);) {
            auto found = line.find(name_key);
            if(found == std::string::npos) continue;
            else {
                found += name_key.size();
                to_add->name = line.substr(found, line.length() - 4 - name_key.size()); //used some magic to fit the string properly :^)
                for (std::string line; getline(device_file, line);) {
                    if (std::regex_search(line, matches, event_nr_rx)) {
                        to_add->evt_n = matches[1].str();
                        devices.push_back(to_add);
                        return true;
                    }
                }
            }
        }
        return false;
    };

    while(get_names());

    if(verbosity) {
        for(auto d : devices) {
            std::cout
                << "Device Name: " << d->name
                << "\nEvent Number: " << d->evt_n
                << "\n";
        }
    }
};

auto select(const std::vector<device_t*>& devices) -> device_t* {
    int no = -1;

    for(int i = 0; i < devices.size(); i++)
        std::cout << i << ") " << devices[i]->name << std::endl;

    do {
        std::cout << "\nSelect your tablet according to its device name: ";
        std::cin >> no;
    } while(no < 0 && no >= devices.size());

    return devices[no];
};

auto listen(device_t* selected_device) -> void {
    const auto event_device = std::string("/dev/input/event") + selected_device->evt_n;
    std::cout << "Using device: " << event_device << std::endl;
    std::ifstream events { event_device, std::ios::in | std::ios::binary };
    if (!events || getuid()) {
        std::cerr << "Could not open device!\nFile not valid or missing sudo rights" << std::endl;
        exit(1);
    }

    input_event event;

    while (events.read((char *)&event, sizeof(event))) {
        if (event.code == magic_evt_code) {
            if(event.value) {
                if(verbosity)
                    std::cout << "Xournal eraser, evt value: " << event.value << std::endl;
                system("xdotool key shift+ctrl+e");
            } else {
                if(verbosity)
                    std::cout << "Xournal pen, evt value: " << event.value << std::endl;
                system("xdotool key shift+ctrl+p");
            }
        }
    }
}

int main(int argc, char** argv) {
    if(argc > 1 && (strcmp(argv[1], "--verbose") || strcmp(argv[1], "-v")))
        verbosity = true;

    std::vector<device_t*> devices;
    pack_devices(devices);
    listen(select(devices));
    return 0;
}