LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

inlinecallbacks

Browse source
This commit is contained in:
Jack Robison 2017-09-07 11:54:39 -04:00
parent 1148a533bf
commit a72fef07c0
No known key found for this signature in database
GPG key ID: 284699E7404E3CFF
1 changed files with 10 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
lbrynet/cryptstream/CryptBlob.py
View file

@ -1,5 +1,6 @@
import binascii import binascii
import logging import logging
from twisted.internet import defer
from cryptography.hazmat.primitives.ciphers import Cipher, modes from cryptography.hazmat.primitives.ciphers import Cipher, modes
from cryptography.hazmat.primitives.ciphers.algorithms import AES from cryptography.hazmat.primitives.ciphers.algorithms import AES
from cryptography.hazmat.primitives.padding import PKCS7 from cryptography.hazmat.primitives.padding import PKCS7
@ -115,23 +116,19 @@ class CryptStreamBlobMaker(object):
self.blob.write(encrypted_data) self.blob.write(encrypted_data)
return done, num_bytes_to_write return done, num_bytes_to_write
@defer.inlineCallbacks
def close(self): def close(self):
log.debug("closing blob %s with plaintext len %s", str(self.blob_num), str(self.length)) log.debug("closing blob %s with plaintext len %s", str(self.blob_num), str(self.length))
if self.length != 0: if self.length != 0:
self._close_buffer() self.length += (AES.block_size / 8) - (self.length % (AES.block_size / 8))
d = self.blob.close() padded_data = self.padder.finalize()
d.addCallback(self._return_info) encrypted_data = self.cipher.update(padded_data) + self.cipher.finalize()
self.blob.write(encrypted_data)
blob_hash = yield self.blob.close()
log.debug("called the finished_callback from CryptStreamBlobMaker.close") log.debug("called the finished_callback from CryptStreamBlobMaker.close")
return d blob = CryptBlobInfo(blob_hash, self.blob_num, self.length, binascii.hexlify(self.iv))
defer.returnValue(blob)
def _close_buffer(self):
self.length += (AES.block_size / 8) - (self.length % (AES.block_size / 8))
padded_data = self.padder.finalize()
encrypted_data = self.cipher.update(padded_data) + self.cipher.finalize()
self.blob.write(encrypted_data)
def _return_info(self, blob_hash):
return CryptBlobInfo(blob_hash, self.blob_num, self.length, binascii.hexlify(self.iv))
def greatest_multiple(a, b): def greatest_multiple(a, b):