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_FILESYSTEM_INTERFACE_H_
18 #define UPDATE_ENGINE_PAYLOAD_GENERATOR_FILESYSTEM_INTERFACE_H_
19 
20 // This class is used to abstract a filesystem and iterate the blocks
21 // associated with the files and filesystem structures.
22 // For the purposes of the update payload generation, a filesystem is a
23 // formatted partition composed by fixed-size blocks, since that's the interface
24 // used in the update payload.
25 
26 #include <sys/stat.h>
27 #include <sys/types.h>
28 #include <unistd.h>
29 
30 #include <memory>
31 #include <string>
32 #include <vector>
33 
34 #include <base/macros.h>
35 #include <brillo/key_value_store.h>
36 #include <puffin/utils.h>
37 
38 #include "update_engine/update_metadata.pb.h"
39 
40 namespace chromeos_update_engine {
41 
42 class FilesystemInterface {
43  public:
44   // This represents a file or pseudo-file in the filesystem. It can include
45   // all sort of files, like symlinks, hardlinks, directories and even a file
46   // entry representing the metadata, free space, journaling data, etc.
47   struct File {
48     File() { memset(&file_stat, 0, sizeof(file_stat)); }
49 
50     // The stat struct for the file. This is invalid (inode 0) for some
51     // pseudo-files.
52     struct stat file_stat;
53 
54     // The absolute path to the file inside the filesystem, for example,
55     // "/usr/bin/bash". For pseudo-files, like blocks associated to internal
56     // filesystem tables or free space, the path doesn't start with a /.
57     std::string name;
58 
59     // The list of all physical blocks holding the data of this file in
60     // the same order as the logical data. All the block numbers shall be
61     // between 0 and GetBlockCount() - 1. The blocks are encoded in extents,
62     // indicating the starting block, and the number of consecutive blocks.
63     std::vector<Extent> extents;
64 
65     // All the deflate locations in the file. These locations are not relative
66     // to the extents. They are relative to the file system itself.
67     std::vector<puffin::BitExtent> deflates;
68   };
69 
70   virtual ~FilesystemInterface() = default;
71 
72   // Returns the size of a block in the filesystem.
73   virtual size_t GetBlockSize() const = 0;
74 
75   // Returns the number of blocks in the filesystem.
76   virtual size_t GetBlockCount() const = 0;
77 
78   // Stores in |files| the list of files and pseudo-files in the filesystem. See
79   // FileInterface for details. The paths returned by this method shall not
80   // be repeated; but the same block could be present in more than one file as
81   // happens for example with hard-linked files, but not limited to those cases.
82   // Returns whether the function succeeded.
83   virtual bool GetFiles(std::vector<File>* files) const = 0;
84 
85   // Load the image settings stored in the filesystem in the
86   // /etc/update_engine.conf file. Returns whether the settings were found.
87   virtual bool LoadSettings(brillo::KeyValueStore* store) const = 0;
88 
89  protected:
90   FilesystemInterface() = default;
91 
92  private:
93   DISALLOW_COPY_AND_ASSIGN(FilesystemInterface);
94 };
95 
96 }  // namespace chromeos_update_engine
97 
98 #endif  // UPDATE_ENGINE_PAYLOAD_GENERATOR_FILESYSTEM_INTERFACE_H_
99