首页 > 解决方案 > 将白名单添加到 python 套接字服务器

问题描述

我想知道如何制作白名单或允许的 IP 列表。有人试图加入我的服务器,他使用 VPN 隐藏他的 IP 并更改它。请帮助我,每次他这样做都会关闭我的服务器。

我的代码是:

import socket
import threading


class ChatServer:
    clients_list = []

    last_received_message = ""

    def __init__(self):
        self.server_socket = None
        self.create_listening_server()

    def create_listening_server(self):
        self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        local_ip = '127.0.0.1'
        local_port = 10319
        # this will allow you to immediately restart a TCP server
        self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        # this makes the server listen to requests coming from other computers on the network
        self.server_socket.bind((local_ip, local_port))
        print("Listening for incoming messages..")
        self.server_socket.listen(5)
        self.receive_messages_in_a_new_thread()

    def receive_messages(self, so):
        while True:
            incoming_buffer = so.recv(256)
            if not incoming_buffer:
                break
            self.last_received_message = incoming_buffer.decode('utf-8')
            self.broadcast_to_all_clients(so)  # send to all clients
        so.close()

    def broadcast_to_all_clients(self, senders_socket):
        for client in self.clients_list:
            socket, (ip, port) = client
            if socket is not senders_socket:
                socket.sendall(self.last_received_message.encode('utf-8'))

    def receive_messages_in_a_new_thread(self):
        while True:
            client = so, (ip, port) = self.server_socket.accept()
            self.add_to_clients_list(client)
            print('Connected to ', ip, ':', str(port))
            t = threading.Thread(target=self.receive_messages, args=(so,))
            t.start()

    def add_to_clients_list(self, client):
        if client not in self.clients_list:
            self.clients_list.append(client)


if __name__ == "__main__":
    ChatServer()

receive_messages_in_a_new_thread(self),我正在获取客户端的 IP。我尝试使用:

whitelist = [“client-ip”, “client-ip”]
if ip not in whitelist:
    so.close()
else:
    pass

但它仍然允许其他 IP 进入。我什至尝试了一个黑名单:

blacklist = [“client-ip”, “client-ip”]
if ip in blacklist:
    so.close()
else:
    pass

我错过了什么?还有什么我应该知道的吗?

标签: pythonsockets

解决方案


推荐阅读