diff --git a/tools/utils.js b/tools/utils.js index 4acbbb76c62508f9fe8117096baca1a3a1dd6046..62924bfa2750a641568a4aa41828ce5d6adf8031 100644 --- a/tools/utils.js +++ b/tools/utils.js @@ -422,6 +422,19 @@ function initServeEvent(server) { ); }); }); + + // 增加 token 验证中间件 + if (store.get('token')) { + server.use((socket, next) => { + const token = socket.handshake.auth.token || socket.handshake.query.token // 获取 token + if (token === store.get('token')) { + // 校验 token + return next() // 验证通过 + } else { + return next(new Error('Authentication error')) // 验证失败 + } + }) + } } /**