1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_ITERATOR_H__
#define CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_ITERATOR_H__
// This class is used to walk a filesystem. It will iterate over every file
// on the same device as the file passed in the ctor. Directories will be
// visited before their children. Children will be visited in no particular
// order.
// The iterator is a forward iterator. It's not random access nor can it be
// decremented.
// Note: If the iterator comes across a mount point where another filesystem
// is mounted, that mount point will be present, but none of its children
// will be. Technically the mount point is on the other filesystem (and
// the Stat() call will verify that), but we return it anyway since:
// 1. Such a folder must exist in the first filesystem if it got used
// as a mount point.
// 2. You probably want to copy if it you're using the iterator to do a
// filesystem copy
// 3. If you don't want that, you can just check Stat().st_dev and skip
// foreign filesystems manually.
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <unistd.h>
#include <string>
#include <set>
#include <vector>
namespace chromeos_update_engine {
class FilesystemIterator {
public:
FilesystemIterator(const std::string& path,
const std::set<std::string>& excl_prefixes);
~FilesystemIterator();
// Returns stat struct for the current file.
struct stat GetStat() const {
return stbuf_;
}
// Returns full path for current file.
std::string GetFullPath() const;
// Returns the path that's part of the iterator. For example, if
// the object were constructed by passing in "/foo/bar" and Path()
// returns "/foo/bar/baz/bat.txt", IterPath would return
// "/baz/bat.txt". When this object is on root (ie, the very first
// path), IterPath will return "", otherwise the first character of
// IterPath will be "/".
std::string GetPartialPath() const;
// Returns name for current file.
std::string GetBasename() const {
return names_.back();
}
// Increments to the next file.
void Increment();
// If we're at the end. If at the end, do not call Stat(), Path(), etc.,
// since this iterator currently isn't pointing to any file at all.
bool IsEnd() const {
return is_end_;
}
// Returns true if the iterator is in an error state.
bool IsErr() const {
return is_err_;
}
private:
// Helper for Increment.
void IncrementInternal();
// Returns true if path exists and it's a directory.
bool DirectoryExists(const std::string& path);
// In general (i.e., not midway through a call to Increment()), there is a
// relationship between dirs_ and names_: dirs[i] == names_[i - 1].
// For example, say we are asked to iterate "/usr/local" and we're currently
// at /usr/local/share/dict/words. dirs_ contains DIR* variables for the
// dirs at: {"/usr/local", ".../share", ".../dict"} and names_ contains:
// {"share", "dict", "words"}. root_path_ contains "/usr/local".
// root_dev_ would be the dev for root_path_
// (and /usr/local/share/dict/words). stbuf_ would be the stbuf for
// /usr/local/share/dict/words.
// All opened directories. If this is empty, we're currently on the root,
// but not descended into the root.
// This will always contain the current directory and all it's ancestors
// in root-to-leaf order. For more details, see comment above.
std::vector<DIR*> dirs_;
// The list of all filenames for the current path that we've descended into.
std::vector<std::string> names_;
// The device of the root path we've been asked to iterate.
dev_t root_dev_;
// The root path we've been asked to iteratate.
std::string root_path_;
// Exclude items w/ this prefix.
std::set<std::string> excl_prefixes_;
// The struct stat of the current file we're at.
struct stat stbuf_;
// Generally false; set to true when we reach the end of files to iterate
// or error occurs.
bool is_end_;
// Generally false; set to true if an error occurrs.
bool is_err_;
};
} // namespace chromeos_update_engine
#endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_FILESYSTEM_ITERATOR_H__
|
