Add support for HMAC, RC4, AES-128, hex encoding, etc
This commit is contained in:
parent
d436384169
commit
07773d8fee
|
|
@ -0,0 +1,7 @@
|
||||||
|
|
||||||
|
crypto: crypto.rc $(wildcard *.rs)
|
||||||
|
rustc crypto.rc
|
||||||
|
rustc --test crypto.rc
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f crypto libcrypto-*.so
|
||||||
|
|
@ -0,0 +1,91 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2013 Jack Lloyd
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
extern mod std;
|
||||||
|
|
||||||
|
pub trait ToHex {
|
||||||
|
pure fn to_hex() -> ~str;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl &[u8]: ToHex {
|
||||||
|
pure fn to_hex() -> ~str {
|
||||||
|
|
||||||
|
let chars = str::chars(~"0123456789ABCDEF");
|
||||||
|
|
||||||
|
let mut s = ~"";
|
||||||
|
|
||||||
|
for uint::range(0, self.len()) |i| {
|
||||||
|
|
||||||
|
let x = self[i];
|
||||||
|
|
||||||
|
let xhi = (x >> 4) & 0x0F;
|
||||||
|
let xlo = (x ) & 0x0F;
|
||||||
|
|
||||||
|
unsafe {
|
||||||
|
str::push_char(&mut s, chars[xhi]);
|
||||||
|
str::push_char(&mut s, chars[xlo]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait FromHex {
|
||||||
|
pure fn from_hex() -> ~[u8];
|
||||||
|
}
|
||||||
|
|
||||||
|
impl &str: FromHex {
|
||||||
|
pure fn from_hex() -> ~[u8] {
|
||||||
|
let mut vec = vec::with_capacity(self.len() / 2);
|
||||||
|
|
||||||
|
for str::each_chari(self) |i,c| {
|
||||||
|
|
||||||
|
let nibble =
|
||||||
|
if c >= '0' && c <= '9' { (c as u8) - 0x30 }
|
||||||
|
else if c >= 'a' && c <= 'f' { (c as u8) - (0x61 - 10) }
|
||||||
|
else if c >= 'A' && c <= 'F' { (c as u8) - (0x41 - 10) }
|
||||||
|
else { fail ~"bad hex character"; };
|
||||||
|
|
||||||
|
if i % 2 == 0 {
|
||||||
|
unsafe {
|
||||||
|
vec::push(&mut vec, nibble << 4);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
vec[i/2] |= nibble;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
vec
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
pub fn test() {
|
||||||
|
|
||||||
|
assert [05u8, 0xffu8, 0x00u8, 0x59u8].to_hex() == ~"05FF0059";
|
||||||
|
|
||||||
|
assert "00FFA9D1F5".from_hex() == ~[0, 0xff, 0xa9, 0xd1, 0xf5];
|
||||||
|
|
||||||
|
assert "00FFA9D1F5".from_hex().to_hex() == ~"00FFA9D1F5";
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,92 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2013 Jack Lloyd
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
use hash::*;
|
||||||
|
|
||||||
|
#[allow(non_camel_case_types)]
|
||||||
|
struct HMAC_CTX {
|
||||||
|
mut md: EVP_MD,
|
||||||
|
mut md_ctx: EVP_MD_CTX,
|
||||||
|
mut i_ctx: EVP_MD_CTX,
|
||||||
|
mut o_ctx: EVP_MD_CTX,
|
||||||
|
mut key_length: libc::c_uint,
|
||||||
|
mut key: [libc::c_uchar * 128]
|
||||||
|
}
|
||||||
|
|
||||||
|
#[link_name = "crypto"]
|
||||||
|
#[abi = "cdecl"]
|
||||||
|
extern mod libcrypto {
|
||||||
|
|
||||||
|
fn HMAC_CTX_init(ctx: *mut HMAC_CTX, key: *u8, keylen: libc::c_int, md: EVP_MD);
|
||||||
|
|
||||||
|
fn HMAC_Update(ctx: *mut HMAC_CTX, input: *u8, len: libc::c_uint);
|
||||||
|
|
||||||
|
fn HMAC_Final(ctx: *mut HMAC_CTX, output: *mut u8, len: *mut libc::c_uint);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct HMAC {
|
||||||
|
priv mut ctx: HMAC_CTX,
|
||||||
|
priv len: uint,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn HMAC(ht: HashType, key: ~[u8]) -> HMAC {
|
||||||
|
unsafe {
|
||||||
|
|
||||||
|
let (evp, mdlen) = evpmd(ht);
|
||||||
|
|
||||||
|
let mut ctx : HMAC_CTX = HMAC_CTX {
|
||||||
|
mut md: ptr::null(),
|
||||||
|
mut md_ctx: ptr::null(),
|
||||||
|
mut i_ctx: ptr::null(),
|
||||||
|
mut o_ctx: ptr::null(),
|
||||||
|
mut key_length: 0,
|
||||||
|
mut key: [0u8, .. 128]
|
||||||
|
};
|
||||||
|
|
||||||
|
libcrypto::HMAC_CTX_init(&mut ctx,
|
||||||
|
vec::raw::to_ptr(key),
|
||||||
|
key.len() as libc::c_int,
|
||||||
|
evp);
|
||||||
|
|
||||||
|
HMAC { ctx: ctx, len: mdlen }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub impl HMAC {
|
||||||
|
fn update(data: &[u8]) unsafe {
|
||||||
|
do vec::as_imm_buf(data) |pdata, len| {
|
||||||
|
libcrypto::HMAC_Update(&mut self.ctx, pdata, len as libc::c_uint)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn final() -> ~[u8] unsafe {
|
||||||
|
let mut res = vec::from_elem(self.len, 0u8);
|
||||||
|
let mut outlen: libc::c_uint = 0;
|
||||||
|
do vec::as_mut_buf(res) |pres, _len| {
|
||||||
|
libcrypto::HMAC_Final(&mut self.ctx, pres, &mut outlen);
|
||||||
|
assert self.len == outlen as uint
|
||||||
|
}
|
||||||
|
res
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let h = HMAC(SHA512, ~[00u8]);
|
||||||
|
|
||||||
|
h.update(~[00u8]);
|
||||||
|
|
||||||
|
io::println(fmt!("%?", h.final()))
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue