tss-esapi 8.0.0-alpha.3

Rust-native wrapper around TSS 2.0 Enhanced System API
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

// Copyright 2022 Contributors to the Parsec project.
// SPDX-License-Identifier: Apache-2.0

use crate::{
    Error, Result,
    ffi::take_from_esys,
    structures::{CreateKeyResult, CreationData, CreationTicket, Digest, Private, Public},
    tss2_esys::{TPM2B_CREATION_DATA, TPM2B_DIGEST, TPM2B_PRIVATE, TPM2B_PUBLIC, TPMT_TK_CREATION},
};
use std::convert::TryFrom;
use std::ptr::null_mut;

/// Struct that handles the output of the
/// Create Esys_Create command and zeroizes
/// the FFI data.
pub(crate) struct CreateCommandOutputHandler {
    ffi_out_public_ptr: *mut TPM2B_PUBLIC,
    ffi_out_private_ptr: *mut TPM2B_PRIVATE,
    ffi_creation_data_ptr: *mut TPM2B_CREATION_DATA,
    ffi_creation_hash_ptr: *mut TPM2B_DIGEST,
    ffi_creation_ticket_ptr: *mut TPMT_TK_CREATION,
}

/// Creates a new CreateCommandOutputHandler
impl CreateCommandOutputHandler {
    pub(crate) fn new() -> Self {
        Self {
            ffi_out_private_ptr: null_mut(),
            ffi_out_public_ptr: null_mut(),
            ffi_creation_data_ptr: null_mut(),
            ffi_creation_hash_ptr: null_mut(),
            ffi_creation_ticket_ptr: null_mut(),
        }
    }

    /// A reference to the where 'outPrivate' output parameter pointer shall be stored.
    pub fn ffi_out_private_ptr(&mut self) -> &mut *mut TPM2B_PRIVATE {
        &mut self.ffi_out_private_ptr
    }

    /// A reference to the where 'outPublic' output parameter pointer shall be stored.
    pub fn ffi_out_public_ptr(&mut self) -> &mut *mut TPM2B_PUBLIC {
        &mut self.ffi_out_public_ptr
    }

    /// A reference to the where 'creationData' output parameter pointer shall be stored.
    pub fn ffi_creation_data_ptr(&mut self) -> &mut *mut TPM2B_CREATION_DATA {
        &mut self.ffi_creation_data_ptr
    }

    /// A reference to the where 'creationHash' output parameter pointer shall be stored.
    pub fn ffi_creation_hash_ptr(&mut self) -> &mut *mut TPM2B_DIGEST {
        &mut self.ffi_creation_hash_ptr
    }

    /// A reference to the where 'creationTicket' output parameter pointer shall be stored.
    pub fn ffi_creation_ticket_ptr(&mut self) -> &mut *mut TPMT_TK_CREATION {
        &mut self.ffi_creation_ticket_ptr
    }
}

impl TryFrom<CreateCommandOutputHandler> for CreateKeyResult {
    type Error = Error;

    fn try_from(ffi_data_handler: CreateCommandOutputHandler) -> Result<CreateKeyResult> {
        // Take and free with Esys_Free; then null out the handler's fields so Drop (if any)
        // won't free them a second time.

        let out_private_owned_result =
            unsafe { take_from_esys(ffi_data_handler.ffi_out_private_ptr) };

        let out_public_owned_result =
            unsafe { take_from_esys(ffi_data_handler.ffi_out_public_ptr) };

        let creation_data_owned_result =
            unsafe { take_from_esys(ffi_data_handler.ffi_creation_data_ptr) };

        let creation_hash_owned_result =
            unsafe { take_from_esys(ffi_data_handler.ffi_creation_hash_ptr) };

        let creation_ticket_owned_result =
            unsafe { take_from_esys(ffi_data_handler.ffi_creation_ticket_ptr) };

        Ok(CreateKeyResult {
            out_private: out_private_owned_result.and_then(Private::try_from)?,
            out_public: out_public_owned_result.and_then(Public::try_from)?,
            creation_data: creation_data_owned_result.and_then(CreationData::try_from)?,
            creation_hash: creation_hash_owned_result.and_then(Digest::try_from)?,
            creation_ticket: creation_ticket_owned_result.and_then(CreationTicket::try_from)?,
        })
    }
}