Readers, writers and factories for subscription and transport updates.

components/net/sf/briar/db/ReadWriteLockDatabaseComponent.java

@@ -259,7 +259,7 @@ class ReadWriteLockDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				try {
 					Collection<BatchId> acks = db.getBatchesToAck(txn, c);
 					Collection<BatchId> sent = new ArrayList<BatchId>();
-					for(BatchId b : acks) if(a.addBatchId(b)) sent.add(b);
+					for(BatchId b : acks) if(a.writeBatchId(b)) sent.add(b);
 					a.finish();
 					db.removeBatchesToAck(txn, c, sent);
 					if(LOG.isLoggable(Level.FINE))
@@ -300,7 +300,7 @@ class ReadWriteLockDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 						while(it.hasNext()) {
 							MessageId m = it.next();
 							byte[] message = db.getMessage(txn, m);
-							if(!b.addMessage(message)) break;
+							if(!b.writeMessage(message)) break;
 							bytesSent += message.length;
 							sent.add(m);
 						}
@@ -349,7 +349,7 @@ class ReadWriteLockDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				Txn txn = db.startTransaction();
 				try {
 					Collection<Group> subs = db.getSubscriptions(txn);
-					s.setSubscriptions(subs);
+					s.writeSubscriptions(subs);
 					if(LOG.isLoggable(Level.FINE))
 						LOG.fine("Added " + subs.size() + " subscriptions");
 					db.commitTransaction(txn);
@@ -378,7 +378,7 @@ class ReadWriteLockDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				Txn txn = db.startTransaction();
 				try {
 					Map<String, String> transports = db.getTransports(txn);
-					t.setTransports(transports);
+					t.writeTransports(transports);
 					if(LOG.isLoggable(Level.FINE))
 						LOG.fine("Added " + transports.size() + " transports");
 					db.commitTransaction(txn);

components/net/sf/briar/db/SynchronizedDatabaseComponent.java

@@ -192,7 +192,7 @@ class SynchronizedDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				try {
 					Collection<BatchId> acks = db.getBatchesToAck(txn, c);
 					Collection<BatchId> sent = new ArrayList<BatchId>();
-					for(BatchId b : acks) if(a.addBatchId(b)) sent.add(b);
+					for(BatchId b : acks) if(a.writeBatchId(b)) sent.add(b);
 					a.finish();
 					db.removeBatchesToAck(txn, c, sent);
 					if(LOG.isLoggable(Level.FINE))
@@ -225,7 +225,7 @@ class SynchronizedDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 						while(it.hasNext()) {
 							MessageId m = it.next();
 							byte[] message = db.getMessage(txn, m);
-							if(!b.addMessage(message)) break;
+							if(!b.writeMessage(message)) break;
 							bytesSent += message.length;
 							sent.add(m);
 						}
@@ -254,7 +254,7 @@ class SynchronizedDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				Txn txn = db.startTransaction();
 				try {
 					Collection<Group> subs = db.getSubscriptions(txn);
-					s.setSubscriptions(subs);
+					s.writeSubscriptions(subs);
 					if(LOG.isLoggable(Level.FINE))
 						LOG.fine("Added " + subs.size() + " subscriptions");
 					db.commitTransaction(txn);
@@ -277,7 +277,7 @@ class SynchronizedDatabaseComponent<Txn> extends DatabaseComponentImpl<Txn> {
 				Txn txn = db.startTransaction();
 				try {
 					Map<String, String> transports = db.getTransports(txn);
-					t.setTransports(transports);
+					t.writeTransports(transports);
 					if(LOG.isLoggable(Level.FINE))
 						LOG.fine("Added " + transports.size() + " transports");
 					db.commitTransaction(txn);

components/net/sf/briar/protocol/AckReader.java

@@ -11,9 +11,11 @@ import net.sf.briar.api.serial.Reader;
 
 class AckReader implements ObjectReader<Ack> {
 
+	private final ObjectReader<BatchId> batchIdReader;
 	private final AckFactory ackFactory;
 
-	AckReader(AckFactory ackFactory) {
+	AckReader(ObjectReader<BatchId> batchIdReader, AckFactory ackFactory) {
+		this.batchIdReader = batchIdReader;
 		this.ackFactory = ackFactory;
 	}
 
@@ -23,7 +25,7 @@ class AckReader implements ObjectReader<Ack> {
 		// Read and digest the data
 		r.addConsumer(counting);
 		r.readUserDefinedTag(Tags.ACK);
-		r.addObjectReader(Tags.BATCH_ID, new BatchIdReader());
+		r.addObjectReader(Tags.BATCH_ID, batchIdReader);
 		Collection<BatchId> batches = r.readList(BatchId.class);
 		r.removeObjectReader(Tags.BATCH_ID);
 		r.removeConsumer(counting);

components/net/sf/briar/protocol/GroupFactoryImpl.java

@@ -20,14 +20,14 @@ class GroupFactoryImpl implements GroupFactory {
 	}
 
 	public Group createGroup(GroupId id, String name, boolean restricted,
-			byte[] b) {
+			byte[] saltOrKey) {
 		if(restricted) {
 			try {
-				PublicKey key = keyParser.parsePublicKey(b);
-				return new GroupImpl(id, name, null, key);
-			} catch (InvalidKeySpecException e) {
+				PublicKey key = keyParser.parsePublicKey(saltOrKey);
+				return new GroupImpl(id, name, key);
+			} catch(InvalidKeySpecException e) {
 				throw new IllegalArgumentException(e);
 			}
-		} else return new GroupImpl(id, name, b, null);
+		} else return new GroupImpl(id, name, saltOrKey);
 	}
 }

components/net/sf/briar/protocol/GroupImpl.java

@@ -1,9 +1,12 @@
 package net.sf.briar.protocol;
 
+import java.io.IOException;
 import java.security.PublicKey;
 
 import net.sf.briar.api.protocol.Group;
 import net.sf.briar.api.protocol.GroupId;
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.serial.Writer;
 
 class GroupImpl implements Group {
 
@@ -12,12 +15,18 @@ class GroupImpl implements Group {
 	private final byte[] salt;
 	private final PublicKey publicKey;
 
-	GroupImpl(GroupId id, String name, byte[] salt, PublicKey publicKey) {
-		assert salt == null || publicKey == null;
+	GroupImpl(GroupId id, String name, byte[] salt) {
 		this.id = id;
 		this.name = name;
 		this.salt = salt;
+		publicKey = null;
+	}
+
+	GroupImpl(GroupId id, String name, PublicKey publicKey) {
+		this.id = id;
+		this.name = name;
 		this.publicKey = publicKey;
+		salt = null;
 	}
 
 	public GroupId getId() {
@@ -40,6 +49,14 @@ class GroupImpl implements Group {
 		return publicKey;
 	}
 
+	public void writeTo(Writer w) throws IOException {
+		w.writeUserDefinedTag(Tags.GROUP);
+		w.writeString(name);
+		w.writeBoolean(isRestricted());
+		if(salt == null) w.writeRaw(publicKey.getEncoded());
+		else w.writeRaw(salt);
+	}
+
 	@Override
 	public boolean equals(Object o) {
 		return o instanceof Group && id.equals(((Group) o).getId());

components/net/sf/briar/protocol/GroupReader.java

@@ -0,0 +1,38 @@
+package net.sf.briar.protocol;
+
+import java.io.IOException;
+import java.security.MessageDigest;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.GroupFactory;
+import net.sf.briar.api.protocol.GroupId;
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.serial.ObjectReader;
+import net.sf.briar.api.serial.Reader;
+
+class GroupReader implements ObjectReader<Group> {
+
+	private final MessageDigest messageDigest;
+	private final GroupFactory groupFactory;
+
+	GroupReader(MessageDigest messageDigest, GroupFactory groupFactory) {
+		this.messageDigest = messageDigest;
+		this.groupFactory = groupFactory;
+	}
+
+	public Group readObject(Reader r) throws IOException {
+		// Initialise the consumer
+		DigestingConsumer digesting = new DigestingConsumer(messageDigest);
+		messageDigest.reset();
+		// Read and digest the data
+		r.addConsumer(digesting);
+		r.readUserDefinedTag(Tags.GROUP);
+		String name = r.readString();
+		boolean restricted = r.readBoolean();
+		byte[] saltOrKey = r.readRaw();
+		r.removeConsumer(digesting);
+		// Build and return the group
+		GroupId id = new GroupId(messageDigest.digest());
+		return groupFactory.createGroup(id, name, restricted, saltOrKey);
+	}
+}

components/net/sf/briar/protocol/MessageEncoderImpl.java

@@ -16,12 +16,15 @@ import net.sf.briar.api.protocol.Tags;
 import net.sf.briar.api.serial.Writer;
 import net.sf.briar.api.serial.WriterFactory;
 
+import com.google.inject.Inject;
+
 class MessageEncoderImpl implements MessageEncoder {
 
 	private final Signature signature;
 	private final MessageDigest messageDigest;
 	private final WriterFactory writerFactory;
 
+	@Inject
 	MessageEncoderImpl(Signature signature, MessageDigest messageDigest,
 			WriterFactory writerFactory) {
 		this.signature = signature;

components/net/sf/briar/protocol/ProtocolModule.java

@@ -1,6 +1,7 @@
 package net.sf.briar.protocol;
 
 import net.sf.briar.api.protocol.GroupFactory;
+import net.sf.briar.api.protocol.MessageEncoder;
 
 import com.google.inject.AbstractModule;
 
@@ -11,5 +12,6 @@ public class ProtocolModule extends AbstractModule {
 		bind(AckFactory.class).to(AckFactoryImpl.class);
 		bind(BatchFactory.class).to(BatchFactoryImpl.class);
 		bind(GroupFactory.class).to(GroupFactoryImpl.class);
+		bind(MessageEncoder.class).to(MessageEncoderImpl.class);
 	}
 }

components/net/sf/briar/protocol/SubscriptionFactory.java

@@ -0,0 +1,11 @@
+package net.sf.briar.protocol;
+
+import java.util.Collection;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.Subscriptions;
+
+interface SubscriptionFactory {
+
+	Subscriptions createSubscriptions(Collection<Group> subs, long timestamp);
+}

components/net/sf/briar/protocol/SubscriptionFactoryImpl.java

@@ -0,0 +1,14 @@
+package net.sf.briar.protocol;
+
+import java.util.Collection;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.Subscriptions;
+
+class SubscriptionFactoryImpl implements SubscriptionFactory {
+
+	public Subscriptions createSubscriptions(Collection<Group> subs,
+			long timestamp) {
+		return new SubscriptionsImpl(subs, timestamp);
+	}
+}

components/net/sf/briar/protocol/SubscriptionReader.java

@@ -0,0 +1,38 @@
+package net.sf.briar.protocol;
+
+import java.io.IOException;
+import java.util.Collection;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.Subscriptions;
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.serial.ObjectReader;
+import net.sf.briar.api.serial.Reader;
+
+class SubscriptionReader implements ObjectReader<Subscriptions> {
+
+	private final ObjectReader<Group> groupReader;
+	private final SubscriptionFactory subscriptionFactory;
+
+	SubscriptionReader(ObjectReader<Group> groupReader,
+			SubscriptionFactory subscriptionFactory) {
+		this.groupReader = groupReader;
+		this.subscriptionFactory = subscriptionFactory;
+	}
+
+	public Subscriptions readObject(Reader r) throws IOException {
+		// Initialise the consumer
+		CountingConsumer counting =
+			new CountingConsumer(Subscriptions.MAX_SIZE);
+		// Read the data
+		r.addConsumer(counting);
+		r.readUserDefinedTag(Tags.SUBSCRIPTIONS);
+		r.addObjectReader(Tags.GROUP, groupReader);
+		Collection<Group> subs = r.readList(Group.class);
+		r.removeObjectReader(Tags.GROUP);
+		long timestamp = r.readInt64();
+		r.removeConsumer(counting);
+		// Build and return the subscriptions update
+		return subscriptionFactory.createSubscriptions(subs, timestamp);
+	}
+}

components/net/sf/briar/protocol/SubscriptionsImpl.java

@@ -0,0 +1,25 @@
+package net.sf.briar.protocol;
+
+import java.util.Collection;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.Subscriptions;
+
+class SubscriptionsImpl implements Subscriptions {
+
+	private final Collection<Group> subs;
+	private final long timestamp;
+
+	SubscriptionsImpl(Collection<Group> subs, long timestamp) {
+		this.subs = subs;
+		this.timestamp = timestamp;
+	}
+
+	public Collection<Group> getSubscriptions() {
+		return subs;
+	}
+
+	public long getTimestamp() {
+		return timestamp;
+	}
+}

components/net/sf/briar/protocol/TransportFactory.java

@@ -0,0 +1,10 @@
+package net.sf.briar.protocol;
+
+import java.util.Map;
+
+import net.sf.briar.api.protocol.Transports;
+
+interface TransportFactory {
+
+	Transports createTransports(Map<String, String> transports, long timestamp);
+}

components/net/sf/briar/protocol/TransportFactoryImpl.java

@@ -0,0 +1,13 @@
+package net.sf.briar.protocol;
+
+import java.util.Map;
+
+import net.sf.briar.api.protocol.Transports;
+
+class TransportFactoryImpl implements TransportFactory {
+
+	public Transports createTransports(Map<String, String> transports,
+			long timestamp) {
+		return new TransportsImpl(transports, timestamp);
+	}
+}

components/net/sf/briar/protocol/TransportReader.java

@@ -0,0 +1,31 @@
+package net.sf.briar.protocol;
+
+import java.io.IOException;
+import java.util.Map;
+
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.protocol.Transports;
+import net.sf.briar.api.serial.ObjectReader;
+import net.sf.briar.api.serial.Reader;
+
+class TransportReader implements ObjectReader<Transports> {
+
+	private final TransportFactory transportFactory;
+
+	TransportReader(TransportFactory transportFactory) {
+		this.transportFactory = transportFactory;
+	}
+
+	public Transports readObject(Reader r) throws IOException {
+		// Initialise the consumer
+		CountingConsumer counting = new CountingConsumer(Transports.MAX_SIZE);
+		// Read the data
+		r.addConsumer(counting);
+		r.readUserDefinedTag(Tags.TRANSPORTS);
+		Map<String, String> transports = r.readMap(String.class, String.class);
+		long timestamp = r.readInt64();
+		r.removeConsumer(counting);
+		// Build and return the transports update
+		return transportFactory.createTransports(transports, timestamp);
+	}
+}

components/net/sf/briar/protocol/TransportsImpl.java

@@ -0,0 +1,24 @@
+package net.sf.briar.protocol;
+
+import java.util.Map;
+
+import net.sf.briar.api.protocol.Transports;
+
+class TransportsImpl implements Transports {
+
+	private final Map<String, String> transports;
+	private final long timestamp;
+
+	TransportsImpl(Map<String, String> transports, long timestamp) {
+		this.transports = transports;
+		this.timestamp = timestamp;
+	}
+
+	public Map<String, String> getTransports() {
+		return transports;
+	}
+
+	public long getTimestamp() {
+		return timestamp;
+	}
+}

components/net/sf/briar/protocol/writers/AckWriterImpl.java

@@ -19,10 +19,10 @@ class AckWriterImpl implements AckWriter {
 
 	AckWriterImpl(OutputStream out, WriterFactory writerFactory) {
 		this.out = out;
-		this.w = writerFactory.createWriter(out);
+		w = writerFactory.createWriter(out);
 	}
 
-	public boolean addBatchId(BatchId b) throws IOException {
+	public boolean writeBatchId(BatchId b) throws IOException {
 		if(finished) throw new IllegalStateException();
 		if(!started) {
 			w.writeUserDefinedTag(Tags.ACK);

components/net/sf/briar/protocol/writers/BatchWriterImpl.java

@@ -31,7 +31,7 @@ class BatchWriterImpl implements BatchWriter {
 		return Batch.MAX_SIZE - 3;
 	}
 
-	public boolean addMessage(byte[] message) throws IOException {
+	public boolean writeMessage(byte[] message) throws IOException {
 		if(finished) throw new IllegalStateException();
 		if(!started) {
 			messageDigest.reset();

components/net/sf/briar/protocol/writers/PacketWriterFactoryImpl.java

@@ -33,12 +33,10 @@ class PacketWriterFactoryImpl implements PacketWriterFactory {
 	}
 
 	public SubscriptionWriter createSubscriptionWriter(OutputStream out) {
-		// TODO Auto-generated method stub
-		return null;
+		return new SubscriptionWriterImpl(out, writerFactory);
 	}
 
 	public TransportWriter createTransportWriter(OutputStream out) {
-		// TODO Auto-generated method stub
-		return null;
+		return new TransportWriterImpl(out, writerFactory);
 	}
 }

components/net/sf/briar/protocol/writers/SubscriptionWriterImpl.java

@@ -0,0 +1,33 @@
+package net.sf.briar.protocol.writers;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Collection;
+
+import net.sf.briar.api.protocol.Group;
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.protocol.writers.SubscriptionWriter;
+import net.sf.briar.api.serial.Writer;
+import net.sf.briar.api.serial.WriterFactory;
+
+class SubscriptionWriterImpl implements SubscriptionWriter {
+
+	private final OutputStream out;
+	private final Writer w;
+
+	private boolean used = false;
+
+	SubscriptionWriterImpl(OutputStream out, WriterFactory writerFactory) {
+		this.out = out;
+		w = writerFactory.createWriter(out);
+	}
+
+	public void writeSubscriptions(Collection<Group> subs) throws IOException {
+		if(used) throw new IllegalStateException();
+		w.writeUserDefinedTag(Tags.SUBSCRIPTIONS);
+		w.writeList(subs);
+		w.writeInt64(System.currentTimeMillis());
+		out.flush();
+		used = true;
+	}
+}

components/net/sf/briar/protocol/writers/TransportWriterImpl.java

@@ -0,0 +1,33 @@
+package net.sf.briar.protocol.writers;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Map;
+
+import net.sf.briar.api.protocol.Tags;
+import net.sf.briar.api.protocol.writers.TransportWriter;
+import net.sf.briar.api.serial.Writer;
+import net.sf.briar.api.serial.WriterFactory;
+
+class TransportWriterImpl implements TransportWriter {
+
+	private final OutputStream out;
+	private final Writer w;
+
+	private boolean used = false;
+
+	TransportWriterImpl(OutputStream out, WriterFactory writerFactory) {
+		this.out = out;
+		w = writerFactory.createWriter(out);
+	}
+
+	public void writeTransports(Map<String, String> transports)
+	throws IOException {
+		if(used) throw new IllegalStateException();
+		w.writeUserDefinedTag(Tags.TRANSPORTS);
+		w.writeMap(transports);
+		w.writeInt64(System.currentTimeMillis());
+		out.flush();
+		used = true;
+	}
+}