86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
package printer
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/mmcloughlin/avo"
|
|
"github.com/mmcloughlin/avo/internal/prnt"
|
|
"github.com/mmcloughlin/avo/operand"
|
|
)
|
|
|
|
// dot is the pesky unicode dot used in Go assembly.
|
|
const dot = "\u00b7"
|
|
|
|
type goasm struct {
|
|
cfg Config
|
|
prnt.Generator
|
|
}
|
|
|
|
func NewGoAsm(cfg Config) Printer {
|
|
return &goasm{cfg: cfg}
|
|
}
|
|
|
|
func (p *goasm) Print(f *avo.File) ([]byte, error) {
|
|
p.header()
|
|
for _, s := range f.Sections {
|
|
switch s := s.(type) {
|
|
case *avo.Function:
|
|
p.function(s)
|
|
case *avo.Global:
|
|
p.global(s)
|
|
default:
|
|
panic("unknown section type")
|
|
}
|
|
}
|
|
return p.Result()
|
|
}
|
|
|
|
func (p *goasm) header() {
|
|
p.Comment(p.cfg.GeneratedWarning())
|
|
p.NL()
|
|
p.include("textflag.h")
|
|
}
|
|
|
|
func (p *goasm) include(path string) {
|
|
p.Printf("#include \"%s\"\n", path)
|
|
}
|
|
|
|
func (p *goasm) function(f *avo.Function) {
|
|
p.NL()
|
|
p.Comment(f.Stub())
|
|
p.Printf("TEXT %s%s(SB),0,$%d-%d\n", dot, f.Name, f.FrameBytes(), f.ArgumentBytes())
|
|
|
|
for _, node := range f.Nodes {
|
|
switch n := node.(type) {
|
|
case *avo.Instruction:
|
|
if len(n.Operands) > 0 {
|
|
p.Printf("\t%s\t%s\n", n.Opcode, joinOperands(n.Operands))
|
|
} else {
|
|
p.Printf("\t%s\n", n.Opcode)
|
|
}
|
|
case avo.Label:
|
|
p.Printf("%s:\n", n)
|
|
default:
|
|
panic("unexpected node type")
|
|
}
|
|
}
|
|
}
|
|
|
|
func (p *goasm) global(g *avo.Global) {
|
|
p.NL()
|
|
for _, d := range g.Data {
|
|
a := operand.NewDataAddr(g.Symbol, d.Offset)
|
|
p.Printf("DATA %s/%d, %s\n", a.Asm(), d.Value.Bytes(), d.Value.Asm())
|
|
}
|
|
// TODO(mbm): replace hardcoded RODATA with an attributes list
|
|
p.Printf("GLOBL %s%s(SB), RODATA, $%d\n", dot, g.Symbol, g.Size)
|
|
}
|
|
|
|
func joinOperands(operands []operand.Op) string {
|
|
asm := make([]string, len(operands))
|
|
for i, op := range operands {
|
|
asm[i] = op.Asm()
|
|
}
|
|
return strings.Join(asm, ", ")
|
|
}
|