feat(prover): implementing --graffiti flag for prover as input to b…

…lock evidence (#209)

Co-authored-by: jeff <[email protected]>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: David <[email protected]>

cmd/flags/prover.go

@@ -59,6 +59,12 @@ var (
 		Category: proverCategory,
 		Value:    false,
 	}
+	Graffiti = &cli.StringFlag{
+		Name:     "graffiti",
+		Usage:    "When string is passed, adds additional graffiti info to proof evidence",
+		Category: proverCategory,
+		Value:    "",
+	}
 )
 
 // All prover flags.
@@ -74,4 +80,5 @@ var ProverFlags = MergeFlags(CommonFlags, []cli.Flag{
 	Dummy,
 	RandomDummyProofDelay,
 	OracleProver,
+	Graffiti,
 })

prover/config.go

@@ -28,6 +28,7 @@ type Config struct {
 	MaxConcurrentProvingJobs        uint
 	Dummy                           bool
 	OracleProver                    bool
+	Graffiti                        string
 	RandomDummyProofDelayLowerBound *time.Duration
 	RandomDummyProofDelayUpperBound *time.Duration
 }
@@ -89,6 +90,7 @@ func NewConfigFromCliContext(c *cli.Context) (*Config, error) {
 		MaxConcurrentProvingJobs:        c.Uint(flags.MaxConcurrentProvingJobs.Name),
 		Dummy:                           c.Bool(flags.Dummy.Name),
 		OracleProver:                    c.Bool(flags.OracleProver.Name),
+		Graffiti:                        c.String(flags.Graffiti.Name),
 		RandomDummyProofDelayLowerBound: randomDummyProofDelayLowerBound,
 		RandomDummyProofDelayUpperBound: randomDummyProofDelayUpperBound,
 	}, nil

prover/config_test.go

@@ -30,6 +30,7 @@ func (s *ProverTestSuite) TestNewConfigFromCliContext() {
 		&cli.BoolFlag{Name: flags.Dummy.Name},
 		&cli.StringFlag{Name: flags.RandomDummyProofDelay.Name},
 		&cli.BoolFlag{Name: flags.OracleProver.Name},
+		&cli.StringFlag{Name: flags.Graffiti.Name},
 	}
 	app.Action = func(ctx *cli.Context) error {
 		c, err := NewConfigFromCliContext(ctx)
@@ -48,6 +49,7 @@ func (s *ProverTestSuite) TestNewConfigFromCliContext() {
 		s.Equal(time.Hour, *c.RandomDummyProofDelayUpperBound)
 		s.True(c.Dummy)
 		s.True(c.OracleProver)
+		s.Equal("", c.Graffiti)
 		s.Nil(new(Prover).InitFromCli(context.Background(), ctx))
 
 		return err
@@ -65,5 +67,6 @@ func (s *ProverTestSuite) TestNewConfigFromCliContext() {
 		"-" + flags.Dummy.Name,
 		"-" + flags.RandomDummyProofDelay.Name, "30m-1h",
 		"-" + flags.OracleProver.Name,
+		"-" + flags.Graffiti.Name, "",
 	}))
 }

prover/proof_submitter/valid_proof_submitter.go

@@ -32,6 +32,7 @@ type ValidProofSubmitter struct {
 	proverAddress     common.Address
 	mutex             *sync.Mutex
 	isOracle          bool
+	graffiti          [32]byte
 }
 
 // NewValidProofSubmitter creates a new ValidProofSubmitter instance.
@@ -43,12 +44,16 @@ func NewValidProofSubmitter(
 	proverPrivKey *ecdsa.PrivateKey,
 	mutex *sync.Mutex,
 	isOracle bool,
+	graffiti string,
 ) (*ValidProofSubmitter, error) {
 	anchorValidator, err := anchorTxValidator.New(taikoL2Address, rpc.L2ChainID, rpc)
 	if err != nil {
 		return nil, err
 	}
 
+	var bytes [32]byte
+	copy(bytes[:], []byte(graffiti))
+
 	return &ValidProofSubmitter{
 		rpc:               rpc,
 		proofProducer:     proofProducer,
@@ -58,6 +63,7 @@ func NewValidProofSubmitter(
 		proverAddress:     crypto.PubkeyToAddress(proverPrivKey.PublicKey),
 		mutex:             mutex,
 		isOracle:          isOracle,
+		graffiti:          bytes,
 	}, nil
 }
 
@@ -102,6 +108,7 @@ func (s *ValidProofSubmitter) SubmitProof(
 		"beneficiary", proofWithHeader.Meta.Beneficiary,
 		"hash", proofWithHeader.Header.Hash(),
 		"proof", common.Bytes2Hex(proofWithHeader.ZkProof),
+		"graffiti", string(s.graffiti[:]),
 	)
 	var (
 		blockID = proofWithHeader.BlockID
@@ -162,7 +169,7 @@ func (s *ValidProofSubmitter) SubmitProof(
 		ParentHash:    block.ParentHash(),
 		BlockHash:     block.Hash(),
 		SignalRoot:    signalRoot,
-		Graffiti:      [32]byte{},
+		Graffiti:      s.graffiti,
 		ParentGasUsed: uint32(parent.GasUsed()),
 		GasUsed:       uint32(block.GasUsed()),
 		Proof:         zkProof,

prover/proof_submitter/valid_proof_submitter_test.go

@@ -46,6 +46,7 @@ func (s *ProofSubmitterTestSuite) SetupTest() {
 		l1ProverPrivKey,
 		&sync.Mutex{},
 		false,
+		"test",
 	)
 	s.Nil(err)
 

prover/prover.go

@@ -160,6 +160,7 @@ func InitFromConfig(ctx context.Context, p *Prover, cfg *Config) (err error) {
 		p.cfg.L1ProverPrivKey,
 		p.submitProofTxMutex,
 		p.cfg.OracleProver,
+		p.cfg.Graffiti,
 	); err != nil {
 		return err
 	}