WebProbe Class Reference

#include <WebProbe.hpp>

Public Types
using	NotifyCb = std::function<void(const std::string&, const std::string&, bool)>

Public Member Functions
	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 Member Functions
std::string	extract_host (const std::string &raw)
std::string	build_local_warning_redirect (const std::string &url) const

Private Attributes
int	warning_port_
std::shared_mutex	mutex_
NotifyCb	notify_cb_

Detailed Description

Definition at line 26 of file WebProbe.hpp.

Member Typedef Documentation

NotifyCb

using WebProbe::NotifyCb = std::function<void(const std::string&, const std::string&, bool)>

Definition at line 28 of file WebProbe.hpp.

Constructor & Destructor Documentation

WebProbe()

WebProbe::WebProbe ( const std::string & blocklist_path_hint = "", int warning_port = 81 )

explicit

Definition at line 139 of file WebProbe.cpp.

    : warning_port_(warning_port)
{
    load_blocklist(blocklist_path_hint);
}

References load_blocklist(), and warning_port_.

~WebProbe()

WebProbe::~WebProbe ( )

default

References build_local_warning_redirect(), check_url(), extract_host(), register_callback(), and reload_blocklist().

Member Function Documentation

build_local_warning_redirect()

std::string WebProbe::build_local_warning_redirect ( const std::string & url ) const

private

Definition at line 178 of file WebProbe.cpp.

                                                                           {
    std::string enc;
    for (unsigned char c : url)
        enc += (c == ' ') ? "%20" : std::string(1, c);
    return "http://127.0.0.2:" + std::to_string(warning_port_) + "/?target=" + enc;
}

References warning_port_.

Referenced by check_url(), and ~WebProbe().

check_url()

std::string WebProbe::check_url

(

const std::string &

url

)

Definition at line 185 of file WebProbe.cpp.

                                                  {
    if (url.empty()) return url;
    std::string host = extract_host(url);
    if (host.empty()) return url;
 
    auto now = now_ms();
    {
        std::shared_lock<std::shared_mutex> lk(g_hot_mtx);
        auto it = g_hot.find(host);
        if (it != g_hot.end() && now < it->second.second)
            return it->second.first ? build_local_warning_redirect(url) : url;
    }
 
 
    if (Backend::instance().isBankDomain(host)) {
        std::unique_lock<std::shared_mutex> lk(g_hot_mtx);
        g_hot[host] = {false, now + TTL_MS};
        return url;
    }
 
    if (is_blocked_host(host)) {
        std::cout << "[WebProbe] 🔴 Bloccato: " << host << "\n";
        std::unique_lock<std::shared_mutex> lk(g_hot_mtx);
        g_hot[host] = {true, now + TTL_MS};
        return build_local_warning_redirect(url);
    }
 
    std::string cached_ip = AsyncDNSResolver::instance().resolve_cached(host);
    if (cached_ip == "127.0.0.2") {
        std::cout << _("[WebProbe] ⚠️ DNS ha restituito 127.0.0.2 per ") << host << "\n";
        std::unique_lock<std::shared_mutex> lk(g_hot_mtx);
        g_hot[host] = {true, now + TTL_MS};
        return build_local_warning_redirect(url);
    }
 
 
    std::unique_lock<std::shared_mutex> lk(g_hot_mtx);
    g_hot[host] = {false, now + TTL_MS};
    return url;
}

References _, build_local_warning_redirect(), extract_host(), g_hot, g_hot_mtx, AsyncDNSResolver::instance(), Backend::instance(), is_blocked_host(), now_ms(), AsyncDNSResolver::resolve_cached(), and TTL_MS.

Referenced by ~WebProbe().

extract_host()

std::string WebProbe::extract_host ( const std::string & raw )

private

Definition at line 147 of file WebProbe.cpp.

                                                     {
    if (raw.empty()) return {};
    std::string s = raw;
    auto notspace = [](int ch){ return !std::isspace(ch); };
    s.erase(s.begin(), std::find_if(s.begin(), s.end(), notspace));
    s.erase(std::find_if(s.rbegin(), s.rend(), notspace).base(), s.end());
    if (s.empty()) return {};
 
    const char* schemes[] = {"http://","https://","ftp://"};
    for (auto sch : schemes) {
        size_t n = std::strlen(sch);
        if (s.rfind(sch, 0) == 0) { s.erase(0, n); break; }
    }
    auto at = s.find('@');
    if (at != std::string::npos) s.erase(0, at + 1);
    size_t slash = s.find_first_of("/?#");
    if (slash != std::string::npos) s = s.substr(0, slash);
    size_t colon = s.rfind(':');
    if (colon != std::string::npos) {
        std::string after = s.substr(colon + 1);
        if (std::all_of(after.begin(), after.end(), ::isdigit))
            s = s.substr(0, colon);
    }
    if (!s.empty() && s.back() == '.') s.pop_back();
 
    std::string ascii = to_ascii_idna(s);
    std::transform(ascii.begin(), ascii.end(), ascii.begin(), ::tolower);
    return ascii;
}

References to_ascii_idna().

Referenced by check_url(), and ~WebProbe().

register_callback()

void WebProbe::register_callback

(

NotifyCb

cb

)

Definition at line 227 of file WebProbe.cpp.

                                            {
    std::unique_lock<std::shared_mutex> lock(mutex_);
    notify_cb_ = std::move(cb);
}

References mutex_, and notify_cb_.

Referenced by ~WebProbe().

reload_blocklist()

void WebProbe::reload_blocklist

(

const std::string &

path_hint

)

Definition at line 232 of file WebProbe.cpp.

                                                     {
    load_blocklist(path);
}

References load_blocklist().

Referenced by ~WebProbe().

Member Data Documentation

mutex_

std::shared_mutex WebProbe::mutex_

mutableprivate

Definition at line 42 of file WebProbe.hpp.

Referenced by register_callback().

notify_cb_

NotifyCb WebProbe::notify_cb_

private

Definition at line 43 of file WebProbe.hpp.

Referenced by register_callback().

warning_port_

int WebProbe::warning_port_

private

Definition at line 41 of file WebProbe.hpp.

Referenced by build_local_warning_redirect(), and WebProbe().

---

The documentation for this class was generated from the following files:

• BastionGuard/src/phishing/WebProbe.hpp
• BastionGuard/src/phishing/WebProbe.cpp