lib/pwn/plugins/transparent_browser.rb in pwn-0.5.163 vs lib/pwn/plugins/transparent_browser.rb in pwn-0.5.164
- old
+ new
@@ -333,19 +333,17 @@
# browser_obj: 'required - browser_obj returned from #open method)'
# )
public_class_method def self.close(opts = {})
browser_obj = opts[:browser_obj]
+ browser = browser_obj[:browser]
+ tor_obj = browser_obj[:tor_obj]
- if browser_obj[:tor_obj]
- tor_obj = browser_obj[:tor_obj]
- PWN::Plugins::Tor.stop(tor_obj: browser_obj[:tor_obj])
- end
+ PWN::Plugins::Tor.stop(tor_obj: browser_obj[:tor_obj]) if tor_obj
- unless browser_obj[:browser].is_a?(RestClient)
- # Close the browser unless this_browser_obj.nil? (thus the &)
- browser_obj[:browser]&.close
- end
+ # Close the browser unless browser.nil? (thus the &)
+ browser&.close unless browser.to_s == 'RestClient'
+
nil
rescue StandardError => e
raise e
end