diff --git a/src/channels/channelBase.ts b/src/channels/channelBase.ts index b6357ba..e94ae28 100644 --- a/src/channels/channelBase.ts +++ b/src/channels/channelBase.ts @@ -48,7 +48,7 @@ class ChannelBase { broadcastMessage(from: ClientBase|null, message: string) { for (let client of this.clients) { if (client != from) { - client.ws.send(message); + client.ws.send(JSON.stringify(message)); logger.accessLog.info(`sent to ${client.id}`, { data: { message: message }}); } else { logger.accessLog.info(`client is same as sender: ${client.id}`, { data: { message: message }}); diff --git a/src/messageManager.ts b/src/messageManager.ts index b04abcb..f70c0b5 100644 --- a/src/messageManager.ts +++ b/src/messageManager.ts @@ -17,7 +17,8 @@ let schema = { user_id: Joi.number().integer(), user_exam_id: Joi.number().integer(), user_name: Joi.string(), - exam_title: Joi.string() + exam_title: Joi.string(), + message: Joi.string(), }; module.exports = { diff --git a/src/server.ts b/src/server.ts index 2016ad4..52c2616 100755 --- a/src/server.ts +++ b/src/server.ts @@ -15,99 +15,81 @@ import ChannelManager from './channelManager'; import ClientBase from './clients/clientBase'; import MHSClient from './clients/sites/mhsClient'; -var privateKey = fs.readFileSync(app.privateKey, 'utf8'); -var certificate = fs.readFileSync(app.certificate, 'utf8'); -var options = {key: privateKey, cert: certificate, hostname: app.hostname}; -const application = express(); -const server = https.createServer(options, application); -const wss = new WebSocket.Server({ noServer: true, maxPayload:250000, host: app.hostname }); +if (app.environment == 'development') { + var privateKey = fs.readFileSync(app.privateKey, 'utf8'); + var certificate = fs.readFileSync(app.certificate, 'utf8'); + var options = {key: privateKey, cert: certificate, hostname: app.hostname}; -application.use(express.json()); -application.use('', routes); + const application = express(); + const server = https.createServer(options, application); + var wss = new WebSocket.Server({ server: server, maxPayload:250000, host: app.hostname }); + + application.use(express.json()); + application.use('', routes); + + server.listen(app.port, () => { + console.log(`Braid v${app.version} is running!\n`); + logger.accessLog.info(`Braid v${app.version} is running!\n`); + }); +} else { + var wss = new WebSocket.Server({ noServer: true, maxPayload:250000, host: app.hostname }); +} let clientManager = new ClientManager(); let channelManager = new ChannelManager(); function connectionManager() { - wss.on('connection', (ws: WebSocket, request: object, args: string) => { - var data = JSON.parse(args).data - logger.accessLog.info(`Client Connected: ${data.user_id}`); + wss.on('connection', (ws: WebSocket, request: any, args: string) => { + var result = JSON.parse(validateJWT(request)); - if (!channelManager.channelExists(data.channel)) { - channelManager.createChannel(data); - } + if (result.error) { + ws.send(`Unable to validate JWT, please try again or contact support...`); + ws.close(); + } else { + var data = result.data; + logger.accessLog.info(`Client Connected: ${data.user_id}`); - if (clientManager.clientExists(data.user_id)) { - var client: ClientBase|MHSClient|null = clientManager.getClient(data.user_id); + if (!channelManager.channelExists(data.channel)) { + channelManager.createChannel(data); + } + + if (clientManager.clientExists(data.user_id)) { + var client: ClientBase|MHSClient|null = clientManager.getClient(data.user_id); + + if (client != null) { + client.replaceWebSocket(ws); + } + } else { + var client: ClientBase|MHSClient|null = clientManager.addClient(data, channelManager, ws); + } if (client != null) { - client.replaceWebSocket(ws); - } - } else { - var client: ClientBase|MHSClient|null = clientManager.addClient(data, channelManager, ws); - } - - if (client != null) { - channelManager.addClientToChannel(client, data.channel); - } - - //remove any channels that have no users in them - logger.accessLog.info(`Purging empty channels...`); - channelManager.purgeEmptyChannels(); - - ws.send(`Hi there, welcome to braid, Measures Web Socket server. Connecting all our services!\nYou are currently in channel: ${data.channel}`); - }); -} - -function verifyConnection(request: any) { - return new Promise((resolve, reject) => { - var data = url.parse(request.url, true).query; - var token = data.token || (app.environment == 'development' ? app.devToken : ''); - var accepted = true; - var result: string; - - jwt.verify(token, app.secret, app.signOptions, function(err, decoded) { - if (err) { - console.log(err); - accepted = false; + channelManager.addClientToChannel(client, data.channel); } - result = JSON.stringify(decoded) - accepted ? resolve(result) : reject('rejected'); - }); - }); -} + //remove any channels that have no users in them + logger.accessLog.info(`Purging empty channels...`); + channelManager.purgeEmptyChannels(); -function serverUpgrade() { - server.on('upgrade', async function upgrade(request, socket, head) { - let args: {}; - - try { - args = await verifyConnection(request); - } catch (e) { - socket.destroy(); - logger.accessLog.info('Connection Terminated'); - return; + ws.send(`Hi there, welcome to braid, Measures Web Socket server. Connecting all our services!\nYou are currently in channel: ${data.channel}`); } - - wss.handleUpgrade(request, socket, head, function done(ws) { - wss.emit('connection', ws, request, args); - }); }); } -function listenForConnection() { - server.listen(app.port, () => { - console.log(`Braid v${app.version} is running!\n`); - logger.accessLog.info(`Braid v${app.version} is running!\n`); - }); +function validateJWT(request: any) { + try { + var query = url.parse(request.url, true).query + var token = query.token || (app.environment == 'development' ? app.devToken : ''); + return JSON.stringify(jwt.verify(token, app.secret, app.signOptions)); + } catch (e) { + console.log(e); + return JSON.stringify({error: e}); + } } function startServer() { connectionManager(); - serverUpgrade(); - listenForConnection(); } startServer();