ruby-netmon-qt 1.0.0

data/lib/netmon-qt/app/components/connstableview.rb

@@ -0,0 +1,239 @@
+require_relative "connstableview/sortfilterproxymodel"
+require_relative "connstableview/statusbar"
+require_relative "connstableview/store"
+
+class ConnsTableView < RubyQt6::Bando::QWidget
+  QSS_BUTTON_AUTOREFRESH = "
+    QPushButton {
+      background: #EFF0F1;
+      color: #7F8C8D;
+      border: 1px solid #BCBEBF;
+      border-bottom: 3px solid #B0B0B0;
+      border-radius: 3px;
+      padding: 6px;
+      font-weight: bold;
+    }
+    QPushButton:checked {
+      background: #FFFFFF;
+      color: #31363B;
+    }
+  "
+
+  q_object do
+    slot "_on_filter_changed()"
+    slot "_on_autorefreshbtn_toggled(bool)"
+    slot "_on_autorefresh_timer_timeout()"
+    slot "_on_tableview_custom_context_menu_requested(QPoint)"
+    slot "_on_endprocess_action_triggered()"
+    slot "_on_whois_action_triggered()"
+    slot "_on_copy_action_triggered()"
+  end
+
+  attr_reader :processfilter, :protocolfilter, :statefilter, :userfilter
+  attr_reader :statusbar
+
+  def initialize
+    super
+
+    @lastindex = nil
+
+    initialize_store
+    initialize_actions
+    initialize_toolbar
+    initialize_tableview
+    initialize_statusbar
+    initialize_timer_autorefresh
+
+    mainlayout = QVBoxLayout.new(self)
+    mainlayout.add_widget(@toolbar)
+    mainlayout.add_widget(@tableview)
+
+    @userfilter.set_current_text(Etc.getpwuid.name)
+    @autorefreshbtn.set_checked(true)
+  end
+
+  private
+
+  def initialize_store
+    @store = Store.new(self)
+  end
+
+  def initialize_actions
+    @endprocess_action = initialize_actions_act(QIcon::ThemeIcon::ApplicationExit, "End Process", :_on_endprocess_action_triggered)
+    @whois_action = initialize_actions_act(QIcon::ThemeIcon::EditFind, "", :_on_whois_action_triggered)
+    @copy_action = initialize_actions_act(QIcon::ThemeIcon::EditCopy, "", :_on_copy_action_triggered)
+  end
+
+  def initialize_actions_act(icon, text, slot)
+    action = QAction.new(QIcon.from_theme(icon), text, self)
+    action.triggered.connect(self, slot)
+    action
+  end
+
+  def initialize_toolbar
+    @processfilter = QComboBox.new
+    @processfilter.set_fixed_width(128)
+    @processfilter.add_item("*")
+    @store.active_processes.each { |process| @processfilter.add_item(process) }
+    @processfilter.current_text_changed.connect(self, :_on_filter_changed)
+
+    @protocolfilter = QComboBox.new
+    @protocolfilter.add_item("*")
+    @store.active_protocols.each { |protocol| @protocolfilter.add_item(protocol) }
+    @protocolfilter.set_fixed_width(128)
+    @protocolfilter.current_text_changed.connect(self, :_on_filter_changed)
+
+    @statefilter = QComboBox.new
+    @statefilter.add_item("*")
+    @store.active_states.each { |state| @statefilter.add_item(state) }
+    @statefilter.set_fixed_width(128)
+    @statefilter.current_text_changed.connect(self, :_on_filter_changed)
+
+    @userfilter = QComboBox.new
+    @userfilter.set_fixed_width(128)
+    @userfilter.add_item("*")
+    @store.active_users.each { |user| @userfilter.add_item(user) }
+    @userfilter.current_text_changed.connect(self, :_on_filter_changed)
+
+    @autorefreshbtn = QPushButton.new("")
+    @autorefreshbtn.set_fixed_width(192)
+    @autorefreshbtn.set_style_sheet(QSS_BUTTON_AUTOREFRESH)
+    @autorefreshbtn.set_checkable(true)
+    @autorefreshbtn.toggled.connect(self, :_on_autorefreshbtn_toggled)
+
+    @toolbar = QWidget.new
+    @toolbarlayout = QHBoxLayout.new(@toolbar)
+    @toolbarlayout.add_widget(initialize_toolbar_label("Process:"))
+    @toolbarlayout.add_widget(@processfilter)
+    @toolbarlayout.add_spacing(8)
+    @toolbarlayout.add_widget(initialize_toolbar_label("Protocol:"))
+    @toolbarlayout.add_widget(@protocolfilter)
+    @toolbarlayout.add_spacing(8)
+    @toolbarlayout.add_widget(initialize_toolbar_label("State:"))
+    @toolbarlayout.add_widget(@statefilter)
+    @toolbarlayout.add_spacing(8)
+    @toolbarlayout.add_widget(initialize_toolbar_label("User:"))
+    @toolbarlayout.add_widget(@userfilter)
+    @toolbarlayout.add_stretch
+    @toolbarlayout.add_widget(@autorefreshbtn)
+  end
+
+  def initialize_toolbar_label(text)
+    label = QLabel.new(text)
+    label.set_fixed_width(64)
+    label
+  end
+
+  def initialize_tableview
+    @tableview = QTableView.new
+    @tableview.set_selection_behavior(QAbstractItemView::SelectRows)
+    @tableview.set_selection_mode(QAbstractItemView::SingleSelection)
+    @tableview.set_context_menu_policy(Qt::ContextMenuPolicy::CustomContextMenu)
+    @tableview.custom_context_menu_requested.connect(self, :_on_tableview_custom_context_menu_requested)
+
+    @storeproxymodel = SortFilterProxyModel.new(self)
+    @storeproxymodel.set_source_model(@store.itemmodel)
+    @tableview.set_model(@storeproxymodel)
+
+    @tableview.horizontal_header.hide_section(0)
+    @tableview.vertical_header.set_visible(false)
+    @tableview.set_column_width(COLUMN_PROCESS_NAME, 128)
+    @tableview.set_column_width(COLUMN_LOCAL_ADDRESS, 192)
+    @tableview.set_column_width(COLUMN_REMOTE_ADDRESS, 192)
+
+    @tableview.sort_by_column(COLUMN_PROCESS_NAME, Qt::AscendingOrder)
+    @tableview.set_sorting_enabled(true)
+  end
+
+  def initialize_statusbar
+    @statusbar = StatusBar.new(@storeproxymodel)
+  end
+
+  def initialize_timer_autorefresh
+    @timer = QTimer.new(self)
+    @timer.set_interval(NetmonQt.settings.GET_connections_check_interval)
+    @timer.timeout.connect(self, :_on_autorefresh_timer_timeout)
+  end
+
+  def update_filter_additem(filter, items)
+    index = filter.count - 1
+    items.reverse_each do |item|
+      loop do
+        case filter.item_text(index) <=> item
+        when +0 then break
+        when +1 then (index -= 1) || next
+        when -1 then filter.insert_item(index + 1, item) || break
+        end
+      end
+    end
+  end
+
+  def _on_filter_changed
+    @storeproxymodel.invalidate
+    @statusbar.refresh
+  end
+
+  def _on_autorefreshbtn_toggled(checked)
+    if checked
+      @autorefreshbtn.set_text("Auto Refresh:  ON")
+      @timer.start
+      _on_autorefresh_timer_timeout
+    else
+      @autorefreshbtn.set_text("Auto Refresh: OFF")
+      @timer.stop
+    end
+  end
+
+  def _on_autorefresh_timer_timeout
+    @store.refresh
+
+    update_filter_additem(@processfilter, @store.active_processes)
+    update_filter_additem(@userfilter, @store.active_users)
+    _on_filter_changed
+  end
+
+  def _on_tableview_custom_context_menu_requested(position)
+    @lastindex = @tableview.index_at(position)
+    return unless @lastindex.valid?
+
+    menu = QMenu.new("", self)
+    menu.set_attribute(Qt::WA_DeleteOnClose)
+
+    pid = @lastindex.sibling_at_column(COLUMN_PROCESS_ID).data.value.to_s
+    remote_address = @lastindex.sibling_at_column(COLUMN_REMOTE_ADDRESS).data.value
+    remote_address = remote_address.contains(".") ? remote_address : "#{remote_address[0, 12]}..."
+
+    @endprocess_action.set_enabled(pid.to_i.nonzero?)
+    @whois_action.set_text("Whois #{remote_address} - via IPinfo")
+    @copy_action.set_text("Copy \"#{@lastindex.data.value}\"")
+
+    menu.add_action(@endprocess_action)
+    menu.add_separator
+    menu.add_action(@whois_action)
+    menu.add_action(@copy_action)
+
+    menu.exec(@tableview.viewport.map_to_global(position))
+  end
+
+  def _on_endprocess_action_triggered
+    return unless @lastindex.valid?
+
+    pid = @lastindex.sibling_at_column(COLUMN_PROCESS_ID).data.value
+    QProcess.execute("kill", QStringList.new << "-9" << pid)
+  end
+
+  def _on_whois_action_triggered
+    return unless @lastindex.valid?
+
+    remote_address = @lastindex.sibling_at_column(COLUMN_REMOTE_ADDRESS).data.value
+    url = QUrl.new("https://ipinfo.io/#{remote_address}")
+    QDesktopServices.open_url(url)
+  end
+
+  def _on_copy_action_triggered
+    return unless @lastindex.valid?
+
+    text = @lastindex.data.value
+    QApplication.clipboard.set_text(text)
+  end
+end

