main.py

import socket

from TCPServer import TCPServer
from UDPServer import UDPServer


def test_udp():
    IP = str(input("Enter IP Address: "))
    SUBNET = str(input("Enter Subnet: "))
    PORT = int(input("Enter Port: "))

    # Test UDP server
    server = UDPServer(IP, 54321, 54322, SUBNET, PORT)
    server()

    # Create a socket for UDP communication
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    client_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

    # Bind the client to a specific port
    client_socket.bind((IP, PORT))
    print(f'{client_socket.getsockname()}')
    while True:
        # Receive data from the server
        data, address = client_socket.recvfrom(1024)
        if data:
            print(f"Received packet from {address}")
            print(f"Data: {data}")


def test_tcp():
    # Create the server:
    IP = str(input("Enter IP Address: "))
    server = TCPServer(IP, 54322, 4)
    server()

    # Test TCP Server
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.bind((IP, 54323))
    print(f'Client on: {client_socket.getsockname()}')

    # Connect to server
    client_socket.connect((IP, 54322))

    client_socket.send(bytes("10\n", "utf-8"))

    while True:
        data, address = client_socket.recvfrom(1024)
        if data:
            print(f"Received packet from {address}")
            print(f"Data: {data}")


if __name__ == "__main__":
    test_tcp()