idb 1.3.1

data/lib/gui/log_widget.rb

@@ -0,0 +1,71 @@
+require_relative 'log_plain_text_edit'
+require_relative 'i_device_syslog_thread'
+require 'htmlentities'
+
+module Idb
+  class LogWidget < Qt::Widget
+
+  def initialize *args
+    super *args
+
+    @start = Qt::PushButton.new "Start"
+    @start.connect(SIGNAL :released) {
+      @start.setEnabled(false)
+      @stop.setEnabled(true)
+      start_log
+    }
+
+    @stop = Qt::PushButton.new "Stop"
+    @stop.setEnabled(false)
+    @stop.connect(SIGNAL :released) {
+      @start.setEnabled(true)
+      @stop.setEnabled(false)
+      stop_log
+    }
+
+    @log_window = LogPlainTextEdit.new
+    @log_window.setReadOnly(true)
+
+    layout = Qt::VBoxLayout.new do |v|
+      v.add_widget(@log_window)
+      v.add_widget(@start)
+      v.add_widget(@stop)
+    end
+    setLayout(layout)
+    @colored = true
+
+  end
+
+
+  def start_log
+    h = HTMLEntities.new
+    @log_window.append_message "Please wait.. Streaming device syslog..."
+    @log_thread = IDeviceSyslogThread.new
+    @log_thread.connect(SIGNAL('new_entry(QString)')) {|line|
+      color = 'black'
+      if @colored
+        if line.include? "<Notice>"
+          color = 'grey'
+        elsif line.include? "<Error>"
+          color = 'red'
+        elsif line.include? "<Warning>"
+          color = 'Orange'
+        elsif line.include? "<Debug>"
+          color = 'Green'
+        end
+      end
+
+
+
+      line = line.encode('utf-8', :invalid => :replace, :undef => :replace, :replace => '_')
+
+      @log_window.append_message "<font color='#{color}'>#{h.encode(line.chomp)}</font>"
+    }
+  end
+
+  def stop_log
+    @log_thread.stop
+  end
+
+  end
+end

data/lib/gui/main_tab_widget.rb