data/lib/netmon-qt/app/components.rb

@@ -0,0 +1 @@
+require_relative "components/connstableview"

data/lib/netmon-qt/app/services/conns_lookup_service.rb

@@ -0,0 +1,46 @@
+class ConnsLookupService
+  def perform
+    conns = {}
+    ino2conn = {}
+
+    %w[tcp tcp6 udp udp6].each do |filename|
+      str = File.read("/proc/net/#{filename}")
+      sockets = Netmon::ProcNet.sockets_form_str(str)
+      sockets.each do |socket|
+        conn = Netmon::Connection.new(socket)
+        conn.protocol = conn_protocol(filename)
+        conns[conn.key] = conn
+        ino2conn[conn.inode] = conn
+      end
+    end
+
+    Dir.glob("/proc/[1-9]*/fd/[1-9]*") do |dir|
+      ino = File.stat(dir).ino
+      conn = ino2conn[ino]
+      next if conn.nil?
+
+      conn.pid = dir.split("/")[2].to_i
+      conn.comm = File.read("/proc/#{conn.pid}/comm").strip
+    rescue Errno::EACCES, Errno::ENOENT
+    end
+
+    conns
+  end
+
+  private
+
+  def conn_protocol(str)
+    case str
+    when "tcp"
+      Netmon::Connection::PROTOCOL_TCP4
+    when "tcp6"
+      Netmon::Connection::PROTOCOL_TCP6
+    when "udp"
+      Netmon::Connection::PROTOCOL_UDP4
+    when "udp6"
+      Netmon::Connection::PROTOCOL_UDP6
+    else
+      ""
+    end
+  end
+end

