ChCh/CloudBot
Archived
10
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
Show stats 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
def encrypt(inp, bot=None, db=None):
def encrypt(inp, bot=None, db=None, notice=None):
"""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)
key = PBKDF2(password, salt)
iv = Random.new().read(AES.block_size);
iv_encoded = base64.b64encode(iv)
text = " ".join(inp.split(" ")[1:])
text = " ".join(split[1:])
cipher = AES.new(key, AES.MODE_CBC, iv)
encrypted = cipher.encrypt(pad(text))
encoded = base64.b64encode(encrypted)
@ -54,15 +61,22 @@ def encrypt(inp, bot=None, db=None):
@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>."""
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)
key = PBKDF2(password, salt)
text = " ".join(inp.split(" ")[1:])
text = " ".join(split[1:])
iv_encoded = db.execute("select iv from encryption where"
" encrypted=?", (text,)).fetchone()[0]