public void BidirectionalCommunicationTest()
{
using (KBREnvironment env = new KBREnvironment (true, true)) {
int nodes = 20;
ECKeyPair[] nodePrivateKeys = new ECKeyPair[nodes];
Key[] nodeKeys = new Key[nodes];
for (int i = 0; i < nodePrivateKeys.Length; i ++) {
nodePrivateKeys[i] = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
nodeKeys[i] = Key.Create (nodePrivateKeys[i]);
}
env.AddNodes (nodeKeys, nodePrivateKeys);
ECKeyPair priv1 = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
ECKeyPair priv2 = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
Key id1 = Key.Create (priv1);
Key id2 = Key.Create (priv2);
ISubscribeInfo subscribeInfo1 = env.AnonymousRouters[0].SubscribeRecipient (id1, priv1);
ISubscribeInfo subscribeInfo2 = env.AnonymousRouters[1].SubscribeRecipient (id2, priv2);
while (true) {
if (subscribeInfo1.Status != SubscribeRouteStatus.Establishing && subscribeInfo2.Status != SubscribeRouteStatus.Establishing)
break;
Thread.Sleep (100);
}
byte[] received1 = null;
object received1_lock = new object ();
AutoResetEvent received1_done = new AutoResetEvent (false);
byte[] received2 = null;
object received2_lock = new object ();
AutoResetEvent received2_done = new AutoResetEvent (false);
AutoResetEvent accepted_done = new AutoResetEvent (false);
IAnonymousSocket sock2 = null;
subscribeInfo2.Accepting += delegate (object sender, AcceptingEventArgs args) {
string msg = args.Payload as string;
Assert.IsNotNull (msg);
Assert.AreEqual ("HELLO", msg);
args.Accept ("HELLO2", null);
};
subscribeInfo2.Accepted += delegate (object sender, AcceptedEventArgs args) {
lock (received2_lock) {
Assert.IsNull (sock2, "2.accepted.#1");
sock2 = args.Socket;
Assert.IsNotNull (sock2, "2.accepted.#2");
accepted_done.Set ();
}
};
IAsyncResult ar = env.AnonymousRouters[0].BeginConnect (id1, id2, AnonymousConnectionType.LowLatency, "HELLO", null, null);
IAnonymousSocket sock1 = env.AnonymousRouters[0].EndConnect (ar);
Assert.IsNotNull (sock1, "1.sock");
Assert.AreEqual ("HELLO2", sock1.PayloadAtEstablishing as string);
Assert.IsTrue (accepted_done.WaitOne (1000), "2.waiting");
Assert.IsNotNull (sock2, "2.sock");
Assert.AreEqual ("HELLO", sock2.PayloadAtEstablishing as string);
sock1.Received += delegate (object sender, DatagramReceiveEventArgs args) {
lock (received1_lock) {
Assert.IsNull (received1, "1.received.#1");
received1 = new byte[args.Size];
Buffer.BlockCopy (args.Buffer, 0, received1, 0, received1.Length);
received1_done.Set ();
}
};
sock2.Received += delegate (object sender2, DatagramReceiveEventArgs args2) {
lock (received2_lock) {
Assert.IsNull (received2, "2.received.#1");
received2 = new byte[args2.Size];
Buffer.BlockCopy (args2.Buffer, 0, received2, 0, received2.Length);
received2_done.Set ();
}
};
sock1.InitializedEventHandlers ();
sock2.InitializedEventHandlers ();
for (int i = 0; i < 3; i ++) {
string msg = "HELLO " + i.ToString ();
sock1.SendTo (Encoding.UTF8.GetBytes (msg), AnonymousRouter.DummyEndPoint);
Assert.IsTrue (received2_done.WaitOne (5000), "2.received.#2");
Assert.IsNotNull (received2, "2.received.#3");
Assert.AreEqual (msg, Encoding.UTF8.GetString (received2));
lock (received2_lock) {
received2 = null;
}
msg = "OK " + i.ToString ();
sock2.SendTo (Encoding.UTF8.GetBytes (msg), AnonymousRouter.DummyEndPoint);
Assert.IsTrue (received1_done.WaitOne (5000), "1.received.#2");
Assert.IsNotNull (received1, "1.received.#3");
Assert.AreEqual (msg, Encoding.UTF8.GetString (received1));
lock (received1_lock) {
received1 = null;
}
}
}
}
public void EstablishFailTest()
{
using (KBREnvironment env = new KBREnvironment (true, true)) {
int nodes = 20;
ECKeyPair[] nodePrivateKeys = new ECKeyPair[nodes];
Key[] nodeKeys = new Key[nodes];
for (int i = 0; i < nodePrivateKeys.Length; i ++) {
nodePrivateKeys[i] = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
nodeKeys[i] = Key.Create (nodePrivateKeys[i]);
}
env.AddNodes (nodeKeys, nodePrivateKeys);
ECKeyPair priv1 = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
ECKeyPair priv2 = ECKeyPair.Create (DefaultAlgorithm.ECDomainName);
Key id1 = Key.Create (priv1);
Key id2 = Key.Create (priv2);
ISubscribeInfo subscribeInfo1 = env.AnonymousRouters[0].SubscribeRecipient (id1, priv1);
ISubscribeInfo subscribeInfo2 = env.AnonymousRouters[1].SubscribeRecipient (id2, priv2);
while (true) {
if (subscribeInfo1.Status != SubscribeRouteStatus.Establishing && subscribeInfo2.Status != SubscribeRouteStatus.Establishing)
break;
Thread.Sleep (100);
}
AutoResetEvent accepted_done = new AutoResetEvent (false);
IAnonymousSocket sock2 = null;
subscribeInfo2.Accepting += delegate (object sender, AcceptingEventArgs args) {
args.Reject ();
};
subscribeInfo2.Accepted += delegate (object sender, AcceptedEventArgs args) {
accepted_done.Set ();
sock2 = args.Socket;
};
try {
IAsyncResult ar = env.AnonymousRouters[0].BeginConnect (id1, id2, AnonymousConnectionType.LowLatency, null, null, null);
env.AnonymousRouters[0].EndConnect (ar);
Assert.Fail ();
} catch (System.Net.Sockets.SocketException) {}
}
}