如何在连接到节点上的 websocket 服务器时传递消息

Question

我这里有这段代码 -

const wss = new SocketServer({ express() });

wss.on('connection', ws => {
    console.log('Client connected');
    ws.on('message', msg => {
        console.log('received: %s', msg);
    });
    ws.on('close', () => console.log('Client disconnected'));
});

现在有一些问题，D-Dos 攻击可能会导致我的服务器崩溃。

在允许陌生人连接到 websocket 之前，有没有办法检查令牌或其他东西？此外，套接字将在另一个域上，因此 cookie 将不起作用。

或者如果用户在连接后没有发送身份验证令牌，有没有办法断开连接。

如果它正在运行并允许每个人无限期地连接，那么这是一个安全风险。服务器可以轻松关闭。

Answer 1

使用 Socket.io 可以实现您需要的设置，请参阅 Authenticating socket io connections

否则你必须自己实现它，但你应该遵循一些标准。我能想到的第一个丑陋的手工解决方案是这样的：

wss.on('connection', ws => {
    console.log('Client connected');

    isAuthenticated = false

    ws.on('message', msg => {
        if (!isAuthenticated ) {
            // check if message is a token
            // else comunicates back the error
        }

        console.log('received: %s', msg);
    });

    ws.on('close', () => console.log('Client disconnected'));
});

对于 DDos 攻击，如果他们有一个令牌，他们可以发送无限的垃圾请求：同样的，你应该自己计算传入的按摩率并实施类似限制策略的东西。