2012-12-03 18:48:01 -06:00
|
|
|
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
2013-03-27 02:40:15 -05:00
|
|
|
//! Base64 binary-to-text encoding
|
|
|
|
|
2012-12-23 16:41:37 -06:00
|
|
|
use core::iter;
|
|
|
|
use core::str;
|
|
|
|
use core::vec;
|
2012-07-03 23:29:45 -05:00
|
|
|
|
2012-09-27 18:43:15 -05:00
|
|
|
pub trait ToBase64 {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn to_base64(&self) -> ~str;
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
|
2013-03-26 22:52:16 -05:00
|
|
|
static CHARS: [char, ..64] = [
|
2013-03-24 01:51:18 -05:00
|
|
|
'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', '+', '/'
|
|
|
|
];
|
|
|
|
|
2013-03-25 15:21:04 -05:00
|
|
|
impl<'self> ToBase64 for &'self [u8] {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn to_base64(&self) -> ~str {
|
2012-07-14 00:57:48 -05:00
|
|
|
let mut s = ~"";
|
2012-11-17 12:13:11 -06:00
|
|
|
unsafe {
|
|
|
|
let len = self.len();
|
|
|
|
str::reserve(&mut s, ((len + 3u) / 4u) * 3u);
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
let mut i = 0u;
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
while i < len - (len % 3u) {
|
|
|
|
let n = (self[i] as uint) << 16u |
|
|
|
|
(self[i + 1u] as uint) << 8u |
|
|
|
|
(self[i + 2u] as uint);
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
// This 24-bit number gets separated into four 6-bit numbers.
|
2013-03-24 01:51:18 -05:00
|
|
|
str::push_char(&mut s, CHARS[(n >> 18u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[(n >> 12u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[(n >> 6u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[n & 63u]);
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
i += 3u;
|
|
|
|
}
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
// Heh, would be cool if we knew this was exhaustive
|
|
|
|
// (the dream of bounded integer types)
|
|
|
|
match len % 3 {
|
|
|
|
0 => (),
|
|
|
|
1 => {
|
|
|
|
let n = (self[i] as uint) << 16u;
|
2013-03-24 01:51:18 -05:00
|
|
|
str::push_char(&mut s, CHARS[(n >> 18u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[(n >> 12u) & 63u]);
|
2012-11-17 12:13:11 -06:00
|
|
|
str::push_char(&mut s, '=');
|
|
|
|
str::push_char(&mut s, '=');
|
|
|
|
}
|
|
|
|
2 => {
|
2012-11-17 13:00:38 -06:00
|
|
|
let n = (self[i] as uint) << 16u |
|
|
|
|
(self[i + 1u] as uint) << 8u;
|
2013-03-24 01:51:18 -05:00
|
|
|
str::push_char(&mut s, CHARS[(n >> 18u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[(n >> 12u) & 63u]);
|
|
|
|
str::push_char(&mut s, CHARS[(n >> 6u) & 63u]);
|
2012-11-17 12:13:11 -06:00
|
|
|
str::push_char(&mut s, '=');
|
|
|
|
}
|
2013-02-11 21:26:38 -06:00
|
|
|
_ => fail!(~"Algebra is broken, please alert the math police")
|
2012-11-17 12:13:11 -06:00
|
|
|
}
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
s
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-03-25 15:21:04 -05:00
|
|
|
impl<'self> ToBase64 for &'self str {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn to_base64(&self) -> ~str {
|
2013-03-07 20:11:09 -06:00
|
|
|
str::to_bytes(*self).to_base64()
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-09-27 18:43:15 -05:00
|
|
|
pub trait FromBase64 {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn from_base64(&self) -> ~[u8];
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
|
2013-02-14 13:47:00 -06:00
|
|
|
impl FromBase64 for ~[u8] {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn from_base64(&self) -> ~[u8] {
|
2013-02-11 21:26:38 -06:00
|
|
|
if self.len() % 4u != 0u { fail!(~"invalid base64 length"); }
|
2012-07-03 23:29:45 -05:00
|
|
|
|
|
|
|
let len = self.len();
|
|
|
|
let mut padding = 0u;
|
|
|
|
|
|
|
|
if len != 0u {
|
|
|
|
if self[len - 1u] == '=' as u8 { padding += 1u; }
|
|
|
|
if self[len - 2u] == '=' as u8 { padding += 1u; }
|
|
|
|
}
|
|
|
|
|
2012-09-21 20:43:30 -05:00
|
|
|
let mut r = vec::with_capacity((len / 4u) * 3u - padding);
|
2012-07-03 23:29:45 -05:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
unsafe {
|
|
|
|
let mut i = 0u;
|
|
|
|
while i < len {
|
|
|
|
let mut n = 0u;
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
for iter::repeat(4u) {
|
|
|
|
let ch = self[i] as char;
|
|
|
|
n <<= 6u;
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
if ch >= 'A' && ch <= 'Z' {
|
|
|
|
n |= (ch as uint) - 0x41u;
|
|
|
|
} else if ch >= 'a' && ch <= 'z' {
|
|
|
|
n |= (ch as uint) - 0x47u;
|
|
|
|
} else if ch >= '0' && ch <= '9' {
|
|
|
|
n |= (ch as uint) + 0x04u;
|
|
|
|
} else if ch == '+' {
|
|
|
|
n |= 0x3Eu;
|
|
|
|
} else if ch == '/' {
|
|
|
|
n |= 0x3Fu;
|
|
|
|
} else if ch == '=' {
|
|
|
|
match len - i {
|
|
|
|
1u => {
|
|
|
|
r.push(((n >> 16u) & 0xFFu) as u8);
|
|
|
|
r.push(((n >> 8u ) & 0xFFu) as u8);
|
|
|
|
return copy r;
|
|
|
|
}
|
|
|
|
2u => {
|
|
|
|
r.push(((n >> 10u) & 0xFFu) as u8);
|
|
|
|
return copy r;
|
|
|
|
}
|
2013-02-11 21:26:38 -06:00
|
|
|
_ => fail!(~"invalid base64 padding")
|
2012-11-17 12:13:11 -06:00
|
|
|
}
|
|
|
|
} else {
|
2013-02-11 21:26:38 -06:00
|
|
|
fail!(~"invalid base64 character");
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
i += 1u;
|
|
|
|
};
|
2012-11-17 13:00:38 -06:00
|
|
|
|
2012-11-17 12:13:11 -06:00
|
|
|
r.push(((n >> 16u) & 0xFFu) as u8);
|
|
|
|
r.push(((n >> 8u ) & 0xFFu) as u8);
|
|
|
|
r.push(((n ) & 0xFFu) as u8);
|
|
|
|
}
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
r
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-02-14 13:47:00 -06:00
|
|
|
impl FromBase64 for ~str {
|
2013-03-21 23:34:30 -05:00
|
|
|
fn from_base64(&self) -> ~[u8] {
|
2013-03-07 20:11:09 -06:00
|
|
|
str::to_bytes(*self).from_base64()
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
2012-12-27 20:24:18 -06:00
|
|
|
use core::str;
|
|
|
|
|
2012-07-03 23:29:45 -05:00
|
|
|
#[test]
|
2013-01-29 14:06:09 -06:00
|
|
|
pub fn test_to_base64() {
|
2013-03-06 15:58:02 -06:00
|
|
|
fail_unless!((~"").to_base64() == ~"");
|
|
|
|
fail_unless!((~"f").to_base64() == ~"Zg==");
|
|
|
|
fail_unless!((~"fo").to_base64() == ~"Zm8=");
|
|
|
|
fail_unless!((~"foo").to_base64() == ~"Zm9v");
|
|
|
|
fail_unless!((~"foob").to_base64() == ~"Zm9vYg==");
|
|
|
|
fail_unless!((~"fooba").to_base64() == ~"Zm9vYmE=");
|
|
|
|
fail_unless!((~"foobar").to_base64() == ~"Zm9vYmFy");
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2013-01-29 14:06:09 -06:00
|
|
|
pub fn test_from_base64() {
|
2013-03-06 15:58:02 -06:00
|
|
|
fail_unless!((~"").from_base64() == str::to_bytes(~""));
|
|
|
|
fail_unless!((~"Zg==").from_base64() == str::to_bytes(~"f"));
|
|
|
|
fail_unless!((~"Zm8=").from_base64() == str::to_bytes(~"fo"));
|
|
|
|
fail_unless!((~"Zm9v").from_base64() == str::to_bytes(~"foo"));
|
|
|
|
fail_unless!((~"Zm9vYg==").from_base64() == str::to_bytes(~"foob"));
|
|
|
|
fail_unless!((~"Zm9vYmE=").from_base64() == str::to_bytes(~"fooba"));
|
|
|
|
fail_unless!((~"Zm9vYmFy").from_base64() == str::to_bytes(~"foobar"));
|
2012-07-03 23:29:45 -05:00
|
|
|
}
|
|
|
|
}
|