mirror of
https://github.com/unode/firefox_decrypt.git
synced 2026-09-27 17:01:50 +02:00
123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import datetime
|
|
import tempfile
|
|
import platform
|
|
from subprocess import run, CalledProcessError, PIPE, STDOUT
|
|
|
|
|
|
class Test:
|
|
def __init__(self):
|
|
self.testdir = os.path.realpath(os.path.dirname(os.path.dirname(__file__)))
|
|
self.platform = platform.system()
|
|
self.interpreter = "python3"
|
|
|
|
self.check_interpreter()
|
|
|
|
def check_interpreter(self):
|
|
out = run([self.interpreter, "--version"], capture_output=True)
|
|
|
|
try:
|
|
major, minor, _ = out.stdout.decode("utf-8").strip().split(" ")[1].split(".")
|
|
valid_version = (int(major) == 3) & (int(minor) >= 9)
|
|
except (ValueError, IndexError, TypeError):
|
|
raise Exception("Couldn't parse version of 'python3 --version'")
|
|
|
|
if not valid_version:
|
|
raise Exception(f"{self.interpreter} binary has version {major}.{minor} but at least 3.9 is required")
|
|
|
|
def run(self, cmd, stdin=None, stderr=STDOUT, workdir=None):
|
|
if stderr == sys.stderr:
|
|
with tempfile.NamedTemporaryFile(mode="w+t") as err:
|
|
try:
|
|
p = run(cmd, check=True, encoding="utf8", cwd=workdir,
|
|
input=stdin, stdout=PIPE, stderr=err)
|
|
except CalledProcessError as e:
|
|
if e.returncode:
|
|
err.flush()
|
|
err.seek(0)
|
|
sys.stderr.write(err.read())
|
|
raise
|
|
|
|
else:
|
|
return p.stdout
|
|
else:
|
|
p = run(cmd, check=True, encoding="utf8", cwd=workdir,
|
|
input=stdin, stdout=PIPE, stderr=stderr)
|
|
|
|
return p.stdout
|
|
|
|
def run_error(self, cmd, returncode, stdin=None, stderr=STDOUT, workdir=None):
|
|
try:
|
|
output = self.run(cmd, stdin=stdin, stderr=stderr, workdir=workdir)
|
|
except CalledProcessError as e:
|
|
if e.returncode != returncode:
|
|
raise ValueError("Expected exit code {} but saw {}".format(returncode, e.returncode))
|
|
else:
|
|
output = e.stdout
|
|
|
|
return output
|
|
|
|
def get_password(self):
|
|
with open(os.path.join(self.get_test_data(), "master_password")) as fh:
|
|
return fh.read()
|
|
|
|
def get_script(self):
|
|
return [self.interpreter, "{}/../firefox_decrypt.py".format(self.testdir)]
|
|
|
|
def get_test_data(self):
|
|
return os.path.join(self.testdir, "test_data")
|
|
|
|
def _get_dir_data(self, subdir, target):
|
|
with open(os.path.join(self.get_test_data(), subdir, "{}.{}".format(target, subdir[:-1]))) as fh:
|
|
return fh.read()
|
|
|
|
def get_user_data(self, target):
|
|
return self._get_dir_data("users", target)
|
|
|
|
def get_output_data(self, target):
|
|
return self._get_dir_data("outputs", target)
|
|
|
|
def get_internal_version(self):
|
|
with open(os.path.join(self.testdir, "..", "CHANGELOG.md")) as fh:
|
|
for line in fh:
|
|
if line.startswith("###") and "." in line:
|
|
return line.strip("# ")
|
|
|
|
def remove_full_pwd(self, output):
|
|
return output.replace(os.path.join(self.testdir, ''), '')
|
|
|
|
def remove_log_date_time(self, input, dropmatches=False):
|
|
output = []
|
|
date = str(datetime.datetime.now().date())
|
|
for line in input.split('\n'):
|
|
if line.startswith(date):
|
|
if not dropmatches:
|
|
output.append(line.split(' ', 2)[-1])
|
|
else:
|
|
output.append(line)
|
|
|
|
return '\n'.join(output)
|
|
|
|
def grep(self, pattern, output, context=0):
|
|
r = re.compile(pattern)
|
|
lines = output.split('\n')
|
|
|
|
acc = []
|
|
for i in range(len(lines)):
|
|
if r.search(lines[i]):
|
|
acc.extend(lines[i-context:1+i+context])
|
|
|
|
return '\n'.join(acc) + '\n'
|
|
|
|
|
|
lib = Test()
|
|
|
|
if __name__ == "__main__":
|
|
pass
|
|
|
|
# vim: ai sts=4 et sw=4
|