114 lines
3.6 KiB
Python
Executable File
114 lines
3.6 KiB
Python
Executable File
#!/bin/python3
|
|
import os
|
|
import sys
|
|
import logging
|
|
import requests
|
|
import json
|
|
import argparse
|
|
|
|
from subject import main
|
|
|
|
from lib.diffie_hellman import *
|
|
|
|
sys.path.append(os.path.abspath("../../"))
|
|
from lib.symmetric_encryption import *
|
|
from lib import digest
|
|
|
|
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 - file
|
|
def addDoc(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('file', nargs='?', default=None)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.session or not args.name or not args.file:
|
|
logger.error("Need session file, document's name and file to upload.")
|
|
sys.exit(1)
|
|
|
|
if (not os.path.isfile(os.path.join(BASE_DIR, args.session))):
|
|
logger.error("File '" + args.session + "' not found.")
|
|
sys.exit(1)
|
|
|
|
if (not os.path.isfile(os.path.join(BASE_DIR, args.file))):
|
|
logger.error("File '" + args.file + "' not found")
|
|
sys.exit(1)
|
|
|
|
#Get session file content
|
|
with open(os.path.join(BASE_DIR, args.session), 'r') as f:
|
|
args.session = json.load(f)
|
|
|
|
#Encrypt content
|
|
key, content = encrypt_file(BASE_DIR + args.file, BASE_DIR + 'encryptedText')
|
|
|
|
derived_key = bytes.fromhex(args.session['derived_key'])
|
|
#Upload document metadata
|
|
doc = json.dumps({'document_name' : args.name, 'key' : key.hex(), 'alg' : 'AES-CFB' })
|
|
try:
|
|
doc = encrypt(doc, derived_key).hex()
|
|
except Exception:
|
|
logger.error("Failed to encrypt the content")
|
|
sys.exit(-1)
|
|
|
|
headers = {
|
|
'Authorization': args.session['token'],
|
|
'Content-Type': 'application/octet-stream'
|
|
}
|
|
|
|
try:
|
|
req = requests.post(f'http://{state['REP_ADDRESS']}/file/upload/metadata',
|
|
data=doc,
|
|
headers=headers)
|
|
req.raise_for_status()
|
|
|
|
except requests.exceptions.HTTPError:
|
|
logger.error("%d: %s", req.status_code, req.json()['error'])
|
|
sys.exit(-1)
|
|
|
|
except requests.exceptions.RequestException as errex:
|
|
logger.error("Failed to obtain response from server.")
|
|
sys.exit(-1)
|
|
|
|
#Upload Document content
|
|
file = {'file' : (BASE_DIR + args.file, content)}
|
|
|
|
try:
|
|
req = requests.post(f'http://{state['REP_ADDRESS']}/file/upload/content',
|
|
files=file,
|
|
headers={'Authorization': args.session['token'], 'File-Checksum' : digest.get_hash(content)})
|
|
req.raise_for_status()
|
|
|
|
except requests.exceptions.HTTPError:
|
|
logger.error("%d: %s", req.status_code, req.json()['error'])
|
|
sys.exit(-1)
|
|
|
|
except requests.exceptions.RequestException as errex:
|
|
logger.error("Failed to obtain response from server.")
|
|
os.remove(BASE_DIR + 'encryptedText')
|
|
sys.exit(1)
|
|
|
|
#Delete temporary file
|
|
os.remove(BASE_DIR + 'encryptedText')
|
|
|
|
if req.status_code == 201:
|
|
logger.info("Document uploaded successfully.")
|
|
sys.exit(0)
|
|
logger.error("Failed to add document.")
|
|
sys.exit(-1)
|
|
|
|
if __name__ == '__main__':
|
|
addDoc(sys.argv[1:]) |