macho/macho-go/internal/wrapper/action/save_imports.go

package action

import (
	"fmt"
	// log "github.com/sirupsen/logrus"

	. "ios-wrapper/internal/wrapper/ofile"
	"ios-wrapper/pkg/protomodel"
)

type saveImports struct{}

func (action *saveImports) withMacho(mf *MachoFile) error {
	calculateHash := func(name string) uint32 {
		var h uint32 = 0x811c9dc5
		for _, s := range name {
			h ^= uint32(s)
			h *= 0x01000193
		}
		return h
	}

	mc := mf.Context()
	symbols := []*protomodel.MachoInfo_BindSymbol{}
	fmt.Println("struct imported_symbol {const char* name; const char* lib; uint32_t hash; int segment_i; uint64_t offset;};")
	fmt.Println("const char* lib_to_resolve = \"main\";")
	fmt.Println("struct imported_symbol imported_table[] = {")
	for _, symbol := range mc.CollectBindSymbols() {
		if symbol.Type() != "lazy" {
			continue
		}
		dylib_hash := calculateHash(symbol.Dylib())
		seg := mc.Segments()[symbol.Segment()]

		var offset uint64

		if symbol.Address() >= seg.Vmaddr() {
			// this is virtual address
			offset = symbol.Address() - seg.Vmaddr()
		} else {
			// this is file address
			offset = symbol.Address() - seg.Fileoff()
		}

		fmt.Printf("{\"%s\", \"%s\", 0x%x, 0x%x, 0x%x},\n",
			symbol.Name(), symbol.Dylib(), dylib_hash, symbol.Segment(), offset)

		symbols = append(symbols,
			&protomodel.MachoInfo_BindSymbol{
				Name:    symbol.Name(),
				Libname: symbol.Dylib(),
				Libhash: dylib_hash,
				Segment: symbol.Segment(),
				Offset:  offset,
			})

	}
	fmt.Println("};")
	fmt.Printf("uint32_t nimports = %d;\n", len(symbols))
	mf.Info().Symbols = symbols
	return nil
}

func (action *saveImports) withFat(ff *FatFile) error {
	return defaultWithFat(action, ff)
}

func NewSaveImportsAction() *saveImports {
	return &saveImports{}
}