mirror of
https://github.com/encounter/decomp-toolkit.git
synced 2025-08-06 04:05:37 +00:00
Luke Street
2e524e6806
This allows handling path conversions in a more structured way, as well as avoiding needless UTF-8 checks. All argument inputs use `Utf8NativePathBuf`, while all config entries use `Utf8UnixPathBuf`, ensuring that we deserialize/serialize using forward slashes. We can omit `.display()` and lossy UTF-8 conversions since all paths are known valid UTF-8.
92 lines
2.2 KiB
Rust
92 lines
2.2 KiB
Rust
use std::io::{stdout, Write};
|
|
|
|
use anyhow::Result;
|
|
use argp::FromArgs;
|
|
use typed_path::Utf8NativePathBuf;
|
|
|
|
use crate::{
|
|
cmd,
|
|
util::{
|
|
alf::AlfFile,
|
|
file::buf_writer,
|
|
path::native_path,
|
|
reader::{Endian, FromReader},
|
|
},
|
|
vfs::open_file,
|
|
};
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
/// Commands for processing NVIDIA Shield TV alf files.
|
|
#[argp(subcommand, name = "alf")]
|
|
pub struct Args {
|
|
#[argp(subcommand)]
|
|
command: SubCommand,
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
#[argp(subcommand)]
|
|
enum SubCommand {
|
|
Info(InfoArgs),
|
|
Hashes(HashesArgs),
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
/// Prints information about an alf file. (Same as `dol info`)
|
|
#[argp(subcommand, name = "info")]
|
|
pub struct InfoArgs {
|
|
#[argp(positional, from_str_fn(native_path))]
|
|
/// alf file
|
|
file: Utf8NativePathBuf,
|
|
}
|
|
|
|
#[derive(FromArgs, PartialEq, Debug)]
|
|
/// Extracts symbol hashes from an alf file.
|
|
#[argp(subcommand, name = "hashes")]
|
|
pub struct HashesArgs {
|
|
#[argp(positional, from_str_fn(native_path))]
|
|
/// alf file
|
|
alf_file: Utf8NativePathBuf,
|
|
#[argp(positional, from_str_fn(native_path))]
|
|
/// output file
|
|
output: Option<Utf8NativePathBuf>,
|
|
}
|
|
|
|
pub fn run(args: Args) -> Result<()> {
|
|
match args.command {
|
|
SubCommand::Info(c_args) => info(c_args),
|
|
SubCommand::Hashes(c_args) => hashes(c_args),
|
|
}
|
|
}
|
|
|
|
fn hashes(args: HashesArgs) -> Result<()> {
|
|
let alf_file = {
|
|
let mut file = open_file(&args.alf_file, true)?;
|
|
AlfFile::from_reader(file.as_mut(), Endian::Little)?
|
|
};
|
|
let mut w: Box<dyn Write> = if let Some(output) = &args.output {
|
|
Box::new(buf_writer(output)?)
|
|
} else {
|
|
Box::new(stdout())
|
|
};
|
|
let mut symbols = alf_file.symbols.clone();
|
|
symbols.sort_by_key(|s| s.address);
|
|
for symbol in symbols {
|
|
writeln!(
|
|
w,
|
|
"{:#010X} | {} | {:?} | {} | {} | {:#X}",
|
|
symbol.address,
|
|
symbol.section,
|
|
symbol.kind,
|
|
symbol.name,
|
|
symbol.demangled_name,
|
|
symbol.size
|
|
)?;
|
|
}
|
|
w.flush()?;
|
|
Ok(())
|
|
}
|
|
|
|
fn info(args: InfoArgs) -> Result<()> {
|
|
cmd::dol::info(cmd::dol::InfoArgs { dol_file: args.file, selfile: None })
|
|
}
|