mirror of
https://github.com/Astatin3/unshell.git
synced 2026-06-08 22:38:01 -06:00
Add manager
This commit is contained in:
Generated
+1
-8
@@ -129,13 +129,6 @@ dependencies = [
|
||||
"winapi-util",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unshell-logger"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"log",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unshell-module-load"
|
||||
version = "0.1.0"
|
||||
@@ -143,7 +136,6 @@ dependencies = [
|
||||
"libloading",
|
||||
"log",
|
||||
"pretty_env_logger",
|
||||
"unshell-logger",
|
||||
"unshell-modules",
|
||||
]
|
||||
|
||||
@@ -152,6 +144,7 @@ name = "unshell-modules"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"libloading",
|
||||
"log",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
|
||||
@@ -9,7 +9,7 @@ edition = "2024"
|
||||
libloading = "0.8.9"
|
||||
log = "0.4.28"
|
||||
pretty_env_logger = "0.5.0"
|
||||
unshell-logger = {path = "../unshell-logger"}
|
||||
# unshell-logger = {path = "../unshell-logger"}
|
||||
unshell-modules = {path = "../unshell-modules"}
|
||||
|
||||
[profile.release]
|
||||
|
||||
@@ -1,10 +1,15 @@
|
||||
// #![no]
|
||||
|
||||
#[macro_use]
|
||||
extern crate log;
|
||||
|
||||
use libloading::{Library, Symbol};
|
||||
use log::{info, warn};
|
||||
use unshell_logger::SetupLogger;
|
||||
use unshell_modules::module_interface;
|
||||
mod manager;
|
||||
|
||||
use std::sync::{Arc, Mutex};
|
||||
|
||||
use unshell_modules::{Module, ModuleError, module_interface};
|
||||
|
||||
use crate::manager::Manager;
|
||||
|
||||
module_interface! {
|
||||
Interface {
|
||||
@@ -14,50 +19,7 @@ module_interface! {
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
#[derive(Debug)]
|
||||
enum ModuleError {
|
||||
LibLoadingError(libloading::Error),
|
||||
LinkError(String),
|
||||
}
|
||||
|
||||
struct Module {
|
||||
name: String,
|
||||
lib: Library,
|
||||
}
|
||||
|
||||
impl Module {
|
||||
pub fn new(path: &str) -> Result<Self, ModuleError> {
|
||||
let lib = unsafe { Library::new(&path) }.map_err(|e| ModuleError::LibLoadingError(e))?;
|
||||
|
||||
Ok(Self {
|
||||
name: path.to_owned(),
|
||||
lib,
|
||||
})
|
||||
}
|
||||
pub fn get_symbol<T>(&self, symbol: &[u8]) -> Result<Symbol<'_, T>, ModuleError> {
|
||||
let symbol = unsafe { self.lib.get::<T>(symbol) }
|
||||
.map_err(|e| ModuleError::LinkError(format!("Failed to load symbol: {}", e)))?;
|
||||
|
||||
Ok(symbol)
|
||||
}
|
||||
pub fn init_logger(&self) {
|
||||
if let Ok(setup_logger) = self.get_symbol::<SetupLogger>(b"setup_logger") {
|
||||
setup_logger(log::logger(), log::max_level()).unwrap();
|
||||
} else {
|
||||
warn!("setup_logger not found");
|
||||
}
|
||||
}
|
||||
pub fn get_interface<T>(&self) -> Result<T, ModuleError> {
|
||||
if let Ok(interface_function) = self.get_symbol::<fn() -> T>(b"interface") {
|
||||
Ok(interface_function())
|
||||
} else {
|
||||
Err(ModuleError::LinkError(format!(
|
||||
"Interface function not found!"
|
||||
)))
|
||||
}
|
||||
}
|
||||
}
|
||||
// const modules: Arc<Mutex<Vec<Module>>> = Arc::new(Mutex::new(Vec::new()));
|
||||
|
||||
fn main() {
|
||||
pretty_env_logger::init();
|
||||
@@ -65,13 +27,19 @@ fn main() {
|
||||
info!("Initalized");
|
||||
|
||||
match || -> Result<(), ModuleError> {
|
||||
let module =
|
||||
Module::new("../unshell-module-test/target/release/libunshell_module_test.so")?;
|
||||
module.init_logger();
|
||||
let args = std::env::args();
|
||||
|
||||
let interface = module.get_interface::<Interface>()?;
|
||||
let mut modules = Vec::new();
|
||||
for arg in args.skip(1) {
|
||||
modules.push(Module::new(&arg)?)
|
||||
}
|
||||
let _manager = Manager::new(modules);
|
||||
|
||||
interface.test1();
|
||||
// for i in 1..args.len() {}
|
||||
|
||||
// let interface = module.get_interface::<Interface>()?;
|
||||
|
||||
// interface.test1();
|
||||
|
||||
Ok(())
|
||||
}() {
|
||||
|
||||
@@ -0,0 +1,31 @@
|
||||
use unshell_modules::{ManagerInterface, Module};
|
||||
|
||||
pub struct Manager {
|
||||
modules: Vec<Module>,
|
||||
}
|
||||
|
||||
impl Manager {
|
||||
pub fn new(modules: Vec<Module>) -> Self {
|
||||
let this = Self { modules };
|
||||
|
||||
let interface = this.get_interface();
|
||||
|
||||
for module in &this.modules {
|
||||
if let Ok(init) = module.get_symbol::<fn(ManagerInterface)>(b"init") {
|
||||
init(interface.clone());
|
||||
} else {
|
||||
warn!("init not found");
|
||||
}
|
||||
}
|
||||
|
||||
this
|
||||
}
|
||||
|
||||
pub fn get_interface(&self) -> ManagerInterface {
|
||||
ManagerInterface::from_raw(Self::test123)
|
||||
}
|
||||
|
||||
pub extern "C" fn test123() {
|
||||
info!("Manager Test Sucsessfull!");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user