From 342e85d37882c7cfa5a0c4ae261107375a4f931b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Magiera?= <magik6k@gmail.com>
Date: Sat, 29 Feb 2020 03:31:25 +0100
Subject: [PATCH] bench: flag to save commit2 input

---
 cmd/lotus-bench/main.go | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/cmd/lotus-bench/main.go b/cmd/lotus-bench/main.go
index 0070ae2df..3bf86e93f 100644
--- a/cmd/lotus-bench/main.go
+++ b/cmd/lotus-bench/main.go
@@ -54,6 +54,12 @@ type SealingResult struct {
 	Unseal     time.Duration
 }
 
+type Commit2In struct {
+	SectorNum int64
+	Phase1Out []byte
+	SectorSize uint64 // for stats
+}
+
 func main() {
 	logging.SetLogLevel("*", "INFO")
 
@@ -97,6 +103,10 @@ func main() {
 				Name:  "skip-unseal",
 				Usage: "skip the unseal portion of the benchmark",
 			},
+			&cli.StringFlag{
+				Name:  "save-commit2-input",
+				Usage: "Save commit2 input to a file",
+			},
 		},
 		Action: func(c *cli.Context) error {
 			if c.Bool("no-gpu") {
@@ -240,6 +250,23 @@ func main() {
 
 				log.Info("Generating PoRep for sector (2)")
 
+				if c.String("save-commit2-input") != "" {
+					c2in := Commit2In{
+						SectorNum: int64(i),
+						Phase1Out:  c1o,
+						SectorSize: 0,
+					}
+
+					b, err := json.Marshal(&c2in)
+					if err != nil {
+						return err
+					}
+
+					if err := ioutil.WriteFile(c.String("save-commit2-input"), b, 0664); err != nil {
+						log.Warnf("%+v", err)
+					}
+				}
+
 				proof, err := sb.SealCommit2(context.TODO(), i, c1o)
 				if err != nil {
 					return err