diff options
| author | Valery Piashchynski <[email protected]> | 2021-06-16 14:04:11 +0300 |
|---|---|---|
| committer | Valery Piashchynski <[email protected]> | 2021-06-16 14:04:11 +0300 |
| commit | 8483f605037317e9ac266f893d8ef80ff3a6a56e (patch) | |
| tree | 98b5c6f5a6fd47b39306b5f30bbaa772b7f59893 /plugins/websockets/plugin.go | |
| parent | 9dc98d43b0c0de3e1e1bd8fdc97c122c7c7c594f (diff) | |
- Add origin check for the websockets
Signed-off-by: Valery Piashchynski <[email protected]>
Diffstat (limited to 'plugins/websockets/plugin.go')
| -rw-r--r-- | plugins/websockets/plugin.go | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/plugins/websockets/plugin.go b/plugins/websockets/plugin.go index 6dfe6ca3..8b708187 100644 --- a/plugins/websockets/plugin.go +++ b/plugins/websockets/plugin.go @@ -82,6 +82,9 @@ func (p *Plugin) Init(cfg config.Configurer, log logger.Logger, server server.Se HandshakeTimeout: time.Second * 60, ReadBufferSize: 1024, WriteBufferSize: 1024, + CheckOrigin: func(r *http.Request) bool { + return isOriginAllowed(r.Header.Get("Origin"), p.cfg) + }, } p.serveExit = make(chan struct{}) p.server = server |