tcp-client-server example

Author: anthwal

tcp-client-server/tcp-client.py

@@ -0,0 +1,23 @@
+import socket
+
+# host and port to connect to
+target_host = "0.0.0.0"
+target_port = 9998
+
+# create socket object with the socket module
+client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+
+# connect to the socket
+client.connect((target_host, target_port))
+
+# send some data
+client.send(b"Some data from the client\r\nHost: localhost\r\n\r\n")
+
+# receive some data
+response = client.recv(4096)
+
+print("buffer response:\n\n", response)
+
+print("\n\nresponse decoded to utf8:\n\n", response.decode("utf8"))
+
+client.close()

tcp-client-server/tcp-server.py

@@ -0,0 +1,34 @@
+import socket
+import threading
+
+IP = "0.0.0.0"
+PORT = 9998
+
+
+def main():
+    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    server.bind((IP, PORT))
+    server.listen(5)
+    print(f"[*] Listening on {IP}:{PORT}")
+    # handler loop of the server, it sends each incoming connection
+    # to a separate thread to be handled
+    while True:
+        client, address = server.accept()
+        print(f"[*] Accepted connection from {address[0]}:{address[1]}")
+        client_handler = threading.Thread(target=handle_client, args=(client,))
+        client_handler.start()
+
+
+def handle_client(client_socket):
+    """
+    function to handle request from the client connected, receives data
+    from the client and sends example acknowledgement string back to
+    the client"""
+    with client_socket as sock:
+        request = sock.recv(4096)
+        print(f'[*] Received: {request.decode("utf-8")}')
+        sock.send(b"ACK")
+
+
+if __name__ == "__main__":
+    main()