build: add cl.Context for linkname

cl/builtin_test.go

@@ -40,10 +40,10 @@ func TestToBackground(t *testing.T) {
 }
 
 func TestCollectSkipNames(t *testing.T) {
-	ctx := &context{skips: make(map[string]none)}
-	ctx.collectSkipNames("//llgo:skipall")
-	ctx.collectSkipNames("//llgo:skip")
-	ctx.collectSkipNames("//llgo:skip abs")
+	ctx := &Context{}
+	ctx.collectSkipNames("pkg", "//llgo:skipall")
+	ctx.collectSkipNames("pkg", "//llgo:skip")
+	ctx.collectSkipNames("pkg", "//llgo:skip abs")
 }
 
 func TestReplaceGoName(t *testing.T) {
@@ -236,6 +236,7 @@ func TestIgnoreName(t *testing.T) {
 
 func TestErrImport(t *testing.T) {
 	var ctx context
+	ctx.Context = NewContext(nil)
 	pkg := types.NewPackage("foo", "foo")
 	ctx.importPkg(pkg, nil)
 
@@ -249,6 +250,7 @@ func TestErrImport(t *testing.T) {
 
 func TestErrInitLinkname(t *testing.T) {
 	var ctx context
+	ctx.Context = NewContext(nil)
 	ctx.initLinkname("//llgo:link abc", func(name string) (string, bool, bool) {
 		return "", false, false
 	})

cl/cltest/cltest.go

@@ -119,12 +119,8 @@ func testFrom(t *testing.T, pkgDir, sel string, byLLGen bool) {
 		t.Fatal("ReadFile failed:", err)
 	}
 	expected := string(b)
-	if byLLGen {
-		if v := llgen.GenFrom(in); v != expected && expected != ";" { // expected == ";" means skipping out.ll
-			t.Fatalf("\n==> got:\n%s\n==> expected:\n%s\n", v, expected)
-		}
-	} else {
-		TestCompileEx(t, nil, in, expected)
+	if v := llgen.GenFrom(in); v != expected && expected != ";" { // expected == ";" means skipping out.ll
+		t.Fatalf("\n==> got:\n%s\n==> expected:\n%s\n", v, expected)
 	}
 }
 

cl/compile.go

@@ -25,6 +25,7 @@ import (
 	"log"
 	"os"
 	"sort"
+	"sync"
 
 	"github.com/goplus/llgo/cl/blocks"
 	"github.com/goplus/llgo/internal/typepatch"
@@ -79,22 +80,42 @@ type pkgInfo struct {
 
 type none = struct{}
 
+type Context struct {
+	prog      llssa.Program
+	link      sync.Map // pkgPath.nameInPkg => linkname
+	skipLines sync.Map // pkgPath => skip lines
+	patches   Patches
+	kinds     map[string]int
+}
+
+func NewContext(prog llssa.Program) *Context {
+	return &Context{
+		prog: prog,
+		kinds: map[string]int{
+			"runtime/cgo": PkgDeclOnly,
+			"unsafe":      PkgDeclOnly,
+		},
+	}
+}
+
+func (p *Context) SetPatches(patches Patches) {
+	p.patches = patches
+}
+
 type context struct {
-	prog   llssa.Program
+	*Context
 	pkg    llssa.Package
 	fn     llssa.Function
 	fset   *token.FileSet
 	goProg *ssa.Program
 	goTyps *types.Package
 	goPkg  *ssa.Package
 	pyMod  string
-	link   map[string]string // pkgPath.nameInPkg => linkname
 	skips  map[string]none
 	loaded map[*types.Package]*pkgInfo // loaded packages
 	bvals  map[ssa.Value]llssa.Expr    // block values
 	vargs  map[*ssa.Alloc][]llssa.Expr // varargs
 
-	patches  Patches
 	blkInfos []blocks.Info
 
 	inits []func()
@@ -771,16 +792,19 @@ type Patches = map[string]Patch
 
 // NewPackage compiles a Go package to LLVM IR package.
 func NewPackage(prog llssa.Program, pkg *ssa.Package, files []*ast.File) (ret llssa.Package, err error) {
-	return NewPackageEx(prog, nil, pkg, files)
+	bctx := NewContext(prog)
+	bctx.ParsePkgSyntax(pkg.Pkg, files, false)
+	return NewPackageEx(bctx, pkg)
 }
 
 // NewPackageEx compiles a Go package to LLVM IR package.
-func NewPackageEx(prog llssa.Program, patches Patches, pkg *ssa.Package, files []*ast.File) (ret llssa.Package, err error) {
+func NewPackageEx(bctx *Context, pkg *ssa.Package) (ret llssa.Package, err error) {
+	prog := bctx.prog
 	pkgProg := pkg.Prog
 	pkgTypes := pkg.Pkg
 	oldTypes := pkgTypes
 	pkgName, pkgPath := pkgTypes.Name(), llssa.PathOf(pkgTypes)
-	patch, hasPatch := patches[pkgPath]
+	patch, hasPatch := bctx.patches[pkgPath]
 	if hasPatch {
 		pkgTypes = patch.Types
 		pkg.Pkg = pkgTypes
@@ -792,25 +816,27 @@ func NewPackageEx(prog llssa.Program, patches Patches, pkg *ssa.Package, files [
 	ret = prog.NewPackage(pkgName, pkgPath)
 
 	ctx := &context{
-		prog:    prog,
+		Context: bctx,
 		pkg:     ret,
 		fset:    pkgProg.Fset,
 		goProg:  pkgProg,
 		goTyps:  pkgTypes,
 		goPkg:   pkg,
-		patches: patches,
-		link:    make(map[string]string),
 		skips:   make(map[string]none),
 		vargs:   make(map[*ssa.Alloc][]llssa.Expr),
 		loaded: map[*types.Package]*pkgInfo{
 			types.Unsafe: {kind: PkgDeclOnly}, // TODO(xsw): PkgNoInit or PkgDeclOnly?
 		},
 	}
 	ctx.initPyModule()
-	ctx.initFiles(pkgPath, files)
 	ret.SetPatch(ctx.patchType)
 
 	if hasPatch {
+		if v, ok := bctx.skipLines.Load(pkgPath); ok {
+			for _, line := range v.([]string) {
+				ctx.parseSkip(line)
+			}
+		}
 		skips := ctx.skips
 		typepatch.Merge(pkgTypes, oldTypes, skips, ctx.skipall)
 		ctx.skips = nil