private bool Add(Node node, bool raiseNodeAdded) { if (node == null) { throw new ArgumentNullException("node"); } Bucket bucket = buckets.Find(delegate(Bucket b) { return(b.CanContain(node)); }); if (bucket != null && bucket.Nodes.Contains(node)) { return(false); } bool added = bucket.Add(node); if (!added && bucket.CanContain(LocalNode)) { if (Split(bucket)) { return(Add(node, raiseNodeAdded)); } } return(added); }
public async Task NodeReplaceTest() { int nodeCount = 0; Bucket b = new Bucket(); for (int i = 0; i < Bucket.MaxCapacity; i++) { Node n = new Node(NodeId.Create(), new IPEndPoint(IPAddress.Any, i)); n.Seen(); b.Add(n); } b.Nodes[3].Seen(TimeSpan.FromDays(5)); b.Nodes[1].Seen(TimeSpan.FromDays(4)); b.Nodes[5].Seen(TimeSpan.FromDays(3)); listener.MessageSent += (data, endpoint) => { engine.MessageLoop.DhtMessageFactory.TryDecodeMessage(BEncodedValue.Decode <BEncodedDictionary> (data), out DhtMessage message); b.Nodes.Sort((l, r) => l.LastSeen.CompareTo(r.LastSeen)); if ((endpoint.Port == 3 && nodeCount == 0) || (endpoint.Port == 1 && nodeCount == 1) || (endpoint.Port == 5 && nodeCount == 2)) { Node n = b.Nodes.Find(no => no.EndPoint.Port == endpoint.Port); n.Seen(); PingResponse response = new PingResponse(n.Id, message.TransactionId); listener.RaiseMessageReceived(response, node.EndPoint); nodeCount++; } }; ReplaceNodeTask task = new ReplaceNodeTask(engine, b, null); await task.Execute().WithTimeout(4000); }
public void NodeReplaceTest() { int nodeCount = 0; Bucket b = new Bucket(); for (int i = 0; i < Bucket.MaxCapacity; i++) { Node n = new Node(NodeId.Create(), new IPEndPoint(IPAddress.Any, i)); n.Seen(); b.Add(n); } b.Nodes[3].Seen(TimeSpan.FromDays(5)); b.Nodes[1].Seen(TimeSpan.FromDays(4)); b.Nodes[5].Seen(TimeSpan.FromDays(3)); listener.MessageSent += (message, endpoint) => { b.Nodes.Sort((l, r) => l.LastSeen.CompareTo(r.LastSeen)); if ((endpoint.Port == 3 && nodeCount == 0) || (endpoint.Port == 1 && nodeCount == 1) || (endpoint.Port == 5 && nodeCount == 2)) { Node n = b.Nodes.Find(delegate(Node no) { return(no.EndPoint.Port == endpoint.Port); }); n.Seen(); PingResponse response = new PingResponse(n.Id, message.TransactionId); listener.RaiseMessageReceived(response, node.EndPoint); nodeCount++; } }; ReplaceNodeTask task = new ReplaceNodeTask(engine, b, null); Assert.IsTrue(task.Execute().Wait(4000), "#10"); }
bool Add(Node node, bool raiseNodeAdded) { if (node == null) { throw new ArgumentNullException(nameof(node)); } Bucket bucket = Buckets.Find(b => b.CanContain(node)) !; if (bucket.Nodes.Contains(node)) { return(false); } bool added = bucket.Add(node); if (!added && bucket.CanContain(LocalNode)) { if (Split(bucket)) { return(Add(node, raiseNodeAdded)); } } return(added); }
public void NodeReplaceTest() { engine.TimeOut = TimeSpan.FromMilliseconds(25); ManualResetEvent handle = new ManualResetEvent(false); Bucket b = new Bucket(); for (int i = 0; i < Bucket.MaxCapacity; i++) { Node n = new Node(NodeId.Create(), new IPEndPoint(IPAddress.Any, i)); n.LastSeen = DateTime.UtcNow; b.Add(n); } b.Nodes[3].LastSeen = DateTime.UtcNow.AddDays(-5); b.Nodes[1].LastSeen = DateTime.UtcNow.AddDays(-4); b.Nodes[5].LastSeen = DateTime.UtcNow.AddDays(-3); engine.MessageLoop.QuerySent += delegate(object o, SendQueryEventArgs e) { if (!e.TimedOut) { return; } b.Nodes.Sort(); if ((e.EndPoint.Port == 3 && nodeCount == 0) || (e.EndPoint.Port == 1 && nodeCount == 1) || (e.EndPoint.Port == 5 && nodeCount == 2)) { Node n = b.Nodes.Find(delegate(Node no) { return(no.EndPoint.Port == e.EndPoint.Port); }); n.Seen(); PingResponse response = new PingResponse(n.Id, e.Query.TransactionId); DhtEngine.MainLoop.Queue(delegate { //System.Threading.Thread.Sleep(100); Console.WriteLine("Faking the receive"); listener.RaiseMessageReceived(response, node.EndPoint); }); nodeCount++; } }; ReplaceNodeTask task = new ReplaceNodeTask(engine, b, null); // FIXME: Need to assert that node 0.0.0.0:0 is the one which failed - i.e. it should be replaced task.Completed += delegate(object o, TaskCompleteEventArgs e) { handle.Set(); }; task.Execute(); Assert.IsTrue(handle.WaitOne(4000, false), "#10"); }