editres_cli 0.1.6

Command line tool for injecting data into executables that use editres
Documentation 
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

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */
#![doc = include_str!("../README.md")]
#![doc(hidden)]

use std::{
    fs::{self, OpenOptions},
    io::{self, Write as _},
    path::{Path, PathBuf},
};

use editres::injector::{inject as base_inject, list as base_list};

pub fn list<P: AsRef<Path>>(path: P) -> anyhow::Result<Vec<(String, bool)>> {
    let content = fs::read(path)?;
    base_list(&content)
}

pub fn write_executable(path: &mut PathBuf, content: &[u8]) -> io::Result<()> {
    let mut open_options = OpenOptions::new();
    open_options.create(true).write(true);
    #[cfg(unix)]
    {
        std::os::unix::fs::OpenOptionsExt::mode(&mut open_options, 0o755);
    }
    #[cfg(windows)]
    {
        path.set_extension("exe");
    }
    let mut file = open_options.open(path)?;
    file.write_all(content)?;
    file.sync_all()?;
    Ok(())
}

pub fn inject(
    path: impl AsRef<Path>,
    resource_name: &str,
    resource_data: &[u8],
    out_path: &mut PathBuf,
) -> anyhow::Result<()> {
    let mut executable_content = fs::read(path)?;
    base_inject(&mut executable_content, resource_name, resource_data)?;
    write_executable(out_path, &executable_content)?;
    Ok(())
}