Directory: | cvmfs/ |
---|---|
File: | cvmfs/fd_table.h |
Date: | 2025-07-06 02:35:01 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 62 | 63 | 98.4% |
Branches: | 40 | 59 | 67.8% |
Line | Branch | Exec | Source |
---|---|---|---|
1 | /** | ||
2 | * This file is part of the CernVM File System. | ||
3 | */ | ||
4 | |||
5 | #ifndef CVMFS_FD_TABLE_H_ | ||
6 | #define CVMFS_FD_TABLE_H_ | ||
7 | |||
8 | #include <errno.h> | ||
9 | #include <inttypes.h> | ||
10 | #include <stdint.h> | ||
11 | |||
12 | #include <cassert> | ||
13 | #include <vector> | ||
14 | |||
15 | #include "util/single_copy.h" | ||
16 | |||
17 | /** | ||
18 | * Maintains integers mapped to custom open file descriptors. File descriptors | ||
19 | * can be added, removed, and accessed. All operations take constant time. The | ||
20 | * maximum file descriptor number needs to be known upfront. | ||
21 | * | ||
22 | * Note that new file descriptors do not necessarily have the smallest available | ||
23 | * number but any number between 0..max_open_fds. | ||
24 | * | ||
25 | * This class is used by a couple of cache managers. | ||
26 | */ | ||
27 | template<class HandleT> | ||
28 | class FdTable : SingleCopy { | ||
29 | public: | ||
30 | 910 | FdTable(unsigned max_open_fds, const HandleT &invalid_handle) | |
31 | 910 | : invalid_handle_(invalid_handle) | |
32 | 910 | , fd_pivot_(0) | |
33 |
1/2✓ Branch 2 taken 910 times.
✗ Branch 3 not taken.
|
910 | , fd_index_(max_open_fds) |
34 |
4/7✓ Branch 3 taken 74 times.
✓ Branch 4 taken 836 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 74 times.
✗ Branch 7 not taken.
✓ Branch 9 taken 74 times.
✗ Branch 10 not taken.
|
1820 | , open_fds_(max_open_fds, FdWrapper(invalid_handle_, 0)) { |
35 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 910 times.
|
910 | assert(max_open_fds > 0); |
36 |
2/2✓ Branch 0 taken 2548076 times.
✓ Branch 1 taken 910 times.
|
2548986 | for (unsigned i = 0; i < max_open_fds; ++i) { |
37 | 2548076 | fd_index_[i] = i; | |
38 | 2548076 | open_fds_[i].index = i; | |
39 | } | ||
40 | 910 | } | |
41 | |||
42 | /** | ||
43 | * Used to restore the state. | ||
44 | */ | ||
45 | 14 | void AssignFrom(const FdTable<HandleT> &other) { | |
46 | 14 | invalid_handle_ = other.invalid_handle_; | |
47 | 14 | fd_pivot_ = other.fd_pivot_; | |
48 | 14 | fd_index_.resize(other.fd_index_.size()); | |
49 |
1/5✗ Branch 2 not taken.
✓ Branch 3 taken 14 times.
✗ Branch 4 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
14 | open_fds_.resize(other.open_fds_.size(), FdWrapper(invalid_handle_, 0)); |
50 |
2/2✓ Branch 1 taken 1792 times.
✓ Branch 2 taken 14 times.
|
1806 | for (unsigned i = 0; i < fd_index_.size(); ++i) { |
51 | 1792 | fd_index_[i] = other.fd_index_[i]; | |
52 | 1792 | open_fds_[i] = other.open_fds_[i]; | |
53 | } | ||
54 | 14 | } | |
55 | |||
56 | /** | ||
57 | * Used to save the state. | ||
58 | */ | ||
59 | 14 | FdTable<HandleT> *Clone() { | |
60 | 14 | FdTable<HandleT> *result = new FdTable<HandleT>(open_fds_.size(), | |
61 |
1/2✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
|
14 | invalid_handle_); |
62 | 14 | result->fd_pivot_ = fd_pivot_; | |
63 |
2/2✓ Branch 1 taken 1792 times.
✓ Branch 2 taken 14 times.
|
1806 | for (unsigned i = 0; i < fd_index_.size(); ++i) { |
64 | 1792 | result->fd_index_[i] = fd_index_[i]; | |
65 | 1792 | result->open_fds_[i] = open_fds_[i]; | |
66 | } | ||
67 | 14 | return result; | |
68 | } | ||
69 | |||
70 | |||
71 | /** | ||
72 | * Registers fd with a currently unused number. If the table is full, | ||
73 | * returns -ENFILE; | ||
74 | */ | ||
75 | 15287255 | int OpenFd(const HandleT &handle) { | |
76 |
2/3✗ Branch 0 not taken.
✓ Branch 1 taken 15009030 times.
✓ Branch 2 taken 278225 times.
|
15287255 | if (handle == invalid_handle_) |
77 | ✗ | return -EINVAL; | |
78 |
2/2✓ Branch 1 taken 2497899 times.
✓ Branch 2 taken 12789356 times.
|
15287255 | if (fd_pivot_ >= fd_index_.size()) |
79 | 2497899 | return -ENFILE; | |
80 | |||
81 | 12789356 | size_t next_fd = fd_index_[fd_pivot_]; | |
82 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 12789356 times.
|
12789356 | assert(next_fd < open_fds_.size()); |
83 |
2/3✗ Branch 1 not taken.
✓ Branch 2 taken 12511170 times.
✓ Branch 3 taken 278186 times.
|
12789356 | assert(open_fds_[next_fd].handle == invalid_handle_); |
84 |
1/2✓ Branch 2 taken 74 times.
✗ Branch 3 not taken.
|
12789356 | open_fds_[next_fd] = FdWrapper(handle, fd_pivot_); |
85 | 12789356 | ++fd_pivot_; | |
86 | 12789356 | return next_fd; | |
87 | } | ||
88 | |||
89 | /** | ||
90 | * For invalid and unused numbers, the invalid handle is returned. | ||
91 | */ | ||
92 | 15548482 | HandleT GetHandle(int fd) { | |
93 |
2/2✓ Branch 1 taken 13066170 times.
✓ Branch 2 taken 2482312 times.
|
15548482 | return IsValid(fd) ? open_fds_[fd].handle : invalid_handle_; |
94 | } | ||
95 | |||
96 | |||
97 | /** | ||
98 | * Releases fd back to the set of available numbers. Gracefully handles | ||
99 | * invalid handles (-EBADFD) | ||
100 | */ | ||
101 | 15269396 | int CloseFd(int fd) { | |
102 |
2/2✓ Branch 1 taken 2480460 times.
✓ Branch 2 taken 12788936 times.
|
15269396 | if (!IsValid(fd)) |
103 | 2480460 | return -EBADF; | |
104 | |||
105 | 12788936 | unsigned index = open_fds_[fd].index; | |
106 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 12788936 times.
|
12788936 | assert(index < fd_index_.size()); |
107 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 12788936 times.
|
12788936 | assert(fd_pivot_ <= fd_index_.size()); |
108 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12788936 times.
|
12788936 | assert(fd_pivot_ > 0); |
109 | 12788936 | open_fds_[fd].handle = invalid_handle_; | |
110 | 12788936 | --fd_pivot_; | |
111 |
2/2✓ Branch 0 taken 7080241 times.
✓ Branch 1 taken 5708695 times.
|
12788936 | if (index < fd_pivot_) { |
112 | 7080241 | unsigned other = fd_index_[fd_pivot_]; | |
113 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 7080241 times.
|
7080241 | assert(other < open_fds_.size()); |
114 |
2/3✗ Branch 1 not taken.
✓ Branch 2 taken 6804090 times.
✓ Branch 3 taken 276151 times.
|
7080241 | assert(open_fds_[other].handle != invalid_handle_); |
115 | 7080241 | open_fds_[other].index = index; | |
116 | 7080241 | fd_index_[index] = other; | |
117 | 7080241 | fd_index_[fd_pivot_] = fd; | |
118 | } | ||
119 | 12788936 | return 0; | |
120 | } | ||
121 | |||
122 | 1792 | unsigned GetMaxFds() const { return fd_index_.size(); } | |
123 | |||
124 | private: | ||
125 | struct FdWrapper { | ||
126 | 12790280 | FdWrapper(HandleT h, unsigned i) : handle(h), index(i) { } | |
127 | |||
128 | HandleT handle; | ||
129 | /** | ||
130 | * Back-pointer into fd_index_, which is needed when closing a file. | ||
131 | */ | ||
132 | unsigned index; | ||
133 | }; | ||
134 | |||
135 | |||
136 | 30817878 | inline bool IsValid(int fd) { | |
137 |
6/6✓ Branch 0 taken 30817818 times.
✓ Branch 1 taken 60 times.
✓ Branch 3 taken 60 times.
✓ Branch 4 taken 30817758 times.
✓ Branch 5 taken 120 times.
✓ Branch 6 taken 30817758 times.
|
30817878 | if ((fd < 0) || (static_cast<unsigned>(fd) >= open_fds_.size())) |
138 | 120 | return false; | |
139 | 30817758 | return open_fds_[fd].handle != invalid_handle_; | |
140 | } | ||
141 | |||
142 | /** | ||
143 | * An unused (available) file descriptor. | ||
144 | */ | ||
145 | HandleT invalid_handle_; | ||
146 | /** | ||
147 | * The index of the first available file descriptor in fd_index_. | ||
148 | */ | ||
149 | unsigned fd_pivot_; | ||
150 | /** | ||
151 | * Maps into open_fds_. Until fd_pivot_, file descriptors are used. As of | ||
152 | * fd_pivot_, points to free file descriptors. Used to acquire new file | ||
153 | * descriptors in constant time. | ||
154 | */ | ||
155 | std::vector<unsigned> fd_index_; | ||
156 | /** | ||
157 | * The file descriptor number mapped to a user-defined file descriptor | ||
158 | * (struct). The fd integer passed to users of the file descriptor table | ||
159 | * points into this array. | ||
160 | */ | ||
161 | std::vector<FdWrapper> open_fds_; | ||
162 | }; // class FdTable | ||
163 | |||
164 | #endif // CVMFS_FD_TABLE_H_ | ||
165 |