From 4d1ba03823da8371482a2a753575737f6ecd16db Mon Sep 17 00:00:00 2001 From: zzz Date: Sun, 8 Jun 2025 10:29:58 -0400 Subject: [PATCH] SSU2: Change packet classification if/else to switch --- .../router/transport/udp/PacketHandler.java | 24 ++++++++++++++----- 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/router/java/src/net/i2p/router/transport/udp/PacketHandler.java b/router/java/src/net/i2p/router/transport/udp/PacketHandler.java index 47d15ea64..11ea8d815 100644 --- a/router/java/src/net/i2p/router/transport/udp/PacketHandler.java +++ b/router/java/src/net/i2p/router/transport/udp/PacketHandler.java @@ -388,29 +388,41 @@ class PacketHandler { // all good SSU2Header.acceptTrialDecrypt(packet, header); - if (type == SSU2Util.SESSION_REQUEST_FLAG_BYTE) { + switch (type) { + case SSU2Util.SESSION_REQUEST_FLAG_BYTE: if (_log.shouldDebug()) _log.debug("Got a Session Request on " + state); _establisher.receiveSessionOrTokenRequest(from, state, packet); - } else if (type == SSU2Util.TOKEN_REQUEST_FLAG_BYTE) { + break; + + case SSU2Util.TOKEN_REQUEST_FLAG_BYTE: if (_log.shouldDebug()) _log.debug("Got a Token Request on " + state); _establisher.receiveSessionOrTokenRequest(from, state, packet); - } else if (type == SSU2Util.SESSION_CONFIRMED_FLAG_BYTE) { + break; + + case SSU2Util.SESSION_CONFIRMED_FLAG_BYTE: if (_log.shouldDebug()) _log.debug("Got a Session Confirmed on " + state); _establisher.receiveSessionConfirmed(state, packet); - } else if (type == SSU2Util.PEER_TEST_FLAG_BYTE) { + break; + + case SSU2Util.PEER_TEST_FLAG_BYTE: if (_log.shouldDebug()) _log.debug("Got a Peer Test"); _testManager.receiveTest(from, packet); - } else if (type == SSU2Util.HOLE_PUNCH_FLAG_BYTE) { + break; + + case SSU2Util.HOLE_PUNCH_FLAG_BYTE: if (_log.shouldDebug()) _log.debug("Got a Hole Punch"); _establisher.receiveHolePunch(from, packet); - } else { + break; + + default: if (_log.shouldWarn()) _log.warn("Got unknown SSU2 message " + header + " from " + from); + break; } return true; }