marek/i2p.i2p
1
0
Fork 0
forked from I2P_Developers/i2p.i2p
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Tests: Add serialize/deserialize test for STBM/OTBRM

Browse Source
This commit is contained in:
zzz
2021-06-13 10:52:42 -04:00
parent 010d1a9953
commit 4c6aa8cfc1
1 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
router/java/test/junit/net/i2p/router/tunnel/pool/BuildMessageTestStandalone.java
View File

@ -21,6 +21,9 @@ import net.i2p.data.TunnelId;
import net.i2p.data.i2np.BuildRequestRecord;
import net.i2p.data.i2np.BuildResponseRecord;
import net.i2p.data.i2np.EncryptedBuildRecord;
import net.i2p.data.i2np.I2NPMessage;
import net.i2p.data.i2np.I2NPMessageException;
import net.i2p.data.i2np.I2NPMessageHandler;
import net.i2p.data.i2np.OutboundTunnelBuildReplyMessage;
import net.i2p.data.i2np.ShortTunnelBuildMessage;
import net.i2p.data.i2np.TunnelBuildMessage;
@ -97,6 +100,17 @@ public class BuildMessageTestStandalone extends TestCase {
"\n" + cfg +
"\n================================================================");
if (testType == 3) {
// test read/write
byte[] data = msg.toByteArray();
try {
I2NPMessage msg2 = (new I2NPMessageHandler(ctx)).readMessage(data);
msg = (ShortTunnelBuildMessage) msg2;
} catch (Exception e) {
log.error("STBM out/in fail", e);
assertTrue(e.toString(), false);
}
}
// now msg is fully encrypted, so lets go through the hops, decrypting and replying
// as necessary
@ -193,7 +207,16 @@ public class BuildMessageTestStandalone extends TestCase {
else
otbrm.setRecord(i, msg.getRecord(i));
}
reply = otbrm;
// test read/write
byte[] data = otbrm.toByteArray();
try {
I2NPMessage msg2 = (new I2NPMessageHandler(ctx)).readMessage(data);
reply = (OutboundTunnelBuildReplyMessage) msg2;
} catch (Exception e) {
reply = null;
log.error("OTBRM out/in fail", e);
assertTrue(e.toString(), false);
}
} else {
reply = new TunnelBuildReplyMessage(ctx);
for (int i = 0; i < TunnelBuildMessage.MAX_RECORD_COUNT; i++) {