tihu 0.1.3

tihu is a toolset written in Rust.
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
93
94
95
96
97
98
99
100
101
102
103
104
105

use super::encoder;
use super::version_data;
use super::LightString;
use chrono::DateTime;
use chrono::Utc;
use log;
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize, Debug)]
pub struct ClientIdV1 {
    rsa_pub_key: LightString,
    expire_time: i64, //过期时间，单位：秒
}

pub enum ClientId {
    V1(ClientIdV1),
}

impl ClientIdV1 {
    pub fn expired(&self) -> bool {
        let curr_time = Utc::now();
        return curr_time >= DateTime::from_timestamp(self.expire_time, 0).unwrap();
    }
}

impl ClientId {
    pub fn new(rsa_pub_key: LightString, expire_time: DateTime<Utc>) -> ClientId {
        ClientId::V1(ClientIdV1 {
            rsa_pub_key,
            expire_time: expire_time.timestamp(),
        })
    }

    fn version(&self) -> u64 {
        match self {
            ClientId::V1(_) => 1,
        }
    }

    pub fn rsa_pub_key(&self) -> &LightString {
        match self {
            ClientId::V1(client_id) => &client_id.rsa_pub_key,
        }
    }

    pub fn expire_time(&self) -> i64 {
        match self {
            ClientId::V1(client_id) => client_id.expire_time,
        }
    }

    pub fn expired(&self) -> bool {
        match self {
            ClientId::V1(client_id) => client_id.expired(),
        }
    }

    pub fn try_decode(
        client_id_data: &str,
        check_signature: impl FnOnce(&str, &[u8], &[u8]) -> Result<bool, LightString>,
    ) -> Result<ClientId, LightString> {
        let (version, chunk) = version_data::try_decode(client_id_data)?;
        match version {
            1 => {
                let (client_id_data, signature) = encoder::decode_chunks::<2>(chunk)?;
                let client_id: ClientIdV1 =
                    serde_json::from_slice(&client_id_data).map_err(|err| {
                        log::error!("反序列化客户端身份数据失败: {}", err);
                        return LightString::from_static("反序列化客户端身份数据失败!");
                    })?;
                if client_id.expired() {
                    return Err(LightString::from_static("客户端身份数据已过期!"));
                }
                if check_signature(&client_id.rsa_pub_key, &client_id_data, &signature)? {
                    return Ok(ClientId::V1(client_id));
                } else {
                    return Err(LightString::from_static("客户端身份数据签名不正确！"));
                }
            }
            _ => {
                return Err(LightString::from_static("未知的客户端身份数据版本!"));
            }
        }
    }

    pub fn encode<B>(
        &self,
        sign: impl FnOnce(&[u8]) -> Result<B, LightString>,
    ) -> Result<String, LightString>
    where
        B: AsRef<[u8]>,
    {
        let client_id_data = match self {
            ClientId::V1(client_id) => serde_json::to_vec(&client_id).map_err(|err| {
                log::error!("序列化客户端身份数据失败: {}", err);
                return LightString::from_static("序列化客户端身份数据失败!");
            })?,
        };
        let signature = sign(&client_id_data)?;
        return version_data::encode(
            self.version(),
            &encoder::encode_chunks(&[&client_id_data, signature.as_ref()], None),
        );
    }
}