summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorVictor Shyba <victor1984@riseup.net>2016-11-29 02:20:03 -0300
committerVictor Shyba <victor1984@riseup.net>2016-11-29 02:20:03 -0300
commit6481c87c32b7d81c177f0a48814dc7fe0cba706f (patch)
tree7e9d323f4eae0735647636a3d2c067d9933f9c1c
parent91870566bbfc642d43e927ea714192b9962f7a7f (diff)
[style] fixes from code-review
Naming, interfaces and other details.
-rw-r--r--client/src/leap/soledad/client/_crypto.py20
1 files changed, 11 insertions, 9 deletions
diff --git a/client/src/leap/soledad/client/_crypto.py b/client/src/leap/soledad/client/_crypto.py
index f6a84b70..574e2b6e 100644
--- a/client/src/leap/soledad/client/_crypto.py
+++ b/client/src/leap/soledad/client/_crypto.py
@@ -195,7 +195,7 @@ class BlobEncryptor(object):
self._aes_fd = BytesIO()
_aes = AESWriter(sym_key, _buffer=self._aes_fd)
- self.__iv = _aes.iv
+ self._iv = _aes.iv
self._hmac_writer = HMACWriter(mac_key)
self._write_preamble()
@@ -203,7 +203,7 @@ class BlobEncryptor(object):
@property
def iv(self):
- return self.__iv
+ return self._iv
def encrypt(self):
"""
@@ -292,16 +292,15 @@ class BlobDecryptor(object):
try:
unpacked_data = PACMAN.unpack(preamble)
- pad, ts, sch, meth, iv, doc_id, rev = unpacked_data
+ magic, ts, sch, meth, iv, doc_id, rev = unpacked_data
except struct.error:
raise InvalidBlob
- if pad != '\x80':
+ if magic != '\x80':
raise InvalidBlob
# TODO check timestamp
if sch != ENC_SCHEME.symkey:
raise InvalidBlob('invalid scheme')
- # TODO should adapt the assymetric-gpg too, rigth?
if meth != ENC_METHOD.aes_256_ctr:
raise InvalidBlob('invalid encryption scheme')
if rev != self.rev:
@@ -339,18 +338,18 @@ class GenericWriter(object):
"""
implements(interfaces.IConsumer)
- def __init__(self, operator, closer, result=None):
+ def __init__(self, process, close, result=None):
self.result = result or BytesIO()
- self.operator, self.closer = operator, closer
+ self.process, self.close = process, close
def write(self, data):
- out = self.operator(data)
+ out = self.process(data)
if out:
self.result.write(out)
return out
def end(self):
- self.result.write(self.closer())
+ self.result.write(self.close())
return self.result.getvalue()
@@ -359,6 +358,7 @@ class HMACWriter(GenericWriter):
A Twisted's Consumer implementation that takes an input file descriptor and
produces a HMAC-SHA512 Message Authentication Code.
"""
+ implements(interfaces.IConsumer)
hashtype = 'sha512'
def __init__(self, key, result=None):
@@ -371,6 +371,8 @@ class AESWriter(GenericWriter):
A Twisted's Consumer implementation that takes an input file descriptor and
applies AES-256 cipher in CTR mode.
"""
+ implements(interfaces.IConsumer)
+
def __init__(self, key, iv=None, _buffer=None, encrypt=True):
if len(key) != 32:
raise EncryptionDecryptionError('key is not 256 bits')