diff options
Diffstat (limited to 'client/src/leap/soledad/client/_blobs.py')
-rw-r--r-- | client/src/leap/soledad/client/_blobs.py | 108 |
1 files changed, 58 insertions, 50 deletions
diff --git a/client/src/leap/soledad/client/_blobs.py b/client/src/leap/soledad/client/_blobs.py index 3de67569..336fe9af 100644 --- a/client/src/leap/soledad/client/_blobs.py +++ b/client/src/leap/soledad/client/_blobs.py @@ -19,7 +19,10 @@ Clientside BlobBackend Storage. """ from copy import copy +from urlparse import urljoin + import os.path +import uuid from io import BytesIO from functools import partial @@ -143,7 +146,7 @@ class BlobManager(object): # concurrently. not sure if we'd gain something. yield self.local.put(doc.blob_id, fd) fd.seek(0) - yield self._encrypt_and_upload(doc.blob_id, fd, up) + yield self._encrypt_and_upload(doc.blob_id, doc.doc_id, doc.rev, fd) @defer.inlineCallbacks def get(self, blob_id, doc_id, rev): @@ -152,7 +155,11 @@ class BlobManager(object): logger.info("Found blob in local database: %s" % blob_id) defer.returnValue(local_blob) - blob, size = yield self._download_and_decrypt(blob_id, doc_id, rev) + result = yield self._download_and_decrypt(blob_id, doc_id, rev) + + if not result: + defer.returnValue(None) + blob, size = result if blob: logger.info("Got decrypted blob of type: %s" % type(blob)) @@ -169,7 +176,7 @@ class BlobManager(object): logger.error('sorry, dunno what happened') @defer.inlineCallbacks - def _encrypt_and_upload(self, blob_id, doc_id, rev, payload): + def _encrypt_and_upload(self, blob_id, doc_id, rev, fd): # TODO ------------------------------------------ # this is wrong, is doing 2 stages. # the crypto producer can be passed to @@ -177,12 +184,14 @@ class BlobManager(object): # try to rewrite as a tube: pass the fd to aes and let aes writer # produce data to the treq request fd. # ------------------------------------------------ + logger.info("Staring upload of blob: %s" % blob_id) doc_info = DocInfo(doc_id, rev) - uri = self.remote + '/' + self.user + '/' + blob_id - crypter = BlobEncryptor(doc_info, payload, secret=self.secret, + uri = urljoin(self.remote, self.user + "/" + blob_id) + crypter = BlobEncryptor(doc_info, fd, secret=self.secret, armor=True) - result = yield crypter.encrypt() - yield treq.put(uri, data=result) + fd = yield crypter.encrypt() + yield treq.put(uri, data=fd) + logger.info("Finished upload: %s" % (blob_id,)) @defer.inlineCallbacks def _download_and_decrypt(self, blob_id, doc_id, rev): @@ -262,21 +271,19 @@ def _sqlcipherInitFactory(fun): return _initialize -# --------------------8<---------------------------------------------- -# class BlobDoc(object): -# -# # TODO probably not needed, but convenient for testing for now. -# -# def __init__(self, doc_id, rev, content, blob_id=None): -# -# self.doc_id = doc_id -# self.rev = rev -# self.is_blob = True -# self.blob_fd = content -# if blob_id is None: -# blob_id = uuid4().get_hex() -# self.blob_id = blob_id -# --------------------8<---------------------------------------------- +class BlobDoc(object): + + # TODO probably not needed, but convenient for testing for now. + + def __init__(self, doc_id, rev, content, blob_id=None): + + self.doc_id = doc_id + self.rev = rev + self.is_blob = True + self.blob_fd = content + if blob_id is None: + blob_id = uuid.uuid4().get_hex() + self.blob_id = blob_id # @@ -293,7 +300,7 @@ def testit(reactor): # parse command line arguments import argparse - usage = "\n cd server/src/leap/soledad/server/ && python _blobs.py" \ + usage = "\n mkdir /tmp/blobs/user && cd server/src/leap/soledad/server/ && python _blobs.py" \ "\n python _blobs.py upload /path/to/file blob_id" \ "\n python _blobs.py download blob_id" @@ -327,47 +334,49 @@ def testit(reactor): # TODO convert these into proper unittests + def _manager(): + manager = BlobManager( + '/tmp/blobs', 'http://localhost:9000/', + 'A' * 32, 'secret', 'user') + return manager + @defer.inlineCallbacks def _upload(blob_id, payload): - logger.info(":: Starting upload only...") - doc_info = DocInfo('mydoc', '1') - logger.info(str(doc_info)) - # use BlobEncryptor intead of BlobManager to only upload and not put - # blob on local db - crypter = BlobEncryptor( - doc_info, open(payload, 'r'), 'A' * 32, armor=True) - logger.info(":: Uploading with encryptor") - result = yield crypter.encrypt() - yield treq.put('http://localhost:9000/user/' + blob_id, data=result) - logger.info(":: Finished upload only.") + logger.info(":: Starting upload only: %s" % str((blob_id, payload))) + manager = _manager() + with open(payload, 'r') as fd: + yield manager._encrypt_and_upload(blob_id, 'mydoc', '1', fd) + logger.info(":: Finished upload only: %s" % str((blob_id, payload))) @defer.inlineCallbacks def _download(blob_id): logger.info(":: Starting download only: %s" % blob_id) - manager = BlobManager( - '/tmp/blobs', 'http://localhost:9000/', - 'A' * 32, 'secret', 'user') + manager = _manager() result = yield manager._download_and_decrypt(blob_id, 'mydoc', '1') logger.info(":: Result of download: %s" % str(result)) - if not result: - logger.info(":: Download failed for: %s" % blob_id) - else: + if result: fd, _ = result logger.info(":: Content of blob %s: %s" % (blob_id, fd.getvalue())) logger.info(":: Finished download only: %s" % blob_id) + @defer.inlineCallbacks def _put(blob_id, payload): - pass + logger.info(":: Starting full put: %s" % blob_id) + manager = _manager() + with open(payload) as fd: + doc = BlobDoc('mydoc', '1', fd, blob_id=blob_id) + result = yield manager.put(doc) + logger.info(":: Result of put: %s" % str(result)) + logger.info(":: Finished full put: %s" % blob_id) @defer.inlineCallbacks def _get(blob_id): - logger.info(":: Starting full get...") - manager = BlobManager( - '/tmp/blobs', 'http://localhost:9000/', - 'A' * 32, 'secret', 'user') - payload = yield manager.get(blob_id, 'mydoc', '1') - logger.info(":: Finished full get.") - defer.returnValue(payload) + logger.info(":: Starting full get: %s" % blob_id) + manager = _manager() + fd = yield manager.get(blob_id, 'mydoc', '1') + if fd: + logger.info(":: Result of get: " + fd.getvalue()) + logger.info(":: Finished full get: %s" % blob_id) if args.action == 'upload': yield _upload(args.blob_id, args.payload) @@ -376,8 +385,7 @@ def testit(reactor): elif args.action == 'put': yield _put(args.blob_id, args.payload) elif args.action == 'get': - result = yield _get(args.blob_id) - logger.info(":: Result of get: " + result.getvalue()) + yield _get(args.blob_id) if __name__ == '__main__': |