1
0
mirror of https://github.com/bitdefender/bddisasm.git synced 2024-12-22 22:18:09 +00:00
bddisasm/tests/core/util.py
2024-09-16 12:47:03 +03:00

82 lines
1.7 KiB
Python

#
# Copyright (c) 2024 Bitdefender
# SPDX-License-Identifier: Apache-2.0
#
import sys
import platform
import subprocess
RETCODE_SUCCESS = 0
RETCODE_FAILURE = 1
RETCODE_ERROR = 2
PRINT_COLOR_NONE = ""
PRINT_COLOR_RED = "\x1b[31m"
PRINT_COLOR_GREEN = "\x1b[32m"
PRINT_COLOR_YELLOW = "\x1b[33m"
PRINT_COLOR_BLUE = "\x1b[34m"
PRINT_COLOR_RESET = "\x1b[0m"
def pr_warn(text: str) -> type(None):
"""
Print the provided text - yellow color.
"""
pr_color(text, PRINT_COLOR_YELLOW)
def pr_error(text: str) -> type(None):
"""
Print the provided text - red color.
"""
pr_color(text, PRINT_COLOR_RED)
def pr_info(text: str) -> type(None):
"""
Print the provided text - green color.
"""
pr_color(text, PRINT_COLOR_GREEN)
def pr_debug(text: str) -> type(None):
"""
Print the provided text - no color.
"""
pr_color(text, PRINT_COLOR_NONE)
def pr_color(text, color) -> type(None):
"""
Print the provided text using the provided color.
"""
sys.stdout.write(color + text + PRINT_COLOR_RESET + "\n")
def rsubprocess(cmd) -> tuple[str, str, int]:
"""
Creates a process using the provided command and returns the output.
Handles the shell for Windows.
"""
shell = False
if platform.system() == "Windows":
shell = True
result = subprocess.run(
cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=shell
)
stdout = result.stdout.decode("utf-8")
stderr = result.stderr.decode("utf-8")
returncode = result.returncode
if returncode != 0 or stderr:
pr_warn("[-] failed to run subprocess : %d" % (returncode))
pr_warn("[-] stderr : %s" % (stderr))
if stderr:
returncode = 1
return stdout, stderr, returncode