1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#![deny(missing_docs)]
extern crate libc;
mod ll;
mod stream;
pub mod block;
pub mod dict;
pub use stream::encoder::{AutoFinishEncoder, Encoder};
pub use stream::decoder::Decoder;
use std::io;
use std::ffi::CStr;
pub fn decode_all(data: &[u8]) -> io::Result<Vec<u8>> {
let mut result = Vec::new();
let mut decoder = try!(Decoder::new(data));
try!(io::copy(&mut decoder, &mut result));
Ok(result)
}
pub fn encode_all(data: &[u8], level: i32) -> io::Result<Vec<u8>> {
let result = Vec::<u8>::new();
let mut encoder = try!(Encoder::new(result, level));
let mut input = data;
try!(io::copy(&mut input, &mut encoder));
encoder.finish()
}
fn parse_code(code: libc::size_t) -> Result<usize, io::Error> {
unsafe {
if ll::ZSTD_isError(code) == 0 {
Ok(code as usize)
} else {
let msg = CStr::from_ptr(ll::ZSTD_getErrorName(code));
let error = io::Error::new(io::ErrorKind::Other,
msg.to_str().unwrap().to_string());
Err(error)
}
}
}
#[test]
fn test_cycle() {
let text = include_str!("ll.rs");
let compressed = encode_all(text.as_bytes(), 1).unwrap();
let decompressed = String::from_utf8(decode_all(&compressed).unwrap())
.unwrap();
assert_eq!(text, &decompressed);
}