feat: expose environment, session, & value pointers

for libs which might want to use unsafe ort-sys functions that don't currently have a safe implementation.

src/environment.rs

@@ -1,4 +1,11 @@
-use std::{cell::UnsafeCell, ffi::CString, sync::atomic::AtomicPtr, sync::Arc};
+use std::{
+	cell::UnsafeCell,
+	ffi::CString,
+	sync::{
+		atomic::{AtomicPtr, Ordering},
+		Arc
+	}
+};
 
 use tracing::debug;
 
@@ -19,11 +26,18 @@ unsafe impl Sync for EnvironmentSingleton {}
 static G_ENV: EnvironmentSingleton = EnvironmentSingleton { cell: UnsafeCell::new(None) };
 
 #[derive(Debug)]
-pub(crate) struct Environment {
+pub struct Environment {
 	pub(crate) execution_providers: Vec<ExecutionProviderDispatch>,
 	pub(crate) env_ptr: AtomicPtr<ort_sys::OrtEnv>
 }
 
+impl Environment {
+	/// Loads the underlying [`ort_sys::OrtEnv`] pointer.
+	pub fn ptr(&self) -> *mut ort_sys::OrtEnv {
+		self.env_ptr.load(Ordering::Relaxed)
+	}
+}
+
 impl Drop for Environment {
 	#[tracing::instrument]
 	fn drop(&mut self) {
@@ -36,7 +50,8 @@ impl Drop for Environment {
 	}
 }
 
-pub(crate) fn get_environment() -> Result<&'static Arc<Environment>> {
+/// Gets a reference to the global environment, creating one if an environment has been committed yet.
+pub fn get_environment() -> Result<&'static Arc<Environment>> {
 	if let Some(c) = unsafe { &*G_ENV.cell.get() } {
 		Ok(c)
 	} else {
@@ -110,7 +125,9 @@ impl EnvironmentBuilder {
 	/// Commit the environment configuration and set the global environment.
 	pub fn commit(self) -> Result<()> {
 		// drop global reference to previous environment
-		drop(unsafe { (*G_ENV.cell.get()).take() });
+		if let Some(env_arc) = unsafe { (*G_ENV.cell.get()).take() } {
+			drop(env_arc);
+		}
 
 		let env_ptr = if let Some(global_thread_pool) = self.global_thread_pool_options {
 			let mut env_ptr: *mut ort_sys::OrtEnv = std::ptr::null_mut();

src/lib.rs

@@ -39,7 +39,7 @@ use tracing::Level;
 
 #[cfg(feature = "load-dynamic")]
 pub use self::environment::init_from;
-pub use self::environment::{init, EnvironmentBuilder, EnvironmentGlobalThreadPoolOptions};
+pub use self::environment::{get_environment, init, Environment, EnvironmentBuilder, EnvironmentGlobalThreadPoolOptions};
 #[cfg(feature = "fetch-models")]
 #[cfg_attr(docsrs, doc(cfg(feature = "fetch-models")))]
 pub use self::error::FetchModelError;

src/session/mod.rs

@@ -449,6 +449,13 @@ pub struct SharedSessionInner {
 	_environment: Arc<Environment>
 }
 
+impl SharedSessionInner {
+	/// Returns the underlying [`ort_sys::OrtSession`] pointer.
+	pub fn ptr(&self) -> *mut ort_sys::OrtSession {
+		self.session_ptr.as_ptr()
+	}
+}
+
 unsafe impl Send for SharedSessionInner {}
 unsafe impl Sync for SharedSessionInner {}
 
@@ -626,7 +633,12 @@ impl Session {
 		IoBinding::new(self)
 	}
 
-	/// Get an shared ([`Arc`]'d) reference to the underlying [`SharedSessionInner`], which holds the
+	/// Returns the underlying [`ort_sys::OrtSession`] pointer.
+	pub fn ptr(&self) -> *mut ort_sys::OrtSession {
+		self.inner.ptr()
+	}
+
+	/// Get a shared ([`Arc`]'d) reference to the underlying [`SharedSessionInner`], which holds the
 	/// [`ort_sys::OrtSession`] pointer and the session allocator.
 	#[must_use]
 	pub fn inner(&self) -> Arc<SharedSessionInner> {

src/value/mod.rs

@@ -269,7 +269,8 @@ impl Value {
 		}
 	}
 
-	pub(crate) fn ptr(&self) -> *mut ort_sys::OrtValue {
+	/// Returns the underlying [`ort_sys::OrtValue`] pointer.
+	pub fn ptr(&self) -> *mut ort_sys::OrtValue {
 		match &self.inner {
 			ValueInner::CppOwnedRef { ptr } | ValueInner::CppOwned { ptr, .. } | ValueInner::RustOwned { ptr, .. } => ptr.as_ptr()
 		}