hatter/tiny-encrypt-rs
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: update wk

Browse Source
This commit is contained in:
Hatter Jiang
2023-12-09 20:14:59 +08:00
parent 2d29e6eaed
commit 9b735f8e48
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/wrap_key.rs
View File

@@ -3,6 +3,8 @@ use serde::{Deserialize, Serialize};
use crate::util; use crate::util;
pub const WRAP_KEY_PREFIX: &str = "WK:";
#[derive(Clone, Debug, Serialize, Deserialize)] #[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")] #[serde(rename_all = "camelCase")]
pub struct WrapKey { pub struct WrapKey {
@@ -37,7 +39,7 @@ impl WrapKeyHeader {
impl WrapKey { impl WrapKey {
pub fn encode(&self) -> XResult<String> { pub fn encode(&self) -> XResult<String> {
let mut buf = String::with_capacity(512); let mut buf = String::with_capacity(512);
buf.push_str("WK:"); buf.push_str(WRAP_KEY_PREFIX);
let header = serde_json::to_string(&self.header)?; let header = serde_json::to_string(&self.header)?;
let header_str = util::encode_base64_url_no_pad(header.as_bytes()); let header_str = util::encode_base64_url_no_pad(header.as_bytes());
buf.push_str(&header_str); buf.push_str(&header_str);
@@ -49,7 +51,7 @@ impl WrapKey {
} }
pub fn parse(wk: &str) -> XResult<WrapKey> { pub fn parse(wk: &str) -> XResult<WrapKey> {
if !wk.starts_with("WK:") { if !wk.starts_with(WRAP_KEY_PREFIX) {
return simple_error!("Wrap key string must starts with WK:"); return simple_error!("Wrap key string must starts with WK:");
} }
let wks = wk.split('.').collect::<Vec<_>>(); let wks = wk.split('.').collect::<Vec<_>>();