tests: port symbol-mangling-hashed to rmake.rs

- Use `object` based test logic instead of processing `nm`
  human-readable textual output.
- Try to expand test coverage to not be limited to only linux + x86_64.

Co-authored-by: binarycat <binarycat@envs.net>
This commit is contained in:
许杰友 Jieyou Xu (Joe) 2025-01-20 19:14:51 +08:00 committed by Jieyou Xu
parent 0bc1b4f4f6
commit 9734ebb9be
9 changed files with 128 additions and 69 deletions

View file

@ -1,2 +1 @@
run-make/split-debuginfo/Makefile
run-make/symbol-mangling-hashed/Makefile

View file

@ -1,48 +0,0 @@
include ../tools.mk
# ignore-cross-compile
# only-linux
# only-x86_64
NM=nm -D
RLIB_NAME=liba_rlib.rlib
DYLIB_NAME=liba_dylib.so
SO_NAME=libb_dylib.so
BIN_NAME=b_bin
ifeq ($(UNAME),Darwin)
NM=nm -gU
RLIB_NAME=liba_rlib.rlib
DYLIB_NAME=liba_dylib.dylib
SO_NAME=libb_dylib.dylib
BIN_NAME=b_bin
endif
ifdef IS_WINDOWS
NM=nm -g
RLIB_NAME=liba_rlib.dll.a
DYLIB_NAME=liba_dylib.dll
SO_NAME=libb_dylib.dll
BIN_NAME=b_bin.exe
endif
all:
$(RUSTC) -C prefer-dynamic -Z unstable-options -C symbol-mangling-version=hashed -C metadata=foo a_dylib.rs
$(RUSTC) -C prefer-dynamic -Z unstable-options -C symbol-mangling-version=hashed -C metadata=bar a_rlib.rs
$(RUSTC) -C prefer-dynamic -L $(TMPDIR) b_dylib.rs
$(RUSTC) -C prefer-dynamic -L $(TMPDIR) b_bin.rs
# Check hashed symbol name
[ "$$($(NM) $(TMPDIR)/$(DYLIB_NAME) | grep -c hello)" -eq "0" ]
[ "$$($(NM) $(TMPDIR)/$(DYLIB_NAME) | grep _RNxC7a_dylib | grep -c ' T ')" -eq "2" ]
[ "$$($(NM) $(TMPDIR)/$(SO_NAME) | grep b_dylib | grep -c hello)" -eq "1" ]
[ "$$($(NM) $(TMPDIR)/$(SO_NAME) | grep _RNxC6a_rlib | grep -c ' T ')" -eq "2" ]
[ "$$($(NM) $(TMPDIR)/$(SO_NAME) | grep _RNxC7a_dylib | grep -c ' U ')" -eq "1" ]
[ "$$($(NM) $(TMPDIR)/$(BIN_NAME) | grep _RNxC6a_rlib | grep -c ' U ')" -eq "1" ]
[ "$$($(NM) $(TMPDIR)/$(BIN_NAME) | grep _RNxC7a_dylib | grep -c ' U ')" -eq "1" ]
[ "$$($(NM) $(TMPDIR)/$(BIN_NAME) | grep b_dylib | grep hello | grep -c ' U ')" -eq "1" ]
$(call RUN,$(BIN_NAME))

View file

@ -1,9 +0,0 @@
extern crate a_dylib;
extern crate a_rlib;
extern crate b_dylib;
fn main() {
a_rlib::hello();
a_dylib::hello();
b_dylib::hello();
}

View file

@ -1,9 +0,0 @@
#![crate_type = "dylib"]
extern crate a_dylib;
extern crate a_rlib;
pub fn hello() {
a_rlib::hello();
a_dylib::hello();
}

View file

@ -0,0 +1,9 @@
extern crate default_dylib;
extern crate hashed_dylib;
extern crate hashed_rlib;
fn main() {
hashed_rlib::hrhello();
hashed_dylib::hdhello();
default_dylib::ddhello();
}

View file

@ -0,0 +1,9 @@
#![crate_type = "dylib"]
extern crate hashed_dylib;
extern crate hashed_rlib;
pub fn ddhello() {
hashed_rlib::hrhello();
hashed_dylib::hdhello();
}

View file

