Merge branch 'master' into vbdev2
Conflicts: meerkat-common/src/main/proto/meerkat/voting.protovbdev2
commit
49551dc36b
|
@ -13,4 +13,6 @@ out
|
||||||
*.prefs
|
*.prefs
|
||||||
*.project
|
*.project
|
||||||
*.classpath
|
*.classpath
|
||||||
bulletin-board-server/local-instances/meerkat.db
|
*.db
|
||||||
|
*.sql
|
||||||
|
.arcconfig
|
||||||
|
|
|
@ -69,7 +69,7 @@ dependencies {
|
||||||
|
|
||||||
test {
|
test {
|
||||||
exclude '**/*IntegrationTest*'
|
exclude '**/*IntegrationTest*'
|
||||||
outputs.upToDateWhen { false }
|
// outputs.upToDateWhen { false }
|
||||||
}
|
}
|
||||||
|
|
||||||
task integrationTest(type: Test) {
|
task integrationTest(type: Test) {
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.protobuf.ByteString;
|
import meerkat.protobuf.BulletinBoardAPI.BatchChunk;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -11,15 +11,13 @@ import java.util.List;
|
||||||
*/
|
*/
|
||||||
public class BatchDataContainer {
|
public class BatchDataContainer {
|
||||||
|
|
||||||
public final byte[] signerId;
|
public final MultiServerBatchIdentifier batchId;
|
||||||
public final int batchId;
|
public final List<BatchChunk> batchChunkList;
|
||||||
public final List<BatchData> batchDataList;
|
|
||||||
public final int startPosition;
|
public final int startPosition;
|
||||||
|
|
||||||
public BatchDataContainer(byte[] signerId, int batchId, List<BatchData> batchDataList, int startPosition) {
|
public BatchDataContainer(MultiServerBatchIdentifier batchId, List<BatchChunk> batchChunkList, int startPosition) {
|
||||||
this.signerId = signerId;
|
|
||||||
this.batchId = batchId;
|
this.batchId = batchId;
|
||||||
this.batchDataList = batchDataList;
|
this.batchChunkList = batchChunkList;
|
||||||
this.startPosition = startPosition;
|
this.startPosition = startPosition;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,168 +1,490 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
import com.google.common.util.concurrent.ListeningScheduledExecutorService;
|
import com.google.protobuf.Timestamp;
|
||||||
import com.google.common.util.concurrent.MoreExecutors;
|
|
||||||
import com.google.protobuf.ByteString;
|
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.protobuf.BulletinBoardAPI;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
import meerkat.protobuf.Voting.*;
|
import meerkat.protobuf.Voting.*;
|
||||||
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Collection;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.Executors;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 03-Mar-16.
|
* Created by Arbel Deutsch Peled on 03-Mar-16.
|
||||||
* This is a full-fledged implementation of a Bulletin Board Client
|
* This is a full-fledged implementation of a Bulletin Board Client
|
||||||
* It provides asynchronous access to several remote servers, as well as a local cache
|
* It provides asynchronous access to several remote servers, as well as a local cache
|
||||||
* Read/write operations are performed on the local server
|
* Read operations are performed on the local server
|
||||||
|
* Batch reads are performed on the local server and, if they fail, also on the remote servers
|
||||||
|
* Write operations are performed on the local server
|
||||||
|
* A Synchronizer is employed in order to keep the remote server up to date
|
||||||
* After any read is carried out, a subscription is made for the specific query to make sure the local DB will be updated
|
* After any read is carried out, a subscription is made for the specific query to make sure the local DB will be updated
|
||||||
* The database also employs a synchronizer which makes sure local data is sent to the remote servers
|
* The database also employs a synchronizer which makes sure local data is sent to the remote servers
|
||||||
*/
|
*/
|
||||||
public class CachedBulletinBoardClient implements SubscriptionAsyncBulletinBoardClient {
|
public class CachedBulletinBoardClient implements SubscriptionBulletinBoardClient {
|
||||||
|
|
||||||
private final BulletinBoardClient localClient;
|
private final AsyncBulletinBoardClient localClient;
|
||||||
private AsyncBulletinBoardClient remoteClient;
|
private final AsyncBulletinBoardClient remoteClient;
|
||||||
private BulletinBoardSubscriber subscriber;
|
private final AsyncBulletinBoardClient queueClient;
|
||||||
|
private final BulletinBoardSubscriber subscriber;
|
||||||
|
private final BulletinBoardSynchronizer synchronizer;
|
||||||
|
|
||||||
private final int threadPoolSize;
|
private Thread syncThread;
|
||||||
private final long failDelayInMilliseconds;
|
|
||||||
private final long subscriptionIntervalInMilliseconds;
|
|
||||||
|
|
||||||
public CachedBulletinBoardClient(BulletinBoardClient localClient,
|
private final static int DEFAULT_WAIT_CAP = 3000;
|
||||||
int threadPoolSize,
|
private final static int DEFAULT_SLEEP_INTERVAL = 3000;
|
||||||
long failDelayInMilliseconds,
|
|
||||||
long subscriptionIntervalInMilliseconds)
|
private class SubscriptionStoreCallback implements FutureCallback<List<BulletinBoardMessage>> {
|
||||||
throws IllegalAccessException, InstantiationException {
|
|
||||||
|
private final FutureCallback<List<BulletinBoardMessage>> callback;
|
||||||
|
|
||||||
|
public SubscriptionStoreCallback(){
|
||||||
|
callback = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public SubscriptionStoreCallback(FutureCallback<List<BulletinBoardMessage>> callback){
|
||||||
|
this.callback = callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
for (BulletinBoardMessage msg : result) {
|
||||||
|
try {
|
||||||
|
|
||||||
|
if (msg.getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
|
||||||
|
// This is a batch message: need to upload batch data as well as the message itself
|
||||||
|
BulletinBoardMessage completeMessage = localClient.readBatchData(msg);
|
||||||
|
|
||||||
|
localClient.postMessage(completeMessage);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// This is a regular message: post it
|
||||||
|
localClient.postMessage(msg);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (CommunicationException ignored) {
|
||||||
|
// TODO: log
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null) {
|
||||||
|
callback.onFailure(t); // This is some hard error that cannot be dealt with
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a Cached Client
|
||||||
|
* Assumes all parameters are initialized
|
||||||
|
* @param localClient is a Client for the local instance
|
||||||
|
* @param remoteClient is a Client for the remote instance(s); Should have endless retries for post operations
|
||||||
|
* @param subscriber is a subscription service to the remote instance(s)
|
||||||
|
* @param queueClient is a client for a local deletable server to be used as a queue for not-yet-uploaded messages
|
||||||
|
*/
|
||||||
|
public CachedBulletinBoardClient(AsyncBulletinBoardClient localClient,
|
||||||
|
AsyncBulletinBoardClient remoteClient,
|
||||||
|
BulletinBoardSubscriber subscriber,
|
||||||
|
DeletableSubscriptionBulletinBoardClient queueClient,
|
||||||
|
int sleepInterval,
|
||||||
|
int waitCap) {
|
||||||
|
|
||||||
this.localClient = localClient;
|
this.localClient = localClient;
|
||||||
this.threadPoolSize = threadPoolSize;
|
this.remoteClient = remoteClient;
|
||||||
this.failDelayInMilliseconds = failDelayInMilliseconds;
|
this.subscriber = subscriber;
|
||||||
this.subscriptionIntervalInMilliseconds = subscriptionIntervalInMilliseconds;
|
this.queueClient = queueClient;
|
||||||
|
|
||||||
remoteClient = new ThreadedBulletinBoardClient();
|
this.synchronizer = new SimpleBulletinBoardSynchronizer(sleepInterval,waitCap);
|
||||||
|
synchronizer.init(queueClient, remoteClient);
|
||||||
|
syncThread = new Thread(synchronizer);
|
||||||
|
syncThread.start();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a Cached Client
|
||||||
|
* Used default values foe the time caps
|
||||||
|
* */
|
||||||
|
public CachedBulletinBoardClient(AsyncBulletinBoardClient localClient,
|
||||||
|
AsyncBulletinBoardClient remoteClient,
|
||||||
|
BulletinBoardSubscriber subscriber,
|
||||||
|
DeletableSubscriptionBulletinBoardClient queue) {
|
||||||
|
|
||||||
|
this(localClient, remoteClient, subscriber, queue, DEFAULT_SLEEP_INTERVAL, DEFAULT_WAIT_CAP);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID postMessage(final BulletinBoardMessage msg, final FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
return localClient.postMessage(msg, new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
remoteClient.postMessage(msg, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postMessage(BulletinBoardMessage msg, FutureCallback<Boolean> callback) {
|
public MessageID postAsBatch(final BulletinBoardMessage msg, final int chunkSize, final FutureCallback<Boolean> callback) {
|
||||||
return null;
|
|
||||||
|
return localClient.postAsBatch(msg, chunkSize, new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
remoteClient.postAsBatch(msg, chunkSize, callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postBatch(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public void onFailure(Throwable t) {
|
||||||
return null;
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
}
|
}
|
||||||
|
});
|
||||||
@Override
|
|
||||||
public void beginBatch(BeginBatchMessage beginBatchMessage, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, int startPosition, FutureCallback<Boolean> callback) {
|
public void beginBatch(final Iterable<String> tags, final FutureCallback<BatchIdentifier> callback) {
|
||||||
|
|
||||||
|
localClient.beginBatch(tags, new FutureCallback<BatchIdentifier>() {
|
||||||
|
|
||||||
|
private BatchIdentifier localIdentifier;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BatchIdentifier result) {
|
||||||
|
|
||||||
|
localIdentifier = result;
|
||||||
|
|
||||||
|
remoteClient.beginBatch(tags, new FutureCallback<BatchIdentifier>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BatchIdentifier result) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(new CachedClientBatchIdentifier(localIdentifier, result));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, int startPosition, FutureCallback<Boolean> callback) {
|
public void postBatchData(final BatchIdentifier batchIdentifier, final List<BatchChunk> batchChunkList,
|
||||||
|
final int startPosition, final FutureCallback<Boolean> callback) throws IllegalArgumentException{
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof CachedClientBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
final CachedClientBatchIdentifier identifier = (CachedClientBatchIdentifier) batchIdentifier;
|
||||||
|
|
||||||
|
localClient.postBatchData(identifier.getLocalIdentifier(), batchChunkList, startPosition, new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
remoteClient.postBatchData(identifier.getRemoteIdentifier(), batchChunkList, startPosition, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
public void postBatchData(final BatchIdentifier batchIdentifier, final List<BatchChunk> batchChunkList, final FutureCallback<Boolean> callback)
|
||||||
|
throws IllegalArgumentException{
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof CachedClientBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
final CachedClientBatchIdentifier identifier = (CachedClientBatchIdentifier) batchIdentifier;
|
||||||
|
|
||||||
|
localClient.postBatchData(identifier.getLocalIdentifier(), batchChunkList, new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
remoteClient.postBatchData(identifier.getRemoteIdentifier(), batchChunkList, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void closeBatch(CloseBatchMessage closeBatchMessage, FutureCallback<Boolean> callback) {
|
public void closeBatch(final BatchIdentifier batchIdentifier, final Timestamp timestamp, final Iterable<Signature> signatures,
|
||||||
|
final FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof CachedClientBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
final CachedClientBatchIdentifier identifier = (CachedClientBatchIdentifier) batchIdentifier;
|
||||||
|
|
||||||
|
localClient.closeBatch(identifier.getLocalIdentifier(), timestamp, signatures, new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
|
||||||
|
remoteClient.closeBatch(identifier.getRemoteIdentifier(), timestamp, signatures, callback);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void getRedundancy(MessageID id, FutureCallback<Float> callback) {
|
public void getRedundancy(MessageID id, FutureCallback<Float> callback) {
|
||||||
|
|
||||||
}
|
remoteClient.getRedundancy(id, callback);
|
||||||
|
|
||||||
@Override
|
|
||||||
public void readMessages(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback) {
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage batchSpecificationMessage, FutureCallback<CompleteBatch> callback) {
|
public void readMessages(MessageFilterList filterList, final FutureCallback<List<BulletinBoardMessage>> callback) {
|
||||||
|
|
||||||
|
localClient.readMessages(filterList, callback);
|
||||||
|
|
||||||
|
subscriber.subscribe(filterList, new SubscriptionStoreCallback(callback));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void readMessage(final MessageID msgID, final FutureCallback<BulletinBoardMessage> callback) {
|
||||||
|
|
||||||
|
localClient.readMessage(msgID, new FutureCallback<BulletinBoardMessage>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BulletinBoardMessage result) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(result); // Read from local client was successful
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
|
// Read from local unsuccessful: try to read from remote
|
||||||
|
|
||||||
|
remoteClient.readMessage(msgID, new FutureCallback<BulletinBoardMessage>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BulletinBoardMessage result) {
|
||||||
|
|
||||||
|
// Read from remote was successful: store in local and return result
|
||||||
|
|
||||||
|
localClient.postMessage(result, null);
|
||||||
|
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(result);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
|
// Read from remote was unsuccessful: report error
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void readBatchData(final BulletinBoardMessage stub, final FutureCallback<BulletinBoardMessage> callback) throws IllegalArgumentException {
|
||||||
|
|
||||||
|
localClient.readBatchData(stub, new FutureCallback<BulletinBoardMessage>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BulletinBoardMessage result) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(result); // Read from local client was successful
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
|
// Read from local unsuccessful: try to read from remote
|
||||||
|
|
||||||
|
remoteClient.readBatchData(stub, new FutureCallback<BulletinBoardMessage>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BulletinBoardMessage result) {
|
||||||
|
|
||||||
|
// Read from remote was successful: store in local and return result
|
||||||
|
|
||||||
|
localClient.postMessage(result, null);
|
||||||
|
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(result);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
|
// Read from remote was unsuccessful: report error
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void querySync(SyncQuery syncQuery, FutureCallback<SyncQueryResponse> callback) {
|
public void querySync(SyncQuery syncQuery, FutureCallback<SyncQueryResponse> callback) {
|
||||||
|
|
||||||
|
localClient.querySync(syncQuery, callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void init(BulletinBoardClientParams clientParams) {
|
/**
|
||||||
|
* This is a stub method
|
||||||
remoteClient.init(clientParams);
|
* All resources are assumed to be initialized
|
||||||
|
*/
|
||||||
ListeningScheduledExecutorService executorService = MoreExecutors.listeningDecorator(Executors.newScheduledThreadPool(threadPoolSize));
|
public void init(BulletinBoardClientParams clientParams) {}
|
||||||
|
|
||||||
List<SubscriptionAsyncBulletinBoardClient> subscriberClients = new ArrayList<>(clientParams.getBulletinBoardAddressCount());
|
|
||||||
|
|
||||||
for (String address : clientParams.getBulletinBoardAddressList()){
|
|
||||||
|
|
||||||
SubscriptionAsyncBulletinBoardClient newClient =
|
|
||||||
new SingleServerBulletinBoardClient(executorService, failDelayInMilliseconds, subscriptionIntervalInMilliseconds);
|
|
||||||
|
|
||||||
newClient.init(clientParams.toBuilder().clearBulletinBoardAddress().addBulletinBoardAddress(address).build());
|
|
||||||
|
|
||||||
subscriberClients.add(newClient);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
subscriber = new ThreadedBulletinBoardSubscriber(subscriberClients, localClient);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
||||||
return null;
|
return localClient.postMessage(msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public float getRedundancy(MessageID id) {
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize) throws CommunicationException {
|
||||||
return 0;
|
MessageID result = localClient.postAsBatch(msg, chunkSize);
|
||||||
|
remoteClient.postAsBatch(msg, chunkSize);
|
||||||
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) {
|
public float getRedundancy(MessageID id) throws CommunicationException {
|
||||||
return null;
|
return remoteClient.getRedundancy(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncQuery generateSyncQuery(GenerateSyncQueryParams GenerateSyncQueryParams) throws CommunicationException {
|
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException {
|
||||||
return null;
|
subscriber.subscribe(filterList, new SubscriptionStoreCallback());
|
||||||
|
return localClient.readMessages(filterList);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readMessage(MessageID msgID) throws CommunicationException {
|
||||||
|
|
||||||
|
BulletinBoardMessage result = null;
|
||||||
|
try {
|
||||||
|
result = localClient.readMessage(msgID);
|
||||||
|
} catch (CommunicationException e) {
|
||||||
|
//TODO: log
|
||||||
|
}
|
||||||
|
|
||||||
|
if (result == null){
|
||||||
|
result = remoteClient.readMessage(msgID);
|
||||||
|
|
||||||
|
if (result != null){
|
||||||
|
localClient.postMessage(result);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readBatchData(BulletinBoardMessage stub) throws CommunicationException, IllegalArgumentException {
|
||||||
|
|
||||||
|
BulletinBoardMessage result = null;
|
||||||
|
try {
|
||||||
|
result = localClient.readBatchData(stub);
|
||||||
|
} catch (CommunicationException e) {
|
||||||
|
//TODO: log
|
||||||
|
}
|
||||||
|
|
||||||
|
if (result == null){
|
||||||
|
result = remoteClient.readBatchData(stub);
|
||||||
|
|
||||||
|
if (result != null){
|
||||||
|
localClient.postMessage(result);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException {
|
||||||
|
return localClient.generateSyncQuery(generateSyncQueryParams);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void close() {
|
public void close() {
|
||||||
|
localClient.close();
|
||||||
|
remoteClient.close();
|
||||||
|
synchronizer.stop();
|
||||||
|
try {
|
||||||
|
syncThread.join();
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
//TODO: log interruption
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void subscribe(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback) {
|
public void subscribe(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback) {
|
||||||
|
subscriber.subscribe(filterList, new SubscriptionStoreCallback(callback));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void subscribe(MessageFilterList filterList, long startEntry, FutureCallback<List<BulletinBoardMessage>> callback) {
|
public void subscribe(MessageFilterList filterList, long startEntry, FutureCallback<List<BulletinBoardMessage>> callback) {
|
||||||
|
subscriber.subscribe(filterList, startEntry, new SubscriptionStoreCallback(callback));
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
|
@ -0,0 +1,29 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
|
||||||
|
import java.util.Arrays;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 17-Jun-16.
|
||||||
|
*/
|
||||||
|
public final class CachedClientBatchIdentifier implements BatchIdentifier {
|
||||||
|
|
||||||
|
// Per-server identifiers
|
||||||
|
private final BatchIdentifier localIdentifier;
|
||||||
|
private final BatchIdentifier remoteIdentifier;
|
||||||
|
|
||||||
|
public CachedClientBatchIdentifier(BatchIdentifier localIdentifier, BatchIdentifier remoteIdentifier) {
|
||||||
|
this.localIdentifier = localIdentifier;
|
||||||
|
this.remoteIdentifier = remoteIdentifier;
|
||||||
|
}
|
||||||
|
|
||||||
|
public BatchIdentifier getLocalIdentifier() {
|
||||||
|
return localIdentifier;
|
||||||
|
}
|
||||||
|
|
||||||
|
public BatchIdentifier getRemoteIdentifier() {
|
||||||
|
return remoteIdentifier;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -1,20 +1,21 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.*;
|
import com.google.common.util.concurrent.*;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.Int64Value;
|
||||||
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.comm.MessageInputStream;
|
import meerkat.comm.MessageInputStream;
|
||||||
import meerkat.comm.MessageInputStream.MessageInputStreamFactory;
|
import meerkat.comm.MessageInputStream.MessageInputStreamFactory;
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.crypto.concrete.SHA256Digest;
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
import meerkat.protobuf.Voting.*;
|
import meerkat.protobuf.Voting.*;
|
||||||
import meerkat.util.BulletinBoardUtils;
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
import javax.ws.rs.NotFoundException;
|
import javax.ws.rs.NotFoundException;
|
||||||
import java.io.ByteArrayInputStream;
|
import java.io.ByteArrayInputStream;
|
||||||
import java.io.ByteArrayOutputStream;
|
import java.io.ByteArrayOutputStream;
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.Callable;
|
import java.util.concurrent.Callable;
|
||||||
import java.util.concurrent.Executors;
|
import java.util.concurrent.Executors;
|
||||||
|
@ -22,28 +23,27 @@ import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 15-Mar-16.
|
* Created by Arbel Deutsch Peled on 15-Mar-16.
|
||||||
* This client is to be used mainly for testing.
|
* This client wraps a BulletinBoardServer in an asynchronous client.
|
||||||
* It wraps a BulletinBoardServer in an asynchronous client.
|
* It is meant to be used as a local cache handler and for testing purposes.
|
||||||
* This means the access to the server is direct (via method calls) instead of through a TCP connection.
|
* This means the access to the server is direct (via method calls) instead of through a TCP connection.
|
||||||
* The client implements both synchronous and asynchronous method calls, but calls to the server itself are performed synchronously.
|
* The client implements both synchronous and asynchronous method calls, but calls to the server itself are performed synchronously.
|
||||||
*/
|
*/
|
||||||
public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardClient{
|
public class LocalBulletinBoardClient implements DeletableSubscriptionBulletinBoardClient {
|
||||||
|
|
||||||
private final BulletinBoardServer server;
|
private final DeletableBulletinBoardServer server;
|
||||||
private final ListeningScheduledExecutorService executorService;
|
private final ListeningScheduledExecutorService executorService;
|
||||||
private final BatchDigest digest;
|
private final BulletinBoardDigest digest;
|
||||||
private final int subsrciptionDelay;
|
private final long subsrciptionDelay;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Initializes an instance of the client
|
* Initializes an instance of the client
|
||||||
* @param server an initialized Bulletin Board Server instance which will perform the actual processing of the requests
|
* @param server an initialized Bulletin Board Server instance which will perform the actual processing of the requests
|
||||||
* @param threadNum is the number of concurrent threads to allocate for the client
|
* @param threadNum is the number of concurrent threads to allocate for the client
|
||||||
* @param subscriptionDelay is the required delay between subscription calls in milliseconds
|
|
||||||
*/
|
*/
|
||||||
public LocalBulletinBoardClient(BulletinBoardServer server, int threadNum, int subscriptionDelay) {
|
public LocalBulletinBoardClient(DeletableBulletinBoardServer server, int threadNum, int subscriptionDelay) {
|
||||||
this.server = server;
|
this.server = server;
|
||||||
this.executorService = MoreExecutors.listeningDecorator(Executors.newScheduledThreadPool(threadNum));
|
this.executorService = MoreExecutors.listeningDecorator(Executors.newScheduledThreadPool(threadNum));
|
||||||
this.digest = new GenericBatchDigest(new SHA256Digest());
|
this.digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
this.subsrciptionDelay = subscriptionDelay;
|
this.subsrciptionDelay = subscriptionDelay;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -57,7 +57,7 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Boolean call() throws Exception {
|
public Boolean call() throws CommunicationException {
|
||||||
return server.postMessage(msg).getValue();
|
return server.postMessage(msg).getValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -75,51 +75,57 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
|
|
||||||
private class CompleteBatchPoster implements Callable<Boolean> {
|
private class CompleteBatchPoster implements Callable<Boolean> {
|
||||||
|
|
||||||
private final CompleteBatch completeBatch;
|
private final BulletinBoardMessage msg;
|
||||||
|
private final int chunkSize;
|
||||||
|
|
||||||
public CompleteBatchPoster(CompleteBatch completeBatch) {
|
public CompleteBatchPoster(BulletinBoardMessage msg, int chunkSize) {
|
||||||
this.completeBatch = completeBatch;
|
this.msg = msg;
|
||||||
|
this.chunkSize = chunkSize;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Boolean call() throws Exception {
|
public Boolean call() throws CommunicationException {
|
||||||
|
|
||||||
if (!server.beginBatch(completeBatch.getBeginBatchMessage()).getValue())
|
BeginBatchMessage beginBatchMessage = BeginBatchMessage.newBuilder()
|
||||||
return false;
|
.addAllTag(msg.getMsg().getTagList())
|
||||||
|
|
||||||
int i=0;
|
|
||||||
for (BatchData data : completeBatch.getBatchDataList()){
|
|
||||||
|
|
||||||
BatchMessage message = BatchMessage.newBuilder()
|
|
||||||
.setSignerId(completeBatch.getSignature().getSignerId())
|
|
||||||
.setBatchId(completeBatch.getBeginBatchMessage().getBatchId())
|
|
||||||
.setSerialNum(i)
|
|
||||||
.setData(data)
|
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
if (!server.postBatchMessage(message).getValue())
|
Int64Value batchId = server.beginBatch(beginBatchMessage);
|
||||||
return false;
|
|
||||||
|
BatchMessage.Builder builder = BatchMessage.newBuilder()
|
||||||
|
.setBatchId(batchId.getValue());
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = BulletinBoardUtils.breakToBatch(msg, chunkSize);
|
||||||
|
|
||||||
|
int i=0;
|
||||||
|
for (BatchChunk chunk : batchChunkList){
|
||||||
|
|
||||||
|
server.postBatchMessage(builder.setSerialNum(i).setData(chunk).build());
|
||||||
|
|
||||||
i++;
|
i++;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return server.closeBatchMessage(completeBatch.getCloseBatchMessage()).getValue();
|
CloseBatchMessage closeBatchMessage = BulletinBoardUtils.generateCloseBatchMessage(batchId, batchChunkList.size(), msg);
|
||||||
|
|
||||||
|
return server.closeBatch(closeBatchMessage).getValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postBatch(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
Futures.addCallback(executorService.schedule(new CompleteBatchPoster(completeBatch), subsrciptionDelay, TimeUnit.MILLISECONDS), callback);
|
Futures.addCallback(executorService.submit(new CompleteBatchPoster(msg, chunkSize)), callback);
|
||||||
|
|
||||||
digest.update(completeBatch);
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
return digest.digestAsMessageID();
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class BatchBeginner implements Callable<Boolean> {
|
private class BatchBeginner implements Callable<SingleServerBatchIdentifier> {
|
||||||
|
|
||||||
private final BeginBatchMessage msg;
|
private final BeginBatchMessage msg;
|
||||||
|
|
||||||
|
@ -129,28 +135,31 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Boolean call() throws Exception {
|
public SingleServerBatchIdentifier call() throws Exception {
|
||||||
return server.beginBatch(msg).getValue();
|
return new SingleServerBatchIdentifier(server.beginBatch(msg));
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void beginBatch(BeginBatchMessage beginBatchMessage, FutureCallback<Boolean> callback) {
|
public void beginBatch(Iterable<String> tags, FutureCallback<BatchIdentifier> callback) {
|
||||||
|
|
||||||
|
BeginBatchMessage beginBatchMessage = BeginBatchMessage.newBuilder()
|
||||||
|
.addAllTag(tags)
|
||||||
|
.build();
|
||||||
|
|
||||||
Futures.addCallback(executorService.submit(new BatchBeginner(beginBatchMessage)), callback);
|
Futures.addCallback(executorService.submit(new BatchBeginner(beginBatchMessage)), callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
private class BatchDataPoster implements Callable<Boolean> {
|
private class BatchDataPoster implements Callable<Boolean> {
|
||||||
|
|
||||||
private final ByteString signerId;
|
private final SingleServerBatchIdentifier batchId;
|
||||||
private final int batchId;
|
private final List<BatchChunk> batchChunkList;
|
||||||
private final List<BatchData> batchDataList;
|
|
||||||
private final int startPosition;
|
private final int startPosition;
|
||||||
|
|
||||||
public BatchDataPoster(ByteString signerId, int batchId, List<BatchData> batchDataList, int startPosition) {
|
public BatchDataPoster(SingleServerBatchIdentifier batchId, List<BatchChunk> batchChunkList, int startPosition) {
|
||||||
this.signerId = signerId;
|
|
||||||
this.batchId = batchId;
|
this.batchId = batchId;
|
||||||
this.batchDataList = batchDataList;
|
this.batchChunkList = batchChunkList;
|
||||||
this.startPosition = startPosition;
|
this.startPosition = startPosition;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -159,11 +168,10 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
public Boolean call() throws Exception {
|
public Boolean call() throws Exception {
|
||||||
|
|
||||||
BatchMessage.Builder msgBuilder = BatchMessage.newBuilder()
|
BatchMessage.Builder msgBuilder = BatchMessage.newBuilder()
|
||||||
.setSignerId(signerId)
|
.setBatchId(batchId.getBatchId().getValue());
|
||||||
.setBatchId(batchId);
|
|
||||||
|
|
||||||
int i = startPosition;
|
int i = startPosition;
|
||||||
for (BatchData data : batchDataList){
|
for (BatchChunk data : batchChunkList){
|
||||||
|
|
||||||
msgBuilder.setSerialNum(i)
|
msgBuilder.setSerialNum(i)
|
||||||
.setData(data);
|
.setData(data);
|
||||||
|
@ -175,6 +183,8 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
batchId.setLength(i);
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -182,24 +192,28 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, int startPosition, FutureCallback<Boolean> callback) {
|
public void postBatchData(BatchIdentifier batchId, List<BatchChunk> batchChunkList, int startPosition, FutureCallback<Boolean> callback)
|
||||||
postBatchData(ByteString.copyFrom(signerId), batchId, batchDataList, startPosition, callback);
|
throws IllegalArgumentException{
|
||||||
|
|
||||||
|
// Cast identifier to usable form
|
||||||
|
|
||||||
|
if (!(batchId instanceof SingleServerBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
SingleServerBatchIdentifier identifier = (SingleServerBatchIdentifier) batchId;
|
||||||
|
|
||||||
|
// Add worker
|
||||||
|
|
||||||
|
Futures.addCallback(executorService.submit(new BatchDataPoster(identifier, batchChunkList, startPosition)), callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
public void postBatchData(BatchIdentifier batchId, List<BatchChunk> batchChunkList, FutureCallback<Boolean> callback) throws IllegalArgumentException{
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
postBatchData(batchId, batchChunkList, 0, callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, int startPosition, FutureCallback<Boolean> callback) {
|
|
||||||
Futures.addCallback(executorService.submit(new BatchDataPoster(signerId, batchId, batchDataList, startPosition)), callback);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
|
||||||
}
|
|
||||||
|
|
||||||
private class BatchCloser implements Callable<Boolean> {
|
private class BatchCloser implements Callable<Boolean> {
|
||||||
|
|
||||||
|
@ -212,14 +226,33 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Boolean call() throws Exception {
|
public Boolean call() throws Exception {
|
||||||
return server.closeBatchMessage(msg).getValue();
|
return server.closeBatch(msg).getValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void closeBatch(CloseBatchMessage closeBatchMessage, FutureCallback<Boolean> callback) {
|
public void closeBatch(BatchIdentifier batchId, Timestamp timestamp, Iterable<Signature> signatures, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
// Cast identifier to usable form
|
||||||
|
|
||||||
|
if (!(batchId instanceof SingleServerBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
SingleServerBatchIdentifier identifier = (SingleServerBatchIdentifier) batchId;
|
||||||
|
|
||||||
|
// Add worker
|
||||||
|
|
||||||
|
CloseBatchMessage closeBatchMessage = CloseBatchMessage.newBuilder()
|
||||||
|
.setBatchId(identifier.getBatchId().getValue())
|
||||||
|
.setBatchLength(identifier.getLength())
|
||||||
|
.setTimestamp(timestamp)
|
||||||
|
.addAllSig(signatures)
|
||||||
|
.build();
|
||||||
|
|
||||||
Futures.addCallback(executorService.submit(new BatchCloser(closeBatchMessage)), callback);
|
Futures.addCallback(executorService.submit(new BatchCloser(closeBatchMessage)), callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class RedundancyGetter implements Callable<Float> {
|
private class RedundancyGetter implements Callable<Float> {
|
||||||
|
@ -310,6 +343,7 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
public void onSuccess(List<BulletinBoardMessage> result) {
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
|
||||||
// Report new messages to user
|
// Report new messages to user
|
||||||
|
if (callback != null)
|
||||||
callback.onSuccess(result);
|
callback.onSuccess(result);
|
||||||
|
|
||||||
MessageFilterList.Builder filterBuilder = filterList.toBuilder();
|
MessageFilterList.Builder filterBuilder = filterList.toBuilder();
|
||||||
|
@ -331,7 +365,7 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
filterList = filterBuilder.build();
|
filterList = filterBuilder.build();
|
||||||
|
|
||||||
// Reschedule job
|
// Reschedule job
|
||||||
Futures.addCallback(executorService.submit(new MessageReader(filterList)), this);
|
Futures.addCallback(executorService.schedule(new MessageReader(filterList), subsrciptionDelay, TimeUnit.MILLISECONDS), this);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -339,6 +373,7 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
// Notify caller about failure and terminate subscription
|
// Notify caller about failure and terminate subscription
|
||||||
|
if (callback != null)
|
||||||
callback.onFailure(t);
|
callback.onFailure(t);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -364,83 +399,122 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
subscribe(filterList, 0, callback);
|
subscribe(filterList, 0, callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
private class CompleteBatchReader implements Callable<CompleteBatch> {
|
private class BatchDataReader implements Callable<List<BatchChunk>> {
|
||||||
|
|
||||||
private final BatchSpecificationMessage batchSpecificationMessage;
|
private final MessageID msgID;
|
||||||
|
|
||||||
public CompleteBatchReader(BatchSpecificationMessage batchSpecificationMessage) {
|
public BatchDataReader(MessageID msgID) {
|
||||||
this.batchSpecificationMessage = batchSpecificationMessage;
|
this.msgID = msgID;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<BatchChunk> call() throws Exception {
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(msgID)
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
ByteArrayOutputStream byteOutputStream = new ByteArrayOutputStream();
|
||||||
|
MessageOutputStream<BatchChunk> batchOutputStream = new MessageOutputStream<>(byteOutputStream);
|
||||||
|
server.readBatch(batchQuery,batchOutputStream);
|
||||||
|
|
||||||
|
MessageInputStream<BatchChunk> inputStream =
|
||||||
|
MessageInputStreamFactory.createMessageInputStream(
|
||||||
|
new ByteArrayInputStream(byteOutputStream.toByteArray()),
|
||||||
|
BatchChunk.class);
|
||||||
|
|
||||||
|
return inputStream.asList();
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private class CompleteBatchReader implements Callable<BulletinBoardMessage> {
|
||||||
|
|
||||||
|
private final MessageID msgID;
|
||||||
|
|
||||||
|
public CompleteBatchReader(MessageID msgID) {
|
||||||
|
this.msgID = msgID;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public CompleteBatch call() throws Exception {
|
public BulletinBoardMessage call() throws Exception {
|
||||||
|
|
||||||
final String[] TAGS_TO_REMOVE = {BulletinBoardConstants.BATCH_TAG, BulletinBoardConstants.BATCH_ID_TAG_PREFIX};
|
// Read message (mat be a stub)
|
||||||
|
|
||||||
CompleteBatch completeBatch = new CompleteBatch(BeginBatchMessage.newBuilder()
|
|
||||||
.setSignerId(batchSpecificationMessage.getSignerId())
|
|
||||||
.setBatchId(batchSpecificationMessage.getBatchId())
|
|
||||||
.build());
|
|
||||||
|
|
||||||
ByteArrayOutputStream byteOutputStream = new ByteArrayOutputStream();
|
|
||||||
MessageOutputStream<BatchData> batchOutputStream = new MessageOutputStream<>(byteOutputStream);
|
|
||||||
server.readBatch(batchSpecificationMessage,batchOutputStream);
|
|
||||||
|
|
||||||
MessageInputStream<BatchData> batchInputStream =
|
|
||||||
MessageInputStreamFactory.createMessageInputStream(
|
|
||||||
new ByteArrayInputStream(byteOutputStream.toByteArray()),
|
|
||||||
BatchData.class);
|
|
||||||
|
|
||||||
completeBatch.appendBatchData(batchInputStream.asList());
|
|
||||||
|
|
||||||
MessageFilterList filterList = MessageFilterList.newBuilder()
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
.addFilter(MessageFilter.newBuilder()
|
.addFilter(MessageFilter.newBuilder()
|
||||||
.setType(FilterType.TAG)
|
.setType(FilterType.MSG_ID)
|
||||||
.setTag(BulletinBoardConstants.BATCH_TAG)
|
.setId(msgID.getID())
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.TAG)
|
|
||||||
.setTag(BulletinBoardConstants.BATCH_ID_TAG_PREFIX + completeBatch.getBeginBatchMessage().getBatchId())
|
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.SIGNER_ID)
|
|
||||||
.setId(completeBatch.getBeginBatchMessage().getSignerId())
|
|
||||||
.build())
|
.build())
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
byteOutputStream = new ByteArrayOutputStream();
|
MessageReader messageReader = new MessageReader(filterList);
|
||||||
MessageOutputStream<BulletinBoardMessage> messageOutputStream = new MessageOutputStream<>(byteOutputStream);
|
List<BulletinBoardMessage> bulletinBoardMessages = messageReader.call();
|
||||||
server.readMessages(filterList,messageOutputStream);
|
|
||||||
|
|
||||||
MessageInputStream<BulletinBoardMessage> messageInputStream =
|
if (bulletinBoardMessages.size() <= 0) {
|
||||||
MessageInputStreamFactory.createMessageInputStream(
|
throw new NotFoundException("Message does not exist");
|
||||||
new ByteArrayInputStream(byteOutputStream.toByteArray()),
|
}
|
||||||
BulletinBoardMessage.class);
|
|
||||||
|
|
||||||
if (!messageInputStream.isAvailable())
|
BulletinBoardMessage msg = bulletinBoardMessages.get(0);
|
||||||
throw new NotFoundException("Batch does not exist");
|
|
||||||
|
|
||||||
BulletinBoardMessage message = messageInputStream.readMessage();
|
if (msg.getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
|
||||||
completeBatch.setBeginBatchMessage(BeginBatchMessage.newBuilder()
|
// Read data
|
||||||
.addAllTag(BulletinBoardUtils.removePrefixTags(message, Arrays.asList(TAGS_TO_REMOVE)))
|
|
||||||
.setSignerId(message.getSig(0).getSignerId())
|
|
||||||
.setBatchId(Integer.parseInt(BulletinBoardUtils.findTagWithPrefix(message, BulletinBoardConstants.BATCH_ID_TAG_PREFIX)))
|
|
||||||
.build());
|
|
||||||
|
|
||||||
completeBatch.setSignature(message.getSig(0));
|
BatchDataReader batchDataReader = new BatchDataReader(msgID);
|
||||||
completeBatch.setTimestamp(message.getMsg().getTimestamp());
|
List<BatchChunk> batchChunkList = batchDataReader.call();
|
||||||
|
|
||||||
return completeBatch;
|
// Combine and return
|
||||||
|
|
||||||
|
return BulletinBoardUtils.gatherBatch(msg, batchChunkList);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
return msg;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private class BatchDataCombiner implements Callable<BulletinBoardMessage> {
|
||||||
|
|
||||||
|
private final BulletinBoardMessage stub;
|
||||||
|
|
||||||
|
public BatchDataCombiner(BulletinBoardMessage stub) {
|
||||||
|
this.stub = stub;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage call() throws Exception {
|
||||||
|
|
||||||
|
MessageID msgID = MessageID.newBuilder().setID(stub.getMsg().getMsgId()).build();
|
||||||
|
|
||||||
|
BatchDataReader batchDataReader = new BatchDataReader(msgID);
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = batchDataReader.call();
|
||||||
|
|
||||||
|
return BulletinBoardUtils.gatherBatch(stub, batchChunkList);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage batchSpecificationMessage, FutureCallback<CompleteBatch> callback) {
|
public void readMessage(MessageID msgID, FutureCallback<BulletinBoardMessage> callback) {
|
||||||
Futures.addCallback(executorService.submit(new CompleteBatchReader(batchSpecificationMessage)), callback);
|
Futures.addCallback(executorService.submit(new CompleteBatchReader(msgID)), callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void readBatchData(BulletinBoardMessage stub, FutureCallback<BulletinBoardMessage> callback) throws IllegalArgumentException {
|
||||||
|
|
||||||
|
if (stub.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID){
|
||||||
|
throw new IllegalArgumentException("Message is not a stub and does not contain the required message ID");
|
||||||
|
}
|
||||||
|
|
||||||
|
Futures.addCallback(executorService.submit(new BatchDataCombiner(stub)),callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class SyncQueryHandler implements Callable<SyncQueryResponse> {
|
private class SyncQueryHandler implements Callable<SyncQueryResponse> {
|
||||||
|
@ -474,18 +548,27 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
@Override
|
@Override
|
||||||
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
||||||
|
|
||||||
try {
|
|
||||||
|
|
||||||
MessagePoster poster = new MessagePoster(msg);
|
MessagePoster poster = new MessagePoster(msg);
|
||||||
poster.call();
|
poster.call();
|
||||||
|
|
||||||
digest.update(msg);
|
digest.update(msg.getMsg());
|
||||||
return digest.digestAsMessageID();
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
} catch (Exception e) {
|
|
||||||
return null;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize) throws CommunicationException {
|
||||||
|
|
||||||
|
CompleteBatchPoster poster = new CompleteBatchPoster(msg, chunkSize);
|
||||||
|
Boolean result = poster.call();
|
||||||
|
|
||||||
|
if (!result)
|
||||||
|
throw new CommunicationException("Batch post failed");
|
||||||
|
|
||||||
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -503,7 +586,7 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) {
|
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException{
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|
||||||
|
@ -511,14 +594,89 @@ public class LocalBulletinBoardClient implements SubscriptionAsyncBulletinBoardC
|
||||||
return reader.call();
|
return reader.call();
|
||||||
|
|
||||||
} catch (Exception e){
|
} catch (Exception e){
|
||||||
return null;
|
throw new CommunicationException("Error reading from server");
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncQuery generateSyncQuery(GenerateSyncQueryParams GenerateSyncQueryParams) throws CommunicationException {
|
public BulletinBoardMessage readMessage(MessageID msgID) throws CommunicationException {
|
||||||
return server.generateSyncQuery(GenerateSyncQueryParams);
|
|
||||||
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgID.getID())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
CompleteBatchReader completeBatchReader = new CompleteBatchReader(msgID);
|
||||||
|
|
||||||
|
try {
|
||||||
|
return completeBatchReader.call();
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new CommunicationException(e.getMessage() + " " + e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readBatchData(BulletinBoardMessage stub) throws CommunicationException, IllegalArgumentException {
|
||||||
|
|
||||||
|
if (stub.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID){
|
||||||
|
throw new IllegalArgumentException("Message is not a stub and does not contain the required message ID");
|
||||||
|
}
|
||||||
|
|
||||||
|
BatchDataCombiner combiner = new BatchDataCombiner(stub);
|
||||||
|
|
||||||
|
try {
|
||||||
|
return combiner.call();
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new CommunicationException(e.getCause() + " " + e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException {
|
||||||
|
return server.generateSyncQuery(generateSyncQueryParams);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void deleteMessage(MessageID msgID, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
Boolean deleted = server.deleteMessage(msgID).getValue();
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(deleted);
|
||||||
|
} catch (CommunicationException e) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(e);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void deleteMessage(long entryNum, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
Boolean deleted = server.deleteMessage(entryNum).getValue();
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(deleted);
|
||||||
|
} catch (CommunicationException e) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(e);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean deleteMessage(MessageID msgID) throws CommunicationException {
|
||||||
|
return server.deleteMessage(msgID).getValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean deleteMessage(long entryNum) throws CommunicationException {
|
||||||
|
return server.deleteMessage(entryNum).getValue();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -0,0 +1,27 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
|
||||||
|
import java.util.Arrays;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 17-Jun-16.
|
||||||
|
*/
|
||||||
|
public final class MultiServerBatchIdentifier implements AsyncBulletinBoardClient.BatchIdentifier {
|
||||||
|
|
||||||
|
// Per-server identifiers
|
||||||
|
private final Iterable<BatchIdentifier> identifiers;
|
||||||
|
|
||||||
|
public MultiServerBatchIdentifier(Iterable<BatchIdentifier> identifiers) {
|
||||||
|
this.identifiers = identifiers;
|
||||||
|
}
|
||||||
|
|
||||||
|
public MultiServerBatchIdentifier(BatchIdentifier[] identifiers) {
|
||||||
|
this.identifiers = Arrays.asList(identifiers);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Iterable<BatchIdentifier> getIdentifiers() {
|
||||||
|
return identifiers;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -18,7 +18,7 @@ import java.util.concurrent.atomic.AtomicInteger;
|
||||||
*/
|
*/
|
||||||
public abstract class MultiServerWorker<IN, OUT> extends BulletinClientWorker<IN> implements Runnable, FutureCallback<OUT>{
|
public abstract class MultiServerWorker<IN, OUT> extends BulletinClientWorker<IN> implements Runnable, FutureCallback<OUT>{
|
||||||
|
|
||||||
private final List<SingleServerBulletinBoardClient> clients;
|
protected final List<SingleServerBulletinBoardClient> clients;
|
||||||
|
|
||||||
protected AtomicInteger minServers; // The minimal number of servers the job must be successful on for the job to be completed
|
protected AtomicInteger minServers; // The minimal number of servers the job must be successful on for the job to be completed
|
||||||
|
|
||||||
|
@ -74,6 +74,7 @@ public abstract class MultiServerWorker<IN, OUT> extends BulletinClientWorker<IN
|
||||||
*/
|
*/
|
||||||
protected void succeed(OUT result){
|
protected void succeed(OUT result){
|
||||||
if (returnedResult.compareAndSet(false, true)) {
|
if (returnedResult.compareAndSet(false, true)) {
|
||||||
|
if (futureCallback != null)
|
||||||
futureCallback.onSuccess(result);
|
futureCallback.onSuccess(result);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -85,18 +86,11 @@ public abstract class MultiServerWorker<IN, OUT> extends BulletinClientWorker<IN
|
||||||
*/
|
*/
|
||||||
protected void fail(Throwable t){
|
protected void fail(Throwable t){
|
||||||
if (returnedResult.compareAndSet(false, true)) {
|
if (returnedResult.compareAndSet(false, true)) {
|
||||||
|
if (futureCallback != null)
|
||||||
futureCallback.onFailure(t);
|
futureCallback.onFailure(t);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Used by implementations to get a Single Server Client iterator
|
|
||||||
* @return the requested iterator
|
|
||||||
*/
|
|
||||||
protected Iterator<SingleServerBulletinBoardClient> getClientIterator() {
|
|
||||||
return clients.iterator();
|
|
||||||
}
|
|
||||||
|
|
||||||
protected int getClientNumber() {
|
protected int getClientNumber() {
|
||||||
return clients.size();
|
return clients.size();
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,13 +2,15 @@ package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.protobuf.BoolValue;
|
import com.google.protobuf.BoolValue;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.ByteString;
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
|
import meerkat.bulletinboard.workers.singleserver.*;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.crypto.Digest;
|
|
||||||
import meerkat.crypto.concrete.SHA256Digest;
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Comm.*;
|
|
||||||
import meerkat.protobuf.Voting.*;
|
import meerkat.protobuf.Voting.*;
|
||||||
import meerkat.rest.*;
|
import meerkat.rest.*;
|
||||||
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -30,7 +32,7 @@ public class SimpleBulletinBoardClient implements BulletinBoardClient{
|
||||||
|
|
||||||
protected Client client;
|
protected Client client;
|
||||||
|
|
||||||
protected Digest digest;
|
protected BulletinBoardDigest digest;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Stores database locations and initializes the web Client
|
* Stores database locations and initializes the web Client
|
||||||
|
@ -45,7 +47,8 @@ public class SimpleBulletinBoardClient implements BulletinBoardClient{
|
||||||
client.register(ProtobufMessageBodyReader.class);
|
client.register(ProtobufMessageBodyReader.class);
|
||||||
client.register(ProtobufMessageBodyWriter.class);
|
client.register(ProtobufMessageBodyWriter.class);
|
||||||
|
|
||||||
digest = new SHA256Digest();
|
// Wrap the Digest into a BatchDigest
|
||||||
|
digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -60,19 +63,16 @@ public class SimpleBulletinBoardClient implements BulletinBoardClient{
|
||||||
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
||||||
|
|
||||||
WebTarget webTarget;
|
WebTarget webTarget;
|
||||||
Response response;
|
Response response = null;
|
||||||
|
|
||||||
// Post message to all databases
|
// Post message to all databases
|
||||||
try {
|
try {
|
||||||
for (String db : meerkatDBs) {
|
for (String db : meerkatDBs) {
|
||||||
webTarget = client.target(db).path(BULLETIN_BOARD_SERVER_PATH).path(POST_MESSAGE_PATH);
|
|
||||||
response = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(Entity.entity(msg, Constants.MEDIATYPE_PROTOBUF));
|
|
||||||
|
|
||||||
// Only consider valid responses
|
SingleServerPostMessageWorker worker = new SingleServerPostMessageWorker(db, msg, 0);
|
||||||
if (response.getStatusInfo() == Response.Status.OK
|
|
||||||
|| response.getStatusInfo() == Response.Status.CREATED) {
|
worker.call();
|
||||||
response.readEntity(BoolValue.class).getValue();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} catch (Exception e) { // Occurs only when server replies with valid status but invalid data
|
} catch (Exception e) { // Occurs only when server replies with valid status but invalid data
|
||||||
throw new CommunicationException("Error accessing database: " + e.getMessage());
|
throw new CommunicationException("Error accessing database: " + e.getMessage());
|
||||||
|
@ -130,35 +130,172 @@ public class SimpleBulletinBoardClient implements BulletinBoardClient{
|
||||||
* @return the list of Bulletin Board messages that are returned from a server
|
* @return the list of Bulletin Board messages that are returned from a server
|
||||||
*/
|
*/
|
||||||
@Override
|
@Override
|
||||||
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) {
|
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException{
|
||||||
|
|
||||||
WebTarget webTarget;
|
|
||||||
Response response;
|
|
||||||
BulletinBoardMessageList messageList;
|
|
||||||
|
|
||||||
// Replace null filter list with blank one.
|
// Replace null filter list with blank one.
|
||||||
if (filterList == null){
|
if (filterList == null){
|
||||||
filterList = MessageFilterList.newBuilder().build();
|
filterList = MessageFilterList.getDefaultInstance();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
String exceptionString = "";
|
||||||
|
|
||||||
for (String db : meerkatDBs) {
|
for (String db : meerkatDBs) {
|
||||||
|
|
||||||
try {
|
try {
|
||||||
webTarget = client.target(db).path(BULLETIN_BOARD_SERVER_PATH).path(READ_MESSAGES_PATH);
|
|
||||||
|
|
||||||
response = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(Entity.entity(filterList, Constants.MEDIATYPE_PROTOBUF));
|
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(db, filterList, 0);
|
||||||
|
|
||||||
messageList = response.readEntity(BulletinBoardMessageList.class);
|
List<BulletinBoardMessage> result = worker.call();
|
||||||
|
|
||||||
if (messageList != null){
|
return result;
|
||||||
return messageList.getMessageList();
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
//TODO: log
|
||||||
|
exceptionString += e.getMessage() + "\n";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
} catch (Exception e) {}
|
throw new CommunicationException("Could not find message in any DB. Errors follow:\n" + exceptionString);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return null;
|
@Override
|
||||||
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize) throws CommunicationException {
|
||||||
|
|
||||||
|
List<BatchChunk> chunkList = BulletinBoardUtils.breakToBatch(msg, chunkSize);
|
||||||
|
|
||||||
|
BeginBatchMessage beginBatchMessage = BulletinBoardUtils.generateBeginBatchMessage(msg);
|
||||||
|
|
||||||
|
boolean posted = false;
|
||||||
|
|
||||||
|
// Post message to all databases
|
||||||
|
|
||||||
|
for (String db : meerkatDBs) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
int pos = 0;
|
||||||
|
|
||||||
|
SingleServerBeginBatchWorker beginBatchWorker = new SingleServerBeginBatchWorker(db, beginBatchMessage, 0);
|
||||||
|
|
||||||
|
Int64Value batchId = beginBatchWorker.call();
|
||||||
|
|
||||||
|
BatchMessage.Builder builder = BatchMessage.newBuilder().setBatchId(batchId.getValue());
|
||||||
|
|
||||||
|
for (BatchChunk batchChunk : chunkList) {
|
||||||
|
|
||||||
|
SingleServerPostBatchWorker postBatchWorker =
|
||||||
|
new SingleServerPostBatchWorker(
|
||||||
|
db,
|
||||||
|
builder.setData(batchChunk).setSerialNum(pos).build(),
|
||||||
|
0);
|
||||||
|
|
||||||
|
postBatchWorker.call();
|
||||||
|
|
||||||
|
pos++;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
CloseBatchMessage closeBatchMessage = BulletinBoardUtils.generateCloseBatchMessage(batchId, chunkList.size(), msg);
|
||||||
|
|
||||||
|
SingleServerCloseBatchWorker closeBatchWorker = new SingleServerCloseBatchWorker(db, closeBatchMessage, 0);
|
||||||
|
|
||||||
|
closeBatchWorker.call();
|
||||||
|
|
||||||
|
posted = true;
|
||||||
|
|
||||||
|
} catch(Exception ignored) {}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!posted){
|
||||||
|
throw new CommunicationException("Could not post to any server");
|
||||||
|
}
|
||||||
|
|
||||||
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readMessage(MessageID msgID) throws CommunicationException {
|
||||||
|
|
||||||
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgID.getID())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(msgID)
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
String exceptionString = "";
|
||||||
|
|
||||||
|
for (String db : meerkatDBs) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
SingleServerReadMessagesWorker messagesWorker = new SingleServerReadMessagesWorker(db, filterList, 0);
|
||||||
|
|
||||||
|
List<BulletinBoardMessage> messages = messagesWorker.call();
|
||||||
|
|
||||||
|
if (messages == null || messages.size() < 1)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
BulletinBoardMessage stub = messages.get(0);
|
||||||
|
|
||||||
|
SingleServerReadBatchWorker batchWorker = new SingleServerReadBatchWorker(db, batchQuery, 0);
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = batchWorker.call();
|
||||||
|
|
||||||
|
return BulletinBoardUtils.gatherBatch(stub, batchChunkList);
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
//TODO: log
|
||||||
|
exceptionString += e.getMessage() + "\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new CommunicationException("Could not find message in any DB. Errors follow:\n" + exceptionString);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readBatchData(BulletinBoardMessage stub) throws CommunicationException, IllegalArgumentException {
|
||||||
|
|
||||||
|
if (stub.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID){
|
||||||
|
throw new IllegalArgumentException("Message is not a stub and does not contain the required message ID");
|
||||||
|
}
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(MessageID.newBuilder()
|
||||||
|
.setID(stub.getMsg().getMsgId())
|
||||||
|
.build())
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
String exceptionString = "";
|
||||||
|
|
||||||
|
for (String db : meerkatDBs) {
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
SingleServerReadBatchWorker batchWorker = new SingleServerReadBatchWorker(db, batchQuery, 0);
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = batchWorker.call();
|
||||||
|
|
||||||
|
return BulletinBoardUtils.gatherBatch(stub, batchChunkList);
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
//TODO: log
|
||||||
|
exceptionString += e.getMessage() + "\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new CommunicationException("Could not find message in any DB. Errors follow:\n" + exceptionString);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,241 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import com.google.protobuf.ByteString;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
|
import java.util.LinkedList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.concurrent.Semaphore;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel on 13/04/2016.
|
||||||
|
* Simple, straightforward implementation of the {@link BulletinBoardSynchronizer} interface
|
||||||
|
*/
|
||||||
|
public class SimpleBulletinBoardSynchronizer implements BulletinBoardSynchronizer {
|
||||||
|
|
||||||
|
private DeletableSubscriptionBulletinBoardClient localClient;
|
||||||
|
private AsyncBulletinBoardClient remoteClient;
|
||||||
|
|
||||||
|
private AtomicBoolean running;
|
||||||
|
private volatile SyncStatus syncStatus;
|
||||||
|
|
||||||
|
private List<FutureCallback<Integer>> messageCountCallbacks;
|
||||||
|
private List<FutureCallback<SyncStatus>> syncStatusCallbacks;
|
||||||
|
|
||||||
|
private static final MessageFilterList EMPTY_FILTER = MessageFilterList.getDefaultInstance();
|
||||||
|
private static final int DEFAULT_SLEEP_INTERVAL = 10000; // 10 Seconds
|
||||||
|
private static final int DEFAULT_WAIT_CAP = 300000; // 5 minutes wait before deciding that the sync has failed fatally
|
||||||
|
|
||||||
|
private final int SLEEP_INTERVAL;
|
||||||
|
private final int WAIT_CAP;
|
||||||
|
|
||||||
|
private Semaphore semaphore;
|
||||||
|
|
||||||
|
private class SyncCallback implements FutureCallback<List<BulletinBoardMessage>> {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
|
||||||
|
// Notify Message Count callbacks if needed
|
||||||
|
|
||||||
|
if (syncStatus != SyncStatus.SYNCHRONIZED || result.size() > 0) {
|
||||||
|
|
||||||
|
for (FutureCallback<Integer> callback : messageCountCallbacks){
|
||||||
|
callback.onSuccess(result.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle upload and status change
|
||||||
|
|
||||||
|
SyncStatus newStatus = SyncStatus.PENDING;
|
||||||
|
|
||||||
|
if (result.size() == 0) {
|
||||||
|
newStatus = SyncStatus.SYNCHRONIZED;
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
else{ // Upload messages
|
||||||
|
|
||||||
|
for (BulletinBoardMessage message : result){
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
if (message.getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
|
||||||
|
// This is a batch message: need to upload batch data as well as the message itself
|
||||||
|
|
||||||
|
BulletinBoardMessage completeMsg = localClient.readBatchData(message);
|
||||||
|
|
||||||
|
remoteClient.postMessage(completeMsg);
|
||||||
|
|
||||||
|
localClient.deleteMessage(completeMsg.getEntryNum());
|
||||||
|
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// This is a regular message: post it
|
||||||
|
remoteClient.postMessage(message);
|
||||||
|
|
||||||
|
localClient.deleteMessage(message.getEntryNum());
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (CommunicationException e) {
|
||||||
|
// This is an error with the local server
|
||||||
|
// TODO: log
|
||||||
|
updateSyncStatus(SyncStatus.SERVER_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
updateSyncStatus(newStatus);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
|
updateSyncStatus(SyncStatus.SERVER_ERROR);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public SimpleBulletinBoardSynchronizer(int sleepInterval, int waitCap) {
|
||||||
|
this.syncStatus = SyncStatus.STOPPED;
|
||||||
|
this.SLEEP_INTERVAL = sleepInterval;
|
||||||
|
this.WAIT_CAP = waitCap;
|
||||||
|
this.running = new AtomicBoolean(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
public SimpleBulletinBoardSynchronizer() {
|
||||||
|
this(DEFAULT_SLEEP_INTERVAL, DEFAULT_WAIT_CAP);
|
||||||
|
}
|
||||||
|
|
||||||
|
private synchronized void updateSyncStatus(SyncStatus newStatus) {
|
||||||
|
|
||||||
|
if (!running.get()) {
|
||||||
|
|
||||||
|
newStatus = SyncStatus.STOPPED;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if (newStatus != syncStatus){
|
||||||
|
|
||||||
|
syncStatus = newStatus;
|
||||||
|
|
||||||
|
for (FutureCallback<SyncStatus> callback : syncStatusCallbacks){
|
||||||
|
if (callback != null)
|
||||||
|
callback.onSuccess(syncStatus);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void init(DeletableSubscriptionBulletinBoardClient localClient, AsyncBulletinBoardClient remoteClient) {
|
||||||
|
|
||||||
|
updateSyncStatus(SyncStatus.STOPPED);
|
||||||
|
|
||||||
|
this.localClient = localClient;
|
||||||
|
this.remoteClient = remoteClient;
|
||||||
|
|
||||||
|
messageCountCallbacks = new LinkedList<>();
|
||||||
|
syncStatusCallbacks = new LinkedList<>();
|
||||||
|
|
||||||
|
semaphore = new Semaphore(0);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SyncStatus getSyncStatus() {
|
||||||
|
return syncStatus;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void subscribeToSyncStatus(FutureCallback<SyncStatus> callback) {
|
||||||
|
syncStatusCallbacks.add(callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<BulletinBoardMessage> getRemainingMessages() throws CommunicationException{
|
||||||
|
return localClient.readMessages(EMPTY_FILTER);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void getRemainingMessages(FutureCallback<List<BulletinBoardMessage>> callback) {
|
||||||
|
localClient.readMessages(EMPTY_FILTER, callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getRemainingMessagesCount() throws CommunicationException {
|
||||||
|
return localClient.readMessages(EMPTY_FILTER).size();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void subscribeToRemainingMessagesCount(FutureCallback<Integer> callback) {
|
||||||
|
messageCountCallbacks.add(callback);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
|
||||||
|
if (running.compareAndSet(false,true)){
|
||||||
|
|
||||||
|
updateSyncStatus(SyncStatus.PENDING);
|
||||||
|
SyncCallback callback = new SyncCallback();
|
||||||
|
|
||||||
|
while (syncStatus != SyncStatus.STOPPED) {
|
||||||
|
|
||||||
|
do {
|
||||||
|
|
||||||
|
localClient.readMessages(EMPTY_FILTER, callback);
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
semaphore.tryAcquire(WAIT_CAP, TimeUnit.MILLISECONDS);
|
||||||
|
//TODO: log hard error. Too much time trying to upload data.
|
||||||
|
|
||||||
|
} catch (InterruptedException ignored) {
|
||||||
|
// We expect an interruption when the upload will complete
|
||||||
|
}
|
||||||
|
|
||||||
|
} while (syncStatus == SyncStatus.PENDING);
|
||||||
|
|
||||||
|
// Database is synced. Wait for new data.
|
||||||
|
|
||||||
|
try {
|
||||||
|
semaphore.tryAcquire(SLEEP_INTERVAL, TimeUnit.MILLISECONDS);
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
//TODO: log (probably nudged)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void nudge() {
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void stop() {
|
||||||
|
|
||||||
|
running.set(false);
|
||||||
|
updateSyncStatus(SyncStatus.STOPPED);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,42 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 16-Jun-16.
|
||||||
|
* Single-server implementation of the BatchIdentifier interface
|
||||||
|
*/
|
||||||
|
final class SingleServerBatchIdentifier implements AsyncBulletinBoardClient.BatchIdentifier {
|
||||||
|
|
||||||
|
private final Int64Value batchId;
|
||||||
|
|
||||||
|
private int length;
|
||||||
|
|
||||||
|
public SingleServerBatchIdentifier(Int64Value batchId) {
|
||||||
|
this.batchId = batchId;
|
||||||
|
length = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
public SingleServerBatchIdentifier(long batchId) {
|
||||||
|
this(Int64Value.newBuilder().setValue(batchId).build());
|
||||||
|
}
|
||||||
|
|
||||||
|
public Int64Value getBatchId() {
|
||||||
|
return batchId;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Overrides the existing length with the new one only if the new length is longer
|
||||||
|
* @param newLength
|
||||||
|
*/
|
||||||
|
public void setLength(int newLength) {
|
||||||
|
if (newLength > length) {
|
||||||
|
length = newLength;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getLength() {
|
||||||
|
return length;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -4,16 +4,19 @@ import com.google.common.util.concurrent.FutureCallback;
|
||||||
import com.google.common.util.concurrent.Futures;
|
import com.google.common.util.concurrent.Futures;
|
||||||
import com.google.common.util.concurrent.ListeningScheduledExecutorService;
|
import com.google.common.util.concurrent.ListeningScheduledExecutorService;
|
||||||
import com.google.common.util.concurrent.MoreExecutors;
|
import com.google.common.util.concurrent.MoreExecutors;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.Int64Value;
|
||||||
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.bulletinboard.workers.singleserver.*;
|
import meerkat.bulletinboard.workers.singleserver.*;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.Crypto;
|
||||||
import meerkat.protobuf.Voting.BulletinBoardClientParams;
|
import meerkat.protobuf.Voting.BulletinBoardClientParams;
|
||||||
import meerkat.util.BulletinBoardUtils;
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
import java.util.Arrays;
|
import javax.ws.rs.client.Client;
|
||||||
|
import java.lang.Iterable;
|
||||||
import java.util.Iterator;
|
import java.util.Iterator;
|
||||||
import java.util.LinkedList;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.Executors;
|
import java.util.concurrent.Executors;
|
||||||
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.TimeUnit;
|
||||||
|
@ -28,19 +31,23 @@ import java.util.concurrent.atomic.AtomicInteger;
|
||||||
* If the list of servers contains more than one server: the server actually used is the first one
|
* If the list of servers contains more than one server: the server actually used is the first one
|
||||||
* The class further implements a delayed access to the server after a communication error occurs
|
* The class further implements a delayed access to the server after a communication error occurs
|
||||||
*/
|
*/
|
||||||
public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient implements SubscriptionAsyncBulletinBoardClient {
|
public class SingleServerBulletinBoardClient implements SubscriptionBulletinBoardClient {
|
||||||
|
|
||||||
|
protected Client client;
|
||||||
|
|
||||||
|
protected BulletinBoardDigest digest;
|
||||||
|
|
||||||
|
private String dbAddress;
|
||||||
|
|
||||||
private final int MAX_RETRIES = 11;
|
private final int MAX_RETRIES = 11;
|
||||||
|
|
||||||
private ListeningScheduledExecutorService executorService;
|
private final ListeningScheduledExecutorService executorService;
|
||||||
|
|
||||||
protected BatchDigest batchDigest;
|
|
||||||
|
|
||||||
private long lastServerErrorTime;
|
private long lastServerErrorTime;
|
||||||
|
|
||||||
private final long failDelayInMilliseconds;
|
private final long FAIL_DELAY_IN_MILLISECONDS;
|
||||||
|
|
||||||
private final long subscriptionIntervalInMilliseconds;
|
private final long SUBSCRIPTION_INTERVAL_IN_MILLISECONDS;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Notify the client that a job has failed
|
* Notify the client that a job has failed
|
||||||
|
@ -53,6 +60,43 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private class SynchronousRetry<OUT> {
|
||||||
|
|
||||||
|
private final SingleServerWorker<?,OUT> worker;
|
||||||
|
|
||||||
|
private String thrown;
|
||||||
|
|
||||||
|
public SynchronousRetry(SingleServerWorker<?,OUT> worker) {
|
||||||
|
this.worker = worker;
|
||||||
|
this.thrown = "Could not contact server. Errors follow:\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
OUT run() throws CommunicationException {
|
||||||
|
|
||||||
|
do {
|
||||||
|
|
||||||
|
try {
|
||||||
|
return worker.call();
|
||||||
|
} catch (Exception e) {
|
||||||
|
thrown += e.getCause() + " " + e.getMessage() + "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
Thread.sleep(FAIL_DELAY_IN_MILLISECONDS);
|
||||||
|
} catch (InterruptedException e) {
|
||||||
|
//TODO: log
|
||||||
|
}
|
||||||
|
|
||||||
|
worker.decMaxRetry();
|
||||||
|
|
||||||
|
} while (worker.isRetry());
|
||||||
|
|
||||||
|
throw new CommunicationException(thrown);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This method adds a worker to the scheduled queue of the threadpool
|
* This method adds a worker to the scheduled queue of the threadpool
|
||||||
* If the server is in an accessible state: the job is submitted for immediate handling
|
* If the server is in an accessible state: the job is submitted for immediate handling
|
||||||
|
@ -64,7 +108,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
|
|
||||||
long timeSinceLastServerError = System.currentTimeMillis() - lastServerErrorTime;
|
long timeSinceLastServerError = System.currentTimeMillis() - lastServerErrorTime;
|
||||||
|
|
||||||
if (timeSinceLastServerError >= failDelayInMilliseconds) {
|
if (timeSinceLastServerError >= FAIL_DELAY_IN_MILLISECONDS) {
|
||||||
|
|
||||||
// Schedule for immediate processing
|
// Schedule for immediate processing
|
||||||
Futures.addCallback(executorService.submit(worker), callback);
|
Futures.addCallback(executorService.submit(worker), callback);
|
||||||
|
@ -74,7 +118,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
// Schedule for processing immediately following delay expiry
|
// Schedule for processing immediately following delay expiry
|
||||||
Futures.addCallback(executorService.schedule(
|
Futures.addCallback(executorService.schedule(
|
||||||
worker,
|
worker,
|
||||||
failDelayInMilliseconds - timeSinceLastServerError,
|
FAIL_DELAY_IN_MILLISECONDS - timeSinceLastServerError,
|
||||||
TimeUnit.MILLISECONDS),
|
TimeUnit.MILLISECONDS),
|
||||||
callback);
|
callback);
|
||||||
|
|
||||||
|
@ -97,6 +141,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(T result) {
|
public void onSuccess(T result) {
|
||||||
|
if (futureCallback != null)
|
||||||
futureCallback.onSuccess(result);
|
futureCallback.onSuccess(result);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -115,6 +160,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
scheduleWorker(worker, this);
|
scheduleWorker(worker, this);
|
||||||
} else {
|
} else {
|
||||||
// No more retries: notify caller about failure
|
// No more retries: notify caller about failure
|
||||||
|
if (futureCallback != null)
|
||||||
futureCallback.onFailure(t);
|
futureCallback.onFailure(t);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -127,14 +173,14 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
* It reports success back to the user only if all of the batch-data were successfully posted
|
* It reports success back to the user only if all of the batch-data were successfully posted
|
||||||
* If any batch-data fails to post: this callback reports failure
|
* If any batch-data fails to post: this callback reports failure
|
||||||
*/
|
*/
|
||||||
class PostBatchDataListCallback implements FutureCallback<Boolean> {
|
class PostBatchChunkListCallback implements FutureCallback<Boolean> {
|
||||||
|
|
||||||
private final FutureCallback<Boolean> callback;
|
private final FutureCallback<Boolean> callback;
|
||||||
|
|
||||||
private AtomicInteger batchDataRemaining;
|
private AtomicInteger batchDataRemaining;
|
||||||
private AtomicBoolean aggregatedResult;
|
private AtomicBoolean aggregatedResult;
|
||||||
|
|
||||||
public PostBatchDataListCallback(int batchDataLength, FutureCallback<Boolean> callback) {
|
public PostBatchChunkListCallback(int batchDataLength, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
this.callback = callback;
|
this.callback = callback;
|
||||||
this.batchDataRemaining = new AtomicInteger(batchDataLength);
|
this.batchDataRemaining = new AtomicInteger(batchDataLength);
|
||||||
|
@ -150,6 +196,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
}
|
}
|
||||||
|
|
||||||
if (batchDataRemaining.decrementAndGet() == 0){
|
if (batchDataRemaining.decrementAndGet() == 0){
|
||||||
|
if (callback != null)
|
||||||
callback.onSuccess(this.aggregatedResult.get());
|
callback.onSuccess(this.aggregatedResult.get());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -158,110 +205,80 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
|
|
||||||
// Notify caller about failure
|
// Notify caller about failure
|
||||||
|
if (callback != null)
|
||||||
callback.onFailure(t);
|
callback.onFailure(t);
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private class ReadBatchCallback implements FutureCallback<List<BatchChunk>> {
|
||||||
|
|
||||||
|
private final BulletinBoardMessage stub;
|
||||||
|
private final FutureCallback<BulletinBoardMessage> callback;
|
||||||
|
|
||||||
|
public ReadBatchCallback(BulletinBoardMessage stub, FutureCallback<BulletinBoardMessage> callback) {
|
||||||
|
this.stub = stub;
|
||||||
|
this.callback = callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(List<BatchChunk> result) {
|
||||||
|
callback.onSuccess(BulletinBoardUtils.gatherBatch(stub, result));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This callback ties together the different parts of a CompleteBatch as they arrive from the server
|
* This callback receives a message which may be a stub
|
||||||
* It assembles a CompleteBatch from the parts and sends it to the user if all parts arrived
|
* If the message is not a stub: it returns it as is to a callback function
|
||||||
* If any part fails to arrive: it invokes the onFailure method
|
* If it is a stub: it schedules a read of the batch data which will return a complete message to the callback function
|
||||||
*/
|
*/
|
||||||
class CompleteBatchReadCallback {
|
class CompleteMessageReadCallback implements FutureCallback<List<BulletinBoardMessage>>{
|
||||||
|
|
||||||
private final FutureCallback<CompleteBatch> callback;
|
private final FutureCallback<BulletinBoardMessage> callback;
|
||||||
|
|
||||||
private List<BatchData> batchDataList;
|
public CompleteMessageReadCallback(FutureCallback<BulletinBoardMessage> callback) {
|
||||||
private BulletinBoardMessage batchMessage;
|
|
||||||
|
|
||||||
private AtomicInteger remainingQueries;
|
|
||||||
private AtomicBoolean failed;
|
|
||||||
|
|
||||||
public CompleteBatchReadCallback(FutureCallback<CompleteBatch> callback) {
|
|
||||||
|
|
||||||
this.callback = callback;
|
this.callback = callback;
|
||||||
|
|
||||||
remainingQueries = new AtomicInteger(2);
|
|
||||||
failed = new AtomicBoolean(false);
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
protected void combineAndReturn() {
|
|
||||||
|
|
||||||
final String[] prefixes = {
|
|
||||||
BulletinBoardConstants.BATCH_ID_TAG_PREFIX,
|
|
||||||
BulletinBoardConstants.BATCH_TAG};
|
|
||||||
|
|
||||||
if (remainingQueries.decrementAndGet() == 0){
|
|
||||||
|
|
||||||
String batchIdStr = BulletinBoardUtils.findTagWithPrefix(batchMessage, BulletinBoardConstants.BATCH_ID_TAG_PREFIX);
|
|
||||||
|
|
||||||
if (batchIdStr == null){
|
|
||||||
callback.onFailure(new CommunicationException("Server returned invalid message with no Batch ID tag"));
|
|
||||||
}
|
|
||||||
|
|
||||||
BeginBatchMessage beginBatchMessage =
|
|
||||||
BeginBatchMessage.newBuilder()
|
|
||||||
.setSignerId(batchMessage.getSig(0).getSignerId())
|
|
||||||
.setBatchId(Integer.parseInt(batchIdStr))
|
|
||||||
.addAllTag(BulletinBoardUtils.removePrefixTags(batchMessage, Arrays.asList(prefixes)))
|
|
||||||
.build();
|
|
||||||
callback.onSuccess(new CompleteBatch(beginBatchMessage, batchDataList, batchMessage.getSig(0)));
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
protected void fail(Throwable t) {
|
|
||||||
if (failed.compareAndSet(false, true)) {
|
|
||||||
callback.onFailure(t);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @return a FutureCallback for the Batch Data List that ties to this object
|
|
||||||
*/
|
|
||||||
public FutureCallback<List<BatchData>> asBatchDataListFutureCallback() {
|
|
||||||
return new FutureCallback<List<BatchData>>() {
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onSuccess(List<BatchData> result) {
|
|
||||||
batchDataList = result;
|
|
||||||
|
|
||||||
combineAndReturn();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onFailure(Throwable t) {
|
|
||||||
fail(t);
|
|
||||||
}
|
|
||||||
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @return a FutureCallback for the Bulletin Board Message that ties to this object
|
|
||||||
*/
|
|
||||||
public FutureCallback<List<BulletinBoardMessage>> asBulletinBoardMessageListFutureCallback() {
|
|
||||||
return new FutureCallback<List<BulletinBoardMessage>>() {
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(List<BulletinBoardMessage> result) {
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
if (result.size() < 1){
|
if (result.size() <= 0) {
|
||||||
onFailure(new IllegalArgumentException("Server returned empty message list"));
|
onFailure(new CommunicationException("Could not find required message on the server."));
|
||||||
return;
|
} else {
|
||||||
|
|
||||||
|
BulletinBoardMessage msg = result.get(0);
|
||||||
|
|
||||||
|
if (msg.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
callback.onSuccess(msg);
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// Create job with MAX retries for retrieval of the Batch Data List
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(MessageID.newBuilder()
|
||||||
|
.setID(msg.getMsg().getMsgId())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerReadBatchWorker batchWorker = new SingleServerReadBatchWorker(dbAddress, batchQuery, MAX_RETRIES);
|
||||||
|
|
||||||
|
scheduleWorker(batchWorker, new ReadBatchCallback(msg, callback));
|
||||||
|
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
batchMessage = result.get(0);
|
|
||||||
|
|
||||||
combineAndReturn();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
fail(t);
|
callback.onFailure(t);
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -289,8 +306,13 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
public void onSuccess(List<BulletinBoardMessage> result) {
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
|
||||||
// Report new messages to user
|
// Report new messages to user
|
||||||
|
if (callback != null)
|
||||||
callback.onSuccess(result);
|
callback.onSuccess(result);
|
||||||
|
|
||||||
|
// Update filter if needed
|
||||||
|
|
||||||
|
if (result.size() > 0) {
|
||||||
|
|
||||||
// Remove last filter from list (MIN_ENTRY one)
|
// Remove last filter from list (MIN_ENTRY one)
|
||||||
filterBuilder.removeFilter(filterBuilder.getFilterCount() - 1);
|
filterBuilder.removeFilter(filterBuilder.getFilterCount() - 1);
|
||||||
|
|
||||||
|
@ -300,11 +322,15 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
.setEntry(result.get(result.size() - 1).getEntryNum() + 1)
|
.setEntry(result.get(result.size() - 1).getEntryNum() + 1)
|
||||||
.build());
|
.build());
|
||||||
|
|
||||||
// Create new worker with updated task
|
}
|
||||||
worker = new SingleServerReadMessagesWorker(worker.serverAddress, filterBuilder.build(), 1);
|
|
||||||
|
|
||||||
// Schedule the worker
|
// Create new worker with updated task
|
||||||
scheduleWorker(worker, this);
|
worker = new SingleServerReadMessagesWorker(worker.serverAddress, filterBuilder.build(), MAX_RETRIES);
|
||||||
|
|
||||||
|
RetryCallback<List<BulletinBoardMessage>> retryCallback = new RetryCallback<>(worker, this);
|
||||||
|
|
||||||
|
// Schedule the worker to run after the given interval has elapsed
|
||||||
|
Futures.addCallback(executorService.schedule(worker, SUBSCRIPTION_INTERVAL_IN_MILLISECONDS, TimeUnit.MILLISECONDS), retryCallback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -315,6 +341,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
fail();
|
fail();
|
||||||
|
|
||||||
// Notify caller about failure and terminate subscription
|
// Notify caller about failure and terminate subscription
|
||||||
|
if (callback != null)
|
||||||
callback.onFailure(t);
|
callback.onFailure(t);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -325,8 +352,8 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
|
|
||||||
this.executorService = executorService;
|
this.executorService = executorService;
|
||||||
|
|
||||||
this.failDelayInMilliseconds = failDelayInMilliseconds;
|
this.FAIL_DELAY_IN_MILLISECONDS = failDelayInMilliseconds;
|
||||||
this.subscriptionIntervalInMilliseconds = subscriptionIntervalInMilliseconds;
|
this.SUBSCRIPTION_INTERVAL_IN_MILLISECONDS = subscriptionIntervalInMilliseconds;
|
||||||
|
|
||||||
// Set server error time to a time sufficiently in the past to make new jobs go through
|
// Set server error time to a time sufficiently in the past to make new jobs go through
|
||||||
lastServerErrorTime = System.currentTimeMillis() - failDelayInMilliseconds;
|
lastServerErrorTime = System.currentTimeMillis() - failDelayInMilliseconds;
|
||||||
|
@ -348,79 +375,289 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
@Override
|
@Override
|
||||||
public void init(BulletinBoardClientParams clientParams) {
|
public void init(BulletinBoardClientParams clientParams) {
|
||||||
|
|
||||||
// Perform usual setup
|
this.digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
super.init(clientParams);
|
|
||||||
|
|
||||||
// Wrap the Digest into a BatchDigest
|
|
||||||
batchDigest = new GenericBatchDigest(digest);
|
|
||||||
|
|
||||||
// Remove all but first DB address
|
// Remove all but first DB address
|
||||||
String dbAddress = meerkatDBs.get(0);
|
this.dbAddress = clientParams.getBulletinBoardAddress(0);
|
||||||
meerkatDBs = new LinkedList<>();
|
|
||||||
meerkatDBs.add(dbAddress);
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Synchronous methods
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
||||||
|
|
||||||
|
SingleServerPostMessageWorker worker = new SingleServerPostMessageWorker(dbAddress, msg, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<Boolean> retry = new SynchronousRetry<>(worker);
|
||||||
|
|
||||||
|
retry.run();
|
||||||
|
|
||||||
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
|
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public float getRedundancy(MessageID id) throws CommunicationException {
|
||||||
|
|
||||||
|
SingleServerGetRedundancyWorker worker = new SingleServerGetRedundancyWorker(dbAddress, id, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<Float> retry = new SynchronousRetry<>(worker);
|
||||||
|
|
||||||
|
return retry.run();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException {
|
||||||
|
|
||||||
|
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(dbAddress, filterList, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<List<BulletinBoardMessage>> retry = new SynchronousRetry<>(worker);
|
||||||
|
|
||||||
|
return retry.run();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize) throws CommunicationException {
|
||||||
|
|
||||||
|
// Begin the batch and obtain identifier
|
||||||
|
|
||||||
|
BeginBatchMessage beginBatchMessage = BeginBatchMessage.newBuilder()
|
||||||
|
.addAllTag(msg.getMsg().getTagList())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerBeginBatchWorker beginBatchWorker = new SingleServerBeginBatchWorker(dbAddress, beginBatchMessage, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<Int64Value> beginRetry = new SynchronousRetry<>(beginBatchWorker);
|
||||||
|
|
||||||
|
Int64Value identifier = beginRetry.run();
|
||||||
|
|
||||||
|
// Post data chunks
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = BulletinBoardUtils.breakToBatch(msg, chunkSize);
|
||||||
|
|
||||||
|
BatchMessage.Builder builder = BatchMessage.newBuilder().setBatchId(identifier.getValue());
|
||||||
|
|
||||||
|
int position = 0;
|
||||||
|
|
||||||
|
for (BatchChunk data : batchChunkList) {
|
||||||
|
|
||||||
|
builder.setSerialNum(position).setData(data);
|
||||||
|
|
||||||
|
SingleServerPostBatchWorker dataWorker = new SingleServerPostBatchWorker(dbAddress, builder.build(), MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<Boolean> dataRetry = new SynchronousRetry<>(dataWorker);
|
||||||
|
|
||||||
|
dataRetry.run();
|
||||||
|
|
||||||
|
// Increment position in batch
|
||||||
|
position++;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close batch
|
||||||
|
|
||||||
|
CloseBatchMessage closeBatchMessage = CloseBatchMessage.newBuilder()
|
||||||
|
.setBatchId(identifier.getValue())
|
||||||
|
.addAllSig(msg.getSigList())
|
||||||
|
.setTimestamp(msg.getMsg().getTimestamp())
|
||||||
|
.setBatchLength(position)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerCloseBatchWorker closeBatchWorker = new SingleServerCloseBatchWorker(dbAddress, closeBatchMessage, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<Boolean> retry = new SynchronousRetry<>(closeBatchWorker);
|
||||||
|
|
||||||
|
retry.run();
|
||||||
|
|
||||||
|
// Calculate ID and return
|
||||||
|
|
||||||
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
|
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readMessage(MessageID msgID) throws CommunicationException {
|
||||||
|
|
||||||
|
// Retrieve message (which may be a stub)
|
||||||
|
|
||||||
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgID.getID())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerReadMessagesWorker stubWorker = new SingleServerReadMessagesWorker(dbAddress, filterList, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<List<BulletinBoardMessage>> retry = new SynchronousRetry<>(stubWorker);
|
||||||
|
|
||||||
|
List<BulletinBoardMessage> messages = retry.run();
|
||||||
|
|
||||||
|
if (messages.size() <= 0) {
|
||||||
|
throw new CommunicationException("Could not find message in database.");
|
||||||
|
}
|
||||||
|
|
||||||
|
BulletinBoardMessage msg = messages.get(0);
|
||||||
|
|
||||||
|
if (msg.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
|
||||||
|
// We retrieved a complete message. Return it.
|
||||||
|
return msg;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// We retrieved a stub. Retrieve data.
|
||||||
|
return readBatchData(msg);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BulletinBoardMessage readBatchData(BulletinBoardMessage stub) throws CommunicationException, IllegalArgumentException {
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(MessageID.newBuilder()
|
||||||
|
.setID(stub.getMsg().getMsgId())
|
||||||
|
.build())
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerReadBatchWorker readBatchWorker = new SingleServerReadBatchWorker(dbAddress, batchQuery, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<List<BatchChunk>> batchRetry = new SynchronousRetry<>(readBatchWorker);
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = batchRetry.run();
|
||||||
|
|
||||||
|
return BulletinBoardUtils.gatherBatch(stub, batchChunkList);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException {
|
||||||
|
|
||||||
|
SingleServerGenerateSyncQueryWorker worker =
|
||||||
|
new SingleServerGenerateSyncQueryWorker(dbAddress, generateSyncQueryParams, MAX_RETRIES);
|
||||||
|
|
||||||
|
SynchronousRetry<SyncQuery> retry = new SynchronousRetry<>(worker);
|
||||||
|
|
||||||
|
return retry.run();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Asynchronous methods
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postMessage(BulletinBoardMessage msg, FutureCallback<Boolean> callback) {
|
public MessageID postMessage(BulletinBoardMessage msg, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
// Create worker with redundancy 1 and MAX_RETRIES retries
|
// Create worker with redundancy 1 and MAX_RETRIES retries
|
||||||
SingleServerPostMessageWorker worker = new SingleServerPostMessageWorker(meerkatDBs.get(0), msg, MAX_RETRIES);
|
SingleServerPostMessageWorker worker = new SingleServerPostMessageWorker(dbAddress, msg, MAX_RETRIES);
|
||||||
|
|
||||||
// Submit worker and create callback
|
// Submit worker and create callback
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
||||||
|
|
||||||
// Calculate the correct message ID and return it
|
// Calculate the correct message ID and return it
|
||||||
batchDigest.reset();
|
digest.reset();
|
||||||
batchDigest.update(msg.getMsg());
|
digest.update(msg.getMsg());
|
||||||
return batchDigest.digestAsMessageID();
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class PostBatchDataCallback implements FutureCallback<Boolean> {
|
private class PostBatchDataCallback implements FutureCallback<Boolean> {
|
||||||
|
|
||||||
private final CompleteBatch completeBatch;
|
private final BulletinBoardMessage msg;
|
||||||
|
private final BatchIdentifier identifier;
|
||||||
private final FutureCallback<Boolean> callback;
|
private final FutureCallback<Boolean> callback;
|
||||||
|
|
||||||
public PostBatchDataCallback(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public PostBatchDataCallback(BulletinBoardMessage msg, BatchIdentifier identifier, FutureCallback<Boolean> callback) {
|
||||||
this.completeBatch = completeBatch;
|
this.msg = msg;
|
||||||
|
this.identifier = identifier;
|
||||||
this.callback = callback;
|
this.callback = callback;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(Boolean msg) {
|
public void onSuccess(Boolean result) {
|
||||||
closeBatch(
|
closeBatch(
|
||||||
completeBatch.getCloseBatchMessage(),
|
identifier,
|
||||||
|
msg.getMsg().getTimestamp(),
|
||||||
|
msg.getSigList(),
|
||||||
callback
|
callback
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
callback.onFailure(t);
|
callback.onFailure(t);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private class BeginBatchCallback implements FutureCallback<Boolean> {
|
private class ContinueBatchCallback implements FutureCallback<BatchIdentifier> {
|
||||||
|
|
||||||
private final CompleteBatch completeBatch;
|
private final BulletinBoardMessage msg;
|
||||||
|
private final int chunkSize;
|
||||||
private final FutureCallback<Boolean> callback;
|
private final FutureCallback<Boolean> callback;
|
||||||
|
|
||||||
public BeginBatchCallback(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public ContinueBatchCallback(BulletinBoardMessage msg, int chunkSize, FutureCallback<Boolean> callback) {
|
||||||
this.completeBatch = completeBatch;
|
this.msg = msg;
|
||||||
|
this.chunkSize = chunkSize;
|
||||||
this.callback = callback;
|
this.callback = callback;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(Boolean msg) {
|
public void onSuccess(BatchIdentifier identifier) {
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunkList = BulletinBoardUtils.breakToBatch(msg, chunkSize);
|
||||||
|
|
||||||
postBatchData(
|
postBatchData(
|
||||||
completeBatch.getBeginBatchMessage().getSignerId(),
|
identifier,
|
||||||
completeBatch.getBeginBatchMessage().getBatchId(),
|
batchChunkList,
|
||||||
completeBatch.getBatchDataList(),
|
|
||||||
0,
|
0,
|
||||||
new PostBatchDataCallback(completeBatch,callback));
|
new PostBatchDataCallback(msg, identifier, callback));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (callback != null)
|
||||||
|
callback.onFailure(t);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
|
beginBatch(
|
||||||
|
msg.getMsg().getTagList(),
|
||||||
|
new ContinueBatchCallback(msg, chunkSize, callback)
|
||||||
|
);
|
||||||
|
|
||||||
|
digest.update(msg);
|
||||||
|
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private class BeginBatchCallback implements FutureCallback<Int64Value> {
|
||||||
|
|
||||||
|
private final FutureCallback<BatchIdentifier> callback;
|
||||||
|
|
||||||
|
public BeginBatchCallback(FutureCallback<BatchIdentifier> callback) {
|
||||||
|
this.callback = callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Int64Value result) {
|
||||||
|
callback.onSuccess(new SingleServerBatchIdentifier(result));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -430,51 +667,53 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postBatch(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public void beginBatch(Iterable<String> tags, FutureCallback<BatchIdentifier> callback) {
|
||||||
|
|
||||||
beginBatch(
|
BeginBatchMessage beginBatchMessage = BeginBatchMessage.newBuilder()
|
||||||
completeBatch.getBeginBatchMessage(),
|
.addAllTag(tags)
|
||||||
new BeginBatchCallback(completeBatch, callback)
|
.build();
|
||||||
);
|
|
||||||
|
|
||||||
batchDigest.update(completeBatch);
|
|
||||||
|
|
||||||
return batchDigest.digestAsMessageID();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void beginBatch(BeginBatchMessage beginBatchMessage, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
// Create worker with redundancy 1 and MAX_RETRIES retries
|
// Create worker with redundancy 1 and MAX_RETRIES retries
|
||||||
SingleServerBeginBatchWorker worker =
|
SingleServerBeginBatchWorker worker =
|
||||||
new SingleServerBeginBatchWorker(meerkatDBs.get(0), beginBatchMessage, MAX_RETRIES);
|
new SingleServerBeginBatchWorker(dbAddress, beginBatchMessage, MAX_RETRIES);
|
||||||
|
|
||||||
// Submit worker and create callback
|
// Submit worker and create callback
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, new BeginBatchCallback(callback)));
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList,
|
|
||||||
int startPosition, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
BatchMessage.Builder builder = BatchMessage.newBuilder()
|
@Override
|
||||||
.setSignerId(signerId)
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList,
|
||||||
.setBatchId(batchId);
|
int startPosition, FutureCallback<Boolean> callback) throws IllegalArgumentException{
|
||||||
|
|
||||||
|
// Cast identifier to usable form
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof SingleServerBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
SingleServerBatchIdentifier identifier = (SingleServerBatchIdentifier) batchIdentifier;
|
||||||
|
|
||||||
|
// Update batch size
|
||||||
|
|
||||||
|
identifier.setLength(startPosition + batchChunkList.size());
|
||||||
|
|
||||||
// Create a unified callback to aggregate successful posts
|
// Create a unified callback to aggregate successful posts
|
||||||
|
|
||||||
PostBatchDataListCallback listCallback = new PostBatchDataListCallback(batchDataList.size(), callback);
|
PostBatchChunkListCallback listCallback = new PostBatchChunkListCallback(batchChunkList.size(), callback);
|
||||||
|
|
||||||
// Iterate through data list
|
// Iterate through data list
|
||||||
|
|
||||||
for (BatchData data : batchDataList) {
|
BatchMessage.Builder builder = BatchMessage.newBuilder()
|
||||||
|
.setBatchId(identifier.getBatchId().getValue());
|
||||||
|
|
||||||
|
for (BatchChunk data : batchChunkList) {
|
||||||
builder.setSerialNum(startPosition).setData(data);
|
builder.setSerialNum(startPosition).setData(data);
|
||||||
|
|
||||||
// Create worker with redundancy 1 and MAX_RETRIES retries
|
// Create worker with redundancy 1 and MAX_RETRIES retries
|
||||||
SingleServerPostBatchWorker worker =
|
SingleServerPostBatchWorker worker =
|
||||||
new SingleServerPostBatchWorker(meerkatDBs.get(0), builder.build(), MAX_RETRIES);
|
new SingleServerPostBatchWorker(dbAddress, builder.build(), MAX_RETRIES);
|
||||||
|
|
||||||
// Create worker with redundancy 1 and MAX_RETRIES retries
|
// Create worker with redundancy 1 and MAX_RETRIES retries
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, listCallback));
|
scheduleWorker(worker, new RetryCallback<>(worker, listCallback));
|
||||||
|
@ -486,33 +725,33 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList, FutureCallback<Boolean> callback)
|
||||||
|
throws IllegalArgumentException {
|
||||||
|
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
postBatchData(batchIdentifier, batchChunkList, 0, callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList,
|
public void closeBatch(BatchIdentifier batchIdentifier, Timestamp timestamp, Iterable<Crypto.Signature> signatures, FutureCallback<Boolean> callback)
|
||||||
int startPosition, FutureCallback<Boolean> callback) {
|
throws IllegalArgumentException {
|
||||||
|
|
||||||
postBatchData(ByteString.copyFrom(signerId), batchId, batchDataList, startPosition, callback);
|
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof SingleServerBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
SingleServerBatchIdentifier identifier = (SingleServerBatchIdentifier) batchIdentifier;
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
CloseBatchMessage closeBatchMessage = CloseBatchMessage.newBuilder()
|
||||||
|
.setBatchId(identifier.getBatchId().getValue())
|
||||||
}
|
.setBatchLength(identifier.getLength())
|
||||||
|
.setTimestamp(timestamp)
|
||||||
@Override
|
.addAllSig(signatures)
|
||||||
public void closeBatch(CloseBatchMessage closeBatchMessage, FutureCallback<Boolean> callback) {
|
.build();
|
||||||
|
|
||||||
// Create worker with redundancy 1 and MAX_RETRIES retries
|
// Create worker with redundancy 1 and MAX_RETRIES retries
|
||||||
SingleServerCloseBatchWorker worker =
|
SingleServerCloseBatchWorker worker =
|
||||||
new SingleServerCloseBatchWorker(meerkatDBs.get(0), closeBatchMessage, MAX_RETRIES);
|
new SingleServerCloseBatchWorker(dbAddress, closeBatchMessage, MAX_RETRIES);
|
||||||
|
|
||||||
// Submit worker and create callback
|
// Submit worker and create callback
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
||||||
|
@ -523,7 +762,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
public void getRedundancy(MessageID id, FutureCallback<Float> callback) {
|
public void getRedundancy(MessageID id, FutureCallback<Float> callback) {
|
||||||
|
|
||||||
// Create worker with no retries
|
// Create worker with no retries
|
||||||
SingleServerGetRedundancyWorker worker = new SingleServerGetRedundancyWorker(meerkatDBs.get(0), id, 1);
|
SingleServerGetRedundancyWorker worker = new SingleServerGetRedundancyWorker(dbAddress, id, 1);
|
||||||
|
|
||||||
// Submit job and create callback
|
// Submit job and create callback
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
||||||
|
@ -534,7 +773,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
public void readMessages(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback) {
|
public void readMessages(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback) {
|
||||||
|
|
||||||
// Create job with no retries
|
// Create job with no retries
|
||||||
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(meerkatDBs.get(0), filterList, 1);
|
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(dbAddress, filterList, 1);
|
||||||
|
|
||||||
// Submit job and create callback
|
// Submit job and create callback
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
||||||
|
@ -542,43 +781,55 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage batchSpecificationMessage, FutureCallback<CompleteBatch> callback) {
|
public void readMessage(MessageID msgID, FutureCallback<BulletinBoardMessage> callback) {
|
||||||
|
|
||||||
// Create job with no retries for retrieval of the Bulletin Board Message that defines the batch
|
// Create job with MAX retries for retrieval of the Bulletin Board Message (which may be a stub)
|
||||||
|
|
||||||
MessageFilterList filterList = MessageFilterList.newBuilder()
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
.addFilter(MessageFilter.newBuilder()
|
.addFilter(MessageFilter.newBuilder()
|
||||||
.setType(FilterType.TAG)
|
.setType(FilterType.MSG_ID)
|
||||||
.setTag(BulletinBoardConstants.BATCH_TAG)
|
.setId(msgID.getID())
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.TAG)
|
|
||||||
.setTag(BulletinBoardConstants.BATCH_ID_TAG_PREFIX + batchSpecificationMessage.getBatchId())
|
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.SIGNER_ID)
|
|
||||||
.setId(batchSpecificationMessage.getSignerId())
|
|
||||||
.build())
|
.build())
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
SingleServerReadMessagesWorker messageWorker = new SingleServerReadMessagesWorker(meerkatDBs.get(0), filterList, 1);
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(msgID)
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
// Create job with no retries for retrieval of the Batch Data List
|
SingleServerReadMessagesWorker messageWorker = new SingleServerReadMessagesWorker(dbAddress, filterList, MAX_RETRIES);
|
||||||
SingleServerReadBatchWorker batchWorker = new SingleServerReadBatchWorker(meerkatDBs.get(0), batchSpecificationMessage, 1);
|
|
||||||
|
|
||||||
// Create callback that will combine the two worker products
|
|
||||||
CompleteBatchReadCallback completeBatchReadCallback = new CompleteBatchReadCallback(callback);
|
|
||||||
|
|
||||||
// Submit jobs with wrapped callbacks
|
// Submit jobs with wrapped callbacks
|
||||||
scheduleWorker(messageWorker, new RetryCallback<>(messageWorker, completeBatchReadCallback.asBulletinBoardMessageListFutureCallback()));
|
scheduleWorker(messageWorker, new RetryCallback<>(messageWorker, new CompleteMessageReadCallback(callback)));
|
||||||
scheduleWorker(batchWorker, new RetryCallback<>(batchWorker, completeBatchReadCallback.asBatchDataListFutureCallback()));
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void readBatchData(BulletinBoardMessage stub, FutureCallback<BulletinBoardMessage> callback) throws IllegalArgumentException{
|
||||||
|
|
||||||
|
if (stub.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
throw new IllegalArgumentException("Message is not a stub and does not contain the required message ID");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create job with MAX retries for retrieval of the Batch Data List
|
||||||
|
|
||||||
|
BatchQuery batchQuery = BatchQuery.newBuilder()
|
||||||
|
.setMsgID(MessageID.newBuilder()
|
||||||
|
.setID(stub.getMsg().getMsgId())
|
||||||
|
.build())
|
||||||
|
.setStartPosition(0)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
SingleServerReadBatchWorker batchWorker = new SingleServerReadBatchWorker(dbAddress, batchQuery, MAX_RETRIES);
|
||||||
|
|
||||||
|
scheduleWorker(batchWorker, new RetryCallback<>(batchWorker, new ReadBatchCallback(stub, callback)));
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void querySync(SyncQuery syncQuery, FutureCallback<SyncQueryResponse> callback) {
|
public void querySync(SyncQuery syncQuery, FutureCallback<SyncQueryResponse> callback) {
|
||||||
|
|
||||||
SingleServerQuerySyncWorker worker = new SingleServerQuerySyncWorker(meerkatDBs.get(0), syncQuery, MAX_RETRIES);
|
SingleServerQuerySyncWorker worker = new SingleServerQuerySyncWorker(dbAddress, syncQuery, MAX_RETRIES);
|
||||||
|
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
scheduleWorker(worker, new RetryCallback<>(worker, callback));
|
||||||
|
|
||||||
|
@ -604,7 +855,7 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
.build());
|
.build());
|
||||||
|
|
||||||
// Create job with no retries
|
// Create job with no retries
|
||||||
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(meerkatDBs.get(0), filterListBuilder.build(), MAX_RETRIES);
|
SingleServerReadMessagesWorker worker = new SingleServerReadMessagesWorker(dbAddress, filterListBuilder.build(), MAX_RETRIES);
|
||||||
|
|
||||||
// Submit job and create callback that retries on failure and handles repeated subscription
|
// Submit job and create callback that retries on failure and handles repeated subscription
|
||||||
scheduleWorker(worker, new RetryCallback<>(worker, new SubscriptionCallback(worker, callback)));
|
scheduleWorker(worker, new RetryCallback<>(worker, new SubscriptionCallback(worker, callback)));
|
||||||
|
@ -618,8 +869,6 @@ public class SingleServerBulletinBoardClient extends SimpleBulletinBoardClient i
|
||||||
@Override
|
@Override
|
||||||
public void close() {
|
public void close() {
|
||||||
|
|
||||||
super.close();
|
|
||||||
|
|
||||||
executorService.shutdown();
|
executorService.shutdown();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,12 +1,11 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
import com.google.protobuf.ByteString;
|
|
||||||
|
|
||||||
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.bulletinboard.workers.multiserver.*;
|
import meerkat.bulletinboard.workers.multiserver.*;
|
||||||
import meerkat.comm.CommunicationException;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
import meerkat.protobuf.Voting.*;
|
import meerkat.protobuf.Voting.*;
|
||||||
|
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
|
@ -31,18 +30,33 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
// Per-server clients
|
// Per-server clients
|
||||||
private List<SingleServerBulletinBoardClient> clients;
|
private List<SingleServerBulletinBoardClient> clients;
|
||||||
|
|
||||||
private BatchDigest batchDigest;
|
private BulletinBoardDigest batchDigest;
|
||||||
|
|
||||||
private final static int POST_MESSAGE_RETRY_NUM = 3;
|
private final static int POST_MESSAGE_RETRY_NUM = 3;
|
||||||
private final static int READ_MESSAGES_RETRY_NUM = 1;
|
private final static int READ_MESSAGES_RETRY_NUM = 1;
|
||||||
private final static int GET_REDUNDANCY_RETRY_NUM = 1;
|
private final static int GET_REDUNDANCY_RETRY_NUM = 1;
|
||||||
|
|
||||||
private static final int SERVER_THREADPOOL_SIZE = 5;
|
private final int SERVER_THREADPOOL_SIZE;
|
||||||
private static final long FAIL_DELAY = 5000;
|
private final long FAIL_DELAY;
|
||||||
private static final long SUBSCRIPTION_INTERVAL = 10000;
|
private final long SUBSCRIPTION_INTERVAL;
|
||||||
|
|
||||||
|
private static final int DEFAULT_SERVER_THREADPOOL_SIZE = 5;
|
||||||
|
private static final long DEFAULT_FAIL_DELAY = 5000;
|
||||||
|
private static final long DEFAULT_SUBSCRIPTION_INTERVAL = 10000;
|
||||||
|
|
||||||
private int minAbsoluteRedundancy;
|
private int minAbsoluteRedundancy;
|
||||||
|
|
||||||
|
|
||||||
|
public ThreadedBulletinBoardClient(int serverThreadpoolSize, long failDelay, long subscriptionInterval) {
|
||||||
|
SERVER_THREADPOOL_SIZE = serverThreadpoolSize;
|
||||||
|
FAIL_DELAY = failDelay;
|
||||||
|
SUBSCRIPTION_INTERVAL = subscriptionInterval;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ThreadedBulletinBoardClient() {
|
||||||
|
this(DEFAULT_SERVER_THREADPOOL_SIZE, DEFAULT_FAIL_DELAY, DEFAULT_SUBSCRIPTION_INTERVAL);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Stores database locations and initializes the web Client
|
* Stores database locations and initializes the web Client
|
||||||
* Stores the required minimum redundancy.
|
* Stores the required minimum redundancy.
|
||||||
|
@ -54,7 +68,7 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
|
|
||||||
super.init(clientParams);
|
super.init(clientParams);
|
||||||
|
|
||||||
batchDigest = new GenericBatchDigest(digest);
|
batchDigest = new GenericBulletinBoardDigest(digest);
|
||||||
|
|
||||||
minAbsoluteRedundancy = (int) (clientParams.getMinRedundancy() * (float) clientParams.getBulletinBoardAddressCount());
|
minAbsoluteRedundancy = (int) (clientParams.getMinRedundancy() * (float) clientParams.getBulletinBoardAddressCount());
|
||||||
|
|
||||||
|
@ -100,28 +114,28 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public MessageID postBatch(CompleteBatch completeBatch, FutureCallback<Boolean> callback) {
|
public MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize, FutureCallback<Boolean> callback) {
|
||||||
|
|
||||||
// Create job
|
// Create job
|
||||||
MultiServerPostBatchWorker worker =
|
MultiServerPostBatchWorker worker =
|
||||||
new MultiServerPostBatchWorker(clients, minAbsoluteRedundancy, completeBatch, POST_MESSAGE_RETRY_NUM, callback);
|
new MultiServerPostBatchWorker(clients, minAbsoluteRedundancy, msg, chunkSize, POST_MESSAGE_RETRY_NUM, callback);
|
||||||
|
|
||||||
// Submit job
|
// Submit job
|
||||||
executorService.submit(worker);
|
executorService.submit(worker);
|
||||||
|
|
||||||
// Calculate the correct message ID and return it
|
// Calculate the correct message ID and return it
|
||||||
batchDigest.reset();
|
batchDigest.reset();
|
||||||
batchDigest.update(completeBatch);
|
batchDigest.update(msg);
|
||||||
return batchDigest.digestAsMessageID();
|
return batchDigest.digestAsMessageID();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void beginBatch(BeginBatchMessage beginBatchMessage, FutureCallback<Boolean> callback) {
|
public void beginBatch(Iterable<String> tags, FutureCallback<BatchIdentifier> callback) {
|
||||||
|
|
||||||
// Create job
|
// Create job
|
||||||
MultiServerBeginBatchWorker worker =
|
MultiServerBeginBatchWorker worker =
|
||||||
new MultiServerBeginBatchWorker(clients, minAbsoluteRedundancy, beginBatchMessage, POST_MESSAGE_RETRY_NUM, callback);
|
new MultiServerBeginBatchWorker(clients, minAbsoluteRedundancy, tags, POST_MESSAGE_RETRY_NUM, callback);
|
||||||
|
|
||||||
// Submit job
|
// Submit job
|
||||||
executorService.submit(worker);
|
executorService.submit(worker);
|
||||||
|
@ -129,10 +143,18 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList,
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList,
|
||||||
int startPosition, FutureCallback<Boolean> callback) {
|
int startPosition, FutureCallback<Boolean> callback) throws IllegalArgumentException {
|
||||||
|
|
||||||
BatchDataContainer batchDataContainer = new BatchDataContainer(signerId, batchId, batchDataList, startPosition);
|
// Cast identifier to usable form
|
||||||
|
|
||||||
|
if (!(batchIdentifier instanceof MultiServerBatchIdentifier)){
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
|
}
|
||||||
|
|
||||||
|
MultiServerBatchIdentifier identifier = (MultiServerBatchIdentifier) batchIdentifier;
|
||||||
|
|
||||||
|
BatchDataContainer batchDataContainer = new BatchDataContainer(identifier, batchChunkList, startPosition);
|
||||||
|
|
||||||
// Create job
|
// Create job
|
||||||
MultiServerPostBatchDataWorker worker =
|
MultiServerPostBatchDataWorker worker =
|
||||||
|
@ -144,33 +166,26 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList, FutureCallback<Boolean> callback)
|
||||||
|
throws IllegalArgumentException {
|
||||||
|
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
postBatchData(batchIdentifier, batchChunkList, 0, callback);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList,
|
public void closeBatch(BatchIdentifier payload, Timestamp timestamp, Iterable<Signature> signatures, FutureCallback<Boolean> callback)
|
||||||
int startPosition, FutureCallback<Boolean> callback) {
|
throws IllegalArgumentException{
|
||||||
|
|
||||||
postBatchData(signerId.toByteArray(), batchId, batchDataList, startPosition, callback);
|
|
||||||
|
|
||||||
|
if (!(payload instanceof MultiServerBatchIdentifier)) {
|
||||||
|
throw new IllegalArgumentException("Error: batch identifier supplied was not created by this class.");
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
MultiServerBatchIdentifier identifier = (MultiServerBatchIdentifier) payload;
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
postBatchData(signerId, batchId, batchDataList, 0, callback);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void closeBatch(CloseBatchMessage closeBatchMessage, FutureCallback<Boolean> callback) {
|
|
||||||
|
|
||||||
// Create job
|
// Create job
|
||||||
MultiServerCloseBatchWorker worker =
|
MultiServerCloseBatchWorker worker =
|
||||||
new MultiServerCloseBatchWorker(clients, minAbsoluteRedundancy, closeBatchMessage, POST_MESSAGE_RETRY_NUM, callback);
|
new MultiServerCloseBatchWorker(clients, minAbsoluteRedundancy, identifier, timestamp, signatures, POST_MESSAGE_RETRY_NUM, callback);
|
||||||
|
|
||||||
// Submit job
|
// Submit job
|
||||||
executorService.submit(worker);
|
executorService.submit(worker);
|
||||||
|
@ -213,11 +228,27 @@ public class ThreadedBulletinBoardClient extends SimpleBulletinBoardClient imple
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage batchSpecificationMessage, FutureCallback<CompleteBatch> callback) {
|
public void readMessage(MessageID msgID, FutureCallback<BulletinBoardMessage> callback) {
|
||||||
|
|
||||||
|
//Create job
|
||||||
|
MultiServerReadMessageWorker worker =
|
||||||
|
new MultiServerReadMessageWorker(clients, minAbsoluteRedundancy, msgID, READ_MESSAGES_RETRY_NUM, callback);
|
||||||
|
|
||||||
|
// Submit job
|
||||||
|
executorService.submit(worker);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void readBatchData(BulletinBoardMessage stub, FutureCallback<BulletinBoardMessage> callback) throws IllegalArgumentException {
|
||||||
|
|
||||||
|
if (stub.getMsg().getDataTypeCase() != UnsignedBulletinBoardMessage.DataTypeCase.MSGID) {
|
||||||
|
throw new IllegalArgumentException("Message is not a stub and does not contain the required message ID");
|
||||||
|
}
|
||||||
|
|
||||||
// Create job
|
// Create job
|
||||||
MultiServerReadBatchWorker worker =
|
MultiServerReadBatchDataWorker worker =
|
||||||
new MultiServerReadBatchWorker(clients, minAbsoluteRedundancy, batchSpecificationMessage, READ_MESSAGES_RETRY_NUM, callback);
|
new MultiServerReadBatchDataWorker(clients, minAbsoluteRedundancy, stub, READ_MESSAGES_RETRY_NUM, callback);
|
||||||
|
|
||||||
// Submit job
|
// Submit job
|
||||||
executorService.submit(worker);
|
executorService.submit(worker);
|
||||||
|
|
|
@ -8,7 +8,6 @@ import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
import static meerkat.protobuf.BulletinBoardAPI.FilterType.*;
|
import static meerkat.protobuf.BulletinBoardAPI.FilterType.*;
|
||||||
|
|
||||||
import java.sql.Time;
|
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
import java.util.concurrent.Semaphore;
|
import java.util.concurrent.Semaphore;
|
||||||
import java.util.concurrent.atomic.AtomicBoolean;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
@ -19,20 +18,22 @@ import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
*/
|
*/
|
||||||
public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber {
|
public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber {
|
||||||
|
|
||||||
protected final Collection<SubscriptionAsyncBulletinBoardClient> clients;
|
protected final Collection<SubscriptionBulletinBoardClient> clients;
|
||||||
protected final BulletinBoardClient localClient;
|
protected final BulletinBoardClient localClient;
|
||||||
|
|
||||||
protected Iterator<SubscriptionAsyncBulletinBoardClient> clientIterator;
|
protected Iterator<SubscriptionBulletinBoardClient> clientIterator;
|
||||||
protected SubscriptionAsyncBulletinBoardClient currentClient;
|
protected SubscriptionBulletinBoardClient currentClient;
|
||||||
|
|
||||||
private long lastServerSwitchTime;
|
private long lastServerSwitchTime;
|
||||||
|
|
||||||
private AtomicBoolean isSyncInProgress;
|
private AtomicBoolean isSyncInProgress;
|
||||||
private Semaphore rescheduleSemaphore;
|
private Semaphore rescheduleSemaphore;
|
||||||
|
|
||||||
|
private AtomicBoolean stopped;
|
||||||
|
|
||||||
private static final Float[] BREAKPOINTS = {0.5f, 0.75f, 0.9f, 0.95f, 0.99f, 0.999f};
|
private static final Float[] BREAKPOINTS = {0.5f, 0.75f, 0.9f, 0.95f, 0.99f, 0.999f};
|
||||||
|
|
||||||
public ThreadedBulletinBoardSubscriber(Collection<SubscriptionAsyncBulletinBoardClient> clients, BulletinBoardClient localClient) {
|
public ThreadedBulletinBoardSubscriber(Collection<SubscriptionBulletinBoardClient> clients, BulletinBoardClient localClient) {
|
||||||
|
|
||||||
this.clients = clients;
|
this.clients = clients;
|
||||||
this.localClient = localClient;
|
this.localClient = localClient;
|
||||||
|
@ -45,6 +46,8 @@ public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber
|
||||||
isSyncInProgress = new AtomicBoolean(false);
|
isSyncInProgress = new AtomicBoolean(false);
|
||||||
rescheduleSemaphore = new Semaphore(1);
|
rescheduleSemaphore = new Semaphore(1);
|
||||||
|
|
||||||
|
stopped = new AtomicBoolean(false);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -132,6 +135,7 @@ public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber
|
||||||
|
|
||||||
//TODO: log
|
//TODO: log
|
||||||
|
|
||||||
|
if (callback != null)
|
||||||
callback.onFailure(e); // Hard error: Cannot guarantee subscription safety
|
callback.onFailure(e); // Hard error: Cannot guarantee subscription safety
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -218,6 +222,7 @@ public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber
|
||||||
public void onSuccess(List<BulletinBoardMessage> result) {
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
|
||||||
// Propagate result to caller
|
// Propagate result to caller
|
||||||
|
if (callback != null)
|
||||||
callback.onSuccess(result);
|
callback.onSuccess(result);
|
||||||
|
|
||||||
// Renew subscription
|
// Renew subscription
|
||||||
|
@ -245,11 +250,11 @@ public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber
|
||||||
super(filterList, callback);
|
super(filterList, callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(List<BulletinBoardMessage> result) {
|
public void onSuccess(List<BulletinBoardMessage> result) {
|
||||||
|
|
||||||
// Propagate result to caller
|
// Propagate result to caller
|
||||||
|
if (callback != null)
|
||||||
callback.onSuccess(result);
|
callback.onSuccess(result);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -268,5 +273,4 @@ public class ThreadedBulletinBoardSubscriber implements BulletinBoardSubscriber
|
||||||
subscribe(filterList, 0, callback);
|
subscribe(filterList, 0, callback);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,28 +1,96 @@
|
||||||
package meerkat.bulletinboard.workers.multiserver;
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import meerkat.bulletinboard.MultiServerBatchIdentifier;
|
||||||
|
import meerkat.bulletinboard.MultiServerWorker;
|
||||||
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BeginBatchMessage;
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
*/
|
*/
|
||||||
public class MultiServerBeginBatchWorker extends MultiServerGenericPostWorker<BeginBatchMessage> {
|
public class MultiServerBeginBatchWorker extends MultiServerWorker<Iterable<String>, BatchIdentifier> {
|
||||||
|
|
||||||
|
private BatchIdentifier[] identifiers;
|
||||||
|
private AtomicInteger remainingServers;
|
||||||
|
|
||||||
public MultiServerBeginBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
public MultiServerBeginBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
int minServers, BeginBatchMessage payload, int maxRetry,
|
int minServers, Iterable<String> payload, int maxRetry,
|
||||||
FutureCallback<Boolean> futureCallback) {
|
FutureCallback<BatchIdentifier> futureCallback) {
|
||||||
|
|
||||||
super(clients, minServers, payload, maxRetry, futureCallback);
|
super(clients, minServers, payload, maxRetry, futureCallback);
|
||||||
|
|
||||||
|
identifiers = new BatchIdentifier[clients.size()];
|
||||||
|
|
||||||
|
for (int i = 0 ; i < identifiers.length ; i++) {
|
||||||
|
identifiers[i] = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
remainingServers = new AtomicInteger(clients.size());
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private class BeginBatchCallback implements FutureCallback<BatchIdentifier> {
|
||||||
|
|
||||||
|
private final int clientNum;
|
||||||
|
|
||||||
|
public BeginBatchCallback(int clientNum) {
|
||||||
|
this.clientNum = clientNum;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void finishPost() {
|
||||||
|
|
||||||
|
if (remainingServers.decrementAndGet() <= 0){
|
||||||
|
|
||||||
|
if (minServers.decrementAndGet() <= 0) {
|
||||||
|
MultiServerBeginBatchWorker.this.onSuccess(new MultiServerBatchIdentifier(identifiers));
|
||||||
|
} else {
|
||||||
|
MultiServerBeginBatchWorker.this.onFailure(new CommunicationException("Could not open batch in enough servers"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void doPost(SingleServerBulletinBoardClient client, BeginBatchMessage payload) {
|
public void onSuccess(BatchIdentifier result) {
|
||||||
client.beginBatch(payload, this);
|
|
||||||
|
identifiers[clientNum] = result;
|
||||||
|
finishPost();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
finishPost();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(BatchIdentifier result) {
|
||||||
|
succeed(result);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
fail(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
|
||||||
|
int clientNum = 0;
|
||||||
|
|
||||||
|
for (SingleServerBulletinBoardClient client : clients){
|
||||||
|
|
||||||
|
client.beginBatch(payload, new BeginBatchCallback(clientNum));
|
||||||
|
|
||||||
|
clientNum++;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,27 +1,83 @@
|
||||||
package meerkat.bulletinboard.workers.multiserver;
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import com.google.protobuf.Timestamp;
|
||||||
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
import meerkat.bulletinboard.BatchDataContainer;
|
||||||
|
import meerkat.bulletinboard.MultiServerBatchIdentifier;
|
||||||
|
import meerkat.bulletinboard.MultiServerWorker;
|
||||||
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.CloseBatchMessage;
|
import meerkat.crypto.DigitalSignature;
|
||||||
|
import meerkat.protobuf.Crypto;
|
||||||
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
|
|
||||||
|
import java.util.Iterator;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
*/
|
*/
|
||||||
public class MultiServerCloseBatchWorker extends MultiServerGenericPostWorker<CloseBatchMessage> {
|
public class MultiServerCloseBatchWorker extends MultiServerWorker<MultiServerBatchIdentifier, Boolean> {
|
||||||
|
|
||||||
|
private final Timestamp timestamp;
|
||||||
|
private final Iterable<Crypto.Signature> signatures;
|
||||||
|
|
||||||
public MultiServerCloseBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
public MultiServerCloseBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
int minServers, CloseBatchMessage payload, int maxRetry,
|
int minServers, MultiServerBatchIdentifier payload, Timestamp timestamp, Iterable<Crypto.Signature> signatures,
|
||||||
FutureCallback<Boolean> futureCallback) {
|
int maxRetry, FutureCallback<Boolean> futureCallback) {
|
||||||
|
|
||||||
super(clients, minServers, payload, maxRetry, futureCallback);
|
super(clients, minServers, payload, maxRetry, futureCallback);
|
||||||
|
|
||||||
|
this.timestamp = timestamp;
|
||||||
|
this.signatures = signatures;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void doPost(SingleServerBulletinBoardClient client, CloseBatchMessage payload) {
|
public void run() {
|
||||||
client.closeBatch(payload, this);
|
|
||||||
|
Iterator<BatchIdentifier> identifierIterator = payload.getIdentifiers().iterator();
|
||||||
|
|
||||||
|
// Iterate through client
|
||||||
|
|
||||||
|
for (SingleServerBulletinBoardClient client : clients) {
|
||||||
|
|
||||||
|
if (identifierIterator.hasNext()) {
|
||||||
|
|
||||||
|
// Fetch the batch identifier supplied by the specific client (may be null if batch open failed on client
|
||||||
|
|
||||||
|
BatchIdentifier identifier = identifierIterator.next();
|
||||||
|
|
||||||
|
if (identifier != null) {
|
||||||
|
|
||||||
|
// Post the data with the matching identifier to the client
|
||||||
|
client.closeBatch(identifier, timestamp, signatures, this);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// Count servers with no batch identifier as failed
|
||||||
|
maxFailedServers.decrementAndGet();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
if (minServers.decrementAndGet() <= 0){
|
||||||
|
succeed(result);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (maxFailedServers.decrementAndGet() <= 0){
|
||||||
|
fail(t);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -35,14 +35,9 @@ public abstract class MultiServerGenericPostWorker<T> extends MultiServerWorker<
|
||||||
public void run() {
|
public void run() {
|
||||||
|
|
||||||
// Iterate through servers
|
// Iterate through servers
|
||||||
|
for (SingleServerBulletinBoardClient client : clients) {
|
||||||
Iterator<SingleServerBulletinBoardClient> clientIterator = getClientIterator();
|
|
||||||
|
|
||||||
while (clientIterator.hasNext()) {
|
|
||||||
|
|
||||||
// Send request to Server
|
// Send request to Server
|
||||||
SingleServerBulletinBoardClient client = clientIterator.next();
|
|
||||||
|
|
||||||
doPost(client, payload);
|
doPost(client, payload);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -14,7 +14,9 @@ import java.util.List;
|
||||||
*/
|
*/
|
||||||
public abstract class MultiServerGenericReadWorker<IN, OUT> extends MultiServerWorker<IN, OUT>{
|
public abstract class MultiServerGenericReadWorker<IN, OUT> extends MultiServerWorker<IN, OUT>{
|
||||||
|
|
||||||
private final Iterator<SingleServerBulletinBoardClient> clientIterator;
|
private Iterator<SingleServerBulletinBoardClient> clientIterator;
|
||||||
|
|
||||||
|
private String errorString;
|
||||||
|
|
||||||
public MultiServerGenericReadWorker(List<SingleServerBulletinBoardClient> clients,
|
public MultiServerGenericReadWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
int minServers, IN payload, int maxRetry,
|
int minServers, IN payload, int maxRetry,
|
||||||
|
@ -22,7 +24,8 @@ public abstract class MultiServerGenericReadWorker<IN, OUT> extends MultiServerW
|
||||||
|
|
||||||
super(clients, true, minServers, payload, maxRetry, futureCallback); // Shuffle clients on creation to balance load
|
super(clients, true, minServers, payload, maxRetry, futureCallback); // Shuffle clients on creation to balance load
|
||||||
|
|
||||||
clientIterator = getClientIterator();
|
clientIterator = clients.iterator();
|
||||||
|
errorString = "";
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -46,7 +49,7 @@ public abstract class MultiServerGenericReadWorker<IN, OUT> extends MultiServerW
|
||||||
doRead(payload, client);
|
doRead(payload, client);
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
fail(new CommunicationException("Could not contact any server"));
|
fail(new CommunicationException("Could not contact any server. Errors follow:\n" + errorString));
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -58,6 +61,8 @@ public abstract class MultiServerGenericReadWorker<IN, OUT> extends MultiServerW
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
|
//TODO: log
|
||||||
|
errorString += t.getCause() + " " + t.getMessage() + "\n";
|
||||||
run(); // Retry with next server
|
run(); // Retry with next server
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -36,13 +36,8 @@ public class MultiServerGetRedundancyWorker extends MultiServerWorker<MessageID,
|
||||||
*/
|
*/
|
||||||
public void run(){
|
public void run(){
|
||||||
|
|
||||||
Iterator<SingleServerBulletinBoardClient> clientIterator = getClientIterator();
|
|
||||||
|
|
||||||
// Iterate through clients
|
// Iterate through clients
|
||||||
|
for (SingleServerBulletinBoardClient client : clients) {
|
||||||
while (clientIterator.hasNext()) {
|
|
||||||
|
|
||||||
SingleServerBulletinBoardClient client = clientIterator.next();
|
|
||||||
|
|
||||||
// Send request to client
|
// Send request to client
|
||||||
client.getRedundancy(payload,this);
|
client.getRedundancy(payload,this);
|
||||||
|
|
|
@ -1,15 +1,18 @@
|
||||||
package meerkat.bulletinboard.workers.multiserver;
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
import meerkat.bulletinboard.MultiServerWorker;
|
||||||
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
import meerkat.bulletinboard.BatchDataContainer;
|
import meerkat.bulletinboard.BatchDataContainer;
|
||||||
|
|
||||||
|
import java.util.Iterator;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
*/
|
*/
|
||||||
public class MultiServerPostBatchDataWorker extends MultiServerGenericPostWorker<BatchDataContainer> {
|
public class MultiServerPostBatchDataWorker extends MultiServerWorker<BatchDataContainer, Boolean> {
|
||||||
|
|
||||||
public MultiServerPostBatchDataWorker(List<SingleServerBulletinBoardClient> clients,
|
public MultiServerPostBatchDataWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
int minServers, BatchDataContainer payload, int maxRetry,
|
int minServers, BatchDataContainer payload, int maxRetry,
|
||||||
|
@ -20,9 +23,50 @@ public class MultiServerPostBatchDataWorker extends MultiServerGenericPostWorker
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void doPost(SingleServerBulletinBoardClient client, BatchDataContainer payload) {
|
public void run() {
|
||||||
client.postBatchData(payload.signerId, payload.batchId, payload.batchDataList, payload.startPosition, this);
|
|
||||||
|
Iterator<BatchIdentifier> identifierIterator = payload.batchId.getIdentifiers().iterator();
|
||||||
|
|
||||||
|
// Iterate through client
|
||||||
|
|
||||||
|
for (SingleServerBulletinBoardClient client : clients) {
|
||||||
|
|
||||||
|
if (identifierIterator.hasNext()) {
|
||||||
|
|
||||||
|
// Fetch the batch identifier supplied by the specific client (may be null if batch open failed on client
|
||||||
|
|
||||||
|
BatchIdentifier identifier = identifierIterator.next();
|
||||||
|
|
||||||
|
if (identifier != null) {
|
||||||
|
|
||||||
|
// Post the data with the matching identifier to the client
|
||||||
|
client.postBatchData(identifier, payload.batchChunkList, payload.startPosition, this);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// Count servers with no batch identifier as failed
|
||||||
|
maxFailedServers.decrementAndGet();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
if (minServers.decrementAndGet() <= 0){
|
||||||
|
succeed(result);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
if (maxFailedServers.decrementAndGet() <= 0){
|
||||||
|
fail(t);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,27 +1,33 @@
|
||||||
package meerkat.bulletinboard.workers.multiserver;
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
import meerkat.bulletinboard.CompleteBatch;
|
|
||||||
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BulletinBoardMessage;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
*/
|
*/
|
||||||
public class MultiServerPostBatchWorker extends MultiServerGenericPostWorker<CompleteBatch> {
|
public class MultiServerPostBatchWorker extends MultiServerGenericPostWorker<BulletinBoardMessage> {
|
||||||
|
|
||||||
|
private final int chunkSize;
|
||||||
|
|
||||||
public MultiServerPostBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
public MultiServerPostBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
int minServers, CompleteBatch payload, int maxRetry,
|
int minServers, BulletinBoardMessage payload, int chunkSize, int maxRetry,
|
||||||
FutureCallback<Boolean> futureCallback) {
|
FutureCallback<Boolean> futureCallback) {
|
||||||
|
|
||||||
super(clients, minServers, payload, maxRetry, futureCallback);
|
super(clients, minServers, payload, maxRetry, futureCallback);
|
||||||
|
|
||||||
|
this.chunkSize = chunkSize;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected void doPost(SingleServerBulletinBoardClient client, CompleteBatch payload) {
|
protected void doPost(SingleServerBulletinBoardClient client, BulletinBoardMessage payload) {
|
||||||
client.postBatch(payload, this);
|
|
||||||
|
client.postAsBatch(payload, chunkSize, this);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,29 @@
|
||||||
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
|
*/
|
||||||
|
public class MultiServerReadBatchDataWorker extends MultiServerGenericReadWorker<BulletinBoardMessage, BulletinBoardMessage> {
|
||||||
|
|
||||||
|
public MultiServerReadBatchDataWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
|
int minServers, BulletinBoardMessage payload, int maxRetry,
|
||||||
|
FutureCallback<BulletinBoardMessage> futureCallback) {
|
||||||
|
|
||||||
|
super(clients, minServers, payload, maxRetry, futureCallback);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void doRead(BulletinBoardMessage payload, SingleServerBulletinBoardClient client) {
|
||||||
|
client.readBatchData(payload, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
|
@ -1,30 +0,0 @@
|
||||||
package meerkat.bulletinboard.workers.multiserver;
|
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
|
||||||
import meerkat.bulletinboard.CompleteBatch;
|
|
||||||
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchSpecificationMessage;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
|
||||||
*/
|
|
||||||
public class MultiServerReadBatchWorker extends MultiServerGenericReadWorker<BatchSpecificationMessage, CompleteBatch> {
|
|
||||||
|
|
||||||
public MultiServerReadBatchWorker(List<SingleServerBulletinBoardClient> clients,
|
|
||||||
int minServers, BatchSpecificationMessage payload, int maxRetry,
|
|
||||||
FutureCallback<CompleteBatch> futureCallback) {
|
|
||||||
|
|
||||||
super(clients, minServers, payload, maxRetry, futureCallback);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected void doRead(BatchSpecificationMessage payload, SingleServerBulletinBoardClient client) {
|
|
||||||
client.readBatch(payload, this);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
|
@ -0,0 +1,30 @@
|
||||||
|
package meerkat.bulletinboard.workers.multiserver;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import meerkat.bulletinboard.SingleServerBulletinBoardClient;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BulletinBoardMessage;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.MessageID;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
|
*/
|
||||||
|
public class MultiServerReadMessageWorker extends MultiServerGenericReadWorker<MessageID, BulletinBoardMessage> {
|
||||||
|
|
||||||
|
public MultiServerReadMessageWorker(List<SingleServerBulletinBoardClient> clients,
|
||||||
|
int minServers, MessageID payload, int maxRetry,
|
||||||
|
FutureCallback<BulletinBoardMessage> futureCallback) {
|
||||||
|
|
||||||
|
super(clients, minServers, payload, maxRetry, futureCallback);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void doRead(MessageID payload, SingleServerBulletinBoardClient client) {
|
||||||
|
client.readMessage(payload, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
|
@ -1,17 +1,53 @@
|
||||||
package meerkat.bulletinboard.workers.singleserver;
|
package meerkat.bulletinboard.workers.singleserver;
|
||||||
|
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
|
import meerkat.bulletinboard.SingleServerWorker;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BeginBatchMessage;
|
import meerkat.protobuf.BulletinBoardAPI.BeginBatchMessage;
|
||||||
|
import meerkat.rest.Constants;
|
||||||
|
|
||||||
|
import javax.ws.rs.ProcessingException;
|
||||||
|
import javax.ws.rs.client.Client;
|
||||||
|
import javax.ws.rs.client.Entity;
|
||||||
|
import javax.ws.rs.client.WebTarget;
|
||||||
|
import javax.ws.rs.core.Response;
|
||||||
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.BEGIN_BATCH_PATH;
|
import static meerkat.bulletinboard.BulletinBoardConstants.BEGIN_BATCH_PATH;
|
||||||
|
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
* Tries to contact server once and perform a post operation
|
* Tries to contact server once and perform a post operation
|
||||||
*/
|
*/
|
||||||
public class SingleServerBeginBatchWorker extends SingleServerGenericPostWorker<BeginBatchMessage> {
|
public class SingleServerBeginBatchWorker extends SingleServerWorker<BeginBatchMessage,Int64Value> {
|
||||||
|
|
||||||
public SingleServerBeginBatchWorker(String serverAddress, BeginBatchMessage payload, int maxRetry) {
|
public SingleServerBeginBatchWorker(String serverAddress, BeginBatchMessage payload, int maxRetry) {
|
||||||
super(serverAddress, BEGIN_BATCH_PATH, payload, maxRetry);
|
super(serverAddress, payload, maxRetry);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Int64Value call() throws Exception {
|
||||||
|
Client client = clientLocal.get();
|
||||||
|
|
||||||
|
WebTarget webTarget = client.target(serverAddress).path(BULLETIN_BOARD_SERVER_PATH).path(BEGIN_BATCH_PATH);
|
||||||
|
Response response = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(
|
||||||
|
Entity.entity(payload, Constants.MEDIATYPE_PROTOBUF));
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
Int64Value result = response.readEntity(Int64Value.class);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
} catch (ProcessingException | IllegalStateException e) {
|
||||||
|
|
||||||
|
// Post to this server failed
|
||||||
|
throw new CommunicationException("Could not contact the server. Original error: " + e.getMessage());
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new CommunicationException("Could not contact the server. Original error: " + e.getMessage());
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
response.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,55 @@
|
||||||
|
package meerkat.bulletinboard.workers.singleserver;
|
||||||
|
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
|
import meerkat.bulletinboard.SingleServerWorker;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.SyncQuery;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.GenerateSyncQueryParams;
|
||||||
|
import meerkat.rest.Constants;
|
||||||
|
|
||||||
|
import javax.ws.rs.ProcessingException;
|
||||||
|
import javax.ws.rs.client.Client;
|
||||||
|
import javax.ws.rs.client.Entity;
|
||||||
|
import javax.ws.rs.client.WebTarget;
|
||||||
|
import javax.ws.rs.core.Response;
|
||||||
|
|
||||||
|
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
||||||
|
import static meerkat.bulletinboard.BulletinBoardConstants.GENERATE_SYNC_QUERY_PATH;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
|
* Tries to contact server once and perform a Sync Query Generation operation
|
||||||
|
*/
|
||||||
|
public class SingleServerGenerateSyncQueryWorker extends SingleServerWorker<GenerateSyncQueryParams,SyncQuery> {
|
||||||
|
|
||||||
|
public SingleServerGenerateSyncQueryWorker(String serverAddress, GenerateSyncQueryParams payload, int maxRetry) {
|
||||||
|
super(serverAddress, payload, maxRetry);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SyncQuery call() throws Exception {
|
||||||
|
|
||||||
|
Client client = clientLocal.get();
|
||||||
|
|
||||||
|
WebTarget webTarget = client.target(serverAddress).path(BULLETIN_BOARD_SERVER_PATH).path(GENERATE_SYNC_QUERY_PATH);
|
||||||
|
|
||||||
|
Response response = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(Entity.entity(payload, Constants.MEDIATYPE_PROTOBUF));
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
SyncQuery result = response.readEntity(SyncQuery.class);
|
||||||
|
return result;
|
||||||
|
|
||||||
|
} catch (ProcessingException | IllegalStateException e) {
|
||||||
|
|
||||||
|
// Post to this server failed
|
||||||
|
throw new CommunicationException("Could not contact the server. Original error: " + e.getMessage());
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new CommunicationException("Could not contact the server. Original error: " + e.getMessage());
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
response.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -6,7 +6,6 @@ import meerkat.comm.MessageInputStream;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.rest.Constants;
|
import meerkat.rest.Constants;
|
||||||
|
|
||||||
import javax.ws.rs.ProcessingException;
|
|
||||||
import javax.ws.rs.client.Client;
|
import javax.ws.rs.client.Client;
|
||||||
import javax.ws.rs.client.Entity;
|
import javax.ws.rs.client.Entity;
|
||||||
import javax.ws.rs.client.WebTarget;
|
import javax.ws.rs.client.WebTarget;
|
||||||
|
@ -14,7 +13,6 @@ import javax.ws.rs.core.Response;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.lang.reflect.InvocationTargetException;
|
|
||||||
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.READ_MESSAGES_PATH;
|
import static meerkat.bulletinboard.BulletinBoardConstants.READ_MESSAGES_PATH;
|
||||||
|
|
|
@ -1,35 +1,28 @@
|
||||||
package meerkat.bulletinboard.workers.singleserver;
|
package meerkat.bulletinboard.workers.singleserver;
|
||||||
|
|
||||||
import meerkat.bulletinboard.CompleteBatch;
|
|
||||||
import meerkat.bulletinboard.SingleServerWorker;
|
import meerkat.bulletinboard.SingleServerWorker;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.comm.MessageInputStream;
|
import meerkat.comm.MessageInputStream;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.rest.Constants;
|
import meerkat.rest.Constants;
|
||||||
|
|
||||||
import javax.ws.rs.ProcessingException;
|
|
||||||
import javax.ws.rs.client.Client;
|
import javax.ws.rs.client.Client;
|
||||||
import javax.ws.rs.client.Entity;
|
import javax.ws.rs.client.Entity;
|
||||||
import javax.ws.rs.client.WebTarget;
|
import javax.ws.rs.client.WebTarget;
|
||||||
import javax.ws.rs.core.GenericType;
|
|
||||||
import javax.ws.rs.core.Response;
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.lang.reflect.InvocationTargetException;
|
import java.lang.reflect.InvocationTargetException;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
import static meerkat.bulletinboard.BulletinBoardConstants.BULLETIN_BOARD_SERVER_PATH;
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.READ_MESSAGES_PATH;
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.READ_BATCH_PATH;
|
import static meerkat.bulletinboard.BulletinBoardConstants.READ_BATCH_PATH;
|
||||||
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.BATCH_ID_TAG_PREFIX;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
* Created by Arbel Deutsch Peled on 27-Dec-15.
|
||||||
*/
|
*/
|
||||||
public class SingleServerReadBatchWorker extends SingleServerWorker<BatchSpecificationMessage, List<BatchData>> {
|
public class SingleServerReadBatchWorker extends SingleServerWorker<BatchQuery, List<BatchChunk>> {
|
||||||
|
|
||||||
public SingleServerReadBatchWorker(String serverAddress, BatchSpecificationMessage payload, int maxRetry) {
|
public SingleServerReadBatchWorker(String serverAddress, BatchQuery payload, int maxRetry) {
|
||||||
super(serverAddress, payload, maxRetry);
|
super(serverAddress, payload, maxRetry);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -39,7 +32,7 @@ public class SingleServerReadBatchWorker extends SingleServerWorker<BatchSpecifi
|
||||||
* @return the complete batch as read from the server
|
* @return the complete batch as read from the server
|
||||||
* @throws CommunicationException if the server's response is invalid
|
* @throws CommunicationException if the server's response is invalid
|
||||||
*/
|
*/
|
||||||
public List<BatchData> call() throws CommunicationException{
|
public List<BatchChunk> call() throws CommunicationException{
|
||||||
|
|
||||||
Client client = clientLocal.get();
|
Client client = clientLocal.get();
|
||||||
|
|
||||||
|
@ -50,11 +43,11 @@ public class SingleServerReadBatchWorker extends SingleServerWorker<BatchSpecifi
|
||||||
webTarget = client.target(serverAddress).path(BULLETIN_BOARD_SERVER_PATH).path(READ_BATCH_PATH);
|
webTarget = client.target(serverAddress).path(BULLETIN_BOARD_SERVER_PATH).path(READ_BATCH_PATH);
|
||||||
InputStream in = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(Entity.entity(payload, Constants.MEDIATYPE_PROTOBUF), InputStream.class);
|
InputStream in = webTarget.request(Constants.MEDIATYPE_PROTOBUF).post(Entity.entity(payload, Constants.MEDIATYPE_PROTOBUF), InputStream.class);
|
||||||
|
|
||||||
MessageInputStream<BatchData> inputStream = null;
|
MessageInputStream<BatchChunk> inputStream = null;
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|
||||||
inputStream = MessageInputStream.MessageInputStreamFactory.createMessageInputStream(in, BatchData.class);
|
inputStream = MessageInputStream.MessageInputStreamFactory.createMessageInputStream(in, BatchChunk.class);
|
||||||
|
|
||||||
return inputStream.asList();
|
return inputStream.asList();
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,318 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import com.google.protobuf.*;
|
||||||
|
import com.google.protobuf.Timestamp;
|
||||||
|
|
||||||
|
import static meerkat.bulletinboard.BulletinBoardSynchronizer.SyncStatus;
|
||||||
|
|
||||||
|
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer;
|
||||||
|
import meerkat.bulletinboard.sqlserver.H2QueryProvider;
|
||||||
|
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.crypto.concrete.ECDSASignature;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.util.BulletinBoardMessageComparator;
|
||||||
|
import meerkat.util.BulletinBoardMessageGenerator;
|
||||||
|
import org.junit.*;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.security.*;
|
||||||
|
import java.security.cert.CertificateException;
|
||||||
|
import java.util.LinkedList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Random;
|
||||||
|
import java.util.concurrent.Semaphore;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.concurrent.TimeoutException;
|
||||||
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
|
||||||
|
import static org.hamcrest.MatcherAssert.assertThat;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel on 6/1/2016.
|
||||||
|
*/
|
||||||
|
public class BulletinBoardSynchronizerTest {
|
||||||
|
|
||||||
|
private static final String REMOTE_SERVER_ADDRESS = "remoteDB";
|
||||||
|
private static final String LOCAL_SERVER_ADDRESS = "localDB";
|
||||||
|
private static int testCount;
|
||||||
|
|
||||||
|
private static final int THREAD_NUM = 3;
|
||||||
|
private static final int SUBSCRIPTION_INTERVAL = 1000;
|
||||||
|
|
||||||
|
private static final int SYNC_SLEEP_INTERVAL = 500;
|
||||||
|
private static final int SYNC_WAIT_CAP = 1000;
|
||||||
|
|
||||||
|
private DeletableSubscriptionBulletinBoardClient localClient;
|
||||||
|
private AsyncBulletinBoardClient remoteClient;
|
||||||
|
|
||||||
|
private BulletinBoardSynchronizer synchronizer;
|
||||||
|
|
||||||
|
private static BulletinBoardMessageGenerator messageGenerator;
|
||||||
|
private static BulletinBoardMessageComparator messageComparator;
|
||||||
|
|
||||||
|
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
||||||
|
private static String KEYFILE_PASSWORD1 = "secret";
|
||||||
|
private static String CERT1_PEM_EXAMPLE = "/certs/enduser-certs/user1.crt";
|
||||||
|
|
||||||
|
private static BulletinBoardSignature[] signers;
|
||||||
|
private static ByteString[] signerIDs;
|
||||||
|
|
||||||
|
private Semaphore semaphore;
|
||||||
|
private List<Throwable> thrown;
|
||||||
|
|
||||||
|
@BeforeClass
|
||||||
|
public static void build() {
|
||||||
|
|
||||||
|
messageGenerator = new BulletinBoardMessageGenerator(new Random(0));
|
||||||
|
messageComparator = new BulletinBoardMessageComparator();
|
||||||
|
|
||||||
|
signers = new BulletinBoardSignature[1];
|
||||||
|
signerIDs = new ByteString[1];
|
||||||
|
|
||||||
|
signers[0] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
signerIDs[0] = signers[0].getSignerID();
|
||||||
|
|
||||||
|
InputStream keyStream = BulletinBoardSynchronizerTest.class.getResourceAsStream(KEYFILE_EXAMPLE);
|
||||||
|
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
KeyStore.Builder keyStoreBuilder = signers[0].getPKCS12KeyStoreBuilder(keyStream, password);
|
||||||
|
|
||||||
|
signers[0].loadSigningCertificate(keyStoreBuilder);
|
||||||
|
|
||||||
|
signers[0].loadVerificationCertificates(BulletinBoardSynchronizerTest.class.getResourceAsStream(CERT1_PEM_EXAMPLE));
|
||||||
|
|
||||||
|
} catch (IOException e) {
|
||||||
|
System.err.println("Failed reading from signature file " + e.getMessage());
|
||||||
|
fail("Failed reading from signature file " + e.getMessage());
|
||||||
|
} catch (CertificateException e) {
|
||||||
|
System.err.println("Failed reading certificate " + e.getMessage());
|
||||||
|
fail("Failed reading certificate " + e.getMessage());
|
||||||
|
} catch (KeyStoreException e) {
|
||||||
|
System.err.println("Failed reading keystore " + e.getMessage());
|
||||||
|
fail("Failed reading keystore " + e.getMessage());
|
||||||
|
} catch (NoSuchAlgorithmException e) {
|
||||||
|
System.err.println("Couldn't find signing algorithm " + e.getMessage());
|
||||||
|
fail("Couldn't find signing algorithm " + e.getMessage());
|
||||||
|
} catch (UnrecoverableKeyException e) {
|
||||||
|
System.err.println("Couldn't find signing key " + e.getMessage());
|
||||||
|
fail("Couldn't find signing key " + e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
signerIDs[0] = signers[0].getSignerID();
|
||||||
|
|
||||||
|
testCount = 0;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void init() throws CommunicationException {
|
||||||
|
|
||||||
|
DeletableBulletinBoardServer remoteServer = new BulletinBoardSQLServer(new H2QueryProvider(REMOTE_SERVER_ADDRESS + testCount));
|
||||||
|
remoteServer.init();
|
||||||
|
|
||||||
|
remoteClient = new LocalBulletinBoardClient(
|
||||||
|
remoteServer,
|
||||||
|
THREAD_NUM,
|
||||||
|
SUBSCRIPTION_INTERVAL);
|
||||||
|
|
||||||
|
DeletableBulletinBoardServer localServer = new BulletinBoardSQLServer(new H2QueryProvider(LOCAL_SERVER_ADDRESS + testCount));
|
||||||
|
localServer.init();
|
||||||
|
|
||||||
|
localClient = new LocalBulletinBoardClient(
|
||||||
|
localServer,
|
||||||
|
THREAD_NUM,
|
||||||
|
SUBSCRIPTION_INTERVAL);
|
||||||
|
|
||||||
|
synchronizer = new SimpleBulletinBoardSynchronizer(SYNC_SLEEP_INTERVAL, SYNC_WAIT_CAP);
|
||||||
|
synchronizer.init(localClient, remoteClient);
|
||||||
|
|
||||||
|
semaphore = new Semaphore(0);
|
||||||
|
thrown = new LinkedList<>();
|
||||||
|
|
||||||
|
testCount++;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private class SyncStatusCallback implements FutureCallback<SyncStatus> {
|
||||||
|
|
||||||
|
private final SyncStatus statusToWaitFor;
|
||||||
|
private AtomicBoolean stillWaiting;
|
||||||
|
|
||||||
|
public SyncStatusCallback(SyncStatus statusToWaitFor) {
|
||||||
|
this.statusToWaitFor = statusToWaitFor;
|
||||||
|
stillWaiting = new AtomicBoolean(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(SyncStatus result) {
|
||||||
|
|
||||||
|
if (result == statusToWaitFor && stillWaiting.compareAndSet(true, false)){
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
thrown.add(t);
|
||||||
|
if (stillWaiting.compareAndSet(true,false)) {
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private class MessageCountCallback implements FutureCallback<Integer> {
|
||||||
|
|
||||||
|
private int[] expectedCounts;
|
||||||
|
private int currentIteration;
|
||||||
|
|
||||||
|
public MessageCountCallback(int[] expectedCounts) {
|
||||||
|
this.expectedCounts = expectedCounts;
|
||||||
|
this.currentIteration = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Integer result) {
|
||||||
|
|
||||||
|
if (currentIteration < expectedCounts.length){
|
||||||
|
if (result != expectedCounts[currentIteration]){
|
||||||
|
onFailure(new AssertionError("Wrong message count. Expected " + expectedCounts[currentIteration] + " but received " + result));
|
||||||
|
currentIteration = expectedCounts.length;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
currentIteration++;
|
||||||
|
|
||||||
|
if (currentIteration == expectedCounts.length)
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
thrown.add(t);
|
||||||
|
semaphore.release();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSync() throws SignatureException, CommunicationException, InterruptedException {
|
||||||
|
|
||||||
|
Timestamp timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(15252162)
|
||||||
|
.setNanos(85914)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
BulletinBoardMessage msg = messageGenerator.generateRandomMessage(signers, timestamp, 10, 10);
|
||||||
|
|
||||||
|
MessageID msgID = localClient.postMessage(msg);
|
||||||
|
|
||||||
|
timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(51511653)
|
||||||
|
.setNanos(3625)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
BulletinBoardMessage batchMessage = messageGenerator.generateRandomMessage(signers,timestamp, 100, 10);
|
||||||
|
|
||||||
|
MessageID batchMsgID = localClient.postAsBatch(batchMessage, 10);
|
||||||
|
|
||||||
|
BulletinBoardMessage test = localClient.readMessage(batchMsgID);
|
||||||
|
|
||||||
|
BulletinBoardMessage stub = localClient.readMessages(MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(batchMsgID.getID())
|
||||||
|
.build())
|
||||||
|
.build()).get(0);
|
||||||
|
|
||||||
|
BulletinBoardMessage test2 = localClient.readBatchData(stub);
|
||||||
|
|
||||||
|
synchronizer.subscribeToSyncStatus(new SyncStatusCallback(SyncStatus.SYNCHRONIZED));
|
||||||
|
|
||||||
|
int[] expectedCounts = {2,0};
|
||||||
|
synchronizer.subscribeToRemainingMessagesCount(new MessageCountCallback(expectedCounts));
|
||||||
|
|
||||||
|
Thread syncThread = new Thread(synchronizer);
|
||||||
|
syncThread.start();
|
||||||
|
|
||||||
|
if (!semaphore.tryAcquire(2, 4000, TimeUnit.MILLISECONDS)) {
|
||||||
|
thrown.add(new TimeoutException("Timeout occurred while waiting for synchronizer to sync."));
|
||||||
|
}
|
||||||
|
|
||||||
|
synchronizer.stop();
|
||||||
|
syncThread.join();
|
||||||
|
|
||||||
|
if (thrown.size() > 0) {
|
||||||
|
for (Throwable t : thrown)
|
||||||
|
System.err.println(t.getMessage());
|
||||||
|
assertThat("Exception thrown by Synchronizer: " + thrown.get(0).getMessage(), false);
|
||||||
|
}
|
||||||
|
|
||||||
|
List<BulletinBoardMessage> msgList = remoteClient.readMessages(MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgID.getID())
|
||||||
|
.build())
|
||||||
|
.build());
|
||||||
|
|
||||||
|
assertThat("Wrong number of messages returned.", msgList.size() == 1);
|
||||||
|
assertThat("Returned message is not equal to original one", messageComparator.compare(msgList.get(0),msg) == 0);
|
||||||
|
|
||||||
|
BulletinBoardMessage returnedBatchMsg = remoteClient.readMessage(batchMsgID);
|
||||||
|
|
||||||
|
assertThat("Returned batch does not equal original one.", messageComparator.compare(returnedBatchMsg, batchMessage) == 0);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testServerError() throws SignatureException, CommunicationException, InterruptedException {
|
||||||
|
|
||||||
|
Timestamp timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(945736256)
|
||||||
|
.setNanos(276788)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
BulletinBoardMessage msg = messageGenerator.generateRandomMessage(signers, timestamp, 10, 10);
|
||||||
|
|
||||||
|
remoteClient.close();
|
||||||
|
|
||||||
|
synchronizer.subscribeToSyncStatus(new SyncStatusCallback(SyncStatus.SERVER_ERROR));
|
||||||
|
|
||||||
|
localClient.postMessage(msg);
|
||||||
|
|
||||||
|
Thread thread = new Thread(synchronizer);
|
||||||
|
|
||||||
|
thread.start();
|
||||||
|
|
||||||
|
if (!semaphore.tryAcquire(4000, TimeUnit.MILLISECONDS)) {
|
||||||
|
thrown.add(new TimeoutException("Timeout occurred while waiting for synchronizer to sync."));
|
||||||
|
}
|
||||||
|
|
||||||
|
synchronizer.stop();
|
||||||
|
thread.join();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void close() {
|
||||||
|
|
||||||
|
if (thrown.size() > 0) {
|
||||||
|
for (Throwable t : thrown) {
|
||||||
|
System.err.println(t.getMessage());
|
||||||
|
}
|
||||||
|
assertThat("Exception thrown by Synchronizer: " + thrown.get(0).getMessage(), false);
|
||||||
|
}
|
||||||
|
|
||||||
|
synchronizer.stop();
|
||||||
|
localClient.close();
|
||||||
|
remoteClient.close();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,111 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer;
|
||||||
|
import meerkat.bulletinboard.sqlserver.H2QueryProvider;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.security.SignatureException;
|
||||||
|
import java.util.LinkedList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel on 6/27/2016.
|
||||||
|
*/
|
||||||
|
public class CachedBulletinBoardClientTest {
|
||||||
|
|
||||||
|
private static final int THREAD_NUM = 3;
|
||||||
|
|
||||||
|
private static final String LOCAL_DB_NAME = "localDB";
|
||||||
|
private static final String REMOTE_DB_NAME = "remoteDB";
|
||||||
|
private static final String QUEUE_DB_NAME = "queueDB";
|
||||||
|
|
||||||
|
private static final int SUBSRCIPTION_DELAY = 500;
|
||||||
|
private static final int SYNC_DELAY = 500;
|
||||||
|
|
||||||
|
// Testers
|
||||||
|
private CachedBulletinBoardClient cachedClient;
|
||||||
|
private GenericBulletinBoardClientTester clientTest;
|
||||||
|
private GenericSubscriptionClientTester subscriptionTester;
|
||||||
|
|
||||||
|
public CachedBulletinBoardClientTest() throws CommunicationException {
|
||||||
|
|
||||||
|
DeletableBulletinBoardServer localServer = new BulletinBoardSQLServer(new H2QueryProvider(LOCAL_DB_NAME));
|
||||||
|
localServer.init();
|
||||||
|
LocalBulletinBoardClient localClient = new LocalBulletinBoardClient(localServer, THREAD_NUM, SUBSRCIPTION_DELAY);
|
||||||
|
|
||||||
|
DeletableBulletinBoardServer remoteServer = new BulletinBoardSQLServer(new H2QueryProvider(REMOTE_DB_NAME));
|
||||||
|
remoteServer.init();
|
||||||
|
LocalBulletinBoardClient remoteClient = new LocalBulletinBoardClient(remoteServer, THREAD_NUM, SUBSRCIPTION_DELAY);
|
||||||
|
|
||||||
|
DeletableBulletinBoardServer queueServer = new BulletinBoardSQLServer(new H2QueryProvider(QUEUE_DB_NAME));
|
||||||
|
queueServer.init();
|
||||||
|
LocalBulletinBoardClient queueClient = new LocalBulletinBoardClient(queueServer, THREAD_NUM, SUBSRCIPTION_DELAY);
|
||||||
|
|
||||||
|
List<SubscriptionBulletinBoardClient> clientList = new LinkedList<>();
|
||||||
|
clientList.add(remoteClient);
|
||||||
|
|
||||||
|
BulletinBoardSubscriber subscriber = new ThreadedBulletinBoardSubscriber(clientList, localClient);
|
||||||
|
|
||||||
|
cachedClient = new CachedBulletinBoardClient(localClient, remoteClient, subscriber, queueClient, SYNC_DELAY, SYNC_DELAY);
|
||||||
|
subscriptionTester = new GenericSubscriptionClientTester(cachedClient);
|
||||||
|
clientTest = new GenericBulletinBoardClientTester(cachedClient, 87351);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test methods
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Takes care of initializing the client and the test resources
|
||||||
|
*/
|
||||||
|
@Before
|
||||||
|
public void init(){
|
||||||
|
|
||||||
|
clientTest.init();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Closes the client and makes sure the test fails when an exception occurred in a separate thread
|
||||||
|
*/
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void close() {
|
||||||
|
|
||||||
|
cachedClient.close();
|
||||||
|
clientTest.close();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testPost() {
|
||||||
|
|
||||||
|
clientTest.testPost();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
|
clientTest.testBatchPost();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCompleteBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
|
clientTest.testCompleteBatchPost();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscription() throws SignatureException, CommunicationException {
|
||||||
|
|
||||||
|
// subscriptionTester.init();
|
||||||
|
// subscriptionTester.subscriptionTest();
|
||||||
|
// subscriptionTester.close();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -5,10 +5,13 @@ import com.google.protobuf.ByteString;
|
||||||
import com.google.protobuf.Timestamp;
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.crypto.concrete.ECDSASignature;
|
import meerkat.crypto.concrete.ECDSASignature;
|
||||||
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Crypto;
|
import meerkat.protobuf.Crypto;
|
||||||
import meerkat.util.BulletinBoardMessageComparator;
|
import meerkat.util.BulletinBoardMessageComparator;
|
||||||
import meerkat.util.BulletinBoardMessageGenerator;
|
import meerkat.util.BulletinBoardMessageGenerator;
|
||||||
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
import meerkat.bulletinboard.AsyncBulletinBoardClient.BatchIdentifier;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
|
@ -28,7 +31,7 @@ public class GenericBulletinBoardClientTester {
|
||||||
|
|
||||||
// Signature resources
|
// Signature resources
|
||||||
|
|
||||||
private GenericBatchDigitalSignature signers[];
|
private BulletinBoardSignature signers[];
|
||||||
private ByteString[] signerIDs;
|
private ByteString[] signerIDs;
|
||||||
|
|
||||||
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
||||||
|
@ -45,28 +48,29 @@ public class GenericBulletinBoardClientTester {
|
||||||
private AsyncBulletinBoardClient bulletinBoardClient;
|
private AsyncBulletinBoardClient bulletinBoardClient;
|
||||||
|
|
||||||
private PostCallback postCallback;
|
private PostCallback postCallback;
|
||||||
private PostCallback failPostCallback = new PostCallback(true,false);
|
|
||||||
|
|
||||||
private RedundancyCallback redundancyCallback;
|
private RedundancyCallback redundancyCallback;
|
||||||
private ReadCallback readCallback;
|
private ReadCallback readCallback;
|
||||||
private ReadBatchCallback readBatchCallback;
|
|
||||||
|
|
||||||
// Sync and misc
|
// Sync and misc
|
||||||
|
|
||||||
private Semaphore jobSemaphore;
|
private Semaphore jobSemaphore;
|
||||||
private Vector<Throwable> thrown;
|
private Vector<Throwable> thrown;
|
||||||
private Random random;
|
private Random random;
|
||||||
|
private BulletinBoardMessageGenerator generator;
|
||||||
|
|
||||||
|
private BulletinBoardDigest digest;
|
||||||
|
|
||||||
// Constructor
|
// Constructor
|
||||||
|
|
||||||
public GenericBulletinBoardClientTester(AsyncBulletinBoardClient bulletinBoardClient){
|
public GenericBulletinBoardClientTester(AsyncBulletinBoardClient bulletinBoardClient, int seed){
|
||||||
|
|
||||||
this.bulletinBoardClient = bulletinBoardClient;
|
this.bulletinBoardClient = bulletinBoardClient;
|
||||||
|
|
||||||
signers = new GenericBatchDigitalSignature[2];
|
signers = new GenericBulletinBoardSignature[2];
|
||||||
signerIDs = new ByteString[signers.length];
|
signerIDs = new ByteString[signers.length];
|
||||||
signers[0] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[0] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
signers[1] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[1] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
|
||||||
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
||||||
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
||||||
|
@ -108,6 +112,10 @@ public class GenericBulletinBoardClientTester {
|
||||||
fail("Couldn't find signing key " + e.getMessage());
|
fail("Couldn't find signing key " + e.getMessage());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
this.random = new Random(seed);
|
||||||
|
this.generator = new BulletinBoardMessageGenerator(random);
|
||||||
|
this.digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Callback definitions
|
// Callback definitions
|
||||||
|
@ -138,16 +146,21 @@ public class GenericBulletinBoardClientTester {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(Boolean msg) {
|
public void onSuccess(Boolean msg) {
|
||||||
|
|
||||||
System.err.println("Post operation completed");
|
System.err.println("Post operation completed");
|
||||||
jobSemaphore.release();
|
|
||||||
//TODO: Change Assert mechanism to exception one
|
|
||||||
if (isAssert) {
|
if (isAssert) {
|
||||||
if (assertValue) {
|
if (assertValue && !msg) {
|
||||||
assertThat("Post operation failed", msg, is(Boolean.TRUE));
|
genericHandleFailure(new AssertionError("Post operation failed"));
|
||||||
|
} else if (!assertValue && msg){
|
||||||
|
genericHandleFailure(new AssertionError("Post operation succeeded unexpectedly"));
|
||||||
} else {
|
} else {
|
||||||
assertThat("Post operation succeeded unexpectedly", msg, is(Boolean.FALSE));
|
jobSemaphore.release();
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
jobSemaphore.release();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -210,21 +223,24 @@ public class GenericBulletinBoardClientTester {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private class ReadBatchCallback implements FutureCallback<CompleteBatch> {
|
private class ReadBatchCallback implements FutureCallback<BulletinBoardMessage>{
|
||||||
|
|
||||||
private CompleteBatch expectedBatch;
|
private BulletinBoardMessage expectedMsg;
|
||||||
|
|
||||||
public ReadBatchCallback(CompleteBatch expectedBatch) {
|
public ReadBatchCallback(BulletinBoardMessage expectedMsg) {
|
||||||
this.expectedBatch = expectedBatch;
|
this.expectedMsg = expectedMsg;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void onSuccess(CompleteBatch batch) {
|
public void onSuccess(BulletinBoardMessage msg) {
|
||||||
|
|
||||||
System.err.println(batch);
|
BulletinBoardMessageComparator msgComparator = new BulletinBoardMessageComparator();
|
||||||
|
|
||||||
|
if (msgComparator.compare(msg, expectedMsg) != 0) {
|
||||||
|
genericHandleFailure(new AssertionError("Batch read returned different message.\nExpected:" + expectedMsg + "\nRecieved:" + msg + "\n"));
|
||||||
|
} else {
|
||||||
jobSemaphore.release();
|
jobSemaphore.release();
|
||||||
|
}
|
||||||
assertThat("Batch returned is incorrect", batch, is(equalTo(expectedBatch)));
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -234,59 +250,6 @@ public class GenericBulletinBoardClientTester {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Randomness generators
|
|
||||||
|
|
||||||
private byte randomByte(){
|
|
||||||
return (byte) random.nextInt();
|
|
||||||
}
|
|
||||||
|
|
||||||
private byte[] randomByteArray(int length) {
|
|
||||||
|
|
||||||
byte[] randomBytes = new byte[length];
|
|
||||||
|
|
||||||
for (int i = 0; i < length ; i++){
|
|
||||||
randomBytes[i] = randomByte();
|
|
||||||
}
|
|
||||||
|
|
||||||
return randomBytes;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
private CompleteBatch createRandomBatch(int signer, int batchId, int length) throws SignatureException {
|
|
||||||
|
|
||||||
CompleteBatch completeBatch = new CompleteBatch();
|
|
||||||
|
|
||||||
// Create data
|
|
||||||
|
|
||||||
completeBatch.setBeginBatchMessage(BeginBatchMessage.newBuilder()
|
|
||||||
.setSignerId(signerIDs[signer])
|
|
||||||
.setBatchId(batchId)
|
|
||||||
.addTag("Test")
|
|
||||||
.build());
|
|
||||||
|
|
||||||
for (int i = 0 ; i < length ; i++){
|
|
||||||
|
|
||||||
BatchData batchData = BatchData.newBuilder()
|
|
||||||
.setData(ByteString.copyFrom(randomByteArray(i)))
|
|
||||||
.build();
|
|
||||||
|
|
||||||
completeBatch.appendBatchData(batchData);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
completeBatch.setTimestamp(Timestamp.newBuilder()
|
|
||||||
.setSeconds(Math.abs(90))
|
|
||||||
.setNanos(50)
|
|
||||||
.build());
|
|
||||||
|
|
||||||
signers[signer].updateContent(completeBatch);
|
|
||||||
|
|
||||||
completeBatch.setSignature(signers[signer].sign());
|
|
||||||
|
|
||||||
return completeBatch;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Test methods
|
// Test methods
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -311,7 +274,13 @@ public class GenericBulletinBoardClientTester {
|
||||||
public void close() {
|
public void close() {
|
||||||
|
|
||||||
if (thrown.size() > 0) {
|
if (thrown.size() > 0) {
|
||||||
|
|
||||||
|
for (Throwable t : thrown){
|
||||||
|
System.err.println(t.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
assert false;
|
assert false;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -319,7 +288,7 @@ public class GenericBulletinBoardClientTester {
|
||||||
/**
|
/**
|
||||||
* Tests the standard post, redundancy and read methods
|
* Tests the standard post, redundancy and read methods
|
||||||
*/
|
*/
|
||||||
public void postTest() {
|
public void testPost() {
|
||||||
|
|
||||||
byte[] b1 = {(byte) 1, (byte) 2, (byte) 3, (byte) 4};
|
byte[] b1 = {(byte) 1, (byte) 2, (byte) 3, (byte) 4};
|
||||||
byte[] b2 = {(byte) 11, (byte) 12, (byte) 13, (byte) 14};
|
byte[] b2 = {(byte) 11, (byte) 12, (byte) 13, (byte) 14};
|
||||||
|
@ -395,59 +364,69 @@ public class GenericBulletinBoardClientTester {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Tests posting a batch by parts
|
* Tests posting a batch by parts
|
||||||
* Also tests not being able to post to a closed batch
|
|
||||||
* @throws CommunicationException, SignatureException, InterruptedException
|
* @throws CommunicationException, SignatureException, InterruptedException
|
||||||
*/
|
*/
|
||||||
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
final int SIGNER = 1;
|
|
||||||
final int BATCH_ID = 100;
|
|
||||||
final int BATCH_LENGTH = 100;
|
final int BATCH_LENGTH = 100;
|
||||||
|
final int CHUNK_SIZE = 10;
|
||||||
|
final int TAG_NUM = 10;
|
||||||
|
|
||||||
CompleteBatch completeBatch = createRandomBatch(SIGNER, BATCH_ID, BATCH_LENGTH);
|
final Timestamp timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(141515)
|
||||||
|
.setNanos(859018)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
final BulletinBoardMessage msg = generator.generateRandomMessage(signers, timestamp, BATCH_LENGTH, TAG_NUM);
|
||||||
|
|
||||||
// Begin batch
|
// Begin batch
|
||||||
|
|
||||||
bulletinBoardClient.beginBatch(completeBatch.getBeginBatchMessage(), postCallback);
|
bulletinBoardClient.beginBatch(msg.getMsg().getTagList(), new FutureCallback<BatchIdentifier>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(final BatchIdentifier identifier) {
|
||||||
|
|
||||||
|
bulletinBoardClient.postBatchData(identifier, BulletinBoardUtils.breakToBatch(msg, CHUNK_SIZE), new FutureCallback<Boolean>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
|
||||||
|
bulletinBoardClient.closeBatch(identifier, msg.getMsg().getTimestamp(), msg.getSigList(), new FutureCallback<Boolean>() {
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Boolean result) {
|
||||||
|
jobSemaphore.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
genericHandleFailure(t);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
genericHandleFailure(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
genericHandleFailure(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
jobSemaphore.acquire();
|
||||||
|
|
||||||
// Post data
|
digest.reset();
|
||||||
|
digest.update(msg);
|
||||||
|
|
||||||
bulletinBoardClient.postBatchData(signerIDs[SIGNER], BATCH_ID, completeBatch.getBatchDataList(), postCallback);
|
bulletinBoardClient.readMessage(digest.digestAsMessageID(), new ReadBatchCallback(msg));
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
|
||||||
|
|
||||||
// Close batch
|
|
||||||
|
|
||||||
CloseBatchMessage closeBatchMessage = completeBatch.getCloseBatchMessage();
|
|
||||||
|
|
||||||
bulletinBoardClient.closeBatch(closeBatchMessage, postCallback);
|
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
|
||||||
|
|
||||||
// Attempt to open batch again
|
|
||||||
|
|
||||||
bulletinBoardClient.beginBatch(completeBatch.getBeginBatchMessage(), failPostCallback);
|
|
||||||
|
|
||||||
// Attempt to add batch data
|
|
||||||
|
|
||||||
bulletinBoardClient.postBatchData(signerIDs[SIGNER], BATCH_ID, completeBatch.getBatchDataList(), failPostCallback);
|
|
||||||
|
|
||||||
jobSemaphore.acquire(2);
|
|
||||||
|
|
||||||
// Read batch data
|
|
||||||
|
|
||||||
BatchSpecificationMessage batchSpecificationMessage =
|
|
||||||
BatchSpecificationMessage.newBuilder()
|
|
||||||
.setSignerId(signerIDs[SIGNER])
|
|
||||||
.setBatchId(BATCH_ID)
|
|
||||||
.setStartPosition(0)
|
|
||||||
.build();
|
|
||||||
|
|
||||||
readBatchCallback = new ReadBatchCallback(completeBatch);
|
|
||||||
|
|
||||||
bulletinBoardClient.readBatch(batchSpecificationMessage, readBatchCallback);
|
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
jobSemaphore.acquire();
|
||||||
|
|
||||||
|
@ -455,62 +434,61 @@ public class GenericBulletinBoardClientTester {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Posts a complete batch message
|
* Posts a complete batch message
|
||||||
* Checks reading of the message
|
* Checks reading of the message in two parts
|
||||||
* @throws CommunicationException, SignatureException, InterruptedException
|
* @throws CommunicationException, SignatureException, InterruptedException
|
||||||
*/
|
*/
|
||||||
public void testCompleteBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
public void testCompleteBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
final int SIGNER = 0;
|
final int BATCH_LENGTH = 100;
|
||||||
final int BATCH_ID = 101;
|
final int CHUNK_SIZE = 99;
|
||||||
final int BATCH_LENGTH = 50;
|
final int TAG_NUM = 8;
|
||||||
|
|
||||||
|
final Timestamp timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(7776151)
|
||||||
|
.setNanos(252616)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
final BulletinBoardMessage msg = generator.generateRandomMessage(signers, timestamp, BATCH_LENGTH, TAG_NUM);
|
||||||
|
|
||||||
// Post batch
|
// Post batch
|
||||||
|
|
||||||
CompleteBatch completeBatch = createRandomBatch(SIGNER, BATCH_ID, BATCH_LENGTH);
|
MessageID msgID = bulletinBoardClient.postAsBatch(msg, CHUNK_SIZE, postCallback);
|
||||||
|
|
||||||
bulletinBoardClient.postBatch(completeBatch,postCallback);
|
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
jobSemaphore.acquire();
|
||||||
|
|
||||||
// Read batch
|
// Read batch
|
||||||
|
|
||||||
BatchSpecificationMessage batchSpecificationMessage =
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
BatchSpecificationMessage.newBuilder()
|
.addFilter(MessageFilter.newBuilder()
|
||||||
.setSignerId(signerIDs[SIGNER])
|
.setType(FilterType.MSG_ID)
|
||||||
.setBatchId(BATCH_ID)
|
.setId(msgID.getID())
|
||||||
.setStartPosition(0)
|
|
||||||
.build();
|
|
||||||
|
|
||||||
readBatchCallback = new ReadBatchCallback(completeBatch);
|
|
||||||
|
|
||||||
bulletinBoardClient.readBatch(batchSpecificationMessage, readBatchCallback);
|
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Tests that an unopened batch cannot be closed
|
|
||||||
* @throws CommunicationException, InterruptedException
|
|
||||||
*/
|
|
||||||
public void testInvalidBatchClose() throws CommunicationException, InterruptedException {
|
|
||||||
|
|
||||||
final int NON_EXISTENT_BATCH_ID = 999;
|
|
||||||
|
|
||||||
CloseBatchMessage closeBatchMessage =
|
|
||||||
CloseBatchMessage.newBuilder()
|
|
||||||
.setBatchId(NON_EXISTENT_BATCH_ID)
|
|
||||||
.setBatchLength(1)
|
|
||||||
.setSig(Crypto.Signature.getDefaultInstance())
|
|
||||||
.setTimestamp(Timestamp.newBuilder()
|
|
||||||
.setSeconds(9)
|
|
||||||
.setNanos(12)
|
|
||||||
.build())
|
.build())
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
// Try to stop the (unopened) batch;
|
bulletinBoardClient.readMessages(filterList, new FutureCallback<List<BulletinBoardMessage>>() {
|
||||||
|
|
||||||
bulletinBoardClient.closeBatch(closeBatchMessage, failPostCallback);
|
@Override
|
||||||
|
public void onSuccess(List<BulletinBoardMessage> msgList) {
|
||||||
|
|
||||||
|
if (msgList.size() != 1) {
|
||||||
|
|
||||||
|
genericHandleFailure(new AssertionError("Wrong number of stubs returned. Expected: 1; Found: " + msgList.size()));
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
BulletinBoardMessage retrievedMsg = msgList.get(0);
|
||||||
|
bulletinBoardClient.readBatchData(retrievedMsg, new ReadBatchCallback(msg));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable t) {
|
||||||
|
genericHandleFailure(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
jobSemaphore.acquire();
|
jobSemaphore.acquire();
|
||||||
|
|
||||||
|
|
|
@ -16,9 +16,6 @@ import java.security.cert.CertificateException;
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
import java.util.concurrent.Semaphore;
|
import java.util.concurrent.Semaphore;
|
||||||
|
|
||||||
import static org.hamcrest.CoreMatchers.is;
|
|
||||||
import static org.hamcrest.CoreMatchers.startsWith;
|
|
||||||
import static org.junit.Assert.assertThat;
|
|
||||||
import static org.junit.Assert.fail;
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -26,7 +23,7 @@ import static org.junit.Assert.fail;
|
||||||
*/
|
*/
|
||||||
public class GenericSubscriptionClientTester {
|
public class GenericSubscriptionClientTester {
|
||||||
|
|
||||||
private GenericBatchDigitalSignature signers[];
|
private BulletinBoardSignature signers[];
|
||||||
private ByteString[] signerIDs;
|
private ByteString[] signerIDs;
|
||||||
|
|
||||||
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
||||||
|
@ -38,7 +35,7 @@ public class GenericSubscriptionClientTester {
|
||||||
private static String CERT1_PEM_EXAMPLE = "/certs/enduser-certs/user1.crt";
|
private static String CERT1_PEM_EXAMPLE = "/certs/enduser-certs/user1.crt";
|
||||||
private static String CERT3_PEM_EXAMPLE = "/certs/enduser-certs/user3.crt";
|
private static String CERT3_PEM_EXAMPLE = "/certs/enduser-certs/user3.crt";
|
||||||
|
|
||||||
private SubscriptionAsyncBulletinBoardClient bulletinBoardClient;
|
private SubscriptionBulletinBoardClient bulletinBoardClient;
|
||||||
|
|
||||||
private Random random;
|
private Random random;
|
||||||
private BulletinBoardMessageGenerator generator;
|
private BulletinBoardMessageGenerator generator;
|
||||||
|
@ -46,14 +43,14 @@ public class GenericSubscriptionClientTester {
|
||||||
private Semaphore jobSemaphore;
|
private Semaphore jobSemaphore;
|
||||||
private Vector<Throwable> thrown;
|
private Vector<Throwable> thrown;
|
||||||
|
|
||||||
public GenericSubscriptionClientTester(SubscriptionAsyncBulletinBoardClient bulletinBoardClient){
|
public GenericSubscriptionClientTester(SubscriptionBulletinBoardClient bulletinBoardClient){
|
||||||
|
|
||||||
this.bulletinBoardClient = bulletinBoardClient;
|
this.bulletinBoardClient = bulletinBoardClient;
|
||||||
|
|
||||||
signers = new GenericBatchDigitalSignature[2];
|
signers = new BulletinBoardSignature[2];
|
||||||
signerIDs = new ByteString[signers.length];
|
signerIDs = new ByteString[signers.length];
|
||||||
signers[0] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[0] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
signers[1] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[1] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
|
||||||
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
||||||
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
||||||
|
@ -181,15 +178,13 @@ public class GenericSubscriptionClientTester {
|
||||||
public void onFailure(Throwable t) {
|
public void onFailure(Throwable t) {
|
||||||
System.err.println(t.getCause() + " " + t.getMessage());
|
System.err.println(t.getCause() + " " + t.getMessage());
|
||||||
thrown.add(t);
|
thrown.add(t);
|
||||||
jobSemaphore.release(expectedMessages.size());
|
jobSemaphore.release();
|
||||||
stage = expectedMessages.size();
|
stage = expectedMessages.size();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void subscriptionTest() throws SignatureException, CommunicationException {
|
public void subscriptionTest() throws SignatureException, CommunicationException {
|
||||||
|
|
||||||
final int FIRST_POST_ID = 201;
|
|
||||||
final int SECOND_POST_ID = 202;
|
|
||||||
final String COMMON_TAG = "SUBSCRIPTION_TEST";
|
final String COMMON_TAG = "SUBSCRIPTION_TEST";
|
||||||
|
|
||||||
List<String> tags = new LinkedList<>();
|
List<String> tags = new LinkedList<>();
|
||||||
|
@ -207,9 +202,9 @@ public class GenericSubscriptionClientTester {
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
List<List<BulletinBoardMessage>> expectedMessages = new ArrayList<>(3);
|
List<List<BulletinBoardMessage>> expectedMessages = new ArrayList<>(3);
|
||||||
expectedMessages.add(new LinkedList<BulletinBoardMessage>());
|
expectedMessages.add(new LinkedList<>());
|
||||||
expectedMessages.add(new LinkedList<BulletinBoardMessage>());
|
expectedMessages.add(new LinkedList<>());
|
||||||
expectedMessages.add(new LinkedList<BulletinBoardMessage>());
|
expectedMessages.add(new LinkedList<>());
|
||||||
expectedMessages.get(0).add(msg1);
|
expectedMessages.get(0).add(msg1);
|
||||||
expectedMessages.get(2).add(msg3);
|
expectedMessages.get(2).add(msg3);
|
||||||
|
|
||||||
|
|
|
@ -7,12 +7,6 @@ import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import java.security.SignatureException;
|
import java.security.SignatureException;
|
||||||
import java.sql.Connection;
|
|
||||||
import java.sql.SQLException;
|
|
||||||
import java.sql.Statement;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import static org.junit.Assert.fail;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 05-Dec-15.
|
* Created by Arbel Deutsch Peled on 05-Dec-15.
|
||||||
|
@ -30,30 +24,14 @@ public class LocalBulletinBoardClientTest {
|
||||||
|
|
||||||
public LocalBulletinBoardClientTest() throws CommunicationException {
|
public LocalBulletinBoardClientTest() throws CommunicationException {
|
||||||
|
|
||||||
H2QueryProvider queryProvider = new H2QueryProvider(DB_NAME) ;
|
H2QueryProvider queryProvider = new H2QueryProvider(DB_NAME);
|
||||||
|
|
||||||
try {
|
DeletableBulletinBoardServer server = new BulletinBoardSQLServer(queryProvider);
|
||||||
|
server.init();
|
||||||
Connection conn = queryProvider.getDataSource().getConnection();
|
|
||||||
Statement stmt = conn.createStatement();
|
|
||||||
|
|
||||||
List<String> deletionQueries = queryProvider.getSchemaDeletionCommands();
|
|
||||||
|
|
||||||
for (String deletionQuery : deletionQueries) {
|
|
||||||
stmt.execute(deletionQuery);
|
|
||||||
}
|
|
||||||
|
|
||||||
} catch (SQLException e) {
|
|
||||||
System.err.println(e.getMessage());
|
|
||||||
throw new CommunicationException(e.getCause() + " " + e.getMessage());
|
|
||||||
}
|
|
||||||
|
|
||||||
BulletinBoardServer server = new BulletinBoardSQLServer(queryProvider);
|
|
||||||
server.init(DB_NAME);
|
|
||||||
|
|
||||||
LocalBulletinBoardClient client = new LocalBulletinBoardClient(server, THREAD_NUM, SUBSRCIPTION_DELAY);
|
LocalBulletinBoardClient client = new LocalBulletinBoardClient(server, THREAD_NUM, SUBSRCIPTION_DELAY);
|
||||||
subscriptionTester = new GenericSubscriptionClientTester(client);
|
subscriptionTester = new GenericSubscriptionClientTester(client);
|
||||||
clientTest = new GenericBulletinBoardClientTester(client);
|
clientTest = new GenericBulletinBoardClientTester(client, 98354);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -81,9 +59,9 @@ public class LocalBulletinBoardClientTest {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void postTest() {
|
public void testPost() {
|
||||||
|
|
||||||
clientTest.postTest();
|
clientTest.testPost();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -100,15 +78,9 @@ public class LocalBulletinBoardClientTest {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testInvalidBatchClose() throws CommunicationException, InterruptedException {
|
|
||||||
|
|
||||||
clientTest.testInvalidBatchClose();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testSubscription() throws SignatureException, CommunicationException {
|
public void testSubscription() throws SignatureException, CommunicationException {
|
||||||
|
|
||||||
subscriptionTester.init();
|
subscriptionTester.init();
|
||||||
subscriptionTester.subscriptionTest();
|
subscriptionTester.subscriptionTest();
|
||||||
subscriptionTester.close();
|
subscriptionTester.close();
|
||||||
|
|
|
@ -0,0 +1,104 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.ListeningExecutorService;
|
||||||
|
import com.google.common.util.concurrent.MoreExecutors;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.Voting.BulletinBoardClientParams;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.security.SignatureException;
|
||||||
|
import java.util.LinkedList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.concurrent.Executors;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 05-Dec-15.
|
||||||
|
*/
|
||||||
|
public class SingleServerBulletinBoardClientIntegrationTest {
|
||||||
|
|
||||||
|
// Server data
|
||||||
|
|
||||||
|
private static final String PROP_GETTY_URL = "gretty.httpBaseURI";
|
||||||
|
private static final String DEFAULT_BASE_URL = "http://localhost:8081";
|
||||||
|
private static final String BASE_URL = System.getProperty(PROP_GETTY_URL, DEFAULT_BASE_URL);
|
||||||
|
|
||||||
|
private static final int THREAD_NUM = 3;
|
||||||
|
private static final long FAIL_DELAY = 3000;
|
||||||
|
private static final long SUBSCRIPTION_INTERVAL = 500;
|
||||||
|
|
||||||
|
// Testers
|
||||||
|
private GenericBulletinBoardClientTester clientTest;
|
||||||
|
private GenericSubscriptionClientTester subscriptionTester;
|
||||||
|
|
||||||
|
public SingleServerBulletinBoardClientIntegrationTest(){
|
||||||
|
|
||||||
|
SingleServerBulletinBoardClient client = new SingleServerBulletinBoardClient(THREAD_NUM, FAIL_DELAY, SUBSCRIPTION_INTERVAL);
|
||||||
|
|
||||||
|
List<String> testDB = new LinkedList<>();
|
||||||
|
testDB.add(BASE_URL);
|
||||||
|
|
||||||
|
client.init(BulletinBoardClientParams.newBuilder()
|
||||||
|
.addAllBulletinBoardAddress(testDB)
|
||||||
|
.setMinRedundancy((float) 1.0)
|
||||||
|
.build());
|
||||||
|
|
||||||
|
clientTest = new GenericBulletinBoardClientTester(client, 981541);
|
||||||
|
subscriptionTester = new GenericSubscriptionClientTester(client);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Test methods
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Takes care of initializing the client and the test resources
|
||||||
|
*/
|
||||||
|
@Before
|
||||||
|
public void init(){
|
||||||
|
|
||||||
|
clientTest.init();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Closes the client and makes sure the test fails when an exception occurred in a separate thread
|
||||||
|
*/
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void close() {
|
||||||
|
|
||||||
|
clientTest.close();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testPost() {
|
||||||
|
|
||||||
|
clientTest.testPost();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
|
clientTest.testBatchPost();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCompleteBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
|
clientTest.testCompleteBatchPost();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSubscription() throws SignatureException, CommunicationException {
|
||||||
|
|
||||||
|
subscriptionTester.init();
|
||||||
|
subscriptionTester.subscriptionTest();
|
||||||
|
subscriptionTester.close();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -28,7 +28,7 @@ public class ThreadedBulletinBoardClientIntegrationTest {
|
||||||
|
|
||||||
public ThreadedBulletinBoardClientIntegrationTest(){
|
public ThreadedBulletinBoardClientIntegrationTest(){
|
||||||
|
|
||||||
ThreadedBulletinBoardClient client = new ThreadedBulletinBoardClient();
|
ThreadedBulletinBoardClient client = new ThreadedBulletinBoardClient(3,0,500);
|
||||||
|
|
||||||
List<String> testDB = new LinkedList<>();
|
List<String> testDB = new LinkedList<>();
|
||||||
testDB.add(BASE_URL);
|
testDB.add(BASE_URL);
|
||||||
|
@ -38,7 +38,7 @@ public class ThreadedBulletinBoardClientIntegrationTest {
|
||||||
.setMinRedundancy((float) 1.0)
|
.setMinRedundancy((float) 1.0)
|
||||||
.build());
|
.build());
|
||||||
|
|
||||||
clientTest = new GenericBulletinBoardClientTester(client);
|
clientTest = new GenericBulletinBoardClientTester(client, 52351);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -66,9 +66,9 @@ public class ThreadedBulletinBoardClientIntegrationTest {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void postTest() {
|
public void testPost() {
|
||||||
|
|
||||||
clientTest.postTest();
|
clientTest.testPost();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -76,6 +76,7 @@ public class ThreadedBulletinBoardClientIntegrationTest {
|
||||||
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
public void testBatchPost() throws CommunicationException, SignatureException, InterruptedException {
|
||||||
|
|
||||||
clientTest.testBatchPost();
|
clientTest.testBatchPost();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
@ -85,11 +86,4 @@ public class ThreadedBulletinBoardClientIntegrationTest {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testInvalidBatchClose() throws CommunicationException, InterruptedException {
|
|
||||||
|
|
||||||
clientTest.testInvalidBatchClose();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -259,4 +259,3 @@ publishing {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
package meerkat.bulletinboard.sqlserver;
|
package meerkat.bulletinboard.sqlserver;
|
||||||
|
|
||||||
|
import java.io.ByteArrayInputStream;
|
||||||
|
import java.io.ByteArrayOutputStream;
|
||||||
import java.sql.*;
|
import java.sql.*;
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
|
|
||||||
|
@ -8,16 +10,16 @@ import com.google.protobuf.*;
|
||||||
import com.google.protobuf.Timestamp;
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.bulletinboard.*;
|
import meerkat.bulletinboard.*;
|
||||||
import meerkat.bulletinboard.sqlserver.mappers.*;
|
import meerkat.bulletinboard.sqlserver.mappers.*;
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.*;
|
|
||||||
|
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
|
|
||||||
|
import meerkat.comm.MessageInputStream;
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.crypto.concrete.ECDSASignature;
|
import meerkat.crypto.DigitalSignature;
|
||||||
import meerkat.crypto.concrete.SHA256Digest;
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Comm.*;
|
import meerkat.protobuf.Comm;
|
||||||
import meerkat.protobuf.Crypto.Signature;
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
import meerkat.protobuf.Crypto.SignatureVerificationKey;
|
import meerkat.protobuf.Crypto.SignatureVerificationKey;
|
||||||
|
|
||||||
|
@ -38,7 +40,7 @@ import org.springframework.jdbc.support.KeyHolder;
|
||||||
/**
|
/**
|
||||||
* This is a generic SQL implementation of the BulletinBoardServer API.
|
* This is a generic SQL implementation of the BulletinBoardServer API.
|
||||||
*/
|
*/
|
||||||
public class BulletinBoardSQLServer implements BulletinBoardServer{
|
public class BulletinBoardSQLServer implements DeletableBulletinBoardServer{
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This interface provides the required implementation-specific data to enable an access to an actual SQL server.
|
* This interface provides the required implementation-specific data to enable an access to an actual SQL server.
|
||||||
|
@ -67,6 +69,16 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
new int[] {Types.BLOB, Types.TIMESTAMP, Types.BLOB}
|
new int[] {Types.BLOB, Types.TIMESTAMP, Types.BLOB}
|
||||||
),
|
),
|
||||||
|
|
||||||
|
DELETE_MSG_BY_ENTRY(
|
||||||
|
new String[] {"EntryNum"},
|
||||||
|
new int[] {Types.INTEGER}
|
||||||
|
),
|
||||||
|
|
||||||
|
DELETE_MSG_BY_ID(
|
||||||
|
new String[] {"MsgId"},
|
||||||
|
new int[] {Types.BLOB}
|
||||||
|
),
|
||||||
|
|
||||||
INSERT_NEW_TAG(
|
INSERT_NEW_TAG(
|
||||||
new String[] {"Tag"},
|
new String[] {"Tag"},
|
||||||
new int[] {Types.VARCHAR}
|
new int[] {Types.VARCHAR}
|
||||||
|
@ -107,39 +119,44 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
new int[] {}
|
new int[] {}
|
||||||
),
|
),
|
||||||
|
|
||||||
GET_BATCH_MESSAGE_ENTRY(
|
|
||||||
new String[] {"SignerId", "BatchId"},
|
|
||||||
new int[] {Types.BLOB, Types.INTEGER}
|
|
||||||
),
|
|
||||||
|
|
||||||
CHECK_BATCH_LENGTH(
|
CHECK_BATCH_LENGTH(
|
||||||
new String[] {"SignerId", "BatchId"},
|
new String[] {"BatchId"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER}
|
new int[] {Types.BLOB, Types.INTEGER}
|
||||||
),
|
),
|
||||||
|
|
||||||
GET_BATCH_MESSAGE_DATA(
|
CHECK_BATCH_OPEN(
|
||||||
new String[] {"SignerId", "BatchId", "StartPosition"},
|
new String[] {"BatchId"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER, Types.INTEGER}
|
new int[] {Types.BLOB, Types.INTEGER}
|
||||||
|
),
|
||||||
|
|
||||||
|
GET_BATCH_MESSAGE_DATA_BY_MSG_ID(
|
||||||
|
new String[] {"MsgId", "StartPosition"},
|
||||||
|
new int[] {Types.BLOB, Types.INTEGER}
|
||||||
|
),
|
||||||
|
|
||||||
|
GET_BATCH_MESSAGE_DATA_BY_BATCH_ID(
|
||||||
|
new String[] {"BatchId", "StartPosition"},
|
||||||
|
new int[] {Types.INTEGER, Types.INTEGER}
|
||||||
),
|
),
|
||||||
|
|
||||||
INSERT_BATCH_DATA(
|
INSERT_BATCH_DATA(
|
||||||
new String[] {"SignerId", "BatchId", "SerialNum", "Data"},
|
new String[] {"BatchId", "SerialNum", "Data"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER, Types.INTEGER, Types.BLOB}
|
new int[] {Types.INTEGER, Types.INTEGER, Types.BLOB}
|
||||||
),
|
),
|
||||||
|
|
||||||
CONNECT_BATCH_TAG(
|
STORE_BATCH_TAGS(
|
||||||
new String[] {"SignerId", "BatchId", "Tag"},
|
new String[] {"Tags"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER, Types.VARCHAR}
|
new int[] {Types.BLOB}
|
||||||
),
|
),
|
||||||
|
|
||||||
GET_BATCH_TAGS(
|
GET_BATCH_TAGS(
|
||||||
new String[] {"SignerId", "BatchId"},
|
new String[] {"BatchId"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER}
|
new int[] {Types.BLOB, Types.INTEGER}
|
||||||
),
|
),
|
||||||
|
|
||||||
REMOVE_BATCH_TAGS(
|
ADD_ENTRY_NUM_TO_BATCH(
|
||||||
new String[] {"SignerId", "BatchId"},
|
new String[] {"BatchId", "EntryNum"},
|
||||||
new int[] {Types.BLOB, Types.INTEGER}
|
new int[] {Types.BLOB, Types.INTEGER, Types.INTEGER}
|
||||||
);
|
);
|
||||||
|
|
||||||
private String[] paramNames;
|
private String[] paramNames;
|
||||||
|
@ -317,8 +334,8 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
protected NamedParameterJdbcTemplate jdbcTemplate;
|
protected NamedParameterJdbcTemplate jdbcTemplate;
|
||||||
|
|
||||||
protected BatchDigest digest;
|
protected BulletinBoardDigest digest;
|
||||||
protected BatchDigitalSignature signer;
|
protected DigitalSignature signer;
|
||||||
|
|
||||||
protected List<SignatureVerificationKey> trusteeSignatureVerificationArray;
|
protected List<SignatureVerificationKey> trusteeSignatureVerificationArray;
|
||||||
protected int minTrusteeSignatures;
|
protected int minTrusteeSignatures;
|
||||||
|
@ -351,11 +368,10 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
* This method initializes the signatures, connects to the DB and creates the schema (if required).
|
* This method initializes the signatures, connects to the DB and creates the schema (if required).
|
||||||
*/
|
*/
|
||||||
@Override
|
@Override
|
||||||
public void init(String meerkatDB) throws CommunicationException {
|
public void init() throws CommunicationException {
|
||||||
// TODO write signature reading part.
|
// TODO write signature reading part.
|
||||||
|
|
||||||
digest = new GenericBatchDigest(new SHA256Digest());
|
digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
signer = new GenericBatchDigitalSignature(new ECDSASignature());
|
|
||||||
|
|
||||||
jdbcTemplate = new NamedParameterJdbcTemplate(sqlQueryProvider.getDataSource());
|
jdbcTemplate = new NamedParameterJdbcTemplate(sqlQueryProvider.getDataSource());
|
||||||
|
|
||||||
|
@ -415,14 +431,17 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
/**
|
/**
|
||||||
* This method posts a messages to the server
|
* This method posts a messages to the server
|
||||||
* @param msg is the message to post
|
* @param msg is the message to post
|
||||||
* @param checkSignature decides whether ot not the method should check the signature before it posts the message
|
* @param precalculatedMsgID is an optional precalculated message ID
|
||||||
* @return TRUE if the post is successful and FALSE otherwise
|
* It is used when the message is the stub of a batch message
|
||||||
|
* In this case the validity of the signature is not checked either
|
||||||
|
* @return -1 if the message is not verified
|
||||||
|
* The entry number of the message if the message is posted
|
||||||
* @throws CommunicationException
|
* @throws CommunicationException
|
||||||
*/
|
*/
|
||||||
public BoolValue postMessage(BulletinBoardMessage msg, boolean checkSignature) throws CommunicationException{
|
private long postMessage(BulletinBoardMessage msg, byte[] precalculatedMsgID) throws CommunicationException{
|
||||||
|
|
||||||
if (checkSignature && !verifyMessage(msg)) {
|
if (precalculatedMsgID != null && !verifyMessage(msg)) {
|
||||||
return boolToBoolValue(false);
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
String sql;
|
String sql;
|
||||||
|
@ -437,12 +456,16 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
List<Signature> signatureList;
|
List<Signature> signatureList;
|
||||||
Signature[] signatures;
|
Signature[] signatures;
|
||||||
|
|
||||||
// Calculate message ID (depending only on the the unsigned message)
|
|
||||||
|
|
||||||
|
if (precalculatedMsgID != null){
|
||||||
|
msgID = precalculatedMsgID;
|
||||||
|
} else{
|
||||||
|
// Calculate message ID (depending only on the the unsigned message)
|
||||||
digest.reset();
|
digest.reset();
|
||||||
digest.update(msg.getMsg());
|
digest.update(msg.getMsg());
|
||||||
|
|
||||||
msgID = digest.digest();
|
msgID = digest.digest();
|
||||||
|
}
|
||||||
|
|
||||||
// Add message to table if needed and store entry number of message.
|
// Add message to table if needed and store entry number of message.
|
||||||
|
|
||||||
|
@ -520,13 +543,64 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
jdbcTemplate.batchUpdate(sql,namedParameterArray);
|
jdbcTemplate.batchUpdate(sql,namedParameterArray);
|
||||||
|
|
||||||
return boolToBoolValue(true);
|
return entryNum;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private void checkConnection() throws CommunicationException {
|
||||||
|
if (jdbcTemplate == null) {
|
||||||
|
throw new CommunicationException("DB connection not initialized");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public BoolValue postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
public BoolValue postMessage(BulletinBoardMessage msg) throws CommunicationException {
|
||||||
return postMessage(msg, true); // Perform a post and check the signature for authenticity
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
|
// Perform a post, calculate the message ID and check the signature for authenticity
|
||||||
|
if (postMessage(msg, null) != -1){
|
||||||
|
return BoolValue.newBuilder().setValue(true).build(); // Message was posted
|
||||||
|
}
|
||||||
|
|
||||||
|
return BoolValue.newBuilder().setValue(false).build(); // Message was not posted
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BoolValue deleteMessage(MessageID msgID) throws CommunicationException {
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
|
String sql = sqlQueryProvider.getSQLString(QueryType.DELETE_MSG_BY_ID);
|
||||||
|
Map namedParameters = new HashMap();
|
||||||
|
|
||||||
|
namedParameters.put(QueryType.DELETE_MSG_BY_ID.getParamName(0),msgID);
|
||||||
|
|
||||||
|
int affectedRows = jdbcTemplate.update(sql, namedParameters);
|
||||||
|
|
||||||
|
//TODO: Log
|
||||||
|
|
||||||
|
return BoolValue.newBuilder().setValue(affectedRows > 0).build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BoolValue deleteMessage(long entryNum) throws CommunicationException {
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
|
String sql = sqlQueryProvider.getSQLString(QueryType.DELETE_MSG_BY_ENTRY);
|
||||||
|
Map namedParameters = new HashMap();
|
||||||
|
|
||||||
|
namedParameters.put(QueryType.DELETE_MSG_BY_ENTRY.getParamName(0),entryNum);
|
||||||
|
|
||||||
|
int affectedRows = jdbcTemplate.update(sql, namedParameters);
|
||||||
|
|
||||||
|
//TODO: Log
|
||||||
|
|
||||||
|
return BoolValue.newBuilder().setValue(affectedRows > 0).build();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -611,10 +685,38 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Private implementation of the message reader for returning result as a list
|
||||||
|
* @param filterList is a filter list that defines which messages the client is interested in
|
||||||
|
* @return the requested list of messages
|
||||||
|
*/
|
||||||
|
private List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException {
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||||||
|
|
||||||
|
readMessages(filterList, new MessageOutputStream<BulletinBoardMessage>(outputStream));
|
||||||
|
|
||||||
|
MessageInputStream<BulletinBoardMessage> inputStream =
|
||||||
|
MessageInputStream.MessageInputStreamFactory.createMessageInputStream(new ByteArrayInputStream(
|
||||||
|
outputStream.toByteArray()),
|
||||||
|
BulletinBoardMessage.class);
|
||||||
|
|
||||||
|
return inputStream.asList();
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new CommunicationException(e.getCause() + " " + e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readMessages(MessageFilterList filterList, MessageOutputStream<BulletinBoardMessage> out) throws CommunicationException {
|
public void readMessages(MessageFilterList filterList, MessageOutputStream<BulletinBoardMessage> out) throws CommunicationException {
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
BulletinBoardMessageList.Builder resultListBuilder = BulletinBoardMessageList.newBuilder();
|
BulletinBoardMessageList.Builder resultListBuilder = BulletinBoardMessageList.newBuilder();
|
||||||
|
|
||||||
// SQL length is roughly 50 characters per filter + 50 for the query itself
|
// SQL length is roughly 50 characters per filter + 50 for the query itself
|
||||||
|
@ -635,39 +737,12 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Int32Value getMessageCount(MessageFilterList filterList) throws CommunicationException {
|
||||||
|
|
||||||
/**
|
checkConnection();
|
||||||
* This method returns a string representation of the tag associated with a batch ID
|
|
||||||
* @param batchId is the given batch ID
|
|
||||||
* @return the String representation of the tag
|
|
||||||
*/
|
|
||||||
private String batchIdToTag(int batchId) {
|
|
||||||
return BATCH_ID_TAG_PREFIX + Integer.toString(batchId);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
BulletinBoardMessageList.Builder resultListBuilder = BulletinBoardMessageList.newBuilder();
|
||||||
/**
|
|
||||||
* This method checks if a specified batch exists and is already closed
|
|
||||||
* @param signerId is the ID of the publisher of the batch
|
|
||||||
* @param batchId is the unique (per signer) batch ID
|
|
||||||
* @return TRUE if the batch is closed and FALSE if it is still open or doesn't exist at all
|
|
||||||
*/
|
|
||||||
private boolean isBatchClosed(ByteString signerId, int batchId) throws CommunicationException {
|
|
||||||
|
|
||||||
MessageFilterList filterList = MessageFilterList.newBuilder()
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.SIGNER_ID)
|
|
||||||
.setId(signerId)
|
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.TAG)
|
|
||||||
.setTag(BATCH_TAG)
|
|
||||||
.build())
|
|
||||||
.addFilter(MessageFilter.newBuilder()
|
|
||||||
.setType(FilterType.TAG)
|
|
||||||
.setTag(batchIdToTag(batchId))
|
|
||||||
.build())
|
|
||||||
.build();
|
|
||||||
|
|
||||||
// SQL length is roughly 50 characters per filter + 50 for the query itself
|
// SQL length is roughly 50 characters per filter + 50 for the query itself
|
||||||
StringBuilder sqlBuilder = new StringBuilder(50 * (filterList.getFilterCount() + 1));
|
StringBuilder sqlBuilder = new StringBuilder(50 * (filterList.getFilterCount() + 1));
|
||||||
|
@ -684,44 +759,73 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
// Run query and stream the output using a MessageCallbackHandler
|
// Run query and stream the output using a MessageCallbackHandler
|
||||||
|
|
||||||
List<Long> count = jdbcTemplate.query(sqlBuilder.toString(), sqlAndParameters.parameters, new LongMapper());
|
List<Long> count = jdbcTemplate.query(sqlBuilder.toString(), sqlAndParameters.parameters, new LongMapper());
|
||||||
|
return Int32Value.newBuilder().setValue(count.get(0).intValue()).build();
|
||||||
return (count.size() > 0) && (count.get(0) > 0);
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This method checks if a specified batch exists and is already closed
|
||||||
|
* @param msgID is the unique ID of the batch message
|
||||||
|
* @return TRUE if the batch is closed and FALSE if it is still open or doesn't exist at all
|
||||||
|
*/
|
||||||
|
private boolean isBatchClosed(MessageID msgID) throws CommunicationException {
|
||||||
|
|
||||||
|
MessageFilterList filterList = MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgID.getID())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
|
||||||
|
List<BulletinBoardMessage> messages = readMessages(filterList);
|
||||||
|
|
||||||
|
if (messages.size() <= 0){
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return (messages.get(0).getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.MSGID);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This method checks if a specified batch exists and is still open
|
||||||
|
* @param batchId is the temporary batch ID
|
||||||
|
* @return TRUE if the batch is closed and FALSE if it is still open or doesn't exist at all
|
||||||
|
*/
|
||||||
|
private boolean isBatchOpen(long batchId) throws CommunicationException {
|
||||||
|
|
||||||
|
String sql = sqlQueryProvider.getSQLString(QueryType.CHECK_BATCH_OPEN);
|
||||||
|
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
|
namedParameters.addValue(QueryType.CHECK_BATCH_OPEN.getParamName(0),batchId);
|
||||||
|
|
||||||
|
List<Long> result = jdbcTemplate.query(sql, namedParameters, new LongMapper());
|
||||||
|
|
||||||
|
return (result.size() > 0 && result.get(0) > 0);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public BoolValue beginBatch(BeginBatchMessage message) throws CommunicationException {
|
public Int64Value beginBatch(BeginBatchMessage message) throws CommunicationException {
|
||||||
|
|
||||||
// Check if batch is closed
|
checkConnection();
|
||||||
if (isBatchClosed(message.getSignerId(), message.getBatchId())) {
|
|
||||||
return BoolValue.newBuilder().setValue(false).build();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add new tags to table
|
// Store tags
|
||||||
ProtocolStringList tagList = message.getTagList();
|
String sql = sqlQueryProvider.getSQLString(QueryType.STORE_BATCH_TAGS);
|
||||||
String[] tags = new String[tagList.size()];
|
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
||||||
tags = tagList.toArray(tags);
|
|
||||||
try {
|
|
||||||
insertNewTags(tags);
|
|
||||||
} catch (SQLException e) {
|
|
||||||
throw new CommunicationException(e.getMessage());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Connect tags
|
namedParameters.addValue(QueryType.STORE_BATCH_TAGS.getParamName(0),message.toByteArray());
|
||||||
String sql = sqlQueryProvider.getSQLString(QueryType.CONNECT_BATCH_TAG);
|
|
||||||
MapSqlParameterSource namedParameters[] = new MapSqlParameterSource[tags.length];
|
|
||||||
|
|
||||||
for (int i=0 ; i < tags.length ; i++) {
|
jdbcTemplate.update(sql,namedParameters);
|
||||||
namedParameters[i] = new MapSqlParameterSource();
|
|
||||||
namedParameters[i].addValue(QueryType.CONNECT_BATCH_TAG.getParamName(0),message.getSignerId().toByteArray());
|
|
||||||
namedParameters[i].addValue(QueryType.CONNECT_BATCH_TAG.getParamName(1),message.getBatchId());
|
|
||||||
namedParameters[i].addValue(QueryType.CONNECT_BATCH_TAG.getParamName(2),tags[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
jdbcTemplate.batchUpdate(sql,namedParameters);
|
KeyHolder keyHolder = new GeneratedKeyHolder();
|
||||||
|
jdbcTemplate.update(sql, namedParameters, keyHolder);
|
||||||
|
|
||||||
return BoolValue.newBuilder().setValue(true).build();
|
long entryNum = keyHolder.getKey().longValue();
|
||||||
|
|
||||||
|
return Int64Value.newBuilder().setValue(entryNum).build();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -729,8 +833,10 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
@Override
|
@Override
|
||||||
public BoolValue postBatchMessage(BatchMessage batchMessage) throws CommunicationException{
|
public BoolValue postBatchMessage(BatchMessage batchMessage) throws CommunicationException{
|
||||||
|
|
||||||
// Check if batch is closed
|
checkConnection();
|
||||||
if (isBatchClosed(batchMessage.getSignerId(), batchMessage.getBatchId())) {
|
|
||||||
|
// Make sure batch is open
|
||||||
|
if (!isBatchOpen(batchMessage.getBatchId())) {
|
||||||
return BoolValue.newBuilder().setValue(false).build();
|
return BoolValue.newBuilder().setValue(false).build();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -738,10 +844,9 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
String sql = sqlQueryProvider.getSQLString(QueryType.INSERT_BATCH_DATA);
|
String sql = sqlQueryProvider.getSQLString(QueryType.INSERT_BATCH_DATA);
|
||||||
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(0),batchMessage.getSignerId().toByteArray());
|
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(0),batchMessage.getBatchId());
|
||||||
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(1),batchMessage.getBatchId());
|
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(1),batchMessage.getSerialNum());
|
||||||
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(2),batchMessage.getSerialNum());
|
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(2),batchMessage.getData().toByteArray());
|
||||||
namedParameters.addValue(QueryType.INSERT_BATCH_DATA.getParamName(3),batchMessage.getData().toByteArray());
|
|
||||||
|
|
||||||
jdbcTemplate.update(sql, namedParameters);
|
jdbcTemplate.update(sql, namedParameters);
|
||||||
|
|
||||||
|
@ -751,12 +856,9 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public BoolValue closeBatchMessage(CloseBatchMessage message) throws CommunicationException {
|
public BoolValue closeBatch(CloseBatchMessage message) throws CommunicationException {
|
||||||
|
|
||||||
ByteString signerId = message.getSig().getSignerId();
|
checkConnection();
|
||||||
int batchId = message.getBatchId();
|
|
||||||
|
|
||||||
KeyHolder keyHolder = new GeneratedKeyHolder();
|
|
||||||
|
|
||||||
// Check batch size
|
// Check batch size
|
||||||
|
|
||||||
|
@ -764,8 +866,7 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.CHECK_BATCH_LENGTH.getParamName(0),signerId.toByteArray());
|
namedParameters.addValue(QueryType.CHECK_BATCH_LENGTH.getParamName(0),message.getBatchId());
|
||||||
namedParameters.addValue(QueryType.CHECK_BATCH_LENGTH.getParamName(1),batchId);
|
|
||||||
|
|
||||||
List<Long> lengthResult = jdbcTemplate.query(sql, namedParameters, new LongMapper());
|
List<Long> lengthResult = jdbcTemplate.query(sql, namedParameters, new LongMapper());
|
||||||
|
|
||||||
|
@ -773,106 +874,87 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
return BoolValue.newBuilder().setValue(false).build();
|
return BoolValue.newBuilder().setValue(false).build();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get Tags and add them to CompleteBatch
|
// Get Tags and add them to BulletinBoardMessage
|
||||||
|
|
||||||
sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_TAGS);
|
sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_TAGS);
|
||||||
namedParameters = new MapSqlParameterSource();
|
namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_TAGS.getParamName(0),signerId.toByteArray());
|
namedParameters.addValue(QueryType.GET_BATCH_TAGS.getParamName(0),message.getBatchId());
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_TAGS.getParamName(1),batchId);
|
|
||||||
|
|
||||||
List<String> tags = jdbcTemplate.query(sql, namedParameters, new StringMapper());
|
List<BeginBatchMessage> beginBatchMessages = jdbcTemplate.query(sql, namedParameters, new BeginBatchMessageMapper());
|
||||||
|
|
||||||
CompleteBatch completeBatch = new CompleteBatch(
|
if (beginBatchMessages == null || beginBatchMessages.size() <= 0 || beginBatchMessages.get(0) == null) {
|
||||||
BeginBatchMessage.newBuilder()
|
return BoolValue.newBuilder().setValue(false).build();
|
||||||
.setSignerId(signerId)
|
}
|
||||||
.setBatchId(batchId)
|
|
||||||
.addAllTag(tags)
|
|
||||||
.build()
|
|
||||||
);
|
|
||||||
|
|
||||||
// Add timestamp to CompleteBatch
|
UnsignedBulletinBoardMessage unsignedMessage = UnsignedBulletinBoardMessage.newBuilder()
|
||||||
completeBatch.setTimestamp(message.getTimestamp());
|
.addAllTag(beginBatchMessages.get(0).getTagList())
|
||||||
|
.setTimestamp(message.getTimestamp())
|
||||||
|
.build();
|
||||||
|
|
||||||
// Add actual batch data to CompleteBatch
|
// Digest the data
|
||||||
|
|
||||||
sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_MESSAGE_DATA);
|
digest.reset();
|
||||||
|
digest.update(unsignedMessage);
|
||||||
|
|
||||||
|
sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID);
|
||||||
namedParameters = new MapSqlParameterSource();
|
namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(0),signerId.toByteArray());
|
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(0),message.getBatchId());
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(1),batchId);
|
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(1),0); // Read from the beginning
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(2),0); // Read from the beginning
|
jdbcTemplate.query(sql, namedParameters, new BatchDataDigestHandler(digest));
|
||||||
|
|
||||||
completeBatch.appendBatchData(jdbcTemplate.query(sql, namedParameters, new BatchDataMapper()));
|
byte[] msgID = digest.digest();
|
||||||
|
|
||||||
// Verify signature
|
//TODO: Signature verification
|
||||||
|
|
||||||
completeBatch.setSignature(message.getSig());
|
|
||||||
|
|
||||||
// try {
|
|
||||||
// TODO: Actual verification
|
|
||||||
// //signer.verify(completeBatch);
|
|
||||||
// } catch (CertificateException | InvalidKeyException | SignatureException e) {
|
|
||||||
// return BoolValue.newBuilder().setValue(false).build();
|
|
||||||
// }
|
|
||||||
|
|
||||||
// Batch verified: finalize it
|
|
||||||
|
|
||||||
// Calculate message ID
|
|
||||||
digest.reset();
|
|
||||||
digest.update(completeBatch);
|
|
||||||
MessageID msgID = MessageID.newBuilder().setID(ByteString.copyFrom(digest.digest())).build();
|
|
||||||
|
|
||||||
// Create Bulletin Board message
|
// Create Bulletin Board message
|
||||||
BulletinBoardMessage bulletinBoardMessage = BulletinBoardMessage.newBuilder()
|
BulletinBoardMessage bulletinBoardMessage = BulletinBoardMessage.newBuilder()
|
||||||
.addSig(message.getSig())
|
|
||||||
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
||||||
.addAllTag(tags)
|
.mergeFrom(unsignedMessage)
|
||||||
.addTag(BATCH_TAG)
|
.setMsgId(ByteString.copyFrom(msgID)))
|
||||||
.addTag(batchIdToTag(batchId))
|
.addAllSig(message.getSigList())
|
||||||
.setData(message.getSig().getSignerId())
|
|
||||||
.setTimestamp(message.getTimestamp())
|
|
||||||
.build())
|
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
// Post message without checking signature validity
|
// Post message with pre-calculated ID and without checking signature validity
|
||||||
postMessage(bulletinBoardMessage, false);
|
long entryNum = postMessage(bulletinBoardMessage, msgID);
|
||||||
|
|
||||||
// Remove tags from temporary table
|
// Add entry num to tag data table
|
||||||
sql = sqlQueryProvider.getSQLString(QueryType.REMOVE_BATCH_TAGS);
|
sql = sqlQueryProvider.getSQLString(QueryType.ADD_ENTRY_NUM_TO_BATCH);
|
||||||
namedParameters = new MapSqlParameterSource();
|
namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.REMOVE_BATCH_TAGS.getParamName(0), signerId.toByteArray());
|
namedParameters.addValue(QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(0), message.getBatchId());
|
||||||
namedParameters.addValue(QueryType.REMOVE_BATCH_TAGS.getParamName(1), batchId);
|
namedParameters.addValue(QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(1), entryNum);
|
||||||
|
|
||||||
jdbcTemplate.update(sql, namedParameters);
|
jdbcTemplate.update(sql, namedParameters);
|
||||||
|
|
||||||
// Return TRUE
|
// Return TRUE
|
||||||
|
|
||||||
return BoolValue.newBuilder().setValue(true).build();
|
return BoolValue.newBuilder().setValue(true).build();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage message, MessageOutputStream<BatchData> out) throws CommunicationException, IllegalArgumentException{
|
public void readBatch(BatchQuery batchQuery, MessageOutputStream<BatchChunk> out) throws CommunicationException, IllegalArgumentException{
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
// Check that batch is closed
|
// Check that batch is closed
|
||||||
if (!isBatchClosed(message.getSignerId(), message.getBatchId())) {
|
if (!isBatchClosed(batchQuery.getMsgID())) {
|
||||||
throw new IllegalArgumentException("No such batch");
|
throw new IllegalArgumentException("No such batch");
|
||||||
}
|
}
|
||||||
|
|
||||||
String sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_MESSAGE_DATA);
|
String sql = sqlQueryProvider.getSQLString(QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID);
|
||||||
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
MapSqlParameterSource namedParameters = new MapSqlParameterSource();
|
||||||
|
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(0),message.getSignerId().toByteArray());
|
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(0),batchQuery.getMsgID().getID().toByteArray());
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(1),message.getBatchId());
|
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(1),batchQuery.getStartPosition());
|
||||||
namedParameters.addValue(QueryType.GET_BATCH_MESSAGE_DATA.getParamName(2),message.getStartPosition());
|
|
||||||
|
|
||||||
jdbcTemplate.query(sql, namedParameters, new BatchDataCallbackHandler(out));
|
jdbcTemplate.query(sql, namedParameters, new BatchDataCallbackHandler(out));
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Finds the entry number of the last entry in the database
|
* Finds the entry number of the last entry in the database
|
||||||
* @return the entry number, or -1 if no entries are found
|
* @return the entry number, or -1 if no entries are found
|
||||||
|
@ -892,7 +974,9 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) {
|
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException{
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
if (generateSyncQueryParams == null
|
if (generateSyncQueryParams == null
|
||||||
|| !generateSyncQueryParams.hasFilterList()
|
|| !generateSyncQueryParams.hasFilterList()
|
||||||
|
@ -935,7 +1019,7 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
checksumChanged = true;
|
checksumChanged = true;
|
||||||
|
|
||||||
checksum.update(message.getMsg().getData());
|
checksum.update(message.getMsg().getMsgId());
|
||||||
|
|
||||||
lastTimestamp = message.getMsg().getTimestamp();
|
lastTimestamp = message.getMsg().getTimestamp();
|
||||||
message = messageIterator.next();
|
message = messageIterator.next();
|
||||||
|
@ -971,6 +1055,8 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
@Override
|
@Override
|
||||||
public SyncQueryResponse querySync(SyncQuery syncQuery) throws CommunicationException {
|
public SyncQueryResponse querySync(SyncQuery syncQuery) throws CommunicationException {
|
||||||
|
|
||||||
|
checkConnection();
|
||||||
|
|
||||||
if (syncQuery == null){
|
if (syncQuery == null){
|
||||||
return SyncQueryResponse.newBuilder()
|
return SyncQueryResponse.newBuilder()
|
||||||
.setLastEntryNum(-1)
|
.setLastEntryNum(-1)
|
||||||
|
@ -1013,7 +1099,7 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
// Advance checksum
|
// Advance checksum
|
||||||
|
|
||||||
ByteString messageID = message.getMsg().getData(); // The data field contains the message ID
|
ByteString messageID = message.getMsg().getMsgId(); // The data field contains the message ID
|
||||||
|
|
||||||
checksum.update(messageID);
|
checksum.update(messageID);
|
||||||
|
|
||||||
|
@ -1039,6 +1125,8 @@ public class BulletinBoardSQLServer implements BulletinBoardServer{
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void close() {}
|
public void close() {
|
||||||
|
jdbcTemplate = null;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,11 +2,7 @@ package meerkat.bulletinboard.sqlserver;
|
||||||
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.FilterType;
|
import meerkat.protobuf.BulletinBoardAPI.FilterType;
|
||||||
import org.apache.commons.dbcp2.BasicDataSource;
|
import org.apache.commons.dbcp2.BasicDataSource;
|
||||||
import org.h2.jdbcx.JdbcDataSource;
|
|
||||||
import javax.naming.Context;
|
|
||||||
import javax.naming.InitialContext;
|
|
||||||
|
|
||||||
import javax.naming.NamingException;
|
|
||||||
import javax.sql.DataSource;
|
import javax.sql.DataSource;
|
||||||
import java.text.MessageFormat;
|
import java.text.MessageFormat;
|
||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
|
@ -30,19 +26,28 @@ public class H2QueryProvider implements BulletinBoardSQLServer.SQLQueryProvider
|
||||||
|
|
||||||
switch(queryType) {
|
switch(queryType) {
|
||||||
case ADD_SIGNATURE:
|
case ADD_SIGNATURE:
|
||||||
return "INSERT INTO SignatureTable (EntryNum, SignerId, Signature)"
|
return MessageFormat.format(
|
||||||
+ " SELECT DISTINCT :EntryNum AS Entry, :SignerId AS Id, :Signature AS Sig FROM UtilityTable AS Temp"
|
"INSERT INTO SignatureTable (EntryNum, SignerId, Signature)"
|
||||||
|
+ " SELECT DISTINCT :{0} AS Entry, :{1} AS Id, :{2} AS Sig FROM UtilityTable AS Temp"
|
||||||
+ " WHERE NOT EXISTS"
|
+ " WHERE NOT EXISTS"
|
||||||
+ " (SELECT 1 FROM SignatureTable AS SubTable WHERE SubTable.SignerId = :SignerId AND SubTable.EntryNum = :EntryNum)";
|
+ " (SELECT 1 FROM SignatureTable AS SubTable WHERE SubTable.EntryNum = :{0} AND SubTable.SignerId = :{1})",
|
||||||
|
QueryType.ADD_SIGNATURE.getParamName(0),
|
||||||
|
QueryType.ADD_SIGNATURE.getParamName(1),
|
||||||
|
QueryType.ADD_SIGNATURE.getParamName(2));
|
||||||
|
|
||||||
case CONNECT_TAG:
|
case CONNECT_TAG:
|
||||||
return "INSERT INTO MsgTagTable (TagId, EntryNum)"
|
return MessageFormat.format(
|
||||||
+ " SELECT DISTINCT TagTable.TagId, :EntryNum AS NewEntry FROM TagTable WHERE Tag = :Tag"
|
"INSERT INTO MsgTagTable (TagId, EntryNum)"
|
||||||
|
+ " SELECT DISTINCT TagTable.TagId, :{0} AS NewEntry FROM TagTable WHERE Tag = :{1}"
|
||||||
+ " AND NOT EXISTS (SELECT 1 FROM MsgTagTable AS SubTable WHERE SubTable.TagId = TagTable.TagId"
|
+ " AND NOT EXISTS (SELECT 1 FROM MsgTagTable AS SubTable WHERE SubTable.TagId = TagTable.TagId"
|
||||||
+ " AND SubTable.EntryNum = :EntryNum)";
|
+ " AND SubTable.EntryNum = :{0})",
|
||||||
|
QueryType.CONNECT_TAG.getParamName(0),
|
||||||
|
QueryType.CONNECT_TAG.getParamName(1));
|
||||||
|
|
||||||
case FIND_MSG_ID:
|
case FIND_MSG_ID:
|
||||||
return "SELECT EntryNum From MsgTable WHERE MsgId = :MsgId";
|
return MessageFormat.format(
|
||||||
|
"SELECT EntryNum From MsgTable WHERE MsgId = :{0}",
|
||||||
|
QueryType.FIND_MSG_ID.getParamName(0));
|
||||||
|
|
||||||
case FIND_TAG_ID:
|
case FIND_TAG_ID:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
|
@ -59,74 +64,85 @@ public class H2QueryProvider implements BulletinBoardSQLServer.SQLQueryProvider
|
||||||
return "SELECT MsgTable.EntryNum, MsgTable.MsgId, MsgTable.ExactTime FROM MsgTable";
|
return "SELECT MsgTable.EntryNum, MsgTable.MsgId, MsgTable.ExactTime FROM MsgTable";
|
||||||
|
|
||||||
case GET_SIGNATURES:
|
case GET_SIGNATURES:
|
||||||
return "SELECT Signature FROM SignatureTable WHERE EntryNum = :EntryNum";
|
return MessageFormat.format(
|
||||||
|
"SELECT Signature FROM SignatureTable WHERE EntryNum = :{0}",
|
||||||
|
QueryType.GET_SIGNATURES.getParamName(0));
|
||||||
|
|
||||||
case INSERT_MSG:
|
case INSERT_MSG:
|
||||||
return "INSERT INTO MsgTable (MsgId, Msg, ExactTime) VALUES(:MsgId,:Msg,:TimeStamp)";
|
return MessageFormat.format(
|
||||||
|
"INSERT INTO MsgTable (MsgId, ExactTime, Msg) VALUES(:{0}, :{1}, :{2})",
|
||||||
|
QueryType.INSERT_MSG.getParamName(0),
|
||||||
|
QueryType.INSERT_MSG.getParamName(1),
|
||||||
|
QueryType.INSERT_MSG.getParamName(2));
|
||||||
|
|
||||||
|
case DELETE_MSG_BY_ENTRY:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"DELETE FROM MsgTable WHERE EntryNum = :{0}",
|
||||||
|
QueryType.DELETE_MSG_BY_ENTRY.getParamName(0));
|
||||||
|
|
||||||
|
case DELETE_MSG_BY_ID:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"DELETE FROM MsgTable WHERE MsgId = :{0}",
|
||||||
|
QueryType.DELETE_MSG_BY_ID.getParamName(0));
|
||||||
|
|
||||||
case INSERT_NEW_TAG:
|
case INSERT_NEW_TAG:
|
||||||
return "INSERT INTO TagTable(Tag) SELECT DISTINCT :Tag AS NewTag FROM UtilityTable WHERE"
|
return MessageFormat.format(
|
||||||
+ " NOT EXISTS (SELECT 1 FROM TagTable AS SubTable WHERE SubTable.Tag = :Tag)";
|
"INSERT INTO TagTable(Tag) SELECT DISTINCT :Tag AS NewTag FROM UtilityTable WHERE"
|
||||||
|
+ " NOT EXISTS (SELECT 1 FROM TagTable AS SubTable WHERE SubTable.Tag = :{0})",
|
||||||
|
QueryType.INSERT_NEW_TAG.getParamName(0));
|
||||||
|
|
||||||
case GET_LAST_MESSAGE_ENTRY:
|
case GET_LAST_MESSAGE_ENTRY:
|
||||||
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_ENTRY:
|
case GET_BATCH_MESSAGE_DATA_BY_MSG_ID:
|
||||||
return MessageFormat.format(
|
|
||||||
"SELECT MsgTable.EntryNum, MsgTable.Msg FROM MsgTable"
|
|
||||||
+ " INNER JOIN SignatureTable ON MsgTable.EntryNum = SignatureTable.EntryNum"
|
|
||||||
+ " INNER JOIN MsgTagTable ON MsgTable.EntryNum = MsgTagTable.EntryNum"
|
|
||||||
+ " INNER JOIN TagTable ON MsgTagTable.TagId = TagTable.TagId"
|
|
||||||
+ " WHERE SignatureTable.SignerId = :{0}"
|
|
||||||
+ " AND TagTable.Tag = :{1}",
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(1));
|
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_DATA:
|
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Data FROM BatchTable"
|
"SELECT Data FROM BatchTable"
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} AND SerialNum >= :{2}"
|
+ " INNER JOIN MsgTable ON MsgTable.EntryNum = BatchTable.EntryNum"
|
||||||
+ " ORDER BY SerialNum ASC",
|
+ " WHERE MsgTable.MsgId = :{0} AND BatchTable.SerialNum >= :{1}"
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(0),
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(1),
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(0),
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(2));
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(1));
|
||||||
|
|
||||||
|
case GET_BATCH_MESSAGE_DATA_BY_BATCH_ID:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"SELECT Data FROM BatchTable"
|
||||||
|
+ " WHERE BatchId = :{0} AND SerialNum >= :{1}"
|
||||||
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(0),
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(1));
|
||||||
|
|
||||||
case INSERT_BATCH_DATA:
|
case INSERT_BATCH_DATA:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTable (SignerId, BatchId, SerialNum, Data)"
|
"INSERT INTO BatchTable (BatchId, SerialNum, Data) VALUES (:{0}, :{1}, :{2})",
|
||||||
+ " VALUES (:{0}, :{1}, :{2}, :{3})",
|
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(2),
|
QueryType.INSERT_BATCH_DATA.getParamName(2));
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(3));
|
|
||||||
|
|
||||||
case CHECK_BATCH_LENGTH:
|
case CHECK_BATCH_LENGTH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT COUNT(Data) AS BatchLength FROM BatchTable"
|
"SELECT COUNT(Data) AS BatchLength FROM BatchTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1}",
|
QueryType.CHECK_BATCH_LENGTH.getParamName(0));
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(0),
|
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(1));
|
|
||||||
|
|
||||||
case CONNECT_BATCH_TAG:
|
case CHECK_BATCH_OPEN:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTagTable (SignerId, BatchId, TagId) SELECT :{0}, :{1}, TagId FROM TagTable"
|
"SELECT COUNT(BatchId) AS batchCount FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE Tag = :{2}",
|
QueryType.CHECK_BATCH_OPEN.getParamName(0));
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(0),
|
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(1),
|
case STORE_BATCH_TAGS:
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(2));
|
return MessageFormat.format(
|
||||||
|
"INSERT INTO BatchTagTable (Tags) VALUES (:{0})",
|
||||||
|
QueryType.STORE_BATCH_TAGS.getParamName(0));
|
||||||
|
|
||||||
case GET_BATCH_TAGS:
|
case GET_BATCH_TAGS:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Tag FROM TagTable INNER JOIN BatchTagTable ON TagTable.TagId = BatchTagTable.TagId"
|
"SELECT Tags FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} ORDER BY Tag ASC",
|
QueryType.GET_BATCH_TAGS.getParamName(0));
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(1));
|
|
||||||
|
|
||||||
case REMOVE_BATCH_TAGS:
|
case ADD_ENTRY_NUM_TO_BATCH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"DELETE FROM BatchTagTable WHERE SignerId = :{0} AND BatchId = :{1}",
|
"UPDATE BatchTable SET EntryNum = :{1} WHERE BatchId = :{0}",
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(0),
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(0),
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(1));
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(1));
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
||||||
|
@ -204,7 +220,7 @@ public class H2QueryProvider implements BulletinBoardSQLServer.SQLQueryProvider
|
||||||
BasicDataSource dataSource = new BasicDataSource();
|
BasicDataSource dataSource = new BasicDataSource();
|
||||||
|
|
||||||
dataSource.setDriverClassName("org.h2.Driver");
|
dataSource.setDriverClassName("org.h2.Driver");
|
||||||
dataSource.setUrl("jdbc:h2:~/" + dbName);
|
dataSource.setUrl("jdbc:h2:mem:" + dbName);
|
||||||
|
|
||||||
return dataSource;
|
return dataSource;
|
||||||
|
|
||||||
|
@ -215,28 +231,30 @@ public class H2QueryProvider implements BulletinBoardSQLServer.SQLQueryProvider
|
||||||
public List<String> getSchemaCreationCommands() {
|
public List<String> getSchemaCreationCommands() {
|
||||||
List<String> list = new LinkedList<String>();
|
List<String> list = new LinkedList<String>();
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS MsgTable (EntryNum INT NOT NULL AUTO_INCREMENT PRIMARY KEY, MsgId TINYBLOB UNIQUE, ExactTime TIMESTAMP, Msg BLOB)");
|
list.add("CREATE TABLE IF NOT EXISTS MsgTable (EntryNum INT NOT NULL AUTO_INCREMENT PRIMARY KEY,"
|
||||||
|
+ " MsgId TINYBLOB UNIQUE, ExactTime TIMESTAMP, Msg BLOB)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INT NOT NULL AUTO_INCREMENT PRIMARY KEY, Tag VARCHAR(50) UNIQUE)");
|
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INT NOT NULL AUTO_INCREMENT PRIMARY KEY, Tag VARCHAR(50) UNIQUE)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum INT, TagId INT,"
|
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum INT, TagId INT,"
|
||||||
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum),"
|
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE,"
|
||||||
+ " FOREIGN KEY (TagId) REFERENCES TagTable(TagId),"
|
+ " FOREIGN KEY (TagId) REFERENCES TagTable(TagId) ON DELETE CASCADE,"
|
||||||
+ " UNIQUE (EntryNum, TagID))");
|
+ " UNIQUE (EntryNum, TagID))");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INT, SignerId TINYBLOB, Signature TINYBLOB UNIQUE,"
|
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INT, SignerId TINYBLOB, Signature TINYBLOB UNIQUE,"
|
||||||
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum))");
|
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE)");
|
||||||
|
|
||||||
list.add("CREATE INDEX IF NOT EXISTS SignerIndex ON SignatureTable(SignerId)");
|
list.add("CREATE INDEX IF NOT EXISTS SignerIndex ON SignatureTable(SignerId)");
|
||||||
list.add("CREATE UNIQUE INDEX IF NOT EXISTS SignerIndex ON SignatureTable(SignerId, EntryNum)");
|
list.add("CREATE UNIQUE INDEX IF NOT EXISTS SignatureIndex ON SignatureTable(SignerId, EntryNum)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS BatchTable (SignerId TINYBLOB, BatchId INT, SerialNum INT, Data BLOB,"
|
list.add("CREATE TABLE IF NOT EXISTS BatchTagTable (BatchId INT AUTO_INCREMENT PRIMARY KEY, Tags BLOB)");
|
||||||
+ " UNIQUE(SignerId, BatchId, SerialNum))");
|
|
||||||
|
list.add("CREATE TABLE IF NOT EXISTS BatchTable (BatchId INT, EntryNum INT, SerialNum INT, Data BLOB,"
|
||||||
|
+ " UNIQUE(BatchId, SerialNum),"
|
||||||
|
+ " FOREIGN KEY (BatchId) REFERENCES BatchTagTable(BatchId) ON DELETE CASCADE)");
|
||||||
|
list.add("CREATE INDEX IF NOT EXISTS BatchDataIndex ON BatchTable(EntryNum, SerialNum)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS BatchTagTable (SignerId TINYBLOB, BatchId INT, TagId INT,"
|
|
||||||
+ " FOREIGN KEY (TagId) REFERENCES TagTable(TagId))");
|
|
||||||
|
|
||||||
list.add("CREATE INDEX IF NOT EXISTS BatchIndex ON BatchTagTable(SignerId, BatchId)");
|
|
||||||
|
|
||||||
// This is used to create a simple table with one entry.
|
// This is used to create a simple table with one entry.
|
||||||
// It is used for implementing a workaround for the missing INSERT IGNORE syntax
|
// It is used for implementing a workaround for the missing INSERT IGNORE syntax
|
||||||
|
@ -251,13 +269,20 @@ public class H2QueryProvider implements BulletinBoardSQLServer.SQLQueryProvider
|
||||||
List<String> list = new LinkedList<String>();
|
List<String> list = new LinkedList<String>();
|
||||||
|
|
||||||
list.add("DROP TABLE IF EXISTS UtilityTable");
|
list.add("DROP TABLE IF EXISTS UtilityTable");
|
||||||
list.add("DROP INDEX IF EXISTS BatchIndex");
|
|
||||||
list.add("DROP TABLE IF EXISTS BatchTagTable");
|
list.add("DROP INDEX IF EXISTS BatchDataIndex");
|
||||||
list.add("DROP TABLE IF EXISTS BatchTable");
|
list.add("DROP TABLE IF EXISTS BatchTable");
|
||||||
list.add("DROP INDEX IF EXISTS SignerIdIndex");
|
|
||||||
|
list.add("DROP INDEX IF EXISTS BatchTagIndex");
|
||||||
|
list.add("DROP TABLE IF EXISTS BatchTagTable");
|
||||||
|
|
||||||
list.add("DROP TABLE IF EXISTS MsgTagTable");
|
list.add("DROP TABLE IF EXISTS MsgTagTable");
|
||||||
|
|
||||||
|
list.add("DROP INDEX IF EXISTS SignerIdIndex");
|
||||||
list.add("DROP TABLE IF EXISTS SignatureTable");
|
list.add("DROP TABLE IF EXISTS SignatureTable");
|
||||||
|
|
||||||
list.add("DROP TABLE IF EXISTS TagTable");
|
list.add("DROP TABLE IF EXISTS TagTable");
|
||||||
|
|
||||||
list.add("DROP TABLE IF EXISTS MsgTable");
|
list.add("DROP TABLE IF EXISTS MsgTable");
|
||||||
|
|
||||||
return list;
|
return list;
|
||||||
|
|
|
@ -1,7 +1,5 @@
|
||||||
package meerkat.bulletinboard.sqlserver;
|
package meerkat.bulletinboard.sqlserver;
|
||||||
|
|
||||||
import com.mysql.jdbc.jdbc2.optional.MysqlDataSource;
|
|
||||||
import meerkat.bulletinboard.BulletinBoardConstants;
|
|
||||||
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer.SQLQueryProvider;
|
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer.SQLQueryProvider;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.FilterType;
|
import meerkat.protobuf.BulletinBoardAPI.FilterType;
|
||||||
import org.apache.commons.dbcp2.BasicDataSource;
|
import org.apache.commons.dbcp2.BasicDataSource;
|
||||||
|
@ -81,6 +79,16 @@ public class MySQLQueryProvider implements SQLQueryProvider {
|
||||||
QueryType.INSERT_MSG.getParamName(1),
|
QueryType.INSERT_MSG.getParamName(1),
|
||||||
QueryType.INSERT_MSG.getParamName(2));
|
QueryType.INSERT_MSG.getParamName(2));
|
||||||
|
|
||||||
|
case DELETE_MSG_BY_ENTRY:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"DELETE IGNORE FROM MsgTable WHERE EntryNum = :{0}",
|
||||||
|
QueryType.DELETE_MSG_BY_ENTRY.getParamName(0));
|
||||||
|
|
||||||
|
case DELETE_MSG_BY_ID:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"DELETE IGNORE FROM MsgTable WHERE MsgId = :{0}",
|
||||||
|
QueryType.DELETE_MSG_BY_ID.getParamName(0));
|
||||||
|
|
||||||
case INSERT_NEW_TAG:
|
case INSERT_NEW_TAG:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT IGNORE INTO TagTable(Tag) VALUES (:{0})",
|
"INSERT IGNORE INTO TagTable(Tag) VALUES (:{0})",
|
||||||
|
@ -89,62 +97,55 @@ public class MySQLQueryProvider implements SQLQueryProvider {
|
||||||
case GET_LAST_MESSAGE_ENTRY:
|
case GET_LAST_MESSAGE_ENTRY:
|
||||||
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_ENTRY:
|
case GET_BATCH_MESSAGE_DATA_BY_MSG_ID:
|
||||||
return MessageFormat.format(
|
|
||||||
"SELECT MsgTable.EntryNum, MsgTable.Msg FROM MsgTable"
|
|
||||||
+ " INNER JOIN SignatureTable ON MsgTable.EntryNum = SignatureTable.EntryNum"
|
|
||||||
+ " INNER JOIN MsgTagTable ON MsgTable.EntryNum = MsgTagTable.EntryNum"
|
|
||||||
+ " INNER JOIN TagTable ON MsgTagTable.TagId = TagTable.TagId"
|
|
||||||
+ " WHERE SignatureTable.SignerId = :{0}"
|
|
||||||
+ " AND TagTable.Tag = :{1}",
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(1));
|
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_DATA:
|
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Data FROM BatchTable"
|
"SELECT Data FROM BatchTable"
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} AND SerialNum >= :{2}"
|
+ " INNER JOIN MsgTable ON MsgTable.EntryNum = BatchTable.EntryNum"
|
||||||
+ " ORDER BY SerialNum ASC",
|
+ " WHERE MsgTable.MsgId = :{0} AND BatchTable.SerialNum >= :{1}"
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(0),
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(1),
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(0),
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(2));
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(1));
|
||||||
|
|
||||||
|
case GET_BATCH_MESSAGE_DATA_BY_BATCH_ID:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"SELECT Data FROM BatchTable"
|
||||||
|
+ " WHERE BatchId = :{0} AND SerialNum >= :{1}"
|
||||||
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(0),
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(1));
|
||||||
|
|
||||||
case INSERT_BATCH_DATA:
|
case INSERT_BATCH_DATA:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTable (SignerId, BatchId, SerialNum, Data)"
|
"INSERT INTO BatchTable (BatchId, SerialNum, Data) VALUES (:{0}, :{1}, :{2})",
|
||||||
+ " VALUES (:{0}, :{1}, :{2}, :{3})",
|
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(2),
|
QueryType.INSERT_BATCH_DATA.getParamName(2));
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(3));
|
|
||||||
|
|
||||||
case CHECK_BATCH_LENGTH:
|
case CHECK_BATCH_LENGTH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT COUNT(Data) AS BatchLength FROM BatchTable"
|
"SELECT COUNT(Data) AS BatchLength FROM BatchTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1}",
|
QueryType.CHECK_BATCH_LENGTH.getParamName(0));
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(0),
|
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(1));
|
|
||||||
|
|
||||||
case CONNECT_BATCH_TAG:
|
case CHECK_BATCH_OPEN:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTagTable (SignerId, BatchId, TagId) SELECT :{0}, :{1}, TagId FROM TagTable"
|
"SELECT COUNT(BatchId) AS batchCount FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE Tag = :{2}",
|
QueryType.CHECK_BATCH_OPEN.getParamName(0));
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(0),
|
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(1),
|
case STORE_BATCH_TAGS:
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(2));
|
return MessageFormat.format(
|
||||||
|
"INSERT INTO BatchTagTable (Tags) VALUES (:{0})",
|
||||||
|
QueryType.STORE_BATCH_TAGS.getParamName(0));
|
||||||
|
|
||||||
case GET_BATCH_TAGS:
|
case GET_BATCH_TAGS:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Tag FROM TagTable INNER JOIN BatchTagTable ON TagTable.TagId = BatchTagTable.TagId"
|
"SELECT Tags FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} ORDER BY Tag ASC",
|
QueryType.GET_BATCH_TAGS.getParamName(0));
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(1));
|
|
||||||
|
|
||||||
case REMOVE_BATCH_TAGS:
|
case ADD_ENTRY_NUM_TO_BATCH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"DELETE FROM BatchTagTable WHERE SignerId = :{0} AND BatchId = :{1}",
|
"UPDATE BatchTable SET EntryNum = :{1} WHERE BatchId = :{0}",
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(0),
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(0),
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(1));
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(1));
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
||||||
|
@ -240,19 +241,21 @@ public class MySQLQueryProvider implements SQLQueryProvider {
|
||||||
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INT NOT NULL AUTO_INCREMENT PRIMARY KEY, Tag VARCHAR(50), UNIQUE(Tag))");
|
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INT NOT NULL AUTO_INCREMENT PRIMARY KEY, Tag VARCHAR(50), UNIQUE(Tag))");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum INT, TagId INT,"
|
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum INT, TagId INT,"
|
||||||
+ " CONSTRAINT FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum),"
|
+ " CONSTRAINT FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE,"
|
||||||
+ " CONSTRAINT FOREIGN KEY (TagId) REFERENCES TagTable(TagId),"
|
+ " CONSTRAINT FOREIGN KEY (TagId) REFERENCES TagTable(TagId) ON DELETE CASCADE,"
|
||||||
+ " CONSTRAINT UNIQUE (EntryNum, TagID))");
|
+ " CONSTRAINT UNIQUE (EntryNum, TagID))");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INT, SignerId TINYBLOB, Signature TINYBLOB,"
|
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INT, SignerId TINYBLOB, Signature TINYBLOB,"
|
||||||
+ " INDEX(SignerId(32)), CONSTRAINT Unique_Signature UNIQUE(SignerId(32), EntryNum),"
|
+ " INDEX(SignerId(32)), CONSTRAINT Unique_Signature UNIQUE(SignerId(32), EntryNum),"
|
||||||
+ " CONSTRAINT FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum))");
|
+ " CONSTRAINT FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE)");
|
||||||
|
|
||||||
|
list.add("CREATE TABLE IF NOT EXISTS BatchTagTable (BatchId INT AUTO_INCREMENT PRIMARY KEY, Tags BLOB)");
|
||||||
|
|
||||||
|
list.add("CREATE TABLE IF NOT EXISTS BatchTable (BatchId INT, EntryNum INT, SerialNum INT, Data BLOB,"
|
||||||
|
+ " CONSTRAINT UNIQUE(BatchId, SerialNum),"
|
||||||
|
+ " CONSTRAINT FOREIGN KEY (BatchId) REFERENCES BatchTagTable(BatchId) ON DELETE CASCADE)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS BatchTable (SignerId TINYBLOB, BatchId INT, SerialNum INT, Data BLOB,"
|
|
||||||
+ " CONSTRAINT Unique_Batch UNIQUE(SignerId(32), BatchId, SerialNum))");
|
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS BatchTagTable (SignerId TINYBLOB, BatchId INT, TagId INT,"
|
|
||||||
+ " INDEX(SignerId(32), BatchId), CONSTRAINT FOREIGN KEY (TagId) REFERENCES TagTable(TagId))");
|
|
||||||
|
|
||||||
return list;
|
return list;
|
||||||
}
|
}
|
||||||
|
@ -261,8 +264,8 @@ public class MySQLQueryProvider implements SQLQueryProvider {
|
||||||
public List<String> getSchemaDeletionCommands() {
|
public List<String> getSchemaDeletionCommands() {
|
||||||
List<String> list = new LinkedList<String>();
|
List<String> list = new LinkedList<String>();
|
||||||
|
|
||||||
list.add("DROP TABLE IF EXISTS BatchTagTable");
|
|
||||||
list.add("DROP TABLE IF EXISTS BatchTable");
|
list.add("DROP TABLE IF EXISTS BatchTable");
|
||||||
|
list.add("DROP TABLE IF EXISTS BatchTagTable");
|
||||||
list.add("DROP TABLE IF EXISTS MsgTagTable");
|
list.add("DROP TABLE IF EXISTS MsgTagTable");
|
||||||
list.add("DROP TABLE IF EXISTS SignatureTable");
|
list.add("DROP TABLE IF EXISTS SignatureTable");
|
||||||
list.add("DROP TABLE IF EXISTS TagTable");
|
list.add("DROP TABLE IF EXISTS TagTable");
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
package meerkat.bulletinboard.sqlserver;
|
package meerkat.bulletinboard.sqlserver;
|
||||||
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import org.apache.commons.dbcp2.BasicDataSource;
|
||||||
import org.sqlite.SQLiteDataSource;
|
import org.sqlite.SQLiteDataSource;
|
||||||
|
|
||||||
import javax.sql.DataSource;
|
import javax.sql.DataSource;
|
||||||
|
@ -60,62 +61,55 @@ public class SQLiteQueryProvider implements BulletinBoardSQLServer.SQLQueryProvi
|
||||||
case GET_LAST_MESSAGE_ENTRY:
|
case GET_LAST_MESSAGE_ENTRY:
|
||||||
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
return "SELECT MAX(MsgTable.EntryNum) FROM MsgTable";
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_ENTRY:
|
case GET_BATCH_MESSAGE_DATA_BY_MSG_ID:
|
||||||
return MessageFormat.format(
|
|
||||||
"SELECT MsgTable.EntryNum, MsgTable.Msg FROM MsgTable"
|
|
||||||
+ " INNER JOIN SignatureTable ON MsgTable.EntryNum = SignatureTable.EntryNum"
|
|
||||||
+ " INNER JOIN MsgTagTable ON MsgTable.EntryNum = MsgTagTable.EntryNum"
|
|
||||||
+ " INNER JOIN TagTable ON MsgTagTable.TagId = TagTable.TagId"
|
|
||||||
+ " WHERE SignatureTable.SignerId = :{0}"
|
|
||||||
+ " AND TagTable.Tag = :{1}",
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_MESSAGE_ENTRY.getParamName(1));
|
|
||||||
|
|
||||||
case GET_BATCH_MESSAGE_DATA:
|
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Data FROM BatchTable"
|
"SELECT Data FROM BatchTable"
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} AND SerialNum >= :{2}"
|
+ " INNER JOIN MsgTable ON MsgTable.EntryNum = BatchTable.EntryNum"
|
||||||
+ " ORDER BY SerialNum ASC",
|
+ " WHERE MsgTable.MsgId = :{0} AND BatchTable.SerialNum >= :{1}"
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(0),
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(1),
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(0),
|
||||||
QueryType.GET_BATCH_MESSAGE_DATA.getParamName(2));
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_MSG_ID.getParamName(1));
|
||||||
|
|
||||||
|
case GET_BATCH_MESSAGE_DATA_BY_BATCH_ID:
|
||||||
|
return MessageFormat.format(
|
||||||
|
"SELECT Data FROM BatchTable"
|
||||||
|
+ " WHERE BatchId = :{0} AND SerialNum >= :{1}"
|
||||||
|
+ " ORDER BY BatchTable.SerialNum ASC",
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(0),
|
||||||
|
QueryType.GET_BATCH_MESSAGE_DATA_BY_BATCH_ID.getParamName(1));
|
||||||
|
|
||||||
case INSERT_BATCH_DATA:
|
case INSERT_BATCH_DATA:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTable (SignerId, BatchId, SerialNum, Data)"
|
"INSERT INTO BatchTable (BatchId, SerialNum, Data) VALUES (:{0}, :{1}, :{2})",
|
||||||
+ " VALUES (:{0}, :{1}, :{2}, :{3})",
|
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
QueryType.INSERT_BATCH_DATA.getParamName(0),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
QueryType.INSERT_BATCH_DATA.getParamName(1),
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(2),
|
QueryType.INSERT_BATCH_DATA.getParamName(2));
|
||||||
QueryType.INSERT_BATCH_DATA.getParamName(3));
|
|
||||||
|
|
||||||
case CHECK_BATCH_LENGTH:
|
case CHECK_BATCH_LENGTH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT COUNT(Data) AS BatchLength FROM BatchTable"
|
"SELECT COUNT(Data) AS BatchLength FROM BatchTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1}",
|
QueryType.CHECK_BATCH_LENGTH.getParamName(0));
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(0),
|
|
||||||
QueryType.CHECK_BATCH_LENGTH.getParamName(1));
|
|
||||||
|
|
||||||
case CONNECT_BATCH_TAG:
|
case CHECK_BATCH_OPEN:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"INSERT INTO BatchTagTable (SignerId, BatchId, TagId) SELECT :{0}, :{1}, TagId FROM TagTable"
|
"SELECT COUNT(BatchId) AS batchCount FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE Tag = :{2}",
|
QueryType.CHECK_BATCH_OPEN.getParamName(0));
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(0),
|
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(1),
|
case STORE_BATCH_TAGS:
|
||||||
QueryType.CONNECT_BATCH_TAG.getParamName(2));
|
return MessageFormat.format(
|
||||||
|
"INSERT INTO BatchTagTable (Tags) VALUES (:{0})",
|
||||||
|
QueryType.STORE_BATCH_TAGS.getParamName(0));
|
||||||
|
|
||||||
case GET_BATCH_TAGS:
|
case GET_BATCH_TAGS:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"SELECT Tag FROM TagTable INNER JOIN BatchTagTable ON TagTable.TagId = BatchTagTable.TagId"
|
"SELECT Tags FROM BatchTagTable WHERE BatchId = :{0}",
|
||||||
+ " WHERE SignerId = :{0} AND BatchId = :{1} ORDER BY Tag ASC",
|
QueryType.GET_BATCH_TAGS.getParamName(0));
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(0),
|
|
||||||
QueryType.GET_BATCH_TAGS.getParamName(1));
|
|
||||||
|
|
||||||
case REMOVE_BATCH_TAGS:
|
case ADD_ENTRY_NUM_TO_BATCH:
|
||||||
return MessageFormat.format(
|
return MessageFormat.format(
|
||||||
"DELETE FROM BatchTagTable WHERE SignerId = :{0} AND BatchId = :{1}",
|
"UPDATE BatchTable SET EntryNum = :{1} WHERE BatchId = :{0}",
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(0),
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(0),
|
||||||
QueryType.REMOVE_BATCH_TAGS.getParamName(1));
|
QueryType.ADD_ENTRY_NUM_TO_BATCH.getParamName(1));
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
throw new IllegalArgumentException("Cannot serve a query of type " + queryType);
|
||||||
|
@ -191,7 +185,8 @@ public class SQLiteQueryProvider implements BulletinBoardSQLServer.SQLQueryProvi
|
||||||
@Override
|
@Override
|
||||||
public DataSource getDataSource() {
|
public DataSource getDataSource() {
|
||||||
|
|
||||||
SQLiteDataSource dataSource = new SQLiteDataSource();
|
BasicDataSource dataSource = new BasicDataSource();
|
||||||
|
dataSource.setDriverClassName("org.sqlite.JDBC");
|
||||||
dataSource.setUrl("jdbc:sqlite:" + dbName);
|
dataSource.setUrl("jdbc:sqlite:" + dbName);
|
||||||
|
|
||||||
return dataSource;
|
return dataSource;
|
||||||
|
@ -205,14 +200,23 @@ public class SQLiteQueryProvider implements BulletinBoardSQLServer.SQLQueryProvi
|
||||||
list.add("CREATE TABLE IF NOT EXISTS MsgTable (EntryNum INTEGER PRIMARY KEY, MsgId BLOB UNIQUE, Msg BLOB)");
|
list.add("CREATE TABLE IF NOT EXISTS MsgTable (EntryNum INTEGER PRIMARY KEY, MsgId BLOB UNIQUE, Msg BLOB)");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INTEGER PRIMARY KEY, Tag varchar(50) UNIQUE)");
|
list.add("CREATE TABLE IF NOT EXISTS TagTable (TagId INTEGER PRIMARY KEY, Tag varchar(50) UNIQUE)");
|
||||||
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum BLOB, TagId INTEGER, FOREIGN KEY (EntryNum)"
|
list.add("CREATE TABLE IF NOT EXISTS MsgTagTable (EntryNum BLOB, TagId INTEGER,"
|
||||||
+ " REFERENCES MsgTable(EntryNum), FOREIGN KEY (TagId) REFERENCES TagTable(TagId), UNIQUE (EntryNum, TagID))");
|
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE,"
|
||||||
|
+ " FOREIGN KEY (TagId) REFERENCES TagTable(TagId) ON DELETE CASCADE,"
|
||||||
|
+ " UNIQUE (EntryNum, TagID))");
|
||||||
|
|
||||||
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INTEGER, SignerId BLOB, Signature BLOB,"
|
list.add("CREATE TABLE IF NOT EXISTS SignatureTable (EntryNum INTEGER, SignerId BLOB, Signature BLOB,"
|
||||||
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum), UNIQUE(SignerId, EntryNum))");
|
+ " FOREIGN KEY (EntryNum) REFERENCES MsgTable(EntryNum) ON DELETE CASCADE,"
|
||||||
|
+ " UNIQUE(SignerId, EntryNum))");
|
||||||
|
|
||||||
list.add("CREATE INDEX IF NOT EXISTS SignerIndex ON SignatureTable(SignerId)");
|
list.add("CREATE INDEX IF NOT EXISTS SignerIndex ON SignatureTable(SignerId)");
|
||||||
|
|
||||||
|
list.add("CREATE TABLE IF NOT EXISTS BatchTagTable (BatchId INTEGER PRIMARY KEY, Tags BLOB)");
|
||||||
|
|
||||||
|
list.add("CREATE TABLE IF NOT EXISTS BatchTable (BatchId INTEGER, EntryNum INTEGER, SerialNum INTEGER, Data BLOB,"
|
||||||
|
+ " UNIQUE(BatchId, SerialNum),"
|
||||||
|
+ " FOREIGN KEY (BatchId) REFERENCES BatchTagTable(BatchId) ON DELETE CASCADE)");
|
||||||
|
|
||||||
return list;
|
return list;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,10 +1,8 @@
|
||||||
package meerkat.bulletinboard.sqlserver.mappers;
|
package meerkat.bulletinboard.sqlserver.mappers;
|
||||||
|
|
||||||
import com.google.protobuf.InvalidProtocolBufferException;
|
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
import meerkat.protobuf.BulletinBoardAPI.BatchChunk;
|
||||||
import org.springframework.jdbc.core.RowCallbackHandler;
|
import org.springframework.jdbc.core.RowCallbackHandler;
|
||||||
import org.springframework.jdbc.core.RowMapper;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.sql.ResultSet;
|
import java.sql.ResultSet;
|
||||||
|
@ -15,16 +13,16 @@ import java.sql.SQLException;
|
||||||
*/
|
*/
|
||||||
public class BatchDataCallbackHandler implements RowCallbackHandler {
|
public class BatchDataCallbackHandler implements RowCallbackHandler {
|
||||||
|
|
||||||
private final MessageOutputStream<BatchData> out;
|
private final MessageOutputStream<BatchChunk> out;
|
||||||
|
|
||||||
public BatchDataCallbackHandler(MessageOutputStream<BatchData> out) {
|
public BatchDataCallbackHandler(MessageOutputStream<BatchChunk> out) {
|
||||||
this.out = out;
|
this.out = out;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void processRow(ResultSet rs) throws SQLException {
|
public void processRow(ResultSet rs) throws SQLException {
|
||||||
try {
|
try {
|
||||||
out.writeMessage(BatchData.parseFrom(rs.getBytes(1)));
|
out.writeMessage(BatchChunk.parseFrom(rs.getBytes(1)));
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
//TODO: Log
|
//TODO: Log
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,31 @@
|
||||||
|
package meerkat.bulletinboard.sqlserver.mappers;
|
||||||
|
|
||||||
|
import meerkat.bulletinboard.BulletinBoardDigest;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BatchChunk;
|
||||||
|
import org.springframework.jdbc.core.RowCallbackHandler;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.sql.ResultSet;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 19-Dec-15.
|
||||||
|
*/
|
||||||
|
public class BatchDataDigestHandler implements RowCallbackHandler {
|
||||||
|
|
||||||
|
private final BulletinBoardDigest digest;
|
||||||
|
|
||||||
|
public BatchDataDigestHandler(BulletinBoardDigest digest) {
|
||||||
|
this.digest = digest;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void processRow(ResultSet rs) throws SQLException {
|
||||||
|
try {
|
||||||
|
BatchChunk batchChunk = BatchChunk.parseFrom(rs.getBytes(1));
|
||||||
|
digest.update(batchChunk.getData().toByteArray());
|
||||||
|
} catch (IOException e) {
|
||||||
|
//TODO: Log
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,26 +0,0 @@
|
||||||
package meerkat.bulletinboard.sqlserver.mappers;
|
|
||||||
|
|
||||||
import com.google.protobuf.ByteString;
|
|
||||||
import com.google.protobuf.InvalidProtocolBufferException;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
|
||||||
import org.springframework.jdbc.core.RowMapper;
|
|
||||||
|
|
||||||
import java.sql.ResultSet;
|
|
||||||
import java.sql.SQLException;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 19-Dec-15.
|
|
||||||
*/
|
|
||||||
public class BatchDataMapper implements RowMapper<BatchData> {
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public BatchData mapRow(ResultSet rs, int rowNum) throws SQLException {
|
|
||||||
|
|
||||||
try {
|
|
||||||
return BatchData.parseFrom(rs.getBytes(1));
|
|
||||||
} catch (InvalidProtocolBufferException e) {
|
|
||||||
return BatchData.getDefaultInstance();
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -0,0 +1,24 @@
|
||||||
|
package meerkat.bulletinboard.sqlserver.mappers;
|
||||||
|
|
||||||
|
import com.google.protobuf.InvalidProtocolBufferException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BeginBatchMessage;
|
||||||
|
import org.springframework.jdbc.core.RowMapper;
|
||||||
|
|
||||||
|
import java.sql.ResultSet;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 20-Dec-15.
|
||||||
|
*/
|
||||||
|
public class BeginBatchMessageMapper implements RowMapper<BeginBatchMessage> {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BeginBatchMessage mapRow(ResultSet rs, int rowNum) throws SQLException {
|
||||||
|
try {
|
||||||
|
return BeginBatchMessage.newBuilder().mergeFrom(rs.getBytes(1)).build();
|
||||||
|
} catch (InvalidProtocolBufferException e) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -21,7 +21,7 @@ public class MessageStubMapper implements RowMapper<BulletinBoardMessage> {
|
||||||
return BulletinBoardMessage.newBuilder()
|
return BulletinBoardMessage.newBuilder()
|
||||||
.setEntryNum(rs.getLong(1))
|
.setEntryNum(rs.getLong(1))
|
||||||
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
||||||
.setData(ByteString.copyFrom(rs.getBytes(2)))
|
.setMsgId(ByteString.copyFrom(rs.getBytes(2)))
|
||||||
.setTimestamp(BulletinBoardUtils.toTimestampProto(rs.getTimestamp(3)))
|
.setTimestamp(BulletinBoardUtils.toTimestampProto(rs.getTimestamp(3)))
|
||||||
.build())
|
.build())
|
||||||
.build();
|
.build();
|
||||||
|
|
|
@ -9,6 +9,8 @@ import javax.ws.rs.core.MediaType;
|
||||||
import javax.ws.rs.core.StreamingOutput;
|
import javax.ws.rs.core.StreamingOutput;
|
||||||
|
|
||||||
import com.google.protobuf.BoolValue;
|
import com.google.protobuf.BoolValue;
|
||||||
|
import com.google.protobuf.Int32Value;
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
import meerkat.bulletinboard.BulletinBoardServer;
|
import meerkat.bulletinboard.BulletinBoardServer;
|
||||||
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer;
|
import meerkat.bulletinboard.sqlserver.BulletinBoardSQLServer;
|
||||||
import meerkat.bulletinboard.sqlserver.H2QueryProvider;
|
import meerkat.bulletinboard.sqlserver.H2QueryProvider;
|
||||||
|
@ -17,13 +19,11 @@ import meerkat.bulletinboard.sqlserver.SQLiteQueryProvider;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Comm.*;
|
|
||||||
import static meerkat.bulletinboard.BulletinBoardConstants.*;
|
import static meerkat.bulletinboard.BulletinBoardConstants.*;
|
||||||
import static meerkat.rest.Constants.*;
|
import static meerkat.rest.Constants.*;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
import java.util.Collection;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* An implementation of the BulletinBoardServer which functions as a WebApp
|
* An implementation of the BulletinBoardServer which functions as a WebApp
|
||||||
|
@ -44,14 +44,6 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
bulletinBoard = (BulletinBoardServer) servletContext.getAttribute(BULLETIN_BOARD_ATTRIBUTE_NAME);
|
bulletinBoard = (BulletinBoardServer) servletContext.getAttribute(BULLETIN_BOARD_ATTRIBUTE_NAME);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* This is the BulletinBoard init method.
|
|
||||||
*/
|
|
||||||
@Override
|
|
||||||
public void init(String meerkatDB) throws CommunicationException {
|
|
||||||
bulletinBoard.init(meerkatDB);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void contextInitialized(ServletContextEvent servletContextEvent) {
|
public void contextInitialized(ServletContextEvent servletContextEvent) {
|
||||||
ServletContext servletContext = servletContextEvent.getServletContext();
|
ServletContext servletContext = servletContextEvent.getServletContext();
|
||||||
|
@ -77,7 +69,7 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
init(dbName);
|
bulletinBoard.init();
|
||||||
servletContext.setAttribute(BULLETIN_BOARD_ATTRIBUTE_NAME, bulletinBoard);
|
servletContext.setAttribute(BULLETIN_BOARD_ATTRIBUTE_NAME, bulletinBoard);
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
|
@ -100,6 +92,16 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
bulletinBoard.readMessages(filterList, out);
|
bulletinBoard.readMessages(filterList, out);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Path(COUNT_MESSAGES_PATH)
|
||||||
|
@POST
|
||||||
|
@Consumes(MEDIATYPE_PROTOBUF)
|
||||||
|
@Produces(MEDIATYPE_PROTOBUF)
|
||||||
|
@Override
|
||||||
|
public Int32Value getMessageCount(MessageFilterList filterList) throws CommunicationException {
|
||||||
|
init();
|
||||||
|
return bulletinBoard.getMessageCount(filterList);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
@Path(READ_MESSAGES_PATH)
|
@Path(READ_MESSAGES_PATH)
|
||||||
@POST
|
@POST
|
||||||
|
@ -133,7 +135,7 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
@Consumes(MEDIATYPE_PROTOBUF)
|
@Consumes(MEDIATYPE_PROTOBUF)
|
||||||
@Produces(MEDIATYPE_PROTOBUF)
|
@Produces(MEDIATYPE_PROTOBUF)
|
||||||
@Override
|
@Override
|
||||||
public BoolValue beginBatch(BeginBatchMessage message) {
|
public Int64Value beginBatch(BeginBatchMessage message) {
|
||||||
try {
|
try {
|
||||||
init();
|
init();
|
||||||
return bulletinBoard.beginBatch(message);
|
return bulletinBoard.beginBatch(message);
|
||||||
|
@ -163,10 +165,10 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
@Consumes(MEDIATYPE_PROTOBUF)
|
@Consumes(MEDIATYPE_PROTOBUF)
|
||||||
@Produces(MEDIATYPE_PROTOBUF)
|
@Produces(MEDIATYPE_PROTOBUF)
|
||||||
@Override
|
@Override
|
||||||
public BoolValue closeBatchMessage(CloseBatchMessage message) {
|
public BoolValue closeBatch(CloseBatchMessage message) {
|
||||||
try {
|
try {
|
||||||
init();
|
init();
|
||||||
return bulletinBoard.closeBatchMessage(message);
|
return bulletinBoard.closeBatch(message);
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
return null;
|
return null;
|
||||||
|
@ -175,10 +177,10 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
|
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void readBatch(BatchSpecificationMessage message, MessageOutputStream<BatchData> out) {
|
public void readBatch(BatchQuery batchQuery, MessageOutputStream<BatchChunk> out) throws CommunicationException, IllegalArgumentException {
|
||||||
try {
|
try {
|
||||||
init();
|
init();
|
||||||
bulletinBoard.readBatch(message, out);
|
bulletinBoard.readBatch(batchQuery, out);
|
||||||
} catch (CommunicationException | IllegalArgumentException e) {
|
} catch (CommunicationException | IllegalArgumentException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
}
|
}
|
||||||
|
@ -205,17 +207,17 @@ public class BulletinBoardWebApp implements BulletinBoardServer, ServletContextL
|
||||||
/**
|
/**
|
||||||
* Wrapper for the readBatch method which streams the output into the response
|
* Wrapper for the readBatch method which streams the output into the response
|
||||||
*/
|
*/
|
||||||
public StreamingOutput readBatch(final BatchSpecificationMessage message) {
|
public StreamingOutput readBatch(final BatchQuery batchQuery) {
|
||||||
|
|
||||||
return new StreamingOutput() {
|
return new StreamingOutput() {
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void write(OutputStream output) throws IOException, WebApplicationException {
|
public void write(OutputStream output) throws IOException, WebApplicationException {
|
||||||
MessageOutputStream<BatchData> out = new MessageOutputStream<>(output);
|
MessageOutputStream<BatchChunk> out = new MessageOutputStream<>(output);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
init();
|
init();
|
||||||
bulletinBoard.readBatch(message, out);
|
bulletinBoard.readBatch(batchQuery, out);
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
//TODO: Log
|
//TODO: Log
|
||||||
out.writeMessage(null);
|
out.writeMessage(null);
|
||||||
|
|
|
@ -1,9 +0,0 @@
|
||||||
syntax = "proto3";
|
|
||||||
|
|
||||||
package meerkat;
|
|
||||||
|
|
||||||
option java_package = "meerkat.protobuf";
|
|
||||||
|
|
||||||
message Boolean {
|
|
||||||
bool value = 1;
|
|
||||||
}
|
|
|
@ -20,17 +20,18 @@ import java.util.*;
|
||||||
import com.google.protobuf.BoolValue;
|
import com.google.protobuf.BoolValue;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.ByteString;
|
||||||
|
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
import com.google.protobuf.Timestamp;
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.comm.MessageInputStream;
|
import meerkat.comm.MessageInputStream;
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.comm.MessageInputStream.MessageInputStreamFactory;
|
import meerkat.comm.MessageInputStream.MessageInputStreamFactory;
|
||||||
import meerkat.crypto.Digest;
|
|
||||||
import meerkat.crypto.concrete.ECDSASignature;
|
import meerkat.crypto.concrete.ECDSASignature;
|
||||||
import meerkat.crypto.concrete.SHA256Digest;
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Comm.*;
|
import meerkat.util.BulletinBoardMessageComparator;
|
||||||
import meerkat.util.BulletinBoardMessageGenerator;
|
import meerkat.util.BulletinBoardMessageGenerator;
|
||||||
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
|
||||||
import static org.junit.Assert.*;
|
import static org.junit.Assert.*;
|
||||||
import static org.hamcrest.CoreMatchers.*;
|
import static org.hamcrest.CoreMatchers.*;
|
||||||
|
@ -39,7 +40,7 @@ import static org.hamcrest.MatcherAssert.assertThat;
|
||||||
public class GenericBulletinBoardServerTest {
|
public class GenericBulletinBoardServerTest {
|
||||||
|
|
||||||
protected BulletinBoardServer bulletinBoardServer;
|
protected BulletinBoardServer bulletinBoardServer;
|
||||||
private GenericBatchDigitalSignature[] signers;
|
private GenericBulletinBoardSignature[] signers;
|
||||||
private ByteString[] signerIDs;
|
private ByteString[] signerIDs;
|
||||||
|
|
||||||
private Random random;
|
private Random random;
|
||||||
|
@ -59,24 +60,18 @@ public class GenericBulletinBoardServerTest {
|
||||||
private String[] tags;
|
private String[] tags;
|
||||||
private byte[][] data;
|
private byte[][] data;
|
||||||
|
|
||||||
private List<CompleteBatch> completeBatches;
|
private List<BulletinBoardMessage> batches;
|
||||||
|
|
||||||
private final ThreadMXBean threadBean = ManagementFactory.getThreadMXBean(); // Used to time the tests
|
private final ThreadMXBean threadBean = ManagementFactory.getThreadMXBean(); // Used to time the tests
|
||||||
|
|
||||||
private BulletinBoardMessageGenerator bulletinBoardMessageGenerator;
|
private BulletinBoardMessageGenerator bulletinBoardMessageGenerator;
|
||||||
|
|
||||||
private Digest digest;
|
private BulletinBoardDigest digest;
|
||||||
|
|
||||||
|
private BulletinBoardMessageComparator comparator;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param bulletinBoardServer is an initialized server.
|
* @param bulletinBoardServer is an initialized server.
|
||||||
* @throws InstantiationException
|
|
||||||
* @throws IllegalAccessException
|
|
||||||
* @throws CertificateException
|
|
||||||
* @throws KeyStoreException
|
|
||||||
* @throws NoSuchAlgorithmException
|
|
||||||
* @throws IOException
|
|
||||||
* @throws UnrecoverableKeyException
|
|
||||||
* @throws CommunicationException
|
|
||||||
*/
|
*/
|
||||||
public void init(BulletinBoardServer bulletinBoardServer) {
|
public void init(BulletinBoardServer bulletinBoardServer) {
|
||||||
|
|
||||||
|
@ -85,10 +80,10 @@ public class GenericBulletinBoardServerTest {
|
||||||
|
|
||||||
this.bulletinBoardServer = bulletinBoardServer;
|
this.bulletinBoardServer = bulletinBoardServer;
|
||||||
|
|
||||||
signers = new GenericBatchDigitalSignature[2];
|
signers = new GenericBulletinBoardSignature[2];
|
||||||
signerIDs = new ByteString[signers.length];
|
signerIDs = new ByteString[signers.length];
|
||||||
signers[0] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[0] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
signers[1] = new GenericBatchDigitalSignature(new ECDSASignature());
|
signers[1] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
|
||||||
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
||||||
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
||||||
|
@ -134,7 +129,9 @@ public class GenericBulletinBoardServerTest {
|
||||||
random = new Random(0);
|
random = new Random(0);
|
||||||
bulletinBoardMessageGenerator = new BulletinBoardMessageGenerator(random);
|
bulletinBoardMessageGenerator = new BulletinBoardMessageGenerator(random);
|
||||||
|
|
||||||
digest = new SHA256Digest();
|
digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
|
|
||||||
|
comparator = new BulletinBoardMessageComparator();
|
||||||
|
|
||||||
long end = threadBean.getCurrentThreadCpuTime();
|
long end = threadBean.getCurrentThreadCpuTime();
|
||||||
System.err.println("Finished initializing GenericBulletinBoardServerTest");
|
System.err.println("Finished initializing GenericBulletinBoardServerTest");
|
||||||
|
@ -142,7 +139,7 @@ public class GenericBulletinBoardServerTest {
|
||||||
|
|
||||||
// Initialize Batch variables
|
// Initialize Batch variables
|
||||||
|
|
||||||
completeBatches = new ArrayList<CompleteBatch>(10);
|
batches = new ArrayList<>(10);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -446,73 +443,48 @@ public class GenericBulletinBoardServerTest {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Tests that posting a message before opening a batch does not work
|
|
||||||
* @throws CommunicationException
|
|
||||||
*/
|
|
||||||
public void testBatchPostAfterClose() throws CommunicationException, SignatureException {
|
|
||||||
|
|
||||||
final int BATCH_ID = 100;
|
private void postAsBatch(BulletinBoardMessage message, int chunkSize, boolean close) throws CommunicationException {
|
||||||
|
|
||||||
|
List<BatchChunk> batchChunks = BulletinBoardUtils.breakToBatch(message, chunkSize);
|
||||||
|
BeginBatchMessage beginBatchMessage = BulletinBoardUtils.generateBeginBatchMessage(message);
|
||||||
|
|
||||||
CompleteBatch completeBatch = new CompleteBatch(Timestamp.newBuilder()
|
|
||||||
.setSeconds(978325)
|
|
||||||
.setNanos(8097234)
|
|
||||||
.build());
|
|
||||||
BoolValue result;
|
BoolValue result;
|
||||||
|
|
||||||
// Create data
|
|
||||||
|
|
||||||
completeBatch.setBeginBatchMessage(BeginBatchMessage.newBuilder()
|
|
||||||
.setSignerId(signerIDs[1])
|
|
||||||
.setBatchId(BATCH_ID)
|
|
||||||
.addTag("Test")
|
|
||||||
.build());
|
|
||||||
|
|
||||||
BatchData batchData = BatchData.newBuilder()
|
|
||||||
.setData(ByteString.copyFrom((new byte[] {1,2,3,4})))
|
|
||||||
.build();
|
|
||||||
|
|
||||||
completeBatch.appendBatchData(batchData);
|
|
||||||
|
|
||||||
signers[1].updateContent(completeBatch);
|
|
||||||
|
|
||||||
completeBatch.setSignature(signers[1].sign());
|
|
||||||
|
|
||||||
// Begin batch
|
// Begin batch
|
||||||
|
|
||||||
result = bulletinBoardServer.beginBatch(completeBatch.getBeginBatchMessage());
|
Int64Value batchId = bulletinBoardServer.beginBatch(beginBatchMessage);
|
||||||
|
|
||||||
assertThat("Was not able to open batch", result.getValue(), is(true));
|
assertThat("Was not able to open batch", batchId.getValue() != -1);
|
||||||
|
|
||||||
// Post data
|
// Post data
|
||||||
|
|
||||||
BatchMessage batchMessage = BatchMessage.newBuilder()
|
BatchMessage batchMessage = BatchMessage.getDefaultInstance();
|
||||||
.setSignerId(signerIDs[1])
|
|
||||||
.setBatchId(BATCH_ID)
|
for (int i = 0 ; i < batchChunks.size() ; i++){
|
||||||
.setData(batchData)
|
|
||||||
|
batchMessage = BatchMessage.newBuilder()
|
||||||
|
.setBatchId(batchId.getValue())
|
||||||
|
.setSerialNum(i)
|
||||||
|
.setData(batchChunks.get(i))
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
result = bulletinBoardServer.postBatchMessage(batchMessage);
|
result = bulletinBoardServer.postBatchMessage(batchMessage);
|
||||||
|
|
||||||
assertThat("Was not able to post batch message", result.getValue(), is(true));
|
assertThat("Was not able to post batch message", result.getValue(), is(true));
|
||||||
|
|
||||||
// Close batch
|
}
|
||||||
|
|
||||||
result = bulletinBoardServer.closeBatchMessage(completeBatch.getCloseBatchMessage());
|
// Close batch
|
||||||
|
if (close) {
|
||||||
|
|
||||||
|
CloseBatchMessage closeBatchMessage = BulletinBoardUtils.generateCloseBatchMessage(batchId, batchChunks.size(), message);
|
||||||
|
|
||||||
|
result = bulletinBoardServer.closeBatch(closeBatchMessage);
|
||||||
|
|
||||||
assertThat("Was not able to close batch", result.getValue(), is(true));
|
assertThat("Was not able to close batch", result.getValue(), is(true));
|
||||||
|
|
||||||
// Attempt to open batch again
|
}
|
||||||
|
|
||||||
result = bulletinBoardServer.beginBatch(completeBatch.getBeginBatchMessage());
|
|
||||||
|
|
||||||
assertThat("Was able to open a closed batch", result.getValue(), is(false));
|
|
||||||
|
|
||||||
// Attempt to add batch data
|
|
||||||
|
|
||||||
result = bulletinBoardServer.postBatchMessage(batchMessage);
|
|
||||||
|
|
||||||
assertThat("Was able to change a closed batch", result.getValue(), is(false));
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -522,110 +494,79 @@ public class GenericBulletinBoardServerTest {
|
||||||
*/
|
*/
|
||||||
public void testPostBatch() throws CommunicationException, SignatureException {
|
public void testPostBatch() throws CommunicationException, SignatureException {
|
||||||
|
|
||||||
CompleteBatch completeBatch = new CompleteBatch(Timestamp.newBuilder()
|
// Create data
|
||||||
.setSeconds(12345)
|
final int BATCH_ID = 200;
|
||||||
.setNanos(1111)
|
final int DATA_SIZE = 10000;
|
||||||
.build());
|
final int CHUNK_SIZE = 100;
|
||||||
int currentBatch = completeBatches.size();
|
final int TAG_NUMBER = 10;
|
||||||
|
|
||||||
BoolValue result;
|
Timestamp timestamp = Timestamp.newBuilder()
|
||||||
|
.setSeconds(5235000)
|
||||||
|
.setNanos(32541)
|
||||||
|
.build();
|
||||||
|
|
||||||
// Define batch data
|
BulletinBoardMessage batch = bulletinBoardMessageGenerator.generateRandomMessage(signers, timestamp, DATA_SIZE, TAG_NUMBER);
|
||||||
|
|
||||||
String[] tempBatchTags = new String[]{randomString(),randomString(),randomString()};
|
// Post batch
|
||||||
byte[][] tempBatchData = new byte[Math.abs(randomByte())][];
|
|
||||||
|
|
||||||
for (int i = 0 ; i < tempBatchData.length ; i++) {
|
postAsBatch(batch, CHUNK_SIZE, true);
|
||||||
|
|
||||||
tempBatchData[i] = new byte[Math.abs(randomByte())];
|
|
||||||
|
|
||||||
for (int j = 0; j < tempBatchData[i].length; j++) {
|
|
||||||
tempBatchData[i][j] = randomByte();
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Begin batch
|
|
||||||
|
|
||||||
completeBatch.setBeginBatchMessage(BeginBatchMessage.newBuilder()
|
|
||||||
.setSignerId(signerIDs[0])
|
|
||||||
.setBatchId(currentBatch)
|
|
||||||
.addAllTag(Arrays.asList(tempBatchTags))
|
|
||||||
.build());
|
|
||||||
|
|
||||||
result = bulletinBoardServer.beginBatch(completeBatch.getBeginBatchMessage());
|
|
||||||
|
|
||||||
assertThat("Could not begin batch " + currentBatch, result.getValue(), is(true));
|
|
||||||
|
|
||||||
// Add batch data and randomize data posting order
|
|
||||||
|
|
||||||
List<Integer> dataOrder = new ArrayList<Integer>(tempBatchData.length);
|
|
||||||
for (int i = 0 ; i < tempBatchData.length ; i++) {
|
|
||||||
dataOrder.add(i);
|
|
||||||
completeBatch.appendBatchData(BatchData.newBuilder()
|
|
||||||
.setData(ByteString.copyFrom(tempBatchData[i]))
|
|
||||||
.build());
|
|
||||||
}
|
|
||||||
Collections.shuffle(dataOrder);
|
|
||||||
|
|
||||||
// Post data
|
|
||||||
|
|
||||||
for (int i = 0 ; i < tempBatchData.length ; i++) {
|
|
||||||
|
|
||||||
int dataIndex = dataOrder.get(i);
|
|
||||||
|
|
||||||
result = bulletinBoardServer.postBatchMessage(BatchMessage.newBuilder()
|
|
||||||
.setSignerId(signerIDs[0])
|
|
||||||
.setBatchId(currentBatch)
|
|
||||||
.setSerialNum(dataIndex)
|
|
||||||
.setData(completeBatch.getBatchDataList().get(dataIndex))
|
|
||||||
.build());
|
|
||||||
|
|
||||||
assertThat("Could not post batch data for batch ID " + currentBatch + " serial number " + dataIndex,
|
|
||||||
result.getValue(), is(true));
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close batch
|
|
||||||
|
|
||||||
signers[0].updateContent(completeBatch);
|
|
||||||
completeBatch.setSignature(signers[0].sign());
|
|
||||||
|
|
||||||
result = bulletinBoardServer.closeBatchMessage(completeBatch.getCloseBatchMessage());
|
|
||||||
|
|
||||||
assertThat("Could not close batch " + currentBatch, result.getValue(), is(true));
|
|
||||||
|
|
||||||
// Update locally stored batches
|
// Update locally stored batches
|
||||||
completeBatches.add(completeBatch);
|
batches.add(batch);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public void testReadBatch() throws CommunicationException {
|
public void testReadBatch() throws CommunicationException {
|
||||||
|
|
||||||
for (CompleteBatch completeBatch : completeBatches) {
|
for (BulletinBoardMessage message : batches) {
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|
||||||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||||||
|
|
||||||
BatchSpecificationMessage batchSpecificationMessage =
|
digest.update(message);
|
||||||
BatchSpecificationMessage.newBuilder()
|
|
||||||
.setSignerId(completeBatch.getBeginBatchMessage().getSignerId())
|
MessageID msgId = digest.digestAsMessageID();
|
||||||
.setBatchId(completeBatch.getBeginBatchMessage().getBatchId())
|
|
||||||
|
MessageFilterList messageFilterList = MessageFilterList.newBuilder()
|
||||||
|
.addFilter(MessageFilter.newBuilder()
|
||||||
|
.setType(FilterType.MSG_ID)
|
||||||
|
.setId(msgId.getID())
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
bulletinBoardServer.readMessages(messageFilterList, new MessageOutputStream<BulletinBoardMessage>(outputStream));
|
||||||
|
|
||||||
|
MessageInputStream<BulletinBoardMessage> messageInputStream =
|
||||||
|
MessageInputStreamFactory.createMessageInputStream(new ByteArrayInputStream(
|
||||||
|
outputStream.toByteArray()),
|
||||||
|
BulletinBoardMessage.class);
|
||||||
|
|
||||||
|
List<BulletinBoardMessage> messageList = messageInputStream.asList();
|
||||||
|
|
||||||
|
assertThat("No stub found for message ID " + msgId.getID().toStringUtf8(), messageList.size() == 1);
|
||||||
|
|
||||||
|
BulletinBoardMessage stub = messageList.get(0);
|
||||||
|
|
||||||
|
BatchQuery batchQuery =
|
||||||
|
BatchQuery.newBuilder()
|
||||||
|
.setMsgID(msgId)
|
||||||
.setStartPosition(0)
|
.setStartPosition(0)
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
bulletinBoardServer.readBatch(batchSpecificationMessage, new MessageOutputStream<BatchData>(outputStream));
|
bulletinBoardServer.readBatch(batchQuery, new MessageOutputStream<BatchChunk>(outputStream));
|
||||||
|
|
||||||
MessageInputStream<BatchData> inputStream =
|
MessageInputStream<BatchChunk> batchInputStream =
|
||||||
MessageInputStreamFactory.createMessageInputStream(new ByteArrayInputStream(
|
MessageInputStreamFactory.createMessageInputStream(new ByteArrayInputStream(
|
||||||
outputStream.toByteArray()),
|
outputStream.toByteArray()),
|
||||||
BatchData.class);
|
BatchChunk.class);
|
||||||
|
|
||||||
List<BatchData> batchDataList = inputStream.asList();
|
List<BatchChunk> batchChunkList = batchInputStream.asList();
|
||||||
|
|
||||||
assertThat("Non-matching batch data for batch " + completeBatch.getBeginBatchMessage().getBatchId(),
|
BulletinBoardMessage retrievedMessage = BulletinBoardUtils.gatherBatch(stub, batchChunkList);
|
||||||
completeBatch.getBatchDataList().equals(batchDataList), is(true));
|
|
||||||
|
assertThat("Non-matching batch data for batch " + msgId.getID().toStringUtf8(),
|
||||||
|
comparator.compare(message, retrievedMessage) == 0);
|
||||||
|
|
||||||
} catch (IOException | NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
|
} catch (IOException | NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
|
||||||
assertThat("Error reading batch data list from input stream", false);
|
assertThat("Error reading batch data list from input stream", false);
|
||||||
|
|
|
@ -7,7 +7,6 @@ import meerkat.comm.CommunicationException;
|
||||||
import org.junit.After;
|
import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.junit.runner.Result;
|
|
||||||
|
|
||||||
import java.lang.management.ManagementFactory;
|
import java.lang.management.ManagementFactory;
|
||||||
import java.lang.management.ThreadMXBean;
|
import java.lang.management.ThreadMXBean;
|
||||||
|
@ -55,7 +54,7 @@ public class H2BulletinBoardServerTest {
|
||||||
|
|
||||||
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(queryProvider);
|
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(queryProvider);
|
||||||
try {
|
try {
|
||||||
bulletinBoardServer.init("");
|
bulletinBoardServer.init();
|
||||||
|
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
|
@ -107,16 +106,6 @@ public class H2BulletinBoardServerTest {
|
||||||
System.err.println("Time of operation: " + (end - start));
|
System.err.println("Time of operation: " + (end - start));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testBatchPostAfterClose() {
|
|
||||||
try{
|
|
||||||
serverTest.testBatchPostAfterClose();
|
|
||||||
} catch (Exception e) {
|
|
||||||
System.err.println(e.getMessage());
|
|
||||||
fail(e.getMessage());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testBatch() {
|
public void testBatch() {
|
||||||
|
|
||||||
|
|
|
@ -8,13 +8,9 @@ import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.lang.management.ManagementFactory;
|
import java.lang.management.ManagementFactory;
|
||||||
import java.lang.management.ThreadMXBean;
|
import java.lang.management.ThreadMXBean;
|
||||||
import java.lang.reflect.InvocationTargetException;
|
|
||||||
import java.security.SignatureException;
|
|
||||||
import java.sql.Connection;
|
import java.sql.Connection;
|
||||||
import java.sql.DriverManager;
|
|
||||||
import java.sql.SQLException;
|
import java.sql.SQLException;
|
||||||
import java.sql.Statement;
|
import java.sql.Statement;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
@ -62,7 +58,7 @@ public class MySQLBulletinBoardServerTest {
|
||||||
|
|
||||||
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(queryProvider);
|
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(queryProvider);
|
||||||
try {
|
try {
|
||||||
bulletinBoardServer.init("");
|
bulletinBoardServer.init();
|
||||||
|
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
|
@ -114,16 +110,6 @@ public class MySQLBulletinBoardServerTest {
|
||||||
System.err.println("Time of operation: " + (end - start));
|
System.err.println("Time of operation: " + (end - start));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testBatchPostAfterClose() {
|
|
||||||
try{
|
|
||||||
serverTest.testBatchPostAfterClose();
|
|
||||||
} catch (Exception e) {
|
|
||||||
System.err.println(e.getMessage());
|
|
||||||
fail(e.getMessage());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testBatch() {
|
public void testBatch() {
|
||||||
|
|
||||||
|
|
|
@ -39,7 +39,7 @@ public class SQLiteBulletinBoardServerTest{
|
||||||
|
|
||||||
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(new SQLiteQueryProvider(testFilename));
|
BulletinBoardServer bulletinBoardServer = new BulletinBoardSQLServer(new SQLiteQueryProvider(testFilename));
|
||||||
try {
|
try {
|
||||||
bulletinBoardServer.init("");
|
bulletinBoardServer.init();
|
||||||
|
|
||||||
} catch (CommunicationException e) {
|
} catch (CommunicationException e) {
|
||||||
System.err.println(e.getMessage());
|
System.err.println(e.getMessage());
|
||||||
|
@ -60,7 +60,7 @@ public class SQLiteBulletinBoardServerTest{
|
||||||
System.err.println("Time of operation: " + (end - start));
|
System.err.println("Time of operation: " + (end - start));
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
// @Test
|
||||||
public void bulkTest() {
|
public void bulkTest() {
|
||||||
System.err.println("Starting bulkTest of SQLiteBulletinBoardServerTest");
|
System.err.println("Starting bulkTest of SQLiteBulletinBoardServerTest");
|
||||||
long start = threadBean.getCurrentThreadCpuTime();
|
long start = threadBean.getCurrentThreadCpuTime();
|
||||||
|
@ -91,6 +91,29 @@ public class SQLiteBulletinBoardServerTest{
|
||||||
System.err.println("Time of operation: " + (end - start));
|
System.err.println("Time of operation: " + (end - start));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// @Test
|
||||||
|
public void testBatch() {
|
||||||
|
|
||||||
|
final int BATCH_NUM = 20;
|
||||||
|
|
||||||
|
try{
|
||||||
|
for (int i = 0 ; i < BATCH_NUM ; i++) {
|
||||||
|
serverTest.testPostBatch();
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
System.err.println(e.getMessage());
|
||||||
|
fail(e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
try{
|
||||||
|
serverTest.testReadBatch();
|
||||||
|
} catch (Exception e) {
|
||||||
|
System.err.println(e.getMessage());
|
||||||
|
fail(e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
@After
|
@After
|
||||||
public void close() {
|
public void close() {
|
||||||
System.err.println("Starting to close SQLiteBulletinBoardServerTest");
|
System.err.println("Starting to close SQLiteBulletinBoardServerTest");
|
||||||
|
|
Binary file not shown.
|
@ -1,6 +1,7 @@
|
||||||
#Fri Jan 29 21:00:29 IST 2016
|
#Tue Aug 05 03:26:05 IDT 2014
|
||||||
distributionBase=GRADLE_USER_HOME
|
distributionBase=GRADLE_USER_HOME
|
||||||
distributionPath=wrapper/dists
|
distributionPath=wrapper/dists
|
||||||
zipStoreBase=GRADLE_USER_HOME
|
zipStoreBase=GRADLE_USER_HOME
|
||||||
zipStorePath=wrapper/dists
|
zipStorePath=wrapper/dists
|
||||||
distributionUrl=https\://services.gradle.org/distributions/gradle-2.9-bin.zip
|
distributionUrl=https\://services.gradle.org/distributions/gradle-2.9-all.zip
|
||||||
|
distributionSha256Sum=4647967f8de78d6d6d8093cdac50f368f8c2b8038f41a5afe1c3bce4c69219a9
|
||||||
|
|
|
@ -42,6 +42,11 @@ case "`uname`" in
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
|
|
||||||
|
# For Cygwin, ensure paths are in UNIX format before anything is touched.
|
||||||
|
if $cygwin ; then
|
||||||
|
[ -n "$JAVA_HOME" ] && JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
|
||||||
|
fi
|
||||||
|
|
||||||
# Attempt to set APP_HOME
|
# Attempt to set APP_HOME
|
||||||
# Resolve links: $0 may be a link
|
# Resolve links: $0 may be a link
|
||||||
PRG="$0"
|
PRG="$0"
|
||||||
|
@ -56,9 +61,9 @@ while [ -h "$PRG" ] ; do
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
SAVED="`pwd`"
|
SAVED="`pwd`"
|
||||||
cd "`dirname \"$PRG\"`/" >/dev/null
|
cd "`dirname \"$PRG\"`/" >&-
|
||||||
APP_HOME="`pwd -P`"
|
APP_HOME="`pwd -P`"
|
||||||
cd "$SAVED" >/dev/null
|
cd "$SAVED" >&-
|
||||||
|
|
||||||
CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
|
CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
|
||||||
|
|
||||||
|
@ -109,7 +114,6 @@ fi
|
||||||
if $cygwin ; then
|
if $cygwin ; then
|
||||||
APP_HOME=`cygpath --path --mixed "$APP_HOME"`
|
APP_HOME=`cygpath --path --mixed "$APP_HOME"`
|
||||||
CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
|
CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
|
||||||
JAVACMD=`cygpath --unix "$JAVACMD"`
|
|
||||||
|
|
||||||
# We build the pattern for arguments to be converted via cygpath
|
# We build the pattern for arguments to be converted via cygpath
|
||||||
ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
|
ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
|
||||||
|
|
|
@ -1,8 +1,9 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.common.util.concurrent.FutureCallback;
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.Timestamp;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.Crypto.Signature;
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -13,6 +14,7 @@ public interface AsyncBulletinBoardClient extends BulletinBoardClient {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Post a message to the bulletin board in an asynchronous manner
|
* Post a message to the bulletin board in an asynchronous manner
|
||||||
|
* The message may be broken up by the client into a batch message, depending on implementation
|
||||||
* @param msg is the message to be posted
|
* @param msg is the message to be posted
|
||||||
* @param callback is a class containing methods to handle the result of the operation
|
* @param callback is a class containing methods to handle the result of the operation
|
||||||
* @return a unique message ID for the message, that can be later used to retrieve the batch
|
* @return a unique message ID for the message, that can be later used to retrieve the batch
|
||||||
|
@ -20,56 +22,58 @@ public interface AsyncBulletinBoardClient extends BulletinBoardClient {
|
||||||
public MessageID postMessage(BulletinBoardMessage msg, FutureCallback<Boolean> callback);
|
public MessageID postMessage(BulletinBoardMessage msg, FutureCallback<Boolean> callback);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Perform an end-to-end post of a signed batch message
|
* Perform an end-to-end post of a message in batch form
|
||||||
* @param completeBatch contains all the data of the batch including the meta-data and the signature
|
* @param completeBatch contains all the data of the batch
|
||||||
|
* @param chunkSize is the maximum size of each chunk of the message in bytes
|
||||||
* @param callback is a class containing methods to handle the result of the operation
|
* @param callback is a class containing methods to handle the result of the operation
|
||||||
* @return a unique identifier for the batch message
|
* @return a unique identifier for the batch message
|
||||||
*/
|
*/
|
||||||
public MessageID postBatch(CompleteBatch completeBatch, FutureCallback<Boolean> callback);
|
public MessageID postAsBatch(BulletinBoardMessage completeBatch, int chunkSize, FutureCallback<Boolean> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An interface for returning an opaque identifier for a batch message
|
||||||
|
* This identifier is used to uniquely identify the batch until it is completely posted and signed
|
||||||
|
* After the batch is fully posted: it is identified by its digest (like any message)
|
||||||
|
* This can be implementation-specific (and not necessarily interchangeable between different implementations)
|
||||||
|
*/
|
||||||
|
public interface BatchIdentifier {}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This message informs the server about the existence of a new batch message and supplies it with the tags associated with it
|
* This message informs the server about the existence of a new batch message and supplies it with the tags associated with it
|
||||||
* @param beginBatchMessage contains the data required to begin the batch
|
* @param tags contains the tags used in the batch
|
||||||
* @param callback is a callback function class for handling results of the operation
|
* @param callback is a callback function class for handling results of the operation
|
||||||
|
* it receives a BatchIdentifier for use in subsequent batch post operations
|
||||||
*/
|
*/
|
||||||
public void beginBatch(BeginBatchMessage beginBatchMessage, FutureCallback<Boolean> callback);
|
public void beginBatch(Iterable<String> tags, FutureCallback<BatchIdentifier> callback);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* This method posts batch data into an (assumed to be open) batch
|
* This method posts batch data into an (assumed to be open) batch
|
||||||
* It does not close the batch
|
* It does not close the batch
|
||||||
* @param signerId is the canonical form for the ID of the sender of this batch
|
* @param batchIdentifier is the temporary batch identifier
|
||||||
* @param batchId is a unique (per signer) ID for this batch
|
* @param batchChunkList is the (canonically ordered) list of data comprising the portion of the batch to be posted
|
||||||
* @param batchDataList is the (canonically ordered) list of data comprising the portion of the batch to be posted
|
|
||||||
* @param startPosition is the location (in the batch) of the first entry in batchDataList
|
* @param startPosition is the location (in the batch) of the first entry in batchDataList
|
||||||
* (optionally used to continue interrupted post operations)
|
* (optionally used to continue interrupted post operations)
|
||||||
* The first position in the batch is position 0
|
* The first position in the batch is position 0
|
||||||
* @param callback is a callback function class for handling results of the operation
|
* @param callback is a callback function class for handling results of the operation
|
||||||
|
* @throws IllegalArgumentException if the batch identifier given was of an illegal format
|
||||||
*/
|
*/
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList,
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList,
|
||||||
int startPosition, FutureCallback<Boolean> callback);
|
int startPosition, FutureCallback<Boolean> callback) throws IllegalArgumentException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Overloading of the postBatchData method which starts at the first position in the batch
|
* Overloading of the postBatchData method which starts at the first position in the batch
|
||||||
*/
|
*/
|
||||||
public void postBatchData(byte[] signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback);
|
public void postBatchData(BatchIdentifier batchIdentifier, List<BatchChunk> batchChunkList, FutureCallback<Boolean> callback)
|
||||||
|
throws IllegalArgumentException;
|
||||||
/**
|
|
||||||
* Overloading of the postBatchData method which uses ByteString
|
|
||||||
*/
|
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList,
|
|
||||||
int startPosition, FutureCallback<Boolean> callback);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Overloading of the postBatchData method which uses ByteString and starts at the first position in the batch
|
|
||||||
*/
|
|
||||||
public void postBatchData(ByteString signerId, int batchId, List<BatchData> batchDataList, FutureCallback<Boolean> callback);
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Attempts to close a batch message
|
* Attempts to close a batch message
|
||||||
* @param closeBatchMessage contains the data required to close the batch
|
* @param batchIdentifier is the temporary batch identifier
|
||||||
* @param callback is a callback function class for handling results of the operation
|
* @param callback is a callback function class for handling results of the operation
|
||||||
|
* @throws IllegalArgumentException if the batch identifier given was of an illegal format
|
||||||
*/
|
*/
|
||||||
public void closeBatch(CloseBatchMessage closeBatchMessage, FutureCallback<Boolean> callback);
|
public void closeBatch(BatchIdentifier batchIdentifier, Timestamp timestamp, Iterable<Signature> signatures, FutureCallback<Boolean> callback)
|
||||||
|
throws IllegalArgumentException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Check how "safe" a given message is in an asynchronous manner
|
* Check how "safe" a given message is in an asynchronous manner
|
||||||
|
@ -83,18 +87,29 @@ public interface AsyncBulletinBoardClient extends BulletinBoardClient {
|
||||||
* Read all messages posted matching the given filter in an asynchronous manner
|
* Read all messages posted matching the given filter in an asynchronous manner
|
||||||
* Note that if messages haven't been "fully posted", this might return a different
|
* Note that if messages haven't been "fully posted", this might return a different
|
||||||
* set of messages in different calls. However, messages that are fully posted
|
* set of messages in different calls. However, messages that are fully posted
|
||||||
* are guaranteed to be included.
|
* are guaranteed to be included
|
||||||
* @param filterList return only messages that match the filters (null means no filtering).
|
* Also: batch messages are returned as stubs.
|
||||||
|
* @param filterList return only messages that match the filters (null means no filtering)
|
||||||
* @param callback is a callback function class for handling results of the operation
|
* @param callback is a callback function class for handling results of the operation
|
||||||
*/
|
*/
|
||||||
public void readMessages(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback);
|
public void readMessages(MessageFilterList filterList, FutureCallback<List<BulletinBoardMessage>> callback);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Read a given batch message from the bulletin board
|
* Read a given message from the bulletin board
|
||||||
* @param batchSpecificationMessage contains the data required to specify a single batch instance
|
* If the message is a batch: returns a complete message containing the batch data as well as the metadata
|
||||||
|
* @param msgID is the ID of the message to be read
|
||||||
* @param callback is a callback class for handling the result of the operation
|
* @param callback is a callback class for handling the result of the operation
|
||||||
*/
|
*/
|
||||||
public void readBatch(BatchSpecificationMessage batchSpecificationMessage, FutureCallback<CompleteBatch> callback);
|
public void readMessage(MessageID msgID, FutureCallback<BulletinBoardMessage> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read batch data for a specific stub message
|
||||||
|
* @param stub is a batch message stub
|
||||||
|
* @param callback is a callback class for handling the result of the operation
|
||||||
|
* @return a new BulletinBoardMessage containing both metadata from the stub and actual data from the server
|
||||||
|
* @throws IllegalArgumentException if the received message is not a stub
|
||||||
|
*/
|
||||||
|
public void readBatchData(BulletinBoardMessage stub, FutureCallback<BulletinBoardMessage> callback) throws IllegalArgumentException;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -1,20 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
import meerkat.crypto.Digest;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 18-Dec-15.
|
|
||||||
* Extends the Digest interface with a method for digesting Batch messages
|
|
||||||
*/
|
|
||||||
public interface BatchDigest extends Digest {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Update the digest with the batch message data (ignore the signature)
|
|
||||||
* @param completeBatch is the batch message that needs to be digested
|
|
||||||
*/
|
|
||||||
public void update(CompleteBatch completeBatch);
|
|
||||||
|
|
||||||
}
|
|
|
@ -1,34 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
import meerkat.crypto.DigitalSignature;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BeginBatchMessage;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
|
||||||
import meerkat.protobuf.Crypto.Signature;
|
|
||||||
|
|
||||||
import java.security.InvalidKeyException;
|
|
||||||
import java.security.SignatureException;
|
|
||||||
import java.security.cert.CertificateException;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 20-Dec-15.
|
|
||||||
* Extends the DigitalSignature interface with methods for signing and authenticating Batch messages
|
|
||||||
*/
|
|
||||||
public interface BatchDigitalSignature extends DigitalSignature {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Appends the batch data to the signed content (ignoring the signature)
|
|
||||||
* @param completeBatch contains all the data about the batch
|
|
||||||
* @throws SignatureException
|
|
||||||
*/
|
|
||||||
public void updateContent(CompleteBatch completeBatch) throws SignatureException;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Performs a complete verification process on the given batch message
|
|
||||||
* @param completeBatch contains the batch data as well as the signature
|
|
||||||
* @return TRUE if the batch is verified and FALSE otherwise
|
|
||||||
* @throws SignatureException | SignatureException | InvalidKeyException when underlying methods do so
|
|
||||||
*/
|
|
||||||
public boolean verify(CompleteBatch completeBatch) throws SignatureException, CertificateException, InvalidKeyException;
|
|
||||||
|
|
||||||
}
|
|
|
@ -5,7 +5,6 @@ import meerkat.protobuf.Voting.*;
|
||||||
|
|
||||||
import static meerkat.protobuf.BulletinBoardAPI.*;
|
import static meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
import java.util.Collection;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -21,6 +20,7 @@ public interface BulletinBoardClient {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Post a message to the bulletin board in a synchronous manner
|
* Post a message to the bulletin board in a synchronous manner
|
||||||
|
* The message may be broken up by the client into a batch message depending on implementation
|
||||||
* @param msg is the message to be posted
|
* @param msg is the message to be posted
|
||||||
* @return a unique message ID for the message, that can be later used to retrieve the batch
|
* @return a unique message ID for the message, that can be later used to retrieve the batch
|
||||||
* @throws CommunicationException
|
* @throws CommunicationException
|
||||||
|
@ -31,28 +31,57 @@ public interface BulletinBoardClient {
|
||||||
* Check how "safe" a given message is in a synchronous manner
|
* Check how "safe" a given message is in a synchronous manner
|
||||||
* @param id is the unique message identifier for retrieval
|
* @param id is the unique message identifier for retrieval
|
||||||
* @return a normalized "redundancy score" from 0 (local only) to 1 (fully published)
|
* @return a normalized "redundancy score" from 0 (local only) to 1 (fully published)
|
||||||
|
* @throws CommunicationException
|
||||||
*/
|
*/
|
||||||
float getRedundancy(MessageID id);
|
float getRedundancy(MessageID id) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Read all messages posted matching the given filter in a synchronous manner
|
* Read all messages posted matching the given filter in a synchronous manner
|
||||||
* Note that if messages haven't been "fully posted", this might return a different
|
* Note that if messages haven't been "fully posted", this might return a different
|
||||||
* set of messages in different calls. However, messages that are fully posted
|
* set of messages in different calls. However, messages that are fully posted
|
||||||
* are guaranteed to be included.
|
* are guaranteed to be included.
|
||||||
|
* Also: batch messages are returned as stubs.
|
||||||
* @param filterList return only messages that match the filters (null means no filtering)
|
* @param filterList return only messages that match the filters (null means no filtering)
|
||||||
* @return the list of messages
|
* @return the list of messages
|
||||||
*/
|
*/
|
||||||
List<BulletinBoardMessage> readMessages(MessageFilterList filterList);
|
List<BulletinBoardMessage> readMessages(MessageFilterList filterList) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Breaks up a bulletin board message into chunks and posts it as a batch message
|
||||||
|
* @param msg is the message to post
|
||||||
|
* @param chunkSize is the maximal chunk size in bytes
|
||||||
|
* @return the unique message ID
|
||||||
|
* @throws CommunicationException if operation is unsuccessful
|
||||||
|
*/
|
||||||
|
MessageID postAsBatch(BulletinBoardMessage msg, int chunkSize) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read a given message from the bulletin board
|
||||||
|
* If the message is a batch: returns a complete message containing the batch data as well as the metadata
|
||||||
|
* @param msgID is the ID of the message to be read
|
||||||
|
* @return the complete message
|
||||||
|
* @throws CommunicationException if operation is unsuccessful
|
||||||
|
*/
|
||||||
|
BulletinBoardMessage readMessage(MessageID msgID) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read batch data for a specific stub message
|
||||||
|
* @param stub is a batch message stub
|
||||||
|
* @return a new BulletinBoardMessage containing both metadata from the stub and actual data from the server
|
||||||
|
* @throws CommunicationException if operation is unsuccessful
|
||||||
|
* @throws IllegalArgumentException if the received message is not a stub
|
||||||
|
*/
|
||||||
|
BulletinBoardMessage readBatchData(BulletinBoardMessage stub) throws CommunicationException, IllegalArgumentException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create a SyncQuery to test against that corresponds with the current server state for a specific filter list
|
* Create a SyncQuery to test against that corresponds with the current server state for a specific filter list
|
||||||
* Should only be called on instances for which the actual server contacted is known (i.e. there is only one server)
|
* Should only be called on instances for which the actual server contacted is known (i.e. there is only one server)
|
||||||
* @param GenerateSyncQueryParams defines the required information needed to generate the query
|
* @param generateSyncQueryParams defines the required information needed to generate the query
|
||||||
* These are represented as fractions of the total number of relevant messages
|
* These are represented as fractions of the total number of relevant messages
|
||||||
* @return The generated SyncQuery
|
* @return The generated SyncQuery
|
||||||
* @throws CommunicationException when no DB can be contacted
|
* @throws CommunicationException when no DB can be contacted
|
||||||
*/
|
*/
|
||||||
SyncQuery generateSyncQuery(GenerateSyncQueryParams GenerateSyncQueryParams) throws CommunicationException;
|
SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Closes all connections, if any.
|
* Closes all connections, if any.
|
||||||
|
|
|
@ -10,6 +10,7 @@ public interface BulletinBoardConstants {
|
||||||
public static final String BULLETIN_BOARD_SERVER_PATH = "/bbserver";
|
public static final String BULLETIN_BOARD_SERVER_PATH = "/bbserver";
|
||||||
public static final String GENERATE_SYNC_QUERY_PATH = "/generatesyncquery";
|
public static final String GENERATE_SYNC_QUERY_PATH = "/generatesyncquery";
|
||||||
public static final String READ_MESSAGES_PATH = "/readmessages";
|
public static final String READ_MESSAGES_PATH = "/readmessages";
|
||||||
|
public static final String COUNT_MESSAGES_PATH = "/countmessages";
|
||||||
public static final String READ_BATCH_PATH = "/readbatch";
|
public static final String READ_BATCH_PATH = "/readbatch";
|
||||||
public static final String POST_MESSAGE_PATH = "/postmessage";
|
public static final String POST_MESSAGE_PATH = "/postmessage";
|
||||||
public static final String BEGIN_BATCH_PATH = "/beginbatch";
|
public static final String BEGIN_BATCH_PATH = "/beginbatch";
|
||||||
|
@ -17,9 +18,4 @@ public interface BulletinBoardConstants {
|
||||||
public static final String CLOSE_BATCH_PATH = "/closebatch";
|
public static final String CLOSE_BATCH_PATH = "/closebatch";
|
||||||
public static final String SYNC_QUERY_PATH = "/syncquery";
|
public static final String SYNC_QUERY_PATH = "/syncquery";
|
||||||
|
|
||||||
// Other Constants
|
|
||||||
|
|
||||||
public static final String BATCH_TAG = "@BATCH";
|
|
||||||
public static final String BATCH_ID_TAG_PREFIX = "BATCHID#";
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,30 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.crypto.Digest;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 18-Dec-15.
|
||||||
|
* Extends the Digest interface with methods for digesting Bulletin Board messages
|
||||||
|
*/
|
||||||
|
public interface BulletinBoardDigest extends Digest {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Update the digest with the message data (ignore the signature)
|
||||||
|
* The digest only uses the part the signatures are computed on for this operation
|
||||||
|
* If the message is a stub: this should be called before digesting the raw data
|
||||||
|
* @param msg is the message that needs to be digested
|
||||||
|
*/
|
||||||
|
public void update(BulletinBoardMessage msg);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Update the digest with the message data (ignore the signature)
|
||||||
|
* The digest only uses the part the signatures are computed on for this operation
|
||||||
|
* If the message is a stub: this should be called before digesting the raw data
|
||||||
|
* @param msg is the message that needs to be digested
|
||||||
|
*/
|
||||||
|
public void update(UnsignedBulletinBoardMessage msg);
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,49 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 13-Apr-16.
|
||||||
|
* This interface is meant to extend a BulletinBoardClient interface/class
|
||||||
|
* It provides it with the ability to delete messages from the Server
|
||||||
|
* This assumes the Server implements the {@link DeletableBulletinBoardServer}
|
||||||
|
*/
|
||||||
|
public interface BulletinBoardMessageDeleter {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from a Bulletin Board Server in a possibly asynchronous manner
|
||||||
|
* Logs this action
|
||||||
|
* @param msgID is the ID of the message to delete
|
||||||
|
* @param callback handles the result of the operation
|
||||||
|
*/
|
||||||
|
public void deleteMessage(MessageID msgID, FutureCallback<Boolean> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from the Bulletin Board in a possibly asynchronous manner
|
||||||
|
* Logs this action
|
||||||
|
* @param entryNum is the serial entry number of the message to delete
|
||||||
|
* @param callback handles the result of the operation
|
||||||
|
*/
|
||||||
|
public void deleteMessage(long entryNum, FutureCallback<Boolean> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from a Bulletin Board Server in a synchronous manner
|
||||||
|
* Logs this action
|
||||||
|
* @param msgID is the ID of the message to delete
|
||||||
|
* @return TRUE if the message was deleted and FALSE if it did not exist on the server
|
||||||
|
* @throws CommunicationException when an error occurs
|
||||||
|
*/
|
||||||
|
public boolean deleteMessage(MessageID msgID) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from the Bulletin Board in a synchronous manner
|
||||||
|
* Logs this action
|
||||||
|
* @param entryNum is the serial entry number of the message to delete
|
||||||
|
* @return TRUE if the message was deleted and FALSE if it did not exist on the server
|
||||||
|
* @throws CommunicationException when an error occurs
|
||||||
|
*/
|
||||||
|
public boolean deleteMessage(long entryNum) throws CommunicationException;
|
||||||
|
|
||||||
|
}
|
|
@ -1,12 +1,12 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
import com.google.protobuf.BoolValue;
|
import com.google.protobuf.BoolValue;
|
||||||
|
import com.google.protobuf.Int32Value;
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
import meerkat.comm.CommunicationException;
|
import meerkat.comm.CommunicationException;
|
||||||
import meerkat.comm.MessageOutputStream;
|
import meerkat.comm.MessageOutputStream;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
import java.util.Collection;
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel on 07/11/15.
|
* Created by Arbel on 07/11/15.
|
||||||
|
@ -22,7 +22,7 @@ public interface BulletinBoardServer{
|
||||||
* It also establishes the connection to the DB
|
* It also establishes the connection to the DB
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
*/
|
*/
|
||||||
public void init(String meerkatDB) throws CommunicationException;
|
public void init() throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Post a message to bulletin board.
|
* Post a message to bulletin board.
|
||||||
|
@ -33,7 +33,7 @@ public interface BulletinBoardServer{
|
||||||
public BoolValue postMessage(BulletinBoardMessage msg) throws CommunicationException;
|
public BoolValue postMessage(BulletinBoardMessage msg) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Read all messages posted matching the given filter
|
* Read all posted messages matching the given filters
|
||||||
* @param filterList return only messages that match the filters (empty list or null means no filtering)
|
* @param filterList return only messages that match the filters (empty list or null means no filtering)
|
||||||
* @param out is an output stream into which the matching messages are written
|
* @param out is an output stream into which the matching messages are written
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
|
@ -41,17 +41,23 @@ public interface BulletinBoardServer{
|
||||||
public void readMessages(MessageFilterList filterList, MessageOutputStream<BulletinBoardMessage> out) throws CommunicationException;
|
public void readMessages(MessageFilterList filterList, MessageOutputStream<BulletinBoardMessage> out) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Informs server about a new batch message
|
* Return the number of posted messages matching the given filters
|
||||||
* @param message contains the required data about the new batch
|
* @param filterList count only messages that match the filters (empty list or null means no filtering)
|
||||||
* @return TRUE if the batch request is accepted amd FALSE otherwise
|
* @return an IntMsg containing the number of messages that match the filter
|
||||||
* Specifically, if such a batch already exists and is not yet closed: the value returned will be TRUE
|
|
||||||
* However, if such a batch exists and is already closed: the value returned will be FALSE
|
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
*/
|
*/
|
||||||
public BoolValue beginBatch(BeginBatchMessage message) throws CommunicationException;
|
public Int32Value getMessageCount(MessageFilterList filterList) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Posts a (part of a) batch message to the bulletin board
|
* Informs server about a new batch message
|
||||||
|
* @param message contains the required data about the new batch
|
||||||
|
* @return a unique batch identifier for the new batch ; -1 if batch creation was unsuccessful
|
||||||
|
* @throws CommunicationException on DB connection error
|
||||||
|
*/
|
||||||
|
public Int64Value beginBatch(BeginBatchMessage message) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Posts a chunk of a batch message to the bulletin board
|
||||||
* Note that the existence and contents of a batch message are not available for reading before the batch is finalized
|
* Note that the existence and contents of a batch message are not available for reading before the batch is finalized
|
||||||
* @param batchMessage contains the (partial) data this message carries as well as meta-data required in order to place the data
|
* @param batchMessage contains the (partial) data this message carries as well as meta-data required in order to place the data
|
||||||
* in the correct position inside the correct batch
|
* in the correct position inside the correct batch
|
||||||
|
@ -63,22 +69,22 @@ public interface BulletinBoardServer{
|
||||||
public BoolValue postBatchMessage(BatchMessage batchMessage) throws CommunicationException;
|
public BoolValue postBatchMessage(BatchMessage batchMessage) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Attempts to stop and finalize a batch message
|
* Attempts to close and finalize a batch message
|
||||||
* @param message contains the data necessary to stop the batch; in particular: the signature for the batch
|
* @param message contains the data necessary to close the batch; in particular: the signature for the batch
|
||||||
* @return TRUE if the batch was successfully closed, FALSE otherwise
|
* @return TRUE if the batch was successfully closed, FALSE otherwise
|
||||||
* Specifically, if the signature is invalid or if some of the batch parts have not yet been submitted: the value returned will be FALSE
|
* Specifically, if the signature is invalid or if some of the batch parts have not yet been submitted: the value returned will be FALSE
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
*/
|
*/
|
||||||
public BoolValue closeBatchMessage(CloseBatchMessage message) throws CommunicationException;
|
public BoolValue closeBatch(CloseBatchMessage message) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Reads a batch message from the server (starting with the supplied position)
|
* Reads a batch message from the server (starting with the supplied position)
|
||||||
* @param message specifies the signer ID and the batch ID to read as well as an (optional) start position
|
* @param batchQuery specifies which batch and what parts of it to retrieve
|
||||||
* @param out is a stream of the ordered batch messages starting from the specified start position (if given) or from the beginning (if omitted)
|
* @param out is a stream of the ordered batch messages starting from the specified start position (if given) or from the beginning (if omitted)
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
* @throws IllegalArgumentException if message does not specify a batch
|
* @throws IllegalArgumentException if message ID does not specify a batch
|
||||||
*/
|
*/
|
||||||
public void readBatch(BatchSpecificationMessage message, MessageOutputStream<BatchData> out) throws CommunicationException, IllegalArgumentException;
|
public void readBatch(BatchQuery batchQuery, MessageOutputStream<BatchChunk> out) throws CommunicationException, IllegalArgumentException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create a SyncQuery to test against that corresponds with the current server state for a specific filter list
|
* Create a SyncQuery to test against that corresponds with the current server state for a specific filter list
|
||||||
|
@ -86,7 +92,7 @@ public interface BulletinBoardServer{
|
||||||
* @return The generated SyncQuery
|
* @return The generated SyncQuery
|
||||||
* @throws CommunicationException on DB connection error
|
* @throws CommunicationException on DB connection error
|
||||||
*/
|
*/
|
||||||
SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException;
|
public SyncQuery generateSyncQuery(GenerateSyncQueryParams generateSyncQueryParams) throws CommunicationException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Queries the database for sync status with respect to a given sync query
|
* Queries the database for sync status with respect to a given sync query
|
||||||
|
|
|
@ -0,0 +1,31 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.crypto.DigitalSignature;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BulletinBoardMessage;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.UnsignedBulletinBoardMessage;
|
||||||
|
import meerkat.protobuf.Crypto;
|
||||||
|
|
||||||
|
import java.security.SignatureException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 18-Dec-15.
|
||||||
|
* Extends the DigitalSignature interface with methods for signing Bulletin Board messages
|
||||||
|
*/
|
||||||
|
public interface BulletinBoardSignature extends DigitalSignature {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add msg to the content stream to be verified / signed
|
||||||
|
* The digest only uses the part the signatures are computed on for this operation
|
||||||
|
* If the message is a stub: this should be called before updating with the raw data
|
||||||
|
* @param msg is the message that needs to be digested
|
||||||
|
*/
|
||||||
|
public void updateContent(BulletinBoardMessage msg) throws SignatureException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add msg to the content stream to be verified / signed
|
||||||
|
* If the message is a stub: this should be called before updating with the raw data
|
||||||
|
* @param msg is the message that needs to be digested
|
||||||
|
*/
|
||||||
|
public void updateContent(UnsignedBulletinBoardMessage msg) throws SignatureException;
|
||||||
|
|
||||||
|
}
|
|
@ -1,22 +1,84 @@
|
||||||
package meerkat.bulletinboard;
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
import com.google.common.util.concurrent.FutureCallback;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by Arbel Deutsch Peled on 08-Mar-16.
|
* Created by Arbel Deutsch Peled on 08-Mar-16.
|
||||||
* This interface defines the behaviour of a bulletin board synchronizer
|
* This interface defines the behaviour of a bulletin board synchronizer
|
||||||
* This is used to make sure that data in a specific instance of a bulletin board server is duplicated to a sufficient percentage of the other servers
|
* This is used to make sure that data in a specific instance of a bulletin board server is duplicated to a sufficient percentage of the other servers
|
||||||
*/
|
*/
|
||||||
public interface BulletinBoardSynchronizer extends Runnable{
|
public interface BulletinBoardSynchronizer extends Runnable {
|
||||||
|
|
||||||
|
public enum SyncStatus{
|
||||||
|
SYNCHRONIZED, // No more messages to upload
|
||||||
|
PENDING, // Synchronizer is querying for data to upload and uploading it as needed
|
||||||
|
SERVER_ERROR, // Synchronizer encountered an error while uploading, but will retry
|
||||||
|
STOPPED // Stopped/Not started by user
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
*
|
* Initializes the synchronizer with the required data to function properly
|
||||||
* @param localClient is a client for the local DB instance
|
* @param localClient is a client for the temporary local storage server which contains only data to be uploaded
|
||||||
* @param remoteClient is a client for the remote DBs
|
* @param remoteClient is a client for the remote servers into which the data needs to be uploaded
|
||||||
* @param minRedundancy
|
|
||||||
*/
|
*/
|
||||||
public void init(BulletinBoardClient localClient, AsyncBulletinBoardClient remoteClient, float minRedundancy);
|
public void init(DeletableSubscriptionBulletinBoardClient localClient, AsyncBulletinBoardClient remoteClient);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the current server synchronization status
|
||||||
|
* @return the current synchronization status
|
||||||
|
*/
|
||||||
|
public SyncStatus getSyncStatus();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a subscription to sync status changes
|
||||||
|
* @param callback is the handler for any status changes
|
||||||
|
*/
|
||||||
|
public void subscribeToSyncStatus(FutureCallback<SyncStatus> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the messages which have not yet been synchronized
|
||||||
|
* @return the list of messages remaining to be synchronized
|
||||||
|
*/
|
||||||
|
public List<BulletinBoardMessage> getRemainingMessages() throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Asynchronously returns the messages which have not yet been synchronized
|
||||||
|
* @param callback is the handler for the list of messages
|
||||||
|
*/
|
||||||
|
public void getRemainingMessages(FutureCallback<List<BulletinBoardMessage>> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the current number of unsynchronized messages
|
||||||
|
* @return the current synchronization status
|
||||||
|
*/
|
||||||
|
public long getRemainingMessagesCount() throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a subscription to changes in the number of unsynchronized messages
|
||||||
|
* @param callback is the handler for any status changes
|
||||||
|
*/
|
||||||
|
public void subscribeToRemainingMessagesCount(FutureCallback<Integer> callback);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Starts the synchronization
|
||||||
|
*/
|
||||||
@Override
|
@Override
|
||||||
public void run();
|
public void run();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Lets the Synchronizer know that there is new data to be uploaded
|
||||||
|
* This is used to reduce the latency between local data-writes and uploads to the remote servers
|
||||||
|
*/
|
||||||
|
public void nudge();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Stops the synchronization
|
||||||
|
*/
|
||||||
|
public void stop();
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,150 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
import com.google.protobuf.Timestamp;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
|
||||||
import meerkat.protobuf.Crypto.*;
|
|
||||||
import meerkat.util.BulletinBoardMessageComparator;
|
|
||||||
|
|
||||||
import java.util.LinkedList;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 14-Dec-15.
|
|
||||||
*
|
|
||||||
* A data structure for holding a complete batch message along with its signature
|
|
||||||
*/
|
|
||||||
public class CompleteBatch {
|
|
||||||
|
|
||||||
private BeginBatchMessage beginBatchMessage;
|
|
||||||
private List<BatchData> batchDataList;
|
|
||||||
private Signature signature;
|
|
||||||
private Timestamp timestamp;
|
|
||||||
|
|
||||||
public CompleteBatch() {
|
|
||||||
batchDataList = new LinkedList<BatchData>();
|
|
||||||
}
|
|
||||||
|
|
||||||
public CompleteBatch(BeginBatchMessage newBeginBatchMessage) {
|
|
||||||
this();
|
|
||||||
beginBatchMessage = newBeginBatchMessage;
|
|
||||||
}
|
|
||||||
|
|
||||||
public CompleteBatch(BeginBatchMessage newBeginBatchMessage, List<BatchData> newDataList) {
|
|
||||||
this(newBeginBatchMessage);
|
|
||||||
appendBatchData(newDataList);
|
|
||||||
}
|
|
||||||
|
|
||||||
public CompleteBatch(BeginBatchMessage newBeginBatchMessage, List<BatchData> newDataList, Signature newSignature) {
|
|
||||||
this(newBeginBatchMessage, newDataList);
|
|
||||||
signature = newSignature;
|
|
||||||
}
|
|
||||||
|
|
||||||
public CompleteBatch(BeginBatchMessage newBeginBatchMessage, List<BatchData> newDataList, Signature newSignature, Timestamp timestamp) {
|
|
||||||
this(newBeginBatchMessage, newDataList, newSignature);
|
|
||||||
this.timestamp = timestamp;
|
|
||||||
}
|
|
||||||
|
|
||||||
public CompleteBatch(Timestamp timestamp) {
|
|
||||||
this();
|
|
||||||
this.timestamp = timestamp;
|
|
||||||
}
|
|
||||||
|
|
||||||
public BeginBatchMessage getBeginBatchMessage() {
|
|
||||||
return beginBatchMessage;
|
|
||||||
}
|
|
||||||
|
|
||||||
public List<BatchData> getBatchDataList() {
|
|
||||||
return batchDataList;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Signature getSignature() {
|
|
||||||
return signature;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Timestamp getTimestamp() {
|
|
||||||
return timestamp;
|
|
||||||
}
|
|
||||||
|
|
||||||
public CloseBatchMessage getCloseBatchMessage() {
|
|
||||||
return CloseBatchMessage.newBuilder()
|
|
||||||
.setBatchId(getBeginBatchMessage().getBatchId())
|
|
||||||
.setBatchLength(getBatchDataList().size())
|
|
||||||
.setSig(getSignature())
|
|
||||||
.setTimestamp(getTimestamp())
|
|
||||||
.build();
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setBeginBatchMessage(BeginBatchMessage beginBatchMessage) {
|
|
||||||
this.beginBatchMessage = beginBatchMessage;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void appendBatchData(BatchData newBatchData) {
|
|
||||||
batchDataList.add(newBatchData);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void appendBatchData(List<BatchData> newBatchDataList) {
|
|
||||||
batchDataList.addAll(newBatchDataList);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setSignature(Signature newSignature) {
|
|
||||||
signature = newSignature;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setTimestamp(Timestamp timestamp) {
|
|
||||||
this.timestamp = timestamp;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean equals(Object other) {
|
|
||||||
|
|
||||||
if (!(other instanceof CompleteBatch)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
CompleteBatch otherBatch = (CompleteBatch) other;
|
|
||||||
|
|
||||||
boolean result = true;
|
|
||||||
|
|
||||||
if (beginBatchMessage == null) {
|
|
||||||
if (otherBatch.getBeginBatchMessage() != null)
|
|
||||||
return false;
|
|
||||||
} else {
|
|
||||||
result = result && beginBatchMessage.equals(otherBatch.getBeginBatchMessage());
|
|
||||||
}
|
|
||||||
|
|
||||||
if (batchDataList == null) {
|
|
||||||
if (otherBatch.getBatchDataList() != null)
|
|
||||||
return false;
|
|
||||||
} else {
|
|
||||||
result = result && batchDataList.equals(otherBatch.getBatchDataList());
|
|
||||||
}
|
|
||||||
|
|
||||||
if (signature == null) {
|
|
||||||
if (otherBatch.getSignature() != null)
|
|
||||||
return false;
|
|
||||||
} else {
|
|
||||||
result = result && signature.equals(otherBatch.getSignature());
|
|
||||||
}
|
|
||||||
|
|
||||||
if (timestamp == null) {
|
|
||||||
if (otherBatch.getTimestamp() != null)
|
|
||||||
return false;
|
|
||||||
} else {
|
|
||||||
result = result && timestamp.equals(otherBatch.getTimestamp());
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String toString() {
|
|
||||||
|
|
||||||
if (beginBatchMessage == null || beginBatchMessage.getSignerId() == null)
|
|
||||||
return "Unspecified batch " + super.toString();
|
|
||||||
|
|
||||||
return "Batch " + beginBatchMessage.getSignerId().toString() + ":" + beginBatchMessage.getBatchId();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
|
@ -0,0 +1,32 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.protobuf.BoolValue;
|
||||||
|
import meerkat.comm.CommunicationException;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 13-Apr-16.
|
||||||
|
*/
|
||||||
|
public interface DeletableBulletinBoardServer extends BulletinBoardServer {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from the Bulletin Board
|
||||||
|
* If the message is a batch: the batch data is deleted as well
|
||||||
|
* Logs this action
|
||||||
|
* @param msgID is the ID of the message to delete
|
||||||
|
* @return a BoolMsg containing the value TRUE if a message was deleted, FALSE if the message does not exist
|
||||||
|
* @throws CommunicationException in case of an error
|
||||||
|
*/
|
||||||
|
public BoolValue deleteMessage(MessageID msgID) throws CommunicationException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Deletes a message from the Bulletin Board
|
||||||
|
* If the message is a batch: the batch data is deleted as well
|
||||||
|
* Logs this action
|
||||||
|
* @param entryNum is the serial entry number of the message to delete
|
||||||
|
* @return a BoolMsg containing the value TRUE if a message was deleted, FALSE if the message does not exist
|
||||||
|
* @throws CommunicationException in case of an error
|
||||||
|
*/
|
||||||
|
public BoolValue deleteMessage(long entryNum) throws CommunicationException;
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,7 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 13-Apr-16.
|
||||||
|
*/
|
||||||
|
public interface DeletableSubscriptionBulletinBoardClient extends SubscriptionBulletinBoardClient, BulletinBoardMessageDeleter {
|
||||||
|
}
|
|
@ -1,61 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
import com.google.protobuf.Message;
|
|
||||||
import meerkat.crypto.Digest;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.MessageID;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 19-Dec-15.
|
|
||||||
* Wrapper class for digesting Batches in a standardized way
|
|
||||||
*/
|
|
||||||
public class GenericBatchDigest implements BatchDigest{
|
|
||||||
|
|
||||||
private Digest digest;
|
|
||||||
|
|
||||||
public GenericBatchDigest(Digest digest) {
|
|
||||||
this.digest = digest;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void update(CompleteBatch completeBatch) {
|
|
||||||
|
|
||||||
update(completeBatch.getBeginBatchMessage());
|
|
||||||
|
|
||||||
for (BatchData batchData : completeBatch.getBatchDataList()) {
|
|
||||||
update(batchData);
|
|
||||||
}
|
|
||||||
|
|
||||||
update(completeBatch.getTimestamp());
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public byte[] digest() {
|
|
||||||
return digest.digest();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public MessageID digestAsMessageID() {
|
|
||||||
return digest.digestAsMessageID();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void update(Message msg) {
|
|
||||||
digest.update(msg);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void reset() {
|
|
||||||
digest.reset();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public GenericBatchDigest clone() throws CloneNotSupportedException{
|
|
||||||
return new GenericBatchDigest(digest.clone());
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
|
@ -1,104 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
import com.google.protobuf.ByteString;
|
|
||||||
import com.google.protobuf.Message;
|
|
||||||
import meerkat.crypto.DigitalSignature;
|
|
||||||
import meerkat.protobuf.BulletinBoardAPI.BatchData;
|
|
||||||
import meerkat.protobuf.Crypto;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.security.InvalidKeyException;
|
|
||||||
import java.security.KeyStore;
|
|
||||||
import java.security.KeyStoreException;
|
|
||||||
import java.security.SignatureException;
|
|
||||||
import java.security.NoSuchAlgorithmException;
|
|
||||||
import java.security.UnrecoverableKeyException;
|
|
||||||
import java.security.cert.CertificateException;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 20-Dec-15.
|
|
||||||
* Wrapper class for signing and verifying Batch signatures in a standardized way
|
|
||||||
*/
|
|
||||||
public class GenericBatchDigitalSignature implements BatchDigitalSignature{
|
|
||||||
|
|
||||||
private DigitalSignature digitalSignature;
|
|
||||||
|
|
||||||
public GenericBatchDigitalSignature(DigitalSignature digitalSignature) {
|
|
||||||
this.digitalSignature = digitalSignature;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void updateContent(CompleteBatch completeBatch) throws SignatureException {
|
|
||||||
|
|
||||||
digitalSignature.updateContent(completeBatch.getBeginBatchMessage());
|
|
||||||
|
|
||||||
for (BatchData batchData : completeBatch.getBatchDataList()) {
|
|
||||||
digitalSignature.updateContent(batchData);
|
|
||||||
}
|
|
||||||
|
|
||||||
digitalSignature.updateContent(completeBatch.getTimestamp());
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean verify(CompleteBatch completeBatch) throws SignatureException, CertificateException, InvalidKeyException {
|
|
||||||
|
|
||||||
digitalSignature.initVerify(completeBatch.getSignature());
|
|
||||||
updateContent(completeBatch);
|
|
||||||
return digitalSignature.verify();
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void loadVerificationCertificates(InputStream certStream) throws CertificateException {
|
|
||||||
digitalSignature.loadVerificationCertificates(certStream);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void clearVerificationCertificates() {
|
|
||||||
digitalSignature.clearVerificationCertificates();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void updateContent(Message msg) throws SignatureException {
|
|
||||||
digitalSignature.updateContent(msg);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Crypto.Signature sign() throws SignatureException {
|
|
||||||
return digitalSignature.sign();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void initVerify(Crypto.Signature sig) throws CertificateException, InvalidKeyException {
|
|
||||||
digitalSignature.initVerify(sig);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean verify() {
|
|
||||||
return digitalSignature.verify();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public KeyStore.Builder getPKCS12KeyStoreBuilder(InputStream keyStream, char[] password)
|
|
||||||
throws IOException, CertificateException, KeyStoreException, NoSuchAlgorithmException {
|
|
||||||
return digitalSignature.getPKCS12KeyStoreBuilder(keyStream, password);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void loadSigningCertificate(KeyStore.Builder keyStoreBuilder) throws IOException, CertificateException, UnrecoverableKeyException {
|
|
||||||
digitalSignature.loadSigningCertificate(keyStoreBuilder);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public ByteString getSignerID() {
|
|
||||||
return digitalSignature.getSignerID();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void clearSigningKey() {
|
|
||||||
digitalSignature.clearSigningKey();
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
|
@ -0,0 +1,74 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.protobuf.ByteString;
|
||||||
|
import com.google.protobuf.BytesValue;
|
||||||
|
import com.google.protobuf.Message;
|
||||||
|
import meerkat.crypto.Digest;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.MessageID;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 19-Dec-15.
|
||||||
|
* Wrapper class for digesting Batches in a standardized way
|
||||||
|
*/
|
||||||
|
public class GenericBulletinBoardDigest implements BulletinBoardDigest {
|
||||||
|
|
||||||
|
private Digest digest;
|
||||||
|
|
||||||
|
public GenericBulletinBoardDigest(Digest digest) {
|
||||||
|
this.digest = digest;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public byte[] digest() {
|
||||||
|
return digest.digest();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MessageID digestAsMessageID() {
|
||||||
|
return digest.digestAsMessageID();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void update(Message msg) {
|
||||||
|
digest.update(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void update(byte[] data) {
|
||||||
|
digest.update(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void reset() {
|
||||||
|
digest.reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public GenericBulletinBoardDigest clone() throws CloneNotSupportedException{
|
||||||
|
return new GenericBulletinBoardDigest(digest.clone());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void update(BulletinBoardMessage msg) {
|
||||||
|
update(msg.getMsg());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void update(UnsignedBulletinBoardMessage msg) {
|
||||||
|
|
||||||
|
for (ByteString tag : msg.getTagList().asByteStringList()){
|
||||||
|
update(tag.toByteArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
update(msg.getTimestamp());
|
||||||
|
|
||||||
|
if (msg.getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.DATA){
|
||||||
|
update(msg.getData().toByteArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,106 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.protobuf.ByteString;
|
||||||
|
import com.google.protobuf.Message;
|
||||||
|
import meerkat.crypto.Digest;
|
||||||
|
import meerkat.crypto.DigitalSignature;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.BulletinBoardMessage;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.MessageID;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.UnsignedBulletinBoardMessage;
|
||||||
|
import meerkat.protobuf.Crypto;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.security.*;
|
||||||
|
import java.security.cert.CertificateException;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 19-Dec-15.
|
||||||
|
* Wrapper class for digesting Batches in a standardized way
|
||||||
|
*/
|
||||||
|
public class GenericBulletinBoardSignature implements BulletinBoardSignature {
|
||||||
|
|
||||||
|
private DigitalSignature signer;
|
||||||
|
|
||||||
|
public GenericBulletinBoardSignature(DigitalSignature signer) {
|
||||||
|
this.signer = signer;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(BulletinBoardMessage msg) throws SignatureException{
|
||||||
|
signer.updateContent(msg.getMsg());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(UnsignedBulletinBoardMessage msg) throws SignatureException{
|
||||||
|
|
||||||
|
for (ByteString tag : msg.getTagList().asByteStringList()){
|
||||||
|
updateContent(tag.toByteArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
updateContent(msg.getTimestamp());
|
||||||
|
|
||||||
|
if (msg.getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.DATA){
|
||||||
|
updateContent(msg.getData().toByteArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void loadVerificationCertificates(InputStream certStream) throws CertificateException {
|
||||||
|
signer.loadVerificationCertificates(certStream);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void clearVerificationCertificates() {
|
||||||
|
signer.clearVerificationCertificates();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(byte[] data) throws SignatureException {
|
||||||
|
signer.updateContent(data);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(Message msg) throws SignatureException {
|
||||||
|
signer.updateContent(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Crypto.Signature sign() throws SignatureException {
|
||||||
|
return signer.sign();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void initVerify(Crypto.Signature sig) throws CertificateException, InvalidKeyException {
|
||||||
|
signer.initVerify(sig);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean verify() {
|
||||||
|
return signer.verify();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public KeyStore.Builder getPKCS12KeyStoreBuilder(InputStream keyStream, char[] password) throws IOException, CertificateException, KeyStoreException, NoSuchAlgorithmException {
|
||||||
|
return signer.getPKCS12KeyStoreBuilder(keyStream, password);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void loadSigningCertificate(KeyStore.Builder keyStoreBuilder) throws IOException, CertificateException, UnrecoverableKeyException {
|
||||||
|
signer.loadSigningCertificate(keyStoreBuilder);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ByteString getSignerID() {
|
||||||
|
return signer.getSignerID();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void clearSigningKey() {
|
||||||
|
signer.clearSigningKey();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -1,7 +0,0 @@
|
||||||
package meerkat.bulletinboard;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Created by Arbel Deutsch Peled on 03-Mar-16.
|
|
||||||
*/
|
|
||||||
public interface SubscriptionAsyncBulletinBoardClient extends AsyncBulletinBoardClient, BulletinBoardSubscriber {
|
|
||||||
}
|
|
|
@ -0,0 +1,7 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 03-Mar-16.
|
||||||
|
*/
|
||||||
|
public interface SubscriptionBulletinBoardClient extends AsyncBulletinBoardClient, BulletinBoardSubscriber {
|
||||||
|
}
|
|
@ -22,6 +22,12 @@ public interface Digest {
|
||||||
*/
|
*/
|
||||||
public MessageID digestAsMessageID();
|
public MessageID digestAsMessageID();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Updates the digest using the given raw data
|
||||||
|
* @param data contains the raw data
|
||||||
|
*/
|
||||||
|
public void update (byte[] data);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Updates the digest using the specified message (in serialized wire form)
|
* Updates the digest using the specified message (in serialized wire form)
|
||||||
*
|
*
|
||||||
|
|
|
@ -39,6 +39,13 @@ public interface DigitalSignature {
|
||||||
*/
|
*/
|
||||||
public void clearVerificationCertificates();
|
public void clearVerificationCertificates();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add raw data to the content stream to be verified / signed.
|
||||||
|
*
|
||||||
|
* @param data
|
||||||
|
* @throws SignatureException
|
||||||
|
*/
|
||||||
|
public void updateContent(byte[] data) throws SignatureException;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Add msg to the content stream to be verified / signed. Each message is (automatically)
|
* Add msg to the content stream to be verified / signed. Each message is (automatically)
|
||||||
|
|
|
@ -140,6 +140,11 @@ public class ECDSASignature implements DigitalSignature {
|
||||||
signer.update(msg.toByteString().asReadOnlyByteBuffer());
|
signer.update(msg.toByteString().asReadOnlyByteBuffer());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(byte[] data) throws SignatureException {
|
||||||
|
signer.update(data);
|
||||||
|
}
|
||||||
|
|
||||||
public void updateContent(InputStream in) throws IOException, SignatureException {
|
public void updateContent(InputStream in) throws IOException, SignatureException {
|
||||||
ByteString inStr = ByteString.readFrom(in);
|
ByteString inStr = ByteString.readFrom(in);
|
||||||
signer.update(inStr.asReadOnlyByteBuffer());
|
signer.update(inStr.asReadOnlyByteBuffer());
|
||||||
|
|
|
@ -80,6 +80,7 @@ public class SHA256Digest implements Digest {
|
||||||
hash.update(msg.asReadOnlyByteBuffer());
|
hash.update(msg.asReadOnlyByteBuffer());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
final public void update(byte[] msg) {
|
final public void update(byte[] msg) {
|
||||||
hash.update(msg);
|
hash.update(msg);
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,6 +5,7 @@ import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
import meerkat.protobuf.Crypto.*;
|
import meerkat.protobuf.Crypto.*;
|
||||||
|
|
||||||
import java.util.Comparator;
|
import java.util.Comparator;
|
||||||
|
import java.util.Iterator;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -24,22 +25,55 @@ public class BulletinBoardMessageComparator implements Comparator<BulletinBoardM
|
||||||
@Override
|
@Override
|
||||||
public int compare(BulletinBoardMessage msg1, BulletinBoardMessage msg2) {
|
public int compare(BulletinBoardMessage msg1, BulletinBoardMessage msg2) {
|
||||||
|
|
||||||
List<Signature> msg1Sigs = msg1.getSigList();
|
|
||||||
List<Signature> msg2Sigs = msg2.getSigList();
|
|
||||||
|
|
||||||
// Compare unsigned message
|
|
||||||
if (!msg1.getMsg().equals(msg2.getMsg())){
|
// Compare Timestamps
|
||||||
|
|
||||||
|
if (!msg1.getMsg().getTimestamp().equals(msg2.getMsg().getTimestamp())){
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Compare signatures
|
// Compare tags (enforce order)
|
||||||
|
|
||||||
if (msg1Sigs.size() != msg2Sigs.size()){
|
List<String> tags1 = msg1.getMsg().getTagList();
|
||||||
|
Iterator<String> tags2 = msg2.getMsg().getTagList().iterator();
|
||||||
|
|
||||||
|
for (String tag : tags1){
|
||||||
|
if (!tags2.hasNext()) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (!tags2.next().equals(tag)){
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compare data
|
||||||
|
|
||||||
|
if (msg1.getMsg().getDataTypeCase() != msg2.getMsg().getDataTypeCase()){
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (Signature sig : msg1Sigs){
|
if (msg1.getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.DATA){
|
||||||
if (!msg2Sigs.contains(sig)) {
|
if (!msg1.getMsg().getData().equals(msg2.getMsg().getData())){
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
} else if (msg1.getMsg().getDataTypeCase() == UnsignedBulletinBoardMessage.DataTypeCase.MSGID){
|
||||||
|
if (!msg1.getMsg().getMsgId().equals(msg2.getMsg().getMsgId())){
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compare signatures (do not enforce order)
|
||||||
|
|
||||||
|
List<Signature> sigs1 = msg1.getSigList();
|
||||||
|
List<Signature> sigs2 = msg2.getSigList();
|
||||||
|
|
||||||
|
if (sigs1.size() != sigs2.size()){
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (Signature sig : sigs1){
|
||||||
|
if (!sigs2.contains(sig)) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -7,7 +7,6 @@ import com.google.protobuf.Timestamp;
|
||||||
|
|
||||||
import java.math.BigInteger;
|
import java.math.BigInteger;
|
||||||
import java.security.SignatureException;
|
import java.security.SignatureException;
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Random;
|
import java.util.Random;
|
||||||
|
@ -28,10 +27,33 @@ public class BulletinBoardMessageGenerator {
|
||||||
return (byte) random.nextInt();
|
return (byte) random.nextInt();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private byte[] randomBytes(int length) {
|
||||||
|
|
||||||
|
byte[] result = new byte[length];
|
||||||
|
|
||||||
|
for (int i = 0; i < length; i++) {
|
||||||
|
result[i] = randomByte();
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
private String randomString(){
|
private String randomString(){
|
||||||
return new BigInteger(130, random).toString(32);
|
return new BigInteger(130, random).toString(32);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private List<String> randomStrings(int length) {
|
||||||
|
|
||||||
|
List<String> result = new LinkedList<>();
|
||||||
|
|
||||||
|
for (int i = 0; i < length; i++) {
|
||||||
|
result.add(randomString());
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Generates a complete instance of a BulletinBoardMessage
|
* Generates a complete instance of a BulletinBoardMessage
|
||||||
* @param signers contains the (possibly multiple) credentials required to sign the message
|
* @param signers contains the (possibly multiple) credentials required to sign the message
|
||||||
|
@ -46,23 +68,16 @@ public class BulletinBoardMessageGenerator {
|
||||||
|
|
||||||
// Generate random data.
|
// Generate random data.
|
||||||
|
|
||||||
byte[] data = new byte[dataSize];
|
|
||||||
String[] newTags = new String[tagNumber];
|
|
||||||
|
|
||||||
for (int i = 0; i < dataSize; i++) {
|
|
||||||
data[i] = randomByte();
|
|
||||||
}
|
|
||||||
|
|
||||||
for (int i = 0; i < tagNumber; i++) {
|
|
||||||
newTags[i] = randomString();
|
|
||||||
}
|
|
||||||
|
|
||||||
UnsignedBulletinBoardMessage unsignedMessage =
|
UnsignedBulletinBoardMessage unsignedMessage =
|
||||||
UnsignedBulletinBoardMessage.newBuilder()
|
UnsignedBulletinBoardMessage.newBuilder()
|
||||||
.setData(ByteString.copyFrom(data))
|
.setData(ByteString.copyFrom(randomBytes(dataSize)))
|
||||||
.setTimestamp(timestamp)
|
.setTimestamp(timestamp)
|
||||||
.addAllTag(tags)
|
.addAllTag(tags)
|
||||||
.addAllTag(Arrays.asList(newTags))
|
.addAllTag(randomStrings(tagNumber))
|
||||||
.build();
|
.build();
|
||||||
|
|
||||||
BulletinBoardMessage.Builder messageBuilder =
|
BulletinBoardMessage.Builder messageBuilder =
|
||||||
|
@ -102,7 +117,6 @@ public class BulletinBoardMessageGenerator {
|
||||||
* @param tagNumber is the number of tags to generate
|
* @param tagNumber is the number of tags to generate
|
||||||
* @return a random, signed Bulletin Board Message containing random data, tags and timestamp
|
* @return a random, signed Bulletin Board Message containing random data, tags and timestamp
|
||||||
*/
|
*/
|
||||||
|
|
||||||
public BulletinBoardMessage generateRandomMessage(DigitalSignature[] signers, int dataSize, int tagNumber)
|
public BulletinBoardMessage generateRandomMessage(DigitalSignature[] signers, int dataSize, int tagNumber)
|
||||||
throws SignatureException {
|
throws SignatureException {
|
||||||
|
|
||||||
|
|
|
@ -1,7 +1,10 @@
|
||||||
package meerkat.util;
|
package meerkat.util;
|
||||||
|
|
||||||
|
import com.google.protobuf.ByteString;
|
||||||
|
import com.google.protobuf.Int64Value;
|
||||||
import meerkat.protobuf.BulletinBoardAPI.*;
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
import java.util.LinkedList;
|
import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
|
@ -104,4 +107,129 @@ public class BulletinBoardUtils {
|
||||||
return new java.sql.Timestamp(protoTimestamp.getSeconds() * 1000 + protoTimestamp.getNanos() / 1000000);
|
return new java.sql.Timestamp(protoTimestamp.getSeconds() * 1000 + protoTimestamp.getNanos() / 1000000);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Breaks up a bulletin board message into chunks
|
||||||
|
* @param msg is the complete message
|
||||||
|
* @return a list of BatchChunks that contains the raw message data
|
||||||
|
*/
|
||||||
|
public static List<BatchChunk> breakToBatch(BulletinBoardMessage msg, int chunkSize) {
|
||||||
|
|
||||||
|
byte[] data = msg.getMsg().getData().toByteArray();
|
||||||
|
|
||||||
|
int chunkNum = data.length / chunkSize;
|
||||||
|
if (data.length % chunkSize != 0)
|
||||||
|
chunkNum++;
|
||||||
|
|
||||||
|
List<BatchChunk> chunkList = new ArrayList<>(chunkNum);
|
||||||
|
|
||||||
|
int location = 0;
|
||||||
|
|
||||||
|
for (int i=0 ; i < chunkNum ; i++) {
|
||||||
|
|
||||||
|
int chunkLength;
|
||||||
|
|
||||||
|
if (i == chunkNum - 1){
|
||||||
|
chunkLength = data.length % chunkSize;
|
||||||
|
if (chunkLength == 0){
|
||||||
|
chunkLength = chunkSize;
|
||||||
|
}
|
||||||
|
} else{
|
||||||
|
chunkLength = chunkSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
chunkList.add(BatchChunk.newBuilder()
|
||||||
|
.setData(ByteString.copyFrom(data, location, chunkLength))
|
||||||
|
.build());
|
||||||
|
|
||||||
|
location += chunkLength;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunkList;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Removes concrete data from the message and turns it into a stub
|
||||||
|
* Note that the stub does not contain the message ID
|
||||||
|
* Therefore, it cannot be used to retrieve the message from a server
|
||||||
|
* @param msg is the original message
|
||||||
|
* @return the message stub
|
||||||
|
*/
|
||||||
|
public static BulletinBoardMessage makeStub(BulletinBoardMessage msg) {
|
||||||
|
|
||||||
|
return BulletinBoardMessage.newBuilder()
|
||||||
|
.mergeFrom(msg)
|
||||||
|
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
||||||
|
.mergeFrom(msg.getMsg())
|
||||||
|
.clearDataType()
|
||||||
|
.clearData()
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Merges a batch chunk list back into a message stub to create a complete Bulletin Board message
|
||||||
|
* @param msgStub is a message stub
|
||||||
|
* @param chunkList contains the (ordered) data of the batch message
|
||||||
|
* @return a complete message containing both data and metadata
|
||||||
|
*/
|
||||||
|
public static BulletinBoardMessage gatherBatch(BulletinBoardMessage msgStub, List<BatchChunk> chunkList) {
|
||||||
|
|
||||||
|
List<ByteString> dataList = new LinkedList<>();
|
||||||
|
|
||||||
|
for (BatchChunk chunk : chunkList){
|
||||||
|
dataList.add(chunk.getData());
|
||||||
|
}
|
||||||
|
|
||||||
|
return BulletinBoardMessage.newBuilder()
|
||||||
|
.mergeFrom(msgStub)
|
||||||
|
.setMsg(UnsignedBulletinBoardMessage.newBuilder()
|
||||||
|
.mergeFrom(msgStub.getMsg())
|
||||||
|
.setData(ByteString.copyFrom(dataList))
|
||||||
|
.build())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Gerenates a BeginBatchMessage Protobuf which is used to begin uploading a message as a batch
|
||||||
|
* @param msg is the Bulletin Board message to be uploaded, which can be a stub or a complete message
|
||||||
|
* @return the required BeginBatchMessage
|
||||||
|
*/
|
||||||
|
public static BeginBatchMessage generateBeginBatchMessage(BulletinBoardMessage msg) {
|
||||||
|
|
||||||
|
if (msg.getSigCount() <= 0){
|
||||||
|
throw new IllegalArgumentException("No signatures found");
|
||||||
|
}
|
||||||
|
|
||||||
|
return BeginBatchMessage.newBuilder()
|
||||||
|
.addAllTag(msg.getMsg().getTagList())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Gerenates a CloseBatchMessage Protobuf which is used to finalize a batch message
|
||||||
|
* @param batchId is the temporary identifier for the message
|
||||||
|
* @param batchLength is the number of chunks in the batch
|
||||||
|
* @param msg is the Bulletin Board message that was uploaded (and can also be a stub of said message)
|
||||||
|
* @throws IllegalArgumentException if the message contains no signatures
|
||||||
|
*/
|
||||||
|
public static CloseBatchMessage generateCloseBatchMessage(Int64Value batchId, int batchLength, BulletinBoardMessage msg) {
|
||||||
|
|
||||||
|
if (msg.getSigCount() <= 0){
|
||||||
|
throw new IllegalArgumentException("No signatures found");
|
||||||
|
}
|
||||||
|
|
||||||
|
return CloseBatchMessage.newBuilder()
|
||||||
|
.setTimestamp(msg.getMsg().getTimestamp())
|
||||||
|
.setBatchLength(batchLength)
|
||||||
|
.setBatchId(batchId.getValue())
|
||||||
|
.addAllSig(msg.getSigList())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -20,8 +20,18 @@ message UnsignedBulletinBoardMessage {
|
||||||
// Timestamp of the message (as defined by client)
|
// Timestamp of the message (as defined by client)
|
||||||
google.protobuf.Timestamp timestamp = 2;
|
google.protobuf.Timestamp timestamp = 2;
|
||||||
|
|
||||||
|
// The payload of the message
|
||||||
|
oneof dataType{
|
||||||
|
|
||||||
|
// A unique message identifier
|
||||||
|
bytes msgId = 3;
|
||||||
|
|
||||||
// The actual content of the message
|
// The actual content of the message
|
||||||
bytes data = 3;
|
bytes data = 4;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
message BulletinBoardMessage {
|
message BulletinBoardMessage {
|
||||||
|
@ -82,42 +92,32 @@ message MessageFilterList {
|
||||||
|
|
||||||
// This message is used to start a batch transfer to the Bulletin Board Server
|
// This message is used to start a batch transfer to the Bulletin Board Server
|
||||||
message BeginBatchMessage {
|
message BeginBatchMessage {
|
||||||
bytes signerId = 1; // Unique signer identifier
|
repeated string tag = 1; // Tags for the batch message
|
||||||
int32 batchId = 2; // Unique identifier for the batch (unique per signer)
|
|
||||||
repeated string tag = 3; // Tags for the batch message
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// This message is used to finalize and sign a batch transfer to the Bulletin Board Server
|
// This message is used to finalize and sign a batch transfer to the Bulletin Board Server
|
||||||
message CloseBatchMessage {
|
message CloseBatchMessage {
|
||||||
int32 batchId = 1; // Unique identifier for the batch (unique per signer)
|
int64 batchId = 1; // Unique temporary identifier for the batch
|
||||||
int32 batchLength = 2; // Number of messages in the batch
|
int32 batchLength = 2; // Number of messages in the batch
|
||||||
google.protobuf.Timestamp timestamp = 3; // Timestamp of the batch (as defined by client)
|
google.protobuf.Timestamp timestamp = 3; // Timestamp of the batch (as defined by client)
|
||||||
meerkat.Signature sig = 4; // Signature on the (ordered) batch messages
|
repeated meerkat.Signature sig = 4; // Signatures on the (ordered) batch messages
|
||||||
}
|
}
|
||||||
|
|
||||||
// Container for single batch message data
|
// Container for single chunk of abatch message
|
||||||
message BatchData {
|
message BatchChunk {
|
||||||
bytes data = 1;
|
bytes data = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// List of BatchData; Only used for testing
|
// List of BatchChunk; Only used for testing
|
||||||
message BatchDataList {
|
message BatchChunkList {
|
||||||
repeated BatchData data = 1;
|
repeated BatchChunk data = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// These messages comprise a batch message
|
// These messages comprise a batch message
|
||||||
message BatchMessage {
|
message BatchMessage {
|
||||||
bytes signerId = 1; // Unique signer identifier
|
int64 batchId = 1; // Unique temporary identifier for the batch
|
||||||
int32 batchId = 2; // Unique identifier for the batch (unique per signer)
|
int32 serialNum = 2; // Location of the message in the batch: starting from 0
|
||||||
int32 serialNum = 3; // Location of the message in the batch: starting from 0
|
BatchChunk data = 3; // Actual data
|
||||||
BatchData data = 4; // Actual data
|
|
||||||
}
|
|
||||||
|
|
||||||
// This message defines which batch to read and from which location to start reading
|
|
||||||
message BatchSpecificationMessage {
|
|
||||||
bytes signerId = 1; // Unique signer identifier
|
|
||||||
int32 batchId = 2; // Unique identifier for the batch (unique per signer)
|
|
||||||
int32 startPosition = 3; // Position in batch to start reading from
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// This message is used to define a single query to the server to ascertain whether or not the server is synched with the client
|
// This message is used to define a single query to the server to ascertain whether or not the server is synched with the client
|
||||||
|
@ -161,3 +161,14 @@ message SyncQueryResponse {
|
||||||
google.protobuf.Timestamp lastTimeOfSync = 2;
|
google.protobuf.Timestamp lastTimeOfSync = 2;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// This message defines a query for retrieval of batch data
|
||||||
|
message BatchQuery {
|
||||||
|
|
||||||
|
// The unique message ID if the batch
|
||||||
|
MessageID msgID = 1;
|
||||||
|
|
||||||
|
// The first chunk to retrieve (0 is the first chunk)
|
||||||
|
int32 startPosition = 2;
|
||||||
|
|
||||||
|
}
|
|
@ -152,4 +152,3 @@ message SimpleCategoriesSelectionData {
|
||||||
repeated CategoryChooser categoryChooser = 2;
|
repeated CategoryChooser categoryChooser = 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,112 @@
|
||||||
|
package meerkat.bulletinboard;
|
||||||
|
|
||||||
|
import com.google.protobuf.ByteString;
|
||||||
|
import meerkat.crypto.concrete.ECDSASignature;
|
||||||
|
import meerkat.crypto.concrete.SHA256Digest;
|
||||||
|
import meerkat.protobuf.BulletinBoardAPI.*;
|
||||||
|
import meerkat.util.BulletinBoardMessageGenerator;
|
||||||
|
import meerkat.util.BulletinBoardUtils;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.security.*;
|
||||||
|
import java.security.cert.CertificateException;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
import static org.hamcrest.MatcherAssert.assertThat;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by Arbel Deutsch Peled on 16-Jun-16.
|
||||||
|
*/
|
||||||
|
public class BulletinBoardDigestTest {
|
||||||
|
|
||||||
|
private static String KEYFILE_EXAMPLE = "/certs/enduser-certs/user1-key-with-password-secret.p12";
|
||||||
|
private static String KEYFILE_EXAMPLE3 = "/certs/enduser-certs/user3-key-with-password-shh.p12";
|
||||||
|
|
||||||
|
private static String KEYFILE_PASSWORD1 = "secret";
|
||||||
|
private static String KEYFILE_PASSWORD3 = "shh";
|
||||||
|
|
||||||
|
private GenericBulletinBoardSignature[] signers;
|
||||||
|
private ByteString[] signerIDs;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void init() {
|
||||||
|
|
||||||
|
signers = new GenericBulletinBoardSignature[2];
|
||||||
|
signerIDs = new ByteString[signers.length];
|
||||||
|
signers[0] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
signers[1] = new GenericBulletinBoardSignature(new ECDSASignature());
|
||||||
|
|
||||||
|
InputStream keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE);
|
||||||
|
char[] password = KEYFILE_PASSWORD1.toCharArray();
|
||||||
|
|
||||||
|
KeyStore.Builder keyStoreBuilder = null;
|
||||||
|
try {
|
||||||
|
keyStoreBuilder = signers[0].getPKCS12KeyStoreBuilder(keyStream, password);
|
||||||
|
|
||||||
|
signers[0].loadSigningCertificate(keyStoreBuilder);
|
||||||
|
|
||||||
|
keyStream = getClass().getResourceAsStream(KEYFILE_EXAMPLE3);
|
||||||
|
password = KEYFILE_PASSWORD3.toCharArray();
|
||||||
|
|
||||||
|
keyStoreBuilder = signers[1].getPKCS12KeyStoreBuilder(keyStream, password);
|
||||||
|
signers[1].loadSigningCertificate(keyStoreBuilder);
|
||||||
|
|
||||||
|
for (int i = 0 ; i < signers.length ; i++) {
|
||||||
|
signerIDs[i] = signers[i].getSignerID();
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (IOException e) {
|
||||||
|
System.err.println("Failed reading from signature file " + e.getMessage());
|
||||||
|
fail("Failed reading from signature file " + e.getMessage());
|
||||||
|
} catch (CertificateException e) {
|
||||||
|
System.err.println("Failed reading certificate " + e.getMessage());
|
||||||
|
fail("Failed reading certificate " + e.getMessage());
|
||||||
|
} catch (KeyStoreException e) {
|
||||||
|
System.err.println("Failed reading keystore " + e.getMessage());
|
||||||
|
fail("Failed reading keystore " + e.getMessage());
|
||||||
|
} catch (NoSuchAlgorithmException e) {
|
||||||
|
System.err.println("Couldn't find signing algorithm " + e.getMessage());
|
||||||
|
fail("Couldn't find signing algorithm " + e.getMessage());
|
||||||
|
} catch (UnrecoverableKeyException e) {
|
||||||
|
System.err.println("Couldn't find signing key " + e.getMessage());
|
||||||
|
fail("Couldn't find signing key " + e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testBatchDigest() throws SignatureException {
|
||||||
|
|
||||||
|
final int MESSAGE_SIZE = 100;
|
||||||
|
final int CHUNK_SIZE = 10;
|
||||||
|
final int TAG_NUM = 10;
|
||||||
|
|
||||||
|
BulletinBoardMessageGenerator generator = new BulletinBoardMessageGenerator(new Random(0));
|
||||||
|
|
||||||
|
BulletinBoardMessage completeMessage = generator.generateRandomMessage(signers, MESSAGE_SIZE, TAG_NUM);
|
||||||
|
|
||||||
|
BulletinBoardMessage stub = BulletinBoardUtils.makeStub(completeMessage);
|
||||||
|
List<BatchChunk> batchChunks = BulletinBoardUtils.breakToBatch(completeMessage, CHUNK_SIZE);
|
||||||
|
|
||||||
|
BulletinBoardDigest digest = new GenericBulletinBoardDigest(new SHA256Digest());
|
||||||
|
|
||||||
|
digest.update(completeMessage);
|
||||||
|
MessageID id1 = digest.digestAsMessageID();
|
||||||
|
|
||||||
|
digest.update(stub);
|
||||||
|
for (BatchChunk batchChunk : batchChunks){
|
||||||
|
digest.update(batchChunk.getData().toByteArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
MessageID id2 = digest.digestAsMessageID();
|
||||||
|
|
||||||
|
assertThat("Digests not equal!", id1.getID().equals(id2.getID()));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -1,4 +1,3 @@
|
||||||
|
|
||||||
plugins {
|
plugins {
|
||||||
id "us.kirchmeier.capsule" version "1.0.1"
|
id "us.kirchmeier.capsule" version "1.0.1"
|
||||||
id 'com.google.protobuf' version '0.7.0'
|
id 'com.google.protobuf' version '0.7.0'
|
||||||
|
@ -197,4 +196,3 @@ publishing {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -57,6 +57,11 @@ public class ToySignature implements DigitalSignature {
|
||||||
throw new UnsupportedOperationException();
|
throw new UnsupportedOperationException();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void updateContent(byte[] data) throws SignatureException {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void initVerify(Signature sig) throws CertificateException, InvalidKeyException {
|
public void initVerify(Signature sig) throws CertificateException, InvalidKeyException {
|
||||||
throw new UnsupportedOperationException();
|
throw new UnsupportedOperationException();
|
||||||
|
|
Loading…
Reference in New Issue