idb 1.3.1

data/lib/gui/qt_ruby_variant.rb

@@ -0,0 +1,16 @@
+module Idb
+  class Qt::RubyVariant < Qt::Variant
+      def initialize(value)
+          super()
+          @value = value
+      end
+
+      attr_accessor :value
+  end
+
+  class Object
+      def to_variant
+          Qt::RubyVariant.new self
+      end
+  end
+end

data/lib/gui/screenshot_wizard.rb

@@ -0,0 +1,169 @@
+require_relative '../lib/screen_shot_util'
+require 'Qt'
+
+module Idb
+module ScreenShotWizard
+  Pages = Hash.new
+  wizard = nil
+
+
+  class ScreenShotWizard < Qt::Wizard
+    attr_accessor :app, :screenshot, :result
+
+
+    def self.new_with_app app
+      wiz = self.new
+      wiz.app = app
+      wiz.screenshot = ScreenShotUtil.new app.app_dir, $device.ops, false
+      wiz
+    end
+
+    def initialize
+      super
+
+      Pages[:intro] = add_page(IntroPage.new)
+      Pages[:open_app] = add_page(OpenAppPage.new self)
+      Pages[:background_app] = add_page(BackgroundAppPage.new self)
+      Pages[:screen_shot_found] = add_page(ScreenShotFoundPage.new self)
+      Pages[:no_screen_shot] = add_page(NoScreenShotPage.new self)
+      set_window_title("Screenshot Wizard")
+      wizard = self
+      show
+    end
+  end
+
+
+  class IntroPage < Qt::WizardPage
+    def initialize
+      super
+
+      setTitle("Introduction")
+      label = Qt::Label.new("This wizard will guide you through the testing for the iOS backgrounding screenshot vulnerability.")
+      label.word_wrap = true
+      layout = Qt::VBoxLayout.new do |v|
+        v.add_widget(label)
+      end
+      setLayout(layout)
+    end
+
+    def validatePage
+      puts "[*] Marking screenshot time"
+      #wiz = parentWidget.parentWidget.parentWidget
+      wizard.screenshot.mark
+      return true
+    end
+  end
+
+  class OpenAppPage < Qt::WizardPage
+
+    def initialize *args
+      super *args
+
+      setTitle("Launch Application")
+      label = Qt::Label.new("Launch the application and navigate to a view that contains potentially sensitive data. Or click below to launch the app automatically.")
+      label.word_wrap = true
+
+      launch_button = Qt::PushButton.new "Launch app"
+      launch_button.connect(SIGNAL :released) { |x|
+        wizard.app.launch
+      }
+
+      layout = Qt::VBoxLayout.new do |v|
+        v.add_widget(label)
+        v.add_widget(launch_button)
+      end
+      setLayout(layout)
+
+    end
+
+
+  end
+
+
+  class BackgroundAppPage < Qt::WizardPage
+
+    def initialize *args
+      super *args
+      setTitle("Background App")
+
+      label = Qt::Label.new("Now Background the app by hitting the home button (XX in the simulator). Then click continue.")
+      label.word_wrap = true
+      layout = Qt::VBoxLayout.new do |v|
+        v.add_widget(label)
+      end
+      setLayout(layout)
+    end
+
+    def validatePage
+      puts "[*] Checking for screenshot"
+      #wiz = parentWidget.parentWidget.parentWidget
+      wizard.result = wizard.screenshot.check
+      puts wizard.result
+      true
+    end
+
+    def nextId
+      #wiz = parentWidget.parentWidget.parentWidget
+      puts "Determining next id"
+      if wizard.result.nil?
+        Pages[:no_screen_shot]
+      else
+        Pages[:screen_shot_found]
+      end
+    end
+
+
+  end
+
+  class NoScreenShotPage < Qt::WizardPage
+    def initialize *args
+      super *args
+      setFinalPage(true)
+      setTitle("No Screenshot Found")
+      label = Qt::Label.new("No new screenshot was detected for this application.")
+      label.word_wrap = true
+      layout = Qt::VBoxLayout.new do |v|
+        v.add_widget(label)
+      end
+      setLayout(layout)
+    end
+
+    def nextId
+      -1
+    end
+  end
+
+  class ScreenShotFoundPage < Qt::WizardPage
+    def initialize *args
+      super *args
+      setFinalPage(true)
+    end
+
+    def initializePage *args
+      super  *args
+
+      #wiz = parentWidget.parentWidget.parentWidget
+      screenshot_file = wizard.app.cache_file wizard.result
+
+      setTitle("Screenshot Found (click to open)")
+      screen = Qt::Pixmap.new screenshot_file
+      button = Qt::PushButton.new
+      button.setFlat(true)
+      button.setIcon(Qt::Icon.new(screen))
+      button.setIconSize(parentWidget.size)
+      button.connect(SIGNAL :released) { |x|
+        $device.ops.open screenshot_file
+      }
+      layout = Qt::VBoxLayout.new do |v|
+        v.add_widget(button)
+      end
+      setLayout(layout)
+    end
+  end
+
+
+  def nextId
+    -1
+  end
+end
+end

