diff --git a/main.go b/main.go index dd0bbe7..42164fd 100644 --- a/main.go +++ b/main.go @@ -197,6 +197,13 @@ var ( upgrade = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, + CheckOrigin: func(r *http.Request) bool { + if r.Header.Get("Origin") == configFile.Origin { + return true + } else { + return false + } + }, } ) @@ -275,6 +282,7 @@ type Bet struct { type Config struct { ApiKey string `json:"key"` + Origin string `json:"origin"` Rate float64 `json:"rate"` Multiplier float64 `json:"multiplier"` }