server.py

import subprocess
import socket
import threading
import os
import ssl

os.environ['PYDEVD_DISABLE_FILE_VALIDATION'] = '1'
def send_certificate(client_socket):
    # Path to the server's certificate file
    cert_file_path = "server.crt"

    # Open the certificate file and read its contents
    with open(cert_file_path, "rb") as cert_file:
        certificate_data = cert_file.read()

    # Send the certificate data to the client
    client_socket.sendall(certificate_data)
def main():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    port = 9999
    ipaddress = "localhost" #use ip address here
    server.bind((ipaddress, port))
    server.listen()
    os.system('cls')
    print(f"Server is listening at port: {port}")
    print("Clients: ")
    while True:
        client, addr = server.accept()
        host = addr
        send_certificate(client)
        try:
            # Wrap the client socket with SSL
            server_ssl = ssl.wrap_socket(
                client,
                server_side=True,
                certfile="./server.crt",
                keyfile="./server.key",
                ssl_version=ssl.PROTOCOL_TLSv1_2
            )
            file_name = receive_file_name(server_ssl)
            thread = threading.Thread(target=receivefile, args=(file_name, server_ssl,host))
            thread.start()
            print(f"Active clients: {threading.activeCount()-1}")
        except ssl.SSLError as e:
            print(f"SSL Error: {e}")
        


def receive_file_name(client_socket):
    file_name = b""  # Initialize an empty byte string to store received data
    while True:
        data = client_socket.recv(12) 
        if not data:  
            break
        file_name += data  # Concatenate the received data
        if b"<END_FILENAME>" in file_name:  # Check if the end marker is present in the received data
            file_name = file_name.replace(b"<END_FILENAME>", b"")  # Remove the end marker
            break
    response_message = f"Filename {file_name.decode()} received successfully!"
    client_socket.sendall(response_message.encode())
    return file_name.decode()  # Decode the byte string to get the file name as a string

def compile_and_run(filename):
    error = None
    output = None
    try:
        # Execute the code and capture the output
        if filename.endswith(".py"):
            result = subprocess.run(['python', filename], capture_output=True, text=True, timeout=10)
            output = result.stdout
            error = result.stderr
        elif filename.endswith(".c"):
            compilation = subprocess.run(['gcc', '-o', filename[:-2], filename], capture_output=True, text=True, timeout=10)
            if (compilation.returncode == 0):
                executable = subprocess.run([filename[:-2] + '.exe'], capture_output=True, text=True, timeout=10)
                output = executable.stdout
                error = executable.stderr
        elif filename.endswith(".rs"):
             result = subprocess.run(['rustc', filename], capture_output=True, text=True, timeout=10)
             output= result.stdout
             error = result.stderr

        else:
            pass

        if error:
            return error
        else:
            return output
    except subprocess.TimeoutExpired:
        return "Error: Timeout expired. Execution took too long."
    except Exception as e:
        return f"Error: {str(e)}"
    
def receivefile(file_name,client,host):
    print("Connection from {}:{}\n".format(host[0],host[1]))
    print("Receiving:\n", file_name)
    file = open(file_name, "wb")
    while True:
        data = client.recv(1024)
        if not data:
            break
        if data.endswith(b"<END>"):
            file.write(data[:-5])  # Remove the "<END>" marker before writing to file
            break
        else:
            file.write(data)
    file.close()
    response_message = "File received successfully!"
    client.sendall(response_message.encode())
    output_final = compile_and_run(file_name)
    compile_and_run(file_name)
    client.sendall(output_final.encode())
main()