kernelpanic/rsfus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Support custom installer disk number.

Browse source
This commit is contained in:
kernelpanic 2023-09-24 22:58:26 -04:00
parent 8ef6666613
commit ebe2ac98f8
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/main.rs
View file

@ -7,13 +7,18 @@ fn main() {
let args: Vec<String> = env::args().collect(); let args: Vec<String> = env::args().collect();
let iso_mount_dir = Path::new(args.get(1) let iso_mount_dir = Path::new(args.get(1)
.expect("Arguments: rsfus (path to XP ISO mount)")) .expect("Arguments: rsfus (path to XP ISO mount) [disk number]"))
.canonicalize().expect("Can't canonicalize path to XP ISO mount."); .canonicalize().expect("Can't canonicalize path to XP ISO mount.");
if !iso_mount_dir.exists() || !iso_mount_dir.is_dir() { if !iso_mount_dir.exists() || !iso_mount_dir.is_dir() {
panic!("XP ISO mount is not a directory."); panic!("XP ISO mount is not a directory.");
} }
let disk_number = if let Some(i) = args.get(2) {
i.parse::<u8>().expect("Not a valid number for disk number!")
} else { 1 };
let setup_source_line = format!("SetupSourceDevice = \"\\device\\harddisk{disk_number}\\partition1\"");
let amd64_basedir = iso_mount_dir.join("amd64"); let amd64_basedir = iso_mount_dir.join("amd64");
let i386_basedir = iso_mount_dir.join("i386"); let i386_basedir = iso_mount_dir.join("i386");
@ -47,8 +52,8 @@ fn main() {
for line in txtsetup_contents.lines() { for line in txtsetup_contents.lines() {
txtsetup_lines.push(line); txtsetup_lines.push(line);
if line == "[SetupData]" { if line == "[SetupData]" {
println!("TXTSETP: Found [SetupData]"); println!("TXTSETP: Found [SetupData], writing \"{setup_source_line}\"");
txtsetup_lines.push(r#"SetupSourceDevice = "\device\harddisk1\partition1""#); txtsetup_lines.push(setup_source_line.as_str());
} }
} }
println!("Writing {txtsetup_copy:?}"); println!("Writing {txtsetup_copy:?}");
@ -83,8 +88,8 @@ fn main() {
for index in 0..(bootmgr_contents.len() - win_nt_bt.len()) { for index in 0..(bootmgr_contents.len() - win_nt_bt.len()) {
if &bootmgr_contents[index..index + rdisk0.len()] == rdisk0 { if &bootmgr_contents[index..index + rdisk0.len()] == rdisk0 {
println!("BOOTMGR: {index:#010x} Patching rdisk(0) --> rdisk(1)"); println!("BOOTMGR: {index:#010x} Patching rdisk(0) --> rdisk({disk_number})");
bootmgr_contents[index + 6] = 0x31; bootmgr_contents[index + 6] = 0x30 + disk_number;
} }
if &bootmgr_contents[index..index + win_nt_bt.len()] == win_nt_bt if &bootmgr_contents[index..index + win_nt_bt.len()] == win_nt_bt