@@ -0,0 +1,179 @@
+require_relative 'local_storage_tab_widget'
+require_relative 'url_handler_widget'
+require_relative 'app_binary_tab_widget'
+require_relative 'log_widget'
+require_relative 'snoop_it_tab_widget'
+require_relative 'cycript_console_widget'
+require_relative 'pasteboard_monitor_widget'
+require_relative 'fs_viewer_tab_widget'
+require_relative 'key_chain_widget'
+require_relative 'tool_widget'
+require 'Qt'
+
+module Idb
+  class MainTabWidget < Qt::TabWidget
+
+
+    def initialize *args
+      super *args
+      @tabs = Hash.new
+
+      @local_storage = LocalStorageTabWidget.new self
+      @local_storage.setEnabled(false)
+      @tabs[:local_storage] = addTab(@local_storage, "Storage")
+      @local_storage.connect(SIGNAL('currentChanged(int)')) { |x|
+       #@local_storage.currentWidget.refresh
+      }
+
+      @url_handler = URLHandlerWidget.new self
+      @url_handler.setEnabled(false)
+      @tabs[:url_handlers] = addTab(@url_handler, "URL Handlers")
+      connect(SIGNAL('currentChanged(int)')) { |x|
+  #      if isTabEnabled(@tabs[:url_handlers]) and
+  #          @tabs[:url_handlers] == currentIndex
+  #
+  #        @url_handler.refresh
+  #      end
+      }
+
+      @app_binary = AppBinaryTabWidget.new self
+      @app_binary.setEnabled(false)
+      @tabs[:app_binary] = addTab(@app_binary, "Binary")
+      @app_binary.connect(SIGNAL('currentChanged(int)')) { |x|
+        @app_binary.currentWidget.refresh
+      }
+
+      @fs_viewer = FsViewerTabWidget.new self
+      @fs_viewer.setEnabled(false)
+      @tabs[:fs_viewer] = addTab(@fs_viewer, "Filesystem")
+
+      @tools = ToolWidget.new self
+      @tools.setEnabled(false)
+      @tabs[:tools] = addTab(@tools, "Tools")
+
+  #    @isnoop = SnoopItTabWidget.new self
+  #    @isnoop.setEnabled(false)
+  #    @tabs[:isnoop] = addTab(@isnoop, "Snoop-It")
+  #    setTabToolTip(@tabs[:isnoop],"The performance of the snoop-it integration is not good enough to enable the tab yet.  If people would find it useful, let me know and I'll put time into it.")
+
+      @log = LogWidget.new self
+      @log.setEnabled(false)
+      @tabs[:log] = addTab(@log, "Log")
+
+      @keychain = KeyChainWidget.new self
+      @keychain.setEnabled(false)
+      @tabs[:keychain] = addTab(@keychain, "Keychain")
+
+  #    @cycript = CycriptConsoleWidget.new self
+  #    @cycript.setEnabled(false)
+  #    @tabs[:cycript] = addTab(@cycript, "Cycript")
+  #    setTabToolTip(@tabs[:cycript],"Cycript needs proper terminal emulation. At the moment this tab is not very useful and disabled.")
+
+      @pasteboard = PasteboardMonitorWidget.new self
+      @pasteboard.setEnabled(false)
+      @tabs[:pasteboard] = addTab(@pasteboard, "Pasteboard")
+
+      disable_all
+    end
+
+    def enableLog
+      @log.setEnabled(true)
+      setTabEnabled(@tabs[:log], true)
+    end
+
+    def enableAppBinary
+      @app_binary.setEnabled(true)
+      setTabEnabled(@tabs[:app_binary], true)
+      @app_binary.enableTabs
+    end
+
+    def enableCycript
+  #    @cycript.setEnabled(true)
+  #    setTabEnabled(@tabs[:cycript], true)
+    end
+
+    def enableLocalStorage
+      @local_storage.setEnabled(true)
+      setTabEnabled(@tabs[:local_storage], true)
+    end
+
+    def enableURLHandlers
+      @url_handler.setEnabled(true)
+      setTabEnabled(@tabs[:url_handlers], true)
+    end
+
+    def enablePasteboard
+      @pasteboard.setEnabled(true)
+      setTabEnabled(@tabs[:pasteboard], true)
+    end
+
+    def enableFSViewer
+      @fs_viewer.setEnabled(true)
+      setTabEnabled(@tabs[:fs_viewer], true)
+    end
+
+    def enableTools
+      @tools.setEnabled(true)
+      setTabEnabled(@tabs[:tools], true)
+    end
+
+
+    def enableKeychain
+      @keychain.setEnabled(true)
+      setTabEnabled(@tabs[:keychain], true)
+    end
+
+
+    def enableDeviceFunctions
+      enableCycript
+      enableLog
+      enablePasteboard
+      enableKeychain
+    end
+
+    def disable_all
+      setTabEnabled(@tabs[:local_storage], false)
+      setTabEnabled(@tabs[:log], false)
+      setTabEnabled(@tabs[:app_binary],false)
+      setTabEnabled(@tabs[:url_handlers],false)
+      setTabEnabled(@tabs[:pasteboard],false)
+  #    setTabEnabled(@tabs[:cycript],false)
+  #    setTabEnabled(@tabs[:isnoop],false)
+      setTabEnabled(@tabs[:fs_viewer],false)
+      setTabEnabled(@tabs[:keychain],false)
+      setTabEnabled(@tabs[:tools],false)
+    end
+
+    def clear
+      @tabs.each { |tab|
+        tab.clear
+      }
+
+    end
+
+    def refresh_current_tab
+      currentWidget.refresh_current_tab
+    end
+
+    def refresh_app_binary
+      enableAppBinary
+    end
+
+    def app_changed
+      clear
+      enableLocalStorage
+      enableURLHandlers
+  #    refresh_current_tab
+      @app_binary.refresh
+      enableFSViewer
+      @fs_viewer.set_start  $selected_app.app_dir
+      enableTools
+      @tools.enable_screenshot
+
+
+    end
+
+
+
+  end
+end

