wehub-resource-sync
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package mcp
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"loov.dev/lensm/internal/comments"
|
|
"loov.dev/lensm/internal/disasm"
|
|
)
|
|
|
|
type Session struct {
|
|
Path string
|
|
File disasm.File
|
|
Comments *comments.Store
|
|
}
|
|
|
|
// LoadFile opens a binary for disassembly. The caller injects an
|
|
// implementation so sessions stay independent of the object-file formats.
|
|
type LoadFile func(path string) (disasm.File, error)
|
|
|
|
func NewSessionWithComments(load LoadFile, path string, commentsPath string, store *comments.Store) (*Session, error) {
|
|
file, err := load(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if store == nil {
|
|
if commentsPath == "" {
|
|
commentsPath = comments.DefaultPath(path)
|
|
}
|
|
var err error
|
|
store, err = comments.Open(commentsPath, path)
|
|
if err != nil {
|
|
_ = file.Close()
|
|
return nil, err
|
|
}
|
|
}
|
|
return &Session{
|
|
Path: comments.CleanPath(path),
|
|
File: file,
|
|
Comments: store,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Session) Close() error {
|
|
if s == nil || s.File == nil {
|
|
return nil
|
|
}
|
|
return s.File.Close()
|
|
}
|
|
|
|
func (s *Session) Funcs() []disasm.Func {
|
|
if s == nil || s.File == nil {
|
|
return nil
|
|
}
|
|
return s.File.Funcs()
|
|
}
|
|
|
|
func (s *Session) FindFunc(name string) disasm.Func {
|
|
for _, fn := range s.Funcs() {
|
|
if fn.Name() == name {
|
|
return fn
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Session) LoadCode(name string, context int) (*disasm.Code, error) {
|
|
fn := s.FindFunc(name)
|
|
if fn == nil {
|
|
return nil, fmt.Errorf("function %q not found", name)
|
|
}
|
|
return fn.Load(disasm.Options{Context: context})
|
|
}
|