mirror of
https://github.com/aljazceru/recon-pipeline.git
synced 2026-01-16 12:54:29 +01:00
b0534ceb27
* removed kali specific installs - test 1 * removed kali specific installs - test 2 * removed kali specific installs - test 2 * removed kali specific installs - test 4 * removed kali specific installs - test 5 * removed kali specific installs - test 6 * removed kali specific installs - test 7 * removed kali specific installs - test 8 * removed kali specific installs - test 9 * removed kali specific installs - test 10 * removed kali specific installs - test 11 * removed kali specific installs - test 12 * removed kali specific installs - test 13 * removed kali specific installs - test 14 * all works locally, fixing up tests et al * trying pipeline again; round 2 * looks good locally; testing pipeline again * test 18? * test 19 * 20 * 21
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import sys
|
|
import subprocess
|
|
from pathlib import Path
|
|
from contextlib import redirect_stdout, redirect_stderr
|
|
|
|
from cmd2.utils import StdSim
|
|
|
|
|
|
def is_kali():
|
|
return any(
|
|
[
|
|
"kali" in x
|
|
for x in subprocess.run("cat /etc/lsb-release".split(), stdout=subprocess.PIPE).stdout.decode().split()
|
|
]
|
|
)
|
|
|
|
|
|
def normalize(block):
|
|
""" Normalize a block of text to perform comparison.
|
|
Strip newlines from the very beginning and very end Then split into separate lines and strip trailing whitespace
|
|
from each line.
|
|
"""
|
|
assert isinstance(block, str)
|
|
block = block.strip("\n")
|
|
return [line.rstrip() for line in block.splitlines()]
|
|
|
|
|
|
def run_cmd(app, cmd):
|
|
""" Clear out and err StdSim buffers, run the command, and return out and err """
|
|
saved_sysout = sys.stdout
|
|
sys.stdout = app.stdout
|
|
|
|
# This will be used to capture app.stdout and sys.stdout
|
|
copy_cmd_stdout = StdSim(app.stdout)
|
|
|
|
# This will be used to capture sys.stderr
|
|
copy_stderr = StdSim(sys.stderr)
|
|
|
|
try:
|
|
app.stdout = copy_cmd_stdout
|
|
with redirect_stdout(copy_cmd_stdout):
|
|
with redirect_stderr(copy_stderr):
|
|
app.onecmd_plus_hooks(cmd)
|
|
finally:
|
|
app.stdout = copy_cmd_stdout.inner_stream
|
|
sys.stdout = saved_sysout
|
|
|
|
out = copy_cmd_stdout.getvalue()
|
|
err = copy_stderr.getvalue()
|
|
return normalize(out), normalize(err)
|
|
|
|
|
|
def setup_install_test(tool=None):
|
|
tools = Path.home() / ".cache" / ".tool-dict.pkl"
|
|
|
|
try:
|
|
tools.unlink()
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
if tool is not None:
|
|
try:
|
|
tool.unlink()
|
|
except (FileNotFoundError, PermissionError):
|
|
pass
|