data/lib/gui/pasteboard_monitor_widget.rb

@@ -0,0 +1,116 @@
+require_relative 'pb_watcher_thread'
+require_relative 'log_plain_text_edit'
+
+module Idb
+  class PasteboardMonitorWidget < Qt::Widget
+
+    def initialize *args
+      super *args
+
+      @pbs_to_watch = Hash.new
+      @pbs_to_watch["general"] = ""
+
+
+      @stop = Qt::PushButton.new "Stop"
+      @stop.setEnabled(false)
+      @stop.connect(SIGNAL :released) {
+        @start.setEnabled(true)
+        @stop.setEnabled(false)
+        stop_log
+        @pb_config.setEnabled(true)
+      }
+
+      @log_window = LogPlainTextEdit.new
+      @log_window.setReadOnly(true)
+
+      @start = Qt::PushButton.new "Start"
+      @start.connect(SIGNAL :released) {
+        unless $device.pbwatcher_installed?
+          error = Qt::MessageBox.new
+          error.setInformativeText("pbwatcher not found on the device. Please visit the status dialog and install it.")
+          error.setIcon(Qt::MessageBox::Critical)
+          error.exec
+        else
+          @pb_config.setEnabled(false)
+          @start.setEnabled(false)
+          @stop.setEnabled(true)
+          launch_process
+        end
+      }
+
+
+      @pb_config = Qt::GroupBox.new self
+      @pb_config.setTitle "Pasteboard Names"
+      @pb_config_layout = Qt::GridLayout.new
+      @pb_config.setLayout @pb_config_layout
+
+      @pb_names = Qt::ListWidget.new @self
+
+      @pb_add = Qt::PushButton.new "Add"
+      @pb_add.connect(SIGNAL :released) {
+        @pbs_to_watch[@pb_text.text] = ""
+        @pb_names.addItem @pb_text.text
+        @pb_text.text = ""
+        @pb_remove.setEnabled(true)
+      }
+
+
+      @pb_remove = Qt::PushButton.new "Remove"
+      @pb_remove.setEnabled(false)
+      @pb_remove.connect(SIGNAL :released) {
+        removed_pb = @pb_names.item(@pb_names.current_row)
+        @pbs_to_watch.delete removed_pb.text unless removed_pb.nil?
+
+        row = @pb_names.current_row
+        @pb_names.takeItem  row unless row.nil?
+        if @pb_names.count == 0
+          @pb_remove.setEnabled(false)
+        end
+      }
+
+      @pb_text = Qt::LineEdit.new
+
+      @pb_config_layout.addWidget @pb_names, 0, 0, 1, 2
+      @pb_config_layout.addWidget @pb_text, 1, 0, 1, 2
+      @pb_config_layout.addWidget @pb_add, 2, 0, 1, 1
+      @pb_config_layout.addWidget @pb_remove, 2, 1, 1, 1
+
+
+      layout = Qt::GridLayout.new do |h|
+        h.add_widget @log_window, 0,0
+        h.add_widget @pb_config, 0,1
+
+        h.add_widget @start, 1, 0, 1, 2
+        h.add_widget @stop, 2, 0, 1, 2
+      end
+      setLayout(layout)
+
+    end
+
+
+    def launch_process
+      h = HTMLEntities.new
+      @log_window.append_message "Please wait.."
+      @pbwatcher_thread = PBWatcherThread.new
+      @pbwatcher_thread.connect(SIGNAL('new_entry(QString)')) {|line|
+        color = 'black'
+        date, time, app, payload = line.split " ", 4
+        pb, data = payload.split ":", 2
+        if @pbs_to_watch[pb] != data
+          @pbs_to_watch[pb] = data
+          new_entry = "#{time} #{pb} => #{data}"
+          new_entry = new_entry.encode('utf-8', :invalid => :replace, :undef => :replace, :replace => '_')
+          @log_window.append_message "<font color='#{color}'>#{h.encode(new_entry.chomp)}</font>"
+        end
+
+
+      }
+      @pbwatcher_thread.start_pbwatcher_thread @pbs_to_watch.select{|x,y| x != 'general' }.map{|x,y| "\"#{x}\""}.join ' '
+
+    end
+
+    def stop_log
+      @pbwatcher_thread.stop
+    end
+  end
+end

