blob: 6272689a31de8b034dbcc33a4156e4f218ec4b74 [file] [log] [blame]
// Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_
#define UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_
#include "update_engine/p2p_manager.h"
#include "update_engine/test_utils.h"
#include "update_engine/utils.h"
#include <string>
#include <vector>
#include <glib.h>
#include <base/logging.h>
#include <base/strings/string_util.h>
#include <base/strings/stringprintf.h>
namespace chromeos_update_engine {
// Configuration for P2PManager for use in unit tests. Instead of
// /var/cache/p2p, a temporary directory is used.
class FakeP2PManagerConfiguration : public P2PManager::Configuration {
public:
FakeP2PManagerConfiguration()
: p2p_client_cmdline_format_(
"p2p-client --get-url={file_id} --minimum-size={minsize}") {
EXPECT_TRUE(utils::MakeTempDirectory("/tmp/p2p-tc.XXXXXX", &p2p_dir_));
SetInitctlStartCommandLine("initctl start p2p");
SetInitctlStopCommandLine("initctl stop p2p");
}
~FakeP2PManagerConfiguration() {
if (p2p_dir_.size() > 0 && !test_utils::RecursiveUnlinkDir(p2p_dir_)) {
PLOG(ERROR) << "Unable to unlink files and directory in " << p2p_dir_;
}
}
// P2PManager::Configuration override
base::FilePath GetP2PDir() override {
return base::FilePath(p2p_dir_);
}
// P2PManager::Configuration override
std::vector<std::string> GetInitctlArgs(bool is_start) override {
return is_start ? initctl_start_args_ : initctl_stop_args_;
}
// P2PManager::Configuration override
std::vector<std::string> GetP2PClientArgs(const std::string &file_id,
size_t minimum_size) override {
std::string formatted_command_line = p2p_client_cmdline_format_;
// Replace {variable} on the passed string.
ReplaceSubstringsAfterOffset(
&formatted_command_line, 0, "{file_id}", file_id);
ReplaceSubstringsAfterOffset(
&formatted_command_line, 0,
"{minsize}", base::StringPrintf("%zu", minimum_size));
return ParseCommandLine(formatted_command_line);
}
// Use |command_line| instead of "initctl start p2p" when attempting
// to start the p2p service.
void SetInitctlStartCommandLine(const std::string &command_line) {
initctl_start_args_ = ParseCommandLine(command_line);
}
// Use |command_line| instead of "initctl stop p2p" when attempting
// to stop the p2p service.
void SetInitctlStopCommandLine(const std::string &command_line) {
initctl_stop_args_ = ParseCommandLine(command_line);
}
// Use |command_line_format| instead of "p2p-client --get-url={file_id}
// --minimum-size={minsize}" when attempting to look up a file using
// p2p-client(1).
//
// The passed |command_line_format| argument can have "{file_id}" and
// "{minsize}" as substrings, that will be replaced by the corresponding
// values passed to GetP2PClientArgs().
void SetP2PClientCommandLine(const std::string &command_line_format) {
p2p_client_cmdline_format_ = command_line_format;
}
private:
// Helper for parsing and splitting |command_line| into an argument
// vector in much the same way a shell would except for not
// supporting wildcards, globs, operators etc. See
// g_shell_parse_argv() for more details. If an error occurs, the
// empty vector is returned.
std::vector<std::string> ParseCommandLine(const std::string &command_line) {
gint argc;
gchar **argv;
std::vector<std::string> ret;
if (!g_shell_parse_argv(command_line.c_str(),
&argc,
&argv,
nullptr)) {
LOG(ERROR) << "Error splitting '" << command_line << "'";
return ret;
}
for (int n = 0; n < argc; n++)
ret.push_back(argv[n]);
g_strfreev(argv);
return ret;
}
// The temporary directory used for p2p.
std::string p2p_dir_;
// Argument vector for starting p2p.
std::vector<std::string> initctl_start_args_;
// Argument vector for stopping p2p.
std::vector<std::string> initctl_stop_args_;
// A string for generating the p2p-client command. See the
// SetP2PClientCommandLine() for details.
std::string p2p_client_cmdline_format_;
DISALLOW_COPY_AND_ASSIGN(FakeP2PManagerConfiguration);
};
} // namespace chromeos_update_engine
#endif // UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_