admin/llgo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cl: isVargs support defer/go

Browse Source
This commit is contained in:
xushiwei
2024-06-11 10:35:27 +08:00
parent 1ecd9af2e1
commit 5e5d149ca5
6 changed files with 213 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -187,7 +187,7 @@ Here are some examples related to Go syntax:
### Garbage Collection (GC)
By default, LLGo implements `gc` based on [bdwgc](https://www.hboehm.info/gc/).
By default, LLGo implements `gc` based on [bdwgc](https://www.hboehm.info/gc/) (also known as [libgc](https://www.hboehm.info/gc/)).
However, you can disable gc by specifying the `nogc` tag. For example:
@@ -214,7 +214,7 @@ Follow these steps to generate the `llgo` command (its usage is the same as the
```sh
brew update # execute if needed
brew install bdw-gc
brew install libgc
brew install llvm@17
go install -v ./...
```

16
cl/_testlibc/defer/in.go Normal file
View File

@@ -0,0 +1,16 @@
package main
import "github.com/goplus/llgo/c"
func f(s string) bool {
return len(s) > 2
}
func main() {
if s := "hello"; f(s) {
defer c.Printf(c.Str("%s\n"), c.AllocaCStr(s))
} else {
defer c.Printf(c.Str("world\n"))
}
defer c.Printf(c.Str("bye\n"))
}

148
cl/_testlibc/defer/out.ll Normal file
View File

@@ -0,0 +1,148 @@
; ModuleID = 'main'
source_filename = "main"
%"github.com/goplus/llgo/internal/runtime.String" = type { ptr, i64 }
%"github.com/goplus/llgo/internal/runtime.Defer" = type { i64, ptr, i64 }
@"main.init$guard" = global ptr null
@__llgo_argc = global ptr null
@__llgo_argv = global ptr null
@0 = private unnamed_addr constant [6 x i8] c"hello\00", align 1
@1 = private unnamed_addr constant [4 x i8] c"%s\0A\00", align 1
@2 = private unnamed_addr constant [6 x i8] c"hello\00", align 1
@__llgo_defer = linkonce global ptr null
@3 = private unnamed_addr constant [7 x i8] c"world\0A\00", align 1
@4 = private unnamed_addr constant [5 x i8] c"bye\0A\00", align 1
define i1 @main.f(%"github.com/goplus/llgo/internal/runtime.String" %0) {
_llgo_0:
%1 = extractvalue %"github.com/goplus/llgo/internal/runtime.String" %0, 1
%2 = icmp sgt i64 %1, 2
ret i1 %2
}
define void @main.init() {
_llgo_0:
%0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"main.init$guard", align 1
call void @"main.init$after"()
br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0
ret void
}
define i32 @main(i32 %0, ptr %1) {
_llgo_0:
store i32 %0, ptr @__llgo_argc, align 4
store ptr %1, ptr @__llgo_argv, align 8
call void @"github.com/goplus/llgo/internal/runtime.init"()
call void @main.init()
%2 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%3 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 0
store ptr @0, ptr %3, align 8
%4 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 1
store i64 5, ptr %4, align 4
%5 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %2, align 8
%6 = call i1 @main.f(%"github.com/goplus/llgo/internal/runtime.String" %5)
%7 = load ptr, ptr @__llgo_defer, align 8
%8 = call ptr @pthread_getspecific(ptr %7)
%9 = alloca i8, i64 24, align 1
%10 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Defer", ptr %9, i32 0, i32 0
store i64 0, ptr %10, align 4
%11 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Defer", ptr %9, i32 0, i32 1
store ptr %8, ptr %11, align 8
%12 = call i32 @pthread_setspecific(ptr %7, ptr %9)
%13 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Defer", ptr %9, i32 0, i32 0
%14 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.Defer", ptr %9, i32 0, i32 2
br i1 %6, label %_llgo_1, label %_llgo_3
_llgo_1: ; preds = %_llgo_0
%15 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%16 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %15, i32 0, i32 0
store ptr @2, ptr %16, align 8
%17 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %15, i32 0, i32 1
store i64 5, ptr %17, align 4
%18 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %15, align 8
%19 = extractvalue %"github.com/goplus/llgo/internal/runtime.String" %18, 1
%20 = add i64 %19, 1
%21 = alloca i8, i64 %20, align 1
%22 = call ptr @"github.com/goplus/llgo/internal/runtime.CStrCopy"(ptr %21, %"github.com/goplus/llgo/internal/runtime.String" %18)
%23 = load i64, ptr %13, align 4
%24 = or i64 %23, 1
store i64 %24, ptr %13, align 4
br label %_llgo_2
_llgo_2: ; preds = %_llgo_3, %_llgo_1
store i64 0, ptr %14, align 4
br label %_llgo_5
_llgo_3: ; preds = %_llgo_0
%25 = load i64, ptr %13, align 4
%26 = or i64 %25, 2
store i64 %26, ptr %13, align 4
br label %_llgo_2
_llgo_4: ; No predecessors!
ret i32 0
_llgo_5: ; preds = %_llgo_2
%27 = load i64, ptr %13, align 4
%28 = call i32 (ptr, ...) @printf(ptr @4)
%29 = and i64 %27, 2
%30 = icmp ne i64 %29, 0
br i1 %30, label %_llgo_7, label %_llgo_8
_llgo_6: ; preds = %_llgo_10
ret i32 0
_llgo_7: ; preds = %_llgo_5
%31 = call i32 (ptr, ...) @printf(ptr @3)
br label %_llgo_8
_llgo_8: ; preds = %_llgo_7, %_llgo_5
%32 = and i64 %27, 1
%33 = icmp ne i64 %32, 0
br i1 %33, label %_llgo_9, label %_llgo_10
_llgo_9: ; preds = %_llgo_8
%34 = call i32 (ptr, ...) @printf(ptr @1, ptr %22)
br label %_llgo_10
_llgo_10: ; preds = %_llgo_9, %_llgo_8
%35 = load %"github.com/goplus/llgo/internal/runtime.Defer", ptr %9, align 8
%36 = extractvalue %"github.com/goplus/llgo/internal/runtime.Defer" %35, 2
%37 = call i32 @pthread_setspecific(ptr %7, i64 %36)
%38 = load i64, ptr %14, align 4
switch i64 %38, label %_llgo_6 [
]
}
declare void @"github.com/goplus/llgo/internal/runtime.init"()
declare ptr @"github.com/goplus/llgo/internal/runtime.CStrCopy"(ptr, %"github.com/goplus/llgo/internal/runtime.String")
declare i32 @printf(ptr, ...)
declare ptr @pthread_getspecific(i32)
declare i32 @pthread_setspecific(i32, ptr)
define void @"main.init$after"() {
_llgo_0:
%0 = load ptr, ptr @__llgo_defer, align 8
%1 = icmp eq ptr %0, null
br i1 %1, label %_llgo_1, label %_llgo_2
_llgo_1: ; preds = %_llgo_0
%2 = call i32 @pthread_key_create(ptr @__llgo_defer, ptr null)
br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0
ret void
}
declare i32 @pthread_key_create(ptr, ptr)

31
cl/builtin_test.go
View File

@@ -21,11 +21,42 @@ import (
"go/constant"
"go/types"
"testing"
"unsafe"
llssa "github.com/goplus/llgo/ssa"
"golang.org/x/tools/go/ssa"
)
func TestIsVargs(t *testing.T) {
if isVargs(nil, ssaAlloc(&ssa.Return{})) {
t.Fatal("isVargs?")
}
if isVargs(nil, ssaAlloc(ssaSlice(&ssa.Go{}))) {
t.Fatal("isVargs?")
}
if isVargs(nil, ssaAlloc(ssaSlice(&ssa.Return{}))) {
t.Fatal("isVargs?")
}
}
func ssaSlice(refs ...ssa.Instruction) *ssa.Slice {
a := &ssa.Slice{}
setRefs(unsafe.Pointer(a), refs...)
return a
}
func ssaAlloc(refs ...ssa.Instruction) *ssa.Alloc {
a := &ssa.Alloc{}
setRefs(unsafe.Pointer(a), refs...)
return a
}
func setRefs(v unsafe.Pointer, refs ...ssa.Instruction) {
off := unsafe.Offsetof(ssa.Alloc{}.Comment) - unsafe.Sizeof([]int(nil))
ptr := uintptr(v) + off
*(*[]ssa.Instruction)(unsafe.Pointer(ptr)) = refs
}
func TestRecvTypeName(t *testing.T) {
if ret := recvTypeName(&ast.IndexExpr{
X: &ast.Ident{Name: "Pointer"},

20
cl/compile.go
View File

@@ -430,9 +430,10 @@ func intVal(v ssa.Value) int64 {
panic("intVal: ssa.Value is not a const int")
}
func (p *context) isVArgs(vx ssa.Value) (ret []llssa.Expr, ok bool) {
if va, vok := vx.(*ssa.Alloc); vok {
ret, ok = p.vargs[va] // varargs: this is a varargs index
func (p *context) isVArgs(v ssa.Value) (ret []llssa.Expr, ok bool) {
switch v := v.(type) {
case *ssa.Alloc:
ret, ok = p.vargs[v] // varargs: this is a varargs index
}
return
}
@@ -455,9 +456,18 @@ func isVargs(ctx *context, v *ssa.Alloc) bool {
lastref := refs[n-1]
if i, ok := lastref.(*ssa.Slice); ok {
if refs = *i.Referrers(); len(refs) == 1 {
if call, ok := refs[0].(*ssa.Call); ok {
return ctx.funcKind(call.Call.Value) == fnHasVArg
var call *ssa.CallCommon
switch ref := refs[0].(type) {
case *ssa.Call:
call = &ref.Call
case *ssa.Defer:
call = &ref.Call
case *ssa.Go:
call = &ref.Call
default:
return false
}
return ctx.funcKind(call.Value) == fnHasVArg
}
}
return false

2
internal/llgen/llgenf.go
View File

@@ -131,5 +131,5 @@ func genZip(dir string, outFile, inFile string) {
}
func inCompilerDir(dir string) bool {
return strings.Contains(dir, "/llgo/cl/")
return strings.Contains(dir, "/cl/_test")
}