在Golang中正确使用io.Copy在两个net.Conn TCP连接之间代理数据的方法是什么？

huangapple
go评论114阅读模式
英文:

Correct usage of io.Copy to proxy data between two net.Conn TCP connections in Golang?

问题

在Golang中，复制两个TCP连接之间的数据有不同的用法。以下是两种常见的模式：

模式一：

  1. package main
  3. import (
  4. 	"io"
  5. 	"net"
  6. 	"sync"
  7. )
  9. func proxy1(conn1, conn2 net.Conn) {
  10. 	defer conn1.Close()
  11. 	defer conn2.Close()
  13. 	var wg sync.WaitGroup
  14. 	wg.Add(2)
  16. 	go func() {
  17. 		io.Copy(conn1, conn2)
  18. 		wg.Done()
  19. 	}()
  20. 	go func() {
  21. 		io.Copy(conn2, conn1)
  22. 		wg.Done()
  23. 	}()
  25. 	wg.Wait()
  26. }

模式二：

  1. package main
  3. import (
  4. 	"io"
  5. 	"net"
  6. 	"sync"
  7. )
  9. func proxy2(conn1, conn2 net.Conn) {
  10. 	var wg sync.WaitGroup
  11. 	wg.Add(2)
  13. 	go func() {
  14. 		io.Copy(conn1, conn2)
  15. 		wg.Done()
  16. 		conn1.Close()
  17. 	}()
  18. 	go func() {
  19. 		io.Copy(conn2, conn1)
  20. 		wg.Done()
  21. 		conn2.Close()
  22. 	}()
  24. 	wg.Wait()
  25. }

这两种模式都可以在测试中正常工作，但是你可能会担心边缘情况。其中，模式一在两个goroutine结束后再关闭连接，而模式二在关闭连接之前就结束了goroutine。具体使用哪种模式取决于你的需求和场景。

英文:

There are conflicting usages of io.Copy and I want to know the correct way to copy data between two TCP connections in Golang?

Some people say to call Close inside the goroutines, and others don't
https://stackoverflow.com/questions/32460618/golang-1-5-io-copy-blocked-with-two-tcpconn
https://github.com/LiamHaworth/go-tproxy/blob/master/example/tproxy_example.go#L159

Here are the two common patterns I've seen online, both seemly work fine in my tests, but I'm worried about edge cases and I cannot figure out which, if any, is correct?

  1. package main
  3. import (
  4. 	"io"
  5. 	"net"
  6. 	"sync"
  7. )
  9. func proxy1(conn1, conn2 net.Conn) {
  10. 	defer conn1.Close()
  11. 	defer conn2.Close()
  13. 	var wg sync.WaitGroup
  14. 	wg.Add(2)
  16. 	go func() {
  17. 		io.Copy(conn1, conn2)
  18. 		wg.Done()
  19. 	}()
  20. 	go func() {
  21. 		io.Copy(conn2, conn1)
  22. 		wg.Done()
  23. 	}()
  25. 	wg.Wait()
  26. }
  28. func proxy2(conn1, conn2 net.Conn) {
  29. 	var wg sync.WaitGroup
  30. 	wg.Add(2)
  32. 	go func() {
  33. 		io.Copy(conn1, conn2)
  34. 		wg.Done()
  35. 		conn1.Close()
  36. 	}()
  37. 	go func() {
  38. 		io.Copy(conn2, conn1)
  39. 		wg.Done()
  40. 		conn2.Close()
  41. 	}()
  43. 	wg.Wait()
  44. }

答案1

得分: 2

代理1：当一个对等方在对等方关闭连接之前等待EOF时，代理会发生死锁。

代理2：代理可以在从连接中读取所有数据之前关闭连接。

以下代码修复了这两个问题：

  1. func proxy(conn1, conn2 *net.TCPConn) {
  2.     defer conn1.Close()
  3.     defer conn2.Close()
  5.     var wg sync.WaitGroup
  6.     wg.Add(2)
  8.     go func() {
  9.         defer wg.Done()
  10.         io.Copy(conn1, conn2)
  11.         // 向对等方发送信号，表示没有更多的数据了。
  12.         conn1.CloseWrite()
  13.     }()
  14.     go func() {
  15.         defer wg.Done()
  16.         io.Copy(conn2, conn1)
  17.         // 向对等方发送信号，表示没有更多的数据了。
  18.         conn2.CloseWrite()
  19.     }()
  21.     wg.Wait()
  22. }
英文:

Proxy 1: The proxy deadlocks when a peer waits for EOF before the peer closes the connection.

Proxy 2: The proxy can close a connection before all data is read from the connection.

This code fixes both problems:

  1. func proxy(conn1, conn2 *net.TCPConn) {
  2. 	defer conn1.Close()
  3. 	defer conn2.Close()
  5. 	var wg sync.WaitGroup
  6. 	wg.Add(2)
  8. 	go func() {
  9. 		defer wg.Done()
  10. 		io.Copy(conn1, conn2)
  11.         // Signal peer that no more data is coming.
  12. 		conn1.CloseWrite()
  13. 	}()
  14. 	go func() {
  15. 		defer wg.Done()
  16. 		io.Copy(conn2, conn1)
  17.         // Signal peer that no more data is coming.
  18. 		conn2.CloseWrite()
  19. 	}()
  21. 	wg.Wait()
  22. }

huangapple
  • 本文由 发表于 2023年2月11日 13:05:31
  • 转载请务必保留本文链接：https://go.coder-hub.com/75418196.html
  • go
  • tcp
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定