From 6cb9489e9acc976ca37350eda9f6feedf692a60a Mon Sep 17 00:00:00 2001
From: mlokr <jdoka@crownsterling.io>
Date: Mon, 9 Sep 2024 22:17:54 +0200
Subject: [PATCH] saving

---
 hblang/README.md                            | 32 ++++++++++++
 hblang/src/codegen.rs                       |  2 +
 hblang/src/lib.rs                           |  7 ++-
 hblang/tests/codegen_tests_request_page.txt | 58 +++++++++++++++++++++
 4 files changed, 97 insertions(+), 2 deletions(-)
 create mode 100644 hblang/tests/codegen_tests_request_page.txt

diff --git a/hblang/README.md b/hblang/README.md
index accc9ce..c4aa842 100644
--- a/hblang/README.md
+++ b/hblang/README.md
@@ -860,3 +860,35 @@ main := fn(): void {
 	return
 }
 ```
+
+#### request_page
+```hb
+request_page := fn(page_count: u8): ^u8 {
+	msg := "\{00}\{01}xxxxxxxx\0"
+	msg_page_count := msg + 1;
+	*msg_page_count = page_count
+	return @eca(^u8, 3, 2, msg, 12)
+}
+
+create_back_buffer := fn(total_pages: int): ^u32 {
+	if total_pages <= 0xFF {
+		return @bitcast(request_page(total_pages))
+	}
+	ptr := request_page(255)
+	remaining := total_pages - 0xFF
+	loop if remaining <= 0 break else {
+		if remaining < 0xFF {
+			request_page(remaining)
+		} else {
+			request_page(0xFF)
+		}
+		remaining -= 0xFF
+	}
+	return @bitcast(ptr)
+}
+
+main := fn(): void {
+	create_back_buffer(10)
+	return
+}
+```
diff --git a/hblang/src/codegen.rs b/hblang/src/codegen.rs
index a4faeb5..760cb03 100644
--- a/hblang/src/codegen.rs
+++ b/hblang/src/codegen.rs
@@ -3379,6 +3379,7 @@ impl Codegen {
                     .filter(|&(i, f)| {
                         task::unpack(f.offset).is_ok()
                             && (f.runtime || self.ct.active())
+                            && (!self.ct.active() || i != 0)
                             && self.is_fully_linked(i as ty::Func)
                     })
                     .map(|(_, f)| {
@@ -3700,5 +3701,6 @@ mod tests {
         some_generic_code => README;
         integer_inference_issues => README;
         writing_into_string => README;
+        request_page => README;
     }
 }
diff --git a/hblang/src/lib.rs b/hblang/src/lib.rs
index 22ec1d1..c17ffc2 100644
--- a/hblang/src/lib.rs
+++ b/hblang/src/lib.rs
@@ -256,7 +256,11 @@ fn disasm(
 
                 let global_offset: u32 = (offset + rel).try_into().unwrap();
                 if let Some(&(name, ..)) = functions.get(&global_offset) {
-                    write!(out, ":{name}")?;
+                    if name.contains('\0') {
+                        write!(out, ":{name:?}")?;
+                    } else {
+                        write!(out, ":{name}")?;
+                    }
                 } else {
                     let local_has_oob = global_offset < off
                         || global_offset > off + len
@@ -571,7 +575,6 @@ pub fn parse_from_fs(extra_threads: usize, root: &str) -> io::Result<Vec<Ast>> {
     };
 
     let execute_task = |(_, path, command): Task, buffer: &mut Vec<u8>| {
-        log::dbg!("{path:?}");
         if let Some(mut command) = command {
             let output = command.output()?;
             if !output.status.success() {
diff --git a/hblang/tests/codegen_tests_request_page.txt b/hblang/tests/codegen_tests_request_page.txt
new file mode 100644
index 0000000..e04f72e
--- /dev/null
+++ b/hblang/tests/codegen_tests_request_page.txt
@@ -0,0 +1,58 @@
+main:
+    ADDI64 r254, r254, -8d
+    ST r31, r254, 0a, 8h
+    LI64 r2, 10d
+    JAL r31, r0, :create_back_buffer
+    LD r31, r254, 0a, 8h
+    ADDI64 r254, r254, 8d
+    JALA r0, r31, 0a
+create_back_buffer:
+    ADDI64 r254, r254, -32d
+    ST r31, r254, 0a, 32h
+    CP r32, r2
+    LI64 r33, 255d
+    JGTS r32, r33, :0
+    CP r2, r32
+    JAL r31, r0, :request_page
+    JMP :1
+ 0: LI64 r2, 255d
+    JAL r31, r0, :request_page
+    CP r33, r1
+    ADDI64 r32, r32, -255d
+ 6: LI64 r34, 0d
+    JGTS r32, r34, :2
+    JMP :3
+ 2: LI64 r34, 255d
+    JLTS r32, r34, :4
+    LI64 r2, 255d
+    JAL r31, r0, :request_page
+    JMP :5
+ 4: CP r2, r32
+    JAL r31, r0, :request_page
+ 5: CP r34, r32
+    ADDI64 r34, r34, -255d
+    CP r32, r34
+    JMP :6
+ 3: CP r1, r33
+ 1: LD r31, r254, 0a, 32h
+    ADDI64 r254, r254, 32d
+    JALA r0, r31, 0a
+request_page:
+    ADDI64 r254, r254, -32d
+    ST r31, r254, 0a, 32h
+    CP r32, r2
+    LRA r33, r0, :"\0\u{1}xxxxxxxx"
+    CP r34, r33
+    ADDI64 r34, r34, 1d
+    ST r32, r34, 0a, 1h
+    LI64 r2, 3d
+    LI64 r3, 2d
+    CP r4, r33
+    LI64 r5, 12d
+    ECA 
+    LD r31, r254, 0a, 32h
+    ADDI64 r254, r254, 32d
+    JALA r0, r31, 0a
+code size: 440
+ret: 42
+status: Ok(())