natpass/code/client/pool/send.go

package pool

import (
	"natpass/code/client/global"
	"natpass/code/network"
	"time"
)

// SendConnectReq send connect request message
func (conn *Conn) SendConnectReq(id string, cfg global.Tunnel) {
	tp := network.ConnectRequest_tcp
	if cfg.Type != "tcp" {
		tp = network.ConnectRequest_udp
	}
	var msg network.Msg
	msg.To = cfg.Target
	msg.XType = network.Msg_connect_req
	msg.Payload = &network.Msg_Creq{
		Creq: &network.ConnectRequest{
			Id:    id,
			Name:  cfg.Name,
			XType: tp,
			Addr:  cfg.RemoteAddr,
			Port:  uint32(cfg.RemotePort),
		},
	}
	select {
	case conn.write <- &msg:
	case <-time.After(global.WriteTimeout):
	}
}

// SendConnectError send connect error response message
func (conn *Conn) SendConnectError(to, id, info string) {
	var msg network.Msg
	msg.To = to
	msg.XType = network.Msg_connect_rep
	msg.Payload = &network.Msg_Crep{
		Crep: &network.ConnectResponse{
			Id:  id,
			Ok:  false,
			Msg: info,
		},
	}
	select {
	case conn.write <- &msg:
	case <-time.After(global.WriteTimeout):
	}
}

// SendConnectOK send connect success response message
func (conn *Conn) SendConnectOK(to, id string) {
	var msg network.Msg
	msg.To = to
	msg.XType = network.Msg_connect_rep
	msg.Payload = &network.Msg_Crep{
		Crep: &network.ConnectResponse{
			Id: id,
			Ok: true,
		},
	}
	select {
	case conn.write <- &msg:
	case <-time.After(global.WriteTimeout):
	}
}

// SendData forward data
func (conn *Conn) SendData(id, target string, data []byte) {
	dup := func(data []byte) []byte {
		ret := make([]byte, len(data))
		copy(ret, data)
		return ret
	}
	var msg network.Msg
	msg.To = target
	msg.XType = network.Msg_forward
	msg.Payload = &network.Msg_XData{
		XData: &network.Data{
			Lid:  id,
			Data: dup(data),
		},
	}
	select {
	case conn.write <- &msg:
	case <-time.After(global.WriteTimeout):
	}
}