aboutsummaryrefslogtreecommitdiffstats
path: root/src/bin
diff options
context:
space:
mode:
Diffstat (limited to 'src/bin')
-rw-r--r--src/bin/bhash.rs76
1 files changed, 76 insertions, 0 deletions
diff --git a/src/bin/bhash.rs b/src/bin/bhash.rs
new file mode 100644
index 0000000..691db2a
--- /dev/null
+++ b/src/bin/bhash.rs
@@ -0,0 +1,76 @@
+// bad-hashish: a tool for recursively, remotely multi-hashing files
+// Copyright (C) 2017 Bryan Newbold <bnewbold@robocracy.org>
+// GPLv3
+
+extern crate bad_hashish;
+
+#[macro_use]
+extern crate error_chain;
+
+extern crate clap;
+extern crate tree_magic;
+extern crate flate2;
+extern crate tar;
+extern crate crypto;
+
+use clap::App;
+use bad_hashish::Result;
+use std::path::Path;
+use std::io::{self, BufReader};
+use std::io::prelude::*;
+use std::fs::File;
+use flate2::read::GzDecoder;
+use tar::Archive;
+use crypto::digest::Digest;
+use crypto::sha1::Sha1;
+
+
+fn run() -> Result<()> {
+
+ let matches = App::new("bad-hashish")
+ .version(env!("CARGO_PKG_VERSION"))
+ .about("a tool for recursively, remotely multi-hashing files")
+ .arg_from_usage("<FILE>... 'files to hash from'")
+ .get_matches();
+
+ for f in matches.values_of("FILE").unwrap() {
+ let path: &Path = Path::new(f);
+ println!("{} ({})", f, tree_magic::from_filepath(path));
+
+ if tree_magic::match_filepath("application/zip", path) {
+ println!("It's a zip.");
+ } else if tree_magic::match_filepath("application/gzip", path) {
+ println!("It's gzip.");
+ let f = File::open(path)?;
+ let mut gz = GzDecoder::new(f)?;
+
+ let mut reader = BufReader::with_capacity(4*1024*1024, gz);
+ let is_tar: bool = {
+ let buf = reader.fill_buf()?;
+ println!("Inside is: {}", tree_magic::from_u8(&buf));
+ tree_magic::match_u8("application/x-tar", &buf)
+ };
+
+ if is_tar {
+ println!("It's a tar inside");
+ let mut a = Archive::new(reader);
+ for inner in a.entries().unwrap() {
+ let mut inner = inner.unwrap();
+ let mut hasher = Sha1::new();
+ let mut buf: [u8; 1*1024*1024] = [0; 1*1024*1024];
+ loop {
+ let got = inner.read(&mut buf[..])?;
+ if got <= 0 { break };
+ hasher.input(&buf[0..got]);
+ }
+ println!("\t{} {:?}", hasher.result_str(), inner.header().path().unwrap());
+ }
+ }
+ }
+ }
+
+ Ok(())
+}
+
+// At least for now...
+quick_main!(run);