gh-hex 0.0.0

A project to create more functional versions of existing hex tools
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

use pager::Pager;
use std::env;
use std::io;
use std::process;

mod chrome;
mod envman;
mod hexdump;
mod print_error;
mod strings;

static ERROR_EXIT_CODE: i32 = -1;
static PROG_NAME: &str = "hex";

fn help() {
    println!("Usage: {} <operation> [filename] [...options]", PROG_NAME);
    println!("Operations:");
    println!(" help - you're looking at it");
    println!(" version - print version");
    println!(" str <file> - extract all strings in the file");
    println!(" dump <file> - hexdump the file");
}

fn main() {
    let args: Vec<String> = env::args().collect();

    if args.len() == 1 {
        help();
        process::exit(ERROR_EXIT_CODE);
    }

    let operation = &args[1];
    match operation.as_str() {
        "help" => help(),
        "version" => println!("0.0.0"),
        "str" => {
            if args.len() < 3 {
                println!("Usage: {} str <filename>", PROG_NAME);
                process::exit(ERROR_EXIT_CODE);
            }

            let should_have_color_support = chrome::should_have_color_support();

            Pager::with_pager("less -R").setup();

            match strings::main_strings(&args[2], should_have_color_support) {
                Ok(_) => {}
                Err(e) => {
                    println!("Error: {}", e);
                    process::exit(ERROR_EXIT_CODE);
                }
            }
        }
        "dump" => {
            if args.len() < 3 {
                println!("Usage: {} dump <filename>", PROG_NAME);
                process::exit(ERROR_EXIT_CODE);
            }

            let should_have_color_support = chrome::should_have_color_support();

            if termion::is_tty(&mut io::stdout()) {
                envman::set_env("LESS", "-Ps| -offset- \\: 0 1  2 3  4 5  6 7  8 9  A B  C D  E F  | 0123456789ABCDEF |");
                // now that's what I call a hack :)
            }

            let mut pager = Pager::with_pager("less -R");
            pager.setup();

            match hexdump::main_hexdump(&args[2], should_have_color_support) {
                Ok(_) => {}
                Err(e) => {
                    println!("Error: {}", e);
                    process::exit(ERROR_EXIT_CODE);
                }
            }
        }
        _ => {
            println!("Unknown operation: {}", operation);
            help();
            process::exit(ERROR_EXIT_CODE);
        }
    }
}