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

cl: varOf

Browse Source
This commit is contained in:
xushiwei
2024-04-23 01:16:31 +08:00
parent 67af68ae10
commit aafe5a8600
9 changed files with 81 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
chore/llgen/llgen.go
View File

@@ -25,8 +25,8 @@ import (
) )
func main() { func main() {
if len(os.Args) != 2 { if len(os.Args) < 2 {
fmt.Fprintln(os.Stderr, "Usage: llgen xxx.go") fmt.Fprintln(os.Stderr, "Usage: llgen xxx.go [pkgPath]")
return return
} }
@@ -35,6 +35,13 @@ func main() {
dir, _ := filepath.Split(inFile) dir, _ := filepath.Split(inFile)
outFile := dir + "out.ll" outFile := dir + "out.ll"
pkgPath := ""
if len(os.Args) == 3 {
pkgPath = os.Args[2]
} else {
pkgPath = llgen.PkgPath(dir)
}
llgen.Init() llgen.Init()
llgen.Do(llgen.PkgPath(dir), inFile, outFile) llgen.Do(pkgPath, inFile, outFile)
} }

6
cl/_testdata/apkg/out.ll
View File

@@ -1,15 +1,15 @@
; ModuleID = 'apkg' ; ModuleID = 'apkg'
source_filename = "apkg" source_filename = "apkg"
@"init$guard" = global ptr null @"apkg.init$guard" = global ptr null
define void @apkg.init() { define void @apkg.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"apkg.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"apkg.init$guard", align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0

6
cl/_testdata/fncall/out.ll
View File

@@ -1,15 +1,15 @@
; ModuleID = 'main' ; ModuleID = 'main'
source_filename = "main" source_filename = "main"
@"init$guard" = global ptr null @"main.init$guard" = global ptr null
define void @main.init() { define void @main.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"main.init$guard", align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0

24
cl/_testdata/importpkg/out.ll
View File

@@ -1,24 +1,24 @@
; ModuleID = 'main' ; ModuleID = 'main'
source_filename = "main" source_filename = "main"
@"init$guard" = global ptr null @"main.init$guard" = global ptr null
@hello = global ptr null @main.hello = global ptr null
define void @main.init() { define void @main.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"main.init$guard", align 1
call void @"github.com/goplus/llgo/cl/internal/stdio.init"() call void @"github.com/goplus/llgo/cl/internal/stdio.init"()
store i8 72, ptr @hello, align 1 store i8 72, ptr @main.hello, align 1
store i8 101, ptr getelementptr inbounds (i8, ptr @hello, i64 1), align 1 store i8 101, ptr getelementptr inbounds (i8, ptr @main.hello, i64 1), align 1
store i8 108, ptr getelementptr inbounds (i8, ptr @hello, i64 2), align 1 store i8 108, ptr getelementptr inbounds (i8, ptr @main.hello, i64 2), align 1
store i8 108, ptr getelementptr inbounds (i8, ptr @hello, i64 3), align 1 store i8 108, ptr getelementptr inbounds (i8, ptr @main.hello, i64 3), align 1
store i8 111, ptr getelementptr inbounds (i8, ptr @hello, i64 4), align 1 store i8 111, ptr getelementptr inbounds (i8, ptr @main.hello, i64 4), align 1
store i8 10, ptr getelementptr inbounds (i8, ptr @hello, i64 5), align 1 store i8 10, ptr getelementptr inbounds (i8, ptr @main.hello, i64 5), align 1
store i8 0, ptr getelementptr inbounds (i8, ptr @hello, i64 6), align 1 store i8 0, ptr getelementptr inbounds (i8, ptr @main.hello, i64 6), align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0
@@ -28,7 +28,7 @@ _llgo_2: ; preds = %_llgo_1, %_llgo_0
define void @main() { define void @main() {
_llgo_0: _llgo_0:
call void @main.init() call void @main.init()
call void (ptr, ...) @printf(ptr @hello) call void (ptr, ...) @printf(ptr @main.hello)
ret void ret void
} }

24
cl/_testdata/printf/out.ll
View File

@@ -1,23 +1,23 @@
; ModuleID = 'main' ; ModuleID = 'main'
source_filename = "main" source_filename = "main"
@"init$guard" = global ptr null @"main.init$guard" = global ptr null
@hello = global ptr null @main.hello = global ptr null
define void @main.init() { define void @main.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"main.init$guard", align 1
store i8 72, ptr @hello, align 1 store i8 72, ptr @main.hello, align 1
store i8 101, ptr getelementptr inbounds (i8, ptr @hello, i64 1), align 1 store i8 101, ptr getelementptr inbounds (i8, ptr @main.hello, i64 1), align 1
store i8 108, ptr getelementptr inbounds (i8, ptr @hello, i64 2), align 1 store i8 108, ptr getelementptr inbounds (i8, ptr @main.hello, i64 2), align 1
store i8 108, ptr getelementptr inbounds (i8, ptr @hello, i64 3), align 1 store i8 108, ptr getelementptr inbounds (i8, ptr @main.hello, i64 3), align 1
store i8 111, ptr getelementptr inbounds (i8, ptr @hello, i64 4), align 1 store i8 111, ptr getelementptr inbounds (i8, ptr @main.hello, i64 4), align 1
store i8 10, ptr getelementptr inbounds (i8, ptr @hello, i64 5), align 1 store i8 10, ptr getelementptr inbounds (i8, ptr @main.hello, i64 5), align 1
store i8 0, ptr getelementptr inbounds (i8, ptr @hello, i64 6), align 1 store i8 0, ptr getelementptr inbounds (i8, ptr @main.hello, i64 6), align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0
@@ -29,6 +29,6 @@ declare void @printf(ptr, ...)
define void @main() { define void @main() {
_llgo_0: _llgo_0:
call void @main.init() call void @main.init()
call void (ptr, ...) @printf(ptr @hello) call void (ptr, ...) @printf(ptr @main.hello)
ret void ret void
} }

16
cl/_testdata/varinit/out.ll
View File

@@ -1,17 +1,17 @@
; ModuleID = 'main' ; ModuleID = 'main'
source_filename = "main" source_filename = "main"
@"init$guard" = global ptr null @"main.init$guard" = global ptr null
@a = global ptr null @main.a = global ptr null
define void @main.init() { define void @main.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"main.init$guard", align 1
store i64 100, ptr @a, align 4 store i64 100, ptr @main.a, align 4
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0
@@ -21,9 +21,9 @@ _llgo_2: ; preds = %_llgo_1, %_llgo_0
define void @main() { define void @main() {
_llgo_0: _llgo_0:
call void @main.init() call void @main.init()
%0 = load i64, ptr @a, align 4 %0 = load i64, ptr @main.a, align 4
%1 = add i64 %0, 1 %1 = add i64 %0, 1
store i64 %1, ptr @a, align 4 store i64 %1, ptr @main.a, align 4
%2 = load i64, ptr @a, align 4 %2 = load i64, ptr @main.a, align 4
ret void ret void
} }

10
cl/compile.go
View File

@@ -91,6 +91,7 @@ type context struct {
pkg llssa.Package pkg llssa.Package
fn llssa.Function fn llssa.Function
fset *token.FileSet fset *token.FileSet
goTyps *types.Package
goPkg *ssa.Package goPkg *ssa.Package
link map[string]string // pkgPath.nameInPkg => linkname link map[string]string // pkgPath.nameInPkg => linkname
loaded map[*types.Package]none // loaded packages loaded map[*types.Package]none // loaded packages
@@ -104,7 +105,8 @@ func (p *context) compileType(pkg llssa.Package, member *ssa.Type) {
// Global variable. // Global variable.
func (p *context) compileGlobal(pkg llssa.Package, gbl *ssa.Global) { func (p *context) compileGlobal(pkg llssa.Package, gbl *ssa.Global) {
name, typ := gbl.Name(), gbl.Type() typ := gbl.Type()
name := fullName(gbl.Pkg.Pkg, gbl.Name())
if debugInstr { if debugInstr {
log.Println("==> NewVar", name, typ) log.Println("==> NewVar", name, typ)
} }
@@ -245,10 +247,7 @@ func (p *context) compileValue(b llssa.Builder, v ssa.Value) llssa.Expr {
fn := p.funcOf(v) fn := p.funcOf(v)
return fn.Expr return fn.Expr
case *ssa.Global: case *ssa.Global:
if v.Pkg != p.goPkg { g := p.varOf(v)
panic("todo")
}
g := p.pkg.VarOf(v.Name())
return g.Expr return g.Expr
case *ssa.Const: case *ssa.Const:
t := v.Type() t := v.Type()
@@ -309,6 +308,7 @@ func NewPackage(prog llssa.Program, pkg *ssa.Package, files []*ast.File) (ret ll
prog: prog, prog: prog,
pkg: ret, pkg: ret,
fset: pkg.Prog.Fset, fset: pkg.Prog.Fset,
goTyps: pkgTypes,
goPkg: pkg, goPkg: pkg,
link: make(map[string]string), link: make(map[string]string),
loaded: make(map[*types.Package]none), loaded: make(map[*types.Package]none),

14
cl/compile_test.go
View File

@@ -118,16 +118,16 @@ var a int
`, `; ModuleID = 'foo' `, `; ModuleID = 'foo'
source_filename = "foo" source_filename = "foo"
@"init$guard" = global ptr null @"foo.init$guard" = global ptr null
@a = global ptr null @foo.a = global ptr null
define void @foo.init() { define void @foo.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"foo.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"foo.init$guard", align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0
@@ -145,15 +145,15 @@ func fn(a int, b float64) int {
`, `; ModuleID = 'foo' `, `; ModuleID = 'foo'
source_filename = "foo" source_filename = "foo"
@"init$guard" = global ptr null @"foo.init$guard" = global ptr null
define void @foo.init() { define void @foo.init() {
_llgo_0: _llgo_0:
%0 = load i1, ptr @"init$guard", align 1 %0 = load i1, ptr @"foo.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1 br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0 _llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"init$guard", align 1 store i1 true, ptr @"foo.init$guard", align 1
br label %_llgo_2 br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0 _llgo_2: ; preds = %_llgo_1, %_llgo_0

26
cl/import.go
View File

@@ -126,11 +126,7 @@ func (p *context) funcName(pkg *types.Package, fn *ssa.Function) string {
} }
func (p *context) funcOf(fn *ssa.Function) llssa.Function { func (p *context) funcOf(fn *ssa.Function) llssa.Function {
pkgTypes := fn.Pkg.Pkg pkgTypes := p.ensureLoaded(fn.Pkg.Pkg)
if _, ok := p.loaded[pkgTypes]; !ok {
p.loaded[pkgTypes] = none{}
p.importPkg(pkgTypes)
}
pkg := p.pkg pkg := p.pkg
name := p.funcName(pkgTypes, fn) name := p.funcName(pkgTypes, fn)
if ret := pkg.FuncOf(name); ret != nil { if ret := pkg.FuncOf(name); ret != nil {
@@ -138,3 +134,23 @@ func (p *context) funcOf(fn *ssa.Function) llssa.Function {
} }
return pkg.NewFunc(name, fn.Signature) return pkg.NewFunc(name, fn.Signature)
} }
func (p *context) varOf(v *ssa.Global) llssa.Global {
pkgTypes := p.ensureLoaded(v.Pkg.Pkg)
pkg := p.pkg
name := fullName(pkgTypes, v.Name())
if ret := pkg.VarOf(name); ret != nil {
return ret
}
return pkg.NewVar(name, v.Type())
}
func (p *context) ensureLoaded(pkgTypes *types.Package) *types.Package {
if p.goTyps != pkgTypes {
if _, ok := p.loaded[pkgTypes]; !ok {
p.loaded[pkgTypes] = none{}
p.importPkg(pkgTypes)
}
}
return pkgTypes
}