package main
import (
"net"
"log"
"fmt"
"os"
"io"
)
func main() {
l, err := net.Listen("tcp", "127.0.0.1:8888")
defer func() {
if err := recover(); err != nil {
log.Fatal(err)
}
}()
if err != nil {
log.Fatal(err)
}
// pxy
dstConn, err := net.Dial("tcp", "192.168.199.201:9999")
if err != nil {
log.Fatal("连接后端失败:", err)
}
defer dstConn.Close()
defer l.Close()
for {
conn, err := l.Accept()
if err != nil || err == io.EOF {
fmt.Println("accept error:", err)
os.Exit(1)
}
fmt.Printf("message %s; %s\n", conn.RemoteAddr(), conn.LocalAddr())
go handleConn(conn, dstConn)
}
}
func handleConn(conn net.Conn, dstConn net.Conn) {
ip := conn.RemoteAddr().String()
defer func() {
fmt.Println("disconnect:", ip)
conn.Close()
}()
fmt.Println("ip:", ip)
data := make([]byte, 128)
for {
_, err := conn.Read(data)
if err != nil || err == io.EOF {
log.Println("连接已断开!")
break
}
// fmt.Printf("收到%d个字符, %s\n", len, data)
// conn.Write(data)
// 转发
go io.Copy(conn, dstConn)
io.Copy(dstConn, conn)
}
}
发现客户端第一次发送的消息总是不转发,以后的消息才转发,问题在哪里?
这里读出来的并没有转发