data/lib/gui/settings_dialog.rb

@@ -0,0 +1,69 @@
+require_relative 'settings_tab_widget'
+
+
+module Idb
+  class SettingsDialog < Qt::Dialog
+
+    def initialize *args
+      super *args
+
+      @layout = Qt::GridLayout.new
+      setLayout(@layout)
+      setWindowTitle("Settings")
+
+      @tabs = SettingsTabWidget.new self
+      @layout.addWidget @tabs, 0,0,1,2
+
+      @save_button = Qt::PushButton.new "Save"
+      @save_button.setDefault true
+
+      @save_button.connect(SIGNAL(:released)) {|x|
+        $settings["ssh_host"] = @tabs.ssh_host.text
+        $settings["ssh_port"] = @tabs.ssh_port.text
+        $settings["ssh_username"] = @tabs.ssh_username.text
+        $settings["ssh_password"] = @tabs.ssh_password.text
+
+        if @tabs.usbmux_radio.isChecked
+          $settings["device_connection_mode"] = "usb"
+        else
+          $settings["device_connection_mode"] = "ssh"
+        end
+
+        $settings["manual_ssh_port"] = @tabs.manual_ssh_port.text
+        $settings["idb_utility_port"] = @tabs.idb_utility_port.text
+        $settings["sqlite_editor"] = @tabs.sqlite_editor.text
+
+        $settings.store
+
+        if not $device.nil? and  forwarders_changed?
+          reply = Qt::MessageBox::question(self, "Reload Port Forwards", "Portforwarding has changed. Do you want to apply the new configuration?<br>(This may interrupt existing connections)", Qt::MessageBox::Yes, Qt::MessageBox::No);
+          if reply == Qt::MessageBox::Yes
+            $device.restart_port_forwarding
+          end
+        end
+
+        accept()
+      }
+      @cancel_button = Qt::PushButton.new "Cancel"
+      @cancel_button.connect(SIGNAL(:released)) {|x|
+        reject()
+      }
+
+      @layout.addWidget @save_button, 2, 1
+      @layout.addWidget @cancel_button, 2, 0
+
+
+
+
+
+
+    end
+  #
+    def forwarders_changed?
+      @tabs.forwarders_changed?
+    end
+
+
+
+  end
+end

data/lib/gui/settings_tab_widget.rb

