rascul
5 years ago
parent
23a5c25726
commit
18f900005b
@ -1,228 +0,0 @@
|
|||||||
use std::fs::{File, Permissions};
|
|
||||||
use std::os::unix::fs::PermissionsExt;
|
|
||||||
use std::path::PathBuf;
|
|
||||||
use std::process::Command;
|
|
||||||
|
|
||||||
use regex::Regex;
|
|
||||||
|
|
||||||
use crate::config::Config;
|
|
||||||
use crate::error::*;
|
|
||||||
use crate::util::{copy_file, set_perms};
|
|
||||||
|
|
||||||
pub struct Files {
|
|
||||||
pub bins: Vec<PathBuf>,
|
|
||||||
pub sbins: Vec<PathBuf>,
|
|
||||||
pub libs: Vec<PathBuf>,
|
|
||||||
pub lib64s: Vec<PathBuf>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Files {
|
|
||||||
pub fn gather(config: &Config) -> MkrootResult<Files> {
|
|
||||||
let mut myfiles = Files {
|
|
||||||
bins: Vec::new(),
|
|
||||||
sbins: Vec::new(),
|
|
||||||
libs: Vec::new(),
|
|
||||||
lib64s: Vec::new(),
|
|
||||||
};
|
|
||||||
|
|
||||||
for file in &config.files {
|
|
||||||
if config.verbose {
|
|
||||||
println!("Checking {}", &file.display());
|
|
||||||
}
|
|
||||||
|
|
||||||
open(&file)?;
|
|
||||||
|
|
||||||
let (libs, lib64s) = libs_from_ldd(&file, &config)?;
|
|
||||||
myfiles.libs.extend(libs);
|
|
||||||
myfiles.lib64s.extend(lib64s);
|
|
||||||
|
|
||||||
if check_sbin(&file) {
|
|
||||||
myfiles.sbins.push(file.to_owned());
|
|
||||||
} else {
|
|
||||||
myfiles.bins.push(file.to_owned());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
myfiles.bins.sort();
|
|
||||||
myfiles.bins.dedup();
|
|
||||||
myfiles.sbins.sort();
|
|
||||||
myfiles.sbins.dedup();
|
|
||||||
myfiles.libs.sort();
|
|
||||||
myfiles.libs.dedup();
|
|
||||||
myfiles.lib64s.sort();
|
|
||||||
myfiles.lib64s.dedup();
|
|
||||||
|
|
||||||
Ok(myfiles)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn copy(&self, config: &Config) -> MkrootResult<()> {
|
|
||||||
let mut target = PathBuf::from(&config.root_dir);
|
|
||||||
target.push("bin");
|
|
||||||
copy_files(&self.bins, &target, 0o755, config.verbose)?;
|
|
||||||
|
|
||||||
let mut target = PathBuf::from(&config.root_dir);
|
|
||||||
target.push("sbin");
|
|
||||||
copy_files(&self.sbins, &target, 0o755, config.verbose)?;
|
|
||||||
|
|
||||||
let mut target = PathBuf::from(&config.root_dir);
|
|
||||||
target.push("libs");
|
|
||||||
copy_files(&self.libs, &target, 0o644, config.verbose)?;
|
|
||||||
|
|
||||||
let mut target = PathBuf::from(&config.root_dir);
|
|
||||||
target.push("lib64");
|
|
||||||
copy_files(&self.lib64s, &target, 0o644, config.verbose)?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn set_linker_permissions(
|
|
||||||
libs: &[PathBuf],
|
|
||||||
dir: &PathBuf,
|
|
||||||
verbose: bool,
|
|
||||||
) -> MkrootResult<()> {
|
|
||||||
for lib in libs {
|
|
||||||
if let Some(fn_osstr) = &lib.file_name() {
|
|
||||||
if let Some(fn_str) = fn_osstr.to_str() {
|
|
||||||
if fn_str.starts_with("ld-linux") {
|
|
||||||
let mut lib = PathBuf::from(&dir);
|
|
||||||
lib.push(&fn_str);
|
|
||||||
|
|
||||||
if verbose {
|
|
||||||
println!("Setting linker {} to mode 0o755", &lib.display());
|
|
||||||
}
|
|
||||||
|
|
||||||
set_perms(&lib, Permissions::from_mode(0o755))?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn open(p: &PathBuf) -> MkrootResult<()> {
|
|
||||||
if let Err(e) = File::open(&p) {
|
|
||||||
return Err(MkrootError::from(format!(
|
|
||||||
"Error opening file ({}): {}",
|
|
||||||
&p.display(),
|
|
||||||
e
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn libs_from_ldd(
|
|
||||||
file: &PathBuf,
|
|
||||||
config: &Config,
|
|
||||||
) -> MkrootResult<(Vec<PathBuf>, Vec<PathBuf>)> {
|
|
||||||
let ldd = ldd(&file, &config)?;
|
|
||||||
|
|
||||||
let mut libs: Vec<PathBuf> = Vec::new();
|
|
||||||
let mut lib64s: Vec<PathBuf> = Vec::new();
|
|
||||||
|
|
||||||
if ldd.is_empty() {
|
|
||||||
return Ok((libs, lib64s));
|
|
||||||
}
|
|
||||||
|
|
||||||
let re = Regex::new(r"(^|.* )(?P<path>/.*) \(0x[[:xdigit:]]{16}\)$")?;
|
|
||||||
|
|
||||||
for line in ldd.lines() {
|
|
||||||
let line = String::from(line.trim());
|
|
||||||
|
|
||||||
if let Some(caps) = re.captures(&line) {
|
|
||||||
if let Some(rematch) = caps.name("path") {
|
|
||||||
let match_path = PathBuf::from(rematch.as_str());
|
|
||||||
|
|
||||||
if config.verbose {
|
|
||||||
println!("Adding {}", &match_path.display());
|
|
||||||
}
|
|
||||||
|
|
||||||
if check_lib64(&match_path) {
|
|
||||||
lib64s.push(match_path);
|
|
||||||
} else {
|
|
||||||
libs.push(match_path);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok((libs, lib64s))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn copy_files(
|
|
||||||
files: &[PathBuf],
|
|
||||||
target: &PathBuf,
|
|
||||||
mode: u32,
|
|
||||||
verbose: bool,
|
|
||||||
) -> MkrootResult<()> {
|
|
||||||
for f in files {
|
|
||||||
let mut t = PathBuf::from(&target);
|
|
||||||
if let Some(filename) = &f.file_name() {
|
|
||||||
t.push(filename);
|
|
||||||
|
|
||||||
if verbose {
|
|
||||||
println!("Copying {} to {}", &f.display(), &t.display());
|
|
||||||
}
|
|
||||||
|
|
||||||
copy_file(&f, &t)?;
|
|
||||||
set_perms(&t, Permissions::from_mode(mode))?;
|
|
||||||
} else if verbose {
|
|
||||||
println!("Skipping {}", &f.display());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn ldd(file: &PathBuf, config: &Config) -> MkrootResult<String> {
|
|
||||||
match Command::new(&config.ldd).arg(file).output() {
|
|
||||||
Ok(output) => {
|
|
||||||
if output.status.success() {
|
|
||||||
Ok(String::from_utf8(output.stdout).unwrap_or_default())
|
|
||||||
} else if config.verbose {
|
|
||||||
let mut out = String::from("ldd failed: ");
|
|
||||||
let stdout = String::from_utf8(output.stdout).unwrap_or_default();
|
|
||||||
let stderr = String::from_utf8(output.stderr).unwrap_or_default();
|
|
||||||
|
|
||||||
if !stdout.is_empty() {
|
|
||||||
out = out + "stdout: " + &stdout + " ";
|
|
||||||
}
|
|
||||||
if !stderr.is_empty() {
|
|
||||||
out = out + "stderr: " + &stderr;
|
|
||||||
}
|
|
||||||
|
|
||||||
println!("{}", out.trim());
|
|
||||||
|
|
||||||
Ok(String::new())
|
|
||||||
} else {
|
|
||||||
Ok(String::new())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(e) => Err(MkrootError::from(format!(
|
|
||||||
"Error running ldd ({}): {}",
|
|
||||||
&config.ldd.display(),
|
|
||||||
e
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn check_lib64(path: &PathBuf) -> bool {
|
|
||||||
for c in path.components() {
|
|
||||||
if c.as_os_str() == "lib64" {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
fn check_sbin(path: &PathBuf) -> bool {
|
|
||||||
for c in path.components() {
|
|
||||||
if c.as_os_str() == "sbin" {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
false
|
|
||||||
}
|
|
||||||
@ -1,3 +0,0 @@
|
|||||||
mod mkroot_files;
|
|
||||||
|
|
||||||
pub use mkroot_files::*;
|
|
||||||
Loading…
Reference in new issue