@ -1,4 +1,4 @@
#![crate_type = "dylib"]
pub fn hello() {
pub fn hdhello() {
println!("hello dylib");
}

View file

@ -1,5 +1,5 @@
#![crate_type = "rlib"]
pub fn hello() {
pub fn hrhello() {
println!("hello rlib");
}

View file

@ -0,0 +1,108 @@
// ignore-tidy-linelength
//! Basic smoke test for the unstable option `-C symbol_mangling_version=hashed` which aims to
//! replace full symbol mangling names based on hash digests to shorten symbol name lengths in
//! dylibs for space savings.
//!
//! # References
//!
//! - MCP #705: Provide option to shorten symbol names by replacing them with a digest:
//! <https://github.com/rust-lang/compiler-team/issues/705>.
//! - Implementation PR: <https://github.com/rust-lang/rust/pull/118636>.
//! - PE format: <https://learn.microsoft.com/en-us/windows/win32/debug/pe-format>.
//@ ignore-cross-compile
#![deny(warnings)]
use run_make_support::symbols::exported_dynamic_symbol_names;
use run_make_support::{bin_name, cwd, dynamic_lib_name, is_darwin, object, rfs, run, rustc};
macro_rules! adjust_symbol_prefix {
($name:literal) => {
if is_darwin() { concat!("_", $name) } else { $name }
};
}
fn main() {
rustc()
.input("hashed_dylib.rs")
.prefer_dynamic()
.arg("-Zunstable-options")
.symbol_mangling_version("hashed")
.metadata("foo")
.run();
rustc()
.input("hashed_rlib.rs")
.prefer_dynamic()
.arg("-Zunstable-options")
.symbol_mangling_version("hashed")
.metadata("bar")
.run();
rustc().input("default_dylib.rs").library_search_path(cwd()).prefer_dynamic().run();
rustc().input("default_bin.rs").library_search_path(cwd()).prefer_dynamic().run();
{
// Check hashed symbol name
let dylib_filename = dynamic_lib_name("hashed_dylib");
println!("checking dylib `{dylib_filename}`");
let dylib_blob = rfs::read(&dylib_filename);
let dylib_file = object::File::parse(&*dylib_blob)
.unwrap_or_else(|e| panic!("failed to parse `{dylib_filename}`: {e}"));
let dynamic_symbols = exported_dynamic_symbol_names(&dylib_file);
if dynamic_symbols.iter().filter(|sym| sym.contains("hdhello")).count() != 0 {
eprintln!("exported dynamic symbols: {:#?}", dynamic_symbols);
panic!("expected no occurrence of `hdhello`");
}
let expected_prefix = adjust_symbol_prefix!("_RNxC12hashed_dylib");
if dynamic_symbols.iter().filter(|sym| sym.starts_with(expected_prefix)).count() != 2 {
eprintln!("exported dynamic symbols: {:#?}", dynamic_symbols);
panic!("expected two dynamic symbols starting with `{expected_prefix}`");
}
}
{
let dylib_filename = dynamic_lib_name("default_dylib");
println!("checking so `{dylib_filename}`");
let dylib_blob = rfs::read(&dylib_filename);
let dylib_file = object::File::parse(&*dylib_blob)
.unwrap_or_else(|e| panic!("failed to parse `{dylib_filename}`: {e}"));
let dynamic_symbols = exported_dynamic_symbol_names(&dylib_file);
if dynamic_symbols
.iter()
.filter(|sym| sym.contains("default_dylib") && sym.contains("ddhello"))
.count()
!= 1
{
eprintln!("exported dynamic symbols: {:#?}", dynamic_symbols);
panic!("expected one occurrence of mangled `ddhello`");
}
let expected_rlib_prefix = adjust_symbol_prefix!("_RNxC11hashed_rlib");
if dynamic_symbols.iter().filter(|sym| sym.starts_with(expected_rlib_prefix)).count() != 2 {
eprintln!("exported dynamic symbols: {:#?}", dynamic_symbols);
panic!("expected two exported symbols starting with `{expected_rlib_prefix}`");
}
let expected_dylib_prefix = adjust_symbol_prefix!("_RNxC12hashed_dylib");
if dynamic_symbols.iter().any(|sym| sym.starts_with("_RNxC12hashed_dylib")) {
eprintln!("exported dynamic symbols: {:#?}", dynamic_symbols);
panic!("did not expect any symbols starting with `{expected_dylib_prefix}`");
}
}
// Check that the final binary can be run.
{
let bin_filename = bin_name("default_bin");
run(&bin_filename);
}
}