@@ -0,0 +1,149 @@
+require_relative 'ssh_port_forward_tab_widget'
+
+module Idb
+  class SettingsTabWidget < Qt::TabWidget
+    attr_accessor :ssh_host, :ssh_port, :ssh_username, :ssh_password, :usbmux_radio, :manual_ssh_port, :idb_utility_port
+    attr_accessor :sqlite_editor
+
+    def initialize *args
+      super *args
+
+      setup_device_config
+      addTab @device_config_tab, "Device Config"
+      setup_forwards
+      addTab @forward_config, "Port Forwarding"
+      setup_external_apps
+      addTab @external_apps_config, "External Editors"
+
+    end
+
+    def setup_device_config
+      @device_config_tab = Qt::Widget.new self
+      device_config_layout = Qt::GridLayout.new
+      @device_config_tab.setLayout device_config_layout
+
+      device_config_layout.addWidget @device_config, 0, 0, 1,2
+
+
+      @connection_widget = Qt::Widget.new self
+      @connection_widget_layout = Qt::GridLayout.new
+      @connection_widget.setLayout(@connection_widget_layout)
+      device_config_layout.addWidget @connection_widget, 0, 0, 1, 2
+
+      @ssh_direct_radio = Qt::RadioButton.new @connection_widget
+      @ssh_direct_radio.setText("SSH directly")
+      @ssh_direct_radio.connect(SIGNAL :released) {
+        @ssh_host.setEnabled(true)
+
+      }
+
+      @usbmux_radio = Qt::RadioButton.new @connection_widget
+      @usbmux_radio.setText("SSH via USB (usbmuxd)")
+      @usbmux_radio.connect(SIGNAL :released) {
+        @ssh_host.setEnabled(false)
+      }
+
+      @connection_widget_layout.addWidget @ssh_direct_radio, 0,0
+      @connection_widget_layout.addWidget @usbmux_radio, 0, 1
+
+
+
+      # ssh username
+      @label_ssh_username = Qt::Label.new  "SSH Username:", self, 0
+      @ssh_username = Qt::LineEdit.new $settings.ssh_username
+      device_config_layout.addWidget @label_ssh_username, 1, 0
+      device_config_layout.addWidget @ssh_username, 1, 1
+
+      # ssh password
+      @label_ssh_password = Qt::Label.new  "SSH Password:", self, 0
+      @ssh_password = Qt::LineEdit.new  $settings.ssh_password
+      device_config_layout.addWidget @label_ssh_password, 2, 0
+      device_config_layout.addWidget @ssh_password, 2, 1
+
+      # ssh host
+      @label_ssh_host = Qt::Label.new  "SSH Host:", self, 0
+      @ssh_host = Qt::LineEdit.new  $settings.ssh_host
+      device_config_layout.addWidget @label_ssh_host, 3, 0
+      device_config_layout.addWidget @ssh_host, 3, 1
+
+      # ssh port
+      @label_ssh_port = Qt::Label.new  "SSH Port:", self, 0
+      @ssh_port = Qt::LineEdit.new  $settings.ssh_port.to_s
+      device_config_layout.addWidget @label_ssh_port, 4, 0
+      device_config_layout.addWidget @ssh_port, 4, 1
+
+
+      if $settings['device_connection_mode'] == "usb"
+        @usbmux_radio.setChecked(true)
+        @ssh_host.setEnabled(false)
+      end
+
+
+      if $settings['device_connection_mode'] == "ssh"
+        @ssh_direct_radio.setChecked(true)
+      end
+
+    end
+
+    def setup_forwards
+
+      @forward_config = Qt::Widget.new self
+      forward_config_layout = Qt::GridLayout.new
+      @forward_config.setLayout forward_config_layout
+
+      @forward_tabs = SSHPortForwardTabWidget.new self
+      forward_config_layout.addWidget @forward_tabs, 0, 0, 1, 2
+
+      # manual SSH port
+      @label_manual_ssh_port = Qt::Label.new  "Port for Manual SSH:", self, 0
+      @manual_ssh_port = Qt::LineEdit.new  $settings.manual_ssh_port.to_s
+      forward_config_layout.addWidget @label_manual_ssh_port, 2, 0
+      forward_config_layout.addWidget @manual_ssh_port, 2, 1
+
+      # idb utility forward port
+      @label_idb_utility_port = Qt::Label.new  "Port for internal idb operations:", self, 0
+      @idb_utility_port = Qt::LineEdit.new  $settings.idb_utility_port.to_s
+      forward_config_layout.addWidget @label_idb_utility_port, 3, 0
+      forward_config_layout.addWidget @idb_utility_port, 3, 1
+
+    end
+
+    def setup_external_apps
+      @external_apps_config = Qt::Widget.new self
+      external_apps_config_layout = Qt::GridLayout.new
+      @external_apps_config.setLayout external_apps_config_layout
+
+
+
+
+
+
+
+
+      # sqlite editor
+      @label_sqlite_editor = Qt::Label.new  "SQLite Editor:", self, 0
+      @sqlite_editor = Qt::Label.new  $settings.sqlite_editor
+      @sqlite_editor_change = Qt::PushButton.new "Change"
+
+      @sqlite_editor_change.connect(SIGNAL :released) {
+        file_dialog = Qt::FileDialog.new
+        file_dialog.setAcceptMode(Qt::FileDialog::AcceptOpen)
+
+        file_dialog.connect(SIGNAL('fileSelected(QString)')) { |x|
+          @sqlite_editor.setText x
+        }
+        file_dialog.exec
+      }
+      external_apps_config_layout.addWidget @label_sqlite_editor, 1, 0
+      external_apps_config_layout.addWidget @sqlite_editor, 1, 1
+      external_apps_config_layout.addWidget @sqlite_editor_change, 1, 2
+
+    end
+
+    def forwarders_changed?
+      @forward_tabs.forwarders_changed?
+    end
+
+
+  end
+end

data/lib/gui/shared_libraries_widget.rb

@@ -0,0 +1,47 @@
+module Idb
+  class SharedLibrariesWidget < Qt::Widget
+      def initialize *args
+        super *args
+
+        @refresh = Qt::PushButton.new "Refresh"
+        @refresh.connect(SIGNAL :released) {
+          refresh
+        }
+
+        @list = Qt::ListWidget.new self
+
+        layout = Qt::VBoxLayout.new do |v|
+          v.add_widget(@list)
+          v.add_widget(@refresh)
+        end
+        setLayout(layout)
+      end
+
+      def clear
+        @list.clear
+      end
+
+      def refresh
+        @list.clear
+        if not $selected_app.binary.nil?
+          shared_lib = $selected_app.binary.get_shared_libraries
+          if shared_lib.nil?
+            item = Qt::ListWidgetItem.new
+            item.setText "Error: otool required"
+            @list.addItem item
+            return
+          end
+          shared_lib.each { |lib|
+            item = Qt::ListWidgetItem.new
+            item.setText lib
+            @list.addItem item
+          }
+          setEnabled(true)
+        else
+          setEnabled(false)
+        end
+
+      end
+
+  end
+end