lib/pwn/plugins/transparent_browser.rb in pwn-0.5.172 vs lib/pwn/plugins/transparent_browser.rb in pwn-0.5.173

@@ -128,12 +128,13 @@
           args.push('--devtools') if with_devtools
           options = Selenium::WebDriver::Firefox::Options.new(
             args: args,
             accept_insecure_certs: true
           )
+
+          options.web_socket_url = true
           options.profile = this_profile
-          # driver = Selenium::WebDriver.for(:firefox, capabilities: options)
           driver = Selenium::WebDriver.for(:firefox, options: options)
           browser_obj[:browser] = Watir::Browser.new(driver)
 
         when :chrome
           this_profile = Selenium::WebDriver::Chrome::Profile.new
@@ -153,12 +154,12 @@
           options = Selenium::WebDriver::Chrome::Options.new(
             args: args,
             accept_insecure_certs: true
           )
 
+          options.web_socket_url = true
           options.profile = this_profile
-          # driver = Selenium::WebDriver.for(:chrome, capabilities: options)
           driver = Selenium::WebDriver.for(:chrome, options: options)
           browser_obj[:browser] = Watir::Browser.new(driver)
 
         when :headless, :headless_firefox
           this_profile = Selenium::WebDriver::Firefox::Profile.new
@@ -216,10 +217,11 @@
           options = Selenium::WebDriver::Firefox::Options.new(
             args: args,
             accept_insecure_certs: true
           )
 
+          options.web_socket_url = true
           options.profile = this_profile
           driver = Selenium::WebDriver.for(:firefox, options: options)
           browser_obj[:browser] = Watir::Browser.new(driver)
 
         when :headless_chrome
@@ -237,10 +239,11 @@
           options = Selenium::WebDriver::Chrome::Options.new(
             args: args,
             accept_insecure_certs: true
           )
 
+          options.web_socket_url = true
           options.profile = this_profile
           driver = Selenium::WebDriver.for(:chrome, options: options)
           browser_obj[:browser] = Watir::Browser.new(driver)
 
         when :rest
@@ -284,10 +287,11 @@
           browser_obj[:browser].goto('about:blank')
           rand_tab = SecureRandom.hex(8)
           browser_obj[:browser].execute_script("document.title = '#{rand_tab}'")
 
           browser_obj[:devtools] = browser_obj[:browser].driver.devtools
+          browser_obj[:bidi] = browser_obj[:browser].driver.bidi
 
           # browser_obj[:devtools].send_cmd('DOM.enable')
           # browser_obj[:devtools].send_cmd('Log.enable')
           # browser_obj[:devtools].send_cmd('Network.enable')
           # browser_obj[:devtools].send_cmd('Page.enable')
@@ -521,10 +525,10 @@
 
         devtools = browser_obj[:devtools]
 
         case action.to_s.downcase.to_sym
         when :pause
-          console('browser_obj': browser_obj, js: 'debugger')
+          console(browser_obj: browser_obj, js: 'debugger')
 
           # devtools.send_cmd('Debugger.enable')
           # devtools.send_cmd(
           #   'Debugger.setInstrumentationBreakpoint',
           #   instrumentation: 'beforeScriptExecution'