USBScanPage.hpp

Go to the documentation of this file.

/*
 * BastionGuard™
 * Copyright (C) 2025–2026 Calogero Scarnà
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * BastionGuard™ is a trademark of Calogero Scarnà.
 * The BastionGuard™ name and branding are not licensed under the GPL.
 */
 
#pragma once
 
#include <gtkmm.h>
#include <set>
#include <queue>
#include <thread>
#include <atomic>
#include <filesystem>
#include <unordered_map>
#include <mutex>
 
#include "../LifetimeGuard.hpp"
 
 
struct USBDeviceInfo {
    std::string name;
    std::string devnode;
    std::string serial;
    std::string uid;
    uint64_t size_bytes = 0;
};
 
 
enum class ScanState {
    Idle,
    Mounting,
    Scanning
};
 
 
class USBScanPage : public Gtk::Box {
public:
    USBScanPage();
    virtual ~USBScanPage();
 
private:
    std::vector<USBDeviceInfo> detect_usb_devices();
    void update_cards();
    Gtk::Widget* build_usb_card(const USBDeviceInfo& dev);
 
    std::string find_mountpoint(const std::string& devnode);
    std::string mount_partition(const std::string& devnode, bool& auto_mounted);
 
    void run_clamdscan_async(const std::string& devnode);
    bool try_start_scan(const std::string& uid);
    void show_scan_result(const std::string& text);
    void show_infected_dialog(const std::string& text);
    std::string build_scan_report(const std::string& raw);
 
    void write_log(const std::string& msg);
    void update_log_view();
    std::string get_log_path();
 
    void start_monitor_thread();
 
    void append_live_output(const std::string& line);
 
    void send_dbus_alert(const std::string& file, const std::string& family);
 
    Gtk::FlowBox flowbox_devices_;
    Gtk::Button btn_refresh_;
    Gtk::CheckButton chk_auto_refresh_;
    Gtk::Label lbl_title_;
    Gtk::Label lbl_status_;
    Gtk::TextView txt_log_;
    Gtk::ScrolledWindow scroller_log_;
 
    std::thread monitor_thread_;
    std::atomic<bool> monitor_running_{false};
    bool initial_scan_done_{false};
    bool ignore_existing_devices_{true};
 
    std::mutex scan_mtx_;
    std::unordered_map<std::string, ScanState> device_state_;
 
    std::mutex log_mtx_;
    std::mutex pending_log_mtx_;
    std::vector<std::string> pending_log_lines_;
    sigc::connection log_idle_connection_;
    sigc::connection auto_refresh_conn_;
 
    class LiveScanDialog* live_scan_dialog_ = nullptr;
    Gtk::TextView* live_scan_textview_ = nullptr;
 
    static std::string human_readable_size(uint64_t bytes);
 
    BastionGuard::AliveFlag alive_ = BastionGuard::make_alive_flag();
};