mirror of https://github.com/rust-lang/rust
111 lines
3.1 KiB
Rust
111 lines
3.1 KiB
Rust
// This test checks that the object files we generate are actually
|
|
// LLVM bitcode files (as used by linker LTO plugins) when compiling with
|
|
// -Clinker-plugin-lto.
|
|
// See https://github.com/rust-lang/rust/pull/50000
|
|
|
|
#![feature(path_file_prefix)]
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use run_make_support::{
|
|
cwd, has_extension, has_prefix, llvm_ar, llvm_bcanalyzer, path, rfs, rust_lib_name, rustc,
|
|
shallow_find_files, static_lib_name,
|
|
};
|
|
|
|
fn main() {
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("staticlib"),
|
|
output: path(static_lib_name("liblib")),
|
|
lto: None,
|
|
emit_obj: false,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("staticlib"),
|
|
output: path(static_lib_name("liblib-fat-lto")),
|
|
lto: Some("fat"),
|
|
emit_obj: false,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("staticlib"),
|
|
output: path(static_lib_name("liblib-thin-lto")),
|
|
lto: Some("thin"),
|
|
emit_obj: false,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("rlib"),
|
|
output: path(rust_lib_name("liblib")),
|
|
lto: None,
|
|
emit_obj: false,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("cdylib"),
|
|
output: path("cdylib.o"),
|
|
lto: None,
|
|
emit_obj: true,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("lib.rs"),
|
|
crate_type: Some("dylib"),
|
|
output: path("rdylib.o"),
|
|
lto: None,
|
|
emit_obj: true,
|
|
});
|
|
check_bitcode(LibBuild {
|
|
source: path("main.rs"),
|
|
crate_type: None,
|
|
output: path("exe.o"),
|
|
lto: None,
|
|
emit_obj: true,
|
|
});
|
|
}
|
|
|
|
#[track_caller]
|
|
fn check_bitcode(instructions: LibBuild) {
|
|
let mut rustc = rustc();
|
|
rustc
|
|
.input(instructions.source)
|
|
.output(&instructions.output)
|
|
.opt_level("2")
|
|
.codegen_units(1)
|
|
.arg("-Clinker-plugin-lto");
|
|
if instructions.emit_obj {
|
|
rustc.emit("obj");
|
|
}
|
|
if let Some(crate_type) = instructions.crate_type {
|
|
rustc.crate_type(crate_type);
|
|
}
|
|
if let Some(lto) = instructions.lto {
|
|
rustc.arg(format!("-Clto={lto}"));
|
|
}
|
|
rustc.run();
|
|
|
|
if instructions.output.extension().unwrap() != "o" {
|
|
// Remove all potential leftover object files, then turn the output into an object file.
|
|
for object in shallow_find_files(cwd(), |path| has_extension(path, "o")) {
|
|
rfs::remove_file(object);
|
|
}
|
|
llvm_ar().extract().arg(&instructions.output).run();
|
|
}
|
|
|
|
for object in shallow_find_files(cwd(), |path| {
|
|
has_prefix(path, instructions.output.file_prefix().unwrap().to_str().unwrap())
|
|
&& has_extension(path, "o")
|
|
}) {
|
|
// All generated object files should be LLVM bitcode files - this will fail otherwise.
|
|
llvm_bcanalyzer().input(object).run();
|
|
}
|
|
}
|
|
|
|
struct LibBuild {
|
|
source: PathBuf,
|
|
crate_type: Option<&'static str>,
|
|
output: PathBuf,
|
|
lto: Option<&'static str>,
|
|
emit_obj: bool,
|
|
}
|