handle sigint gracefully

Author: BoLaMN

dist/dhcp/index.js

@@ -18,7 +18,7 @@ module.exports = function(ip, acsurl, acspass) {
   acsurl = toHexArray(acsurl);
   acspass = acspass ? toHexArray(acspass) : [84, 101, 108, 115, 116, 114, 97];
   vendor = [1, acsurl.length].concat(acsurl.concat([2, acspass.length].concat(acspass)));
-  return server.createServer({
+  server.createServer({
     range: [ip + '.10', ip + '.15'],
     forceOptions: ['router', 'hostname', 'vendor'],
     randomIP: true,
@@ -42,4 +42,5 @@ module.exports = function(ip, acsurl, acspass) {
     }
     return console.log('!!! ERROR', err, data);
   }).listen(67);
+  return server;
 };

dist/http/index.js

@@ -82,5 +82,6 @@ module.exports = function(ip, port, url) {
       return console.error(e);
     }
   });
-  return srv.listen(port);
+  srv.listen(port);
+  return srv;
 };

dist/index.js

@@ -1,4 +1,4 @@
-var args, ask, dhcpd, httpd, ip, pkg, port, tftp;
+var args, ask, dhcpd, httpd, ip, pkg, port, servers, tftp;
 
 pkg = require('../package.json');
 
@@ -18,18 +18,30 @@ port = require('./get-port');
 
 tftp = require('./tftp');
 
+servers = [];
+
 if (args.tftp) {
-  tftp(args);
+  servers.push.apply(servers, tftp(args));
 } else if (args.dhcponly) {
-  dhcpd(ip, args.acsurl, args.acspass);
+  servers.push(dhcpd(ip, args.acsurl, args.acspass));
 } else {
   ask(ip).then(port).then(function(p) {
     var u, url;
     u = new URL(args.acsurl || ("http://" + ip));
     u.port = p;
     url = u.toString();
     console.log("listening for cwmp requests at " + url);
-    dhcpd(ip, url, args.acspass);
-    return httpd(ip, p, url);
+    servers.push(dhcpd(ip, url, args.acspass));
+    return servers.push(httpd(ip, p, url));
   });
 }
+
+process.on('SIGINT', function() {
+  console.log("\nshutting down servers from SIGINT (Ctrl+C)");
+  servers.forEach(function(server) {
+    return server.close();
+  });
+  return setTimeout(function() {
+    return process.exit();
+  }, 2000);
+});

dist/tftp.js

@@ -44,7 +44,7 @@ ProgressIndicator = (function(superClass) {
 })(Transform);
 
 module.exports = function(arg) {
-  var addr, eth, ip, network, server, tftp;
+  var addr, dhcpd, eth, ip, network, server, tftp;
   eth = arg.eth, ip = arg.ip, tftp = arg.tftp;
   if (eth != null) {
     network = ips.find(function(arg1) {
@@ -59,7 +59,7 @@ module.exports = function(arg) {
   addr = ip.split('.');
   addr.pop();
   addr = addr.join('.');
-  dhcp.createServer({
+  dhcpd = dhcp.createServer({
     range: [addr + '.10', addr + '.15'],
     forceOptions: ['router', 'hostname', 'bootFile'],
     randomIP: true,
@@ -124,5 +124,6 @@ module.exports = function(arg) {
     return console.error('ERROR:', err);
   });
   console.log('Starting tftp server, listening on ' + ip + ':69');
-  return server.listen();
+  server.listen();
+  return [dhcpd, server];
 };

package.json

@@ -1,6 +1,6 @@
 {
   "name": "tch-exploit",
-  "version": "2.0.1-rc4",
+  "version": "2.0.1-rc5",
   "main": "dist/index.js",
   "bin": "dist/index.js",
   "scripts": {

src/dhcp/index.coffee

@@ -44,3 +44,5 @@ module.exports = (ip, acsurl, acspass) ->
 
       console.log '!!! ERROR', err, data
     .listen 67
+
+  server

src/http/index.coffee

@@ -93,3 +93,5 @@ module.exports = (ip, port, url) ->
     else console.error e
 
   srv.listen port
+
+  srv

src/index.coffee

@@ -21,10 +21,12 @@ httpd = require './http'
 port = require './get-port'
 tftp = require './tftp'
 
+servers = []
+
 if args.tftp
-  tftp args
+  servers.push tftp(args)...
 else if args.dhcponly
-  dhcpd ip, args.acsurl, args.acspass
+  servers.push dhcpd ip, args.acsurl, args.acspass
 else
   ask ip
   .then port
@@ -36,5 +38,15 @@ else
 
     console.log "listening for cwmp requests at #{ url }"
 
-    dhcpd ip, url, args.acspass
-    httpd ip, p, url
+    servers.push dhcpd ip, url, args.acspass
+    servers.push httpd ip, p, url
+
+process.on 'SIGINT', ->
+  console.log "\nshutting down servers from SIGINT (Ctrl+C)"
+  
+  servers.forEach (server) ->
+    server.close()
+
+  setTimeout ->
+    process.exit()
+  , 2000

src/tftp.coffee

@@ -8,13 +8,13 @@ dhcp = require './dhcp/server'
 ips = require('./ips')()
 
 
-class ProgressIndicator extends Transform 
+class ProgressIndicator extends Transform
   constructor: (@size, options) ->
     super options
-    
+
     @last = 0
     @bytes = 0
-  
+
   _transform: (chunk, encoding, cb) ->
     @bytes += chunk.length
 
@@ -35,14 +35,14 @@ class ProgressIndicator extends Transform
 module.exports = ({ eth, ip, tftp }) ->
   if eth?
     network = ips.find ({ name }) -> name is eth
-  
-  ip ?= network?.address 
+
+  ip ?= network?.address
 
   addr = ip.split '.'
   addr.pop()
   addr = addr.join '.'
 
-  dhcp
+  dhcpd = dhcp
     .createServer
       range: [
         addr + '.10'
@@ -54,7 +54,7 @@ module.exports = ({ eth, ip, tftp }) ->
       router: [ ip ]
       hostname: 'second.gateway'
       broadcast: addr + '.255'
-      bootFile: (req, res) -> 
+      bootFile: (req, res) ->
         console.log req, res
 
         path.basename tftp
@@ -123,3 +123,5 @@ module.exports = ({ eth, ip, tftp }) ->
   console.log 'Starting tftp server, listening on ' + ip + ':69'
 
   server.listen()
+
+  [ dhcpd, server ]