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

use blob.open_for_reading() in StreamBlobDecryptor

Browse source
This commit is contained in:
Kay Kurokawa 2017-09-29 14:43:03 -04:00
parent e07c2f7bd8
commit 1360345395
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lbrynet/cryptstream/CryptBlob.py
View file

@ -1,6 +1,6 @@
import binascii import binascii
import logging import logging
from twisted.internet import defer from twisted.internet import defer, threads
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
@ -46,6 +46,10 @@ class StreamBlobDecryptor(object):
write_func - function that takes decrypted string as write_func - function that takes decrypted string as
arugment and writes it somewhere arugment and writes it somewhere
Returns:
deferred that returns after decrypting blob and writing content
""" """
def remove_padding(data): def remove_padding(data):
@ -67,13 +71,17 @@ class StreamBlobDecryptor(object):
last_chunk = self.cipher.update(data_to_decrypt) + self.cipher.finalize() last_chunk = self.cipher.update(data_to_decrypt) + self.cipher.finalize()
write_func(remove_padding(last_chunk)) write_func(remove_padding(last_chunk))
def decrypt_bytes(data):
read_handle = self.blob.open_for_reading()
def decrypt_bytes():
data = read_handle.read()
self.buff += data self.buff += data
self.len_read += len(data) self.len_read += len(data)
write_bytes() write_bytes()
finish_decrypt()
d = self.blob.read(decrypt_bytes) d = threads.deferToThread(decrypt_bytes)
d.addCallback(lambda _: finish_decrypt())
return d return d