1 files changed, 96 insertions, 0 deletions
diff --git a/chromium/chrome/browser/signin/reauth_tab_helper.cc b/chromium/chrome/browser/signin/reauth_tab_helper.cc
new file mode 100644
index 00000000000..e8bd1d8b1c8
--- /dev/null
+++ b/chromium/chrome/browser/signin/reauth_tab_helper.cc
@@ -0,0 +1,96 @@
+// Copyright 2020 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.
+
+#include "chrome/browser/signin/reauth_tab_helper.h"
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/signin/reauth_result.h"
+#include "content/public/browser/navigation_handle.h"
+#include "net/http/http_status_code.h"
+#include "url/origin.h"
+
+namespace signin {
+
+namespace {
+
+bool IsExpectedResponseCode(int response_code) {
+  return response_code == net::HTTP_OK || response_code == net::HTTP_NO_CONTENT;
+}
+
+}  // namespace
+
+// static
+void ReauthTabHelper::CreateForWebContents(content::WebContents* web_contents,
+                                           const GURL& reauth_url,
+                                           ReauthCallback callback) {
+  DCHECK(web_contents);
+  if (!FromWebContents(web_contents)) {
+    web_contents->SetUserData(
+        UserDataKey(), base::WrapUnique(new ReauthTabHelper(
+                           web_contents, reauth_url, std::move(callback))));
+  } else {
+    std::move(callback).Run(signin::ReauthResult::kCancelled);
+  }
+}
+
+ReauthTabHelper::~ReauthTabHelper() = default;
+
+void ReauthTabHelper::CompleteReauth(signin::ReauthResult result) {
+  if (callback_)
+    std::move(callback_).Run(result);
+}
+
+void ReauthTabHelper::DidFinishNavigation(
+    content::NavigationHandle* navigation_handle) {
+  if (!navigation_handle->IsInPrimaryMainFrame())
+    return;
+
+  is_within_reauth_origin_ &=
+      url::IsSameOriginWith(reauth_url_, navigation_handle->GetURL());
+
+  if (navigation_handle->IsErrorPage()) {
+    has_last_committed_error_page_ = true;
+    return;
+  }
+
+  has_last_committed_error_page_ = false;
+
+  GURL::Replacements replacements;
+  replacements.ClearQuery();
+  GURL url_without_query =
+      navigation_handle->GetURL().ReplaceComponents(replacements);
+  if (url_without_query != reauth_url_)
+    return;
+
+  if (!navigation_handle->GetResponseHeaders() ||
+      !IsExpectedResponseCode(
+          navigation_handle->GetResponseHeaders()->response_code())) {
+    CompleteReauth(signin::ReauthResult::kUnexpectedResponse);
+  }
+
+  CompleteReauth(signin::ReauthResult::kSuccess);
+}
+
+void ReauthTabHelper::WebContentsDestroyed() {
+  CompleteReauth(signin::ReauthResult::kDismissedByUser);
+}
+
+bool ReauthTabHelper::is_within_reauth_origin() {
+  return is_within_reauth_origin_;
+}
+
+bool ReauthTabHelper::has_last_committed_error_page() {
+  return has_last_committed_error_page_;
+}
+
+ReauthTabHelper::ReauthTabHelper(content::WebContents* web_contents,
+                                 const GURL& reauth_url,
+                                 ReauthCallback callback)
+    : content::WebContentsUserData<ReauthTabHelper>(*web_contents),
+      content::WebContentsObserver(web_contents),
+      reauth_url_(reauth_url),
+      callback_(std::move(callback)) {}
+
+WEB_CONTENTS_USER_DATA_KEY_IMPL(ReauthTabHelper);
+
+}  // namespace signin