storm/fenrir
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update terminalManagement

Browse Source
This commit is contained in:
chrys87 2017-08-22 13:08:48 +02:00 committed by GitHub
parent 5c9e984043
commit 0a1ca95836
1 changed files with 52 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
play zone/terminalManagement
View File

@ -1,11 +1,13 @@
import os
import struct
import sys
import pty
import tty
import shlex
import signal
import select
from pathlib import Path
import asyncio
import pyte
class Terminal:
@ -30,7 +32,7 @@ class Terminal:
return {"c": (cursor.x, cursor.y), "lines": lines}
def open_terminal(command="bash", columns=80, lines=24):
def open_terminal(command="bash -i", columns=80, lines=24):
p_pid, master_fd = pty.fork()
if p_pid == 0: # Child.
argv = shlex.split(command)
@ -41,6 +43,7 @@ def open_terminal(command="bash", columns=80, lines=24):
p_out = os.fdopen(master_fd, "w+b", 0)
return Terminal(columns, lines, p_out), p_pid, p_out
def convert_to_text(dump):
lines = dump['lines']
strLines = ''
@ -50,36 +53,54 @@ def convert_to_text(dump):
strLines += c[0]
return strLines
t,p,o = open_terminal()
t.feed(o.read(10000))
convert_to_text(t.dump())
t.feed(b'ls\r')
t.feed(o.read(10000))
convert_to_text(t.dump())
def HandleTerminal():
terminal, p_pid, p_out, master_fd = open_terminal()
while True:
try:
r, w, x = select.select([p_out, master_fd],[],[],500000)
terminal, p_pid, p_out = open_terminal()
signal_pipe = os.pipe()
Running = True
tty.setraw(0)
while Running:
try:
r, w, x = select.select([sys.stdin, p_out,signal_pipe[0]],[],[],0)
if r == []:
continue
for fd in r:
print(fd,r)
msg = fd.read(65536)
if fd == signal_pipe[0]:
msgBytes = os.read(signal_pipe[0], 1)
if fd == sys.stdin:
msgBytes = os.read(sys.stdin.fileno(), 65536)
terminal.feed(msgBytes)
p_out.write(msgBytes)
if fd == p_out:
terminal.feed(msg)
master_fd.write(msg)
if fd == master_fd:
terminal.feed(msg)
p_out.write(msg
print(terminal.screen.display)
try:
msgBytes = p_out.read(65536)
except (EOFError, OSError):
sys.exit(0)
terminal.feed(msgBytes)
os.write(sys.stdout.fileno(), msgBytes)
#print(terminal.screen.display)
except Exception as e: # Process died?
print(e)
#finally:
# os.kill(p_pid, signal.SIGTERM)
# p_out.close()
Running = False
except Exception as e: # Process died?
print(e)
Running = False
finally:
p_out.close()
os.kill(p_pid, signal.SIGTERM)
def get_terminal_size(fd):
s = struct.pack('HHHH', 0, 0, 0, 0)
rows, cols, _, _ = struct.unpack('HHHH', fcntl.ioctl(fd, termios.TIOCGWINSZ, s))
return rows, cols
def resize_terminal(fd):
s = struct.pack('HHHH', 0, 0, 0, 0)
s = fcntl.ioctl(0, termios.TIOCGWINSZ, s)
fcntl.ioctl(fd, termios.TIOCSWINSZ, s)
rows, cols, _, _ = struct.unpack('hhhh', s)
return rows, cols
if __name__ == "__main__":
HandleTerminal()