diff --git a/client/control.go b/client/control.go index dcf9b5f..4201f44 100644 --- a/client/control.go +++ b/client/control.go @@ -271,9 +271,9 @@ func (ctl *Control) reader() { var xfrpReader io.Reader if config.ClientCommonCfg.UseEncryption { - xfrpReader := crypto.NewReader(ctl.conn, []byte(config.ClientCommonCfg.PrivilegeToken)) + xfrpReader = crypto.NewReader(ctl.conn, []byte(config.ClientCommonCfg.PrivilegeToken)) } else { - xfrpReader := ctl.conn; + xfrpReader = ctl.conn; } for { if m, err := msg.ReadMsg(xfrpReader); err != nil { diff --git a/server/control.go b/server/control.go index ef8ab6a..73b1e89 100644 --- a/server/control.go +++ b/server/control.go @@ -235,9 +235,9 @@ func (ctl *Control) reader() { var xfrpReader io.Reader if config.ServerCommonCfg.UseEncryption { - xfrpReader := crypto.NewReader(ctl.conn, []byte(config.ServerCommonCfg.PrivilegeToken)) + xfrpReader = crypto.NewReader(ctl.conn, []byte(config.ServerCommonCfg.PrivilegeToken)) } else { - xfrpReader := ctl.conn; + xfrpReader = ctl.conn; } for { if m, err := msg.ReadMsg(xfrpReader); err != nil {