use of co.rsk.scoring.PeerScoringManager in project rskj by rsksmart.
the class NodeMessageHandlerTest method processInvalidPoWMessageUsingProcessor.
// TODO: Difficulty in RegTest is so small that this test will sometimes pass and other times fail
// This should be executed in a special mode where difficulty is high.
@Ignore
public void processInvalidPoWMessageUsingProcessor() throws UnknownHostException {
SimpleMessageChannel sender = new SimpleMessageChannel();
PeerScoringManager scoring = createPeerScoringManager();
SimpleBlockProcessor sbp = new SimpleBlockProcessor();
NodeMessageHandler processor = new NodeMessageHandler(config, sbp, null, null, null, null, scoring, new ProofOfWorkRule(config).setFallbackMiningEnabled(false));
Block block = BlockChainBuilder.ofSize(1, true).getBestBlock();
byte[] mergedMiningHeader = block.getBitcoinMergedMiningHeader();
// change merged mining nonce.
mergedMiningHeader[76] += 3;
Message message = new BlockMessage(block);
processor.processMessage(sender, message);
Assert.assertNotNull(sbp.getBlocks());
Assert.assertEquals(0, sbp.getBlocks().size());
Assert.assertFalse(scoring.isEmpty());
PeerScoring pscoring = scoring.getPeerScoring(sender.getPeerNodeID());
Assert.assertNotNull(pscoring);
Assert.assertFalse(pscoring.isEmpty());
Assert.assertEquals(1, pscoring.getTotalEventCounter());
Assert.assertEquals(1, pscoring.getEventCounter(EventType.INVALID_BLOCK));
}
use of co.rsk.scoring.PeerScoringManager in project rskj by rsksmart.
the class NodeMessageHandlerTest method processTransactionsMessage.
@Test
public void processTransactionsMessage() throws UnknownHostException {
PeerScoringManager scoring = createPeerScoringManager();
final SimpleChannelManager channelManager = new SimpleChannelManager();
TxHandler txmock = mock(TxHandler.class);
TransactionPool state = mock(TransactionPool.class);
Mockito.when(state.addTransactions(any())).thenAnswer(i -> i.getArguments()[0]);
BlockProcessor blockProcessor = mock(BlockProcessor.class);
Mockito.when(blockProcessor.hasBetterBlockToSync()).thenReturn(false);
final NodeMessageHandler handler = new NodeMessageHandler(config, blockProcessor, null, channelManager, state, txmock, scoring, new ProofOfWorkRule(config).setFallbackMiningEnabled(false));
final SimpleMessageChannel sender = new SimpleMessageChannel();
sender.setPeerNodeID(new NodeID(new byte[] { 1 }));
final SimpleMessageChannel sender2 = new SimpleMessageChannel();
sender2.setPeerNodeID(new NodeID(new byte[] { 2 }));
final List<Transaction> txs = TransactionUtils.getTransactions(10);
Mockito.when(txmock.retrieveValidTxs(any(List.class))).thenReturn(txs);
final TransactionsMessage message = new TransactionsMessage(txs);
handler.processMessage(sender, message);
Assert.assertNotNull(channelManager.getTransactions());
Assert.assertEquals(10, channelManager.getTransactions().size());
for (int k = 0; k < 10; k++) Assert.assertSame(txs.get(k), channelManager.getTransactions().get(k));
Assert.assertNotNull(channelManager.getLastSkip());
Assert.assertEquals(1, channelManager.getLastSkip().size());
Assert.assertTrue(channelManager.getLastSkip().contains(sender.getPeerNodeID()));
handler.processMessage(sender2, message);
Assert.assertNotNull(channelManager.getLastSkip());
Assert.assertEquals(2, channelManager.getLastSkip().size());
Assert.assertTrue(channelManager.getLastSkip().contains(sender.getPeerNodeID()));
Assert.assertTrue(channelManager.getLastSkip().contains(sender2.getPeerNodeID()));
Assert.assertFalse(scoring.isEmpty());
PeerScoring pscoring = scoring.getPeerScoring(sender.getPeerNodeID());
Assert.assertNotNull(pscoring);
Assert.assertFalse(pscoring.isEmpty());
Assert.assertEquals(10, pscoring.getTotalEventCounter());
Assert.assertEquals(10, pscoring.getEventCounter(EventType.VALID_TRANSACTION));
pscoring = scoring.getPeerScoring(sender2.getPeerNodeID());
Assert.assertNotNull(pscoring);
Assert.assertFalse(pscoring.isEmpty());
Assert.assertEquals(10, pscoring.getTotalEventCounter());
Assert.assertEquals(10, pscoring.getEventCounter(EventType.VALID_TRANSACTION));
}
use of co.rsk.scoring.PeerScoringManager in project rskj by rsksmart.
the class NodeMessageHandlerTest method processBlockMessageUsingProcessor.
@Test
public void processBlockMessageUsingProcessor() throws UnknownHostException {
SimpleMessageChannel sender = new SimpleMessageChannel();
PeerScoringManager scoring = createPeerScoringManager();
SimpleBlockProcessor sbp = new SimpleBlockProcessor();
NodeMessageHandler processor = new NodeMessageHandler(config, sbp, null, null, null, null, scoring, new ProofOfWorkRule(config).setFallbackMiningEnabled(false));
Block block = BlockChainBuilder.ofSize(1, true).getBestBlock();
Message message = new BlockMessage(block);
processor.processMessage(sender, message);
Assert.assertNotNull(sbp.getBlocks());
Assert.assertEquals(1, sbp.getBlocks().size());
Assert.assertSame(block, sbp.getBlocks().get(0));
Assert.assertFalse(scoring.isEmpty());
PeerScoring pscoring = scoring.getPeerScoring(sender.getPeerNodeID());
Assert.assertNotNull(pscoring);
Assert.assertFalse(pscoring.isEmpty());
Assert.assertEquals(1, pscoring.getTotalEventCounter());
Assert.assertEquals(1, pscoring.getEventCounter(EventType.VALID_BLOCK));
pscoring = scoring.getPeerScoring(sender.getAddress());
Assert.assertNotNull(pscoring);
Assert.assertFalse(pscoring.isEmpty());
Assert.assertEquals(1, pscoring.getTotalEventCounter());
Assert.assertEquals(1, pscoring.getEventCounter(EventType.VALID_BLOCK));
}
use of co.rsk.scoring.PeerScoringManager in project rskj by rsksmart.
the class Web3ImplScoringTest method banningUsingLocalIPV4AndMaskThrowsException.
@Test(expected = JsonRpcInvalidParamException.class)
public void banningUsingLocalIPV4AndMaskThrowsException() throws UnknownHostException {
PeerScoringManager peerScoringManager = createPeerScoringManager();
Web3Impl web3 = createWeb3(peerScoringManager);
// generate a random local IPv4 address
InetAddress address = generateLocalIPAddressV4();
Assert.assertTrue(peerScoringManager.hasGoodReputation(address));
web3.sco_banAddress(address.getHostAddress() + "/8");
}
use of co.rsk.scoring.PeerScoringManager in project rskj by rsksmart.
the class Web3ImplScoringTest method getEmptyPeerList.
@Test
public void getEmptyPeerList() {
PeerScoringManager peerScoringManager = createPeerScoringManager();
Web3Impl web3 = createWeb3(peerScoringManager);
PeerScoringInformation[] result = web3.sco_peerList();
Assert.assertNotNull(result);
Assert.assertEquals(0, result.length);
}
Aggregations