Implement fs wrapper for run_make_support

This commit is contained in:
Oneirical 2024-06-06 20:47:00 +02:00
parent 20ba13c38e
commit c84afee898
36 changed files with 211 additions and 108 deletions

View File

@ -0,0 +1,113 @@
use std::fs;
use std::path::Path;
/// A wrapper around [`std::fs::remove_file`] which includes the file path in the panic message..
#[track_caller]
pub fn remove_file<P: AsRef<Path>>(path: P) {
fs::remove_file(path.as_ref())
.expect(&format!("the file in path \"{}\" could not be removed", path.as_ref().display()));
}
/// A wrapper around [`std::fs::copy`] which includes the file path in the panic message.
#[track_caller]
pub fn copy<P: AsRef<Path>, Q: AsRef<Path>>(from: P, to: Q) {
fs::copy(from.as_ref(), to.as_ref()).expect(&format!(
"the file \"{}\" could not be copied over to \"{}\"",
from.as_ref().display(),
to.as_ref().display(),
));
}
/// A wrapper around [`std::fs::File::create`] which includes the file path in the panic message..
#[track_caller]
pub fn create_file<P: AsRef<Path>>(path: P) {
fs::File::create(path.as_ref())
.expect(&format!("the file in path \"{}\" could not be created", path.as_ref().display()));
}
/// A wrapper around [`std::fs::read`] which includes the file path in the panic message..
#[track_caller]
pub fn read<P: AsRef<Path>>(path: P) -> Vec<u8> {
fs::read(path.as_ref())
.expect(&format!("the file in path \"{}\" could not be read", path.as_ref().display()))
}
/// A wrapper around [`std::fs::read_to_string`] which includes the file path in the panic message..
#[track_caller]
pub fn read_to_string<P: AsRef<Path>>(path: P) -> String {
fs::read_to_string(path.as_ref()).expect(&format!(
"the file in path \"{}\" could not be read into a String",
path.as_ref().display()
))
}
/// A wrapper around [`std::fs::read_dir`] which includes the file path in the panic message..
#[track_caller]
pub fn read_dir<P: AsRef<Path>>(path: P) -> fs::ReadDir {
fs::read_dir(path.as_ref())
.expect(&format!("the directory in path \"{}\" could not be read", path.as_ref().display()))
}
/// A wrapper around [`std::fs::write`] which includes the file path in the panic message..
#[track_caller]
pub fn write<P: AsRef<Path>, C: AsRef<[u8]>>(path: P, contents: C) {
fs::write(path.as_ref(), contents.as_ref()).expect(&format!(
"the file in path \"{}\" could not be written to",
path.as_ref().display()
));
}
/// A wrapper around [`std::fs::remove_dir_all`] which includes the file path in the panic message..
#[track_caller]
pub fn remove_dir_all<P: AsRef<Path>>(path: P) {
fs::remove_dir_all(path.as_ref()).expect(&format!(
"the directory in path \"{}\" could not be removed alongside all its contents",
path.as_ref().display(),
));
}
/// A wrapper around [`std::fs::create_dir`] which includes the file path in the panic message..
#[track_caller]
pub fn create_dir<P: AsRef<Path>>(path: P) {
fs::create_dir(path.as_ref()).expect(&format!(
"the directory in path \"{}\" could not be created",
path.as_ref().display()
));
}
/// A wrapper around [`std::fs::create_dir_all`] which includes the file path in the panic message..
#[track_caller]
pub fn create_dir_all<P: AsRef<Path>>(path: P) {
fs::create_dir_all(path.as_ref()).expect(&format!(
"the directory (and all its parents) in path \"{}\" could not be created",
path.as_ref().display()
));
}
/// A wrapper around [`std::fs::metadata`] which includes the file path in the panic message..
#[track_caller]
pub fn metadata<P: AsRef<Path>>(path: P) -> fs::Metadata {
fs::metadata(path.as_ref()).expect(&format!(
"the file's metadata in path \"{}\" could not be read",
path.as_ref().display()
))
}
/// A wrapper around [`std::fs::rename`] which includes the file path in the panic message.
#[track_caller]
pub fn rename<P: AsRef<Path>, Q: AsRef<Path>>(from: P, to: Q) {
fs::rename(from.as_ref(), to.as_ref()).expect(&format!(
"the file \"{}\" could not be moved over to \"{}\"",
from.as_ref().display(),
to.as_ref().display(),
));
}
/// A wrapper around [`std::fs::set_permissions`] which includes the file path in the panic message.
#[track_caller]
pub fn set_permissions<P: AsRef<Path>>(path: P, perm: fs::Permissions) {
fs::set_permissions(path.as_ref(), perm).expect(&format!(
"the file's permissions in path \"{}\" could not be changed",
path.as_ref().display()
));
}

View File

