mirror of
https://github.com/antonioCoco/SharPyShell
synced 2026-06-08 13:11:44 +00:00
antonioCoco
26 lines
808 B
Python
26 lines
808 B
Python
from utils.Singleton import Singleton
|
|
from Crypto.Cipher import AES
|
|
from Crypto.Util.Padding import pad
|
|
from Crypto.Util.Padding import unpad
|
|
|
|
|
|
class ChannelAES(Singleton):
|
|
hashed_password = None
|
|
IV = None
|
|
BS = 16
|
|
|
|
def __init__(self, password):
|
|
self.hashed_password = bytes.fromhex(password)
|
|
self.IV = self.hashed_password[0:self.BS]
|
|
|
|
def encrypt(self, plain_data):
|
|
plain_data_pad = pad(plain_data, self.BS)
|
|
aes = AES.new(self.hashed_password, AES.MODE_CBC, self.IV)
|
|
encrypted_data = aes.encrypt(plain_data_pad)
|
|
return encrypted_data
|
|
|
|
def decrypt(self, encrypted_data):
|
|
aes = AES.new(self.hashed_password, AES.MODE_CBC, self.IV)
|
|
decrypted_data = aes.decrypt(encrypted_data)
|
|
return unpad(decrypted_data, self.BS)
|