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 {
FileFile48     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     // If true, the file is already compressed on the disk, so we don't need to
66     // parse it again for deflates. For example, image .gz files inside a
67     // compressed SquashFS image. They might have already been compressed by the
68     // mksquashfs, so we can't really parse the file and look for deflate
69     // compressed parts anymore.
70     bool is_compressed = false;
71 
72     // All the deflate locations in the file. These locations are not relative
73     // to the extents. They are relative to the file system itself.
74     std::vector<puffin::BitExtent> deflates;
75   };
76 
77   virtual ~FilesystemInterface() = default;
78 
79   // Returns the size of a block in the filesystem.
80   virtual size_t GetBlockSize() const = 0;
81 
82   // Returns the number of blocks in the filesystem.
83   virtual size_t GetBlockCount() const = 0;
84 
85   // Stores in |files| the list of files and pseudo-files in the filesystem. See
86   // FileInterface for details. The paths returned by this method shall not
87   // be repeated; but the same block could be present in more than one file as
88   // happens for example with hard-linked files, but not limited to those cases.
89   // Returns whether the function succeeded.
90   virtual bool GetFiles(std::vector<File>* files) const = 0;
91 
92   // Load the image settings stored in the filesystem in the
93   // /etc/update_engine.conf file. Returns whether the settings were found.
94   virtual bool LoadSettings(brillo::KeyValueStore* store) const = 0;
95 
96  protected:
97   FilesystemInterface() = default;
98 
99  private:
100   DISALLOW_COPY_AND_ASSIGN(FilesystemInterface);
101 };
102 
103 }  // namespace chromeos_update_engine
104 
105 #endif  // UPDATE_ENGINE_PAYLOAD_GENERATOR_FILESYSTEM_INTERFACE_H_
106