TCP-Kommunikation mit Socket-Modul-Python3

Bitte beachten Sie den vorherigen Artikel. Python3-Socket-Modul und Socket-Kommunikationsfluss Ein einfaches Beispiel für die TCP-Kommunikation mit dem Socket-Modul von Python3

Umgebung

Python: 3.6 Erforderliche Module: colorama

Ausführungsergebnis und Quellcode

Serverseitig (für Kommunikation mit zwei Sockets)

import socket
from colored_print import print_msg


SERVER_ADDRESS = ('192.168.1.200', 8000)

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(SERVER_ADDRESS)

sock.listen(0)

for i in range(2):
    try:
        conn_sock, client_address = sock.accept()
        print_msg('ac', 'The connection accepted.')
        print_msg('i', '{}:{} --------> {}:{}'
                  .format(client_address[0], client_address[1],
                          SERVER_ADDRESS[0], SERVER_ADDRESS[1]))

        # Receiving process
        amount_received = 0
        MSGLEN = int(conn_sock.recv(4))
        print_msg('i', 'MSGLEN: {}'.format(MSGLEN))

        while amount_received < MSGLEN:
            data = conn_sock.recv(min(MSGLEN - amount_received, 32))
            print_msg('i', 'received: {}'.format(data))
            amount_received += len(data)
            if not data:
                raise RuntimeError('The connected socket broken.')

            # Sending process
            conn_sock.send(data)
            print_msg('i', 'send: {}'.format(data))
    finally:
        conn_sock.close()
        print_msg('cl', 'The connection closed.')

Client-Seite

import socket
from colored_print import print_msg


SERVER_ADDRESS = ('192.168.1.200', 8000)
MSG = 'Alice was beginning to get very tired of sitting by her sister on the bank, and of having nothing to do.'
MSGLEN = len(MSG)

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

try:
    sock.connect(SERVER_ADDRESS)
    print_msg('cn', 'The connection accepted')

    print_msg('i', 'MSGLEN: {}'.format(MSGLEN))
    sock.send(bytes(str(MSGLEN), 'utf-8').zfill(4))

    # Sending process
    sock.send(bytes(MSG, 'utf-8'))

    # Receiving process
    chunks = []
    amount_received = 0
    while amount_received < MSGLEN:
        data = sock.recv(min(MSGLEN - amount_received, 32))
        print_msg('i', 'received: {}'.format(data))
        amount_received += len(data)
        if not data:
            raise RuntimeError('The connected socket broken.')
        chunks.append(data)
finally:
    sock.close()
    print_msg('cl', 'The connection closed.')

result = '''
Result: [
    MSGLEN: {},
    Send: {},
    Received: {}
]
'''.format(MSGLEN, MSG, chunks)

print_msg('i', result)

Kommentar

Die Unterbrechung der Recv-Schleife ist

1. Pause, wenn "b" kommt.
2. Wenn Sie die Länge der Nachricht kennen, unterbrechen Sie diese, wenn die Länge der empfangenen Nachricht der Länge der Nachricht entspricht.

Es gibt verschiedene Typen, aber das Problem mit 1. ist, dass "b" übergeben wird, selbst wenn die Kommunikation während der Kommunikation unterbrochen wird. Dieses Mal habe ich die Regel aufgestellt, dass die ersten 4 zu sendenden Bytes die Länge der Nachricht sind.

Serverseite

Eine Socket-Kommunikation von "[ACCEPT]" zu "[CLOSE]".

Senden Sie zurück, sobald Sie eine Nachricht von einem Kunden erhalten.

Client-Seite

Eine Socket-Kommunikation von "[CONNECT]" zu "[CLOSE]".

colored_print Legen Sie den folgenden Code in dasselbe Verzeichnis.

colored_print.py

from colorama import Fore, Style


def print_msg(header, msg):
    '''header are i that is INFO or e that is ERROR'''

    if header == 'i':
        print(Fore.GREEN + '[INFO]',
              Style.RESET_ALL + msg)
    elif header == 'e':
        print(Fore.RED + '[ERROR]',
              Style.RESET_ALL + msg)
    elif header == 'ac':
        print(Fore.BLUE + '[ACCEPT]',
              Style.RESET_ALL + msg)
    elif header == 'cn':
        print(Fore.BLUE + '[CONNECT]',
              Style.RESET_ALL + msg)
    elif header == 'cl':
        print(Fore.BLUE + '[CLOSE]',
              Style.RESET_ALL + msg)
    else:
        print(Fore.RED + 'ERROR: header is an invalid value.'
              + Style.RESET_ALL)