data/lib/netmon-qt/app/services.rb

@@ -0,0 +1 @@
+require_relative "services/conns_lookup_service"

data/lib/netmon-qt/app/views/mainwindow.rb

@@ -0,0 +1,40 @@
+class MainWindow < RubyQt6::Bando::QMainWindow
+  q_object do
+  end
+
+  def initialize
+    super
+
+    initialize_central_widget
+    initialize_statusbar
+
+    set_context_menu_policy(Qt::NoContextMenu)
+    NetmonQt.settings.GET_mainwindow_geometry_and_restore_to(self)
+  end
+
+  def close_event(evt)
+    NetmonQt.settings.PUT_mainwindow_geometry(self)
+
+    _close_event(evt)
+  end
+
+  private
+
+  def initialize_central_widget
+    @connstableview = ConnsTableView.new
+
+    centralwidget = QWidget.new
+    mainlayout = QHBoxLayout.new(centralwidget)
+    mainlayout.add_widget(@connstableview)
+
+    set_central_widget(centralwidget)
+  end
+
+  def initialize_statusbar
+    statusbar = QStatusBar.new
+    statusbar.set_contents_margins(12, 0, 12, 0)
+    statusbar.add_permanent_widget(@connstableview.statusbar, 1)
+
+    set_status_bar(statusbar)
+  end
+end

data/lib/netmon-qt/app/views.rb

@@ -0,0 +1 @@
+require_relative "views/mainwindow"

data/lib/netmon-qt/app.rb

