compress the extracted information

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

@@ -1,7 +1,8 @@
 package action
 
 import (
-	"fmt"
+	// "fmt"
+	"sort"
 	// log "github.com/sirupsen/logrus"
 
 	. "ios-wrapper/internal/wrapper/ofile"
@@ -11,25 +12,46 @@ import (
 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
-	}
+	// 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() {
+	// symbols_storage := []*protomodel.MachoInfo_AllImportedSymbols{}
+	symbols_raw := mc.CollectBindSymbols()
+	sort.Slice(symbols_raw, func(i, j int) bool {
+		orderedByLibrary := symbols_raw[i].Dylib() < symbols_raw[j].Dylib()
+		if symbols_raw[i].Dylib() == symbols_raw[j].Dylib() {
+			orderedBySymbol := symbols_raw[i].Name() < symbols_raw[j].Name()
+			return orderedBySymbol
+		}
+		return orderedByLibrary
+	})
+
+	libs := []string{}
+	symbols := []string{}
+	tables := []*protomodel.MachoInfo_LibraryImportedSymbols{}
+
+	var current_table *protomodel.MachoInfo_LibraryImportedSymbols
+
+	current_lib := ""
+	current_symbol := ""
+
+	current_lib_idx := -1
+	current_symbol_idx := -1
+
+	// now we expect everything is sorted and easier to build strings tables
+	// this is not fully optimized, there can be repeated symbol name in different libraries
+	for _, symbol := range symbols_raw {
 		if symbol.Type() != "lazy" {
 			continue
 		}
-		dylib_hash := calculateHash(symbol.Dylib())
+		// 	dylib_hash := calculateHash(symbol.Dylib())
 		seg := mc.Segments()[symbol.Segment()]
 
 		var offset uint64
@@ -42,22 +64,48 @@ func (action *saveImports) withMacho(mf *MachoFile) error {
 			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,
+		if current_lib != symbol.Dylib() {
+			current_lib_idx += len(current_lib) + 1
+			current_lib = symbol.Dylib()
+			libs = append(libs, symbol.Dylib())
+			tables = append(tables, &protomodel.MachoInfo_LibraryImportedSymbols{
+				LibIndex: uint32(current_lib_idx),
+				Nsymbols: 0,
+				Symbols:  []*protomodel.MachoInfo_SymbolTable{},
 			})
+			current_table = tables[len(tables)-1]
+		}
+		if current_symbol != symbol.Name() {
+			current_symbol_idx += len(current_symbol) + 1
+			current_symbol = symbol.Name()
+			symbols = append(symbols, symbol.Name())
+		}
+		current_table.Nsymbols += 1
+		current_table.Symbols = append(current_table.Symbols, &protomodel.MachoInfo_SymbolTable{
+			SymbolIndex:  uint32(current_symbol_idx),
+			SegmentIndex: symbol.Segment(),
+			Offset:       uint32(offset),
+		})
 
+		// fmt.Printf("{\"%s\", \"%s\", 0x%x, 0x%x, 0x%x},\n",
+		// 	symbol.Name(), symbol.Dylib(), 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
+	mf.Info().Symbols = &protomodel.MachoInfo_AllImportedSymbols{
+		Libs:    libs,
+		Symbols: symbols,
+		Tables:  tables,
+	}
+
+	mf.Info().Main = mc.Main()
 	return nil
 }