GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/ingestion/item.cc
Date: 2025-06-22 02:36:02
Exec Total Coverage
Lines: 142 142 100.0%
Branches: 35 64 54.7%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 */
4
5 #include "item.h"
6
7 #include <algorithm>
8 #include <cassert>
9 #include <cstdlib>
10 #include <cstring>
11
12 #include "ingestion/ingestion_source.h"
13 #include "item_mem.h"
14 #include "util/concurrency.h"
15 #include "util/smalloc.h"
16
17 332427 FileItem::FileItem(IngestionSource *source,
18 uint64_t min_chunk_size,
19 uint64_t avg_chunk_size,
20 uint64_t max_chunk_size,
21 zlib::Algorithms compression_algorithm,
22 shash::Algorithms hash_algorithm,
23 shash::Suffix hash_suffix,
24 bool may_have_chunks,
25 332427 bool has_legacy_bulk_chunk)
26 332427 : source_(source)
27 332427 , compression_algorithm_(compression_algorithm)
28 332427 , hash_algorithm_(hash_algorithm)
29 332427 , hash_suffix_(hash_suffix)
30 332427 , has_legacy_bulk_chunk_(has_legacy_bulk_chunk)
31 332427 , size_(kSizeUnknown)
32 332427 , may_have_chunks_(may_have_chunks)
33
1/2
✓ Branch 1 taken 332427 times.
✗ Branch 2 not taken.
 332427 , chunk_detector_(min_chunk_size, avg_chunk_size, max_chunk_size)
34
1/2
✓ Branch 1 taken 332427 times.
✗ Branch 2 not taken.
 332427 , bulk_hash_(hash_algorithm)
