noctra/vgpu_unlock-rs
1
0
Fork 0
mirror of https://github.com/mbilker/vgpu_unlock-rs.git synced 2025-06-08 13:28:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add individual profile override support

Browse Source
This commit is contained in:
Matt Bilker 2021-10-09 12:09:56 +00:00
parent 80e3397a44
commit b9a16c426e
No known key found for this signature in database
GPG Key ID: 69ADF8AEB6C8B5D1
1 changed files with 30 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/lib.rs
View File

@ -6,6 +6,7 @@
//! - DualCoder, snowman, Felix, Elec for vGPU profile modification at runtime //! - DualCoder, snowman, Felix, Elec for vGPU profile modification at runtime
use std::borrow::Cow; use std::borrow::Cow;
use std::collections::HashMap;
use std::env; use std::env;
use std::fmt; use std::fmt;
use std::fs; use std::fs;
@ -134,6 +135,12 @@ struct VgpuConfig {
license_type: [u8; 1156], license_type: [u8; 1156],
} }
#[derive(Deserialize)]
struct Config<'a> {
#[serde(borrow)]
profile: HashMap<&'a str, VgpuProfileOverride<'a>>,
}
#[derive(Deserialize)] #[derive(Deserialize)]
struct VgpuProfileOverride<'a> { struct VgpuProfileOverride<'a> {
gpu_type: Option<u32>, gpu_type: Option<u32>,
@ -301,7 +308,7 @@ pub unsafe extern "C" fn ioctl(fd: RawFd, request: c_ulong, argp: *mut c_void) -
let config = &mut *(io_data.result as *mut VgpuConfig); let config = &mut *(io_data.result as *mut VgpuConfig);
info!("{:#?}", config); info!("{:#?}", config);
if !apply_profile_override(config) { if !handle_profile_override(config) {
error!("Failed to apply profile override"); error!("Failed to apply profile override");
return -1; return -1;
} }
@ -338,7 +345,7 @@ pub fn from_c_str<'a>(value: &'a [u8]) -> Cow<'a, str> {
String::from_utf8_lossy(&value[..len]) String::from_utf8_lossy(&value[..len])
} }
fn apply_profile_override(config: &mut VgpuConfig) -> bool { fn handle_profile_override(config: &mut VgpuConfig) -> bool {
const DEFAULT_CONFIG_PATH: &'static str = "/etc/vgpu_unlock/profile_override.toml"; const DEFAULT_CONFIG_PATH: &'static str = "/etc/vgpu_unlock/profile_override.toml";
let config_path = match env::var_os("VGPU_UNLOCK_PROFILE_OVERRIDE_CONFIG_PATH") { let config_path = match env::var_os("VGPU_UNLOCK_PROFILE_OVERRIDE_CONFIG_PATH") {
@ -352,7 +359,7 @@ fn apply_profile_override(config: &mut VgpuConfig) -> bool {
return false; return false;
} }
}; };
let config_override: VgpuProfileOverride = match toml::from_str(&config_overrides) { let config_overrides: Config = match toml::from_str(&config_overrides) {
Ok(config) => config, Ok(config) => config,
Err(e) => { Err(e) => {
error!("Failed to decode config: {}", e); error!("Failed to decode config: {}", e);
@ -360,13 +367,29 @@ fn apply_profile_override(config: &mut VgpuConfig) -> bool {
} }
}; };
let gpu_type = config.gpu_type; let gpu_type = format!("nvidia-{}", config.gpu_type);
if let Some(config_override) = config_overrides.profile.get(gpu_type.as_str()) {
info!("Applying profile {} overrides", gpu_type);
if !apply_profile_override(config, &gpu_type, config_override) {
return false;
}
}
true
}
fn apply_profile_override(
config: &mut VgpuConfig,
gpu_type: &str,
config_override: &VgpuProfileOverride,
) -> bool {
macro_rules! handle_copy_overrides { macro_rules! handle_copy_overrides {
($field:ident) => { ($field:ident) => {
if let Some(value) = config_override.$field { if let Some(value) = config_override.$field {
info!( info!(
"Patching nvidia-{}/{}: {} -> {}", "Patching {}/{}: {} -> {}",
gpu_type, gpu_type,
stringify!($field), stringify!($field),
config.$field, config.$field,
@ -390,7 +413,7 @@ fn apply_profile_override(config: &mut VgpuConfig) -> bool {
// Use `len - 1` to account for the required NULL terminator. // Use `len - 1` to account for the required NULL terminator.
if value_bytes.len() > config.$field.len() - 1 { if value_bytes.len() > config.$field.len() - 1 {
error!( error!(
"Patching nvidia-{}/{}: value '{}' is too long", "Patching {}/{}: value '{}' is too long",
gpu_type, gpu_type,
stringify!($field), stringify!($field),
value value
@ -399,7 +422,7 @@ fn apply_profile_override(config: &mut VgpuConfig) -> bool {
return false; return false;
} else { } else {
info!( info!(
"Patching nvidia-{}/{}: '{}' -> '{}'", "Patching {}/{}: '{}' -> '{}'",
gpu_type, gpu_type,
stringify!($field), stringify!($field),
from_c_str(&config.$field), from_c_str(&config.$field),