erase objc method names

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

@@ -68,7 +68,7 @@ func (action *saveImports) saveToInfo(mf *MachoFile) error {
 	// 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" {
+		if !symbol.SafeForRemoval() {
 			continue
 		}
 
@@ -138,6 +138,16 @@ func (action *saveImports) saveToInfo(mf *MachoFile) error {
 	}
 
 	mf.Info().Main = mc.Main()
+
+  selectors_list := []*protomodel.MachoInfo_Selector{}
+  for _, sel := range mc.CollectSpecialSelectors() {
+    selectors_list = append(selectors_list, &protomodel.MachoInfo_Selector{
+      Idx: uint32(sel.Idx()),
+      Name: sel.Name(),
+    })
+  }
+  mf.Info().SpecialSelectors = selectors_list
+
 	return nil
 }
 

macho-go/internal/wrapper/cli.go

@@ -299,6 +299,21 @@ func bcell2header(bfile string, header string) {
 		}
 		fmt.Fprintf(w, "};\n")
 		fmt.Fprintf(w, "uint32_t n_instructions = %d;\n", n_instructions)
+
+		fmt.Fprintf(w, "__attribute__((section(\"__DATA,bshield\")))\n")
+		fmt.Fprintf(w, "uint32_t special_selectors_idx[] = {\n")
+    for _, selector := range info.GetSpecialSelectors() {
+      fmt.Fprintf(w, "%x,\n", selector.Idx)
+    }
+		fmt.Fprintf(w, "};\n")
+
+		fmt.Fprintf(w, "__attribute__((section(\"__DATA,bshield\")))\n")
+		fmt.Fprintf(w, "char* special_selectors_name[] = {\n")
+    for _, selector := range info.GetSpecialSelectors() {
+      fmt.Fprintf(w, "\"%s\",\n", selector.Name)
+    }
+		fmt.Fprintf(w, "};\n")
+		fmt.Fprintf(w, "uint32_t n_selectors = %d;\n", len(info.GetSpecialSelectors()))
 	}
 	fmt.Fprintf(w, "}// namespace bshield_data\n")
 	w.Flush()