| // Copyright 2020 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package test |
| |
| import ( |
| "bufio" |
| "cmd/compile/internal/abi" |
| "cmd/compile/internal/base" |
| "cmd/compile/internal/ssagen" |
| "cmd/compile/internal/typecheck" |
| "cmd/compile/internal/types" |
| "cmd/internal/obj" |
| "cmd/internal/obj/x86" |
| "cmd/internal/src" |
| "fmt" |
| "os" |
| "testing" |
| ) |
| |
| // AMD64 registers available: |
| // - integer: RAX, RBX, RCX, RDI, RSI, R8, R9, r10, R11 |
| // - floating point: X0 - X14 |
| var configAMD64 = abi.NewABIConfig(9, 15, 0) |
| |
| func TestMain(m *testing.M) { |
| ssagen.Arch.LinkArch = &x86.Linkamd64 |
| ssagen.Arch.REGSP = x86.REGSP |
| ssagen.Arch.MAXWIDTH = 1 << 50 |
| types.MaxWidth = ssagen.Arch.MAXWIDTH |
| base.Ctxt = obj.Linknew(ssagen.Arch.LinkArch) |
| base.Ctxt.DiagFunc = base.Errorf |
| base.Ctxt.DiagFlush = base.FlushErrors |
| base.Ctxt.Bso = bufio.NewWriter(os.Stdout) |
| types.LocalPkg = types.NewPkg("", "local") |
| types.LocalPkg.Prefix = `""` |
| types.PtrSize = ssagen.Arch.LinkArch.PtrSize |
| types.RegSize = ssagen.Arch.LinkArch.RegSize |
| typecheck.InitUniverse() |
| os.Exit(m.Run()) |
| } |
| |
| func TestABIUtilsBasic1(t *testing.T) { |
| |
| // func(x int32) int32 |
| i32 := types.Types[types.TINT32] |
| ft := mkFuncType(nil, []*types.Type{i32}, []*types.Type{i32}) |
| |
| // expected results |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: int32 |
| OUT 0: R{ I0 } spilloffset: -1 typ: int32 |
| offsetToSpillArea: 0 spillAreaSize: 8 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsBasic2(t *testing.T) { |
| // func(p1 int8, p2 int16, p3 int32, p4 int64, |
| // p5 float32, p6 float32, p7 float64, p8 float64, |
| // p9 int8, p10 int16, p11 int32, p12 int64, |
| // p13 float32, p14 float32, p15 float64, p16 float64, |
| // p17 complex128, p18 complex128, p19 complex12, p20 complex128, |
| // p21 complex64, p22 int8, p23 in16, p24 int32, p25 int64, |
| // p26 int8, p27 in16, p28 int32, p29 int64) |
| // (r1 int32, r2 float64, r3 float64) { |
| i8 := types.Types[types.TINT8] |
| i16 := types.Types[types.TINT16] |
| i32 := types.Types[types.TINT32] |
| i64 := types.Types[types.TINT64] |
| f32 := types.Types[types.TFLOAT32] |
| f64 := types.Types[types.TFLOAT64] |
| c64 := types.Types[types.TCOMPLEX64] |
| c128 := types.Types[types.TCOMPLEX128] |
| ft := mkFuncType(nil, |
| []*types.Type{ |
| i8, i16, i32, i64, |
| f32, f32, f64, f64, |
| i8, i16, i32, i64, |
| f32, f32, f64, f64, |
| c128, c128, c128, c128, c64, |
| i8, i16, i32, i64, |
| i8, i16, i32, i64}, |
| []*types.Type{i32, f64, f64}) |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: int8 |
| IN 1: R{ I1 } spilloffset: 2 typ: int16 |
| IN 2: R{ I2 } spilloffset: 4 typ: int32 |
| IN 3: R{ I3 } spilloffset: 8 typ: int64 |
| IN 4: R{ F0 } spilloffset: 16 typ: float32 |
| IN 5: R{ F1 } spilloffset: 20 typ: float32 |
| IN 6: R{ F2 } spilloffset: 24 typ: float64 |
| IN 7: R{ F3 } spilloffset: 32 typ: float64 |
| IN 8: R{ I4 } spilloffset: 40 typ: int8 |
| IN 9: R{ I5 } spilloffset: 42 typ: int16 |
| IN 10: R{ I6 } spilloffset: 44 typ: int32 |
| IN 11: R{ I7 } spilloffset: 48 typ: int64 |
| IN 12: R{ F4 } spilloffset: 56 typ: float32 |
| IN 13: R{ F5 } spilloffset: 60 typ: float32 |
| IN 14: R{ F6 } spilloffset: 64 typ: float64 |
| IN 15: R{ F7 } spilloffset: 72 typ: float64 |
| IN 16: R{ F8 F9 } spilloffset: 80 typ: complex128 |
| IN 17: R{ F10 F11 } spilloffset: 96 typ: complex128 |
| IN 18: R{ F12 F13 } spilloffset: 112 typ: complex128 |
| IN 19: R{ } offset: 0 typ: complex128 |
| IN 20: R{ } offset: 16 typ: complex64 |
| IN 21: R{ I8 } spilloffset: 128 typ: int8 |
| IN 22: R{ } offset: 24 typ: int16 |
| IN 23: R{ } offset: 28 typ: int32 |
| IN 24: R{ } offset: 32 typ: int64 |
| IN 25: R{ } offset: 40 typ: int8 |
| IN 26: R{ } offset: 42 typ: int16 |
| IN 27: R{ } offset: 44 typ: int32 |
| IN 28: R{ } offset: 48 typ: int64 |
| OUT 0: R{ I0 } spilloffset: -1 typ: int32 |
| OUT 1: R{ F0 } spilloffset: -1 typ: float64 |
| OUT 2: R{ F1 } spilloffset: -1 typ: float64 |
| offsetToSpillArea: 56 spillAreaSize: 136 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsArrays(t *testing.T) { |
| // func(p1 [1]int32, p2 [0]int32, p3 [1][1]int32, p4 [2]int32) |
| // (r1 [2]int32, r2 [1]int32, r3 [0]int32, r4 [1][1]int32) { |
| i32 := types.Types[types.TINT32] |
| ae := types.NewArray(i32, 0) |
| a1 := types.NewArray(i32, 1) |
| a2 := types.NewArray(i32, 2) |
| aa1 := types.NewArray(a1, 1) |
| ft := mkFuncType(nil, []*types.Type{a1, ae, aa1, a2}, |
| []*types.Type{a2, a1, ae, aa1}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: [1]int32 |
| IN 1: R{ } offset: 0 typ: [0]int32 |
| IN 2: R{ I1 } spilloffset: 4 typ: [1][1]int32 |
| IN 3: R{ } offset: 0 typ: [2]int32 |
| OUT 0: R{ } offset: 8 typ: [2]int32 |
| OUT 1: R{ I0 } spilloffset: -1 typ: [1]int32 |
| OUT 2: R{ } offset: 16 typ: [0]int32 |
| OUT 3: R{ I1 } spilloffset: -1 typ: [1][1]int32 |
| offsetToSpillArea: 16 spillAreaSize: 8 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsStruct1(t *testing.T) { |
| // type s struct { f1 int8; f2 int8; f3 struct {}; f4 int8; f5 int16) } |
| // func(p1 int6, p2 s, p3 int64) |
| // (r1 s, r2 int8, r3 int32) { |
| i8 := types.Types[types.TINT8] |
| i16 := types.Types[types.TINT16] |
| i32 := types.Types[types.TINT32] |
| i64 := types.Types[types.TINT64] |
| s := mkstruct([]*types.Type{i8, i8, mkstruct([]*types.Type{}), i8, i16}) |
| ft := mkFuncType(nil, []*types.Type{i8, s, i64}, |
| []*types.Type{s, i8, i32}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: int8 |
| IN 1: R{ I1 I2 I3 I4 } spilloffset: 2 typ: struct { int8; int8; struct {}; int8; int16 } |
| IN 2: R{ I5 } spilloffset: 8 typ: int64 |
| OUT 0: R{ I0 I1 I2 I3 } spilloffset: -1 typ: struct { int8; int8; struct {}; int8; int16 } |
| OUT 1: R{ I4 } spilloffset: -1 typ: int8 |
| OUT 2: R{ I5 } spilloffset: -1 typ: int32 |
| offsetToSpillArea: 0 spillAreaSize: 16 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsStruct2(t *testing.T) { |
| // type s struct { f1 int64; f2 struct { } } |
| // type fs struct { f1 float64; f2 s; f3 struct { } } |
| // func(p1 s, p2 s, p3 fs) |
| // (r1 fs, r2 fs) |
| f64 := types.Types[types.TFLOAT64] |
| i64 := types.Types[types.TINT64] |
| s := mkstruct([]*types.Type{i64, mkstruct([]*types.Type{})}) |
| fs := mkstruct([]*types.Type{f64, s, mkstruct([]*types.Type{})}) |
| ft := mkFuncType(nil, []*types.Type{s, s, fs}, |
| []*types.Type{fs, fs}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: struct { int64; struct {} } |
| IN 1: R{ I1 } spilloffset: 16 typ: struct { int64; struct {} } |
| IN 2: R{ F0 I2 } spilloffset: 32 typ: struct { float64; struct { int64; struct {} }; struct {} } |
| OUT 0: R{ F0 I0 } spilloffset: -1 typ: struct { float64; struct { int64; struct {} }; struct {} } |
| OUT 1: R{ F1 I1 } spilloffset: -1 typ: struct { float64; struct { int64; struct {} }; struct {} } |
| offsetToSpillArea: 0 spillAreaSize: 64 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| // TestABIUtilsEmptyFieldAtEndOfStruct is testing to make sure |
| // the abi code is doing the right thing for struct types that have |
| // a trailing zero-sized field (where the we need to add padding). |
| func TestABIUtilsEmptyFieldAtEndOfStruct(t *testing.T) { |
| // type s struct { f1 [2]int64; f2 struct { } } |
| // type s2 struct { f1 [3]int16; f2 struct { } } |
| // type fs struct { f1 float64; f s; f3 struct { } } |
| // func(p1 s, p2 s, p3 fs) (r1 fs, r2 fs) |
| f64 := types.Types[types.TFLOAT64] |
| i64 := types.Types[types.TINT64] |
| i16 := types.Types[types.TINT16] |
| tb := types.Types[types.TBOOL] |
| ab2 := types.NewArray(tb, 2) |
| a2 := types.NewArray(i64, 2) |
| a3 := types.NewArray(i16, 3) |
| s := mkstruct([]*types.Type{a2, mkstruct([]*types.Type{})}) |
| s2 := mkstruct([]*types.Type{a3, mkstruct([]*types.Type{})}) |
| fs := mkstruct([]*types.Type{f64, s, mkstruct([]*types.Type{})}) |
| ft := mkFuncType(nil, []*types.Type{s, ab2, s2, fs, fs}, |
| []*types.Type{fs, ab2, fs}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ } offset: 0 typ: struct { [2]int64; struct {} } |
| IN 1: R{ } offset: 24 typ: [2]bool |
| IN 2: R{ } offset: 26 typ: struct { [3]int16; struct {} } |
| IN 3: R{ } offset: 40 typ: struct { float64; struct { [2]int64; struct {} }; struct {} } |
| IN 4: R{ } offset: 80 typ: struct { float64; struct { [2]int64; struct {} }; struct {} } |
| OUT 0: R{ } offset: 120 typ: struct { float64; struct { [2]int64; struct {} }; struct {} } |
| OUT 1: R{ } offset: 160 typ: [2]bool |
| OUT 2: R{ } offset: 168 typ: struct { float64; struct { [2]int64; struct {} }; struct {} } |
| offsetToSpillArea: 208 spillAreaSize: 0 |
| `) |
| |
| abitest(t, ft, exp) |
| |
| // Check to make sure that NumParamRegs yields 2 and not 3 |
| // for struct "s" (e.g. that it handles the padding properly). |
| nps := configAMD64.NumParamRegs(s) |
| if nps != 2 { |
| t.Errorf("NumParams(%v) returned %d expected %d\n", |
| s, nps, 2) |
| } |
| } |
| |
| func TestABIUtilsSliceString(t *testing.T) { |
| // func(p1 []int32, p2 int8, p3 []int32, p4 int8, p5 string, |
| // p6 int64, p6 []intr32) (r1 string, r2 int64, r3 string, r4 []int32) |
| i32 := types.Types[types.TINT32] |
| sli32 := types.NewSlice(i32) |
| str := types.Types[types.TSTRING] |
| i8 := types.Types[types.TINT8] |
| i64 := types.Types[types.TINT64] |
| ft := mkFuncType(nil, []*types.Type{sli32, i8, sli32, i8, str, i8, i64, sli32}, |
| []*types.Type{str, i64, str, sli32}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 I1 I2 } spilloffset: 0 typ: []int32 |
| IN 1: R{ I3 } spilloffset: 24 typ: int8 |
| IN 2: R{ I4 I5 I6 } spilloffset: 32 typ: []int32 |
| IN 3: R{ I7 } spilloffset: 56 typ: int8 |
| IN 4: R{ } offset: 0 typ: string |
| IN 5: R{ I8 } spilloffset: 57 typ: int8 |
| IN 6: R{ } offset: 16 typ: int64 |
| IN 7: R{ } offset: 24 typ: []int32 |
| OUT 0: R{ I0 I1 } spilloffset: -1 typ: string |
| OUT 1: R{ I2 } spilloffset: -1 typ: int64 |
| OUT 2: R{ I3 I4 } spilloffset: -1 typ: string |
| OUT 3: R{ I5 I6 I7 } spilloffset: -1 typ: []int32 |
| offsetToSpillArea: 48 spillAreaSize: 64 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsMethod(t *testing.T) { |
| // type s1 struct { f1 int16; f2 int16; f3 int16 } |
| // func(p1 *s1, p2 [7]*s1, p3 float64, p4 int16, p5 int16, p6 int16) |
| // (r1 [7]*s1, r2 float64, r3 int64) |
| i16 := types.Types[types.TINT16] |
| i64 := types.Types[types.TINT64] |
| f64 := types.Types[types.TFLOAT64] |
| s1 := mkstruct([]*types.Type{i16, i16, i16}) |
| ps1 := types.NewPtr(s1) |
| a7 := types.NewArray(ps1, 7) |
| ft := mkFuncType(s1, []*types.Type{ps1, a7, f64, i16, i16, i16}, |
| []*types.Type{a7, f64, i64}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 I1 I2 } spilloffset: 0 typ: struct { int16; int16; int16 } |
| IN 1: R{ I3 } spilloffset: 8 typ: *struct { int16; int16; int16 } |
| IN 2: R{ } offset: 0 typ: [7]*struct { int16; int16; int16 } |
| IN 3: R{ F0 } spilloffset: 16 typ: float64 |
| IN 4: R{ I4 } spilloffset: 24 typ: int16 |
| IN 5: R{ I5 } spilloffset: 26 typ: int16 |
| IN 6: R{ I6 } spilloffset: 28 typ: int16 |
| OUT 0: R{ } offset: 56 typ: [7]*struct { int16; int16; int16 } |
| OUT 1: R{ F0 } spilloffset: -1 typ: float64 |
| OUT 2: R{ I0 } spilloffset: -1 typ: int64 |
| offsetToSpillArea: 112 spillAreaSize: 32 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABIUtilsInterfaces(t *testing.T) { |
| // type s1 { f1 int16; f2 int16; f3 bool) |
| // type nei interface { ...() string } |
| // func(p1 s1, p2 interface{}, p3 interface{}, p4 nei, |
| // p5 *interface{}, p6 nei, p7 int64) |
| // (r1 interface{}, r2 nei, r3 bool) |
| ei := types.Types[types.TINTER] // interface{} |
| pei := types.NewPtr(ei) // *interface{} |
| fldt := mkFuncType(types.FakeRecvType(), []*types.Type{}, |
| []*types.Type{types.Types[types.TSTRING]}) |
| field := types.NewField(src.NoXPos, typecheck.Lookup("F"), fldt) |
| nei := types.NewInterface(types.LocalPkg, []*types.Field{field}, false) |
| i16 := types.Types[types.TINT16] |
| tb := types.Types[types.TBOOL] |
| s1 := mkstruct([]*types.Type{i16, i16, tb}) |
| ft := mkFuncType(nil, []*types.Type{s1, ei, ei, nei, pei, nei, i16}, |
| []*types.Type{ei, nei, pei}) |
| |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 I1 I2 } spilloffset: 0 typ: struct { int16; int16; bool } |
| IN 1: R{ I3 I4 } spilloffset: 8 typ: interface {} |
| IN 2: R{ I5 I6 } spilloffset: 24 typ: interface {} |
| IN 3: R{ I7 I8 } spilloffset: 40 typ: interface { F() string } |
| IN 4: R{ } offset: 0 typ: *interface {} |
| IN 5: R{ } offset: 8 typ: interface { F() string } |
| IN 6: R{ } offset: 24 typ: int16 |
| OUT 0: R{ I0 I1 } spilloffset: -1 typ: interface {} |
| OUT 1: R{ I2 I3 } spilloffset: -1 typ: interface { F() string } |
| OUT 2: R{ I4 } spilloffset: -1 typ: *interface {} |
| offsetToSpillArea: 32 spillAreaSize: 56 |
| `) |
| |
| abitest(t, ft, exp) |
| } |
| |
| func TestABINumParamRegs(t *testing.T) { |
| i8 := types.Types[types.TINT8] |
| i16 := types.Types[types.TINT16] |
| i32 := types.Types[types.TINT32] |
| i64 := types.Types[types.TINT64] |
| f32 := types.Types[types.TFLOAT32] |
| f64 := types.Types[types.TFLOAT64] |
| c64 := types.Types[types.TCOMPLEX64] |
| c128 := types.Types[types.TCOMPLEX128] |
| |
| s := mkstruct([]*types.Type{i8, i8, mkstruct([]*types.Type{}), i8, i16}) |
| a := types.NewArray(s, 3) |
| |
| nrtest(t, i8, 1) |
| nrtest(t, i16, 1) |
| nrtest(t, i32, 1) |
| nrtest(t, i64, 1) |
| nrtest(t, f32, 1) |
| nrtest(t, f64, 1) |
| nrtest(t, c64, 2) |
| nrtest(t, c128, 2) |
| nrtest(t, s, 4) |
| nrtest(t, a, 12) |
| |
| } |
| |
| func TestABIUtilsComputePadding(t *testing.T) { |
| // type s1 { f1 int8; f2 int16; f3 struct{}; f4 int32; f5 int64 } |
| i8 := types.Types[types.TINT8] |
| i16 := types.Types[types.TINT16] |
| i32 := types.Types[types.TINT32] |
| i64 := types.Types[types.TINT64] |
| emptys := mkstruct([]*types.Type{}) |
| s1 := mkstruct([]*types.Type{i8, i16, emptys, i32, i64}) |
| // func (p1 int32, p2 s1, p3 emptys, p4 [1]int32) |
| a1 := types.NewArray(i32, 1) |
| ft := mkFuncType(nil, []*types.Type{i32, s1, emptys, a1}, []*types.Type{}) |
| |
| // Run abitest() just to document what we're expected to see. |
| exp := makeExpectedDump(` |
| IN 0: R{ I0 } spilloffset: 0 typ: int32 |
| IN 1: R{ I1 I2 I3 I4 } spilloffset: 8 typ: struct { int8; int16; struct {}; int32; int64 } |
| IN 2: R{ } offset: 0 typ: struct {} |
| IN 3: R{ I5 } spilloffset: 24 typ: [1]int32 |
| offsetToSpillArea: 0 spillAreaSize: 32 |
| `) |
| abitest(t, ft, exp) |
| |
| // Analyze with full set of registers, then call ComputePadding |
| // on the second param, verifying the results. |
| regRes := configAMD64.ABIAnalyze(ft, false) |
| padding := make([]uint64, 32) |
| parm := regRes.InParams()[1] |
| padding = parm.ComputePadding(padding) |
| want := "[1 1 1 0]" |
| got := fmt.Sprintf("%+v", padding) |
| if got != want { |
| t.Errorf("padding mismatch: wanted %q got %q\n", got, want) |
| } |
| } |