PyBitmessage-2021-04-27/src/pyelliptic/cipher.py

91 lines
3.1 KiB
Python
Raw Normal View History

2019-09-23 14:21:58 +02:00
"""
2019-12-21 10:43:29 +01:00
Symmetric Encryption
2019-09-23 14:21:58 +02:00
"""
2013-01-16 17:52:52 +01:00
# Copyright (C) 2011 Yann GUIBET <yannguibet@gmail.com>
# See LICENSE for details.
from .openssl import OpenSSL
2013-01-16 17:52:52 +01:00
2019-09-23 14:21:58 +02:00
# pylint: disable=redefined-builtin
class Cipher(object):
2013-01-16 17:52:52 +01:00
"""
2019-12-21 10:43:29 +01:00
Main class for encryption
2013-01-16 17:52:52 +01:00
import pyelliptic
iv = pyelliptic.Cipher.gen_IV('aes-256-cfb')
ctx = pyelliptic.Cipher("secretkey", iv, 1, ciphername='aes-256-cfb')
ciphertext = ctx.update('test1')
ciphertext += ctx.update('test2')
ciphertext += ctx.final()
ctx2 = pyelliptic.Cipher("secretkey", iv, 0, ciphername='aes-256-cfb')
print ctx2.ciphering(ciphertext)
"""
def __init__(self, key, iv, do, ciphername='aes-256-cbc'):
"""
do == 1 => Encrypt; do == 0 => Decrypt
"""
self.cipher = OpenSSL.get_cipher(ciphername)
self.ctx = OpenSSL.EVP_CIPHER_CTX_new()
if do == 1 or do == 0:
k = OpenSSL.malloc(key, len(key))
IV = OpenSSL.malloc(iv, len(iv))
OpenSSL.EVP_CipherInit_ex(
self.ctx, self.cipher.get_pointer(), 0, k, IV, do)
else:
raise Exception("RTFM ...")
@staticmethod
def get_all_cipher():
"""
static method, returns all ciphers available
"""
return OpenSSL.cipher_algo.keys()
@staticmethod
def get_blocksize(ciphername):
2019-09-23 14:21:58 +02:00
"""This Method returns cipher blocksize"""
2013-01-16 17:52:52 +01:00
cipher = OpenSSL.get_cipher(ciphername)
return cipher.get_blocksize()
@staticmethod
def gen_IV(ciphername):
2019-09-23 14:21:58 +02:00
"""Generate random initialization vector"""
2013-01-16 17:52:52 +01:00
cipher = OpenSSL.get_cipher(ciphername)
return OpenSSL.rand(cipher.get_blocksize())
def update(self, input):
2019-09-23 14:21:58 +02:00
"""Update result with more data"""
2013-01-16 17:52:52 +01:00
i = OpenSSL.c_int(0)
buffer = OpenSSL.malloc(b"", len(input) + self.cipher.get_blocksize())
inp = OpenSSL.malloc(input, len(input))
if OpenSSL.EVP_CipherUpdate(self.ctx, OpenSSL.byref(buffer),
OpenSSL.byref(i), inp, len(input)) == 0:
raise Exception("[OpenSSL] EVP_CipherUpdate FAIL ...")
2019-12-21 10:43:29 +01:00
return buffer.raw[0:i.value] # pylint: disable=invalid-slice-index
2013-01-16 17:52:52 +01:00
def final(self):
2019-09-23 14:21:58 +02:00
"""Returning the final value"""
2013-01-16 17:52:52 +01:00
i = OpenSSL.c_int(0)
buffer = OpenSSL.malloc(b"", self.cipher.get_blocksize())
if (OpenSSL.EVP_CipherFinal_ex(self.ctx, OpenSSL.byref(buffer),
OpenSSL.byref(i))) == 0:
raise Exception("[OpenSSL] EVP_CipherFinal_ex FAIL ...")
2019-12-21 10:43:29 +01:00
return buffer.raw[0:i.value] # pylint: disable=invalid-slice-index
2013-01-16 17:52:52 +01:00
def ciphering(self, input):
"""
Do update and final in one method
"""
buff = self.update(input)
return buff + self.final()
def __del__(self):
2019-09-23 14:21:58 +02:00
# pylint: disable=protected-access
if OpenSSL._hexversion > 0x10100000 and not OpenSSL._libreSSL:
OpenSSL.EVP_CIPHER_CTX_reset(self.ctx)
else:
OpenSSL.EVP_CIPHER_CTX_cleanup(self.ctx)
2013-01-16 17:52:52 +01:00
OpenSSL.EVP_CIPHER_CTX_free(self.ctx)