1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
package com.seaofnodes.simple;
import com.seaofnodes.simple.codegen.CodeGen;
import com.seaofnodes.simple.codegen.StaticData;
import com.seaofnodes.simple.type.TypeStruct;
import com.seaofnodes.simple.type.TypeInteger;
import com.seaofnodes.simple.util.Ary;
import com.seaofnodes.isa.eval.EvalArm64;
import com.seaofnodes.isa.eval.EvalRisc5;
import java.nio.file.Files;
import java.nio.file.Path;
import org.junit.Test;
import static org.junit.Assert.*;
public class StaticDataTest {
private static final String SRC="val n=42; val f={int x -> x+n;}; return 0;";
private static StaticData.ObjectData root(CodeGen code) {
return code._encoding._data.entries().stream()
.filter(obj -> obj.symbol.endsWith("Test.$class")).findFirst().orElseThrow();
}
// Read the image before <clinit>, then call the function through its data slot.
@Test public void testEmulatorImage() {
for( String cpu : new String[]{"arm","riscv"} ) {
CodeGen code=new CodeGen(SRC).driver(cpu,"SystemV",true,false);
var e=code._encoding;
var obj=root(code);
TypeStruct ts=(TypeStruct)obj.type;
int cp=(e._bits.size()+15)&-16, sd=(cp+e._cpool.size()+15)&-16;
byte[] image=new byte[1<<20];
System.arraycopy(e._bits.buf(),0,image,0,e._bits.size());
System.arraycopy(e._cpool.buf(),0,image,cp,e._cpool.size());
System.arraycopy(e._sdata.buf(),0,image,sd,e._sdata.size());
int n=sd+obj.offset+ts.offset(ts.find("n"));
int f=sd+obj.offset+ts.offset(ts.find("f"));
if( cpu.equals("arm") ) {
var arm=new EvalArm64(image,1<<16);
assertEquals(42,arm.ld4s(n));
arm._pc=arm.ld4s(f);
arm.regs[0]=7;
assertEquals(0,arm.step(1000));
assertEquals(49,arm.regs[0]);
} else {
var r5=new EvalRisc5(image,1<<16);
assertEquals(42,r5.ld4s(n));
r5._pc=r5.ld4s(f);
r5.regs[10]=7;
assertEquals(0,r5.step(1000));
assertEquals(49,r5.regs[10]);
}
}
}
@Test public void testNativeImage() throws Exception {
Path dir=Path.of("build/objs/static-data");
Files.createDirectories(dir);
CodeGen code=new CodeGen(null,dir.toString(),null,"Test",SRC,123L,true,
TypeInteger.BOT)
.driver(TestC.CPU_PORT,TestC.CALL_CONVENTION,false,false);
TypeStruct ts=(TypeStruct)root(code).type;
Path driver=dir.resolve("driver.c");
Files.writeString(driver,"""
#include <stdint.h>
#include <string.h>
extern unsigned char data[] __asm__("Test.$class");
typedef int64_t (__attribute__((CALL_CONV)) *fun)(int64_t);
int main(void) {
int64_t n; uint32_t f;
memcpy(&n,data+%d,sizeof(n));
memcpy(&f,data+%d,sizeof(f));
return n!=42 || ((fun)(uintptr_t)f)(7)!=49;
}
""".formatted(ts.offset(ts.find("n")),ts.offset(ts.find("f"))));
String exe=dir.resolve(TestC.OS.startsWith("Windows")?"test.exe":"test").toString();
TestC.linkExe(dir.resolve("Test.o").toString(),
TestC.CALL_CONVENTION.equals("win64")?"ms_abi":"sysv_abi",
driver.toString(),null,exe);
assertEquals("",TestC.exec(TestC.TEST_TIMEOUT_SECONDS,exe));
}
// Two separately compiled users must reference the same mutable class object.
@Test public void testSharedClass() throws Exception {
Path dir=Path.of("build/objs/static-data-shared");
Files.createDirectories(dir);
Path lib=dir.resolve("lib");
Files.createDirectories(lib);
var paths=new Ary<>(new String[]{lib.toString()});
String[] names={"Globals","A","B"};
String[] sources={"int count=0;",
"val a={ -> Globals; };", "val b={ -> Globals; };"};
for( int i=0; i<names.length; i++ )
new CodeGen(null,(i==0?lib:dir).toString(),i==0?null:paths,names[i],sources[i],123L,true,TypeInteger.BOT)
.driver(TestC.CPU_PORT,TestC.CALL_CONVENTION,false,false);
Path driver=dir.resolve("driver.c");
Files.writeString(driver,"""
#include <stdint.h>
__attribute__((CALL_CONV)) extern int64_t *a(void),*b(void);
int main(void) {
int64_t *x=a(),*y=b();
if(x!=y) return 1;
*x=37;
return *y!=37;
}
""");
String exe=dir.resolve(TestC.OS.startsWith("Windows")?"test.exe":"test").toString();
TestC.linkExe(lib.resolve("Globals.o").toString(),
TestC.CALL_CONVENTION.equals("win64")?"ms_abi":"sysv_abi",
driver.toString(),new Ary<>(new String[]{dir+"/A.o",dir+"/B.o"}),exe);
assertEquals("",TestC.exec(TestC.TEST_TIMEOUT_SECONDS,exe));
}
}
|
