net.TCPConn 允许在 FIN 数据包后写入

2024-04-23

我正在尝试为一些服务器端代码编写单元测试,但我在确定关闭测试用例时遇到了困难。环回 TCP 连接似乎无法正确处理干净关闭。我在一个示例应用程序中重现了这一点,该应用程序按顺序执行以下操作:

  1. 创建客户端和服务器连接。
  2. 通过从客户端向服务器成功发送消息来验证连接。
  3. 使用通道告诉服务器调用 conn.Close() 并等待该调用完成。
  4. (尝试)通过再次在客户端连接上调用 Write 来验证连接是否完全中断。

步骤4成功,没有错误。我尝试使用 json.Encoder 和对 TCPConn.Write 的裸调用。我用WireShark检查了流量。服务器发送了一个 FIN 数据包,但客户端从未发送过(即使有 1 秒的睡眠)。服务器甚至发送了一个 RST 数据包来响应 (4),并且客户端 conn.Write 仍然返回 nil 错误。

这看起来完全是疯了。我在这里错过了什么吗?目前运行 Go v1.2.1/Darwin

编辑:强制性重现

package main

import (
  "bufio"
  "fmt"
  "net"
)

var (
  loopback = make(chan string)
  shouldClose = make(chan struct{})
  didClose = make(chan struct{})
)

func serve(listener *net.TCPListener) {
  conn, err := listener.Accept()
  if err != nil {
    panic(err)
  }

  s := bufio.NewScanner(conn)
  if !s.Scan() {
    panic(fmt.Sprint("Failed to scan for line: ", s.Err()))
  }

  loopback <- s.Text() + "\n"

  <-shouldClose
  conn.Close()
  close(didClose)

  if s.Scan() {
    panic("Expected error reading from a socket closed on this side")
  }
}

func main() {
  listener, err := net.ListenTCP("tcp", &net.TCPAddr{})
  if err != nil {
    panic(err)
  }
  go serve(listener)

  conn, err := net.Dial("tcp", listener.Addr().String())
  if err != nil {
    panic(fmt.Sprint("Dialer got error ", err))
  }

  oracle := "Mic check\n"
  if _, err = conn.Write([]byte(oracle)); err != nil {
    panic(fmt.Sprint("Dialer failed to write oracle: ", err))
  }

  test := <-loopback
  if test != oracle {
    panic("Server did not receive the value sent by the client")
  }

  close(shouldClose)
  <-didClose

  // For giggles, I can also add a <-time.After(500 * time.Millisecond)
  if _, err = conn.Write([]byte("This should fail after active disconnect")); err == nil {
    panic("Sender 'successfully' wrote to a closed socket")
  }
}

这就是 TCP 连接主动关闭的工作原理。当客户端检测到服务器已关闭时,预计会关闭其一半的连接。

就您而言,您不是关闭客户端而是发送更多数据。这会导致服务器发送 RST 数据包以强制关闭连接,因为收到的消息无效。

如果您仍然不确定,这里是等效的 python 客户端+服务器,它显示相同的行为。 (我发现使用 python 很有帮助,因为它紧密遵循底层 BSD 套接字 API,而不使用 C)

Server:

import socket, time

server = socket.socket()
server.bind(("127.0.0.1", 9999))
server.listen(1)
sock, addr = server.accept()
msg = sock.recv(1024)
print msg
print "closing"
sock.close()
time.sleep(3)
print "done"

Client:

import socket, time

sock = socket.socket()
sock.connect(("127.0.0.1", 9999))
sock.send("test\n")
time.sleep(1)
print "sending again!"
sock.send("no error here")
time.sleep(1)
print "sending one last time"
sock.send("broken pipe this time")

要正确检测连接上的远程关闭,您应该这样做Read(),并寻找一个io.EOF返回错误。

// we technically need to try and read at least one byte, 
// or we will get an EOF even if the connection isn't closed.
buff := make([]byte, 1)
    if _, err := conn.Read(buff); err != io.EOF {
    panic("connection not closed")
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

net.TCPConn 允许在 FIN 数据包后写入 的相关文章

随机推荐