WebProbe.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 <string>
#include <functional>
#include <shared_mutex>
 
class WebProbe {
public:
    using NotifyCb = std::function<void(const std::string&, const std::string&, bool)>;
 
    explicit WebProbe(const std::string& blocklist_path_hint = "", int warning_port = 81);
    ~WebProbe();
 
    std::string check_url(const std::string& url);
    void reload_blocklist(const std::string& path_hint);
    void register_callback(NotifyCb cb);
 
private:
    std::string extract_host(const std::string& raw);
    std::string build_local_warning_redirect(const std::string& url) const;
 
    int warning_port_;
    mutable std::shared_mutex mutex_;
    NotifyCb notify_cb_;
};