python套接字服务器/客户端协议与不稳定的客户端连接

时间:2012-10-15 14:35:35

标签: python sockets

我有一个线程python套接字服务器,为每个连接打开一个新线程。

线程是基于问答的非常简单的通信。 基本上客户端发送初始数据传输,服务器将其运行一个外部应用程序,对传输执行操作并返回服务器将发回的回复,并且循环将再次开始,直到客户端断开连接。

现在因为客户端将在移动电话上,因此连接不稳定,我不再连接开放线程,并且因为循环以recv开始,所以很难以这种方式打破丢失的连接。

我正在考虑在recv之前添加一个send来测试连接是否仍处于活动状态但是如果客户端在我的故障安全发送后断开连接,这可能根本不起作用,因为客户端每5秒发送一次数据流。

我注意到recv有时会破坏但并非总是如此,在这种情况下,我会使用资源留给僵尸线程。

这也可能是我的系统被DOS操作的一个可靠漏洞。 我查看了python手册和Googled,因为我周四试图找到一些东西,但我找到的大多数东西都与客户端和非阻塞模式有关。

有人能指出我正确的方向来解决这个问题吗?

代码示例:

监听器:

serversocket = socket(AF_INET, SOCK_STREAM)
serversocket.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
serversocket.bind(addr)
serversocket.listen(2)
logg("Binded to port: " + str(port))

# Listening Loop
while 1:
  clientsocket, clientaddr = serversocket.accept()
  threading.Thread(target=handler, args=(clientsocket, clientaddr,port,)).start()

# This is useless as it will never get here
serversocket.close()

处理程序:

  # Socket connection handler (Threaded)
  def handler(clientsocket, clientaddr, port):
    clientsocket.settimeout(15)

    # Loop till client closes connection or connection drops
    while 1:
      stream = ''
      while 1:
        ending = stream[-6:] # get stream ending
        if ending == '.$$$$.':
          break

        try:
          data = clientsocket.recv(1)
        except:
          sys.exit()

        if not data:
          sys.exit()
          # this is the usual point where thread is closed when a client closes connection normally

        stream += data

      # Clear the line ending
      stream = base64.b64encode(stream[:-6])

      # Send data to be processed
      re = getreply(stream)

      # Send response to client
      try:
        clientsocket.send(re + str('.$$$$.'))
      except:
        sys.exit()

正如您所看到的,有三个条件,如果连接失败,至少有一个条件应该触发退出,但有时它们不会。

2 个答案:

答案 0 :(得分:2)

很抱歉,但我认为threaded在这种情况下的想法并不好。因为你不需要在这些线程中处理/做很多东西(工作者?)而且大多数时候这些线程都在等待socket(是阻塞操作,不是吗?)我建议你阅读关于event-driven programming。根据套接字,这种模式非常有用,因为你可以在一个线程中完成所有工作。您一次与一个套接字进行通信,但其余连接只是等待数据,因此几乎没有丢失。当您发送几个字节时,您只需检查可能需要携带另一个连接。你可以阅读selectepoll

在python中有几个库可以玩这个:

我在一些项目中使用了龙卷风,这项任务非常好。 Libev也很好,但它是一个c-wrapper所以它有点low-level(但对于某些任务非常好)。

答案 1 :(得分:0)

因此,您应该socket.settimeout(float)使用clientsocket,就像建议的评论之一一样。

您没有看到任何区别的原因是,当您调用socket.recv(bufsize[, flags])并且超时用尽时,将抛出socket.timeout异常,您将捕获该异常并退出。

try:
      data = clientsocket.recv(1)
except:
      sys.exit()

应该是这样的:

try:
      data = clientsocket.recv(1)
except timeout:
      #timeout occurred
      #handle it
      clientsocket.close()
      sys.exit()