mirror of
https://github.com/darkrenaissance/darkfi.git
synced 2026-04-28 03:00:18 -04:00
util/encoding: Add base64 encoder and decoder.
This commit is contained in:
parazyd
267
src/util/encoding/base64.rs
Normal file
267
src/util/encoding/base64.rs
Normal file
@@ -0,0 +1,267 @@
|
||||
/* This file is part of DarkFi (https://dark.fi)
|
||||
*
|
||||
* Copyright (C) 2020-2023 Dyne.org foundation
|
||||
* Copyright (C) 2021-2023 Kavan Mevada (MIT) (https://github.com/kavanmevada/base64cr)
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of the
|
||||
* License, or (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#![forbid(unsafe_code)]
|
||||
|
||||
macro_rules! seq4 {
|
||||
[$($e:expr),*] => { [$($e,$e,$e,$e,)*] }
|
||||
}
|
||||
|
||||
macro_rules! rep4 {
|
||||
[$($e:expr),*] => { [$($e,)*$($e,)*$($e,)*$($e,)*] }
|
||||
}
|
||||
|
||||
static E0: [char; 256] = seq4![
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S',
|
||||
'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l',
|
||||
'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4',
|
||||
'5', '6', '7', '8', '9', '+', '/'
|
||||
];
|
||||
|
||||
static E1: [char; 256] = rep4![
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S',
|
||||
'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l',
|
||||
'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4',
|
||||
'5', '6', '7', '8', '9', '+', '/'
|
||||
];
|
||||
|
||||
static E2: [char; 256] = E1;
|
||||
|
||||
const FF: u32 = 33554431;
|
||||
|
||||
static D0: [u32; 256] = [
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, 248, FF, FF, FF,
|
||||
252, 208, 212, 216, 220, 224, 228, 232, 236, 240, 244, FF, FF, FF, FF, FF, FF, FF, 0, 4, 8, 12,
|
||||
16, 20, 24, 28, 32, 36, 40, 44, 48, 52, 56, 60, 64, 68, 72, 76, 80, 84, 88, 92, 96, 100, FF,
|
||||
FF, FF, FF, FF, FF, 104, 108, 112, 116, 120, 124, 128, 132, 136, 140, 144, 148, 152, 156, 160,
|
||||
164, 168, 172, 176, 180, 184, 188, 192, 196, 200, 204, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF,
|
||||
];
|
||||
|
||||
static D1: [u32; 256] = [
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, 57347, FF, FF, FF,
|
||||
61443, 16387, 20483, 24579, 28675, 32771, 36867, 40963, 45059, 49155, 53251, FF, FF, FF, FF,
|
||||
FF, FF, FF, 0, 4096, 8192, 12288, 16384, 20480, 24576, 28672, 32768, 36864, 40960, 45056,
|
||||
49152, 53248, 57344, 61440, 1, 4097, 8193, 12289, 16385, 20481, 24577, 28673, 32769, 36865, FF,
|
||||
FF, FF, FF, FF, FF, 40961, 45057, 49153, 53249, 57345, 61441, 2, 4098, 8194, 12290, 16386,
|
||||
20482, 24578, 28674, 32770, 36866, 40962, 45058, 49154, 53250, 57346, 61442, 3, 4099, 8195,
|
||||
12291, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
];
|
||||
|
||||
static D2: [u32; 256] = [
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, 8392448, FF, FF,
|
||||
FF, 12586752, 3328, 4197632, 8391936, 12586240, 3584, 4197888, 8392192, 12586496, 3840,
|
||||
4198144, FF, FF, FF, FF, FF, FF, FF, 0, 4194304, 8388608, 12582912, 256, 4194560, 8388864,
|
||||
12583168, 512, 4194816, 8389120, 12583424, 768, 4195072, 8389376, 12583680, 1024, 4195328,
|
||||
8389632, 12583936, 1280, 4195584, 8389888, 12584192, 1536, 4195840, FF, FF, FF, FF, FF, FF,
|
||||
8390144, 12584448, 1792, 4196096, 8390400, 12584704, 2048, 4196352, 8390656, 12584960, 2304,
|
||||
4196608, 8390912, 12585216, 2560, 4196864, 8391168, 12585472, 2816, 4197120, 8391424, 12585728,
|
||||
3072, 4197376, 8391680, 12585984, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
];
|
||||
|
||||
static D3: [u32; 256] = [
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, 4063232, FF, FF,
|
||||
FF, 4128768, 3407872, 3473408, 3538944, 3604480, 3670016, 3735552, 3801088, 3866624, 3932160,
|
||||
3997696, FF, FF, FF, FF, FF, FF, FF, 0, 65536, 131072, 196608, 262144, 327680, 393216, 458752,
|
||||
524288, 589824, 655360, 720896, 786432, 851968, 917504, 983040, 1048576, 1114112, 1179648,
|
||||
1245184, 1310720, 1376256, 1441792, 1507328, 1572864, 1638400, FF, FF, FF, FF, FF, FF, 1703936,
|
||||
1769472, 1835008, 1900544, 1966080, 2031616, 2097152, 2162688, 2228224, 2293760, 2359296,
|
||||
2424832, 2490368, 2555904, 2621440, 2686976, 2752512, 2818048, 2883584, 2949120, 3014656,
|
||||
3080192, 3145728, 3211264, 3276800, 3342336, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF, FF,
|
||||
FF,
|
||||
];
|
||||
|
||||
/// Encode a byte slice into a base64 string
|
||||
pub fn encode(data: &[u8]) -> String {
|
||||
let len = data.len();
|
||||
|
||||
let mut dest = vec![0u8; ((4 * len / 3) + 3) & !3];
|
||||
|
||||
let mut i = 0;
|
||||
let mut j = 0;
|
||||
|
||||
if len > 2 {
|
||||
while i < len - 2 {
|
||||
let t1 = data[i];
|
||||
let t2 = data[i + 1];
|
||||
let t3 = data[i + 2];
|
||||
|
||||
dest[j] = E0[t1 as usize] as u8;
|
||||
dest[j + 1] = E1[(((t1 & 0x03) << 4) | ((t2 >> 4) & 0x0F)) as usize] as u8;
|
||||
dest[j + 2] = E1[(((t2 & 0x0F) << 2) | ((t3 >> 6) & 0x03)) as usize] as u8;
|
||||
dest[j + 3] = E2[t3 as usize] as u8;
|
||||
|
||||
i += 3;
|
||||
j += 4;
|
||||
}
|
||||
}
|
||||
match len - i {
|
||||
0 => {}
|
||||
1 => {
|
||||
let t1 = data[i];
|
||||
|
||||
dest[j] = E0[t1 as usize] as u8;
|
||||
dest[j + 1] = E1[((t1 & 0x03) << 4) as usize] as u8;
|
||||
dest[j + 2] = b'=';
|
||||
dest[j + 3] = b'=';
|
||||
}
|
||||
_ => {
|
||||
/* case 2 */
|
||||
let t1 = data[i] as usize;
|
||||
let t2 = data[i + 1] as usize;
|
||||
|
||||
dest[j] = E0[t1] as u8;
|
||||
dest[j + 1] = E1[((t1 & 0x03) << 4) | ((t2 >> 4) & 0x0F)] as u8;
|
||||
dest[j + 2] = E2[(t2 & 0x0F) << 2] as u8;
|
||||
dest[j + 3] = b'=';
|
||||
}
|
||||
}
|
||||
|
||||
String::from_utf8(dest).unwrap()
|
||||
}
|
||||
|
||||
/// Tries to decode a base64 string into a byte vector.
|
||||
/// Returns `None` if something fails.
|
||||
pub fn decode(data: &str) -> Option<Vec<u8>> {
|
||||
if !data.is_ascii() {
|
||||
return None
|
||||
}
|
||||
|
||||
let data = data.as_bytes();
|
||||
|
||||
let mut len = data.len();
|
||||
|
||||
if data[len - 1] == b'=' {
|
||||
len -= 1;
|
||||
if data[len - 1] == b'=' {
|
||||
len -= 1;
|
||||
}
|
||||
}
|
||||
|
||||
let mut dest = vec![0u8; (3 * (data.len() / 4)) - (data.len() - len)];
|
||||
|
||||
let leftover = len % 4;
|
||||
let chunks = if leftover == 0 { len / 4 - 1 } else { len / 4 };
|
||||
|
||||
let mut j = 0;
|
||||
let mut k = 0;
|
||||
for _ in 0..chunks {
|
||||
let x: u32 = D0[data[k] as usize] |
|
||||
D1[data[1 + k] as usize] |
|
||||
D2[data[2 + k] as usize] |
|
||||
D3[data[3 + k] as usize];
|
||||
|
||||
dest[j] = x as u8;
|
||||
dest[j + 1] = (x >> 8) as u8;
|
||||
dest[j + 2] = (x >> 16) as u8;
|
||||
|
||||
j += 3;
|
||||
k += 4;
|
||||
}
|
||||
|
||||
match leftover {
|
||||
0 => {
|
||||
let x: u32 = D0[data[k] as usize] |
|
||||
D1[data[1 + k] as usize] |
|
||||
D2[data[2 + k] as usize] |
|
||||
D3[data[3 + k] as usize];
|
||||
|
||||
dest[j] = x as u8;
|
||||
dest[j + 1] = (x >> 8) as u8;
|
||||
dest[j + 2] = (x >> 16) as u8;
|
||||
|
||||
// (chunks + 1) * 3)
|
||||
return Some(dest)
|
||||
}
|
||||
1 => {
|
||||
/* with padding this is an impossible case */
|
||||
let x: u32 = D0[data[k] as usize]; // i.e. first char/byte in int
|
||||
dest[j] = x as u8;
|
||||
}
|
||||
2 => {
|
||||
// * case 2, 1 output byte */
|
||||
let x: u32 = D0[data[k] as usize] | D1[data[1 + k] as usize]; // i.e. first char
|
||||
dest[j] = x as u8;
|
||||
}
|
||||
_ => {
|
||||
let x: u32 = D0[data[k] as usize] | D1[data[1 + k] as usize] | D2[data[2 + k] as usize]; /* 0x3c */
|
||||
dest[j] = x as u8;
|
||||
dest[j + 1] = (x >> 8) as u8;
|
||||
}
|
||||
}
|
||||
|
||||
// 3 * chunks + (6 * leftover) / 8
|
||||
Some(dest)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
pub fn b64_encdec() {
|
||||
const EXAMPLES: [(&[u8], &str); 3] = [
|
||||
(b"abc123!?$*&()'-=@~", "YWJjMTIzIT8kKiYoKSctPUB+"),
|
||||
(b"gm world", "Z20gd29ybGQ="),
|
||||
(
|
||||
b"Man is distinguished, not only by his reason, but by this singular passion from \
|
||||
other animals, which is a lust of the mind, that by a perseverance of delight \
|
||||
in the continued and indefatigable generation of knowledge, exceeds the short \
|
||||
vehemence of any carnal pleasure.",
|
||||
"TWFuIGlzIGRpc3Rpbmd1aXNoZWQsIG5vdCBvbmx5IGJ5IGhpcyByZWFzb24sIGJ1dCBieSB0aGlz\
|
||||
IHNpbmd1bGFyIHBhc3Npb24gZnJvbSBvdGhlciBhbmltYWxzLCB3aGljaCBpcyBhIGx1c3Qgb2Yg\
|
||||
dGhlIG1pbmQsIHRoYXQgYnkgYSBwZXJzZXZlcmFuY2Ugb2YgZGVsaWdodCBpbiB0aGUgY29udGlu\
|
||||
dWVkIGFuZCBpbmRlZmF0aWdhYmxlIGdlbmVyYXRpb24gb2Yga25vd2xlZGdlLCBleGNlZWRzIHRo\
|
||||
ZSBzaG9ydCB2ZWhlbWVuY2Ugb2YgYW55IGNhcm5hbCBwbGVhc3VyZS4=",
|
||||
),
|
||||
];
|
||||
|
||||
for &(input, answer) in EXAMPLES.iter() {
|
||||
let res = encode(&input);
|
||||
assert_eq!(answer, res);
|
||||
|
||||
let res = decode(&answer).unwrap();
|
||||
assert_eq!(input, res);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -18,3 +18,6 @@
|
||||
|
||||
/// Base32 encoding and decoding
|
||||
pub mod base32;
|
||||
|
||||
/// Base64 encoding and decoding
|
||||
pub mod base64;
|
||||
|
||||
Reference in New Issue
Block a user