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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
//! Low level interface for Key-Value (KV) Stores
//!
//! The KV store implementation in Tock has three levels, described below.
//!
//! 1 - Hardware Level:
//! This level is the interface that writes a buffer to the hardware. This will
//! generally be writing to flash, although in theory it would be possible to
//! write to other mediums.
//!
//! An example of the HIL used here is the Tock Flash HIL.
//!
//! 2 - KV System Level:
//! This level can be thought of like a file system. It is responsible for
//! taking save/load operations and generating a buffer to pass to level 1
//! This level is also in charge of generating hashes and checksums.
//!
//! This level allows generating a key hash but otherwise operates on
//! hashed keys. This level is not responsible for permission checks.
//!
//! This file describes the HIL for this level.
//!
//! 3 - KV Store:
//! This is a user friendly high level API. This API is used inside the kernel
//! and exposed to applications to allow KV operations. The API from this level
//! should be high level, for example set/get/delete on unhashed keys.
//! This level is in charge of enforcing permissions.
//!
//! This level is also in charge of generating the key hash by calling into
//! level 2.
//!
//! The expected setup inside Tock will look like this:
//! +-----------------------+
//! |                       |
//! |  Capsule using K-V    |
//! |                       |
//! +-----------------------+
//!
//!    capsules::kv_store
//!
//! +-----------------------+
//! |                       |
//! |  K-V in Tock          |
//! |                       |
//! +-----------------------+
//!
//!    hil::kv_system (this file)
//!
//! +-----------------------+
//! |                       |
//! |  K-V library          |
//! |                       |
//! +-----------------------+
//!
//!    hil::flash

use crate::ErrorCode;

/// The type of keys, this should define the output size of the digest
/// operations.
pub trait KeyType: Eq + Copy + Clone + Sized + AsRef<[u8]> + AsMut<[u8]> {}

impl KeyType for [u8; 8] {}

/// Implement this trait and use `set_client()` in order to receive callbacks.
pub trait StoreClient<K: KeyType> {
    /// This callback is called when the get operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    /// `ret_buf`: The ret_buf buffer
    fn get_complete(
        &self,
        result: Result<(), ErrorCode>,
        key: &'static mut [u8],
        ret_buf: &'static mut [u8],
    );

    /// This callback is called when the set operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    /// `value`: The value buffer
    fn set_complete(
        &self,
        result: Result<(), ErrorCode>,
        key: &'static mut [u8],
        value: &'static mut [u8],
    );

    /// This callback is called when the delete operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    fn delete_complete(&self, result: Result<(), ErrorCode>, key: &'static mut [u8]);
}

/// Implement this trait and use `set_client()` in order to receive callbacks.
pub trait Client<K: KeyType> {
    /// This callback is called when the append_key operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `unhashed_key`: The unhashed_key buffer
    /// `key_buf`: The key_buf buffer
    fn generate_key_complete(
        &self,
        result: Result<(), ErrorCode>,
        unhashed_key: &'static mut [u8],
        key_buf: &'static mut K,
    );

    /// This callback is called when the append_key operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    /// `value`: The value buffer
    fn append_key_complete(
        &self,
        result: Result<(), ErrorCode>,
        key: &'static mut K,
        value: &'static mut [u8],
    );

    /// This callback is called when the get_value operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    /// `ret_buf`: The ret_buf buffer
    fn get_value_complete(
        &self,
        result: Result<(), ErrorCode>,
        key: &'static mut K,
        ret_buf: &'static mut [u8],
    );

    /// This callback is called when the invalidate_key operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    /// `key`: The key buffer
    fn invalidate_key_complete(&self, result: Result<(), ErrorCode>, key: &'static mut K);

    /// This callback is called when the garbage_collect operation completes
    ///
    /// `result`: Nothing on success, 'ErrorCode' on error
    fn garbage_collect_complete(&self, result: Result<(), ErrorCode>);
}

pub trait KVSystem<'a> {
    /// The type of the hashed key. For example '[u8; 8]'.
    type K: KeyType;

