add Go bindings

2015-08-27 21:13:50 -07:00
parent b335cf016c
commit c5bcb02847
14 changed files with 2720 additions and 0 deletions
--- a/bindings/go/unicorn/x86_test.go
+++ b/bindings/go/unicorn/x86_test.go
@@ -0,0 +1,145 @@
+package unicorn
+
+import (
+	"testing"
+)
+
+var ADDRESS uint64 = 0x1000000
+
+func MakeUc(mode int, code string) (*Uc, error) {
+	mu, err := NewUc(UC_ARCH_X86, mode)
+	if err != nil {
+		return nil, err
+	}
+	if err := mu.MemMap(ADDRESS, 2*1024*1024); err != nil {
+		return nil, err
+	}
+	if err := mu.MemWrite(ADDRESS, []byte(code)); err != nil {
+		return nil, err
+	}
+	if err := mu.RegWrite(UC_X86_REG_ECX, 0x1234); err != nil {
+		return nil, err
+	}
+	if err := mu.RegWrite(UC_X86_REG_EDX, 0x7890); err != nil {
+		return nil, err
+	}
+	return mu, nil
+}
+
+func TestX86(t *testing.T) {
+	code := "\x41\x4a"
+	mu, err := MakeUc(UC_MODE_32, code)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := mu.Start(ADDRESS, ADDRESS+uint64(len(code))); err != nil {
+		t.Fatal(err)
+	}
+	ecx, _ := mu.RegRead(UC_X86_REG_ECX)
+	edx, _ := mu.RegRead(UC_X86_REG_EDX)
+	if ecx != 0x1235 || edx != 0x788f {
+		t.Fatal("Bad register values.")
+	}
+}
+
+func TestX86InvalidRead(t *testing.T) {
+	code := "\x8B\x0D\xAA\xAA\xAA\xAA\x41\x4a"
+	mu, err := MakeUc(UC_MODE_32, code)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = mu.Start(ADDRESS, ADDRESS+uint64(len(code)))
+	if err.(UcError) != UC_ERR_MEM_READ {
+		t.Fatal("Expected UC_ERR_MEM_READ")
+	}
+	ecx, _ := mu.RegRead(UC_X86_REG_ECX)
+	edx, _ := mu.RegRead(UC_X86_REG_EDX)
+	if ecx != 0x1234 || edx != 0x7890 {
+		t.Fatal("Bad register values.")
+	}
+}
+
+func TestX86InvalidWrite(t *testing.T) {
+	code := "\x89\x0D\xAA\xAA\xAA\xAA\x41\x4a"
+	mu, err := MakeUc(UC_MODE_32, code)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = mu.Start(ADDRESS, ADDRESS+uint64(len(code)))
+	if err.(UcError) != UC_ERR_MEM_WRITE {
+		t.Fatal("Expected UC_ERR_MEM_WRITE")
+	}
+	ecx, _ := mu.RegRead(UC_X86_REG_ECX)
+	edx, _ := mu.RegRead(UC_X86_REG_EDX)
+	if ecx != 0x1234 || edx != 0x7890 {
+		t.Fatal("Bad register values.")
+	}
+}
+
+func TestX86InOut(t *testing.T) {
+	code := "\x41\xE4\x3F\x4a\xE6\x46\x43"
+	mu, err := MakeUc(UC_MODE_32, code)
+	if err != nil {
+		t.Fatal(err)
+	}
+	var outVal uint64
+	var inCalled, outCalled bool
+	mu.HookAdd(UC_HOOK_INSN, func(mu *Uc, port, size uint32) uint32 {
+		inCalled = true
+		switch size {
+		case 1:
+			return 0xf1
+		case 2:
+			return 0xf2
+		case 4:
+			return 0xf4
+		default:
+			return 0
+		}
+	}, UC_X86_INS_IN)
+	mu.HookAdd(UC_HOOK_INSN, func(uc *Uc, port, size, value uint32) {
+		outCalled = true
+		var err error
+		switch size {
+		case 1:
+			outVal, err = mu.RegRead(UC_X86_REG_AL)
+		case 2:
+			outVal, err = mu.RegRead(UC_X86_REG_AX)
+		case 4:
+			outVal, err = mu.RegRead(UC_X86_REG_EAX)
+		}
+		if err != nil {
+			t.Fatal(err)
+		}
+	}, UC_X86_INS_OUT)
+	if err := mu.Start(ADDRESS, ADDRESS+uint64(len(code))); err != nil {
+		t.Fatal(err)
+	}
+	if !inCalled || !outCalled {
+		t.Fatal("Ports not accessed.")
+	}
+	if outVal != 0xf1 {
+		t.Fatal("Incorrect OUT value.")
+	}
+}
+
+func TestX86Syscall(t *testing.T) {
+	code := "\x0f\x05"
+	mu, err := MakeUc(UC_MODE_64, code)
+	if err != nil {
+		t.Fatal(err)
+	}
+	mu.HookAdd(UC_HOOK_INSN, func(mu *Uc) {
+		rax, _ := mu.RegRead(UC_X86_REG_RAX)
+		mu.RegWrite(UC_X86_REG_RAX, rax+1)
+	}, UC_X86_INS_SYSCALL)
+	mu.RegWrite(UC_X86_REG_RAX, 0x100)
+	err = mu.Start(ADDRESS, ADDRESS+uint64(len(code)))
+	if err != nil {
+		t.Fatal(err)
+	}
+	v, _ := mu.RegRead(UC_X86_REG_RAX)
+	if v != 0x101 {
+		t.Fatal("Incorrect syscall return value.")
+	}
+}