forked from faucetsdn/ryu
-
Notifications
You must be signed in to change notification settings - Fork 6
/
echo_server.py
50 lines (42 loc) · 1.29 KB
/
echo_server.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
#!/usr/bin/env python
"""
An echo server that uses select to handle multiple clients at a time.
Entering any line of input at the terminal will exit the server.
"""
import select
import socket
import sys
if len(sys.argv)!=2:
print("You need to specify a listening port!")
sys.exit()
host = ''
port = int(sys.argv[1])
backlog = 5
size = 1024
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((host,port))
server.listen(backlog)
input = [server,sys.stdin] if sys.stdin.isatty() else [server]
running = 1
print("Press any key to stop the server...")
while running:
inputready,outputready,exceptready = select.select(input,[],[])
for s in inputready:
if s == server:
# handle the server socket
client, address = server.accept()
print("New client at "+address[0]+":"+str(address[1]))
input.append(client)
elif s == sys.stdin:
# handle standard input
junk = sys.stdin.readline()
running = 0
else:
# handle all other sockets
data = s.recv(size)
if data:
s.send("[from h"+sys.argv[1][0]+"]: "+str(data) )
else:
s.close()
input.remove(s)
server.close()