aa430587cc
This brings the Java API up to par with Python feature-wise and substantially simplifies the hook implementation, enabling proper bounds-checked hooks. The rewrite strives for compatibility with the previous API, but there are some breaking changes. It is possible to push closer to full backwards compatibility if required, at the cost of reintroducing some of the suboptimal designs. Here are the main points of breakage: - ReadHook and WriteHook are gone, replaced simply by MemHook. Hooking valid memory accesses now requires a type parameter. This enables fetch and read-after hooks with a unified API and a single callback object. - mem_read now takes an int, not a long. We are unable to allocate more than 2GB in a single request anyway (Java limitation). - Instruction hooks now require specifying the instruction explicitly, instead of guessing based on the hook type. This is necessary to distinguish sysenter/syscall and ARM64 mrs/msr/sys/sysl, without excessively bloating the library with redundant hook types. Bounds must also be specified, to support bounds-checked instruction hooks. - Reading object-type registers (any register larger than 64 bits, or registers with special formats) requires a second argument to reg_read. This allows us to provide a fast reg_read that returns a long for the common cases, while still supporting a more general reg_read for other registers. - mem_map_ptr is rewritten to take a *direct* java.nio.Buffer, which enables many more use cases than a simple byte array, and improves performance (a byte array cannot really be used as a mapped buffer without GC-pinning it, which hurts the GC performance). - Context handling API is redesigned to be safer and more object-oriented. A lot of bugs are fixed with this implementation: - Unicorn instances can be properly garbage-collected, instead of hanging around forever in the Unicorn.unicorns table. - Hooks no longer fire outside of their bounds (#1164), and in fact, hook bounds are properly respected (previously, all hooks were just registered globally to all addresses). - Hooks are substantially faster, as they are now dispatched directly via a single method call rather than being indirected through invokeCallbacks. - Loading vector registers works now, rather than crashing the VM (#1539). Several features are now enabled in the Java implementation: - All of the current ctl_* calls are implemented. - mmio_map is implemented. - New virtual TLB mode is implemented. - reading/writing Context registers is implemented. - New hook types are added: TcgOpcodeHook, EdgeGeneratedHook, InvalidInstructionHook, TlbFillHook, and the instruction hooks Arm64SysHook, CpuidHook. - All known special registers are supported.
75 lines
2.5 KiB
Java
75 lines
2.5 KiB
Java
package tests;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
import java.math.BigInteger;
|
|
|
|
import org.junit.Test;
|
|
|
|
import unicorn.Unicorn;
|
|
import unicorn.UnicornException;
|
|
import unicorn.CodeHook;
|
|
|
|
public class RegressionTests {
|
|
/** Test for GH #1539: Unable to read ARM64 v or q register using java binding */
|
|
@Test
|
|
public void testARM64VReg() {
|
|
Unicorn uc = new Unicorn(Unicorn.UC_ARCH_ARM64, Unicorn.UC_MODE_ARM);
|
|
uc.reg_write(Unicorn.UC_ARM64_REG_X0, 0x1);
|
|
uc.reg_write(Unicorn.UC_ARM64_REG_V0, BigInteger.valueOf(0x1234));
|
|
uc.reg_read(Unicorn.UC_ARM64_REG_X0);
|
|
assertEquals("V0 value", BigInteger.valueOf(0x1234),
|
|
uc.reg_read(Unicorn.UC_ARM64_REG_V0, null)); // should not crash
|
|
assertEquals("V0 low byte", 0x34,
|
|
uc.reg_read(Unicorn.UC_ARM64_REG_B0));
|
|
assertEquals("V0 low halfword", 0x1234,
|
|
uc.reg_read(Unicorn.UC_ARM64_REG_H0));
|
|
uc.close();
|
|
}
|
|
|
|
/** Test for GH #1164: Java binding use CodeHook on Windows, will invoke callback before every instruction */
|
|
@Test
|
|
public void testCodeHookRunsOnce() {
|
|
byte[] ARM_CODE =
|
|
{ 55, 0, (byte) 0xa0, (byte) 0xe3, 3, 16, 66, (byte) 0xe0 }; // mov r0, #0x37; sub r1, r2, r3
|
|
int ADDRESS = 0x10000;
|
|
final int[] hook_count = { 0 };
|
|
|
|
Unicorn u = new Unicorn(Unicorn.UC_ARCH_ARM, Unicorn.UC_MODE_ARM);
|
|
u.mem_map(ADDRESS, 2 * 1024 * 1024, Unicorn.UC_PROT_ALL);
|
|
u.mem_write(ADDRESS, ARM_CODE);
|
|
u.hook_add((CodeHook) (uc, address, size, user) -> hook_count[0] += 1,
|
|
ADDRESS, ADDRESS, null);
|
|
|
|
u.emu_start(ADDRESS, ADDRESS + ARM_CODE.length, 0, 0);
|
|
assertEquals("Hook should only be called once", 1, hook_count[0]);
|
|
|
|
u.close();
|
|
}
|
|
|
|
/** Test that close() can be called multiple times without crashing */
|
|
@Test
|
|
public void testCloseIdempotent() {
|
|
Unicorn u = new Unicorn(Unicorn.UC_ARCH_ARM, Unicorn.UC_MODE_ARM);
|
|
u.close();
|
|
u.close();
|
|
}
|
|
|
|
/** Test that Unicorn instances are properly garbage-collected */
|
|
@Test
|
|
public void testUnicornsWillGC() {
|
|
final boolean[] close_called = { false };
|
|
|
|
new Unicorn(Unicorn.UC_ARCH_ARM, Unicorn.UC_MODE_ARM) {
|
|
@Override
|
|
public void close() throws UnicornException {
|
|
close_called[0] = true;
|
|
super.close();
|
|
}
|
|
};
|
|
System.gc();
|
|
System.runFinalization();
|
|
assertEquals("close() was called", true, close_called[0]);
|
|
}
|
|
}
|