public void MakeAndReadFragment()
{
var arec = new DatabaseLookupMessage(
new I2PIdentHash(true),
new I2PIdentHash(true),
DatabaseLookupMessage.LookupTypes.Normal);
var msg = new TunnelMessageRouter(
arec,
new I2PIdentHash(true));
var refmsgdata = msg.Message.CreateHeader16.HeaderAndPayload;
var fragments = TunnelDataMessage.MakeFragments(
new TunnelMessage[] { msg },
BufUtils.RandomUint());
var mkmsg = new TunnelDataFragmentReassembly();
var recvtmsgs = mkmsg.Process(fragments.Shuffle().ToArray(), out var _);
foreach (var rmsg in recvtmsgs)
{
var rmsgdata = rmsg.Message.CreateHeader16.HeaderAndPayload;
Assert.IsTrue(msg.Delivery == rmsg.Delivery);
Assert.IsTrue(refmsgdata == rmsgdata);
}
}
public void MakeAndReadFragments5_2()
{
var origmsgs = new List <TunnelMessage>();
for (int i = 0; i < 5; ++i)
{
var adatarec = new DataMessage(new BufLen(BufUtils.RandomBytes(2048)));
var amsg = new TunnelMessageRouter(
adatarec,
new I2PIdentHash(true));
origmsgs.Add(amsg);
}
var msgs = TunnelDataMessage.MakeFragments(origmsgs, BufUtils.RandomUint());
var mkmsg = new TunnelDataFragmentReassembly();
var recvtmsgs = msgs
.Chunk(a => 2)
.Shuffle()
.Select(m => mkmsg.Process(m, out var _))
.SelectMany(b => b);
Assert.IsTrue(origmsgs.All(o => recvtmsgs.Any(m =>
m.Delivery == o.Delivery &&
m.Message.CreateHeader16.HeaderAndPayload == o.Message.CreateHeader16.HeaderAndPayload
)));
}
public void TestSimpleTunnelDataCreation()
{
var smalldata = BufUtils.RandomBytes(38);
var srcmsgs = new List <TunnelMessage>();
srcmsgs.Add(new TunnelMessageLocal(new DeliveryStatusMessage(1234)));
var msgfrags = TunnelDataMessage.MakeFragments(srcmsgs, 0x3e5c);
Assert.IsTrue(msgfrags.Count() == 1);
var firstmsg = msgfrags.First();
var serialized = firstmsg.CreateHeader16.HeaderAndPayload;
var recovered = I2NPMessage.ReadHeader16(new BufRefLen(serialized));
var reassembler = new TunnelDataFragmentReassembly();
var reassembledmsgs = reassembler.Process(
new TunnelDataMessage[]
{
(TunnelDataMessage)recovered.Message
}, out var _);
Assert.IsTrue(reassembledmsgs.Count() == 1);
var firstrecinstr = reassembledmsgs.First();
Assert.IsTrue(firstrecinstr.Delivery == TunnelMessage.DeliveryTypes.Local);
Assert.IsTrue(firstrecinstr.Message.MessageType == I2NPMessage.MessageTypes.DeliveryStatus);
Assert.IsTrue(((DeliveryStatusMessage)firstrecinstr.Message).StatusMessageId == 1234);
}
public void MakeAndReadFragmentsWithSerialize()
{
var origmsgs = new List <TunnelMessage>();
for (int i = 0; i < 200; ++i)
{
switch (BufUtils.RandomInt(3))
{
case 0:
var adatarec = new DataMessage(
new BufLen(
BufUtils.RandomBytes(2048 + BufUtils.RandomInt(1024))));
origmsgs.Add(new TunnelMessageLocal(adatarec));
break;
case 1:
var arec = new DatabaseLookupMessage(
new I2PIdentHash(true),
new I2PIdentHash(true),
DatabaseLookupMessage.LookupTypes.RouterInfo);
origmsgs.Add(new TunnelMessageRouter(
arec,
new I2PIdentHash(true)));
break;
case 2:
var adatarec2 = new DataMessage(
new BufLen(
BufUtils.RandomBytes(2048 + BufUtils.RandomInt(1024))));
origmsgs.Add(new TunnelMessageTunnel(adatarec2,
new I2PIdentHash(true),
BufUtils.RandomUint()));
break;
}
}
var msgs = TunnelDataMessage.MakeFragments(origmsgs, BufUtils.RandomUint());
var recvlist = new List <TunnelDataMessage>();
foreach (var msg in msgs)
{
recvlist.Add((TunnelDataMessage)I2NPMessage.ReadHeader16(
new BufRefLen(msg.CreateHeader16.HeaderAndPayload)).Message);
}
var mkmsg = new TunnelDataFragmentReassembly();
var recvtmsgs = mkmsg.Process(recvlist, out var _);
foreach (var rmsg in recvtmsgs)
{
Assert.IsTrue(origmsgs.SingleOrDefault(m =>
m.Delivery == rmsg.Delivery &&
m.Message.CreateHeader16.HeaderAndPayload == rmsg.Message.CreateHeader16.HeaderAndPayload
) != null);
}
}
public void MakeAndReadFragments()
{
var origmsgs = new List <TunnelMessage>();
for (int i = 0; i < 200; ++i)
{
switch (BufUtils.RandomInt(3))
{
case 0:
var adatarec = new DataMessage(new BufLen(BufUtils.Random(2048 + BufUtils.RandomInt(1024))));
origmsgs.Add(new TunnelMessageTunnel(
adatarec.Header16,
new I2PIdentHash(true),
BufUtils.RandomUint()));
break;
case 1:
var arec = new DatabaseLookupMessage(
new I2PIdentHash(true),
new I2PIdentHash(true),
DatabaseLookupMessage.LookupTypes.Normal);
origmsgs.Add(new TunnelMessageRouter(arec.Header16, new I2PIdentHash(true)));
break;
case 2:
var adatarec2 = new DataMessage(new BufLen(BufUtils.Random(2048 + BufUtils.RandomInt(1024))));
origmsgs.Add(new TunnelMessageLocal(adatarec2.Header16));
break;
}
}
var msgs = TunnelDataMessage.MakeFragments(origmsgs, BufUtils.RandomUint());
var mkmsg = new TunnelDataFragmentReassembly();
var recvtmsgs = mkmsg.Process(msgs);
foreach (var rmsg in recvtmsgs)
{
Assert.IsTrue(origmsgs.SingleOrDefault(m =>
m.Delivery == rmsg.Delivery &&
m.Header.HeaderAndPayload == rmsg.Header.HeaderAndPayload
) != null);
}
}
public void TestSingleLargeTunnelDataCreation()
{
var sourcedata = new BufLen(BufUtils.RandomBytes(9000));
var srcmsgs = new List <TunnelMessage>();
srcmsgs.Add(new TunnelMessageTunnel(new DataMessage(sourcedata), new I2PIdentHash(true), 4242));
var msgfrags = TunnelDataMessage.MakeFragments(srcmsgs, 0x3e5c);
Assert.IsTrue(msgfrags.Count() == 10);
var serbuf = new List <byte>();
foreach (var frag in msgfrags)
{
serbuf.AddRange(frag.CreateHeader16.HeaderAndPayload);
}
var serbufarray = serbuf.ToArray();
var reassembler = new TunnelDataFragmentReassembly();
var reader = new BufRefLen(serbufarray);
var readmsgs = new List <TunnelDataMessage>();
while (reader.Length > 0)
{
readmsgs.Add((TunnelDataMessage)(I2NPMessage.ReadHeader16(reader)).Message);
}
var reassembledmsgs = reassembler.Process(readmsgs, out var _);
Assert.IsTrue(reassembledmsgs.Count() == 1);
Assert.IsTrue(reassembler.BufferedFragmentCount == 0);
var firstrecinstr = reassembledmsgs.First();
Assert.IsTrue(firstrecinstr.Delivery == TunnelMessage.DeliveryTypes.Tunnel);
Assert.IsTrue(((TunnelMessageTunnel)firstrecinstr).Tunnel == 4242);
Assert.IsTrue(firstrecinstr.Message.MessageType == I2NPMessage.MessageTypes.Data);
var datamsg = (DataMessage)firstrecinstr.Message;
Assert.IsTrue(datamsg.DataMessagePayloadLength == sourcedata.Length);
Assert.IsTrue(datamsg.DataMessagePayload == sourcedata);
}
public void MakeAndReadFragments100()
{
var origmsgs = new List <TunnelMessage>();
for (int i = 0; i < 100; ++i)
{
switch (BufUtils.RandomInt(3))
{
case 0:
var adatarec = new DataMessage(new BufLen(BufUtils.RandomBytes(2048)));
origmsgs.Add(new TunnelMessageTunnel(
adatarec,
new I2PIdentHash(true),
BufUtils.RandomUint()));
break;
case 1:
var arec = new DatabaseLookupMessage(
new I2PIdentHash(true),
new I2PIdentHash(true),
DatabaseLookupMessage.LookupTypes.Normal);
origmsgs.Add(new TunnelMessageRouter(
arec,
new I2PIdentHash(true)));
break;
case 2:
var adatarec2 = new DataMessage(
new BufLen(
BufUtils.RandomBytes(2048 + BufUtils.RandomInt(1024))));
origmsgs.Add(new TunnelMessageLocal(adatarec2));
break;
}
}
var msgs = TunnelDataMessage.MakeFragments(origmsgs, BufUtils.RandomUint());
var mkmsg = new TunnelDataFragmentReassembly();
var chunks = msgs
.Shuffle()
.ToArray()
.Chunk(a => 1 + BufUtils.RandomInt(10))
.Shuffle();
var recvtmsgs = chunks.SelectMany(c => mkmsg.Process(c, out var _)).ToArray();
Assert.IsTrue(origmsgs.All(o => recvtmsgs.Any(m =>
m.Delivery == o.Delivery &&
m.Message.CreateHeader16.HeaderAndPayload == o.Message.CreateHeader16.HeaderAndPayload
)));
var mkmsg2 = new TunnelDataFragmentReassembly();
var chunks2 = msgs
.Shuffle()
.ToArray()
.Skip(1)
.Chunk(a => 1 + BufUtils.RandomInt(10))
.Shuffle();
var recvtmsgs2 = chunks2.SelectMany(c => mkmsg2.Process(c, out var _)).ToArray();
Assert.IsFalse(origmsgs.All(o => recvtmsgs2.Any(m =>
m.Delivery == o.Delivery &&
m.Message.CreateHeader16.HeaderAndPayload == o.Message.CreateHeader16.HeaderAndPayload
)));
}
