mirror
/
bd-jb
mirror of https://github.com/TheOfficialFloW/bd-jb
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Split mapPayload into jitMap and jitCopy.

This commit is contained in:
Andy Nguyen 2021-11-03 17:19:40 +01:00
parent 39ab5ea346
commit b45af6fec5
1 changed files with 41 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
com/bdjb/JIT.java
View File

@ -138,6 +138,44 @@ public final class JIT {
return (x + align - 1) & ~(align - 1);
}
public long jitMap(long size, long alignment) {
long name = api.malloc(4);
api.strcpy(name, "jit");
long blob = api.call(BufferBlob__create, name, size);
api.free(name);
if (blob == 0) {
throw new IllegalStateException("Could not map JIT memory.");
}
long code = blob + api.read32(blob + 0x20);
return align(code, alignment);
}
public void jitCopy(long dest, byte[] src, long n) {
long req = api.malloc(COMPILER_AGENT_REQUEST_SIZE);
long resp = api.malloc(API.INT8_SIZE);
for (long i = 0; i < n; i += CHUNK_SIZE) {
byte[] chunk = new byte[CHUNK_SIZE];
System.arraycopy(src, (int) i, chunk, 0, (int) Math.min(n - i, CHUNK_SIZE));
api.memset(req, 0, COMPILER_AGENT_REQUEST_SIZE);
api.memcpy(req + 0x00, chunk, Math.min(n - i, CHUNK_SIZE));
api.write64(req + 0x38, dest + i - 0x28);
api.call(write, compilerAgentSocket, req, COMPILER_AGENT_REQUEST_SIZE);
api.write8(resp, (byte) 0);
api.call(read, compilerAgentSocket, resp, API.INT8_SIZE);
if (api.read8(resp) != ACK_MAGIC_NUMBER) {
throw new IllegalStateException("Wrong compiler resp.");
}
}
api.free(resp);
api.free(req);
}
public long mapPayload(String path, long dataSectionOffset) throws Exception {
RandomAccessFile file = new RandomAccessFile(path, "r");
@ -155,45 +193,20 @@ public final class JIT {
}
// Allocate JIT memory.
long name = api.malloc(4);
api.strcpy(name, "jit");
long blob = api.call(BufferBlob__create, name, size);
api.free(name);
if (blob == 0) {
throw new IllegalStateException("Could not allocate JIT memory.");
}
long code = blob + api.read32(blob + 0x20);
long address = align(code, ALIGNMENT);
long address = jitMap(size, ALIGNMENT);
byte[] chunk = new byte[CHUNK_SIZE];
long request = api.malloc(COMPILER_AGENT_REQUEST_SIZE);
long response = api.malloc(API.INT8_SIZE);
// Copy .text section.
for (long i = 0; i < dataSectionOffset; i += CHUNK_SIZE) {
api.memset(chunk, 0, CHUNK_SIZE);
file.seek(i);
file.read(chunk, 0, (int) Math.min(dataSectionOffset - i, CHUNK_SIZE));
int read = file.read(chunk, 0, (int) Math.min(dataSectionOffset - i, CHUNK_SIZE));
api.memset(request, 0, COMPILER_AGENT_REQUEST_SIZE);
api.memcpy(request + 0x00, chunk, CHUNK_SIZE);
api.write64(request + 0x38, address + i - 0x28);
api.call(write, compilerAgentSocket, request, COMPILER_AGENT_REQUEST_SIZE);
api.write8(response, (byte) 0);
api.call(read, compilerAgentSocket, response, API.INT8_SIZE);
if (api.read8(response) != ACK_MAGIC_NUMBER) {
throw new IllegalStateException("Wrong compiler response.");
}
jitCopy(address + i, chunk, read);
}
api.free(response);
api.free(request);
// Map the .data section as RW.
if (api.call(
mmap,