mirror of
https://github.com/python/cpython.git
synced 2025-01-27 11:33:55 +08:00
5e38aae91b
and the passed buffer was exactly 1024 bytes long, the buffer wouldn't be updated back after the system call. Original patch by Brian Brazil.
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
import array
|
|
import unittest
|
|
from test.support import run_unittest, import_module, get_attribute
|
|
import os, struct
|
|
fcntl = import_module('fcntl')
|
|
termios = import_module('termios')
|
|
get_attribute(termios, 'TIOCGPGRP') #Can't run tests without this feature
|
|
|
|
try:
|
|
tty = open("/dev/tty", "r")
|
|
except IOError:
|
|
raise unittest.SkipTest("Unable to open /dev/tty")
|
|
else:
|
|
# Skip if another process is in foreground
|
|
r = fcntl.ioctl(tty, termios.TIOCGPGRP, " ")
|
|
tty.close()
|
|
rpgrp = struct.unpack("i", r)[0]
|
|
if rpgrp not in (os.getpgrp(), os.getsid(0)):
|
|
raise unittest.SkipTest("Neither the process group nor the session "
|
|
"are attached to /dev/tty")
|
|
del tty, r, rpgrp
|
|
|
|
try:
|
|
import pty
|
|
except ImportError:
|
|
pty = None
|
|
|
|
class IoctlTests(unittest.TestCase):
|
|
def test_ioctl(self):
|
|
# If this process has been put into the background, TIOCGPGRP returns
|
|
# the session ID instead of the process group id.
|
|
ids = (os.getpgrp(), os.getsid(0))
|
|
tty = open("/dev/tty", "r")
|
|
r = fcntl.ioctl(tty, termios.TIOCGPGRP, " ")
|
|
rpgrp = struct.unpack("i", r)[0]
|
|
self.assertIn(rpgrp, ids)
|
|
|
|
def _check_ioctl_mutate_len(self, nbytes=None):
|
|
buf = array.array('i')
|
|
intsize = buf.itemsize
|
|
ids = (os.getpgrp(), os.getsid(0))
|
|
# A fill value unlikely to be in `ids`
|
|
fill = -12345
|
|
if nbytes is not None:
|
|
# Extend the buffer so that it is exactly `nbytes` bytes long
|
|
buf.extend([fill] * (nbytes // intsize))
|
|
self.assertEqual(len(buf) * intsize, nbytes) # sanity check
|
|
else:
|
|
buf.append(fill)
|
|
with open("/dev/tty", "r") as tty:
|
|
r = fcntl.ioctl(tty, termios.TIOCGPGRP, buf, 1)
|
|
rpgrp = buf[0]
|
|
self.assertEquals(r, 0)
|
|
self.assertIn(rpgrp, ids)
|
|
|
|
def test_ioctl_mutate(self):
|
|
self._check_ioctl_mutate_len()
|
|
|
|
def test_ioctl_mutate_1024(self):
|
|
# Issue #9758: a mutable buffer of exactly 1024 bytes wouldn't be
|
|
# copied back after the system call.
|
|
self._check_ioctl_mutate_len(1024)
|
|
|
|
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')
|
|
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)
|
|
finally:
|
|
os.close(mfd)
|
|
os.close(sfd)
|
|
|
|
def test_main():
|
|
run_unittest(IoctlTests)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|