1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/webauthn/authenticator_request_scheduler.h"
#include "base/memory/weak_ptr.h"
#include "base/supports_user_data.h"
#include "chrome/browser/webauthn/chrome_authenticator_request_delegate.h"
#include "content/public/browser/web_contents.h"
namespace {
// Holds a weak pointer to the active request in a WebContents, if any.
class ActiveRequestWeakHolder : public base::SupportsUserData::Data {
public:
ActiveRequestWeakHolder() = default;
ActiveRequestWeakHolder(const ActiveRequestWeakHolder&) = delete;
ActiveRequestWeakHolder& operator=(const ActiveRequestWeakHolder&) = delete;
~ActiveRequestWeakHolder() override = default;
static ActiveRequestWeakHolder* EnsureForWebContents(
content::WebContents* web_contents) {
static constexpr char kActiveRequestDataKey[] =
"ActiveAuthenticatorRequestKey";
if (!web_contents->GetUserData(kActiveRequestDataKey)) {
web_contents->SetUserData(kActiveRequestDataKey,
std::make_unique<ActiveRequestWeakHolder>());
}
return static_cast<ActiveRequestWeakHolder*>(
web_contents->GetUserData(kActiveRequestDataKey));
}
base::WeakPtr<ChromeAuthenticatorRequestDelegate>& request() {
return request_;
}
private:
base::WeakPtr<ChromeAuthenticatorRequestDelegate> request_;
};
} // namespace
// static
std::unique_ptr<ChromeAuthenticatorRequestDelegate>
AuthenticatorRequestScheduler::CreateRequestDelegate(
content::RenderFrameHost* render_frame_host) {
// RenderFrameHosts which are not exposed to the user can't create
// authenticator request delegate.
if (!render_frame_host->IsActive())
return nullptr;
auto* const web_contents =
content::WebContents::FromRenderFrameHost(render_frame_host);
auto* const active_request_holder =
ActiveRequestWeakHolder::EnsureForWebContents(web_contents);
if (active_request_holder->request())
return nullptr;
auto request =
std::make_unique<ChromeAuthenticatorRequestDelegate>(render_frame_host);
active_request_holder->request() = request->AsWeakPtr();
return request;
}
// static
ChromeAuthenticatorRequestDelegate*
AuthenticatorRequestScheduler::GetRequestDelegate(
content::WebContents* web_contents) {
return ActiveRequestWeakHolder::EnsureForWebContents(web_contents)
->request()
.get();
}
|