use of com.biglybt.pif.ddb.DistributedDatabase in project BiglyBT by BiglySoftware.
the class DHTPlugin method initialize.
@Override
public void initialize(PluginInterface _plugin_interface) {
status = STATUS_INITALISING;
plugin_interface = _plugin_interface;
dht_data_port = UDPNetworkManager.getSingleton().getUDPNonDataListeningPortNumber();
log = plugin_interface.getLogger().getTimeStampedChannel(PLUGIN_NAME);
UIManager ui_manager = plugin_interface.getUIManager();
final BasicPluginViewModel model = ui_manager.createBasicPluginViewModel(PLUGIN_RESOURCE_ID);
model.setConfigSectionID(PLUGIN_CONFIGSECTION_ID);
BasicPluginConfigModel config = ui_manager.createBasicPluginConfigModel(ConfigSection.SECTION_PLUGINS, PLUGIN_CONFIGSECTION_ID);
config.addLabelParameter2("dht.info");
final BooleanParameter enabled_param = config.addBooleanParameter2("dht.enabled", "dht.enabled", true);
plugin_interface.getPluginconfig().addListener(new PluginConfigListener() {
@Override
public void configSaved() {
int new_dht_data_port = UDPNetworkManager.getSingleton().getUDPNonDataListeningPortNumber();
if (new_dht_data_port != dht_data_port) {
changePort(new_dht_data_port);
}
}
});
LabelParameter reseed_label = config.addLabelParameter2("dht.reseed.label");
final StringParameter reseed_ip = config.addStringParameter2("dht.reseed.ip", "dht.reseed.ip", "");
final IntParameter reseed_port = config.addIntParameter2("dht.reseed.port", "dht.reseed.port", 0);
reseed = config.addActionParameter2("dht.reseed.info", "dht.reseed");
reseed.setEnabled(false);
config.createGroup("dht.reseed.group", new Parameter[] { reseed_label, reseed_ip, reseed_port, reseed });
final BooleanParameter ipfilter_logging_param = config.addBooleanParameter2("dht.ipfilter.log", "dht.ipfilter.log", true);
ipfilter_logging[0] = ipfilter_logging_param.getValue();
ipfilter_logging_param.addListener(new ParameterListener() {
@Override
public void parameterChanged(Parameter p) {
ipfilter_logging[0] = ipfilter_logging_param.getValue();
}
});
warn_user = config.addBooleanParameter2("dht.warn.user", "dht.warn.user", true);
prefer_i2p = config.addBooleanParameter2("dht.prefer.i2p", "dht.prefer.i2p", false);
BooleanParameter sleeping = config.addBooleanParameter2("dht.is.sleeping", "dht.is.sleeping", false);
AERunStateHandler.addListener(new AERunStateHandler.RunStateChangeListener() {
@Override
public void runStateChanged(long run_state) {
sleeping.setValue(AERunStateHandler.isDHTSleeping());
}
}, true);
sleeping.addListener(new ParameterListener() {
@Override
public void parameterChanged(Parameter param) {
AERunStateHandler.setDHTSleeping(sleeping.getValue());
}
});
final BooleanParameter advanced = config.addBooleanParameter2("dht.advanced", "dht.advanced", false);
LabelParameter advanced_label = config.addLabelParameter2("dht.advanced.label");
final StringParameter override_ip = config.addStringParameter2("dht.override.ip", "dht.override.ip", "");
config.createGroup("dht.advanced.group", new Parameter[] { advanced_label, override_ip });
advanced.addEnabledOnSelection(advanced_label);
advanced.addEnabledOnSelection(override_ip);
final StringParameter command = config.addStringParameter2("dht.execute.command", "dht.execute.command", "print");
ActionParameter execute = config.addActionParameter2("dht.execute.info", "dht.execute");
final BooleanParameter logging = config.addBooleanParameter2("dht.logging", "dht.logging", false);
config.createGroup("dht.diagnostics.group", new Parameter[] { command, execute, logging });
logging.addListener(new ParameterListener() {
@Override
public void parameterChanged(Parameter param) {
if (dhts != null) {
for (int i = 0; i < dhts.length; i++) {
dhts[i].setLogging(logging.getValue());
}
}
}
});
final DHTPluginOperationListener log_polistener = new DHTPluginOperationListener() {
@Override
public boolean diversified() {
return (true);
}
@Override
public void starts(byte[] key) {
}
@Override
public void valueRead(DHTPluginContact originator, DHTPluginValue value) {
log.log("valueRead: " + new String(value.getValue()) + " from " + originator.getName() + "/" + originator.getAddress() + ", flags=" + Integer.toHexString(value.getFlags() & 0x00ff));
if ((value.getFlags() & DHTPlugin.FLAG_STATS) != 0) {
DHTPluginKeyStats stats = decodeStats(value);
log.log(" stats: size=" + (stats == null ? "null" : stats.getSize()));
}
}
@Override
public void valueWritten(DHTPluginContact target, DHTPluginValue value) {
log.log("valueWritten:" + new String(value.getValue()) + " to " + target.getName() + "/" + target.getAddress());
}
@Override
public void complete(byte[] key, boolean timeout_occurred) {
log.log("complete: timeout = " + timeout_occurred);
}
};
execute.addListener(new ParameterListener() {
@Override
public void parameterChanged(Parameter param) {
AEThread2 t = new AEThread2("DHT:commandrunner", true) {
@Override
public void run() {
try {
if (dhts == null) {
return;
}
String c = command.getValue().trim();
String lc = c.toLowerCase();
if (lc.equals("suspend")) {
if (!setSuspended(true)) {
Debug.out("Suspend failed");
}
return;
} else if (lc.equals("resume")) {
if (!setSuspended(false)) {
Debug.out("Resume failed");
}
return;
} else if (lc.equals("bridge_put")) {
try {
List<DistributedDatabase> ddbs = plugin_interface.getUtilities().getDistributedDatabases(new String[] { AENetworkClassifier.AT_I2P });
DistributedDatabase ddb = ddbs.get(0);
DistributedDatabaseKey key = ddb.createKey("fred");
key.setFlags(DistributedDatabaseKey.FL_BRIDGED);
ddb.write(new DistributedDatabaseListener() {
@Override
public void event(DistributedDatabaseEvent event) {
// TODO Auto-generated method stub
}
}, key, ddb.createValue("bill"));
} catch (Throwable e) {
e.printStackTrace();
}
return;
}
for (int i = 0; i < dhts.length; i++) {
DHT dht = dhts[i].getDHT();
DHTTransportUDP transport = (DHTTransportUDP) dht.getTransport();
if (lc.equals("print")) {
dht.print(true);
dhts[i].logStats();
} else if (lc.equals("pingall")) {
if (i == 1) {
dht.getControl().pingAll();
}
} else if (lc.equals("versions")) {
List<DHTRouterContact> contacts = dht.getRouter().getAllContacts();
Map<Byte, Integer> counts = new TreeMap<>();
for (DHTRouterContact r : contacts) {
DHTControlContact contact = (DHTControlContact) r.getAttachment();
byte v = contact.getTransportContact().getProtocolVersion();
Integer count = counts.get(v);
if (count == null) {
counts.put(v, 1);
} else {
counts.put(v, count + 1);
}
}
log.log("Net " + dht.getTransport().getNetwork());
int total = contacts.size();
if (total == 0) {
log.log(" no contacts");
} else {
String ver = "";
for (Map.Entry<Byte, Integer> entry : counts.entrySet()) {
ver += (ver.length() == 0 ? "" : ", ") + entry.getKey() + "=" + 100 * entry.getValue() / total + "%";
}
log.log(" contacts=" + total + ": " + ver);
}
} else if (lc.equals("testca")) {
((DHTTransportUDPImpl) transport).testExternalAddressChange();
} else if (lc.equals("testnd")) {
((DHTTransportUDPImpl) transport).testNetworkAlive(false);
} else if (lc.equals("testna")) {
((DHTTransportUDPImpl) transport).testNetworkAlive(true);
} else {
int pos = c.indexOf(' ');
if (pos != -1) {
String lhs = lc.substring(0, pos);
String rhs = c.substring(pos + 1);
if (lhs.equals("set")) {
pos = rhs.indexOf('=');
if (pos != -1) {
DHTPlugin.this.put(rhs.substring(0, pos).getBytes(), "DHT Plugin: set", rhs.substring(pos + 1).getBytes(), (byte) 0, log_polistener);
}
} else if (lhs.equals("get")) {
DHTPlugin.this.get(rhs.getBytes("UTF-8"), "DHT Plugin: get", (byte) 0, 1, 10000, true, false, log_polistener);
} else if (lhs.equals("query")) {
DHTPlugin.this.get(rhs.getBytes("UTF-8"), "DHT Plugin: get", DHTPlugin.FLAG_STATS, 1, 10000, true, false, log_polistener);
} else if (lhs.equals("punch")) {
Map originator_data = new HashMap();
originator_data.put("hello", "mum");
DHTNATPuncher puncher = dht.getNATPuncher();
if (puncher != null) {
puncher.punch("Test", transport.getLocalContact(), null, originator_data);
}
} else if (lhs.equals("stats")) {
try {
pos = rhs.lastIndexOf(":");
DHTTransportContact contact;
if (pos == -1) {
contact = transport.getLocalContact();
} else {
String host = rhs.substring(0, pos);
int port = Integer.parseInt(rhs.substring(pos + 1));
contact = transport.importContact(new InetSocketAddress(host, port), transport.getProtocolVersion(), false);
}
log.log("Stats request to " + contact.getName());
DHTTransportFullStats stats = contact.getStats();
log.log("Stats:" + (stats == null ? "<null>" : stats.getString()));
DHTControlActivity[] activities = dht.getControl().getActivities();
for (int j = 0; j < activities.length; j++) {
log.log(" act:" + activities[j].getString());
}
} catch (Throwable e) {
Debug.printStackTrace(e);
}
}
}
}
}
} catch (Throwable e) {
Debug.out(e);
}
}
};
t.start();
}
});
reseed.addListener(new ParameterListener() {
@Override
public void parameterChanged(Parameter param) {
reseed.setEnabled(false);
AEThread2 t = new AEThread2("DHT:reseeder", true) {
@Override
public void run() {
try {
String ip = reseed_ip.getValue().trim();
if (dhts == null) {
return;
}
int port = reseed_port.getValue();
for (int i = 0; i < dhts.length; i++) {
DHTPluginImpl dht = dhts[i];
if (ip.length() == 0 || port == 0) {
dht.checkForReSeed(true);
} else {
DHTTransportContact seed = dht.importSeed(ip, port);
if (seed != null) {
dht.integrateDHT(false, seed);
}
}
}
} finally {
reseed.setEnabled(true);
}
}
};
t.start();
}
});
model.getActivity().setVisible(false);
model.getProgress().setVisible(false);
log.addListener(new LoggerChannelListener() {
@Override
public void messageLogged(int type, String message) {
model.getLogArea().appendText(message + "\n");
}
@Override
public void messageLogged(String str, Throwable error) {
model.getLogArea().appendText(error.toString() + "\n");
}
});
dht_log = new DHTLogger() {
@Override
public void log(String str) {
log.log(str);
}
@Override
public void log(Throwable e) {
log.log(e);
}
@Override
public void log(int log_type, String str) {
if (isEnabled(log_type)) {
log.log(str);
}
}
@Override
public boolean isEnabled(int log_type) {
if (log_type == DHTLogger.LT_IP_FILTER) {
return ipfilter_logging[0];
}
return (true);
}
@Override
public PluginInterface getPluginInterface() {
return (log.getLogger().getPluginInterface());
}
};
if (!enabled_param.getValue()) {
model.getStatus().setText("Disabled");
status = STATUS_DISABLED;
init_sem.releaseForever();
return;
}
setPluginInfo();
plugin_interface.addListener(new PluginListener() {
@Override
public void initializationComplete() {
PluginInterface pi_upnp = plugin_interface.getPluginManager().getPluginInterfaceByClass(UPnPPlugin.class);
if (pi_upnp == null) {
log.log("UPnP plugin not found, can't map port");
} else {
upnp_mapping = ((UPnPPlugin) pi_upnp.getPlugin()).addMapping(plugin_interface.getPluginName(), false, dht_data_port, true);
}
String ip = null;
if (advanced.getValue()) {
ip = override_ip.getValue().trim();
if (ip.length() == 0) {
ip = null;
}
}
initComplete(model.getStatus(), logging.getValue(), ip);
}
@Override
public void closedownInitiated() {
if (dhts != null) {
for (int i = 0; i < dhts.length; i++) {
dhts[i].closedownInitiated();
}
}
saveClockSkew();
}
@Override
public void closedownComplete() {
}
});
final int sample_frequency = 60 * 1000;
// every 15 mins
final int sample_stats_ticks = 15;
plugin_interface.getUtilities().createTimer("DHTStats", true).addPeriodicEvent(sample_frequency, new UTTimerEventPerformer() {
@Override
public void perform(UTTimerEvent event) {
if (dhts != null) {
for (int i = 0; i < dhts.length; i++) {
dhts[i].updateStats(sample_stats_ticks);
}
}
setPluginInfo();
saveClockSkew();
}
});
}
use of com.biglybt.pif.ddb.DistributedDatabase in project BiglyBT by BiglySoftware.
the class RelatedContentManager method checkMixSearcher.
private RelatedContentSearcher checkMixSearcher() {
synchronized (searchers) {
if (mix_searcher == null) {
List<DistributedDatabase> ddbs = DDBaseImpl.getDDBs(new String[] { AENetworkClassifier.AT_PUBLIC, AENetworkClassifier.AT_I2P });
for (DistributedDatabase ddb : ddbs) {
if (ddb.getNetwork() == AENetworkClassifier.AT_I2P) {
DHTPluginInterface i2p_mix_dht = ddb.getDHTPlugin();
mix_searcher = new RelatedContentSearcher(RelatedContentManager.this, transfer_type, i2p_mix_dht, false);
}
}
}
return (mix_searcher);
}
}
use of com.biglybt.pif.ddb.DistributedDatabase in project BiglyBT by BiglySoftware.
the class RelatedContentManager method checkI2PSearcher.
void checkI2PSearcher(boolean force) {
synchronized (searchers) {
if (added_i2p_searcher) {
return;
}
if (!force) {
DownloadManager dm = plugin_interface.getDownloadManager();
Download[] downloads = dm.getDownloads();
boolean found = false;
for (Download download : downloads) {
String[] nets = PluginCoreUtils.unwrap(download).getDownloadState().getNetworks();
if (nets.length == 1 && nets[0] == AENetworkClassifier.AT_I2P) {
found = true;
break;
}
}
if (!found) {
return;
}
}
List<DistributedDatabase> ddbs = DDBaseImpl.getDDBs(new String[] { AENetworkClassifier.AT_I2P });
for (DistributedDatabase ddb : ddbs) {
if (ddb.getNetwork() == AENetworkClassifier.AT_I2P) {
DHTPluginInterface i2p_dht = ddb.getDHTPlugin();
RelatedContentSearcher i2p_searcher = new RelatedContentSearcher(RelatedContentManager.this, transfer_type, i2p_dht, false);
searchers.add(i2p_searcher);
added_i2p_searcher = true;
}
}
}
}
Aggregations