luago/main.go

// Author: simon
// Author: ynwdlxm@163.com
// Date: 2024/4/13 17:14
// Desc:

package main

import (
    "fmt"
    "luago/binchunk"
    "os"
)

func main() {
    if len(os.Args) > 1 {
        data, err := os.ReadFile(os.Args[1])
        if err != nil {
            panic(err)
        }
        proto := binchunk.Undump(data)
        list(proto)

    }
}

func list(f *binchunk.Prototype) {
    printHeader(f)
    printCode(f)
    printDetail(f)
    for _, p := range f.Protos {
        list(p)
    }
}

// 打印常量表、局部变量表和 Upvalue 表
func printDetail(f *binchunk.Prototype) {
    fmt.Printf("constants (%d):\n", len(f.Constants))
    for i, k := range f.Constants {
        fmt.Printf("\t%d\t%s\n", i+1, constantToString(k))
    }

    fmt.Printf("locals (%d):\n", len(f.LocVars))
    for i, locVar := range f.LocVars {
        fmt.Printf("\t%d\t%s\t%d\t%d\n", i, locVar.VarName, locVar.StartPC+1, locVar.EndPC+1)
    }

    fmt.Printf("upvalues (%d):\n", len(f.Upvalues))
    for i, upval := range f.Upvalues {
        fmt.Printf("\t%d\t%s\t%d\t%d\n", i, upvalName(f, i), upval.Instack, upval.Idx)
    }

}

// 根据 Upvalue 索引从调试信息里找出 Upvalue 的名字
func upvalName(f *binchunk.Prototype, idx int) string {
    if len(f.UpvalueNames) > 0 {
        return f.UpvalueNames[idx]
    }
    return "-"
}

func constantToString(k interface{}) string {
    switch k.(type) {
    case nil:
        return "nil"
    case bool:
        return fmt.Sprintf("%t", k)
    case float64:
        return fmt.Sprintf("%g", k)
    case int64:
        return fmt.Sprintf("%d", k)
    case string:
        return fmt.Sprintf("%q", k)
    default:
        return "?"
    }
}

func printCode(f *binchunk.Prototype) {
    for pc, c := range f.Code {
        line := "-"
        if len(f.LineInfo) > 0 {
            line = fmt.Sprintf("%d", f.LineInfo[pc])
        }
        // 序号、行号和 十六进制表示
        fmt.Printf("\t%d\t[%s]\t0X%08X\n", pc+1, line, c)
    }
}

func printHeader(f *binchunk.Prototype) {
    funcType := "main"
    if f.LineDefined > 0 {
        funcType = "function"
    }

    varargFlag := ""
    if f.IsVararg > 0 {
        varargFlag = "+"
    }

    fmt.Printf("\n%s <%s:%d, %d> (%d instructions)\n", funcType, f.Source, f.LineDefined, f.LastLineDefined, len(f.Code))

    fmt.Printf("%d%s params, %d slots, %d upvalues, ", f.NumParams, varargFlag, f.MaxStackSize, len(f.Upvalues))

    fmt.Printf("%d locals, %d constants, %d functions\n", len(f.LocVars), len(f.Constants), len(f.Protos))
}