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
|
import os
import os.path
import eventlet
from eventlet import backdoor
from eventlet.green import socket
import tests
class BackdoorTest(tests.LimitedTestCase):
def test_server(self):
listener = socket.socket()
listener.bind(('localhost', 0))
listener.listen(50)
serv = eventlet.spawn(backdoor.backdoor_server, listener)
client = socket.socket()
client.connect(('localhost', listener.getsockname()[1]))
self._run_test_on_client_and_server(client, serv)
def _run_test_on_client_and_server(self, client, server_thread):
f = client.makefile('rw')
assert 'Python' in f.readline()
f.readline() # build info
f.readline() # help info
assert 'InteractiveConsole' in f.readline()
self.assertEqual('>>> ', f.read(4))
f.write('print("hi")\n')
f.flush()
self.assertEqual('hi\n', f.readline())
self.assertEqual('>>> ', f.read(4))
f.close()
client.close()
server_thread.kill()
# wait for the console to discover that it's dead
eventlet.sleep(0.1)
@tests.skip_if_no_ipv6
def test_server_on_ipv6_socket(self):
listener = socket.socket(socket.AF_INET6)
listener.bind(('::', 0))
listener.listen(5)
serv = eventlet.spawn(backdoor.backdoor_server, listener)
client = socket.socket(socket.AF_INET6)
client.connect(listener.getsockname())
self._run_test_on_client_and_server(client, serv)
def test_server_on_unix_socket(self):
SOCKET_PATH = '/tmp/eventlet_backdoor_test.socket'
if os.path.exists(SOCKET_PATH):
os.unlink(SOCKET_PATH)
listener = socket.socket(socket.AF_UNIX)
listener.bind(SOCKET_PATH)
listener.listen(5)
serv = eventlet.spawn(backdoor.backdoor_server, listener)
client = socket.socket(socket.AF_UNIX)
client.connect(SOCKET_PATH)
self._run_test_on_client_and_server(client, serv)
def test_quick_client_disconnect(self):
listener = socket.socket()
listener.bind(('localhost', 0))
listener.listen(50)
serv = eventlet.spawn(backdoor.backdoor_server, listener)
client = socket.socket()
client.connect(('localhost', listener.getsockname()[1]))
client.close()
# can still reconnect; server is running
client = socket.socket()
client.connect(('localhost', listener.getsockname()[1]))
client.close()
serv.kill()
# wait for the console to discover that it's dead
eventlet.sleep(0.1)
|
