Python Клиент-сервер как справиться с таймаутом - PullRequest
Новая
       11

Python Клиент-сервер как справиться с таймаутом

0 голосов
/ 26 февраля 2020

У меня есть клиент-сервер TCP, где клиент отправляет команды оболочки на сервер, а сервер отвечает выводом указанной команды. Некоторые команды, такие как date или cd, не работают, потому что они интерактивны, я полагаю, и мой код не справляется с этим. Всякий раз, когда я отправляю упомянутые команды, мой клиент, кажется, зависает и просто говорит, что отправка ... с моим сервером никогда ничего не получает.

Я хочу выяснить, как я могу справиться с этим, возможно, с тайм-аутом, когда, если для отправки команды потребуется более 5 секунд, клиент вызовет таймаут или просто обработает его и напечатает сообщение о том, что Команда не была выполнена успешно, пока клиент оставался подключенным к серверу.

Вот мой код

Клиент: 

import socket


# Client
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # creates TCP Socket

# local host and port
LOCAL_HOST = '127.0.0.1'

PORT = 5313
BUFFER_SIZE = 5000  # size of message

sock.settimeout(5)

# connect socket to ip and port
sock.connect((LOCAL_HOST, PORT))
print("Connected to server\n")
print("Enter quit to close connection\n")
while True:

    message = input("Please enter a command:\n")  # ask user to input message
    if message == 'quit':
        break
    if len(message) == 0:
        print("Please enter something")
        message = input("Please enter a command:\n")
    print("Sending %s" % message)
    sock.send(str.encode(message))  # send message
    command = str(sock.recv(BUFFER_SIZE), "utf-8")  # receive message

    print("received %s" % command)
print("closing connection with server")
sock.close()

Сервер: 

import socket
import subprocess


# Server
# Some commands do not work such as cd. Date does not work on windows because it is interactive.

# creates TCP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sock.settimeout(5)

# port and server ip(localhost)
LOCAL_HOST = ''
PORT = 5313

BUFFER_SIZE = 5000  # size of message

# test connection
print("Server starting up on %s with port number %s" % (LOCAL_HOST, PORT))
# bind socket to ip and port
sock.bind((LOCAL_HOST, PORT))
# listen to socket
sock.listen(1)

# socket will accept connection and client address

print("Waiting for connection")  # waiting for connection
connection, address = sock.accept()  # accept connection with client address
print("Connected to", address)  # connected by address
while True:

    command = connection.recv(BUFFER_SIZE)  # receive message from client

    if not command:
        break

    if len(command) > 0:

        terminal = subprocess.Popen(command[:].decode("utf-8"), shell=True, stdout=subprocess.PIPE,
                                    stderr=subprocess.PIPE, stdin=subprocess.PIPE)
        output = terminal.stdout.read() + terminal.stderr.read()
        output_as_string = str(output, "utf-8")
        connection.send(str.encode("0:") + str.encode(output_as_string))
        print(output_as_string)

print("Closing Server")
sock.close()
connection.close()

1 Ответ

1 голос
/ 26 февраля 2020

Есть разные способы сделать это, но, вероятно, самый простой способ - вызвать метод subprocess.check_output () , который позволяет вам указать значение времени ожидания в секундах, а также возвращает текст, который сгенерированный дочерний процесс, то есть: 

[...]

while True:
    command = connection.recv(BUFFER_SIZE)  # receive message from client
    if not command:
        break
    command = command.decode("utf-8").strip()
    if len(command) > 0:
        try:
           output_as_string = subprocess.check_output(command, stderr=subprocess.STDOUT, shell=True, timeout=5.0).decode("utf-8")
        except subprocess.TimeoutExpired:
           output_as_string = "Sub-process timed out!\r\n"

        connection.send(str.encode("0:") + str.encode(output_as_string))

[...]
...