1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
// ws.go
package main
import (
"encoding/json"
"log"
"net/http"
"time"
"unicode/utf8"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
// okay while you're behind nginx/caddy
CheckOrigin: func(r *http.Request) bool { return true },
}
func wsHandler(hub *Hub) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Printf("upgrade error: %v", err)
return
}
// Expect first message to be auth
_, firstMsg, err := conn.ReadMessage()
if err != nil {
log.Printf("failed to read auth message: %v", err)
conn.Close()
return
}
var auth IncomingMessage
if err := json.Unmarshal(firstMsg, &auth); err != nil {
log.Printf("invalid auth json: %v", err)
_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "bad_json"})
conn.Close()
return
}
if auth.Type != "auth" || auth.Token == "" {
_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "missing_token"})
conn.Close()
return
}
// Lookup token
hub.mu.RLock()
tc, ok := hub.tokens[auth.Token]
hub.mu.RUnlock()
if !ok || tc.Revoked {
_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "invalid_token"})
conn.Close()
return
}
if auth.ServerID == "" {
auth.ServerID = "default"
}
if auth.Nick == "" {
auth.Nick = tc.Name
}
// Enforce nickname length limit on initial/auth nick.
if utf8.RuneCountInString(auth.Nick) > maxNickLen {
log.Printf("auth nick %q too long, falling back to internal name %q", auth.Nick, tc.Name)
auth.Nick = tc.Name
}
if utf8.RuneCountInString(auth.Nick) > maxNickLen {
// still too long, truncate internal name
log.Printf("internal name %q too long, truncating to %d runes", auth.Nick, maxNickLen)
auth.Nick = truncateRunes(auth.Nick, maxNickLen)
}
client := &Client{
hub: hub,
conn: conn,
send: make(chan []byte, 32),
token: tc,
nick: auth.Nick,
serverID: auth.ServerID,
playerUUID: auth.PlayerUUID,
lastSeen: time.Now(),
}
hub.register(client)
if err := conn.WriteJSON(AuthOK{
Type: "auth_ok",
InternalName: tc.Name,
}); err != nil {
log.Printf("failed to send auth_ok: %v", err)
hub.unregister(client)
conn.Close()
return
}
go client.writePump()
client.readPump()
}
}
|
