init: 自 zomaintain/backend/zonat 平移独立成库; module path 改为 git.zeroonesoft.cn/golib/zonat; 内部 import 同步重写

agent/forward.go

@@ -0,0 +1,52 @@
+package agent
+
+import (
+	"io"
+	"net"
+)
+
+// handleData 处理节点主动打开的数据流：
+// 首帧 CmdTarget 携带目标信息 → 拨号本地 TCP 目标 → 回执 → 双向管道。
+func (a *Agent) handleData(stream net.Conn) {
+	defer stream.Close()
+	f, err := ReadFrame(stream)
+	if err != nil {
+		return
+	}
+	if f.Cmd() != CmdTarget {
+		a.logger().Warnf("agent 数据流首帧命令错误 cmd=%d", f.Cmd())
+		return
+	}
+	t := Tunnel{}
+	if err := f.Unmarshal(&t); err != nil {
+		return
+	}
+
+	conn, err := net.Dial("tcp", t.TargetAddr())
+	if err != nil {
+		a.logger().Warnf("agent 拨号本地目标失败 target=%s err=%s", t.TargetAddr(), err.Error())
+		rf := NewFrame(FrameVersion, CmdTarget, f.StreamID())
+		_ = rf.Marshal(Ret{Code: 1, Msg: err.Error()})
+		_ = WriteFrame(stream, rf)
+		return
+	}
+	defer conn.Close()
+
+	rf := NewFrame(FrameVersion, CmdTarget, f.StreamID())
+	_ = rf.Marshal(Ret{Code: 0, Msg: "ok"})
+	if err := WriteFrame(stream, rf); err != nil {
+		return
+	}
+
+	pipe(conn, stream)
+}
+
+// pipe 双向拷贝，任一方向结束即关闭两端
+func pipe(a, b net.Conn) {
+	go func() {
+		io.Copy(b, a)
+		b.Close()
+	}()
+	io.Copy(a, b)
+	a.Close()
+}