https://git.ruby-lang.org/ruby.git/commit/?id=51c84f0033

From 51c84f003391a45460d188be155a3e85057234fd Mon Sep 17 00:00:00 2001
From: Maxime Chevalier-Boisvert <maxime.chevalierboisvert@s...>
Date: Wed, 7 Jul 2021 15:26:58 -0400
Subject: Code page allocation code

---
 yjit_asm.c | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 yjit_asm.h |  6 +++++-
 2 files changed, 58 insertions(+), 1 deletion(-)

diff --git a/yjit_asm.c b/yjit_asm.c
index da640e6d7d..1b44587e3c 100644
--- a/yjit_asm.c
+++ b/yjit_asm.c
@@ -218,6 +218,59 @@ uint8_t* alloc_exec_mem(uint32_t mem_size) https://github.com/ruby/ruby/blob/trunk/yjit_asm.c#L218
 #endif
 }
 
+// Size of code pages to allocate
+#define CODE_PAGE_SIZE 16 * 1024
+
+// How many code pages to allocate at once
+#define PAGES_PER_ALLOC 512
+
+typedef struct free_list_node
+{
+    uint8_t* page_ptr;
+
+    struct free_list_node *next;
+
+} freelist_t;
+
+freelist_t *freelist = NULL;
+
+// Allocate a single code page from a pool of free pages
+uint8_t* alloc_code_page()
+{
+    fprintf(stderr, "allocating code page\n");
+
+    // If the free list is empty
+    if (!freelist) {
+        // Allocate many pages at once
+        uint8_t* code_chunk = alloc_exec_mem(PAGES_PER_ALLOC * CODE_PAGE_SIZE);
+
+        // Do this in reverse order so we allocate our pages in order
+        for (int i = PAGES_PER_ALLOC - 1; i >= 0; --i) {
+            freelist_t* node = malloc(sizeof(freelist_t));
+            node->page_ptr = code_chunk + i * CODE_PAGE_SIZE;
+            node->next = freelist;
+            freelist = node;
+        }
+    }
+
+    freelist_t* free_node = freelist;
+    uint8_t* page_ptr = freelist->page_ptr;
+
+    freelist = freelist->next;
+    free(free_node);
+
+    return page_ptr;
+}
+
+// Put a code page back into the allocation pool
+void free_code_page(uint8_t* page_ptr)
+{
+    freelist_t* node = malloc(sizeof(freelist_t));
+    node->page_ptr = page_ptr;
+    node->next = freelist;
+    freelist = node;
+}
+
 // Initialize a code block object
 void cb_init(codeblock_t* cb, uint8_t* mem_block, uint32_t mem_size)
 {
diff --git a/yjit_asm.h b/yjit_asm.h
index 15c670fec7..c94b54240c 100644
--- a/yjit_asm.h
+++ b/yjit_asm.h
@@ -240,8 +240,12 @@ x86opnd_t const_ptr_opnd(const void *ptr); https://github.com/ruby/ruby/blob/trunk/yjit_asm.h#L240
      sizeof(((struct_type*)0)->member_name[0]) * idx)  \
 )
 
-// Code block methods
+// Machine code allocation
 uint8_t* alloc_exec_mem(uint32_t mem_size);
+uint8_t* alloc_code_page();
+void free_code_page(uint8_t* page_ptr);
+
+// Code block methods
 void cb_init(codeblock_t* cb, uint8_t* mem_block, uint32_t mem_size);
 void cb_align_pos(codeblock_t* cb, uint32_t multiple);
 void cb_set_pos(codeblock_t* cb, uint32_t pos);
-- 
cgit v1.2.1


--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/