linux的tftp

import socket
import os

SERVER_IP = '0.0.0.0'
SERVER_PORT = 69

def handle_client(client_socket):
    data = client_socket.recv(1024)
    if data[0] == 1:  # Read request
        filename = data[2:data.find(b'\x00', 2)].decode()
        mode = data[data.find(b'\x00', 2) + 1:data.find(b'\x00', data.find(b'\x00', 2) + 1)].decode()
        if mode != 'octet':
            client_socket.send(b'501')
            return
        if os.path.isfile(filename):
            with open(filename, 'rb') as f:
                while True:
                    data = f.read(1024)
                    if not data:
                        break
                    client_socket.send(data)
        else:
            client_socket.send(b'404')
    elif data[0] == 2:  # Write request
        filename = data[2:data.find(b'\x00', 2)].decode()
        mode = data[data.find(b'\x00', 2) + 1:data.find(b'\x00', data.find(b'\x00', 2) + 1)].decode()
        if mode != 'octet':
            client_socket.send(b'501')
            return
        with open(filename, 'wb') as f:
            while True:
                data = client_socket.recv(1024)
                if data == b'\x00':
                    break
                f.write(data)
        client_socket.send(b'226')
    client_socket.close()

def main():
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    server_socket.bind((SERVER_IP, SERVER_PORT))
    print(f'TFTP Server started on {SERVER_IP}:{SERVER_PORT}')
    while True:
        data, addr = server_socket.recvfrom(1024)
        print(f'Received request from {addr}')
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        client_socket.bind((SERVER_IP, 0))
        client_socket.sendto(b'334', addr)
        handle_client(client_socket)

if __name__ == '__main__':
    main()

import socket

SERVER_IP = '192.168.1.1'
SERVER_PORT = 69

def tftp_get(filename):
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    request = bytearray([1]) + filename.encode() + b'\x00octet\x00'
    client_socket.sendto(request, (SERVER_IP, SERVER_PORT))
    data, addr = client_socket.recvfrom(1024)
    if data[0] == 3:
        with open(filename, 'wb') as f:
            while True:
                data, addr = client_socket.recvfrom(1024)
                if data == b'\x00':
                    break
                f.write(data)
        print(f'File {filename} downloaded successfully')
    else:
        print(f'Failed to download file {filename}')
    client_socket.close()

def tftp_put(filename):
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    request = bytearray([2]) + filename.encode() + b'\x00octet\x00'
    client_socket.sendto(request, (SERVER_IP, SERVER_PORT))
    data, addr = client_socket.recvfrom(1024)
    if data[0] == 3:
        with open(filename, 'rb') as f:
            while True:
                data = f.read(1024)
                if not data:
                    break
                client_socket.sendto(data, addr)
        data, addr = client_socket.recvfrom(1024)
        if data == b'226':
            print(f'File {filename} uploaded successfully')
        else:
            print(f'Failed to upload file {filename}')
    else:
        print(f'Failed to upload file {filename}')
    client_socket.close()

if __name__ == '__main__':
    tftp_get('test.txt')
    tftp_put('test.txt')