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

NTCP2: Refactor padding size calculation

Browse Source 
Avoid possible NPEs (ticket #2286)
Bundle up to 5 tunnel messages
Use read buffer to send RI and termination
Temp buf doesn't need 2 bytes for length
RI size check
Log tweaks
This commit is contained in:
zzz
2018-07-08 11:00:54 +00:00
parent fcf82ea580
commit 950ca71a34
4 changed files with 103 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
history.txt
View File

@ -1,3 +1,16 @@
2018-07-08 zzz
* NTCP2: Avoid possible NPEs (ticket #2286)
2018-07-06 zzz
* NTCP: Read all available data when able (ticket #2243)
* SSU: Change remaining acks from List to Set (ticket #2258)
2018-07-05 zzz
* i2psnark:
- Fix IOOBE when stopping torrent that is allocating (ticket #2273)
- Fix comments wrapping (ticket #2284)
* NTCP2: Increase max message size
2018-07-04 zzz
* NTCP: Don't advertise interface address when configured for force-firewalled

2
router/java/src/net/i2p/router/RouterVersion.java
View File

@ -18,7 +18,7 @@ public class RouterVersion {
/** deprecated */
public final static String ID = "Monotone";
public final static String VERSION = CoreVersion.VERSION;
public final static long BUILD = 5;
public final static long BUILD = 6;
/** for example "-test" */
public final static String EXTRA = "";

6
router/java/src/net/i2p/router/transport/ntcp/NTCP2Options.java
View File

@ -63,8 +63,8 @@ class NTCP2Options {
@Override
public String toString() {
return "Padding options: send min/max %: (" + (_sendMin * 100) + ", " + (_sendMax * 100) +
") recv min/max %: ( " + (_recvMin * 100) + ", " + (_recvMax * 100) +
") dummy send/recv B/s: ( " + _sendDummy + ", " + _recvDummy +
") delay send/recv ms: ( " + _sendDelay + ", " + _recvDelay + ')';
") recv min/max %: (" + (_recvMin * 100) + ", " + (_recvMax * 100) +
") dummy send/recv B/s: (" + _sendDummy + ", " + _recvDummy +
") delay send/recv ms: (" + _sendDelay + ", " + _recvDelay + ')';
}
}

86
router/java/src/net/i2p/router/transport/ntcp/NTCPConnection.java
View File

@ -186,7 +186,8 @@ public class NTCPConnection implements Closeable {
private static final long NTCP2_FAIL_TIMEOUT = 10*1000;
private static final long NTCP2_TERMINATION_CLOSE_DELAY = 50;
// don't make combined messages too big, to minimize latency
private static final int NTCP2_PREFERRED_PAYLOAD_MAX = 5000;
// Tunnel data msgs are 1024 + 4 + 9 + 3 = 1040, allow 5
private static final int NTCP2_PREFERRED_PAYLOAD_MAX = 5 * 1040;
static final int REASON_UNSPEC = 0;
static final int REASON_TERMINATION = 1;
static final int REASON_TIMEOUT = 2;
@ -897,9 +898,29 @@ public class NTCPConnection implements Closeable {
}
int availForPad = BUFFER_SIZE - (size + NTCP2Payload.BLOCK_HEADER_SIZE);
if (availForPad > 0) {
int padlen = getPaddingSize(size, availForPad);
// all zeros is fine here
//Block block = new NTCP2Payload.PaddingBlock(_context, padlen);
Block block = new NTCP2Payload.PaddingBlock(padlen);
blocks.add(block);
size += block.getTotalLength();
}
sendNTCP2(buf.unencrypted, blocks);
}
/**
* NTCP2 only
*
* @param dataSize the total size of the data we are sending
* @param availForPad the available size for padding, not including padding block header,
* must be greater than zero
* @return min 0 max availForPad
* @since 0.9.36
*/
private int getPaddingSize(int dataSize, int availForPad) {
// what we want to send, calculated in proportion to data size
int minSend = (int) (size * _paddingConfig.getSendMin());
int maxSend = (int) (size * _paddingConfig.getSendMax());
int minSend = (int) (dataSize * _paddingConfig.getSendMin());
int maxSend = (int) (dataSize * _paddingConfig.getSendMax());
// the absolute min and max we can send
int min = Math.min(minSend, availForPad);
int max = Math.min(maxSend, availForPad);
@ -917,7 +938,7 @@ public class NTCPConnection implements Closeable {
padlen += _context.random().nextInt(1 + range);
if (_log.shouldWarn())
_log.warn("Padding params:" +
" data size: " + size +
" data size: " + dataSize +
" avail: " + availForPad +
" minSend: " + minSend +
" maxSend: " + maxSend +
@ -925,13 +946,7 @@ public class NTCPConnection implements Closeable {
" max: " + max +
" range: " + range +
" padlen: " + padlen);
// all zeros is fine here
//Block block = new NTCP2Payload.PaddingBlock(_context, padlen);
Block block = new NTCP2Payload.PaddingBlock(padlen);
blocks.add(block);
size += block.getTotalLength();
}
sendNTCP2(buf.unencrypted, blocks);
return padlen;
}
/**
@ -940,7 +955,10 @@ public class NTCPConnection implements Closeable {
* @since 0.9.36
*/
private void sendOurRouterInfo(boolean shouldFlood) {
sendRouterInfo(_context.router().getRouterInfo(), shouldFlood);
RouterInfo ri = _context.router().getRouterInfo();
if (ri == null)
return;
sendRouterInfo(ri, shouldFlood);
}
/**
@ -953,18 +971,27 @@ public class NTCPConnection implements Closeable {
if (_log.shouldWarn())
_log.warn("Sending router info for: " + ri.getHash() + " flood? " + shouldFlood);
List<Block> blocks = new ArrayList<Block>(2);
int plen = 2;
Block block = new NTCP2Payload.RIBlock(ri, shouldFlood);
plen += block.getTotalLength();
int size = block.getTotalLength();
if (size > BUFFER_SIZE) {
if (_log.shouldWarn())
_log.warn("RI too big: " + ri);
return;
}
blocks.add(block);
int padlen = 1 + _context.random().nextInt(PADDING_MAX);
int availForPad = BUFFER_SIZE - (size + NTCP2Payload.BLOCK_HEADER_SIZE);
if (availForPad > 0) {
int padlen = getPaddingSize(size, availForPad);
// all zeros is fine here
//block = new NTCP2Payload.PaddingBlock(_context, padlen);
block = new NTCP2Payload.PaddingBlock(padlen);
plen += block.getTotalLength();
size += block.getTotalLength();
blocks.add(block);
byte[] tmp = new byte[plen];
sendNTCP2(tmp, blocks);
}
// use a "read buf" for the temp array
ByteArray dataBuf = acquireReadBuf();
sendNTCP2(dataBuf.getData(), blocks);
releaseReadBuf(dataBuf);
}
/**
@ -978,18 +1005,21 @@ public class NTCPConnection implements Closeable {
if (_log.shouldWarn())
_log.warn("Sending termination, reason: " + reason + ", vaild frames rcvd: " + validFramesRcvd);
List<Block> blocks = new ArrayList<Block>(2);
int plen = 2;
Block block = new NTCP2Payload.TerminationBlock(reason, validFramesRcvd);
plen += block.getTotalLength();
int plen = block.getTotalLength();
blocks.add(block);
int padlen = 1 + _context.random().nextInt(PADDING_MAX);
int padlen = getPaddingSize(plen, PADDING_MAX);
if (padlen > 0) {
// all zeros is fine here
//block = new NTCP2Payload.PaddingBlock(_context, padlen);
block = new NTCP2Payload.PaddingBlock(padlen);
plen += block.getTotalLength();
blocks.add(block);
byte[] tmp = new byte[plen];
sendNTCP2(tmp, blocks);
}
// use a "read buf" for the temp array
ByteArray dataBuf = acquireReadBuf();
sendNTCP2(dataBuf.getData(), blocks);
releaseReadBuf(dataBuf);
}
/**
@ -998,10 +1028,15 @@ public class NTCPConnection implements Closeable {
* passes it to the pumper for writing.
*
* @param tmp to be used for output of NTCP2Payload.writePayload(),
* must have room for 2 byte length and block output
* must have room for block output. May be released immediately on return.
* @since 0.9.36
*/
private synchronized void sendNTCP2(byte[] tmp, List<Block> blocks) {
if (_sender == null) {
if (_log.shouldWarn())
_log.warn("sender gone", new Exception());
return;
}
int payloadlen = NTCP2Payload.writePayload(tmp, 0, blocks);
int framelen = payloadlen + OutboundNTCP2State.MAC_SIZE;
// TODO use a buffer
@ -2039,9 +2074,6 @@ public class NTCPConnection implements Closeable {
_transport.messageReceived(msg, _remotePeer, null, timeToRecv, size);
_lastReceiveTime = _context.clock().now();
_messagesRead.incrementAndGet();
// TEST send back. null RI for target, not necesary
//if (_context.getBooleanProperty("i2np.ntcp2.loopback"))
// send(new OutNetMessage(_context, msg, _context.clock().now() + 10*1000, OutNetMessage.PRIORITY_MY_DATA, null));
}
public void gotOptions(byte[] options, boolean isHandshake) {