cleanups, javadoc

2009-12-30 22:38:37 +00:00
parent cefc1f130d
commit 883b53de0a
7 changed files with 21 additions and 9 deletions
--- a/router/java/test/net/i2p/router/networkdb/kademlia/RouterGenerator.java
+++ b/router/java/test/net/i2p/router/networkdb/kademlia/RouterGenerator.java
@ -0,0 +1,136 @@
+package net.i2p.router.networkdb.kademlia;
+/*
+ * 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.io.File;
+import java.io.FileOutputStream;
+import java.math.BigInteger;
+import java.util.HashSet;
+import java.util.Properties;
+import java.util.Set;
+
+import net.i2p.crypto.KeyGenerator;
+import net.i2p.data.Certificate;
+import net.i2p.data.PrivateKey;
+import net.i2p.data.PublicKey;
+import net.i2p.data.RouterAddress;
+import net.i2p.data.RouterIdentity;
+import net.i2p.data.RouterInfo;
+import net.i2p.data.SigningPrivateKey;
+import net.i2p.data.SigningPublicKey;
+import net.i2p.util.Clock;
+
+public class RouterGenerator {
+    public static void main(String args[]) {
+	RouterGenerator gen = new RouterGenerator();
+	switch (args.length) {
+	    case 0:
+		gen.createRouters(10000, "dummyRouters");
+		break;
+	    case 1:
+		gen.createRouters(10000, args[0]);
+		break;
+	    case 2:
+		try { gen.createRouters(Integer.parseInt(args[1]), args[0]); } catch (NumberFormatException nfe) { nfe.printStackTrace(); }
+		break;
+	}
+    }
+    
+    private void createRouters(int numRouters, String outDir) {
+	File dir = new File(outDir);
+	if (!dir.exists())
+	    dir.mkdirs();
+	int numSuccess = 0;
+	for (int i = 1; numSuccess < numRouters; i++) {
+	    RouterInfo ri = createRouterInfo(i);
+	    String hash = ri.getIdentity().getHash().toBase64();
+	    if (!hash.startsWith("fwI")) {
+		System.out.print(".");
+		if ( (i % 100) == 0) System.out.println();
+		continue;
+	    }
+	    
+	    System.out.println("Router " + i + " created: \t" + hash);
+	    numSuccess++;
+	    
+	    FileOutputStream fos = null;
+	    try {
+		fos = new FileOutputStream(new File(dir, "routerInfo-" + hash + ".dat"));
+		ri.writeBytes(fos);
+	    } catch (Exception e) {
+		System.err.println("Error writing router - " + e.getMessage());
+		e.printStackTrace();
+		return;
+	    } finally {
+		if (fos != null) try { fos.close(); } catch (Exception e) {}
+	    }
+	}
+    }
+    
+    private static PublicKey pubkey = null;
+    private static PrivateKey privkey = null;
+    private static SigningPublicKey signingPubKey = null;
+    private static SigningPrivateKey signingPrivKey = null;
+    private static Object keypair[] = KeyGenerator.getInstance().generatePKIKeypair();
+    private static Object signingKeypair[] = KeyGenerator.getInstance().generateSigningKeypair();
+    
+    static {
+	pubkey = (PublicKey)keypair[0];
+	privkey = (PrivateKey)keypair[1];
+	signingPubKey = (SigningPublicKey)signingKeypair[0];
+	signingPrivKey = (SigningPrivateKey)signingKeypair[1];
+    }
+    
+    
+    static RouterInfo createRouterInfo(int num) {
+	RouterInfo info = new RouterInfo();
+	try {
+	    info.setAddresses(createAddresses(num));
+	    info.setOptions(new Properties());
+	    info.setPeers(new HashSet());
+	    info.setPublished(Clock.getInstance().now());
+	    RouterIdentity ident = new RouterIdentity();
+	    BigInteger bv = new BigInteger(""+num);
+	    Certificate cert = new Certificate(Certificate.CERTIFICATE_TYPE_NULL, bv.toByteArray());
+	    ident.setCertificate(cert);
+	    ident.setPublicKey(pubkey);
+	    ident.setSigningPublicKey(signingPubKey);
+	    info.setIdentity(ident);
+	    
+	    info.sign(signingPrivKey);
+	} catch (Exception e) {
+	    System.err.println("Error building router " + num + ": " + e.getMessage());
+	    e.printStackTrace();
+	}
+	return info;
+    }
+    
+    static Set createAddresses(int num) {
+	Set addresses = new HashSet();
+	RouterAddress addr = createTCPAddress(num);
+	if (addr != null)
+	    addresses.add(addr);
+	return addresses;
+    }
+    
+    private static RouterAddress createTCPAddress(int num) {
+	RouterAddress addr = new RouterAddress();
+	addr.setCost(10);
+	addr.setExpiration(null);
+	Properties props = new Properties();
+	String name = "blah.random.host.org";
+	String port = "" + (1024+num);
+	props.setProperty("host", name);
+	props.setProperty("port", port);
+	addr.setOptions(props);
+	addr.setTransportStyle("TCP");
+	return addr;
+    }
+    
+}