mirror of https://github.com/r4sas/PBinCLI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
225 lines
7.5 KiB
225 lines
7.5 KiB
import signal, sys |
|
from urllib.parse import parse_qsl |
|
|
|
from pbincli.api import Shortener |
|
from pbincli.format import Paste |
|
from pbincli.utils import PBinCLIError, check_writable, json_encode, uri_validator, validate_url_ending, validate_path_ending |
|
|
|
|
|
def signal_handler(sig, frame): |
|
print('Keyboard interrupt received, terminating…') |
|
sys.exit(0) |
|
|
|
signal.signal(signal.SIGINT, signal_handler) |
|
|
|
|
|
def send(args, api_client, settings=None): |
|
if settings['short']: |
|
shortener = Shortener(settings) |
|
|
|
if not args.notext: |
|
if args.text: |
|
text = args.text |
|
elif args.stdin: |
|
print("Reading text from stdin…") |
|
text = args.stdin.read() |
|
elif not args.file: |
|
PBinCLIError("Nothing to send!") |
|
else: |
|
text = "" |
|
|
|
print("Preparing paste…") |
|
paste = Paste(args.debug) |
|
|
|
if args.verbose: print("Used server: {}".format(api_client.getServer())) |
|
|
|
# get from server supported paste format version and update object |
|
if args.verbose: print("Getting supported paste format version from server…") |
|
version = api_client.getVersion() |
|
paste.setVersion(version) |
|
|
|
if args.verbose: print("Filling paste with data…") |
|
# set compression type, works only on v2 pastes |
|
if version == 2: |
|
paste.setCompression(settings['compression']) |
|
|
|
# add text in paste (if it provided) |
|
paste.setText(text) |
|
|
|
# If we set PASSWORD variable |
|
if args.password: |
|
paste.setPassword(args.password) |
|
|
|
# If we set FILE variable |
|
if args.file: |
|
paste.setAttachment(args.file) |
|
|
|
if args.verbose: print("Encrypting paste…") |
|
paste.encrypt( |
|
formatter = settings['format'], |
|
burnafterreading = settings['burn'], |
|
discussion = settings['discus'], |
|
expiration = settings['expire']) |
|
|
|
if args.verbose: print("Sending request to server…") |
|
request = paste.getJSON() |
|
|
|
if args.debug: print("Passphrase:\t{}\nRequest:\t{}".format(paste.getHash(), request)) |
|
|
|
# If we use dry option, exit now |
|
if args.dry: sys.exit(0) |
|
|
|
print("Uploading paste…") |
|
result = api_client.post(request) |
|
|
|
if args.debug: print("Response:\t{}\n".format(result)) |
|
|
|
# Paste was sent. Checking for returned status code |
|
if not result['status']: # return code is zero |
|
passphrase = paste.getHash() |
|
|
|
# Paste information |
|
print("Paste uploaded!\nPasteID:\t{}\nPassword:\t{}\nDelete token:\t{}".format( |
|
result['id'], |
|
passphrase, |
|
result['deletetoken'])) |
|
|
|
# Paste link |
|
print("\nLink:\t\t{}?{}#{}".format( |
|
settings['server'], |
|
result['id'], |
|
passphrase)) |
|
|
|
# Paste deletion link |
|
print("Delete Link:\t{}?pasteid={}&deletetoken={}".format( |
|
settings['server'], |
|
result['id'], |
|
result['deletetoken'])) |
|
|
|
# Print links to mirrors if present |
|
if settings['mirrors']: |
|
print("\nMirrors:") |
|
urls = settings['mirrors'].split(',') |
|
for x in urls: |
|
print("\t\t{}?{}#{}".format( |
|
validate_url_ending(x), |
|
result['id'], |
|
passphrase)) |
|
|
|
elif result['status']: # return code is other then zero |
|
PBinCLIError("Something went wrong…\nError:\t\t{}".format(result['message'])) |
|
else: # or here no status field in response or it is empty |
|
PBinCLIError("Something went wrong…\nError: Empty response.") |
|
|
|
if settings['short']: |
|
print("\nQuerying URL shortening service…") |
|
shortener.getlink("{}?{}#{}".format( |
|
settings['server'], |
|
result['id'], |
|
passphrase)) |
|
|
|
|
|
def get(args, api_client, settings=None): |
|
parseduri, isuri = uri_validator(args.pasteinfo) |
|
|
|
if isuri and parseduri.query and parseduri.fragment: |
|
api_client.server = args.pasteinfo.split("?")[0] |
|
pasteid = parseduri.query |
|
passphrase = parseduri.fragment |
|
elif parseduri.path and parseduri.path != "/" and parseduri.fragment: |
|
pasteid = parseduri.path |
|
passphrase = parseduri.fragment |
|
else: |
|
PBinCLIError("Provided info hasn't contain valid URL or PasteID#Passphrase string") |
|
|
|
if args.verbose: print("Used server: {}".format(api_client.getServer())) |
|
if args.debug: print("PasteID:\t{}\nPassphrase:\t{}".format(pasteid, passphrase)) |
|
|
|
paste = Paste(args.debug) |
|
|
|
if args.password: |
|
paste.setPassword(args.password) |
|
if args.debug: print("Password:\t{}".format(args.password)) |
|
|
|
if args.verbose: print("Requesting paste from server…") |
|
result = api_client.get(pasteid) |
|
|
|
if args.debug: print("Response:\t{}\n".format(result)) |
|
|
|
# Paste was received. Checking received status code |
|
if not result['status']: # return code is zero |
|
print("Paste received! Decoding…") |
|
|
|
version = result['v'] if 'v' in result else 1 |
|
paste.setVersion(version) |
|
|
|
if version == 2: |
|
if args.debug: print("Authentication data:\t{}".format(result['adata'])) |
|
|
|
paste.setHash(passphrase) |
|
paste.loadJSON(result) |
|
paste.decrypt() |
|
|
|
text = paste.getText() |
|
|
|
if args.debug: print("Decoded text size: {}\n".format(len(text))) |
|
|
|
if len(text): |
|
if args.debug: print("{}\n".format(text.decode())) |
|
if settings['output']: |
|
paste_path = validate_path_ending(settings['output']) + "paste-" + pasteid + ".txt" |
|
else: |
|
paste_path = "paste-" + pasteid + ".txt" |
|
|
|
print("Found text in paste. Saving it to {}".format(paste_path)) |
|
|
|
check_writable(paste_path) |
|
with open(paste_path, "wb") as f: |
|
f.write(text) |
|
f.close() |
|
|
|
attachment, attachment_name = paste.getAttachment() |
|
|
|
if attachment: |
|
if settings['output']: |
|
attachment_path = validate_path_ending(settings['output']) + attachment_name |
|
else: |
|
attachment_path = attachment_name |
|
|
|
print("Found file, attached to paste. Saving it to {}\n".format(attachment_path)) |
|
|
|
check_writable(attachment_path) |
|
with open(attachment_path, "wb") as f: |
|
f.write(attachment) |
|
f.close() |
|
|
|
if version == 1 and 'meta' in result and 'burnafterreading' in result['meta'] and result['meta']['burnafterreading']: |
|
print("Burn afrer reading flag found. Deleting paste…") |
|
api_client.delete(json_encode({'pasteid':pasteid,'deletetoken':'burnafterreading'})) |
|
|
|
elif result['status']: # return code is other then zero |
|
PBinCLIError("Something went wrong…\nError:\t\t{}".format(result['message'])) |
|
else: # or here no status field in response or it is empty |
|
PBinCLIError("Something went wrong…\nError: Empty response.") |
|
|
|
|
|
def delete(args, api_client, settings=None): |
|
parseduri, isuri = uri_validator(args.pasteinfo) |
|
|
|
if isuri: |
|
api_client.server = args.pasteinfo.split("?")[0] |
|
query = dict(parse_qsl(parseduri.query)) |
|
else: |
|
query = dict(parse_qsl(args.pasteinfo)) |
|
|
|
if 'pasteid' in query and 'deletetoken' in query: |
|
pasteid = query['pasteid'] |
|
token = query['deletetoken'] |
|
else: |
|
PBinCLIError("Provided info hasn't contain required information") |
|
|
|
if args.verbose: print("Used server: {}".format(api_client.getServer())) |
|
if args.debug: print("PasteID:\t{}\nToken:\t\t{}".format(pasteid, token)) |
|
|
|
print("Requesting paste deletion…") |
|
api_client.delete(json_encode({'pasteid':pasteid,'deletetoken':token}))
|
|
|