forked from r4sas/PBinCLI
partly realised send command
This commit is contained in:
parent
c88f678bbf
commit
8014c386ac
@ -20,7 +20,7 @@ def main():
|
|||||||
send_parser.add_argument("-e", "--expire", default="1day", action="store", help="expiration of paste (default: 1day)")
|
send_parser.add_argument("-e", "--expire", default="1day", action="store", help="expiration of paste (default: 1day)")
|
||||||
send_parser.add_argument("-f", "--format", default="plaintext", action="store", help="format of paste (default: plaintext)")
|
send_parser.add_argument("-f", "--format", default="plaintext", action="store", help="format of paste (default: plaintext)")
|
||||||
send_parser.add_argument("-p", "--password", default=None, help="password for crypting paste")
|
send_parser.add_argument("-p", "--password", default=None, help="password for crypting paste")
|
||||||
send_parser.add_argument("filename", type=argparse.FileType('r'), help="filename (example: image.jpg)")
|
send_parser.add_argument("filename", help="filename (example: image.jpg)")
|
||||||
send_parser.set_defaults(func=pbincli.actions.send)
|
send_parser.set_defaults(func=pbincli.actions.send)
|
||||||
|
|
||||||
# parse arguments
|
# parse arguments
|
||||||
|
@ -1,7 +1,27 @@
|
|||||||
"""Action functions for argparser"""
|
"""Action functions for argparser"""
|
||||||
import pbincli.sjcl_gcm
|
import base64
|
||||||
import pbincli.actions
|
import pbincli.actions
|
||||||
|
from Crypto.Hash import SHA256
|
||||||
|
from Crypto.Random import get_random_bytes
|
||||||
|
from pbincli.sjcl_gcm import SJCL
|
||||||
from pbincli.utils import PBinCLIException, check_readable, check_writable
|
from pbincli.utils import PBinCLIException, check_readable, check_writable
|
||||||
|
from zlib import compress
|
||||||
|
|
||||||
def send(args):
|
def send(args):
|
||||||
print("Meow!")
|
""" Sub-command for sending paste """
|
||||||
|
check_readable(args.filename)
|
||||||
|
with open(args.filename, "rb") as f:
|
||||||
|
contents = f.read()
|
||||||
|
file = base64.b64encode(compress(contents))
|
||||||
|
|
||||||
|
passphrase = base64.b64encode(get_random_bytes(32))
|
||||||
|
if not args.password:
|
||||||
|
password = passphrase
|
||||||
|
else:
|
||||||
|
p = SHA256.new()
|
||||||
|
p.update(args.password.encode("UTF-8"))
|
||||||
|
password = passphrase + p.hexdigest().encode("UTF-8")
|
||||||
|
|
||||||
|
data = SJCL().encrypt(file, password)
|
||||||
|
'''Here we must run function post from pbincli.transports'''
|
||||||
|
print(data)
|
||||||
|
@ -2,19 +2,13 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
Edited SJCL.py library (https://github.com/berlincode/sjcl/)
|
Edited SJCL.py library (https://github.com/berlincode/sjcl/)
|
||||||
for AES.MODE_GCM support, work begin.
|
for AES.MODE_GCM support, work begin.
|
||||||
|
|
||||||
pip3.5 install pycryptodome
|
pip3.5 install pycryptodome
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from Crypto.Hash import SHA256, HMAC
|
from Crypto.Hash import SHA256, HMAC
|
||||||
#pip3.5 install pycryptodome
|
|
||||||
from Crypto.Protocol.KDF import PBKDF2
|
from Crypto.Protocol.KDF import PBKDF2
|
||||||
from Crypto.Cipher import AES
|
from Crypto.Cipher import AES
|
||||||
from Crypto.Random import get_random_bytes
|
from Crypto.Random import get_random_bytes
|
||||||
@ -112,7 +106,7 @@ class SJCL(object):
|
|||||||
|
|
||||||
return plaintext
|
return plaintext
|
||||||
|
|
||||||
def encrypt(self, plaintext, passphrase, count=1000, dkLen=16):
|
def encrypt(self, plaintext, passphrase, count=10000, dkLen=32):
|
||||||
# dkLen = key length in bytes
|
# dkLen = key length in bytes
|
||||||
|
|
||||||
check_mode_gcm() # check gcm support
|
check_mode_gcm() # check gcm support
|
||||||
|
Loading…
x
Reference in New Issue
Block a user