-
Notifications
You must be signed in to change notification settings - Fork 42
/
shell.py
88 lines (75 loc) · 2.43 KB
/
shell.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#! /usr/bin/env python
"""
a remote python shell
for injection into startserver.py
"""
import os
import select
import socket
import sys
from threading import Thread
from traceback import print_exc
def clientside():
print("client side starting")
host, port = sys.argv[1].split(":")
port = int(port)
myself = open(os.path.abspath(sys.argv[0])).read()
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((host, port))
sock.sendall(repr(myself) + "\n")
print("send boot string")
inputlist = [sock, sys.stdin]
try:
while 1:
r, w, e = select.select(inputlist, [], [])
if sys.stdin in r:
line = raw_input() # noqa:F821
sock.sendall(line + "\n")
if sock in r:
line = sock.recv(4096)
sys.stdout.write(line)
sys.stdout.flush()
except BaseException:
import traceback
print(traceback.print_exc())
sys.exit(1)
class promptagent(Thread):
def __init__(self, clientsock):
print("server side starting")
super.__init__()
self.clientsock = clientsock
def run(self):
print("Entering thread prompt loop")
clientfile = self.clientsock.makefile("w")
filein = self.clientsock.makefile("r")
loc = self.clientsock.getsockname()
while 1:
try:
clientfile.write("%s %s >>> " % loc)
clientfile.flush()
line = filein.readline()
if not line:
raise EOFError("nothing")
if line.strip():
oldout, olderr = sys.stdout, sys.stderr
sys.stdout, sys.stderr = clientfile, clientfile
try:
try:
exec(compile(line + "\n", "<remote pyin>", "single"))
except BaseException:
print_exc()
finally:
sys.stdout = oldout
sys.stderr = olderr
clientfile.flush()
except EOFError:
sys.stderr.write("connection close, prompt thread returns")
break
self.clientsock.close()
sock = globals().get("clientsock")
if sock is not None:
prompter = promptagent(sock)
prompter.start()
print("promptagent - thread started")
else:
clientside()