mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
YJIT: Fix page rounding for icache busting
Previously, we found the current page by rounding the current pointer to the closest smaller page size. This is incorrect because pages are relative to the start of the address we reserve. For example, if the starting address is 12KiB modulo the 16KiB page size, once we have more than 4KiB of code, calculating with the address would incorrectly give us page 1 when we're actually still on page 0. Previously, I can reproduce crashes with: make btest RUN_OPTS=--yjit-code-page-size=32 on ARM64 macOS, where system page sizes are 16KiB.
This commit is contained in:
parent
8bbcb75377
commit
87bb0bee6b
Notes:
git
2022-10-21 21:07:03 +00:00
Merged: https://github.com/ruby/ruby/pull/6607 Merged-By: XrXr
1 changed files with 7 additions and 5 deletions
|
@ -217,12 +217,14 @@ impl CodeBlock {
|
|||
#[cfg(any(feature = "disasm", target_arch = "aarch64"))]
|
||||
pub fn writable_addrs(&self, start_ptr: CodePtr, end_ptr: CodePtr) -> Vec<(usize, usize)> {
|
||||
let mut addrs = vec![];
|
||||
let mut start = start_ptr.raw_ptr() as usize;
|
||||
let codeblock_end = self.get_ptr(self.get_mem_size()).raw_ptr() as usize;
|
||||
let end = std::cmp::min(end_ptr.raw_ptr() as usize, codeblock_end);
|
||||
let mut start = start_ptr.into_usize();
|
||||
let region_start = self.get_ptr(0).into_usize();
|
||||
let region_end = self.get_ptr(self.get_mem_size()).into_usize();
|
||||
let end = std::cmp::min(end_ptr.into_usize(), region_end);
|
||||
while start < end {
|
||||
let current_page = start / self.page_size * self.page_size;
|
||||
let page_end = std::cmp::min(end, current_page + self.page_end()) as usize;
|
||||
let current_page = region_start +
|
||||
(start.saturating_sub(region_start) / self.page_size * self.page_size);
|
||||
let page_end = std::cmp::min(end, current_page + self.page_end());
|
||||
addrs.push((start, page_end));
|
||||
start = current_page + self.page_size + self.page_start();
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue