def syntax_highlight(code, filename='.asm'):
# No syntax highlight if pygment is not installed
if not pygments:
return code
filename = os.path.basename(filename)
formatter = pygments.formatters.Terminal256Formatter(
style=str(style)
)
lexer = None
# If source code is asm, use our customized lexer.
# Note: We can not register our Lexer to pygments and use their APIs,
# since the pygment only search the lexers installed via setuptools.
for glob_pat in PwntoolsLexer.filenames:
pat = '^' + glob_pat.replace('.', r'\.').replace('*', r'.*') + '$'
if re.match(pat, filename):
lexer = PwntoolsLexer()
break
if not lexer:
try:
lexer = pygments.lexers.guess_lexer_for_filename(filename, code)
except pygments.util.ClassNotFound:
# no lexer for this file or invalid style
pass
if lexer:
code = pygments.highlight(code, lexer, formatter).rstrip()
return code
try:
import pygments
import pygments.lexers
import pygments.formatters
from pwndbg.color.lexer import PwntoolsLexer
except ImportError:
pygments = None
pwndbg.config.Parameter('syntax-highlight', True,
'Source code / assembly syntax highlight')
style = theme.Parameter(
'syntax-highlight-style', 'monokai',
'Source code / assembly syntax highlight stylename of pygments module')
formatter = pygments.formatters.Terminal256Formatter(style=str(style))
pwntools_lexer = PwntoolsLexer()
lexer_cache = {}
@pwndbg.config.Trigger([style])
def check_style():
global formatter
try:
formatter = pygments.formatters.Terminal256Formatter(style=str(style))
# Reset the highlighted source cache
from pwndbg.commands.context import get_highlight_source
get_highlight_source._reset()
except pygments.util.ClassNotFound:
print(
message.warn(