forked from coolaj86/telebit.js
618 lines
19 KiB
JavaScript
Executable File
618 lines
19 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
(function () {
|
|
'use strict';
|
|
|
|
var pkg = require('../package.json');
|
|
|
|
var url = require('url');
|
|
var path = require('path');
|
|
var os = require('os');
|
|
var common = require('../lib/cli-common.js');
|
|
var http = require('http');
|
|
var YAML = require('js-yaml');
|
|
var recase = require('recase').create({});
|
|
var camelCopy = recase.camelCopy.bind(recase);
|
|
var snakeCopy = recase.snakeCopy.bind(recase);
|
|
var state = { homedir: os.homedir(), servernames: {}, ports: {} };
|
|
|
|
var argv = process.argv.slice(2);
|
|
|
|
var confIndex = argv.indexOf('--config');
|
|
var confpath;
|
|
var confargs;
|
|
if (-1 === confIndex) {
|
|
confIndex = argv.indexOf('-c');
|
|
}
|
|
if (-1 !== confIndex) {
|
|
confargs = argv.splice(confIndex, 2);
|
|
confpath = confargs[1];
|
|
}
|
|
|
|
function help() {
|
|
console.info('');
|
|
console.info('Telebit Daemon v' + pkg.version);
|
|
console.info('');
|
|
console.info('Usage:');
|
|
console.info('');
|
|
console.info('\ttelebitd --config <path>');
|
|
console.info('\tex: telebitd --config ~/.config/telebit/telebitd.yml');
|
|
console.info('');
|
|
console.info('');
|
|
console.info('Config:');
|
|
console.info('');
|
|
console.info('\tSee https://git.coolaj86.com/coolaj86/telebit.js');
|
|
console.info('');
|
|
console.info('');
|
|
}
|
|
|
|
var verstr = '' + pkg.name + ' v' + pkg.version;
|
|
if (-1 === confIndex) {
|
|
// We have two possible valid paths if no --config is given (i.e. run from an npm-only install)
|
|
// * {install}/etc/telebitd.yml
|
|
// * ~/.config/telebit/telebitd.yml
|
|
// We'll asume the later since the installers include --config in the system launcher script
|
|
confpath = path.join(state.homedir, '.config/telebit/telebitd.yml');
|
|
verstr += ' (--config "' + confpath + '")';
|
|
}
|
|
console.info(verstr + '\n');
|
|
|
|
if (-1 !== argv.indexOf('-h') || -1 !== argv.indexOf('--help')) {
|
|
help();
|
|
process.exit(0);
|
|
}
|
|
if (!confpath || /^--/.test(confpath)) {
|
|
help();
|
|
process.exit(1);
|
|
}
|
|
var tokenpath = path.join(path.dirname(confpath), 'access_token.txt');
|
|
var token;
|
|
try {
|
|
token = require('fs').readFileSync(tokenpath, 'ascii').trim();
|
|
} catch(e) {
|
|
// ignore
|
|
}
|
|
var controlServer;
|
|
|
|
var tun;
|
|
function serveControls() {
|
|
if (!state.config.disable) {
|
|
if (state.config.relay && (state.config.token || state.config.agreeTos)) {
|
|
tun = rawTunnel();
|
|
}
|
|
}
|
|
controlServer = http.createServer(function (req, res) {
|
|
var opts = url.parse(req.url, true);
|
|
if (opts.query._body) {
|
|
try {
|
|
opts.body = JSON.parse(opts.query._body, true);
|
|
} catch(e) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"?_body={{bad_format}}"}}');
|
|
return;
|
|
}
|
|
}
|
|
|
|
function listSuccess() {
|
|
var dumpy = {
|
|
servernames: state.servernames
|
|
, ports: state.ports
|
|
, ssh: state.config.sshAuto || 'disabled'
|
|
};
|
|
|
|
if (/\btelebit\.cloud\b/i.test(conf.relay) && state.config.email && !state.token) {
|
|
dumpy.code = "AWAIT_AUTH";
|
|
dumpy.message = "Check your email. You must verify your email address to activate this device.";
|
|
}
|
|
|
|
res.end(YAML.safeDump(dumpy));
|
|
}
|
|
|
|
function sshSuccess() {
|
|
fs.writeFile(confpath, YAML.safeDump(snakeCopy(state.config)), function (err) {
|
|
if (err) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"Could not save config file. Perhaps you\'re not running as root?"}}');
|
|
return;
|
|
}
|
|
res.end('{"success":true}');
|
|
});
|
|
}
|
|
|
|
if (/\b(init|config)\b/.test(opts.path)) {
|
|
var conf = {};
|
|
var fresh;
|
|
if (!opts.body) {
|
|
res.statusCode = 422;
|
|
res.end('{"error":{"message":"needs more arguments"}}');
|
|
return;
|
|
}
|
|
// relay, email, agree_tos, servernames, ports
|
|
//
|
|
opts.body.forEach(function (opt) {
|
|
var parts = opt.split(/,/);
|
|
conf[parts[0]] = parts[1];
|
|
});
|
|
if (!state.config.relay || !state.config.email || !state.config.agreeTos) {
|
|
fresh = true;
|
|
}
|
|
|
|
// TODO camelCase query
|
|
state.config.email = conf.email || state.config.email || '';
|
|
if ('undefined' !== typeof conf.agree_tos) {
|
|
state.config.agreeTos = conf.agree_tos;
|
|
}
|
|
state.config.relay = conf.relay || state.config.relay || '';
|
|
state.config.token = conf.token || state.config.token || null;
|
|
state.config.secret = conf.secret || state.config.secret || null;
|
|
if ('undefined' !== typeof conf.newsletter) {
|
|
state.config.newsletter = conf.newsletter;
|
|
}
|
|
if ('undefined' !== typeof conf.community_member) {
|
|
state.config.communityMember = conf.community_member;
|
|
}
|
|
if ('undefined' !== typeof conf.telemetry) {
|
|
state.config.telemetry = conf.telemetry;
|
|
}
|
|
conf.servernames = (conf.servernames||'').split(/,/g);
|
|
conf.ports = (conf.ports||'').split(/,/g);
|
|
conf.servernames.forEach(function (key) {
|
|
if (!state.config.servernames[key]) {
|
|
state.config.servernames[key] = {};
|
|
}
|
|
});
|
|
conf.ports.forEach(function (key) {
|
|
if (!state.config.ports[key]) {
|
|
state.config.ports[key] = {};
|
|
}
|
|
});
|
|
|
|
if (!state.config.relay || !state.config.email || !state.config.agreeTos) {
|
|
res.statusCode = 400;
|
|
res.end('{"error":{"code":"E_CONFIG","message":"Missing important config file params. Please run \'telebit init\'"}}');
|
|
return;
|
|
}
|
|
|
|
if (tun) {
|
|
tun.end(function () {
|
|
tun = rawTunnel();
|
|
});
|
|
tun = null;
|
|
setTimeout(function () {
|
|
if (!tun) { tun = rawTunnel(); }
|
|
}, 3000);
|
|
} else {
|
|
tun = rawTunnel();
|
|
}
|
|
|
|
fs.writeFile(confpath, YAML.safeDump(snakeCopy(state.config)), function (err) {
|
|
if (err) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"Could not save config file after init: ' + err.message.replace(/"/g, "'")
|
|
+ '.\nPerhaps check that the file exists and your user has permissions to write it?"}}');
|
|
return;
|
|
}
|
|
|
|
// TODO check for message from remote about email
|
|
if (/\btelebit\.cloud\b/i.test(conf.relay) && state.config.email && !state.token) {
|
|
res.statusCode = 200;
|
|
res.end('{"success":true,"code":"AWAIT_AUTH","message":"Check your email. You must verify your email address to activate this device."}');
|
|
} else {
|
|
res.statusCode = 200;
|
|
res.end('{"success":true}');
|
|
}
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
if (!state.config.relay || !state.config.email || !state.config.agreeTos) {
|
|
res.statusCode = 400;
|
|
res.end('{"error":{"code":"E_CONFIG","message":"Invalid config file. Please run \'telebit init\'"}}');
|
|
return;
|
|
}
|
|
|
|
//
|
|
// With proper config
|
|
//
|
|
if (/http/.test(opts.path)) {
|
|
if (!opts.body) {
|
|
res.statusCode = 422;
|
|
res.end('{"error":{"message":"needs more arguments"}}');
|
|
return;
|
|
}
|
|
if (opts.body[1]) {
|
|
if (!state.servernames[opts.body[1]]) {
|
|
res.statusCode = 400;
|
|
res.end('{"error":{"message":"bad servername \'' + opts.body[1] + '\'"');
|
|
return;
|
|
}
|
|
state.servernames[opts.body[1]].handler = opts.body[0];
|
|
} else {
|
|
Object.keys(state.servernames).forEach(function (key) {
|
|
state.servernames[key].handler = opts.body[0];
|
|
});
|
|
}
|
|
res.end('{"success":true}');
|
|
return;
|
|
}
|
|
|
|
if (/tcp/.test(opts.path)) {
|
|
if (!opts.body) {
|
|
res.statusCode = 422;
|
|
res.end('{"error":{"message":"needs more arguments"}}');
|
|
return;
|
|
}
|
|
|
|
// portnum
|
|
if (opts.body[1]) {
|
|
if (!state.ports[opts.body[1]]) {
|
|
res.statusCode = 400;
|
|
res.end('{"error":{"message":"bad port \'' + opts.body[1] + '\'"');
|
|
return;
|
|
}
|
|
// forward-to port-or-module
|
|
state.ports[opts.body[1]].handler = opts.body[0];
|
|
} else {
|
|
Object.keys(state.ports).forEach(function (key) {
|
|
state.ports[key].handler = opts.body[0];
|
|
});
|
|
}
|
|
res.end('{"success":true}');
|
|
return;
|
|
}
|
|
|
|
if (/save|commit/.test(opts.path)) {
|
|
state.config.servernames = state.servernames;
|
|
state.config.ports = state.ports;
|
|
fs.writeFile(confpath, YAML.safeDump(snakeCopy(state.config)), function (err) {
|
|
if (err) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"Could not save config file. Perhaps you\'re not running as root?"}}');
|
|
return;
|
|
}
|
|
listSuccess();
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (/ssh/.test(opts.path)) {
|
|
var sshAuto;
|
|
if (!opts.body) {
|
|
res.statusCode = 422;
|
|
res.end('{"error":{"message":"needs more arguments"}}');
|
|
return;
|
|
}
|
|
|
|
sshAuto = opts.body[0];
|
|
if (-1 !== [ 'false', 'none', 'off', 'disable' ].indexOf(sshAuto)) {
|
|
state.config.sshAuto = false;
|
|
sshSuccess();
|
|
return;
|
|
}
|
|
if (-1 !== [ 'true', 'auto', 'on', 'enable' ].indexOf(sshAuto)) {
|
|
state.config.sshAuto = 22;
|
|
sshSuccess();
|
|
return;
|
|
}
|
|
sshAuto = parseInt(sshAuto, 10);
|
|
if (!sshAuto || sshAuto <= 0 || sshAuto > 65535) {
|
|
res.statusCode = 400;
|
|
res.end('{"error":{"message":"bad ssh_auto option \'' + opts.body[0] + '\'"');
|
|
return;
|
|
}
|
|
state.config.sshAuto = sshAuto;
|
|
sshSuccess();
|
|
return;
|
|
}
|
|
|
|
if (/enable/.test(opts.path)) {
|
|
delete state.config.disable;// = undefined;
|
|
if (!tun) { tun = rawTunnel(); }
|
|
fs.writeFile(confpath, YAML.safeDump(snakeCopy(state.config)), function (err) {
|
|
if (err) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"Could not save config file. Perhaps you\'re not running as root?"}}');
|
|
return;
|
|
}
|
|
listSuccess();
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (/disable/.test(opts.path)) {
|
|
state.config.disable = true;
|
|
if (tun) { tun.end(); tun = null; }
|
|
fs.writeFile(confpath, YAML.safeDump(snakeCopy(state.config)), function (err) {
|
|
if (err) {
|
|
res.statusCode = 500;
|
|
res.end('{"error":{"message":"Could not save config file. Perhaps you\'re not running as root?"}}');
|
|
return;
|
|
}
|
|
res.end('{"success":true}');
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (/status/.test(opts.path)) {
|
|
res.end(
|
|
'{"status":' + (state.config.disable ? 'disabled' : 'enabled')
|
|
+ ',"ready":' + ((state.config.relay && (state.config.token || state.config.agreeTos)) ? 'true' : 'false')
|
|
+ '}');
|
|
return;
|
|
}
|
|
|
|
if (/restart/.test(opts.path)) {
|
|
tun.end();
|
|
res.end('{"success":true}');
|
|
controlServer.close(function () {
|
|
// TODO closeAll other things
|
|
process.nextTick(function () {
|
|
// system daemon will restart the process
|
|
process.exit(22); // use non-success exit code
|
|
});
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (/list/.test(opts.path)) {
|
|
listSuccess();
|
|
return;
|
|
}
|
|
|
|
res.end('{"error":{"message":"unrecognized rpc"}}');
|
|
});
|
|
var pipename = common.pipename(state.config);
|
|
var fs = require('fs');
|
|
if (fs.existsSync(pipename)) {
|
|
fs.unlinkSync(pipename);
|
|
}
|
|
// mask is so that processes owned by other users
|
|
// can speak to this process, which is probably root-owned
|
|
var oldUmask = process.umask(0x0000);
|
|
controlServer.listen({
|
|
path: pipename
|
|
, writableAll: true
|
|
, readableAll: true
|
|
, exclusive: false
|
|
}, function () {
|
|
process.umask(oldUmask);
|
|
});
|
|
}
|
|
|
|
function parseConfig(err, text) {
|
|
var config;
|
|
|
|
function run() {
|
|
state._confpath = confpath;
|
|
if (!state.config.servernames) {
|
|
state.config.servernames = {};
|
|
}
|
|
if (!state.config.ports) {
|
|
state.config.ports = {};
|
|
}
|
|
state.servernames = JSON.parse(JSON.stringify(state.config.servernames));
|
|
state.ports = JSON.parse(JSON.stringify(state.config.ports));
|
|
|
|
serveControls();
|
|
}
|
|
|
|
if (err) {
|
|
console.warn("\nCouldn't load config:\n\n\t" + err.message + "\n");
|
|
state.config = {};
|
|
run();
|
|
return;
|
|
}
|
|
|
|
try {
|
|
config = JSON.parse(text);
|
|
} catch(e1) {
|
|
try {
|
|
config = YAML.safeLoad(text);
|
|
} catch(e2) {
|
|
console.error(e1.message);
|
|
console.error(e2.message);
|
|
process.exit(1);
|
|
return;
|
|
}
|
|
}
|
|
|
|
state.config = camelCopy(config);
|
|
if (state.config.token && token) {
|
|
console.warn();
|
|
console.warn("Found two tokens:");
|
|
console.warn();
|
|
console.warn("\t1. " + tokenpath);
|
|
console.warn("\n2. " + confpath);
|
|
console.warn();
|
|
console.warn("Choosing the first.");
|
|
console.warn();
|
|
}
|
|
state.token = token;
|
|
|
|
run();
|
|
}
|
|
|
|
function connectTunnel() {
|
|
function sigHandler() {
|
|
console.info('Received kill signal. Attempting to exit cleanly...');
|
|
|
|
// We want to handle cleanup properly unless something is broken in our cleanup process
|
|
// that prevents us from exitting, in which case we want the user to be able to send
|
|
// the signal again and exit the way it normally would.
|
|
process.removeListener('SIGINT', sigHandler);
|
|
tun.end();
|
|
controlServer.close();
|
|
}
|
|
process.on('SIGINT', sigHandler);
|
|
state.net = state.net || {
|
|
createConnection: function (info, cb) {
|
|
// data is the hello packet / first chunk
|
|
// info = { data, servername, port, host, remoteFamily, remoteAddress, remotePort }
|
|
var net = require('net');
|
|
// socket = { write, push, end, events: [ 'readable', 'data', 'error', 'end' ] };
|
|
var socket = net.createConnection({ port: info.port, host: info.host }, cb);
|
|
return socket;
|
|
}
|
|
};
|
|
|
|
state.greenlock = state.config.greenlock || {};
|
|
state.sortingHat = state.config.sortingHat || path.resolve(__dirname, '..', 'lib/sorting-hat.js');
|
|
|
|
// TODO sortingHat.print(); ?
|
|
|
|
if (state.config.email && !state.token) {
|
|
console.info();
|
|
console.info('==================================');
|
|
console.info('= HEY! LISTEN! =');
|
|
console.info('==================================');
|
|
console.info('= =');
|
|
console.info('= 1. Open your email =');
|
|
console.info('= 2. Click the magic login link =');
|
|
console.info('= 3. Check back here for deets =');
|
|
console.info('= =');
|
|
console.info('==================================');
|
|
console.info();
|
|
}
|
|
// TODO Check undefined vs false for greenlock config
|
|
var remote = require('../');
|
|
state.handlers = {
|
|
grant: function (grants) {
|
|
console.info("");
|
|
console.info("Connect to your device by any of the following means:");
|
|
console.info("");
|
|
grants.forEach(function (arr) {
|
|
if ('https' === arr[0]) {
|
|
if (!state.servernames[arr[1]]) {
|
|
state.servernames[arr[1]] = {};
|
|
}
|
|
} else if ('tcp' === arr[0]) {
|
|
if (!state.ports[arr[2]]) {
|
|
state.ports[arr[2]] = {};
|
|
}
|
|
}
|
|
|
|
if ('ssh+https' === arr[0]) {
|
|
console.info("SSH+HTTPS");
|
|
} else if ('ssh' === arr[0]) {
|
|
console.info("SSH");
|
|
} else if ('tcp' === arr[0]) {
|
|
console.info("TCP");
|
|
} else if ('https' === arr[0]) {
|
|
console.info("HTTPS");
|
|
}
|
|
console.info('\t' + arr[0] + '://' + arr[1] + (arr[2] ? (':' + arr[2]) : ''));
|
|
if ('ssh+https' === arr[0]) {
|
|
console.info("\tex: ssh -o ProxyCommand='openssl s_client -connect %h:%p -quiet' " + arr[1] + " -p 443\n");
|
|
} else if ('ssh' === arr[0]) {
|
|
console.info("\tex: ssh " + arr[1] + " -p " + arr[2] + "\n");
|
|
} else if ('tcp' === arr[0]) {
|
|
console.info("\tex: netcat " + arr[1] + " " + arr[2] + "\n");
|
|
} else if ('https' === arr[0]) {
|
|
console.info("\tex: curl https://" + arr[1] + "\n");
|
|
}
|
|
});
|
|
}
|
|
, access_token: function (opts) {
|
|
console.info("Updating '" + tokenpath + "' with new token:");
|
|
try {
|
|
require('fs').writeFileSync(tokenpath, opts.jwt);
|
|
} catch (e) {
|
|
console.error("Token not saved:");
|
|
console.error(e);
|
|
}
|
|
}
|
|
};
|
|
state.greenlockConfig = {
|
|
version: state.greenlock.version || 'draft-11'
|
|
, server: state.greenlock.server || 'https://acme-v02.api.letsencrypt.org/directory'
|
|
, communityMember: state.greenlock.communityMember || state.config.communityMember
|
|
, telemetry: state.greenlock.telemetry || state.config.telemetry
|
|
, configDir: state.greenlock.configDir || path.resolve(__dirname, '..', 'etc/acme/')
|
|
// TODO, store: require(state.greenlock.store.name || 'le-store-certbot').create(state.greenlock.store.options || {})
|
|
, approveDomains: function (opts, certs, cb) {
|
|
// Certs being renewed are listed in certs.altnames
|
|
if (certs) {
|
|
opts.domains = certs.altnames;
|
|
cb(null, { options: opts, certs: certs });
|
|
return;
|
|
}
|
|
|
|
// by virtue of the fact that it's being tunneled through a
|
|
// trusted source that is already checking, we're good
|
|
//if (-1 !== state.config.servernames.indexOf(opts.domains[0])) {
|
|
opts.email = state.greenlock.email || state.config.email;
|
|
opts.agreeTos = state.greenlock.agree || state.config.agreeTos;
|
|
cb(null, { options: opts, certs: certs });
|
|
return;
|
|
//}
|
|
|
|
//cb(new Error("servername not found in allowed list"));
|
|
}
|
|
};
|
|
state.insecure = state.config.relay_ignore_invalid_certificates;
|
|
// { relay, config, servernames, ports, sortingHat, net, insecure, token, handlers, greenlockConfig }
|
|
|
|
var tun = remote.connect({
|
|
relay: state.relay
|
|
, config: state.config
|
|
, sortingHat: state.sortingHat
|
|
, net: state.net
|
|
, insecure: state.insecure
|
|
, token: state.token
|
|
, servernames: state.servernames
|
|
, ports: state.ports
|
|
, handlers: state.handlers
|
|
, greenlockConfig: state.greenlockConfig
|
|
});
|
|
|
|
return tun;
|
|
}
|
|
|
|
function rawTunnel() {
|
|
if (state.config.disable || !state.config.relay || !(state.config.token || state.config.agreeTos)) {
|
|
return;
|
|
}
|
|
|
|
state.relay = state.config.relay;
|
|
if (!state.relay) {
|
|
throw new Error("'" + state._confpath + "' is missing 'relay'");
|
|
}
|
|
|
|
/*
|
|
if (!(state.config.secret || state.config.token)) {
|
|
console.error("You must use --secret or --token with --relay");
|
|
process.exit(1);
|
|
return;
|
|
}
|
|
*/
|
|
|
|
var location = url.parse(state.relay);
|
|
if (!location.protocol || /\./.test(location.protocol)) {
|
|
state.relay = 'wss://' + state.relay;
|
|
location = url.parse(state.relay);
|
|
}
|
|
var aud = location.hostname + (location.port ? ':' + location.port : '');
|
|
state.relay = location.protocol + '//' + aud;
|
|
|
|
if (!state.config.token && state.config.secret) {
|
|
var jwt = require('jsonwebtoken');
|
|
var tokenData = {
|
|
domains: Object.keys(state.config.servernames || {}).filter(function (name) { return /\./.test(name); })
|
|
, aud: aud
|
|
, iss: Math.round(Date.now() / 1000)
|
|
};
|
|
|
|
state.token = jwt.sign(tokenData, state.config.secret);
|
|
}
|
|
state.token = state.token || state.config.token;
|
|
|
|
// TODO sign token with own private key, including public key and thumbprint
|
|
// (much like ACME JOSE account)
|
|
|
|
return connectTunnel();
|
|
}
|
|
|
|
require('fs').readFile(confpath, 'utf8', parseConfig);
|
|
|
|
}());
|