    /// Set the client
    fn set_client(&self, client: &'a dyn Client<Self::K>);

    /// Generate key
    ///
    /// `unhashed_key`: A unhashed key that should be hashed.
    /// `key_buf`: A buffer to store the hashed key output.
    ///
    /// On success returns nothing.
    /// On error the unhashed_key, key_buf and `Result<(), ErrorCode>` will be returned.
    fn generate_key(
        &self,
        unhashed_key: &'static mut [u8],
        key_buf: &'static mut Self::K,
    ) -> Result<
        (),
        (
            &'static mut [u8],
            &'static mut Self::K,
            Result<(), ErrorCode>,
        ),
    >;

    /// Appends the key/value pair.
    ///
    /// `key`: A hashed key. This key will be used in future to retrieve
    ///        or remove the `value`.
    /// `value`: A buffer containing the data to be stored to flash.
    ///
    /// On success nothing will be returned.
    /// On error the key, value and a `Result<(), ErrorCode>` will be returned.
    ///
    /// The possible `Result<(), ErrorCode>`s are:
    ///    `BUSY`: An operation is already in progress
    ///    `INVAL`: An invalid parameter was passed
    ///    `NODEVICE`: No KV store was setup
    ///    `ENOSUPPORT`: The key could not be added due to a collision.
    ///    `NOMEM`: The key could not be added due to no more space.
    fn append_key(
        &self,
        key: &'static mut Self::K,
        value: &'static mut [u8],
    ) -> Result<
        (),
        (
            &'static mut Self::K,
            &'static mut [u8],
            Result<(), ErrorCode>,
        ),
    >;

    /// Retrieves the value from a specified key.
    ///
    /// `key`: A hashed key. This key will be used to retrieve the `value`.
    /// `ret_buf`: A buffer to store the value to.
    ///
    /// On success nothing will be returned.
    /// On error the key, ret_buf and a `Result<(), ErrorCode>` will be returned.
    ///
    /// The possible `Result<(), ErrorCode>`s are:
    ///    `BUSY`: An operation is already in progress
    ///    `INVAL`: An invalid parameter was passed
    ///    `NODEVICE`: No KV store was setup
    ///    `ENOSUPPORT`: The key could not be found.
    fn get_value(
        &self,
        key: &'static mut Self::K,
        ret_buf: &'static mut [u8],
    ) -> Result<
        (),
        (
            &'static mut Self::K,
            &'static mut [u8],
            Result<(), ErrorCode>,
        ),
    >;

    /// Invalidates the key in flash storage
    ///
    /// `key`: A hashed key. This key will be used to remove the `value`.
    ///
    /// On success nothing will be returned.
    /// On error the key and a `Result<(), ErrorCode>` will be returned.
    ///
    /// The possible `Result<(), ErrorCode>`s are:
    ///    `BUSY`: An operation is already in progress
    ///    `INVAL`: An invalid parameter was passed
    ///    `NODEVICE`: No KV store was setup
    ///    `ENOSUPPORT`: The key could not be found.
    fn invalidate_key(
        &self,
        key: &'static mut Self::K,
    ) -> Result<(), (&'static mut Self::K, Result<(), ErrorCode>)>;

    /// Perform a garbage collection on the KV Store
    ///
    /// For implementations that don't require garbage collecting
    /// this can just be a NOP that returns 'Ok(0)'.
    ///
    /// On success the number of bytes freed will be returned.
    /// On error a `Result<(), ErrorCode>` will be returned.
    ///
    /// The possible `Result<(), ErrorCode>`s are:
    ///    `BUSY`: An operation is already in progress
    ///    `INVAL`: An invalid parameter was passed
    ///    `NODEVICE`: No KV store was setup
    fn garbage_collect(&self) -> Result<usize, Result<(), ErrorCode>>;
}