Skip to content

Commit fcb1809

Browse files
committed
Auto merge of #118344 - saethlin:rmeta-header-pos, r=WaffleLapkin
Use a u64 for the rmeta root position Waffle noticed this in rust-lang/rust#117301 (comment) We've upgraded the other file offsets to u64, and this one only costs 4 bytes per file. Also the way the truncation was being done before was extremely easy to miss, I sure missed it! It's not clear to me if not having this change effectively made the other upgrades from u32 to u64 ineffective, but we can have it now. r? `@WaffleLapkin`
2 parents 9290748 + d3bf675 commit fcb1809

File tree

1 file changed

+18
-13
lines changed

1 file changed

+18
-13
lines changed

crates/proc-macro-api/src/version.rs

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,8 @@ fn read_section<'a>(dylib_binary: &'a [u8], section_name: &str) -> io::Result<&'
8585
}
8686

8787
/// Check the version of rustc that was used to compile a proc macro crate's
88-
///
8988
/// binary file.
89+
///
9090
/// A proc macro crate binary's ".rustc" section has following byte layout:
9191
/// * [b'r',b'u',b's',b't',0,0,0,5] is the first 8 bytes
9292
/// * ff060000 734e6150 is followed, it's the snappy format magic bytes,
@@ -96,8 +96,8 @@ fn read_section<'a>(dylib_binary: &'a [u8], section_name: &str) -> io::Result<&'
9696
/// The bytes you get after decompressing the snappy format portion has
9797
/// following layout:
9898
/// * [b'r',b'u',b's',b't',0,0,0,5] is the first 8 bytes(again)
99-
/// * [crate root bytes] next 4 bytes is to store crate root position,
100-
/// according to rustc's source code comment
99+
/// * [crate root bytes] next 8 bytes (4 in old versions) is to store
100+
/// crate root position, according to rustc's source code comment
101101
/// * [length byte] next 1 byte tells us how many bytes we should read next
102102
/// for the version string's utf8 bytes
103103
/// * [version string bytes encoded in utf8] <- GET THIS BOI
@@ -119,14 +119,19 @@ pub fn read_version(dylib_path: &AbsPath) -> io::Result<String> {
119119
}
120120
let version = u32::from_be_bytes([dot_rustc[4], dot_rustc[5], dot_rustc[6], dot_rustc[7]]);
121121
// Last supported version is:
122-
// https://github.com/rust-lang/rust/commit/0696e79f2740ad89309269b460579e548a5cd632
123-
let snappy_portion = match version {
124-
5 | 6 => &dot_rustc[8..],
122+
// https://github.com/rust-lang/rust/commit/b94cfefc860715fb2adf72a6955423d384c69318
123+
let (snappy_portion, bytes_before_version) = match version {
124+
5 | 6 => (&dot_rustc[8..], 13),
125125
7 | 8 => {
126126
let len_bytes = &dot_rustc[8..12];
127127
let data_len = u32::from_be_bytes(len_bytes.try_into().unwrap()) as usize;
128-
&dot_rustc[12..data_len + 12]
128+
(&dot_rustc[12..data_len + 12], 13)
129129
}
130+
9 => {
131+
let len_bytes = &dot_rustc[8..16];
132+
let data_len = u64::from_le_bytes(len_bytes.try_into().unwrap()) as usize;
133+
(&dot_rustc[16..data_len + 12], 17)
134+
}
130135
_ => {
131136
return Err(io::Error::new(
132137
io::ErrorKind::InvalidData,
@@ -142,15 +147,15 @@ pub fn read_version(dylib_path: &AbsPath) -> io::Result<String> {
142147
Box::new(SnapDecoder::new(snappy_portion))
143148
};
144149

145-
// the bytes before version string bytes, so this basically is:
150+
// We're going to skip over the bytes before the version string, so basically:
146151
// 8 bytes for [b'r',b'u',b's',b't',0,0,0,5]
147-
// 4 bytes for [crate root bytes]
152+
// 4 or 8 bytes for [crate root bytes]
148153
// 1 byte for length of version string
149-
// so 13 bytes in total, and we should check the 13th byte
154+
// so 13 or 17 bytes in total, and we should check the last of those bytes
150155
// to know the length
151-
let mut bytes_before_version = [0u8; 13];
152-
uncompressed.read_exact(&mut bytes_before_version)?;
153-
let length = bytes_before_version[12];
156+
let mut bytes = [0u8; 17];
157+
uncompressed.read_exact(&mut bytes[..bytes_before_version])?;
158+
let length = bytes[bytes_before_version - 1];
154159

155160
let mut version_string_utf8 = vec![0u8; length as usize];
156161
uncompressed.read_exact(&mut version_string_utf8)?;

0 commit comments

Comments
 (0)