944133de6e
llcppsymg:symbo test llcppsymg:exist symb file test llcppsymg:GenSymbolTabledata llcppsymg:GenSymbolTableData test llcppsymg:full symg operation test
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
package symbol
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"strings"
|
|
"unsafe"
|
|
|
|
"github.com/goplus/llgo/c"
|
|
"github.com/goplus/llgo/c/cjson"
|
|
"github.com/goplus/llgo/chore/_xtool/llcppsymg/config"
|
|
"github.com/goplus/llgo/chore/_xtool/llcppsymg/parse"
|
|
"github.com/goplus/llgo/chore/llcppg/types"
|
|
"github.com/goplus/llgo/xtool/nm"
|
|
)
|
|
|
|
// finds the intersection of symbols from the dynamic library's symbol table and the symbols parsed from header files.
|
|
// It returns a list of symbols that can be externally linked.
|
|
func GetCommonSymbols(dylibSymbols []*nm.Symbol, headerSymbols map[string]*parse.SymbolInfo) []*types.SymbolInfo {
|
|
var commonSymbols []*types.SymbolInfo
|
|
for _, dylibSym := range dylibSymbols {
|
|
symName := strings.TrimPrefix(dylibSym.Name, "_")
|
|
if symInfo, ok := headerSymbols[symName]; ok {
|
|
symbolInfo := &types.SymbolInfo{
|
|
Mangle: symName,
|
|
CPP: symInfo.ProtoName,
|
|
Go: symInfo.GoName,
|
|
}
|
|
commonSymbols = append(commonSymbols, symbolInfo)
|
|
}
|
|
}
|
|
return commonSymbols
|
|
}
|
|
|
|
func ReadExistingSymbolTable(fileName string) (map[string]types.SymbolInfo, bool) {
|
|
if _, err := os.Stat(fileName); err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
data, err := os.ReadFile(fileName)
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
parsedJSON := cjson.ParseBytes(data)
|
|
if parsedJSON == nil {
|
|
return nil, false
|
|
}
|
|
|
|
existingSymbols := make(map[string]types.SymbolInfo)
|
|
arraySize := parsedJSON.GetArraySize()
|
|
|
|
for i := 0; i < int(arraySize); i++ {
|
|
item := parsedJSON.GetArrayItem(c.Int(i))
|
|
symbol := types.SymbolInfo{
|
|
Mangle: config.GetStringItem(item, "mangle", ""),
|
|
CPP: config.GetStringItem(item, "c++", ""),
|
|
Go: config.GetStringItem(item, "go", ""),
|
|
}
|
|
existingSymbols[symbol.Mangle] = symbol
|
|
}
|
|
|
|
return existingSymbols, true
|
|
}
|
|
|
|
func GenSymbolTableData(commonSymbols []*types.SymbolInfo, existingSymbols map[string]types.SymbolInfo) ([]byte, error) {
|
|
// todo(zzy): len(existingSymbols) !=0
|
|
// https://github.com/goplus/llgo/issues/808 will cause unexpected panic
|
|
// https://github.com/goplus/llgo/pull/793 this pr can fix it
|
|
for i := range commonSymbols {
|
|
if existingSymbol, exists := existingSymbols[commonSymbols[i].Mangle]; exists {
|
|
commonSymbols[i].Go = existingSymbol.Go
|
|
}
|
|
}
|
|
|
|
root := cjson.Array()
|
|
defer root.Delete()
|
|
|
|
for _, symbol := range commonSymbols {
|
|
item := cjson.Object()
|
|
item.SetItem(c.Str("mangle"), cjson.String(c.AllocaCStr(symbol.Mangle)))
|
|
item.SetItem(c.Str("c++"), cjson.String(c.AllocaCStr(symbol.CPP)))
|
|
item.SetItem(c.Str("go"), cjson.String(c.AllocaCStr(symbol.Go)))
|
|
root.AddItem(item)
|
|
}
|
|
|
|
cStr := root.Print()
|
|
if cStr == nil {
|
|
return nil, errors.New("symbol table is empty")
|
|
}
|
|
defer c.Free(unsafe.Pointer(cStr))
|
|
result := []byte(c.GoString(cStr))
|
|
return result, nil
|
|
}
|
|
|
|
func GenerateAndUpdateSymbolTable(symbols []*nm.Symbol, headerInfos map[string]*parse.SymbolInfo, symbFile string) ([]byte, error) {
|
|
commonSymbols := GetCommonSymbols(symbols, headerInfos)
|
|
|
|
existSymbols, _ := ReadExistingSymbolTable(symbFile)
|
|
|
|
symbolData, err := GenSymbolTableData(commonSymbols, existSymbols)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return symbolData, nil
|
|
}
|