Support profile level remote command job.
Create a new profile picker helper class. It can accept
`ProfileManager` for CBCM remote command. Or a `Profile` instance
for profile level remote command.
Note that one job can only be one of the two states above.
The helper class now is only used for `ClearBrowsingDataJob`.
Change-Id: I06f5298ce8c0779892822c4e745b68f7e0204b40
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4776180
Reviewed-by: Anthony Vallée-Dubois <[email protected]>
Commit-Queue: Owen Min <[email protected]>
Cr-Commit-Position: refs/heads/main@{#1184313}
diff --git a/chrome/browser/enterprise/remote_commands/job_profile_picker.cc b/chrome/browser/enterprise/remote_commands/job_profile_picker.cc
new file mode 100644
index 0000000..fceeb2d
--- /dev/null
+++ b/chrome/browser/enterprise/remote_commands/job_profile_picker.cc
@@ -0,0 +1,81 @@
+// Copyright 2023 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/enterprise/remote_commands/job_profile_picker.h"
+
+#include "base/values.h"
+#include "build/build_config.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_attributes_storage.h"
+#include "chrome/browser/profiles/profile_manager.h"
+
+namespace enterprise_commands {
+namespace {
+
+const char kProfilePathField[] = "profile_path";
+
+} // namespace
+
+JobProfilePicker::JobProfilePicker(Profile* profile)
+ : profile_or_profile_manager_(profile) {}
+JobProfilePicker::JobProfilePicker(ProfileManager* profile_manager)
+ : profile_or_profile_manager_(profile_manager) {}
+
+JobProfilePicker::~JobProfilePicker() = default;
+
+bool JobProfilePicker::ParseCommandPayload(
+ const base::Value::Dict& command_payload) {
+ if (absl::holds_alternative<raw_ptr<Profile>>(profile_or_profile_manager_)) {
+ return true;
+ }
+
+ const std::string* path = command_payload.FindString(kProfilePathField);
+ if (!path) {
+ return false;
+ }
+
+ // On Windows, file paths are wstring as opposed to string on other platforms.
+ // On POSIX platforms other than MacOS and ChromeOS, the encoding is unknown.
+ //
+ // This path is sent from the server, which obtained it from Chrome in a
+ // previous report, and Chrome casts the path as UTF8 using UTF8Unsafe before
+ // sending it (see BrowserReportGeneratorDesktop::GenerateProfileInfo).
+ // Because of that, the best thing we can do everywhere is try to get the
+ // path from UTF8, and ending up with an invalid path will fail later in
+ // RunImpl when we attempt to get the profile from the path.
+ profile_path_ = base::FilePath::FromUTF8Unsafe(*path);
+#if BUILDFLAG(IS_WIN)
+ // For Windows machines, the path that Chrome reports for the profile is
+ // "Normalized" to all lower-case on the reporting server. This means that
+ // when the server sends the command, the path will be all lower case and
+ // the profile manager won't be able to use it as a key. To avoid this issue,
+ // This code will iterate over all profile paths and find the one that matches
+ // in a case-insensitive comparison. If this doesn't find one, RunImpl will
+ // fail in the same manner as if the profile didn't exist, which is the
+ // expected behavior.
+ ProfileAttributesStorage& storage =
+ absl::get<raw_ptr<ProfileManager>>(profile_or_profile_manager_)
+ ->GetProfileAttributesStorage();
+ for (ProfileAttributesEntry* entry : storage.GetAllProfilesAttributes()) {
+ base::FilePath entry_path = entry->GetPath();
+
+ if (base::FilePath::CompareEqualIgnoreCase(profile_path_.value(),
+ entry_path.value())) {
+ profile_path_ = entry_path;
+ break;
+ }
+ }
+#endif
+ return true;
+}
+
+Profile* JobProfilePicker::GetProfile() {
+ if (absl::holds_alternative<raw_ptr<Profile>>(profile_or_profile_manager_)) {
+ return absl::get<raw_ptr<Profile>>(profile_or_profile_manager_);
+ }
+ return absl::get<raw_ptr<ProfileManager>>(profile_or_profile_manager_)
+ ->GetProfileByPath(profile_path_);
+}
+
+} // namespace enterprise_commands
