Another unit test for the Tor plugin.

This commit is contained in:
akwizgran
2012-02-06 23:41:30 +00:00
parent fbf5c78f91
commit 5656dc8df8
2 changed files with 33 additions and 3 deletions

View File

@@ -26,7 +26,6 @@ import org.silvertunnel.netlib.api.util.TcpipNetAddress;
import org.silvertunnel.netlib.layer.tor.TorHiddenServicePortPrivateNetAddress;
import org.silvertunnel.netlib.layer.tor.TorHiddenServicePrivateNetAddress;
import org.silvertunnel.netlib.layer.tor.TorNetLayerUtil;
import org.silvertunnel.netlib.layer.tor.TorNetServerSocket;
import org.silvertunnel.netlib.layer.tor.util.Encryption;
import org.silvertunnel.netlib.layer.tor.util.RSAKeyPair;
@@ -94,9 +93,9 @@ class TorPlugin implements DuplexPlugin {
NetLayer nl = netFactory.getNetLayerById(NetLayerIDs.TOR);
nl.waitUntilReady();
// Publish the hidden service
TorNetServerSocket ss;
NetServerSocket ss;
try {
ss = (TorNetServerSocket) nl.createNetServerSocket(null, addrPort);
ss = nl.createNetServerSocket(null, addrPort);
} catch(IOException e) {
if(LOG.isLoggable(Level.WARNING)) LOG.warning(e.toString());
return;

View File

@@ -49,6 +49,37 @@ public class TorPluginTest extends BriarTestCase {
clientPlugin.stop();
}
@Test
public void testStoreAndRetrievePrivateKey() throws Exception {
Executor e = Executors.newCachedThreadPool();
// Start a plugin instance with no private key
Callback callback = new Callback();
TorPlugin plugin = new TorPlugin(e, callback, 0L);
plugin.start();
// The plugin should create a hidden service... eventually
callback.latch.await(5, TimeUnit.MINUTES);
String onion = callback.local.get("onion");
assertNotNull(onion);
assertTrue(onion.endsWith(".onion"));
// Get the PEM-encoded private key and stop the plugin
String privateKey = callback.config.get("privateKey");
assertNotNull(privateKey);
plugin.stop();
// Start another instance, reusing the private key
callback = new Callback();
callback.config.put("privateKey", privateKey);
plugin = new TorPlugin(e, callback, 0L);
plugin.start();
// The plugin should create a hidden service... eventually
callback.latch.await(5, TimeUnit.MINUTES);
// The onion URL should be the same
assertEquals(onion, callback.local.get("onion"));
// The private key should be the same
assertEquals(privateKey, callback.config.get("privateKey"));
// Stop the plugin
plugin.stop();
}
private static class Callback implements DuplexPluginCallback {
private final Map<ContactId, TransportProperties> remote =