package cli import ( "bufio" "encoding/hex" "encoding/json" "fmt" "io/ioutil" "os" "strings" "github.com/filecoin-project/go-address" types "github.com/filecoin-project/lotus/chain/types" "github.com/filecoin-project/lotus/chain/wallet" "github.com/filecoin-project/specs-actors/actors/crypto" "gopkg.in/urfave/cli.v2" ) var walletCmd = &cli.Command{ Name: "wallet", Usage: "Manage wallet", Subcommands: []*cli.Command{ walletNew, walletList, walletBalance, walletExport, walletImport, walletGetDefault, walletSetDefault, walletSign, walletVerify, }, } var walletNew = &cli.Command{ Name: "new", Usage: "Generate a new key of the given type", ArgsUsage: "[bls|secp256k1 (default secp256k1)]", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) t := cctx.Args().First() if t == "" { t = "secp256k1" } nk, err := api.WalletNew(ctx, wallet.ActSigType(t)) if err != nil { return err } fmt.Println(nk.String()) return nil }, } var walletList = &cli.Command{ Name: "list", Usage: "List wallet address", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) addrs, err := api.WalletList(ctx) if err != nil { return err } for _, addr := range addrs { fmt.Println(addr.String()) } return nil }, } var walletBalance = &cli.Command{ Name: "balance", Usage: "Get account balance", ArgsUsage: "[address]", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) var addr address.Address if cctx.Args().First() != "" { addr, err = address.NewFromString(cctx.Args().First()) } else { addr, err = api.WalletDefaultAddress(ctx) } if err != nil { return err } balance, err := api.WalletBalance(ctx, addr) if err != nil { return err } fmt.Printf("%s\n", types.FIL(balance)) return nil }, } var walletGetDefault = &cli.Command{ Name: "default", Usage: "Get default wallet address", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) addr, err := api.WalletDefaultAddress(ctx) if err != nil { return err } fmt.Printf("%s\n", addr.String()) return nil }, } var walletSetDefault = &cli.Command{ Name: "set-default", Usage: "Set default wallet address", ArgsUsage: "[address]", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) if !cctx.Args().Present() { return fmt.Errorf("must pass address to set as default") } addr, err := address.NewFromString(cctx.Args().First()) if err != nil { return err } return api.WalletSetDefault(ctx, addr) }, } var walletExport = &cli.Command{ Name: "export", Usage: "export keys", ArgsUsage: "[address]", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) if !cctx.Args().Present() { return fmt.Errorf("must specify key to export") } addr, err := address.NewFromString(cctx.Args().First()) if err != nil { return err } ki, err := api.WalletExport(ctx, addr) if err != nil { return err } b, err := json.Marshal(ki) if err != nil { return err } fmt.Println(hex.EncodeToString(b)) return nil }, } var walletImport = &cli.Command{ Name: "import", Usage: "import keys", ArgsUsage: "[ (optional, will read from stdin if omitted)]", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) var hexdata []byte if !cctx.Args().Present() || cctx.Args().First() == "-" { reader := bufio.NewReader(os.Stdin) fmt.Print("Enter private key: ") indata, err := reader.ReadBytes('\n') if err != nil { return err } hexdata = indata } else { fdata, err := ioutil.ReadFile(cctx.Args().First()) if err != nil { return err } hexdata = fdata } data, err := hex.DecodeString(strings.TrimSpace(string(hexdata))) if err != nil { return err } var ki types.KeyInfo if err := json.Unmarshal(data, &ki); err != nil { return err } addr, err := api.WalletImport(ctx, &ki) if err != nil { return err } fmt.Printf("imported key %s successfully!\n", addr) return nil }, } var walletSign = &cli.Command{ Name: "sign", Usage: "sign a message", ArgsUsage: " ", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) if !cctx.Args().Present() || cctx.NArg() != 2 { return fmt.Errorf("must specify signing address and message to sign") } addr, err := address.NewFromString(cctx.Args().First()) if err != nil { return err } msg, err := hex.DecodeString(cctx.Args().Get(1)) if err != nil { return err } sig, err := api.WalletSign(ctx, addr, msg) if err != nil { return err } sigBytes := append([]byte{byte(sig.Type)}, sig.Data...) fmt.Println(hex.EncodeToString(sigBytes)) return nil }, } var walletVerify = &cli.Command{ Name: "verify", Usage: "verify the signature of a message", ArgsUsage: " ", Action: func(cctx *cli.Context) error { api, closer, err := GetFullNodeAPI(cctx) if err != nil { return err } defer closer() ctx := ReqContext(cctx) if !cctx.Args().Present() || cctx.NArg() != 3 { return fmt.Errorf("must specify signing address, message, and signature to verify") } addr, err := address.NewFromString(cctx.Args().First()) if err != nil { return err } msg, err := hex.DecodeString(cctx.Args().Get(1)) if err != nil { return err } sigBytes, err := hex.DecodeString(cctx.Args().Get(2)) if err != nil { return err } var sig crypto.Signature if err := sig.UnmarshalBinary(sigBytes); err != nil { return err } if api.WalletVerify(ctx, addr, msg, &sig) { fmt.Println("valid") return nil } else { fmt.Println("invalid") return NewCliError("CLI Verify called with invalid signature") } }, }