feat(ws): controller endpoint with auth

This commit is contained in:
leandrofars 2024-02-06 00:14:54 -03:00
parent c94ab348b5
commit 52ed7bfd8d

View File

@ -130,8 +130,27 @@ func (c *Client) writePump() {
} }
} }
// Handle USP Controller events
func ServeController(w http.ResponseWriter, r *http.Request, token string) {
recv_token := r.URL.Query().Get("token")
if recv_token != token {
w.WriteHeader(http.StatusUnauthorized)
w.Write([]byte("Unauthorized"))
return
}
_, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
}
// Handle USP Agent events // Handle USP Agent events
func ServeAgent(w http.ResponseWriter, r *http.Request) { func ServeAgent(w http.ResponseWriter, r *http.Request) {
//TODO: find out a way to authenticate agents
header := http.Header{ header := http.Header{
"Sec-Websocket-Protocol": {uspVersion}, "Sec-Websocket-Protocol": {uspVersion},
"Sec-Websocket-Version": {wsVersion}, "Sec-Websocket-Version": {wsVersion},