@@ -0,0 +1,26 @@
+require_relative "config"
+require_relative "lib"
+
+require_relative "app/components"
+require_relative "app/services"
+require_relative "app/views"
+
+module NetmonQt
+  class Application
+    def self.run
+      new.exec
+    end
+
+    def initialize
+      @app = QApplication.new
+      @mainwindow = MainWindow.new
+    end
+
+    def exec
+      Contrib::SigHandler.new(@app)
+
+      @mainwindow.show
+      @app.exec
+    end
+  end
+end

data/lib/netmon-qt/config/initializers/qt6.rb

@@ -0,0 +1,8 @@
+QApplication.set_organization_name("souk4711")
+QApplication.set_application_name("NetmonQt")
+
+root_path = File.expand_path("../..", __dir__)
+QDir.add_search_path("assets", File.join(root_path, "app/assets"))
+
+icon = QIcon.new("assets:/app.svg")
+QApplication.set_window_icon(icon)

data/lib/netmon-qt/config/settings.rb

@@ -0,0 +1,33 @@
+module NetmonQt
+  class Settings
+    def GET_mainwindow_geometry_and_restore_to(mainwindow)
+      settings = QSettings.new
+      mainwindow.restore_geometry(settings.value("MainWindow/geometry", QByteArray.new("")))
+      mainwindow.restore_state(settings.value("MainWindow/windowState", QByteArray.new("")))
+    end
+
+    def PUT_mainwindow_geometry(mainwindow)
+      settings = QSettings.new
+      settings.set_value("MainWindow/geometry", mainwindow.save_geometry)
+      settings.set_value("MainWindow/windowState", mainwindow.save_state)
+    end
+
+    def GET_connections_check_interval
+      default_connections_check_interval
+    end
+
+    def GET_geolite2_mmdb
+      default_geolite2_mmdb
+    end
+
+    private
+
+    def default_connections_check_interval
+      4_000
+    end
+
+    def default_geolite2_mmdb
+      "/usr/share/GeoIP/GeoLite2-Country.mmdb"
+    end
+  end
+end

data/lib/netmon-qt/config.rb

@@ -0,0 +1,8 @@
+require_relative "config/initializers/qt6"
+require_relative "config/settings"
+
+module NetmonQt
+  def self.settings
+    @settings ||= Settings.new
+  end
+end

data/lib/netmon-qt/lib/contrib/sighandler.rb

@@ -0,0 +1,24 @@
+module Contrib
+  class SigHandler < RubyQt6::Bando::QObject
+    q_object do
+      slot "_noop()"
+    end
+
+    def initialize(parent)
+      super
+
+      ["INT", "TERM"].each do |sig|
+        Signal.trap(sig) { |_| QApplication.quit }
+      end
+
+      timer = QTimer.new(self)
+      timer.timeout.connect(self, :_noop)
+      timer.start(1_000)
+    end
+
+    private
+
+    def _noop
+    end
+  end
+end

data/lib/netmon-qt/lib/contrib.rb

@@ -0,0 +1 @@
+require_relative "contrib/sighandler"

data/lib/netmon-qt/lib/netmon/connection.rb

@@ -0,0 +1,57 @@
+module Netmon
+  class Connection
+    # rubocop:disable Layout
+    PROTOCOL_TCP  = "TCP"
+    PROTOCOL_UDP  = "UDP"
+    PROTOCOL_TCP4 = "TCP v4"
+    PROTOCOL_TCP6 = "TCP v6"
+    PROTOCOL_UDP4 = "UDP v4"
+    PROTOCOL_UDP6 = "UDP v6"
+
+    STATE_ESTABLISHED = "ESTABLISHED"
+    STATE_LISTEN      = "LISTEN"
+    STATE_SYN_SENT    = "SYN_SENT"
+    STATE_SYN_RECV    = "SYN_RECV"
+    STATE_FIN_WAIT1   = "FIN_WAIT1"
+    STATE_FIN_WAIT2   = "FIN_WAIT2"
+    STATE_CLOSING     = "CLOSING"
+    STATE_CLOSE_WAIT  = "CLOSE_WAIT"
+    STATE_LAST_ACK    = "LAST_ACK"
+    STATE_TIME_WAIT   = "TIME_WAIT"
+    STATE_CLOSE       = "CLOSE"
+
+    COMM_DEFUNCT = "<defunct>"
+    COMM_UNKNOWN = "<unknown>"
+    # rubocop:enable Layout
+
+    attr_accessor :protocol
+    attr_accessor :local_address, :local_port
+    attr_accessor :remote_address, :remote_port
+    attr_accessor :state, :uid, :inode
+    attr_accessor :pid, :comm
+
+    def initialize(attrs)
+      attrs.each do |attr, value|
+        public_send("#{attr}=", value)
+      end
+    end
+
+    def key
+      "#{protocol}: #{local_address}:#{local_port} <=> #{remote_address}:#{remote_port}"
+    end
+
+    def uname
+      Etc.getpwuid(uid).name
+    end
+
+    def pid_text
+      return pid.to_s if pid
+      inode.zero? ? COMM_DEFUNCT : COMM_UNKNOWN
+    end
+
+    def comm_text
+      return comm if comm
+      inode.zero? ? COMM_DEFUNCT : COMM_UNKNOWN
+    end
+  end
+end

