DEV Community

Cover image for TCP client/server with Python
pikoTutorial
pikoTutorial

Posted on • Originally published at pikotutorial.com

TCP client/server with Python

Welcome to the next pikoTutorial !

The minimal TCP server

A TCP server listens for incoming connections on a specified port and communicates with clients over that connection.

from socket import socket, AF_INET, SOCK_STREAM
from argparse import ArgumentParser
# define command line interface
parser = ArgumentParser()
parser.add_argument('-ip', help='IP')
parser.add_argument('-server_port', type=int, help='Server port')
# parse command line options
args = parser.parse_args()
# create a TCP socket
with socket(AF_INET, SOCK_STREAM) as server_socket:
    # bind server with a specific network address
    # which consists of IP and port number
    server_socket.bind((args.ip, args.server_port))
    # start listening for requests
    server_socket.listen()
    print(f'TCP server up and listening on {args.ip}:{args.server_port}')
    # accept an incoming connection
    client_socket, client_address = server_socket.accept()
    print(f'Connection established with {client_address[0]}:{client_address[1]}')
    # receive data from the client
    data: bytes = client_socket.recv(1024)
    print(f'Received message: {data.decode()}')
    # send response to client
    client_socket.send('Message received'.encode())
Enter fullscreen mode Exit fullscreen mode

The minimal TCP client

A TCP client establishes a connection with a server, sends data, and receives responses.

from socket import socket, AF_INET, SOCK_STREAM
from argparse import ArgumentParser
# define command line interface
parser = ArgumentParser()
parser.add_argument('-ip', help='IP')
parser.add_argument('-server_port', type=int, help='Server port')
# parse command line options
args = parser.parse_args()
# create a TCP socket
with socket(AF_INET, SOCK_STREAM) as client_socket:
    # connect to the server
    client_socket.connect((args.ip, args.server_port))
    # send message to the server
    print(f'Sending message to {args.ip}:{args.server_port}')
    client_socket.send('Hello from TCP client!'.encode())
    # receive response from server
    data: bytes = client_socket.recv(1024)
    print(f'Received response from server: {data.decode()}')
Enter fullscreen mode Exit fullscreen mode

Bidirectional communication

You can now create the following setup:

Image description

Note for beginners: on the image above, I didn’t put any specific number of the client’s port because in the first example that number is assigned by operating system at random. If you want to know how to specify the client’s port as well, you’ll find it in the second example.

Run server with the command below:

python3 server.py -ip 127.0.0.1 -server_port 12345
Enter fullscreen mode Exit fullscreen mode

And the client:

python3 client.py -ip 127.0.0.1 -server_port 12345
Enter fullscreen mode Exit fullscreen mode

You’ll see the following output in the server console:

TCP server up and listening on 127.0.0.1:12345
Connection established with 127.0.0.1:36486
Received message: Hello from TCP client!
Enter fullscreen mode Exit fullscreen mode

And on the client side, you’ll get:

Sending message to 127.0.0.1:12345
Received response from server: Message received
Enter fullscreen mode Exit fullscreen mode

Specifying client's port

In the previous example, although we defined the server port, the client’s port was randomly assigned by the operating system. If you need to specify the client’s port, you can bind it manually before connecting to the server:

from socket import socket, AF_INET, SOCK_STREAM
from argparse import ArgumentParser
# define command line interface
parser = ArgumentParser()
parser.add_argument('-ip', help='IP')
parser.add_argument('-server_port', type=int, help='Server port')
parser.add_argument('-client_port', type=int, help='Client port')
# parse command line options
args = parser.parse_args()
# create a TCP socket
with socket(AF_INET, SOCK_STREAM) as client_socket:
    # bind client to the specific address
    client_socket.bind((args.ip, args.client_port))
    # connect to the server
    client_socket.connect((args.ip, args.server_port))
    # send message to the server
    print(f'Sending message to {args.ip}:{args.server_port}')
    client_socket.send('Hello from TCP client!'.encode())
    # receive response from server
    data: bytes = client_socket.recv(1024)
    print(f'Received response from server: {data.decode()}')
Enter fullscreen mode Exit fullscreen mode

If you run the above script with the following command:

python3 client.py -ip 127.0.0.1 -server_port 12345 -client_port 56789
Enter fullscreen mode Exit fullscreen mode

You’ll get the connection as presented on the image below:

Image description

You'll see the client port number in the server's logs:

TCP server up and listening on 127.0.0.1:12345
Connection established with 127.0.0.1:56789
Received message: Hello from TCP client!
Enter fullscreen mode Exit fullscreen mode

Top comments (0)