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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
extern crate clap;
extern crate libflate;
use clap::App;
use clap::Arg;
use clap::SubCommand;
use libflate::gzip;
use libflate::zlib;
use std::fs;
use std::io;
use std::io::Read;
use std::io::Write;
use std::process;
fn main() {
let matches = App::new("deflate")
.arg(
Arg::with_name("INPUT")
.short("i")
.long("input")
.value_name("FILE")
.takes_value(true)
.default_value("-"),
)
.arg(
Arg::with_name("OUTPUT")
.short("o")
.long("output")
.value_name("FILE")
.takes_value(true)
.default_value("-"),
)
.arg(Arg::with_name("VERBOSE").short("v").long("verbose"))
.subcommand(SubCommand::with_name("copy"))
.subcommand(
SubCommand::with_name("byte-read").arg(
Arg::with_name("UNIT")
.short("u")
.long("unit")
.takes_value(true)
.default_value("1"),
),
)
.subcommand(SubCommand::with_name("gzip-decode"))
.subcommand(SubCommand::with_name("gzip-decode-multi"))
.subcommand(SubCommand::with_name("gzip-encode"))
.subcommand(SubCommand::with_name("zlib-decode"))
.subcommand(SubCommand::with_name("zlib-encode"))
.get_matches();
let input_filename = matches.value_of("INPUT").unwrap();
let input: Box<io::Read> = if input_filename == "-" {
Box::new(io::stdin())
} else {
Box::new(
fs::File::open(input_filename).expect(&format!("Can't open file: {}", input_filename)),
)
};
let mut input = io::BufReader::new(input);
let output_filename = matches.value_of("OUTPUT").unwrap();
let output: Box<io::Write> = if output_filename == "-" {
Box::new(io::stdout())
} else if output_filename == "/dev/null" {
Box::new(io::sink())
} else {
Box::new(
fs::File::create(output_filename)
.expect(&format!("Can't create file: {}", output_filename)),
)
};
let mut output = io::BufWriter::new(output);
let verbose = matches.is_present("VERBOSE");
if let Some(_matches) = matches.subcommand_matches("copy") {
io::copy(&mut input, &mut output).expect("Coyping failed");
} else if let Some(matches) = matches.subcommand_matches("byte-read") {
let unit = matches
.value_of("UNIT")
.and_then(|x| x.parse::<usize>().ok())
.unwrap();
let mut buf = vec![0; unit];
let mut reader = input;
let mut count = 0;
while let Ok(size) = reader.read(&mut buf) {
if size == 0 {
break;
}
count += size;
}
println!("COUNT: {}", count);
} else if let Some(_matches) = matches.subcommand_matches("gzip-decode") {
let mut decoder = gzip::Decoder::new(input).expect("Read GZIP header failed");
if verbose {
let _ = writeln!(&mut io::stderr(), "HEADER: {:?}", decoder.header());
}
io::copy(&mut decoder, &mut output).expect("Decoding GZIP stream failed");
} else if let Some(_matches) = matches.subcommand_matches("gzip-decode-multi") {
let mut decoder = gzip::MultiDecoder::new(input).expect("Read GZIP header failed");
io::copy(&mut decoder, &mut output).expect("Decoding GZIP stream failed");
} else if let Some(_matches) = matches.subcommand_matches("gzip-encode") {
let mut encoder = gzip::Encoder::new(output).unwrap();
io::copy(&mut input, &mut encoder).expect("Encoding GZIP stream failed");
encoder.finish().into_result().unwrap();
} else if let Some(_matches) = matches.subcommand_matches("zlib-decode") {
let mut decoder = zlib::Decoder::new(input).expect("Read ZLIB header failed");
if verbose {
let _ = writeln!(&mut io::stderr(), "HEADER: {:?}", decoder.header());
}
io::copy(&mut decoder, &mut output).expect("Decoding ZLIB stream failed");
} else if let Some(_matches) = matches.subcommand_matches("zlib-encode") {
let mut encoder = zlib::Encoder::new(output).unwrap();
io::copy(&mut input, &mut encoder).expect("Encoding ZLIB stream failed");
encoder.finish().into_result().unwrap();
} else {
println!("{}", matches.usage());
process::exit(1);
}
}
|
