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
30namespace chromeos_update_engine {
31
32class 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().
50  void SetMinorVersion(int minor_version) {
51    minor_version_ = minor_version;
52  }
53
54 private:
55  FakeFilesystem() = default;
56
57  uint64_t block_size_;
58  uint64_t block_count_;
59  int minor_version_{-1};
60
61  std::vector<File> files_;
62
63  DISALLOW_COPY_AND_ASSIGN(FakeFilesystem);
64};
65
66}  // namespace chromeos_update_engine
67
68#endif  // UPDATE_ENGINE_PAYLOAD_GENERATOR_FAKE_FILESYSTEM_H_
69