tss_esapi/constants/
capabilities.rs

1use crate::{Error, Result, WrapperErrorKind};
2use log::error;
3use num_derive::{FromPrimitive, ToPrimitive};
4use num_traits::{FromPrimitive, ToPrimitive};
5use std::convert::{From, TryFrom};
6
7use crate::{
8    constants::tss::{
9        TPM2_CAP_ACT, TPM2_CAP_ALGS, TPM2_CAP_AUDIT_COMMANDS, TPM2_CAP_AUTH_POLICIES,
10        TPM2_CAP_COMMANDS, TPM2_CAP_ECC_CURVES, TPM2_CAP_HANDLES, TPM2_CAP_PCRS,
11        TPM2_CAP_PCR_PROPERTIES, TPM2_CAP_PP_COMMANDS, TPM2_CAP_TPM_PROPERTIES,
12    },
13    tss2_esys::TPM2_CAP,
14};
15
16// Enum representing the different TPM Capability Type values.
17#[derive(FromPrimitive, ToPrimitive, Debug, Copy, Clone, PartialEq, Eq)]
18#[repr(u32)]
19pub enum CapabilityType {
20    Algorithms = TPM2_CAP_ALGS,
21    Handles = TPM2_CAP_HANDLES,
22    Command = TPM2_CAP_COMMANDS,
23    PpCommands = TPM2_CAP_PP_COMMANDS,
24    AuditCommands = TPM2_CAP_AUDIT_COMMANDS,
25    AssignedPcr = TPM2_CAP_PCRS,
26    TpmProperties = TPM2_CAP_TPM_PROPERTIES,
27    PcrProperties = TPM2_CAP_PCR_PROPERTIES,
28    EccCurves = TPM2_CAP_ECC_CURVES,
29    AuthPolicies = TPM2_CAP_AUTH_POLICIES,
30    Act = TPM2_CAP_ACT,
31}
32
33impl From<CapabilityType> for TPM2_CAP {
34    fn from(capability_type: CapabilityType) -> TPM2_CAP {
35        // The values are well defined so this cannot fail.
36        capability_type.to_u32().unwrap()
37    }
38}
39
40impl TryFrom<TPM2_CAP> for CapabilityType {
41    type Error = Error;
42    fn try_from(tpm_capability_type: TPM2_CAP) -> Result<CapabilityType> {
43        CapabilityType::from_u32(tpm_capability_type).ok_or_else(|| {
44            error!(
45                "value = {} did not match any CapabilityType.",
46                tpm_capability_type
47            );
48            Error::local_error(WrapperErrorKind::InvalidParam)
49        })
50    }
51}