forked from I2P_Developers/i2p.i2p
RoutingKeyGenerator:
- Move from core to RouterKeyGenerator in router.jar - Leave RoutingKeyGenerator as a simple abstract class - DatabaseEntry now uses timestamp instead of mod data to determine if mod data has changed. Don't expose mod data to DatabaseEntry any more. - I2PAppContext.routingKeyGenerator() now returns null; you must be in RouterContext to get a generator.
This commit is contained in:
224
router/java/src/net/i2p/data/router/RouterKeyGenerator.java
Normal file
224
router/java/src/net/i2p/data/router/RouterKeyGenerator.java
Normal file
@@ -0,0 +1,224 @@
|
||||
package net.i2p.data.router;
|
||||
|
||||
/*
|
||||
* free (adj.): unencumbered; not under the control of others
|
||||
* Written by jrandom in 2003 and released into the public domain
|
||||
* with no warranty of any kind, either expressed or implied.
|
||||
* It probably won't make your computer catch on fire, or eat
|
||||
* your children, but it might. Use at your own risk.
|
||||
*
|
||||
*/
|
||||
|
||||
import java.text.SimpleDateFormat;
|
||||
import java.util.Calendar;
|
||||
import java.util.Date;
|
||||
import java.util.Arrays;
|
||||
import java.util.GregorianCalendar;
|
||||
import java.util.Locale;
|
||||
import java.util.TimeZone;
|
||||
|
||||
import net.i2p.I2PAppContext;
|
||||
import net.i2p.crypto.SHA256Generator;
|
||||
import net.i2p.data.Hash;
|
||||
import net.i2p.data.RoutingKeyGenerator;
|
||||
import net.i2p.util.HexDump;
|
||||
import net.i2p.util.Log;
|
||||
|
||||
/**
|
||||
* Component to manage the munging of hashes into routing keys - given a hash,
|
||||
* perform some consistent transformation against it and return the result.
|
||||
* This transformation is fed by the current "mod data".
|
||||
*
|
||||
* Right now the mod data is the current date (GMT) as a string: "yyyyMMdd",
|
||||
* and the transformation takes the original hash, appends the bytes of that mod data,
|
||||
* then returns the SHA256 of that concatenation.
|
||||
*
|
||||
* Do we want this to simply do the XOR of the SHA256 of the current mod data and
|
||||
* the key? does that provide the randomization we need? It'd save an SHA256 op.
|
||||
* Bah, too much effort to think about for so little gain. Other algorithms may come
|
||||
* into play layer on about making periodic updates to the routing key for data elements
|
||||
* to mess with Sybil. This may be good enough though.
|
||||
*
|
||||
* Also - the method generateDateBasedModData() should be called after midnight GMT
|
||||
* once per day to generate the correct routing keys!
|
||||
*
|
||||
* @since 0.9.16 moved from net.i2p.data.RoutingKeyGenerator..
|
||||
*
|
||||
*/
|
||||
public class RouterKeyGenerator extends RoutingKeyGenerator {
|
||||
private final Log _log;
|
||||
private final I2PAppContext _context;
|
||||
|
||||
public RouterKeyGenerator(I2PAppContext context) {
|
||||
_log = context.logManager().getLog(RoutingKeyGenerator.class);
|
||||
_context = context;
|
||||
// ensure non-null mod data
|
||||
generateDateBasedModData();
|
||||
}
|
||||
|
||||
private volatile byte _currentModData[];
|
||||
private volatile byte _nextModData[];
|
||||
private volatile long _nextMidnight;
|
||||
private volatile long _lastChanged;
|
||||
|
||||
private final static Calendar _cal = GregorianCalendar.getInstance(TimeZone.getTimeZone("GMT"));
|
||||
private static final String FORMAT = "yyyyMMdd";
|
||||
private static final int LENGTH = FORMAT.length();
|
||||
private final static SimpleDateFormat _fmt = new SimpleDateFormat(FORMAT, Locale.US);
|
||||
static {
|
||||
// make sure GMT is set, azi2phelper Vuze plugin is disabling static JVM TZ setting in Router.java
|
||||
_fmt.setCalendar(_cal);
|
||||
}
|
||||
|
||||
/**
|
||||
* The current (today's) mod data.
|
||||
* Warning - not a copy, do not corrupt.
|
||||
*
|
||||
* @return non-null, 8 bytes
|
||||
*/
|
||||
public byte[] getModData() {
|
||||
return _currentModData;
|
||||
}
|
||||
|
||||
/**
|
||||
* Tomorrow's mod data.
|
||||
* Warning - not a copy, do not corrupt.
|
||||
* For debugging use only.
|
||||
*
|
||||
* @return non-null, 8 bytes
|
||||
* @since 0.9.10
|
||||
*/
|
||||
public byte[] getNextModData() {
|
||||
return _nextModData;
|
||||
}
|
||||
|
||||
public long getLastChanged() {
|
||||
return _lastChanged;
|
||||
}
|
||||
|
||||
/**
|
||||
* How long until midnight (ms)
|
||||
*
|
||||
* @return could be slightly negative
|
||||
* @since 0.9.10 moved from UpdateRoutingKeyModifierJob
|
||||
*/
|
||||
public long getTimeTillMidnight() {
|
||||
return _nextMidnight - _context.clock().now();
|
||||
}
|
||||
|
||||
/**
|
||||
* Set _cal to midnight for the time given.
|
||||
* Caller must synch.
|
||||
* @since 0.9.10
|
||||
*/
|
||||
private void setCalToPreviousMidnight(long now) {
|
||||
_cal.setTime(new Date(now));
|
||||
_cal.set(Calendar.YEAR, _cal.get(Calendar.YEAR)); // gcj <= 4.0 workaround
|
||||
_cal.set(Calendar.DAY_OF_YEAR, _cal.get(Calendar.DAY_OF_YEAR)); // gcj <= 4.0 workaround
|
||||
_cal.set(Calendar.HOUR_OF_DAY, 0);
|
||||
_cal.set(Calendar.MINUTE, 0);
|
||||
_cal.set(Calendar.SECOND, 0);
|
||||
_cal.set(Calendar.MILLISECOND, 0);
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate mod data from _cal.
|
||||
* Caller must synch.
|
||||
* @since 0.9.10
|
||||
*/
|
||||
private byte[] generateModDataFromCal() {
|
||||
Date today = _cal.getTime();
|
||||
|
||||
String modVal = _fmt.format(today);
|
||||
if (modVal.length() != LENGTH)
|
||||
throw new IllegalStateException();
|
||||
byte[] mod = new byte[LENGTH];
|
||||
for (int i = 0; i < LENGTH; i++)
|
||||
mod[i] = (byte)(modVal.charAt(i) & 0xFF);
|
||||
return mod;
|
||||
}
|
||||
|
||||
/**
|
||||
* Update the current modifier data with some bytes derived from the current
|
||||
* date (yyyyMMdd in GMT)
|
||||
*
|
||||
* @return true if changed
|
||||
*/
|
||||
public synchronized boolean generateDateBasedModData() {
|
||||
long now = _context.clock().now();
|
||||
setCalToPreviousMidnight(now);
|
||||
byte[] mod = generateModDataFromCal();
|
||||
boolean changed = !Arrays.equals(_currentModData, mod);
|
||||
if (changed) {
|
||||
// add a day and store next midnight and mod data for convenience
|
||||
_cal.add(Calendar.DATE, 1);
|
||||
_nextMidnight = _cal.getTime().getTime();
|
||||
byte[] next = generateModDataFromCal();
|
||||
_currentModData = mod;
|
||||
_nextModData = next;
|
||||
// ensure version is bumped
|
||||
if (_lastChanged == now)
|
||||
now++;
|
||||
_lastChanged = now;
|
||||
if (_log.shouldLog(Log.INFO))
|
||||
_log.info("Routing modifier generated: " + HexDump.dump(mod));
|
||||
}
|
||||
return changed;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate a modified (yet consistent) hash from the origKey by generating the
|
||||
* SHA256 of the targetKey with the current modData appended to it
|
||||
*
|
||||
* This makes Sybil's job a lot harder, as she needs to essentially take over the
|
||||
* whole keyspace.
|
||||
*
|
||||
* @throws IllegalArgumentException if origKey is null
|
||||
*/
|
||||
public Hash getRoutingKey(Hash origKey) {
|
||||
return getKey(origKey, _currentModData);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the routing key using tomorrow's modData, not today's
|
||||
*
|
||||
* @since 0.9.10
|
||||
*/
|
||||
public Hash getNextRoutingKey(Hash origKey) {
|
||||
return getKey(origKey, _nextModData);
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate a modified (yet consistent) hash from the origKey by generating the
|
||||
* SHA256 of the targetKey with the specified modData appended to it
|
||||
*
|
||||
* @throws IllegalArgumentException if origKey is null
|
||||
*/
|
||||
private static Hash getKey(Hash origKey, byte[] modData) {
|
||||
if (origKey == null) throw new IllegalArgumentException("Original key is null");
|
||||
byte modVal[] = new byte[Hash.HASH_LENGTH + LENGTH];
|
||||
System.arraycopy(origKey.getData(), 0, modVal, 0, Hash.HASH_LENGTH);
|
||||
System.arraycopy(modData, 0, modVal, Hash.HASH_LENGTH, LENGTH);
|
||||
return SHA256Generator.getInstance().calculateHash(modVal);
|
||||
}
|
||||
|
||||
/****
|
||||
public static void main(String args[]) {
|
||||
Hash k1 = new Hash();
|
||||
byte k1d[] = new byte[Hash.HASH_LENGTH];
|
||||
RandomSource.getInstance().nextBytes(k1d);
|
||||
k1.setData(k1d);
|
||||
|
||||
for (int i = 0; i < 10; i++) {
|
||||
System.out.println("K1: " + k1);
|
||||
Hash k1m = RoutingKeyGenerator.getInstance().getRoutingKey(k1);
|
||||
System.out.println("MOD: " + new String(RoutingKeyGenerator.getInstance().getModData()));
|
||||
System.out.println("K1M: " + k1m);
|
||||
}
|
||||
try {
|
||||
Thread.sleep(2000);
|
||||
} catch (Throwable t) { // nop
|
||||
}
|
||||
}
|
||||
****/
|
||||
}
|
@@ -1070,7 +1070,7 @@ public class Router implements RouterClock.ClockShiftListener {
|
||||
return;
|
||||
_eventLog.addEvent(EventLog.CLOCK_SHIFT, Long.toString(delta));
|
||||
// update the routing key modifier
|
||||
_context.routingKeyGenerator().generateDateBasedModData();
|
||||
_context.routerKeyGenerator().generateDateBasedModData();
|
||||
if (_context.commSystem().countActivePeers() <= 0)
|
||||
return;
|
||||
if (delta > 0)
|
||||
|
@@ -10,7 +10,9 @@ import java.util.concurrent.CopyOnWriteArrayList;
|
||||
import net.i2p.I2PAppContext;
|
||||
import net.i2p.app.ClientAppManager;
|
||||
import net.i2p.data.Hash;
|
||||
import net.i2p.data.RoutingKeyGenerator;
|
||||
import net.i2p.data.router.RouterInfo;
|
||||
import net.i2p.data.router.RouterKeyGenerator;
|
||||
import net.i2p.internal.InternalClientManager;
|
||||
import net.i2p.router.client.ClientManagerFacadeImpl;
|
||||
import net.i2p.router.crypto.TransientSessionKeyManager;
|
||||
@@ -65,6 +67,7 @@ public class RouterContext extends I2PAppContext {
|
||||
//private MessageStateMonitor _messageStateMonitor;
|
||||
private RouterThrottle _throttle;
|
||||
private RouterAppManager _appManager;
|
||||
private RouterKeyGenerator _routingKeyGenerator;
|
||||
private final Set<Runnable> _finalShutdownTasks;
|
||||
// split up big lock on this to avoid deadlocks
|
||||
private volatile boolean _initialized;
|
||||
@@ -183,6 +186,7 @@ public class RouterContext extends I2PAppContext {
|
||||
_messageHistory = new MessageHistory(this);
|
||||
_messageRegistry = new OutboundMessageRegistry(this);
|
||||
//_messageStateMonitor = new MessageStateMonitor(this);
|
||||
_routingKeyGenerator = new RouterKeyGenerator(this);
|
||||
if (!getBooleanProperty("i2p.dummyNetDb"))
|
||||
_netDb = new FloodfillNetworkDatabaseFacade(this); // new KademliaNetworkDatabaseFacade(this);
|
||||
else
|
||||
@@ -582,4 +586,35 @@ public class RouterContext extends I2PAppContext {
|
||||
_sessionKeyManagerInitialized = true;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Determine how much do we want to mess with the keys to turn them
|
||||
* into something we can route. This is context specific because we
|
||||
* may want to test out how things react when peers don't agree on
|
||||
* how to skew.
|
||||
*
|
||||
* Returns same thing as routerKeyGenerator()
|
||||
*
|
||||
* @return non-null
|
||||
* @since 0.9.16 Overrides I2PAppContext. Returns non-null in RouterContext and null in I2PAppcontext.
|
||||
*/
|
||||
@Override
|
||||
public RoutingKeyGenerator routingKeyGenerator() {
|
||||
return _routingKeyGenerator;
|
||||
}
|
||||
|
||||
/**
|
||||
* Determine how much do we want to mess with the keys to turn them
|
||||
* into something we can route. This is context specific because we
|
||||
* may want to test out how things react when peers don't agree on
|
||||
* how to skew.
|
||||
*
|
||||
* Returns same thing as routingKeyGenerator()
|
||||
*
|
||||
* @return non-null
|
||||
* @since 0.9.16
|
||||
*/
|
||||
public RouterKeyGenerator routerKeyGenerator() {
|
||||
return _routingKeyGenerator;
|
||||
}
|
||||
}
|
||||
|
@@ -13,6 +13,7 @@ import net.i2p.data.TunnelId;
|
||||
import net.i2p.data.i2np.DatabaseLookupMessage;
|
||||
import net.i2p.data.i2np.DatabaseStoreMessage;
|
||||
import net.i2p.data.router.RouterInfo;
|
||||
import net.i2p.data.router.RouterKeyGenerator;
|
||||
import net.i2p.router.Job;
|
||||
import net.i2p.router.JobImpl;
|
||||
import net.i2p.router.OutNetMessage;
|
||||
@@ -176,16 +177,17 @@ public class FloodfillNetworkDatabaseFacade extends KademliaNetworkDatabaseFacad
|
||||
*/
|
||||
public void flood(DatabaseEntry ds) {
|
||||
Hash key = ds.getHash();
|
||||
Hash rkey = _context.routingKeyGenerator().getRoutingKey(key);
|
||||
RouterKeyGenerator gen = _context.routerKeyGenerator();
|
||||
Hash rkey = gen.getRoutingKey(key);
|
||||
FloodfillPeerSelector sel = (FloodfillPeerSelector)getPeerSelector();
|
||||
List<Hash> peers = sel.selectFloodfillParticipants(rkey, MAX_TO_FLOOD, getKBuckets());
|
||||
// todo key cert skip?
|
||||
long until = _context.routingKeyGenerator().getTimeTillMidnight();
|
||||
long until = gen.getTimeTillMidnight();
|
||||
if (until < NEXT_RKEY_LS_ADVANCE_TIME ||
|
||||
(ds.getType() == DatabaseEntry.KEY_TYPE_ROUTERINFO && until < NEXT_RKEY_RI_ADVANCE_TIME)) {
|
||||
// to avoid lookup failures after midnight, also flood to some closest to the
|
||||
// to avoid lookup faulures after midnight, also flood to some closest to the
|
||||
// next routing key for a period of time before midnight.
|
||||
Hash nkey = _context.routingKeyGenerator().getNextRoutingKey(key);
|
||||
Hash nkey = gen.getNextRoutingKey(key);
|
||||
List<Hash> nextPeers = sel.selectFloodfillParticipants(nkey, NEXT_FLOOD_QTY, getKBuckets());
|
||||
int i = 0;
|
||||
for (Hash h : nextPeers) {
|
||||
|
@@ -8,7 +8,7 @@ package net.i2p.router.tasks;
|
||||
*
|
||||
*/
|
||||
|
||||
import net.i2p.data.RoutingKeyGenerator;
|
||||
import net.i2p.data.router.RouterKeyGenerator;
|
||||
import net.i2p.router.JobImpl;
|
||||
import net.i2p.router.RouterContext;
|
||||
import net.i2p.util.Log;
|
||||
@@ -33,7 +33,7 @@ public class UpdateRoutingKeyModifierJob extends JobImpl {
|
||||
public String getName() { return "Update Routing Key Modifier"; }
|
||||
|
||||
public void runJob() {
|
||||
RoutingKeyGenerator gen = getContext().routingKeyGenerator();
|
||||
RouterKeyGenerator gen = getContext().routerKeyGenerator();
|
||||
// make sure we requeue quickly if just before midnight
|
||||
long delay = Math.max(5, Math.min(MAX_DELAY_FAILSAFE, gen.getTimeTillMidnight()));
|
||||
// TODO tell netdb if mod data changed?
|
||||
|
Reference in New Issue
Block a user