fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
laodaming 2023-09-05 12:22:32 +08:00
parent 29c509f658
commit 038ac93db4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/websocket/internal/logic/datatransferlogic.go
View File

@ -84,7 +84,7 @@ var (
//允许跨域的origin //允许跨域的origin
mapAllowOrigin = map[string]struct{}{ mapAllowOrigin = map[string]struct{}{
"https://www.fusen.3718.cn": struct{}{}, "https://www.fusen.3718.cn": struct{}{},
//"http://www.fusen.3718.cn": struct{}{}, "http://www.fusen.3718.cn": struct{}{},
} }
) )
@ -117,11 +117,11 @@ type wsConnectItem struct {
// 请求建立连接，升级websocket协议 // 请求建立连接，升级websocket协议
func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) { func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) {
//判断是不是允许的origin //判断是不是允许的origin
if _, ok := mapAllowOrigin[r.Header.Get("Origin")]; !ok { /*if _, ok := mapAllowOrigin[r.Header.Get("Origin")]; !ok {
upgrader.CheckOrigin = func(r *http.Request) bool { upgrader.CheckOrigin = func(r *http.Request) bool {
return false return false
} }
} }*/
//把子协议携带的token设置到标准token头信息中 //把子协议携带的token设置到标准token头信息中
token := r.Header.Get("Sec-Websocket-Protocol") token := r.Header.Get("Sec-Websocket-Protocol")
//有token是正常用户，无则是白板用户，也可以连接 //有token是正常用户，无则是白板用户，也可以连接