node.js - SocketIO + MySQL认证

我正试图通过socketio通过mysql数据库进行身份验证。我已经建立了连接，可以毫无问题地查询结果，但由于某种原因，我无法将用户是否经过身份验证传递到socketio的connection部分。我的想法是我的应用程序有主机和观众。如果在QueryString中连接到应用程序而不发送密码，则该应用程序假定其为查看器并接受连接。如果发送了密码，则会对照数据库进行检查，并接受/拒绝连接。我想要一个变量传递到connection中，这样我就可以在我的应用程序事件中使用它。这是我目前所掌握的，但显然data.query['ishost']并没有进入应用程序。

sio.configure(function() {
    sio.set('authorization', function (data, accept) {
        UserID = data.query['username'];

        try {
            UserID = UserID.toLowerCase();
        } catch(err) {
            return accept("No WebBot Specified. ("+err+")", false);
        }

        // if not sending a password, skip authorization and connect as a viewer
        if (data.query['password'] === 'undefined')
        {
            return accept(null, true);
        }
        // if sending a password, attempt authorization and connect as a host
        else
        {
            client.query(
            'SELECT * FROM web_users WHERE username = "'+UserID+'" LIMIT 1',
              function selectCb(err, results, fields) {
                if (err) {
                  throw err;
                }
                // Found match, hash password and check against DB
                if (results.length != 0)
                {
                    // Passwords match, authenticate.
                    if (hex_md5(data.query['password']) == results[0]['password'])
                    {
                        data.query['ishost'] = true;
                        accept(null, true);
                    }
                    // Passwords don't match, do not authenticate
                    else
                    {
                        data.query['ishost'] = false;
                        return accept("Invalid Password", false);
                    }
                }
                // No match found, add to DB then authenticate
                else
                {
                    client.query(
                        'INSERT INTO web_users (username, password) VALUES ("'+UserID+'", "'+hex_md5(data.query['password'])+'")', null);

                    data.query['ishost'] = "1";
                    accept(null, true);
                }

                client.end();
              }
            );

            // Should never reach this
            return accept("Hacking Attempt", false);
        }

        // Definitely should never reach this
        return accept("Hacking Attempt", false);
    });
});

写入data.query可以通过握手数据访问它。但由于某些原因，它没有通过应用程序传递它。感谢您的帮助。

最佳答案

很接近，不过我建议在设置查询字符串param的同时设置请求头。授权函数中的data变量是握手数据，其中包含可以使用的请求头和cookie信息。下面是一个设置cookie的示例：
在服务器上

io.configure(function() {
    io.set('authorization', function(handshake, callback) {
        var cookie, token, authPair, parts;

        // check for headers
        if (handshake.headers.cookie &&
            handshake.headers.cookie.split('=')[0]=='myapp') {

            // found request cookie, parse it
            cookie   = handshake.headers.cookie;
            token    = cookie.split(/\s+/).pop() || '';
            authPair = new Buffer(token, 'base64').toString();
            parts    = authPair.split(/:/);

            if (parts.length>=1) {
                // assume username & pass provided, check against db
                // parts[0] is username, parts[1] is password
                // .... {db checks}, then if valid....
                callback(null, true);
            } else if(parts.length==1) {
                // assume only username was provided @ parts[0]
                callback(null,true);
            } else {
                // not what we were expecting
                callback(null, false);
            }
        }
        else {
            // auth failed
            callback(null, false);
        }
    });
});

在客户身上
在调用socket.connect之前，请使用您的身份/用户信息设置cookie：

function writeCookie(value, days) {
    var date, expires;

    // days indicates how long the user's session should last
    if (days) {
        date = new Date();
        date.setTime(date.getTime()+(days*24*60*60*1000));
        expires = "; expires="+date.toGMTString();
    } else {
        expires = "";
    }
    document.cookie = "myapp="+Base64.encode(value)+expires+"; path=/";
};

// for a 'viewer' user:
writeCookie('usernameHere', 1);

// for the 'host' user:
writeCookie('usernameHere:passwordHere', 1);

除非您的浏览器支持btoa()，否则您需要在客户端使用base64库。
需要注意的是，这不是一个好的身份验证结构。直接在查询字符串或头信息中传递用户凭据是不安全的。不过，这种方法使您更接近一种更安全的方法。我建议您查看一个auth库，比如passport.js或everyauth。您可以在此代码中进行sub，以利用这些库在运行检查时存储的会话信息。

关于node.js - SocketIO + MySQL认证，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/10609564/