Add startup method wrapper

Signed-off-by: Patrick Uiterwijk <[email protected]>

Author: puiterwijk

src/context.rs

@@ -6,7 +6,7 @@ use crate::{
         algorithm::{Cipher, HashingAlgorithm},
         tags::PropertyTag,
         tss::{TPMA_SESSION_DECRYPT, TPMA_SESSION_ENCRYPT},
-        types::{capability::CapabilityType, session::SessionType},
+        types::{capability::CapabilityType, session::SessionType, startup::StartupType},
     },
     handles::{
         handle_conversion::TryIntoNotNone, AuthHandle, KeyHandle, NvIndexHandle, ObjectHandle,
@@ -130,6 +130,18 @@ impl Context {
         }
     }
 
+    /// Send a TPM2_STARTUP command to the TPM
+    pub fn startup(&mut self, startup_type: StartupType) -> Result<()> {
+        let ret = unsafe { Esys_Startup(self.mut_context(), startup_type.into()) };
+        let ret = Error::from_tss_rc(ret);
+        if ret.is_success() {
+            Ok(())
+        } else {
+            error!("Error while starting up TPM: {}", ret);
+            Err(ret)
+        }
+    }
+
     /// Start new authentication session and return the Session object
     /// associated with the session.
     ///

tests/context_tests.rs

@@ -39,7 +39,7 @@ use tss_esapi::{
         algorithm::{Cipher, HashingAlgorithm},
         tags::PropertyTag,
         tss::*,
-        types::{capability::CapabilityType, session::SessionType},
+        types::{capability::CapabilityType, session::SessionType, startup::StartupType},
     },
     handles::{
         AuthHandle, KeyHandle, NvIndexHandle, NvIndexTpmHandle, ObjectHandle, PcrHandle,
@@ -475,6 +475,16 @@ mod test_clear {
     }
 }
 
+mod test_startup {
+    use super::*;
+
+    #[test]
+    fn test_startup() {
+        let mut context = create_ctx_without_session();
+        context.startup(StartupType::Clear).unwrap();
+    }
+}
+
 mod test_pcr_extend_reset {
     use super::*;