1 // 2 // Copyright (C) 2015 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_PAYLOAD_GENERATOR_FAKE_FILESYSTEM_H_ 18 #define UPDATE_ENGINE_PAYLOAD_GENERATOR_FAKE_FILESYSTEM_H_ 19 20 // A fake filesystem interface implementation allowing the user to add arbitrary 21 // files/metadata. 22 23 #include "update_engine/payload_generator/filesystem_interface.h" 24 25 #include <string> 26 #include <vector> 27 28 #include "update_engine/update_metadata.pb.h" 29 30 namespace chromeos_update_engine { 31 32 class FakeFilesystem : public FilesystemInterface { 33 public: 34 FakeFilesystem(uint64_t block_size, uint64_t block_count); 35 virtual ~FakeFilesystem() = default; 36 37 // FilesystemInterface overrides. 38 size_t GetBlockSize() const override; 39 size_t GetBlockCount() const override; 40 bool GetFiles(std::vector<File>* files) const override; 41 bool LoadSettings(brillo::KeyValueStore* store) const override; 42 43 // Fake methods. 44 45 // Add a file to the list of fake files. 46 void AddFile(const std::string& filename, const std::vector<Extent>& extents); 47 48 // Sets the PAYLOAD_MINOR_VERSION key stored by LoadSettings(). Use a negative 49 // value to produce an error in LoadSettings(). SetMinorVersion(int minor_version)50 void SetMinorVersion(int minor_version) { minor_version_ = minor_version; } 51 52 private: 53 FakeFilesystem() = default; 54 55 uint64_t block_size_; 56 uint64_t block_count_; 57 int minor_version_{-1}; 58 59 std::vector<File> files_; 60 61 DISALLOW_COPY_AND_ASSIGN(FakeFilesystem); 62 }; 63 64 } // namespace chromeos_update_engine 65 66 #endif // UPDATE_ENGINE_PAYLOAD_GENERATOR_FAKE_FILESYSTEM_H_ 67