func TestSayHi(t *testing.T) {
opts := NewOptions()
opts.ClientTimeout = 60 * time.Second
tcpAddr, gfw := mustStartGFW(opts)
defer os.RemoveAll(opts.DataPath)
defer gfw.Exit()
conn, err := mustConnectGFW(tcpAddr)
Equal(t, err, nil)
defer conn.Close()
reqSayHi := &onepiece.SayHiRequest{
Msg: "hi",
}
buffer, err := proto.Marshal(reqSayHi)
_, err = protocol.SendFramedResponse(conn, int32(onepiece.NetMsgID_SAYHI), buffer)
Equal(t, err, nil)
msgid, buffer, err := protocol.ReadUnpackedResponse(conn)
Equal(t, err, nil)
Equal(t, msgid, int32(onepiece.NetMsgID_SAYHI))
resSayHi := &onepiece.SayHiResponse{}
err = proto.Unmarshal(buffer, resSayHi)
Equal(t, err, nil)
Equal(t, resSayHi.Msg, "叔叔不约!")
}
func (p *tcpServer) Handle(clientConn net.Conn) {
p.ctx.gfw.Info("TCP: new client(%s)", clientConn.RemoteAddr())
buf := make([]byte, 4)
_, err := io.ReadFull(clientConn, buf)
if err != nil {
p.ctx.gfw.Error("ERROR: failed to read protocol version - %s", err)
}
protocolMagic := string(buf)
p.ctx.gfw.Info("CLIENT(%s): desired protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)
var prot protocol.Protocol
switch protocolMagic {
case " V1":
prot = &protocolV1{ctx: p.ctx}
default:
protocol.SendFramedResponse(clientConn, int32(onepiece.NetMsgID_ERROR), []byte("E_BAD_PROTOCOL"))
clientConn.Close()
p.ctx.gfw.Info("ERROR: client(%s) bad protocol magic '%s'",
clientConn.RemoteAddr(), protocolMagic)
return
}
err = prot.IOLoop(clientConn)
if err != nil {
p.ctx.gfw.Error("ERROR: client(%s) - %s", clientConn.RemoteAddr(), err)
return
}
}
func (p *protocolV1) Send(client *clientV1, frameType int32, data []byte) error {
client.Lock()
var zeroTime time.Time
if client.HeartbeatInterval > 0 {
client.SetWriteDeadline(time.Now().Add(client.HeartbeatInterval))
} else {
client.SetWriteDeadline(zeroTime)
}
_, err := protocol.SendFramedResponse(client.Writer, frameType, data)
if err != nil {
client.Unlock()
return err
}
err = client.Flush()
client.Unlock()
return err
}