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

extern crate boxxy;

use std::env;
use std::fs::File;
use boxxy::{Shell, Toolbox, Interface};

/* Rust doesn't directly expose __attribute__((constructor)), but this
 * is how GNU implements it.
 * Props to https://github.com/geofft/redhook */
#[link_section=".init_array"]
pub static INITIALIZE_CTOR: extern fn() = ::initialize;

/*
// alternative hook
#[link_section=".ctors"]
#[no_mangle]
pub static CALL_BOXXY: extern fn() = ::initialize;
*/

extern fn initialize() {
    env::remove_var("LD_PRELOAD");

    if let Ok(cmd) = env::var("AUTOBOXXY") {
        let toolbox = Toolbox::new();
        let mut shell = Shell::new(toolbox);
        if let Ok(target) = env::var("AUTOBOXXY_OUTPUT") {
            let output = File::create(target).unwrap();
            shell.hotswap(Interface::file(None, Some(output)));
        }
        shell.exec_once(&cmd);
        std::process::exit(0);
    }
}