when set _func.deferreturn check isDirectCall

Author: pkujhd

link/deferreturn.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/pkujhd/goloader/constants"
 	"github.com/pkujhd/goloader/objabi/dataindex"
+	"github.com/pkujhd/goloader/objabi/reloctype"
 )
 
 func (linker *Linker) addDeferReturn(_func *_func, module *moduledata) (err error) {
@@ -17,7 +18,7 @@ func (linker *Linker) addDeferReturn(_func *_func, module *moduledata) (err erro
 	if Func != nil && len(Func.FuncData) > dataindex.FUNCDATA_OpenCodedDeferInfo {
 		sym := linker.SymMap[funcName]
 		for _, r := range sym.Reloc {
-			if r.SymName == constants.RuntimeDeferReturn {
+			if r.SymName == constants.RuntimeDeferReturn && reloctype.IsDirectCall(r.Type) {
 				//../cmd/link/internal/ld/pcln.go:pclntab
 				switch linker.Arch.Name {
 				case sys.Arch386.Name, sys.ArchAMD64.Name:

link/relocate.go

@@ -328,7 +328,7 @@ func (linker *Linker) relocate(codeModule *CodeModule, symbolMap, symPtr map[str
 					reloctype.R_ARM64_PCREL_LDST8, reloctype.R_ARM64_PCREL_LDST16,
 					reloctype.R_ARM64_PCREL_LDST32, reloctype.R_ARM64_PCREL_LDST64:
 					if symkind.IsText(symbol.Kind) {
-						err = fmt.Errorf("impossible!Sym:%s locate not in code segment!\n", loc.SymName)
+						return fmt.Errorf("impossible!Sym:%s locate not in code segment!\n", loc.SymName)
 					}
 					err = linker.relocateADRP(relocByte[loc.Offset:], loc, segment, symAddr)
 				case reloctype.R_ADDR, reloctype.R_WEAKADDR:
@@ -346,7 +346,7 @@ func (linker *Linker) relocate(codeModule *CodeModule, symbolMap, symPtr map[str
 						if sym, ok := linker.SymMap[loc.SymName]; ok {
 							offset = sym.Offset + loc.Add
 						} else {
-							err = fmt.Errorf("symName:%s relocateType:%s, offset:%d is overflow!\n", loc.SymName, reloctype.RelocTypeString(loc.Type), offset)
+							return fmt.Errorf("symName:%s relocateType:%s, offset:%d is overflow!\n", loc.SymName, reloctype.RelocTypeString(loc.Type), offset)
 						}
 					}
 					if symkind.IsROData(linker.SymMap[loc.SymName].Kind) || symkind.IsNoPtrData(linker.SymMap[loc.SymName].Kind) {

objabi/reloctype/reloctype.go

@@ -0,0 +1,9 @@
+package reloctype
+
+func IsDirectCall(r int) bool {
+	switch r {
+	case R_CALL, R_CALLARM, R_CALLARM64:
+		return true
+	}
+	return false
+}