137 lines
3.6 KiB
Go
137 lines
3.6 KiB
Go
package mailservice
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
"encoding/binary"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/p2p/discover"
|
|
|
|
whisper "github.com/ethereum/go-ethereum/whisper/whisperv6"
|
|
)
|
|
|
|
const (
|
|
// defaultWorkTime is a work time reported in messages sent to MailServer nodes.
|
|
defaultWorkTime = 5
|
|
)
|
|
|
|
var (
|
|
// ErrInvalidMailServerPeer is returned when it fails to parse enode from params.
|
|
ErrInvalidMailServerPeer = errors.New("invalid mailServerPeer value")
|
|
// ErrInvalidSymKeyID is returned when it fails to get a symmetric key.
|
|
ErrInvalidSymKeyID = errors.New("invalid symKeyID value")
|
|
)
|
|
|
|
// PublicAPI defines a MailServer public API.
|
|
type PublicAPI struct {
|
|
provider ServiceProvider
|
|
log log.Logger
|
|
}
|
|
|
|
// NewPublicAPI returns a new PublicAPI.
|
|
func NewPublicAPI(provider ServiceProvider) *PublicAPI {
|
|
return &PublicAPI{
|
|
provider: provider,
|
|
log: log.New("package", "status-go/geth/mailservice.PublicAPI"),
|
|
}
|
|
}
|
|
|
|
// MessagesRequest is a payload send to a MailServer to get messages.
|
|
type MessagesRequest struct {
|
|
// MailServerPeer is MailServer's enode address.
|
|
MailServerPeer string `json:"mailServerPeer"`
|
|
|
|
// From is a lower bound of time range (optional).
|
|
// Default is 24 hours back from now.
|
|
From uint32 `json:"from"`
|
|
|
|
// To is a upper bound of time range (optional).
|
|
// Default is now.
|
|
To uint32 `json:"to"`
|
|
|
|
// Topic is a regular Whisper topic.
|
|
Topic whisper.TopicType `json:"topic"`
|
|
|
|
// SymKeyID is an ID of a symmetric key to authenticate to MailServer.
|
|
// It's derived from MailServer password.
|
|
SymKeyID string `json:"symKeyID"`
|
|
}
|
|
|
|
func setMessagesRequestDefaults(r *MessagesRequest) {
|
|
// set From and To defaults
|
|
if r.From == 0 && r.To == 0 {
|
|
r.From = uint32(time.Now().UTC().Add(-24 * time.Hour).Unix())
|
|
r.To = uint32(time.Now().UTC().Unix())
|
|
}
|
|
}
|
|
|
|
// RequestMessages sends a request for historic messages to a MailServer.
|
|
func (api *PublicAPI) RequestMessages(_ context.Context, r MessagesRequest) (bool, error) {
|
|
api.log.Info("RequestMessages", "request", r)
|
|
|
|
setMessagesRequestDefaults(&r)
|
|
|
|
shh, err := api.provider.WhisperService()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
node, err := api.provider.Node()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
mailServerNode, err := discover.ParseNode(r.MailServerPeer)
|
|
if err != nil {
|
|
return false, fmt.Errorf("%v: %v", ErrInvalidMailServerPeer, err)
|
|
}
|
|
|
|
symKey, err := shh.GetSymKey(r.SymKeyID)
|
|
if err != nil {
|
|
return false, fmt.Errorf("%v: %v", ErrInvalidSymKeyID, err)
|
|
}
|
|
|
|
envelope, err := makeEnvelop(makePayload(r), symKey, node.Server().PrivateKey, shh.MinPow())
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if err := shh.RequestHistoricMessages(mailServerNode.ID[:], envelope); err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// makeEnvelop makes an envelop for a historic messages request.
|
|
// Symmetric key is used to authenticate to MailServer.
|
|
// PK is the current node ID.
|
|
func makeEnvelop(payload []byte, symKey []byte, nodeID *ecdsa.PrivateKey, pow float64) (*whisper.Envelope, error) {
|
|
params := whisper.MessageParams{
|
|
PoW: pow,
|
|
Payload: payload,
|
|
KeySym: symKey,
|
|
WorkTime: defaultWorkTime,
|
|
Src: nodeID,
|
|
}
|
|
message, err := whisper.NewSentMessage(¶ms)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return message.Wrap(¶ms)
|
|
}
|
|
|
|
// makePayload makes a specific payload for MailServer to request historic messages.
|
|
func makePayload(r MessagesRequest) []byte {
|
|
// first 8 bytes are lowed and upper bounds as uint32
|
|
data := make([]byte, 8+whisper.TopicLength)
|
|
binary.BigEndian.PutUint32(data, r.From)
|
|
binary.BigEndian.PutUint32(data[4:], r.To)
|
|
copy(data[8:], r.Topic[:])
|
|
return data
|
|
}
|