Skip to content

Commit d2fe83d

Browse files
divankaralabe
authored andcommitted
whisper/mailserver: pass init error to the caller (#16671)
* whisper/mailserver: pass init error to the caller * whisper/mailserver: add returns to fmt.Errorf * whisper/mailserver: check err in mailserver init test
1 parent 16f3c31 commit d2fe83d

File tree

3 files changed

+14
-9
lines changed

3 files changed

+14
-9
lines changed

cmd/wnode/main.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,9 @@ func initialize() {
271271

272272
if *mailServerMode {
273273
shh.RegisterServer(&mailServer)
274-
mailServer.Init(shh, *argDBPath, msPassword, *argServerPoW)
274+
if err := mailServer.Init(shh, *argDBPath, msPassword, *argServerPoW); err != nil {
275+
utils.Fatalf("Failed to init MailServer: %s", err)
276+
}
275277
}
276278

277279
server = &p2p.Server{

whisper/mailserver/mailserver.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@ import (
2020
"encoding/binary"
2121
"fmt"
2222

23-
"github.com/ethereum/go-ethereum/cmd/utils"
2423
"github.com/ethereum/go-ethereum/common"
2524
"github.com/ethereum/go-ethereum/crypto"
2625
"github.com/ethereum/go-ethereum/log"
@@ -54,32 +53,33 @@ func NewDbKey(t uint32, h common.Hash) *DBKey {
5453
return &k
5554
}
5655

57-
func (s *WMailServer) Init(shh *whisper.Whisper, path string, password string, pow float64) {
56+
func (s *WMailServer) Init(shh *whisper.Whisper, path string, password string, pow float64) error {
5857
var err error
5958
if len(path) == 0 {
60-
utils.Fatalf("DB file is not specified")
59+
return fmt.Errorf("DB file is not specified")
6160
}
6261

6362
if len(password) == 0 {
64-
utils.Fatalf("Password is not specified for MailServer")
63+
return fmt.Errorf("password is not specified")
6564
}
6665

6766
s.db, err = leveldb.OpenFile(path, nil)
6867
if err != nil {
69-
utils.Fatalf("Failed to open DB file: %s", err)
68+
return fmt.Errorf("open DB file: %s", err)
7069
}
7170

7271
s.w = shh
7372
s.pow = pow
7473

7574
MailServerKeyID, err := s.w.AddSymKeyFromPassword(password)
7675
if err != nil {
77-
utils.Fatalf("Failed to create symmetric key for MailServer: %s", err)
76+
return fmt.Errorf("create symmetric key: %s", err)
7877
}
7978
s.key, err = s.w.GetSymKey(MailServerKeyID)
8079
if err != nil {
81-
utils.Fatalf("Failed to save symmetric key for MailServer")
80+
return fmt.Errorf("save symmetric key: %s", err)
8281
}
82+
return nil
8383
}
8484

8585
func (s *WMailServer) Close() {

whisper/mailserver/server_test.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,10 @@ func TestMailServer(t *testing.T) {
9292
shh = whisper.New(&whisper.DefaultConfig)
9393
shh.RegisterServer(&server)
9494

95-
server.Init(shh, dir, password, powRequirement)
95+
err = server.Init(shh, dir, password, powRequirement)
96+
if err != nil {
97+
t.Fatal(err)
98+
}
9699
defer server.Close()
97100

98101
keyID, err = shh.AddSymKeyFromPassword(password)

0 commit comments

Comments
 (0)