elb-cli 0.4.0

Command-line utility that inspects ELF files, prints their dependencies and patches RPATH, RUNPATH and interpreter.
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

#![doc = include_str!("../README.md")]

use clap::Parser;
use std::path::PathBuf;
use std::process::ExitCode;

use elb::Elf;
use fs_err::File;

mod deps;
mod formatting;
mod logger;
mod patch;
mod relocate;
mod show;

use self::deps::*;
use self::formatting::*;
use self::logger::*;
use self::patch::*;
use self::relocate::*;
use self::show::*;

#[derive(clap::Parser)]
#[clap(version)]
struct Args {
    /// Verbose output.
    #[clap(short = 'v', long = "verbose")]
    verbose: bool,

    #[clap(flatten)]
    common: CommonArgs,

    #[command(subcommand)]
    command: Command,
}

#[derive(clap::Subcommand)]
enum Command {
    /// Show file contents.
    Show(ShowArgs),
    /// Validate the file.
    Check {
        /// ELF file.
        #[clap(value_name = "ELF file")]
        file: PathBuf,
    },
    /// Print dependencies.
    Deps(DepsArgs),
    /// Relocate ELF with all of its dependencies.
    Relocate(RelocateArgs),
    /// Modify ELF file.
    Patch(PatchArgs),
}

#[derive(clap::Args)]
pub struct CommonArgs {
    /// Memory page size.
    #[clap(long = "page-size", value_name = "NUM", default_value_t = 4096)]
    page_size: u64,
}

fn main() -> ExitCode {
    match do_main() {
        Ok(_) => ExitCode::SUCCESS,
        Err(e) => {
            eprintln!("{e}");
            ExitCode::FAILURE
        }
    }
}

fn do_main() -> Result<(), Box<dyn std::error::Error>> {
    let args = Args::parse();
    Logger::init(args.verbose)?;
    match args.command {
        Command::Show(show_args) => show(args.common, show_args),
        Command::Check { file } => check(args.common, file),
        Command::Deps(deps_args) => deps(args.common, deps_args),
        Command::Relocate(relocate_args) => relocate(args.common, relocate_args),
        Command::Patch(patch_args) => patch(args.common, patch_args),
    }
}

fn check(common: CommonArgs, file: PathBuf) -> Result<(), Box<dyn std::error::Error>> {
    let mut file = File::open(&file)?;
    let _elf = Elf::read(&mut file, common.page_size)?;
    Ok(())
}