From 10dee28fbd21ddc224561c9ba8aef578f50c096e Mon Sep 17 00:00:00 2001
From: jvazquez-r7 <juan_vazquez@rapid7.com>
Date: Wed, 3 Sep 2014 12:01:48 -0500
Subject: [PATCH] Add http socket to the module sockets and allow the framework
 to cleanup

---
 .../admin/appletv/appletv_display_image.rb    | 43 +++++++++----------
 .../admin/appletv/appletv_display_video.rb    | 43 +++++++++----------
 2 files changed, 40 insertions(+), 46 deletions(-)

diff --git a/modules/auxiliary/admin/appletv/appletv_display_image.rb b/modules/auxiliary/admin/appletv/appletv_display_image.rb
index e532504cea..06a11af16f 100644
--- a/modules/auxiliary/admin/appletv/appletv_display_image.rb
+++ b/modules/auxiliary/admin/appletv/appletv_display_image.rb
@@ -62,33 +62,30 @@ class Metasploit4 < Msf::Auxiliary
   def send_image_request(opts)
     http = nil
 
-    begin
-      http = Rex::Proto::Http::Client.new(
-        rhost,
-        rport.to_i,
-        {
-          'Msf' => framework,
-          'MsfExploit' => self
-        },
-        ssl,
-        ssl_version,
-        proxies,
-        datastore['USERNAME'],
-        datastore['PASSWORD']
-      )
+    http = Rex::Proto::Http::Client.new(
+      rhost,
+      rport.to_i,
+      {
+        'Msf' => framework,
+        'MsfExploit' => self
+      },
+      ssl,
+      ssl_version,
+      proxies,
+      datastore['USERNAME'],
+      datastore['PASSWORD']
+    )
+    add_socket(http)
 
-      http.set_config('agent' => datastore['UserAgent'])
+    http.set_config('agent' => datastore['UserAgent'])
 
-      req = http.request_raw(opts)
-      res = http.send_recv(req)
+    req = http.request_raw(opts)
+    res = http.send_recv(req)
 
-      sleep(datastore['TIME']) if res.code == 200
-      http.close
-    ensure
-      cleanup
-    end
+    Rex.sleep(datastore['TIME']) if res.code == 200
+    http.close
 
-    http
+    res
   end
 
 
diff --git a/modules/auxiliary/admin/appletv/appletv_display_video.rb b/modules/auxiliary/admin/appletv/appletv_display_video.rb
index 787d630d81..208f46db99 100644
--- a/modules/auxiliary/admin/appletv/appletv_display_video.rb
+++ b/modules/auxiliary/admin/appletv/appletv_display_video.rb
@@ -65,32 +65,29 @@ class Metasploit4 < Msf::Auxiliary
   def send_video_request(opts)
     http = nil
 
-    begin
-      http = Rex::Proto::Http::Client.new(
-        rhost,
-        rport.to_i,
-        {
-          'Msf' => framework,
-          'MsfExploit' => self
-        },
-        ssl,
-        ssl_version,
-        proxies,
-        datastore['USERNAME'],
-        datastore['PASSWORD']
-      )
+    http = Rex::Proto::Http::Client.new(
+      rhost,
+      rport.to_i,
+      {
+        'Msf' => framework,
+        'MsfExploit' => self
+      },
+      ssl,
+      ssl_version,
+      proxies,
+      datastore['USERNAME'],
+      datastore['PASSWORD']
+    )
+    add_socket(http)
 
-      http.set_config('agent' => datastore['UserAgent'])
+    http.set_config('agent' => datastore['UserAgent'])
 
-      req = http.request_raw(opts)
-      res = http.send_recv(req)
-      sleep(datastore['TIME']) if res.code == 200
-      http.close
-    ensure
-      cleanup
-    end
+    req = http.request_raw(opts)
+    res = http.send_recv(req)
+    Rex.sleep(datastore['TIME']) if res.code == 200
+    http.close
 
-    http
+    res
   end