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 // The default constructor.
  24 UrlRequestInterceptor::UrlRequestInterceptor(QObject *parentObjectPointer) : QWebEngineUrlRequestInterceptor(parentObjectPointer) {}
  25
  26 void UrlRequestInterceptor::interceptRequest(QWebEngineUrlRequestInfo &urlRequestInfo)
  27 {
  28     // Handle the request according to the navigation type.
  29     switch (urlRequestInfo.navigationType())
  30     {
  31         case QWebEngineUrlRequestInfo::NavigationTypeLink:
  32         case QWebEngineUrlRequestInfo::NavigationTypeTyped:
  33         case QWebEngineUrlRequestInfo::NavigationTypeBackForward:
  34         case QWebEngineUrlRequestInfo::NavigationTypeRedirect:
  35         {
  36             // Only check the hosts if the main URL is changing.
  37             if (urlRequestInfo.resourceType() == QWebEngineUrlRequestInfo::ResourceTypeMainFrame)
  38             {
  39                 // Get the hosts.
  40                 QString requestingHost = urlRequestInfo.initiator().host();
  41                 QString requestedHost = urlRequestInfo.requestUrl().host();
  42
  43                 // Reapply the domain settings if the host is changing.
  44                 if (requestingHost != requestedHost)
  45                 {
  46                     emit applyDomainSettings(requestedHost);
  47                 }
  48             }
  49
  50             break;
  51         }
  52
  53         default:
  54             // Do nothing.
  55             break;
  56     }
  57
  58     // Handle the request according to the resource type.
  59     switch (urlRequestInfo.resourceType())
  60     {
  61         // A naughty HTTP ping request.
  62         case QWebEngineUrlRequestInfo::ResourceTypePing:
  63         {
  64             // Block HTTP ping requests.
  65             urlRequestInfo.block(true);
  66
  67             break;
  68         }
  69
  70         default:
  71         {
  72             // Do nothing.
  73             break;
  74         }
  75     }
  76 }