This commit is contained in:
Travis Burtrum 2017-09-08 01:16:34 -04:00
parent 7dd042f34e
commit 2b0fda1c29
4 changed files with 351 additions and 25 deletions

View File

@ -13,6 +13,7 @@ keywords = ["linux", "input"]
libc = "0.2"
nix = "0.5"
uinput-sys = "0.1"
getopts = "*"
[dependencies.libudev]
optional = true

View File

@ -6,6 +6,7 @@ use nix::{self, fcntl, unistd, Errno};
use nix::sys::stat;
use ffi::*;
use {Result as Res, Error, Device};
use std::collections::hash_map::Values;
#[cfg(feature = "udev")]
use udev;
@ -87,13 +88,17 @@ impl Builder {
self
}
pub fn event(mut self) -> Res<Self> {
pub fn event(mut self, key_codes: Values<&str, *const c_int>) -> Res<Self> {
self.abs = None;
//let test_ev_key : c_int = EV_KEY as c_int;
unsafe {
//try!(Errno::result(ui_set_evbit(self.fd, EV_KEY)));
//try!(Errno::result(ui_set_keybit(self.fd, KEY_H)));
ui_set_evbit(self.fd, &EV_KEY as *const c_int)?;
ui_set_keybit(self.fd, &KEY_H as *const c_int)?;
ui_set_evbit(self.fd, EV_KEY as *const c_int)?;
//ui_set_keybit(self.fd, KEY_H as *const c_int)?;
for key_code in key_codes {
ui_set_keybit(self.fd, *key_code)?;
}
//try!(ui_set_keybit(self.fd, &KEY_H));
}
Ok(self)

View File

@ -28,6 +28,13 @@ impl Device {
value: value as i32,
};
self.write_event(event)
}
}
#[doc(hidden)]
pub fn write_event(&self, mut event: input_event) -> Res<()> {
unsafe {
gettimeofday(&mut event.time, ptr::null_mut());
let ptr = &event as *const _ as *const u8;

View File

@ -1,35 +1,348 @@
extern crate uinput;
extern crate uinput_sys as ffi;
extern crate libc;
extern crate getopts;
#[macro_use]
extern crate nix;
use uinput::Device;
use ffi::*;
use libc::{c_int, input_event};
use std::thread;
use std::time::Duration;
use std::collections::HashMap;
use std::process::{exit, Command};
use std::fs::File;
use std::io::{Read};
use std::{env, mem};
use std::os::unix::io::AsRawFd;
use getopts::Options;
const VERSION: &'static str = env!("CARGO_PKG_VERSION");
const EV_KEY_U16 : u16 = EV_KEY as u16;
#[derive(Debug)]
struct Config {
device_file: String,
log_file: String
}
impl Config {
fn new(device_file: String, log_file: String) -> Self {
Config { device_file: device_file, log_file: log_file }
}
}
fn main() {
let mut device = uinput::default().expect("1")
.name("test").expect("2")
.event().expect("3")
//.event(uinput::event::Keyboard::All).unwrap()
.create().expect("4");
thread::sleep(Duration::from_secs(1));
let key_map = KeyMap::key_map();
device.click(EV_KEY, KEY_H).unwrap();
/*
device.click(&keyboard::Key::H).unwrap();
device.click(&keyboard::Key::E).unwrap();
device.click(&keyboard::Key::L).unwrap();
device.click(&keyboard::Key::L).unwrap();
device.click(&keyboard::Key::O).unwrap();
device.click(&keyboard::Key::Space).unwrap();
device.click(&keyboard::Key::W).unwrap();
device.click(&keyboard::Key::O).unwrap();
device.click(&keyboard::Key::R).unwrap();
device.click(&keyboard::Key::L).unwrap();
device.click(&keyboard::Key::D).unwrap();
device.click(&keyboard::Key::Enter).unwrap();
*/
//println!("key_map: {:?}", key_map);
device.synchronize().unwrap();
let device = uinput::default().expect("1")
.name("test").expect("2")
.event(key_map.values()).expect("3")
//.event(uinput::event::Keyboard::All).unwrap()
.create().expect("4");
let mut key_map = KeyMap::new();
key_map.map(KEY_A, KEY_B);
thread::sleep(Duration::from_secs(1));
//device.click(EV_KEY, KEY_H).unwrap();
//device.synchronize().unwrap();
let config = parse_args();
println!("Config: {:?}", config);
let mut input_device = InputDevice::open(&config.device_file);
input_device.grab();
loop {
let event = input_device.read_event();
if event.type_ == EV_KEY_U16 {
key_map.send_event(event, &device);
/*
println!("type: {} code: {}", event.type_, event.code);
if event.code == KEY_A as u16 {
event.code = KEY_B as u16;
}
*/
} else {
device.write_event(event).expect("could not write event?");
}
}
}
fn parse_args() -> Config {
fn print_usage(program: &str, opts: Options) {
let brief = format!("Usage: {} [options]", program);
println!("{}", opts.usage(&brief));
}
let args: Vec<_> = env::args().collect();
let mut opts = Options::new();
opts.optflag("h", "help", "prints this help message");
opts.optflag("v", "version", "prints the version");
opts.optopt("d", "device", "specify the device file", "DEVICE");
opts.optopt("f", "file", "specify the file to log to", "FILE");
let matches = opts.parse(&args[1..]).unwrap_or_else(|e| panic!("{}", e));
if matches.opt_present("h") {
print_usage(&args[0], opts);
exit(0);
}
if matches.opt_present("v") {
println!("{}", VERSION);
exit(0);
}
let device_file = matches.opt_str("d").unwrap_or_else(|| get_default_device());
let log_file = matches.opt_str("f").unwrap_or("keys.log".to_owned());
Config::new(device_file, log_file)
}
fn get_default_device() -> String {
let mut filenames = get_keyboard_device_filenames();
println!("Detected devices: {:?}", filenames);
if filenames.len() == 1 {
filenames.swap_remove(0)
} else {
panic!("The following keyboard devices were detected: {:?}. Please select one using \
the `-d` flag", filenames);
}
}
// Detects and returns the name of the keyboard device file. This function uses
// the fact that all device information is shown in /proc/bus/input/devices and
// the keyboard device file should always have an EV of 120013
fn get_keyboard_device_filenames() -> Vec<String> {
let mut command_str = "grep -E 'Handlers|EV' /proc/bus/input/devices".to_string();
command_str.push_str("| grep -B1 120013");
command_str.push_str("| grep -Eo event[0-9]+");
let res = Command::new("sh").arg("-c").arg(command_str).output().unwrap_or_else(|e| {
panic!("{}", e);
});
let res_str = std::str::from_utf8(&res.stdout).unwrap();
let mut filenames = Vec::new();
for file in res_str.trim().split('\n') {
let mut filename = "/dev/input/".to_string();
filename.push_str(file);
filenames.push(filename);
}
filenames
}
// inputdevice stuff
ioctl!(write eviocgrab with b'E', 0x90; c_int);
// TODO: use size_of_input_event instead of hard-coding 24.
const SIZE_OF_INPUT_EVENT : usize = 24;//mem::size_of::<input_event>();
struct InputDevice {
device_file : File,
buf : [u8; SIZE_OF_INPUT_EVENT],
}
impl InputDevice {
pub fn open(device_file : &str) -> Self {
let device_file = File::open(device_file).unwrap_or_else(|e| panic!("{}", e));
InputDevice {
device_file: device_file,
buf: [0u8; SIZE_OF_INPUT_EVENT],
}
}
pub fn read_event(&mut self) -> input_event {
let num_bytes = self.device_file.read(&mut self.buf).unwrap_or_else(|e| panic!("{}", e));
if num_bytes != SIZE_OF_INPUT_EVENT {
panic!("Error while reading from device file");
}
let event: input_event = unsafe { mem::transmute(self.buf) };
event
}
pub fn grab(&mut self) {
unsafe {
eviocgrab(self.device_file.as_raw_fd(), 1 as *const c_int).expect("no grab?");
}
}
pub fn release(&mut self) {
unsafe {
eviocgrab(self.device_file.as_raw_fd(), 0 as *const c_int).expect("no release?");
}
}
}
impl Drop for InputDevice {
fn drop(&mut self) {
self.release();
}
}
// keymapper stuff
trait KeyMapper {
fn send_event(&self, event: input_event, device: &Device);
}
// 249 is one more than KEY_MICMUTE which is max key in uinput-sys event.rs
const KEY_MAX : usize = 249;
struct KeyMap {
keymap: [u16; KEY_MAX],
}
impl KeyMap {
pub fn key_map() -> HashMap<&'static str, *const c_int> {
[
("Reserved", KEY_RESERVED),
("Esc", KEY_ESC),
("_1", KEY_1),
("_2", KEY_2),
("_3", KEY_3),
("_4", KEY_4),
("_5", KEY_5),
("_6", KEY_6),
("_7", KEY_7),
("_8", KEY_8),
("_9", KEY_9),
("_0", KEY_10),
("Minus", KEY_MINUS),
("Equal", KEY_EQUAL),
("BackSpace", KEY_BACKSPACE),
("Tab", KEY_TAB),
("Q", KEY_Q),
("W", KEY_W),
("E", KEY_E),
("R", KEY_R),
("T", KEY_T),
("Y", KEY_Y),
("U", KEY_U),
("I", KEY_I),
("O", KEY_O),
("P", KEY_P),
("LeftBrace", KEY_LEFTBRACE),
("RightBrace", KEY_RIGHTBRACE),
("Enter", KEY_ENTER),
("LeftControl", KEY_LEFTCTRL),
("A", KEY_A),
("S", KEY_S),
("D", KEY_D),
("F", KEY_F),
("G", KEY_G),
("H", KEY_H),
("J", KEY_J),
("K", KEY_K),
("L", KEY_L),
("SemiColon", KEY_SEMICOLON),
("Apostrophe", KEY_APOSTROPHE),
("Grave", KEY_GRAVE),
("LeftShift", KEY_LEFTSHIFT),
("BackSlash", KEY_BACKSLASH),
("Z", KEY_Z),
("X", KEY_X),
("C", KEY_C),
("V", KEY_V),
("B", KEY_B),
("N", KEY_N),
("M", KEY_M),
("Comma", KEY_COMMA),
("Dot", KEY_DOT),
("Slash", KEY_SLASH),
("RightShift", KEY_RIGHTSHIFT),
("LeftAlt", KEY_LEFTALT),
("Space", KEY_SPACE),
("CapsLock", KEY_CAPSLOCK),
("F1", KEY_F1),
("F2", KEY_F2),
("F3", KEY_F3),
("F4", KEY_F4),
("F5", KEY_F5),
("F6", KEY_F6),
("F7", KEY_F7),
("F8", KEY_F8),
("F9", KEY_F9),
("F10", KEY_F10),
("NumLock", KEY_NUMLOCK),
("ScrollLock", KEY_SCROLLLOCK),
("F11", KEY_F11),
("F12", KEY_F12),
("RightControl", KEY_RIGHTCTRL),
("SysRq", KEY_SYSRQ),
("RightAlt", KEY_RIGHTALT),
("LineFeed", KEY_LINEFEED),
("Home", KEY_HOME),
("Up", KEY_UP),
("PageUp", KEY_PAGEUP),
("Left", KEY_LEFT),
("Right", KEY_RIGHT),
("End", KEY_END),
("Down", KEY_DOWN),
("PageDown", KEY_PAGEDOWN),
("Insert", KEY_INSERT),
("Delete", KEY_DELETE),
("LeftMeta", KEY_LEFTMETA),
("RightMeta", KEY_RIGHTMETA),
("ScrollUp", KEY_SCROLLUP),
("ScrollDown", KEY_SCROLLDOWN),
("F13", KEY_F13),
("F14", KEY_F14),
("F15", KEY_F15),
("F16", KEY_F16),
("F17", KEY_F17),
("F18", KEY_F18),
("F19", KEY_F19),
("F20", KEY_F20),
("F21", KEY_F21),
("F22", KEY_F22),
("F23", KEY_F23),
("F24", KEY_F24),
].iter().cloned().map(|(m, v)| (m, v as *const c_int)).collect()
}
pub fn new() -> Self {
let mut keymap = [0u16; KEY_MAX];
// which is rustier
/*
for x in 0..KEY_MAX {
keymap[x as usize] = x as u16;
}
*/
for (x, v) in keymap.iter_mut().enumerate() {
*v = x as u16;
}
println!("keymap: {:?}", &keymap[..]);
KeyMap {
keymap: keymap
}
}
pub fn map(&mut self, from : c_int, to: c_int) {
self.keymap[from as usize] = to as u16;
}
}
impl KeyMapper for KeyMap {
fn send_event(&self, mut event: input_event, device: &Device) {
event.code = self.keymap[event.code as usize];
device.write_event(event).expect("could not write event?");
}
}