term: hotfix

Gallopsled · Dec 11, 2023 · 07ea0fa · 07ea0fa
1 parent 84b6c82
commit 07ea0fa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pwnlib/ui.py b/pwnlib/ui.py
@@ -42,8 +42,11 @@ def ehook(*args):
     if "coverage" in sys.modules:
         cmd = "import coverage; coverage.process_startup()\n" + cmd
         env.setdefault("COVERAGE_PROCESS_START", ".coveragerc")
+    env['COLUMNS'] = '80'
+    env['ROWS'] = '24'
     p = process([sys.executable, "-c", cmd], env=env, stderr=subprocess.PIPE)
     # late initialization can lead to EINTR in many places
+    fcntl.ioctl(p.stdout.fileno(), termios.TIOCSWINSZ, struct.pack('HH', 24, 80))
     return p
 
 def yesno(prompt, default=None):