Zns 2 l2

cmd/flags/flags.go

@@ -71,4 +71,14 @@ var (
 		Usage: "Metrics HTTP server listening port",
 		Value: 6060,
 	}
+	CommandFlag = &cli.StringFlag{
+		Name:    "command",
+		Aliases: []string{"m"},
+		Usage:   "the config file",
+	}
+	RecoveryFromHistoryFlag = &cli.BoolFlag{
+		Name:  "recoveryFromHistory",
+		Value: true,
+		Usage: "read account and nft history record from the database, or read account and nft from the database",
+	}
 )

cmd/zkbnb/main.go

@@ -2,6 +2,8 @@ package main
 
 import (
 	"fmt"
+	"github.com/bnb-chain/zkbnb/tools/exodusexit/generateproof"
+	"github.com/bnb-chain/zkbnb/tools/exodusexit/performexodus"
 	"github.com/bnb-chain/zkbnb/tools/query"
 	"os"
 	"runtime"
@@ -184,6 +186,59 @@ func main() {
 					return apiserver.Run(cCtx.String(flags.ConfigFlag.Name))
 				},
 			},
+			{
+				Name:  "generateproof",
+				Usage: "Run generateproof service",
+				Flags: []cli.Flag{
+					flags.CommandFlag,
+					flags.ConfigFlag,
+				},
+				Action: func(cCtx *cli.Context) error {
+					if !cCtx.IsSet(flags.CommandFlag.Name) {
+						return cli.ShowSubcommandHelp(cCtx)
+					}
+					if !cCtx.IsSet(flags.ConfigFlag.Name) {
+						return cli.ShowSubcommandHelp(cCtx)
+					}
+					if cCtx.String(flags.CommandFlag.Name) == "init" {
+						return dbinitializer.InitializeExodusExit(
+							cCtx.String(flags.ConfigFlag.Name),
+						)
+					}
+					if cCtx.String(flags.CommandFlag.Name) == "run" {
+						err := dbinitializer.InitializeExodusExit(
+							cCtx.String(flags.ConfigFlag.Name),
+						)
+						if err != nil {
+							return err
+						}
+						return generateproof.Run(cCtx.String(flags.ConfigFlag.Name))
+					}
+					if cCtx.String(flags.CommandFlag.Name) == "continue" {
+						return generateproof.Run(cCtx.String(flags.ConfigFlag.Name))
+					}
+					return nil
+				},
+			},
+			{
+				Name:  "performexodus",
+				Usage: "Run performexodus service",
+				Flags: []cli.Flag{
+					flags.CommandFlag,
+					flags.ConfigFlag,
+				},
+				Action: func(cCtx *cli.Context) error {
+					if !cCtx.IsSet(flags.CommandFlag.Name) {
+						return cli.ShowSubcommandHelp(cCtx)
+					}
+					if !cCtx.IsSet(flags.ConfigFlag.Name) {
+						return cli.ShowSubcommandHelp(cCtx)
+					}
+					return performexodus.Run(cCtx.String(flags.ConfigFlag.Name))
+
+					return nil
+				},
+			},
 			// tools
 			{
 				Name:  "db",
@@ -226,6 +281,7 @@ func main() {
 							flags.BlockHeightFlag,
 							flags.ServiceNameFlag,
 							flags.BatchSizeFlag,
+							flags.RecoveryFromHistoryFlag,
 						},
 						Action: func(cCtx *cli.Context) error {
 							if !cCtx.IsSet(flags.ServiceNameFlag.Name) ||
@@ -238,6 +294,7 @@ func main() {
 								cCtx.Int64(flags.BlockHeightFlag.Name),
 								cCtx.String(flags.ServiceNameFlag.Name),
 								cCtx.Int(flags.BatchSizeFlag.Name),
+								cCtx.Bool(flags.RecoveryFromHistoryFlag.Name),
 							)
 							return nil
 						},
@@ -256,6 +313,7 @@ func main() {
 							flags.BlockHeightFlag,
 							flags.ServiceNameFlag,
 							flags.BatchSizeFlag,
+							flags.RecoveryFromHistoryFlag,
 						},
 						Action: func(cCtx *cli.Context) error {
 							if !cCtx.IsSet(flags.ServiceNameFlag.Name) ||
@@ -268,6 +326,7 @@ func main() {
 								cCtx.Int64(flags.BlockHeightFlag.Name),
 								cCtx.String(flags.ServiceNameFlag.Name),
 								cCtx.Int(flags.BatchSizeFlag.Name),
+								cCtx.Bool(flags.RecoveryFromHistoryFlag.Name),
 							)
 							return nil
 						},

common/abicoder/abicoder.go

@@ -0,0 +1,66 @@
+package abicoder
+
+import (
+	"fmt"
+	"github.com/ethereum/go-ethereum/accounts/abi"
+)
+
+func (d *ABIDecoder) getArguments(name string, data []byte) (Arguments, error) {
+	// since there can't be naming collisions with contracts and events,
+	// we need to decide whether we're calling a method or an event
+	args := make([]Argument, 0)
+	if method, ok := d.myabi.Methods[name]; ok {
+		if len(data)%32 != 0 {
+			return nil, fmt.Errorf("abicoder: improperly formatted output: %s - Bytes: [%+v]", string(data), data)
+		}
+		for _, input := range method.Inputs {
+			newInput := Argument{Name: input.Name, Type: input.Type, Indexed: input.Indexed}
+			args = append(args, newInput)
+		}
+	}
+	if args == nil {
+		return nil, fmt.Errorf("abicoder: could not locate named method or event: %s", name)
+	}
+	return args, nil
+}
+
+//
+// Unpack unpacks the output according to the abicoder specification.
+func (d *ABIDecoder) Unpack(name string, data []byte) ([]interface{}, error) {
+	args, err := d.getArguments(name, data)
+	if err != nil {
+		return nil, err
+	}
+	return args.Unpack(data)
+}
+
+func (d *ABIDecoder) UnpackIntoInterface(v interface{}, name string, data []byte) error {
+	args, err := d.getArguments(name, data)
+	if err != nil {
+		return err
+	}
+	unpacked, err := args.Unpack(data)
+	if err != nil {
+		return err
+	}
+	return args.Copy(v, unpacked)
+}
+
+//
+// UnpackIntoMap unpacks a log into the provided map[string]interface{}.
+func (d *ABIDecoder) UnpackIntoMap(v map[string]interface{}, name string, data []byte) (err error) {
+	args, err := d.getArguments(name, data)
+	if err != nil {
+		return err
+	}
+	return args.UnpackIntoMap(v, data)
+}
+
+// ABIDecoder ethereum transaction data decoder
+type ABIDecoder struct {
+	myabi abi.ABI
+}
+
+func NewABIDecoder(abi abi.ABI) *ABIDecoder {
+	return &ABIDecoder{myabi: abi}
+}