package main import ( "bufio" "encoding/hex" "errors" "fmt" "io" "os" "strings" "github.com/spf13/cobra" "github.com/tendermint/tendermint/libs/log" tmos "github.com/tendermint/tendermint/libs/os" abcicli "github.com/tendermint/tendermint/abci/client" "github.com/tendermint/tendermint/abci/example/code" "github.com/tendermint/tendermint/abci/example/counter" "github.com/tendermint/tendermint/abci/example/kvstore" "github.com/tendermint/tendermint/abci/server" servertest "github.com/tendermint/tendermint/abci/tests/server" "github.com/tendermint/tendermint/abci/types" "github.com/tendermint/tendermint/abci/version" "github.com/tendermint/tendermint/proto/tendermint/crypto" ) // client is a global variable so it can be reused by the console var ( client abcicli.Client logger log.Logger ) // flags var ( // global flagAddress string flagAbci string flagVerbose bool // for the println output flagLogLevel string // for the logger // query flagPath string flagHeight int flagProve bool // counter flagSerial bool // kvstore flagPersist string ) var RootCmd = &cobra.Command{ Use: "abci-cli", Short: "the ABCI CLI tool wraps an ABCI client", Long: "the ABCI CLI tool wraps an ABCI client and is used for testing ABCI servers", PersistentPreRunE: func(cmd *cobra.Command, args []string) error { switch cmd.Use { case "counter", "kvstore": // for the examples apps, don't pre-run return nil case "version": // skip running for version command return nil } if logger == nil { allowLevel, err := log.AllowLevel(flagLogLevel) if err != nil { return err } logger = log.NewFilter(log.NewTMLogger(log.NewSyncWriter(os.Stdout)), allowLevel) } if client == nil { var err error client, err = abcicli.NewClient(flagAddress, flagAbci, false) if err != nil { return err } client.SetLogger(logger.With("module", "abci-client")) if err := client.Start(); err != nil { return err } } return nil }, } // Structure for data passed to print response. type response struct { // generic abci response Data []byte Code uint32 Info string Log string Query *queryResponse } type queryResponse struct { Key []byte Value []byte Height int64 ProofOps *crypto.ProofOps } func Execute() error { addGlobalFlags() addCommands() return RootCmd.Execute() } func addGlobalFlags() { RootCmd.PersistentFlags().StringVarP(&flagAddress, "address", "", "tcp://0.0.0.0:26658", "address of application socket") RootCmd.PersistentFlags().StringVarP(&flagAbci, "abci", "", "socket", "either socket or grpc") RootCmd.PersistentFlags().BoolVarP(&flagVerbose, "verbose", "v", false, "print the command and results as if it were a console session") RootCmd.PersistentFlags().StringVarP(&flagLogLevel, "log_level", "", "debug", "set the logger level") } func addQueryFlags() { queryCmd.PersistentFlags().StringVarP(&flagPath, "path", "", "/store", "path to prefix query with") queryCmd.PersistentFlags().IntVarP(&flagHeight, "height", "", 0, "height to query the blockchain at") queryCmd.PersistentFlags().BoolVarP(&flagProve, "prove", "", false, "whether or not to return a merkle proof of the query result") } func addCounterFlags() { counterCmd.PersistentFlags().BoolVarP(&flagSerial, "serial", "", false, "enforce incrementing (serial) transactions") } func addKVStoreFlags() { kvstoreCmd.PersistentFlags().StringVarP(&flagPersist, "persist", "", "", "directory to use for a database") } func addCommands() { RootCmd.AddCommand(batchCmd) RootCmd.AddCommand(consoleCmd) RootCmd.AddCommand(echoCmd) RootCmd.AddCommand(infoCmd) RootCmd.AddCommand(setOptionCmd) RootCmd.AddCommand(deliverTxCmd) RootCmd.AddCommand(checkTxCmd) RootCmd.AddCommand(commitCmd) RootCmd.AddCommand(versionCmd) RootCmd.AddCommand(testCmd) addQueryFlags() RootCmd.AddCommand(queryCmd) // examples addCounterFlags() RootCmd.AddCommand(counterCmd) addKVStoreFlags() RootCmd.AddCommand(kvstoreCmd) } var batchCmd = &cobra.Command{ Use: "batch", Short: "run a batch of abci commands against an application", Long: `run a batch of abci commands against an application This command is run by piping in a file containing a series of commands you'd like to run: abci-cli batch < example.file where example.file looks something like: set_option serial on check_tx 0x00 check_tx 0xff deliver_tx 0x00 check_tx 0x00 deliver_tx 0x01 deliver_tx 0x04 info `, Args: cobra.ExactArgs(0), RunE: cmdBatch, } var consoleCmd = &cobra.Command{ Use: "console", Short: "start an interactive ABCI console for multiple commands", Long: `start an interactive ABCI console for multiple commands This command opens an interactive console for running any of the other commands without opening a new connection each time `, Args: cobra.ExactArgs(0), ValidArgs: []string{"echo", "info", "set_option", "deliver_tx", "check_tx", "commit", "query"}, RunE: cmdConsole, } var echoCmd = &cobra.Command{ Use: "echo", Short: "have the application echo a message", Long: "have the application echo a message", Args: cobra.ExactArgs(1), RunE: cmdEcho, } var infoCmd = &cobra.Command{ Use: "info", Short: "get some info about the application", Long: "get some info about the application", Args: cobra.ExactArgs(0), RunE: cmdInfo, } var setOptionCmd = &cobra.Command{ Use: "set_option", Short: "set an option on the application", Long: "set an option on the application", Args: cobra.ExactArgs(2), RunE: cmdSetOption, } var deliverTxCmd = &cobra.Command{ Use: "deliver_tx", Short: "deliver a new transaction to the application", Long: "deliver a new transaction to the application", Args: cobra.ExactArgs(1), RunE: cmdDeliverTx, } var checkTxCmd = &cobra.Command{ Use: "check_tx", Short: "validate a transaction", Long: "validate a transaction", Args: cobra.ExactArgs(1), RunE: cmdCheckTx, } var commitCmd = &cobra.Command{ Use: "commit", Short: "commit the application state and return the Merkle root hash", Long: "commit the application state and return the Merkle root hash", Args: cobra.ExactArgs(0), RunE: cmdCommit, } var versionCmd = &cobra.Command{ Use: "version", Short: "print ABCI console version", Long: "print ABCI console version", Args: cobra.ExactArgs(0), RunE: func(cmd *cobra.Command, args []string) error { fmt.Println(version.Version) return nil }, } var queryCmd = &cobra.Command{ Use: "query", Short: "query the application state", Long: "query the application state", Args: cobra.ExactArgs(1), RunE: cmdQuery, } var counterCmd = &cobra.Command{ Use: "counter", Short: "ABCI demo example", Long: "ABCI demo example", Args: cobra.ExactArgs(0), RunE: cmdCounter, } var kvstoreCmd = &cobra.Command{ Use: "kvstore", Short: "ABCI demo example", Long: "ABCI demo example", Args: cobra.ExactArgs(0), RunE: cmdKVStore, } var testCmd = &cobra.Command{ Use: "test", Short: "run integration tests", Long: "run integration tests", Args: cobra.ExactArgs(0), RunE: cmdTest, } // Generates new Args array based off of previous call args to maintain flag persistence func persistentArgs(line []byte) []string { // generate the arguments to run from original os.Args // to maintain flag arguments args := os.Args args = args[:len(args)-1] // remove the previous command argument if len(line) > 0 { // prevents introduction of extra space leading to argument parse errors args = append(args, strings.Split(string(line), " ")...) } return args } //-------------------------------------------------------------------------------- func compose(fs []func() error) error { if len(fs) == 0 { return nil } err := fs[0]() if err == nil { return compose(fs[1:]) } return err } func cmdTest(cmd *cobra.Command, args []string) error { return compose( []func() error{ func() error { return servertest.InitChain(client) }, func() error { return servertest.SetOption(client, "serial", "on") }, func() error { return servertest.Commit(client, nil) }, func() error { return servertest.DeliverTx(client, []byte("abc"), code.CodeTypeBadNonce, nil) }, func() error { return servertest.Commit(client, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x00}, code.CodeTypeOK, nil) }, func() error { return servertest.Commit(client, []byte{0, 0, 0, 0, 0, 0, 0, 1}) }, func() error { return servertest.DeliverTx(client, []byte{0x00}, code.CodeTypeBadNonce, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x01}, code.CodeTypeOK, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x00, 0x02}, code.CodeTypeOK, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x00, 0x03}, code.CodeTypeOK, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x00, 0x00, 0x04}, code.CodeTypeOK, nil) }, func() error { return servertest.DeliverTx(client, []byte{0x00, 0x00, 0x06}, code.CodeTypeBadNonce, nil) }, func() error { return servertest.Commit(client, []byte{0, 0, 0, 0, 0, 0, 0, 5}) }, }) } func cmdBatch(cmd *cobra.Command, args []string) error { bufReader := bufio.NewReader(os.Stdin) LOOP: for { line, more, err := bufReader.ReadLine() switch { case more: return errors.New("input line is too long") case err == io.EOF: break LOOP case len(line) == 0: continue case err != nil: return err } cmdArgs := persistentArgs(line) if err := muxOnCommands(cmd, cmdArgs); err != nil { return err } fmt.Println() } return nil } func cmdConsole(cmd *cobra.Command, args []string) error { for { fmt.Printf("> ") bufReader := bufio.NewReader(os.Stdin) line, more, err := bufReader.ReadLine() if more { return errors.New("input is too long") } else if err != nil { return err } pArgs := persistentArgs(line) if err := muxOnCommands(cmd, pArgs); err != nil { return err } } } func muxOnCommands(cmd *cobra.Command, pArgs []string) error { if len(pArgs) < 2 { return errors.New("expecting persistent args of the form: abci-cli [command] <...>") } // TODO: this parsing is fragile args := []string{} for i := 0; i < len(pArgs); i++ { arg := pArgs[i] // check for flags if strings.HasPrefix(arg, "-") { // if it has an equal, we can just skip if strings.Contains(arg, "=") { continue } // if its a boolean, we can just skip _, err := cmd.Flags().GetBool(strings.TrimLeft(arg, "-")) if err == nil { continue } // otherwise, we need to skip the next one too i++ continue } // append the actual arg args = append(args, arg) } var subCommand string var actualArgs []string if len(args) > 1 { subCommand = args[1] } if len(args) > 2 { actualArgs = args[2:] } cmd.Use = subCommand // for later print statements ... switch strings.ToLower(subCommand) { case "check_tx": return cmdCheckTx(cmd, actualArgs) case "commit": return cmdCommit(cmd, actualArgs) case "deliver_tx": return cmdDeliverTx(cmd, actualArgs) case "echo": return cmdEcho(cmd, actualArgs) case "info": return cmdInfo(cmd, actualArgs) case "query": return cmdQuery(cmd, actualArgs) case "set_option": return cmdSetOption(cmd, actualArgs) default: return cmdUnimplemented(cmd, pArgs) } } func cmdUnimplemented(cmd *cobra.Command, args []string) error { msg := "unimplemented command" if len(args) > 0 { msg += fmt.Sprintf(" args: [%s]", strings.Join(args, " ")) } printResponse(cmd, args, response{ Code: codeBad, Log: msg, }) fmt.Println("Available commands:") fmt.Printf("%s: %s\n", echoCmd.Use, echoCmd.Short) fmt.Printf("%s: %s\n", infoCmd.Use, infoCmd.Short) fmt.Printf("%s: %s\n", checkTxCmd.Use, checkTxCmd.Short) fmt.Printf("%s: %s\n", deliverTxCmd.Use, deliverTxCmd.Short) fmt.Printf("%s: %s\n", queryCmd.Use, queryCmd.Short) fmt.Printf("%s: %s\n", commitCmd.Use, commitCmd.Short) fmt.Printf("%s: %s\n", setOptionCmd.Use, setOptionCmd.Short) fmt.Println("Use \"[command] --help\" for more information about a command.") return nil } // Have the application echo a message func cmdEcho(cmd *cobra.Command, args []string) error { msg := "" if len(args) > 0 { msg = args[0] } res, err := client.EchoSync(msg) if err != nil { return err } printResponse(cmd, args, response{ Data: []byte(res.Message), }) return nil } // Get some info from the application func cmdInfo(cmd *cobra.Command, args []string) error { var version string if len(args) == 1 { version = args[0] } res, err := client.InfoSync(types.RequestInfo{Version: version}) if err != nil { return err } printResponse(cmd, args, response{ Data: []byte(res.Data), }) return nil } const codeBad uint32 = 10 // Set an option on the application func cmdSetOption(cmd *cobra.Command, args []string) error { if len(args) < 2 { printResponse(cmd, args, response{ Code: codeBad, Log: "want at least arguments of the form: ", }) return nil } key, val := args[0], args[1] _, err := client.SetOptionSync(types.RequestSetOption{Key: key, Value: val}) if err != nil { return err } printResponse(cmd, args, response{Log: "OK (SetOption doesn't return anything.)"}) // NOTE: Nothing to show... return nil } // Append a new tx to application func cmdDeliverTx(cmd *cobra.Command, args []string) error { if len(args) == 0 { printResponse(cmd, args, response{ Code: codeBad, Log: "want the tx", }) return nil } txBytes, err := stringOrHexToBytes(args[0]) if err != nil { return err } res, err := client.DeliverTxSync(types.RequestDeliverTx{Tx: txBytes}) if err != nil { return err } printResponse(cmd, args, response{ Code: res.Code, Data: res.Data, Info: res.Info, Log: res.Log, }) return nil } // Validate a tx func cmdCheckTx(cmd *cobra.Command, args []string) error { if len(args) == 0 { printResponse(cmd, args, response{ Code: codeBad, Info: "want the tx", }) return nil } txBytes, err := stringOrHexToBytes(args[0]) if err != nil { return err } res, err := client.CheckTxSync(types.RequestCheckTx{Tx: txBytes}) if err != nil { return err } printResponse(cmd, args, response{ Code: res.Code, Data: res.Data, Info: res.Info, Log: res.Log, }) return nil } // Get application Merkle root hash func cmdCommit(cmd *cobra.Command, args []string) error { res, err := client.CommitSync() if err != nil { return err } printResponse(cmd, args, response{ Data: res.Data, }) return nil } // Query application state func cmdQuery(cmd *cobra.Command, args []string) error { if len(args) == 0 { printResponse(cmd, args, response{ Code: codeBad, Info: "want the query", Log: "", }) return nil } queryBytes, err := stringOrHexToBytes(args[0]) if err != nil { return err } resQuery, err := client.QuerySync(types.RequestQuery{ Data: queryBytes, Path: flagPath, Height: int64(flagHeight), Prove: flagProve, }) if err != nil { return err } printResponse(cmd, args, response{ Code: resQuery.Code, Info: resQuery.Info, Log: resQuery.Log, Query: &queryResponse{ Key: resQuery.Key, Value: resQuery.Value, Height: resQuery.Height, ProofOps: resQuery.ProofOps, }, }) return nil } func cmdCounter(cmd *cobra.Command, args []string) error { app := counter.NewApplication(flagSerial) logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)) // Start the listener srv, err := server.NewServer(flagAddress, flagAbci, app) if err != nil { return err } srv.SetLogger(logger.With("module", "abci-server")) if err := srv.Start(); err != nil { return err } // Stop upon receiving SIGTERM or CTRL-C. tmos.TrapSignal(logger, func() { // Cleanup if err := srv.Stop(); err != nil { logger.Error("Error while stopping server", "err", err) } }) // Run forever. select {} } func cmdKVStore(cmd *cobra.Command, args []string) error { logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)) // Create the application - in memory or persisted to disk var app types.Application if flagPersist == "" { app = kvstore.NewApplication() } else { app = kvstore.NewPersistentKVStoreApplication(flagPersist) app.(*kvstore.PersistentKVStoreApplication).SetLogger(logger.With("module", "kvstore")) } // Start the listener srv, err := server.NewServer(flagAddress, flagAbci, app) if err != nil { return err } srv.SetLogger(logger.With("module", "abci-server")) if err := srv.Start(); err != nil { return err } // Stop upon receiving SIGTERM or CTRL-C. tmos.TrapSignal(logger, func() { // Cleanup if err := srv.Stop(); err != nil { logger.Error("Error while stopping server", "err", err) } }) // Run forever. select {} } //-------------------------------------------------------------------------------- func printResponse(cmd *cobra.Command, args []string, rsp response) { if flagVerbose { fmt.Println(">", cmd.Use, strings.Join(args, " ")) } // Always print the status code. if rsp.Code == types.CodeTypeOK { fmt.Printf("-> code: OK\n") } else { fmt.Printf("-> code: %d\n", rsp.Code) } if len(rsp.Data) != 0 { // Do no print this line when using the commit command // because the string comes out as gibberish if cmd.Use != "commit" { fmt.Printf("-> data: %s\n", rsp.Data) } fmt.Printf("-> data.hex: 0x%X\n", rsp.Data) } if rsp.Log != "" { fmt.Printf("-> log: %s\n", rsp.Log) } if rsp.Query != nil { fmt.Printf("-> height: %d\n", rsp.Query.Height) if rsp.Query.Key != nil { fmt.Printf("-> key: %s\n", rsp.Query.Key) fmt.Printf("-> key.hex: %X\n", rsp.Query.Key) } if rsp.Query.Value != nil { fmt.Printf("-> value: %s\n", rsp.Query.Value) fmt.Printf("-> value.hex: %X\n", rsp.Query.Value) } if rsp.Query.ProofOps != nil { fmt.Printf("-> proof: %#v\n", rsp.Query.ProofOps) } } } // NOTE: s is interpreted as a string unless prefixed with 0x func stringOrHexToBytes(s string) ([]byte, error) { if len(s) > 2 && strings.ToLower(s[:2]) == "0x" { b, err := hex.DecodeString(s[2:]) if err != nil { err = fmt.Errorf("error decoding hex argument: %s", err.Error()) return nil, err } return b, nil } if !strings.HasPrefix(s, "\"") || !strings.HasSuffix(s, "\"") { err := fmt.Errorf("invalid string arg: \"%s\". Must be quoted or a \"0x\"-prefixed hex string", s) return nil, err } return []byte(s[1 : len(s)-1]), nil }