106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package signed
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"git.lumeweb.com/LumeWeb/libs5-go/interfaces"
|
|
"git.lumeweb.com/LumeWeb/libs5-go/net"
|
|
"git.lumeweb.com/LumeWeb/libs5-go/protocol/base"
|
|
"github.com/vmihailenco/msgpack/v5"
|
|
)
|
|
|
|
var _ base.IncomingMessageTyped = (*HandshakeDone)(nil)
|
|
|
|
type HandshakeDone struct {
|
|
challenge []byte
|
|
networkId string
|
|
base.IncomingMessageTypedImpl
|
|
base.IncomingMessageHandler
|
|
supportedFeatures int
|
|
}
|
|
|
|
func (m *HandshakeDone) SetChallenge(challenge []byte) {
|
|
m.challenge = challenge
|
|
}
|
|
|
|
func (m *HandshakeDone) SetNetworkId(networkId string) {
|
|
m.networkId = networkId
|
|
}
|
|
|
|
func NewHandshakeDone() *HandshakeDone {
|
|
return &HandshakeDone{challenge: nil, networkId: "", supportedFeatures: -1}
|
|
}
|
|
|
|
func (h HandshakeDone) HandleMessage(node interfaces.Node, peer *net.Peer, verifyId bool) error {
|
|
if !node.IsStarted() {
|
|
err := (*peer).End()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if !bytes.Equal((*peer).Challenge(), h.challenge) {
|
|
return errors.New("Invalid challenge")
|
|
}
|
|
/*
|
|
if !verifyId {
|
|
(*peer).SetId(h)
|
|
} else {
|
|
if !peer.ID.Equals(pId) {
|
|
return errInvalidChallenge
|
|
}
|
|
}
|
|
|
|
peer.isConnected = true
|
|
|
|
supportedFeatures := data.UnpackInt()
|
|
|
|
if supportedFeatures != 3 {
|
|
return errors.New("Remote node does not support required features")
|
|
}
|
|
|
|
node.Services.P2P.Peers[peer.ID.String()] = peer
|
|
node.Services.P2P.ReconnectDelay[peer.ID.String()] = 1
|
|
|
|
connectionUrisCount := data.UnpackInt()
|
|
|
|
peer.ConnectionUris = make([]*url.URL, 0)
|
|
for i := 0; i < connectionUrisCount; i++ {
|
|
uriStr := data.UnpackString()
|
|
uri, err := url.Parse(uriStr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
peer.ConnectionUris = append(peer.ConnectionUris, uri)
|
|
}
|
|
|
|
// Log information - Assuming a logging method exists
|
|
node.Logger.Info(fmt.Sprintf("[+] %s (%s)", peer.ID.String(), peer.RenderLocationUri().String()))
|
|
|
|
// Send peer lists and emit 'peerConnected' event
|
|
// Assuming appropriate methods exist in node.Services.P2P
|
|
node.Services.P2P.SendPublicPeersToPeer(peer)
|
|
*/
|
|
return nil
|
|
}
|
|
|
|
func (h HandshakeDone) DecodeMessage(dec *msgpack.Decoder) error {
|
|
|
|
challenge, err := dec.DecodeBytes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
h.challenge = challenge
|
|
|
|
supportedFeatures, err := dec.DecodeInt()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
h.supportedFeatures = supportedFeatures
|
|
return nil
|
|
}
|