Chapter 25: Moduleslinear-chapter25

1 files changed, 23 insertions, 21 deletions

diff --git a/src/test/java/com/seaofnodes/simple/TestArm64.java b/src/test/java/com/seaofnodes/simple/TestArm64.java
index a1c02ede..14864d64 100644
--- a/src/test/java/com/seaofnodes/simple/TestArm64.java
+++ b/src/test/java/com/seaofnodes/simple/TestArm64.java
@@ -1,11 +1,9 @@
 package com.seaofnodes.simple;
 
-
 import com.seaofnodes.simple.codegen.CodeGen;
+import com.seaofnodes.simple.node.FunNode;
 import com.seaofnodes.simple.node.cpus.arm.arm;
-import junit.framework.Assert;
-import java.nio.file.Files;
-import java.nio.file.Path;
+import com.seaofnodes.simple.util.BAOS;
 import java.io.IOException;
 
 import static org.junit.Assert.assertEquals;
@@ -13,36 +11,40 @@ import static org.junit.Assert.assertEquals;
 // Runs 32-bit ARM code in an emulator
 
 public class TestArm64 {
-    public static EvalArm64 build( String file, int arg, int spills, boolean print ) throws IOException {
-        String dir = "src/test/java/com/seaofnodes/simple/progs";
-        String src = Files.readString(Path.of(dir + "/"+file+".smp"));
-        return build(dir,src, file, arg, spills, print);
-    }
 
-    public static EvalArm64 build( String name, String src, int arg, int spills, boolean print ) throws IOException {
-        String dir = "src/test/java/com/seaofnodes/simple/progs";
-        return build(dir, src ,name, arg, spills, print);
-    }
-
-    // Compile and run a simple program
-    public static EvalArm64 build( String dir, String src, String file, int arg, int spills, boolean print ) throws IOException {
+    public static EvalArm64 build( String main, String src, int arg, int spills, boolean print ) throws IOException {
         // Compile and export Simple
-        CodeGen code = new CodeGen(src).driver("arm", "SystemV",null);
-        if( print ) { code.print_as_hex(); System.out.print(code.asm()); }
+        CodeGen code = new CodeGen(src).driver("arm", "SystemV",true,false);
 
         // Allocation quality not degraded
         int delta = spills>>3;
         if( delta == 0 ) delta =1;
-        if( spills != -1 ) assertEquals("Expect spills:", spills, code._regAlloc._spillScaled, delta);
+        if( spills != -1 && !CodeGen.iterSeedOverridden() ) assertEquals("Expect spills:", spills, code._regAlloc._spillScaled, delta);
 
         // Image
         byte[] image = new byte[1<<20]; // A megabyte (1024*1024 bytes)
         EvalArm64 ARM = new EvalArm64(image, 1<<16);
-        // Code at offset 0
-        System.arraycopy(code._encoding.bits(), 0, image, 0, code._encoding.bits().length);
+        int start = 0;          // Code at offset 0
+        int cpool = fill(code._encoding._bits ,image,start);
+        int sdata = fill(code._encoding._cpool,image,cpool);
+        int end   = fill(code._encoding._sdata,image,sdata);
+
         // Initial incoming int arg
         ARM.regs[arm.X0] = arg;
         // malloc memory starts at 64K and goes up
+
+        // Look up starting PC or zero if not given
+        if( main!=null )
+            for( FunNode fun : code._linker )
+                if( fun != null && !fun.isDead() && fun._compunit == code.compunit() && main.equals(fun._name) )
+                    ARM._pc = code._encoding.opStart(fun);
+
         return ARM;
     }
+
+    // Fill bits into image; return end point rounded up.
+    private static int fill( BAOS bits, byte[] image, int off) {
+        System.arraycopy(bits.buf(), 0, image, off, bits.size());
+        return (off+bits.size() + 15) & -16;
+    }
 }