asusctl/rog-platform/src/hid_raw.rs

use std::cell::UnsafeCell;
use std::fs::OpenOptions;
use std::io::Write;
use std::path::PathBuf;

use log::{info, warn};
use udev::Device;

use crate::error::{PlatformError, Result};

#[derive(Debug)]
pub struct HidRaw {
    path: UnsafeCell<PathBuf>,
    prod_id: String,
}

impl HidRaw {
    pub fn new(id_product: &str) -> Result<Self> {
        let mut enumerator = udev::Enumerator::new().map_err(|err| {
            warn!("{}", err);
            PlatformError::Udev("enumerator failed".into(), err)
        })?;

        enumerator.match_subsystem("hidraw").map_err(|err| {
            warn!("{}", err);
            PlatformError::Udev("match_subsystem failed".into(), err)
        })?;

        for device in enumerator
            .scan_devices()
            .map_err(|e| PlatformError::IoPath("enumerator".to_owned(), e))?
        {
            if let Some(parent) = device
                .parent_with_subsystem_devtype("usb", "usb_device")
                .map_err(|e| {
                    PlatformError::IoPath(device.devpath().to_string_lossy().to_string(), e)
                })?
            {
                if let Some(parent) = parent.attribute_value("idProduct") {
                    if parent == id_product {
                        if let Some(dev_node) = device.devnode() {
                            info!("Using device at: {:?} for hidraw control", dev_node);
                            return Ok(Self {
                                path: UnsafeCell::new(dev_node.to_owned()),
                                prod_id: id_product.to_string(),
                            });
                        }
                    }
                }
            } else {
                // Try to see if there is a virtual device created with uhid for testing
                let dev_path = device.devpath().to_string_lossy();
                if dev_path.contains("virtual") && dev_path.contains(&id_product.to_uppercase()) {
                    if let Some(dev_node) = device.devnode() {
                        info!(
                            "Using device at: {:?} for <TODO: label control> control",
                            dev_node
                        );
                        return Ok(Self {
                            path: UnsafeCell::new(dev_node.to_owned()),
                            prod_id: id_product.to_string(),
                        });
                    }
                }
            }
        }
        Err(PlatformError::MissingFunction(format!(
            "hidraw dev {} not found",
            id_product
        )))
    }

    pub fn write_bytes(&self, message: &[u8]) -> Result<()> {
        let mut path = unsafe { &*(self.path.get()) };
        let mut file = match OpenOptions::new().write(true).open(path) {
            Ok(f) => f,
            Err(e) => {
                warn!("write_bytes failed for {:?}, trying again: {e}", self.path);
                unsafe {
                    *(self.path.get()) = (*(Self::new(&self.prod_id)?.path.get())).clone();
                    path = &mut *(self.path.get());
                }
                OpenOptions::new()
                    .write(true)
                    .open(path)
                    .map_err(|e| PlatformError::IoPath(path.to_string_lossy().to_string(), e))?
            }
        };
        file.write_all(message)
            .map_err(|e| PlatformError::IoPath(path.to_string_lossy().to_string(), e))
    }

    pub fn set_wakeup_disabled(&self) -> Result<()> {
        let path = unsafe { &*(self.path.get()) };
        let mut dev = Device::from_syspath(path)?;
        Ok(dev.set_attribute_value("power/wakeup", "disabled")?)
    }
}