110 lines
3.3 KiB
Python
Executable File
110 lines
3.3 KiB
Python
Executable File
#!/bin/python3
|
|
import os
|
|
import sys
|
|
import logging
|
|
import requests
|
|
import json
|
|
import argparse
|
|
|
|
from lib.diffie_hellman import decrypt
|
|
from subject import main
|
|
|
|
sys.path.append(os.path.abspath("../../"))
|
|
from lib import digest
|
|
from lib import symmetric_encryption
|
|
|
|
logging.basicConfig(format='%(levelname)s\t- %(message)s')
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.INFO)
|
|
|
|
state = main(sys.argv)
|
|
|
|
BASE_DIR = os.path.join(os.path.expanduser('~'), '.sio/')
|
|
|
|
# session file - document name - output file(optional)
|
|
def getDoc(args):
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("-k", '--key', nargs=1, help="Path to the key file")
|
|
parser.add_argument("-r", '--repo', nargs=1, help="Address:Port of the repository")
|
|
parser.add_argument("-v", '--verbose', help="Increase verbosity", action="store_true")
|
|
|
|
parser.add_argument('session', nargs='?', default=None)
|
|
parser.add_argument('name', nargs='?', default=None)
|
|
parser.add_argument('output', nargs='?', default=None)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.session or not args.name:
|
|
logger.error("Need session file and document's name.")
|
|
sys.exit(-1)
|
|
|
|
if (not os.path.isfile(BASE_DIR + args.session)):
|
|
logger.error("File '" + args.session + "' not found.")
|
|
sys.exit(-1)
|
|
|
|
# Get session file content
|
|
with open(BASE_DIR + args.session, 'r') as f:
|
|
args.session = json.load(f)
|
|
|
|
# Get Document metadata
|
|
|
|
doc_name = digest.get_hash(bytes(args.name, encoding='utf-8'))
|
|
|
|
try:
|
|
metadata = requests.get(f'http://{state['REP_ADDRESS']}/file/get/' + doc_name + '/metadata', headers={'Authorization': args.session['token']})
|
|
metadata.raise_for_status()
|
|
|
|
except requests.exceptions.HTTPError:
|
|
logger.error("%d: %s", metadata.status_code, metadata.json()['error'])
|
|
sys.exit(-1)
|
|
|
|
except requests.exceptions.RequestException as errex:
|
|
logger.error("Failed to obtain response from server.")
|
|
sys.exit(1)
|
|
|
|
if metadata.status_code == 200:
|
|
try:
|
|
metadata = json.loads(decrypt(bytes.fromhex(metadata.text), bytes.fromhex(args.session['derived_key'])).decode('utf-8'))
|
|
except Exception:
|
|
logger.error("Failed to decrypt the content")
|
|
sys.exit(1)
|
|
else:
|
|
logger.error("Failed to get metadata")
|
|
sys.exit(-1)
|
|
|
|
try:
|
|
file = requests.get(f'http://{state['REP_ADDRESS']}/file/get/' + metadata['file_handle'] + '/content')
|
|
file.raise_for_status()
|
|
|
|
except requests.exceptions.HTTPError:
|
|
logger.error("%d: %s", file.status_code, file.json()['error'])
|
|
sys.exit(-1)
|
|
|
|
except requests.exceptions.RequestException as errex:
|
|
logger.error("Failed to obtain response from server.")
|
|
sys.exit(-1)
|
|
|
|
file = file.content
|
|
|
|
if not digest.get_hash(file) == metadata['file_handle']:
|
|
logger.error("File's integrity was lost.")
|
|
sys.exit(-1)
|
|
|
|
with open(BASE_DIR + 'encrypted_file', 'wb') as f:
|
|
f.write(file)
|
|
|
|
content = symmetric_encryption.decrypt_file(bytes.fromhex(metadata['key']), BASE_DIR + 'encrypted_file')
|
|
os.remove(BASE_DIR + 'encrypted_file')
|
|
|
|
if args.output:
|
|
with open(BASE_DIR + args.output, 'w') as f:
|
|
f.write(content)
|
|
|
|
else:
|
|
sys.stdout.write(content)
|
|
|
|
sys.exit(0)
|
|
|
|
if __name__ == '__main__':
|
|
getDoc(sys.argv[1:]) |