You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

193 lines
4.3 KiB

9 years ago
8 years ago
9 years ago
9 years ago
9 years ago
9 years ago
9 years ago
9 years ago
9 years ago
7 years ago
7 years ago
9 years ago
9 years ago
9 years ago
8 years ago
8 years ago
9 years ago
9 years ago
9 years ago
9 years ago
[common] use temp intead of {filePath}.new The problem with {filePath}.new is that it is not safe for concurrent use! Calling this function with the same params results in the following error: ``` panic: Panicked on a Crisis: rename /root/.tendermint_test/consensus_replay_test/priv_validator.json.new /root/.tendermint_test/consensus_replay_test/priv_validator.json: no such file or directory goroutine 47860 [running]: github.com/tendermint/tendermint/vendor/github.com/tendermint/tmlibs/common.PanicCrisis(0xcba800, 0xc42152d640) /go/src/github.com/tendermint/tendermint/vendor/github.com/tendermint/tmlibs/common/errors.go:33 +0x10f github.com/tendermint/tendermint/types.(*PrivValidator).save(0xc42235f2c0) /go/src/github.com/tendermint/tendermint/types/priv_validator.go:165 +0x159 github.com/tendermint/tendermint/types.(*PrivValidator).signBytesHRS(0xc42235f2c0, 0x6, 0x0, 0xc424e88f03, 0xc429908580, 0xca, 0x155, 0x80, 0xc424e88f00, 0x7f4ecafc88d0, ...) /go/src/github.com/tendermint/tendermint/types/priv_validator.go:249 +0x2bb github.com/tendermint/tendermint/types.(*PrivValidator).SignVote(0xc42235f2c0, 0xc4228c7460, 0xf, 0xc424e88f00, 0x0, 0x0) /go/src/github.com/tendermint/tendermint/types/priv_validator.go:186 +0x1a2 github.com/tendermint/tendermint/consensus.(*ConsensusState).signVote(0xc424efd520, 0xc400000002, 0xc422d5e3c0, 0x14, 0x20, 0x1, 0xc4247b6560, 0x14, 0x20, 0x0, ...) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:1556 +0x35e github.com/tendermint/tendermint/consensus.(*ConsensusState).signAddVote(0xc424efd520, 0x2, 0xc422d5e3c0, 0x14, 0x20, 0x1, 0xc4247b6560, 0x14, 0x20, 0xc42001b300) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:1568 +0x200 github.com/tendermint/tendermint/consensus.(*ConsensusState).enterPrecommit(0xc424efd520, 0x6, 0x0) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:1082 +0x13a4 github.com/tendermint/tendermint/consensus.(*ConsensusState).addVote(0xc424efd520, 0xc424e88780, 0x0, 0x0, 0x39, 0x1dc, 0x28c) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:1477 +0x1be5 github.com/tendermint/tendermint/consensus.(*ConsensusState).tryAddVote(0xc424efd520, 0xc424e88780, 0x0, 0x0, 0xd7fb00, 0xc42152ce00) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:1382 +0x93 github.com/tendermint/tendermint/consensus.(*ConsensusState).handleMsg(0xc424efd520, 0xcb58e0, 0xc42547feb8, 0x0, 0x0, 0x6, 0x0, 0x4, 0xed10ca07e, 0x3077bfea, ...) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:660 +0x9fc github.com/tendermint/tendermint/consensus.(*ConsensusState).receiveRoutine(0xc424efd520, 0x0) github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:615 +0x5f5 created by github.com/tendermint/tendermint/consensus.(*ConsensusState).OnStart github.com/tendermint/tendermint/consensus/_test/_obj_test/state.go:332 +0x4a7 exit status 2 FAIL github.com/tendermint/tendermint/consensus 76.644s make: *** [test_integrations] Error 1 ``` See https://github.com/tendermint/tendermint/pull/568
7 years ago
9 years ago
9 years ago
8 years ago
9 years ago
  1. package common
  2. import (
  3. "bufio"
  4. "fmt"
  5. "io"
  6. "io/ioutil"
  7. "os"
  8. "os/exec"
  9. "os/signal"
  10. "path/filepath"
  11. "strings"
  12. "syscall"
  13. )
  14. var gopath string
  15. // GoPath returns GOPATH env variable value. If it is not set, this function
  16. // will try to call `go env GOPATH` subcommand.
  17. func GoPath() string {
  18. if gopath != "" {
  19. return gopath
  20. }
  21. path := os.Getenv("GOPATH")
  22. if len(path) == 0 {
  23. goCmd := exec.Command("go", "env", "GOPATH")
  24. out, err := goCmd.Output()
  25. if err != nil {
  26. panic(fmt.Sprintf("failed to determine gopath: %v", err))
  27. }
  28. path = string(out)
  29. }
  30. gopath = path
  31. return path
  32. }
  33. // TrapSignal catches the SIGTERM and executes cb function. After that it exits
  34. // with code 1.
  35. func TrapSignal(cb func()) {
  36. c := make(chan os.Signal, 1)
  37. signal.Notify(c, os.Interrupt, syscall.SIGTERM)
  38. go func() {
  39. for sig := range c {
  40. fmt.Printf("captured %v, exiting...\n", sig)
  41. if cb != nil {
  42. cb()
  43. }
  44. os.Exit(1)
  45. }
  46. }()
  47. select {}
  48. }
  49. // Kill the running process by sending itself SIGTERM.
  50. func Kill() error {
  51. p, err := os.FindProcess(os.Getpid())
  52. if err != nil {
  53. return err
  54. }
  55. return p.Signal(syscall.SIGTERM)
  56. }
  57. func Exit(s string) {
  58. fmt.Printf(s + "\n")
  59. os.Exit(1)
  60. }
  61. func EnsureDir(dir string, mode os.FileMode) error {
  62. if _, err := os.Stat(dir); os.IsNotExist(err) {
  63. err := os.MkdirAll(dir, mode)
  64. if err != nil {
  65. return fmt.Errorf("Could not create directory %v. %v", dir, err)
  66. }
  67. }
  68. return nil
  69. }
  70. func IsDirEmpty(name string) (bool, error) {
  71. f, err := os.Open(name)
  72. if err != nil {
  73. if os.IsNotExist(err) {
  74. return true, err
  75. }
  76. // Otherwise perhaps a permission
  77. // error or some other error.
  78. return false, err
  79. }
  80. defer f.Close()
  81. _, err = f.Readdirnames(1) // Or f.Readdir(1)
  82. if err == io.EOF {
  83. return true, nil
  84. }
  85. return false, err // Either not empty or error, suits both cases
  86. }
  87. func FileExists(filePath string) bool {
  88. _, err := os.Stat(filePath)
  89. return !os.IsNotExist(err)
  90. }
  91. func ReadFile(filePath string) ([]byte, error) {
  92. return ioutil.ReadFile(filePath)
  93. }
  94. func MustReadFile(filePath string) []byte {
  95. fileBytes, err := ioutil.ReadFile(filePath)
  96. if err != nil {
  97. Exit(Fmt("MustReadFile failed: %v", err))
  98. return nil
  99. }
  100. return fileBytes
  101. }
  102. func WriteFile(filePath string, contents []byte, mode os.FileMode) error {
  103. return ioutil.WriteFile(filePath, contents, mode)
  104. }
  105. func MustWriteFile(filePath string, contents []byte, mode os.FileMode) {
  106. err := WriteFile(filePath, contents, mode)
  107. if err != nil {
  108. Exit(Fmt("MustWriteFile failed: %v", err))
  109. }
  110. }
  111. // WriteFileAtomic creates a temporary file with data and the perm given and
  112. // swaps it atomically with filename if successful.
  113. func WriteFileAtomic(filename string, data []byte, perm os.FileMode) error {
  114. var (
  115. dir = filepath.Dir(filename)
  116. tempFile = filepath.Join(dir, "write-file-atomic-"+RandStr(32))
  117. // Override in case it does exist, create in case it doesn't and force kernel
  118. // flush, which still leaves the potential of lingering disk cache.
  119. flag = os.O_WRONLY | os.O_CREATE | os.O_SYNC | os.O_TRUNC
  120. )
  121. f, err := os.OpenFile(tempFile, flag, perm)
  122. if err != nil {
  123. return err
  124. }
  125. // Clean up in any case. Defer stacking order is last-in-first-out.
  126. defer os.Remove(f.Name())
  127. defer f.Close()
  128. if n, err := f.Write(data); err != nil {
  129. return err
  130. } else if n < len(data) {
  131. return io.ErrShortWrite
  132. }
  133. return os.Rename(f.Name(), filename)
  134. }
  135. //--------------------------------------------------------------------------------
  136. func Tempfile(prefix string) (*os.File, string) {
  137. file, err := ioutil.TempFile("", prefix)
  138. if err != nil {
  139. PanicCrisis(err)
  140. }
  141. return file, file.Name()
  142. }
  143. func Tempdir(prefix string) (*os.File, string) {
  144. tempDir := os.TempDir() + "/" + prefix + RandStr(12)
  145. err := EnsureDir(tempDir, 0700)
  146. if err != nil {
  147. panic(Fmt("Error creating temp dir: %v", err))
  148. }
  149. dir, err := os.Open(tempDir)
  150. if err != nil {
  151. panic(Fmt("Error opening temp dir: %v", err))
  152. }
  153. return dir, tempDir
  154. }
  155. //--------------------------------------------------------------------------------
  156. func Prompt(prompt string, defaultValue string) (string, error) {
  157. fmt.Print(prompt)
  158. reader := bufio.NewReader(os.Stdin)
  159. line, err := reader.ReadString('\n')
  160. if err != nil {
  161. return defaultValue, err
  162. } else {
  163. line = strings.TrimSpace(line)
  164. if line == "" {
  165. return defaultValue, nil
  166. }
  167. return line, nil
  168. }
  169. }