src/interceptors/UrlRequestInterceptor.cpp

   1 /*
   2  * Copyright © 2022 Soren Stoutner <soren@stoutner.com>.
   3  *
   4  * This file is part of Privacy Browser PC <https://www.stoutner.com/privacy-browser-pc>.
   5  *
   6  * Privacy Browser PC is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Privacy Browser PC is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Privacy Browser PC.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 // Application headers.
  21 #include "UrlRequestInterceptor.h"
  22
  23 // KDE Framework headers.
  24 #include <KLocalizedString>
  25
  26 // Qt framework headers.
  27 #include <QMessageBox>
  28
  29 // Construct the class.
  30 UrlRequestInterceptor::UrlRequestInterceptor(QObject *parentObjectPointer) : QWebEngineUrlRequestInterceptor(parentObjectPointer) {}
  31
  32 void UrlRequestInterceptor::interceptRequest(QWebEngineUrlRequestInfo &urlRequestInfo)
  33 {
  34     // Handle the request according to the navigation type.
  35     switch (urlRequestInfo.navigationType())
  36     {
  37         case QWebEngineUrlRequestInfo::NavigationTypeLink:
  38         case QWebEngineUrlRequestInfo::NavigationTypeTyped:
  39         case QWebEngineUrlRequestInfo::NavigationTypeBackForward:
  40         case QWebEngineUrlRequestInfo::NavigationTypeRedirect:
  41         {
  42             // Only check the hosts if the main URL is changing.
  43             if (urlRequestInfo.resourceType() == QWebEngineUrlRequestInfo::ResourceTypeMainFrame)
  44             {
  45                 // Get the hosts.
  46                 QString requestingHost = urlRequestInfo.initiator().host();
  47                 QString requestedHost = urlRequestInfo.requestUrl().host();
  48
  49                 // Reapply the domain settings if the host is changing.
  50                 if (requestingHost != requestedHost)
  51                     emit applyDomainSettings(requestedHost);
  52             }
  53
  54             break;
  55         }
  56
  57         default:
  58             // Do nothing.
  59             break;
  60     }
  61
  62     // Handle the request according to the resource type.
  63     switch (urlRequestInfo.resourceType())
  64     {
  65         // A naughty HTTP ping request.
  66         case QWebEngineUrlRequestInfo::ResourceTypePing:
  67         {
  68             // Block HTTP ping requests.
  69             urlRequestInfo.block(true);
  70
  71             // Instantiate an HTTP ping blocked message box.
  72             QMessageBox httpPingBlockedMessageBox;
  73
  74             // Set the icon.
  75             httpPingBlockedMessageBox.setIcon(QMessageBox::Information);
  76
  77             // Set the window title.
  78             httpPingBlockedMessageBox.setWindowTitle(i18nc("HTTP Ping blocked dialog title", "HTTP Ping Blocked"));
  79
  80             // Set the text.
  81             httpPingBlockedMessageBox.setText(i18nc("HTTP Ping blocked dialog text", "This request has been blocked because it sends a naughty HTTP ping to %1.",
  82                                                     urlRequestInfo.requestUrl().toString()));
  83
  84             // Set the standard button.
  85             httpPingBlockedMessageBox.setStandardButtons(QMessageBox::Ok);
  86
  87             // Display the message box.
  88             httpPingBlockedMessageBox.exec();
  89
  90             break;
  91         }
  92
  93         default:
  94         {
  95             // Do nothing.
  96             break;
  97         }
  98     }
  99 }