[Feat] Enable LTO

Signed-off-by: ZhouGuangyuan <zhouguangyuan.xian@gmail.com>

Author: zhouguangyuan0718

_demo/embed/targetsbuild/build.sh

@@ -111,6 +111,7 @@ case "$test_dir" in
 			"riscv32"
 			"riscv64"
 			"rp2040"
+			"nintendoswitch" # undefined symbol under lto, should not work when no-lto
 		)
 		;;
 	defer)

internal/build/plan9asm.go

@@ -254,6 +254,10 @@ func filterPlan9AsmFuncs(pkgPath, goos, goarch string, funcs []plan9asm.Func, re
 		if pkgPath == "syscall" && goos == "linux" && (goarch == "arm64" || goarch == "amd64") && strings.HasSuffix(resolved, "rawVforkSyscall") {
 			continue
 		}
+		if pkgPath == "syscall" && goos == "darwin" && (goarch == "arm64" || goarch == "amd64") &&
+			(strings.HasSuffix(resolved, "RawSyscall") || strings.HasSuffix(resolved, "RawSyscall6")) {
+			continue
+		}
 		keep = append(keep, fn)
 	}
 	return keep

internal/crosscompile/crosscompile.go

@@ -225,8 +225,9 @@ func use(goos, goarch string, wasiThreads, forceEspClang bool) (export Export, e
 			"-Wl,--error-limit=0",
 			"-fuse-ld=lld",
 			// Enable ICF (Identical Code Folding) to reduce binary size
-			"-Xlinker",
-			"--icf=safe",
+			"-Wl,--icf=safe",
+			// Enable ThinLTO, Using default lto kind(thinlto).
+			"-Wl,--lto-O0",
 		}
 		if clangRoot != "" {
 			clangLib := filepath.Join(clangRoot, "lib")
@@ -249,6 +250,7 @@ func use(goos, goarch string, wasiThreads, forceEspClang bool) (export Export, e
 		export.CCFLAGS = []string{
 			"-Qunused-arguments",
 			"-Wno-unused-command-line-argument",
+			"-flto=thin",
 		}
 
 		// Add sysroot for macOS only
@@ -499,6 +501,13 @@ func UseTarget(targetName string) (export Export, err error) {
 	expandedCFlags := env.ExpandEnvSlice(config.CFlags, envs)
 	cflags = append(cflags, expandedCFlags...)
 
+	if config.Linker == "ld.lld" {
+		// Enable ThinLTO, Using default lto kind(thinlto).
+		ldflags = append(ldflags, "--lto-O0")
+		cflags = append(cflags, "-flto=thin")
+		ccflags = append(ccflags, "-flto=thin")
+	}
+
 	// The following parameters are inspired by tinygo/builder/library.go
 	// Handle CPU configuration
 	if cpu != "" {
@@ -545,6 +554,8 @@ func UseTarget(targetName string) (export Export, err error) {
 		ccflags = append(ccflags, "-fforce-enable-int128")
 	case "riscv64":
 		ccflags = append(ccflags, "-march=rv64gc")
+		// codegen option should be added to ldflags for lto
+		ldflags = append(ldflags, "-mllvm", "-march=rv64gc")
 	case "mips":
 		ccflags = append(ccflags, "-fno-pic")
 	}
@@ -574,9 +585,13 @@ func UseTarget(targetName string) (export Export, err error) {
 	// Handle code generation configuration
 	if config.CodeModel != "" {
 		ccflags = append(ccflags, "-mcmodel="+config.CodeModel)
+		// codegen option should be added to ldflags for lto
+		ldflags = append(ldflags, "-mllvm", "-code-model="+config.CodeModel)
 	}
 	if config.TargetABI != "" {
 		ccflags = append(ccflags, "-mabi="+config.TargetABI)
+		// codegen option should be added to ldflags for lto
+		ldflags = append(ldflags, "-mllvm", "-target-abi="+config.TargetABI)
 	}
 	if config.RelocationModel != "" {
 		switch config.RelocationModel {

ssa/datastruct.go

@@ -342,6 +342,10 @@ func (b Builder) Slice(x, low, high, max Expr) (ret Expr) {
 	if lowIsNil {
 		low = prog.IntVal(0, prog.Int())
 	}
+	low = b.fitIntSize(low)
+	if !high.IsNil() {
+		high = b.fitIntSize(high)
+	}
 	switch t := x.raw.Type.Underlying().(type) {
 	case *types.Basic:
 		if t.Kind() != types.String {
@@ -382,6 +386,7 @@ func (b Builder) Slice(x, low, high, max Expr) (ret Expr) {
 	if max.IsNil() {
 		max = nCap
 	}
+	max = b.fitIntSize(max)
 	ret.impl = b.InlineCall(b.Pkg.rtFunc("NewSlice3"), base, nEltSize, nCap, low, high, max).impl
 	return
 }

ssa/decl.go

@@ -211,6 +211,10 @@ func (p Package) NewFuncEx(name string, sig *types.Signature, bg Background, has
 	if instantiated {
 		fn.SetLinkage(llvm.LinkOnceAnyLinkage)
 	}
+	if p.Prog.requiresFramePointer() {
+		attr := p.mod.Context().CreateStringAttribute("frame-pointer", "all")
+		fn.AddFunctionAttr(attr)
+	}
 	if p.isPreservedName(name) {
 		p.markLLVMUsed(fn)
 	}
@@ -345,6 +349,10 @@ func (p Function) Block(idx int) BasicBlock {
 	return p.blks[idx]
 }
 
+func (p Program) requiresFramePointer() bool {
+	return p.target != nil && p.target.GOOS == "darwin" && p.target.GOARCH == "arm64"
+}
+
 // SetRecover sets the recover block for the function.
 func (p Function) SetRecover(blk BasicBlock) {
 	p.recov = blk

ssa/ssa_test.go

@@ -134,6 +134,36 @@ func TestNewFuncExLLVMUsed(t *testing.T) {
 	}
 }
 
+func TestNewFuncExFramePointerAttrOnDarwinArm64(t *testing.T) {
+	prog := NewProgram(&Target{GOOS: "darwin", GOARCH: "arm64"})
+	pkg := prog.NewPackage("main", "main")
+	sig := types.NewSignatureType(nil, nil, nil, nil, nil, false)
+
+	fn := pkg.NewFunc("Foo", sig, InGo)
+	attr := fn.impl.GetStringAttributeAtIndex(-1, "frame-pointer")
+	if attr.IsNil() {
+		t.Fatal("missing frame-pointer function attribute")
+	}
+	if got := pkg.String(); !strings.Contains(got, `"frame-pointer"="non-leaf"`) {
+		t.Fatalf("module missing frame-pointer attribute:\n%s", got)
+	}
+}
+
+func TestNewFuncExFramePointerAttrNotForcedOnOtherTargets(t *testing.T) {
+	prog := NewProgram(&Target{GOOS: "linux", GOARCH: "amd64"})
+	pkg := prog.NewPackage("main", "main")
+	sig := types.NewSignatureType(nil, nil, nil, nil, nil, false)
+
+	fn := pkg.NewFunc("Foo", sig, InGo)
+	attr := fn.impl.GetStringAttributeAtIndex(-1, "frame-pointer")
+	if !attr.IsNil() {
+		t.Fatal("unexpected frame-pointer function attribute")
+	}
+	if got := pkg.String(); strings.Contains(got, `"frame-pointer"="non-leaf"`) {
+		t.Fatalf("unexpected frame-pointer attribute in module:\n%s", got)
+	}
+}
+
 func TestSetBlock(t *testing.T) {
 	defer func() {
 		if r := recover(); r == nil {