Question

我试图创建一个基于套接字的聊天室，该套接字适用于Windows。

我有一个服务器脚本：

# chat_server.py

import sys
import socket
import select

HOST = ''
SOCKET_LIST = []
RECV_BUFFER = 4096
PORT = 9009

def chat_server():

    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    server_socket.bind((HOST, PORT))
    server_socket.listen(10)
    threads = []

    #add server socket object to the list of readable connections
    SOCKET_LIST.append(server_socket)

    print("Chat server started on port " + str(PORT))

    while 1:
        # get the list sockets wich are ready to be read through select
        # 4th arg, tiome_out = 0 : poll and never block
        ready_to_read,ready_to_write,in_error = select.select(SOCKET_LIST,[],[],0)

        for sock in ready_to_read:
            # a new connection request recieved
            if sock == server_socket:
                sockfd, addr = server_socket.accept()
                SOCKET_LIST.append(sockfd)
                print("Cient (%s, %s) connected" % addr)

                broadcast(server_socket, sockfd, "[%s:%s] entered our chatting room\n" % addr)


            # a message from a client, not a new connection
            else:
                # process data recieved from client,
                try:
                    #receiving data from the socket.
                    data, addr = sock.recvfrom(RECV_BUFFER)
                    if data:
                        # there is something in the socket
                        broadcast(server_socket, sock, "\r" + '[' + str(sock.getpeername()) + ']' + data)
                    else:
                        # remove the socket that's broken
                        if sock in SOCKET_LIST:
                            SOCKET_LIST.remove(sock)

                        # at this stage, no data means probably the connection has been broken
                        broadcast(server_socket, sock, "Client (%s, %s) is offline\n" % addr)

                    # exception
                except:
                    broadcast(server_socket, sock, "Client (%s, %s) is offline\n" % addr)
                    continue

    server_socket.close()

# broadcast chat messages to all connected clients
def broadcast (server_socket, sock, message):
    for socket in SOCKET_LIST:
        # send the message only to peer
        if socket != server_socket and socket != sock :
            try :
                socket.send(message)
            except :
                # broken socket connection
                socket.close()
                # broken socket, remove it
                if socket in SOCKET_LIST:
                    SOCKET_LIST.remove(socket)

if __name__ == "___main__":

    sys.exit(chat_server())

chat_server()

客户端脚本：

# chat_client.py

import sys
import socket
import select
from threading import Thread

def chat_client():
    if(len(sys.argv) < 3):
        print('Usage: python chat_client.py hostname port')
        sys.exit()

    host = sys.argv[1]
    port = int(sys.argv[2])

    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(2)

    #connect to remote host
    try:
        s.connect((host,port))
    except:
        print('Unable to connect')
        sys.exit()

    print('Connected to remote host. You can start sending messages')
    sys.stdout.write('[Me] '); sys.stdout.flush()

    sock_send = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock_send.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    sock_send.bind((host, port))

    def send_msg(sock):
        while True:
            # user entered a message
            s.send(sys.stdin.buffer.readline())
            sys.stdout.write('[Me] '); sys.stdout.flush()

    def recv_msg(sock):
        while True:
            # incoming message from remote server, s
                data, addr = sock.recvfrom(1024)
                if not data :
                    print('\nDisconnected from chat server')
                    sys.exit()
                else:
                    #print data
                    sys.stdout.write(data)
                    sys.stdout.write('[Me] '); sys.stdout.flush()

    Thread(target=send_msg, args=(sock_send,)).start()
    Thread(target=recv_msg, args=(sock_send,)).start()

if __name__ == "__main__":

    sys.exit(chat_client())

程序执行：

$ python chat_server.py

$ python chat_client.py localhost 9009

如果我运行代码，我就不会得到任何错误。当我同时运行多个客户端时，它们都正确连接到服务器，但是一个客户端没有得到另一个客户端编写的文本。

我认为服务器的广播功能有问题，但我不确定它是什么。

我已经搜索了类似的问题，但我找不到任何有用的方法来解决这个问题。请帮助！

基于套接字的聊天程序无法正常工作

0 个答案: