llgo/ssa: eh.go (exception handling)
This commit is contained in:
@@ -19,7 +19,6 @@ package ssa
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"go/token"
|
||||
"go/types"
|
||||
"log"
|
||||
"strings"
|
||||
@@ -136,191 +135,6 @@ func notInit(instr llvm.Value) bool {
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
const (
|
||||
deferKey = "__llgo_defer"
|
||||
)
|
||||
|
||||
type aDefer struct {
|
||||
nextBit int // next defer bit
|
||||
key Expr // pthread TLS key
|
||||
data Expr // pointer to runtime.Defer
|
||||
bitsPtr Expr // pointer to defer bits
|
||||
rundPtr Expr // pointer to RunDefers index
|
||||
procBlk BasicBlock // deferProc block
|
||||
stmts []func(bits Expr)
|
||||
runsNext []BasicBlock // next blocks of RunDefers
|
||||
}
|
||||
|
||||
/*
|
||||
// func(uintptr)
|
||||
func (p Program) tyDeferFunc() *types.Signature {
|
||||
if p.deferFnTy == nil {
|
||||
paramUintptr := types.NewParam(token.NoPos, nil, "", p.Uintptr().raw.Type)
|
||||
params := types.NewTuple(paramUintptr)
|
||||
p.deferFnTy = types.NewSignatureType(nil, nil, nil, params, nil, false)
|
||||
}
|
||||
return p.deferFnTy
|
||||
}
|
||||
*/
|
||||
|
||||
func (p Package) deferInit() {
|
||||
keyVar := p.VarOf(deferKey)
|
||||
if keyVar == nil {
|
||||
return
|
||||
}
|
||||
prog := p.Prog
|
||||
keyNil := prog.Null(prog.DeferPtrPtr())
|
||||
keyVar.Init(keyNil)
|
||||
keyVar.impl.SetLinkage(llvm.LinkOnceAnyLinkage)
|
||||
|
||||
b := p.afterBuilder()
|
||||
eq := b.BinOp(token.EQL, b.Load(keyVar.Expr), keyNil)
|
||||
b.IfThen(eq, func() {
|
||||
b.pthreadKeyCreate(keyVar.Expr, prog.Null(prog.VoidPtr()))
|
||||
})
|
||||
}
|
||||
|
||||
func (p Package) newDeferKey() Global {
|
||||
return p.NewVarEx(deferKey, p.Prog.DeferPtrPtr())
|
||||
}
|
||||
|
||||
func (b Builder) deferKey() Expr {
|
||||
return b.Load(b.Pkg.newDeferKey().Expr)
|
||||
}
|
||||
|
||||
func (b Builder) getDefer() *aDefer {
|
||||
self := b.Func
|
||||
if self.defer_ == nil {
|
||||
// TODO(xsw): if in pkg.init?
|
||||
// 0: proc func(uintptr)
|
||||
// 1: bits uintptr
|
||||
// 2: link *Defer
|
||||
// 3: rund int
|
||||
b.SetBlockEx(b.blk, AtStart, false)
|
||||
prog := b.Prog
|
||||
key := b.deferKey()
|
||||
deferfn := prog.Null(prog.VoidPtr())
|
||||
zero := prog.Val(uintptr(0))
|
||||
link := b.pthreadGetspecific(key)
|
||||
ptr := b.aggregateAlloca(prog.Defer(), deferfn.impl, zero.impl, link.impl)
|
||||
deferData := Expr{ptr, prog.DeferPtr()}
|
||||
b.pthreadSetspecific(key, deferData)
|
||||
b.SetBlockEx(b.blk, AtEnd, false)
|
||||
self.defer_ = &aDefer{
|
||||
key: key,
|
||||
data: deferData,
|
||||
bitsPtr: b.FieldAddr(deferData, 1),
|
||||
rundPtr: b.FieldAddr(deferData, 3),
|
||||
procBlk: self.MakeBlock(),
|
||||
}
|
||||
}
|
||||
return self.defer_
|
||||
}
|
||||
|
||||
// Defer emits a defer instruction.
|
||||
func (b Builder) Defer(kind DoAction, fn Expr, args ...Expr) {
|
||||
if debugInstr {
|
||||
logCall("Defer", fn, args)
|
||||
}
|
||||
var prog Program
|
||||
var nextbit Expr
|
||||
var self = b.getDefer()
|
||||
switch kind {
|
||||
case DeferInCond:
|
||||
prog = b.Prog
|
||||
next := self.nextBit
|
||||
self.nextBit++
|
||||
bits := b.Load(self.bitsPtr)
|
||||
nextbit = prog.Val(uintptr(1 << next))
|
||||
b.Store(self.bitsPtr, b.BinOp(token.OR, bits, nextbit))
|
||||
case DeferAlways:
|
||||
// nothing to do
|
||||
default:
|
||||
panic("todo: DeferInLoop is not supported")
|
||||
}
|
||||
self.stmts = append(self.stmts, func(bits Expr) {
|
||||
switch kind {
|
||||
case DeferInCond:
|
||||
zero := prog.Val(uintptr(0))
|
||||
has := b.BinOp(token.NEQ, b.BinOp(token.AND, bits, nextbit), zero)
|
||||
b.IfThen(has, func() {
|
||||
b.Call(fn, args...)
|
||||
})
|
||||
case DeferAlways:
|
||||
b.Call(fn, args...)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// RunDefers emits instructions to run deferred instructions.
|
||||
func (b Builder) RunDefers() {
|
||||
prog := b.Prog
|
||||
self := b.getDefer()
|
||||
b.Store(self.rundPtr, prog.Val(len(self.runsNext)))
|
||||
b.Jump(self.procBlk)
|
||||
|
||||
blk := b.Func.MakeBlock()
|
||||
self.runsNext = append(self.runsNext, blk)
|
||||
|
||||
b.SetBlockEx(blk, AtEnd, false)
|
||||
b.blk.last = blk.last
|
||||
}
|
||||
|
||||
func (p Function) endDefer(b Builder) {
|
||||
self := p.defer_
|
||||
if self == nil {
|
||||
return
|
||||
}
|
||||
nexts := self.runsNext
|
||||
n := len(nexts)
|
||||
if n == 0 {
|
||||
return
|
||||
}
|
||||
b.SetBlockEx(self.procBlk, AtEnd, true)
|
||||
bits := b.Load(self.bitsPtr)
|
||||
stmts := self.stmts
|
||||
for i := len(stmts) - 1; i >= 0; i-- {
|
||||
stmts[i](bits)
|
||||
}
|
||||
|
||||
link := b.getField(b.Load(self.data), 2)
|
||||
b.pthreadSetspecific(self.key, link)
|
||||
|
||||
prog := b.Prog
|
||||
rund := b.Load(self.rundPtr)
|
||||
sw := b.impl.CreateSwitch(rund.impl, nexts[0].first, n-1)
|
||||
for i := 1; i < n; i++ {
|
||||
sw.AddCase(prog.Val(i).impl, nexts[i].first)
|
||||
}
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
/*
|
||||
// Recover emits a recover instruction.
|
||||
func (b Builder) Recover() (v Expr) {
|
||||
if debugInstr {
|
||||
log.Println("Recover")
|
||||
}
|
||||
prog := b.Prog
|
||||
return prog.Zero(prog.Eface())
|
||||
}
|
||||
*/
|
||||
|
||||
// Panic emits a panic instruction.
|
||||
func (b Builder) Panic(v Expr) {
|
||||
if debugInstr {
|
||||
log.Printf("Panic %v\n", v.impl)
|
||||
}
|
||||
b.Call(b.Pkg.rtFunc("TracePanic"), v)
|
||||
b.impl.CreateUnreachable()
|
||||
}
|
||||
|
||||
// Unreachable emits an unreachable instruction.
|
||||
func (b Builder) Unreachable() {
|
||||
b.impl.CreateUnreachable()
|
||||
}
|
||||
|
||||
// Return emits a return instruction.
|
||||
func (b Builder) Return(results ...Expr) {
|
||||
if debugInstr {
|
||||
|
||||
Reference in New Issue
Block a user