35
1/2
✓ Branch 2 taken 332427 times.
✗ Branch 3 not taken.
 664854 , chunks_(1) {
36 332427 const int retval = pthread_mutex_init(&lock_, NULL);
37
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 332427 times.
 332427 assert(retval == 0);
38 332427 atomic_init64(&nchunks_in_fly_);
39 332427 atomic_init32(&is_fully_chunked_);
40 332427 }
41
42 331941 FileItem::~FileItem() { pthread_mutex_destroy(&lock_); }
43
44 252733 void FileItem::RegisterChunk(const FileChunk &file_chunk) {
45 252733 const MutexLockGuard lock_guard(lock_);
46
47
2/2
✓ Branch 1 taken 857 times.
✓ Branch 2 taken 251876 times.
 252733 switch (file_chunk.content_hash().suffix) {
48 857 case shash::kSuffixPartial:
49
1/2
✓ Branch 1 taken 857 times.
✗ Branch 2 not taken.
 857 chunks_.PushBack(file_chunk);
50 857 break;
51
52 251876 default:
53
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 251876 times.
 251876 assert(file_chunk.offset() == 0);
54
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 251876 times.
 251876 assert(file_chunk.size() == size_);
55 251876 bulk_hash_ = file_chunk.content_hash();
56 251876 break;
57 }
58 252733 atomic_dec64(&nchunks_in_fly_);
59 252733 }
60
61
62 //------------------------------------------------------------------------------
63
64
65 322288 ChunkItem::ChunkItem(FileItem *file_item, uint64_t offset)
66 322130 : file_item_(file_item)
67 322130 , offset_(offset)
68 322130 , size_(0)
69 322130 , is_bulk_chunk_(false)
70 322130 , upload_handle_(NULL)
71
1/2
✓ Branch 4 taken 322272 times.
✗ Branch 5 not taken.
 322288 , compressor_(NULL) {
72 322272 hash_ctx_.algorithm = file_item->hash_algorithm();
73
1/2
✓ Branch 1 taken 322115 times.
✗ Branch 2 not taken.
 322154 hash_ctx_.size = shash::GetContextSize(hash_ctx_.algorithm);
74 322115 hash_ctx_.buffer = hash_ctx_buffer_;
75
1/2
✓ Branch 1 taken 322118 times.
✗ Branch 2 not taken.
 322115 shash::Init(hash_ctx_);
76 322118 hash_value_.algorithm = hash_ctx_.algorithm;
77 322118 hash_value_.suffix = shash::kSuffixPartial;
78 322118 file_item_->IncNchunksInFly();
79 322709 }
80
81
82 251237 void ChunkItem::MakeBulkChunk() {
83 251237 is_bulk_chunk_ = true;
84 251237 hash_value_.suffix = file_item_->hash_suffix();
85 251142 }
86
87
88 586811 zlib::Compressor *ChunkItem::GetCompressor() {
89
2/2
✓ Branch 1 taken 251017 times.
✓ Branch 2 taken 336294 times.
 586811 if (!compressor_.IsValid()) {
90 compressor_ = zlib::Compressor::Construct(
91
2/4
✓ Branch 2 taken 251874 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 251758 times.
✗ Branch 6 not taken.
 251017 file_item_->compression_algorithm());
92 }
93 588052 return compressor_.weak_ref();
94 }
95
96
97 251624 void ChunkItem::ReleaseCompressor() { compressor_.Destroy(); }
98
99
100 //------------------------------------------------------------------------------
101
102 atomic_int64 BlockItem::managed_bytes_ = 0;
103
104
105 104112 BlockItem::BlockItem(ItemAllocator *allocator)
106 104112 : allocator_(allocator)
107 104112 , type_(kBlockHollow)
108 104112 , tag_(-1)
109 104112 , file_item_(NULL)
110 104112 , chunk_item_(NULL)
111 104112 , data_(NULL)
112 104112 , capacity_(0)
113 104112 , size_(0) { }
114
115
116 2649717 BlockItem::BlockItem(int64_t tag, ItemAllocator *allocator)
117 2638969 : allocator_(allocator)
118 2638969 , type_(kBlockHollow)
119 2638969 , tag_(tag)
120 2638969 , file_item_(NULL)
121 2638969 , chunk_item_(NULL)
122 2638969 , data_(NULL)
123 2638969 , capacity_(0)
124 2649717 , size_(0) {
125
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2638969 times.
 2638969 assert(tag_ >= 0);
126 2638969 }
127
128
129 2743327 BlockItem::~BlockItem() {
130
2/2
✓ Branch 0 taken 1429186 times.
✓ Branch 1 taken 1314141 times.
 2743327 if (data_)
131 1429186 allocator_->Free(data_);
132 2744423 atomic_xadd64(&managed_bytes_, -static_cast<int64_t>(capacity_));
133 2762702 }
134
135
136 277232 void BlockItem::Discharge() {
137 277232 data_ = NULL;
138 277232 size_ = capacity_ = 0;
139 277232 }
140
141
142 821749 void BlockItem::MakeStop() {
143
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 821749 times.
 821749 assert(type_ == kBlockHollow);
144 821749 type_ = kBlockStop;
145 821749 }
146
147
148 419936 void BlockItem::MakeData(uint32_t capacity) {
149
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 419936 times.
 419936 assert(type_ == kBlockHollow);
150
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 419936 times.
 419936 assert(allocator_ != NULL);
151
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 419936 times.
 419936 assert(capacity > 0);
152
153 419936 type_ = kBlockData;
154 419936 capacity_ = capacity;
155 419936 data_ = reinterpret_cast<unsigned char *>(allocator_->Malloc(capacity_));
156 421138 atomic_xadd64(&managed_bytes_, static_cast<int64_t>(capacity_));
157 421306 }
158
159
160 /**
161 * Move data from one block to another.
162 */
163 277143 void BlockItem::MakeDataMove(BlockItem *other) {
164
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 277143 times.
 277143 assert(type_ == kBlockHollow);
165
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 277143 times.
 277143 assert(other->type_ == kBlockData);
166
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 277143 times.
 277143 assert(other->size_ > 0);
167
168 277143 type_ = kBlockData;
169 277143 capacity_ = size_ = other->size_;
170 277143 data_ = other->data_;
171 277143 allocator_ = other->allocator_;
172
173 277143 other->Discharge();
174 277134 }
175
176
177 /**
178 * Copy a piece of one block's data into a new block.
179 */
180 1175177 void BlockItem::MakeDataCopy(const unsigned char *data, uint32_t size) {
181
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1175177 times.
 1175177 assert(type_ == kBlockHollow);
182
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1175177 times.
 1175177 assert(allocator_ != NULL);
183
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1175177 times.
 1175177 assert(size > 0);
184
185 1175177 type_ = kBlockData;
186 1175177 capacity_ = size_ = size;
187 1175177 data_ = reinterpret_cast<unsigned char *>(allocator_->Malloc(capacity_));
188 1176993 memcpy(data_, data, size);
189 1176993 atomic_xadd64(&managed_bytes_, static_cast<int64_t>(capacity_));
190 1176123 }
191
192
193 167605 void BlockItem::Reset() {
194
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 167605 times.
 167605 assert(type_ == kBlockData);
195
196 167605 atomic_xadd64(&managed_bytes_, -static_cast<int64_t>(capacity_));
197 167612 allocator_->Free(data_);
198 167608 data_ = NULL;
199 167608 size_ = capacity_ = 0;
200 167608 type_ = kBlockHollow;
201 167608 }
202
203
204 1954915 void BlockItem::SetChunkItem(ChunkItem *value) {
205
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1954915 times.
 1954915 assert(value != NULL);
206
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1954915 times.
 1954915 assert(chunk_item_ == NULL);
207 1954915 chunk_item_ = value;
208 1954915 }
209
210
211 2632796 void BlockItem::SetFileItem(FileItem *value) {
212
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2632796 times.
 2632796 assert(value != NULL);
213
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2632796 times.
 2632796 assert(file_item_ == NULL);
214 2632796 file_item_ = value;
215 2632796 }
216
217
218 7168 uint32_t BlockItem::Write(void *buf, uint32_t count) {
219
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7168 times.
 7168 assert(type_ == kBlockData);
220
221 7168 const uint32_t remaining = capacity_ - size_;
222 7168 const uint32_t nbytes = std::min(remaining, count);
223 7168 memcpy(data_ + size_, buf, nbytes);
224 7168 size_ += nbytes;
225 7168 return nbytes;
226 }
227