ios/chrome/browser/ssl/ios_ssl_error_handler.h

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_CHROME_BROWSER_SSL_IOS_SSL_ERROR_HANDLER_H_
#define IOS_CHROME_BROWSER_SSL_IOS_SSL_ERROR_HANDLER_H_

#include "base/callback_forward.h"
#include "base/memory/weak_ptr.h"
#include "base/timer/timer.h"
#include "components/captive_portal/captive_portal_detector.h"
#include "components/captive_portal/captive_portal_types.h"
#import "ios/web/public/web_state/web_state_user_data.h"
#include "url/gurl.h"

namespace net {
class SSLInfo;
}  // namespace net

namespace web {
class WebState;
}  // namespace web

// This class is responsible for deciding what type of interstitial to show for
// an SSL validation error.
class IOSSSLErrorHandler : public web::WebStateUserData<IOSSSLErrorHandler> {
 public:
  // Entry point for the class.
  static void HandleSSLError(web::WebState* web_state,
                             int cert_error,
                             const net::SSLInfo& info,
                             const GURL& request_url,
                             bool overridable,
                             const base::Callback<void(bool)>& callback);
  ~IOSSSLErrorHandler() override;

 private:
  // Creates an error handler for the given |web_state| and |request_url|.
  // The |cert_error| and SSL |info| represent the SSL error detected which
  // triggered the display of the SSL interstitial. If |overridable| is true,
  // the interstitial will allow the error to be ignored in order to proceed to
  // |request_url|. |callback| will be called after the user is done interacting
  // with this interstitial.
  IOSSSLErrorHandler(web::WebState* web_state,
                     int cert_error,
                     const net::SSLInfo& info,
                     const GURL& request_url,
                     bool overridable,
                     const base::Callback<void(bool)>& callback);

  // Begins captive portal detection to determine which interstitial should be
  // displayed.
  void StartHandlingError();
  // Presents the appropriate interstitial based on the |results| of the captive
  // portal detection.
  void HandleCaptivePortalDetectionResult(
      const captive_portal::CaptivePortalDetector::Results& results);
  // Displays an SSL error page interstitial.
  void ShowSSLInterstitial();
  // Displays a Captive Portal interstitial. The |landing_url| is the web page
  // which allows the user to complete their connection to the network.
  void ShowCaptivePortalInterstitial(const GURL& landing_url);
  // Detects the current Captive Portal state and records the result with
  // |LogCaptivePortalResult|.
  static void RecordCaptivePortalState(web::WebState* web_state);
  // Records a metric to classify if SSL errors are due to a Captive Portal
  // state.
  static void LogCaptivePortalResult(
      captive_portal::CaptivePortalResult result);
  // Called on interstitial dismissal.
  static void InterstitialWasDismissed(
      web::WebState* web_state,
      const base::Callback<void(bool)>& callback,
      bool proceed);

  // The WebState associated with this error handler.
  web::WebState* const web_state_ = nullptr;
  // The ssl certificate error.
  const int cert_error_ = 0;
  // The ssl certificate details.
  const net::SSLInfo ssl_info_;
  // The request the user was loading which triggered the certificate error.
  const GURL request_url_;
  // Whether or not the user can ignore this error in order to continue loading
  // |request_url_|.
  const bool overridable_ = false;
  // The callback to run after the user is done interacting with this
  // interstitial. |proceed| will be true if the user wants to procced with the
  // page load of |request_url_|, false otherwise.
  base::Callback<void(bool proceed)> callback_;
  // A timer to display the SSL interstitial if the captive portal detection
  // takes too long.
  base::OneShotTimer timer_;

  base::WeakPtrFactory<IOSSSLErrorHandler> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(IOSSSLErrorHandler);
};

#endif  // IOS_CHROME_BROWSER_SSL_IOS_SSL_ERROR_HANDLER_H_