private Message Launch(PeerConnector peerConnector, PingMessage message)
{
var payload = peerConnector.Execute(message.Serialize());
var result = _messageParser.Parse(payload.ToArray());
if (result.GetCommandName() != Constants.MessageNames.Pong)
{
return(null);
}
return(result as PongMessage);
}
private void Launch(PeerConnector peerConnector, InventoryMessage message)
{
var payload = peerConnector.Execute(message.Serialize()); // S : INVENTORY.
var response = _messageParser.Parse(payload.ToArray());
if (response.GetCommandName() == Constants.MessageNames.GetData) // R : GETDATA.
{
var inventories = Execute(response as GetDataMessage);
foreach (var inventory in inventories)
{
peerConnector.Execute(inventory.Serialize()); // S : TX & B.
}
}
}
private void RemovePeer(PeerConnector peerConnector, IpAddress ipAdr)
{
if (peerConnector == null)
{
throw new ArgumentNullException(nameof(peerConnector));
}
if (ipAdr == null)
{
throw new ArgumentNullException(nameof(ipAdr));
}
_peers = new ConcurrentBag <PeerConnector>(_peers.Except(new[] { peerConnector }));
_peersRepository.RemovePeer(ipAdr);
peerConnector.Dispose();
}
private void Launch(PeerConnector peerConnector, MemPoolMessage message)
{
var payload = peerConnector.Execute(message.Serialize()); // S : MEMPOOL.
var response = _messageParser.Parse(payload.ToArray());
if (response.GetCommandName() != Constants.MessageNames.Inventory)
{
return;
}
response = Execute(response as InventoryMessage); // S : GETDATA.
if (response == null)
{
return;
}
peerConnector.Execute(response.Serialize());
}
private Task ConnectToPeer(string host, ServiceFlags serviceFlag)
{
if (string.IsNullOrWhiteSpace(host))
{
throw new ArgumentNullException(nameof(host));
}
return(Task.Factory.StartNew(() =>
{
var peerConnector = new PeerConnector(_network, this, _messageCoordinator);
try
{
var manualResetEvent = new ManualResetEvent(false);
peerConnector.ConnectEvent += (s, i) =>
{
AddPeer(s, i);
manualResetEvent.Set();
};
if (_isSeedNode)
{
peerConnector.TimeOutEvent += RemovePeer;
}
else
{
peerConnector.TimeOutEvent += Disconnect;
}
peerConnector.Connect(host, serviceFlag);
manualResetEvent.WaitOne();
}
catch (PeerConnectorException)
{
if (_isSeedNode)
{
RemovePeer(peerConnector, peerConnector.GetCurrentIpAddress());
}
else
{
Disconnect();
}
throw;
}
}));
}
public Message Launch(PeerConnector peerConnector, Message message)
{
if (message.GetCommandName() == Constants.MessageNames.GetBlocks)
{
Launch(peerConnector, message as GetBlocksMessage);
}
if (message.GetCommandName() == Constants.MessageNames.Inventory)
{
Launch(peerConnector, message as InventoryMessage);
}
if (message.GetCommandName() == Constants.MessageNames.MemPool)
{
Launch(peerConnector, message as MemPoolMessage);
}
if (message.GetCommandName() == Constants.MessageNames.GetAddr)
{
return(Launch(peerConnector, message as GetAddressMessage));
}
if (message.GetCommandName() == Constants.MessageNames.Addr)
{
Launch(peerConnector, message as AddrMessage);
}
if (message.GetCommandName() == Constants.MessageNames.Version)
{
return(Launch(peerConnector, message as VersionMessage));
}
if (message.GetCommandName() == Constants.MessageNames.Ping)
{
return(Launch(peerConnector, message as PingMessage));
}
if (message.GetCommandName() == Constants.MessageNames.Block)
{
Launch(peerConnector, message as BlockMessage);
}
return(null);
}
private Message Launch(PeerConnector peerConnector, VersionMessage message)
{
var payload = peerConnector.Execute(message.Serialize());
var result = _messageParser.Parse(payload.ToArray());
if (result.GetCommandName() == Constants.MessageNames.Version)
{
var versionMessage = result as VersionMessage;
var verackMessage = new VerackMessage(message.MessageHeader.Network);
payload = peerConnector.Execute(verackMessage.Serialize());
result = _messageParser.Parse(payload.ToArray());
if (result.GetCommandName() == Constants.MessageNames.Verack)
{
return(result);
}
}
return(null);
}
private void Launch(PeerConnector peerConnector, GetBlocksMessage message)
{
var payload = peerConnector.Execute(message.Serialize());
var result = _messageParser.Parse(payload.ToArray());
if (result.GetCommandName() != Constants.MessageNames.Inventory)
{
return;
}
var getDataMessage = Execute(result as InventoryMessage);
if (getDataMessage == null)
{
return;
}
peerConnector.Execute(getDataMessage.Serialize());
}
public Message Receive(Message message, PeerConnector peer, P2PNetworkConnector p2pNetworkConnector)
{
var blockChain = _blockChainStore.GetBlockChain();
var smartContract = _smartContractStore.GetSmartContracts();
if (message.GetCommandName() == Constants.MessageNames.Version) // RETURNS VERSION.
{
var msg = message as VersionMessage;
var instance = PeersStore.Instance();
var transmittingNode = instance.GetMyIpAddress();
var receivingNode = msg.TransmittingNode;
return(new VersionMessage(transmittingNode, receivingNode, msg.Nonce, msg.UserAgent, msg.StartHeight, msg.Relay, msg.MessageHeader.Network));
}
if (message.GetCommandName() == Constants.MessageNames.Ping) // RETURNS PONG.
{
var msg = message as PingMessage;
var pong = new PongMessage(msg.Nonce, msg.MessageHeader.Network);
return(pong);
}
if (message.GetCommandName() == Constants.MessageNames.Verack) // RETURNS VERACK MESSAGE.
{
var msg = message as VerackMessage;
return(new VerackMessage(msg.MessageHeader.Network));
}
if (message.GetCommandName() == Constants.MessageNames.Addr) // RETURNS THE ADDRS.
{
var msg = message as AddrMessage;
if (msg.IpAddresses != null)
{
foreach (var ipAddress in msg.IpAddresses)
{
PeerEventStore.Instance().NewPeer(ipAddress);
}
}
return(null);
}
if (message.GetCommandName() == Constants.MessageNames.GetAddr) // RETURNS THE ADDRS.
{
var msg = message as GetAddressMessage;
var ipAdrLst = _peersStorage.GetAll();
var response = new AddrMessage(new CompactSize {
Size = (ulong)ipAdrLst.Count()
}, msg.MessageHeader.Network);
foreach (var ipAdr in ipAdrLst)
{
response.IpAddresses.Add(ipAdr);
}
return(response);
}
if (message.GetCommandName() == Constants.MessageNames.MemPool) // RETURNS THE INVENTORY.
{
var msg = message as MemPoolMessage;
var memoryPool = MemoryPool.Instance();
var txIds = memoryPool.GetTransactions().Select(t => t.Transaction.GetTxId());
var inventories = new List <Inventory>();
foreach (var txId in txIds)
{
inventories.Add(new Inventory(InventoryTypes.MSG_TX, txId));
}
return(new InventoryMessage(inventories, msg.MessageHeader.Network));
}
if (message.GetCommandName() == Constants.MessageNames.GetData) // RETURNS ALL THE DATA TO THE PEER.
{
var msg = message as GetDataMessage;
var messages = Execute(msg);
if (messages != null)
{
foreach (var m in messages)
{
peer.Execute(m.Serialize());
}
}
return(null);
}
if (message.GetCommandName() == Constants.MessageNames.Transaction) // ADD TRANSACTION INTO MEMORY POOL & BROADCAST IT.
{
var msg = message as TransactionMessage;
AddTransaction(msg.Transaction);
MemoryPool.Instance().Remove(msg.Transaction);
p2pNetworkConnector.Broadcast(msg.Transaction, msg.MessageHeader.Ipv6);
return(null);
}
if (message.GetCommandName() == Constants.MessageNames.Block) // ADD THE BLOCK.
{
var msg = message as BlockMessage;
_blockValidator.Check(msg.Block);
if (msg.Block.Transactions != null)
{
MemoryPool.Instance().Remove(msg.Block.Transactions.Select(tx => tx.GetTxId()));
}
smartContract.AddBlock(msg.Block);
smartContract.Commit();
blockChain.AddBlock(msg.Block);
}
if (message.GetCommandName() == Constants.MessageNames.NotFound) // SOME INVENTORIES ARE NOT FOUND.
{
return(null);
}
if (message.GetCommandName() == Constants.MessageNames.GetBlocks) // RETURN THE BLOCKS : https://bitcoin.org/en/developer-reference#getblocks
{
var msg = message as GetBlocksMessage;
int lastBlockHeight = -1;
foreach (var blockHash in msg.BlockHashes)
{
lastBlockHeight = blockChain.GetBlockHeight(blockHash);
if (lastBlockHeight != -1)
{
goto Found;
}
}
Found:
var currentBlockHeight = blockChain.GetCurrentBlockHeight();
if (currentBlockHeight == lastBlockHeight)
{
return(new InventoryMessage(new List <Inventory>(), msg.MessageHeader.Network));
}
var nbBlocks = currentBlockHeight - lastBlockHeight;
if (lastBlockHeight == -1)
{
nbBlocks = currentBlockHeight - 1;
}
if (nbBlocks > Constants.DEFAULT_MAX_GET_INVENTORIES)
{
nbBlocks = Constants.DEFAULT_MAX_GET_INVENTORIES;
}
var blocks = blockChain.GetLastBlocks(nbBlocks);
var inventories = new List <Inventory>();
foreach (var block in blocks)
{
inventories.Add(new Inventory(InventoryTypes.MSG_BLOCK, block.GetHashHeader()));
}
return(new InventoryMessage(inventories, msg.MessageHeader.Network));
}
return(null);
}
private Message Launch(PeerConnector peerConnector, GetAddressMessage message)
{
var payload = peerConnector.Execute(message.Serialize()); // S : GETADDR.
return(_messageParser.Parse(payload.ToArray()));
}
private void Launch(PeerConnector peerConnector, AddrMessage message)
{
peerConnector.Execute(message.Serialize());
}