7250413177

Committed 18 Dec 2023 03:44PM UTC coverage: 71.866% (-2.7%) from 74.55%

Build # 7250413177

Build Type

Pull #2297

github

Committed by

web-flow

Commit Message

Merge fbc1d8e0b into c7649c95e

Pull Request Pull Request #2297: add "retguard" property and display it with checksec

Run Details

4328 of 7244 branches covered (0.0%)

5 of 6 new or added lines in 1 file covered. (83.33%)

464 existing lines in 9 files now uncovered.

12381 of 17228 relevant lines covered (71.87%)

0.72 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

90.74

/pwnlib/py2compat.py

"""
Compatibility layer with python 2, allowing us to write normal code.
Beware, some monkey-patching is done.
"""

import os
import shutil
import sys
try:
    import fcntl
    import termios
except ImportError:
    pass

from collections import namedtuple
from struct import Struct

def py2_monkey_patch(module):
    def decorator(f):
        if sys.version_info < (3,):
            f.__module__ = module.__name__
            setattr(module, f.__name__, f)
    return decorator

# python3 -c 'import shutil,inspect; print(inspect.getsource(shutil.get_terminal_size))'
@py2_monkey_patch(shutil)
def get_terminal_size(fallback=(80, 24)):
    """Get the size of the terminal window.

    For each of the two dimensions, the environment variable, COLUMNS
    and LINES respectively, is checked. If the variable is defined and
    the value is a positive integer, it is used.

    When COLUMNS or LINES is not defined, which is the common case,
    the terminal connected to sys.__stdout__ is queried
    by invoking os.get_terminal_size.

    If the terminal size cannot be successfully queried, either because
    the system doesn't support querying, or because we are not
    connected to a terminal, the value given in fallback parameter
    is used. Fallback defaults to (80, 24) which is the default
    size used by many terminal emulators.

    The value returned is a named tuple of type os.terminal_size.
    """
    # columns, lines are the working values
    try:
        columns = int(os.environ['COLUMNS'])
    except (KeyError, ValueError):
        columns = 0

    try:
        lines = int(os.environ['LINES'])
    except (KeyError, ValueError):
        lines = 0

    # only query if necessary
    if columns <= 0 or lines <= 0:
        try:
            size = os.get_terminal_size(sys.__stdout__.fileno())
        except (AttributeError, ValueError, IOError):
            # stdout is None, closed, detached, or not a terminal, or
            # os.get_terminal_size() is unsupported
            size = os.terminal_size(fallback)
        if columns <= 0:
            columns = size.columns
        if lines <= 0:
            lines = size.lines

    return os.terminal_size((columns, lines))

@py2_monkey_patch(os)
class terminal_size(tuple):
    @property
    def columns(self):
        return self[0]

    @property
    def lines(self):
        return self[1]

    def __repr__(self):
        return 'os.terminal_size(columns=%r, lines=%r)' % self

terminal_size = namedtuple('terminal_size', 'columns lines')

termsize = Struct('HHHH')

@py2_monkey_patch(os)
def get_terminal_size(fd):  # pylint: disable=function-redefined
    arr = b'\0' * termsize.size
    arr = fcntl.ioctl(fd, termios.TIOCGWINSZ, arr)
    lines, columns, xpixel, ypixel = termsize.unpack(arr)
    return os.terminal_size((columns, lines))

1	"""
2	Compatibility layer with python 2, allowing us to write normal code.
3	Beware, some monkey-patching is done.
4	"""
5
6	import os	1✔
7	import shutil	1✔
8	import sys	1✔
9	try:	1✔
10	import fcntl	1✔
11	import termios	1✔
12	except ImportError:	×
13	pass	×
14
15	from collections import namedtuple	1✔
16	from struct import Struct	1✔
17
18	def py2_monkey_patch(module):	1✔
19	def decorator(f):	1✔
20	if sys.version_info < (3,):	1✔
21	f.__module__ = module.__name__	1✔
22	setattr(module, f.__name__, f)	1✔
23	return decorator	1✔
24
25	# python3 -c 'import shutil,inspect; print(inspect.getsource(shutil.get_terminal_size))'
26	@py2_monkey_patch(shutil)	1✔
27	def get_terminal_size(fallback=(80, 24)):	1✔
28	"""Get the size of the terminal window.
29
30	For each of the two dimensions, the environment variable, COLUMNS
31	and LINES respectively, is checked. If the variable is defined and
32	the value is a positive integer, it is used.
33
34	When COLUMNS or LINES is not defined, which is the common case,
35	the terminal connected to sys.__stdout__ is queried
36	by invoking os.get_terminal_size.
37
38	If the terminal size cannot be successfully queried, either because
39	the system doesn't support querying, or because we are not
40	connected to a terminal, the value given in fallback parameter
41	is used. Fallback defaults to (80, 24) which is the default
42	size used by many terminal emulators.
43
44	The value returned is a named tuple of type os.terminal_size.
45	"""
46	# columns, lines are the working values
47	try:	1✔
48	columns = int(os.environ['COLUMNS'])	1✔
49	except (KeyError, ValueError):	1✔
50	columns = 0	1✔
51
52	try:	1✔
53	lines = int(os.environ['LINES'])	1✔
54	except (KeyError, ValueError):	1✔
55	lines = 0	1✔
56
57	# only query if necessary
58	if columns <= 0 or lines <= 0:	1!
59	try:	1✔
60	size = os.get_terminal_size(sys.__stdout__.fileno())	1✔
61	except (AttributeError, ValueError, IOError):	1✔
62	# stdout is None, closed, detached, or not a terminal, or
63	# os.get_terminal_size() is unsupported
64	size = os.terminal_size(fallback)	1✔
65	if columns <= 0:	1!
66	columns = size.columns	1✔
67	if lines <= 0:	1!
68	lines = size.lines	1✔
69
70	return os.terminal_size((columns, lines))	1✔
71
72	@py2_monkey_patch(os)	1✔
73	class terminal_size(tuple):	1✔
74	@property	1✔
75	def columns(self):	1✔
76	return self[0]	1✔
77
78	@property	1✔
79	def lines(self):	1✔
80	return self[1]	1✔
81
82	def __repr__(self):	1✔
83	return 'os.terminal_size(columns=%r, lines=%r)' % self	×
84
85	terminal_size = namedtuple('terminal_size', 'columns lines')	1✔
86
87	termsize = Struct('HHHH')	1✔
88
89	@py2_monkey_patch(os)	1✔
90	def get_terminal_size(fd): # pylint: disable=function-redefined	1✔
91	arr = b'\0' * termsize.size	1✔
92	arr = fcntl.ioctl(fd, termios.TIOCGWINSZ, arr)	1✔
UNCOV 93	lines, columns, xpixel, ypixel = termsize.unpack(arr)	×
UNCOV 94	return os.terminal_size((columns, lines))	×

Gallopsled / pwntools / 7250413177

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous