GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/pack.cc
Date: 2025-12-07 02:35:36
Exec Total Coverage
Lines: 283 307 92.2%
Branches: 202 332 60.8%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 */
4
5 #include "pack.h"
6
7 #include <algorithm>
8 #include <cassert>
9 #include <cstring>
10 #include <map>
11
12 #include "util/concurrency.h"
13 #include "util/exception.h"
14 #include "util/platform.h"
15 #include "util/smalloc.h"
16 #include "util/string.h"
17
18 using namespace std; // NOLINT
19
20 namespace { // some private utility functions used by ObjectPackProducer
21
22 2174 void InitializeHeader(const int version, const int num_objects,
23 const size_t pack_size, std::string *header) {
24
1/2
✓ Branch 0 taken 2174 times.
✗ Branch 1 not taken.
 2174 if (header) {
25
2/4
✓ Branch 2 taken 2174 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 2174 times.
✗ Branch 6 not taken.
 2174 *header = "V" + StringifyInt(version) + "\n";
26
3/6
✓ Branch 2 taken 2174 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 2174 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 2174 times.
✗ Branch 9 not taken.
 2174 *header += "S" + StringifyInt(pack_size) + "\n";
27
3/6
✓ Branch 2 taken 2174 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 2174 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 2174 times.
✗ Branch 9 not taken.
 2174 *header += "N" + StringifyInt(num_objects) + "\n";
28 2174 *header += "--\n";
29 }
30 2174 }
31
32 165281 void AppendItemToHeader(ObjectPack::BucketContentType object_type,
33 const std::string &hash_str, const size_t object_size,
34 const std::string &object_name, std::string *header) {
35 // If the item type is kName, the "item_name" parameter should not be empty
36
4/6
✓ Branch 0 taken 81 times.
✓ Branch 1 taken 165200 times.
✓ Branch 2 taken 81 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 81 times.
✗ Branch 6 not taken.
 165281 assert((object_type == ObjectPack::kCas)
37 || ((object_type == ObjectPack::kNamed) && (!object_name.empty())));
38
1/2
✓ Branch 2 taken 165281 times.
✗ Branch 3 not taken.
 165281 std::string line_prefix = "";
39
1/2
✓ Branch 2 taken 165281 times.
✗ Branch 3 not taken.
 165281 std::string line_suffix = "";
40
2/3
✓ Branch 0 taken 81 times.
✓ Branch 1 taken 165200 times.
✗ Branch 2 not taken.
 165281 switch (object_type) {
41 81 case ObjectPack::kNamed:
42
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 line_prefix = "N ";
43
3/6
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 81 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 81 times.
✗ Branch 9 not taken.
 81 line_suffix = std::string(" ") + Base64Url(object_name);
44 81 break;
45 165200 case ObjectPack::kCas:
46
1/2
✓ Branch 1 taken 165200 times.
✗ Branch 2 not taken.
 165200 line_prefix = "C ";
47 165200 break;
48 ✗ default:
49 ✗ PANIC(kLogStderr, "Unknown object pack type to be added to header.");
50 }
51
1/2
✓ Branch 0 taken 165281 times.
✗ Branch 1 not taken.
 165281 if (header) {
52
4/8
✓ Branch 1 taken 165281 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 165281 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 165281 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 165281 times.
✗ Branch 11 not taken.
 330562 *header += line_prefix + hash_str + " " + StringifyInt(object_size)
53
3/6
✓ Branch 1 taken 165281 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 165281 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 165281 times.
✗ Branch 8 not taken.
 165281 + line_suffix + "\n";
54 }
55 165281 }
56
57 } // namespace
58
59 2710865 ObjectPack::Bucket::Bucket()
60 2710865 : content(reinterpret_cast<unsigned char *>(smalloc(kInitialSize)))
61 2710865 , size(0)
62 2710865 , capacity(kInitialSize)
63 2710865 , content_type(kEmpty)
64 5421730 , name() { }
65
66 10568 void ObjectPack::Bucket::Add(const void *buf, const uint64_t buf_size) {
67
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10568 times.
 10568 if (buf_size == 0)
68 ✗ return;
69
70
2/2
✓ Branch 0 taken 112645 times.
✓ Branch 1 taken 10568 times.
 123213 while (size + buf_size > capacity) {
71 112645 capacity *= 2;
72 112645 content = reinterpret_cast<unsigned char *>(srealloc(content, capacity));
73 }
74 10568 memcpy(content + size, buf, buf_size);
75 10568 size += buf_size;
76 }
77
78 2710865 ObjectPack::Bucket::~Bucket() { free(content); }
79
80 //------------------------------------------------------------------------------
81
82 2412 ObjectPack::ObjectPack(const uint64_t limit) : limit_(limit), size_(0) {
83 2412 InitLock();
84 2412 }
85
86 2366 ObjectPack::~ObjectPack() {
87 4732 for (std::set<BucketHandle>::const_iterator i = open_buckets_.begin(),
88 2366 iEnd = open_buckets_.end();
89
2/2
✓ Branch 1 taken 1782 times.
✓ Branch 2 taken 2366 times.
 4148 i != iEnd;
90 1782 ++i) {
91
1/2
✓ Branch 1 taken 1782 times.
✗ Branch 2 not taken.
 1782 delete *i;
92 }
93
94
2/2
✓ Branch 1 taken 2708975 times.
✓ Branch 2 taken 2366 times.
 2711341 for (unsigned i = 0; i < buckets_.size(); ++i)
95
1/2
✓ Branch 1 taken 2708975 times.
✗ Branch 2 not taken.
 2708975 delete buckets_[i];
96 2366 pthread_mutex_destroy(lock_);
97 2366 free(lock_);
98 2366 }
99
100 3494 void ObjectPack::AddToBucket(const void *buf, const uint64_t size,
101 const ObjectPack::BucketHandle handle) {
102 3494 handle->Add(buf, size);
103 3494 }
104
105 2710838 ObjectPack::BucketHandle ObjectPack::NewBucket() {
106
2/4
✓ Branch 1 taken 2710838 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 2710838 times.
✗ Branch 5 not taken.
 2710838 BucketHandle handle = new Bucket();
107
108 2710838 const MutexLockGuard mutex_guard(lock_);
109
1/2
✓ Branch 1 taken 2710838 times.
✗ Branch 2 not taken.
 2710838 open_buckets_.insert(handle);
110 2710838 return handle;
111 2710838 }
112
113 /**
114 * Can only fail due to insufficient remaining space in the ObjectPack.
115 */
116 2711520 bool ObjectPack::CommitBucket(const BucketContentType type,
117 const shash::Any &id,
118 const ObjectPack::BucketHandle handle,
119 const std::string &name) {
120 2711520 handle->id = id;
121
122 2711520 handle->content_type = type;
123
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2711520 times.
 2711520 if (type == kNamed) {
124 ✗ handle->name = name;
125 }
126
127 2711520 const MutexLockGuard mutex_guard(lock_);
128
2/2
✓ Branch 1 taken 27 times.
✓ Branch 2 taken 2711493 times.
 2711520 if (buckets_.size() >= kMaxObjects)
129 27 return false;
130
2/2
✓ Branch 0 taken 2518 times.
✓ Branch 1 taken 2708975 times.
 2711493 if (size_ + handle->size > limit_)
131 2518 return false;
132
1/2
✓ Branch 1 taken 2708975 times.
✗ Branch 2 not taken.
 2708975 open_buckets_.erase(handle);
133
1/2
✓ Branch 1 taken 2708975 times.
✗ Branch 2 not taken.
 2708975 buckets_.push_back(handle);
134 2708975 size_ += handle->size;
135 2708975 return true;
136 2711520 }
137
138 81 void ObjectPack::DiscardBucket(const BucketHandle handle) {
139 81 const MutexLockGuard mutex_guard(lock_);
140
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 open_buckets_.erase(handle);
141
1/2
✓ Branch 0 taken 81 times.
✗ Branch 1 not taken.
 81 delete handle;
142 81 }
143
144 2412 void ObjectPack::InitLock() {
145 2412 lock_ = reinterpret_cast<pthread_mutex_t *>(smalloc(sizeof(pthread_mutex_t)));
146 2412 const int retval = pthread_mutex_init(lock_, NULL);
147
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2412 times.
 2412 assert(retval == 0);
148 2412 }
149
150 /**
151 * If a commit failed, an open Bucket can be transferred to another ObjectPack
152 * with more space.
153 */
154 1039 void ObjectPack::TransferBucket(const ObjectPack::BucketHandle handle,
155 ObjectPack *other) {
156 1039 const MutexLockGuard mutex_guard(lock_);
157
1/2
✓ Branch 1 taken 1039 times.
✗ Branch 2 not taken.
 1039 open_buckets_.erase(handle);
158
1/2
✓ Branch 1 taken 1039 times.
✗ Branch 2 not taken.
 1039 other->open_buckets_.insert(handle);
159 1039 }
160
161 1007428 unsigned char *ObjectPack::BucketContent(size_t idx) const {
162
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1007428 times.
 1007428 assert(idx < buckets_.size());
163 1007428 return buckets_[idx]->content;
164 }
165
166 1172628 uint64_t ObjectPack::BucketSize(size_t idx) const {
167
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1172628 times.
 1172628 assert(idx < buckets_.size());
168 1172628 return buckets_[idx]->size;
169 }
170
171 165200 const shash::Any &ObjectPack::BucketId(size_t idx) const {
172
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 165200 times.
 165200 assert(idx < buckets_.size());
173 165200 return buckets_[idx]->id;
174 }
175
176 //------------------------------------------------------------------------------
177
178 /**
179 * Hash over the header. The hash algorithm needs to be provided by hash.
180 */
181 1947 void ObjectPackProducer::GetDigest(shash::Any *hash) {
182
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1947 times.
 1947 assert(hash);
183 1947 shash::HashString(header_, hash);
184 1947 }
185
186 2093 ObjectPackProducer::ObjectPackProducer(ObjectPack *pack)
187 2093 : pack_(pack), big_file_(NULL), pos_(0), idx_(0), pos_in_bucket_(0) {
188 2093 const unsigned N = pack->GetNoObjects();
189 // rough guess, most likely a little too much
190
1/2
✓ Branch 1 taken 2093 times.
✗ Branch 2 not taken.
 2093 header_.reserve(30 + N * (2 * shash::kMaxDigestSize + 5));
191
192
1/2
✓ Branch 2 taken 2093 times.
✗ Branch 3 not taken.
 2093 InitializeHeader(2, N, pack->size(), &header_);
193
194
2/2
✓ Branch 0 taken 165200 times.
✓ Branch 1 taken 2093 times.
 167293 for (unsigned i = 0; i < N; ++i) {
195
3/6
✓ Branch 2 taken 165200 times.
✗ Branch 3 not taken.
✓ Branch 7 taken 165200 times.
✗ Branch 8 not taken.
✓ Branch 10 taken 165200 times.
✗ Branch 11 not taken.
 165200 AppendItemToHeader(ObjectPack::kCas, pack->BucketId(i).ToString(true),
196 pack->BucketSize(i), "", &header_);
197 }
198 2093 }
199
200 81 ObjectPackProducer::ObjectPackProducer(const shash::Any &id, FILE *big_file,
201 81 const std::string &file_name)
202 81 : pack_(NULL), big_file_(big_file), pos_(0), idx_(0), pos_in_bucket_(0) {
203 81 const int fd = fileno(big_file_);
204
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
 81 assert(fd >= 0);
205 platform_stat64 info;
206 81 const int retval = platform_fstat(fd, &info);
207
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 81 times.
 81 assert(retval == 0);
208
209
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 InitializeHeader(2, 1, info.st_size, &header_);
210
211
2/4
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 81 times.
✗ Branch 5 not taken.
 81 AppendItemToHeader(ObjectPack::kNamed, id.ToString(true), info.st_size,
212 file_name, &header_);
213
214
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 rewind(big_file);
215 81 }
216
217 /**
218 * Copies as many bytes as possible into buf. If the returned number of bytes
219 * is shorter than buf_size, everything has been produced.
220 */
221 851887 unsigned ObjectPackProducer::ProduceNext(const unsigned buf_size,
222 unsigned char *buf) {
223 851887 const unsigned remaining_in_header = (pos_ < header_.size())
224 8924 ? (header_.size() - pos_)
225
2/2
✓ Branch 0 taken 8924 times.
✓ Branch 1 taken 842963 times.
 860811 : 0;
226 851887 const unsigned nbytes_header = std::min(remaining_in_header, buf_size);
227
2/2
✓ Branch 0 taken 8924 times.
✓ Branch 1 taken 842963 times.
 851887 if (nbytes_header) {
228 8924 memcpy(buf, header_.data() + pos_, nbytes_header);
229 8924 pos_ += nbytes_header;
230 }
231
232 851887 unsigned remaining_in_buf = buf_size - nbytes_header;
233
2/2
✓ Branch 0 taken 6831 times.
✓ Branch 1 taken 845056 times.
 851887 if (remaining_in_buf == 0)
234 6831 return nbytes_header;
235 845056 unsigned nbytes_payload = 0;
236
237
2/2
✓ Branch 0 taken 513 times.
✓ Branch 1 taken 844543 times.
 845056 if (big_file_) {
238
1/2
✓ Branch 1 taken 513 times.
✗ Branch 2 not taken.
 513 const size_t nbytes = fread(buf + nbytes_header, 1, remaining_in_buf,
239 big_file_);
240 513 nbytes_payload = nbytes;
241 513 pos_ += nbytes_payload;
242
2/2
✓ Branch 1 taken 844267 times.
✓ Branch 2 taken 276 times.
 844543 } else if (idx_ < pack_->GetNoObjects()) {
243 // Copy a few buckets more
244
6/6
✓ Branch 0 taken 1009413 times.
✓ Branch 1 taken 842282 times.
✓ Branch 3 taken 1007428 times.
✓ Branch 4 taken 1985 times.
✓ Branch 5 taken 1007428 times.
✓ Branch 6 taken 844267 times.
 1851695 while ((remaining_in_buf) > 0 && (idx_ < pack_->GetNoObjects())) {
245 1007428 const unsigned remaining_in_bucket = pack_->BucketSize(idx_)
246 1007428 - pos_in_bucket_;
247 1007428 const unsigned nbytes = std::min(remaining_in_buf, remaining_in_bucket);
248 2014856 memcpy(buf + nbytes_header + nbytes_payload,
249 1007428 pack_->BucketContent(idx_) + pos_in_bucket_, nbytes);
250
251 1007428 pos_in_bucket_ += nbytes;
252 1007428 nbytes_payload += nbytes;
253 1007428 remaining_in_buf -= nbytes;
254
2/2
✓ Branch 0 taken 165200 times.
✓ Branch 1 taken 842228 times.
 1007428 if (nbytes == remaining_in_bucket) {
255 165200 pos_in_bucket_ = 0;
256 165200 idx_++;
257 }
258 }
259 }
260
261 845056 return nbytes_header + nbytes_payload;
262 }
263
264 //------------------------------------------------------------------------------
265
266 1974 ObjectPackConsumer::ObjectPackConsumer(const shash::Any &expected_digest,
267 1974 const unsigned expected_header_size)
268 1974 : expected_digest_(expected_digest)
269 1974 , expected_header_size_(expected_header_size)
270 1974 , pos_(0)
271 1974 , idx_(0)
272 1974 , pos_in_object_(0)
273 1974 , pos_in_accu_(0)
274 1974 , state_(ObjectPackBuild::kStateContinue)
275 1974 , size_(0) {
276 // Upper limit of 100B per entry
277
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1974 times.
 1974 if (expected_header_size > (100 * ObjectPack::kMaxObjects)) {
278 ✗ state_ = ObjectPackBuild::kStateHeaderTooBig;
279 ✗ return;
280 }
281
282
1/2
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
 1974 raw_header_.reserve(expected_header_size);
283 }
284
285 /**
286 * At the end of the function, pos_ will have progressed by buf_size (unless
287 * the buffer contains trailing garbage bytes.
288 */
289 850422 ObjectPackBuild::State ObjectPackConsumer::ConsumeNext(
290 const unsigned buf_size, const unsigned char *buf) {
291
2/2
✓ Branch 0 taken 1809 times.
✓ Branch 1 taken 848613 times.
 850422 if (buf_size == 0)
292 1809 return state_;
293
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 848613 times.
 848613 if (state_ == ObjectPackBuild::kStateDone) {
294 ✗ state_ = ObjectPackBuild::kStateTrailingBytes;
295 ✗ return state_;
296 }
297
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 848613 times.
 848613 if (state_ != ObjectPackBuild::kStateContinue)
298 ✗ return state_;
299
300
2/2
✓ Branch 0 taken 6591 times.
✓ Branch 1 taken 842022 times.
 848613 const unsigned remaining_in_header = (pos_ < expected_header_size_)
301 6591 ? (expected_header_size_ - pos_)
302 : 0;
303 848613 const unsigned nbytes_header = std::min(remaining_in_header, buf_size);
304
2/2
✓ Branch 0 taken 6591 times.
✓ Branch 1 taken 842022 times.
 848613 if (nbytes_header) {
305
2/4
✓ Branch 2 taken 6591 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 6591 times.
✗ Branch 6 not taken.
 6591 raw_header_ += string(reinterpret_cast<const char *>(buf), nbytes_header);
306 6591 pos_ += nbytes_header;
307 }
308
309
2/2
✓ Branch 0 taken 4617 times.
✓ Branch 1 taken 843996 times.
 848613 if (pos_ < expected_header_size_)
310 4617 return ObjectPackBuild::kStateContinue;
311
312 // This condition can only be true once through the lifetime of the
313 // Consumer.
314
3/4
✓ Branch 0 taken 1974 times.
✓ Branch 1 taken 842022 times.
✓ Branch 2 taken 1974 times.
✗ Branch 3 not taken.
 843996 if (nbytes_header && (pos_ == expected_header_size_)) {
315
1/2
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
 1974 shash::Any digest(expected_digest_.algorithm);
316
1/2
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
 1974 shash::HashString(raw_header_, &digest);
317
2/4
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1974 times.
 1974 if (digest != expected_digest_) {
318 ✗ state_ = ObjectPackBuild::kStateCorrupt;
319 54 return state_;
320 } else {
321
1/2
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
 1974 const bool retval = ParseHeader();
322
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1974 times.
 1974 if (!retval) {
323 ✗ state_ = ObjectPackBuild::kStateBadFormat;
324 ✗ return state_;
325 }
326 // We don't need the raw string anymore
327 1974 raw_header_.clear();
328 }
329
330 // Empty pack?
331
6/6
✓ Branch 0 taken 81 times.
✓ Branch 1 taken 1893 times.
✓ Branch 3 taken 54 times.
✓ Branch 4 taken 27 times.
✓ Branch 5 taken 54 times.
✓ Branch 6 taken 1920 times.
 1974 if ((buf_size == nbytes_header) && (index_.size() == 0)) {
332 54 state_ = ObjectPackBuild::kStateDone;
333 54 return state_;
334 }
335 }
336
337 843942 const unsigned remaining_in_buf = buf_size - nbytes_header;
338 843942 const unsigned char *payload = buf + nbytes_header;
339
1/2
✓ Branch 1 taken 843942 times.
✗ Branch 2 not taken.
 843942 return ConsumePayload(remaining_in_buf, payload);
340 }
341
342 /**
343 * Informs listeners for small complete objects. For large objects, buffers
344 * the
345 * input into reasonably sized chunks. buf can contain both a chunk of data
346 * that needs to be added to the consumer's accumulator and a bunch of
347 * complete small objects. We use the accumulator only if necessary to avoid
348 * unnecessary memory copies.
349 */
350 843942 ObjectPackBuild::State ObjectPackConsumer::ConsumePayload(
351 const unsigned buf_size, const unsigned char *buf) {
352 843942 uint64_t pos_in_buf = 0;
353 1973193 while ((idx_ < index_.size())
354
8/8
✓ Branch 0 taken 1971273 times.
✓ Branch 1 taken 1920 times.
✓ Branch 2 taken 842049 times.
✓ Branch 3 taken 1129224 times.
✓ Branch 5 taken 27 times.
✓ Branch 6 taken 842022 times.
✓ Branch 7 taken 1129251 times.
✓ Branch 8 taken 843942 times.
 1973193 && ((pos_in_buf < buf_size) || (index_[idx_].size == 0))) {
355 // Fill the accumulator or process next small object
356 uint64_t nbytes; // How many bytes are consumed in this iteration
357 1129251 const uint64_t remaining_in_buf = buf_size - pos_in_buf;
358 1129251 const uint64_t remaining_in_object = index_[idx_].size - pos_in_object_;
359 1129251 const bool is_small_rest = remaining_in_buf < kAccuSize;
360
361 // We use the accumulator if there is already something in or if we have a
362 // small piece of data of a larger object.
363 1129251 nbytes = std::min(remaining_in_object, remaining_in_buf);
364
2/2
✓ Branch 0 taken 858549 times.
✓ Branch 1 taken 270702 times.
 1129251 if ((pos_in_accu_ > 0)
365
4/4
✓ Branch 0 taken 696330 times.
✓ Branch 1 taken 162219 times.
✓ Branch 2 taken 125064 times.
✓ Branch 3 taken 571266 times.
 858549 || ((remaining_in_buf < remaining_in_object) && is_small_rest)) {
366 395766 const uint64_t remaining_in_accu = kAccuSize - pos_in_accu_;
367 395766 nbytes = std::min(remaining_in_accu, nbytes);
368 395766 memcpy(accumulator_ + pos_in_accu_, buf + pos_in_buf, nbytes);
369 395766 pos_in_accu_ += nbytes;
370
4/4
✓ Branch 0 taken 273510 times.
✓ Branch 1 taken 122256 times.
✓ Branch 2 taken 2808 times.
✓ Branch 3 taken 270702 times.
 395766 if ((pos_in_accu_ == kAccuSize) || (nbytes == remaining_in_object)) {
371
1/2
✓ Branch 1 taken 125064 times.
✗ Branch 2 not taken.
 125064 NotifyListeners(ObjectPackBuild::Event(
372
1/2
✓ Branch 3 taken 125064 times.
✗ Branch 4 not taken.
 125064 index_[idx_].id, index_[idx_].size, pos_in_accu_, accumulator_,
373 125064 index_[idx_].entry_type, index_[idx_].entry_name));
374 125064 pos_in_accu_ = 0;
375 }
376 395766 } else { // directly trigger listeners using buf
377
1/2
✓ Branch 1 taken 733485 times.
✗ Branch 2 not taken.
 733485 NotifyListeners(ObjectPackBuild::Event(
378
1/2
✓ Branch 3 taken 733485 times.
✗ Branch 4 not taken.
 733485 index_[idx_].id, index_[idx_].size, nbytes, buf + pos_in_buf,
379 733485 index_[idx_].entry_type, index_[idx_].entry_name));
380 }
381
382 1129251 pos_in_buf += nbytes;
383 1129251 pos_in_object_ += nbytes;
384
2/2
✓ Branch 0 taken 165054 times.
✓ Branch 1 taken 964197 times.
 1129251 if (nbytes == remaining_in_object) {
385 165054 idx_++;
386 165054 pos_in_object_ = 0;
387 }
388 }
389
390 843942 pos_ += buf_size;
391
392
2/2
✓ Branch 1 taken 1920 times.
✓ Branch 2 taken 842022 times.
 843942 if (idx_ == index_.size())
393
1/2
✓ Branch 0 taken 1920 times.
✗ Branch 1 not taken.
 1920 state_ = (pos_in_buf == buf_size) ? ObjectPackBuild::kStateDone
394 : ObjectPackBuild::kStateTrailingBytes;
395 else
396 842022 state_ = ObjectPackBuild::kStateContinue;
397 843942 return state_;
398 }
399
400 1974 bool ObjectPackConsumer::ParseHeader() {
401 1974 map<char, string> header;
402 const unsigned char *data = reinterpret_cast<const unsigned char *>(
403 1974 raw_header_.data());
404
1/2
✓ Branch 2 taken 1974 times.
✗ Branch 3 not taken.
 1974 ParseKeyvalMem(data, raw_header_.size(), &header);
405
2/5
✓ Branch 2 taken 1974 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 1974 times.
 1974 if (header.find('V') == header.end())
406 ✗ return false;
407
3/7
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1974 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 1974 times.
 1974 if (header['V'] != "2")
408 ✗ return false;
409
2/4
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1974 times.
✗ Branch 5 not taken.
 1974 size_ = String2Uint64(header['S']);
410
2/4
✓ Branch 1 taken 1974 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1974 times.
✗ Branch 5 not taken.
 1974 const unsigned nobjects = String2Uint64(header['N']);
411
412
2/2
✓ Branch 0 taken 54 times.
✓ Branch 1 taken 1920 times.
 1974 if (nobjects == 0)
413 54 return true;
414
415 // Build the object index
416 1920 const size_t separator_idx = raw_header_.find("--\n");
417
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1920 times.
 1920 if (separator_idx == string::npos)
418 ✗ return false;
419 1920 unsigned index_idx = separator_idx + 3;
420
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 1920 times.
 1920 if (index_idx >= raw_header_.size())
421 ✗ return false;
422
423 1920 uint64_t sum_size = 0;
424 do {
425 165054 const unsigned remaining_in_header = raw_header_.size() - index_idx;
426 330108 const string line = GetLineMem(raw_header_.data() + index_idx,
427
1/2
✓ Branch 2 taken 165054 times.
✗ Branch 3 not taken.
 165054 remaining_in_header);
428
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 165054 times.
 165054 if (line == "")
429 ✗ break;
430
431
1/2
✓ Branch 1 taken 165054 times.
✗ Branch 2 not taken.
 165054 IndexEntry entry;
432
2/4
✓ Branch 1 taken 165054 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 165054 times.
 165054 if (!ParseItem(line, &entry, &sum_size)) {
433 ✗ break;
434 }
435
436
1/2
✓ Branch 1 taken 165054 times.
✗ Branch 2 not taken.
 165054 index_.push_back(entry);
437 165054 index_idx += line.size() + 1;
438
4/6
✓ Branch 1 taken 165054 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 165054 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 163134 times.
✓ Branch 8 taken 1920 times.
 330108 } while (index_idx < raw_header_.size());
439
440
2/4
✓ Branch 1 taken 1920 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1920 times.
✗ Branch 4 not taken.
 1920 return (nobjects == index_.size()) && (size_ == sum_size);
441 1974 }
442
443 165054 bool ObjectPackConsumer::ParseItem(const std::string &line,
444 ObjectPackConsumer::IndexEntry *entry,
445 uint64_t *sum_size) {
446
2/4
✓ Branch 0 taken 165054 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 165054 times.
 165054 if (!entry || !sum_size) {
447 ✗ return false;
448 }
449
450
2/2
✓ Branch 1 taken 165027 times.
✓ Branch 2 taken 27 times.
 165054 if (line[0] == 'C') { // CAS blob
451 165027 const ObjectPack::BucketContentType entry_type = ObjectPack::kCas;
452
453 // We could use SplitString but we can have many lines so we do something
454 // more efficient here
455 165027 const size_t separator = line.find(' ', 2);
456
3/6
✓ Branch 0 taken 165027 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 165027 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 165027 times.
 165027 if ((separator == string::npos) || (separator == (line.size() - 1))) {
457 ✗ return false;
458 }
459
460
2/4
✓ Branch 1 taken 165027 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 165027 times.
✗ Branch 5 not taken.
 165027 const uint64_t size = String2Uint64(line.substr(separator + 1));
461 165027 *sum_size += size;
462
463 // Warning do not construct a HexPtr with an rvalue!
464 // The constructor takes the address of its argument.
465
1/2
✓ Branch 1 taken 165027 times.
✗ Branch 2 not taken.
 165027 const std::string hash_string = line.substr(2, separator - 2);
466 165027 const shash::HexPtr hex_ptr(hash_string);
467
468
1/2
✓ Branch 1 taken 165027 times.
✗ Branch 2 not taken.
 165027 entry->id = shash::MkFromSuffixedHexPtr(hex_ptr);
469 165027 entry->size = size;
470 165027 entry->entry_type = entry_type;
471
1/2
✓ Branch 1 taken 165027 times.
✗ Branch 2 not taken.
 165027 entry->entry_name = "";
472
1/2
✓ Branch 2 taken 27 times.
✗ Branch 3 not taken.
 165054 } else if (line[0] == 'N') { // Named file
473 27 const ObjectPack::BucketContentType entry_type = ObjectPack::kNamed;
474
475 // First separator, before the size field
476 27 const size_t separator1 = line.find(' ', 2);
477
3/6
✓ Branch 0 taken 27 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 27 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 27 times.
 27 if ((separator1 == string::npos) || (separator1 == (line.size() - 1))) {
478 ✗ return false;
479 }
480
481 // Second separator, before the name field
482 27 const size_t separator2 = line.find(' ', separator1 + 1);
483
1/2
✓ Branch 0 taken 27 times.
✗ Branch 1 not taken.
 27 if ((separator1 == 0) || (separator1 == string::npos)
484
3/6
✓ Branch 0 taken 27 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 27 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 27 times.
 54 || (separator1 == (line.size() - 1))) {
485 ✗ return false;
486 }
487
488
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 27 const uint64_t size = String2Uint64(
489
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 54 line.substr(separator1 + 1, separator2 - separator1 - 1));
490
491 27 std::string name;
492
3/7
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 27 times.
✗ Branch 5 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 27 times.
 27 if (!Debase64(line.substr(separator2 + 1), &name)) {
493 ✗ return false;
494 }
495
496 27 *sum_size += size;
497
498 // Warning do not construct a HexPtr with an rvalue!
499 // The constructor takes the address of its argument.
500
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 27 const std::string hash_string = line.substr(2, separator1 - 2);
501 27 const shash::HexPtr hex_ptr(hash_string);
502
503
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 27 entry->id = shash::MkFromSuffixedHexPtr(hex_ptr);
504 27 entry->size = size;
505 27 entry->entry_type = entry_type;
506
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 27 entry->entry_name = name;
507
1/2
✓ Branch 2 taken 27 times.
✗ Branch 3 not taken.
 27 } else { // Error
508 ✗ return false;
509 }
510
511 165054 return true;
512 }
513