use crate::internal_prelude::*;
use std::path::Path;
use std::sync::RwLock;
lazy_static::lazy_static! {
static ref WITNESS_LOCK: RwLock<bool> = RwLock::new(false);
}
pub struct InitializationGuard();
impl InitializationGuard {
fn new_init() -> InitializationGuard {
InitializationGuard()
}
}
impl Drop for InitializationGuard {
fn drop(&mut self) {
let mut guard = WITNESS_LOCK.write().expect("Mutex Poisoned");
*guard = false;
unsafe { syz_shutdown() };
}
}
pub fn initialize() -> Result<InitializationGuard> {
let mut guard = WITNESS_LOCK.write().expect("Mutex poisoned");
check_error(unsafe { syz_initialize() })?;
*guard = true;
Ok(InitializationGuard::new_init())
}
pub struct LibraryConfig {
config: syz_LibraryConfig,
libsndfile_path_string: Option<std::ffi::CString>,
}
#[derive(Clone, Copy, Default, Debug)]
#[repr(i32)]
pub enum LogLevel {
#[default]
Error = SYZ_LOG_LEVEL_ERROR as i32,
Warn = SYZ_LOG_LEVEL_WARN as i32,
Info = SYZ_LOG_LEVEL_INFO as i32,
Debug = SYZ_LOG_LEVEL_DEBUG as i32,
}
impl LibraryConfig {
pub fn new() -> LibraryConfig {
let mut out = LibraryConfig {
config: Default::default(),
libsndfile_path_string: None,
};
unsafe { syz_libraryConfigSetDefaults(&mut out.config as *mut syz_LibraryConfig) };
out
}
pub fn log_level(&mut self, level: LogLevel) {
self.config.log_level = unsafe { std::mem::transmute(level) };
}
pub fn log_to_stderr(&mut self) {
self.config.logging_backend = SYZ_LOGGING_BACKEND_STDERR;
}
pub fn load_libsndfile<P: AsRef<Path>>(&mut self, path: P) -> Result<()> {
let c_str = std::ffi::CString::new(
path.as_ref()
.to_str()
.ok_or_else(|| Error::rust_error("Path isn't valid UTF-8"))?,
)
.map_err(|_| Error::rust_error("Unable to convert path to C string"))?;
self.config.libsndfile_path = c_str.as_ptr();
self.libsndfile_path_string = Some(c_str);
Ok(())
}
pub fn initialize(self) -> Result<InitializationGuard> {
let mut guard = WITNESS_LOCK.write().expect("Mutex poisoned");
check_error(unsafe { syz_initializeWithConfig(&self.config as *const syz_LibraryConfig) })?;
*guard = true;
Ok(InitializationGuard::new_init())
}
}
impl Default for LibraryConfig {
fn default() -> LibraryConfig {
LibraryConfig::new()
}
}
pub(crate) fn with_witness<T>(closure: impl FnOnce() -> Result<T>) -> Result<T> {
let guard = WITNESS_LOCK.read().expect("Mutex Poisoned");
if !*guard {
return Err(Error::rust_error("Synthizer is not initialized"));
}
closure()
}