344 lines
10 KiB
JavaScript
Executable File
344 lines
10 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
(function () {
|
|
'use strict';
|
|
|
|
var pkg = require('../package.json');
|
|
|
|
var argv = process.argv.slice(2);
|
|
var relay = require('../');
|
|
var Greenlock = require('greenlock');
|
|
|
|
var confIndex = argv.indexOf('--config');
|
|
var confpath;
|
|
if (-1 === confIndex) {
|
|
confIndex = argv.indexOf('-c');
|
|
}
|
|
confpath = argv[confIndex + 1];
|
|
|
|
function help() {
|
|
console.info('');
|
|
console.info('Usage:');
|
|
console.info('');
|
|
console.info('\ttelebit-relay --config <path>');
|
|
console.info('');
|
|
console.info('Example:');
|
|
console.info('');
|
|
console.info('\ttelebit-relay --config /opt/telebit-relay/etc/telebit-relay.yml');
|
|
console.info('');
|
|
console.info('Config:');
|
|
console.info('');
|
|
console.info('\tSee https://git.coolaj86.com/coolaj86/telebit-relay.js');
|
|
console.info('');
|
|
console.info('');
|
|
process.exit(0);
|
|
}
|
|
|
|
if (-1 === confIndex || -1 !== argv.indexOf('-h') || -1 !== argv.indexOf('--help')) {
|
|
help();
|
|
}
|
|
if (!confpath || /^--/.test(confpath)) {
|
|
help();
|
|
}
|
|
|
|
function applyConfig(config) {
|
|
var state = { defaults: {}, ports: [ 80, 443 ], tcp: {} };
|
|
if ('undefined' !== typeof Promise) {
|
|
state.Promise = Promise;
|
|
} else {
|
|
state.Promise = require('bluebird');
|
|
}
|
|
state.tlsOptions = {}; // TODO just close the sockets that would use this early? or use the admin servername
|
|
state.config = config;
|
|
state.servernames = config.servernames || [];
|
|
state.secret = state.config.secret;
|
|
if (!state.secret) {
|
|
state.secret = require('crypto').randomBytes(16).toString('hex');
|
|
console.info("");
|
|
console.info("Secret for this session:");
|
|
console.info("");
|
|
console.info("\t" + state.secret);
|
|
console.info("");
|
|
console.info("");
|
|
}
|
|
if (!state.config.greenlock) {
|
|
state.config.greenlock = {};
|
|
}
|
|
if (!state.config.greenlock.configDir) {
|
|
state.config.greenlock.configDir = require('os').homedir() + require('path').sep + 'acme';
|
|
}
|
|
|
|
function approveDomains(opts, certs, cb) {
|
|
if (state.debug) { console.log('[debug] approveDomains', opts.domains); }
|
|
// This is where you check your database and associated
|
|
// email addresses with domains and agreements and such
|
|
|
|
// The domains being approved for the first time are listed in opts.domains
|
|
// Certs being renewed are listed in certs.altnames
|
|
if (certs) {
|
|
opts.domains = certs.altnames;
|
|
cb(null, { options: opts, certs: certs });
|
|
return;
|
|
}
|
|
|
|
if (!state.validHosts) { state.validHosts = {}; }
|
|
if (!state.validHosts[opts.domains[0]] && state.config.vhost) {
|
|
if (state.debug) { console.log('[sni] vhost checking is turned on'); }
|
|
var vhost = state.config.vhost.replace(/:hostname/, opts.domains[0]);
|
|
require('fs').readdir(vhost, function (err, nodes) {
|
|
if (state.debug) { console.log('[sni] checking fs vhost', opts.domains[0], !err); }
|
|
if (err) { check(); return; }
|
|
if (nodes) { approve(); }
|
|
});
|
|
return;
|
|
}
|
|
|
|
function approve() {
|
|
state.validHosts[opts.domains[0]] = true;
|
|
opts.email = state.config.email;
|
|
opts.agreeTos = state.config.agreeTos;
|
|
opts.communityMember = state.config.communityMember || state.config.greenlock.communityMember;
|
|
opts.challenges = {
|
|
// TODO dns-01
|
|
'http-01': require('le-challenge-fs').create({ webrootPath: '/tmp/acme-challenges' })
|
|
};
|
|
opts.communityMember = state.config.communityMember;
|
|
cb(null, { options: opts, certs: certs });
|
|
}
|
|
|
|
function check() {
|
|
if (state.debug) { console.log('[sni] checking servername'); }
|
|
if (-1 !== state.servernames.indexOf(opts.domain) || -1 !== (state._servernames||[]).indexOf(opts.domain)) {
|
|
approve();
|
|
} else {
|
|
cb(new Error("failed the approval chain '" + opts.domains[0] + "'"));
|
|
}
|
|
}
|
|
|
|
check();
|
|
}
|
|
|
|
state.greenlock = Greenlock.create({
|
|
|
|
version: state.config.greenlock.version || 'draft-11'
|
|
, server: state.config.greenlock.server || 'https://acme-v02.api.letsencrypt.org/directory'
|
|
|
|
, store: require('le-store-certbot').create({ debug: state.config.debug || state.config.greenlock.debug, webrootPath: '/tmp/acme-challenges' })
|
|
|
|
, approveDomains: approveDomains
|
|
, telemetry: state.config.telemetry || state.config.greenlock.telemetry
|
|
, configDir: state.config.greenlock.configDir
|
|
, debug: state.config.debug || state.config.greenlock.debug
|
|
});
|
|
|
|
try {
|
|
// TODO specify extensions in config file
|
|
state.extensions = require('../lib/extensions');
|
|
} catch(e) {
|
|
if ('ENOENT' !== e.code || state.debug) { console.log('[DEBUG] no extensions loaded', e); }
|
|
state.extensions = {};
|
|
}
|
|
require('../lib/handlers').create(state); // adds directly to config for now...
|
|
|
|
//require('cluster-store').create().then(function (store) {
|
|
//program.store = store;
|
|
|
|
|
|
state.authenticate = function (opts) {
|
|
if (state.extensions.authenticate) {
|
|
try {
|
|
return state.extensions.authenticate({
|
|
state: state
|
|
, auth: opts.auth
|
|
});
|
|
} catch(e) {
|
|
console.error('Extension Error:');
|
|
console.error(e);
|
|
}
|
|
}
|
|
return state.defaults.authenticate(opts.auth);
|
|
};
|
|
|
|
// default authenticator for single-user setup
|
|
// (i.e. personal use on DO, Vultr, or RPi)
|
|
state.defaults.authenticate = function onAuthenticate(jwtoken) {
|
|
return state.Promise.resolve().then(function () {
|
|
var jwt = require('jsonwebtoken');
|
|
var auth;
|
|
var token;
|
|
var decoded;
|
|
|
|
try {
|
|
token = jwt.verify(jwtoken, state.secret);
|
|
} catch (e) {
|
|
token = null;
|
|
}
|
|
|
|
return token;
|
|
});
|
|
};
|
|
|
|
var net = require('net');
|
|
var netConnHandlers = relay.create(state); // { tcp, ws }
|
|
var WebSocketServer = require('ws').Server;
|
|
var wss = new WebSocketServer({ server: (state.httpTunnelServer || state.httpServer) });
|
|
wss.on('connection', netConnHandlers.ws);
|
|
state.ports.forEach(function (port) {
|
|
if (state.tcp[port]) {
|
|
console.warn("[cli] skipping previously added port " + port);
|
|
return;
|
|
}
|
|
state.tcp[port] = net.createServer();
|
|
state.tcp[port].listen(port, function () {
|
|
console.info('[cli] Listening for TCP connections on', port);
|
|
});
|
|
state.tcp[port].on('connection', netConnHandlers.tcp);
|
|
});
|
|
//});
|
|
}
|
|
|
|
require('fs').readFile(confpath, 'utf8', function (err, text) {
|
|
var config;
|
|
|
|
var recase = require('recase').create({});
|
|
var camelCopy = recase.camelCopy.bind(recase);
|
|
|
|
if (err) {
|
|
console.error("\nCouldn't load config:\n\n\t" + err.message + "\n");
|
|
process.exit(1);
|
|
return;
|
|
}
|
|
|
|
try {
|
|
config = JSON.parse(text);
|
|
} catch(e1) {
|
|
try {
|
|
config = require('js-yaml').safeLoad(text);
|
|
} catch(e2) {
|
|
console.error(e1.message);
|
|
console.error(e2.message);
|
|
process.exit(1);
|
|
return;
|
|
}
|
|
}
|
|
|
|
applyConfig(camelCopy(config));
|
|
});
|
|
|
|
|
|
|
|
function adjustArgs() {
|
|
function collectServernames(val, memo) {
|
|
var lowerCase = val.split(/,/).map(function (servername) {
|
|
return servername.toLowerCase();
|
|
});
|
|
|
|
return memo.concat(lowerCase);
|
|
}
|
|
|
|
function collectProxies(val, memo) {
|
|
var vals = val.split(/,/g);
|
|
vals.map(function (location) {
|
|
// http:john.example.com:3000
|
|
// http://john.example.com:3000
|
|
var parts = location.split(':');
|
|
if (1 === parts.length) {
|
|
parts[1] = parts[0];
|
|
parts[0] = 'wss';
|
|
}
|
|
if (2 === parts.length) {
|
|
if (/\./.test(parts[0])) {
|
|
parts[2] = parts[1];
|
|
parts[1] = parts[0];
|
|
parts[0] = 'wss';
|
|
}
|
|
if (!/\./.test(parts[1])) {
|
|
throw new Error("bad --serve option Example: wss://tunnel.example.com:1337");
|
|
}
|
|
}
|
|
parts[0] = parts[0].toLowerCase();
|
|
parts[1] = parts[1].toLowerCase().replace(/(\/\/)?/, '') || '*';
|
|
parts[2] = parseInt(parts[2], 10) || 0;
|
|
if (!parts[2]) {
|
|
// TODO grab OS list of standard ports?
|
|
if (-1 !== [ 'ws', 'http' ].indexOf(parts[0])) {
|
|
//parts[2] = 80;
|
|
}
|
|
else if (-1 !== [ 'wss', 'https' ].indexOf(parts[0])) {
|
|
//parts[2] = 443;
|
|
}
|
|
else {
|
|
throw new Error("port must be specified - ex: tls:*:1337");
|
|
}
|
|
}
|
|
|
|
return {
|
|
protocol: parts[0]
|
|
, hostname: parts[1]
|
|
, port: parts[2]
|
|
};
|
|
}).forEach(function (val) {
|
|
memo.push(val);
|
|
});
|
|
|
|
return memo;
|
|
}
|
|
|
|
function collectPorts(val, memo) {
|
|
return memo.concat(val.split(/,/g).map(Number).filter(Boolean));
|
|
}
|
|
|
|
program
|
|
.version(pkg.version)
|
|
.option('--agree-tos', "Accept the Daplie and Let's Encrypt Terms of Service")
|
|
.option('--email <EMAIL>', "Email to use for Daplie and Let's Encrypt accounts")
|
|
.option('--serve <URL>', 'comma separated list of <proto>:<//><servername>:<port> to which matching incoming http and https should forward (reverse proxy). Ex: https://john.example.com,tls:*:1337', collectProxies, [ ])
|
|
.option('--ports <PORT>', 'comma separated list of ports on which to listen. Ex: 80,443,1337', collectPorts, [ ])
|
|
.option('--servernames <STRING>', 'comma separated list of servernames to use for the admin interface. Ex: tunnel.example.com,tunnel.example.net', collectServernames, [ ])
|
|
.option('--secret <STRING>', 'the same secret used by telebit-relay (used for JWT authentication)')
|
|
.parse(process.argv)
|
|
;
|
|
|
|
var portsMap = {};
|
|
var servernamesMap = {};
|
|
program.serve.forEach(function (proxy) {
|
|
servernamesMap[proxy.hostname] = true;
|
|
if (proxy.port) {
|
|
portsMap[proxy.port] = true;
|
|
}
|
|
});
|
|
program.servernames.forEach(function (name) {
|
|
servernamesMap[name] = true;
|
|
});
|
|
program.ports.forEach(function (port) {
|
|
portsMap[port] = true;
|
|
});
|
|
|
|
program.servernames = Object.keys(servernamesMap);
|
|
if (!program.servernames.length) {
|
|
throw new Error('You must give this server at least one servername for its admin interface. Example:\n\n\t--servernames tunnel.example.com,tunnel.example.net');
|
|
}
|
|
|
|
program.ports = Object.keys(portsMap);
|
|
if (!program.ports.length) {
|
|
program.ports = [ 80, 443 ];
|
|
}
|
|
|
|
if (!program.secret) {
|
|
// TODO randomly generate and store in file?
|
|
console.warn("[SECURITY] you must provide --secret '" + require('crypto').randomBytes(16).toString('hex') + "'");
|
|
process.exit(1);
|
|
return;
|
|
}
|
|
|
|
//program.tlsOptions.SNICallback = program.greenlock.httpsOptions.SNICallback;
|
|
/*
|
|
program.middleware = program.greenlock.middleware(function (req, res) {
|
|
res.end('Hello, World!');
|
|
});
|
|
*/
|
|
}
|
|
|
|
//adjustArgs();
|
|
|
|
}());
|