Python_FTP/server.py

import os
import socket
import ssl
import sys
import threading
import struct
import time

BUFFER_SIZE = 1024
SERVER_FILE = "serverfile"

def handle_list(conn, args):
    #conn.send(b'150 Opening ASCII mode data connection for file list.\n')
    #home_directory = os.path.expanduser('~')
    os.chdir(os.path.abspath(SERVER_FILE))
    files = os.listdir(os.getcwd())
    for file in files:
        file_info = os.stat(file)
        file_mode = oct(file_info.st_mode)[-3:]
        file_size = file_info.st_size
        file_time = file_info.st_mtime
        file_date = time.strftime('%b %d %H:%M', time.gmtime(file_time))
        file_name = file.encode('utf-8') + b'\r\n'
        conn.sendall(file_mode.encode('utf-8') + b' 1 user user ' + str(file_size).encode('utf-8') + b' ' + file_date.encode('utf-8') + b' ' + file_name)

    if not files:
        conn.sendall(b'226 No files in directory.\n')
    os.chdir("../")
    conn.sendall(b'226 Transfer complete.\n')

def handle_upload(conn, args):
    filename = args[1]
    filesize = int(args[2])
    os.chdir(os.path.abspath(SERVER_FILE))
    print(f'Uploading {filename} ({filesize} bytes)')

    bytes_received = 0
    print("\nReceiving...")
    with open(os.path.join(os.getcwd(), filename), 'wb') as f:
        while bytes_received < filesize:
            data = conn.recv(BUFFER_SIZE)
            if not data:
                break
            f.write(data)
            bytes_received += BUFFER_SIZE

    os.chdir("../")
    conn.sendall(b'Upload done')


def handle_download(conn, args):
    filename = args[1]
    os.chdir(os.path.abspath(SERVER_FILE))
    filesize = os.path.getsize(filename)
    #conn.send(struct.pack('i', filesize))
    conn.sendall(str(filesize).encode('utf-8'))
    bytes_sent = 0
    print("\nSending...")
    with open(filename, 'rb') as f:
        while bytes_sent < filesize:
            data = f.read(BUFFER_SIZE)
            conn.sendall(data)
            bytes_sent += BUFFER_SIZE

    os.chdir("../")
    conn.sendall(b'Download done')


def handle_delete(conn, args):
    filename = args[1]
    os.remove(filename)
    conn.sendall(b'Delete done')


def handle_rename(conn, args):
    os.chdir(os.path.abspath(SERVER_FILE))
    old_filename = args[1]
    new_filename = args[2]
    os.rename(old_filename, new_filename)
    os.chdir("../")
    conn.sendall(f"completed".encode('utf-8'))


def handle_quit(conn, args):
    conn.sendall(b'Quitting')
    conn.close()
    return True


def handle_user(conn, args):
    conn.sendall(b'230 Login successful.\n')


def handle_pwd(conn, args):
    current_dir = os.getcwd()
    conn.sendall(f'257 "{current_dir}" is the current directory.\n'.encode())
    handle_list(conn, args)

def handle_port(conn, args):
    conn.sendall(b'200 OK\n')

commands = {
    'LIST': handle_list,
    'UPLD': handle_upload,
    'DWLD': handle_download,
    'DELF': handle_delete,
    'RNTO': handle_rename,
    'QUIT': handle_quit,
    'USER': handle_user,
    'PWD': handle_pwd,
    'TYPE': lambda conn, args: conn.sendall(b'200 OK\n'),
    'PORT': handle_port,
    'CDUP': lambda conn, args: conn.sendall(b'200 OK\n'),
}


def handle_connection(conn):
    conn.send(b'220 Welcome to the FTP server.\nList of executable commands:\nLIST: List files\nUPLD <file_name>:  '
              b'Upload file\nDWLD <file_name>: Download file\nDELF <file_name>: Delete file\nRNTO <old_name> '
              b'<new_name>: Rename file\nQUIT: Exit')

    while True:

        data = conn.recv(BUFFER_SIZE).decode()
        print(f'Server received: {data}')
        if not data:
            break
        args = data.split()
        command = args[0]
        print(f'Command: {command}')
        if command in commands:
            if commands == handle_quit:
                conn.sendall(b'Closing Connection')
                conn.close()
                break
            commands[command](conn, args)
        else:
            conn.sendall(b'Invalid command.\n')

        # For FTP client testing
        #handle_pwd(conn, [])
        #handle_list(conn, [])

        #if command != 'QUIT':
        #    handle_list(conn, args)


def main():
    # FTP Server should be run with a port number as an argument
    # if len(sys.argv) != 2:
    #    print("Usage: python3 server.py <port>")
    #    return

    #context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
    #context.load_cert_chain("certificate.pem", "privatekey.pem")

    port = 2001
    #server = context.wrap_socket(socket.socket(socket.AF_INET, socket.SOCK_STREAM), server_side=True)
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind(('', port))
    server.listen(1)
    print("Listening on port", port)

    while True:
        try:
            conn, addr = server.accept()
            print("Connected to", addr)
            # Create a new thread to handle the connection
            threading.Thread(target=handle_connection, args=(conn,)).start()
        except KeyboardInterrupt:
            print("Shutting down server")
            server.close()
            return


if __name__ == '__main__':
    main()