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

builtin: llgo.advance

Browse Source
This commit is contained in:
xushiwei
2024-05-01 23:57:19 +08:00
parent cf02f4a34f
commit df1e4708f5
6 changed files with 85 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
cl/compile.go
View File

@@ -300,6 +300,15 @@ func cstr(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
panic("cstr(<string-literal>): invalid arguments")
}
func (p *context) advance(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
if len(args) == 2 {
ptr := p.compileValue(b, args[0])
offset := p.compileValue(b, args[1])
return b.Advance(ptr, offset)
}
panic("advance(p ptr, offset int): invalid arguments")
}
// func alloca(size uintptr) unsafe.Pointer
func (p *context) alloca(b llssa.Builder, args []ssa.Value) (ret llssa.Expr) {
if len(args) == 1 {
@@ -355,6 +364,8 @@ func (p *context) compileInstrOrValue(b llssa.Builder, iv instrOrValue, asValue
ret = b.Call(fn.Expr, args...)
case llgoCstr:
ret = cstr(b, call.Args)
case llgoAdvance:
ret = p.advance(b, call.Args)
case llgoAlloca:
ret = p.alloca(b, call.Args)
case llgoAllocaCStr:

3
cl/import.go
View File

@@ -200,6 +200,7 @@ const (
llgoCstr = llgoInstrBase + 1
llgoAlloca = llgoInstrBase + 2
llgoAllocaCStr = llgoInstrBase + 3
llgoAdvance = llgoInstrBase + 4
)
func (p *context) funcName(pkg *types.Package, fn *ssa.Function, ignore bool) (string, int) {
@@ -237,6 +238,8 @@ func (p *context) funcOf(fn *ssa.Function) (ret llssa.Function, ftype int) {
switch name {
case "cstr":
ftype = llgoCstr
case "advance":
ftype = llgoAdvance
case "alloca":
ftype = llgoAlloca
case "allocaCStr":

3
internal/runtime/c/c.go
View File

@@ -42,6 +42,9 @@ var Stderr FilePtr
//go:linkname Str llgo.cstr
func Str(string) *Char
//go:linkname Advance llgo.advance
func Advance(ptr Pointer, offset int) Pointer
//go:linkname Alloca llgo.alloca
func Alloca(size uintptr) Pointer

71
internal/runtime/llgo_autogen.ll
View File

@@ -13,8 +13,7 @@ source_filename = "github.com/goplus/llgo/internal/runtime"
@"github.com/goplus/llgo/internal/runtime.init$guard" = global ptr null
@"github.com/goplus/llgo/internal/runtime.sizeBasicTypes" = global ptr null
@0 = private unnamed_addr constant [21 x i8] c"I2Int: type mismatch\00", align 1
@1 = private unnamed_addr constant [5 x i8] c"todo\00", align 1
@2 = private unnamed_addr constant [11 x i8] c"panic: %s\0A\00", align 1
@1 = private unnamed_addr constant [11 x i8] c"panic: %s\0A\00", align 1
define ptr @"github.com/goplus/llgo/internal/runtime.Alloc"(i64 %0) {
_llgo_0:
@@ -269,32 +268,62 @@ _llgo_0:
define %"github.com/goplus/llgo/internal/runtime.String" @"github.com/goplus/llgo/internal/runtime.StringCat"(%"github.com/goplus/llgo/internal/runtime.Slice" %0) {
_llgo_0:
%1 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%2 = call i64 @"github.com/goplus/llgo/internal/runtime.SliceLen"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
%2 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%3 = call i64 @"github.com/goplus/llgo/internal/runtime.SliceLen"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
br label %_llgo_1
_llgo_1: ; preds = %_llgo_2, %_llgo_0
%3 = phi i64 [ 0, %_llgo_0 ], [ %12, %_llgo_2 ]
%4 = phi i64 [ -1, %_llgo_0 ], [ %5, %_llgo_2 ]
%5 = add i64 %4, 1
%6 = icmp slt i64 %5, %2
br i1 %6, label %_llgo_2, label %_llgo_3
%4 = phi i64 [ 0, %_llgo_0 ], [ %13, %_llgo_2 ]
%5 = phi i64 [ -1, %_llgo_0 ], [ %6, %_llgo_2 ]
%6 = add i64 %5, 1
%7 = icmp slt i64 %6, %3
br i1 %7, label %_llgo_2, label %_llgo_3
_llgo_2: ; preds = %_llgo_1
%7 = call ptr @"github.com/goplus/llgo/internal/runtime.SliceData"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
%8 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %7, i64 %5
%9 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %8, align 8
store %"github.com/goplus/llgo/internal/runtime.String" %9, ptr %1, align 8
%10 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %1, i32 0, i32 1
%11 = load i64, ptr %10, align 4
%12 = add i64 %3, %11
%8 = call ptr @"github.com/goplus/llgo/internal/runtime.SliceData"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
%9 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %8, i64 %6
%10 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %9, align 8
store %"github.com/goplus/llgo/internal/runtime.String" %10, ptr %2, align 8
%11 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 1
%12 = load i64, ptr %11, align 4
%13 = add i64 %4, %12
br label %_llgo_1
_llgo_3: ; preds = %_llgo_1
%13 = call ptr @"github.com/goplus/llgo/internal/runtime.Alloc"(i64 %3)
%14 = call %"github.com/goplus/llgo/internal/runtime.String" @"github.com/goplus/llgo/internal/runtime.NewString"(ptr @1, i64 4)
%15 = call %"github.com/goplus/llgo/internal/runtime.iface" @"github.com/goplus/llgo/internal/runtime.MakeAnyString"(%"github.com/goplus/llgo/internal/runtime.String" %14)
call void @"github.com/goplus/llgo/internal/runtime.TracePanic"(%"github.com/goplus/llgo/internal/runtime.iface" %15)
unreachable
%14 = call ptr @"github.com/goplus/llgo/internal/runtime.Alloc"(i64 %4)
%15 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %1, i32 0, i32 0
store ptr %14, ptr %15, align 8
%16 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %1, i32 0, i32 1
store i64 %4, ptr %16, align 4
%17 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%18 = call i64 @"github.com/goplus/llgo/internal/runtime.SliceLen"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
br label %_llgo_4
_llgo_4: ; preds = %_llgo_5, %_llgo_3
%19 = phi ptr [ %14, %_llgo_3 ], [ %33, %_llgo_5 ]
%20 = phi i64 [ -1, %_llgo_3 ], [ %21, %_llgo_5 ]
%21 = add i64 %20, 1
%22 = icmp slt i64 %21, %18
br i1 %22, label %_llgo_5, label %_llgo_6
_llgo_5: ; preds = %_llgo_4
%23 = call ptr @"github.com/goplus/llgo/internal/runtime.SliceData"(%"github.com/goplus/llgo/internal/runtime.Slice" %0)
%24 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %23, i64 %21
%25 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %24, align 8
store %"github.com/goplus/llgo/internal/runtime.String" %25, ptr %17, align 8
%26 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %17, i32 0, i32 0
%27 = load ptr, ptr %26, align 8
%28 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %17, i32 0, i32 1
%29 = load i64, ptr %28, align 4
%30 = call ptr @memcpy(ptr %19, ptr %27, i64 %29)
%31 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %17, i32 0, i32 1
%32 = load i64, ptr %31, align 4
%33 = getelementptr i8, ptr %19, i64 %32
br label %_llgo_4
_llgo_6: ; preds = %_llgo_4
%34 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %1, align 8
ret %"github.com/goplus/llgo/internal/runtime.String" %34
}
define ptr @"github.com/goplus/llgo/internal/runtime.StringData"(%"github.com/goplus/llgo/internal/runtime.String" %0) {
@@ -341,7 +370,7 @@ _llgo_2: ; preds = %_llgo_0
%15 = alloca i8, i64 %14, align 1
%16 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %11, align 8
%17 = call ptr @"github.com/goplus/llgo/internal/runtime.CStrCopy"(ptr %15, %"github.com/goplus/llgo/internal/runtime.String" %16)
%18 = call i32 (ptr, ...) @printf(ptr @2, ptr %17)
%18 = call i32 (ptr, ...) @printf(ptr @1, ptr %17)
br label %_llgo_1
}

13
internal/runtime/z_string.go
View File

@@ -56,14 +56,19 @@ func StringData(s String) unsafe.Pointer {
}
// StringCat concatenates strings.
func StringCat(args ...String) String {
func StringCat(args ...String) (ret String) {
n := 0
for _, v := range args {
n += v.len
}
ret := Alloc(uintptr(n))
_ = ret
panic("todo")
dest := Alloc(uintptr(n))
ret.data = dest
ret.len = n
for _, v := range args {
c.Memcpy(dest, v.data, uintptr(v.len))
dest = c.Advance(dest, v.len)
}
return
}
// -----------------------------------------------------------------------------

9
ssa/expr.go
View File

@@ -364,6 +364,15 @@ func (b Builder) Phi(t Type) Phi {
// -----------------------------------------------------------------------------
// Advance returns the pointer ptr advanced by offset bytes.
func (b Builder) Advance(ptr Expr, offset Expr) Expr {
if debugInstr {
log.Printf("Advance %v, %v\n", ptr.impl, offset.impl)
}
ret := llvm.CreateGEP(b.impl, b.Prog.tyInt8(), ptr.impl, []llvm.Value{offset.impl})
return Expr{ret, ptr.Type}
}
// Load returns the value at the pointer ptr.
func (b Builder) Load(ptr Expr) Expr {
if debugInstr {