forked from pcewebpython/echo-server
-
Notifications
You must be signed in to change notification settings - Fork 0
/
echo_client.py
58 lines (52 loc) · 2.33 KB
/
echo_client.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
import socket
import sys
import traceback
def client(msg, log_buffer=sys.stderr):
server_address = ('localhost', 10000)
# TODO: Replace the following line with your code which will instantiate
# a TCP socket with IPv4 Addressing, call the socket you make 'sock'
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_IP)
print('connecting to {0} port {1}'.format(*server_address), file=log_buffer)
# TODO: connect your socket to the server here.
sock.connect(server_address)
# you can use this variable to accumulate the entire message received back
# from the server
received_message = ''
# this try/finally block exists purely to allow us to close the socket
# when we are finished with it
try:
print('sending "{0}"'.format(msg), file=log_buffer)
# TODO: send your message to the server here.
sock.sendall(msg.encode('utf-8'))
# TODO: the server should be sending you back your message as a series
# of 16-byte chunks. Accumulate the chunks you get to build the
# entire reply from the server. Make sure that you have received
# the entire message and then you can break the loop.
#
# Log each chunk you receive. Use the print statement below to
# do it. This will help in debugging problems
num_bytes = 16
while True:
chunk = sock.recv(num_bytes)
print('received "{0}"'.format(chunk.decode('utf8')), file=log_buffer)
received_message += chunk.decode('utf8')
if len(chunk) < num_bytes:
break
except Exception as e:
traceback.print_exc()
sys.exit(1)
finally:
# TODO: after you break out of the loop receiving echoed chunks from
# the server you will want to close your client socket.
print('closing socket', file=log_buffer)
sock.close()
# TODO: when all is said and done, you should return the entire reply
# you received from the server as the return value of this function.
return received_message
if __name__ == '__main__':
if len(sys.argv) != 2:
usage = '\nusage: python echo_client.py "this is my message"\n'
print(usage, file=sys.stderr)
sys.exit(1)
msg = sys.argv[1]
client(msg)