public NodesGroup(
Network network,
NodeConnectionParameters connectionParameters = null,
NodeRequirement requirements = null)
{
AllowSameGroup = false;
MaximumNodeConnection = 8;
_Network = network;
cs = new object();
_ConnectedNodes = new NodesCollection();
_ConnectionParameters = connectionParameters ?? new NodeConnectionParameters();
_ConnectionParameters = _ConnectionParameters.Clone();
_Requirements = requirements ?? new NodeRequirement();
_Disconnect = new CancellationTokenSource();
}
public NodeConnectionParameters(NodeConnectionParameters other)
{
Version = other.Version;
IsRelay = other.IsRelay;
Services = other.Services;
ConnectCancellation = other.ConnectCancellation;
UserAgent = other.UserAgent;
AddressFrom = other.AddressFrom;
Nonce = other.Nonce;
Advertize = other.Advertize;
PreferredTransactionOptions = other.PreferredTransactionOptions;
EndpointConnector = other.EndpointConnector.Clone();
SocketSettings = other.SocketSettings.Clone();
foreach (var behavior in other.TemplateBehaviors)
{
TemplateBehaviors.Add(behavior.Clone());
}
}
public NodeServer(Network network, uint?version = null,
int internalPort = -1)
{
AllowLocalPeers = true;
InboundNodeConnectionParameters = new NodeConnectionParameters();
internalPort = internalPort == -1 ? network.DefaultPort : internalPort;
_LocalEndpoint = new IPEndPoint(IPAddress.Parse("0.0.0.0").MapToIPv6Ex(), internalPort);
MaxConnections = 125;
_Network = network;
_ExternalEndpoint = new IPEndPoint(_LocalEndpoint.Address, Network.DefaultPort);
_Version = version == null ? network.MaxP2PVersion : version.Value;
var listener = new EventLoopMessageListener <IncomingMessage>(ProcessMessage);
_MessageProducer.AddMessageListener(listener);
OwnResource(listener);
_ConnectedNodes = new NodesCollection();
_ConnectedNodes.Added += _Nodes_NodeAdded;
_ConnectedNodes.Removed += _Nodes_NodeRemoved;
_ConnectedNodes.MessageProducer.AddMessageListener(listener);
}
internal void DiscoverPeers(Network network, NodeConnectionParameters parameters, int peerToFind)
{
Logs.NodeServer.LogTrace("Discovering nodes");
int found = 0;
{
while (found < peerToFind)
{
parameters.ConnectCancellation.ThrowIfCancellationRequested();
Logs.NodeServer.LogTrace("Remaining peer to get {remainingPeerCount}", (-found + peerToFind));
List <NetworkAddress> peers = new List <NetworkAddress>();
peers.AddRange(this.GetAddr());
if (peers.Count == 0)
{
PopulateTableWithDNSNodes(network, peers).GetAwaiter().GetResult();
PopulateTableWithHardNodes(network, peers);
peers = new List <NetworkAddress>(peers.OrderBy(a => RandomUtils.GetInt32()));
if (peers.Count == 0)
{
return;
}
}
CancellationTokenSource peerTableFull = new CancellationTokenSource();
CancellationToken loopCancel = CancellationTokenSource.CreateLinkedTokenSource(peerTableFull.Token, parameters.ConnectCancellation).Token;
try
{
Parallel.ForEach(peers, new ParallelOptions()
{
MaxDegreeOfParallelism = 5,
CancellationToken = loopCancel,
}, p =>
{
using (CancellationTokenSource timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)))
using (var cancelConnection = CancellationTokenSource.CreateLinkedTokenSource(timeout.Token, loopCancel))
{
Node n = null;
try
{
var param2 = parameters.Clone();
param2.ConnectCancellation = cancelConnection.Token;
var addrman = param2.TemplateBehaviors.Find <AddressManagerBehavior>();
param2.TemplateBehaviors.Clear();
param2.TemplateBehaviors.Add(addrman);
n = Node.Connect(network, p.Endpoint, param2);
n.VersionHandshake(cancelConnection.Token);
n.MessageReceived += (s, a) =>
{
var addr = (a.Message.Payload as AddrPayload);
if (addr != null)
{
Interlocked.Add(ref found, addr.Addresses.Length);
if (found >= peerToFind)
{
peerTableFull.Cancel();
}
}
};
n.SendMessageAsync(new GetAddrPayload());
loopCancel.WaitHandle.WaitOne(2000);
}
catch
{
}
finally
{
if (n != null)
{
n.DisconnectAsync();
}
}
}
if (found >= peerToFind)
{
peerTableFull.Cancel();
}
else
{
Logs.NodeServer.LogInformation("Need {neededPeerCount} more peers", (-found + peerToFind));
}
});
}
catch (OperationCanceledException)
{
if (parameters.ConnectCancellation.IsCancellationRequested)
{
throw;
}
}
}
}
}