@ -8,6 +8,7 @@ pub mod clang;
mod command; mod command;
pub mod diff; pub mod diff;
mod drop_bomb; mod drop_bomb;
pub mod fs_wrapper;
pub mod llvm_readobj; pub mod llvm_readobj;
pub mod run; pub mod run;
pub mod rustc; pub mod rustc;
@ -148,7 +149,7 @@ pub fn dynamic_lib_extension() -> &'static str {
} }
} }
/// Construct a rust library (rlib) name. /// Generate the name a rust library (rlib) would have.
pub fn rust_lib_name(name: &str) -> String { pub fn rust_lib_name(name: &str) -> String {
format!("lib{name}.rlib") format!("lib{name}.rlib")
} }
@ -223,15 +224,15 @@ pub fn copy_dir_all(src: impl AsRef<Path>, dst: impl AsRef<Path>) {
fn copy_dir_all_inner(src: impl AsRef<Path>, dst: impl AsRef<Path>) -> io::Result<()> { fn copy_dir_all_inner(src: impl AsRef<Path>, dst: impl AsRef<Path>) -> io::Result<()> {
let dst = dst.as_ref(); let dst = dst.as_ref();
if !dst.is_dir() { if !dst.is_dir() {
fs::create_dir_all(&dst)?; std::fs::create_dir_all(&dst)?;
} }
for entry in fs::read_dir(src)? { for entry in std::fs::read_dir(src)? {
let entry = entry?; let entry = entry?;
let ty = entry.file_type()?; let ty = entry.file_type()?;
if ty.is_dir() { if ty.is_dir() {
copy_dir_all_inner(entry.path(), dst.join(entry.file_name()))?; copy_dir_all_inner(entry.path(), dst.join(entry.file_name()))?;
} else { } else {
fs::copy(entry.path(), dst.join(entry.file_name()))?; std::fs::copy(entry.path(), dst.join(entry.file_name()))?;
} }
} }
Ok(()) Ok(())
@ -250,13 +251,6 @@ pub fn copy_dir_all(src: impl AsRef<Path>, dst: impl AsRef<Path>) {
/// Check that all files in `dir1` exist and have the same content in `dir2`. Panic otherwise. /// Check that all files in `dir1` exist and have the same content in `dir2`. Panic otherwise.
pub fn recursive_diff(dir1: impl AsRef<Path>, dir2: impl AsRef<Path>) { pub fn recursive_diff(dir1: impl AsRef<Path>, dir2: impl AsRef<Path>) {
fn read_file(path: &Path) -> Vec<u8> {
match fs::read(path) {
Ok(c) => c,
Err(e) => panic!("Failed to read `{}`: {:?}", path.display(), e),
}
}
let dir2 = dir2.as_ref(); let dir2 = dir2.as_ref();
read_dir(dir1, |entry_path| { read_dir(dir1, |entry_path| {
let entry_name = entry_path.file_name().unwrap(); let entry_name = entry_path.file_name().unwrap();
@ -264,8 +258,8 @@ pub fn recursive_diff(dir1: impl AsRef<Path>, dir2: impl AsRef<Path>) {
recursive_diff(&entry_path, &dir2.join(entry_name)); recursive_diff(&entry_path, &dir2.join(entry_name));
} else { } else {
let path2 = dir2.join(entry_name); let path2 = dir2.join(entry_name);
let file1 = read_file(&entry_path); let file1 = fs_wrapper::read(&entry_path);
let file2 = read_file(&path2); let file2 = fs_wrapper::read(&path2);
// We don't use `assert_eq!` because they are `Vec<u8>`, so not great for display. // We don't use `assert_eq!` because they are `Vec<u8>`, so not great for display.
// Why not using String? Because there might be minified files or even potentially // Why not using String? Because there might be minified files or even potentially
@ -281,7 +275,7 @@ pub fn recursive_diff(dir1: impl AsRef<Path>, dir2: impl AsRef<Path>) {
} }
pub fn read_dir<F: Fn(&Path)>(dir: impl AsRef<Path>, callback: F) { pub fn read_dir<F: Fn(&Path)>(dir: impl AsRef<Path>, callback: F) {
for entry in fs::read_dir(dir).unwrap() { for entry in fs_wrapper::read_dir(dir) {
callback(&entry.unwrap().path()); callback(&entry.unwrap().path());
} }
} }

View File

@ -5,7 +5,7 @@
use std::path::PathBuf; use std::path::PathBuf;
use run_make_support::{aux_build, rustc, source_root}; use run_make_support::{aux_build, fs_wrapper, rustc, source_root};
fn main() { fn main() {
aux_build().input("stable.rs").emit("metadata").run(); aux_build().input("stable.rs").emit("metadata").run();
@ -13,7 +13,7 @@ fn main() {
let output = let output =
rustc().input("main.rs").emit("metadata").extern_("stable", "libstable.rmeta").run(); rustc().input("main.rs").emit("metadata").extern_("stable", "libstable.rmeta").run();
let version = std::fs::read_to_string(source_root().join("src/version")).unwrap(); let version = fs_wrapper::read_to_string(source_root().join("src/version"));
let expected_string = format!("stable since {}", version.trim()); let expected_string = format!("stable since {}", version.trim());
output.assert_stderr_contains(expected_string); output.assert_stderr_contains(expected_string);
} }

View File

@ -3,9 +3,9 @@
//@ ignore-cross-compile //@ ignore-cross-compile
use std::fs::remove_file; use run_make_support::{
cc, cwd, dynamic_lib_extension, fs_wrapper, is_msvc, read_dir, run, run_fail, rustc,
use run_make_support::{cc, cwd, dynamic_lib_extension, is_msvc, read_dir, run, run_fail, rustc}; };
fn main() { fn main() {
rustc().input("foo.rs").run(); rustc().input("foo.rs").run();
@ -28,7 +28,7 @@ fn main() {
name.ends_with(".so") || name.ends_with(".dll") || name.ends_with(".dylib") name.ends_with(".so") || name.ends_with(".dll") || name.ends_with(".dylib")
}) })
{ {
remove_file(path).unwrap(); fs_wrapper::remove_file(path);
} }
}); });
run_fail("bar"); run_fail("bar");

View File

@ -3,6 +3,7 @@
//@ ignore-cross-compile //@ ignore-cross-compile
use run_make_support::fs_wrapper::remove_file;
use run_make_support::{cc, extra_c_flags, run, rustc, static_lib_name}; use run_make_support::{cc, extra_c_flags, run, rustc, static_lib_name};
use std::fs; use std::fs;
@ -10,6 +11,6 @@ fn main() {
rustc().input("foo.rs").run(); rustc().input("foo.rs").run();
cc().input("bar.c").input(static_lib_name("foo")).out_exe("bar").args(&extra_c_flags()).run(); cc().input("bar.c").input(static_lib_name("foo")).out_exe("bar").args(&extra_c_flags()).run();
run("bar"); run("bar");
fs::remove_file(static_lib_name("foo")); remove_file(static_lib_name("foo"));
run("bar"); run("bar");
} }

View File

@ -10,9 +10,7 @@
//@ ignore-cross-compile //@ ignore-cross-compile
use std::fs::remove_file; use run_make_support::{cc, cwd, dynamic_lib_name, fs_wrapper, is_msvc, run, rustc};
use run_make_support::{cc, cwd, dynamic_lib_name, is_msvc, run, rustc};
fn main() { fn main() {
rustc().input("bar.rs").run(); rustc().input("bar.rs").run();
@ -25,7 +23,7 @@ fn main() {
} }
run("foo"); run("foo");
remove_file(dynamic_lib_name("foo")).unwrap(); fs_wrapper::remove_file(dynamic_lib_name("foo"));
rustc().input("foo.rs").arg("-Clto").run(); rustc().input("foo.rs").arg("-Clto").run();
run("foo"); run("foo");

View File

@ -14,6 +14,7 @@
#![deny(warnings)] #![deny(warnings)]
use run_make_support::fs_wrapper::{read, read_dir};
use run_make_support::object::read::archive::ArchiveFile; use run_make_support::object::read::archive::ArchiveFile;
use run_make_support::object::read::Object; use run_make_support::object::read::Object;
use run_make_support::object::ObjectSection; use run_make_support::object::ObjectSection;
@ -55,8 +56,7 @@ fn main() {
cmd.run(); cmd.run();
let rlibs_path = target_dir.join(target).join("debug").join("deps"); let rlibs_path = target_dir.join(target).join("debug").join("deps");
let compiler_builtins_rlib = std::fs::read_dir(rlibs_path) let compiler_builtins_rlib = read_dir(rlibs_path)
.unwrap()
.find_map(|e| { .find_map(|e| {
let path = e.unwrap().path(); let path = e.unwrap().path();
let file_name = path.file_name().unwrap().to_str().unwrap(); let file_name = path.file_name().unwrap().to_str().unwrap();
@ -70,7 +70,7 @@ fn main() {
// rlib files are archives, where the archive members each a CGU, and we also have one called // rlib files are archives, where the archive members each a CGU, and we also have one called
// lib.rmeta which is the encoded metadata. Each of the CGUs is an object file. // lib.rmeta which is the encoded metadata. Each of the CGUs is an object file.
let data = std::fs::read(compiler_builtins_rlib).unwrap(); let data = read(compiler_builtins_rlib);
let mut defined_symbols = HashSet::new(); let mut defined_symbols = HashSet::new();
let mut undefined_relocations = HashSet::new(); let mut undefined_relocations = HashSet::new();

View File

@ -1,13 +1,11 @@
// Tests that const prop lints interrupting codegen don't leave `.o` files around. // Tests that const prop lints interrupting codegen don't leave `.o` files around.
use std::fs; use run_make_support::{cwd, fs_wrapper, rustc};
use run_make_support::{cwd, rustc};
fn main() { fn main() {
rustc().input("input.rs").run_fail().assert_exit_code(1); rustc().input("input.rs").run_fail().assert_exit_code(1);
for entry in fs::read_dir(cwd()).unwrap() { for entry in fs_wrapper::read_dir(cwd()) {
let entry = entry.unwrap(); let entry = entry.unwrap();
let path = entry.path(); let path = entry.path();

View File

@ -1,13 +1,13 @@
// Check that valid binaries are persisted by running them, regardless of whether the // Check that valid binaries are persisted by running them, regardless of whether the
// --run or --no-run option is used. // --run or --no-run option is used.
use run_make_support::fs_wrapper::{create_dir, remove_dir_all};
use run_make_support::{run, rustc, rustdoc}; use run_make_support::{run, rustc, rustdoc};
use std::fs::{create_dir, remove_dir_all};
use std::path::Path; use std::path::Path;
fn setup_test_env<F: FnOnce(&Path, &Path)>(callback: F) { fn setup_test_env<F: FnOnce(&Path, &Path)>(callback: F) {
let out_dir = Path::new("doctests"); let out_dir = Path::new("doctests");
create_dir(&out_dir).expect("failed to create doctests folder"); create_dir(&out_dir);
rustc().input("t.rs").crate_type("rlib").run(); rustc().input("t.rs").crate_type("rlib").run();
callback(&out_dir, Path::new("libt.rlib")); callback(&out_dir, Path::new("libt.rlib"));
remove_dir_all(out_dir); remove_dir_all(out_dir);
@ -43,9 +43,9 @@ fn main() {
check_generated_binaries(); check_generated_binaries();
}); });
// Behavior with --test-run-directory with relative paths. // Behavior with --test-run-directory with relative paths.
setup_test_env(|_out_dir, extern_path| { setup_test_env(|_out_dir, _extern_path| {
let run_dir_path = Path::new("rundir"); let run_dir_path = Path::new("rundir");
create_dir(&run_dir_path).expect("failed to create rundir folder"); create_dir(&run_dir_path);
rustdoc() rustdoc()
.input("t.rs") .input("t.rs")

View File

@ -1,12 +1,12 @@
// Tests behavior of rustdoc `--runtool`. // Tests behavior of rustdoc `--runtool`.
use run_make_support::fs_wrapper::{create_dir, remove_dir_all};
use run_make_support::{rustc, rustdoc}; use run_make_support::{rustc, rustdoc};
use std::fs::{create_dir, remove_dir_all};
use std::path::PathBuf; use std::path::PathBuf;
fn mkdir(name: &str) -> PathBuf { fn mkdir(name: &str) -> PathBuf {
let dir = PathBuf::from(name); let dir = PathBuf::from(name);
create_dir(&dir).expect("failed to create doctests folder"); create_dir(&dir);
dir dir
} }

View File

@ -1,7 +1,6 @@
use std::fs::create_dir;
use std::path::Path; use std::path::Path;
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
fn emit_and_check(out_dir: &Path, out_file: &str, format: &str) { fn emit_and_check(out_dir: &Path, out_file: &str, format: &str) {
let out_file = out_dir.join(out_file); let out_file = out_dir.join(out_file);
@ -12,7 +11,7 @@ fn emit_and_check(out_dir: &Path, out_file: &str, format: &str) {
fn main() { fn main() {
let out_dir = Path::new("emit"); let out_dir = Path::new("emit");
create_dir(&out_dir).unwrap(); fs_wrapper::create_dir(&out_dir);
emit_and_check(&out_dir, "libfoo.s", "asm"); emit_and_check(&out_dir, "libfoo.s", "asm");
emit_and_check(&out_dir, "libfoo.bc", "llvm-bc"); emit_and_check(&out_dir, "libfoo.bc", "llvm-bc");

View File

@ -13,15 +13,14 @@
//@ ignore-nvptx64-nvidia-cuda //@ ignore-nvptx64-nvidia-cuda
// FIXME: can't find crate for `std` // FIXME: can't find crate for `std`
use run_make_support::fs_wrapper as fs;
use run_make_support::rustc; use run_make_support::rustc;
use std::fs;
fn main() { fn main() {
// FIXME(Oneirical): Use run_make_support::fs_wrapper here. fs::create_dir("src");
fs::create_dir("src").unwrap(); fs::create_dir("incr");
fs::create_dir("incr").unwrap(); fs::copy("a.rs", "src/main.rs");
fs::copy("a.rs", "src/main.rs").unwrap();
rustc().incremental("incr").input("src/main.rs").run(); rustc().incremental("incr").input("src/main.rs").run();
fs::copy("b.rs", "src/main.rs").unwrap(); fs::copy("b.rs", "src/main.rs");
rustc().incremental("incr").input("src/main.rs").run(); rustc().incremental("incr").input("src/main.rs").run();
} }

View File

@ -3,8 +3,8 @@
#[cfg(unix)] #[cfg(unix)]
extern crate libc; extern crate libc;
use run_make_support::aux_build; use run_make_support::{aux_build, fs_wrapper};
use std::fs;
#[cfg(unix)] #[cfg(unix)]
use std::os::unix::fs::PermissionsExt; use std::os::unix::fs::PermissionsExt;
use std::path::Path; use std::path::Path;
@ -20,7 +20,7 @@ fn main() {
} }
fn verify(path: &Path) { fn verify(path: &Path) {
let perm = fs::metadata(path).unwrap().permissions(); let perm = fs_wrapper::metadata(path).permissions();
assert!(!perm.readonly()); assert!(!perm.readonly());

View File

@ -13,7 +13,6 @@
//@ ignore-cross-compile //@ ignore-cross-compile
use run_make_support::{run_in_tmpdir, rustc}; use run_make_support::{run_in_tmpdir, rustc};
use std::fs;
fn main() { fn main() {
run_in_tmpdir(|| { run_in_tmpdir(|| {

View File

@ -1,3 +1,4 @@
use run_make_support::fs_wrapper;
use run_make_support::rustc; use run_make_support::rustc;
fn main() { fn main() {
@ -6,6 +7,6 @@ fn main() {
#[cfg(windows)] #[cfg(windows)]
let non_unicode: std::ffi::OsString = std::os::windows::ffi::OsStringExt::from_wide(&[0xD800]); let non_unicode: std::ffi::OsString = std::os::windows::ffi::OsStringExt::from_wide(&[0xD800]);
let output = rustc().input("non_unicode_env.rs").env("NON_UNICODE_VAR", non_unicode).run_fail(); let output = rustc().input("non_unicode_env.rs").env("NON_UNICODE_VAR", non_unicode).run_fail();
let expected = std::fs::read_to_string("non_unicode_env.stderr").unwrap(); let expected = fs_wrapper::read_to_string("non_unicode_env.stderr");
output.assert_stderr_equals(expected); output.assert_stderr_equals(expected);
} }

View File

@ -1,4 +1,4 @@
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
fn main() { fn main() {
#[cfg(unix)] #[cfg(unix)]
@ -17,8 +17,8 @@ fn main() {
} }
let incr_dir = "incr-dir"; let incr_dir = "incr-dir";
rustc().input("foo.rs").incremental(&incr_dir).run(); rustc().input("foo.rs").incremental(&incr_dir).run();
for crate_dir in std::fs::read_dir(&incr_dir).unwrap() { for crate_dir in fs_wrapper::read_dir(&incr_dir) {
std::fs::create_dir(crate_dir.unwrap().path().join(&non_unicode)).unwrap(); fs_wrapper::create_dir(crate_dir.unwrap().path().join(&non_unicode));
} }
rustc().input("foo.rs").incremental(&incr_dir).run(); rustc().input("foo.rs").incremental(&incr_dir).run();
} }

View File

@ -10,7 +10,7 @@ use std::ffi::OsString;
use std::iter::FromIterator; use std::iter::FromIterator;
use std::path::PathBuf; use std::path::PathBuf;
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
struct PrintCfg { struct PrintCfg {
target: &'static str, target: &'static str,
@ -96,7 +96,7 @@ fn check(PrintCfg { target, includes, disallow }: PrintCfg) {
rustc().target(target).arg(print_arg).run(); rustc().target(target).arg(print_arg).run();
let output = std::fs::read_to_string(&tmp_path).unwrap(); let output = fs_wrapper::read_to_string(&tmp_path);
check_(&output, includes, disallow); check_(&output, includes, disallow);
} }

View File

@ -4,7 +4,7 @@
use std::ffi::OsString; use std::ffi::OsString;
use std::path::PathBuf; use std::path::PathBuf;
use run_make_support::{rustc, target}; use run_make_support::{fs_wrapper, rustc, target};
struct Option<'a> { struct Option<'a> {
target: &'a str, target: &'a str,
@ -49,7 +49,7 @@ fn check(args: Option) {
rustc().target(args.target).arg(print_arg).run(); rustc().target(args.target).arg(print_arg).run();
std::fs::read_to_string(&tmp_path).unwrap() fs_wrapper::read_to_string(&tmp_path)
}; };
check_(&stdout, args.includes); check_(&stdout, args.includes);

View File

@ -3,7 +3,7 @@
use gimli::{AttributeValue, EndianRcSlice, Reader, RunTimeEndian}; use gimli::{AttributeValue, EndianRcSlice, Reader, RunTimeEndian};
use object::{Object, ObjectSection}; use object::{Object, ObjectSection};
use run_make_support::{gimli, object, rustc}; use run_make_support::{fs_wrapper, gimli, object, rustc};
use std::collections::HashMap; use std::collections::HashMap;
use std::path::PathBuf; use std::path::PathBuf;
use std::rc::Rc; use std::rc::Rc;
@ -19,8 +19,7 @@ fn main() {
.join("DWARF") .join("DWARF")
.join("repr128"); .join("repr128");
let output = let output =
std::fs::read(if dsym_location.try_exists().unwrap() { dsym_location } else { output }) fs_wrapper::read(if dsym_location.try_exists().unwrap() { dsym_location } else { output });
.unwrap();
let obj = object::File::parse(output.as_slice()).unwrap(); let obj = object::File::parse(output.as_slice()).unwrap();
let endian = if obj.is_little_endian() { RunTimeEndian::Little } else { RunTimeEndian::Big }; let endian = if obj.is_little_endian() { RunTimeEndian::Little } else { RunTimeEndian::Big };
let dwarf = gimli::Dwarf::load(|section| -> Result<_, ()> { let dwarf = gimli::Dwarf::load(|section| -> Result<_, ()> {

View File

@ -7,8 +7,8 @@
//@ ignore-cross-compile //@ ignore-cross-compile
use run_make_support::rustc; use run_make_support::{bin_name, fs_wrapper, rustc};
use std::fs; use std::path::Path;
fn compile(output_file: &str, emit: Option<&str>) { fn compile(output_file: &str, emit: Option<&str>) {
let mut rustc = rustc(); let mut rustc = rustc();
@ -28,11 +28,15 @@ fn main() {
("link-output", Some("link")), ("link-output", Some("link")),
("obj-output", Some("obj")), ("obj-output", Some("obj")),
("dep-output", Some("dep-info")), ("dep-output", Some("dep-info")),
("multi-output", Some("asm,obj")),
]; ];
for (output_file, emit) in flags { for (output_file, emit) in flags {
fs::remove_file(output_file).unwrap_or_default(); // In the None case, bin_name is required for successful Windows compilation.
let output_file = &bin_name(output_file);
compile(output_file, emit); compile(output_file, emit);
fs::remove_file(output_file); assert!(Path::new(output_file).is_file());
} }
compile("multi-output", Some("asm,obj"));
assert!(Path::new("multi-output.s").is_file());
assert!(Path::new("multi-output.o").is_file());
} }

View File

@ -5,11 +5,12 @@
// the renamed library. // the renamed library.
// See https://github.com/rust-lang/rust/pull/49253 // See https://github.com/rust-lang/rust/pull/49253
use run_make_support::fs_wrapper;
use run_make_support::rustc; use run_make_support::rustc;
use std::fs;
fn main() { fn main() {
rustc().extra_filename("-hash").input("foo.rs").run(); rustc().extra_filename("-hash").input("foo.rs").run();
rustc().input("bar.rs").run(); rustc().input("bar.rs").run();
fs::rename("libfoo-hash.rlib", "libfoo-another-hash.rlib").unwrap(); fs_wrapper::rename("libfoo-hash.rlib", "libfoo-another-hash.rlib");
rustc().input("baz.rs").run(); rustc().input("baz.rs").run();
} }

View File

@ -1,12 +1,10 @@
use run_make_support::{htmldocck, rustc, rustdoc, source_root}; use run_make_support::{fs_wrapper, htmldocck, rustc, rustdoc, source_root};
use std::fs::read_dir;
use std::path::Path; use std::path::Path;
pub fn scrape(extra_args: &[&str]) { pub fn scrape(extra_args: &[&str]) {
let out_dir = Path::new("rustdoc"); let out_dir = Path::new("rustdoc");
let crate_name = "foobar"; let crate_name = "foobar";
let deps = read_dir("examples") let deps = fs_wrapper::read_dir("examples")
.unwrap()
.filter_map(|entry| entry.ok().map(|e| e.path())) .filter_map(|entry| entry.ok().map(|e| e.path()))
.filter(|path| path.is_file() && path.extension().is_some_and(|ext| ext == "rs")) .filter(|path| path.is_file() && path.extension().is_some_and(|ext| ext == "rs"))
.collect::<Vec<_>>(); .collect::<Vec<_>>();

View File

@ -1,10 +1,10 @@
use run_make_support::rustdoc; use run_make_support::{fs_wrapper, rustdoc};
use std::iter;
use std::path::Path; use std::path::Path;
use std::{fs, iter};
fn generate_a_lot_of_cfgs(path: &Path) { fn generate_a_lot_of_cfgs(path: &Path) {
let content = iter::repeat("--cfg=a\n").take(100_000).collect::<String>(); let content = iter::repeat("--cfg=a\n").take(100_000).collect::<String>();
fs::write(path, content.as_bytes()).expect("failed to create args file"); fs_wrapper::write(path, content.as_bytes());
} }
fn main() { fn main() {

View File

@ -1,15 +1,15 @@
// Test that rustdoc will properly load in a theme file and display it in the theme selector. // Test that rustdoc will properly load in a theme file and display it in the theme selector.
use run_make_support::{htmldocck, rustdoc, source_root}; use run_make_support::{fs_wrapper, htmldocck, rustdoc, source_root};
use std::path::Path; use std::path::Path;
fn main() { fn main() {
let out_dir = Path::new("rustdoc-themes"); let out_dir = Path::new("rustdoc-themes");
let test_css = "test.css"; let test_css = "test.css";
let no_script = let no_script = fs_wrapper::read_to_string(
std::fs::read_to_string(source_root().join("src/librustdoc/html/static/css/noscript.css")) source_root().join("src/librustdoc/html/static/css/noscript.css"),
.unwrap(); );
let mut test_content = String::new(); let mut test_content = String::new();
let mut found_begin_light = false; let mut found_begin_light = false;
@ -24,8 +24,8 @@ fn main() {
} }
} }
assert!(!test_content.is_empty()); assert!(!test_content.is_empty());
std::fs::create_dir_all(&out_dir).unwrap(); fs_wrapper::create_dir_all(&out_dir);
std::fs::write(&test_css, test_content).unwrap(); fs_wrapper::write(&test_css, test_content);
rustdoc().output(&out_dir).input("foo.rs").arg("--theme").arg(&test_css).run(); rustdoc().output(&out_dir).input("foo.rs").arg("--theme").arg(&test_css).run();
htmldocck().arg(out_dir).arg("foo.rs").run(); htmldocck().arg(out_dir).arg("foo.rs").run();

View File

@ -1,4 +1,4 @@
use std::fs::copy; use run_make_support::fs_wrapper::copy;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use run_make_support::{copy_dir_all, recursive_diff, rustdoc}; use run_make_support::{copy_dir_all, recursive_diff, rustdoc};
@ -38,7 +38,7 @@ fn main() {
assert!(out_dir.join("foobar.json").is_file()); assert!(out_dir.join("foobar.json").is_file());
// Copy first json output to check if it's exactly same after second compilation. // Copy first json output to check if it's exactly same after second compilation.
copy(out_dir.join("foobar.json"), tmp_out_dir.join("foobar.json")).unwrap(); copy(out_dir.join("foobar.json"), tmp_out_dir.join("foobar.json"));
// Generate json doc on the same output. // Generate json doc on the same output.
generate_docs(&out_dir, JsonOutput::Yes); generate_docs(&out_dir, JsonOutput::Yes);

View File

@ -1,13 +1,13 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::HashMap; use std::collections::HashMap;
fn main() { fn main() {
rustc().input("foo.rs").target("wasm32-wasip1").run(); rustc().input("foo.rs").target("wasm32-wasip1").run();
rustc().input("bar.rs").target("wasm32-wasip1").arg("-Clto").opt().run(); rustc().input("bar.rs").target("wasm32-wasip1").arg("-Clto").opt().run();
let file = std::fs::read("bar.wasm").unwrap(); let file = fs_wrapper::read("bar.wasm");
let mut custom = HashMap::new(); let mut custom = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {

View File

@ -1,18 +1,18 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::HashMap; use std::collections::HashMap;
use std::path::Path; use std::path::Path;
fn main() { fn main() {
rustc().input("foo.rs").target("wasm32-wasip1").opt().run(); rustc().input("foo.rs").target("wasm32-wasip1").opt().run();
verify(&Path::new("foo.wasm")); verify(Path::new("foo.wasm"));
} }
fn verify(path: &Path) { fn verify(path: &Path) {
eprintln!("verify {path:?}"); eprintln!("verify {path:?}");
let file = std::fs::read(&path).unwrap(); let file = fs_wrapper::read(&path);
let mut custom = HashMap::new(); let mut custom = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::HashMap; use std::collections::HashMap;
use std::path::Path; use std::path::Path;
use wasmparser::ExternalKind::*; use wasmparser::ExternalKind::*;
@ -33,7 +33,7 @@ fn test(args: &[&str]) {
fn verify_exports(path: &Path, exports: &[(&str, wasmparser::ExternalKind)]) { fn verify_exports(path: &Path, exports: &[(&str, wasmparser::ExternalKind)]) {
println!("verify {path:?}"); println!("verify {path:?}");
let file = std::fs::read(path).unwrap(); let file = fs_wrapper::read(path);
let mut wasm_exports = HashMap::new(); let mut wasm_exports = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {
let payload = payload.unwrap(); let payload = payload.unwrap();

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::HashMap; use std::collections::HashMap;
use wasmparser::TypeRef::Func; use wasmparser::TypeRef::Func;
@ -8,7 +8,7 @@ fn main() {
rustc().input("foo.rs").target("wasm32-wasip1").run(); rustc().input("foo.rs").target("wasm32-wasip1").run();
rustc().input("bar.rs").target("wasm32-wasip1").arg("-Clto").opt().run(); rustc().input("bar.rs").target("wasm32-wasip1").arg("-Clto").opt().run();
let file = std::fs::read("bar.wasm").unwrap(); let file = fs_wrapper::read("bar.wasm");
let mut imports = HashMap::new(); let mut imports = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {

View File

@ -1,7 +1,7 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
#![deny(warnings)] #![deny(warnings)]
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
fn main() { fn main() {
test("a"); test("a");
@ -15,7 +15,7 @@ fn test(cfg: &str) {
rustc().input("foo.rs").target("wasm32-wasip1").arg("-Clto").opt().cfg(cfg).run(); rustc().input("foo.rs").target("wasm32-wasip1").arg("-Clto").opt().cfg(cfg).run();
let bytes = std::fs::read("foo.wasm").unwrap(); let bytes = fs_wrapper::read("foo.wasm");
println!("{}", bytes.len()); println!("{}", bytes.len());
assert!(bytes.len() < 40_000); assert!(bytes.len() < 40_000);
} }

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::HashMap; use std::collections::HashMap;
fn main() { fn main() {
@ -13,7 +13,7 @@ fn main() {
.arg("-Copt-level=z") .arg("-Copt-level=z")
.run(); .run();
let file = std::fs::read("main.wasm").unwrap(); let file = fs_wrapper::read("main.wasm");
let mut imports = HashMap::new(); let mut imports = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {

View File

@ -1,12 +1,12 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
#![deny(warnings)] #![deny(warnings)]
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
fn main() { fn main() {
rustc().input("foo.rs").target("wasm32-wasip1").arg("-Clto").opt().run(); rustc().input("foo.rs").target("wasm32-wasip1").arg("-Clto").opt().run();
let bytes = std::fs::read("foo.wasm").unwrap(); let bytes = fs_wrapper::read("foo.wasm");
println!("{}", bytes.len()); println!("{}", bytes.len());
assert!(bytes.len() < 50_000); assert!(bytes.len() < 50_000);
} }

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::collections::{HashMap, HashSet}; use std::collections::{HashMap, HashSet};
use std::path::Path; use std::path::Path;
@ -23,7 +23,7 @@ fn test(file: &str, args: &[&str], expected_imports: &[(&str, &[&str])]) {
rustc().input(file).target("wasm32-wasip1").args(args).run(); rustc().input(file).target("wasm32-wasip1").args(args).run();
let file = std::fs::read(Path::new(file).with_extension("wasm")).unwrap(); let file = fs_wrapper::read(Path::new(file).with_extension("wasm"));
let mut imports = HashMap::new(); let mut imports = HashMap::new();
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::path::Path; use std::path::Path;
fn main() { fn main() {
@ -17,7 +17,7 @@ fn main() {
fn verify_symbols(path: &Path) { fn verify_symbols(path: &Path) {
eprintln!("verify {path:?}"); eprintln!("verify {path:?}");
let file = std::fs::read(&path).unwrap(); let file = fs_wrapper::read(&path);
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {
let payload = payload.unwrap(); let payload = payload.unwrap();

View File

@ -1,6 +1,6 @@
//@ only-wasm32-wasip1 //@ only-wasm32-wasip1
use run_make_support::{rustc, wasmparser}; use run_make_support::{fs_wrapper, rustc, wasmparser};
use std::path::Path; use std::path::Path;
fn main() { fn main() {
@ -16,7 +16,7 @@ fn main() {
fn verify_symbols(path: &Path) { fn verify_symbols(path: &Path) {
eprintln!("verify {path:?}"); eprintln!("verify {path:?}");
let file = std::fs::read(&path).unwrap(); let file = fs_wrapper::read(&path);
for payload in wasmparser::Parser::new(0).parse_all(&file) { for payload in wasmparser::Parser::new(0).parse_all(&file) {
let payload = payload.unwrap(); let payload = payload.unwrap();

View File

@ -3,8 +3,7 @@
// Tests that WS2_32.dll is not unnecessarily linked, see issue #85441 // Tests that WS2_32.dll is not unnecessarily linked, see issue #85441
use run_make_support::object::{self, read::Object}; use run_make_support::object::{self, read::Object};
use run_make_support::rustc; use run_make_support::{fs_wrapper, rustc};
use std::fs;
fn main() { fn main() {
rustc().input("empty.rs").run(); rustc().input("empty.rs").run();
@ -15,7 +14,7 @@ fn main() {
} }
fn links_ws2_32(exe: &str) -> bool { fn links_ws2_32(exe: &str) -> bool {
let binary_data = fs::read(exe).unwrap(); let binary_data = fs_wrapper::read(exe);
let file = object::File::parse(&*binary_data).unwrap(); let file = object::File::parse(&*binary_data).unwrap();
for import in file.imports().unwrap() { for import in file.imports().unwrap() {
if import.library().eq_ignore_ascii_case(b"WS2_32.dll") { if import.library().eq_ignore_ascii_case(b"WS2_32.dll") {