mirror of
https://code.briarproject.org/briar/briar.git
synced 2026-02-12 18:59:06 +01:00
Compare commits
136 Commits
peer-disco
...
avoid-depr
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2bf3736b2d | ||
|
|
443043ae09 | ||
|
|
a03953563f | ||
|
|
033fd2d3b4 | ||
|
|
011d8e1df7 | ||
|
|
ef5e2dad72 | ||
|
|
f35e87c8ad | ||
|
|
e4940a046a | ||
|
|
0a666df164 | ||
|
|
6fb4b95b18 | ||
|
|
5567982fb4 | ||
|
|
25e50ceb10 | ||
|
|
1495daf977 | ||
|
|
badc6da649 | ||
|
|
e065d45d16 | ||
|
|
d0c53f1310 | ||
|
|
e1084ffadd | ||
|
|
2bd2f67693 | ||
|
|
c2b0a4b8d1 | ||
|
|
ee19d2f574 | ||
|
|
e9ec5734e2 | ||
|
|
7b1c6f3fdd | ||
|
|
d689cf776c | ||
|
|
f0fd1844dd | ||
|
|
d16a301fc4 | ||
|
|
3ab88181eb | ||
|
|
802e599f09 | ||
|
|
a6bd59d3c9 | ||
|
|
b04b724028 | ||
|
|
71b0408fe6 | ||
|
|
2d38bd5734 | ||
|
|
ff5da8404a | ||
|
|
75615a4e7f | ||
|
|
96e32ad64e | ||
|
|
0fec5d7783 | ||
|
|
ee74b3774b | ||
|
|
c783a2f352 | ||
|
|
77aa5401f3 | ||
|
|
99686f5316 | ||
|
|
f5b4f6e071 | ||
|
|
a2de841e6a | ||
|
|
1f94c2d4e8 | ||
|
|
413ce29c0c | ||
|
|
c67f758c90 | ||
|
|
339524500b | ||
|
|
03811f78fa | ||
|
|
fc86c46456 | ||
|
|
7ae86d70af | ||
|
|
63e3c661a3 | ||
|
|
4f54bd90fb | ||
|
|
706c03aa8b | ||
|
|
c42a987927 | ||
|
|
297dbe0b16 | ||
|
|
4130662e1f | ||
|
|
c08bdf96cd | ||
|
|
8bb534564f | ||
|
|
5e60a717fc | ||
|
|
dd1509350c | ||
|
|
465ba3d337 | ||
|
|
7561c5039e | ||
|
|
242d6f8a0e | ||
|
|
c554847b54 | ||
|
|
d30b250389 | ||
|
|
ecea2c587d | ||
|
|
43a91e2e57 | ||
|
|
ea288b998b | ||
|
|
48dc598ca3 | ||
|
|
e2d63ac6a4 | ||
|
|
afc85cdf52 | ||
|
|
b2a1ea84f8 | ||
|
|
fcc26c093b | ||
|
|
5a741bf13b | ||
|
|
5dc460851b | ||
|
|
b805514f70 | ||
|
|
69d94c9f29 | ||
|
|
53d4b7a0df | ||
|
|
648f26542c | ||
|
|
dcb5f95934 | ||
|
|
730d553b0a | ||
|
|
7736a3b6fc | ||
|
|
95f427863d | ||
|
|
ff8a422638 | ||
|
|
78d7fc2106 | ||
|
|
cc943be540 | ||
|
|
6eb77465f6 | ||
|
|
35d1b406f7 | ||
|
|
2add63657e | ||
|
|
d3751fbead | ||
|
|
4aaa8c3b93 | ||
|
|
5b04527c54 | ||
|
|
7d6b65913a | ||
|
|
36747acac1 | ||
|
|
e8dbc00712 | ||
|
|
d3d7212b08 | ||
|
|
2919657b4a | ||
|
|
0c338b362e | ||
|
|
8dd993dd9d | ||
|
|
1b2b50d91b | ||
|
|
ee9c771045 | ||
|
|
9e6d67f13d | ||
|
|
710b6d18ce | ||
|
|
dd4aa67643 | ||
|
|
79482d5e3a | ||
|
|
ee0bf7218c | ||
|
|
c1101c7fe1 | ||
|
|
708452713d | ||
|
|
c80d3196af | ||
|
|
d1c2eb89a1 | ||
|
|
c4273d22ed | ||
|
|
21f3a9f3c7 | ||
|
|
0281eec0da | ||
|
|
d3fd309609 | ||
|
|
f2f278c393 | ||
|
|
e204d5a996 | ||
|
|
876efee1a8 | ||
|
|
8fd9a40ffb | ||
|
|
fb918457d4 | ||
|
|
b5fe55faf3 | ||
|
|
7320099494 | ||
|
|
346bec94e8 | ||
|
|
856ec61759 | ||
|
|
f61e2b399e | ||
|
|
6135f9152f | ||
|
|
84584d4d3c | ||
|
|
17239810c8 | ||
|
|
9eee58657e | ||
|
|
76425455b8 | ||
|
|
9ea7140a7f | ||
|
|
bde9800c89 | ||
|
|
4e5b6ed3e0 | ||
|
|
77d037f061 | ||
|
|
676f5faef4 | ||
|
|
8e21068465 | ||
|
|
ed50582e27 | ||
|
|
1546a05568 | ||
|
|
618ab1f1ec |
@@ -38,7 +38,7 @@ configurations {
|
|||||||
|
|
||||||
dependencies {
|
dependencies {
|
||||||
implementation project(path: ':bramble-core', configuration: 'default')
|
implementation project(path: ':bramble-core', configuration: 'default')
|
||||||
tor 'org.briarproject:tor-android:0.3.5.9@zip'
|
tor 'org.briarproject:tor-android:0.3.5.10@zip'
|
||||||
tor 'org.briarproject:obfs4proxy-android:0.0.11-2@zip'
|
tor 'org.briarproject:obfs4proxy-android:0.0.11-2@zip'
|
||||||
|
|
||||||
annotationProcessor 'com.google.dagger:dagger-compiler:2.24'
|
annotationProcessor 'com.google.dagger:dagger-compiler:2.24'
|
||||||
|
|||||||
@@ -6,6 +6,8 @@ import android.content.Context;
|
|||||||
import android.content.Intent;
|
import android.content.Intent;
|
||||||
import android.content.IntentFilter;
|
import android.content.IntentFilter;
|
||||||
import android.net.ConnectivityManager;
|
import android.net.ConnectivityManager;
|
||||||
|
import android.net.Network;
|
||||||
|
import android.net.NetworkCapabilities;
|
||||||
import android.net.NetworkInfo;
|
import android.net.NetworkInfo;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
@@ -32,12 +34,15 @@ import static android.content.Intent.ACTION_SCREEN_OFF;
|
|||||||
import static android.content.Intent.ACTION_SCREEN_ON;
|
import static android.content.Intent.ACTION_SCREEN_ON;
|
||||||
import static android.net.ConnectivityManager.CONNECTIVITY_ACTION;
|
import static android.net.ConnectivityManager.CONNECTIVITY_ACTION;
|
||||||
import static android.net.ConnectivityManager.TYPE_WIFI;
|
import static android.net.ConnectivityManager.TYPE_WIFI;
|
||||||
|
import static android.net.NetworkCapabilities.NET_CAPABILITY_INTERNET;
|
||||||
|
import static android.net.NetworkCapabilities.TRANSPORT_WIFI;
|
||||||
import static android.net.wifi.p2p.WifiP2pManager.WIFI_P2P_THIS_DEVICE_CHANGED_ACTION;
|
import static android.net.wifi.p2p.WifiP2pManager.WIFI_P2P_THIS_DEVICE_CHANGED_ACTION;
|
||||||
import static android.os.Build.VERSION.SDK_INT;
|
import static android.os.Build.VERSION.SDK_INT;
|
||||||
import static android.os.PowerManager.ACTION_DEVICE_IDLE_MODE_CHANGED;
|
import static android.os.PowerManager.ACTION_DEVICE_IDLE_MODE_CHANGED;
|
||||||
import static java.util.concurrent.TimeUnit.MINUTES;
|
import static java.util.concurrent.TimeUnit.MINUTES;
|
||||||
import static java.util.concurrent.TimeUnit.SECONDS;
|
import static java.util.concurrent.TimeUnit.SECONDS;
|
||||||
import static java.util.logging.Level.INFO;
|
import static java.util.logging.Level.INFO;
|
||||||
|
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
||||||
|
|
||||||
@MethodsNotNullByDefault
|
@MethodsNotNullByDefault
|
||||||
@ParametersNotNullByDefault
|
@ParametersNotNullByDefault
|
||||||
@@ -52,7 +57,8 @@ class AndroidNetworkManager implements NetworkManager, Service {
|
|||||||
|
|
||||||
private final ScheduledExecutorService scheduler;
|
private final ScheduledExecutorService scheduler;
|
||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
private final Context appContext;
|
private final Application app;
|
||||||
|
private final ConnectivityManager connectivityManager;
|
||||||
private final AtomicReference<Future<?>> connectivityCheck =
|
private final AtomicReference<Future<?>> connectivityCheck =
|
||||||
new AtomicReference<>();
|
new AtomicReference<>();
|
||||||
private final AtomicBoolean used = new AtomicBoolean(false);
|
private final AtomicBoolean used = new AtomicBoolean(false);
|
||||||
@@ -64,7 +70,9 @@ class AndroidNetworkManager implements NetworkManager, Service {
|
|||||||
EventBus eventBus, Application app) {
|
EventBus eventBus, Application app) {
|
||||||
this.scheduler = scheduler;
|
this.scheduler = scheduler;
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
this.appContext = app.getApplicationContext();
|
this.app = app;
|
||||||
|
connectivityManager = (ConnectivityManager)
|
||||||
|
requireNonNull(app.getSystemService(CONNECTIVITY_SERVICE));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -79,25 +87,34 @@ class AndroidNetworkManager implements NetworkManager, Service {
|
|||||||
filter.addAction(WIFI_AP_STATE_CHANGED_ACTION);
|
filter.addAction(WIFI_AP_STATE_CHANGED_ACTION);
|
||||||
filter.addAction(WIFI_P2P_THIS_DEVICE_CHANGED_ACTION);
|
filter.addAction(WIFI_P2P_THIS_DEVICE_CHANGED_ACTION);
|
||||||
if (SDK_INT >= 23) filter.addAction(ACTION_DEVICE_IDLE_MODE_CHANGED);
|
if (SDK_INT >= 23) filter.addAction(ACTION_DEVICE_IDLE_MODE_CHANGED);
|
||||||
appContext.registerReceiver(networkStateReceiver, filter);
|
app.registerReceiver(networkStateReceiver, filter);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void stopService() {
|
public void stopService() {
|
||||||
if (networkStateReceiver != null)
|
if (networkStateReceiver != null) {
|
||||||
appContext.unregisterReceiver(networkStateReceiver);
|
app.unregisterReceiver(networkStateReceiver);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public NetworkStatus getNetworkStatus() {
|
public NetworkStatus getNetworkStatus() {
|
||||||
ConnectivityManager cm = (ConnectivityManager)
|
if (SDK_INT >= 23) {
|
||||||
appContext.getSystemService(CONNECTIVITY_SERVICE);
|
Network net = connectivityManager.getActiveNetwork();
|
||||||
if (cm == null) throw new AssertionError();
|
if (net == null) return new NetworkStatus(false, false);
|
||||||
NetworkInfo net = cm.getActiveNetworkInfo();
|
NetworkCapabilities caps =
|
||||||
|
connectivityManager.getNetworkCapabilities(net);
|
||||||
|
if (caps == null) return new NetworkStatus(false, false);
|
||||||
|
boolean connected = caps.hasCapability(NET_CAPABILITY_INTERNET);
|
||||||
|
boolean wifi = caps.hasTransport(TRANSPORT_WIFI);
|
||||||
|
return new NetworkStatus(connected, wifi);
|
||||||
|
} else {
|
||||||
|
NetworkInfo net = connectivityManager.getActiveNetworkInfo();
|
||||||
boolean connected = net != null && net.isConnected();
|
boolean connected = net != null && net.isConnected();
|
||||||
boolean wifi = connected && net.getType() == TYPE_WIFI;
|
boolean wifi = connected && net.getType() == TYPE_WIFI;
|
||||||
return new NetworkStatus(connected, wifi);
|
return new NetworkStatus(connected, wifi);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private void updateConnectionStatus() {
|
private void updateConnectionStatus() {
|
||||||
eventBus.broadcast(new NetworkStatusEvent(getNetworkStatus()));
|
eventBus.broadcast(new NetworkStatusEvent(getNetworkStatus()));
|
||||||
|
|||||||
@@ -8,39 +8,30 @@ import android.content.BroadcastReceiver;
|
|||||||
import android.content.Context;
|
import android.content.Context;
|
||||||
import android.content.Intent;
|
import android.content.Intent;
|
||||||
import android.content.IntentFilter;
|
import android.content.IntentFilter;
|
||||||
import android.os.ParcelUuid;
|
|
||||||
import android.os.Parcelable;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.Pair;
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.PluginException;
|
import org.briarproject.bramble.api.plugin.PluginException;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
import org.briarproject.bramble.api.properties.TransportProperties;
|
|
||||||
import org.briarproject.bramble.api.system.AndroidExecutor;
|
import org.briarproject.bramble.api.system.AndroidExecutor;
|
||||||
import org.briarproject.bramble.api.system.Clock;
|
import org.briarproject.bramble.api.system.Clock;
|
||||||
import org.briarproject.bramble.util.AndroidUtils;
|
import org.briarproject.bramble.util.AndroidUtils;
|
||||||
import org.briarproject.bramble.util.IoUtils;
|
import org.briarproject.bramble.util.IoUtils;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.lang.reflect.InvocationTargetException;
|
|
||||||
import java.lang.reflect.Method;
|
|
||||||
import java.nio.ByteBuffer;
|
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.HashMap;
|
import java.util.Collection;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Set;
|
|
||||||
import java.util.TreeSet;
|
|
||||||
import java.util.UUID;
|
import java.util.UUID;
|
||||||
import java.util.concurrent.BlockingQueue;
|
import java.util.concurrent.BlockingQueue;
|
||||||
import java.util.concurrent.ExecutionException;
|
import java.util.concurrent.ExecutionException;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
import java.util.concurrent.LinkedBlockingQueue;
|
import java.util.concurrent.LinkedBlockingQueue;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
import java.util.logging.Logger;
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
@@ -57,25 +48,15 @@ import static android.bluetooth.BluetoothAdapter.SCAN_MODE_NONE;
|
|||||||
import static android.bluetooth.BluetoothAdapter.STATE_OFF;
|
import static android.bluetooth.BluetoothAdapter.STATE_OFF;
|
||||||
import static android.bluetooth.BluetoothAdapter.STATE_ON;
|
import static android.bluetooth.BluetoothAdapter.STATE_ON;
|
||||||
import static android.bluetooth.BluetoothDevice.ACTION_FOUND;
|
import static android.bluetooth.BluetoothDevice.ACTION_FOUND;
|
||||||
import static android.bluetooth.BluetoothDevice.ACTION_UUID;
|
|
||||||
import static android.bluetooth.BluetoothDevice.DEVICE_TYPE_LE;
|
import static android.bluetooth.BluetoothDevice.DEVICE_TYPE_LE;
|
||||||
import static android.bluetooth.BluetoothDevice.EXTRA_DEVICE;
|
import static android.bluetooth.BluetoothDevice.EXTRA_DEVICE;
|
||||||
import static android.bluetooth.BluetoothDevice.EXTRA_UUID;
|
|
||||||
import static android.os.Build.VERSION.SDK_INT;
|
import static android.os.Build.VERSION.SDK_INT;
|
||||||
import static java.nio.ByteOrder.LITTLE_ENDIAN;
|
|
||||||
import static java.util.Arrays.asList;
|
|
||||||
import static java.util.Collections.emptyList;
|
|
||||||
import static java.util.Collections.shuffle;
|
import static java.util.Collections.shuffle;
|
||||||
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
||||||
import static java.util.logging.Level.INFO;
|
import static java.util.logging.Level.INFO;
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
import static java.util.logging.Logger.getLogger;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
|
||||||
import static org.briarproject.bramble.api.plugin.BluetoothConstants.PROP_ADDRESS;
|
|
||||||
import static org.briarproject.bramble.api.plugin.BluetoothConstants.PROP_UUID;
|
|
||||||
import static org.briarproject.bramble.util.LogUtils.logException;
|
|
||||||
import static org.briarproject.bramble.util.PrivacyUtils.scrubMacAddress;
|
import static org.briarproject.bramble.util.PrivacyUtils.scrubMacAddress;
|
||||||
import static org.briarproject.bramble.util.StringUtils.isNullOrEmpty;
|
|
||||||
|
|
||||||
@MethodsNotNullByDefault
|
@MethodsNotNullByDefault
|
||||||
@ParametersNotNullByDefault
|
@ParametersNotNullByDefault
|
||||||
@@ -84,10 +65,9 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
private static final Logger LOG =
|
private static final Logger LOG =
|
||||||
getLogger(AndroidBluetoothPlugin.class.getName());
|
getLogger(AndroidBluetoothPlugin.class.getName());
|
||||||
|
|
||||||
private static final int MIN_DEVICE_DISCOVERY_MS = 2_000;
|
private static final int MAX_DISCOVERY_MS = 10_000;
|
||||||
private static final int MAX_DEVICE_DISCOVERY_MS = 30_000;
|
|
||||||
private static final int MAX_SERVICE_DISCOVERY_MS = 15_000;
|
|
||||||
|
|
||||||
|
private final ScheduledExecutorService scheduler;
|
||||||
private final AndroidExecutor androidExecutor;
|
private final AndroidExecutor androidExecutor;
|
||||||
private final Context appContext;
|
private final Context appContext;
|
||||||
private final Clock clock;
|
private final Clock clock;
|
||||||
@@ -99,11 +79,14 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
private volatile BluetoothAdapter adapter = null;
|
private volatile BluetoothAdapter adapter = null;
|
||||||
|
|
||||||
AndroidBluetoothPlugin(BluetoothConnectionLimiter connectionLimiter,
|
AndroidBluetoothPlugin(BluetoothConnectionLimiter connectionLimiter,
|
||||||
Executor ioExecutor, AndroidExecutor androidExecutor,
|
TimeoutMonitor timeoutMonitor, Executor ioExecutor,
|
||||||
Context appContext, SecureRandom secureRandom, Clock clock,
|
SecureRandom secureRandom, ScheduledExecutorService scheduler,
|
||||||
Backoff backoff, PluginCallback callback, int maxLatency) {
|
AndroidExecutor androidExecutor, Context appContext, Clock clock,
|
||||||
super(connectionLimiter, ioExecutor, secureRandom, backoff, callback,
|
Backoff backoff, PluginCallback callback, int maxLatency,
|
||||||
maxLatency);
|
int maxIdleTime) {
|
||||||
|
super(connectionLimiter, timeoutMonitor, ioExecutor, secureRandom,
|
||||||
|
backoff, callback, maxLatency, maxIdleTime);
|
||||||
|
this.scheduler = scheduler;
|
||||||
this.androidExecutor = androidExecutor;
|
this.androidExecutor = androidExecutor;
|
||||||
this.appContext = appContext;
|
this.appContext = appContext;
|
||||||
this.clock = clock;
|
this.clock = clock;
|
||||||
@@ -160,31 +143,12 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
@Override
|
@Override
|
||||||
void disableAdapterIfEnabledByUs() {
|
void disableAdapterIfEnabledByUs() {
|
||||||
if (isAdapterEnabled() && wasEnabledByUs) {
|
if (isAdapterEnabled() && wasEnabledByUs) {
|
||||||
cancelDiscoverability();
|
|
||||||
if (adapter.disable()) LOG.info("Disabling Bluetooth");
|
if (adapter.disable()) LOG.info("Disabling Bluetooth");
|
||||||
else LOG.info("Could not disable Bluetooth");
|
else LOG.info("Could not disable Bluetooth");
|
||||||
wasEnabledByUs = false;
|
wasEnabledByUs = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private void cancelDiscoverability() {
|
|
||||||
if (adapter.getScanMode() == SCAN_MODE_CONNECTABLE_DISCOVERABLE) {
|
|
||||||
try {
|
|
||||||
Method setDiscoverableTimeout = BluetoothAdapter.class
|
|
||||||
.getDeclaredMethod("setDiscoverableTimeout", int.class);
|
|
||||||
setDiscoverableTimeout.setAccessible(true);
|
|
||||||
setDiscoverableTimeout.invoke(adapter, 1);
|
|
||||||
LOG.info("Cancelled discoverability");
|
|
||||||
} catch (NoSuchMethodException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
} catch (IllegalAccessException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
} catch (InvocationTargetException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
void setEnabledByUs() {
|
void setEnabledByUs() {
|
||||||
wasEnabledByUs = true;
|
wasEnabledByUs = true;
|
||||||
@@ -220,9 +184,10 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
return wrapSocket(ss.accept());
|
return wrapSocket(ss.accept());
|
||||||
}
|
}
|
||||||
|
|
||||||
private DuplexTransportConnection wrapSocket(BluetoothSocket s) {
|
private DuplexTransportConnection wrapSocket(BluetoothSocket s)
|
||||||
return new AndroidBluetoothTransportConnection(this,
|
throws IOException {
|
||||||
connectionLimiter, s);
|
return new AndroidBluetoothTransportConnection(this, connectionLimiter,
|
||||||
|
timeoutMonitor, appContext, scheduler, s);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -250,8 +215,7 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
@Nullable
|
@Nullable
|
||||||
DuplexTransportConnection discoverAndConnect(String uuid) {
|
DuplexTransportConnection discoverAndConnect(String uuid) {
|
||||||
if (adapter == null) return null;
|
if (adapter == null) return null;
|
||||||
for (BluetoothDevice d : discoverDevices()) {
|
for (String address : discoverDevices()) {
|
||||||
String address = d.getAddress();
|
|
||||||
try {
|
try {
|
||||||
if (LOG.isLoggable(INFO))
|
if (LOG.isLoggable(INFO))
|
||||||
LOG.info("Connecting to " + scrubMacAddress(address));
|
LOG.info("Connecting to " + scrubMacAddress(address));
|
||||||
@@ -267,184 +231,10 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
private Collection<String> discoverDevices() {
|
||||||
public boolean supportsDiscovery() {
|
List<String> addresses = new ArrayList<>();
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void discoverPeers(
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> properties) {
|
|
||||||
// Discover all nearby devices
|
|
||||||
List<BluetoothDevice> devices = discoverDevices();
|
|
||||||
if (devices.isEmpty()) {
|
|
||||||
LOG.info("No devices discovered");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> discovered =
|
|
||||||
new ArrayList<>();
|
|
||||||
Map<String, Pair<TransportProperties, DiscoveryHandler>> byUuid =
|
|
||||||
new HashMap<>();
|
|
||||||
Map<String, Pair<TransportProperties, DiscoveryHandler>> byAddress =
|
|
||||||
new HashMap<>();
|
|
||||||
for (Pair<TransportProperties, DiscoveryHandler> pair : properties) {
|
|
||||||
TransportProperties p = pair.getFirst();
|
|
||||||
String uuid = p.get(PROP_UUID);
|
|
||||||
if (!isNullOrEmpty(uuid)) {
|
|
||||||
byUuid.put(uuid, pair);
|
|
||||||
String address = p.get(PROP_ADDRESS);
|
|
||||||
if (!isNullOrEmpty(address)) byAddress.put(address, pair);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
List<BluetoothDevice> unknown = new ArrayList<>(devices);
|
|
||||||
for (BluetoothDevice d : devices) {
|
|
||||||
Pair<TransportProperties, DiscoveryHandler> pair =
|
|
||||||
byAddress.remove(d.getAddress());
|
|
||||||
if (pair == null) {
|
|
||||||
// Try cached UUIDs
|
|
||||||
for (String uuid : getUuids(d)) {
|
|
||||||
pair = byUuid.remove(uuid);
|
|
||||||
if (pair != null) {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Matched "
|
|
||||||
+ scrubMacAddress(d.getAddress())
|
|
||||||
+ " by cached UUID");
|
|
||||||
}
|
|
||||||
TransportProperties p =
|
|
||||||
new TransportProperties(pair.getFirst());
|
|
||||||
p.put(PROP_ADDRESS, d.getAddress());
|
|
||||||
discovered.add(new Pair<>(p, pair.getSecond()));
|
|
||||||
unknown.remove(d);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Matched " + scrubMacAddress(d.getAddress())
|
|
||||||
+ " by address");
|
|
||||||
}
|
|
||||||
discovered.add(pair);
|
|
||||||
unknown.remove(d);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (unknown.isEmpty()) {
|
|
||||||
LOG.info("All discovered devices are known, not fetching UUIDs");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Fetch up-to-date UUIDs
|
|
||||||
if (LOG.isLoggable(INFO))
|
|
||||||
LOG.info("Fetching UUIDs for " + unknown.size() + " devices");
|
|
||||||
BlockingQueue<Intent> intents = new LinkedBlockingQueue<>();
|
BlockingQueue<Intent> intents = new LinkedBlockingQueue<>();
|
||||||
QueueingReceiver receiver = new QueueingReceiver(intents);
|
DiscoveryReceiver receiver = new DiscoveryReceiver(intents);
|
||||||
IntentFilter filter = new IntentFilter();
|
|
||||||
filter.addAction(ACTION_UUID);
|
|
||||||
appContext.registerReceiver(receiver, filter);
|
|
||||||
try {
|
|
||||||
List<BluetoothDevice> pending = new ArrayList<>();
|
|
||||||
for (BluetoothDevice d : unknown) {
|
|
||||||
if (d.fetchUuidsWithSdp()) {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Fetching UUIDs for "
|
|
||||||
+ scrubMacAddress(d.getAddress()));
|
|
||||||
}
|
|
||||||
pending.add(d);
|
|
||||||
} else {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Failed to fetch UUIDs for "
|
|
||||||
+ scrubMacAddress(d.getAddress()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
long now = clock.currentTimeMillis();
|
|
||||||
long end = now + MAX_SERVICE_DISCOVERY_MS;
|
|
||||||
while (now < end && !pending.isEmpty()) {
|
|
||||||
Intent i = intents.poll(end - now, MILLISECONDS);
|
|
||||||
if (i == null) break;
|
|
||||||
BluetoothDevice d = requireNonNull(
|
|
||||||
i.getParcelableExtra(EXTRA_DEVICE));
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Fetched UUIDs for "
|
|
||||||
+ scrubMacAddress(d.getAddress()));
|
|
||||||
}
|
|
||||||
Set<String> uuids = getUuids(d);
|
|
||||||
Parcelable[] extra = i.getParcelableArrayExtra(EXTRA_UUID);
|
|
||||||
if (extra != null) {
|
|
||||||
for (Parcelable p : extra) {
|
|
||||||
uuids.addAll(getUuidStrings((ParcelUuid) p));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (String uuid : uuids) {
|
|
||||||
Pair<TransportProperties, DiscoveryHandler> pair =
|
|
||||||
byUuid.remove(uuid);
|
|
||||||
if (pair != null) {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Matched "
|
|
||||||
+ scrubMacAddress(d.getAddress())
|
|
||||||
+ " by fetched UUID");
|
|
||||||
}
|
|
||||||
TransportProperties p =
|
|
||||||
new TransportProperties(pair.getFirst());
|
|
||||||
p.put(PROP_ADDRESS, d.getAddress());
|
|
||||||
discovered.add(new Pair<>(p, pair.getSecond()));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
pending.remove(d);
|
|
||||||
now = clock.currentTimeMillis();
|
|
||||||
}
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
if (pending.isEmpty()) {
|
|
||||||
LOG.info("Finished fetching UUIDs");
|
|
||||||
} else {
|
|
||||||
LOG.info("Failed to fetch UUIDs for " + pending.size()
|
|
||||||
+ " devices");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (InterruptedException e) {
|
|
||||||
LOG.info("Interrupted while fetching UUIDs");
|
|
||||||
Thread.currentThread().interrupt();
|
|
||||||
} finally {
|
|
||||||
appContext.unregisterReceiver(receiver);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
LOG.info("Discovered " + discovered.size() + " contacts");
|
|
||||||
}
|
|
||||||
for (Pair<TransportProperties, DiscoveryHandler> pair : discovered) {
|
|
||||||
pair.getSecond().handleDevice(pair.getFirst());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private Set<String> getUuids(BluetoothDevice d) {
|
|
||||||
Set<String> strings = new TreeSet<>();
|
|
||||||
ParcelUuid[] uuids = d.getUuids();
|
|
||||||
if (uuids == null) return strings;
|
|
||||||
for (ParcelUuid u : uuids) strings.addAll(getUuidStrings(u));
|
|
||||||
return strings;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Workaround for https://code.google.com/p/android/issues/detail?id=197341
|
|
||||||
private List<String> getUuidStrings(ParcelUuid u) {
|
|
||||||
UUID forwards = u.getUuid();
|
|
||||||
ByteBuffer buf = ByteBuffer.allocate(16);
|
|
||||||
buf.putLong(forwards.getLeastSignificantBits());
|
|
||||||
buf.putLong(forwards.getMostSignificantBits());
|
|
||||||
buf.rewind();
|
|
||||||
buf.order(LITTLE_ENDIAN);
|
|
||||||
UUID backwards = new UUID(buf.getLong(), buf.getLong());
|
|
||||||
return asList(forwards.toString(), backwards.toString());
|
|
||||||
}
|
|
||||||
|
|
||||||
private List<BluetoothDevice> discoverDevices() {
|
|
||||||
if (adapter.isDiscovering()) {
|
|
||||||
LOG.info("Already discovering");
|
|
||||||
return emptyList();
|
|
||||||
}
|
|
||||||
List<BluetoothDevice> devices = new ArrayList<>();
|
|
||||||
BlockingQueue<Intent> intents = new LinkedBlockingQueue<>();
|
|
||||||
QueueingReceiver receiver = new QueueingReceiver(intents);
|
|
||||||
IntentFilter filter = new IntentFilter();
|
IntentFilter filter = new IntentFilter();
|
||||||
filter.addAction(ACTION_DISCOVERY_STARTED);
|
filter.addAction(ACTION_DISCOVERY_STARTED);
|
||||||
filter.addAction(ACTION_DISCOVERY_FINISHED);
|
filter.addAction(ACTION_DISCOVERY_FINISHED);
|
||||||
@@ -452,9 +242,8 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
appContext.registerReceiver(receiver, filter);
|
appContext.registerReceiver(receiver, filter);
|
||||||
try {
|
try {
|
||||||
if (adapter.startDiscovery()) {
|
if (adapter.startDiscovery()) {
|
||||||
long start = clock.currentTimeMillis();
|
long now = clock.currentTimeMillis();
|
||||||
long end = start + MAX_DEVICE_DISCOVERY_MS;
|
long end = now + MAX_DISCOVERY_MS;
|
||||||
long now = start;
|
|
||||||
while (now < end) {
|
while (now < end) {
|
||||||
Intent i = intents.poll(end - now, MILLISECONDS);
|
Intent i = intents.poll(end - now, MILLISECONDS);
|
||||||
if (i == null) break;
|
if (i == null) break;
|
||||||
@@ -463,26 +252,17 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
LOG.info("Discovery started");
|
LOG.info("Discovery started");
|
||||||
} else if (ACTION_DISCOVERY_FINISHED.equals(action)) {
|
} else if (ACTION_DISCOVERY_FINISHED.equals(action)) {
|
||||||
LOG.info("Discovery finished");
|
LOG.info("Discovery finished");
|
||||||
now = clock.currentTimeMillis();
|
|
||||||
if (now - start < MIN_DEVICE_DISCOVERY_MS) {
|
|
||||||
LOG.info("Discovery finished quickly, retrying");
|
|
||||||
if (!adapter.startDiscovery()) {
|
|
||||||
LOG.info("Could not restart discovery");
|
|
||||||
break;
|
break;
|
||||||
}
|
|
||||||
} else {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
} else if (ACTION_FOUND.equals(action)) {
|
} else if (ACTION_FOUND.equals(action)) {
|
||||||
BluetoothDevice d = requireNonNull(
|
BluetoothDevice d = i.getParcelableExtra(EXTRA_DEVICE);
|
||||||
i.getParcelableExtra(EXTRA_DEVICE));
|
|
||||||
// Ignore Bluetooth LE devices
|
// Ignore Bluetooth LE devices
|
||||||
if (SDK_INT < 18 || d.getType() != DEVICE_TYPE_LE) {
|
if (SDK_INT < 18 || d.getType() != DEVICE_TYPE_LE) {
|
||||||
if (LOG.isLoggable(INFO)) {
|
String address = d.getAddress();
|
||||||
LOG.info("Discovered "
|
if (LOG.isLoggable(INFO))
|
||||||
+ scrubMacAddress(d.getAddress()));
|
LOG.info("Discovered " +
|
||||||
}
|
scrubMacAddress(address));
|
||||||
if (!devices.contains(d)) devices.add(d);
|
if (!addresses.contains(address))
|
||||||
|
addresses.add(address);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
now = clock.currentTimeMillis();
|
now = clock.currentTimeMillis();
|
||||||
@@ -498,9 +278,9 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
adapter.cancelDiscovery();
|
adapter.cancelDiscovery();
|
||||||
appContext.unregisterReceiver(receiver);
|
appContext.unregisterReceiver(receiver);
|
||||||
}
|
}
|
||||||
// Shuffle the devices so we don't always try the same one first
|
// Shuffle the addresses so we don't always try the same one first
|
||||||
shuffle(devices);
|
shuffle(addresses);
|
||||||
return devices;
|
return addresses;
|
||||||
}
|
}
|
||||||
|
|
||||||
private class BluetoothStateReceiver extends BroadcastReceiver {
|
private class BluetoothStateReceiver extends BroadcastReceiver {
|
||||||
@@ -521,11 +301,11 @@ class AndroidBluetoothPlugin extends BluetoothPlugin<BluetoothServerSocket> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static class QueueingReceiver extends BroadcastReceiver {
|
private static class DiscoveryReceiver extends BroadcastReceiver {
|
||||||
|
|
||||||
private final BlockingQueue<Intent> intents;
|
private final BlockingQueue<Intent> intents;
|
||||||
|
|
||||||
private QueueingReceiver(BlockingQueue<Intent> intents) {
|
private DiscoveryReceiver(BlockingQueue<Intent> intents) {
|
||||||
this.intents = intents;
|
this.intents = intents;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package org.briarproject.bramble.plugin.bluetooth;
|
|||||||
import android.content.Context;
|
import android.content.Context;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
||||||
@@ -15,6 +16,7 @@ import org.briarproject.bramble.api.system.Clock;
|
|||||||
|
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
|
||||||
import javax.annotation.concurrent.Immutable;
|
import javax.annotation.concurrent.Immutable;
|
||||||
|
|
||||||
@@ -25,28 +27,34 @@ import static org.briarproject.bramble.api.plugin.BluetoothConstants.ID;
|
|||||||
public class AndroidBluetoothPluginFactory implements DuplexPluginFactory {
|
public class AndroidBluetoothPluginFactory implements DuplexPluginFactory {
|
||||||
|
|
||||||
private static final int MAX_LATENCY = 30 * 1000; // 30 seconds
|
private static final int MAX_LATENCY = 30 * 1000; // 30 seconds
|
||||||
|
private static final int MAX_IDLE_TIME = 30 * 1000; // 30 seconds
|
||||||
private static final int MIN_POLLING_INTERVAL = 60 * 1000; // 1 minute
|
private static final int MIN_POLLING_INTERVAL = 60 * 1000; // 1 minute
|
||||||
private static final int MAX_POLLING_INTERVAL = 10 * 60 * 1000; // 10 mins
|
private static final int MAX_POLLING_INTERVAL = 10 * 60 * 1000; // 10 mins
|
||||||
private static final double BACKOFF_BASE = 1.2;
|
private static final double BACKOFF_BASE = 1.2;
|
||||||
|
|
||||||
private final Executor ioExecutor;
|
private final Executor ioExecutor;
|
||||||
|
private final ScheduledExecutorService scheduler;
|
||||||
private final AndroidExecutor androidExecutor;
|
private final AndroidExecutor androidExecutor;
|
||||||
private final Context appContext;
|
private final Context appContext;
|
||||||
private final SecureRandom secureRandom;
|
private final SecureRandom secureRandom;
|
||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
private final Clock clock;
|
private final Clock clock;
|
||||||
|
private final TimeoutMonitor timeoutMonitor;
|
||||||
private final BackoffFactory backoffFactory;
|
private final BackoffFactory backoffFactory;
|
||||||
|
|
||||||
public AndroidBluetoothPluginFactory(Executor ioExecutor,
|
public AndroidBluetoothPluginFactory(Executor ioExecutor,
|
||||||
|
ScheduledExecutorService scheduler,
|
||||||
AndroidExecutor androidExecutor, Context appContext,
|
AndroidExecutor androidExecutor, Context appContext,
|
||||||
SecureRandom secureRandom, EventBus eventBus, Clock clock,
|
SecureRandom secureRandom, EventBus eventBus, Clock clock,
|
||||||
BackoffFactory backoffFactory) {
|
TimeoutMonitor timeoutMonitor, BackoffFactory backoffFactory) {
|
||||||
this.ioExecutor = ioExecutor;
|
this.ioExecutor = ioExecutor;
|
||||||
|
this.scheduler = scheduler;
|
||||||
this.androidExecutor = androidExecutor;
|
this.androidExecutor = androidExecutor;
|
||||||
this.appContext = appContext;
|
this.appContext = appContext;
|
||||||
this.secureRandom = secureRandom;
|
this.secureRandom = secureRandom;
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
this.clock = clock;
|
this.clock = clock;
|
||||||
|
this.timeoutMonitor = timeoutMonitor;
|
||||||
this.backoffFactory = backoffFactory;
|
this.backoffFactory = backoffFactory;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -63,12 +71,13 @@ public class AndroidBluetoothPluginFactory implements DuplexPluginFactory {
|
|||||||
@Override
|
@Override
|
||||||
public DuplexPlugin createPlugin(PluginCallback callback) {
|
public DuplexPlugin createPlugin(PluginCallback callback) {
|
||||||
BluetoothConnectionLimiter connectionLimiter =
|
BluetoothConnectionLimiter connectionLimiter =
|
||||||
new BluetoothConnectionLimiterImpl();
|
new BluetoothConnectionLimiterImpl(eventBus);
|
||||||
Backoff backoff = backoffFactory.createBackoff(MIN_POLLING_INTERVAL,
|
Backoff backoff = backoffFactory.createBackoff(MIN_POLLING_INTERVAL,
|
||||||
MAX_POLLING_INTERVAL, BACKOFF_BASE);
|
MAX_POLLING_INTERVAL, BACKOFF_BASE);
|
||||||
AndroidBluetoothPlugin plugin = new AndroidBluetoothPlugin(
|
AndroidBluetoothPlugin plugin = new AndroidBluetoothPlugin(
|
||||||
connectionLimiter, ioExecutor, androidExecutor, appContext,
|
connectionLimiter, timeoutMonitor, ioExecutor, secureRandom,
|
||||||
secureRandom, clock, backoff, callback, MAX_LATENCY);
|
scheduler, androidExecutor, appContext, clock, backoff,
|
||||||
|
callback, MAX_LATENCY, MAX_IDLE_TIME);
|
||||||
eventBus.addListener(plugin);
|
eventBus.addListener(plugin);
|
||||||
return plugin;
|
return plugin;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,33 +1,60 @@
|
|||||||
package org.briarproject.bramble.plugin.bluetooth;
|
package org.briarproject.bramble.plugin.bluetooth;
|
||||||
|
|
||||||
import android.bluetooth.BluetoothSocket;
|
import android.bluetooth.BluetoothSocket;
|
||||||
|
import android.content.Context;
|
||||||
|
import android.os.PowerManager;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.AbstractDuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.AbstractDuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.util.RenewableWakeLock;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
|
||||||
|
import static android.content.Context.POWER_SERVICE;
|
||||||
|
import static android.os.PowerManager.PARTIAL_WAKE_LOCK;
|
||||||
|
import static java.util.concurrent.TimeUnit.MINUTES;
|
||||||
|
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
||||||
|
import static org.briarproject.bramble.api.plugin.BluetoothConstants.PROP_ADDRESS;
|
||||||
|
import static org.briarproject.bramble.util.AndroidUtils.getWakeLockTag;
|
||||||
|
import static org.briarproject.bramble.util.AndroidUtils.isValidBluetoothAddress;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
class AndroidBluetoothTransportConnection
|
class AndroidBluetoothTransportConnection
|
||||||
extends AbstractDuplexTransportConnection {
|
extends AbstractDuplexTransportConnection {
|
||||||
|
|
||||||
private final BluetoothConnectionLimiter connectionManager;
|
private final BluetoothConnectionLimiter connectionLimiter;
|
||||||
|
private final RenewableWakeLock wakeLock;
|
||||||
private final BluetoothSocket socket;
|
private final BluetoothSocket socket;
|
||||||
|
private final InputStream in;
|
||||||
|
|
||||||
AndroidBluetoothTransportConnection(Plugin plugin,
|
AndroidBluetoothTransportConnection(Plugin plugin,
|
||||||
BluetoothConnectionLimiter connectionManager,
|
BluetoothConnectionLimiter connectionLimiter,
|
||||||
BluetoothSocket socket) {
|
TimeoutMonitor timeoutMonitor, Context appContext,
|
||||||
|
ScheduledExecutorService scheduler, BluetoothSocket socket)
|
||||||
|
throws IOException {
|
||||||
super(plugin);
|
super(plugin);
|
||||||
this.connectionManager = connectionManager;
|
this.connectionLimiter = connectionLimiter;
|
||||||
this.socket = socket;
|
this.socket = socket;
|
||||||
|
in = timeoutMonitor.createTimeoutInputStream(
|
||||||
|
socket.getInputStream(), plugin.getMaxIdleTime() * 2);
|
||||||
|
PowerManager powerManager = (PowerManager)
|
||||||
|
requireNonNull(appContext.getSystemService(POWER_SERVICE));
|
||||||
|
String tag = getWakeLockTag(appContext);
|
||||||
|
wakeLock = new RenewableWakeLock(powerManager, scheduler,
|
||||||
|
PARTIAL_WAKE_LOCK, tag, 1, MINUTES);
|
||||||
|
wakeLock.acquire();
|
||||||
|
String address = socket.getRemoteDevice().getAddress();
|
||||||
|
if (isValidBluetoothAddress(address)) remote.put(PROP_ADDRESS, address);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected InputStream getInputStream() throws IOException {
|
protected InputStream getInputStream() {
|
||||||
return socket.getInputStream();
|
return in;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -40,7 +67,8 @@ class AndroidBluetoothTransportConnection
|
|||||||
try {
|
try {
|
||||||
socket.close();
|
socket.close();
|
||||||
} finally {
|
} finally {
|
||||||
connectionManager.connectionClosed(this);
|
wakeLock.release();
|
||||||
|
connectionLimiter.connectionClosed(this);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +1,17 @@
|
|||||||
package org.briarproject.bramble.plugin.tcp;
|
package org.briarproject.bramble.plugin.tcp;
|
||||||
|
|
||||||
|
import android.annotation.TargetApi;
|
||||||
import android.content.Context;
|
import android.content.Context;
|
||||||
import android.net.ConnectivityManager;
|
import android.net.ConnectivityManager;
|
||||||
|
import android.net.LinkAddress;
|
||||||
|
import android.net.LinkProperties;
|
||||||
import android.net.Network;
|
import android.net.Network;
|
||||||
import android.net.NetworkInfo;
|
import android.net.NetworkCapabilities;
|
||||||
import android.net.wifi.WifiInfo;
|
import android.net.wifi.WifiInfo;
|
||||||
import android.net.wifi.WifiManager;
|
import android.net.wifi.WifiManager;
|
||||||
|
|
||||||
import org.briarproject.bramble.PoliteExecutor;
|
import org.briarproject.bramble.PoliteExecutor;
|
||||||
|
import org.briarproject.bramble.api.Pair;
|
||||||
import org.briarproject.bramble.api.event.Event;
|
import org.briarproject.bramble.api.event.Event;
|
||||||
import org.briarproject.bramble.api.network.event.NetworkStatusEvent;
|
import org.briarproject.bramble.api.network.event.NetworkStatusEvent;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
@@ -17,7 +21,10 @@ import org.briarproject.bramble.api.settings.Settings;
|
|||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.InetAddress;
|
import java.net.InetAddress;
|
||||||
|
import java.net.InterfaceAddress;
|
||||||
|
import java.net.NetworkInterface;
|
||||||
import java.net.Socket;
|
import java.net.Socket;
|
||||||
|
import java.net.SocketException;
|
||||||
import java.net.UnknownHostException;
|
import java.net.UnknownHostException;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
@@ -28,15 +35,17 @@ import javax.net.SocketFactory;
|
|||||||
|
|
||||||
import static android.content.Context.CONNECTIVITY_SERVICE;
|
import static android.content.Context.CONNECTIVITY_SERVICE;
|
||||||
import static android.content.Context.WIFI_SERVICE;
|
import static android.content.Context.WIFI_SERVICE;
|
||||||
import static android.net.ConnectivityManager.TYPE_WIFI;
|
import static android.net.NetworkCapabilities.TRANSPORT_WIFI;
|
||||||
import static android.os.Build.VERSION.SDK_INT;
|
import static android.os.Build.VERSION.SDK_INT;
|
||||||
import static java.util.Collections.emptyList;
|
import static java.util.Collections.emptyList;
|
||||||
|
import static java.util.Collections.list;
|
||||||
import static java.util.Collections.singletonList;
|
import static java.util.Collections.singletonList;
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
import static java.util.logging.Logger.getLogger;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.api.plugin.Plugin.State.ACTIVE;
|
import static org.briarproject.bramble.api.plugin.Plugin.State.ACTIVE;
|
||||||
import static org.briarproject.bramble.api.plugin.Plugin.State.INACTIVE;
|
import static org.briarproject.bramble.api.plugin.Plugin.State.INACTIVE;
|
||||||
import static org.briarproject.bramble.util.IoUtils.tryToClose;
|
import static org.briarproject.bramble.util.IoUtils.tryToClose;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
class AndroidLanTcpPlugin extends LanTcpPlugin {
|
class AndroidLanTcpPlugin extends LanTcpPlugin {
|
||||||
@@ -83,21 +92,125 @@ class AndroidLanTcpPlugin extends LanTcpPlugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected List<InetAddress> getUsableLocalInetAddresses() {
|
protected List<InetAddress> getUsableLocalInetAddresses(boolean ipv4) {
|
||||||
// If the device doesn't have wifi, don't open any sockets
|
InetAddress addr = getWifiAddress(ipv4);
|
||||||
if (wifiManager == null) return emptyList();
|
return addr == null ? emptyList() : singletonList(addr);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nullable
|
||||||
|
private InetAddress getWifiAddress(boolean ipv4) {
|
||||||
|
Pair<InetAddress, Boolean> wifi = getWifiIpv4Address();
|
||||||
|
if (ipv4) return wifi == null ? null : wifi.getFirst();
|
||||||
|
// If there's no wifi IPv4 address, we might be a client on an
|
||||||
|
// IPv6-only wifi network. We can only detect this on API 21+
|
||||||
|
if (wifi == null) {
|
||||||
|
return SDK_INT >= 21 ? getWifiClientIpv6Address() : null;
|
||||||
|
}
|
||||||
|
// Use the wifi IPv4 address to determine which interface's IPv6
|
||||||
|
// address we should return (if the interface has a suitable address)
|
||||||
|
return getIpv6AddressForInterface(wifi.getFirst());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a {@link Pair} where the first element is the IPv4 address of
|
||||||
|
* the wifi interface and the second element is true if this device is
|
||||||
|
* providing an access point, or false if this device is a client. Returns
|
||||||
|
* null if this device isn't connected to wifi as an access point or client.
|
||||||
|
*/
|
||||||
|
@Nullable
|
||||||
|
private Pair<InetAddress, Boolean> getWifiIpv4Address() {
|
||||||
|
if (wifiManager == null) return null;
|
||||||
// If we're connected to a wifi network, return its address
|
// If we're connected to a wifi network, return its address
|
||||||
WifiInfo info = wifiManager.getConnectionInfo();
|
WifiInfo info = wifiManager.getConnectionInfo();
|
||||||
if (info != null && info.getIpAddress() != 0) {
|
if (info != null && info.getIpAddress() != 0) {
|
||||||
return singletonList(intToInetAddress(info.getIpAddress()));
|
return new Pair<>(intToInetAddress(info.getIpAddress()), false);
|
||||||
}
|
}
|
||||||
// If we're running an access point, return its address
|
List<InterfaceAddress> ifAddrs = getLocalInterfaceAddresses();
|
||||||
for (InetAddress addr : getLocalInetAddresses()) {
|
// If we're providing a normal access point, return its address
|
||||||
if (addr.equals(WIFI_AP_ADDRESS)) return singletonList(addr);
|
for (InterfaceAddress ifAddr : ifAddrs) {
|
||||||
if (addr.equals(WIFI_DIRECT_AP_ADDRESS)) return singletonList(addr);
|
if (isAndroidWifiApAddress(ifAddr)) {
|
||||||
|
return new Pair<>(ifAddr.getAddress(), true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// If we're providing a wifi direct access point, return its address
|
||||||
|
for (InterfaceAddress ifAddr : ifAddrs) {
|
||||||
|
if (isAndroidWifiDirectApAddress(ifAddr)) {
|
||||||
|
return new Pair<>(ifAddr.getAddress(), true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Not connected to wifi
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns true if the given address belongs to a network provided by an
|
||||||
|
* Android access point (including the access point's own address).
|
||||||
|
* <p>
|
||||||
|
* The access point's address is usually 192.168.43.1, but at least one
|
||||||
|
* device (Honor 8A) may use other addresses in the range 192.168.43.0/24.
|
||||||
|
*/
|
||||||
|
private boolean isAndroidWifiApAddress(InterfaceAddress ifAddr) {
|
||||||
|
if (ifAddr.getNetworkPrefixLength() != 24) return false;
|
||||||
|
byte[] ip = ifAddr.getAddress().getAddress();
|
||||||
|
return ip.length == 4
|
||||||
|
&& ip[0] == (byte) 192
|
||||||
|
&& ip[1] == (byte) 168
|
||||||
|
&& ip[2] == (byte) 43;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns true if the given address belongs to a network provided by an
|
||||||
|
* Android wifi direct legacy mode access point (including the access
|
||||||
|
* point's own address).
|
||||||
|
*/
|
||||||
|
private boolean isAndroidWifiDirectApAddress(InterfaceAddress ifAddr) {
|
||||||
|
if (ifAddr.getNetworkPrefixLength() != 24) return false;
|
||||||
|
byte[] ip = ifAddr.getAddress().getAddress();
|
||||||
|
return ip.length == 4
|
||||||
|
&& ip[0] == (byte) 192
|
||||||
|
&& ip[1] == (byte) 168
|
||||||
|
&& ip[2] == (byte) 49;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a link-local IPv6 address for the wifi client interface, or null
|
||||||
|
* if there's no such interface or it doesn't have a suitable address.
|
||||||
|
*/
|
||||||
|
@TargetApi(21)
|
||||||
|
@Nullable
|
||||||
|
private InetAddress getWifiClientIpv6Address() {
|
||||||
|
for (Network net : connectivityManager.getAllNetworks()) {
|
||||||
|
NetworkCapabilities caps =
|
||||||
|
connectivityManager.getNetworkCapabilities(net);
|
||||||
|
if (caps == null || !caps.hasTransport(TRANSPORT_WIFI)) continue;
|
||||||
|
LinkProperties props = connectivityManager.getLinkProperties(net);
|
||||||
|
if (props == null) continue;
|
||||||
|
for (LinkAddress linkAddress : props.getLinkAddresses()) {
|
||||||
|
InetAddress addr = linkAddress.getAddress();
|
||||||
|
if (isIpv6LinkLocalAddress(addr)) return addr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a link-local IPv6 address for the interface with the given IPv4
|
||||||
|
* address, or null if the interface doesn't have a suitable address.
|
||||||
|
*/
|
||||||
|
@Nullable
|
||||||
|
private InetAddress getIpv6AddressForInterface(InetAddress ipv4) {
|
||||||
|
try {
|
||||||
|
NetworkInterface iface = NetworkInterface.getByInetAddress(ipv4);
|
||||||
|
if (iface == null) return null;
|
||||||
|
for (InetAddress addr : list(iface.getInetAddresses())) {
|
||||||
|
if (isIpv6LinkLocalAddress(addr)) return addr;
|
||||||
|
}
|
||||||
|
// No suitable address
|
||||||
|
return null;
|
||||||
|
} catch (SocketException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
return null;
|
||||||
}
|
}
|
||||||
// No suitable addresses
|
|
||||||
return emptyList();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private InetAddress intToInetAddress(int ip) {
|
private InetAddress intToInetAddress(int ip) {
|
||||||
@@ -119,10 +232,12 @@ class AndroidLanTcpPlugin extends LanTcpPlugin {
|
|||||||
private SocketFactory getSocketFactory() {
|
private SocketFactory getSocketFactory() {
|
||||||
if (SDK_INT < 21) return SocketFactory.getDefault();
|
if (SDK_INT < 21) return SocketFactory.getDefault();
|
||||||
for (Network net : connectivityManager.getAllNetworks()) {
|
for (Network net : connectivityManager.getAllNetworks()) {
|
||||||
NetworkInfo info = connectivityManager.getNetworkInfo(net);
|
NetworkCapabilities caps =
|
||||||
if (info != null && info.getType() == TYPE_WIFI)
|
connectivityManager.getNetworkCapabilities(net);
|
||||||
|
if (caps != null && caps.hasTransport(TRANSPORT_WIFI)) {
|
||||||
return net.getSocketFactory();
|
return net.getSocketFactory();
|
||||||
}
|
}
|
||||||
|
}
|
||||||
LOG.warning("Could not find suitable socket factory");
|
LOG.warning("Could not find suitable socket factory");
|
||||||
return SocketFactory.getDefault();
|
return SocketFactory.getDefault();
|
||||||
}
|
}
|
||||||
@@ -137,9 +252,20 @@ class AndroidLanTcpPlugin extends LanTcpPlugin {
|
|||||||
connectionStatusExecutor.execute(() -> {
|
connectionStatusExecutor.execute(() -> {
|
||||||
State s = getState();
|
State s = getState();
|
||||||
if (s != ACTIVE && s != INACTIVE) return;
|
if (s != ACTIVE && s != INACTIVE) return;
|
||||||
List<InetAddress> addrs = getLocalInetAddresses();
|
Pair<InetAddress, Boolean> wifi = getPreferredWifiAddress();
|
||||||
if (addrs.contains(WIFI_AP_ADDRESS)
|
if (wifi == null) {
|
||||||
|| addrs.contains(WIFI_DIRECT_AP_ADDRESS)) {
|
LOG.info("Not connected to wifi");
|
||||||
|
socketFactory = SocketFactory.getDefault();
|
||||||
|
// Server sockets may not have been closed automatically when
|
||||||
|
// interface was taken down. If any sockets are open, closing
|
||||||
|
// them here will cause the sockets to be cleared and the state
|
||||||
|
// to be updated in acceptContactConnections()
|
||||||
|
if (s == ACTIVE) {
|
||||||
|
LOG.info("Closing server sockets");
|
||||||
|
tryToClose(state.getServerSocket(true), LOG, WARNING);
|
||||||
|
tryToClose(state.getServerSocket(false), LOG, WARNING);
|
||||||
|
}
|
||||||
|
} else if (wifi.getSecond()) {
|
||||||
LOG.info("Providing wifi hotspot");
|
LOG.info("Providing wifi hotspot");
|
||||||
// There's no corresponding Network object and thus no way
|
// There's no corresponding Network object and thus no way
|
||||||
// to get a suitable socket factory, so we won't be able to
|
// to get a suitable socket factory, so we won't be able to
|
||||||
@@ -147,16 +273,6 @@ class AndroidLanTcpPlugin extends LanTcpPlugin {
|
|||||||
// has internet access
|
// has internet access
|
||||||
socketFactory = SocketFactory.getDefault();
|
socketFactory = SocketFactory.getDefault();
|
||||||
if (s == INACTIVE) bind();
|
if (s == INACTIVE) bind();
|
||||||
} else if (addrs.isEmpty()) {
|
|
||||||
LOG.info("Not connected to wifi");
|
|
||||||
socketFactory = SocketFactory.getDefault();
|
|
||||||
// Server socket may not have been closed automatically when
|
|
||||||
// interface was taken down. Socket will be cleared and state
|
|
||||||
// updated in acceptContactConnections()
|
|
||||||
if (s == ACTIVE) {
|
|
||||||
LOG.info("Closing server socket");
|
|
||||||
tryToClose(state.getServerSocket(), LOG, WARNING);
|
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
LOG.info("Connected to wifi");
|
LOG.info("Connected to wifi");
|
||||||
socketFactory = getSocketFactory();
|
socketFactory = getSocketFactory();
|
||||||
@@ -164,4 +280,23 @@ class AndroidLanTcpPlugin extends LanTcpPlugin {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a {@link Pair} where the first element is an IP address (IPv4 if
|
||||||
|
* available, otherwise IPv6) of the wifi interface and the second element
|
||||||
|
* is true if this device is providing an access point, or false if this
|
||||||
|
* device is a client. Returns null if this device isn't connected to wifi
|
||||||
|
* as an access point or client.
|
||||||
|
*/
|
||||||
|
@Nullable
|
||||||
|
private Pair<InetAddress, Boolean> getPreferredWifiAddress() {
|
||||||
|
Pair<InetAddress, Boolean> wifi = getWifiIpv4Address();
|
||||||
|
// If there's no wifi IPv4 address, we might be a client on an
|
||||||
|
// IPv6-only wifi network. We can only detect this on API 21+
|
||||||
|
if (wifi == null && SDK_INT >= 21) {
|
||||||
|
InetAddress ipv6 = getWifiClientIpv6Address();
|
||||||
|
if (ipv6 != null) return new Pair<>(ipv6, false);
|
||||||
|
}
|
||||||
|
return wifi;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -27,6 +27,7 @@ import static android.content.Context.MODE_PRIVATE;
|
|||||||
import static android.content.Context.POWER_SERVICE;
|
import static android.content.Context.POWER_SERVICE;
|
||||||
import static android.os.PowerManager.PARTIAL_WAKE_LOCK;
|
import static android.os.PowerManager.PARTIAL_WAKE_LOCK;
|
||||||
import static java.util.concurrent.TimeUnit.MINUTES;
|
import static java.util.concurrent.TimeUnit.MINUTES;
|
||||||
|
import static org.briarproject.bramble.util.AndroidUtils.getWakeLockTag;
|
||||||
|
|
||||||
@MethodsNotNullByDefault
|
@MethodsNotNullByDefault
|
||||||
@ParametersNotNullByDefault
|
@ParametersNotNullByDefault
|
||||||
@@ -53,7 +54,7 @@ class AndroidTorPlugin extends TorPlugin {
|
|||||||
appContext.getSystemService(POWER_SERVICE);
|
appContext.getSystemService(POWER_SERVICE);
|
||||||
if (pm == null) throw new AssertionError();
|
if (pm == null) throw new AssertionError();
|
||||||
wakeLock = new RenewableWakeLock(pm, scheduler, PARTIAL_WAKE_LOCK,
|
wakeLock = new RenewableWakeLock(pm, scheduler, PARTIAL_WAKE_LOCK,
|
||||||
getWakeLockTag(), 1, MINUTES);
|
getWakeLockTag(appContext), 1, MINUTES);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -84,17 +85,4 @@ class AndroidTorPlugin extends TorPlugin {
|
|||||||
super.stop();
|
super.stop();
|
||||||
wakeLock.release();
|
wakeLock.release();
|
||||||
}
|
}
|
||||||
|
|
||||||
private String getWakeLockTag() {
|
|
||||||
PackageManager pm = appContext.getPackageManager();
|
|
||||||
for (PackageInfo info : pm.getInstalledPackages(0)) {
|
|
||||||
String name = info.packageName.toLowerCase();
|
|
||||||
if (name.startsWith("com.huawei.powergenie")) {
|
|
||||||
return "LocationManagerService";
|
|
||||||
} else if (name.startsWith("com.evenwell.powermonitor")) {
|
|
||||||
return "AudioIn";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return getClass().getSimpleName();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ package org.briarproject.bramble.util;
|
|||||||
import android.annotation.SuppressLint;
|
import android.annotation.SuppressLint;
|
||||||
import android.bluetooth.BluetoothAdapter;
|
import android.bluetooth.BluetoothAdapter;
|
||||||
import android.content.Context;
|
import android.content.Context;
|
||||||
|
import android.content.pm.PackageInfo;
|
||||||
|
import android.content.pm.PackageManager;
|
||||||
import android.os.Build;
|
import android.os.Build;
|
||||||
import android.provider.Settings;
|
import android.provider.Settings;
|
||||||
|
|
||||||
@@ -71,7 +73,7 @@ public class AndroidUtils {
|
|||||||
return new Pair<>("", "");
|
return new Pair<>("", "");
|
||||||
}
|
}
|
||||||
|
|
||||||
private static boolean isValidBluetoothAddress(@Nullable String address) {
|
public static boolean isValidBluetoothAddress(@Nullable String address) {
|
||||||
return !StringUtils.isNullOrEmpty(address)
|
return !StringUtils.isNullOrEmpty(address)
|
||||||
&& BluetoothAdapter.checkBluetoothAddress(address)
|
&& BluetoothAdapter.checkBluetoothAddress(address)
|
||||||
&& !address.equals(FAKE_BLUETOOTH_ADDRESS);
|
&& !address.equals(FAKE_BLUETOOTH_ADDRESS);
|
||||||
@@ -117,4 +119,17 @@ public class AndroidUtils {
|
|||||||
if (SDK_INT < 24) return new String[] {"image/jpeg", "image/png"};
|
if (SDK_INT < 24) return new String[] {"image/jpeg", "image/png"};
|
||||||
else return new String[] {"image/jpeg", "image/png", "image/gif"};
|
else return new String[] {"image/jpeg", "image/png", "image/gif"};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static String getWakeLockTag(Context ctx) {
|
||||||
|
PackageManager pm = ctx.getPackageManager();
|
||||||
|
for (PackageInfo info : pm.getInstalledPackages(0)) {
|
||||||
|
String name = info.packageName.toLowerCase();
|
||||||
|
if (name.startsWith("com.huawei.powergenie")) {
|
||||||
|
return "LocationManagerService";
|
||||||
|
} else if (name.startsWith("com.evenwell.powermonitor")) {
|
||||||
|
return "AudioIn";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ctx.getPackageName();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -70,7 +70,7 @@ dependencyVerification {
|
|||||||
'org.bouncycastle:bcpkix-jdk15on:1.56:bcpkix-jdk15on-1.56.jar:7043dee4e9e7175e93e0b36f45b1ec1ecb893c5f755667e8b916eb8dd201c6ca',
|
'org.bouncycastle:bcpkix-jdk15on:1.56:bcpkix-jdk15on-1.56.jar:7043dee4e9e7175e93e0b36f45b1ec1ecb893c5f755667e8b916eb8dd201c6ca',
|
||||||
'org.bouncycastle:bcprov-jdk15on:1.56:bcprov-jdk15on-1.56.jar:963e1ee14f808ffb99897d848ddcdb28fa91ddda867eb18d303e82728f878349',
|
'org.bouncycastle:bcprov-jdk15on:1.56:bcprov-jdk15on-1.56.jar:963e1ee14f808ffb99897d848ddcdb28fa91ddda867eb18d303e82728f878349',
|
||||||
'org.briarproject:obfs4proxy-android:0.0.11-2:obfs4proxy-android-0.0.11-2.zip:57e55cbe87aa2aac210fdbb6cd8cdeafe15f825406a08ebf77a8b787aa2c6a8a',
|
'org.briarproject:obfs4proxy-android:0.0.11-2:obfs4proxy-android-0.0.11-2.zip:57e55cbe87aa2aac210fdbb6cd8cdeafe15f825406a08ebf77a8b787aa2c6a8a',
|
||||||
'org.briarproject:tor-android:0.3.5.9:tor-android-0.3.5.9.zip:853b0440feccd6904bd03e6b2de53a62ebcde1d58068beeadc447a7dff950bc8',
|
'org.briarproject:tor-android:0.3.5.10:tor-android-0.3.5.10.zip:edd83bf557fcff2105eaa0bdb3f607a6852ebe7360920929ae3039dd5f4774c5',
|
||||||
'org.checkerframework:checker-compat-qual:2.5.3:checker-compat-qual-2.5.3.jar:d76b9afea61c7c082908023f0cbc1427fab9abd2df915c8b8a3e7a509bccbc6d',
|
'org.checkerframework:checker-compat-qual:2.5.3:checker-compat-qual-2.5.3.jar:d76b9afea61c7c082908023f0cbc1427fab9abd2df915c8b8a3e7a509bccbc6d',
|
||||||
'org.checkerframework:checker-qual:2.5.2:checker-qual-2.5.2.jar:64b02691c8b9d4e7700f8ee2e742dce7ea2c6e81e662b7522c9ee3bf568c040a',
|
'org.checkerframework:checker-qual:2.5.2:checker-qual-2.5.2.jar:64b02691c8b9d4e7700f8ee2e742dce7ea2c6e81e662b7522c9ee3bf568c040a',
|
||||||
'org.codehaus.groovy:groovy-all:2.4.15:groovy-all-2.4.15.jar:51d6c4e71782e85674239189499854359d380fb75e1a703756e3aaa5b98a5af0',
|
'org.codehaus.groovy:groovy-all:2.4.15:groovy-all-2.4.15.jar:51d6c4e71782e85674239189499854359d380fb75e1a703756e3aaa5b98a5af0',
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
package org.briarproject.bramble.api.account;
|
package org.briarproject.bramble.api.account;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.crypto.DecryptionException;
|
||||||
import org.briarproject.bramble.api.crypto.SecretKey;
|
import org.briarproject.bramble.api.crypto.SecretKey;
|
||||||
import org.briarproject.bramble.api.identity.IdentityManager;
|
import org.briarproject.bramble.api.identity.IdentityManager;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
@@ -13,7 +14,8 @@ public interface AccountManager {
|
|||||||
* Returns true if the manager has the database key. This will be false
|
* Returns true if the manager has the database key. This will be false
|
||||||
* before {@link #createAccount(String, String)} or {@link #signIn(String)}
|
* before {@link #createAccount(String, String)} or {@link #signIn(String)}
|
||||||
* has been called, and true after {@link #createAccount(String, String)}
|
* has been called, and true after {@link #createAccount(String, String)}
|
||||||
* or {@link #signIn(String)} has returned true, until the process exits.
|
* or {@link #signIn(String)} has returned true, until
|
||||||
|
* {@link #deleteAccount()} is called or the process exits.
|
||||||
*/
|
*/
|
||||||
boolean hasDatabaseKey();
|
boolean hasDatabaseKey();
|
||||||
|
|
||||||
@@ -22,25 +24,22 @@ public interface AccountManager {
|
|||||||
* before {@link #createAccount(String, String)} or {@link #signIn(String)}
|
* before {@link #createAccount(String, String)} or {@link #signIn(String)}
|
||||||
* has been called, and non-null after
|
* has been called, and non-null after
|
||||||
* {@link #createAccount(String, String)} or {@link #signIn(String)} has
|
* {@link #createAccount(String, String)} or {@link #signIn(String)} has
|
||||||
* returned true, until the process exits.
|
* returned true, until {@link #deleteAccount()} is called or the process
|
||||||
|
* exits.
|
||||||
*/
|
*/
|
||||||
@Nullable
|
@Nullable
|
||||||
SecretKey getDatabaseKey();
|
SecretKey getDatabaseKey();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns true if the encrypted database key can be loaded from disk, and
|
* Returns true if the encrypted database key can be loaded from disk.
|
||||||
* the database directory exists and is a directory.
|
|
||||||
*/
|
*/
|
||||||
boolean accountExists();
|
boolean accountExists();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Creates an identity with the given name and registers it with the
|
* Creates an identity with the given name and registers it with the
|
||||||
* {@link IdentityManager}. Creates a database key, encrypts it with the
|
* {@link IdentityManager}. Creates a database key, encrypts it with the
|
||||||
* given password and stores it on disk.
|
* given password and stores it on disk. {@link #accountExists()} will
|
||||||
* <p/>
|
* return true after this method returns true.
|
||||||
* This method does not create the database directory, so
|
|
||||||
* {@link #accountExists()} will continue to return false until the
|
|
||||||
* database directory is created.
|
|
||||||
*/
|
*/
|
||||||
boolean createAccount(String name, String password);
|
boolean createAccount(String name, String password);
|
||||||
|
|
||||||
@@ -54,17 +53,19 @@ public interface AccountManager {
|
|||||||
* Loads the encrypted database key from disk and decrypts it with the
|
* Loads the encrypted database key from disk and decrypts it with the
|
||||||
* given password.
|
* given password.
|
||||||
*
|
*
|
||||||
* @return true if the database key was successfully loaded and decrypted.
|
* @throws DecryptionException If the database key could not be loaded and
|
||||||
|
* decrypted.
|
||||||
*/
|
*/
|
||||||
boolean signIn(String password);
|
void signIn(String password) throws DecryptionException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Loads the encrypted database key from disk, decrypts it with the old
|
* Loads the encrypted database key from disk, decrypts it with the old
|
||||||
* password, encrypts it with the new password, and stores it on disk,
|
* password, encrypts it with the new password, and stores it on disk,
|
||||||
* replacing the old key.
|
* replacing the old key.
|
||||||
*
|
*
|
||||||
* @return true if the database key was successfully loaded, re-encrypted
|
* @throws DecryptionException If the database key could not be loaded and
|
||||||
* and stored.
|
* decrypted.
|
||||||
*/
|
*/
|
||||||
boolean changePassword(String oldPassword, String newPassword);
|
void changePassword(String oldPassword, String newPassword)
|
||||||
|
throws DecryptionException;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,8 +1,11 @@
|
|||||||
package org.briarproject.bramble.api.plugin;
|
package org.briarproject.bramble.api.connection;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
@@ -0,0 +1,130 @@
|
|||||||
|
package org.briarproject.bramble.api.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
|
||||||
|
import java.util.Collection;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Keeps track of which contacts are currently connected by which transports.
|
||||||
|
*/
|
||||||
|
@NotNullByDefault
|
||||||
|
public interface ConnectionRegistry {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Registers an incoming connection from the given contact over the given
|
||||||
|
* transport. The connection's {@link Priority priority} can be set later
|
||||||
|
* via {@link #setPriority(ContactId, TransportId, InterruptibleConnection,
|
||||||
|
* Priority)} if a priority record is received from the contact.
|
||||||
|
* <p>
|
||||||
|
* Broadcasts {@link ConnectionOpenedEvent}. Also broadcasts
|
||||||
|
* {@link ContactConnectedEvent} if this is the only connection with the
|
||||||
|
* contact.
|
||||||
|
*/
|
||||||
|
void registerIncomingConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Registers an outgoing connection to the given contact over the given
|
||||||
|
* transport.
|
||||||
|
* <p>
|
||||||
|
* Broadcasts {@link ConnectionOpenedEvent}. Also broadcasts
|
||||||
|
* {@link ContactConnectedEvent} if this is the only connection with the
|
||||||
|
* contact.
|
||||||
|
* <p>
|
||||||
|
* If the registry has any "better" connections with the given contact, the
|
||||||
|
* given connection will be interrupted. If the registry has any "worse"
|
||||||
|
* connections with the given contact, those connections will be
|
||||||
|
* interrupted.
|
||||||
|
* <p>
|
||||||
|
* Connection A is considered "better" than connection B if both
|
||||||
|
* connections have had their priorities set, and either A's transport is
|
||||||
|
* {@link PluginConfig#getTransportPreferences() preferred} to B's, or
|
||||||
|
* they use the same transport and A has higher {@link Priority priority}
|
||||||
|
* than B.
|
||||||
|
* <p>
|
||||||
|
* For backward compatibility, connections without priorities are not
|
||||||
|
* considered better or worse than other connections.
|
||||||
|
*/
|
||||||
|
void registerOutgoingConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, Priority priority);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unregisters a connection with the given contact over the given transport.
|
||||||
|
* <p>
|
||||||
|
* Broadcasts {@link ConnectionClosedEvent}. Also broadcasts
|
||||||
|
* {@link ContactDisconnectedEvent} if this is the only connection with
|
||||||
|
* the contact.
|
||||||
|
*/
|
||||||
|
void unregisterConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, boolean incoming, boolean exception);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets the {@link Priority priority} of a connection that was previously
|
||||||
|
* registered via {@link #registerIncomingConnection(ContactId, TransportId,
|
||||||
|
* InterruptibleConnection)}.
|
||||||
|
* <p>
|
||||||
|
* If the registry has any "better" connections with the given contact, the
|
||||||
|
* given connection will be interrupted. If the registry has any "worse"
|
||||||
|
* connections with the given contact, those connections will be
|
||||||
|
* interrupted.
|
||||||
|
* <p>
|
||||||
|
* Connection A is considered "better" than connection B if both
|
||||||
|
* connections have had their priorities set, and either A's transport is
|
||||||
|
* {@link PluginConfig#getTransportPreferences() preferred} to B's, or
|
||||||
|
* they use the same transport and A has higher {@link Priority priority}
|
||||||
|
* than B.
|
||||||
|
* <p>
|
||||||
|
* For backward compatibility, connections without priorities are not
|
||||||
|
* considered better or worse than other connections.
|
||||||
|
*/
|
||||||
|
void setPriority(ContactId c, TransportId t, InterruptibleConnection conn,
|
||||||
|
Priority priority);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns any contacts that are connected via the given transport.
|
||||||
|
*/
|
||||||
|
Collection<ContactId> getConnectedContacts(TransportId t);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns any contacts that are connected via the given transport or any
|
||||||
|
* {@link PluginConfig#getTransportPreferences() better} transport.
|
||||||
|
*/
|
||||||
|
Collection<ContactId> getConnectedOrBetterContacts(TransportId t);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns true if the given contact is connected via the given transport.
|
||||||
|
*/
|
||||||
|
boolean isConnected(ContactId c, TransportId t);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns true if the given contact is connected via any transport.
|
||||||
|
*/
|
||||||
|
boolean isConnected(ContactId c);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Registers a connection with the given pending contact. Broadcasts
|
||||||
|
* {@link RendezvousConnectionOpenedEvent} if this is the only connection
|
||||||
|
* with the pending contact.
|
||||||
|
*
|
||||||
|
* @return True if this is the only connection with the pending contact,
|
||||||
|
* false if it is redundant and should be closed
|
||||||
|
*/
|
||||||
|
boolean registerConnection(PendingContactId p);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unregisters a connection with the given pending contact. Broadcasts
|
||||||
|
* {@link RendezvousConnectionClosedEvent}.
|
||||||
|
*/
|
||||||
|
void unregisterConnection(PendingContactId p, boolean success);
|
||||||
|
}
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
package org.briarproject.bramble.api.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A duplex sync connection that can be closed by interrupting its outgoing
|
||||||
|
* sync session.
|
||||||
|
*/
|
||||||
|
@NotNullByDefault
|
||||||
|
public interface InterruptibleConnection {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Interrupts the connection's outgoing sync session. If the underlying
|
||||||
|
* transport connection is alive and the remote peer is cooperative, this
|
||||||
|
* should result in both sync sessions ending and the connection being
|
||||||
|
* cleanly closed.
|
||||||
|
*/
|
||||||
|
void interruptOutgoingSession();
|
||||||
|
}
|
||||||
@@ -142,16 +142,17 @@ public interface CryptoComponent {
|
|||||||
/**
|
/**
|
||||||
* Decrypts and authenticates the given ciphertext that has been read from
|
* Decrypts and authenticates the given ciphertext that has been read from
|
||||||
* storage. The encryption and authentication keys are derived from the
|
* storage. The encryption and authentication keys are derived from the
|
||||||
* given password. Returns null if the ciphertext cannot be decrypted and
|
* given password.
|
||||||
* authenticated (for example, if the password is wrong).
|
|
||||||
*
|
*
|
||||||
* @param keyStrengthener Used to strengthen the password-based key. If
|
* @param keyStrengthener Used to strengthen the password-based key. If
|
||||||
* null, or if strengthening was not used when encrypting the ciphertext,
|
* null, or if strengthening was not used when encrypting the ciphertext,
|
||||||
* the password-based key will not be strengthened
|
* the password-based key will not be strengthened
|
||||||
|
* @throws DecryptionException If the ciphertext cannot be decrypted and
|
||||||
|
* authenticated (for example, if the password is wrong).
|
||||||
*/
|
*/
|
||||||
@Nullable
|
|
||||||
byte[] decryptWithPassword(byte[] ciphertext, String password,
|
byte[] decryptWithPassword(byte[] ciphertext, String password,
|
||||||
@Nullable KeyStrengthener keyStrengthener);
|
@Nullable KeyStrengthener keyStrengthener)
|
||||||
|
throws DecryptionException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns true if the given ciphertext was encrypted using a strengthened
|
* Returns true if the given ciphertext was encrypted using a strengthened
|
||||||
|
|||||||
@@ -0,0 +1,17 @@
|
|||||||
|
package org.briarproject.bramble.api.crypto;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
public class DecryptionException extends Exception {
|
||||||
|
|
||||||
|
private final DecryptionResult result;
|
||||||
|
|
||||||
|
public DecryptionException(DecryptionResult result) {
|
||||||
|
this.result = result;
|
||||||
|
}
|
||||||
|
|
||||||
|
public DecryptionResult getDecryptionResult() {
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
package org.briarproject.bramble.api.crypto;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The result of a password-based decryption operation.
|
||||||
|
*/
|
||||||
|
public enum DecryptionResult {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decryption succeeded.
|
||||||
|
*/
|
||||||
|
SUCCESS,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decryption failed because the format of the ciphertext was invalid.
|
||||||
|
*/
|
||||||
|
INVALID_CIPHERTEXT,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decryption failed because the {@link KeyStrengthener} used for
|
||||||
|
* encryption was not available for decryption.
|
||||||
|
*/
|
||||||
|
KEY_STRENGTHENER_ERROR,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decryption failed because the password used for decryption did not match
|
||||||
|
* the password used for encryption.
|
||||||
|
*/
|
||||||
|
INVALID_PASSWORD
|
||||||
|
}
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
package org.briarproject.bramble.api.io;
|
||||||
|
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
public interface TimeoutMonitor {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns an {@link InputStream} that wraps the given stream and allows
|
||||||
|
* read timeouts to be detected.
|
||||||
|
*
|
||||||
|
* @param timeoutMs The read timeout in milliseconds. Timeouts will be
|
||||||
|
* detected eventually but are not guaranteed to be detected immediately.
|
||||||
|
*/
|
||||||
|
InputStream createTimeoutInputStream(InputStream in, long timeoutMs);
|
||||||
|
}
|
||||||
@@ -1,67 +0,0 @@
|
|||||||
package org.briarproject.bramble.api.plugin;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
|
||||||
|
|
||||||
import java.util.Collection;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Keeps track of which contacts are currently connected by which transports.
|
|
||||||
*/
|
|
||||||
@NotNullByDefault
|
|
||||||
public interface ConnectionRegistry {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Registers a connection with the given contact over the given transport.
|
|
||||||
* Broadcasts {@link ConnectionOpenedEvent}. Also broadcasts
|
|
||||||
* {@link ContactConnectedEvent} if this is the only connection with the
|
|
||||||
* contact.
|
|
||||||
*/
|
|
||||||
void registerConnection(ContactId c, TransportId t, boolean incoming);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Unregisters a connection with the given contact over the given transport.
|
|
||||||
* Broadcasts {@link ConnectionClosedEvent}. Also broadcasts
|
|
||||||
* {@link ContactDisconnectedEvent} if this is the only connection with
|
|
||||||
* the contact.
|
|
||||||
*/
|
|
||||||
void unregisterConnection(ContactId c, TransportId t, boolean incoming);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns any contacts that are connected via the given transport.
|
|
||||||
*/
|
|
||||||
Collection<ContactId> getConnectedContacts(TransportId t);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns true if the given contact is connected via the given transport.
|
|
||||||
*/
|
|
||||||
boolean isConnected(ContactId c, TransportId t);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns true if the given contact is connected via any transport.
|
|
||||||
*/
|
|
||||||
boolean isConnected(ContactId c);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Registers a connection with the given pending contact. Broadcasts
|
|
||||||
* {@link RendezvousConnectionOpenedEvent} if this is the only connection
|
|
||||||
* with the pending contact.
|
|
||||||
*
|
|
||||||
* @return True if this is the only connection with the pending contact,
|
|
||||||
* false if it is redundant and should be closed
|
|
||||||
*/
|
|
||||||
boolean registerConnection(PendingContactId p);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Unregisters a connection with the given pending contact. Broadcasts
|
|
||||||
* {@link RendezvousConnectionClosedEvent}.
|
|
||||||
*/
|
|
||||||
void unregisterConnection(PendingContactId p, boolean success);
|
|
||||||
}
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
package org.briarproject.bramble.api.plugin;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|
||||||
import org.briarproject.bramble.api.properties.TransportProperties;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* An interface for handling peers discovered by transport plugins.
|
|
||||||
*/
|
|
||||||
@NotNullByDefault
|
|
||||||
public interface DiscoveryHandler {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Handles a peer discovered by a transport plugin.
|
|
||||||
*
|
|
||||||
* @param p A set of properties describing the discovered peer.
|
|
||||||
*/
|
|
||||||
void handleDevice(TransportProperties p);
|
|
||||||
}
|
|
||||||
@@ -7,7 +7,9 @@ public interface LanTcpConstants {
|
|||||||
// Transport properties (shared with contacts)
|
// Transport properties (shared with contacts)
|
||||||
String PROP_IP_PORTS = "ipPorts";
|
String PROP_IP_PORTS = "ipPorts";
|
||||||
String PROP_PORT = "port";
|
String PROP_PORT = "port";
|
||||||
|
String PROP_IPV6 = "ipv6";
|
||||||
|
|
||||||
// A local setting
|
// Local settings (not shared with contacts)
|
||||||
String PREF_LAN_IP_PORTS = "ipPorts";
|
String PREF_LAN_IP_PORTS = "ipPorts";
|
||||||
|
String PREF_IPV6 = "ipv6";
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.properties.TransportProperties;
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
import org.briarproject.bramble.api.settings.SettingsManager;
|
import org.briarproject.bramble.api.settings.SettingsManager;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.Collection;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
public interface Plugin {
|
public interface Plugin {
|
||||||
@@ -106,5 +106,6 @@ public interface Plugin {
|
|||||||
* Attempts to create connections using the given transport properties,
|
* Attempts to create connections using the given transport properties,
|
||||||
* passing any created connections to the corresponding handlers.
|
* passing any created connections to the corresponding handlers.
|
||||||
*/
|
*/
|
||||||
void poll(List<Pair<TransportProperties, ConnectionHandler>> properties);
|
void poll(Collection<Pair<TransportProperties, ConnectionHandler>>
|
||||||
|
properties);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,6 +5,8 @@ import org.briarproject.bramble.api.plugin.duplex.DuplexPluginFactory;
|
|||||||
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
||||||
|
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
public interface PluginConfig {
|
public interface PluginConfig {
|
||||||
@@ -14,4 +16,11 @@ public interface PluginConfig {
|
|||||||
Collection<SimplexPluginFactory> getSimplexFactories();
|
Collection<SimplexPluginFactory> getSimplexFactories();
|
||||||
|
|
||||||
boolean shouldPoll();
|
boolean shouldPoll();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a map representing transport preferences. For each entry in the
|
||||||
|
* map, connections via the transports identified by the value are
|
||||||
|
* preferred to connections via the transport identified by the key.
|
||||||
|
*/
|
||||||
|
Map<TransportId, List<TransportId>> getTransportPreferences();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
@@ -14,6 +15,8 @@ import java.util.concurrent.atomic.AtomicBoolean;
|
|||||||
public abstract class AbstractDuplexTransportConnection
|
public abstract class AbstractDuplexTransportConnection
|
||||||
implements DuplexTransportConnection {
|
implements DuplexTransportConnection {
|
||||||
|
|
||||||
|
protected final TransportProperties remote = new TransportProperties();
|
||||||
|
|
||||||
private final Plugin plugin;
|
private final Plugin plugin;
|
||||||
private final Reader reader;
|
private final Reader reader;
|
||||||
private final Writer writer;
|
private final Writer writer;
|
||||||
@@ -44,6 +47,11 @@ public abstract class AbstractDuplexTransportConnection
|
|||||||
return writer;
|
return writer;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public TransportProperties getRemoteProperties() {
|
||||||
|
return remote;
|
||||||
|
}
|
||||||
|
|
||||||
private class Reader implements TransportConnectionReader {
|
private class Reader implements TransportConnectionReader {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|||||||
@@ -1,18 +1,14 @@
|
|||||||
package org.briarproject.bramble.api.plugin.duplex;
|
package org.briarproject.bramble.api.plugin.duplex;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.Pair;
|
|
||||||
import org.briarproject.bramble.api.data.BdfList;
|
import org.briarproject.bramble.api.data.BdfList;
|
||||||
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.properties.TransportProperties;
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
import org.briarproject.bramble.api.rendezvous.KeyMaterialSource;
|
import org.briarproject.bramble.api.rendezvous.KeyMaterialSource;
|
||||||
import org.briarproject.bramble.api.rendezvous.RendezvousEndpoint;
|
import org.briarproject.bramble.api.rendezvous.RendezvousEndpoint;
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -62,16 +58,4 @@ public interface DuplexPlugin extends Plugin {
|
|||||||
@Nullable
|
@Nullable
|
||||||
RendezvousEndpoint createRendezvousEndpoint(KeyMaterialSource k,
|
RendezvousEndpoint createRendezvousEndpoint(KeyMaterialSource k,
|
||||||
boolean alice, ConnectionHandler incoming);
|
boolean alice, ConnectionHandler incoming);
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns true if the plugin supports peer discovery.
|
|
||||||
*/
|
|
||||||
boolean supportsDiscovery();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Attempts to discover peers using the given transport properties, passing
|
|
||||||
* any discovered peers to the corresponding handlers.
|
|
||||||
*/
|
|
||||||
void discoverPeers(List<Pair<TransportProperties, DiscoveryHandler>>
|
|
||||||
properties);
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ package org.briarproject.bramble.api.plugin.duplex;
|
|||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* An interface for reading and writing data over a duplex transport. The
|
* An interface for reading and writing data over a duplex transport. The
|
||||||
@@ -23,4 +24,10 @@ public interface DuplexTransportConnection {
|
|||||||
* for writing to the connection.
|
* for writing to the connection.
|
||||||
*/
|
*/
|
||||||
TransportConnectionWriter getWriter();
|
TransportConnectionWriter getWriter();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a possibly empty set of {@link TransportProperties} describing
|
||||||
|
* the remote peer.
|
||||||
|
*/
|
||||||
|
TransportProperties getRemoteProperties();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,13 +13,14 @@ public class ConnectionClosedEvent extends Event {
|
|||||||
|
|
||||||
private final ContactId contactId;
|
private final ContactId contactId;
|
||||||
private final TransportId transportId;
|
private final TransportId transportId;
|
||||||
private final boolean incoming;
|
private final boolean incoming, exception;
|
||||||
|
|
||||||
public ConnectionClosedEvent(ContactId contactId, TransportId transportId,
|
public ConnectionClosedEvent(ContactId contactId, TransportId transportId,
|
||||||
boolean incoming) {
|
boolean incoming, boolean exception) {
|
||||||
this.contactId = contactId;
|
this.contactId = contactId;
|
||||||
this.transportId = transportId;
|
this.transportId = transportId;
|
||||||
this.incoming = incoming;
|
this.incoming = incoming;
|
||||||
|
this.exception = exception;
|
||||||
}
|
}
|
||||||
|
|
||||||
public ContactId getContactId() {
|
public ContactId getContactId() {
|
||||||
@@ -33,4 +34,8 @@ public class ConnectionClosedEvent extends Event {
|
|||||||
public boolean isIncoming() {
|
public boolean isIncoming() {
|
||||||
return incoming;
|
return incoming;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public boolean isException() {
|
||||||
|
return exception;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,4 +11,28 @@ public interface TransportPropertyConstants {
|
|||||||
* The maximum length of a property's key or value in UTF-8 bytes.
|
* The maximum length of a property's key or value in UTF-8 bytes.
|
||||||
*/
|
*/
|
||||||
int MAX_PROPERTY_LENGTH = 100;
|
int MAX_PROPERTY_LENGTH = 100;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Message metadata key for the transport ID of a local or remote update,
|
||||||
|
* as a BDF string.
|
||||||
|
*/
|
||||||
|
String MSG_KEY_TRANSPORT_ID = "transportId";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Message metadata key for the version number of a local or remote update,
|
||||||
|
* as a BDF long.
|
||||||
|
*/
|
||||||
|
String MSG_KEY_VERSION = "version";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Message metadata key for whether an update is local or remote, as a BDF
|
||||||
|
* boolean.
|
||||||
|
*/
|
||||||
|
String MSG_KEY_LOCAL = "local";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Group metadata key for any discovered transport properties of the
|
||||||
|
* contact, as a BDF dictionary.
|
||||||
|
*/
|
||||||
|
String GROUP_KEY_DISCOVERED = "discovered";
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -34,6 +34,14 @@ public interface TransportPropertyManager {
|
|||||||
void addRemoteProperties(Transaction txn, ContactId c,
|
void addRemoteProperties(Transaction txn, ContactId c,
|
||||||
Map<TransportId, TransportProperties> props) throws DbException;
|
Map<TransportId, TransportProperties> props) throws DbException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Stores the given properties discovered from an incoming transport
|
||||||
|
* connection. They will be overridden by any properties received while
|
||||||
|
* adding the contact or synced from the contact.
|
||||||
|
*/
|
||||||
|
void addRemotePropertiesFromConnection(ContactId c, TransportId t,
|
||||||
|
TransportProperties props) throws DbException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns the local transport properties for all transports.
|
* Returns the local transport properties for all transports.
|
||||||
*/
|
*/
|
||||||
|
|||||||
@@ -0,0 +1,23 @@
|
|||||||
|
package org.briarproject.bramble.api.sync;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.Immutable;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A record containing a nonce for choosing between redundant sessions.
|
||||||
|
*/
|
||||||
|
@Immutable
|
||||||
|
@NotNullByDefault
|
||||||
|
public class Priority {
|
||||||
|
|
||||||
|
private final byte[] nonce;
|
||||||
|
|
||||||
|
public Priority(byte[] nonce) {
|
||||||
|
this.nonce = nonce;
|
||||||
|
}
|
||||||
|
|
||||||
|
public byte[] getNonce() {
|
||||||
|
return nonce;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,13 @@
|
|||||||
|
package org.briarproject.bramble.api.sync;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An interface for handling a {@link Priority} record received by an
|
||||||
|
* incoming {@link SyncSession}.
|
||||||
|
*/
|
||||||
|
@NotNullByDefault
|
||||||
|
public interface PriorityHandler {
|
||||||
|
|
||||||
|
void handle(Priority p);
|
||||||
|
}
|
||||||
@@ -10,4 +10,5 @@ public interface RecordTypes {
|
|||||||
byte OFFER = 2;
|
byte OFFER = 2;
|
||||||
byte REQUEST = 3;
|
byte REQUEST = 3;
|
||||||
byte VERSIONS = 4;
|
byte VERSIONS = 4;
|
||||||
|
byte PRIORITY = 5;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -49,4 +49,10 @@ public interface SyncConstants {
|
|||||||
* simultaneously.
|
* simultaneously.
|
||||||
*/
|
*/
|
||||||
int MAX_SUPPORTED_VERSIONS = 10;
|
int MAX_SUPPORTED_VERSIONS = 10;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The length of the priority nonce used for choosing between redundant
|
||||||
|
* connections.
|
||||||
|
*/
|
||||||
|
int PRIORITY_NONCE_BYTES = 16;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -28,4 +28,8 @@ public interface SyncRecordReader {
|
|||||||
boolean hasVersions() throws IOException;
|
boolean hasVersions() throws IOException;
|
||||||
|
|
||||||
Versions readVersions() throws IOException;
|
Versions readVersions() throws IOException;
|
||||||
|
|
||||||
|
boolean hasPriority() throws IOException;
|
||||||
|
|
||||||
|
Priority readPriority() throws IOException;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,5 +17,7 @@ public interface SyncRecordWriter {
|
|||||||
|
|
||||||
void writeVersions(Versions v) throws IOException;
|
void writeVersions(Versions v) throws IOException;
|
||||||
|
|
||||||
|
void writePriority(Priority p) throws IOException;
|
||||||
|
|
||||||
void flush() throws IOException;
|
void flush() throws IOException;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,18 +2,23 @@ package org.briarproject.bramble.api.sync;
|
|||||||
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.transport.StreamWriter;
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
public interface SyncSessionFactory {
|
public interface SyncSessionFactory {
|
||||||
|
|
||||||
SyncSession createIncomingSession(ContactId c, InputStream in);
|
SyncSession createIncomingSession(ContactId c, InputStream in,
|
||||||
|
PriorityHandler handler);
|
||||||
|
|
||||||
SyncSession createSimplexOutgoingSession(ContactId c, int maxLatency,
|
SyncSession createSimplexOutgoingSession(ContactId c, TransportId t,
|
||||||
StreamWriter streamWriter);
|
int maxLatency, StreamWriter streamWriter);
|
||||||
|
|
||||||
SyncSession createDuplexOutgoingSession(ContactId c, int maxLatency,
|
SyncSession createDuplexOutgoingSession(ContactId c, TransportId t,
|
||||||
int maxIdleTime, StreamWriter streamWriter);
|
int maxLatency, int maxIdleTime, StreamWriter streamWriter,
|
||||||
|
@Nullable Priority priority);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,26 @@
|
|||||||
|
package org.briarproject.bramble.api.sync.event;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.event.Event;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.Immutable;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An event that is broadcast when all sync connections using a given
|
||||||
|
* transport should be closed.
|
||||||
|
*/
|
||||||
|
@Immutable
|
||||||
|
@NotNullByDefault
|
||||||
|
public class CloseSyncConnectionsEvent extends Event {
|
||||||
|
|
||||||
|
private final TransportId transportId;
|
||||||
|
|
||||||
|
public CloseSyncConnectionsEvent(TransportId transportId) {
|
||||||
|
this.transportId = transportId;
|
||||||
|
}
|
||||||
|
|
||||||
|
public TransportId getTransportId() {
|
||||||
|
return transportId;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -117,4 +117,10 @@ public class IoUtils {
|
|||||||
throw new IOException(e);
|
throw new IOException(e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static boolean isNonEmptyDirectory(File f) {
|
||||||
|
if (!f.isDirectory()) return false;
|
||||||
|
File[] children = f.listFiles();
|
||||||
|
return children != null && children.length > 0;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,13 +2,17 @@ package org.briarproject.bramble.util;
|
|||||||
|
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
|
||||||
import java.net.Inet6Address;
|
import java.net.Inet4Address;
|
||||||
import java.net.InetAddress;
|
import java.net.InetAddress;
|
||||||
import java.net.InetSocketAddress;
|
import java.net.InetSocketAddress;
|
||||||
import java.net.SocketAddress;
|
import java.net.SocketAddress;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.isNullOrEmpty;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.isValidMac;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.toHexString;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
public class PrivacyUtils {
|
public class PrivacyUtils {
|
||||||
|
|
||||||
@@ -19,7 +23,7 @@ public class PrivacyUtils {
|
|||||||
|
|
||||||
@Nullable
|
@Nullable
|
||||||
public static String scrubMacAddress(@Nullable String address) {
|
public static String scrubMacAddress(@Nullable String address) {
|
||||||
if (address == null || address.length() == 0) return null;
|
if (isNullOrEmpty(address) || !isValidMac(address)) return address;
|
||||||
// this is a fake address we need to know about
|
// this is a fake address we need to know about
|
||||||
if (address.equals("02:00:00:00:00:00")) return address;
|
if (address.equals("02:00:00:00:00:00")) return address;
|
||||||
// keep first and last octet of MAC address
|
// keep first and last octet of MAC address
|
||||||
@@ -27,39 +31,37 @@ public class PrivacyUtils {
|
|||||||
+ address.substring(14, 17);
|
+ address.substring(14, 17);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
|
||||||
public static String scrubInetAddress(InetAddress address) {
|
public static String scrubInetAddress(InetAddress address) {
|
||||||
// don't scrub link and site local addresses
|
if (address instanceof Inet4Address) {
|
||||||
if (address.isLinkLocalAddress() || address.isSiteLocalAddress())
|
// Don't scrub local IPv4 addresses
|
||||||
return address.toString();
|
if (address.isLoopbackAddress() || address.isLinkLocalAddress() ||
|
||||||
// completely scrub IPv6 addresses
|
address.isSiteLocalAddress()) {
|
||||||
if (address instanceof Inet6Address) return "[scrubbed]";
|
return address.getHostAddress();
|
||||||
// keep first and last octet of IPv4 addresses
|
}
|
||||||
return scrubInetAddress(address.toString());
|
// Keep first and last octet of non-local IPv4 addresses
|
||||||
|
return scrubIpv4Address(address.getAddress());
|
||||||
|
} else {
|
||||||
|
// Keep first and last octet of IPv6 addresses
|
||||||
|
return scrubIpv6Address(address.getAddress());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
private static String scrubIpv4Address(byte[] ipv4) {
|
||||||
public static String scrubInetAddress(@Nullable String address) {
|
return (ipv4[0] & 0xFF) + ".[scrubbed]." + (ipv4[3] & 0xFF);
|
||||||
if (address == null) return null;
|
}
|
||||||
|
|
||||||
int firstDot = address.indexOf(".");
|
private static String scrubIpv6Address(byte[] ipv6) {
|
||||||
if (firstDot == -1) return "[scrubbed]";
|
String hex = toHexString(ipv6).toLowerCase();
|
||||||
String prefix = address.substring(0, firstDot + 1);
|
return hex.substring(0, 2) + "[scrubbed]" + hex.substring(30);
|
||||||
int lastDot = address.lastIndexOf(".");
|
|
||||||
String suffix = address.substring(lastDot, address.length());
|
|
||||||
return prefix + "[scrubbed]" + suffix;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
|
||||||
public static String scrubSocketAddress(InetSocketAddress address) {
|
public static String scrubSocketAddress(InetSocketAddress address) {
|
||||||
InetAddress inetAddress = address.getAddress();
|
return scrubInetAddress(address.getAddress());
|
||||||
return scrubInetAddress(inetAddress);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
|
||||||
public static String scrubSocketAddress(SocketAddress address) {
|
public static String scrubSocketAddress(SocketAddress address) {
|
||||||
if (address instanceof InetSocketAddress)
|
if (address instanceof InetSocketAddress)
|
||||||
return scrubSocketAddress((InetSocketAddress) address);
|
return scrubSocketAddress((InetSocketAddress) address);
|
||||||
return scrubInetAddress(address.toString());
|
return "[scrubbed]";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package org.briarproject.bramble;
|
package org.briarproject.bramble;
|
||||||
|
|
||||||
import org.briarproject.bramble.client.ClientModule;
|
import org.briarproject.bramble.client.ClientModule;
|
||||||
|
import org.briarproject.bramble.connection.ConnectionModule;
|
||||||
import org.briarproject.bramble.contact.ContactModule;
|
import org.briarproject.bramble.contact.ContactModule;
|
||||||
import org.briarproject.bramble.crypto.CryptoExecutorModule;
|
import org.briarproject.bramble.crypto.CryptoExecutorModule;
|
||||||
import org.briarproject.bramble.crypto.CryptoModule;
|
import org.briarproject.bramble.crypto.CryptoModule;
|
||||||
@@ -9,6 +10,7 @@ import org.briarproject.bramble.db.DatabaseExecutorModule;
|
|||||||
import org.briarproject.bramble.db.DatabaseModule;
|
import org.briarproject.bramble.db.DatabaseModule;
|
||||||
import org.briarproject.bramble.event.EventModule;
|
import org.briarproject.bramble.event.EventModule;
|
||||||
import org.briarproject.bramble.identity.IdentityModule;
|
import org.briarproject.bramble.identity.IdentityModule;
|
||||||
|
import org.briarproject.bramble.io.IoModule;
|
||||||
import org.briarproject.bramble.keyagreement.KeyAgreementModule;
|
import org.briarproject.bramble.keyagreement.KeyAgreementModule;
|
||||||
import org.briarproject.bramble.lifecycle.LifecycleModule;
|
import org.briarproject.bramble.lifecycle.LifecycleModule;
|
||||||
import org.briarproject.bramble.plugin.PluginModule;
|
import org.briarproject.bramble.plugin.PluginModule;
|
||||||
@@ -27,6 +29,7 @@ import dagger.Module;
|
|||||||
|
|
||||||
@Module(includes = {
|
@Module(includes = {
|
||||||
ClientModule.class,
|
ClientModule.class,
|
||||||
|
ConnectionModule.class,
|
||||||
ContactModule.class,
|
ContactModule.class,
|
||||||
CryptoModule.class,
|
CryptoModule.class,
|
||||||
CryptoExecutorModule.class,
|
CryptoExecutorModule.class,
|
||||||
@@ -35,6 +38,7 @@ import dagger.Module;
|
|||||||
DatabaseExecutorModule.class,
|
DatabaseExecutorModule.class,
|
||||||
EventModule.class,
|
EventModule.class,
|
||||||
IdentityModule.class,
|
IdentityModule.class,
|
||||||
|
IoModule.class,
|
||||||
KeyAgreementModule.class,
|
KeyAgreementModule.class,
|
||||||
LifecycleModule.class,
|
LifecycleModule.class,
|
||||||
PluginModule.class,
|
PluginModule.class,
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package org.briarproject.bramble.account;
|
|||||||
|
|
||||||
import org.briarproject.bramble.api.account.AccountManager;
|
import org.briarproject.bramble.api.account.AccountManager;
|
||||||
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
||||||
|
import org.briarproject.bramble.api.crypto.DecryptionException;
|
||||||
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
||||||
import org.briarproject.bramble.api.crypto.SecretKey;
|
import org.briarproject.bramble.api.crypto.SecretKey;
|
||||||
import org.briarproject.bramble.api.db.DatabaseConfig;
|
import org.briarproject.bramble.api.db.DatabaseConfig;
|
||||||
@@ -17,6 +18,7 @@ import java.io.FileInputStream;
|
|||||||
import java.io.FileOutputStream;
|
import java.io.FileOutputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStreamReader;
|
import java.io.InputStreamReader;
|
||||||
|
import java.nio.charset.Charset;
|
||||||
import java.util.logging.Logger;
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
@@ -24,6 +26,7 @@ import javax.annotation.concurrent.GuardedBy;
|
|||||||
import javax.inject.Inject;
|
import javax.inject.Inject;
|
||||||
|
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_CIPHERTEXT;
|
||||||
import static org.briarproject.bramble.util.LogUtils.logException;
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
import static org.briarproject.bramble.util.StringUtils.fromHexString;
|
import static org.briarproject.bramble.util.StringUtils.fromHexString;
|
||||||
import static org.briarproject.bramble.util.StringUtils.toHexString;
|
import static org.briarproject.bramble.util.StringUtils.toHexString;
|
||||||
@@ -95,7 +98,7 @@ class AccountManagerImpl implements AccountManager {
|
|||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
||||||
new FileInputStream(f), "UTF-8"));
|
new FileInputStream(f), Charset.forName("UTF-8")));
|
||||||
String key = reader.readLine();
|
String key = reader.readLine();
|
||||||
reader.close();
|
reader.close();
|
||||||
return key;
|
return key;
|
||||||
@@ -147,7 +150,7 @@ class AccountManagerImpl implements AccountManager {
|
|||||||
@GuardedBy("stateChangeLock")
|
@GuardedBy("stateChangeLock")
|
||||||
private void writeDbKeyToFile(String key, File f) throws IOException {
|
private void writeDbKeyToFile(String key, File f) throws IOException {
|
||||||
FileOutputStream out = new FileOutputStream(f);
|
FileOutputStream out = new FileOutputStream(f);
|
||||||
out.write(key.getBytes("UTF-8"));
|
out.write(key.getBytes(Charset.forName("UTF-8")));
|
||||||
out.flush();
|
out.flush();
|
||||||
out.close();
|
out.close();
|
||||||
}
|
}
|
||||||
@@ -155,8 +158,7 @@ class AccountManagerImpl implements AccountManager {
|
|||||||
@Override
|
@Override
|
||||||
public boolean accountExists() {
|
public boolean accountExists() {
|
||||||
synchronized (stateChangeLock) {
|
synchronized (stateChangeLock) {
|
||||||
return loadEncryptedDatabaseKey() != null
|
return loadEncryptedDatabaseKey() != null;
|
||||||
&& databaseConfig.getDatabaseDirectory().isDirectory();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -193,31 +195,24 @@ class AccountManagerImpl implements AccountManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean signIn(String password) {
|
public void signIn(String password) throws DecryptionException {
|
||||||
synchronized (stateChangeLock) {
|
synchronized (stateChangeLock) {
|
||||||
SecretKey key = loadAndDecryptDatabaseKey(password);
|
databaseKey = loadAndDecryptDatabaseKey(password);
|
||||||
if (key == null) return false;
|
|
||||||
databaseKey = key;
|
|
||||||
return true;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@GuardedBy("stateChangeLock")
|
@GuardedBy("stateChangeLock")
|
||||||
@Nullable
|
private SecretKey loadAndDecryptDatabaseKey(String password)
|
||||||
private SecretKey loadAndDecryptDatabaseKey(String password) {
|
throws DecryptionException {
|
||||||
String hex = loadEncryptedDatabaseKey();
|
String hex = loadEncryptedDatabaseKey();
|
||||||
if (hex == null) {
|
if (hex == null) {
|
||||||
LOG.warning("Failed to load encrypted database key");
|
LOG.warning("Failed to load encrypted database key");
|
||||||
return null;
|
throw new DecryptionException(INVALID_CIPHERTEXT);
|
||||||
}
|
}
|
||||||
byte[] ciphertext = fromHexString(hex);
|
byte[] ciphertext = fromHexString(hex);
|
||||||
KeyStrengthener keyStrengthener = databaseConfig.getKeyStrengthener();
|
KeyStrengthener keyStrengthener = databaseConfig.getKeyStrengthener();
|
||||||
byte[] plaintext = crypto.decryptWithPassword(ciphertext, password,
|
byte[] plaintext = crypto.decryptWithPassword(ciphertext, password,
|
||||||
keyStrengthener);
|
keyStrengthener);
|
||||||
if (plaintext == null) {
|
|
||||||
LOG.info("Failed to decrypt database key");
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
SecretKey key = new SecretKey(plaintext);
|
SecretKey key = new SecretKey(plaintext);
|
||||||
// If the DB key was encrypted with a weak key and a key strengthener
|
// If the DB key was encrypted with a weak key and a key strengthener
|
||||||
// is now available, re-encrypt the DB key with a strengthened key
|
// is now available, re-encrypt the DB key with a strengthened key
|
||||||
@@ -230,10 +225,11 @@ class AccountManagerImpl implements AccountManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean changePassword(String oldPassword, String newPassword) {
|
public void changePassword(String oldPassword, String newPassword)
|
||||||
|
throws DecryptionException {
|
||||||
synchronized (stateChangeLock) {
|
synchronized (stateChangeLock) {
|
||||||
SecretKey key = loadAndDecryptDatabaseKey(oldPassword);
|
SecretKey key = loadAndDecryptDatabaseKey(oldPassword);
|
||||||
return key != null && encryptAndStoreDatabaseKey(key, newPassword);
|
encryptAndStoreDatabaseKey(key, newPassword);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,79 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static java.util.logging.Logger.getLogger;
|
||||||
|
import static org.briarproject.bramble.api.transport.TransportConstants.TAG_LENGTH;
|
||||||
|
import static org.briarproject.bramble.util.IoUtils.read;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
abstract class Connection {
|
||||||
|
|
||||||
|
protected static final Logger LOG = getLogger(Connection.class.getName());
|
||||||
|
|
||||||
|
final KeyManager keyManager;
|
||||||
|
final ConnectionRegistry connectionRegistry;
|
||||||
|
final StreamReaderFactory streamReaderFactory;
|
||||||
|
final StreamWriterFactory streamWriterFactory;
|
||||||
|
|
||||||
|
Connection(KeyManager keyManager, ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory) {
|
||||||
|
this.keyManager = keyManager;
|
||||||
|
this.connectionRegistry = connectionRegistry;
|
||||||
|
this.streamReaderFactory = streamReaderFactory;
|
||||||
|
this.streamWriterFactory = streamWriterFactory;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nullable
|
||||||
|
StreamContext recogniseTag(TransportConnectionReader reader,
|
||||||
|
TransportId transportId) {
|
||||||
|
StreamContext ctx;
|
||||||
|
try {
|
||||||
|
byte[] tag = readTag(reader.getInputStream());
|
||||||
|
return keyManager.getStreamContext(transportId, tag);
|
||||||
|
} catch (IOException | DbException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private byte[] readTag(InputStream in) throws IOException {
|
||||||
|
byte[] tag = new byte[TAG_LENGTH];
|
||||||
|
read(in, tag);
|
||||||
|
return tag;
|
||||||
|
}
|
||||||
|
|
||||||
|
void disposeOnError(TransportConnectionReader reader, boolean recognised) {
|
||||||
|
try {
|
||||||
|
reader.dispose(true, recognised);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void disposeOnError(TransportConnectionWriter writer) {
|
||||||
|
try {
|
||||||
|
writer.dispose(true);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,114 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.security.SecureRandom;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.Immutable;
|
||||||
|
import javax.inject.Inject;
|
||||||
|
|
||||||
|
@Immutable
|
||||||
|
@NotNullByDefault
|
||||||
|
class ConnectionManagerImpl implements ConnectionManager {
|
||||||
|
|
||||||
|
private final Executor ioExecutor;
|
||||||
|
private final KeyManager keyManager;
|
||||||
|
private final StreamReaderFactory streamReaderFactory;
|
||||||
|
private final StreamWriterFactory streamWriterFactory;
|
||||||
|
private final SyncSessionFactory syncSessionFactory;
|
||||||
|
private final HandshakeManager handshakeManager;
|
||||||
|
private final ContactExchangeManager contactExchangeManager;
|
||||||
|
private final ConnectionRegistry connectionRegistry;
|
||||||
|
private final TransportPropertyManager transportPropertyManager;
|
||||||
|
private final SecureRandom secureRandom;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
ConnectionManagerImpl(@IoExecutor Executor ioExecutor,
|
||||||
|
KeyManager keyManager, StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
HandshakeManager handshakeManager,
|
||||||
|
ContactExchangeManager contactExchangeManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
SecureRandom secureRandom) {
|
||||||
|
this.ioExecutor = ioExecutor;
|
||||||
|
this.keyManager = keyManager;
|
||||||
|
this.streamReaderFactory = streamReaderFactory;
|
||||||
|
this.streamWriterFactory = streamWriterFactory;
|
||||||
|
this.syncSessionFactory = syncSessionFactory;
|
||||||
|
this.handshakeManager = handshakeManager;
|
||||||
|
this.contactExchangeManager = contactExchangeManager;
|
||||||
|
this.connectionRegistry = connectionRegistry;
|
||||||
|
this.transportPropertyManager = transportPropertyManager;
|
||||||
|
this.secureRandom = secureRandom;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageIncomingConnection(TransportId t,
|
||||||
|
TransportConnectionReader r) {
|
||||||
|
ioExecutor.execute(new IncomingSimplexSyncConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
syncSessionFactory, transportPropertyManager, t, r));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageIncomingConnection(TransportId t,
|
||||||
|
DuplexTransportConnection d) {
|
||||||
|
ioExecutor.execute(new IncomingDuplexSyncConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
syncSessionFactory, transportPropertyManager, ioExecutor,
|
||||||
|
t, d));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageIncomingConnection(PendingContactId p, TransportId t,
|
||||||
|
DuplexTransportConnection d) {
|
||||||
|
ioExecutor.execute(new IncomingHandshakeConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
handshakeManager, contactExchangeManager, this, p, t, d));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageOutgoingConnection(ContactId c, TransportId t,
|
||||||
|
TransportConnectionWriter w) {
|
||||||
|
ioExecutor.execute(new OutgoingSimplexSyncConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
syncSessionFactory, transportPropertyManager, c, t, w));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageOutgoingConnection(ContactId c, TransportId t,
|
||||||
|
DuplexTransportConnection d) {
|
||||||
|
ioExecutor.execute(new OutgoingDuplexSyncConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
syncSessionFactory, transportPropertyManager, ioExecutor,
|
||||||
|
secureRandom, c, t, d));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void manageOutgoingConnection(PendingContactId p, TransportId t,
|
||||||
|
DuplexTransportConnection d) {
|
||||||
|
ioExecutor.execute(new OutgoingHandshakeConnection(keyManager,
|
||||||
|
connectionRegistry, streamReaderFactory, streamWriterFactory,
|
||||||
|
handshakeManager, contactExchangeManager, this, p, t, d));
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,26 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
|
||||||
|
import javax.inject.Singleton;
|
||||||
|
|
||||||
|
import dagger.Module;
|
||||||
|
import dagger.Provides;
|
||||||
|
|
||||||
|
@Module
|
||||||
|
public class ConnectionModule {
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
ConnectionManager provideConnectionManager(
|
||||||
|
ConnectionManagerImpl connectionManager) {
|
||||||
|
return connectionManager;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
@Singleton
|
||||||
|
ConnectionRegistry provideConnectionRegistry(
|
||||||
|
ConnectionRegistryImpl connectionRegistry) {
|
||||||
|
return connectionRegistry;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,283 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.Bytes;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.connection.InterruptibleConnection;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
import javax.annotation.concurrent.GuardedBy;
|
||||||
|
import javax.annotation.concurrent.ThreadSafe;
|
||||||
|
import javax.inject.Inject;
|
||||||
|
|
||||||
|
import static java.util.Collections.emptyList;
|
||||||
|
import static java.util.logging.Level.INFO;
|
||||||
|
import static java.util.logging.Logger.getLogger;
|
||||||
|
|
||||||
|
@ThreadSafe
|
||||||
|
@NotNullByDefault
|
||||||
|
class ConnectionRegistryImpl implements ConnectionRegistry {
|
||||||
|
|
||||||
|
private static final Logger LOG =
|
||||||
|
getLogger(ConnectionRegistryImpl.class.getName());
|
||||||
|
|
||||||
|
private final EventBus eventBus;
|
||||||
|
private final Map<TransportId, List<TransportId>> transportPrefs;
|
||||||
|
|
||||||
|
private final Object lock = new Object();
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private final Map<ContactId, List<ConnectionRecord>> contactConnections;
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private final Set<PendingContactId> connectedPendingContacts;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
ConnectionRegistryImpl(EventBus eventBus, PluginConfig pluginConfig) {
|
||||||
|
this.eventBus = eventBus;
|
||||||
|
transportPrefs = pluginConfig.getTransportPreferences();
|
||||||
|
contactConnections = new HashMap<>();
|
||||||
|
connectedPendingContacts = new HashSet<>();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void registerIncomingConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn) {
|
||||||
|
registerConnection(c, t, conn, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void registerOutgoingConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, Priority priority) {
|
||||||
|
registerConnection(c, t, conn, false);
|
||||||
|
setPriority(c, t, conn, priority);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void registerConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, boolean incoming) {
|
||||||
|
if (LOG.isLoggable(INFO)) {
|
||||||
|
if (incoming) LOG.info("Incoming connection registered: " + t);
|
||||||
|
else LOG.info("Outgoing connection registered: " + t);
|
||||||
|
}
|
||||||
|
boolean firstConnection;
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ConnectionRecord> recs = contactConnections.get(c);
|
||||||
|
if (recs == null) {
|
||||||
|
recs = new ArrayList<>();
|
||||||
|
contactConnections.put(c, recs);
|
||||||
|
}
|
||||||
|
firstConnection = recs.isEmpty();
|
||||||
|
recs.add(new ConnectionRecord(t, conn));
|
||||||
|
}
|
||||||
|
eventBus.broadcast(new ConnectionOpenedEvent(c, t, incoming));
|
||||||
|
if (firstConnection) {
|
||||||
|
LOG.info("Contact connected");
|
||||||
|
eventBus.broadcast(new ContactConnectedEvent(c));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setPriority(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, Priority priority) {
|
||||||
|
if (LOG.isLoggable(INFO)) LOG.info("Setting connection priority: " + t);
|
||||||
|
List<InterruptibleConnection> toInterrupt;
|
||||||
|
boolean interruptNewConnection = false;
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ConnectionRecord> recs = contactConnections.get(c);
|
||||||
|
if (recs == null) throw new IllegalArgumentException();
|
||||||
|
toInterrupt = new ArrayList<>(recs.size());
|
||||||
|
for (ConnectionRecord rec : recs) {
|
||||||
|
if (rec.conn == conn) {
|
||||||
|
// Store the priority of this connection
|
||||||
|
rec.priority = priority;
|
||||||
|
} else if (rec.priority != null) {
|
||||||
|
int compare = compareConnections(t, priority,
|
||||||
|
rec.transportId, rec.priority);
|
||||||
|
if (compare == -1) {
|
||||||
|
// The old connection is better than the new one
|
||||||
|
interruptNewConnection = true;
|
||||||
|
} else if (compare == 1 && !rec.interrupted) {
|
||||||
|
// The new connection is better than the old one
|
||||||
|
toInterrupt.add(rec.conn);
|
||||||
|
rec.interrupted = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (interruptNewConnection) {
|
||||||
|
LOG.info("Interrupting new connection");
|
||||||
|
conn.interruptOutgoingSession();
|
||||||
|
}
|
||||||
|
for (InterruptibleConnection old : toInterrupt) {
|
||||||
|
LOG.info("Interrupting old connection");
|
||||||
|
old.interruptOutgoingSession();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private int compareConnections(TransportId tA, Priority pA, TransportId tB,
|
||||||
|
Priority pB) {
|
||||||
|
if (getBetterTransports(tA).contains(tB)) return -1;
|
||||||
|
if (getBetterTransports(tB).contains(tA)) return 1;
|
||||||
|
return tA.equals(tB) ? Bytes.compare(pA.getNonce(), pB.getNonce()) : 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<TransportId> getBetterTransports(TransportId t) {
|
||||||
|
List<TransportId> better = transportPrefs.get(t);
|
||||||
|
return better == null ? emptyList() : better;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void unregisterConnection(ContactId c, TransportId t,
|
||||||
|
InterruptibleConnection conn, boolean incoming, boolean exception) {
|
||||||
|
if (LOG.isLoggable(INFO)) {
|
||||||
|
if (incoming) LOG.info("Incoming connection unregistered: " + t);
|
||||||
|
else LOG.info("Outgoing connection unregistered: " + t);
|
||||||
|
}
|
||||||
|
boolean lastConnection;
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ConnectionRecord> recs = contactConnections.get(c);
|
||||||
|
if (recs == null || !recs.remove(new ConnectionRecord(t, conn)))
|
||||||
|
throw new IllegalArgumentException();
|
||||||
|
lastConnection = recs.isEmpty();
|
||||||
|
}
|
||||||
|
eventBus.broadcast(
|
||||||
|
new ConnectionClosedEvent(c, t, incoming, exception));
|
||||||
|
if (lastConnection) {
|
||||||
|
LOG.info("Contact disconnected");
|
||||||
|
eventBus.broadcast(new ContactDisconnectedEvent(c));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Collection<ContactId> getConnectedContacts(TransportId t) {
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ContactId> contactIds = new ArrayList<>();
|
||||||
|
for (Entry<ContactId, List<ConnectionRecord>> e :
|
||||||
|
contactConnections.entrySet()) {
|
||||||
|
for (ConnectionRecord rec : e.getValue()) {
|
||||||
|
if (rec.transportId.equals(t)) {
|
||||||
|
contactIds.add(e.getKey());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (LOG.isLoggable(INFO)) {
|
||||||
|
LOG.info(contactIds.size() + " contacts connected: " + t);
|
||||||
|
}
|
||||||
|
return contactIds;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Collection<ContactId> getConnectedOrBetterContacts(TransportId t) {
|
||||||
|
synchronized (lock) {
|
||||||
|
List<TransportId> better = getBetterTransports(t);
|
||||||
|
List<ContactId> contactIds = new ArrayList<>();
|
||||||
|
for (Entry<ContactId, List<ConnectionRecord>> e :
|
||||||
|
contactConnections.entrySet()) {
|
||||||
|
for (ConnectionRecord rec : e.getValue()) {
|
||||||
|
if (rec.transportId.equals(t) ||
|
||||||
|
better.contains(rec.transportId)) {
|
||||||
|
contactIds.add(e.getKey());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (LOG.isLoggable(INFO)) {
|
||||||
|
LOG.info(contactIds.size()
|
||||||
|
+ " contacts connected or better: " + t);
|
||||||
|
}
|
||||||
|
return contactIds;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isConnected(ContactId c, TransportId t) {
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ConnectionRecord> recs = contactConnections.get(c);
|
||||||
|
if (recs == null) return false;
|
||||||
|
for (ConnectionRecord rec : recs) {
|
||||||
|
if (rec.transportId.equals(t)) return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isConnected(ContactId c) {
|
||||||
|
synchronized (lock) {
|
||||||
|
List<ConnectionRecord> recs = contactConnections.get(c);
|
||||||
|
return recs != null && !recs.isEmpty();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean registerConnection(PendingContactId p) {
|
||||||
|
boolean added;
|
||||||
|
synchronized (lock) {
|
||||||
|
added = connectedPendingContacts.add(p);
|
||||||
|
}
|
||||||
|
if (added) eventBus.broadcast(new RendezvousConnectionOpenedEvent(p));
|
||||||
|
return added;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void unregisterConnection(PendingContactId p, boolean success) {
|
||||||
|
synchronized (lock) {
|
||||||
|
if (!connectedPendingContacts.remove(p))
|
||||||
|
throw new IllegalArgumentException();
|
||||||
|
}
|
||||||
|
eventBus.broadcast(new RendezvousConnectionClosedEvent(p, success));
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class ConnectionRecord {
|
||||||
|
|
||||||
|
private final TransportId transportId;
|
||||||
|
private final InterruptibleConnection conn;
|
||||||
|
@GuardedBy("lock")
|
||||||
|
@Nullable
|
||||||
|
private Priority priority = null;
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private boolean interrupted = false;
|
||||||
|
|
||||||
|
private ConnectionRecord(TransportId transportId,
|
||||||
|
InterruptibleConnection conn) {
|
||||||
|
this.transportId = transportId;
|
||||||
|
this.conn = conn;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean equals(Object o) {
|
||||||
|
if (o instanceof ConnectionRecord) {
|
||||||
|
return conn == ((ConnectionRecord) o).conn;
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int hashCode() {
|
||||||
|
return conn.hashCode();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,109 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.connection.InterruptibleConnection;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
import javax.annotation.concurrent.GuardedBy;
|
||||||
|
|
||||||
|
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
abstract class DuplexSyncConnection extends SyncConnection
|
||||||
|
implements InterruptibleConnection {
|
||||||
|
|
||||||
|
final Executor ioExecutor;
|
||||||
|
final TransportId transportId;
|
||||||
|
final TransportConnectionReader reader;
|
||||||
|
final TransportConnectionWriter writer;
|
||||||
|
final TransportProperties remote;
|
||||||
|
|
||||||
|
private final Object interruptLock = new Object();
|
||||||
|
|
||||||
|
@GuardedBy("interruptLock")
|
||||||
|
@Nullable
|
||||||
|
private SyncSession outgoingSession = null;
|
||||||
|
@GuardedBy("interruptLock")
|
||||||
|
private boolean interruptWaiting = false;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void interruptOutgoingSession() {
|
||||||
|
SyncSession out = null;
|
||||||
|
synchronized (interruptLock) {
|
||||||
|
if (outgoingSession == null) interruptWaiting = true;
|
||||||
|
else out = outgoingSession;
|
||||||
|
}
|
||||||
|
if (out != null) out.interrupt();
|
||||||
|
}
|
||||||
|
|
||||||
|
void setOutgoingSession(SyncSession outgoingSession) {
|
||||||
|
boolean interruptWasWaiting = false;
|
||||||
|
synchronized (interruptLock) {
|
||||||
|
this.outgoingSession = outgoingSession;
|
||||||
|
if (interruptWaiting) {
|
||||||
|
interruptWasWaiting = true;
|
||||||
|
interruptWaiting = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (interruptWasWaiting) outgoingSession.interrupt();
|
||||||
|
}
|
||||||
|
|
||||||
|
DuplexSyncConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
Executor ioExecutor, TransportId transportId,
|
||||||
|
DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, syncSessionFactory,
|
||||||
|
transportPropertyManager);
|
||||||
|
this.ioExecutor = ioExecutor;
|
||||||
|
this.transportId = transportId;
|
||||||
|
reader = connection.getReader();
|
||||||
|
writer = connection.getWriter();
|
||||||
|
remote = connection.getRemoteProperties();
|
||||||
|
}
|
||||||
|
|
||||||
|
void onReadError(boolean recognised) {
|
||||||
|
disposeOnError(reader, recognised);
|
||||||
|
disposeOnError(writer);
|
||||||
|
interruptOutgoingSession();
|
||||||
|
}
|
||||||
|
|
||||||
|
void onWriteError() {
|
||||||
|
disposeOnError(reader, true);
|
||||||
|
disposeOnError(writer);
|
||||||
|
}
|
||||||
|
|
||||||
|
SyncSession createDuplexOutgoingSession(StreamContext ctx,
|
||||||
|
TransportConnectionWriter w, @Nullable Priority priority)
|
||||||
|
throws IOException {
|
||||||
|
StreamWriter streamWriter = streamWriterFactory.createStreamWriter(
|
||||||
|
w.getOutputStream(), ctx);
|
||||||
|
ContactId c = requireNonNull(ctx.getContactId());
|
||||||
|
return syncSessionFactory.createDuplexOutgoingSession(c,
|
||||||
|
ctx.getTransportId(), w.getMaxLatency(), w.getMaxIdleTime(),
|
||||||
|
streamWriter, priority);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,72 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
abstract class HandshakeConnection extends Connection {
|
||||||
|
|
||||||
|
final HandshakeManager handshakeManager;
|
||||||
|
final ContactExchangeManager contactExchangeManager;
|
||||||
|
final ConnectionManager connectionManager;
|
||||||
|
final PendingContactId pendingContactId;
|
||||||
|
final TransportId transportId;
|
||||||
|
final DuplexTransportConnection connection;
|
||||||
|
final TransportConnectionReader reader;
|
||||||
|
final TransportConnectionWriter writer;
|
||||||
|
|
||||||
|
HandshakeConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
HandshakeManager handshakeManager,
|
||||||
|
ContactExchangeManager contactExchangeManager,
|
||||||
|
ConnectionManager connectionManager,
|
||||||
|
PendingContactId pendingContactId,
|
||||||
|
TransportId transportId, DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory);
|
||||||
|
this.handshakeManager = handshakeManager;
|
||||||
|
this.contactExchangeManager = contactExchangeManager;
|
||||||
|
this.connectionManager = connectionManager;
|
||||||
|
this.pendingContactId = pendingContactId;
|
||||||
|
this.transportId = transportId;
|
||||||
|
this.connection = connection;
|
||||||
|
reader = connection.getReader();
|
||||||
|
writer = connection.getWriter();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nullable
|
||||||
|
StreamContext allocateStreamContext(PendingContactId pendingContactId,
|
||||||
|
TransportId transportId) {
|
||||||
|
try {
|
||||||
|
return keyManager.getStreamContext(pendingContactId, transportId);
|
||||||
|
} catch (DbException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void onError(boolean recognised) {
|
||||||
|
disposeOnError(reader, recognised);
|
||||||
|
disposeOnError(writer);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,107 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class IncomingDuplexSyncConnection extends DuplexSyncConnection
|
||||||
|
implements Runnable {
|
||||||
|
|
||||||
|
IncomingDuplexSyncConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
Executor ioExecutor, TransportId transportId,
|
||||||
|
DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, syncSessionFactory,
|
||||||
|
transportPropertyManager, ioExecutor, transportId, connection);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Read and recognise the tag
|
||||||
|
StreamContext ctx = recogniseTag(reader, transportId);
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.info("Unrecognised tag");
|
||||||
|
onReadError(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ContactId contactId = ctx.getContactId();
|
||||||
|
if (contactId == null) {
|
||||||
|
LOG.warning("Expected contact tag, got rendezvous tag");
|
||||||
|
onReadError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ctx.isHandshakeMode()) {
|
||||||
|
// TODO: Support handshake mode for contacts
|
||||||
|
LOG.warning("Received handshake tag, expected rotation mode");
|
||||||
|
onReadError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
connectionRegistry.registerIncomingConnection(contactId, transportId,
|
||||||
|
this);
|
||||||
|
// Start the outgoing session on another thread
|
||||||
|
ioExecutor.execute(() -> runOutgoingSession(contactId));
|
||||||
|
try {
|
||||||
|
// Store any transport properties discovered from the connection
|
||||||
|
transportPropertyManager.addRemotePropertiesFromConnection(
|
||||||
|
contactId, transportId, remote);
|
||||||
|
// Update the connection registry when we receive our priority
|
||||||
|
PriorityHandler handler = p -> connectionRegistry.setPriority(
|
||||||
|
contactId, transportId, this, p);
|
||||||
|
// Create and run the incoming session
|
||||||
|
createIncomingSession(ctx, reader, handler).run();
|
||||||
|
reader.dispose(false, true);
|
||||||
|
interruptOutgoingSession();
|
||||||
|
connectionRegistry.unregisterConnection(contactId, transportId,
|
||||||
|
this, true, false);
|
||||||
|
} catch (DbException | IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onReadError(true);
|
||||||
|
connectionRegistry.unregisterConnection(contactId, transportId,
|
||||||
|
this, true, true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void runOutgoingSession(ContactId contactId) {
|
||||||
|
// Allocate a stream context
|
||||||
|
StreamContext ctx = allocateStreamContext(contactId, transportId);
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.warning("Could not allocate stream context");
|
||||||
|
onWriteError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
// Create and run the outgoing session
|
||||||
|
SyncSession out = createDuplexOutgoingSession(ctx, writer, null);
|
||||||
|
setOutgoingSession(out);
|
||||||
|
out.run();
|
||||||
|
writer.dispose(false);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onWriteError();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@@ -0,0 +1,93 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager.HandshakeResult;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class IncomingHandshakeConnection extends HandshakeConnection
|
||||||
|
implements Runnable {
|
||||||
|
|
||||||
|
IncomingHandshakeConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
HandshakeManager handshakeManager,
|
||||||
|
ContactExchangeManager contactExchangeManager,
|
||||||
|
ConnectionManager connectionManager,
|
||||||
|
PendingContactId pendingContactId,
|
||||||
|
TransportId transportId, DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, handshakeManager, contactExchangeManager,
|
||||||
|
connectionManager, pendingContactId, transportId, connection);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Read and recognise the tag
|
||||||
|
StreamContext ctxIn = recogniseTag(reader, transportId);
|
||||||
|
if (ctxIn == null) {
|
||||||
|
LOG.info("Unrecognised tag");
|
||||||
|
onError(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
PendingContactId inPendingContactId = ctxIn.getPendingContactId();
|
||||||
|
if (inPendingContactId == null) {
|
||||||
|
LOG.warning("Expected rendezvous tag, got contact tag");
|
||||||
|
onError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Allocate the outgoing stream context
|
||||||
|
StreamContext ctxOut =
|
||||||
|
allocateStreamContext(pendingContactId, transportId);
|
||||||
|
if (ctxOut == null) {
|
||||||
|
LOG.warning("Could not allocate stream context");
|
||||||
|
onError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Close the connection if it's redundant
|
||||||
|
if (!connectionRegistry.registerConnection(pendingContactId)) {
|
||||||
|
LOG.info("Redundant rendezvous connection");
|
||||||
|
onError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Handshake and exchange contacts
|
||||||
|
try {
|
||||||
|
InputStream in = streamReaderFactory.createStreamReader(
|
||||||
|
reader.getInputStream(), ctxIn);
|
||||||
|
// Flush the output stream to send the outgoing stream header
|
||||||
|
StreamWriter out = streamWriterFactory.createStreamWriter(
|
||||||
|
writer.getOutputStream(), ctxOut);
|
||||||
|
out.getOutputStream().flush();
|
||||||
|
HandshakeResult result =
|
||||||
|
handshakeManager.handshake(pendingContactId, in, out);
|
||||||
|
contactExchangeManager.exchangeContacts(pendingContactId,
|
||||||
|
connection, result.getMasterKey(), result.isAlice(), false);
|
||||||
|
connectionRegistry.unregisterConnection(pendingContactId, true);
|
||||||
|
// Reuse the connection as a transport connection
|
||||||
|
connectionManager.manageIncomingConnection(transportId, connection);
|
||||||
|
} catch (IOException | DbException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onError(true);
|
||||||
|
connectionRegistry.unregisterConnection(pendingContactId, false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,79 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class IncomingSimplexSyncConnection extends SyncConnection implements Runnable {
|
||||||
|
|
||||||
|
private final TransportId transportId;
|
||||||
|
private final TransportConnectionReader reader;
|
||||||
|
|
||||||
|
IncomingSimplexSyncConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
TransportId transportId, TransportConnectionReader reader) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, syncSessionFactory,
|
||||||
|
transportPropertyManager);
|
||||||
|
this.transportId = transportId;
|
||||||
|
this.reader = reader;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Read and recognise the tag
|
||||||
|
StreamContext ctx = recogniseTag(reader, transportId);
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.info("Unrecognised tag");
|
||||||
|
onError(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ContactId contactId = ctx.getContactId();
|
||||||
|
if (contactId == null) {
|
||||||
|
LOG.warning("Received rendezvous stream, expected contact");
|
||||||
|
onError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ctx.isHandshakeMode()) {
|
||||||
|
// TODO: Support handshake mode for contacts
|
||||||
|
LOG.warning("Received handshake tag, expected rotation mode");
|
||||||
|
onError(true);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
// We don't expect to receive a priority for this connection
|
||||||
|
PriorityHandler handler = p ->
|
||||||
|
LOG.info("Ignoring priority for simplex connection");
|
||||||
|
// Create and run the incoming session
|
||||||
|
createIncomingSession(ctx, reader, handler).run();
|
||||||
|
reader.dispose(false, true);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onError(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void onError(boolean recognised) {
|
||||||
|
disposeOnError(reader, recognised);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@@ -0,0 +1,140 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.security.SecureRandom;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.api.sync.SyncConstants.PRIORITY_NONCE_BYTES;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class OutgoingDuplexSyncConnection extends DuplexSyncConnection
|
||||||
|
implements Runnable {
|
||||||
|
|
||||||
|
private final SecureRandom secureRandom;
|
||||||
|
private final ContactId contactId;
|
||||||
|
|
||||||
|
OutgoingDuplexSyncConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
Executor ioExecutor, SecureRandom secureRandom, ContactId contactId,
|
||||||
|
TransportId transportId, DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, syncSessionFactory,
|
||||||
|
transportPropertyManager, ioExecutor, transportId, connection);
|
||||||
|
this.secureRandom = secureRandom;
|
||||||
|
this.contactId = contactId;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Allocate a stream context
|
||||||
|
StreamContext ctx = allocateStreamContext(contactId, transportId);
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.warning("Could not allocate stream context");
|
||||||
|
onWriteError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ctx.isHandshakeMode()) {
|
||||||
|
// TODO: Support handshake mode for contacts
|
||||||
|
LOG.warning("Cannot use handshake mode stream context");
|
||||||
|
onWriteError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Start the incoming session on another thread
|
||||||
|
Priority priority = generatePriority();
|
||||||
|
ioExecutor.execute(() -> runIncomingSession(priority));
|
||||||
|
try {
|
||||||
|
// Create and run the outgoing session
|
||||||
|
SyncSession out =
|
||||||
|
createDuplexOutgoingSession(ctx, writer, priority);
|
||||||
|
setOutgoingSession(out);
|
||||||
|
out.run();
|
||||||
|
writer.dispose(false);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onWriteError();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void runIncomingSession(Priority priority) {
|
||||||
|
// Read and recognise the tag
|
||||||
|
StreamContext ctx = recogniseTag(reader, transportId);
|
||||||
|
// Unrecognised tags are suspicious in this case
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.warning("Unrecognised tag for returning stream");
|
||||||
|
onReadError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Check that the stream comes from the expected contact
|
||||||
|
ContactId inContactId = ctx.getContactId();
|
||||||
|
if (inContactId == null) {
|
||||||
|
LOG.warning("Expected contact tag, got rendezvous tag");
|
||||||
|
onReadError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!contactId.equals(inContactId)) {
|
||||||
|
LOG.warning("Wrong contact ID for returning stream");
|
||||||
|
onReadError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (ctx.isHandshakeMode()) {
|
||||||
|
// TODO: Support handshake mode for contacts
|
||||||
|
LOG.warning("Received handshake tag, expected rotation mode");
|
||||||
|
onReadError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
connectionRegistry.registerOutgoingConnection(contactId, transportId,
|
||||||
|
this, priority);
|
||||||
|
try {
|
||||||
|
// Store any transport properties discovered from the connection
|
||||||
|
transportPropertyManager.addRemotePropertiesFromConnection(
|
||||||
|
contactId, transportId, remote);
|
||||||
|
// We don't expect to receive a priority for this connection
|
||||||
|
PriorityHandler handler = p ->
|
||||||
|
LOG.info("Ignoring priority for outgoing connection");
|
||||||
|
// Create and run the incoming session
|
||||||
|
createIncomingSession(ctx, reader, handler).run();
|
||||||
|
reader.dispose(false, true);
|
||||||
|
interruptOutgoingSession();
|
||||||
|
connectionRegistry.unregisterConnection(contactId, transportId,
|
||||||
|
this, false, false);
|
||||||
|
} catch (DbException | IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onReadError();
|
||||||
|
connectionRegistry.unregisterConnection(contactId, transportId,
|
||||||
|
this, false, true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void onReadError() {
|
||||||
|
// 'Recognised' is always true for outgoing connections
|
||||||
|
onReadError(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
private Priority generatePriority() {
|
||||||
|
byte[] nonce = new byte[PRIORITY_NONCE_BYTES];
|
||||||
|
secureRandom.nextBytes(nonce);
|
||||||
|
return new Priority(nonce);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,115 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.Contact;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager;
|
||||||
|
import org.briarproject.bramble.api.contact.HandshakeManager.HandshakeResult;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class OutgoingHandshakeConnection extends HandshakeConnection
|
||||||
|
implements Runnable {
|
||||||
|
|
||||||
|
OutgoingHandshakeConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
HandshakeManager handshakeManager,
|
||||||
|
ContactExchangeManager contactExchangeManager,
|
||||||
|
ConnectionManager connectionManager,
|
||||||
|
PendingContactId pendingContactId,
|
||||||
|
TransportId transportId, DuplexTransportConnection connection) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, handshakeManager, contactExchangeManager,
|
||||||
|
connectionManager, pendingContactId, transportId, connection);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Allocate the outgoing stream context
|
||||||
|
StreamContext ctxOut =
|
||||||
|
allocateStreamContext(pendingContactId, transportId);
|
||||||
|
if (ctxOut == null) {
|
||||||
|
LOG.warning("Could not allocate stream context");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Flush the output stream to send the outgoing stream header
|
||||||
|
StreamWriter out;
|
||||||
|
try {
|
||||||
|
out = streamWriterFactory.createStreamWriter(
|
||||||
|
writer.getOutputStream(), ctxOut);
|
||||||
|
out.getOutputStream().flush();
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Read and recognise the tag
|
||||||
|
StreamContext ctxIn = recogniseTag(reader, transportId);
|
||||||
|
// Unrecognised tags are suspicious in this case
|
||||||
|
if (ctxIn == null) {
|
||||||
|
LOG.warning("Unrecognised tag for returning stream");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Check that the stream comes from the expected pending contact
|
||||||
|
PendingContactId inPendingContactId = ctxIn.getPendingContactId();
|
||||||
|
if (inPendingContactId == null) {
|
||||||
|
LOG.warning("Expected rendezvous tag, got contact tag");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!inPendingContactId.equals(pendingContactId)) {
|
||||||
|
LOG.warning("Wrong pending contact ID for returning stream");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Close the connection if it's redundant
|
||||||
|
if (!connectionRegistry.registerConnection(pendingContactId)) {
|
||||||
|
LOG.info("Redundant rendezvous connection");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Handshake and exchange contacts
|
||||||
|
try {
|
||||||
|
InputStream in = streamReaderFactory.createStreamReader(
|
||||||
|
reader.getInputStream(), ctxIn);
|
||||||
|
HandshakeResult result =
|
||||||
|
handshakeManager.handshake(pendingContactId, in, out);
|
||||||
|
Contact contact = contactExchangeManager.exchangeContacts(
|
||||||
|
pendingContactId, connection, result.getMasterKey(),
|
||||||
|
result.isAlice(), false);
|
||||||
|
connectionRegistry.unregisterConnection(pendingContactId, true);
|
||||||
|
// Reuse the connection as a transport connection
|
||||||
|
connectionManager.manageOutgoingConnection(contact.getId(),
|
||||||
|
transportId, connection);
|
||||||
|
} catch (IOException | DbException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onError();
|
||||||
|
connectionRegistry.unregisterConnection(pendingContactId, false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void onError() {
|
||||||
|
// 'Recognised' is always true for outgoing connections
|
||||||
|
onError(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,78 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class OutgoingSimplexSyncConnection extends SyncConnection implements Runnable {
|
||||||
|
|
||||||
|
private final ContactId contactId;
|
||||||
|
private final TransportId transportId;
|
||||||
|
private final TransportConnectionWriter writer;
|
||||||
|
|
||||||
|
OutgoingSimplexSyncConnection(KeyManager keyManager,
|
||||||
|
ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager,
|
||||||
|
ContactId contactId, TransportId transportId,
|
||||||
|
TransportConnectionWriter writer) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory, syncSessionFactory,
|
||||||
|
transportPropertyManager);
|
||||||
|
this.contactId = contactId;
|
||||||
|
this.transportId = transportId;
|
||||||
|
this.writer = writer;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
// Allocate a stream context
|
||||||
|
StreamContext ctx = allocateStreamContext(contactId, transportId);
|
||||||
|
if (ctx == null) {
|
||||||
|
LOG.warning("Could not allocate stream context");
|
||||||
|
onError();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
// Create and run the outgoing session
|
||||||
|
createSimplexOutgoingSession(ctx, writer).run();
|
||||||
|
writer.dispose(false);
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
onError();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void onError() {
|
||||||
|
disposeOnError(writer);
|
||||||
|
}
|
||||||
|
|
||||||
|
private SyncSession createSimplexOutgoingSession(StreamContext ctx,
|
||||||
|
TransportConnectionWriter w) throws IOException {
|
||||||
|
StreamWriter streamWriter = streamWriterFactory.createStreamWriter(
|
||||||
|
w.getOutputStream(), ctx);
|
||||||
|
ContactId c = requireNonNull(ctx.getContactId());
|
||||||
|
return syncSessionFactory.createSimplexOutgoingSession(c,
|
||||||
|
ctx.getTransportId(), w.getMaxLatency(), streamWriter);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportPropertyManager;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
|
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.KeyManager;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamContext;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
||||||
|
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.WARNING;
|
||||||
|
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class SyncConnection extends Connection {
|
||||||
|
|
||||||
|
final SyncSessionFactory syncSessionFactory;
|
||||||
|
final TransportPropertyManager transportPropertyManager;
|
||||||
|
|
||||||
|
SyncConnection(KeyManager keyManager, ConnectionRegistry connectionRegistry,
|
||||||
|
StreamReaderFactory streamReaderFactory,
|
||||||
|
StreamWriterFactory streamWriterFactory,
|
||||||
|
SyncSessionFactory syncSessionFactory,
|
||||||
|
TransportPropertyManager transportPropertyManager) {
|
||||||
|
super(keyManager, connectionRegistry, streamReaderFactory,
|
||||||
|
streamWriterFactory);
|
||||||
|
this.syncSessionFactory = syncSessionFactory;
|
||||||
|
this.transportPropertyManager = transportPropertyManager;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nullable
|
||||||
|
StreamContext allocateStreamContext(ContactId contactId,
|
||||||
|
TransportId transportId) {
|
||||||
|
try {
|
||||||
|
return keyManager.getStreamContext(contactId, transportId);
|
||||||
|
} catch (DbException e) {
|
||||||
|
logException(LOG, WARNING, e);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
SyncSession createIncomingSession(StreamContext ctx,
|
||||||
|
TransportConnectionReader r, PriorityHandler handler)
|
||||||
|
throws IOException {
|
||||||
|
InputStream streamReader = streamReaderFactory.createStreamReader(
|
||||||
|
r.getInputStream(), ctx);
|
||||||
|
ContactId c = requireNonNull(ctx.getContactId());
|
||||||
|
return syncSessionFactory
|
||||||
|
.createIncomingSession(c, streamReader, handler);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -7,6 +7,7 @@ import net.i2p.crypto.eddsa.KeyPairGenerator;
|
|||||||
import org.briarproject.bramble.api.crypto.AgreementPrivateKey;
|
import org.briarproject.bramble.api.crypto.AgreementPrivateKey;
|
||||||
import org.briarproject.bramble.api.crypto.AgreementPublicKey;
|
import org.briarproject.bramble.api.crypto.AgreementPublicKey;
|
||||||
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
||||||
|
import org.briarproject.bramble.api.crypto.DecryptionException;
|
||||||
import org.briarproject.bramble.api.crypto.KeyPair;
|
import org.briarproject.bramble.api.crypto.KeyPair;
|
||||||
import org.briarproject.bramble.api.crypto.KeyParser;
|
import org.briarproject.bramble.api.crypto.KeyParser;
|
||||||
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
||||||
@@ -39,6 +40,9 @@ import static java.lang.System.arraycopy;
|
|||||||
import static java.util.logging.Level.INFO;
|
import static java.util.logging.Level.INFO;
|
||||||
import static org.briarproject.bramble.api.crypto.CryptoConstants.KEY_TYPE_AGREEMENT;
|
import static org.briarproject.bramble.api.crypto.CryptoConstants.KEY_TYPE_AGREEMENT;
|
||||||
import static org.briarproject.bramble.api.crypto.CryptoConstants.KEY_TYPE_SIGNATURE;
|
import static org.briarproject.bramble.api.crypto.CryptoConstants.KEY_TYPE_SIGNATURE;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_CIPHERTEXT;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_PASSWORD;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.KEY_STRENGTHENER_ERROR;
|
||||||
import static org.briarproject.bramble.util.ByteUtils.INT_32_BYTES;
|
import static org.briarproject.bramble.util.ByteUtils.INT_32_BYTES;
|
||||||
import static org.briarproject.bramble.util.LogUtils.logDuration;
|
import static org.briarproject.bramble.util.LogUtils.logDuration;
|
||||||
import static org.briarproject.bramble.util.LogUtils.now;
|
import static org.briarproject.bramble.util.LogUtils.now;
|
||||||
@@ -359,16 +363,17 @@ class CryptoComponentImpl implements CryptoComponent {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@Nullable
|
|
||||||
public byte[] decryptWithPassword(byte[] input, String password,
|
public byte[] decryptWithPassword(byte[] input, String password,
|
||||||
@Nullable KeyStrengthener keyStrengthener) {
|
@Nullable KeyStrengthener keyStrengthener)
|
||||||
|
throws DecryptionException {
|
||||||
AuthenticatedCipher cipher = new XSalsa20Poly1305AuthenticatedCipher();
|
AuthenticatedCipher cipher = new XSalsa20Poly1305AuthenticatedCipher();
|
||||||
int macBytes = cipher.getMacBytes();
|
int macBytes = cipher.getMacBytes();
|
||||||
// The input contains the format version, salt, cost parameter, IV,
|
// The input contains the format version, salt, cost parameter, IV,
|
||||||
// ciphertext and MAC
|
// ciphertext and MAC
|
||||||
if (input.length < 1 + PBKDF_SALT_BYTES + INT_32_BYTES
|
if (input.length < 1 + PBKDF_SALT_BYTES + INT_32_BYTES
|
||||||
+ STORAGE_IV_BYTES + macBytes)
|
+ STORAGE_IV_BYTES + macBytes) {
|
||||||
return null; // Invalid input
|
throw new DecryptionException(INVALID_CIPHERTEXT);
|
||||||
|
}
|
||||||
int inputOff = 0;
|
int inputOff = 0;
|
||||||
// Format version
|
// Format version
|
||||||
byte formatVersion = input[inputOff];
|
byte formatVersion = input[inputOff];
|
||||||
@@ -376,7 +381,7 @@ class CryptoComponentImpl implements CryptoComponent {
|
|||||||
// Check whether we support this format version
|
// Check whether we support this format version
|
||||||
if (formatVersion != PBKDF_FORMAT_SCRYPT &&
|
if (formatVersion != PBKDF_FORMAT_SCRYPT &&
|
||||||
formatVersion != PBKDF_FORMAT_SCRYPT_STRENGTHENED) {
|
formatVersion != PBKDF_FORMAT_SCRYPT_STRENGTHENED) {
|
||||||
return null;
|
throw new DecryptionException(INVALID_CIPHERTEXT);
|
||||||
}
|
}
|
||||||
// Salt
|
// Salt
|
||||||
byte[] salt = new byte[PBKDF_SALT_BYTES];
|
byte[] salt = new byte[PBKDF_SALT_BYTES];
|
||||||
@@ -385,8 +390,9 @@ class CryptoComponentImpl implements CryptoComponent {
|
|||||||
// Cost parameter
|
// Cost parameter
|
||||||
long cost = ByteUtils.readUint32(input, inputOff);
|
long cost = ByteUtils.readUint32(input, inputOff);
|
||||||
inputOff += INT_32_BYTES;
|
inputOff += INT_32_BYTES;
|
||||||
if (cost < 2 || cost > Integer.MAX_VALUE)
|
if (cost < 2 || cost > Integer.MAX_VALUE) {
|
||||||
return null; // Invalid cost parameter
|
throw new DecryptionException(INVALID_CIPHERTEXT);
|
||||||
|
}
|
||||||
// IV
|
// IV
|
||||||
byte[] iv = new byte[STORAGE_IV_BYTES];
|
byte[] iv = new byte[STORAGE_IV_BYTES];
|
||||||
arraycopy(input, inputOff, iv, 0, iv.length);
|
arraycopy(input, inputOff, iv, 0, iv.length);
|
||||||
@@ -394,8 +400,10 @@ class CryptoComponentImpl implements CryptoComponent {
|
|||||||
// Derive the decryption key from the password
|
// Derive the decryption key from the password
|
||||||
SecretKey key = passwordBasedKdf.deriveKey(password, salt, (int) cost);
|
SecretKey key = passwordBasedKdf.deriveKey(password, salt, (int) cost);
|
||||||
if (formatVersion == PBKDF_FORMAT_SCRYPT_STRENGTHENED) {
|
if (formatVersion == PBKDF_FORMAT_SCRYPT_STRENGTHENED) {
|
||||||
if (keyStrengthener == null || !keyStrengthener.isInitialised())
|
if (keyStrengthener == null || !keyStrengthener.isInitialised()) {
|
||||||
return null; // Can't derive the same strengthened key
|
// Can't derive the same strengthened key
|
||||||
|
throw new DecryptionException(KEY_STRENGTHENER_ERROR);
|
||||||
|
}
|
||||||
key = keyStrengthener.strengthenKey(key);
|
key = keyStrengthener.strengthenKey(key);
|
||||||
}
|
}
|
||||||
// Initialise the cipher
|
// Initialise the cipher
|
||||||
@@ -411,7 +419,7 @@ class CryptoComponentImpl implements CryptoComponent {
|
|||||||
cipher.process(input, inputOff, inputLen, output, 0);
|
cipher.process(input, inputOff, inputLen, output, 0);
|
||||||
return output;
|
return output;
|
||||||
} catch (GeneralSecurityException e) {
|
} catch (GeneralSecurityException e) {
|
||||||
return null; // Invalid ciphertext
|
throw new DecryptionException(INVALID_PASSWORD);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -25,6 +25,7 @@ import static java.util.logging.Level.INFO;
|
|||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
import static java.util.logging.Logger.getLogger;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.db.JdbcUtils.tryToClose;
|
import static org.briarproject.bramble.db.JdbcUtils.tryToClose;
|
||||||
|
import static org.briarproject.bramble.util.IoUtils.isNonEmptyDirectory;
|
||||||
import static org.briarproject.bramble.util.LogUtils.logFileOrDir;
|
import static org.briarproject.bramble.util.LogUtils.logFileOrDir;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -69,8 +70,9 @@ class H2Database extends JdbcDatabase {
|
|||||||
LOG.info("Contents of account directory before opening DB:");
|
LOG.info("Contents of account directory before opening DB:");
|
||||||
logFileOrDir(LOG, INFO, dir.getParentFile());
|
logFileOrDir(LOG, INFO, dir.getParentFile());
|
||||||
}
|
}
|
||||||
boolean reopen = !dir.mkdirs();
|
boolean reopen = isNonEmptyDirectory(dir);
|
||||||
if (LOG.isLoggable(INFO)) LOG.info("Reopening DB: " + reopen);
|
if (LOG.isLoggable(INFO)) LOG.info("Reopening DB: " + reopen);
|
||||||
|
if (!reopen && dir.mkdirs()) LOG.info("Created database directory");
|
||||||
super.open("org.h2.Driver", reopen, key, listener);
|
super.open("org.h2.Driver", reopen, key, listener);
|
||||||
if (LOG.isLoggable(INFO)) {
|
if (LOG.isLoggable(INFO)) {
|
||||||
LOG.info("Contents of account directory after opening DB:");
|
LOG.info("Contents of account directory after opening DB:");
|
||||||
|
|||||||
@@ -20,9 +20,11 @@ import java.util.logging.Logger;
|
|||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
import javax.inject.Inject;
|
import javax.inject.Inject;
|
||||||
|
|
||||||
|
import static java.util.logging.Level.INFO;
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
import static java.util.logging.Logger.getLogger;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.db.JdbcUtils.tryToClose;
|
import static org.briarproject.bramble.db.JdbcUtils.tryToClose;
|
||||||
|
import static org.briarproject.bramble.util.IoUtils.isNonEmptyDirectory;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Contains all the HSQLDB-specific code for the database.
|
* Contains all the HSQLDB-specific code for the database.
|
||||||
@@ -64,7 +66,10 @@ class HyperSqlDatabase extends JdbcDatabase {
|
|||||||
public boolean open(SecretKey key, @Nullable MigrationListener listener)
|
public boolean open(SecretKey key, @Nullable MigrationListener listener)
|
||||||
throws DbException {
|
throws DbException {
|
||||||
this.key = key;
|
this.key = key;
|
||||||
boolean reopen = !config.getDatabaseDirectory().mkdirs();
|
File dir = config.getDatabaseDirectory();
|
||||||
|
boolean reopen = isNonEmptyDirectory(dir);
|
||||||
|
if (LOG.isLoggable(INFO)) LOG.info("Reopening DB: " + reopen);
|
||||||
|
if (!reopen && dir.mkdirs()) LOG.info("Created database directory");
|
||||||
super.open("org.hsqldb.jdbc.JDBCDriver", reopen, key, listener);
|
super.open("org.hsqldb.jdbc.JDBCDriver", reopen, key, listener);
|
||||||
return reopen;
|
return reopen;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,18 @@
|
|||||||
|
package org.briarproject.bramble.io;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
|
|
||||||
|
import javax.inject.Singleton;
|
||||||
|
|
||||||
|
import dagger.Module;
|
||||||
|
import dagger.Provides;
|
||||||
|
|
||||||
|
@Module
|
||||||
|
public class IoModule {
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
@Singleton
|
||||||
|
TimeoutMonitor provideTimeoutMonitor(TimeoutMonitorImpl timeoutMonitor) {
|
||||||
|
return timeoutMonitor;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,104 @@
|
|||||||
|
package org.briarproject.bramble.io;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.system.Clock;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.GuardedBy;
|
||||||
|
|
||||||
|
@NotNullByDefault
|
||||||
|
class TimeoutInputStream extends InputStream {
|
||||||
|
|
||||||
|
private final Clock clock;
|
||||||
|
private final InputStream in;
|
||||||
|
private final long timeoutMs;
|
||||||
|
private final CloseListener listener;
|
||||||
|
private final Object lock = new Object();
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private long readStartedMs = -1;
|
||||||
|
|
||||||
|
TimeoutInputStream(Clock clock, InputStream in, long timeoutMs,
|
||||||
|
CloseListener listener) {
|
||||||
|
this.clock = clock;
|
||||||
|
this.in = in;
|
||||||
|
this.timeoutMs = timeoutMs;
|
||||||
|
this.listener = listener;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int read() throws IOException {
|
||||||
|
synchronized (lock) {
|
||||||
|
readStartedMs = clock.currentTimeMillis();
|
||||||
|
}
|
||||||
|
int input = in.read();
|
||||||
|
synchronized (lock) {
|
||||||
|
readStartedMs = -1;
|
||||||
|
}
|
||||||
|
return input;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int read(byte[] b) throws IOException {
|
||||||
|
return read(b, 0, b.length);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int read(byte[] b, int off, int len) throws IOException {
|
||||||
|
synchronized (lock) {
|
||||||
|
readStartedMs = clock.currentTimeMillis();
|
||||||
|
}
|
||||||
|
int read = in.read(b, off, len);
|
||||||
|
synchronized (lock) {
|
||||||
|
readStartedMs = -1;
|
||||||
|
}
|
||||||
|
return read;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close() throws IOException {
|
||||||
|
try {
|
||||||
|
in.close();
|
||||||
|
} finally {
|
||||||
|
listener.onClose(this);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int available() throws IOException {
|
||||||
|
return in.available();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void mark(int readlimit) {
|
||||||
|
in.mark(readlimit);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean markSupported() {
|
||||||
|
return in.markSupported();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void reset() throws IOException {
|
||||||
|
in.reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long skip(long n) throws IOException {
|
||||||
|
return in.skip(n);
|
||||||
|
}
|
||||||
|
|
||||||
|
boolean hasTimedOut() {
|
||||||
|
synchronized (lock) {
|
||||||
|
return readStartedMs != -1 &&
|
||||||
|
clock.currentTimeMillis() - readStartedMs > timeoutMs;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
interface CloseListener {
|
||||||
|
|
||||||
|
void onClose(TimeoutInputStream closed);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,96 @@
|
|||||||
|
package org.briarproject.bramble.io;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
|
import org.briarproject.bramble.api.system.Clock;
|
||||||
|
import org.briarproject.bramble.api.system.Scheduler;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
import java.util.concurrent.Future;
|
||||||
|
import java.util.concurrent.ScheduledExecutorService;
|
||||||
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.GuardedBy;
|
||||||
|
import javax.inject.Inject;
|
||||||
|
|
||||||
|
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
||||||
|
import static java.util.concurrent.TimeUnit.SECONDS;
|
||||||
|
import static java.util.logging.Level.INFO;
|
||||||
|
import static java.util.logging.Logger.getLogger;
|
||||||
|
import static org.briarproject.bramble.util.LogUtils.logException;
|
||||||
|
|
||||||
|
class TimeoutMonitorImpl implements TimeoutMonitor {
|
||||||
|
|
||||||
|
private static final Logger LOG =
|
||||||
|
getLogger(TimeoutMonitorImpl.class.getName());
|
||||||
|
|
||||||
|
private static final long CHECK_INTERVAL_MS = SECONDS.toMillis(10);
|
||||||
|
|
||||||
|
private final ScheduledExecutorService scheduler;
|
||||||
|
private final Executor ioExecutor;
|
||||||
|
private final Clock clock;
|
||||||
|
private final Object lock = new Object();
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private final List<TimeoutInputStream> streams = new ArrayList<>();
|
||||||
|
|
||||||
|
@GuardedBy("lock")
|
||||||
|
private Future<?> task = null;
|
||||||
|
|
||||||
|
@Inject
|
||||||
|
TimeoutMonitorImpl(@Scheduler ScheduledExecutorService scheduler,
|
||||||
|
@IoExecutor Executor ioExecutor, Clock clock) {
|
||||||
|
this.scheduler = scheduler;
|
||||||
|
this.ioExecutor = ioExecutor;
|
||||||
|
this.clock = clock;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public InputStream createTimeoutInputStream(InputStream in,
|
||||||
|
long timeoutMs) {
|
||||||
|
TimeoutInputStream stream = new TimeoutInputStream(clock, in,
|
||||||
|
timeoutMs, this::removeStream);
|
||||||
|
synchronized (lock) {
|
||||||
|
if (streams.isEmpty()) {
|
||||||
|
task = scheduler.scheduleWithFixedDelay(this::checkTimeouts,
|
||||||
|
CHECK_INTERVAL_MS, CHECK_INTERVAL_MS, MILLISECONDS);
|
||||||
|
}
|
||||||
|
streams.add(stream);
|
||||||
|
}
|
||||||
|
return stream;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void removeStream(TimeoutInputStream stream) {
|
||||||
|
Future<?> toCancel = null;
|
||||||
|
synchronized (lock) {
|
||||||
|
if (streams.remove(stream) && streams.isEmpty()) {
|
||||||
|
toCancel = task;
|
||||||
|
task = null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (toCancel != null) toCancel.cancel(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Scheduler
|
||||||
|
private void checkTimeouts() {
|
||||||
|
ioExecutor.execute(() -> {
|
||||||
|
List<TimeoutInputStream> snapshot;
|
||||||
|
synchronized (lock) {
|
||||||
|
snapshot = new ArrayList<>(streams);
|
||||||
|
}
|
||||||
|
for (TimeoutInputStream stream : snapshot) {
|
||||||
|
if (stream.hasTimedOut()) {
|
||||||
|
LOG.info("Input stream has timed out");
|
||||||
|
try {
|
||||||
|
stream.close();
|
||||||
|
} catch (IOException e) {
|
||||||
|
logException(LOG, INFO, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,694 +0,0 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.contact.Contact;
|
|
||||||
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
|
||||||
import org.briarproject.bramble.api.contact.HandshakeManager;
|
|
||||||
import org.briarproject.bramble.api.contact.HandshakeManager.HandshakeResult;
|
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
|
||||||
import org.briarproject.bramble.api.db.DbException;
|
|
||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
|
||||||
import org.briarproject.bramble.api.sync.SyncSession;
|
|
||||||
import org.briarproject.bramble.api.sync.SyncSessionFactory;
|
|
||||||
import org.briarproject.bramble.api.transport.KeyManager;
|
|
||||||
import org.briarproject.bramble.api.transport.StreamContext;
|
|
||||||
import org.briarproject.bramble.api.transport.StreamReaderFactory;
|
|
||||||
import org.briarproject.bramble.api.transport.StreamWriter;
|
|
||||||
import org.briarproject.bramble.api.transport.StreamWriterFactory;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.util.concurrent.Executor;
|
|
||||||
import java.util.logging.Logger;
|
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
|
||||||
import javax.inject.Inject;
|
|
||||||
|
|
||||||
import static java.util.logging.Level.WARNING;
|
|
||||||
import static java.util.logging.Logger.getLogger;
|
|
||||||
import static org.briarproject.bramble.api.nullsafety.NullSafety.requireNonNull;
|
|
||||||
import static org.briarproject.bramble.api.transport.TransportConstants.TAG_LENGTH;
|
|
||||||
import static org.briarproject.bramble.util.IoUtils.read;
|
|
||||||
import static org.briarproject.bramble.util.LogUtils.logException;
|
|
||||||
|
|
||||||
@NotNullByDefault
|
|
||||||
class ConnectionManagerImpl implements ConnectionManager {
|
|
||||||
|
|
||||||
private static final Logger LOG =
|
|
||||||
getLogger(ConnectionManagerImpl.class.getName());
|
|
||||||
|
|
||||||
private final Executor ioExecutor;
|
|
||||||
private final KeyManager keyManager;
|
|
||||||
private final StreamReaderFactory streamReaderFactory;
|
|
||||||
private final StreamWriterFactory streamWriterFactory;
|
|
||||||
private final SyncSessionFactory syncSessionFactory;
|
|
||||||
private final HandshakeManager handshakeManager;
|
|
||||||
private final ContactExchangeManager contactExchangeManager;
|
|
||||||
private final ConnectionRegistry connectionRegistry;
|
|
||||||
|
|
||||||
@Inject
|
|
||||||
ConnectionManagerImpl(@IoExecutor Executor ioExecutor,
|
|
||||||
KeyManager keyManager, StreamReaderFactory streamReaderFactory,
|
|
||||||
StreamWriterFactory streamWriterFactory,
|
|
||||||
SyncSessionFactory syncSessionFactory,
|
|
||||||
HandshakeManager handshakeManager,
|
|
||||||
ContactExchangeManager contactExchangeManager,
|
|
||||||
ConnectionRegistry connectionRegistry) {
|
|
||||||
this.ioExecutor = ioExecutor;
|
|
||||||
this.keyManager = keyManager;
|
|
||||||
this.streamReaderFactory = streamReaderFactory;
|
|
||||||
this.streamWriterFactory = streamWriterFactory;
|
|
||||||
this.syncSessionFactory = syncSessionFactory;
|
|
||||||
this.handshakeManager = handshakeManager;
|
|
||||||
this.contactExchangeManager = contactExchangeManager;
|
|
||||||
this.connectionRegistry = connectionRegistry;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageIncomingConnection(TransportId t,
|
|
||||||
TransportConnectionReader r) {
|
|
||||||
ioExecutor.execute(new ManageIncomingSimplexConnection(t, r));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageIncomingConnection(TransportId t,
|
|
||||||
DuplexTransportConnection d) {
|
|
||||||
ioExecutor.execute(new ManageIncomingDuplexConnection(t, d));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageIncomingConnection(PendingContactId p, TransportId t,
|
|
||||||
DuplexTransportConnection d) {
|
|
||||||
ioExecutor.execute(new ManageIncomingHandshakeConnection(p, t, d));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageOutgoingConnection(ContactId c, TransportId t,
|
|
||||||
TransportConnectionWriter w) {
|
|
||||||
ioExecutor.execute(new ManageOutgoingSimplexConnection(c, t, w));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageOutgoingConnection(ContactId c, TransportId t,
|
|
||||||
DuplexTransportConnection d) {
|
|
||||||
ioExecutor.execute(new ManageOutgoingDuplexConnection(c, t, d));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void manageOutgoingConnection(PendingContactId p, TransportId t,
|
|
||||||
DuplexTransportConnection d) {
|
|
||||||
ioExecutor.execute(new ManageOutgoingHandshakeConnection(p, t, d));
|
|
||||||
}
|
|
||||||
|
|
||||||
private byte[] readTag(InputStream in) throws IOException {
|
|
||||||
byte[] tag = new byte[TAG_LENGTH];
|
|
||||||
read(in, tag);
|
|
||||||
return tag;
|
|
||||||
}
|
|
||||||
|
|
||||||
private SyncSession createIncomingSession(StreamContext ctx,
|
|
||||||
TransportConnectionReader r) throws IOException {
|
|
||||||
InputStream streamReader = streamReaderFactory.createStreamReader(
|
|
||||||
r.getInputStream(), ctx);
|
|
||||||
ContactId c = requireNonNull(ctx.getContactId());
|
|
||||||
return syncSessionFactory.createIncomingSession(c, streamReader);
|
|
||||||
}
|
|
||||||
|
|
||||||
private SyncSession createSimplexOutgoingSession(StreamContext ctx,
|
|
||||||
TransportConnectionWriter w) throws IOException {
|
|
||||||
StreamWriter streamWriter = streamWriterFactory.createStreamWriter(
|
|
||||||
w.getOutputStream(), ctx);
|
|
||||||
ContactId c = requireNonNull(ctx.getContactId());
|
|
||||||
return syncSessionFactory.createSimplexOutgoingSession(c,
|
|
||||||
w.getMaxLatency(), streamWriter);
|
|
||||||
}
|
|
||||||
|
|
||||||
private SyncSession createDuplexOutgoingSession(StreamContext ctx,
|
|
||||||
TransportConnectionWriter w) throws IOException {
|
|
||||||
StreamWriter streamWriter = streamWriterFactory.createStreamWriter(
|
|
||||||
w.getOutputStream(), ctx);
|
|
||||||
ContactId c = requireNonNull(ctx.getContactId());
|
|
||||||
return syncSessionFactory.createDuplexOutgoingSession(c,
|
|
||||||
w.getMaxLatency(), w.getMaxIdleTime(), streamWriter);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void disposeOnError(TransportConnectionReader reader,
|
|
||||||
boolean recognised) {
|
|
||||||
try {
|
|
||||||
reader.dispose(true, recognised);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void disposeOnError(TransportConnectionWriter writer) {
|
|
||||||
try {
|
|
||||||
writer.dispose(true);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageIncomingSimplexConnection implements Runnable {
|
|
||||||
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final TransportConnectionReader reader;
|
|
||||||
|
|
||||||
private ManageIncomingSimplexConnection(TransportId transportId,
|
|
||||||
TransportConnectionReader reader) {
|
|
||||||
this.transportId = transportId;
|
|
||||||
this.reader = reader;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Read and recognise the tag
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
byte[] tag = readTag(reader.getInputStream());
|
|
||||||
ctx = keyManager.getStreamContext(transportId, tag);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.info("Unrecognised tag");
|
|
||||||
onError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
ContactId contactId = ctx.getContactId();
|
|
||||||
if (contactId == null) {
|
|
||||||
LOG.warning("Received rendezvous stream, expected contact");
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx.isHandshakeMode()) {
|
|
||||||
// TODO: Support handshake mode for contacts
|
|
||||||
LOG.warning("Received handshake tag, expected rotation mode");
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
connectionRegistry.registerConnection(contactId, transportId, true);
|
|
||||||
try {
|
|
||||||
// Create and run the incoming session
|
|
||||||
createIncomingSession(ctx, reader).run();
|
|
||||||
reader.dispose(false, true);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError(true);
|
|
||||||
} finally {
|
|
||||||
connectionRegistry.unregisterConnection(contactId, transportId,
|
|
||||||
true);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onError(boolean recognised) {
|
|
||||||
disposeOnError(reader, recognised);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageOutgoingSimplexConnection implements Runnable {
|
|
||||||
|
|
||||||
private final ContactId contactId;
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final TransportConnectionWriter writer;
|
|
||||||
|
|
||||||
private ManageOutgoingSimplexConnection(ContactId contactId,
|
|
||||||
TransportId transportId, TransportConnectionWriter writer) {
|
|
||||||
this.contactId = contactId;
|
|
||||||
this.transportId = transportId;
|
|
||||||
this.writer = writer;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Allocate a stream context
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
ctx = keyManager.getStreamContext(contactId, transportId);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.warning("Could not allocate stream context");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
connectionRegistry.registerConnection(contactId, transportId,
|
|
||||||
false);
|
|
||||||
try {
|
|
||||||
// Create and run the outgoing session
|
|
||||||
createSimplexOutgoingSession(ctx, writer).run();
|
|
||||||
writer.dispose(false);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
} finally {
|
|
||||||
connectionRegistry.unregisterConnection(contactId, transportId,
|
|
||||||
false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onError() {
|
|
||||||
disposeOnError(writer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageIncomingDuplexConnection implements Runnable {
|
|
||||||
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final TransportConnectionReader reader;
|
|
||||||
private final TransportConnectionWriter writer;
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
private volatile SyncSession outgoingSession = null;
|
|
||||||
|
|
||||||
private ManageIncomingDuplexConnection(TransportId transportId,
|
|
||||||
DuplexTransportConnection connection) {
|
|
||||||
this.transportId = transportId;
|
|
||||||
reader = connection.getReader();
|
|
||||||
writer = connection.getWriter();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Read and recognise the tag
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
byte[] tag = readTag(reader.getInputStream());
|
|
||||||
ctx = keyManager.getStreamContext(transportId, tag);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onReadError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.info("Unrecognised tag");
|
|
||||||
onReadError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
ContactId contactId = ctx.getContactId();
|
|
||||||
if (contactId == null) {
|
|
||||||
LOG.warning("Expected contact tag, got rendezvous tag");
|
|
||||||
onReadError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx.isHandshakeMode()) {
|
|
||||||
// TODO: Support handshake mode for contacts
|
|
||||||
LOG.warning("Received handshake tag, expected rotation mode");
|
|
||||||
onReadError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
connectionRegistry.registerConnection(contactId, transportId, true);
|
|
||||||
// Start the outgoing session on another thread
|
|
||||||
ioExecutor.execute(() -> runOutgoingSession(contactId));
|
|
||||||
try {
|
|
||||||
// Create and run the incoming session
|
|
||||||
createIncomingSession(ctx, reader).run();
|
|
||||||
reader.dispose(false, true);
|
|
||||||
// Interrupt the outgoing session so it finishes cleanly
|
|
||||||
SyncSession out = outgoingSession;
|
|
||||||
if (out != null) out.interrupt();
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onReadError(true);
|
|
||||||
} finally {
|
|
||||||
connectionRegistry.unregisterConnection(contactId, transportId,
|
|
||||||
true);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void runOutgoingSession(ContactId contactId) {
|
|
||||||
// Allocate a stream context
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
ctx = keyManager.getStreamContext(contactId, transportId);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onWriteError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.warning("Could not allocate stream context");
|
|
||||||
onWriteError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
// Create and run the outgoing session
|
|
||||||
SyncSession out = createDuplexOutgoingSession(ctx, writer);
|
|
||||||
outgoingSession = out;
|
|
||||||
out.run();
|
|
||||||
writer.dispose(false);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onWriteError();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onReadError(boolean recognised) {
|
|
||||||
disposeOnError(reader, recognised);
|
|
||||||
disposeOnError(writer);
|
|
||||||
// Interrupt the outgoing session so it finishes
|
|
||||||
SyncSession out = outgoingSession;
|
|
||||||
if (out != null) out.interrupt();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onWriteError() {
|
|
||||||
disposeOnError(reader, true);
|
|
||||||
disposeOnError(writer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageOutgoingDuplexConnection implements Runnable {
|
|
||||||
|
|
||||||
private final ContactId contactId;
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final TransportConnectionReader reader;
|
|
||||||
private final TransportConnectionWriter writer;
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
private volatile SyncSession outgoingSession = null;
|
|
||||||
|
|
||||||
private ManageOutgoingDuplexConnection(ContactId contactId,
|
|
||||||
TransportId transportId, DuplexTransportConnection connection) {
|
|
||||||
this.contactId = contactId;
|
|
||||||
this.transportId = transportId;
|
|
||||||
reader = connection.getReader();
|
|
||||||
writer = connection.getWriter();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Allocate a stream context
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
ctx = keyManager.getStreamContext(contactId, transportId);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onWriteError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.warning("Could not allocate stream context");
|
|
||||||
onWriteError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx.isHandshakeMode()) {
|
|
||||||
// TODO: Support handshake mode for contacts
|
|
||||||
LOG.warning("Cannot use handshake mode stream context");
|
|
||||||
onWriteError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Start the incoming session on another thread
|
|
||||||
ioExecutor.execute(this::runIncomingSession);
|
|
||||||
try {
|
|
||||||
// Create and run the outgoing session
|
|
||||||
SyncSession out = createDuplexOutgoingSession(ctx, writer);
|
|
||||||
outgoingSession = out;
|
|
||||||
out.run();
|
|
||||||
writer.dispose(false);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onWriteError();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void runIncomingSession() {
|
|
||||||
// Read and recognise the tag
|
|
||||||
StreamContext ctx;
|
|
||||||
try {
|
|
||||||
byte[] tag = readTag(reader.getInputStream());
|
|
||||||
ctx = keyManager.getStreamContext(transportId, tag);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onReadError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Unrecognised tags are suspicious in this case
|
|
||||||
if (ctx == null) {
|
|
||||||
LOG.warning("Unrecognised tag for returning stream");
|
|
||||||
onReadError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Check that the stream comes from the expected contact
|
|
||||||
ContactId inContactId = ctx.getContactId();
|
|
||||||
if (inContactId == null) {
|
|
||||||
LOG.warning("Expected contact tag, got rendezvous tag");
|
|
||||||
onReadError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (!contactId.equals(inContactId)) {
|
|
||||||
LOG.warning("Wrong contact ID for returning stream");
|
|
||||||
onReadError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctx.isHandshakeMode()) {
|
|
||||||
// TODO: Support handshake mode for contacts
|
|
||||||
LOG.warning("Received handshake tag, expected rotation mode");
|
|
||||||
onReadError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
connectionRegistry.registerConnection(contactId, transportId,
|
|
||||||
false);
|
|
||||||
try {
|
|
||||||
// Create and run the incoming session
|
|
||||||
createIncomingSession(ctx, reader).run();
|
|
||||||
reader.dispose(false, true);
|
|
||||||
// Interrupt the outgoing session so it finishes cleanly
|
|
||||||
SyncSession out = outgoingSession;
|
|
||||||
if (out != null) out.interrupt();
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onReadError();
|
|
||||||
} finally {
|
|
||||||
connectionRegistry.unregisterConnection(contactId, transportId,
|
|
||||||
false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onReadError() {
|
|
||||||
// 'Recognised' is always true for outgoing connections
|
|
||||||
disposeOnError(reader, true);
|
|
||||||
disposeOnError(writer);
|
|
||||||
// Interrupt the outgoing session so it finishes
|
|
||||||
SyncSession out = outgoingSession;
|
|
||||||
if (out != null) out.interrupt();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onWriteError() {
|
|
||||||
disposeOnError(reader, true);
|
|
||||||
disposeOnError(writer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageIncomingHandshakeConnection implements Runnable {
|
|
||||||
|
|
||||||
private final PendingContactId pendingContactId;
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final DuplexTransportConnection connection;
|
|
||||||
private final TransportConnectionReader reader;
|
|
||||||
private final TransportConnectionWriter writer;
|
|
||||||
|
|
||||||
private ManageIncomingHandshakeConnection(
|
|
||||||
PendingContactId pendingContactId, TransportId transportId,
|
|
||||||
DuplexTransportConnection connection) {
|
|
||||||
this.pendingContactId = pendingContactId;
|
|
||||||
this.transportId = transportId;
|
|
||||||
this.connection = connection;
|
|
||||||
reader = connection.getReader();
|
|
||||||
writer = connection.getWriter();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Read and recognise the tag
|
|
||||||
StreamContext ctxIn;
|
|
||||||
try {
|
|
||||||
byte[] tag = readTag(reader.getInputStream());
|
|
||||||
ctxIn = keyManager.getStreamContext(transportId, tag);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctxIn == null) {
|
|
||||||
LOG.info("Unrecognised tag");
|
|
||||||
onError(false);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
PendingContactId inPendingContactId = ctxIn.getPendingContactId();
|
|
||||||
if (inPendingContactId == null) {
|
|
||||||
LOG.warning("Expected rendezvous tag, got contact tag");
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Allocate the outgoing stream context
|
|
||||||
StreamContext ctxOut;
|
|
||||||
try {
|
|
||||||
ctxOut = keyManager.getStreamContext(pendingContactId,
|
|
||||||
transportId);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctxOut == null) {
|
|
||||||
LOG.warning("Could not allocate stream context");
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Close the connection if it's redundant
|
|
||||||
if (!connectionRegistry.registerConnection(pendingContactId)) {
|
|
||||||
LOG.info("Redundant rendezvous connection");
|
|
||||||
onError(true);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Handshake and exchange contacts
|
|
||||||
try {
|
|
||||||
InputStream in = streamReaderFactory.createStreamReader(
|
|
||||||
reader.getInputStream(), ctxIn);
|
|
||||||
// Flush the output stream to send the outgoing stream header
|
|
||||||
StreamWriter out = streamWriterFactory.createStreamWriter(
|
|
||||||
writer.getOutputStream(), ctxOut);
|
|
||||||
out.getOutputStream().flush();
|
|
||||||
HandshakeResult result = handshakeManager.handshake(
|
|
||||||
pendingContactId, in, out);
|
|
||||||
Contact contact = contactExchangeManager.exchangeContacts(
|
|
||||||
pendingContactId, connection, result.getMasterKey(),
|
|
||||||
result.isAlice(), false);
|
|
||||||
connectionRegistry.unregisterConnection(pendingContactId, true);
|
|
||||||
// Reuse the connection as a transport connection
|
|
||||||
manageOutgoingConnection(contact.getId(), transportId,
|
|
||||||
connection);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError(true);
|
|
||||||
connectionRegistry.unregisterConnection(pendingContactId,
|
|
||||||
false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onError(boolean recognised) {
|
|
||||||
disposeOnError(reader, recognised);
|
|
||||||
disposeOnError(writer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ManageOutgoingHandshakeConnection implements Runnable {
|
|
||||||
|
|
||||||
private final PendingContactId pendingContactId;
|
|
||||||
private final TransportId transportId;
|
|
||||||
private final DuplexTransportConnection connection;
|
|
||||||
private final TransportConnectionReader reader;
|
|
||||||
private final TransportConnectionWriter writer;
|
|
||||||
|
|
||||||
private ManageOutgoingHandshakeConnection(
|
|
||||||
PendingContactId pendingContactId, TransportId transportId,
|
|
||||||
DuplexTransportConnection connection) {
|
|
||||||
this.pendingContactId = pendingContactId;
|
|
||||||
this.transportId = transportId;
|
|
||||||
this.connection = connection;
|
|
||||||
reader = connection.getReader();
|
|
||||||
writer = connection.getWriter();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void run() {
|
|
||||||
// Allocate the outgoing stream context
|
|
||||||
StreamContext ctxOut;
|
|
||||||
try {
|
|
||||||
ctxOut = keyManager.getStreamContext(pendingContactId,
|
|
||||||
transportId);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (ctxOut == null) {
|
|
||||||
LOG.warning("Could not allocate stream context");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Flush the output stream to send the outgoing stream header
|
|
||||||
StreamWriter out;
|
|
||||||
try {
|
|
||||||
out = streamWriterFactory.createStreamWriter(
|
|
||||||
writer.getOutputStream(), ctxOut);
|
|
||||||
out.getOutputStream().flush();
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Read and recognise the tag
|
|
||||||
StreamContext ctxIn;
|
|
||||||
try {
|
|
||||||
byte[] tag = readTag(reader.getInputStream());
|
|
||||||
ctxIn = keyManager.getStreamContext(transportId, tag);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Unrecognised tags are suspicious in this case
|
|
||||||
if (ctxIn == null) {
|
|
||||||
LOG.warning("Unrecognised tag for returning stream");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Check that the stream comes from the expected pending contact
|
|
||||||
PendingContactId inPendingContactId = ctxIn.getPendingContactId();
|
|
||||||
if (inPendingContactId == null) {
|
|
||||||
LOG.warning("Expected rendezvous tag, got contact tag");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
if (!inPendingContactId.equals(pendingContactId)) {
|
|
||||||
LOG.warning("Wrong pending contact ID for returning stream");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Close the connection if it's redundant
|
|
||||||
if (!connectionRegistry.registerConnection(pendingContactId)) {
|
|
||||||
LOG.info("Redundant rendezvous connection");
|
|
||||||
onError();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// Handshake and exchange contacts
|
|
||||||
try {
|
|
||||||
InputStream in = streamReaderFactory.createStreamReader(
|
|
||||||
reader.getInputStream(), ctxIn);
|
|
||||||
HandshakeResult result = handshakeManager.handshake(
|
|
||||||
pendingContactId, in, out);
|
|
||||||
Contact contact = contactExchangeManager.exchangeContacts(
|
|
||||||
pendingContactId, connection, result.getMasterKey(),
|
|
||||||
result.isAlice(), false);
|
|
||||||
connectionRegistry.unregisterConnection(pendingContactId, true);
|
|
||||||
// Reuse the connection as a transport connection
|
|
||||||
manageOutgoingConnection(contact.getId(), transportId,
|
|
||||||
connection);
|
|
||||||
} catch (IOException | DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
onError();
|
|
||||||
connectionRegistry.unregisterConnection(pendingContactId,
|
|
||||||
false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void onError() {
|
|
||||||
// 'Recognised' is always true for outgoing connections
|
|
||||||
disposeOnError(reader, true);
|
|
||||||
disposeOnError(writer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,150 +0,0 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.Multiset;
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
|
||||||
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Collection;
|
|
||||||
import java.util.Collections;
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.HashSet;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Set;
|
|
||||||
import java.util.logging.Logger;
|
|
||||||
|
|
||||||
import javax.annotation.concurrent.GuardedBy;
|
|
||||||
import javax.annotation.concurrent.ThreadSafe;
|
|
||||||
import javax.inject.Inject;
|
|
||||||
|
|
||||||
import static java.util.logging.Level.INFO;
|
|
||||||
import static java.util.logging.Logger.getLogger;
|
|
||||||
|
|
||||||
@ThreadSafe
|
|
||||||
@NotNullByDefault
|
|
||||||
class ConnectionRegistryImpl implements ConnectionRegistry {
|
|
||||||
|
|
||||||
private static final Logger LOG =
|
|
||||||
getLogger(ConnectionRegistryImpl.class.getName());
|
|
||||||
|
|
||||||
private final EventBus eventBus;
|
|
||||||
|
|
||||||
private final Object lock = new Object();
|
|
||||||
@GuardedBy("lock")
|
|
||||||
private final Map<TransportId, Multiset<ContactId>> contactConnections;
|
|
||||||
@GuardedBy("lock")
|
|
||||||
private final Multiset<ContactId> contactCounts;
|
|
||||||
@GuardedBy("lock")
|
|
||||||
private final Set<PendingContactId> connectedPendingContacts;
|
|
||||||
|
|
||||||
@Inject
|
|
||||||
ConnectionRegistryImpl(EventBus eventBus) {
|
|
||||||
this.eventBus = eventBus;
|
|
||||||
contactConnections = new HashMap<>();
|
|
||||||
contactCounts = new Multiset<>();
|
|
||||||
connectedPendingContacts = new HashSet<>();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void registerConnection(ContactId c, TransportId t,
|
|
||||||
boolean incoming) {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
if (incoming) LOG.info("Incoming connection registered: " + t);
|
|
||||||
else LOG.info("Outgoing connection registered: " + t);
|
|
||||||
}
|
|
||||||
boolean firstConnection = false;
|
|
||||||
synchronized (lock) {
|
|
||||||
Multiset<ContactId> m = contactConnections.get(t);
|
|
||||||
if (m == null) {
|
|
||||||
m = new Multiset<>();
|
|
||||||
contactConnections.put(t, m);
|
|
||||||
}
|
|
||||||
m.add(c);
|
|
||||||
if (contactCounts.add(c) == 1) firstConnection = true;
|
|
||||||
}
|
|
||||||
eventBus.broadcast(new ConnectionOpenedEvent(c, t, incoming));
|
|
||||||
if (firstConnection) {
|
|
||||||
LOG.info("Contact connected");
|
|
||||||
eventBus.broadcast(new ContactConnectedEvent(c));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void unregisterConnection(ContactId c, TransportId t,
|
|
||||||
boolean incoming) {
|
|
||||||
if (LOG.isLoggable(INFO)) {
|
|
||||||
if (incoming) LOG.info("Incoming connection unregistered: " + t);
|
|
||||||
else LOG.info("Outgoing connection unregistered: " + t);
|
|
||||||
}
|
|
||||||
boolean lastConnection = false;
|
|
||||||
synchronized (lock) {
|
|
||||||
Multiset<ContactId> m = contactConnections.get(t);
|
|
||||||
if (m == null || !m.contains(c))
|
|
||||||
throw new IllegalArgumentException();
|
|
||||||
m.remove(c);
|
|
||||||
if (contactCounts.remove(c) == 0) lastConnection = true;
|
|
||||||
}
|
|
||||||
eventBus.broadcast(new ConnectionClosedEvent(c, t, incoming));
|
|
||||||
if (lastConnection) {
|
|
||||||
LOG.info("Contact disconnected");
|
|
||||||
eventBus.broadcast(new ContactDisconnectedEvent(c));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Collection<ContactId> getConnectedContacts(TransportId t) {
|
|
||||||
synchronized (lock) {
|
|
||||||
Multiset<ContactId> m = contactConnections.get(t);
|
|
||||||
if (m == null) return Collections.emptyList();
|
|
||||||
List<ContactId> ids = new ArrayList<>(m.keySet());
|
|
||||||
if (LOG.isLoggable(INFO))
|
|
||||||
LOG.info(ids.size() + " contacts connected: " + t);
|
|
||||||
return ids;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean isConnected(ContactId c, TransportId t) {
|
|
||||||
synchronized (lock) {
|
|
||||||
Multiset<ContactId> m = contactConnections.get(t);
|
|
||||||
return m != null && m.contains(c);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean isConnected(ContactId c) {
|
|
||||||
synchronized (lock) {
|
|
||||||
return contactCounts.contains(c);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean registerConnection(PendingContactId p) {
|
|
||||||
boolean added;
|
|
||||||
synchronized (lock) {
|
|
||||||
added = connectedPendingContacts.add(p);
|
|
||||||
}
|
|
||||||
if (added) eventBus.broadcast(new RendezvousConnectionOpenedEvent(p));
|
|
||||||
return added;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void unregisterConnection(PendingContactId p, boolean success) {
|
|
||||||
synchronized (lock) {
|
|
||||||
if (!connectedPendingContacts.remove(p))
|
|
||||||
throw new IllegalArgumentException();
|
|
||||||
}
|
|
||||||
eventBus.broadcast(new RendezvousConnectionClosedEvent(p, success));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,12 +1,12 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
package org.briarproject.bramble.plugin;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
import org.briarproject.bramble.api.db.DbException;
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.lifecycle.Service;
|
import org.briarproject.bramble.api.lifecycle.Service;
|
||||||
import org.briarproject.bramble.api.lifecycle.ServiceException;
|
import org.briarproject.bramble.api.lifecycle.ServiceException;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.Plugin.State;
|
import org.briarproject.bramble.api.plugin.Plugin.State;
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
|
|||||||
@@ -3,8 +3,6 @@ package org.briarproject.bramble.plugin;
|
|||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.lifecycle.LifecycleManager;
|
import org.briarproject.bramble.api.lifecycle.LifecycleManager;
|
||||||
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginConfig;
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
import org.briarproject.bramble.api.plugin.PluginManager;
|
import org.briarproject.bramble.api.plugin.PluginManager;
|
||||||
|
|
||||||
@@ -29,20 +27,6 @@ public class PluginModule {
|
|||||||
return new BackoffFactoryImpl();
|
return new BackoffFactoryImpl();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Provides
|
|
||||||
@Singleton
|
|
||||||
ConnectionManager provideConnectionManager(
|
|
||||||
ConnectionManagerImpl connectionManager) {
|
|
||||||
return connectionManager;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
|
||||||
@Singleton
|
|
||||||
ConnectionRegistry provideConnectionRegistry(
|
|
||||||
ConnectionRegistryImpl connectionRegistry) {
|
|
||||||
return connectionRegistry;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
@Provides
|
||||||
@Singleton
|
@Singleton
|
||||||
PluginManager providePluginManager(LifecycleManager lifecycleManager,
|
PluginManager providePluginManager(LifecycleManager lifecycleManager,
|
||||||
|
|||||||
@@ -1,6 +1,8 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
package org.briarproject.bramble.plugin;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.Pair;
|
import org.briarproject.bramble.api.Pair;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
import org.briarproject.bramble.api.contact.event.ContactAddedEvent;
|
import org.briarproject.bramble.api.contact.event.ContactAddedEvent;
|
||||||
import org.briarproject.bramble.api.db.DbException;
|
import org.briarproject.bramble.api.db.DbException;
|
||||||
@@ -9,9 +11,6 @@ import org.briarproject.bramble.api.event.EventListener;
|
|||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.PluginManager;
|
import org.briarproject.bramble.api.plugin.PluginManager;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
@@ -33,7 +32,6 @@ import java.security.SecureRandom;
|
|||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Map.Entry;
|
import java.util.Map.Entry;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
@@ -100,8 +98,8 @@ class PollerImpl implements Poller, EventListener {
|
|||||||
ConnectionClosedEvent c = (ConnectionClosedEvent) e;
|
ConnectionClosedEvent c = (ConnectionClosedEvent) e;
|
||||||
// Reschedule polling, the polling interval may have decreased
|
// Reschedule polling, the polling interval may have decreased
|
||||||
reschedule(c.getTransportId());
|
reschedule(c.getTransportId());
|
||||||
if (!c.isIncoming()) {
|
// If an outgoing connection failed, try to reconnect
|
||||||
// Connect to the disconnected contact
|
if (!c.isIncoming() && c.isException()) {
|
||||||
connectToContact(c.getContactId(), c.getTransportId());
|
connectToContact(c.getContactId(), c.getTransportId());
|
||||||
}
|
}
|
||||||
} else if (e instanceof ConnectionOpenedEvent) {
|
} else if (e instanceof ConnectionOpenedEvent) {
|
||||||
@@ -212,20 +210,18 @@ class PollerImpl implements Poller, EventListener {
|
|||||||
@IoExecutor
|
@IoExecutor
|
||||||
private void poll(Plugin p) {
|
private void poll(Plugin p) {
|
||||||
TransportId t = p.getId();
|
TransportId t = p.getId();
|
||||||
if (LOG.isLoggable(INFO)) LOG.info("Polling " + t);
|
if (LOG.isLoggable(INFO)) LOG.info("Polling plugin " + t);
|
||||||
try {
|
try {
|
||||||
Map<ContactId, TransportProperties> remote =
|
Map<ContactId, TransportProperties> remote =
|
||||||
transportPropertyManager.getRemoteProperties(t);
|
transportPropertyManager.getRemoteProperties(t);
|
||||||
Collection<ContactId> connected =
|
Collection<ContactId> connected =
|
||||||
connectionRegistry.getConnectedContacts(t);
|
connectionRegistry.getConnectedOrBetterContacts(t);
|
||||||
List<Pair<TransportProperties, ConnectionHandler>> properties =
|
Collection<Pair<TransportProperties, ConnectionHandler>>
|
||||||
new ArrayList<>();
|
properties = new ArrayList<>();
|
||||||
for (Entry<ContactId, TransportProperties> e : remote.entrySet()) {
|
for (Entry<ContactId, TransportProperties> e : remote.entrySet()) {
|
||||||
ContactId c = e.getKey();
|
ContactId c = e.getKey();
|
||||||
if (!connected.contains(c)) {
|
if (!connected.contains(c))
|
||||||
ConnHandler handler = new ConnHandler(c, t);
|
properties.add(new Pair<>(e.getValue(), new Handler(c, t)));
|
||||||
properties.add(new Pair<>(e.getValue(), handler));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
if (!properties.isEmpty()) p.poll(properties);
|
if (!properties.isEmpty()) p.poll(properties);
|
||||||
} catch (DbException e) {
|
} catch (DbException e) {
|
||||||
@@ -233,30 +229,6 @@ class PollerImpl implements Poller, EventListener {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@IoExecutor
|
|
||||||
private void discover(DuplexPlugin p) {
|
|
||||||
TransportId t = p.getId();
|
|
||||||
if (LOG.isLoggable(INFO)) LOG.info("Discovering peers for " + t);
|
|
||||||
try {
|
|
||||||
Map<ContactId, TransportProperties> remote =
|
|
||||||
transportPropertyManager.getRemoteProperties(t);
|
|
||||||
Collection<ContactId> connected =
|
|
||||||
connectionRegistry.getConnectedContacts(t);
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> properties =
|
|
||||||
new ArrayList<>();
|
|
||||||
for (Entry<ContactId, TransportProperties> e : remote.entrySet()) {
|
|
||||||
ContactId c = e.getKey();
|
|
||||||
if (!connected.contains(c)) {
|
|
||||||
DiscoHandler handler = new DiscoHandler(c, p);
|
|
||||||
properties.add(new Pair<>(e.getValue(), handler));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!properties.isEmpty()) p.discoverPeers(properties);
|
|
||||||
} catch (DbException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private class ScheduledPollTask {
|
private class ScheduledPollTask {
|
||||||
|
|
||||||
private final PollTask task;
|
private final PollTask task;
|
||||||
@@ -296,23 +268,16 @@ class PollerImpl implements Poller, EventListener {
|
|||||||
int delay = plugin.getPollingInterval();
|
int delay = plugin.getPollingInterval();
|
||||||
if (randomiseNext) delay = (int) (delay * random.nextDouble());
|
if (randomiseNext) delay = (int) (delay * random.nextDouble());
|
||||||
schedule(plugin, delay, false);
|
schedule(plugin, delay, false);
|
||||||
// FIXME: Revert
|
|
||||||
if (plugin instanceof DuplexPlugin) {
|
|
||||||
DuplexPlugin d = (DuplexPlugin) plugin;
|
|
||||||
if (d.supportsDiscovery()) discover(d);
|
|
||||||
else poll(d);
|
|
||||||
} else {
|
|
||||||
poll(plugin);
|
poll(plugin);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
private class ConnHandler implements ConnectionHandler {
|
private class Handler implements ConnectionHandler {
|
||||||
|
|
||||||
private final ContactId contactId;
|
private final ContactId contactId;
|
||||||
private final TransportId transportId;
|
private final TransportId transportId;
|
||||||
|
|
||||||
private ConnHandler(ContactId contactId, TransportId transportId) {
|
private Handler(ContactId contactId, TransportId transportId) {
|
||||||
this.contactId = contactId;
|
this.contactId = contactId;
|
||||||
this.transportId = transportId;
|
this.transportId = transportId;
|
||||||
}
|
}
|
||||||
@@ -335,27 +300,4 @@ class PollerImpl implements Poller, EventListener {
|
|||||||
transportId, w);
|
transportId, w);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private class DiscoHandler implements DiscoveryHandler {
|
|
||||||
|
|
||||||
private final ContactId contactId;
|
|
||||||
private final DuplexPlugin plugin;
|
|
||||||
|
|
||||||
private DiscoHandler(ContactId contactId, DuplexPlugin plugin) {
|
|
||||||
this.contactId = contactId;
|
|
||||||
this.plugin = plugin;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void handleDevice(TransportProperties p) {
|
|
||||||
LOG.info("Discovered contact via " + plugin.getId());
|
|
||||||
ioExecutor.execute(() -> {
|
|
||||||
DuplexTransportConnection c = plugin.createConnection(p);
|
|
||||||
if (c != null) {
|
|
||||||
connectionManager.manageOutgoingConnection(contactId,
|
|
||||||
plugin.getId(), c);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,17 +23,9 @@ interface BluetoothConnectionLimiter {
|
|||||||
boolean canOpenContactConnection();
|
boolean canOpenContactConnection();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Informs the limiter that a contact connection has been opened. The
|
* Informs the limiter that the given connection has been opened.
|
||||||
* limiter may close the new connection if key agreement is in progress.
|
|
||||||
* <p/>
|
|
||||||
* Returns false if the limiter has closed the new connection.
|
|
||||||
*/
|
*/
|
||||||
boolean contactConnectionOpened(DuplexTransportConnection conn);
|
void connectionOpened(DuplexTransportConnection conn);
|
||||||
|
|
||||||
/**
|
|
||||||
* Informs the limiter that a key agreement connection has been opened.
|
|
||||||
*/
|
|
||||||
void keyAgreementConnectionOpened(DuplexTransportConnection conn);
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Informs the limiter that the given connection has been closed.
|
* Informs the limiter that the given connection has been closed.
|
||||||
|
|||||||
@@ -1,46 +1,48 @@
|
|||||||
package org.briarproject.bramble.plugin.bluetooth;
|
package org.briarproject.bramble.plugin.bluetooth;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.sync.event.CloseSyncConnectionsEvent;
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.logging.Logger;
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
|
import javax.annotation.concurrent.GuardedBy;
|
||||||
import javax.annotation.concurrent.ThreadSafe;
|
import javax.annotation.concurrent.ThreadSafe;
|
||||||
|
|
||||||
import static java.util.logging.Level.INFO;
|
import static java.util.logging.Level.INFO;
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.util.LogUtils.logException;
|
import static org.briarproject.bramble.api.plugin.BluetoothConstants.ID;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
@ThreadSafe
|
@ThreadSafe
|
||||||
class BluetoothConnectionLimiterImpl implements BluetoothConnectionLimiter {
|
class BluetoothConnectionLimiterImpl implements BluetoothConnectionLimiter {
|
||||||
|
|
||||||
private static final Logger LOG =
|
private static final Logger LOG =
|
||||||
Logger.getLogger(BluetoothConnectionLimiterImpl.class.getName());
|
getLogger(BluetoothConnectionLimiterImpl.class.getName());
|
||||||
|
|
||||||
|
private final EventBus eventBus;
|
||||||
|
|
||||||
private final Object lock = new Object();
|
private final Object lock = new Object();
|
||||||
// The following are locking: lock
|
@GuardedBy("lock")
|
||||||
private final LinkedList<DuplexTransportConnection> connections =
|
private final List<DuplexTransportConnection> connections =
|
||||||
new LinkedList<>();
|
new LinkedList<>();
|
||||||
|
@GuardedBy("lock")
|
||||||
private boolean keyAgreementInProgress = false;
|
private boolean keyAgreementInProgress = false;
|
||||||
|
|
||||||
|
BluetoothConnectionLimiterImpl(EventBus eventBus) {
|
||||||
|
this.eventBus = eventBus;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void keyAgreementStarted() {
|
public void keyAgreementStarted() {
|
||||||
List<DuplexTransportConnection> close;
|
|
||||||
synchronized (lock) {
|
synchronized (lock) {
|
||||||
keyAgreementInProgress = true;
|
keyAgreementInProgress = true;
|
||||||
close = new ArrayList<>(connections);
|
|
||||||
connections.clear();
|
|
||||||
}
|
}
|
||||||
if (LOG.isLoggable(INFO)) {
|
LOG.info("Key agreement started");
|
||||||
LOG.info("Key agreement started, closing " + close.size() +
|
eventBus.broadcast(new CloseSyncConnectionsEvent(ID));
|
||||||
" connections");
|
|
||||||
}
|
|
||||||
for (DuplexTransportConnection conn : close) tryToClose(conn);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -65,46 +67,24 @@ class BluetoothConnectionLimiterImpl implements BluetoothConnectionLimiter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean contactConnectionOpened(DuplexTransportConnection conn) {
|
public void connectionOpened(DuplexTransportConnection conn) {
|
||||||
boolean accept = true;
|
|
||||||
synchronized (lock) {
|
synchronized (lock) {
|
||||||
if (keyAgreementInProgress) {
|
|
||||||
LOG.info("Refusing contact connection during key agreement");
|
|
||||||
accept = false;
|
|
||||||
} else {
|
|
||||||
LOG.info("Accepting contact connection");
|
|
||||||
connections.add(conn);
|
connections.add(conn);
|
||||||
|
if (LOG.isLoggable(INFO)) {
|
||||||
|
LOG.info("Connection opened, " + connections.size() + " open");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (!accept) tryToClose(conn);
|
|
||||||
return accept;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void keyAgreementConnectionOpened(DuplexTransportConnection conn) {
|
|
||||||
synchronized (lock) {
|
|
||||||
LOG.info("Accepting key agreement connection");
|
|
||||||
connections.add(conn);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void tryToClose(DuplexTransportConnection conn) {
|
|
||||||
try {
|
|
||||||
conn.getWriter().dispose(false);
|
|
||||||
conn.getReader().dispose(false, false);
|
|
||||||
} catch (IOException e) {
|
|
||||||
logException(LOG, WARNING, e);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void connectionClosed(DuplexTransportConnection conn) {
|
public void connectionClosed(DuplexTransportConnection conn) {
|
||||||
synchronized (lock) {
|
synchronized (lock) {
|
||||||
connections.remove(conn);
|
connections.remove(conn);
|
||||||
if (LOG.isLoggable(INFO))
|
if (LOG.isLoggable(INFO)) {
|
||||||
LOG.info("Connection closed, " + connections.size() + " open");
|
LOG.info("Connection closed, " + connections.size() + " open");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void allConnectionsClosed() {
|
public void allConnectionsClosed() {
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import org.briarproject.bramble.api.Pair;
|
|||||||
import org.briarproject.bramble.api.data.BdfList;
|
import org.briarproject.bramble.api.data.BdfList;
|
||||||
import org.briarproject.bramble.api.event.Event;
|
import org.briarproject.bramble.api.event.Event;
|
||||||
import org.briarproject.bramble.api.event.EventListener;
|
import org.briarproject.bramble.api.event.EventListener;
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.keyagreement.KeyAgreementConnection;
|
import org.briarproject.bramble.api.keyagreement.KeyAgreementConnection;
|
||||||
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
||||||
import org.briarproject.bramble.api.keyagreement.event.KeyAgreementListeningEvent;
|
import org.briarproject.bramble.api.keyagreement.event.KeyAgreementListeningEvent;
|
||||||
@@ -15,7 +16,6 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.PluginException;
|
import org.briarproject.bramble.api.plugin.PluginException;
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
@@ -32,7 +32,7 @@ import org.briarproject.bramble.api.settings.event.SettingsUpdatedEvent;
|
|||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
import java.util.List;
|
import java.util.Collection;
|
||||||
import java.util.UUID;
|
import java.util.UUID;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
import java.util.concurrent.atomic.AtomicBoolean;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
@@ -68,12 +68,13 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
getLogger(BluetoothPlugin.class.getName());
|
getLogger(BluetoothPlugin.class.getName());
|
||||||
|
|
||||||
final BluetoothConnectionLimiter connectionLimiter;
|
final BluetoothConnectionLimiter connectionLimiter;
|
||||||
|
final TimeoutMonitor timeoutMonitor;
|
||||||
|
|
||||||
private final Executor ioExecutor;
|
private final Executor ioExecutor;
|
||||||
private final SecureRandom secureRandom;
|
private final SecureRandom secureRandom;
|
||||||
private final Backoff backoff;
|
private final Backoff backoff;
|
||||||
private final PluginCallback callback;
|
private final PluginCallback callback;
|
||||||
private final int maxLatency;
|
private final int maxLatency, maxIdleTime;
|
||||||
private final AtomicBoolean used = new AtomicBoolean(false);
|
private final AtomicBoolean used = new AtomicBoolean(false);
|
||||||
|
|
||||||
protected final PluginState state = new PluginState();
|
protected final PluginState state = new PluginState();
|
||||||
@@ -113,14 +114,17 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
abstract DuplexTransportConnection discoverAndConnect(String uuid);
|
abstract DuplexTransportConnection discoverAndConnect(String uuid);
|
||||||
|
|
||||||
BluetoothPlugin(BluetoothConnectionLimiter connectionLimiter,
|
BluetoothPlugin(BluetoothConnectionLimiter connectionLimiter,
|
||||||
Executor ioExecutor, SecureRandom secureRandom,
|
TimeoutMonitor timeoutMonitor, Executor ioExecutor,
|
||||||
Backoff backoff, PluginCallback callback, int maxLatency) {
|
SecureRandom secureRandom, Backoff backoff,
|
||||||
|
PluginCallback callback, int maxLatency, int maxIdleTime) {
|
||||||
this.connectionLimiter = connectionLimiter;
|
this.connectionLimiter = connectionLimiter;
|
||||||
|
this.timeoutMonitor = timeoutMonitor;
|
||||||
this.ioExecutor = ioExecutor;
|
this.ioExecutor = ioExecutor;
|
||||||
this.secureRandom = secureRandom;
|
this.secureRandom = secureRandom;
|
||||||
this.backoff = backoff;
|
this.backoff = backoff;
|
||||||
this.callback = callback;
|
this.callback = callback;
|
||||||
this.maxLatency = maxLatency;
|
this.maxLatency = maxLatency;
|
||||||
|
this.maxIdleTime = maxIdleTime;
|
||||||
}
|
}
|
||||||
|
|
||||||
void onAdapterEnabled() {
|
void onAdapterEnabled() {
|
||||||
@@ -153,8 +157,7 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public int getMaxIdleTime() {
|
public int getMaxIdleTime() {
|
||||||
// Bluetooth detects dead connections so we don't need keepalives
|
return maxIdleTime;
|
||||||
return Integer.MAX_VALUE;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -233,8 +236,8 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
LOG.info("Connection received");
|
LOG.info("Connection received");
|
||||||
|
connectionLimiter.connectionOpened(conn);
|
||||||
backoff.reset();
|
backoff.reset();
|
||||||
if (connectionLimiter.contactConnectionOpened(conn))
|
|
||||||
callback.handleConnection(conn);
|
callback.handleConnection(conn);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -267,8 +270,8 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void poll(
|
public void poll(Collection<Pair<TransportProperties, ConnectionHandler>>
|
||||||
List<Pair<TransportProperties, ConnectionHandler>> properties) {
|
properties) {
|
||||||
if (getState() != ACTIVE) return;
|
if (getState() != ACTIVE) return;
|
||||||
backoff.increment();
|
backoff.increment();
|
||||||
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
||||||
@@ -282,12 +285,9 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
String uuid = p.get(PROP_UUID);
|
String uuid = p.get(PROP_UUID);
|
||||||
if (isNullOrEmpty(uuid)) return;
|
if (isNullOrEmpty(uuid)) return;
|
||||||
ioExecutor.execute(() -> {
|
ioExecutor.execute(() -> {
|
||||||
if (getState() != ACTIVE) return;
|
|
||||||
if (!connectionLimiter.canOpenContactConnection()) return;
|
|
||||||
DuplexTransportConnection d = createConnection(p);
|
DuplexTransportConnection d = createConnection(p);
|
||||||
if (d != null) {
|
if (d != null) {
|
||||||
backoff.reset();
|
backoff.reset();
|
||||||
if (connectionLimiter.contactConnectionOpened(d))
|
|
||||||
h.handleConnection(d);
|
h.handleConnection(d);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
@@ -333,9 +333,8 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
String uuid = p.get(PROP_UUID);
|
String uuid = p.get(PROP_UUID);
|
||||||
if (isNullOrEmpty(uuid)) return null;
|
if (isNullOrEmpty(uuid)) return null;
|
||||||
DuplexTransportConnection conn = connect(address, uuid);
|
DuplexTransportConnection conn = connect(address, uuid);
|
||||||
if (conn == null) return null;
|
if (conn != null) connectionLimiter.connectionOpened(conn);
|
||||||
// TODO: Why don't we reset the backoff here?
|
return conn;
|
||||||
return connectionLimiter.contactConnectionOpened(conn) ? conn : null;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -391,7 +390,7 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
LOG.info("Connecting to key agreement UUID " + uuid);
|
LOG.info("Connecting to key agreement UUID " + uuid);
|
||||||
conn = connect(address, uuid);
|
conn = connect(address, uuid);
|
||||||
}
|
}
|
||||||
if (conn != null) connectionLimiter.keyAgreementConnectionOpened(conn);
|
if (conn != null) connectionLimiter.connectionOpened(conn);
|
||||||
return conn;
|
return conn;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -412,17 +411,6 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
throw new UnsupportedOperationException();
|
throw new UnsupportedOperationException();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean supportsDiscovery() {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void discoverPeers(
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> properties) {
|
|
||||||
throw new UnsupportedOperationException();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void eventOccurred(Event e) {
|
public void eventOccurred(Event e) {
|
||||||
if (e instanceof EnableBluetoothEvent) {
|
if (e instanceof EnableBluetoothEvent) {
|
||||||
@@ -471,7 +459,7 @@ abstract class BluetoothPlugin<SS> implements DuplexPlugin, EventListener {
|
|||||||
public KeyAgreementConnection accept() throws IOException {
|
public KeyAgreementConnection accept() throws IOException {
|
||||||
DuplexTransportConnection conn = acceptConnection(ss);
|
DuplexTransportConnection conn = acceptConnection(ss);
|
||||||
if (LOG.isLoggable(INFO)) LOG.info(ID + ": Incoming connection");
|
if (LOG.isLoggable(INFO)) LOG.info(ID + ": Incoming connection");
|
||||||
connectionLimiter.keyAgreementConnectionOpened(conn);
|
connectionLimiter.connectionOpened(conn);
|
||||||
return new KeyAgreementConnection(conn, ID);
|
return new KeyAgreementConnection(conn, ID);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ import org.briarproject.bramble.api.settings.Settings;
|
|||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.Inet4Address;
|
import java.net.Inet4Address;
|
||||||
|
import java.net.Inet6Address;
|
||||||
import java.net.InetAddress;
|
import java.net.InetAddress;
|
||||||
import java.net.InetSocketAddress;
|
import java.net.InetSocketAddress;
|
||||||
import java.net.InterfaceAddress;
|
import java.net.InterfaceAddress;
|
||||||
@@ -21,8 +22,9 @@ import java.net.ServerSocket;
|
|||||||
import java.net.Socket;
|
import java.net.Socket;
|
||||||
import java.net.UnknownHostException;
|
import java.net.UnknownHostException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
|
import java.util.Deque;
|
||||||
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Random;
|
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
import java.util.logging.Logger;
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
@@ -30,39 +32,48 @@ import javax.annotation.Nullable;
|
|||||||
|
|
||||||
import static java.lang.Integer.parseInt;
|
import static java.lang.Integer.parseInt;
|
||||||
import static java.util.Collections.addAll;
|
import static java.util.Collections.addAll;
|
||||||
|
import static java.util.Collections.emptyList;
|
||||||
import static java.util.Collections.sort;
|
import static java.util.Collections.sort;
|
||||||
import static java.util.logging.Level.INFO;
|
import static java.util.logging.Level.INFO;
|
||||||
import static java.util.logging.Level.WARNING;
|
import static java.util.logging.Level.WARNING;
|
||||||
import static java.util.logging.Logger.getLogger;
|
import static java.util.logging.Logger.getLogger;
|
||||||
import static org.briarproject.bramble.api.keyagreement.KeyAgreementConstants.TRANSPORT_ID_LAN;
|
import static org.briarproject.bramble.api.keyagreement.KeyAgreementConstants.TRANSPORT_ID_LAN;
|
||||||
import static org.briarproject.bramble.api.plugin.LanTcpConstants.ID;
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.ID;
|
||||||
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PREF_IPV6;
|
||||||
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PREF_LAN_IP_PORTS;
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PREF_LAN_IP_PORTS;
|
||||||
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PROP_IPV6;
|
||||||
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PROP_IP_PORTS;
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PROP_IP_PORTS;
|
||||||
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PROP_PORT;
|
import static org.briarproject.bramble.api.plugin.LanTcpConstants.PROP_PORT;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MAX_PROPERTY_LENGTH;
|
||||||
import static org.briarproject.bramble.util.ByteUtils.MAX_16_BIT_UNSIGNED;
|
import static org.briarproject.bramble.util.ByteUtils.MAX_16_BIT_UNSIGNED;
|
||||||
import static org.briarproject.bramble.util.IoUtils.tryToClose;
|
import static org.briarproject.bramble.util.IoUtils.tryToClose;
|
||||||
import static org.briarproject.bramble.util.PrivacyUtils.scrubSocketAddress;
|
import static org.briarproject.bramble.util.PrivacyUtils.scrubSocketAddress;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.fromHexString;
|
||||||
import static org.briarproject.bramble.util.StringUtils.isNullOrEmpty;
|
import static org.briarproject.bramble.util.StringUtils.isNullOrEmpty;
|
||||||
import static org.briarproject.bramble.util.StringUtils.join;
|
import static org.briarproject.bramble.util.StringUtils.join;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.toHexString;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.utf8IsTooLong;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
class LanTcpPlugin extends TcpPlugin {
|
class LanTcpPlugin extends TcpPlugin {
|
||||||
|
|
||||||
private static final Logger LOG = getLogger(LanTcpPlugin.class.getName());
|
private static final Logger LOG = getLogger(LanTcpPlugin.class.getName());
|
||||||
|
|
||||||
private static final int MAX_ADDRESSES = 4;
|
|
||||||
private static final String SEPARATOR = ",";
|
private static final String SEPARATOR = ",";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The IP address of an Android device providing a wifi access point.
|
* The IP address of an Android device providing a wifi access point.
|
||||||
|
* <p>
|
||||||
|
* Most devices use this address, but at least one device (Honor 8A) may
|
||||||
|
* use other addresses in the range 192.168.43.0/24.
|
||||||
*/
|
*/
|
||||||
protected static final InetAddress WIFI_AP_ADDRESS;
|
private static final InetAddress WIFI_AP_ADDRESS;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The IP address of an Android device providing a wifi direct
|
* The IP address of an Android device providing a wifi direct
|
||||||
* legacy mode access point.
|
* legacy mode access point.
|
||||||
*/
|
*/
|
||||||
protected static final InetAddress WIFI_DIRECT_AP_ADDRESS;
|
private static final InetAddress WIFI_DIRECT_AP_ADDRESS;
|
||||||
|
|
||||||
static {
|
static {
|
||||||
try {
|
try {
|
||||||
@@ -99,22 +110,22 @@ class LanTcpPlugin extends TcpPlugin {
|
|||||||
protected void initialisePortProperty() {
|
protected void initialisePortProperty() {
|
||||||
TransportProperties p = callback.getLocalProperties();
|
TransportProperties p = callback.getLocalProperties();
|
||||||
if (isNullOrEmpty(p.get(PROP_PORT))) {
|
if (isNullOrEmpty(p.get(PROP_PORT))) {
|
||||||
int port = new Random().nextInt(32768) + 32768;
|
int port = chooseEphemeralPort();
|
||||||
p.put(PROP_PORT, String.valueOf(port));
|
p.put(PROP_PORT, String.valueOf(port));
|
||||||
callback.mergeLocalProperties(p);
|
callback.mergeLocalProperties(p);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected List<InetSocketAddress> getLocalSocketAddresses() {
|
protected List<InetSocketAddress> getLocalSocketAddresses(boolean ipv4) {
|
||||||
TransportProperties p = callback.getLocalProperties();
|
TransportProperties p = callback.getLocalProperties();
|
||||||
int preferredPort = parsePortProperty(p.get(PROP_PORT));
|
int preferredPort = parsePortProperty(p.get(PROP_PORT));
|
||||||
String oldIpPorts = p.get(PROP_IP_PORTS);
|
String oldIpPorts = p.get(PROP_IP_PORTS);
|
||||||
List<InetSocketAddress> olds = parseSocketAddresses(oldIpPorts);
|
List<InetSocketAddress> olds = parseIpv4SocketAddresses(oldIpPorts);
|
||||||
|
|
||||||
List<InetSocketAddress> locals = new ArrayList<>();
|
List<InetSocketAddress> locals = new ArrayList<>();
|
||||||
List<InetSocketAddress> fallbacks = new ArrayList<>();
|
List<InetSocketAddress> fallbacks = new ArrayList<>();
|
||||||
for (InetAddress local : getUsableLocalInetAddresses()) {
|
for (InetAddress local : getUsableLocalInetAddresses(ipv4)) {
|
||||||
// If we've used this address before, try to use the same port
|
// If we've used this address before, try to use the same port
|
||||||
int port = preferredPort;
|
int port = preferredPort;
|
||||||
for (InetSocketAddress old : olds) {
|
for (InetSocketAddress old : olds) {
|
||||||
@@ -140,17 +151,17 @@ class LanTcpPlugin extends TcpPlugin {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private List<InetSocketAddress> parseSocketAddresses(String ipPorts) {
|
private List<InetSocketAddress> parseIpv4SocketAddresses(String ipPorts) {
|
||||||
List<InetSocketAddress> addresses = new ArrayList<>();
|
List<InetSocketAddress> addresses = new ArrayList<>();
|
||||||
if (isNullOrEmpty(ipPorts)) return addresses;
|
if (isNullOrEmpty(ipPorts)) return addresses;
|
||||||
for (String ipPort : ipPorts.split(SEPARATOR)) {
|
for (String ipPort : ipPorts.split(SEPARATOR)) {
|
||||||
InetSocketAddress a = parseSocketAddress(ipPort);
|
InetSocketAddress a = parseIpv4SocketAddress(ipPort);
|
||||||
if (a != null) addresses.add(a);
|
if (a != null) addresses.add(a);
|
||||||
}
|
}
|
||||||
return addresses;
|
return addresses;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected List<InetAddress> getUsableLocalInetAddresses() {
|
protected List<InetAddress> getUsableLocalInetAddresses(boolean ipv4) {
|
||||||
List<InterfaceAddress> ifAddrs =
|
List<InterfaceAddress> ifAddrs =
|
||||||
new ArrayList<>(getLocalInterfaceAddresses());
|
new ArrayList<>(getLocalInterfaceAddresses());
|
||||||
// Prefer longer network prefixes
|
// Prefer longer network prefixes
|
||||||
@@ -159,50 +170,74 @@ class LanTcpPlugin extends TcpPlugin {
|
|||||||
List<InetAddress> addrs = new ArrayList<>();
|
List<InetAddress> addrs = new ArrayList<>();
|
||||||
for (InterfaceAddress ifAddr : ifAddrs) {
|
for (InterfaceAddress ifAddr : ifAddrs) {
|
||||||
InetAddress addr = ifAddr.getAddress();
|
InetAddress addr = ifAddr.getAddress();
|
||||||
if (isAcceptableAddress(addr)) addrs.add(addr);
|
if (isAcceptableAddress(addr, ipv4)) addrs.add(addr);
|
||||||
}
|
}
|
||||||
return addrs;
|
return addrs;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void setLocalSocketAddress(InetSocketAddress a) {
|
protected void setLocalSocketAddress(InetSocketAddress a, boolean ipv4) {
|
||||||
|
if (ipv4) setLocalIpv4SocketAddress(a);
|
||||||
|
else setLocalIpv6SocketAddress(a);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void setLocalIpv4SocketAddress(InetSocketAddress a) {
|
||||||
String ipPort = getIpPortString(a);
|
String ipPort = getIpPortString(a);
|
||||||
|
updateRecentAddresses(PREF_LAN_IP_PORTS, PROP_IP_PORTS, ipPort);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void setLocalIpv6SocketAddress(InetSocketAddress a) {
|
||||||
|
String hex = toHexString(a.getAddress().getAddress());
|
||||||
|
updateRecentAddresses(PREF_IPV6, PROP_IPV6, hex);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void updateRecentAddresses(String settingKey, String propertyKey,
|
||||||
|
String item) {
|
||||||
// Get the list of recently used addresses
|
// Get the list of recently used addresses
|
||||||
String setting = callback.getSettings().get(PREF_LAN_IP_PORTS);
|
String setting = callback.getSettings().get(settingKey);
|
||||||
List<String> recent = new ArrayList<>();
|
Deque<String> recent = new LinkedList<>();
|
||||||
if (!isNullOrEmpty(setting))
|
if (!isNullOrEmpty(setting)) {
|
||||||
addAll(recent, setting.split(SEPARATOR));
|
addAll(recent, setting.split(SEPARATOR));
|
||||||
// Is the address already in the list?
|
}
|
||||||
if (recent.remove(ipPort)) {
|
if (recent.remove(item)) {
|
||||||
// Move the address to the start of the list
|
// Move the item to the start of the list
|
||||||
recent.add(0, ipPort);
|
recent.addFirst(item);
|
||||||
setting = join(recent, SEPARATOR);
|
setting = join(recent, SEPARATOR);
|
||||||
} else {
|
} else {
|
||||||
// Add the address to the start of the list
|
// Add the item to the start of the list
|
||||||
recent.add(0, ipPort);
|
recent.addFirst(item);
|
||||||
// Drop the least recently used address if the list is full
|
// Drop items from the end of the list if it's too long to encode
|
||||||
if (recent.size() > MAX_ADDRESSES)
|
|
||||||
recent = recent.subList(0, MAX_ADDRESSES);
|
|
||||||
setting = join(recent, SEPARATOR);
|
setting = join(recent, SEPARATOR);
|
||||||
|
while (utf8IsTooLong(setting, MAX_PROPERTY_LENGTH)) {
|
||||||
|
recent.removeLast();
|
||||||
|
setting = join(recent, SEPARATOR);
|
||||||
|
}
|
||||||
// Update the list of addresses shared with contacts
|
// Update the list of addresses shared with contacts
|
||||||
List<String> shared = new ArrayList<>(recent);
|
|
||||||
sort(shared);
|
|
||||||
String property = join(shared, SEPARATOR);
|
|
||||||
TransportProperties properties = new TransportProperties();
|
TransportProperties properties = new TransportProperties();
|
||||||
properties.put(PROP_IP_PORTS, property);
|
properties.put(propertyKey, setting);
|
||||||
callback.mergeLocalProperties(properties);
|
callback.mergeLocalProperties(properties);
|
||||||
}
|
}
|
||||||
// Save the setting
|
// Save the setting
|
||||||
Settings settings = new Settings();
|
Settings settings = new Settings();
|
||||||
settings.put(PREF_LAN_IP_PORTS, setting);
|
settings.put(settingKey, setting);
|
||||||
callback.mergeSettings(settings);
|
callback.mergeSettings(settings);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
protected boolean isIpv6LinkLocalAddress(InetAddress a) {
|
||||||
|
return a instanceof Inet6Address && a.isLinkLocalAddress();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected List<InetSocketAddress> getRemoteSocketAddresses(
|
protected List<InetSocketAddress> getRemoteSocketAddresses(
|
||||||
|
TransportProperties p, boolean ipv4) {
|
||||||
|
if (ipv4) return getRemoteIpv4SocketAddresses(p);
|
||||||
|
else return getRemoteIpv6SocketAddresses(p);
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<InetSocketAddress> getRemoteIpv4SocketAddresses(
|
||||||
TransportProperties p) {
|
TransportProperties p) {
|
||||||
String ipPorts = p.get(PROP_IP_PORTS);
|
String ipPorts = p.get(PROP_IP_PORTS);
|
||||||
List<InetSocketAddress> remotes = parseSocketAddresses(ipPorts);
|
List<InetSocketAddress> remotes = parseIpv4SocketAddresses(ipPorts);
|
||||||
int port = parsePortProperty(p.get(PROP_PORT));
|
int port = parsePortProperty(p.get(PROP_PORT));
|
||||||
// If the contact has a preferred port, we can guess their IP:port when
|
// If the contact has a preferred port, we can guess their IP:port when
|
||||||
// they're providing a wifi access point
|
// they're providing a wifi access point
|
||||||
@@ -217,20 +252,52 @@ class LanTcpPlugin extends TcpPlugin {
|
|||||||
return remotes;
|
return remotes;
|
||||||
}
|
}
|
||||||
|
|
||||||
private boolean isAcceptableAddress(InetAddress a) {
|
private List<InetSocketAddress> getRemoteIpv6SocketAddresses(
|
||||||
|
TransportProperties p) {
|
||||||
|
List<InetAddress> addrs = parseIpv6Addresses(p.get(PROP_IPV6));
|
||||||
|
int port = parsePortProperty(p.get(PROP_PORT));
|
||||||
|
if (addrs.isEmpty() || port == 0) return emptyList();
|
||||||
|
List<InetSocketAddress> remotes = new ArrayList<>();
|
||||||
|
for (InetAddress addr : addrs) {
|
||||||
|
remotes.add(new InetSocketAddress(addr, port));
|
||||||
|
}
|
||||||
|
return remotes;
|
||||||
|
}
|
||||||
|
|
||||||
|
private List<InetAddress> parseIpv6Addresses(String property) {
|
||||||
|
if (isNullOrEmpty(property)) return emptyList();
|
||||||
|
try {
|
||||||
|
List<InetAddress> addrs = new ArrayList<>();
|
||||||
|
for (String hex : property.split(SEPARATOR)) {
|
||||||
|
byte[] ip = fromHexString(hex);
|
||||||
|
if (ip.length == 16) addrs.add(InetAddress.getByAddress(ip));
|
||||||
|
}
|
||||||
|
return addrs;
|
||||||
|
} catch (IllegalArgumentException | UnknownHostException e) {
|
||||||
|
return emptyList();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean isAcceptableAddress(InetAddress a, boolean ipv4) {
|
||||||
|
if (ipv4) {
|
||||||
// Accept link-local and site-local IPv4 addresses
|
// Accept link-local and site-local IPv4 addresses
|
||||||
boolean ipv4 = a instanceof Inet4Address;
|
boolean isIpv4 = a instanceof Inet4Address;
|
||||||
boolean loop = a.isLoopbackAddress();
|
|
||||||
boolean link = a.isLinkLocalAddress();
|
boolean link = a.isLinkLocalAddress();
|
||||||
boolean site = a.isSiteLocalAddress();
|
boolean site = a.isSiteLocalAddress();
|
||||||
return ipv4 && !loop && (link || site);
|
return isIpv4 && (link || site);
|
||||||
|
} else {
|
||||||
|
// Accept link-local IPv6 addresses
|
||||||
|
return isIpv6LinkLocalAddress(a);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected boolean isConnectable(InterfaceAddress local,
|
protected boolean isConnectable(InterfaceAddress local,
|
||||||
InetSocketAddress remote) {
|
InetSocketAddress remote) {
|
||||||
if (remote.getPort() == 0) return false;
|
if (remote.getPort() == 0) return false;
|
||||||
if (!isAcceptableAddress(remote.getAddress())) return false;
|
InetAddress remoteAddress = remote.getAddress();
|
||||||
|
boolean ipv4 = local.getAddress() instanceof Inet4Address;
|
||||||
|
if (!isAcceptableAddress(remoteAddress, ipv4)) return false;
|
||||||
// Try to determine whether the address is on the same LAN as us
|
// Try to determine whether the address is on the same LAN as us
|
||||||
byte[] localIp = local.getAddress().getAddress();
|
byte[] localIp = local.getAddress().getAddress();
|
||||||
byte[] remoteIp = remote.getAddress().getAddress();
|
byte[] remoteIp = remote.getAddress().getAddress();
|
||||||
@@ -288,10 +355,17 @@ class LanTcpPlugin extends TcpPlugin {
|
|||||||
return new LanKeyAgreementListener(descriptor, ss);
|
return new LanKeyAgreementListener(descriptor, ss);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private List<InetSocketAddress> getLocalSocketAddresses() {
|
||||||
|
List<InetSocketAddress> addrs = new ArrayList<>();
|
||||||
|
addrs.addAll(getLocalSocketAddresses(true));
|
||||||
|
addrs.addAll(getLocalSocketAddresses(false));
|
||||||
|
return addrs;
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public DuplexTransportConnection createKeyAgreementConnection(
|
public DuplexTransportConnection createKeyAgreementConnection(
|
||||||
byte[] commitment, BdfList descriptor) {
|
byte[] commitment, BdfList descriptor) {
|
||||||
ServerSocket ss = state.getServerSocket();
|
ServerSocket ss = state.getServerSocket(true);
|
||||||
if (ss == null) return null;
|
if (ss == null) return null;
|
||||||
InterfaceAddress local = getLocalInterfaceAddress(ss.getInetAddress());
|
InterfaceAddress local = getLocalInterfaceAddress(ss.getInetAddress());
|
||||||
if (local == null) {
|
if (local == null) {
|
||||||
|
|||||||
@@ -54,11 +54,13 @@ class PortMapperImpl implements PortMapper {
|
|||||||
shutdownManager.addShutdownHook(() -> deleteMapping(port));
|
shutdownManager.addShutdownHook(() -> deleteMapping(port));
|
||||||
}
|
}
|
||||||
String externalString = gateway.getExternalIPAddress();
|
String externalString = gateway.getExternalIPAddress();
|
||||||
if (LOG.isLoggable(INFO))
|
if (externalString == null) {
|
||||||
LOG.info(
|
LOG.info("External address not available");
|
||||||
"External address " + scrubInetAddress(externalString));
|
} else {
|
||||||
if (externalString != null)
|
|
||||||
external = InetAddress.getByName(externalString);
|
external = InetAddress.getByName(externalString);
|
||||||
|
if (LOG.isLoggable(INFO))
|
||||||
|
LOG.info("External address " + scrubInetAddress(external));
|
||||||
|
}
|
||||||
} catch (IOException | SAXException e) {
|
} catch (IOException | SAXException e) {
|
||||||
logException(LOG, WARNING, e);
|
logException(LOG, WARNING, e);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,15 +2,16 @@ package org.briarproject.bramble.plugin.tcp;
|
|||||||
|
|
||||||
import org.briarproject.bramble.PoliteExecutor;
|
import org.briarproject.bramble.PoliteExecutor;
|
||||||
import org.briarproject.bramble.api.Pair;
|
import org.briarproject.bramble.api.Pair;
|
||||||
|
import org.briarproject.bramble.api.data.BdfList;
|
||||||
import org.briarproject.bramble.api.event.Event;
|
import org.briarproject.bramble.api.event.Event;
|
||||||
import org.briarproject.bramble.api.event.EventListener;
|
import org.briarproject.bramble.api.event.EventListener;
|
||||||
|
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexPlugin;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexPlugin;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
@@ -77,20 +78,22 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
* Returns zero or more socket addresses on which the plugin should listen,
|
* Returns zero or more socket addresses on which the plugin should listen,
|
||||||
* in order of preference. At most one of the addresses will be bound.
|
* in order of preference. At most one of the addresses will be bound.
|
||||||
*/
|
*/
|
||||||
protected abstract List<InetSocketAddress> getLocalSocketAddresses();
|
protected abstract List<InetSocketAddress> getLocalSocketAddresses(
|
||||||
|
boolean ipv4);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Adds the address on which the plugin is listening to the transport
|
* Adds the address on which the plugin is listening to the transport
|
||||||
* properties.
|
* properties.
|
||||||
*/
|
*/
|
||||||
protected abstract void setLocalSocketAddress(InetSocketAddress a);
|
protected abstract void setLocalSocketAddress(InetSocketAddress a,
|
||||||
|
boolean ipv4);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns zero or more socket addresses for connecting to a contact with
|
* Returns zero or more socket addresses for connecting to a contact with
|
||||||
* the given transport properties.
|
* the given transport properties.
|
||||||
*/
|
*/
|
||||||
protected abstract List<InetSocketAddress> getRemoteSocketAddresses(
|
protected abstract List<InetSocketAddress> getRemoteSocketAddresses(
|
||||||
TransportProperties p);
|
TransportProperties p, boolean ipv4);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns true if connections to the given address can be attempted.
|
* Returns true if connections to the given address can be attempted.
|
||||||
@@ -135,8 +138,14 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
protected void bind() {
|
protected void bind() {
|
||||||
bindExecutor.execute(() -> {
|
bindExecutor.execute(() -> {
|
||||||
if (getState() != INACTIVE) return;
|
if (getState() != INACTIVE) return;
|
||||||
|
bind(true);
|
||||||
|
bind(false);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
private void bind(boolean ipv4) {
|
||||||
ServerSocket ss = null;
|
ServerSocket ss = null;
|
||||||
for (InetSocketAddress addr : getLocalSocketAddresses()) {
|
for (InetSocketAddress addr : getLocalSocketAddresses(ipv4)) {
|
||||||
try {
|
try {
|
||||||
ss = new ServerSocket();
|
ss = new ServerSocket();
|
||||||
ss.bind(addr);
|
ss.bind(addr);
|
||||||
@@ -151,7 +160,7 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
LOG.info("Could not bind server socket");
|
LOG.info("Could not bind server socket");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (!state.setServerSocket(ss)) {
|
if (!state.setServerSocket(ss, ipv4)) {
|
||||||
LOG.info("Closing redundant server socket");
|
LOG.info("Closing redundant server socket");
|
||||||
tryToClose(ss, LOG, WARNING);
|
tryToClose(ss, LOG, WARNING);
|
||||||
return;
|
return;
|
||||||
@@ -159,11 +168,11 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
backoff.reset();
|
backoff.reset();
|
||||||
InetSocketAddress local =
|
InetSocketAddress local =
|
||||||
(InetSocketAddress) ss.getLocalSocketAddress();
|
(InetSocketAddress) ss.getLocalSocketAddress();
|
||||||
setLocalSocketAddress(local);
|
setLocalSocketAddress(local, ipv4);
|
||||||
if (LOG.isLoggable(INFO))
|
if (LOG.isLoggable(INFO))
|
||||||
LOG.info("Listening on " + scrubSocketAddress(local));
|
LOG.info("Listening on " + scrubSocketAddress(local));
|
||||||
acceptContactConnections(ss);
|
ServerSocket finalSocket = ss;
|
||||||
});
|
ioExecutor.execute(() -> acceptContactConnections(finalSocket, ipv4));
|
||||||
}
|
}
|
||||||
|
|
||||||
String getIpPortString(InetSocketAddress a) {
|
String getIpPortString(InetSocketAddress a) {
|
||||||
@@ -173,7 +182,7 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
return addr + ":" + a.getPort();
|
return addr + ":" + a.getPort();
|
||||||
}
|
}
|
||||||
|
|
||||||
private void acceptContactConnections(ServerSocket ss) {
|
private void acceptContactConnections(ServerSocket ss, boolean ipv4) {
|
||||||
while (true) {
|
while (true) {
|
||||||
Socket s;
|
Socket s;
|
||||||
try {
|
try {
|
||||||
@@ -182,12 +191,13 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
// This is expected when the server socket is closed
|
// This is expected when the server socket is closed
|
||||||
LOG.info("Server socket closed");
|
LOG.info("Server socket closed");
|
||||||
state.clearServerSocket(ss);
|
state.clearServerSocket(ss, ipv4);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (LOG.isLoggable(INFO))
|
if (LOG.isLoggable(INFO)) {
|
||||||
LOG.info("Connection from " +
|
LOG.info("Connection from " +
|
||||||
scrubSocketAddress(s.getRemoteSocketAddress()));
|
scrubSocketAddress(s.getRemoteSocketAddress()));
|
||||||
|
}
|
||||||
backoff.reset();
|
backoff.reset();
|
||||||
callback.handleConnection(new TcpTransportConnection(this, s));
|
callback.handleConnection(new TcpTransportConnection(this, s));
|
||||||
}
|
}
|
||||||
@@ -195,8 +205,7 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void stop() {
|
public void stop() {
|
||||||
ServerSocket ss = state.setStopped();
|
for (ServerSocket ss : state.setStopped()) tryToClose(ss, LOG, WARNING);
|
||||||
tryToClose(ss, LOG, WARNING);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -220,8 +229,8 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void poll(
|
public void poll(Collection<Pair<TransportProperties, ConnectionHandler>>
|
||||||
List<Pair<TransportProperties, ConnectionHandler>> properties) {
|
properties) {
|
||||||
if (getState() != ACTIVE) return;
|
if (getState() != ACTIVE) return;
|
||||||
backoff.increment();
|
backoff.increment();
|
||||||
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
||||||
@@ -241,14 +250,22 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public DuplexTransportConnection createConnection(TransportProperties p) {
|
public DuplexTransportConnection createConnection(TransportProperties p) {
|
||||||
ServerSocket ss = state.getServerSocket();
|
DuplexTransportConnection c = createConnection(p, true);
|
||||||
|
if (c != null) return c;
|
||||||
|
return createConnection(p, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nullable
|
||||||
|
private DuplexTransportConnection createConnection(TransportProperties p,
|
||||||
|
boolean ipv4) {
|
||||||
|
ServerSocket ss = state.getServerSocket(ipv4);
|
||||||
if (ss == null) return null;
|
if (ss == null) return null;
|
||||||
InterfaceAddress local = getLocalInterfaceAddress(ss.getInetAddress());
|
InterfaceAddress local = getLocalInterfaceAddress(ss.getInetAddress());
|
||||||
if (local == null) {
|
if (local == null) {
|
||||||
LOG.warning("No interface for server socket");
|
LOG.warning("No interface for server socket");
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
for (InetSocketAddress remote : getRemoteSocketAddresses(p)) {
|
for (InetSocketAddress remote : getRemoteSocketAddresses(p, ipv4)) {
|
||||||
// Don't try to connect to our own address
|
// Don't try to connect to our own address
|
||||||
if (!canConnectToOwnAddress() &&
|
if (!canConnectToOwnAddress() &&
|
||||||
remote.getAddress().equals(ss.getInetAddress())) {
|
remote.getAddress().equals(ss.getInetAddress())) {
|
||||||
@@ -273,11 +290,12 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
LOG.info("Connected to " + scrubSocketAddress(remote));
|
LOG.info("Connected to " + scrubSocketAddress(remote));
|
||||||
return new TcpTransportConnection(this, s);
|
return new TcpTransportConnection(this, s);
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
if (LOG.isLoggable(INFO))
|
if (LOG.isLoggable(INFO)) {
|
||||||
LOG.info("Could not connect to " +
|
LOG.info("Could not connect to " +
|
||||||
scrubSocketAddress(remote));
|
scrubSocketAddress(remote));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -298,8 +316,12 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
return new Socket();
|
return new Socket();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int chooseEphemeralPort() {
|
||||||
|
return 32768 + (int) (Math.random() * 32768);
|
||||||
|
}
|
||||||
|
|
||||||
@Nullable
|
@Nullable
|
||||||
InetSocketAddress parseSocketAddress(String ipPort) {
|
InetSocketAddress parseIpv4SocketAddress(String ipPort) {
|
||||||
if (isNullOrEmpty(ipPort)) return null;
|
if (isNullOrEmpty(ipPort)) return null;
|
||||||
String[] split = ipPort.split(":");
|
String[] split = ipPort.split(":");
|
||||||
if (split.length != 2) return null;
|
if (split.length != 2) return null;
|
||||||
@@ -310,18 +332,27 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
InetAddress a = InetAddress.getByName(addr);
|
InetAddress a = InetAddress.getByName(addr);
|
||||||
int p = Integer.parseInt(port);
|
int p = Integer.parseInt(port);
|
||||||
return new InetSocketAddress(a, p);
|
return new InetSocketAddress(a, p);
|
||||||
} catch (UnknownHostException e) {
|
} catch (UnknownHostException | NumberFormatException e) {
|
||||||
if (LOG.isLoggable(WARNING))
|
|
||||||
// not scrubbing to enable us to find the problem
|
|
||||||
LOG.warning("Invalid address: " + addr);
|
|
||||||
return null;
|
|
||||||
} catch (NumberFormatException e) {
|
|
||||||
if (LOG.isLoggable(WARNING))
|
|
||||||
LOG.warning("Invalid port: " + port);
|
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean supportsKeyAgreement() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public KeyAgreementListener createKeyAgreementListener(byte[] commitment) {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DuplexTransportConnection createKeyAgreementConnection(
|
||||||
|
byte[] commitment, BdfList descriptor) {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean supportsRendezvous() {
|
public boolean supportsRendezvous() {
|
||||||
return false;
|
return false;
|
||||||
@@ -333,17 +364,6 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
throw new UnsupportedOperationException();
|
throw new UnsupportedOperationException();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean supportsDiscovery() {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void discoverPeers(
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> properties) {
|
|
||||||
throw new UnsupportedOperationException();
|
|
||||||
}
|
|
||||||
|
|
||||||
List<InterfaceAddress> getLocalInterfaceAddresses() {
|
List<InterfaceAddress> getLocalInterfaceAddresses() {
|
||||||
List<InterfaceAddress> addrs = new ArrayList<>();
|
List<InterfaceAddress> addrs = new ArrayList<>();
|
||||||
for (NetworkInterface iface : getNetworkInterfaces()) {
|
for (NetworkInterface iface : getNetworkInterfaces()) {
|
||||||
@@ -383,13 +403,13 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
@IoExecutor
|
@IoExecutor
|
||||||
private void onSettingsUpdated(Settings settings) {
|
private void onSettingsUpdated(Settings settings) {
|
||||||
boolean enabledByUser = settings.getBoolean(PREF_PLUGIN_ENABLE, false);
|
boolean enabledByUser = settings.getBoolean(PREF_PLUGIN_ENABLE, false);
|
||||||
ServerSocket ss = state.setEnabledByUser(enabledByUser);
|
List<ServerSocket> toClose = state.setEnabledByUser(enabledByUser);
|
||||||
State s = getState();
|
State s = getState();
|
||||||
if (ss != null) {
|
if (!toClose.isEmpty()) {
|
||||||
LOG.info("Disabled by user, closing server socket");
|
LOG.info("Disabled by user, closing server sockets");
|
||||||
tryToClose(ss, LOG, WARNING);
|
for (ServerSocket ss : toClose) tryToClose(ss, LOG, WARNING);
|
||||||
} else if (s == INACTIVE) {
|
} else if (s == INACTIVE) {
|
||||||
LOG.info("Enabled by user, opening server socket");
|
LOG.info("Enabled by user, opening server sockets");
|
||||||
bind();
|
bind();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -403,7 +423,7 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
|
|
||||||
@GuardedBy("this")
|
@GuardedBy("this")
|
||||||
@Nullable
|
@Nullable
|
||||||
private ServerSocket serverSocket = null;
|
private ServerSocket serverSocketV4 = null, serverSocketV6 = null;
|
||||||
|
|
||||||
synchronized void setStarted(boolean enabledByUser) {
|
synchronized void setStarted(boolean enabledByUser) {
|
||||||
started = true;
|
started = true;
|
||||||
@@ -411,48 +431,68 @@ abstract class TcpPlugin implements DuplexPlugin, EventListener {
|
|||||||
callback.pluginStateChanged(getState());
|
callback.pluginStateChanged(getState());
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
synchronized List<ServerSocket> setStopped() {
|
||||||
synchronized ServerSocket setStopped() {
|
|
||||||
stopped = true;
|
stopped = true;
|
||||||
ServerSocket ss = serverSocket;
|
List<ServerSocket> toClose = clearServerSockets();
|
||||||
serverSocket = null;
|
|
||||||
callback.pluginStateChanged(getState());
|
callback.pluginStateChanged(getState());
|
||||||
return ss;
|
return toClose;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
@GuardedBy("this")
|
||||||
synchronized ServerSocket setEnabledByUser(boolean enabledByUser) {
|
private List<ServerSocket> clearServerSockets() {
|
||||||
|
List<ServerSocket> toClose = new ArrayList<>(2);
|
||||||
|
if (serverSocketV4 != null) {
|
||||||
|
toClose.add(serverSocketV4);
|
||||||
|
serverSocketV4 = null;
|
||||||
|
}
|
||||||
|
if (serverSocketV6 != null) {
|
||||||
|
toClose.add(serverSocketV6);
|
||||||
|
serverSocketV6 = null;
|
||||||
|
}
|
||||||
|
return toClose;
|
||||||
|
}
|
||||||
|
|
||||||
|
synchronized List<ServerSocket> setEnabledByUser(
|
||||||
|
boolean enabledByUser) {
|
||||||
this.enabledByUser = enabledByUser;
|
this.enabledByUser = enabledByUser;
|
||||||
ServerSocket ss = null;
|
List<ServerSocket> toClose = enabledByUser
|
||||||
if (!enabledByUser) {
|
? emptyList() : clearServerSockets();
|
||||||
ss = serverSocket;
|
|
||||||
serverSocket = null;
|
|
||||||
}
|
|
||||||
callback.pluginStateChanged(getState());
|
callback.pluginStateChanged(getState());
|
||||||
return ss;
|
return toClose;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nullable
|
@Nullable
|
||||||
synchronized ServerSocket getServerSocket() {
|
synchronized ServerSocket getServerSocket(boolean ipv4) {
|
||||||
return serverSocket;
|
return ipv4 ? serverSocketV4 : serverSocketV6;
|
||||||
}
|
}
|
||||||
|
|
||||||
synchronized boolean setServerSocket(ServerSocket ss) {
|
synchronized boolean setServerSocket(ServerSocket ss, boolean ipv4) {
|
||||||
if (stopped || serverSocket != null) return false;
|
if (stopped) return false;
|
||||||
serverSocket = ss;
|
if (ipv4) {
|
||||||
|
if (serverSocketV4 != null) return false;
|
||||||
|
serverSocketV4 = ss;
|
||||||
|
} else {
|
||||||
|
if (serverSocketV6 != null) return false;
|
||||||
|
serverSocketV6 = ss;
|
||||||
|
}
|
||||||
callback.pluginStateChanged(getState());
|
callback.pluginStateChanged(getState());
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
synchronized void clearServerSocket(ServerSocket ss) {
|
synchronized void clearServerSocket(ServerSocket ss, boolean ipv4) {
|
||||||
if (serverSocket == ss) serverSocket = null;
|
if (ipv4) {
|
||||||
|
if (serverSocketV4 == ss) serverSocketV4 = null;
|
||||||
|
} else {
|
||||||
|
if (serverSocketV6 == ss) serverSocketV6 = null;
|
||||||
|
}
|
||||||
callback.pluginStateChanged(getState());
|
callback.pluginStateChanged(getState());
|
||||||
}
|
}
|
||||||
|
|
||||||
synchronized State getState() {
|
synchronized State getState() {
|
||||||
if (!started || stopped) return STARTING_STOPPING;
|
if (!started || stopped) return STARTING_STOPPING;
|
||||||
if (!enabledByUser) return DISABLED;
|
if (!enabledByUser) return DISABLED;
|
||||||
return serverSocket == null ? INACTIVE : ACTIVE;
|
if (serverSocketV4 != null || serverSocketV6 != null) return ACTIVE;
|
||||||
|
return INACTIVE;
|
||||||
}
|
}
|
||||||
|
|
||||||
synchronized int getReasonsDisabled() {
|
synchronized int getReasonsDisabled() {
|
||||||
|
|||||||
@@ -1,13 +1,10 @@
|
|||||||
package org.briarproject.bramble.plugin.tcp;
|
package org.briarproject.bramble.plugin.tcp;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.data.BdfList;
|
|
||||||
import org.briarproject.bramble.api.keyagreement.KeyAgreementListener;
|
|
||||||
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
|
||||||
import org.briarproject.bramble.api.properties.TransportProperties;
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
|
|
||||||
import java.net.Inet4Address;
|
import java.net.Inet4Address;
|
||||||
@@ -46,26 +43,11 @@ class WanTcpPlugin extends TcpPlugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean supportsKeyAgreement() {
|
protected List<InetSocketAddress> getLocalSocketAddresses(boolean ipv4) {
|
||||||
return false;
|
if (!ipv4) return emptyList();
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public KeyAgreementListener createKeyAgreementListener(byte[] commitment) {
|
|
||||||
throw new UnsupportedOperationException();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public DuplexTransportConnection createKeyAgreementConnection(
|
|
||||||
byte[] commitment, BdfList descriptor) {
|
|
||||||
throw new UnsupportedOperationException();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected List<InetSocketAddress> getLocalSocketAddresses() {
|
|
||||||
// Use the same address and port as last time if available
|
// Use the same address and port as last time if available
|
||||||
TransportProperties p = callback.getLocalProperties();
|
TransportProperties p = callback.getLocalProperties();
|
||||||
InetSocketAddress old = parseSocketAddress(p.get(PROP_IP_PORT));
|
InetSocketAddress old = parseIpv4SocketAddress(p.get(PROP_IP_PORT));
|
||||||
List<InetSocketAddress> addrs = new LinkedList<>();
|
List<InetSocketAddress> addrs = new LinkedList<>();
|
||||||
for (InetAddress a : getLocalInetAddresses()) {
|
for (InetAddress a : getLocalInetAddresses()) {
|
||||||
if (isAcceptableAddress(a)) {
|
if (isAcceptableAddress(a)) {
|
||||||
@@ -95,14 +77,11 @@ class WanTcpPlugin extends TcpPlugin {
|
|||||||
return ipv4 && !loop && !link && !site;
|
return ipv4 && !loop && !link && !site;
|
||||||
}
|
}
|
||||||
|
|
||||||
private int chooseEphemeralPort() {
|
|
||||||
return 32768 + (int) (Math.random() * 32768);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected List<InetSocketAddress> getRemoteSocketAddresses(
|
protected List<InetSocketAddress> getRemoteSocketAddresses(
|
||||||
TransportProperties p) {
|
TransportProperties p, boolean ipv4) {
|
||||||
InetSocketAddress parsed = parseSocketAddress(p.get(PROP_IP_PORT));
|
if (!ipv4) return emptyList();
|
||||||
|
InetSocketAddress parsed = parseIpv4SocketAddress(p.get(PROP_IP_PORT));
|
||||||
if (parsed == null) return emptyList();
|
if (parsed == null) return emptyList();
|
||||||
return singletonList(parsed);
|
return singletonList(parsed);
|
||||||
}
|
}
|
||||||
@@ -115,7 +94,8 @@ class WanTcpPlugin extends TcpPlugin {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void setLocalSocketAddress(InetSocketAddress a) {
|
protected void setLocalSocketAddress(InetSocketAddress a, boolean ipv4) {
|
||||||
|
if (!ipv4) throw new AssertionError();
|
||||||
if (mappingResult != null && mappingResult.isUsable()) {
|
if (mappingResult != null && mappingResult.isUsable()) {
|
||||||
// Advertise the external address to contacts
|
// Advertise the external address to contacts
|
||||||
if (a.equals(mappingResult.getInternal())) {
|
if (a.equals(mappingResult.getInternal())) {
|
||||||
|
|||||||
@@ -19,7 +19,6 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.DiscoveryHandler;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.PluginException;
|
import org.briarproject.bramble.api.plugin.PluginException;
|
||||||
import org.briarproject.bramble.api.plugin.TorConstants;
|
import org.briarproject.bramble.api.plugin.TorConstants;
|
||||||
@@ -288,7 +287,7 @@ abstract class TorPlugin implements DuplexPlugin, EventHandler, EventListener {
|
|||||||
bind();
|
bind();
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: Remove after a reasonable migration period (added 2020-01-16)
|
// TODO: Remove after a reasonable migration period (added 2020-06-25)
|
||||||
private Settings migrateSettings(Settings settings) {
|
private Settings migrateSettings(Settings settings) {
|
||||||
int network = settings.getInt(PREF_TOR_NETWORK,
|
int network = settings.getInt(PREF_TOR_NETWORK,
|
||||||
PREF_TOR_NETWORK_AUTOMATIC);
|
PREF_TOR_NETWORK_AUTOMATIC);
|
||||||
@@ -550,8 +549,8 @@ abstract class TorPlugin implements DuplexPlugin, EventHandler, EventListener {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void poll(
|
public void poll(Collection<Pair<TransportProperties, ConnectionHandler>>
|
||||||
List<Pair<TransportProperties, ConnectionHandler>> properties) {
|
properties) {
|
||||||
if (getState() != ACTIVE) return;
|
if (getState() != ACTIVE) return;
|
||||||
backoff.increment();
|
backoff.increment();
|
||||||
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
for (Pair<TransportProperties, ConnectionHandler> p : properties) {
|
||||||
@@ -685,17 +684,6 @@ abstract class TorPlugin implements DuplexPlugin, EventHandler, EventListener {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean supportsDiscovery() {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void discoverPeers(
|
|
||||||
List<Pair<TransportProperties, DiscoveryHandler>> properties) {
|
|
||||||
throw new UnsupportedOperationException();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void circuitStatus(String status, String id, String path) {
|
public void circuitStatus(String status, String id, String path) {
|
||||||
if (status.equals("BUILT") &&
|
if (status.equals("BUILT") &&
|
||||||
|
|||||||
@@ -37,6 +37,11 @@ import javax.annotation.Nullable;
|
|||||||
import javax.annotation.concurrent.Immutable;
|
import javax.annotation.concurrent.Immutable;
|
||||||
import javax.inject.Inject;
|
import javax.inject.Inject;
|
||||||
|
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.GROUP_KEY_DISCOVERED;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_LOCAL;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_TRANSPORT_ID;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_VERSION;
|
||||||
|
|
||||||
@Immutable
|
@Immutable
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
class TransportPropertyManagerImpl implements TransportPropertyManager,
|
class TransportPropertyManagerImpl implements TransportPropertyManager,
|
||||||
@@ -111,10 +116,10 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
try {
|
try {
|
||||||
// Find the latest update for this transport, if any
|
// Find the latest update for this transport, if any
|
||||||
BdfDictionary d = metadataParser.parse(meta);
|
BdfDictionary d = metadataParser.parse(meta);
|
||||||
TransportId t = new TransportId(d.getString("transportId"));
|
TransportId t = new TransportId(d.getString(MSG_KEY_TRANSPORT_ID));
|
||||||
LatestUpdate latest = findLatest(txn, m.getGroupId(), t, false);
|
LatestUpdate latest = findLatest(txn, m.getGroupId(), t, false);
|
||||||
if (latest != null) {
|
if (latest != null) {
|
||||||
if (d.getLong("version") > latest.version) {
|
if (d.getLong(MSG_KEY_VERSION) > latest.version) {
|
||||||
// This update is newer - delete the previous update
|
// This update is newer - delete the previous update
|
||||||
db.deleteMessage(txn, latest.messageId);
|
db.deleteMessage(txn, latest.messageId);
|
||||||
db.deleteMessageMetadata(txn, latest.messageId);
|
db.deleteMessageMetadata(txn, latest.messageId);
|
||||||
@@ -140,6 +145,27 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void addRemotePropertiesFromConnection(ContactId c, TransportId t,
|
||||||
|
TransportProperties props) throws DbException {
|
||||||
|
if (props.isEmpty()) return;
|
||||||
|
try {
|
||||||
|
db.transaction(false, txn -> {
|
||||||
|
Group g = getContactGroup(db.getContact(txn, c));
|
||||||
|
BdfDictionary meta = clientHelper.getGroupMetadataAsDictionary(
|
||||||
|
txn, g.getId());
|
||||||
|
BdfDictionary discovered =
|
||||||
|
meta.getOptionalDictionary(GROUP_KEY_DISCOVERED);
|
||||||
|
if (discovered == null) discovered = new BdfDictionary();
|
||||||
|
discovered.putAll(props);
|
||||||
|
meta.put(GROUP_KEY_DISCOVERED, discovered);
|
||||||
|
clientHelper.mergeGroupMetadata(txn, g.getId(), meta);
|
||||||
|
});
|
||||||
|
} catch (FormatException e) {
|
||||||
|
throw new DbException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Map<TransportId, TransportProperties> getLocalProperties()
|
public Map<TransportId, TransportProperties> getLocalProperties()
|
||||||
throws DbException {
|
throws DbException {
|
||||||
@@ -203,12 +229,26 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
Group g = getContactGroup(c);
|
Group g = getContactGroup(c);
|
||||||
try {
|
try {
|
||||||
// Find the latest remote update
|
// Find the latest remote update
|
||||||
|
TransportProperties remote;
|
||||||
LatestUpdate latest = findLatest(txn, g.getId(), t, false);
|
LatestUpdate latest = findLatest(txn, g.getId(), t, false);
|
||||||
if (latest == null) return new TransportProperties();
|
if (latest == null) {
|
||||||
|
remote = new TransportProperties();
|
||||||
|
} else {
|
||||||
// Retrieve and parse the latest remote properties
|
// Retrieve and parse the latest remote properties
|
||||||
BdfList message =
|
BdfList message =
|
||||||
clientHelper.getMessageAsList(txn, latest.messageId);
|
clientHelper.getMessageAsList(txn, latest.messageId);
|
||||||
return parseProperties(message);
|
remote = parseProperties(message);
|
||||||
|
}
|
||||||
|
// Merge in any discovered properties
|
||||||
|
BdfDictionary meta =
|
||||||
|
clientHelper.getGroupMetadataAsDictionary(txn, g.getId());
|
||||||
|
BdfDictionary d = meta.getOptionalDictionary(GROUP_KEY_DISCOVERED);
|
||||||
|
if (d == null) return remote;
|
||||||
|
TransportProperties merged =
|
||||||
|
clientHelper.parseAndValidateTransportProperties(d);
|
||||||
|
// Received properties override discovered properties
|
||||||
|
merged.putAll(remote);
|
||||||
|
return merged;
|
||||||
} catch (FormatException e) {
|
} catch (FormatException e) {
|
||||||
throw new DbException(e);
|
throw new DbException(e);
|
||||||
}
|
}
|
||||||
@@ -281,9 +321,9 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
long now = clock.currentTimeMillis();
|
long now = clock.currentTimeMillis();
|
||||||
Message m = clientHelper.createMessage(g, now, body);
|
Message m = clientHelper.createMessage(g, now, body);
|
||||||
BdfDictionary meta = new BdfDictionary();
|
BdfDictionary meta = new BdfDictionary();
|
||||||
meta.put("transportId", t.getString());
|
meta.put(MSG_KEY_TRANSPORT_ID, t.getString());
|
||||||
meta.put("version", version);
|
meta.put(MSG_KEY_VERSION, version);
|
||||||
meta.put("local", local);
|
meta.put(MSG_KEY_LOCAL, local);
|
||||||
clientHelper.addLocalMessage(txn, m, meta, shared, false);
|
clientHelper.addLocalMessage(txn, m, meta, shared, false);
|
||||||
} catch (FormatException e) {
|
} catch (FormatException e) {
|
||||||
throw new RuntimeException(e);
|
throw new RuntimeException(e);
|
||||||
@@ -302,8 +342,9 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
.getMessageMetadataAsDictionary(txn, localGroup.getId());
|
.getMessageMetadataAsDictionary(txn, localGroup.getId());
|
||||||
for (Entry<MessageId, BdfDictionary> e : metadata.entrySet()) {
|
for (Entry<MessageId, BdfDictionary> e : metadata.entrySet()) {
|
||||||
BdfDictionary meta = e.getValue();
|
BdfDictionary meta = e.getValue();
|
||||||
TransportId t = new TransportId(meta.getString("transportId"));
|
TransportId t =
|
||||||
long version = meta.getLong("version");
|
new TransportId(meta.getString(MSG_KEY_TRANSPORT_ID));
|
||||||
|
long version = meta.getLong(MSG_KEY_VERSION);
|
||||||
latestUpdates.put(t, new LatestUpdate(e.getKey(), version));
|
latestUpdates.put(t, new LatestUpdate(e.getKey(), version));
|
||||||
}
|
}
|
||||||
return latestUpdates;
|
return latestUpdates;
|
||||||
@@ -316,9 +357,10 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
|
|||||||
clientHelper.getMessageMetadataAsDictionary(txn, g);
|
clientHelper.getMessageMetadataAsDictionary(txn, g);
|
||||||
for (Entry<MessageId, BdfDictionary> e : metadata.entrySet()) {
|
for (Entry<MessageId, BdfDictionary> e : metadata.entrySet()) {
|
||||||
BdfDictionary meta = e.getValue();
|
BdfDictionary meta = e.getValue();
|
||||||
if (meta.getString("transportId").equals(t.getString())
|
if (meta.getString(MSG_KEY_TRANSPORT_ID).equals(t.getString())
|
||||||
&& meta.getBoolean("local") == local) {
|
&& meta.getBoolean(MSG_KEY_LOCAL) == local) {
|
||||||
return new LatestUpdate(e.getKey(), meta.getLong("version"));
|
return new LatestUpdate(e.getKey(),
|
||||||
|
meta.getLong(MSG_KEY_VERSION));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return null;
|
return null;
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package org.briarproject.bramble.rendezvous;
|
|||||||
|
|
||||||
import org.briarproject.bramble.PoliteExecutor;
|
import org.briarproject.bramble.PoliteExecutor;
|
||||||
import org.briarproject.bramble.api.Pair;
|
import org.briarproject.bramble.api.Pair;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
import org.briarproject.bramble.api.contact.PendingContact;
|
import org.briarproject.bramble.api.contact.PendingContact;
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
import org.briarproject.bramble.api.contact.PendingContactState;
|
import org.briarproject.bramble.api.contact.PendingContactState;
|
||||||
@@ -23,7 +24,6 @@ import org.briarproject.bramble.api.lifecycle.Service;
|
|||||||
import org.briarproject.bramble.api.lifecycle.ServiceException;
|
import org.briarproject.bramble.api.lifecycle.ServiceException;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.PluginManager;
|
import org.briarproject.bramble.api.plugin.PluginManager;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
|
|||||||
@@ -11,13 +11,17 @@ import org.briarproject.bramble.api.event.EventListener;
|
|||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.lifecycle.event.LifecycleEvent;
|
import org.briarproject.bramble.api.lifecycle.event.LifecycleEvent;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.TransportInactiveEvent;
|
||||||
import org.briarproject.bramble.api.sync.Ack;
|
import org.briarproject.bramble.api.sync.Ack;
|
||||||
import org.briarproject.bramble.api.sync.Message;
|
import org.briarproject.bramble.api.sync.Message;
|
||||||
import org.briarproject.bramble.api.sync.Offer;
|
import org.briarproject.bramble.api.sync.Offer;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
import org.briarproject.bramble.api.sync.Request;
|
import org.briarproject.bramble.api.sync.Request;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
||||||
import org.briarproject.bramble.api.sync.SyncSession;
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
import org.briarproject.bramble.api.sync.Versions;
|
import org.briarproject.bramble.api.sync.Versions;
|
||||||
|
import org.briarproject.bramble.api.sync.event.CloseSyncConnectionsEvent;
|
||||||
import org.briarproject.bramble.api.sync.event.GroupVisibilityUpdatedEvent;
|
import org.briarproject.bramble.api.sync.event.GroupVisibilityUpdatedEvent;
|
||||||
import org.briarproject.bramble.api.sync.event.MessageRequestedEvent;
|
import org.briarproject.bramble.api.sync.event.MessageRequestedEvent;
|
||||||
import org.briarproject.bramble.api.sync.event.MessageSharedEvent;
|
import org.briarproject.bramble.api.sync.event.MessageSharedEvent;
|
||||||
@@ -35,6 +39,7 @@ import java.util.concurrent.atomic.AtomicBoolean;
|
|||||||
import java.util.concurrent.atomic.AtomicLong;
|
import java.util.concurrent.atomic.AtomicLong;
|
||||||
import java.util.logging.Logger;
|
import java.util.logging.Logger;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
import javax.annotation.concurrent.ThreadSafe;
|
import javax.annotation.concurrent.ThreadSafe;
|
||||||
|
|
||||||
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
||||||
@@ -71,9 +76,12 @@ class DuplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
private final Clock clock;
|
private final Clock clock;
|
||||||
private final ContactId contactId;
|
private final ContactId contactId;
|
||||||
|
private final TransportId transportId;
|
||||||
private final int maxLatency, maxIdleTime;
|
private final int maxLatency, maxIdleTime;
|
||||||
private final StreamWriter streamWriter;
|
private final StreamWriter streamWriter;
|
||||||
private final SyncRecordWriter recordWriter;
|
private final SyncRecordWriter recordWriter;
|
||||||
|
@Nullable
|
||||||
|
private final Priority priority;
|
||||||
private final BlockingQueue<ThrowingRunnable<IOException>> writerTasks;
|
private final BlockingQueue<ThrowingRunnable<IOException>> writerTasks;
|
||||||
|
|
||||||
private final AtomicBoolean generateAckQueued = new AtomicBoolean(false);
|
private final AtomicBoolean generateAckQueued = new AtomicBoolean(false);
|
||||||
@@ -86,18 +94,21 @@ class DuplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
private volatile boolean interrupted = false;
|
private volatile boolean interrupted = false;
|
||||||
|
|
||||||
DuplexOutgoingSession(DatabaseComponent db, Executor dbExecutor,
|
DuplexOutgoingSession(DatabaseComponent db, Executor dbExecutor,
|
||||||
EventBus eventBus, Clock clock, ContactId contactId, int maxLatency,
|
EventBus eventBus, Clock clock, ContactId contactId,
|
||||||
int maxIdleTime, StreamWriter streamWriter,
|
TransportId transportId, int maxLatency, int maxIdleTime,
|
||||||
SyncRecordWriter recordWriter) {
|
StreamWriter streamWriter, SyncRecordWriter recordWriter,
|
||||||
|
@Nullable Priority priority) {
|
||||||
this.db = db;
|
this.db = db;
|
||||||
this.dbExecutor = dbExecutor;
|
this.dbExecutor = dbExecutor;
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
this.clock = clock;
|
this.clock = clock;
|
||||||
this.contactId = contactId;
|
this.contactId = contactId;
|
||||||
|
this.transportId = transportId;
|
||||||
this.maxLatency = maxLatency;
|
this.maxLatency = maxLatency;
|
||||||
this.maxIdleTime = maxIdleTime;
|
this.maxIdleTime = maxIdleTime;
|
||||||
this.streamWriter = streamWriter;
|
this.streamWriter = streamWriter;
|
||||||
this.recordWriter = recordWriter;
|
this.recordWriter = recordWriter;
|
||||||
|
this.priority = priority;
|
||||||
writerTasks = new LinkedBlockingQueue<>();
|
writerTasks = new LinkedBlockingQueue<>();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -108,6 +119,8 @@ class DuplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
try {
|
try {
|
||||||
// Send our supported protocol versions
|
// Send our supported protocol versions
|
||||||
recordWriter.writeVersions(new Versions(SUPPORTED_VERSIONS));
|
recordWriter.writeVersions(new Versions(SUPPORTED_VERSIONS));
|
||||||
|
// Send our connection priority, if this is an outgoing connection
|
||||||
|
if (priority != null) recordWriter.writePriority(priority);
|
||||||
// Start a query for each type of record
|
// Start a query for each type of record
|
||||||
generateAck();
|
generateAck();
|
||||||
generateBatch();
|
generateBatch();
|
||||||
@@ -223,6 +236,12 @@ class DuplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
} else if (e instanceof LifecycleEvent) {
|
} else if (e instanceof LifecycleEvent) {
|
||||||
LifecycleEvent l = (LifecycleEvent) e;
|
LifecycleEvent l = (LifecycleEvent) e;
|
||||||
if (l.getLifecycleState() == STOPPING) interrupt();
|
if (l.getLifecycleState() == STOPPING) interrupt();
|
||||||
|
} else if (e instanceof CloseSyncConnectionsEvent) {
|
||||||
|
CloseSyncConnectionsEvent c = (CloseSyncConnectionsEvent) e;
|
||||||
|
if (c.getTransportId().equals(transportId)) interrupt();
|
||||||
|
} else if (e instanceof TransportInactiveEvent) {
|
||||||
|
TransportInactiveEvent t = (TransportInactiveEvent) e;
|
||||||
|
if (t.getTransportId().equals(transportId)) interrupt();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -15,6 +15,8 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.sync.Ack;
|
import org.briarproject.bramble.api.sync.Ack;
|
||||||
import org.briarproject.bramble.api.sync.Message;
|
import org.briarproject.bramble.api.sync.Message;
|
||||||
import org.briarproject.bramble.api.sync.Offer;
|
import org.briarproject.bramble.api.sync.Offer;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
import org.briarproject.bramble.api.sync.Request;
|
import org.briarproject.bramble.api.sync.Request;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
||||||
import org.briarproject.bramble.api.sync.SyncSession;
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
@@ -47,17 +49,19 @@ class IncomingSession implements SyncSession, EventListener {
|
|||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
private final ContactId contactId;
|
private final ContactId contactId;
|
||||||
private final SyncRecordReader recordReader;
|
private final SyncRecordReader recordReader;
|
||||||
|
private final PriorityHandler priorityHandler;
|
||||||
|
|
||||||
private volatile boolean interrupted = false;
|
private volatile boolean interrupted = false;
|
||||||
|
|
||||||
IncomingSession(DatabaseComponent db, Executor dbExecutor,
|
IncomingSession(DatabaseComponent db, Executor dbExecutor,
|
||||||
EventBus eventBus, ContactId contactId,
|
EventBus eventBus, ContactId contactId,
|
||||||
SyncRecordReader recordReader) {
|
SyncRecordReader recordReader, PriorityHandler priorityHandler) {
|
||||||
this.db = db;
|
this.db = db;
|
||||||
this.dbExecutor = dbExecutor;
|
this.dbExecutor = dbExecutor;
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
this.contactId = contactId;
|
this.contactId = contactId;
|
||||||
this.recordReader = recordReader;
|
this.recordReader = recordReader;
|
||||||
|
this.priorityHandler = priorityHandler;
|
||||||
}
|
}
|
||||||
|
|
||||||
@IoExecutor
|
@IoExecutor
|
||||||
@@ -86,6 +90,9 @@ class IncomingSession implements SyncSession, EventListener {
|
|||||||
} else if (recordReader.hasVersions()) {
|
} else if (recordReader.hasVersions()) {
|
||||||
Versions v = recordReader.readVersions();
|
Versions v = recordReader.readVersions();
|
||||||
dbExecutor.execute(new ReceiveVersions(v));
|
dbExecutor.execute(new ReceiveVersions(v));
|
||||||
|
} else if (recordReader.hasPriority()) {
|
||||||
|
Priority p = recordReader.readPriority();
|
||||||
|
priorityHandler.handle(p);
|
||||||
} else {
|
} else {
|
||||||
// unknown records are ignored in RecordReader#eof()
|
// unknown records are ignored in RecordReader#eof()
|
||||||
throw new FormatException();
|
throw new FormatException();
|
||||||
|
|||||||
@@ -11,11 +11,14 @@ import org.briarproject.bramble.api.event.EventListener;
|
|||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.lifecycle.event.LifecycleEvent;
|
import org.briarproject.bramble.api.lifecycle.event.LifecycleEvent;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.TransportInactiveEvent;
|
||||||
import org.briarproject.bramble.api.sync.Ack;
|
import org.briarproject.bramble.api.sync.Ack;
|
||||||
import org.briarproject.bramble.api.sync.Message;
|
import org.briarproject.bramble.api.sync.Message;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
||||||
import org.briarproject.bramble.api.sync.SyncSession;
|
import org.briarproject.bramble.api.sync.SyncSession;
|
||||||
import org.briarproject.bramble.api.sync.Versions;
|
import org.briarproject.bramble.api.sync.Versions;
|
||||||
|
import org.briarproject.bramble.api.sync.event.CloseSyncConnectionsEvent;
|
||||||
import org.briarproject.bramble.api.transport.StreamWriter;
|
import org.briarproject.bramble.api.transport.StreamWriter;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
@@ -56,6 +59,7 @@ class SimplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
private final Executor dbExecutor;
|
private final Executor dbExecutor;
|
||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
private final ContactId contactId;
|
private final ContactId contactId;
|
||||||
|
private final TransportId transportId;
|
||||||
private final int maxLatency;
|
private final int maxLatency;
|
||||||
private final StreamWriter streamWriter;
|
private final StreamWriter streamWriter;
|
||||||
private final SyncRecordWriter recordWriter;
|
private final SyncRecordWriter recordWriter;
|
||||||
@@ -65,12 +69,14 @@ class SimplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
private volatile boolean interrupted = false;
|
private volatile boolean interrupted = false;
|
||||||
|
|
||||||
SimplexOutgoingSession(DatabaseComponent db, Executor dbExecutor,
|
SimplexOutgoingSession(DatabaseComponent db, Executor dbExecutor,
|
||||||
EventBus eventBus, ContactId contactId, int maxLatency,
|
EventBus eventBus, ContactId contactId, TransportId transportId,
|
||||||
StreamWriter streamWriter, SyncRecordWriter recordWriter) {
|
int maxLatency, StreamWriter streamWriter,
|
||||||
|
SyncRecordWriter recordWriter) {
|
||||||
this.db = db;
|
this.db = db;
|
||||||
this.dbExecutor = dbExecutor;
|
this.dbExecutor = dbExecutor;
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
this.contactId = contactId;
|
this.contactId = contactId;
|
||||||
|
this.transportId = transportId;
|
||||||
this.maxLatency = maxLatency;
|
this.maxLatency = maxLatency;
|
||||||
this.streamWriter = streamWriter;
|
this.streamWriter = streamWriter;
|
||||||
this.recordWriter = recordWriter;
|
this.recordWriter = recordWriter;
|
||||||
@@ -123,6 +129,12 @@ class SimplexOutgoingSession implements SyncSession, EventListener {
|
|||||||
} else if (e instanceof LifecycleEvent) {
|
} else if (e instanceof LifecycleEvent) {
|
||||||
LifecycleEvent l = (LifecycleEvent) e;
|
LifecycleEvent l = (LifecycleEvent) e;
|
||||||
if (l.getLifecycleState() == STOPPING) interrupt();
|
if (l.getLifecycleState() == STOPPING) interrupt();
|
||||||
|
} else if (e instanceof CloseSyncConnectionsEvent) {
|
||||||
|
CloseSyncConnectionsEvent c = (CloseSyncConnectionsEvent) e;
|
||||||
|
if (c.getTransportId().equals(transportId)) interrupt();
|
||||||
|
} else if (e instanceof TransportInactiveEvent) {
|
||||||
|
TransportInactiveEvent t = (TransportInactiveEvent) e;
|
||||||
|
if (t.getTransportId().equals(transportId)) interrupt();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ import org.briarproject.bramble.api.sync.Message;
|
|||||||
import org.briarproject.bramble.api.sync.MessageFactory;
|
import org.briarproject.bramble.api.sync.MessageFactory;
|
||||||
import org.briarproject.bramble.api.sync.MessageId;
|
import org.briarproject.bramble.api.sync.MessageId;
|
||||||
import org.briarproject.bramble.api.sync.Offer;
|
import org.briarproject.bramble.api.sync.Offer;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
import org.briarproject.bramble.api.sync.Request;
|
import org.briarproject.bramble.api.sync.Request;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
||||||
import org.briarproject.bramble.api.sync.Versions;
|
import org.briarproject.bramble.api.sync.Versions;
|
||||||
@@ -26,10 +27,12 @@ import javax.annotation.concurrent.NotThreadSafe;
|
|||||||
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.MESSAGE;
|
import static org.briarproject.bramble.api.sync.RecordTypes.MESSAGE;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
||||||
|
import static org.briarproject.bramble.api.sync.RecordTypes.PRIORITY;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_SUPPORTED_VERSIONS;
|
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_SUPPORTED_VERSIONS;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.MESSAGE_HEADER_LENGTH;
|
import static org.briarproject.bramble.api.sync.SyncConstants.MESSAGE_HEADER_LENGTH;
|
||||||
|
import static org.briarproject.bramble.api.sync.SyncConstants.PRIORITY_NONCE_BYTES;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
||||||
|
|
||||||
@NotThreadSafe
|
@NotThreadSafe
|
||||||
@@ -48,7 +51,7 @@ class SyncRecordReaderImpl implements SyncRecordReader {
|
|||||||
|
|
||||||
private static boolean isKnownRecordType(byte type) {
|
private static boolean isKnownRecordType(byte type) {
|
||||||
return type == ACK || type == MESSAGE || type == OFFER ||
|
return type == ACK || type == MESSAGE || type == OFFER ||
|
||||||
type == REQUEST || type == VERSIONS;
|
type == REQUEST || type == VERSIONS || type == PRIORITY;
|
||||||
}
|
}
|
||||||
|
|
||||||
private final MessageFactory messageFactory;
|
private final MessageFactory messageFactory;
|
||||||
@@ -174,4 +177,23 @@ class SyncRecordReaderImpl implements SyncRecordReader {
|
|||||||
nextRecord = null;
|
nextRecord = null;
|
||||||
return supported;
|
return supported;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean hasPriority() throws IOException {
|
||||||
|
return !eof() && getNextRecordType() == PRIORITY;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Priority readPriority() throws IOException {
|
||||||
|
if (!hasPriority()) throw new FormatException();
|
||||||
|
return new Priority(readNonce());
|
||||||
|
}
|
||||||
|
|
||||||
|
private byte[] readNonce() throws IOException {
|
||||||
|
if (nextRecord == null) throw new AssertionError();
|
||||||
|
byte[] payload = nextRecord.getPayload();
|
||||||
|
if (payload.length != PRIORITY_NONCE_BYTES) throw new FormatException();
|
||||||
|
nextRecord = null;
|
||||||
|
return payload;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import org.briarproject.bramble.api.sync.Message;
|
|||||||
import org.briarproject.bramble.api.sync.MessageFactory;
|
import org.briarproject.bramble.api.sync.MessageFactory;
|
||||||
import org.briarproject.bramble.api.sync.MessageId;
|
import org.briarproject.bramble.api.sync.MessageId;
|
||||||
import org.briarproject.bramble.api.sync.Offer;
|
import org.briarproject.bramble.api.sync.Offer;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
import org.briarproject.bramble.api.sync.Request;
|
import org.briarproject.bramble.api.sync.Request;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
||||||
import org.briarproject.bramble.api.sync.Versions;
|
import org.briarproject.bramble.api.sync.Versions;
|
||||||
@@ -20,6 +21,7 @@ import javax.annotation.concurrent.NotThreadSafe;
|
|||||||
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.MESSAGE;
|
import static org.briarproject.bramble.api.sync.RecordTypes.MESSAGE;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
||||||
|
import static org.briarproject.bramble.api.sync.RecordTypes.PRIORITY;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
||||||
@@ -73,6 +75,12 @@ class SyncRecordWriterImpl implements SyncRecordWriter {
|
|||||||
writeRecord(VERSIONS);
|
writeRecord(VERSIONS);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void writePriority(Priority p) throws IOException {
|
||||||
|
writer.writeRecord(
|
||||||
|
new Record(PROTOCOL_VERSION, PRIORITY, p.getNonce()));
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void flush() throws IOException {
|
public void flush() throws IOException {
|
||||||
writer.flush();
|
writer.flush();
|
||||||
|
|||||||
@@ -5,6 +5,9 @@ import org.briarproject.bramble.api.db.DatabaseComponent;
|
|||||||
import org.briarproject.bramble.api.db.DatabaseExecutor;
|
import org.briarproject.bramble.api.db.DatabaseExecutor;
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
import org.briarproject.bramble.api.sync.PriorityHandler;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordReaderFactory;
|
import org.briarproject.bramble.api.sync.SyncRecordReaderFactory;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
import org.briarproject.bramble.api.sync.SyncRecordWriter;
|
||||||
@@ -18,6 +21,7 @@ import java.io.InputStream;
|
|||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
import javax.annotation.concurrent.Immutable;
|
import javax.annotation.concurrent.Immutable;
|
||||||
import javax.inject.Inject;
|
import javax.inject.Inject;
|
||||||
|
|
||||||
@@ -46,29 +50,32 @@ class SyncSessionFactoryImpl implements SyncSessionFactory {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncSession createIncomingSession(ContactId c, InputStream in) {
|
public SyncSession createIncomingSession(ContactId c, InputStream in,
|
||||||
|
PriorityHandler handler) {
|
||||||
SyncRecordReader recordReader =
|
SyncRecordReader recordReader =
|
||||||
recordReaderFactory.createRecordReader(in);
|
recordReaderFactory.createRecordReader(in);
|
||||||
return new IncomingSession(db, dbExecutor, eventBus, c, recordReader);
|
return new IncomingSession(db, dbExecutor, eventBus, c, recordReader,
|
||||||
|
handler);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncSession createSimplexOutgoingSession(ContactId c,
|
public SyncSession createSimplexOutgoingSession(ContactId c, TransportId t,
|
||||||
int maxLatency, StreamWriter streamWriter) {
|
int maxLatency, StreamWriter streamWriter) {
|
||||||
OutputStream out = streamWriter.getOutputStream();
|
OutputStream out = streamWriter.getOutputStream();
|
||||||
SyncRecordWriter recordWriter =
|
SyncRecordWriter recordWriter =
|
||||||
recordWriterFactory.createRecordWriter(out);
|
recordWriterFactory.createRecordWriter(out);
|
||||||
return new SimplexOutgoingSession(db, dbExecutor, eventBus, c,
|
return new SimplexOutgoingSession(db, dbExecutor, eventBus, c, t,
|
||||||
maxLatency, streamWriter, recordWriter);
|
maxLatency, streamWriter, recordWriter);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncSession createDuplexOutgoingSession(ContactId c, int maxLatency,
|
public SyncSession createDuplexOutgoingSession(ContactId c, TransportId t,
|
||||||
int maxIdleTime, StreamWriter streamWriter) {
|
int maxLatency, int maxIdleTime, StreamWriter streamWriter,
|
||||||
|
@Nullable Priority priority) {
|
||||||
OutputStream out = streamWriter.getOutputStream();
|
OutputStream out = streamWriter.getOutputStream();
|
||||||
SyncRecordWriter recordWriter =
|
SyncRecordWriter recordWriter =
|
||||||
recordWriterFactory.createRecordWriter(out);
|
recordWriterFactory.createRecordWriter(out);
|
||||||
return new DuplexOutgoingSession(db, dbExecutor, eventBus, clock, c,
|
return new DuplexOutgoingSession(db, dbExecutor, eventBus, clock, c, t,
|
||||||
maxLatency, maxIdleTime, streamWriter, recordWriter);
|
maxLatency, maxIdleTime, streamWriter, recordWriter, priority);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package org.briarproject.bramble.account;
|
package org.briarproject.bramble.account;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
import org.briarproject.bramble.api.crypto.CryptoComponent;
|
||||||
|
import org.briarproject.bramble.api.crypto.DecryptionException;
|
||||||
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
||||||
import org.briarproject.bramble.api.crypto.SecretKey;
|
import org.briarproject.bramble.api.crypto.SecretKey;
|
||||||
import org.briarproject.bramble.api.db.DatabaseConfig;
|
import org.briarproject.bramble.api.db.DatabaseConfig;
|
||||||
@@ -19,12 +20,15 @@ import java.io.FileInputStream;
|
|||||||
import java.io.FileOutputStream;
|
import java.io.FileOutputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStreamReader;
|
import java.io.InputStreamReader;
|
||||||
|
import java.nio.charset.Charset;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
import static junit.framework.Assert.assertFalse;
|
import static junit.framework.Assert.assertFalse;
|
||||||
import static junit.framework.Assert.assertNull;
|
import static junit.framework.Assert.assertNull;
|
||||||
import static junit.framework.Assert.assertTrue;
|
import static junit.framework.Assert.assertTrue;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_CIPHERTEXT;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_PASSWORD;
|
||||||
import static org.briarproject.bramble.test.TestUtils.deleteTestDirectory;
|
import static org.briarproject.bramble.test.TestUtils.deleteTestDirectory;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getIdentity;
|
import static org.briarproject.bramble.test.TestUtils.getIdentity;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
||||||
@@ -35,6 +39,7 @@ import static org.briarproject.bramble.util.StringUtils.toHexString;
|
|||||||
import static org.junit.Assert.assertArrayEquals;
|
import static org.junit.Assert.assertArrayEquals;
|
||||||
import static org.junit.Assert.assertEquals;
|
import static org.junit.Assert.assertEquals;
|
||||||
import static org.junit.Assert.assertNotNull;
|
import static org.junit.Assert.assertNotNull;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
public class AccountManagerImplTest extends BrambleMockTestCase {
|
public class AccountManagerImplTest extends BrambleMockTestCase {
|
||||||
|
|
||||||
@@ -83,8 +88,13 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testSignInReturnsFalseIfDbKeyCannotBeLoaded() {
|
public void testSignInThrowsExceptionIfDbKeyCannotBeLoaded() {
|
||||||
assertFalse(accountManager.signIn(password));
|
try {
|
||||||
|
accountManager.signIn(password);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_CIPHERTEXT, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
assertFalse(accountManager.hasDatabaseKey());
|
assertFalse(accountManager.hasDatabaseKey());
|
||||||
|
|
||||||
assertFalse(keyFile.exists());
|
assertFalse(keyFile.exists());
|
||||||
@@ -92,11 +102,11 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testSignInReturnsFalseIfPasswordIsWrong() throws Exception {
|
public void testSignInThrowsExceptionIfPasswordIsWrong() throws Exception {
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
oneOf(crypto).decryptWithPassword(encryptedKey, password,
|
oneOf(crypto).decryptWithPassword(encryptedKey, password,
|
||||||
keyStrengthener);
|
keyStrengthener);
|
||||||
will(returnValue(null));
|
will(throwException(new DecryptionException(INVALID_PASSWORD)));
|
||||||
}});
|
}});
|
||||||
|
|
||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
storeDatabaseKey(keyFile, encryptedKeyHex);
|
||||||
@@ -105,7 +115,12 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
||||||
|
|
||||||
assertFalse(accountManager.signIn(password));
|
try {
|
||||||
|
accountManager.signIn(password);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_PASSWORD, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
assertFalse(accountManager.hasDatabaseKey());
|
assertFalse(accountManager.hasDatabaseKey());
|
||||||
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
@@ -128,7 +143,7 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
||||||
|
|
||||||
assertTrue(accountManager.signIn(password));
|
accountManager.signIn(password);
|
||||||
assertTrue(accountManager.hasDatabaseKey());
|
assertTrue(accountManager.hasDatabaseKey());
|
||||||
SecretKey decrypted = accountManager.getDatabaseKey();
|
SecretKey decrypted = accountManager.getDatabaseKey();
|
||||||
assertNotNull(decrypted);
|
assertNotNull(decrypted);
|
||||||
@@ -157,7 +172,7 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
||||||
|
|
||||||
assertTrue(accountManager.signIn(password));
|
accountManager.signIn(password);
|
||||||
assertTrue(accountManager.hasDatabaseKey());
|
assertTrue(accountManager.hasDatabaseKey());
|
||||||
SecretKey decrypted = accountManager.getDatabaseKey();
|
SecretKey decrypted = accountManager.getDatabaseKey();
|
||||||
assertNotNull(decrypted);
|
assertNotNull(decrypted);
|
||||||
@@ -239,55 +254,6 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
assertFalse(keyBackupFile.exists());
|
assertFalse(keyBackupFile.exists());
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testAccountExistsReturnsFalseIfDbDirectoryDoesNotExist()
|
|
||||||
throws Exception {
|
|
||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
|
||||||
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
|
||||||
|
|
||||||
assertFalse(dbDir.exists());
|
|
||||||
|
|
||||||
assertFalse(accountManager.accountExists());
|
|
||||||
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
|
||||||
assertFalse(dbDir.exists());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testAccountExistsReturnsFalseIfDbDirectoryIsNotDirectory()
|
|
||||||
throws Exception {
|
|
||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
|
||||||
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
|
||||||
|
|
||||||
assertTrue(dbDir.createNewFile());
|
|
||||||
assertFalse(dbDir.isDirectory());
|
|
||||||
|
|
||||||
assertFalse(accountManager.accountExists());
|
|
||||||
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
|
||||||
assertTrue(dbDir.exists());
|
|
||||||
assertFalse(dbDir.isDirectory());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testAccountExistsReturnsTrueIfDbDirectoryIsDirectory()
|
|
||||||
throws Exception {
|
|
||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
|
||||||
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
|
||||||
|
|
||||||
assertTrue(dbDir.mkdirs());
|
|
||||||
assertTrue(dbDir.isDirectory());
|
|
||||||
|
|
||||||
assertTrue(accountManager.accountExists());
|
|
||||||
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
|
||||||
assertTrue(dbDir.exists());
|
|
||||||
assertTrue(dbDir.isDirectory());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testCreateAccountStoresDbKey() throws Exception {
|
public void testCreateAccountStoresDbKey() throws Exception {
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
@@ -315,26 +281,36 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testChangePasswordReturnsFalseIfDbKeyCannotBeLoaded() {
|
public void testChangePasswordThrowsExceptionIfDbKeyCannotBeLoaded() {
|
||||||
assertFalse(accountManager.changePassword(password, newPassword));
|
try {
|
||||||
|
accountManager.changePassword(password, newPassword);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_CIPHERTEXT, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
|
|
||||||
assertFalse(keyFile.exists());
|
assertFalse(keyFile.exists());
|
||||||
assertFalse(keyBackupFile.exists());
|
assertFalse(keyBackupFile.exists());
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testChangePasswordReturnsFalseIfPasswordIsWrong()
|
public void testChangePasswordThrowsExceptionIfPasswordIsWrong()
|
||||||
throws Exception {
|
throws Exception {
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
oneOf(crypto).decryptWithPassword(encryptedKey, password,
|
oneOf(crypto).decryptWithPassword(encryptedKey, password,
|
||||||
keyStrengthener);
|
keyStrengthener);
|
||||||
will(returnValue(null));
|
will(throwException(new DecryptionException(INVALID_PASSWORD)));
|
||||||
}});
|
}});
|
||||||
|
|
||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
storeDatabaseKey(keyFile, encryptedKeyHex);
|
||||||
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
||||||
|
|
||||||
assertFalse(accountManager.changePassword(password, newPassword));
|
try {
|
||||||
|
accountManager.changePassword(password, newPassword);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_PASSWORD, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
|
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
assertEquals(encryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
||||||
@@ -357,7 +333,7 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
storeDatabaseKey(keyFile, encryptedKeyHex);
|
storeDatabaseKey(keyFile, encryptedKeyHex);
|
||||||
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
storeDatabaseKey(keyBackupFile, encryptedKeyHex);
|
||||||
|
|
||||||
assertTrue(accountManager.changePassword(password, newPassword));
|
accountManager.changePassword(password, newPassword);
|
||||||
|
|
||||||
assertEquals(newEncryptedKeyHex, loadDatabaseKey(keyFile));
|
assertEquals(newEncryptedKeyHex, loadDatabaseKey(keyFile));
|
||||||
assertEquals(newEncryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
assertEquals(newEncryptedKeyHex, loadDatabaseKey(keyBackupFile));
|
||||||
@@ -366,7 +342,7 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
private void storeDatabaseKey(File f, String hex) throws IOException {
|
private void storeDatabaseKey(File f, String hex) throws IOException {
|
||||||
f.getParentFile().mkdirs();
|
f.getParentFile().mkdirs();
|
||||||
FileOutputStream out = new FileOutputStream(f);
|
FileOutputStream out = new FileOutputStream(f);
|
||||||
out.write(hex.getBytes("UTF-8"));
|
out.write(hex.getBytes(Charset.forName("UTF-8")));
|
||||||
out.flush();
|
out.flush();
|
||||||
out.close();
|
out.close();
|
||||||
}
|
}
|
||||||
@@ -374,7 +350,7 @@ public class AccountManagerImplTest extends BrambleMockTestCase {
|
|||||||
@Nullable
|
@Nullable
|
||||||
private String loadDatabaseKey(File f) throws IOException {
|
private String loadDatabaseKey(File f) throws IOException {
|
||||||
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
BufferedReader reader = new BufferedReader(new InputStreamReader(
|
||||||
new FileInputStream(f), "UTF-8"));
|
new FileInputStream(f), Charset.forName("UTF-8")));
|
||||||
String hex = reader.readLine();
|
String hex = reader.readLine();
|
||||||
reader.close();
|
reader.close();
|
||||||
return hex;
|
return hex;
|
||||||
|
|||||||
@@ -0,0 +1,611 @@
|
|||||||
|
package org.briarproject.bramble.connection;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
|
import org.briarproject.bramble.api.connection.InterruptibleConnection;
|
||||||
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
|
import org.briarproject.bramble.api.contact.PendingContactId;
|
||||||
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
||||||
|
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
|
import org.briarproject.bramble.test.BrambleMockTestCase;
|
||||||
|
import org.jmock.Expectations;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.util.Collection;
|
||||||
|
|
||||||
|
import static java.util.Collections.emptyList;
|
||||||
|
import static java.util.Collections.emptyMap;
|
||||||
|
import static java.util.Collections.singletonList;
|
||||||
|
import static java.util.Collections.singletonMap;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
||||||
|
import static org.briarproject.bramble.util.StringUtils.fromHexString;
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.assertFalse;
|
||||||
|
import static org.junit.Assert.assertTrue;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
public class ConnectionRegistryImplTest extends BrambleMockTestCase {
|
||||||
|
|
||||||
|
private final EventBus eventBus = context.mock(EventBus.class);
|
||||||
|
private final PluginConfig pluginConfig = context.mock(PluginConfig.class);
|
||||||
|
private final InterruptibleConnection conn1 =
|
||||||
|
context.mock(InterruptibleConnection.class, "conn1");
|
||||||
|
private final InterruptibleConnection conn2 =
|
||||||
|
context.mock(InterruptibleConnection.class, "conn2");
|
||||||
|
private final InterruptibleConnection conn3 =
|
||||||
|
context.mock(InterruptibleConnection.class, "conn3");
|
||||||
|
|
||||||
|
private final ContactId contactId1 = getContactId();
|
||||||
|
private final ContactId contactId2 = getContactId();
|
||||||
|
private final TransportId transportId1 = getTransportId();
|
||||||
|
private final TransportId transportId2 = getTransportId();
|
||||||
|
private final TransportId transportId3 = getTransportId();
|
||||||
|
private final PendingContactId pendingContactId =
|
||||||
|
new PendingContactId(getRandomId());
|
||||||
|
|
||||||
|
private final Priority low =
|
||||||
|
new Priority(fromHexString("00000000000000000000000000000000"));
|
||||||
|
private final Priority high =
|
||||||
|
new Priority(fromHexString("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"));
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterMultipleConnections() {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(emptyMap()));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// The registry should be empty
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
assertFalse(c.isConnected(contactId1));
|
||||||
|
assertFalse(c.isConnected(contactId1, transportId1));
|
||||||
|
assertFalse(c.isConnected(contactId1, transportId2));
|
||||||
|
assertFalse(c.isConnected(contactId1, transportId3));
|
||||||
|
|
||||||
|
// Check that a registered connection shows up - this should
|
||||||
|
// broadcast a ConnectionOpenedEvent and a ContactConnectedEvent
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn1);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
assertTrue(c.isConnected(contactId1));
|
||||||
|
assertTrue(c.isConnected(contactId1, transportId1));
|
||||||
|
|
||||||
|
// Register another connection with the same contact and transport -
|
||||||
|
// this should broadcast a ConnectionOpenedEvent and lookup should be
|
||||||
|
// unaffected
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn2);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
assertTrue(c.isConnected(contactId1));
|
||||||
|
assertTrue(c.isConnected(contactId1, transportId1));
|
||||||
|
|
||||||
|
// Unregister one of the connections - this should broadcast a
|
||||||
|
// ConnectionClosedEvent and lookup should be unaffected
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.unregisterConnection(contactId1, transportId1, conn1, true, false);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
assertTrue(c.isConnected(contactId1));
|
||||||
|
assertTrue(c.isConnected(contactId1, transportId1));
|
||||||
|
|
||||||
|
// Unregister the other connection - this should broadcast a
|
||||||
|
// ConnectionClosedEvent and a ContactDisconnectedEvent
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(
|
||||||
|
ContactDisconnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.unregisterConnection(contactId1, transportId1, conn2, true, false);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
assertFalse(c.isConnected(contactId1));
|
||||||
|
assertFalse(c.isConnected(contactId1, transportId1));
|
||||||
|
|
||||||
|
// Try to unregister the connection again - exception should be thrown
|
||||||
|
try {
|
||||||
|
c.unregisterConnection(contactId1, transportId1, conn2,
|
||||||
|
true, false);
|
||||||
|
fail();
|
||||||
|
} catch (IllegalArgumentException expected) {
|
||||||
|
// Expected
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterMultipleContacts() {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(emptyMap()));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Register two contacts with one transport, then one of the contacts
|
||||||
|
// with a second transport - this should broadcast three
|
||||||
|
// ConnectionOpenedEvents and two ContactConnectedEvents
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
exactly(3).of(eventBus).broadcast(with(any(
|
||||||
|
ConnectionOpenedEvent.class)));
|
||||||
|
exactly(2).of(eventBus).broadcast(with(any(
|
||||||
|
ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn1);
|
||||||
|
c.registerIncomingConnection(contactId2, transportId1, conn2);
|
||||||
|
c.registerIncomingConnection(contactId2, transportId2, conn3);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertTrue(c.isConnected(contactId1));
|
||||||
|
assertTrue(c.isConnected(contactId2));
|
||||||
|
|
||||||
|
assertTrue(c.isConnected(contactId1, transportId1));
|
||||||
|
assertFalse(c.isConnected(contactId1, transportId2));
|
||||||
|
|
||||||
|
assertTrue(c.isConnected(contactId2, transportId1));
|
||||||
|
assertTrue(c.isConnected(contactId2, transportId2));
|
||||||
|
|
||||||
|
Collection<ContactId> connected = c.getConnectedContacts(transportId1);
|
||||||
|
assertEquals(2, connected.size());
|
||||||
|
assertTrue(connected.contains(contactId1));
|
||||||
|
assertTrue(connected.contains(contactId2));
|
||||||
|
|
||||||
|
connected = c.getConnectedOrBetterContacts(transportId1);
|
||||||
|
assertEquals(2, connected.size());
|
||||||
|
assertTrue(connected.contains(contactId1));
|
||||||
|
assertTrue(connected.contains(contactId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId2),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId2),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testConnectionsAreNotInterruptedUnlessPriorityIsSet() {
|
||||||
|
// Prefer transport 2 to transport 1
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(
|
||||||
|
singletonMap(transportId1, singletonList(transportId2))));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Connect via transport 1 (worse than 2) with no priority set
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn1);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 2 (better than 1) and set priority to high -
|
||||||
|
// the old connection should not be interrupted, despite using a worse
|
||||||
|
// transport, to remain compatible with old peers
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId2, conn2, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 3 (no preference) and set priority to high -
|
||||||
|
// again, no interruptions are expected
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId3, conn3, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNewConnectionIsInterruptedIfOldConnectionUsesBetterTransport() {
|
||||||
|
// Prefer transport 1 to transport 2
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(
|
||||||
|
singletonMap(transportId2, singletonList(transportId1))));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Connect via transport 1 (better than 2) and set priority to low
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId1, conn1, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// The contact is not connected via transport 2 but is connected via a
|
||||||
|
// better transport
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 2 (worse than 1) and set priority to high -
|
||||||
|
// the new connection should be interrupted because it uses a worse
|
||||||
|
// transport
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(conn2).interruptOutgoingSession();
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId2, conn2, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 3 (no preference) and set priority to low -
|
||||||
|
// no further interruptions
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId3, conn3, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
|
||||||
|
// Unregister the interrupted connection (transport 2)
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.unregisterConnection(contactId1, transportId2, conn2, true, false);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// The contact is not connected via transport 2 but is connected via a
|
||||||
|
// better transport
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testOldConnectionIsInterruptedIfNewConnectionUsesBetterTransport() {
|
||||||
|
// Prefer transport 2 to transport 1
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(
|
||||||
|
singletonMap(transportId1, singletonList(transportId2))));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Connect via transport 1 (worse than 2) and set priority to high
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId1, conn1, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(emptyList(), c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 2 (better than 1) and set priority to low -
|
||||||
|
// the old connection should be interrupted because it uses a worse
|
||||||
|
// transport
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(conn1).interruptOutgoingSession();
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId2, conn2, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
// Connect via transport 3 (no preference) and set priority to high -
|
||||||
|
// no further interruptions
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId3, conn3, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
|
||||||
|
// Unregister the interrupted connection (transport 1)
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.unregisterConnection(contactId1, transportId1, conn1, true, false);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
// The contact is not connected via transport 1 but is connected via a
|
||||||
|
// better transport
|
||||||
|
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId2));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId2));
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId3));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId3));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNewConnectionIsInterruptedIfOldConnectionHasHigherPriority() {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(emptyMap()));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Register a connection with high priority
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId1, conn1, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// Register another connection via the same transport (no priority yet)
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn2);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// Set the priority of the second connection to low - the second
|
||||||
|
// connection should be interrupted
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(conn2).interruptOutgoingSession();
|
||||||
|
}});
|
||||||
|
c.setPriority(contactId1, transportId1, conn2, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// Register a third connection with low priority - it should also be
|
||||||
|
// interrupted
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(conn3).interruptOutgoingSession();
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId1, conn3, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testOldConnectionIsInterruptedIfNewConnectionHasHigherPriority() {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(emptyMap()));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
// Register a connection with low priority
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerOutgoingConnection(contactId1, transportId1, conn1, low);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// Register another connection via the same transport (no priority yet)
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.registerIncomingConnection(contactId1, transportId1, conn2);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
|
||||||
|
// Set the priority of the second connection to high - the first
|
||||||
|
// connection should be interrupted
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(conn1).interruptOutgoingSession();
|
||||||
|
}});
|
||||||
|
c.setPriority(contactId1, transportId1, conn2, high);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedContacts(transportId1));
|
||||||
|
assertEquals(singletonList(contactId1),
|
||||||
|
c.getConnectedOrBetterContacts(transportId1));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRegisterAndUnregisterPendingContacts() {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(pluginConfig).getTransportPreferences();
|
||||||
|
will(returnValue(emptyMap()));
|
||||||
|
}});
|
||||||
|
|
||||||
|
ConnectionRegistry c =
|
||||||
|
new ConnectionRegistryImpl(eventBus, pluginConfig);
|
||||||
|
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(
|
||||||
|
RendezvousConnectionOpenedEvent.class)));
|
||||||
|
}});
|
||||||
|
assertTrue(c.registerConnection(pendingContactId));
|
||||||
|
assertFalse(c.registerConnection(pendingContactId)); // Redundant
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(eventBus).broadcast(with(any(
|
||||||
|
RendezvousConnectionClosedEvent.class)));
|
||||||
|
}});
|
||||||
|
c.unregisterConnection(pendingContactId, true);
|
||||||
|
context.assertIsSatisfied();
|
||||||
|
|
||||||
|
try {
|
||||||
|
c.unregisterConnection(pendingContactId, true);
|
||||||
|
fail();
|
||||||
|
} catch (IllegalArgumentException expected) {
|
||||||
|
// Expected
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -2,13 +2,13 @@ package org.briarproject.bramble.contact;
|
|||||||
|
|
||||||
import org.briarproject.bramble.BrambleCoreEagerSingletons;
|
import org.briarproject.bramble.BrambleCoreEagerSingletons;
|
||||||
import org.briarproject.bramble.BrambleCoreModule;
|
import org.briarproject.bramble.BrambleCoreModule;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
import org.briarproject.bramble.api.contact.ContactExchangeManager;
|
||||||
import org.briarproject.bramble.api.contact.ContactManager;
|
import org.briarproject.bramble.api.contact.ContactManager;
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.identity.IdentityManager;
|
import org.briarproject.bramble.api.identity.IdentityManager;
|
||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.lifecycle.LifecycleManager;
|
import org.briarproject.bramble.api.lifecycle.LifecycleManager;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.test.BrambleCoreIntegrationTestModule;
|
import org.briarproject.bramble.test.BrambleCoreIntegrationTestModule;
|
||||||
|
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
|
|||||||
@@ -1,25 +1,35 @@
|
|||||||
package org.briarproject.bramble.crypto;
|
package org.briarproject.bramble.crypto;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.crypto.DecryptionException;
|
||||||
|
import org.briarproject.bramble.api.crypto.KeyStrengthener;
|
||||||
|
import org.briarproject.bramble.api.crypto.SecretKey;
|
||||||
import org.briarproject.bramble.system.SystemClock;
|
import org.briarproject.bramble.system.SystemClock;
|
||||||
import org.briarproject.bramble.test.BrambleTestCase;
|
import org.briarproject.bramble.test.BrambleMockTestCase;
|
||||||
import org.briarproject.bramble.test.TestSecureRandomProvider;
|
import org.briarproject.bramble.test.TestSecureRandomProvider;
|
||||||
import org.briarproject.bramble.test.TestUtils;
|
import org.jmock.Expectations;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import java.util.Random;
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_CIPHERTEXT;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.INVALID_PASSWORD;
|
||||||
|
import static org.briarproject.bramble.api.crypto.DecryptionResult.KEY_STRENGTHENER_ERROR;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getSecretKey;
|
||||||
import static org.junit.Assert.assertArrayEquals;
|
import static org.junit.Assert.assertArrayEquals;
|
||||||
import static org.junit.Assert.assertNull;
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
public class PasswordBasedEncryptionTest extends BrambleTestCase {
|
public class PasswordBasedEncryptionTest extends BrambleMockTestCase {
|
||||||
|
|
||||||
|
private final KeyStrengthener keyStrengthener =
|
||||||
|
context.mock(KeyStrengthener.class);
|
||||||
|
|
||||||
private final CryptoComponentImpl crypto =
|
private final CryptoComponentImpl crypto =
|
||||||
new CryptoComponentImpl(new TestSecureRandomProvider(),
|
new CryptoComponentImpl(new TestSecureRandomProvider(),
|
||||||
new ScryptKdf(new SystemClock()));
|
new ScryptKdf(new SystemClock()));
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testEncryptionAndDecryption() {
|
public void testEncryptionAndDecryption() throws Exception {
|
||||||
byte[] input = TestUtils.getRandomBytes(1234);
|
byte[] input = getRandomBytes(1234);
|
||||||
String password = "password";
|
String password = "password";
|
||||||
byte[] ciphertext = crypto.encryptWithPassword(input, password, null);
|
byte[] ciphertext = crypto.encryptWithPassword(input, password, null);
|
||||||
byte[] output = crypto.decryptWithPassword(ciphertext, password, null);
|
byte[] output = crypto.decryptWithPassword(ciphertext, password, null);
|
||||||
@@ -27,14 +37,80 @@ public class PasswordBasedEncryptionTest extends BrambleTestCase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testInvalidCiphertextReturnsNull() {
|
public void testInvalidFormatVersionThrowsException() {
|
||||||
byte[] input = TestUtils.getRandomBytes(1234);
|
byte[] input = getRandomBytes(1234);
|
||||||
String password = "password";
|
String password = "password";
|
||||||
byte[] ciphertext = crypto.encryptWithPassword(input, password, null);
|
byte[] ciphertext = crypto.encryptWithPassword(input, password, null);
|
||||||
// Modify the ciphertext
|
|
||||||
int position = new Random().nextInt(ciphertext.length);
|
// Modify the format version
|
||||||
ciphertext[position] = (byte) (ciphertext[position] ^ 0xFF);
|
ciphertext[0] ^= (byte) 0xFF;
|
||||||
byte[] output = crypto.decryptWithPassword(ciphertext, password, null);
|
try {
|
||||||
assertNull(output);
|
crypto.decryptWithPassword(ciphertext, password, null);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_CIPHERTEXT, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testInvalidPasswordThrowsException() {
|
||||||
|
byte[] input = getRandomBytes(1234);
|
||||||
|
byte[] ciphertext = crypto.encryptWithPassword(input, "password", null);
|
||||||
|
|
||||||
|
// Try to decrypt with the wrong password
|
||||||
|
try {
|
||||||
|
crypto.decryptWithPassword(ciphertext, "wrong", null);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(INVALID_PASSWORD, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testMissingKeyStrengthenerThrowsException() {
|
||||||
|
SecretKey strengthened = getSecretKey();
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(keyStrengthener).strengthenKey(with(any(SecretKey.class)));
|
||||||
|
will(returnValue(strengthened));
|
||||||
|
}});
|
||||||
|
|
||||||
|
// Use the key strengthener during encryption
|
||||||
|
byte[] input = getRandomBytes(1234);
|
||||||
|
String password = "password";
|
||||||
|
byte[] ciphertext =
|
||||||
|
crypto.encryptWithPassword(input, password, keyStrengthener);
|
||||||
|
|
||||||
|
// The key strengthener is missing during decryption
|
||||||
|
try {
|
||||||
|
crypto.decryptWithPassword(ciphertext, password, null);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(KEY_STRENGTHENER_ERROR, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testKeyStrengthenerFailureThrowsException() {
|
||||||
|
SecretKey strengthened = getSecretKey();
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
oneOf(keyStrengthener).strengthenKey(with(any(SecretKey.class)));
|
||||||
|
will(returnValue(strengthened));
|
||||||
|
oneOf(keyStrengthener).isInitialised();
|
||||||
|
will(returnValue(false));
|
||||||
|
}});
|
||||||
|
|
||||||
|
// Use the key strengthener during encryption
|
||||||
|
byte[] input = getRandomBytes(1234);
|
||||||
|
String password = "password";
|
||||||
|
byte[] ciphertext =
|
||||||
|
crypto.encryptWithPassword(input, password, keyStrengthener);
|
||||||
|
|
||||||
|
// The key strengthener fails during decryption
|
||||||
|
try {
|
||||||
|
crypto.decryptWithPassword(ciphertext, password, keyStrengthener);
|
||||||
|
fail();
|
||||||
|
} catch (DecryptionException expected) {
|
||||||
|
assertEquals(KEY_STRENGTHENER_ERROR, expected.getDecryptionResult());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,143 @@
|
|||||||
|
package org.briarproject.bramble.io;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.test.BrambleTestCase;
|
||||||
|
import org.briarproject.bramble.test.SettableClock;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
import java.util.concurrent.atomic.AtomicLong;
|
||||||
|
|
||||||
|
import static java.util.concurrent.TimeUnit.MINUTES;
|
||||||
|
import static java.util.concurrent.TimeUnit.SECONDS;
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.assertFalse;
|
||||||
|
import static org.junit.Assert.assertTrue;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
public class TimeoutInputStreamTest extends BrambleTestCase {
|
||||||
|
|
||||||
|
private static final long TIMEOUT_MS = MINUTES.toMillis(1);
|
||||||
|
|
||||||
|
private final long now = System.currentTimeMillis();
|
||||||
|
|
||||||
|
private AtomicLong time;
|
||||||
|
private UnresponsiveInputStream in;
|
||||||
|
private AtomicBoolean listenerCalled;
|
||||||
|
private TimeoutInputStream stream;
|
||||||
|
private CountDownLatch readReturned;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() {
|
||||||
|
time = new AtomicLong(now);
|
||||||
|
in = new UnresponsiveInputStream();
|
||||||
|
listenerCalled = new AtomicBoolean(false);
|
||||||
|
stream = new TimeoutInputStream(new SettableClock(time), in,
|
||||||
|
TIMEOUT_MS, stream -> listenerCalled.set(true));
|
||||||
|
readReturned = new CountDownLatch(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testTimeoutIsReportedIfReadDoesNotReturn() throws Exception {
|
||||||
|
startReading();
|
||||||
|
try {
|
||||||
|
// The stream should not report a timeout
|
||||||
|
assertFalse(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// Time passes
|
||||||
|
time.set(now + TIMEOUT_MS);
|
||||||
|
|
||||||
|
// The stream still shouldn't report a timeout
|
||||||
|
assertFalse(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// Time passes
|
||||||
|
time.set(now + TIMEOUT_MS + 1);
|
||||||
|
|
||||||
|
// The stream should report a timeout
|
||||||
|
assertTrue(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// The listener should not have been called yet
|
||||||
|
assertFalse(listenerCalled.get());
|
||||||
|
|
||||||
|
// Close the stream
|
||||||
|
stream.close();
|
||||||
|
|
||||||
|
// The listener should have been called
|
||||||
|
assertTrue(listenerCalled.get());
|
||||||
|
} finally {
|
||||||
|
// Allow the read to return
|
||||||
|
in.readFinished.countDown();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testTimeoutIsNotReportedIfReadReturns() throws Exception {
|
||||||
|
startReading();
|
||||||
|
try {
|
||||||
|
// The stream should not report a timeout
|
||||||
|
assertFalse(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// Time passes
|
||||||
|
time.set(now + TIMEOUT_MS);
|
||||||
|
|
||||||
|
// The stream still shouldn't report a timeout
|
||||||
|
assertFalse(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// Allow the read to finish and wait for it to return
|
||||||
|
in.readFinished.countDown();
|
||||||
|
readReturned.await(10, SECONDS);
|
||||||
|
|
||||||
|
// Time passes
|
||||||
|
time.set(now + TIMEOUT_MS + 1);
|
||||||
|
|
||||||
|
// The stream should not report a timeout as the read has returned
|
||||||
|
assertFalse(stream.hasTimedOut());
|
||||||
|
|
||||||
|
// The listener should not have been called yet
|
||||||
|
assertFalse(listenerCalled.get());
|
||||||
|
|
||||||
|
// Close the stream
|
||||||
|
stream.close();
|
||||||
|
|
||||||
|
// The listener should have been called
|
||||||
|
assertTrue(listenerCalled.get());
|
||||||
|
} finally {
|
||||||
|
// Allow the read to return in case an assertion was thrown
|
||||||
|
in.readFinished.countDown();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void startReading() throws Exception {
|
||||||
|
// Start a background thread to read from the unresponsive stream
|
||||||
|
new Thread(() -> {
|
||||||
|
try {
|
||||||
|
assertEquals(123, stream.read());
|
||||||
|
readReturned.countDown();
|
||||||
|
} catch (IOException e) {
|
||||||
|
fail();
|
||||||
|
}
|
||||||
|
}).start();
|
||||||
|
// Wait for the background thread to start reading
|
||||||
|
assertTrue(in.readStarted.await(10, SECONDS));
|
||||||
|
}
|
||||||
|
|
||||||
|
private class UnresponsiveInputStream extends InputStream {
|
||||||
|
|
||||||
|
private final CountDownLatch readStarted = new CountDownLatch(1);
|
||||||
|
private final CountDownLatch readFinished = new CountDownLatch(1);
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int read() throws IOException {
|
||||||
|
readStarted.countDown();
|
||||||
|
try {
|
||||||
|
readFinished.await();
|
||||||
|
return 123;
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
throw new IOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,149 +0,0 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
|
||||||
|
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
|
||||||
import org.briarproject.bramble.api.contact.PendingContactId;
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ConnectionOpenedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactConnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.plugin.event.ContactDisconnectedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionClosedEvent;
|
|
||||||
import org.briarproject.bramble.api.rendezvous.event.RendezvousConnectionOpenedEvent;
|
|
||||||
import org.briarproject.bramble.test.BrambleMockTestCase;
|
|
||||||
import org.jmock.Expectations;
|
|
||||||
import org.junit.Test;
|
|
||||||
|
|
||||||
import java.util.Collection;
|
|
||||||
|
|
||||||
import static java.util.Collections.emptyList;
|
|
||||||
import static java.util.Collections.singletonList;
|
|
||||||
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
|
||||||
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
|
||||||
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
|
||||||
import static org.junit.Assert.assertEquals;
|
|
||||||
import static org.junit.Assert.assertFalse;
|
|
||||||
import static org.junit.Assert.assertTrue;
|
|
||||||
import static org.junit.Assert.fail;
|
|
||||||
|
|
||||||
public class ConnectionRegistryImplTest extends BrambleMockTestCase {
|
|
||||||
|
|
||||||
private final EventBus eventBus = context.mock(EventBus.class);
|
|
||||||
|
|
||||||
private final ContactId contactId = getContactId();
|
|
||||||
private final ContactId contactId1 = getContactId();
|
|
||||||
private final TransportId transportId = getTransportId();
|
|
||||||
private final TransportId transportId1 = getTransportId();
|
|
||||||
private final PendingContactId pendingContactId =
|
|
||||||
new PendingContactId(getRandomId());
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testRegisterAndUnregister() {
|
|
||||||
ConnectionRegistry c = new ConnectionRegistryImpl(eventBus);
|
|
||||||
|
|
||||||
// The registry should be empty
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId));
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
|
||||||
|
|
||||||
// Check that a registered connection shows up - this should
|
|
||||||
// broadcast a ConnectionOpenedEvent and a ContactConnectedEvent
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
|
||||||
oneOf(eventBus).broadcast(with(any(ContactConnectedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.registerConnection(contactId, transportId, true);
|
|
||||||
assertEquals(singletonList(contactId),
|
|
||||||
c.getConnectedContacts(transportId));
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
// Register an identical connection - this should broadcast a
|
|
||||||
// ConnectionOpenedEvent and lookup should be unaffected
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(ConnectionOpenedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.registerConnection(contactId, transportId, true);
|
|
||||||
assertEquals(singletonList(contactId),
|
|
||||||
c.getConnectedContacts(transportId));
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
// Unregister one of the connections - this should broadcast a
|
|
||||||
// ConnectionClosedEvent and lookup should be unaffected
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.unregisterConnection(contactId, transportId, true);
|
|
||||||
assertEquals(singletonList(contactId),
|
|
||||||
c.getConnectedContacts(transportId));
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
// Unregister the other connection - this should broadcast a
|
|
||||||
// ConnectionClosedEvent and a ContactDisconnectedEvent
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(ConnectionClosedEvent.class)));
|
|
||||||
oneOf(eventBus).broadcast(with(any(
|
|
||||||
ContactDisconnectedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.unregisterConnection(contactId, transportId, true);
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId));
|
|
||||||
assertEquals(emptyList(), c.getConnectedContacts(transportId1));
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
// Try to unregister the connection again - exception should be thrown
|
|
||||||
try {
|
|
||||||
c.unregisterConnection(contactId, transportId, true);
|
|
||||||
fail();
|
|
||||||
} catch (IllegalArgumentException expected) {
|
|
||||||
// Expected
|
|
||||||
}
|
|
||||||
|
|
||||||
// Register both contacts with one transport, one contact with both -
|
|
||||||
// this should broadcast three ConnectionOpenedEvents and two
|
|
||||||
// ContactConnectedEvents
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
exactly(3).of(eventBus).broadcast(with(any(
|
|
||||||
ConnectionOpenedEvent.class)));
|
|
||||||
exactly(2).of(eventBus).broadcast(with(any(
|
|
||||||
ContactConnectedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.registerConnection(contactId, transportId, true);
|
|
||||||
c.registerConnection(contactId1, transportId, true);
|
|
||||||
c.registerConnection(contactId1, transportId1, true);
|
|
||||||
Collection<ContactId> connected = c.getConnectedContacts(transportId);
|
|
||||||
assertEquals(2, connected.size());
|
|
||||||
assertTrue(connected.contains(contactId));
|
|
||||||
assertTrue(connected.contains(contactId1));
|
|
||||||
assertEquals(singletonList(contactId1),
|
|
||||||
c.getConnectedContacts(transportId1));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testRegisterAndUnregisterPendingContacts() {
|
|
||||||
ConnectionRegistry c = new ConnectionRegistryImpl(eventBus);
|
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(
|
|
||||||
RendezvousConnectionOpenedEvent.class)));
|
|
||||||
}});
|
|
||||||
assertTrue(c.registerConnection(pendingContactId));
|
|
||||||
assertFalse(c.registerConnection(pendingContactId)); // Redundant
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
|
||||||
oneOf(eventBus).broadcast(with(any(
|
|
||||||
RendezvousConnectionClosedEvent.class)));
|
|
||||||
}});
|
|
||||||
c.unregisterConnection(pendingContactId, true);
|
|
||||||
context.assertIsSatisfied();
|
|
||||||
|
|
||||||
try {
|
|
||||||
c.unregisterConnection(pendingContactId, true);
|
|
||||||
fail();
|
|
||||||
} catch (IllegalArgumentException expected) {
|
|
||||||
// Expected
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
package org.briarproject.bramble.plugin;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginCallback;
|
import org.briarproject.bramble.api.plugin.PluginCallback;
|
||||||
import org.briarproject.bramble.api.plugin.PluginConfig;
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
import org.briarproject.bramble.api.plugin.PluginException;
|
import org.briarproject.bramble.api.plugin.PluginException;
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
package org.briarproject.bramble.plugin;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionRegistry;
|
||||||
import org.briarproject.bramble.api.contact.ContactId;
|
import org.briarproject.bramble.api.contact.ContactId;
|
||||||
import org.briarproject.bramble.api.contact.event.ContactAddedEvent;
|
import org.briarproject.bramble.api.contact.event.ContactAddedEvent;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionRegistry;
|
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.PluginManager;
|
import org.briarproject.bramble.api.plugin.PluginManager;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
@@ -38,7 +38,7 @@ import static java.util.Collections.emptyList;
|
|||||||
import static java.util.Collections.singletonList;
|
import static java.util.Collections.singletonList;
|
||||||
import static java.util.Collections.singletonMap;
|
import static java.util.Collections.singletonMap;
|
||||||
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
import static java.util.concurrent.TimeUnit.MILLISECONDS;
|
||||||
import static org.briarproject.bramble.test.ListMatcher.listOf;
|
import static org.briarproject.bramble.test.CollectionMatcher.collectionOf;
|
||||||
import static org.briarproject.bramble.test.PairMatcher.pairOf;
|
import static org.briarproject.bramble.test.PairMatcher.pairOf;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
||||||
@@ -157,7 +157,21 @@ public class PollerImplTest extends BrambleMockTestCase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testRescheduleAndReconnectOnConnectionClosed()
|
public void testRescheduleOnOutgoingConnectionClosed() {
|
||||||
|
DuplexPlugin plugin = context.mock(DuplexPlugin.class);
|
||||||
|
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(plugin).getId();
|
||||||
|
will(returnValue(transportId));
|
||||||
|
}});
|
||||||
|
expectReschedule(plugin);
|
||||||
|
|
||||||
|
poller.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
|
||||||
|
false, false));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRescheduleAndReconnectOnOutgoingConnectionFailed()
|
||||||
throws Exception {
|
throws Exception {
|
||||||
DuplexPlugin plugin = context.mock(DuplexPlugin.class);
|
DuplexPlugin plugin = context.mock(DuplexPlugin.class);
|
||||||
DuplexTransportConnection duplexConnection =
|
DuplexTransportConnection duplexConnection =
|
||||||
@@ -166,45 +180,40 @@ public class PollerImplTest extends BrambleMockTestCase {
|
|||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
allowing(plugin).getId();
|
allowing(plugin).getId();
|
||||||
will(returnValue(transportId));
|
will(returnValue(transportId));
|
||||||
// reschedule()
|
|
||||||
// Get the plugin
|
|
||||||
oneOf(pluginManager).getPlugin(transportId);
|
|
||||||
will(returnValue(plugin));
|
|
||||||
// The plugin supports polling
|
|
||||||
oneOf(plugin).shouldPoll();
|
|
||||||
will(returnValue(true));
|
|
||||||
// Get the plugin
|
|
||||||
oneOf(pluginManager).getPlugin(transportId);
|
|
||||||
will(returnValue(plugin));
|
|
||||||
// The plugin supports polling
|
|
||||||
oneOf(plugin).shouldPoll();
|
|
||||||
will(returnValue(true));
|
|
||||||
// Schedule the next poll
|
|
||||||
oneOf(plugin).getPollingInterval();
|
|
||||||
will(returnValue(pollingInterval));
|
|
||||||
oneOf(clock).currentTimeMillis();
|
|
||||||
will(returnValue(now));
|
|
||||||
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
|
||||||
with((long) pollingInterval), with(MILLISECONDS));
|
|
||||||
will(returnValue(future));
|
|
||||||
// connectToContact()
|
|
||||||
// Check whether the contact is already connected
|
|
||||||
oneOf(connectionRegistry).isConnected(contactId, transportId);
|
|
||||||
will(returnValue(false));
|
|
||||||
// Get the transport properties
|
|
||||||
oneOf(transportPropertyManager).getRemoteProperties(contactId,
|
|
||||||
transportId);
|
|
||||||
will(returnValue(properties));
|
|
||||||
// Connect to the contact
|
|
||||||
oneOf(plugin).createConnection(properties);
|
|
||||||
will(returnValue(duplexConnection));
|
|
||||||
// Pass the connection to the connection manager
|
|
||||||
oneOf(connectionManager).manageOutgoingConnection(contactId,
|
|
||||||
transportId, duplexConnection);
|
|
||||||
}});
|
}});
|
||||||
|
expectReschedule(plugin);
|
||||||
|
expectReconnect(plugin, duplexConnection);
|
||||||
|
|
||||||
poller.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
|
poller.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
|
||||||
false));
|
false, true));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRescheduleOnIncomingConnectionClosed() {
|
||||||
|
DuplexPlugin plugin = context.mock(DuplexPlugin.class);
|
||||||
|
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(plugin).getId();
|
||||||
|
will(returnValue(transportId));
|
||||||
|
}});
|
||||||
|
expectReschedule(plugin);
|
||||||
|
|
||||||
|
poller.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
|
||||||
|
true, false));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRescheduleOnIncomingConnectionFailed() {
|
||||||
|
DuplexPlugin plugin = context.mock(DuplexPlugin.class);
|
||||||
|
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
allowing(plugin).getId();
|
||||||
|
will(returnValue(transportId));
|
||||||
|
}});
|
||||||
|
expectReschedule(plugin);
|
||||||
|
|
||||||
|
poller.eventOccurred(new ConnectionClosedEvent(contactId, transportId,
|
||||||
|
true, false));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
@@ -351,17 +360,14 @@ public class PollerImplTest extends BrambleMockTestCase {
|
|||||||
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
||||||
with((long) (pollingInterval * 0.5)), with(MILLISECONDS));
|
with((long) (pollingInterval * 0.5)), with(MILLISECONDS));
|
||||||
will(returnValue(future));
|
will(returnValue(future));
|
||||||
// FIXME: Revert
|
|
||||||
oneOf(plugin).supportsDiscovery();
|
|
||||||
will(returnValue(false));
|
|
||||||
// Get the transport properties and connected contacts
|
// Get the transport properties and connected contacts
|
||||||
oneOf(transportPropertyManager).getRemoteProperties(transportId);
|
oneOf(transportPropertyManager).getRemoteProperties(transportId);
|
||||||
will(returnValue(singletonMap(contactId, properties)));
|
will(returnValue(singletonMap(contactId, properties)));
|
||||||
oneOf(connectionRegistry).getConnectedContacts(transportId);
|
oneOf(connectionRegistry).getConnectedOrBetterContacts(transportId);
|
||||||
will(returnValue(emptyList()));
|
will(returnValue(emptyList()));
|
||||||
// Poll the plugin
|
// Poll the plugin
|
||||||
oneOf(plugin).poll(with(listOf(pairOf(
|
oneOf(plugin).poll(with(collectionOf(
|
||||||
equal(properties), any(ConnectionHandler.class)))));
|
pairOf(equal(properties), any(ConnectionHandler.class)))));
|
||||||
}});
|
}});
|
||||||
|
|
||||||
poller.eventOccurred(new TransportActiveEvent(transportId));
|
poller.eventOccurred(new TransportActiveEvent(transportId));
|
||||||
@@ -397,13 +403,10 @@ public class PollerImplTest extends BrambleMockTestCase {
|
|||||||
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
||||||
with((long) (pollingInterval * 0.5)), with(MILLISECONDS));
|
with((long) (pollingInterval * 0.5)), with(MILLISECONDS));
|
||||||
will(returnValue(future));
|
will(returnValue(future));
|
||||||
// FIXME: Revert
|
|
||||||
oneOf(plugin).supportsDiscovery();
|
|
||||||
will(returnValue(false));
|
|
||||||
// Get the transport properties and connected contacts
|
// Get the transport properties and connected contacts
|
||||||
oneOf(transportPropertyManager).getRemoteProperties(transportId);
|
oneOf(transportPropertyManager).getRemoteProperties(transportId);
|
||||||
will(returnValue(singletonMap(contactId, properties)));
|
will(returnValue(singletonMap(contactId, properties)));
|
||||||
oneOf(connectionRegistry).getConnectedContacts(transportId);
|
oneOf(connectionRegistry).getConnectedOrBetterContacts(transportId);
|
||||||
will(returnValue(singletonList(contactId)));
|
will(returnValue(singletonList(contactId)));
|
||||||
// All contacts are connected, so don't poll the plugin
|
// All contacts are connected, so don't poll the plugin
|
||||||
}});
|
}});
|
||||||
@@ -437,4 +440,48 @@ public class PollerImplTest extends BrambleMockTestCase {
|
|||||||
poller.eventOccurred(new TransportActiveEvent(transportId));
|
poller.eventOccurred(new TransportActiveEvent(transportId));
|
||||||
poller.eventOccurred(new TransportInactiveEvent(transportId));
|
poller.eventOccurred(new TransportInactiveEvent(transportId));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private void expectReschedule(Plugin plugin) {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
// Get the plugin
|
||||||
|
oneOf(pluginManager).getPlugin(transportId);
|
||||||
|
will(returnValue(plugin));
|
||||||
|
// The plugin supports polling
|
||||||
|
oneOf(plugin).shouldPoll();
|
||||||
|
will(returnValue(true));
|
||||||
|
// Schedule the next poll
|
||||||
|
oneOf(plugin).getPollingInterval();
|
||||||
|
will(returnValue(pollingInterval));
|
||||||
|
oneOf(clock).currentTimeMillis();
|
||||||
|
will(returnValue(now));
|
||||||
|
oneOf(scheduler).schedule(with(any(Runnable.class)),
|
||||||
|
with((long) pollingInterval), with(MILLISECONDS));
|
||||||
|
will(returnValue(future));
|
||||||
|
}});
|
||||||
|
}
|
||||||
|
|
||||||
|
private void expectReconnect(DuplexPlugin plugin,
|
||||||
|
DuplexTransportConnection duplexConnection) throws Exception {
|
||||||
|
context.checking(new Expectations() {{
|
||||||
|
// Get the plugin
|
||||||
|
oneOf(pluginManager).getPlugin(transportId);
|
||||||
|
will(returnValue(plugin));
|
||||||
|
// The plugin supports polling
|
||||||
|
oneOf(plugin).shouldPoll();
|
||||||
|
will(returnValue(true));
|
||||||
|
// Check whether the contact is already connected
|
||||||
|
oneOf(connectionRegistry).isConnected(contactId, transportId);
|
||||||
|
will(returnValue(false));
|
||||||
|
// Get the transport properties
|
||||||
|
oneOf(transportPropertyManager).getRemoteProperties(contactId,
|
||||||
|
transportId);
|
||||||
|
will(returnValue(properties));
|
||||||
|
// Connect to the contact
|
||||||
|
oneOf(plugin).createConnection(properties);
|
||||||
|
will(returnValue(duplexConnection));
|
||||||
|
// Pass the connection to the connection manager
|
||||||
|
oneOf(connectionManager).manageOutgoingConnection(contactId,
|
||||||
|
transportId, duplexConnection);
|
||||||
|
}});
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -24,14 +24,18 @@ import org.briarproject.bramble.test.DbExpectations;
|
|||||||
import org.jmock.Expectations;
|
import org.jmock.Expectations;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import java.util.Collections;
|
|
||||||
import java.util.LinkedHashMap;
|
import java.util.LinkedHashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
|
|
||||||
import static java.util.Arrays.asList;
|
import static java.util.Arrays.asList;
|
||||||
|
import static java.util.Collections.emptyMap;
|
||||||
import static java.util.Collections.singletonList;
|
import static java.util.Collections.singletonList;
|
||||||
import static java.util.Collections.singletonMap;
|
import static java.util.Collections.singletonMap;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.GROUP_KEY_DISCOVERED;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_LOCAL;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_TRANSPORT_ID;
|
||||||
|
import static org.briarproject.bramble.api.properties.TransportPropertyConstants.MSG_KEY_VERSION;
|
||||||
import static org.briarproject.bramble.api.properties.TransportPropertyManager.CLIENT_ID;
|
import static org.briarproject.bramble.api.properties.TransportPropertyManager.CLIENT_ID;
|
||||||
import static org.briarproject.bramble.api.properties.TransportPropertyManager.MAJOR_VERSION;
|
import static org.briarproject.bramble.api.properties.TransportPropertyManager.MAJOR_VERSION;
|
||||||
import static org.briarproject.bramble.api.sync.Group.Visibility.SHARED;
|
import static org.briarproject.bramble.api.sync.Group.Visibility.SHARED;
|
||||||
@@ -186,25 +190,25 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
Message message = getMessage(contactGroupId);
|
Message message = getMessage(contactGroupId);
|
||||||
Metadata meta = new Metadata();
|
Metadata meta = new Metadata();
|
||||||
BdfDictionary metaDictionary = BdfDictionary.of(
|
BdfDictionary metaDictionary = BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 2),
|
new BdfEntry(MSG_KEY_VERSION, 2),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
);
|
);
|
||||||
Map<MessageId, BdfDictionary> messageMetadata =
|
Map<MessageId, BdfDictionary> messageMetadata =
|
||||||
new LinkedHashMap<>();
|
new LinkedHashMap<>();
|
||||||
// A remote update for another transport should be ignored
|
// A remote update for another transport should be ignored
|
||||||
MessageId barUpdateId = new MessageId(getRandomId());
|
MessageId barUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "bar"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "bar"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
));
|
));
|
||||||
// A local update for the same transport should be ignored
|
// A local update for the same transport should be ignored
|
||||||
MessageId localUpdateId = new MessageId(getRandomId());
|
MessageId localUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata.put(localUpdateId, BdfDictionary.of(
|
messageMetadata.put(localUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
@@ -228,18 +232,18 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
Metadata meta = new Metadata();
|
Metadata meta = new Metadata();
|
||||||
// Version 4 is being delivered
|
// Version 4 is being delivered
|
||||||
BdfDictionary metaDictionary = BdfDictionary.of(
|
BdfDictionary metaDictionary = BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 4),
|
new BdfEntry(MSG_KEY_VERSION, 4),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
);
|
);
|
||||||
Map<MessageId, BdfDictionary> messageMetadata =
|
Map<MessageId, BdfDictionary> messageMetadata =
|
||||||
new LinkedHashMap<>();
|
new LinkedHashMap<>();
|
||||||
// An older remote update for the same transport should be deleted
|
// An older remote update for the same transport should be deleted
|
||||||
MessageId fooVersion3 = new MessageId(getRandomId());
|
MessageId fooVersion3 = new MessageId(getRandomId());
|
||||||
messageMetadata.put(fooVersion3, BdfDictionary.of(
|
messageMetadata.put(fooVersion3, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 3),
|
new BdfEntry(MSG_KEY_VERSION, 3),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
));
|
));
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
@@ -265,18 +269,18 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
Metadata meta = new Metadata();
|
Metadata meta = new Metadata();
|
||||||
// Version 3 is being delivered
|
// Version 3 is being delivered
|
||||||
BdfDictionary metaDictionary = BdfDictionary.of(
|
BdfDictionary metaDictionary = BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 3),
|
new BdfEntry(MSG_KEY_VERSION, 3),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
);
|
);
|
||||||
Map<MessageId, BdfDictionary> messageMetadata =
|
Map<MessageId, BdfDictionary> messageMetadata =
|
||||||
new LinkedHashMap<>();
|
new LinkedHashMap<>();
|
||||||
// A newer remote update for the same transport should not be deleted
|
// A newer remote update for the same transport should not be deleted
|
||||||
MessageId fooVersion4 = new MessageId(getRandomId());
|
MessageId fooVersion4 = new MessageId(getRandomId());
|
||||||
messageMetadata.put(fooVersion4, BdfDictionary.of(
|
messageMetadata.put(fooVersion4, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 4),
|
new BdfEntry(MSG_KEY_VERSION, 4),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
));
|
));
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
@@ -342,9 +346,9 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
// A local update for another transport should be ignored
|
// A local update for another transport should be ignored
|
||||||
MessageId barUpdateId = new MessageId(getRandomId());
|
MessageId barUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "bar"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "bar"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
|
|
||||||
context.checking(new DbExpectations() {{
|
context.checking(new DbExpectations() {{
|
||||||
@@ -366,16 +370,16 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
// A local update for another transport should be ignored
|
// A local update for another transport should be ignored
|
||||||
MessageId barUpdateId = new MessageId(getRandomId());
|
MessageId barUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "bar"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "bar"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
// A local update for the right transport should be returned
|
// A local update for the right transport should be returned
|
||||||
MessageId fooUpdateId = new MessageId(getRandomId());
|
MessageId fooUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata.put(fooUpdateId, BdfDictionary.of(
|
messageMetadata.put(fooUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
BdfList fooUpdate = BdfList.of("foo", 1, fooPropertiesDict);
|
BdfList fooUpdate = BdfList.of("foo", 1, fooPropertiesDict);
|
||||||
|
|
||||||
@@ -405,28 +409,28 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
List<Contact> contacts = asList(contact1, contact2);
|
List<Contact> contacts = asList(contact1, contact2);
|
||||||
Group contactGroup1 = getGroup(CLIENT_ID, MAJOR_VERSION);
|
Group contactGroup1 = getGroup(CLIENT_ID, MAJOR_VERSION);
|
||||||
Group contactGroup2 = getGroup(CLIENT_ID, MAJOR_VERSION);
|
Group contactGroup2 = getGroup(CLIENT_ID, MAJOR_VERSION);
|
||||||
Map<MessageId, BdfDictionary> messageMetadata2 =
|
Map<MessageId, BdfDictionary> messageMetadata =
|
||||||
new LinkedHashMap<>();
|
new LinkedHashMap<>();
|
||||||
// A remote update for another transport should be ignored
|
// A remote update for another transport should be ignored
|
||||||
MessageId barUpdateId = new MessageId(getRandomId());
|
MessageId barUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata2.put(barUpdateId, BdfDictionary.of(
|
messageMetadata.put(barUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "bar"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "bar"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
));
|
));
|
||||||
// A local update for the right transport should be ignored
|
// A local update for the right transport should be ignored
|
||||||
MessageId localUpdateId = new MessageId(getRandomId());
|
MessageId localUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata2.put(localUpdateId, BdfDictionary.of(
|
messageMetadata.put(localUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
// A remote update for the right transport should be returned
|
// A remote update for the right transport should be returned
|
||||||
MessageId fooUpdateId = new MessageId(getRandomId());
|
MessageId fooUpdateId = new MessageId(getRandomId());
|
||||||
messageMetadata2.put(fooUpdateId, BdfDictionary.of(
|
messageMetadata.put(fooUpdateId, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", false)
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
));
|
));
|
||||||
BdfList fooUpdate = BdfList.of("foo", 1, fooPropertiesDict);
|
BdfList fooUpdate = BdfList.of("foo", 1, fooPropertiesDict);
|
||||||
|
|
||||||
@@ -440,19 +444,25 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
will(returnValue(contactGroup1));
|
will(returnValue(contactGroup1));
|
||||||
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
||||||
contactGroup1.getId());
|
contactGroup1.getId());
|
||||||
will(returnValue(Collections.emptyMap()));
|
will(returnValue(emptyMap()));
|
||||||
|
oneOf(clientHelper).getGroupMetadataAsDictionary(txn,
|
||||||
|
contactGroup1.getId());
|
||||||
|
will(returnValue(new BdfDictionary()));
|
||||||
// Second contact: returns an update
|
// Second contact: returns an update
|
||||||
oneOf(contactGroupFactory).createContactGroup(CLIENT_ID,
|
oneOf(contactGroupFactory).createContactGroup(CLIENT_ID,
|
||||||
MAJOR_VERSION, contact2);
|
MAJOR_VERSION, contact2);
|
||||||
will(returnValue(contactGroup2));
|
will(returnValue(contactGroup2));
|
||||||
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
||||||
contactGroup2.getId());
|
contactGroup2.getId());
|
||||||
will(returnValue(messageMetadata2));
|
will(returnValue(messageMetadata));
|
||||||
oneOf(clientHelper).getMessageAsList(txn, fooUpdateId);
|
oneOf(clientHelper).getMessageAsList(txn, fooUpdateId);
|
||||||
will(returnValue(fooUpdate));
|
will(returnValue(fooUpdate));
|
||||||
oneOf(clientHelper).parseAndValidateTransportProperties(
|
oneOf(clientHelper).parseAndValidateTransportProperties(
|
||||||
fooPropertiesDict);
|
fooPropertiesDict);
|
||||||
will(returnValue(fooProperties));
|
will(returnValue(fooProperties));
|
||||||
|
oneOf(clientHelper).getGroupMetadataAsDictionary(txn,
|
||||||
|
contactGroup2.getId());
|
||||||
|
will(returnValue(new BdfDictionary()));
|
||||||
}});
|
}});
|
||||||
|
|
||||||
TransportPropertyManagerImpl t = createInstance();
|
TransportPropertyManagerImpl t = createInstance();
|
||||||
@@ -463,6 +473,62 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
assertEquals(fooProperties, properties.get(contact2.getId()));
|
assertEquals(fooProperties, properties.get(contact2.getId()));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testReceivePropertiesOverrideDiscoveredProperties()
|
||||||
|
throws Exception {
|
||||||
|
Transaction txn = new Transaction(null, true);
|
||||||
|
Contact contact = getContact();
|
||||||
|
List<Contact> contacts = singletonList(contact);
|
||||||
|
Group contactGroup = getGroup(CLIENT_ID, MAJOR_VERSION);
|
||||||
|
MessageId updateId = new MessageId(getRandomId());
|
||||||
|
Map<MessageId, BdfDictionary> messageMetadata = singletonMap(updateId,
|
||||||
|
BdfDictionary.of(
|
||||||
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
|
new BdfEntry(MSG_KEY_LOCAL, false)
|
||||||
|
));
|
||||||
|
BdfList update = BdfList.of("foo", 1, fooPropertiesDict);
|
||||||
|
TransportProperties discovered = new TransportProperties();
|
||||||
|
discovered.put("fooKey1", "overridden");
|
||||||
|
discovered.put("fooKey3", "fooValue3");
|
||||||
|
BdfDictionary discoveredDict = new BdfDictionary(discovered);
|
||||||
|
BdfDictionary groupMeta = BdfDictionary.of(
|
||||||
|
new BdfEntry(GROUP_KEY_DISCOVERED, discoveredDict)
|
||||||
|
);
|
||||||
|
TransportProperties merged = new TransportProperties();
|
||||||
|
merged.putAll(fooProperties);
|
||||||
|
merged.put("fooKey3", "fooValue3");
|
||||||
|
|
||||||
|
context.checking(new DbExpectations() {{
|
||||||
|
oneOf(db).transactionWithResult(with(true), withDbCallable(txn));
|
||||||
|
oneOf(db).getContacts(txn);
|
||||||
|
will(returnValue(contacts));
|
||||||
|
// One update
|
||||||
|
oneOf(contactGroupFactory).createContactGroup(CLIENT_ID,
|
||||||
|
MAJOR_VERSION, contact);
|
||||||
|
will(returnValue(contactGroup));
|
||||||
|
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
||||||
|
contactGroup.getId());
|
||||||
|
will(returnValue(messageMetadata));
|
||||||
|
oneOf(clientHelper).getMessageAsList(txn, updateId);
|
||||||
|
will(returnValue(update));
|
||||||
|
oneOf(clientHelper).parseAndValidateTransportProperties(
|
||||||
|
fooPropertiesDict);
|
||||||
|
will(returnValue(fooProperties));
|
||||||
|
oneOf(clientHelper).getGroupMetadataAsDictionary(txn,
|
||||||
|
contactGroup.getId());
|
||||||
|
will(returnValue(groupMeta));
|
||||||
|
oneOf(clientHelper).parseAndValidateTransportProperties(
|
||||||
|
discoveredDict);
|
||||||
|
will(returnValue(discovered));
|
||||||
|
}});
|
||||||
|
|
||||||
|
TransportPropertyManagerImpl t = createInstance();
|
||||||
|
Map<ContactId, TransportProperties> properties =
|
||||||
|
t.getRemoteProperties(new TransportId("foo"));
|
||||||
|
assertEquals(merged, properties.get(contact.getId()));
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testMergingUnchangedPropertiesDoesNotCreateUpdate()
|
public void testMergingUnchangedPropertiesDoesNotCreateUpdate()
|
||||||
throws Exception {
|
throws Exception {
|
||||||
@@ -470,9 +536,9 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
MessageId updateId = new MessageId(getRandomId());
|
MessageId updateId = new MessageId(getRandomId());
|
||||||
Map<MessageId, BdfDictionary> messageMetadata = singletonMap(updateId,
|
Map<MessageId, BdfDictionary> messageMetadata = singletonMap(updateId,
|
||||||
BdfDictionary.of(
|
BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
));
|
));
|
||||||
BdfList update = BdfList.of("foo", 1, fooPropertiesDict);
|
BdfList update = BdfList.of("foo", 1, fooPropertiesDict);
|
||||||
|
|
||||||
@@ -505,7 +571,7 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
// There are no existing properties to merge with
|
// There are no existing properties to merge with
|
||||||
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
||||||
localGroup.getId());
|
localGroup.getId());
|
||||||
will(returnValue(Collections.emptyMap()));
|
will(returnValue(emptyMap()));
|
||||||
// Store the new properties in the local group, version 1
|
// Store the new properties in the local group, version 1
|
||||||
expectStoreMessage(txn, localGroup.getId(), "foo",
|
expectStoreMessage(txn, localGroup.getId(), "foo",
|
||||||
fooPropertiesDict, 1, true, false);
|
fooPropertiesDict, 1, true, false);
|
||||||
@@ -517,7 +583,7 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
will(returnValue(contactGroup));
|
will(returnValue(contactGroup));
|
||||||
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
oneOf(clientHelper).getMessageMetadataAsDictionary(txn,
|
||||||
contactGroup.getId());
|
contactGroup.getId());
|
||||||
will(returnValue(Collections.emptyMap()));
|
will(returnValue(emptyMap()));
|
||||||
expectStoreMessage(txn, contactGroup.getId(), "foo",
|
expectStoreMessage(txn, contactGroup.getId(), "foo",
|
||||||
fooPropertiesDict, 1, true, true);
|
fooPropertiesDict, 1, true, true);
|
||||||
}});
|
}});
|
||||||
@@ -532,9 +598,9 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
Contact contact = getContact();
|
Contact contact = getContact();
|
||||||
Group contactGroup = getGroup(CLIENT_ID, MAJOR_VERSION);
|
Group contactGroup = getGroup(CLIENT_ID, MAJOR_VERSION);
|
||||||
BdfDictionary oldMetadata = BdfDictionary.of(
|
BdfDictionary oldMetadata = BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 1),
|
new BdfEntry(MSG_KEY_VERSION, 1),
|
||||||
new BdfEntry("local", true)
|
new BdfEntry(MSG_KEY_LOCAL, true)
|
||||||
);
|
);
|
||||||
MessageId localGroupUpdateId = new MessageId(getRandomId());
|
MessageId localGroupUpdateId = new MessageId(getRandomId());
|
||||||
Map<MessageId, BdfDictionary> localGroupMessageMetadata =
|
Map<MessageId, BdfDictionary> localGroupMessageMetadata =
|
||||||
@@ -589,14 +655,14 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
// The latest update for transport "foo" should be returned
|
// The latest update for transport "foo" should be returned
|
||||||
MessageId fooVersion999 = new MessageId(getRandomId());
|
MessageId fooVersion999 = new MessageId(getRandomId());
|
||||||
messageMetadata.put(fooVersion999, BdfDictionary.of(
|
messageMetadata.put(fooVersion999, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "foo"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "foo"),
|
||||||
new BdfEntry("version", 999)
|
new BdfEntry(MSG_KEY_VERSION, 999)
|
||||||
));
|
));
|
||||||
// The latest update for transport "bar" should be returned
|
// The latest update for transport "bar" should be returned
|
||||||
MessageId barVersion3 = new MessageId(getRandomId());
|
MessageId barVersion3 = new MessageId(getRandomId());
|
||||||
messageMetadata.put(barVersion3, BdfDictionary.of(
|
messageMetadata.put(barVersion3, BdfDictionary.of(
|
||||||
new BdfEntry("transportId", "bar"),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, "bar"),
|
||||||
new BdfEntry("version", 3)
|
new BdfEntry(MSG_KEY_VERSION, 3)
|
||||||
));
|
));
|
||||||
BdfList fooUpdate = BdfList.of("foo", 999, fooPropertiesDict);
|
BdfList fooUpdate = BdfList.of("foo", 999, fooPropertiesDict);
|
||||||
BdfList barUpdate = BdfList.of("bar", 3, barPropertiesDict);
|
BdfList barUpdate = BdfList.of("bar", 3, barPropertiesDict);
|
||||||
@@ -627,9 +693,9 @@ public class TransportPropertyManagerImplTest extends BrambleMockTestCase {
|
|||||||
Message message = getMessage(g);
|
Message message = getMessage(g);
|
||||||
long timestamp = message.getTimestamp();
|
long timestamp = message.getTimestamp();
|
||||||
BdfDictionary meta = BdfDictionary.of(
|
BdfDictionary meta = BdfDictionary.of(
|
||||||
new BdfEntry("transportId", transportId),
|
new BdfEntry(MSG_KEY_TRANSPORT_ID, transportId),
|
||||||
new BdfEntry("version", version),
|
new BdfEntry(MSG_KEY_VERSION, version),
|
||||||
new BdfEntry("local", local)
|
new BdfEntry(MSG_KEY_LOCAL, local)
|
||||||
);
|
);
|
||||||
|
|
||||||
context.checking(new Expectations() {{
|
context.checking(new Expectations() {{
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
package org.briarproject.bramble.rendezvous;
|
package org.briarproject.bramble.rendezvous;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.connection.ConnectionManager;
|
||||||
import org.briarproject.bramble.api.contact.PendingContact;
|
import org.briarproject.bramble.api.contact.PendingContact;
|
||||||
import org.briarproject.bramble.api.contact.PendingContactState;
|
import org.briarproject.bramble.api.contact.PendingContactState;
|
||||||
import org.briarproject.bramble.api.contact.event.PendingContactAddedEvent;
|
import org.briarproject.bramble.api.contact.event.PendingContactAddedEvent;
|
||||||
@@ -13,7 +14,6 @@ import org.briarproject.bramble.api.db.Transaction;
|
|||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
import org.briarproject.bramble.api.identity.IdentityManager;
|
import org.briarproject.bramble.api.identity.IdentityManager;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
import org.briarproject.bramble.api.plugin.ConnectionHandler;
|
||||||
import org.briarproject.bramble.api.plugin.ConnectionManager;
|
|
||||||
import org.briarproject.bramble.api.plugin.PluginManager;
|
import org.briarproject.bramble.api.plugin.PluginManager;
|
||||||
import org.briarproject.bramble.api.plugin.TransportId;
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexPlugin;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexPlugin;
|
||||||
@@ -49,7 +49,7 @@ import static org.briarproject.bramble.api.contact.PendingContactState.OFFLINE;
|
|||||||
import static org.briarproject.bramble.api.contact.PendingContactState.WAITING_FOR_CONNECTION;
|
import static org.briarproject.bramble.api.contact.PendingContactState.WAITING_FOR_CONNECTION;
|
||||||
import static org.briarproject.bramble.rendezvous.RendezvousConstants.POLLING_INTERVAL_MS;
|
import static org.briarproject.bramble.rendezvous.RendezvousConstants.POLLING_INTERVAL_MS;
|
||||||
import static org.briarproject.bramble.rendezvous.RendezvousConstants.RENDEZVOUS_TIMEOUT_MS;
|
import static org.briarproject.bramble.rendezvous.RendezvousConstants.RENDEZVOUS_TIMEOUT_MS;
|
||||||
import static org.briarproject.bramble.test.ListMatcher.listOf;
|
import static org.briarproject.bramble.test.CollectionMatcher.collectionOf;
|
||||||
import static org.briarproject.bramble.test.PairMatcher.pairOf;
|
import static org.briarproject.bramble.test.PairMatcher.pairOf;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getAgreementPrivateKey;
|
import static org.briarproject.bramble.test.TestUtils.getAgreementPrivateKey;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getAgreementPublicKey;
|
import static org.briarproject.bramble.test.TestUtils.getAgreementPublicKey;
|
||||||
@@ -196,7 +196,7 @@ public class RendezvousPollerImplTest extends BrambleMockTestCase {
|
|||||||
oneOf(clock).currentTimeMillis();
|
oneOf(clock).currentTimeMillis();
|
||||||
will(returnValue(beforeExpiry));
|
will(returnValue(beforeExpiry));
|
||||||
oneOf(eventBus).broadcast(with(any(RendezvousPollEvent.class)));
|
oneOf(eventBus).broadcast(with(any(RendezvousPollEvent.class)));
|
||||||
oneOf(plugin).poll(with(listOf(pairOf(
|
oneOf(plugin).poll(with(collectionOf(pairOf(
|
||||||
equal(transportProperties),
|
equal(transportProperties),
|
||||||
any(ConnectionHandler.class)))));
|
any(ConnectionHandler.class)))));
|
||||||
}});
|
}});
|
||||||
@@ -248,7 +248,7 @@ public class RendezvousPollerImplTest extends BrambleMockTestCase {
|
|||||||
oneOf(clock).currentTimeMillis();
|
oneOf(clock).currentTimeMillis();
|
||||||
will(returnValue(beforeExpiry));
|
will(returnValue(beforeExpiry));
|
||||||
oneOf(eventBus).broadcast(with(any(RendezvousPollEvent.class)));
|
oneOf(eventBus).broadcast(with(any(RendezvousPollEvent.class)));
|
||||||
oneOf(plugin).poll(with(listOf(pairOf(
|
oneOf(plugin).poll(with(collectionOf(pairOf(
|
||||||
equal(transportProperties),
|
equal(transportProperties),
|
||||||
any(ConnectionHandler.class)))));
|
any(ConnectionHandler.class)))));
|
||||||
}});
|
}});
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import org.briarproject.bramble.api.contact.ContactId;
|
|||||||
import org.briarproject.bramble.api.db.DatabaseComponent;
|
import org.briarproject.bramble.api.db.DatabaseComponent;
|
||||||
import org.briarproject.bramble.api.db.Transaction;
|
import org.briarproject.bramble.api.db.Transaction;
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.sync.Ack;
|
import org.briarproject.bramble.api.sync.Ack;
|
||||||
import org.briarproject.bramble.api.sync.GroupId;
|
import org.briarproject.bramble.api.sync.GroupId;
|
||||||
import org.briarproject.bramble.api.sync.Message;
|
import org.briarproject.bramble.api.sync.Message;
|
||||||
@@ -23,6 +24,7 @@ import static org.briarproject.bramble.api.sync.SyncConstants.MAX_MESSAGE_IDS;
|
|||||||
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
import static org.briarproject.bramble.test.TestUtils.getContactId;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getMessage;
|
import static org.briarproject.bramble.test.TestUtils.getMessage;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
||||||
|
|
||||||
public class SimplexOutgoingSessionTest extends BrambleMockTestCase {
|
public class SimplexOutgoingSessionTest extends BrambleMockTestCase {
|
||||||
|
|
||||||
@@ -36,14 +38,15 @@ public class SimplexOutgoingSessionTest extends BrambleMockTestCase {
|
|||||||
|
|
||||||
private final Executor dbExecutor = new ImmediateExecutor();
|
private final Executor dbExecutor = new ImmediateExecutor();
|
||||||
private final ContactId contactId = getContactId();
|
private final ContactId contactId = getContactId();
|
||||||
|
private final TransportId transportId = getTransportId();
|
||||||
private final Message message = getMessage(new GroupId(getRandomId()));
|
private final Message message = getMessage(new GroupId(getRandomId()));
|
||||||
private final MessageId messageId = message.getId();
|
private final MessageId messageId = message.getId();
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testNothingToSend() throws Exception {
|
public void testNothingToSend() throws Exception {
|
||||||
SimplexOutgoingSession session = new SimplexOutgoingSession(db,
|
SimplexOutgoingSession session = new SimplexOutgoingSession(db,
|
||||||
dbExecutor, eventBus, contactId, MAX_LATENCY, streamWriter,
|
dbExecutor, eventBus, contactId, transportId, MAX_LATENCY,
|
||||||
recordWriter);
|
streamWriter, recordWriter);
|
||||||
Transaction noAckTxn = new Transaction(null, false);
|
Transaction noAckTxn = new Transaction(null, false);
|
||||||
Transaction noMsgTxn = new Transaction(null, false);
|
Transaction noMsgTxn = new Transaction(null, false);
|
||||||
|
|
||||||
@@ -76,8 +79,8 @@ public class SimplexOutgoingSessionTest extends BrambleMockTestCase {
|
|||||||
public void testSomethingToSend() throws Exception {
|
public void testSomethingToSend() throws Exception {
|
||||||
Ack ack = new Ack(singletonList(messageId));
|
Ack ack = new Ack(singletonList(messageId));
|
||||||
SimplexOutgoingSession session = new SimplexOutgoingSession(db,
|
SimplexOutgoingSession session = new SimplexOutgoingSession(db,
|
||||||
dbExecutor, eventBus, contactId, MAX_LATENCY, streamWriter,
|
dbExecutor, eventBus, contactId, transportId, MAX_LATENCY,
|
||||||
recordWriter);
|
streamWriter, recordWriter);
|
||||||
Transaction ackTxn = new Transaction(null, false);
|
Transaction ackTxn = new Transaction(null, false);
|
||||||
Transaction noAckTxn = new Transaction(null, false);
|
Transaction noAckTxn = new Transaction(null, false);
|
||||||
Transaction msgTxn = new Transaction(null, false);
|
Transaction msgTxn = new Transaction(null, false);
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import org.briarproject.bramble.api.record.RecordReader;
|
|||||||
import org.briarproject.bramble.api.sync.Ack;
|
import org.briarproject.bramble.api.sync.Ack;
|
||||||
import org.briarproject.bramble.api.sync.MessageFactory;
|
import org.briarproject.bramble.api.sync.MessageFactory;
|
||||||
import org.briarproject.bramble.api.sync.Offer;
|
import org.briarproject.bramble.api.sync.Offer;
|
||||||
|
import org.briarproject.bramble.api.sync.Priority;
|
||||||
import org.briarproject.bramble.api.sync.Request;
|
import org.briarproject.bramble.api.sync.Request;
|
||||||
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
import org.briarproject.bramble.api.sync.SyncRecordReader;
|
||||||
import org.briarproject.bramble.api.sync.Versions;
|
import org.briarproject.bramble.api.sync.Versions;
|
||||||
@@ -24,11 +25,14 @@ import javax.annotation.Nullable;
|
|||||||
import static org.briarproject.bramble.api.record.Record.MAX_RECORD_PAYLOAD_BYTES;
|
import static org.briarproject.bramble.api.record.Record.MAX_RECORD_PAYLOAD_BYTES;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
import static org.briarproject.bramble.api.sync.RecordTypes.ACK;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
import static org.briarproject.bramble.api.sync.RecordTypes.OFFER;
|
||||||
|
import static org.briarproject.bramble.api.sync.RecordTypes.PRIORITY;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
import static org.briarproject.bramble.api.sync.RecordTypes.REQUEST;
|
||||||
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
import static org.briarproject.bramble.api.sync.RecordTypes.VERSIONS;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_MESSAGE_IDS;
|
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_MESSAGE_IDS;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_SUPPORTED_VERSIONS;
|
import static org.briarproject.bramble.api.sync.SyncConstants.MAX_SUPPORTED_VERSIONS;
|
||||||
|
import static org.briarproject.bramble.api.sync.SyncConstants.PRIORITY_NONCE_BYTES;
|
||||||
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
import static org.briarproject.bramble.api.sync.SyncConstants.PROTOCOL_VERSION;
|
||||||
|
import static org.briarproject.bramble.test.TestUtils.getRandomBytes;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
import static org.briarproject.bramble.test.TestUtils.getRandomId;
|
||||||
import static org.junit.Assert.assertEquals;
|
import static org.junit.Assert.assertEquals;
|
||||||
import static org.junit.Assert.assertFalse;
|
import static org.junit.Assert.assertFalse;
|
||||||
@@ -119,6 +123,31 @@ public class SyncRecordReaderImplTest extends BrambleMockTestCase {
|
|||||||
reader.readVersions();
|
reader.readVersions();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test(expected = FormatException.class)
|
||||||
|
public void testFormatExceptionIfPriorityNonceIsTooSmall()
|
||||||
|
throws Exception {
|
||||||
|
expectReadRecord(createPriority(PRIORITY_NONCE_BYTES - 1));
|
||||||
|
|
||||||
|
reader.readPriority();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(expected = FormatException.class)
|
||||||
|
public void testFormatExceptionIfPriorityNonceIsTooLarge()
|
||||||
|
throws Exception {
|
||||||
|
expectReadRecord(createPriority(PRIORITY_NONCE_BYTES + 1));
|
||||||
|
|
||||||
|
reader.readPriority();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNoFormatExceptionIfPriorityNonceIsCorrectSize()
|
||||||
|
throws Exception {
|
||||||
|
expectReadRecord(createPriority(PRIORITY_NONCE_BYTES));
|
||||||
|
|
||||||
|
Priority priority = reader.readPriority();
|
||||||
|
assertEquals(PRIORITY_NONCE_BYTES, priority.getNonce().length);
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testEofReturnsTrueWhenAtEndOfStream() throws Exception {
|
public void testEofReturnsTrueWhenAtEndOfStream() throws Exception {
|
||||||
expectReadRecord(createAck());
|
expectReadRecord(createAck());
|
||||||
@@ -173,6 +202,11 @@ public class SyncRecordReaderImplTest extends BrambleMockTestCase {
|
|||||||
return new Record(PROTOCOL_VERSION, VERSIONS, payload);
|
return new Record(PROTOCOL_VERSION, VERSIONS, payload);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private Record createPriority(int nonceBytes) {
|
||||||
|
byte[] payload = getRandomBytes(nonceBytes);
|
||||||
|
return new Record(PROTOCOL_VERSION, PRIORITY, payload);
|
||||||
|
}
|
||||||
|
|
||||||
private byte[] createPayload() throws Exception {
|
private byte[] createPayload() throws Exception {
|
||||||
ByteArrayOutputStream payload = new ByteArrayOutputStream();
|
ByteArrayOutputStream payload = new ByteArrayOutputStream();
|
||||||
while (payload.size() + UniqueId.LENGTH <= MAX_RECORD_PAYLOAD_BYTES) {
|
while (payload.size() + UniqueId.LENGTH <= MAX_RECORD_PAYLOAD_BYTES) {
|
||||||
|
|||||||
@@ -5,24 +5,24 @@ import org.hamcrest.BaseMatcher;
|
|||||||
import org.hamcrest.Description;
|
import org.hamcrest.Description;
|
||||||
import org.hamcrest.Matcher;
|
import org.hamcrest.Matcher;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.Collection;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
@NotNullByDefault
|
@NotNullByDefault
|
||||||
public class ListMatcher<T> extends BaseMatcher<List<T>> {
|
public class CollectionMatcher<T> extends BaseMatcher<Collection<T>> {
|
||||||
|
|
||||||
private final Matcher<T> elementMatcher;
|
private final Matcher<T> elementMatcher;
|
||||||
|
|
||||||
public ListMatcher(Matcher<T> elementMatcher) {
|
public CollectionMatcher(Matcher<T> elementMatcher) {
|
||||||
this.elementMatcher = elementMatcher;
|
this.elementMatcher = elementMatcher;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public boolean matches(@Nullable Object item) {
|
public boolean matches(@Nullable Object item) {
|
||||||
if (!(item instanceof List)) return false;
|
if (!(item instanceof Collection)) return false;
|
||||||
List list = (List) item;
|
Collection collection = (Collection) item;
|
||||||
for (Object element : list) {
|
for (Object element : collection) {
|
||||||
if (!elementMatcher.matches(element)) return false;
|
if (!elementMatcher.matches(element)) return false;
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
@@ -33,7 +33,7 @@ public class ListMatcher<T> extends BaseMatcher<List<T>> {
|
|||||||
description.appendText("matches a collection");
|
description.appendText("matches a collection");
|
||||||
}
|
}
|
||||||
|
|
||||||
public static <T> ListMatcher<T> listOf(Matcher<T> t) {
|
public static <T> CollectionMatcher<T> collectionOf(Matcher<T> t) {
|
||||||
return new ListMatcher<>(t);
|
return new CollectionMatcher<>(t);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -4,6 +4,7 @@ import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
|||||||
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
|
||||||
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
|
||||||
|
import org.briarproject.bramble.api.properties.TransportProperties;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
@@ -37,6 +38,11 @@ public class TestDuplexTransportConnection
|
|||||||
return writer;
|
return writer;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public TransportProperties getRemoteProperties() {
|
||||||
|
return new TransportProperties();
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Creates and returns a pair of TestDuplexTransportConnections that are
|
* Creates and returns a pair of TestDuplexTransportConnections that are
|
||||||
* connected to each other.
|
* connected to each other.
|
||||||
|
|||||||
@@ -10,12 +10,15 @@ import org.briarproject.bramble.api.plugin.simplex.SimplexPlugin;
|
|||||||
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
||||||
|
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
import dagger.Module;
|
import dagger.Module;
|
||||||
import dagger.Provides;
|
import dagger.Provides;
|
||||||
|
|
||||||
|
import static java.util.Collections.emptyMap;
|
||||||
import static java.util.Collections.singletonList;
|
import static java.util.Collections.singletonList;
|
||||||
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
import static org.briarproject.bramble.test.TestUtils.getTransportId;
|
||||||
|
|
||||||
@@ -85,6 +88,12 @@ public class TestPluginConfigModule {
|
|||||||
public boolean shouldPoll() {
|
public boolean shouldPoll() {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Map<TransportId, List<TransportId>> getTransportPreferences() {
|
||||||
|
return emptyMap();
|
||||||
|
}
|
||||||
|
|
||||||
};
|
};
|
||||||
return pluginConfig;
|
return pluginConfig;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ dependencies {
|
|||||||
implementation fileTree(dir: 'libs', include: '*.jar')
|
implementation fileTree(dir: 'libs', include: '*.jar')
|
||||||
implementation 'net.java.dev.jna:jna:4.5.2'
|
implementation 'net.java.dev.jna:jna:4.5.2'
|
||||||
implementation 'net.java.dev.jna:jna-platform:4.5.2'
|
implementation 'net.java.dev.jna:jna-platform:4.5.2'
|
||||||
tor 'org.briarproject:tor:0.3.5.9@zip'
|
tor 'org.briarproject:tor:0.3.5.10@zip'
|
||||||
tor 'org.briarproject:obfs4proxy:0.0.7@zip'
|
tor 'org.briarproject:obfs4proxy:0.0.7@zip'
|
||||||
|
|
||||||
annotationProcessor 'com.google.dagger:dagger-compiler:2.24'
|
annotationProcessor 'com.google.dagger:dagger-compiler:2.24'
|
||||||
|
|||||||
@@ -1,11 +1,15 @@
|
|||||||
package org.briarproject.bramble.plugin;
|
package org.briarproject.bramble.plugin;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
import org.briarproject.bramble.api.lifecycle.IoExecutor;
|
||||||
import org.briarproject.bramble.api.lifecycle.ShutdownManager;
|
import org.briarproject.bramble.api.lifecycle.ShutdownManager;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
||||||
|
import org.briarproject.bramble.api.plugin.BluetoothConstants;
|
||||||
|
import org.briarproject.bramble.api.plugin.LanTcpConstants;
|
||||||
import org.briarproject.bramble.api.plugin.PluginConfig;
|
import org.briarproject.bramble.api.plugin.PluginConfig;
|
||||||
|
import org.briarproject.bramble.api.plugin.TransportId;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.DuplexPluginFactory;
|
import org.briarproject.bramble.api.plugin.duplex.DuplexPluginFactory;
|
||||||
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
import org.briarproject.bramble.api.plugin.simplex.SimplexPluginFactory;
|
||||||
import org.briarproject.bramble.api.reliability.ReliabilityLayerFactory;
|
import org.briarproject.bramble.api.reliability.ReliabilityLayerFactory;
|
||||||
@@ -16,6 +20,8 @@ import org.briarproject.bramble.plugin.tcp.WanTcpPluginFactory;
|
|||||||
|
|
||||||
import java.security.SecureRandom;
|
import java.security.SecureRandom;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
import java.util.concurrent.Executor;
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
import dagger.Module;
|
import dagger.Module;
|
||||||
@@ -23,6 +29,8 @@ import dagger.Provides;
|
|||||||
|
|
||||||
import static java.util.Arrays.asList;
|
import static java.util.Arrays.asList;
|
||||||
import static java.util.Collections.emptyList;
|
import static java.util.Collections.emptyList;
|
||||||
|
import static java.util.Collections.singletonList;
|
||||||
|
import static java.util.Collections.singletonMap;
|
||||||
|
|
||||||
@Module
|
@Module
|
||||||
public class DesktopPluginModule extends PluginModule {
|
public class DesktopPluginModule extends PluginModule {
|
||||||
@@ -31,10 +39,10 @@ public class DesktopPluginModule extends PluginModule {
|
|||||||
PluginConfig getPluginConfig(@IoExecutor Executor ioExecutor,
|
PluginConfig getPluginConfig(@IoExecutor Executor ioExecutor,
|
||||||
SecureRandom random, BackoffFactory backoffFactory,
|
SecureRandom random, BackoffFactory backoffFactory,
|
||||||
ReliabilityLayerFactory reliabilityFactory,
|
ReliabilityLayerFactory reliabilityFactory,
|
||||||
ShutdownManager shutdownManager, EventBus eventBus) {
|
ShutdownManager shutdownManager, EventBus eventBus,
|
||||||
DuplexPluginFactory bluetooth =
|
TimeoutMonitor timeoutMonitor) {
|
||||||
new JavaBluetoothPluginFactory(ioExecutor, random, eventBus,
|
DuplexPluginFactory bluetooth = new JavaBluetoothPluginFactory(
|
||||||
backoffFactory);
|
ioExecutor, random, eventBus, timeoutMonitor, backoffFactory);
|
||||||
DuplexPluginFactory modem = new ModemPluginFactory(ioExecutor,
|
DuplexPluginFactory modem = new ModemPluginFactory(ioExecutor,
|
||||||
reliabilityFactory);
|
reliabilityFactory);
|
||||||
DuplexPluginFactory lan = new LanTcpPluginFactory(ioExecutor, eventBus,
|
DuplexPluginFactory lan = new LanTcpPluginFactory(ioExecutor, eventBus,
|
||||||
@@ -60,6 +68,13 @@ public class DesktopPluginModule extends PluginModule {
|
|||||||
public boolean shouldPoll() {
|
public boolean shouldPoll() {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Map<TransportId, List<TransportId>> getTransportPreferences() {
|
||||||
|
// Prefer LAN to Bluetooth
|
||||||
|
return singletonMap(BluetoothConstants.ID,
|
||||||
|
singletonList(LanTcpConstants.ID));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
return pluginConfig;
|
return pluginConfig;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
package org.briarproject.bramble.plugin.bluetooth;
|
package org.briarproject.bramble.plugin.bluetooth;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.MethodsNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.ParametersNotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
@@ -34,10 +35,11 @@ class JavaBluetoothPlugin extends BluetoothPlugin<StreamConnectionNotifier> {
|
|||||||
private volatile LocalDevice localDevice = null;
|
private volatile LocalDevice localDevice = null;
|
||||||
|
|
||||||
JavaBluetoothPlugin(BluetoothConnectionLimiter connectionManager,
|
JavaBluetoothPlugin(BluetoothConnectionLimiter connectionManager,
|
||||||
Executor ioExecutor, SecureRandom secureRandom,
|
TimeoutMonitor timeoutMonitor, Executor ioExecutor,
|
||||||
Backoff backoff, PluginCallback callback, int maxLatency) {
|
SecureRandom secureRandom, Backoff backoff,
|
||||||
super(connectionManager, ioExecutor, secureRandom, backoff, callback,
|
PluginCallback callback, int maxLatency, int maxIdleTime) {
|
||||||
maxLatency);
|
super(connectionManager, timeoutMonitor, ioExecutor, secureRandom,
|
||||||
|
backoff, callback, maxLatency, maxIdleTime);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -119,7 +121,9 @@ class JavaBluetoothPlugin extends BluetoothPlugin<StreamConnectionNotifier> {
|
|||||||
return "btspp://" + address + ":" + uuid + ";name=RFCOMM";
|
return "btspp://" + address + ":" + uuid + ";name=RFCOMM";
|
||||||
}
|
}
|
||||||
|
|
||||||
private DuplexTransportConnection wrapSocket(StreamConnection s) {
|
private DuplexTransportConnection wrapSocket(StreamConnection s)
|
||||||
return new JavaBluetoothTransportConnection(this, connectionLimiter, s);
|
throws IOException {
|
||||||
|
return new JavaBluetoothTransportConnection(this, connectionLimiter,
|
||||||
|
timeoutMonitor, s);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package org.briarproject.bramble.plugin.bluetooth;
|
package org.briarproject.bramble.plugin.bluetooth;
|
||||||
|
|
||||||
import org.briarproject.bramble.api.event.EventBus;
|
import org.briarproject.bramble.api.event.EventBus;
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Backoff;
|
import org.briarproject.bramble.api.plugin.Backoff;
|
||||||
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
import org.briarproject.bramble.api.plugin.BackoffFactory;
|
||||||
@@ -21,22 +22,25 @@ import static org.briarproject.bramble.api.plugin.BluetoothConstants.ID;
|
|||||||
public class JavaBluetoothPluginFactory implements DuplexPluginFactory {
|
public class JavaBluetoothPluginFactory implements DuplexPluginFactory {
|
||||||
|
|
||||||
private static final int MAX_LATENCY = 30 * 1000; // 30 seconds
|
private static final int MAX_LATENCY = 30 * 1000; // 30 seconds
|
||||||
|
private static final int MAX_IDLE_TIME = 30 * 1000; // 30 seconds
|
||||||
private static final int MIN_POLLING_INTERVAL = 60 * 1000; // 1 minute
|
private static final int MIN_POLLING_INTERVAL = 60 * 1000; // 1 minute
|
||||||
private static final int MAX_POLLING_INTERVAL = 10 * 60 * 1000; // 10 mins
|
private static final int MAX_POLLING_INTERVAL = 10 * 60 * 1000; // 10 mins
|
||||||
private static final double BACKOFF_BASE = 1.2;
|
private static final double BACKOFF_BASE = 1.2;
|
||||||
|
|
||||||
private final Executor ioExecutor;
|
private final Executor ioExecutor;
|
||||||
private final SecureRandom secureRandom;
|
private final SecureRandom secureRandom;
|
||||||
private final BackoffFactory backoffFactory;
|
|
||||||
private final EventBus eventBus;
|
private final EventBus eventBus;
|
||||||
|
private final TimeoutMonitor timeoutMonitor;
|
||||||
|
private final BackoffFactory backoffFactory;
|
||||||
|
|
||||||
public JavaBluetoothPluginFactory(Executor ioExecutor,
|
public JavaBluetoothPluginFactory(Executor ioExecutor,
|
||||||
SecureRandom secureRandom, EventBus eventBus,
|
SecureRandom secureRandom, EventBus eventBus,
|
||||||
BackoffFactory backoffFactory) {
|
TimeoutMonitor timeoutMonitor, BackoffFactory backoffFactory) {
|
||||||
this.ioExecutor = ioExecutor;
|
this.ioExecutor = ioExecutor;
|
||||||
this.secureRandom = secureRandom;
|
this.secureRandom = secureRandom;
|
||||||
this.backoffFactory = backoffFactory;
|
|
||||||
this.eventBus = eventBus;
|
this.eventBus = eventBus;
|
||||||
|
this.timeoutMonitor = timeoutMonitor;
|
||||||
|
this.backoffFactory = backoffFactory;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -52,11 +56,12 @@ public class JavaBluetoothPluginFactory implements DuplexPluginFactory {
|
|||||||
@Override
|
@Override
|
||||||
public DuplexPlugin createPlugin(PluginCallback callback) {
|
public DuplexPlugin createPlugin(PluginCallback callback) {
|
||||||
BluetoothConnectionLimiter connectionLimiter =
|
BluetoothConnectionLimiter connectionLimiter =
|
||||||
new BluetoothConnectionLimiterImpl();
|
new BluetoothConnectionLimiterImpl(eventBus);
|
||||||
Backoff backoff = backoffFactory.createBackoff(MIN_POLLING_INTERVAL,
|
Backoff backoff = backoffFactory.createBackoff(MIN_POLLING_INTERVAL,
|
||||||
MAX_POLLING_INTERVAL, BACKOFF_BASE);
|
MAX_POLLING_INTERVAL, BACKOFF_BASE);
|
||||||
JavaBluetoothPlugin plugin = new JavaBluetoothPlugin(connectionLimiter,
|
JavaBluetoothPlugin plugin = new JavaBluetoothPlugin(connectionLimiter,
|
||||||
ioExecutor, secureRandom, backoff, callback, MAX_LATENCY);
|
timeoutMonitor, ioExecutor, secureRandom, backoff, callback,
|
||||||
|
MAX_LATENCY, MAX_IDLE_TIME);
|
||||||
eventBus.addListener(plugin);
|
eventBus.addListener(plugin);
|
||||||
return plugin;
|
return plugin;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
package org.briarproject.bramble.plugin.bluetooth;
|
package org.briarproject.bramble.plugin.bluetooth;
|
||||||
|
|
||||||
|
import org.briarproject.bramble.api.io.TimeoutMonitor;
|
||||||
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
import org.briarproject.bramble.api.nullsafety.NotNullByDefault;
|
||||||
import org.briarproject.bramble.api.plugin.Plugin;
|
import org.briarproject.bramble.api.plugin.Plugin;
|
||||||
import org.briarproject.bramble.api.plugin.duplex.AbstractDuplexTransportConnection;
|
import org.briarproject.bramble.api.plugin.duplex.AbstractDuplexTransportConnection;
|
||||||
@@ -14,20 +15,24 @@ import javax.microedition.io.StreamConnection;
|
|||||||
class JavaBluetoothTransportConnection
|
class JavaBluetoothTransportConnection
|
||||||
extends AbstractDuplexTransportConnection {
|
extends AbstractDuplexTransportConnection {
|
||||||
|
|
||||||
private final BluetoothConnectionLimiter connectionManager;
|
private final BluetoothConnectionLimiter connectionLimiter;
|
||||||
private final StreamConnection stream;
|
private final StreamConnection stream;
|
||||||
|
private final InputStream in;
|
||||||
|
|
||||||
JavaBluetoothTransportConnection(Plugin plugin,
|
JavaBluetoothTransportConnection(Plugin plugin,
|
||||||
BluetoothConnectionLimiter connectionManager,
|
BluetoothConnectionLimiter connectionLimiter,
|
||||||
StreamConnection stream) {
|
TimeoutMonitor timeoutMonitor,
|
||||||
|
StreamConnection stream) throws IOException {
|
||||||
super(plugin);
|
super(plugin);
|
||||||
|
this.connectionLimiter = connectionLimiter;
|
||||||
this.stream = stream;
|
this.stream = stream;
|
||||||
this.connectionManager = connectionManager;
|
in = timeoutMonitor.createTimeoutInputStream(
|
||||||
|
stream.openInputStream(), plugin.getMaxIdleTime() * 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected InputStream getInputStream() throws IOException {
|
protected InputStream getInputStream() {
|
||||||
return stream.openInputStream();
|
return in;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
@@ -40,7 +45,7 @@ class JavaBluetoothTransportConnection
|
|||||||
try {
|
try {
|
||||||
stream.close();
|
stream.close();
|
||||||
} finally {
|
} finally {
|
||||||
connectionManager.connectionClosed(this);
|
connectionLimiter.connectionClosed(this);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user