1 //
2 // Copyright (C) 2013 The Android Open Source Project
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 //      http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16 
17 #ifndef UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_
18 #define UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_
19 
20 #include "update_engine/common/test_utils.h"
21 #include "update_engine/common/utils.h"
22 #include "update_engine/p2p_manager.h"
23 
24 #include <string>
25 #include <vector>
26 
27 #include <base/files/file_util.h>
28 #include <base/logging.h>
29 #include <base/strings/string_number_conversions.h>
30 #include <base/strings/string_util.h>
31 
32 namespace chromeos_update_engine {
33 
34 // Configuration for P2PManager for use in unit tests. Instead of
35 // /var/cache/p2p, a temporary directory is used.
36 class FakeP2PManagerConfiguration : public P2PManager::Configuration {
37  public:
FakeP2PManagerConfiguration()38   FakeP2PManagerConfiguration() {
39     EXPECT_TRUE(utils::MakeTempDirectory("p2p-tc.XXXXXX", &p2p_dir_));
40   }
41 
~FakeP2PManagerConfiguration()42   ~FakeP2PManagerConfiguration() {
43     if (p2p_dir_.size() > 0 &&
44         !base::DeleteFile(base::FilePath(p2p_dir_), true)) {
45       PLOG(ERROR) << "Unable to unlink files and directory in " << p2p_dir_;
46     }
47   }
48 
49   // P2PManager::Configuration override
GetP2PDir()50   base::FilePath GetP2PDir() override {
51     return base::FilePath(p2p_dir_);
52   }
53 
54   // P2PManager::Configuration override
GetInitctlArgs(bool is_start)55   std::vector<std::string> GetInitctlArgs(bool is_start) override {
56     return is_start ? initctl_start_args_ : initctl_stop_args_;
57   }
58 
59   // P2PManager::Configuration override
GetP2PClientArgs(const std::string & file_id,size_t minimum_size)60   std::vector<std::string> GetP2PClientArgs(const std::string &file_id,
61                                             size_t minimum_size) override {
62     std::vector<std::string> formatted_command = p2p_client_cmd_format_;
63     // Replace {variable} on the passed string.
64     std::string str_minimum_size = std::to_string(minimum_size);
65     for (std::string& arg : formatted_command) {
66       base::ReplaceSubstringsAfterOffset(&arg, 0, "{file_id}", file_id);
67       base::ReplaceSubstringsAfterOffset(&arg, 0, "{minsize}",
68                                          str_minimum_size);
69     }
70     return formatted_command;
71   }
72 
73   // Use |command_line| instead of "initctl start p2p" when attempting
74   // to start the p2p service.
SetInitctlStartCommand(const std::vector<std::string> & command)75   void SetInitctlStartCommand(const std::vector<std::string>& command) {
76     initctl_start_args_ = command;
77   }
78 
79   // Use |command_line| instead of "initctl stop p2p" when attempting
80   // to stop the p2p service.
SetInitctlStopCommand(const std::vector<std::string> & command)81   void SetInitctlStopCommand(const std::vector<std::string>& command) {
82     initctl_stop_args_ = command;
83   }
84 
85   // Use |command_format| instead of "p2p-client --get-url={file_id}
86   // --minimum-size={minsize}" when attempting to look up a file using
87   // p2p-client(1).
88   //
89   // The passed |command_format| argument can have "{file_id}" and "{minsize}"
90   // as substrings of any of its elements, that will be replaced by the
91   // corresponding values passed to GetP2PClientArgs().
SetP2PClientCommand(const std::vector<std::string> & command_format)92   void SetP2PClientCommand(const std::vector<std::string>& command_format) {
93     p2p_client_cmd_format_ = command_format;
94   }
95 
96  private:
97   // The temporary directory used for p2p.
98   std::string p2p_dir_;
99 
100   // Argument vector for starting p2p.
101   std::vector<std::string> initctl_start_args_{"initctl", "start", "p2p"};
102 
103   // Argument vector for stopping p2p.
104   std::vector<std::string> initctl_stop_args_{"initctl", "stop", "p2p"};
105 
106   // A string for generating the p2p-client command. See the
107   // SetP2PClientCommandLine() for details.
108   std::vector<std::string> p2p_client_cmd_format_{
109       "p2p-client", "--get-url={file_id}", "--minimum-size={minsize}"};
110 
111   DISALLOW_COPY_AND_ASSIGN(FakeP2PManagerConfiguration);
112 };
113 
114 }  // namespace chromeos_update_engine
115 
116 #endif  // UPDATE_ENGINE_FAKE_P2P_MANAGER_CONFIGURATION_H_
117