- Switch back from storing NID to full NodeInfo for outgoing tokens so they don't get expired early

- Announce only to the single closest DHT peer
- Increase random port range
- Decrease max local tracker and DHT size
This commit is contained in:
zzz
2012-06-04 14:15:38 +00:00
parent 121491a3be
commit d5cb443925
4 changed files with 17 additions and 16 deletions

View File

@ -374,7 +374,8 @@ public class TrackerClient extends I2PAppThread
// announce ourselves while the token is still good // announce ourselves while the token is still good
// FIXME this needs to be in its own thread // FIXME this needs to be in its own thread
if (!stop) { if (!stop) {
int good = _util.getDHT().announce(snark.getInfoHash(), 8, 5*60*1000); // announce only to the 1 closest
int good = _util.getDHT().announce(snark.getInfoHash(), 1, 5*60*1000);
_util.debug("Sent " + good + " good announces to DHT", Snark.INFO); _util.debug("Sent " + good + " good announces to DHT", Snark.INFO);
} }

View File

@ -39,7 +39,7 @@ class DHTNodes extends ConcurrentHashMap<NID, NodeInfo> {
private static final long MAX_EXPIRE_TIME = 60*60*1000; private static final long MAX_EXPIRE_TIME = 60*60*1000;
private static final long MIN_EXPIRE_TIME = 5*60*1000; private static final long MIN_EXPIRE_TIME = 5*60*1000;
private static final long DELTA_EXPIRE_TIME = 7*60*1000; private static final long DELTA_EXPIRE_TIME = 7*60*1000;
private static final int MAX_PEERS = 9999; private static final int MAX_PEERS = 999;
public DHTNodes(I2PAppContext ctx) { public DHTNodes(I2PAppContext ctx) {
super(); super();

View File

@ -34,7 +34,7 @@ class DHTTracker {
private static final long MAX_EXPIRE_TIME = 95*60*1000; private static final long MAX_EXPIRE_TIME = 95*60*1000;
private static final long MIN_EXPIRE_TIME = 5*60*1000; private static final long MIN_EXPIRE_TIME = 5*60*1000;
private static final long DELTA_EXPIRE_TIME = 7*60*1000; private static final long DELTA_EXPIRE_TIME = 7*60*1000;
private static final int MAX_PEERS = 9999; private static final int MAX_PEERS = 2000;
DHTTracker(I2PAppContext ctx) { DHTTracker(I2PAppContext ctx) {
_context = ctx; _context = ctx;

View File

@ -92,7 +92,7 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
/** index to sent queries awaiting reply */ /** index to sent queries awaiting reply */
private final ConcurrentHashMap<MsgID, ReplyWaiter> _sentQueries; private final ConcurrentHashMap<MsgID, ReplyWaiter> _sentQueries;
/** index to outgoing tokens we generated, sent in reply to a get_peers query */ /** index to outgoing tokens we generated, sent in reply to a get_peers query */
private final ConcurrentHashMap<Token, NID> _outgoingTokens; private final ConcurrentHashMap<Token, NodeInfo> _outgoingTokens;
/** index to incoming opaque tokens, received in a peers or nodes reply */ /** index to incoming opaque tokens, received in a peers or nodes reply */
private final ConcurrentHashMap<NID, Token> _incomingTokens; private final ConcurrentHashMap<NID, Token> _incomingTokens;
@ -148,8 +148,9 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
_incomingTokens = new ConcurrentHashMap(); _incomingTokens = new ConcurrentHashMap();
// Construct my NodeInfo // Construct my NodeInfo
// ports can really be fixed, just do this for testing // Pick ports over a big range to marginally increase security
_qPort = 30000 + ctx.random().nextInt(99); // If we add a search DHT, adjust to stay out of each other's way
_qPort = 2555 + ctx.random().nextInt(61111);
_rPort = _qPort + 1; _rPort = _qPort + 1;
_myID = new byte[NID.HASH_LENGTH]; _myID = new byte[NID.HASH_LENGTH];
ctx.random().nextBytes(_myID); ctx.random().nextBytes(_myID);
@ -291,6 +292,7 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
/** /**
* Get peers for a torrent. * Get peers for a torrent.
* This is an iterative lookup in the DHT.
* Blocking! * Blocking!
* Caller should run in a thread. * Caller should run in a thread.
* *
@ -404,13 +406,16 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
} }
/** /**
* Announce to the closest DHT peers. * Announce to the closest peers in the local DHT.
* This is NOT iterative - call getPeers() first to get the closest
* peers into the local DHT.
* Blocking unless maxWait <= 0 * Blocking unless maxWait <= 0
* Caller should run in a thread. * Caller should run in a thread.
* This also automatically announces ourself to our local tracker. * This also automatically announces ourself to our local tracker.
* For best results do a getPeers() first so we have tokens. * For best results do a getPeers() first so we have tokens.
* *
* @param ih the Info Hash (torrent) * @param ih the Info Hash (torrent)
* @param max maximum number of peers to announce to
* @param maxWait the maximum total time to wait (ms) or 0 to do all in parallel and return immediately. * @param maxWait the maximum total time to wait (ms) or 0 to do all in parallel and return immediately.
* @return the number of successful announces, not counting ourselves. * @return the number of successful announces, not counting ourselves.
*/ */
@ -842,6 +847,7 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
} }
try { try {
// TODO I2CP per-packet options
boolean success = _session.sendMessage(dest, payload, 0, payload.length, null, null, 60*1000, boolean success = _session.sendMessage(dest, payload, 0, payload.length, null, null, 60*1000,
repliable ? I2PSession.PROTO_DATAGRAM : I2PSession.PROTO_DATAGRAM_RAW, repliable ? I2PSession.PROTO_DATAGRAM : I2PSession.PROTO_DATAGRAM_RAW,
fromPort, toPort); fromPort, toPort);
@ -1041,7 +1047,7 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
_log.info("Rcvd get_peers from: " + nInfo + " for: " + ih); _log.info("Rcvd get_peers from: " + nInfo + " for: " + ih);
// generate and save random token // generate and save random token
Token token = new Token(_context); Token token = new Token(_context);
_outgoingTokens.put(token, nInfo.getNID()); _outgoingTokens.put(token, nInfo);
if (_log.shouldLog(Log.INFO)) if (_log.shouldLog(Log.INFO))
_log.info("Stored new OB token: " + token + " for: " + nInfo); _log.info("Stored new OB token: " + token + " for: " + nInfo);
@ -1074,20 +1080,14 @@ public class KRPC implements I2PSessionMuxedListener, DHT {
*/ */
private void receiveAnnouncePeer(MsgID msgID, InfoHash ih, byte[] tok) throws InvalidBEncodingException { private void receiveAnnouncePeer(MsgID msgID, InfoHash ih, byte[] tok) throws InvalidBEncodingException {
Token token = new Token(tok); Token token = new Token(tok);
NID nid = _outgoingTokens.get(token); NodeInfo nInfo = _outgoingTokens.get(token);
if (nid == null) { if (nInfo == null) {
if (_log.shouldLog(Log.WARN)) if (_log.shouldLog(Log.WARN))
_log.warn("Unknown token in announce_peer: " + token); _log.warn("Unknown token in announce_peer: " + token);
if (_log.shouldLog(Log.INFO)) if (_log.shouldLog(Log.INFO))
_log.info("Current known tokens: " + _outgoingTokens.keySet()); _log.info("Current known tokens: " + _outgoingTokens.keySet());
return; return;
} }
NodeInfo nInfo = _knownNodes.get(nid);
if (nInfo == null) {
if (_log.shouldLog(Log.WARN))
_log.warn("Unknown node in announce_peer for: " + nid);
return;
}
if (_log.shouldLog(Log.INFO)) if (_log.shouldLog(Log.INFO))
_log.info("Rcvd announce from: " + nInfo + " for: " + ih); _log.info("Rcvd announce from: " + nInfo + " for: " + ih);