ChCh/CloudBot
Archived
11
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

anti-asshat

Browse source
This commit is contained in:
Luke Rogers 2013-10-01 13:18:41 +13:00
parent 2ae2a8575a
commit 72dce244b4
1 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

30
plugins/encrypt.py
View file

@ -30,18 +30,25 @@ def get_salt(bot):
@hook.command @hook.command
def encrypt(inp, bot=None, db=None): def encrypt(inp, bot=None, db=None, notice=None):
"""encrypt <pass> <string> -- Encrypts <string> with <pass>.""" """encrypt <pass> <string> -- Encrypts <string> with <pass>."""
db_init(db) if not db_ready:
db_init(db)
password = inp.split(" ")[0] split = inp.split(" ")
if len(split) == 1:
notice(encrypt.__doc__)
return
password = split[0]
salt = get_salt(bot) salt = get_salt(bot)
key = PBKDF2(password, salt) key = PBKDF2(password, salt)
iv = Random.new().read(AES.block_size); iv = Random.new().read(AES.block_size);
iv_encoded = base64.b64encode(iv) iv_encoded = base64.b64encode(iv)
text = " ".join(inp.split(" ")[1:]) text = " ".join(split[1:])
cipher = AES.new(key, AES.MODE_CBC, iv) cipher = AES.new(key, AES.MODE_CBC, iv)
encrypted = cipher.encrypt(pad(text)) encrypted = cipher.encrypt(pad(text))
encoded = base64.b64encode(encrypted) encoded = base64.b64encode(encrypted)
@ -54,15 +61,22 @@ def encrypt(inp, bot=None, db=None):
@hook.command @hook.command
def decrypt(inp, bot=None, db=None): def decrypt(inp, bot=None, db=None, notice=None):
"""decrypt <pass> <string> -- Decrypts <string> with <pass>.""" """decrypt <pass> <string> -- Decrypts <string> with <pass>."""
db_init(db) if not db_ready:
db_init(db)
password = inp.split(" ")[0] split = inp.split(" ")
if len(split) == 1:
notice(decrypt.__doc__)
return
password = split[0]
salt = get_salt(bot) salt = get_salt(bot)
key = PBKDF2(password, salt) key = PBKDF2(password, salt)
text = " ".join(inp.split(" ")[1:]) text = " ".join(split[1:])
iv_encoded = db.execute("select iv from encryption where" iv_encoded = db.execute("select iv from encryption where"
" encrypted=?", (text,)).fetchone()[0] " encrypted=?", (text,)).fetchone()[0]