utils.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. // Copyright 2017 The go-ethereum Authors
  2. // This file is part of go-ethereum.
  3. //
  4. // go-ethereum is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // go-ethereum is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU General Public License
  15. // along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
  16. package main
  17. import (
  18. "encoding/json"
  19. "fmt"
  20. "io/ioutil"
  21. "strings"
  22. "github.com/ethereum/go-ethereum/cmd/utils"
  23. "github.com/ethereum/go-ethereum/crypto"
  24. "gopkg.in/urfave/cli.v1"
  25. )
  26. // getPassphrase obtains a passphrase given by the user. It first checks the
  27. // --passfile command line flag and ultimately prompts the user for a
  28. // passphrase.
  29. func getPassphrase(ctx *cli.Context, confirmation bool) string {
  30. // Look for the --passwordfile flag.
  31. passphraseFile := ctx.String(passphraseFlag.Name)
  32. if passphraseFile != "" {
  33. content, err := ioutil.ReadFile(passphraseFile)
  34. if err != nil {
  35. utils.Fatalf("Failed to read password file '%s': %v",
  36. passphraseFile, err)
  37. }
  38. return strings.TrimRight(string(content), "\r\n")
  39. }
  40. // Otherwise prompt the user for the passphrase.
  41. return utils.GetPassPhrase("", confirmation)
  42. }
  43. // signHash is a helper function that calculates a hash for the given message
  44. // that can be safely used to calculate a signature from.
  45. //
  46. // The hash is calulcated as
  47. // keccak256("\x19Ethereum Signed Message:\n"${message length}${message}).
  48. //
  49. // This gives context to the signed message and prevents signing of transactions.
  50. func signHash(data []byte) []byte {
  51. msg := fmt.Sprintf("\x19Ethereum Signed Message:\n%d%s", len(data), data)
  52. return crypto.Keccak256([]byte(msg))
  53. }
  54. // mustPrintJSON prints the JSON encoding of the given object and
  55. // exits the program with an error message when the marshaling fails.
  56. func mustPrintJSON(jsonObject interface{}) {
  57. str, err := json.MarshalIndent(jsonObject, "", " ")
  58. if err != nil {
  59. utils.Fatalf("Failed to marshal JSON object: %v", err)
  60. }
  61. fmt.Println(string(str))
  62. }