Commit 226b12d1 authored by Petar Radovic's avatar Petar Radovic

renamed handshake proto messages

parent da8b085c
...@@ -6,6 +6,7 @@ package handshake ...@@ -6,6 +6,7 @@ package handshake
import ( import (
"fmt" "fmt"
"github.com/ethersphere/bee/pkg/logging" "github.com/ethersphere/bee/pkg/logging"
"github.com/ethersphere/bee/pkg/p2p" "github.com/ethersphere/bee/pkg/p2p"
"github.com/ethersphere/bee/pkg/p2p/libp2p/internal/handshake/pb" "github.com/ethersphere/bee/pkg/p2p/libp2p/internal/handshake/pb"
...@@ -35,8 +36,8 @@ func New(overlay swarm.Address, networkID int32, logger logging.Logger) *Service ...@@ -35,8 +36,8 @@ func New(overlay swarm.Address, networkID int32, logger logging.Logger) *Service
func (s *Service) Handshake(stream p2p.Stream) (i *Info, err error) { func (s *Service) Handshake(stream p2p.Stream) (i *Info, err error) {
w, r := protobuf.NewWriterAndReader(stream) w, r := protobuf.NewWriterAndReader(stream)
var resp pb.ShakeHandAck var resp pb.SynPlusAck
if err := w.WriteMsg(&pb.ShakeHand{ if err := w.WriteMsg(&pb.Syn{
Address: s.overlay.Bytes(), Address: s.overlay.Bytes(),
NetworkID: s.networkID, NetworkID: s.networkID,
}); err != nil { }); err != nil {
...@@ -47,18 +48,18 @@ func (s *Service) Handshake(stream p2p.Stream) (i *Info, err error) { ...@@ -47,18 +48,18 @@ func (s *Service) Handshake(stream p2p.Stream) (i *Info, err error) {
return nil, fmt.Errorf("read message: %w", err) return nil, fmt.Errorf("read message: %w", err)
} }
if err := w.WriteMsg(&pb.Ack{Address: resp.ShakeHand.Address}); err != nil { if err := w.WriteMsg(&pb.Ack{Address: resp.Syn.Address}); err != nil {
return nil, fmt.Errorf("ack: write message: %w", err) return nil, fmt.Errorf("ack: write message: %w", err)
} }
address := swarm.NewAddress(resp.ShakeHand.Address) address := swarm.NewAddress(resp.Syn.Address)
s.logger.Tracef("handshake finished for peer %s", address) s.logger.Tracef("handshake finished for peer %s", address)
return &Info{ return &Info{
Address: address, Address: address,
NetworkID: resp.ShakeHand.NetworkID, NetworkID: resp.Syn.NetworkID,
Light: resp.ShakeHand.Light, Light: resp.Syn.Light,
}, nil }, nil
} }
...@@ -66,13 +67,13 @@ func (s *Service) Handle(stream p2p.Stream) (i *Info, err error) { ...@@ -66,13 +67,13 @@ func (s *Service) Handle(stream p2p.Stream) (i *Info, err error) {
w, r := protobuf.NewWriterAndReader(stream) w, r := protobuf.NewWriterAndReader(stream)
defer stream.Close() defer stream.Close()
var req pb.ShakeHand var req pb.Syn
if err := r.ReadMsg(&req); err != nil { if err := r.ReadMsg(&req); err != nil {
return nil, fmt.Errorf("read message: %w", err) return nil, fmt.Errorf("read message: %w", err)
} }
if err := w.WriteMsg(&pb.ShakeHandAck{ if err := w.WriteMsg(&pb.SynPlusAck{
ShakeHand: &pb.ShakeHand{ Syn: &pb.Syn{
Address: s.overlay.Bytes(), Address: s.overlay.Bytes(),
NetworkID: s.networkID, NetworkID: s.networkID,
}, },
......
...@@ -41,8 +41,8 @@ func TestHandshake(t *testing.T) { ...@@ -41,8 +41,8 @@ func TestHandshake(t *testing.T) {
stream2 := mock.NewStream(&buffer2, &buffer1) stream2 := mock.NewStream(&buffer2, &buffer1)
w, r := protobuf.NewWriterAndReader(stream2) w, r := protobuf.NewWriterAndReader(stream2)
if err := w.WriteMsg(&pb.ShakeHandAck{ if err := w.WriteMsg(&pb.SynPlusAck{
ShakeHand: &pb.ShakeHand{ Syn: &pb.Syn{
Address: expectedInfo.Address.Bytes(), Address: expectedInfo.Address.Bytes(),
NetworkID: expectedInfo.NetworkID, NetworkID: expectedInfo.NetworkID,
Light: expectedInfo.Light, Light: expectedInfo.Light,
...@@ -64,7 +64,7 @@ func TestHandshake(t *testing.T) { ...@@ -64,7 +64,7 @@ func TestHandshake(t *testing.T) {
} }
}) })
t.Run("ERROR - shakehand write error ", func(t *testing.T) { t.Run("ERROR - Syn write error ", func(t *testing.T) {
testErr := errors.New("test error") testErr := errors.New("test error")
expectedErr := fmt.Errorf("write message: %w", testErr) expectedErr := fmt.Errorf("write message: %w", testErr)
stream := &mock.StreamMock{} stream := &mock.StreamMock{}
...@@ -79,7 +79,7 @@ func TestHandshake(t *testing.T) { ...@@ -79,7 +79,7 @@ func TestHandshake(t *testing.T) {
} }
}) })
t.Run("ERROR - shakehand read error ", func(t *testing.T) { t.Run("ERROR - Syn read error ", func(t *testing.T) {
testErr := errors.New("test error") testErr := errors.New("test error")
expectedErr := fmt.Errorf("read message: %w", testErr) expectedErr := fmt.Errorf("read message: %w", testErr)
stream := mock.NewStream(nil, &bytes.Buffer{}) stream := mock.NewStream(nil, &bytes.Buffer{})
...@@ -110,8 +110,8 @@ func TestHandshake(t *testing.T) { ...@@ -110,8 +110,8 @@ func TestHandshake(t *testing.T) {
stream2 := mock.NewStream(&buffer2, &buffer1) stream2 := mock.NewStream(&buffer2, &buffer1)
w, _ := protobuf.NewWriterAndReader(stream2) w, _ := protobuf.NewWriterAndReader(stream2)
if err := w.WriteMsg(&pb.ShakeHandAck{ if err := w.WriteMsg(&pb.SynPlusAck{
ShakeHand: &pb.ShakeHand{ Syn: &pb.Syn{
Address: expectedInfo.Address.Bytes(), Address: expectedInfo.Address.Bytes(),
NetworkID: expectedInfo.NetworkID, NetworkID: expectedInfo.NetworkID,
Light: expectedInfo.Light, Light: expectedInfo.Light,
...@@ -157,7 +157,7 @@ func TestHandle(t *testing.T) { ...@@ -157,7 +157,7 @@ func TestHandle(t *testing.T) {
stream2 := mock.NewStream(&buffer2, &buffer1) stream2 := mock.NewStream(&buffer2, &buffer1)
w, _ := protobuf.NewWriterAndReader(stream2) w, _ := protobuf.NewWriterAndReader(stream2)
if err := w.WriteMsg(&pb.ShakeHand{ if err := w.WriteMsg(&pb.Syn{
Address: node2Info.Address.Bytes(), Address: node2Info.Address.Bytes(),
NetworkID: node2Info.NetworkID, NetworkID: node2Info.NetworkID,
Light: node2Info.Light, Light: node2Info.Light,
...@@ -177,15 +177,15 @@ func TestHandle(t *testing.T) { ...@@ -177,15 +177,15 @@ func TestHandle(t *testing.T) {
testInfo(t, *res, node2Info) testInfo(t, *res, node2Info)
_, r := protobuf.NewWriterAndReader(stream2) _, r := protobuf.NewWriterAndReader(stream2)
var got pb.ShakeHandAck var got pb.SynPlusAck
if err := r.ReadMsg(&got); err != nil { if err := r.ReadMsg(&got); err != nil {
t.Fatal(err) t.Fatal(err)
} }
testInfo(t, nodeInfo, Info{ testInfo(t, nodeInfo, Info{
Address: swarm.NewAddress(got.ShakeHand.Address), Address: swarm.NewAddress(got.Syn.Address),
NetworkID: got.ShakeHand.NetworkID, NetworkID: got.Syn.NetworkID,
Light: got.ShakeHand.Light, Light: got.Syn.Light,
}) })
}) })
...@@ -211,7 +211,7 @@ func TestHandle(t *testing.T) { ...@@ -211,7 +211,7 @@ func TestHandle(t *testing.T) {
stream := mock.NewStream(&buffer, &buffer) stream := mock.NewStream(&buffer, &buffer)
stream.SetWriteErr(testErr, 1) stream.SetWriteErr(testErr, 1)
w, _ := protobuf.NewWriterAndReader(stream) w, _ := protobuf.NewWriterAndReader(stream)
if err := w.WriteMsg(&pb.ShakeHand{ if err := w.WriteMsg(&pb.Syn{
Address: node1Addr.Bytes(), Address: node1Addr.Bytes(),
NetworkID: 0, NetworkID: 0,
Light: false, Light: false,
...@@ -244,7 +244,7 @@ func TestHandle(t *testing.T) { ...@@ -244,7 +244,7 @@ func TestHandle(t *testing.T) {
stream2 := mock.NewStream(&buffer2, &buffer1) stream2 := mock.NewStream(&buffer2, &buffer1)
stream1.SetReadErr(testErr, 1) stream1.SetReadErr(testErr, 1)
w, _ := protobuf.NewWriterAndReader(stream2) w, _ := protobuf.NewWriterAndReader(stream2)
if err := w.WriteMsg(&pb.ShakeHand{ if err := w.WriteMsg(&pb.Syn{
Address: node2Info.Address.Bytes(), Address: node2Info.Address.Bytes(),
NetworkID: node2Info.NetworkID, NetworkID: node2Info.NetworkID,
Light: node2Info.Light, Light: node2Info.Light,
......
...@@ -5,14 +5,14 @@ syntax = "proto3"; ...@@ -5,14 +5,14 @@ syntax = "proto3";
package pb; package pb;
message ShakeHand { message Syn {
bytes Address = 1; bytes Address = 1;
int32 NetworkID = 2; int32 NetworkID = 2;
bool Light = 3; bool Light = 3;
} }
message ShakeHandAck { message SynPlusAck {
ShakeHand ShakeHand = 1; Syn Syn = 1;
Ack Ack = 2; Ack Ack = 2;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment