blob: fcdd4db41ff3184a67e7a00bf858f0e5419a215f [file] [log] [blame]
// Copyright 2018 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 <set>
#include <string>
#include "base/command_line.h"
#include "base/strings/string16.h"
namespace chrome_cleaner {
// Registers the application to execute again after a reboot.
class PostRebootRegistration {
explicit PostRebootRegistration(const base::string16& product_shortname);
// Register the current running app to be called post reboot with the provided
// command line switches. Returns false if failed.
bool RegisterRunOnceOnRestart(const std::string& cleanup_id,
const base::CommandLine& switches);
// Unregisters the current running app from RunOnce, to prevent running it
// again.
void UnregisterRunOnceOnRestart();
// Returns the value stored by |RegisterRunOnceOnRestart|, or an empty string
// if there is none.
base::string16 RunOnceOnRestartRegisteredValue();
// Fills |out_command_line| with the switches of the post-reboot run by
// reading from a cleanup-id dependent registry key. The registry key is then
// deleted. Returns false if reading or deleting the registry key failed.
bool ReadRunOncePostRebootCommandLine(const std::string& cleanup_id,
base::CommandLine* out_command_line);
// Test functions should have access to GetPostRebootSwitchKeyPath.
friend bool RunOnceOverrideCommandLineContains(const std::string& cleanup_id,
const wchar_t* sub_string);
// Returns the registry key path in which the full post-reboot command line
// switches can be found.
static base::string16 GetPostRebootSwitchKeyPath();
base::string16 product_shortname_;
} // namespace chrome_cleaner