rust-libutee 0.1.0

Rust library for UTEE-related functionality.
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

use crate::api::tee_api_panic::TEE_Panic;
use crate::syscalls::syscall_table::{_utee_get_time, _utee_set_ta_time, _utee_wait};
use crate::tee_api_defines::{
    TEE_ERROR_CANCEL, TEE_ERROR_OUT_OF_MEMORY, TEE_ERROR_OVERFLOW, TEE_ERROR_STORAGE_NO_SPACE,
    TEE_ERROR_TIME_NEEDS_RESET, TEE_ERROR_TIME_NOT_SET, TEE_SUCCESS,
};
use crate::tee_api_types::{TEE_Result, TEE_Time};
use crate::utee_types::utee_time_category;

#[unsafe(no_mangle)]
pub extern "C" fn TEE_GetSystemTime(time: *mut TEE_Time) {
    let res = unsafe { _utee_get_time(utee_time_category::UTEE_TIME_CAT_SYSTEM as u64, time) };

    // 检查结果是否为 TEE_SUCCESS (0)
    if res != TEE_SUCCESS as usize {
        TEE_Panic(res as TEE_Result);
    }
}

#[unsafe(no_mangle)]
pub extern "C" fn TEE_GetTAPersistentTime(time: *mut TEE_Time) -> TEE_Result {
    let res =
        unsafe { _utee_get_time(utee_time_category::UTEE_TIME_CAT_TA_PERSISTENT as u64, time) };

    // 如果结果不是成功且不是溢出错误，则将时间设置为0
    if res != TEE_ERROR_OVERFLOW as usize {
        unsafe {
            (*time).seconds = 0;
            (*time).millis = 0;
        }
    }

    // 检查是否需要panic
    if res != TEE_SUCCESS as usize
        && res != TEE_ERROR_TIME_NOT_SET as usize
        && res != TEE_ERROR_TIME_NEEDS_RESET as usize
        && res != TEE_ERROR_OVERFLOW as usize
        && res != TEE_ERROR_OUT_OF_MEMORY as usize
    {
        TEE_Panic(res as TEE_Result);
    }

    res as TEE_Result
}

#[unsafe(no_mangle)]
pub extern "C" fn TEE_GetREETime(time: *mut TEE_Time) {
    let res = unsafe { _utee_get_time(utee_time_category::UTEE_TIME_CAT_REE as u64, time) };

    if res != TEE_SUCCESS as usize {
        TEE_Panic(res as TEE_Result);
    }
}

#[unsafe(no_mangle)]
pub extern "C" fn TEE_Wait(timeout: u32) -> TEE_Result {
    let res = unsafe { _utee_wait(timeout as u64) };

    if res != TEE_SUCCESS as usize && res != TEE_ERROR_CANCEL as usize {
        TEE_Panic(res as TEE_Result);
    }

    res as TEE_Result
}

#[unsafe(no_mangle)]
pub extern "C" fn TEE_SetTAPersistentTime(time: *const TEE_Time) -> TEE_Result {
    let res = unsafe { _utee_set_ta_time(time) };

    if res != TEE_SUCCESS as usize
        && res != TEE_ERROR_OUT_OF_MEMORY as usize
        && res != TEE_ERROR_STORAGE_NO_SPACE as usize
    {
        TEE_Panic(res as TEE_Result);
    }

    res as TEE_Result
}