From e17a48fae2eeea67b891d98a8419ab9b973f3161 Mon Sep 17 00:00:00 2001
From: Brendan Coles <bcoles@gmail.com>
Date: Mon, 22 Jun 2015 09:35:31 +0000
Subject: [PATCH] rescue TypeError, ArgumentError

---
 .../main/network_stack/handlers/dynamicreconstruction.rb | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/core/main/network_stack/handlers/dynamicreconstruction.rb b/core/main/network_stack/handlers/dynamicreconstruction.rb
index 82e60cdae..4c57017bd 100644
--- a/core/main/network_stack/handlers/dynamicreconstruction.rb
+++ b/core/main/network_stack/handlers/dynamicreconstruction.rb
@@ -32,14 +32,17 @@ module BeEF
                     'Content-Type' => 'text/javascript',
                     'Access-Control-Allow-Origin' => '*',
                     'Access-Control-Allow-Methods' => 'POST, GET'
-
-            PQ << {
+            begin
+              PQ << {
                 :beefhook => params[:bh],
                 :stream_id => Integer(params[:sid]),
                 :packet_id => Integer(params[:pid]),
                 :packet_count => Integer(params[:pc]),
                 :data => params[:d]
-            }
+              }
+            rescue TypeError, ArgumentError => e
+              print_error "Hooked browser returned an invalid argument: #{e}"
+            end
 
             Thread.new {
               check_packets()