data/lib/netmon-qt/lib/netmon/localaddresscheck.rb

@@ -0,0 +1,19 @@
+require "ipaddr"
+
+module Netmon
+  CIDR_LIST = [
+    IPAddr.new("224.0.0.0/24"), # Local Multicast - IPv4
+    IPAddr.new("ff02::/16"),    # Local Multicast - IPv6
+    IPAddr.new("0.0.0.0"),      # Unspecified     - IPv4
+    IPAddr.new("::")            # Unspecified     - IPv6
+  ].freeze
+
+  def self.local_address?(address)
+    ip = IPAddr.new(address)
+    return true if ip.link_local?
+    return true if ip.loopback?
+    return true if ip.private?
+    return true if CIDR_LIST.any? { |cidr| cidr.include?(address) }
+    false
+  end
+end

data/lib/netmon-qt/lib/netmon/procnet.rb

@@ -0,0 +1,49 @@
+module Netmon
+  module ProcNet
+    TCP_STATE = {
+      "01" => Connection::STATE_ESTABLISHED,
+      "02" => Connection::STATE_SYN_SENT,
+      "03" => Connection::STATE_SYN_RECV,
+      "04" => Connection::STATE_FIN_WAIT1,
+      "05" => Connection::STATE_FIN_WAIT2,
+      "06" => Connection::STATE_TIME_WAIT,
+      "07" => Connection::STATE_CLOSE,
+      "08" => Connection::STATE_CLOSE_WAIT,
+      "09" => Connection::STATE_LAST_ACK,
+      "0A" => Connection::STATE_LISTEN,
+      "0B" => Connection::STATE_CLOSING
+    }.freeze
+
+    def self.sockets_form_str(str)
+      str.lines[1..].map do |line|
+        linesplit = line.split
+        local_address, local_port = linesplit[1].split(":")
+        remote_address, remote_port = linesplit[2].split(":")
+
+        {
+          local_address: format_address(local_address),
+          local_port: format_port(local_port),
+          remote_address: format_address(remote_address),
+          remote_port: format_port(remote_port),
+          state: TCP_STATE[linesplit[3]],
+          uid: linesplit[7].to_i,
+          inode: linesplit[9].to_i
+        }
+      end
+    end
+
+    def self.format_address(str)
+      case str.size
+      when 8 then str.scan(/.{2}/).reverse.map { |s| s.to_i(16) }.join(".")
+      when 32 then str.scan(/.{8}/).map { |s| s.scan(/.{2}/).reverse }.join.scan(/.{4}/).join(":")
+      else raise "Invalid address str: #{str}"
+      end
+    end
+    private_class_method :format_address
+
+    def self.format_port(str)
+      str.to_i(16)
+    end
+    private_class_method :format_port
+  end
+end

data/lib/netmon-qt/lib/netmon.rb

@@ -0,0 +1,3 @@
+require_relative "netmon/connection"
+require_relative "netmon/localaddresscheck"
+require_relative "netmon/procnet"

data/lib/netmon-qt/lib.rb

@@ -0,0 +1,2 @@
+require_relative "lib/contrib"
+require_relative "lib/netmon"

data/lib/netmon-qt/version.rb

@@ -0,0 +1,3 @@
+module NetmonQt
+  VERSION = "1.0.0"
+end

data/lib/netmon-qt.rb

@@ -0,0 +1,7 @@
+require "etc"
+
+require "maxmind/db"
+require "qt6/qtwidgets"
+
+require_relative "netmon-qt/app"
+require_relative "netmon-qt/version"