Skip to content

Commit

Permalink
pythongh-119770: Make termios ioctl() constants positive (python#119840)
Browse files Browse the repository at this point in the history
  • Loading branch information
vstinner authored and noahbkim committed Jul 11, 2024
1 parent 753fc34 commit 4c3bdae
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 18 deletions.
22 changes: 7 additions & 15 deletions Lib/test/test_ioctl.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,23 +66,15 @@ def test_ioctl_mutate_2048(self):
# Test with a larger buffer, just for the record.
self._check_ioctl_mutate_len(2048)

def test_ioctl_signed_unsigned_code_param(self):
if not pty:
raise unittest.SkipTest('pty module required')
@unittest.skipIf(pty is None, 'pty module required')
def test_ioctl_set_window_size(self):
mfd, sfd = pty.openpty()
try:
if termios.TIOCSWINSZ < 0:
set_winsz_opcode_maybe_neg = termios.TIOCSWINSZ
set_winsz_opcode_pos = termios.TIOCSWINSZ & 0xffffffff
else:
set_winsz_opcode_pos = termios.TIOCSWINSZ
set_winsz_opcode_maybe_neg, = struct.unpack("i",
struct.pack("I", termios.TIOCSWINSZ))

our_winsz = struct.pack("HHHH",80,25,0,0)
# test both with a positive and potentially negative ioctl code
new_winsz = fcntl.ioctl(mfd, set_winsz_opcode_pos, our_winsz)
new_winsz = fcntl.ioctl(mfd, set_winsz_opcode_maybe_neg, our_winsz)
# (rows, columns, xpixel, ypixel)
our_winsz = struct.pack("HHHH", 20, 40, 0, 0)
result = fcntl.ioctl(mfd, termios.TIOCSWINSZ, our_winsz)
new_winsz = struct.unpack("HHHH", result)
self.assertEqual(new_winsz[:2], (20, 40))
finally:
os.close(mfd)
os.close(sfd)
Expand Down
9 changes: 9 additions & 0 deletions Lib/test/test_termios.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,15 @@ def test_constants(self):
self.assertLess(termios.VTIME, termios.NCCS)
self.assertLess(termios.VMIN, termios.NCCS)

def test_ioctl_constants(self):
# gh-119770: ioctl() constants must be positive
for name in dir(termios):
if not name.startswith('TIO'):
continue
value = getattr(termios, name)
with self.subTest(name=name):
self.assertGreaterEqual(value, 0)

def test_exception(self):
self.assertTrue(issubclass(termios.error, Exception))
self.assertFalse(issubclass(termios.error, OSError))
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Make :mod:`termios` ``ioctl()`` constants positive. Patch by Victor Stinner.
18 changes: 15 additions & 3 deletions Modules/termios.c
Original file line number Diff line number Diff line change
Expand Up @@ -1352,9 +1352,21 @@ termios_exec(PyObject *mod)
}

while (constant->name != NULL) {
if (PyModule_AddIntConstant(
mod, constant->name, constant->value) < 0) {
return -1;
if (strncmp(constant->name, "TIO", 3) == 0) {
// gh-119770: Convert value to unsigned int for ioctl() constants,
// constants can be negative on macOS whereas ioctl() expects an
// unsigned long 'request'.
unsigned int value = constant->value & UINT_MAX;
if (PyModule_Add(mod, constant->name,
PyLong_FromUnsignedLong(value)) < 0) {
return -1;
}
}
else {
if (PyModule_AddIntConstant(
mod, constant->name, constant->value) < 0) {
return -1;
}
}
++constant;
}
Expand Down

0 comments on commit 4c3bdae

Please sign in to comment.