Leahnaya/pokemon-showdown
1
0
Fork 0
mirror of https://github.com/smogon/pokemon-showdown.git synced 2026-09-30 21:29:19 -05:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Refactor Verifier to use the PM API

Browse Source
This commit is contained in:
Ivo Julca
2016-03-23 02:22:23 -05:00
parent ce1fd87efd
commit 7ac62d3cdb
2 changed files with 45 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
app.js
View File

@@ -115,6 +115,7 @@ global.Rooms = require('./rooms.js');
delete process.send; // in case we're a child process
global.Verifier = require('./verifier.js');
Verifier.PM.spawn();
global.CommandParser = require('./command-parser.js');

85
verifier.js
View File

@@ -14,55 +14,58 @@
'use strict';
// Because I don't want two files, we're going to fork ourselves.
const crypto = require('crypto');
const ProcessManager = require('./process-manager');
if (!process.send) {
// This is the parent
const PM = exports.PM = new ProcessManager({
maxProcesses: 1,
execFile: 'verifier.js',
onMessageUpstream: function (message) {
// Protocol:
// success: "[id]|1"
// failure: "[id]|0"
let pipeIndex = message.indexOf('|');
let id = +message.substr(0, pipeIndex);
let result = Boolean(~~message.slice(pipeIndex + 1));
let guid = 1;
let resolvers = Object.create(null);
let child = exports.child = require('child_process').fork('verifier.js', {cwd: __dirname});
exports.verify = function (data, signature) {
return new Promise((resolve, reject) => {
let localGuid = guid++;
resolvers[localGuid] = resolve;
child.send({data: data, sig: signature, guid: localGuid});
});
};
child.on('message', response => {
if (response.guid in resolvers) {
resolvers[response.guid](response.success);
delete resolvers[response.guid];
if (this.pendingTasks.has(id)) {
this.pendingTasks.get(id)(result);
this.pendingTasks.delete(id);
this.release();
}
});
} else {
// This is the child
},
onMessageDownstream: function (message) {
// protocol:
// "[id]|{data, sig}"
let pipeIndex = message.indexOf('|');
let id = message.substr(0, pipeIndex);
global.Config = require('./config/config.js');
let crypto = require('crypto');
let keyalgo = Config.loginserverkeyalgo;
let pkey = Config.loginserverpublickey;
process.on('message', message => {
let verifier = crypto.createVerify(keyalgo);
verifier.update(message.data);
let data = JSON.parse(message.slice(pipeIndex + 1));
process.send(id + '|' + this.receive(data));
},
receive: function (data) {
let verifier = crypto.createVerify(Config.loginserverkeyalgo);
verifier.update(data.data);
let success = false;
try {
success = verifier.verify(pkey, message.sig, 'hex');
success = verifier.verify(Config.loginserverpublickey, data.sig, 'hex');
} catch (e) {}
process.send({
success: success,
guid: message.guid,
});
});
process.on('disconnect', () => {
process.exit();
});
return success ? 1 : 0;
},
});
if (process.send && module === process.mainModule) {
// This is a child process!
global.Config = require('./config/config.js');
require('./repl.js').start('verifier', cmd => eval(cmd));
process.on('message', message => PM.onMessageDownstream(message));
process.on('disconnect', () => process.exit());
}
exports.verify = function (data, signature) {
return PM.send({data: data, sig: signature});
};