Function TestFunctionAssembly

internal/binutils/disasm_test.go:104–160

TestFunctionAssembly tests the FunctionAssembly routine by using a fake objdump script.

(t *testing.T)

// TestFunctionAssembly tests the FunctionAssembly routine by using a
// fake objdump script.
func TestFunctionAssembly(t *testing.T) {
	type testcase struct {
		s    plugin.Sym
		asm  string
		want []plugin.Inst
	}
	testcases := []testcase{
		{
			plugin.Sym{Name: []string{"symbol1"}, Start: 0x1000, End: 0x1FFF},
			"  1000: instruction one\n  1001: instruction two\n  1002: instruction three\n  1003: instruction four",
			[]plugin.Inst{
				{Addr: 0x1000, Text: "instruction one"},
				{Addr: 0x1001, Text: "instruction two"},
				{Addr: 0x1002, Text: "instruction three"},
				{Addr: 0x1003, Text: "instruction four"},
			},
		},
		{
			plugin.Sym{Name: []string{"symbol2"}, Start: 0x2000, End: 0x2FFF},
			"  2000: instruction one\n  2001: instruction two",
			[]plugin.Inst{
				{Addr: 0x2000, Text: "instruction one"},
				{Addr: 0x2001, Text: "instruction two"},
			},
		},
		{
			plugin.Sym{Name: []string{"_main"}, Start: 0x30000, End: 0x3FFF},
			"_main:\n; /tmp/hello.c:3\n30001:	push   %rbp",
			[]plugin.Inst{
				{Addr: 0x30001, Text: "push   %rbp", Function: "_main", File: "/tmp/hello.c", Line: 3},
			},
		},
		{
			plugin.Sym{Name: []string{"main"}, Start: 0x4000, End: 0x4FFF},
			"000000000040052d <main>:\nmain():\n/tmp/hello.c:3\n40001:	push   %rbp",
			[]plugin.Inst{
				{Addr: 0x40001, Text: "push   %rbp", Function: "main", File: "/tmp/hello.c", Line: 3},
			},
		},
	}

	for _, tc := range testcases {
		insts, err := disassemble([]byte(tc.asm))
		if err != nil {
			t.Fatalf("FunctionAssembly: %v", err)
		}

		if len(insts) != len(tc.want) {
			t.Errorf("Unexpected number of assembly instructions %d (want %d)\n", len(insts), len(tc.want))
		}
		for i := range insts {
			if insts[i] != tc.want[i] {
				t.Errorf("Expected symbol %v, got %v\n", tc.want[i], insts[i])
			}
		}
	}
}