5c0ec720af
Just getting my feet wet... Change-Id: Ia1db0c69fe7d5ea5cb5585853d0688ef97f2680a Reviewed-on: https://cl.tvl.fyi/c/depot/+/4739 Reviewed-by: wpcarro <wpcarro@gmail.com> Autosubmit: wpcarro <wpcarro@gmail.com> Tested-by: BuildkiteCI
60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
|
|
import sys
|
|
import base64
|
|
from cryptography.fernet import Fernet
|
|
|
|
|
|
|
|
usage_msg = "Usage: "+ sys.argv[0] +" (-e/-d) [file]"
|
|
help_msg = usage_msg + "\n" +\
|
|
"Examples:\n" +\
|
|
" To decrypt a file named 'pole.txt', do: " +\
|
|
"'$ python "+ sys.argv[0] +" -d pole.txt'\n"
|
|
|
|
|
|
|
|
if len(sys.argv) < 2 or len(sys.argv) > 4:
|
|
print(usage_msg)
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
if sys.argv[1] == "-e":
|
|
if len(sys.argv) < 4:
|
|
sim_sala_bim = input("Please enter the password:")
|
|
else:
|
|
sim_sala_bim = sys.argv[3]
|
|
|
|
ssb_b64 = base64.b64encode(sim_sala_bim.encode())
|
|
c = Fernet(ssb_b64)
|
|
|
|
with open(sys.argv[2], "rb") as f:
|
|
data = f.read()
|
|
data_c = c.encrypt(data)
|
|
sys.stdout.write(data_c.decode())
|
|
|
|
|
|
elif sys.argv[1] == "-d":
|
|
if len(sys.argv) < 4:
|
|
sim_sala_bim = input("Please enter the password:")
|
|
else:
|
|
sim_sala_bim = sys.argv[3]
|
|
|
|
ssb_b64 = base64.b64encode(sim_sala_bim.encode())
|
|
c = Fernet(ssb_b64)
|
|
|
|
with open(sys.argv[2], "r") as f:
|
|
data = f.read()
|
|
data_c = c.decrypt(data.encode())
|
|
sys.stdout.buffer.write(data_c)
|
|
|
|
|
|
elif sys.argv[1] == "-h" or sys.argv[1] == "--help":
|
|
print(help_msg)
|
|
sys.exit(1)
|
|
|
|
|
|
else:
|
|
print("Unrecognized first argument: "+ sys.argv[1])
|
|
print("Please use '-e', '-d', or '-h'.")
|
|
|