forked from 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.
33 lines
890 B
33 lines
890 B
import json, ntpath, os, zlib |
|
from base64 import b64encode, b64decode |
|
|
|
class PBinCLIException(Exception): |
|
pass |
|
|
|
|
|
def path_leaf(path): |
|
head, tail = ntpath.split(path) |
|
return tail or ntpath.basename(head) |
|
|
|
|
|
def check_readable(f): |
|
# Checks if path exists and readable |
|
if not os.path.exists(f) or not os.access(f, os.R_OK): |
|
raise PBinCLIException("Error accessing path: {}".format(f)) |
|
|
|
|
|
def check_writable(f): |
|
# Checks if path is writable |
|
if not os.access(os.path.dirname(f) or ".", os.W_OK): |
|
raise PBinCLIException("Path is not writable: {}".format(f)) |
|
|
|
|
|
def decompress(s): |
|
return zlib.decompress(bytearray(map(ord, b64decode(s.encode('utf-8')).decode('utf-8'))), -zlib.MAX_WBITS) |
|
|
|
|
|
def compress(s): |
|
co = zlib.compressobj(wbits=-zlib.MAX_WBITS) |
|
b = co.compress(s) + co.flush() |
|
|
|
return b64encode(''.join(map(chr, b)).encode('utf-8'))
|
|
|