查找从客户端选择的端口的最佳方法

Question

我的程序使用地址('0.0.0.0', 0) 随机选择服务器端端口，并将唯一的code 发送到客户端，其中套接字试图连接到范围内的每个端口并检查相同的@987654324 @ 确认正确的服务器端口。

问题在于遍历所有端口（@98​​7654325@），即使使用ThreadPool，尝试连接每个端口也非常慢。

---

这只是展示我正在尝试做的一个例子。我的主程序通过互联网而不是本地主机托管。

server.py

import socket 

CODE = 'Code123' # just an example

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
addr = ('0.0.0.0', 0)
server.bind(addr)

# Can see which port is assigned
print(server)

def start():
    server.listen(5) # connect to 5 connection.
    while True:
        conn, addr = server.accept()
        conn.send(CODE.encode('utf-8'))
        print("Code Sent")

start()

cilent.py

import socket
from multiprocessing.pool import ThreadPool

code = 'Code123' # just an example
server = socket.gethostbyname(socket.gethostname())
port_left = 65336
found = False

def connect(port):
    global port_left, code, server, found
    if found: return
    port_left -= 1
    try:
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client.connect((server, port))
        if client.recv(len(code)).decode('utf-8') == code:
            found = True
            return client
        else: client.close()
    except Exception as e:
        print(e, port_left)
    return None

pool = ThreadPool(10)
values = [('c', i) for i in set(pool.map(connect, range(1024, 65336))) if i]
client = dict(values).get('c')
pool.close()
pool.join()
print(client)

---

有没有更好的方法来实现这个目标或提高我现有代码的性能？

Answer 1

不是一个直接的答案，但评论太长了。

这不是随机端口的预期使用方式。它们用于FTP协议中的数据连接方式（被动模式）：

• 客户端在已知端口 (21) 上打开控制连接
• 它通过控制连接向服务器请求随机数据端口
• 它打开数据连接知道它的端口

应避免在随机端口上打开连接，因为如果服务器没有立即响应，客户端必须等待超时来确定服务器是否只是忙。

因此，如果您想使用随机端口，我的建议是保留一个众所周知的端口，客户端将在该端口上获取真正的随机端口。

Answer 2

nmap 提供了执行此操作的方法。这是来自this page的示例代码：

import nmap 

# take the range of ports to  
# be scanned 
begin = 75
end = 80

# assign the target ip to be scanned to 
# a variable 
target = '127.0.0.1'

# instantiate a PortScanner object 
scanner = nmap.PortScanner() 

for i in range(begin,end+1): 

    # scan the target port 
    res = scanner.scan(target,str(i)) 

    # the result is a dictionary containing  
    # several information we only need to 
    # check if the port is opened or closed 
    # so we will access only that information  
    # in the dictionary 
    res = res['scan'][target]['tcp'][i]['state'] 

    print(f'port {i} is {res}.') 

输出：

port 75 is closed.
port 76 is closed.
port 77 is closed.
port 78 is closed.
port 79 is closed.
port 80 is open.

但是，我会说随机分配服务器端口是不切实际的，您最好只分配一个固定端口。