summaryrefslogtreecommitdiff
path: root/client/src/leap/soledad
diff options
context:
space:
mode:
Diffstat (limited to 'client/src/leap/soledad')
-rw-r--r--client/src/leap/soledad/client/_blobs.py108
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__':