internal/cabi: option optimize

internal/build/build.go

@@ -303,8 +303,10 @@ func Do(args []string, conf *Config) ([]Package, error) {
 	})
 
 	buildMode := ssaBuildMode
+	cabiOptimize := true
 	if IsDbgEnabled() {
 		buildMode |= ssa.GlobalDebug
+		cabiOptimize = false
 	}
 	if !IsOptimizeEnabled() {
 		buildMode |= ssa.NaiveForm
@@ -324,7 +326,7 @@ func Do(args []string, conf *Config) ([]Package, error) {
 		needPyInit:   make(map[*packages.Package]bool),
 		buildConf:    conf,
 		crossCompile: export,
-		cTransformer: cabi.NewTransformer(prog, conf.AbiMode),
+		cTransformer: cabi.NewTransformer(prog, conf.AbiMode, cabiOptimize),
 	}
 	pkgs, err := buildAllPkgs(ctx, initial, verbose)
 	check(err)

internal/cabi/cabi.go

@@ -15,7 +15,7 @@ const (
 	ModeAllFunc
 )
 
-func NewTransformer(prog ssa.Program, mode Mode) *Transformer {
+func NewTransformer(prog ssa.Program, mode Mode, optimize bool) *Transformer {
 	target := prog.Target()
 	tr := &Transformer{
 		prog:     prog,
@@ -23,6 +23,7 @@ func NewTransformer(prog ssa.Program, mode Mode) *Transformer {
 		GOOS:     target.GOOS,
 		GOARCH:   target.GOARCH,
 		mode:     mode,
+		optimize: optimize,
 	}
 	switch target.GOARCH {
 	case "amd64":
@@ -48,6 +49,7 @@ type Transformer struct {
 	GOARCH   string
 	sys      TypeInfoSys
 	mode     Mode
+	optimize bool
 }
 
 func (p *Transformer) isCFunc(name string) bool {
@@ -369,13 +371,17 @@ func (p *Transformer) transformFuncBody(m llvm.Module, ctx llvm.Context, info *F
 			// store %typ %1, ptr %2, align 4
 			nv = b.CreateLoad(ti.Type, params[index], "")
 			// replace %0 to %2
+			if p.optimize {
 				replaceAllocaInstrs(fn.Param(i), params[index])
+			}
 		case AttrWidthType:
 			iptr := llvm.CreateAlloca(b, ti.Type1)
 			b.CreateStore(params[index], iptr)
 			ptr := b.CreateBitCast(iptr, llvm.PointerType(ti.Type, 0), "")
 			nv = b.CreateLoad(ti.Type, ptr, "")
+			if p.optimize {
 				replaceAllocaInstrs(fn.Param(i), ptr)
+			}
 		case AttrWidthType2:
 			typ := llvm.StructType([]llvm.Type{ti.Type1, ti.Type2}, false)
 			iptr := llvm.CreateAlloca(b, typ)
@@ -384,7 +390,9 @@ func (p *Transformer) transformFuncBody(m llvm.Module, ctx llvm.Context, info *F
 			b.CreateStore(params[index], b.CreateStructGEP(typ, iptr, 1, ""))
 			ptr := b.CreateBitCast(iptr, llvm.PointerType(ti.Type, 0), "")
 			nv = b.CreateLoad(ti.Type, ptr, "")
+			if p.optimize {
 				replaceAllocaInstrs(fn.Param(i), ptr)
+			}
 		case AttrExtract:
 			nsubs := ti.Type.StructElementTypesCount()
 			nv = llvm.Undef(ti.Type)
@@ -426,23 +434,28 @@ func (p *Transformer) transformFuncBody(m llvm.Module, ctx llvm.Context, info *F
 				// %2 = load %typ, ptr %1
 				// store %typ %2, ptr %0 # llvm.memcpy(ptr %0, ptr %1, i64 size, i1 false)
 				// ret void
+				if p.optimize {
 					if load := ret.IsALoadInst(); !load.IsNil() {
 						p.callMemcpy(m, ctx, b, params[0], ret.Operand(0), info.Return.Size)
-				} else {
-					b.CreateStore(ret, params[0])
+						rv = b.CreateRetVoid()
+						break
 					}
+				}
+				b.CreateStore(ret, params[0])
 				rv = b.CreateRetVoid()
 			case AttrWidthType, AttrWidthType2:
+				if p.optimize {
 					if load := ret.IsALoadInst(); !load.IsNil() {
 						iptr := b.CreateBitCast(ret.Operand(0), llvm.PointerType(nft.ReturnType(), 0), "")
 						rv = b.CreateRet(b.CreateLoad(nft.ReturnType(), iptr, ""))
-				} else {
+						break
+					}
+				}
 				ptr := llvm.CreateAlloca(b, info.Return.Type)
 				b.CreateStore(ret, ptr)
 				iptr := b.CreateBitCast(ptr, llvm.PointerType(nft.ReturnType(), 0), "")
 				rv = b.CreateRet(b.CreateLoad(nft.ReturnType(), iptr, ""))
 			}
-			}
 			instr.ReplaceAllUsesWith(rv)
 			instr.EraseFromParentAsInstruction()
 		}
@@ -478,7 +491,7 @@ func (p *Transformer) transformCallInstr(m llvm.Module, ctx llvm.Context, call l
 		case AttrVoid:
 			// none
 		case AttrPointer:
-			// check ptr
+			if p.optimize {
 				if rv := param.IsALoadInst(); !rv.IsNil() {
 					ptr := rv.Operand(0)
 					if p.sys.SupportByVal() {
@@ -488,11 +501,12 @@ func (p *Transformer) transformCallInstr(m llvm.Module, ctx llvm.Context, call l
 						p.callMemcpy(m, ctx, b, nptr, ptr, ti.Size)
 						nparams = append(nparams, nptr)
 					}
-			} else {
+					break
+				}
+			}
 			ptr := createAlloca(ti.Type)
 			b.CreateStore(param, ptr)
 			nparams = append(nparams, ptr)
-			}
 		case AttrWidthType:
 			ptr := createAlloca(ti.Type)
 			b.CreateStore(param, ptr)