remove chainpad server
ansuz
9 years ago
parent
46b009d20f
commit
b09b73fcc6
@ -1,197 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2014 XWiki SAS
|
|
||||||
*
|
|
||||||
* This program is free software: you can redistribute it and/or modify
|
|
||||||
* it under the terms of the GNU Affero General Public License as published by
|
|
||||||
* the Free Software Foundation, either version 3 of the License, or
|
|
||||||
* (at your option) any later version.
|
|
||||||
*
|
|
||||||
* This program is distributed in the hope that it will be useful,
|
|
||||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
* GNU Affero General Public License for more details.
|
|
||||||
*
|
|
||||||
* You should have received a copy of the GNU Affero General Public License
|
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
// triggers jsh because we're not in a browser and WebSocket is a thing
|
|
||||||
var WebSocket = require('ws'); // jshint ignore:line
|
|
||||||
|
|
||||||
var REGISTER = 0;
|
|
||||||
var REGISTER_ACK = 1;
|
|
||||||
var PATCH = 2;
|
|
||||||
var DISCONNECT = 3;
|
|
||||||
var PING = 4;
|
|
||||||
var PONG = 5;
|
|
||||||
|
|
||||||
var parseMessage = function (msg) {
|
|
||||||
var res ={};
|
|
||||||
// two or more? use a for
|
|
||||||
['pass','user','channelId','content'].forEach(function(attr){
|
|
||||||
var len=msg.slice(0,msg.indexOf(':')),
|
|
||||||
// taking an offset lets us slice out the prop
|
|
||||||
// and saves us one string copy
|
|
||||||
o=len.length+1,
|
|
||||||
prop=res[attr]=msg.slice(o,Number(len)+o);
|
|
||||||
// slice off the property and its descriptor
|
|
||||||
msg = msg.slice(prop.length+o);
|
|
||||||
});
|
|
||||||
// content is the only attribute that's not a string
|
|
||||||
res.content=JSON.parse(res.content);
|
|
||||||
return res;
|
|
||||||
};
|
|
||||||
|
|
||||||
// get the password off the message before sending it to other clients.
|
|
||||||
var popPassword = function (msg) {
|
|
||||||
var passLen = msg.substring(0,msg.indexOf(':'));
|
|
||||||
return msg.substring(passLen.length+1 + Number(passLen));
|
|
||||||
};
|
|
||||||
|
|
||||||
var sendMsg = function (msg, socket) {
|
|
||||||
socket.send(msg);
|
|
||||||
};
|
|
||||||
|
|
||||||
var sendChannelMessage = function (ctx, channel, msg, cb) {
|
|
||||||
ctx.store.message(channel.name, msg, function () {
|
|
||||||
channel.forEach(function (user) {
|
|
||||||
try {
|
|
||||||
sendMsg(msg, user.socket);
|
|
||||||
} catch (e) {
|
|
||||||
console.log(e.stack);
|
|
||||||
try { user.socket.close(); } catch (e) { }
|
|
||||||
}
|
|
||||||
});
|
|
||||||
if (cb) { cb(); }
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
var mkMessage = function (user, channel, content) {
|
|
||||||
content = JSON.stringify(content);
|
|
||||||
return user.length + ':' + user +
|
|
||||||
channel.length + ':' + channel +
|
|
||||||
content.length + ':' + content;
|
|
||||||
};
|
|
||||||
|
|
||||||
var dropClient = function (ctx, userpass) {
|
|
||||||
var client = ctx.registeredClients[userpass];
|
|
||||||
if (client.socket.readyState !== WebSocket.CLOSING
|
|
||||||
&& client.socket.readyState !== WebSocket.CLOSED)
|
|
||||||
{
|
|
||||||
try {
|
|
||||||
client.socket.close();
|
|
||||||
} catch (e) {
|
|
||||||
console.log("Failed to disconnect ["+client.userName+"], attempting to terminate");
|
|
||||||
try {
|
|
||||||
client.socket.terminate();
|
|
||||||
} catch (ee) {
|
|
||||||
console.log("Failed to terminate ["+client.userName+"] *shrug*");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (var i = 0; i < client.channels.length; i++) {
|
|
||||||
var chanName = client.channels[i];
|
|
||||||
var chan = ctx.channels[chanName];
|
|
||||||
var idx = chan.indexOf(client);
|
|
||||||
if (idx < 0) { continue; }
|
|
||||||
console.log("Removing ["+client.userName+"] from channel ["+chanName+"]");
|
|
||||||
chan.splice(idx, 1);
|
|
||||||
if (chan.length === 0) {
|
|
||||||
console.log("Removing empty channel ["+chanName+"]");
|
|
||||||
delete ctx.channels[chanName];
|
|
||||||
} else {
|
|
||||||
sendChannelMessage(ctx, chan, mkMessage(client.userName, chanName, [DISCONNECT,0]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
delete ctx.registeredClients[userpass];
|
|
||||||
};
|
|
||||||
|
|
||||||
var handleMessage = function (ctx, socket, msg) {
|
|
||||||
var parsed = parseMessage(msg);
|
|
||||||
var userPass = parsed.user + ':' + parsed.pass;
|
|
||||||
msg = popPassword(msg);
|
|
||||||
|
|
||||||
if (parsed.content[0] === REGISTER) {
|
|
||||||
if (parsed.user.length === 0) { throw new Error(); }
|
|
||||||
console.log("[" + userPass + "] registered");
|
|
||||||
var user = ctx.registeredClients[userPass] = ctx.registeredClients[userPass] || {
|
|
||||||
channels: [parsed.channelId],
|
|
||||||
userName: parsed.user
|
|
||||||
};
|
|
||||||
if (user.socket && user.socket !== socket) { user.socket.close(); }
|
|
||||||
user.socket = socket;
|
|
||||||
|
|
||||||
var chan = ctx.channels[parsed.channelId] = ctx.channels[parsed.channelId] || [];
|
|
||||||
var newChan = (chan.length === 0);
|
|
||||||
chan.name = parsed.channelId;
|
|
||||||
|
|
||||||
// we send a register ack right away but then we fallthrough
|
|
||||||
// to let other users know that we were registered.
|
|
||||||
sendMsg(mkMessage('', parsed.channelId, [1,0]), socket);
|
|
||||||
|
|
||||||
var sendMsgs = function () {
|
|
||||||
sendChannelMessage(ctx, chan, msg, function () {
|
|
||||||
chan.push(user);
|
|
||||||
ctx.store.getMessages(chan.name, function (msg) {
|
|
||||||
try {
|
|
||||||
sendMsg(msg, socket);
|
|
||||||
} catch (e) {
|
|
||||||
console.log(e.stack);
|
|
||||||
try { socket.close(); } catch (e) { }
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
};
|
|
||||||
if (newChan) {
|
|
||||||
sendChannelMessage(ctx, chan, mkMessage('', chan.name, [DISCONNECT,0]), sendMsgs);
|
|
||||||
} else {
|
|
||||||
sendMsgs();
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (parsed.content[0] === PING) {
|
|
||||||
// 31:xwiki:XWiki.Admin-141475016907510:RWJ5xF2+SL17:[5,1414752676547]
|
|
||||||
// 1:y31:xwiki:XWiki.Admin-141475016907510:RWJ5xF2+SL17:[4,1414752676547]
|
|
||||||
sendMsg(mkMessage(parsed.user, parsed.channelId, [ PONG, parsed.content[1] ]), socket);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
var client = ctx.registeredClients[userPass];
|
|
||||||
if (typeof(client) === 'undefined') { throw new Error('unregistered'); }
|
|
||||||
|
|
||||||
var channel = ctx.channels[parsed.channelId];
|
|
||||||
if (typeof(channel) === 'undefined') { throw new Error('no such channel'); }
|
|
||||||
|
|
||||||
if (channel.indexOf(client) === -1) { throw new Error('client not in channel'); }
|
|
||||||
|
|
||||||
sendChannelMessage(ctx, channel, msg);
|
|
||||||
};
|
|
||||||
|
|
||||||
var create = module.exports.create = function (socketServer, store) {
|
|
||||||
var ctx = {
|
|
||||||
registeredClients: {},
|
|
||||||
channels: {},
|
|
||||||
store: store
|
|
||||||
};
|
|
||||||
|
|
||||||
socketServer.on('connection', function(socket) {
|
|
||||||
if(socket.upgradeReq.url !== '/cryptpad_websocket_old') { return; }
|
|
||||||
socket.on('message', function(message) {
|
|
||||||
try {
|
|
||||||
handleMessage(ctx, socket, message);
|
|
||||||
} catch (e) {
|
|
||||||
console.log(e.stack);
|
|
||||||
try { socket.close(); } catch (e) { }
|
|
||||||
}
|
|
||||||
});
|
|
||||||
socket.on('close', function (evt) {
|
|
||||||
for (var client in ctx.registeredClients) {
|
|
||||||
if (ctx.registeredClients[client].socket === socket) {
|
|
||||||
dropClient(ctx, client);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
});
|
|
||||||
};
|
|
||||||
Loading…
Reference in New Issue