Include unique device ID in transport updates.

This commit is contained in:
akwizgran
2016-01-27 12:16:09 +00:00
parent 88475bdd54
commit c776d1e893
8 changed files with 79 additions and 18 deletions

View File

@@ -0,0 +1,16 @@
package org.briarproject.api;
import java.util.Arrays;
/** Type-safe wrapper for a byte array that uniquely identifies a device. */
public class DeviceId extends UniqueId {
public DeviceId(byte[] id) {
super(id);
}
@Override
public boolean equals(Object o) {
return o instanceof DeviceId && Arrays.equals(id, ((DeviceId) o).id);
}
}

View File

@@ -1,5 +1,6 @@
package org.briarproject.api.db; package org.briarproject.api.db;
import org.briarproject.api.DeviceId;
import org.briarproject.api.TransportId; import org.briarproject.api.TransportId;
import org.briarproject.api.contact.Contact; import org.briarproject.api.contact.Contact;
import org.briarproject.api.contact.ContactId; import org.briarproject.api.contact.ContactId;
@@ -137,6 +138,9 @@ public interface DatabaseComponent {
/** Returns all contacts associated with the given local pseudonym. */ /** Returns all contacts associated with the given local pseudonym. */
Collection<ContactId> getContacts(AuthorId a) throws DbException; Collection<ContactId> getContacts(AuthorId a) throws DbException;
/** Returns the unique ID for this device. */
DeviceId getDeviceId() throws DbException;
/** Returns the group with the given ID, if the user subscribes to it. */ /** Returns the group with the given ID, if the user subscribes to it. */
Group getGroup(GroupId g) throws DbException; Group getGroup(GroupId g) throws DbException;

View File

@@ -1,5 +1,6 @@
package org.briarproject.db; package org.briarproject.db;
import org.briarproject.api.DeviceId;
import org.briarproject.api.TransportId; import org.briarproject.api.TransportId;
import org.briarproject.api.contact.Contact; import org.briarproject.api.contact.Contact;
import org.briarproject.api.contact.ContactId; import org.briarproject.api.contact.ContactId;
@@ -256,6 +257,13 @@ interface Database<T> {
*/ */
Collection<ContactId> getContacts(T txn, AuthorId a) throws DbException; Collection<ContactId> getContacts(T txn, AuthorId a) throws DbException;
/**
* Returns the unique ID for this device.
* <p>
* Locking: read.
*/
DeviceId getDeviceId(T txn) throws DbException;
/** /**
* Returns the amount of free storage space available to the database, in * Returns the amount of free storage space available to the database, in
* bytes. This is based on the minimum of the space available on the device * bytes. This is based on the minimum of the space available on the device

View File

@@ -1,5 +1,6 @@
package org.briarproject.db; package org.briarproject.db;
import org.briarproject.api.DeviceId;
import org.briarproject.api.TransportId; import org.briarproject.api.TransportId;
import org.briarproject.api.contact.Contact; import org.briarproject.api.contact.Contact;
import org.briarproject.api.contact.ContactId; import org.briarproject.api.contact.ContactId;
@@ -542,6 +543,23 @@ class DatabaseComponentImpl<T> implements DatabaseComponent {
} }
} }
public DeviceId getDeviceId() throws DbException {
lock.readLock().lock();
try {
T txn = db.startTransaction();
try {
DeviceId id = db.getDeviceId(txn);
db.commitTransaction(txn);
return id;
} catch (DbException e) {
db.abortTransaction(txn);
throw e;
}
} finally {
lock.readLock().unlock();
}
}
public Group getGroup(GroupId g) throws DbException { public Group getGroup(GroupId g) throws DbException {
lock.readLock().lock(); lock.readLock().lock();
try { try {

View File

@@ -1,5 +1,6 @@
package org.briarproject.db; package org.briarproject.db;
import org.briarproject.api.DeviceId;
import org.briarproject.api.TransportId; import org.briarproject.api.TransportId;
import org.briarproject.api.UniqueId; import org.briarproject.api.UniqueId;
import org.briarproject.api.contact.Contact; import org.briarproject.api.contact.Contact;
@@ -484,6 +485,11 @@ abstract class JdbcDatabase implements Database<Connection> {
if (interrupted) Thread.currentThread().interrupt(); if (interrupted) Thread.currentThread().interrupt();
} }
public DeviceId getDeviceId(Connection txn) throws DbException {
Settings s = getSettings(txn, DEVICE_SETTINGS_NAMESPACE);
return new DeviceId(StringUtils.fromHexString(s.get(DEVICE_ID_KEY)));
}
public ContactId addContact(Connection txn, Author remote, AuthorId local) public ContactId addContact(Connection txn, Author remote, AuthorId local)
throws DbException { throws DbException {
PreparedStatement ps = null; PreparedStatement ps = null;

View File

@@ -2,6 +2,7 @@ package org.briarproject.properties;
import com.google.inject.Inject; import com.google.inject.Inject;
import org.briarproject.api.DeviceId;
import org.briarproject.api.FormatException; import org.briarproject.api.FormatException;
import org.briarproject.api.TransportId; import org.briarproject.api.TransportId;
import org.briarproject.api.UniqueId; import org.briarproject.api.UniqueId;
@@ -43,6 +44,7 @@ import java.util.Map.Entry;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.logging.Logger; import java.util.logging.Logger;
import static java.util.logging.Level.INFO;
import static java.util.logging.Level.WARNING; import static java.util.logging.Level.WARNING;
import static org.briarproject.api.properties.TransportPropertyConstants.MAX_PROPERTY_LENGTH; import static org.briarproject.api.properties.TransportPropertyConstants.MAX_PROPERTY_LENGTH;
import static org.briarproject.api.sync.SyncConstants.MESSAGE_HEADER_LENGTH; import static org.briarproject.api.sync.SyncConstants.MESSAGE_HEADER_LENGTH;
@@ -100,9 +102,10 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
db.addContactGroup(c, g); db.addContactGroup(c, g);
db.setVisibility(g.getId(), Collections.singletonList(c)); db.setVisibility(g.getId(), Collections.singletonList(c));
// Copy the latest local properties into the group // Copy the latest local properties into the group
DeviceId dev = db.getDeviceId();
Map<TransportId, TransportProperties> local = getLocalProperties(); Map<TransportId, TransportProperties> local = getLocalProperties();
for (Entry<TransportId, TransportProperties> e : local.entrySet()) for (Entry<TransportId, TransportProperties> e : local.entrySet())
storeMessage(g.getId(), e.getKey(), e.getValue(), 0); storeMessage(g.getId(), dev, e.getKey(), e.getValue(), 0);
} catch (DbException e) { } catch (DbException e) {
if (LOG.isLoggable(WARNING)) LOG.log(WARNING, e.toString(), e); if (LOG.isLoggable(WARNING)) LOG.log(WARNING, e.toString(), e);
} catch (FormatException e) { } catch (FormatException e) {
@@ -141,9 +144,10 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
return out.toByteArray(); return out.toByteArray();
} }
private void storeMessage(GroupId g, TransportId t, TransportProperties p, private void storeMessage(GroupId g, DeviceId dev, TransportId t,
long version) throws DbException, IOException { TransportProperties p, long version) throws DbException,
byte[] body = encodeProperties(t, p, version); IOException {
byte[] body = encodeProperties(dev, t, p, version);
long now = clock.currentTimeMillis(); long now = clock.currentTimeMillis();
Message m = messageFactory.createMessage(g, now, body); Message m = messageFactory.createMessage(g, now, body);
BdfDictionary d = new BdfDictionary(); BdfDictionary d = new BdfDictionary();
@@ -153,13 +157,13 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
db.addLocalMessage(m, CLIENT_ID, metadataEncoder.encode(d)); db.addLocalMessage(m, CLIENT_ID, metadataEncoder.encode(d));
} }
private byte[] encodeProperties(TransportId t, TransportProperties p, private byte[] encodeProperties(DeviceId dev, TransportId t,
long version) { TransportProperties p, long version) {
ByteArrayOutputStream out = new ByteArrayOutputStream(); ByteArrayOutputStream out = new ByteArrayOutputStream();
BdfWriter w = bdfWriterFactory.createWriter(out); BdfWriter w = bdfWriterFactory.createWriter(out);
try { try {
// TODO: Device ID
w.writeListStart(); w.writeListStart();
w.writeRaw(dev.getBytes());
w.writeString(t.getString()); w.writeString(t.getString());
w.writeInteger(version); w.writeInteger(version);
w.writeDictionary(p); w.writeDictionary(p);
@@ -232,8 +236,8 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
ByteArrayInputStream in = new ByteArrayInputStream(raw, ByteArrayInputStream in = new ByteArrayInputStream(raw,
MESSAGE_HEADER_LENGTH, raw.length - MESSAGE_HEADER_LENGTH); MESSAGE_HEADER_LENGTH, raw.length - MESSAGE_HEADER_LENGTH);
BdfReader r = bdfReaderFactory.createReader(in); BdfReader r = bdfReaderFactory.createReader(in);
// TODO: Device ID
r.readListStart(); r.readListStart();
r.skipRaw(); // Device ID
r.skipString(); // Transport ID r.skipString(); // Transport ID
r.skipInteger(); // Version r.skipInteger(); // Version
r.readDictionaryStart(); r.readDictionaryStart();
@@ -242,9 +246,6 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
String value = r.readString(MAX_PROPERTY_LENGTH); String value = r.readString(MAX_PROPERTY_LENGTH);
p.put(key, value); p.put(key, value);
} }
r.readDictionaryEnd();
r.readListEnd();
if (!r.eof()) throw new FormatException();
return p; return p;
} }
@@ -310,14 +311,19 @@ class TransportPropertyManagerImpl implements TransportPropertyManager,
p = old; p = old;
} }
// Store the merged properties in the local group // Store the merged properties in the local group
DeviceId dev = db.getDeviceId();
long version = latest == null ? 0 : latest.version + 1; long version = latest == null ? 0 : latest.version + 1;
storeMessage(localGroup.getId(), t, p, version); storeMessage(localGroup.getId(), dev, t, p, version);
if (LOG.isLoggable(INFO)) {
LOG.info("Stored local properties for " + t.getString()
+ ", version " + version);
}
// Store the merged properties in each contact's group // Store the merged properties in each contact's group
for (Contact c : db.getContacts()) { for (Contact c : db.getContacts()) {
Group g = getContactGroup(c); Group g = getContactGroup(c);
latest = findLatest(g.getId(), true).get(t); latest = findLatest(g.getId(), true).get(t);
version = latest == null ? 0 : latest.version + 1; version = latest == null ? 0 : latest.version + 1;
storeMessage(g.getId(), t, p, version); storeMessage(g.getId(), dev, t, p, version);
} }
} catch (IOException e) { } catch (IOException e) {
throw new DbException(e); throw new DbException(e);

View File

@@ -1,6 +1,7 @@
package org.briarproject.properties; package org.briarproject.properties;
import org.briarproject.api.FormatException; import org.briarproject.api.FormatException;
import org.briarproject.api.UniqueId;
import org.briarproject.api.data.BdfDictionary; import org.briarproject.api.data.BdfDictionary;
import org.briarproject.api.data.BdfReader; import org.briarproject.api.data.BdfReader;
import org.briarproject.api.data.BdfReaderFactory; import org.briarproject.api.data.BdfReaderFactory;
@@ -53,10 +54,11 @@ class TransportPropertyValidator implements MessageValidator {
ByteArrayInputStream in = new ByteArrayInputStream(raw, ByteArrayInputStream in = new ByteArrayInputStream(raw,
MESSAGE_HEADER_LENGTH, raw.length - MESSAGE_HEADER_LENGTH); MESSAGE_HEADER_LENGTH, raw.length - MESSAGE_HEADER_LENGTH);
BdfReader r = bdfReaderFactory.createReader(in); BdfReader r = bdfReaderFactory.createReader(in);
// TODO: Device ID
r.readListStart(); r.readListStart();
String id = r.readString(MAX_TRANSPORT_ID_LENGTH); byte[] deviceId = r.readRaw(UniqueId.LENGTH);
if (id.length() == 0) throw new FormatException(); if (deviceId.length != UniqueId.LENGTH) throw new FormatException();
String transportId = r.readString(MAX_TRANSPORT_ID_LENGTH);
if (transportId.length() == 0) throw new FormatException();
long version = r.readInteger(); long version = r.readInteger();
if (version < 0) throw new FormatException(); if (version < 0) throw new FormatException();
r.readDictionaryStart(); r.readDictionaryStart();
@@ -71,7 +73,7 @@ class TransportPropertyValidator implements MessageValidator {
if (!r.eof()) throw new FormatException(); if (!r.eof()) throw new FormatException();
// Return the metadata // Return the metadata
BdfDictionary d = new BdfDictionary(); BdfDictionary d = new BdfDictionary();
d.put("transportId", id); d.put("transportId", transportId);
d.put("version", version); d.put("version", version);
d.put("local", false); d.put("local", false);
return metadataEncoder.encode(d); return metadataEncoder.encode(d);

View File

@@ -65,7 +65,8 @@ public class StringUtils {
/** Converts the given hex string to a byte array. */ /** Converts the given hex string to a byte array. */
public static byte[] fromHexString(String hex) { public static byte[] fromHexString(String hex) {
int len = hex.length(); int len = hex.length();
if (len % 2 != 0) throw new IllegalArgumentException("Not a hex string"); if (len % 2 != 0)
throw new IllegalArgumentException("Not a hex string");
byte[] bytes = new byte[len / 2]; byte[] bytes = new byte[len / 2];
for (int i = 0, j = 0; i < len; i += 2, j++) { for (int i = 0, j = 0; i < len; i += 2, j++) {
int high = hexDigitToInt(hex.charAt(i)); int high = hexDigitToInt(hex.charAt(i));