data/lib/gui/path_list_widget_item.rb

@@ -0,0 +1,5 @@
+module Idb
+  class PathListWidgetItem < Qt::ListWidgetItem
+  attr_accessor :full_path
+  end
+end

data/lib/gui/pb_watcher_thread.rb

@@ -0,0 +1,63 @@
+module Idb
+  class PBWatcherThread < Qt::Object
+    signals "new_entry(QString)"
+
+    def initialize *args
+      super *args
+      $terminate_pbwatcher_thread = false
+    end
+
+
+    def stop
+      $terminate_pbwatcher_thread = true
+    end
+
+
+    def start_pbwatcher_thread pbs
+      @pbwatcher_thread = Thread.new do
+        channel = $device.ssh.open_channel do |ch|
+          ch.request_pty do |ch, success|
+          cmd = "/var/root/pbwatcher 1 #{pbs}"
+          $log.info "Executing pbwatcher: #{cmd}"
+          ch.exec cmd do |ch, success|
+            $log.error "could not execute command" unless success
+
+            # "on_data" is called when the process writes something to stdout
+            ch.on_data do |c, data|
+              emit new_entry(data)
+            end
+
+             # "on_extended_data" is called when the process writes something to stderr
+             ch.on_extended_data do |c, type, data|
+               emit new_entry(data)
+             end
+
+             ch.on_close { |ch|
+               $log.info "pbwatcher terminated"
+             }
+          end
+            end
+        end
+
+        loop do
+          #TODO mutex to protect device?
+          #even better: make one central thread that calls process.
+          # and all functions using it call it to ensure its running. or auto start it.
+          sleep 0.5
+          $device.ssh.process
+          #$device.ssh.process 0
+          if $terminate_pbwatcher_thread
+            $log.info "Terminating pbwatcher"
+            channel.close
+            break
+          end
+        end
+        $log.info "Terminating thread"
+      end
+
+
+    end
+
+
+  end
+end

data/lib/gui/plist_file_widget.rb

@@ -0,0 +1,66 @@
+require_relative 'path_list_widget_item'
+
+module Idb
+  class PlistFileWidget < Qt::Widget
+
+    def initialize *args
+      super *args
+
+      @refresh = Qt::PushButton.new "Refresh"
+      @refresh.connect(SIGNAL :released) {
+        refresh
+      }
+
+
+      @list = Qt::ListWidget.new self
+      @list.connect(SIGNAL('itemDoubleClicked(QListWidgetItem*)')) { |item|
+        cache_name =  $selected_app.cache_file item.full_path
+        if cache_name.nil?
+          $log.error "File #{item.full_path} could not be downloaded. Either the file does not exist (e.g., dead symlink) or there is a permission problem."
+        else
+          $device.ops.open cache_name
+        end
+      }
+  #    @list.setContextMenuPolicy(Qt::CustomContextMenu);
+  #    @list.connect(SIGNAL('customContextMenuRequested(QPoint)')) { |item|
+  #      menu = Qt::Menu.new("Context menu", self)
+  #      menu.addAction(Qt::Action.new("Hello", self));
+  #      menu.exec(mapToGlobal(pos));
+  #
+  #    }
+
+
+      # "Launch app"
+
+      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
+      plist_files = $selected_app.find_plist_files
+      plist_files.each { |full_path|
+        item = PathListWidgetItem.new
+        if $device.simulator?
+          item.setText full_path.sub($selected_app.app_dir,'')
+        else
+          pc = $device.protection_class full_path
+          item.setText full_path.sub($selected_app.app_dir,'') + " => " + pc.strip
+        end
+        item.full_path = full_path
+        @list.addItem item
+      }
+    end
+
+
+
+
+  end
+end