1 // Copyright 2023, The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //     http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 //! Errors and relating functions thrown in this library.
16 
17 use open_dice_cbor_bindgen::DiceResult;
18 use std::{fmt, result};
19 
20 #[cfg(feature = "std")]
21 use std::error::Error;
22 
23 /// Error type used by DICE.
24 #[derive(Debug)]
25 pub enum DiceError {
26     /// Provided input was invalid.
27     InvalidInput,
28     /// Provided buffer was too small.
29     BufferTooSmall(usize),
30     /// Platform error.
31     PlatformError,
32 }
33 
34 /// This makes `DiceError` accepted by anyhow.
35 #[cfg(feature = "std")]
36 impl Error for DiceError {}
37 
38 impl fmt::Display for DiceError {
fmt(&self, f: &mut fmt::Formatter) -> fmt::Result39     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
40         match self {
41             Self::InvalidInput => write!(f, "Invalid input"),
42             Self::BufferTooSmall(buffer_required_size) => {
43                 write!(f, "Buffer too small; need {buffer_required_size} bytes")
44             }
45             Self::PlatformError => write!(f, "Platform error"),
46         }
47     }
48 }
49 
50 /// DICE result type.
51 pub type Result<T> = result::Result<T, DiceError>;
52 
53 /// Checks the given `DiceResult`. Returns an error if it's not OK.
check_result(result: DiceResult, buffer_required_size: usize) -> Result<()>54 pub(crate) fn check_result(result: DiceResult, buffer_required_size: usize) -> Result<()> {
55     match result {
56         DiceResult::kDiceResultOk => Ok(()),
57         DiceResult::kDiceResultInvalidInput => Err(DiceError::InvalidInput),
58         DiceResult::kDiceResultBufferTooSmall => {
59             Err(DiceError::BufferTooSmall(buffer_required_size))
60         }
61         DiceResult::kDiceResultPlatformError => Err(DiceError::PlatformError),
62     }
63 }
64