Merge branch 'master' into develop

isekai-main
Julian Lam 2 years ago
commit b551b0ad47

@ -379,7 +379,7 @@ authenticationController.onSuccessfulLogin = async function (req, uid) {
}), }),
user.auth.addSession(uid, req.sessionID), user.auth.addSession(uid, req.sessionID),
user.updateLastOnlineTime(uid), user.updateLastOnlineTime(uid),
user.updateOnlineUsers(uid), user.onUserOnline(uid, Date.now()),
analytics.increment('logins'), analytics.increment('logins'),
db.incrObjectFieldBy('global', 'loginCount', 1), db.incrObjectFieldBy('global', 'loginCount', 1),
]); ]);

@ -112,48 +112,49 @@ async function onMessage(socket, payload) {
return winston.warn('[socket.io] Empty payload'); return winston.warn('[socket.io] Empty payload');
} }
const eventName = payload.data[0]; let eventName = payload.data[0];
const params = typeof payload.data[1] === 'function' ? {} : payload.data[1]; const params = typeof payload.data[1] === 'function' ? {} : payload.data[1];
const callback = typeof payload.data[payload.data.length - 1] === 'function' ? payload.data[payload.data.length - 1] : function () {}; const callback = typeof payload.data[payload.data.length - 1] === 'function' ? payload.data[payload.data.length - 1] : function () {};
if (!eventName) { try {
return winston.warn('[socket.io] Empty method name'); if (!eventName) {
} return winston.warn('[socket.io] Empty method name');
}
if (typeof eventName !== 'string') {
const escapedName = validator.escape(String(eventName));
return callback({ message: `[[error:invalid-event, ${escapedName}]]` });
}
const parts = eventName.split('.'); if (typeof eventName !== 'string') {
const namespace = parts[0]; eventName = typeof eventName;
const methodToCall = parts.reduce((prev, cur) => { const escapedName = validator.escape(eventName);
if (prev !== null && prev[cur] && (!prev.hasOwnProperty || prev.hasOwnProperty(cur))) { return callback({ message: `[[error:invalid-event, ${escapedName}]]` });
return prev[cur];
} }
return null;
}, Namespaces);
if (!methodToCall || typeof methodToCall !== 'function') { const parts = eventName.split('.');
if (process.env.NODE_ENV === 'development') { const namespace = parts[0];
winston.warn(`[socket.io] Unrecognized message: ${eventName}`); const methodToCall = parts.reduce((prev, cur) => {
if (prev !== null && prev[cur] && (!prev.hasOwnProperty || prev.hasOwnProperty(cur))) {
return prev[cur];
}
return null;
}, Namespaces);
if (!methodToCall || typeof methodToCall !== 'function') {
if (process.env.NODE_ENV === 'development') {
winston.warn(`[socket.io] Unrecognized message: ${eventName}`);
}
const escapedName = validator.escape(String(eventName));
return callback({ message: `[[error:invalid-event, ${escapedName}]]` });
} }
const escapedName = validator.escape(String(eventName));
return callback({ message: `[[error:invalid-event, ${escapedName}]]` });
}
socket.previousEvents = socket.previousEvents || []; socket.previousEvents = socket.previousEvents || [];
socket.previousEvents.push(eventName); socket.previousEvents.push(eventName);
if (socket.previousEvents.length > 20) { if (socket.previousEvents.length > 20) {
socket.previousEvents.shift(); socket.previousEvents.shift();
} }
if (!eventName.startsWith('admin.') && ratelimit.isFlooding(socket)) { if (!eventName.startsWith('admin.') && ratelimit.isFlooding(socket)) {
winston.warn(`[socket.io] Too many emits! Disconnecting uid : ${socket.uid}. Events : ${socket.previousEvents}`); winston.warn(`[socket.io] Too many emits! Disconnecting uid : ${socket.uid}. Events : ${socket.previousEvents}`);
return socket.disconnect(); return socket.disconnect();
} }
try {
await checkMaintenance(socket); await checkMaintenance(socket);
await validateSession(socket, '[[error:revalidate-failure]]'); await validateSession(socket, '[[error:revalidate-failure]]');

@ -27,9 +27,13 @@ module.exports = function (User) {
if (now - parseInt(userOnlineTime, 10) < 300000) { if (now - parseInt(userOnlineTime, 10) < 300000) {
return; return;
} }
await db.sortedSetAdd('users:online', now, uid); await User.onUserOnline(uid, now);
topics.pushUnreadCount(uid); topics.pushUnreadCount(uid);
plugins.hooks.fire('action:user.online', { uid: uid, timestamp: now }); };
User.onUserOnline = async (uid, timestamp) => {
await db.sortedSetAdd('users:online', timestamp, uid);
plugins.hooks.fire('action:user.online', { uid, timestamp });
}; };
User.isOnline = async function (uid) { User.isOnline = async function (uid) {

@ -110,8 +110,7 @@ describe('socket.io', () => {
it('should return error for invalid eventName type', (done) => { it('should return error for invalid eventName type', (done) => {
const eventName = ['topics.loadMoreTags']; const eventName = ['topics.loadMoreTags'];
io.emit(eventName, (err) => { io.emit(eventName, (err) => {
const eventAsString = String(eventName); assert.strictEqual(err.message, `[[error:invalid-event, object]]`);
assert.strictEqual(err.message, `[[error:invalid-event, ${eventAsString}]]`);
done(); done();
}); });
}); });

Loading…
Cancel
Save