blob: 67a39a7b23804b8c6fad0170b51cf57caee6fff6 [file] [log] [blame]
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/in_session_auth/webauthn_request_registrar_impl.h"
#include <string>
#include "ash/shell.h"
#include "ash/wm/mru_window_tracker.h"
#include "ash/wm/window_properties.h"
#include "base/bind.h"
#include "base/strings/string_number_conversions.h"
#include "ui/aura/window.h"
#include "ui/base/class_property.h"
namespace ash {
namespace {
uint32_t g_current_request_id = 0;
} // namespace
WebAuthnRequestRegistrarImpl::WebAuthnRequestRegistrarImpl() = default;
WebAuthnRequestRegistrarImpl::~WebAuthnRequestRegistrarImpl() = default;
WebAuthnRequestRegistrarImpl::GenerateRequestIdCallback
WebAuthnRequestRegistrarImpl::GetRegisterCallback(aura::Window* window) {
// If the window is nullptr, e.g. IsUVPAA() is called shortly before or after
// navigation, the render_frame_host may not be initialized properly, we
// return a dumb callback. In the worst case, if it's MakeCredential or
// GetAssertion, the Chrome OS auth dialog will not show up and the operation
// fails gracefully.
if (!window) {
return base::BindRepeating([] { return std::string(); });
}
window_tracker_.Add(window);
// base::Unretained() is safe here because WebAuthnRequestRegistrarImpl
// has the same lifetime as Shell and is released at
// PostMainMessageLoopRun stage. The callback should not be invoked
// after main message loop tearing down.
return base::BindRepeating(&WebAuthnRequestRegistrarImpl::DoRegister,
base::Unretained(this), window);
}
std::string WebAuthnRequestRegistrarImpl::DoRegister(aura::Window* window) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
g_current_request_id++;
std::string request_id = base::NumberToString(g_current_request_id);
// If |window| is still valid, associate it with the new request id.
// If |window| is gone, the incremented id will fail the request later,
// which is ok.
if (window_tracker_.Contains(window)) {
window->SetProperty(kWebAuthnRequestId, new std::string(request_id));
}
return request_id;
}
aura::Window* WebAuthnRequestRegistrarImpl::GetWindowForRequestId(
std::string request_id) {
if (request_id.empty()) {
return nullptr;
}
MruWindowTracker::WindowList windows =
Shell::Get()->mru_window_tracker()->BuildMruWindowList(kAllDesks);
for (aura::Window* window : windows) {
std::string* window_request_id = window->GetProperty(kWebAuthnRequestId);
if (window_request_id && *window_request_id == request_id) {
return window;
}
}
return nullptr;
}
} // namespace ash