@@ -16,10 +16,13 @@ import (
1616
1717func HandleStream (stream net.Conn , tokenStr string ) {
1818 var err error
19- tokenModel , err := models .DecodeUnverifiedToken (tokenStr )
20- if err != nil {
21- log .Println (err .Error ())
22- return
19+ var tokenModel * models.TokenClaims
20+ if tokenStr != "" {
21+ tokenModel , err = models .DecodeUnverifiedToken (tokenStr )
22+ if err != nil {
23+ log .Println (err .Error ())
24+ //return
25+ }
2326 }
2427 rawMsg , err := msg .ReadMsg (stream )
2528 if err != nil {
@@ -105,6 +108,10 @@ func HandleStream(stream net.Conn, tokenStr string) {
105108 }
106109 case * models.NewSubSession :
107110 {
111+ if tokenStr == "" {
112+ stream .Close ()
113+ return
114+ }
108115 //:TODO 新创建一个全新的子连接
109116 log .Printf ("newSubSession" )
110117 //snappyConn, err := modelsSnappy.Convert(stream, []byte("BUDIS**$(&CHSKCNNCJSH"))
@@ -125,6 +132,10 @@ func HandleStream(stream net.Conn, tokenStr string) {
125132
126133 case * models.RequestNewWorkConn :
127134 {
135+ if tokenStr == "" {
136+ stream .Close ()
137+ return
138+ }
128139 log .Println ("server请求一个新的工作连接" )
129140 stream .Close ()
130141 go newWorkConn (tokenStr )
@@ -144,6 +155,10 @@ func HandleStream(stream net.Conn, tokenStr string) {
144155 case * models.ReqNewP2PCtrlAsServer :
145156 {
146157 log .Printf ("作为listener方式从洞中获取kcp连接" )
158+ if tokenModel == nil {
159+ stream .Close ()
160+ return
161+ }
147162 go func () {
148163 session , listener , err := gateway .MakeP2PSessionAsServer (stream , m , tokenModel )
149164 if err != nil {
@@ -163,6 +178,10 @@ func HandleStream(stream net.Conn, tokenStr string) {
163178 case * models.ReqNewP2PCtrlAsClient :
164179 {
165180 log .Printf ("作为dial方式从从洞中创建kcp连接" )
181+ if tokenModel == nil {
182+ stream .Close ()
183+ return
184+ }
166185 go func () {
167186 session , listener , err := gateway .MakeP2PSessionAsClient (stream , m , tokenModel )
168187 if err != nil {
0 commit comments