PeerAuthenticationResolver

dataRetriever/errors.go

@@ -229,3 +229,12 @@ var ErrNilHeartbeatPool = errors.New("nil heartbeat pool")
 
 // ErrNotFound signals that a data is missing
 var ErrNotFound = errors.New("data not found")
+
+// ErrNilNodesCoordinator signals a nil nodes coordinator has been provided
+var ErrNilNodesCoordinator = errors.New("nil nodes coordinator")
+
+// InvalidChunkIndex signals that an invalid chunk was provided
+var InvalidChunkIndex = errors.New("invalid chunk index")
+
+// ErrInvalidNumOfPeerAuthentication signals that an invalid number of peer authentication was provided
+var ErrInvalidNumOfPeerAuthentication = errors.New("invalid num of peer authentication")

dataRetriever/interface.go

@@ -156,6 +156,7 @@ type MiniBlocksResolver interface {
 // PeerAuthenticationResolver defines what a peer authentication resolver should do
 type PeerAuthenticationResolver interface {
 	Resolver
+	RequestDataFromChunk(chunkIndex uint32, epoch uint32) error
 	RequestDataFromHashArray(hashes [][]byte, epoch uint32) error
 }
 
@@ -420,3 +421,9 @@ type SelfShardIDProvider interface {
 	SelfId() uint32
 	IsInterfaceNil() bool
 }
+
+// NodesCoordinator provides Validator methods needed for the peer processing
+type NodesCoordinator interface {
+	GetAllEligibleValidatorsPublicKeys(epoch uint32) (map[uint32][][]byte, error)
+	IsInterfaceNil() bool
+}

dataRetriever/mock/nodesCoordinatorStub.go

@@ -0,0 +1,20 @@
+package mock
+
+// NodesCoordinatorStub -
+type NodesCoordinatorStub struct {
+	GetAllEligibleValidatorsPublicKeysCalled func(epoch uint32) (map[uint32][][]byte, error)
+}
+
+// GetAllEligibleValidatorsPublicKeys -
+func (nc *NodesCoordinatorStub) GetAllEligibleValidatorsPublicKeys(epoch uint32) (map[uint32][][]byte, error) {
+	if nc.GetAllEligibleValidatorsPublicKeysCalled != nil {
+		return nc.GetAllEligibleValidatorsPublicKeysCalled(epoch)
+	}
+
+	return nil, nil
+}
+
+// IsInterfaceNil -
+func (nc *NodesCoordinatorStub) IsInterfaceNil() bool {
+	return nc == nil
+}

dataRetriever/requestData.proto

@@ -19,6 +19,8 @@ enum RequestDataType {
 	NonceType      = 3;
 	// EpochType indicates that the request data object is of type epoch
 	EpochType      = 4;
+	// ChunkType indicates that the request data object is of type chunk
+	ChunkType      = 5;
 }
 
 // RequestData holds the requested data

dataRetriever/resolvers/common_test.go

@@ -11,3 +11,14 @@ func createRequestMsg(dataType dataRetriever.RequestDataType, val []byte) p2p.Me
 	buff, _ := marshalizer.Marshal(&dataRetriever.RequestData{Type: dataType, Value: val})
 	return &mock.P2PMessageMock{DataField: buff}
 }
+
+func createRequestMsgWithChunkIndex(dataType dataRetriever.RequestDataType, val []byte, epoch uint32, chunkIndex uint32) p2p.MessageP2P {
+	marshalizer := &mock.MarshalizerMock{}
+	buff, _ := marshalizer.Marshal(&dataRetriever.RequestData{
+		Type:       dataType,
+		Value:      val,
+		Epoch:      epoch,
+		ChunkIndex: chunkIndex,
+	})
+	return &mock.P2PMessageMock{DataField: buff}
+}