summaryrefslogtreecommitdiffstats
path: root/vendor/ct-codecs/src/base64.rs
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/ct-codecs/src/base64.rs')
-rw-r--r--vendor/ct-codecs/src/base64.rs378
1 files changed, 378 insertions, 0 deletions
diff --git a/vendor/ct-codecs/src/base64.rs b/vendor/ct-codecs/src/base64.rs
new file mode 100644
index 000000000..1e031de28
--- /dev/null
+++ b/vendor/ct-codecs/src/base64.rs
@@ -0,0 +1,378 @@
+use crate::error::*;
+use crate::{Decoder, Encoder};
+
+struct Base64Impl;
+
+#[derive(Copy, Clone, Debug, Eq, PartialEq)]
+enum Base64Variant {
+ Original = 1,
+ OriginalNoPadding = 3,
+ UrlSafe = 5,
+ UrlSafeNoPadding = 7,
+}
+
+enum VariantMask {
+ NoPadding = 2,
+ UrlSafe = 4,
+}
+
+impl Base64Impl {
+ #[inline]
+ fn _eq(x: u8, y: u8) -> u8 {
+ !(((0u16.wrapping_sub((x as u16) ^ (y as u16))) >> 8) as u8)
+ }
+
+ #[inline]
+ fn _gt(x: u8, y: u8) -> u8 {
+ (((y as u16).wrapping_sub(x as u16)) >> 8) as u8
+ }
+
+ #[inline]
+ fn _ge(x: u8, y: u8) -> u8 {
+ !Self::_gt(y, x)
+ }
+
+ #[inline]
+ fn _lt(x: u8, y: u8) -> u8 {
+ Self::_gt(y, x)
+ }
+
+ #[inline]
+ fn _le(x: u8, y: u8) -> u8 {
+ Self::_ge(y, x)
+ }
+
+ #[inline]
+ fn b64_byte_to_char(x: u8) -> u8 {
+ (Self::_lt(x, 26) & (x.wrapping_add(b'A')))
+ | (Self::_ge(x, 26) & Self::_lt(x, 52) & (x.wrapping_add(b'a'.wrapping_sub(26))))
+ | (Self::_ge(x, 52) & Self::_lt(x, 62) & (x.wrapping_add(b'0'.wrapping_sub(52))))
+ | (Self::_eq(x, 62) & b'+')
+ | (Self::_eq(x, 63) & b'/')
+ }
+
+ #[inline]
+ fn b64_char_to_byte(c: u8) -> u8 {
+ let x = (Self::_ge(c, b'A') & Self::_le(c, b'Z') & (c.wrapping_sub(b'A')))
+ | (Self::_ge(c, b'a') & Self::_le(c, b'z') & (c.wrapping_sub(b'a'.wrapping_sub(26))))
+ | (Self::_ge(c, b'0') & Self::_le(c, b'9') & (c.wrapping_sub(b'0'.wrapping_sub(52))))
+ | (Self::_eq(c, b'+') & 62)
+ | (Self::_eq(c, b'/') & 63);
+ x | (Self::_eq(x, 0) & (Self::_eq(c, b'A') ^ 0xff))
+ }
+
+ #[inline]
+ fn b64_byte_to_urlsafe_char(x: u8) -> u8 {
+ (Self::_lt(x, 26) & (x.wrapping_add(b'A')))
+ | (Self::_ge(x, 26) & Self::_lt(x, 52) & (x.wrapping_add(b'a'.wrapping_sub(26))))
+ | (Self::_ge(x, 52) & Self::_lt(x, 62) & (x.wrapping_add(b'0'.wrapping_sub(52))))
+ | (Self::_eq(x, 62) & b'-')
+ | (Self::_eq(x, 63) & b'_')
+ }
+
+ #[inline]
+ fn b64_urlsafe_char_to_byte(c: u8) -> u8 {
+ let x = (Self::_ge(c, b'A') & Self::_le(c, b'Z') & (c.wrapping_sub(b'A')))
+ | (Self::_ge(c, b'a') & Self::_le(c, b'z') & (c.wrapping_sub(b'a'.wrapping_sub(26))))
+ | (Self::_ge(c, b'0') & Self::_le(c, b'9') & (c.wrapping_sub(b'0'.wrapping_sub(52))))
+ | (Self::_eq(c, b'-') & 62)
+ | (Self::_eq(c, b'_') & 63);
+ x | (Self::_eq(x, 0) & (Self::_eq(c, b'A') ^ 0xff))
+ }
+
+ #[inline]
+ fn encoded_len(bin_len: usize, variant: Base64Variant) -> Result<usize, Error> {
+ let nibbles = bin_len / 3;
+ let rounded = nibbles * 3;
+ let pad = bin_len - rounded;
+ Ok(nibbles.checked_mul(4).ok_or(Error::Overflow)?
+ + ((pad | (pad >> 1)) & 1)
+ * (4 - (!((((variant as usize) & 2) >> 1).wrapping_sub(1)) & (3 - pad)))
+ + 1)
+ }
+
+ pub fn encode<'t>(
+ b64: &'t mut [u8],
+ bin: &[u8],
+ variant: Base64Variant,
+ ) -> Result<&'t [u8], Error> {
+ let bin_len = bin.len();
+ let b64_maxlen = b64.len();
+ let mut acc_len = 0usize;
+ let mut b64_pos = 0usize;
+ let mut acc = 0u16;
+
+ let nibbles = bin_len / 3;
+ let remainder = bin_len - 3 * nibbles;
+ let mut b64_len = nibbles * 4;
+ if remainder != 0 {
+ if (variant as u16 & VariantMask::NoPadding as u16) == 0 {
+ b64_len += 4;
+ } else {
+ b64_len += 2 + (remainder >> 1);
+ }
+ }
+ if b64_maxlen < b64_len {
+ return Err(Error::Overflow);
+ }
+ if (variant as u16 & VariantMask::UrlSafe as u16) != 0 {
+ for &v in bin {
+ acc = (acc << 8) + v as u16;
+ acc_len += 8;
+ while acc_len >= 6 {
+ acc_len -= 6;
+ b64[b64_pos] = Self::b64_byte_to_urlsafe_char(((acc >> acc_len) & 0x3f) as u8);
+ b64_pos += 1;
+ }
+ }
+ if acc_len > 0 {
+ b64[b64_pos] =
+ Self::b64_byte_to_urlsafe_char(((acc << (6 - acc_len)) & 0x3f) as u8);
+ b64_pos += 1;
+ }
+ } else {
+ for &v in bin {
+ acc = (acc << 8) + v as u16;
+ acc_len += 8;
+ while acc_len >= 6 {
+ acc_len -= 6;
+ b64[b64_pos] = Self::b64_byte_to_char(((acc >> acc_len) & 0x3f) as u8);
+ b64_pos += 1;
+ }
+ }
+ if acc_len > 0 {
+ b64[b64_pos] = Self::b64_byte_to_char(((acc << (6 - acc_len)) & 0x3f) as u8);
+ b64_pos += 1;
+ }
+ }
+ while b64_pos < b64_len {
+ b64[b64_pos] = b'=';
+ b64_pos += 1
+ }
+ Ok(&b64[..b64_pos])
+ }
+
+ fn skip_padding<'t>(
+ b64: &'t [u8],
+ mut padding_len: usize,
+ ignore: Option<&[u8]>,
+ ) -> Result<&'t [u8], Error> {
+ let b64_len = b64.len();
+ let mut b64_pos = 0usize;
+ while padding_len > 0 {
+ if b64_pos > b64_len {
+ return Err(Error::InvalidInput);
+ }
+ let c = b64[b64_pos];
+ if c == b'=' {
+ padding_len -= 1
+ } else {
+ match ignore {
+ Some(ignore) if ignore.contains(&c) => {}
+ _ => return Err(Error::InvalidInput),
+ }
+ }
+ b64_pos += 1
+ }
+ Ok(&b64[b64_pos..])
+ }
+
+ pub fn decode<'t>(
+ bin: &'t mut [u8],
+ b64: &[u8],
+ ignore: Option<&[u8]>,
+ variant: Base64Variant,
+ ) -> Result<&'t [u8], Error> {
+ let bin_maxlen = bin.len();
+ let is_urlsafe = (variant as u16 & VariantMask::UrlSafe as u16) != 0;
+ let mut acc = 0u16;
+ let mut acc_len = 0usize;
+ let mut bin_pos = 0usize;
+ let mut premature_end = None;
+ for (b64_pos, &c) in b64.iter().enumerate() {
+ let d = if is_urlsafe {
+ Self::b64_urlsafe_char_to_byte(c)
+ } else {
+ Self::b64_char_to_byte(c)
+ };
+ if d == 0xff {
+ match ignore {
+ Some(ignore) if ignore.contains(&c) => continue,
+ _ => {
+ premature_end = Some(b64_pos);
+ break;
+ }
+ }
+ }
+ acc = (acc << 6) + d as u16;
+ acc_len += 6;
+ if acc_len >= 8 {
+ acc_len -= 8;
+ if bin_pos >= bin_maxlen {
+ return Err(Error::Overflow);
+ }
+ bin[bin_pos] = (acc >> acc_len) as u8;
+ bin_pos += 1;
+ }
+ }
+ if acc_len > 4 || (acc & ((1u16 << acc_len).wrapping_sub(1))) != 0 {
+ return Err(Error::InvalidInput);
+ }
+ let padding_len = acc_len / 2;
+ if let Some(premature_end) = premature_end {
+ let remaining = if variant as u16 & VariantMask::NoPadding as u16 == 0 {
+ Self::skip_padding(&b64[premature_end..], padding_len, ignore)?
+ } else {
+ &b64[premature_end..]
+ };
+ match ignore {
+ None => {
+ if !remaining.is_empty() {
+ return Err(Error::InvalidInput);
+ }
+ }
+ Some(ignore) => {
+ for &c in remaining {
+ if !ignore.contains(&c) {
+ return Err(Error::InvalidInput);
+ }
+ }
+ }
+ }
+ } else if variant as u16 & VariantMask::NoPadding as u16 == 0 && padding_len != 0 {
+ return Err(Error::InvalidInput);
+ }
+ Ok(&bin[..bin_pos])
+ }
+}
+
+pub struct Base64;
+pub struct Base64NoPadding;
+pub struct Base64UrlSafe;
+pub struct Base64UrlSafeNoPadding;
+
+impl Encoder for Base64 {
+ #[inline]
+ fn encoded_len(bin_len: usize) -> Result<usize, Error> {
+ Base64Impl::encoded_len(bin_len, Base64Variant::Original)
+ }
+
+ #[inline]
+ fn encode<IN: AsRef<[u8]>>(b64: &mut [u8], bin: IN) -> Result<&[u8], Error> {
+ Base64Impl::encode(b64, bin.as_ref(), Base64Variant::Original)
+ }
+}
+
+impl Decoder for Base64 {
+ #[inline]
+ fn decode<'t, IN: AsRef<[u8]>>(
+ bin: &'t mut [u8],
+ b64: IN,
+ ignore: Option<&[u8]>,
+ ) -> Result<&'t [u8], Error> {
+ Base64Impl::decode(bin, b64.as_ref(), ignore, Base64Variant::Original)
+ }
+}
+
+impl Encoder for Base64NoPadding {
+ #[inline]
+ fn encoded_len(bin_len: usize) -> Result<usize, Error> {
+ Base64Impl::encoded_len(bin_len, Base64Variant::OriginalNoPadding)
+ }
+
+ #[inline]
+ fn encode<IN: AsRef<[u8]>>(b64: &mut [u8], bin: IN) -> Result<&[u8], Error> {
+ Base64Impl::encode(b64, bin.as_ref(), Base64Variant::OriginalNoPadding)
+ }
+}
+
+impl Decoder for Base64NoPadding {
+ #[inline]
+ fn decode<'t, IN: AsRef<[u8]>>(
+ bin: &'t mut [u8],
+ b64: IN,
+ ignore: Option<&[u8]>,
+ ) -> Result<&'t [u8], Error> {
+ Base64Impl::decode(bin, b64.as_ref(), ignore, Base64Variant::OriginalNoPadding)
+ }
+}
+
+impl Encoder for Base64UrlSafe {
+ #[inline]
+ fn encoded_len(bin_len: usize) -> Result<usize, Error> {
+ Base64Impl::encoded_len(bin_len, Base64Variant::UrlSafe)
+ }
+
+ #[inline]
+ fn encode<IN: AsRef<[u8]>>(b64: &mut [u8], bin: IN) -> Result<&[u8], Error> {
+ Base64Impl::encode(b64, bin.as_ref(), Base64Variant::UrlSafe)
+ }
+}
+
+impl Decoder for Base64UrlSafe {
+ #[inline]
+ fn decode<'t, IN: AsRef<[u8]>>(
+ bin: &'t mut [u8],
+ b64: IN,
+ ignore: Option<&[u8]>,
+ ) -> Result<&'t [u8], Error> {
+ Base64Impl::decode(bin, b64.as_ref(), ignore, Base64Variant::UrlSafe)
+ }
+}
+
+impl Encoder for Base64UrlSafeNoPadding {
+ #[inline]
+ fn encoded_len(bin_len: usize) -> Result<usize, Error> {
+ Base64Impl::encoded_len(bin_len, Base64Variant::UrlSafeNoPadding)
+ }
+
+ #[inline]
+ fn encode<IN: AsRef<[u8]>>(b64: &mut [u8], bin: IN) -> Result<&[u8], Error> {
+ Base64Impl::encode(b64, bin.as_ref(), Base64Variant::UrlSafeNoPadding)
+ }
+}
+
+impl Decoder for Base64UrlSafeNoPadding {
+ #[inline]
+ fn decode<'t, IN: AsRef<[u8]>>(
+ bin: &'t mut [u8],
+ b64: IN,
+ ignore: Option<&[u8]>,
+ ) -> Result<&'t [u8], Error> {
+ Base64Impl::decode(bin, b64.as_ref(), ignore, Base64Variant::UrlSafeNoPadding)
+ }
+}
+
+#[cfg(feature = "std")]
+#[test]
+fn test_base64() {
+ let bin = [1u8, 5, 11, 15, 19, 131, 122];
+ let expected = "AQULDxODeg==";
+ let b64 = Base64::encode_to_string(&bin).unwrap();
+ assert_eq!(b64, expected);
+ let bin2 = Base64::decode_to_vec(&b64, None).unwrap();
+ assert_eq!(bin, &bin2[..]);
+}
+
+#[cfg(feature = "std")]
+#[test]
+fn test_base64_mising_padding() {
+ let missing_padding = "AA";
+ assert!(Base64::decode_to_vec(&missing_padding, None).is_err());
+ assert!(Base64NoPadding::decode_to_vec(&missing_padding, None).is_ok());
+ let missing_padding = "AAA";
+ assert!(Base64::decode_to_vec(&missing_padding, None).is_err());
+ assert!(Base64NoPadding::decode_to_vec(&missing_padding, None).is_ok());
+}
+
+#[test]
+fn test_base64_no_std() {
+ let bin = [1u8, 5, 11, 15, 19, 131, 122];
+ let expected = [65, 81, 85, 76, 68, 120, 79, 68, 101, 103, 61, 61];
+ let mut b64 = [0u8; 12];
+ let b64 = Base64::encode(&mut b64, &bin).unwrap();
+ assert_eq!(b64, expected);
+ let mut bin2 = [0u8; 7];
+ let bin2 = Base64::decode(&mut bin2, &b64, None).unwrap();
+ assert_eq!(bin, bin2);
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-18 10:00:49 +0000