crabapples
/
websockify
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add option of providing authorization plugin

This commit is contained in:
Sam Frances 2017-05-05 12:46:38 +01:00
parent f3a0ce06a9
commit 845f9f2f9f
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
other/js/websockify.js
View File

@ -22,6 +22,7 @@ var argv = require('optimist').argv,
webServer, wsServer,
source_host, source_port, target_host, target_port,
auth_plugin, websocket_server_opts,
web_path = null;
@ -140,7 +141,7 @@ try {
throw("illegal port");
}
} catch(e) {
console.error("websockify.js [--web web_dir] [--cert cert.pem [--key key.pem]] [source_addr:]source_port target_addr:target_port");
console.error("websockify.js [--web web_dir] [--cert cert.pem [--key key.pem]] [--auth-plugin module.plugin [--auth-source auth_source]] [source_addr:]source_port target_addr:target_port");
process.exit(2);
}
@ -161,7 +162,23 @@ if (argv.cert) {
console.log(" - Running in unencrypted HTTP (ws://) mode");
webServer = http.createServer(http_request);
}
if (argv["auth-plugin"]) {
let auth_plugin_arg = argv["auth-plugin"].split(".")
let plugin_name = auth_plugin_arg.pop();
let module_path = auth_plugin_arg.join(".");
let auth_plugin = require(module_path)[plugin_name];
let auth_source = argv["auth-source"] || undefined;
websocket_server_opts = {
server: webServer,
verifyClient: auth_plugin(auth_source)
};
} else {
websocket_server_opts = {server: webServer};
}
webServer.listen(source_port, function() {
wsServer = new WebSocketServer({server: webServer});
wsServer = new WebSocketServer(websocket_server_opts);
wsServer.on('connection', new_client);
});