github.com/cloudwego/frugal@v0.1.7/internal/loader/loader_amd64_test.go

/*
 * Copyright 2022 ByteDance Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package loader

import (
    `fmt`
    `reflect`
    `runtime`
    `testing`
    `unsafe`

    `github.com/chenzhuoyu/iasm/x86_64`
    `github.com/cloudwego/frugal/internal/rt`
    `github.com/stretchr/testify/assert`
    `github.com/stretchr/testify/require`
    `golang.org/x/arch/x86/x86asm`
)

type funcInfo struct {
    *_Func
    datap *_ModuleData
}

func (self funcInfo) entry() uintptr {
    if runtime.Version() <= "go1.17" {
        return *(*uintptr)(unsafe.Pointer(self._Func))
    }
    off := uintptr(*(*uint32)(unsafe.Pointer(self._Func)))
    off += self.datap.text
    return off
}

//go:linkname findfunc runtime.findfunc
//goland:noinspection GoUnusedParameter
func findfunc(pc uintptr) funcInfo

//go:linkname pcdatavalue2 runtime.pcdatavalue2
//goland:noinspection GoUnusedParameter
func pcdatavalue2(f funcInfo, table uint32, targetpc uintptr) (int32, uintptr)

func TestLoader_Load(t *testing.T) {
    var src string
    var asm x86_64.Assembler
    if runtime.Version() < "go1.17" { src += `
        movq 8(%rsp), %rax`
    }
    src += `
        movq $1234, (%rax)
        ret`
    require.NoError(t, asm.Assemble(src))
    v0 := 0
    cc := asm.Code()
    fp := Loader(cc).Load("test", rt.Frame{})
    (*(*func(*int))(unsafe.Pointer(&fp)))(&v0)
    pc := *(*uintptr)(fp)
    assert.Equal(t, 1234, v0)
    assert.Equal(t, fmt.Sprintf("(frugal).test_%x", pc), runtime.FuncForPC(pc).Name())
    file, line := runtime.FuncForPC(pc).FileLine(pc + 1)
    assert.Equal(t, "(jit-generated)", file)
    assert.Equal(t, 1, line)
    smi, startpc := pcdatavalue2(findfunc(pc), _PCDATA_StackMapIndex, pc + uintptr(len(cc)) - 1)
    assert.Equal(t, int32(0), smi)
    assert.Equal(t, pc, startpc)
    aup, startpc2 := pcdatavalue2(findfunc(pc), _PCDATA_UnsafePoint, pc + uintptr(len(cc)) - 1)
    assert.Equal(t, int32(_PCDATA_UnsafePointUnsafe), aup)
    assert.Equal(t, pc, startpc2)
}

func mkpointer() *int {
    ret := new(int)
    *ret = 1234
    runtime.SetFinalizer(ret, func(_ *int) {
        println("ret has been recycled")
    })
    println("ret is allocated")
    return ret
}

func collect() {
    println("start collecting")
    for i := 1; i < 1000; i++ {
        runtime.GC()
    }
    println("done collecting")
}

func TestLoader_StackMap(t *testing.T) {
    var asm x86_64.Assembler
    var smb rt.StackMapBuilder
    src := `
        subq    $24, %rsp
        movq    %rbp, 16(%rsp)
        leaq    16(%rsp), %rbp
        
        movq    $` + fmt.Sprintf("%p", mkpointer) + `, %r12
        callq   %r12`
    if runtime.Version() < "go1.17" { src += `
        movq    (%rsp), %rax`
    }
    src += `
        movq    %rax, 8(%rsp)
        movq    $0x123, (%rsp)
        movq    $` + fmt.Sprintf("%p", collect) + `, %r12
        callq   %r12
        movq    16(%rsp), %rbp
        addq    $24, %rsp
        ret`
    require.NoError(t, asm.Assemble(src))
    smb.AddField(true)
    cc := asm.Code()
    fp := Loader(cc).Load("test_with_stackmap", rt.Frame {
        SpTab: []rt.Stack {
            { Sp:  0, Nb: 4 },
            { Sp: 24, Nb: uintptr(len(cc) - 5) },
            { Sp:  0, Nb: 0 },
        },
        ArgSize   : 0,
        ArgPtrs   : new(rt.StackMapBuilder).Build(),
        LocalPtrs : smb.Build(),
    })
    dumpfunction(*(*func())(unsafe.Pointer(&fp)))
    println("enter function")
    (*(*func())(unsafe.Pointer(&fp)))()
    println("leave function")
    collect()
}

//go:linkname step runtime.step
//goland:noinspection GoUnusedParameter
func step(p []byte, pc *uintptr, val *int32, first bool) (newp []byte, ok bool)

func dumpfunction(f interface{}) {
    fp := rt.FuncAddr(f)
    fn := findfunc(uintptr(fp))
    var name string
    if runtime.Version() >= "go1.16" {
        name = "pctab"
    } else {
        name = "pclntable"
    }
    datap := reflect.ValueOf(fn.datap)
    ff, ok := datap.Type().Elem().FieldByName(name)
    if !ok {
        panic("no such field: pctab")
    }
    p := (*(*[]byte)(unsafe.Pointer(uintptr(unsafe.Pointer(fn.datap)) + ff.Offset)))[fn.pcsp:]
    pc := fn.entry()
    val := int32(-1)
    lastpc := uintptr(0)
    for {
        var ok bool
        lastpc = pc
        p, ok = step(p, &pc, &val, pc == fn.entry())
        if !ok {
            break
        }
        fmt.Printf("%#x = %#x\n", lastpc, val)
    }
    pc = 0
    lastpc -= fn.entry()
    for pc <= lastpc {
        pp := unsafe.Pointer(uintptr(fp) + pc)
        fx := runtime.FuncForPC(uintptr(pp))
        if fx.Name() != "" && fx.Entry() == uintptr(pp) {
            println("----", fx.Name(), "----")
        }
        ins, err := x86asm.Decode(rt.BytesFrom(pp, 15, 15), 64)
        if err != nil {
            panic(err)
        }
        fmt.Printf("%#x %s\n", uintptr(pp), x86asm.GNUSyntax(ins, uint64(uintptr(pp)), func(u uint64) (string, uint64) {
            v := runtime.FuncForPC(uintptr(u))
            if v == nil {
                return "", 0
            }
            return v.Name(), uint64(v.Entry())
        }))
        pc += uintptr(ins.Len)
    }
}

func TestLoader_PCSPDelta(t *testing.T) {
    dumpfunction(moduledataverify1)
}