2017-01-31 02:16:20 -08:00
|
|
|
// Copyright 2016 The go-ethereum Authors
|
|
|
|
// This file is part of go-ethereum.
|
|
|
|
//
|
|
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
// GNU General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
package mailserver
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"encoding/binary"
|
2017-02-22 04:10:07 -08:00
|
|
|
"fmt"
|
2017-01-31 02:16:20 -08:00
|
|
|
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
2017-02-13 04:15:20 -08:00
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
2017-02-22 04:10:07 -08:00
|
|
|
"github.com/ethereum/go-ethereum/log"
|
2017-01-31 02:16:20 -08:00
|
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
|
|
whisper "github.com/ethereum/go-ethereum/whisper/whisperv5"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb/util"
|
|
|
|
)
|
|
|
|
|
|
|
|
const MailServerKeyName = "958e04ab302fb36ad2616a352cbac79d"
|
|
|
|
|
|
|
|
type WMailServer struct {
|
|
|
|
db *leveldb.DB
|
|
|
|
w *whisper.Whisper
|
|
|
|
pow float64
|
|
|
|
key []byte
|
|
|
|
}
|
|
|
|
|
|
|
|
type DBKey struct {
|
|
|
|
timestamp uint32
|
|
|
|
hash common.Hash
|
|
|
|
raw []byte
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewDbKey(t uint32, h common.Hash) *DBKey {
|
|
|
|
const sz = common.HashLength + 4
|
|
|
|
var k DBKey
|
|
|
|
k.timestamp = t
|
|
|
|
k.hash = h
|
|
|
|
k.raw = make([]byte, sz)
|
|
|
|
binary.BigEndian.PutUint32(k.raw, k.timestamp)
|
|
|
|
copy(k.raw[4:], k.hash[:])
|
|
|
|
return &k
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *WMailServer) Init(shh *whisper.Whisper, path string, password string, pow float64) {
|
|
|
|
var err error
|
|
|
|
if len(path) == 0 {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Crit(fmt.Sprintf("DB file is not specified"))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(password) == 0 {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Crit(fmt.Sprintf("Password is not specified for MailServer"))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
s.db, err = leveldb.OpenFile(path, nil)
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Crit(fmt.Sprintf("Failed to open DB file: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
s.w = shh
|
|
|
|
s.pow = pow
|
|
|
|
|
|
|
|
err = s.w.AddSymKey(MailServerKeyName, []byte(password))
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Crit(fmt.Sprintf("Failed to create symmetric key for MailServer: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
s.key = s.w.GetSymKey(MailServerKeyName)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *WMailServer) Close() {
|
|
|
|
if s.db != nil {
|
|
|
|
s.db.Close()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *WMailServer) Archive(env *whisper.Envelope) {
|
|
|
|
key := NewDbKey(env.Expiry-env.TTL, env.Hash())
|
|
|
|
rawEnvelope, err := rlp.EncodeToBytes(env)
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprintf("rlp.EncodeToBytes failed: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
} else {
|
|
|
|
err = s.db.Put(key.raw, rawEnvelope, nil)
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprintf("Writing to DB failed: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *WMailServer) DeliverMail(peer *whisper.Peer, request *whisper.Envelope) {
|
2017-02-13 04:15:20 -08:00
|
|
|
if peer == nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprint("Whisper peer is nil"))
|
2017-01-31 02:16:20 -08:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2017-02-13 04:15:20 -08:00
|
|
|
ok, lower, upper, topic := s.validateRequest(peer.ID(), request)
|
|
|
|
if ok {
|
|
|
|
s.processRequest(peer, lower, upper, topic)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *WMailServer) processRequest(peer *whisper.Peer, lower, upper uint32, topic whisper.TopicType) []*whisper.Envelope {
|
|
|
|
ret := make([]*whisper.Envelope, 0)
|
2017-01-31 02:16:20 -08:00
|
|
|
var err error
|
|
|
|
var zero common.Hash
|
|
|
|
var empty whisper.TopicType
|
|
|
|
kl := NewDbKey(lower, zero)
|
|
|
|
ku := NewDbKey(upper, zero)
|
|
|
|
i := s.db.NewIterator(&util.Range{Start: kl.raw, Limit: ku.raw}, nil)
|
|
|
|
defer i.Release()
|
|
|
|
|
|
|
|
for i.Next() {
|
|
|
|
var envelope whisper.Envelope
|
|
|
|
err = rlp.DecodeBytes(i.Value(), &envelope)
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprintf("RLP decoding failed: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
if topic == empty || envelope.Topic == topic {
|
2017-02-13 04:15:20 -08:00
|
|
|
if peer == nil {
|
|
|
|
// used for test purposes
|
|
|
|
ret = append(ret, &envelope)
|
|
|
|
} else {
|
|
|
|
err = s.w.SendP2PDirect(peer, &envelope)
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprintf("Failed to send direct message to peer: %s", err))
|
2017-02-13 04:15:20 -08:00
|
|
|
return nil
|
|
|
|
}
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
err = i.Error()
|
|
|
|
if err != nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Error(fmt.Sprintf("Level DB iterator error: %s", err))
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
2017-02-13 04:15:20 -08:00
|
|
|
|
|
|
|
return ret
|
2017-01-31 02:16:20 -08:00
|
|
|
}
|
|
|
|
|
2017-02-13 04:15:20 -08:00
|
|
|
func (s *WMailServer) validateRequest(peerID []byte, request *whisper.Envelope) (bool, uint32, uint32, whisper.TopicType) {
|
2017-01-31 02:16:20 -08:00
|
|
|
var topic whisper.TopicType
|
|
|
|
if s.pow > 0.0 && request.PoW() < s.pow {
|
|
|
|
return false, 0, 0, topic
|
|
|
|
}
|
|
|
|
|
|
|
|
f := whisper.Filter{KeySym: s.key}
|
|
|
|
decrypted := request.Open(&f)
|
|
|
|
if decrypted == nil {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Warn(fmt.Sprintf("Failed to decrypt p2p request"))
|
2017-01-31 02:16:20 -08:00
|
|
|
return false, 0, 0, topic
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(decrypted.Payload) < 8 {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Warn(fmt.Sprintf("Undersized p2p request"))
|
2017-01-31 02:16:20 -08:00
|
|
|
return false, 0, 0, topic
|
|
|
|
}
|
|
|
|
|
2017-02-13 04:15:20 -08:00
|
|
|
src := crypto.FromECDSAPub(decrypted.Src)
|
|
|
|
if len(src)-len(peerID) == 1 {
|
|
|
|
src = src[1:]
|
|
|
|
}
|
|
|
|
if !bytes.Equal(peerID, src) {
|
2017-02-22 04:10:07 -08:00
|
|
|
log.Warn(fmt.Sprintf("Wrong signature of p2p request"))
|
2017-01-31 02:16:20 -08:00
|
|
|
return false, 0, 0, topic
|
|
|
|
}
|
|
|
|
|
|
|
|
lower := binary.BigEndian.Uint32(decrypted.Payload[:4])
|
|
|
|
upper := binary.BigEndian.Uint32(decrypted.Payload[4:8])
|
|
|
|
|
|
|
|
if len(decrypted.Payload) >= 8+whisper.TopicLength {
|
|
|
|
topic = whisper.BytesToTopic(decrypted.Payload[8:])
|
|
|
|
}
|
|
|
|
|
|
|
|
return true, lower, upper, topic
|
|
|
|
}
|