fix: Apply black

pull/1942/head
CristhianMotoche 2 years ago committed by Disconnect3d
parent 5a21adf388
commit fe1b539145

@ -6,22 +6,24 @@ import pwndbg.commands
import pwndbg.gdblib.file
import pwndbg.wrappers.checksec
NEW_LINE = '\n'
NEW_LINE = "\n"
class colors:
RED = "\033[31m"
GREEN = "\033[32m"
BLUE = "\033[34m"
RESET = "\033[0m"
class colors():
RED = '\033[31m'
GREEN = '\033[32m'
BLUE = '\033[34m'
RESET = '\033[0m'
def color_line(line: str) -> str:
return (
line
.replace('*', colors.BLUE + '*' + colors.RESET)
.replace(':', ':' + colors.GREEN)
.replace('No', colors.RED + 'No')
line.replace("*", colors.BLUE + "*" + colors.RESET)
.replace(":", ":" + colors.GREEN)
.replace("No", colors.RED + "No")
) + colors.RESET
def color_lines(lines: list[str]) -> str:
return NEW_LINE.join(list(map(color_line, lines)))
@ -29,4 +31,10 @@ def color_lines(lines: list[str]) -> str:
@pwndbg.commands.ArgparsedCommand("Prints out the binary security settings using `checksec`.")
@pwndbg.commands.OnlyWithFile
def checksec() -> None:
print(color_lines(cast(str, pwndbg.wrappers.checksec.get_raw_out(pwndbg.gdblib.file.get_proc_exe_file())).split(NEW_LINE)))
print(
color_lines(
cast(
str, pwndbg.wrappers.checksec.get_raw_out(pwndbg.gdblib.file.get_proc_exe_file())
).split(NEW_LINE)
)
)

Loading…
Cancel
Save