about summary refs log tree commit diff
path: root/tests/run-make/symbol-mangling-hashed/rmake.rs
blob: 136e6b9fa3a95607fdc1e2d811abf7bbe34544a6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
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);
    }
}