套接字 + MySQL 身份验证

Question

我正在尝试通过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可通过握手数据访问。但由于某种原因，它没有通过应用程序传递它。任何帮助不胜感激，谢谢。

Answer 1

你已经很接近了，尽管我建议设置请求标头而不是设置查询字符串参数。授权函数中的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 库。

请务必注意，这不是一个好的身份验证结构。在查询字符串或标头信息中直接传递用户凭据是不安全的。不过，此方法使您更接近更安全的方法。我建议查看像护照这样的身份验证库.js或者每个。您可以潜入此代码以利用这些库在运行检查时存储的会话信息。