Directory: | cvmfs/ |
---|---|
File: | cvmfs/ingestion/item.cc |
Date: | 2025-06-29 02:35:41 |
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 | 9844372 | 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 | 9844372 | bool has_legacy_bulk_chunk) | |
26 | 9844372 | : source_(source) | |
27 | 9844372 | , compression_algorithm_(compression_algorithm) | |
28 | 9844372 | , hash_algorithm_(hash_algorithm) | |
29 | 9844372 | , hash_suffix_(hash_suffix) | |
30 | 9844372 | , has_legacy_bulk_chunk_(has_legacy_bulk_chunk) | |
31 | 9844372 | , size_(kSizeUnknown) | |
32 | 9844372 | , may_have_chunks_(may_have_chunks) | |
33 |
1/2✓ Branch 1 taken 9844372 times.
✗ Branch 2 not taken.
|
9844372 | , chunk_detector_(min_chunk_size, avg_chunk_size, max_chunk_size) |
34 |
1/2✓ Branch 1 taken 9844372 times.
✗ Branch 2 not taken.
|
9844372 | , bulk_hash_(hash_algorithm) |
35 |
1/2✓ Branch 2 taken 9844372 times.
✗ Branch 3 not taken.
|
19688744 | , chunks_(1) { |
36 | 9844372 | const int retval = pthread_mutex_init(&lock_, NULL); | |
37 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9844372 times.
|
9844372 | assert(retval == 0); |
38 | 9844372 | atomic_init64(&nchunks_in_fly_); | |
39 | 9844372 | atomic_init32(&is_fully_chunked_); | |
40 | 9844372 | } | |
41 | |||
42 | 9838981 | FileItem::~FileItem() { pthread_mutex_destroy(&lock_); } | |
43 | |||
44 | 9769614 | void FileItem::RegisterChunk(const FileChunk &file_chunk) { | |
45 | 9769614 | const MutexLockGuard lock_guard(lock_); | |
46 | |||
47 |
2/2✓ Branch 1 taken 17423 times.
✓ Branch 2 taken 9752191 times.
|
9769614 | switch (file_chunk.content_hash().suffix) { |
48 | 17423 | case shash::kSuffixPartial: | |
49 |
1/2✓ Branch 1 taken 17423 times.
✗ Branch 2 not taken.
|
17423 | chunks_.PushBack(file_chunk); |
50 | 17423 | break; | |
51 | |||
52 | 9752191 | default: | |
53 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 9752191 times.
|
9752191 | assert(file_chunk.offset() == 0); |
54 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 9752191 times.
|
9752191 | assert(file_chunk.size() == size_); |
55 | 9752191 | bulk_hash_ = file_chunk.content_hash(); | |
56 | 9752191 | break; | |
57 | } | ||
58 | 9769614 | atomic_dec64(&nchunks_in_fly_); | |
59 | 9769614 | } | |
60 | |||
61 | |||
62 | //------------------------------------------------------------------------------ | ||
63 | |||
64 | |||
65 | 9984481 | ChunkItem::ChunkItem(FileItem *file_item, uint64_t offset) | |
66 | 9981634 | : file_item_(file_item) | |
67 | 9981634 | , offset_(offset) | |
68 | 9981634 | , size_(0) | |
69 | 9981634 | , is_bulk_chunk_(false) | |
70 | 9981634 | , upload_handle_(NULL) | |
71 |
1/2✓ Branch 4 taken 9986158 times.
✗ Branch 5 not taken.
|
9984481 | , compressor_(NULL) { |
72 | 9986158 | hash_ctx_.algorithm = file_item->hash_algorithm(); | |
73 |
1/2✓ Branch 1 taken 9980893 times.
✗ Branch 2 not taken.
|
9982102 | hash_ctx_.size = shash::GetContextSize(hash_ctx_.algorithm); |
74 | 9980893 | hash_ctx_.buffer = hash_ctx_buffer_; | |
75 |
1/2✓ Branch 1 taken 9981751 times.
✗ Branch 2 not taken.
|
9980893 | shash::Init(hash_ctx_); |
76 | 9981751 | hash_value_.algorithm = hash_ctx_.algorithm; | |
77 | 9981751 | hash_value_.suffix = shash::kSuffixPartial; | |
78 | 9981751 | file_item_->IncNchunksInFly(); | |
79 | 9994075 | } | |
80 | |||
81 | |||
82 | 9734584 | void ChunkItem::MakeBulkChunk() { | |
83 | 9734584 | is_bulk_chunk_ = true; | |
84 | 9734584 | hash_value_.suffix = file_item_->hash_suffix(); | |
85 | 9731620 | } | |
86 | |||
87 | |||
88 | 22545936 | zlib::Compressor *ChunkItem::GetCompressor() { | |
89 |
2/2✓ Branch 1 taken 9733049 times.
✓ Branch 2 taken 12819127 times.
|
22545936 | if (!compressor_.IsValid()) { |
90 | compressor_ = zlib::Compressor::Construct( | ||
91 |
2/4✓ Branch 2 taken 9754226 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 9751613 times.
✗ Branch 6 not taken.
|
9733049 | file_item_->compression_algorithm()); |
92 | } | ||
93 | 22570740 | return compressor_.weak_ref(); | |
94 | } | ||
95 | |||
96 | |||
97 | 9746387 | void ChunkItem::ReleaseCompressor() { compressor_.Destroy(); } | |
98 | |||
99 | |||
100 | //------------------------------------------------------------------------------ | ||
101 | |||
102 | atomic_int64 BlockItem::managed_bytes_ = 0; | ||
103 | |||
104 | |||
105 | 144704 | BlockItem::BlockItem(ItemAllocator *allocator) | |
106 | 144704 | : allocator_(allocator) | |
107 | 144704 | , type_(kBlockHollow) | |
108 | 144704 | , tag_(-1) | |
109 | 144704 | , file_item_(NULL) | |
110 | 144704 | , chunk_item_(NULL) | |
111 | 144704 | , data_(NULL) | |
112 | 144704 | , capacity_(0) | |
113 | 144704 | , size_(0) { } | |
114 | |||
115 | |||
116 | 71466290 | BlockItem::BlockItem(int64_t tag, ItemAllocator *allocator) | |
117 | 71121718 | : allocator_(allocator) | |
118 | 71121718 | , type_(kBlockHollow) | |
119 | 71121718 | , tag_(tag) | |
120 | 71121718 | , file_item_(NULL) | |
121 | 71121718 | , chunk_item_(NULL) | |
122 | 71121718 | , data_(NULL) | |
123 | 71121718 | , capacity_(0) | |
124 | 71466290 | , size_(0) { | |
125 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 71121718 times.
|
71121718 | assert(tag_ >= 0); |
126 | 71121718 | } | |
127 | |||
128 | |||
129 | 71213583 | BlockItem::~BlockItem() { | |
130 |
2/2✓ Branch 0 taken 30224499 times.
✓ Branch 1 taken 40989084 times.
|
71213583 | if (data_) |
131 | 30224499 | allocator_->Free(data_); | |
132 | 71244510 | atomic_xadd64(&managed_bytes_, -static_cast<int64_t>(capacity_)); | |
133 | 71789263 | } | |
134 | |||
135 | |||
136 | 10755992 | void BlockItem::Discharge() { | |
137 | 10755992 | data_ = NULL; | |
138 | 10755992 | size_ = capacity_ = 0; | |
139 | 10755992 | } | |
140 | |||
141 | |||
142 | 29358017 | void BlockItem::MakeStop() { | |
143 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 29358017 times.
|
29358017 | assert(type_ == kBlockHollow); |
144 | 29358017 | type_ = kBlockStop; | |
145 | 29358017 | } | |
146 | |||
147 | |||
148 | 15075638 | void BlockItem::MakeData(uint32_t capacity) { | |
149 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15075638 times.
|
15075638 | assert(type_ == kBlockHollow); |
150 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15075638 times.
|
15075638 | assert(allocator_ != NULL); |
151 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15075638 times.
|
15075638 | assert(capacity > 0); |
152 | |||
153 | 15075638 | type_ = kBlockData; | |
154 | 15075638 | capacity_ = capacity; | |
155 | 15075638 | data_ = reinterpret_cast<unsigned char *>(allocator_->Malloc(capacity_)); | |
156 | 15118031 | atomic_xadd64(&managed_bytes_, static_cast<int64_t>(capacity_)); | |
157 | 15121892 | } | |
158 | |||
159 | |||
160 | /** | ||
161 | * Move data from one block to another. | ||
162 | */ | ||
163 | 10754783 | void BlockItem::MakeDataMove(BlockItem *other) { | |
164 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10754783 times.
|
10754783 | assert(type_ == kBlockHollow); |
165 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10754783 times.
|
10754783 | assert(other->type_ == kBlockData); |
166 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 10754783 times.
|
10754783 | assert(other->size_ > 0); |
167 | |||
168 | 10754783 | type_ = kBlockData; | |
169 | 10754783 | capacity_ = size_ = other->size_; | |
170 | 10754783 | data_ = other->data_; | |
171 | 10754783 | allocator_ = other->allocator_; | |
172 | |||
173 | 10754783 | other->Discharge(); | |
174 | 10754237 | } | |
175 | |||
176 | |||
177 | /** | ||
178 | * Copy a piece of one block's data into a new block. | ||
179 | */ | ||
180 | 16644461 | void BlockItem::MakeDataCopy(const unsigned char *data, uint32_t size) { | |
181 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 16644461 times.
|
16644461 | assert(type_ == kBlockHollow); |
182 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 16644461 times.
|
16644461 | assert(allocator_ != NULL); |
183 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 16644461 times.
|
16644461 | assert(size > 0); |
184 | |||
185 | 16644461 | type_ = kBlockData; | |
186 | 16644461 | capacity_ = size_ = size; | |
187 | 16644461 | data_ = reinterpret_cast<unsigned char *>(allocator_->Malloc(capacity_)); | |
188 | 16672799 | memcpy(data_, data, size); | |
189 | 16672799 | atomic_xadd64(&managed_bytes_, static_cast<int64_t>(capacity_)); | |
190 | 16667533 | } | |
191 | |||
192 | |||
193 | 1528400 | void BlockItem::Reset() { | |
194 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1528400 times.
|
1528400 | assert(type_ == kBlockData); |
195 | |||
196 | 1528400 | atomic_xadd64(&managed_bytes_, -static_cast<int64_t>(capacity_)); | |
197 | 1529024 | allocator_->Free(data_); | |
198 | 1528907 | data_ = NULL; | |
199 | 1528907 | size_ = capacity_ = 0; | |
200 | 1528907 | type_ = kBlockHollow; | |
201 | 1528907 | } | |
202 | |||
203 | |||
204 | 49436030 | void BlockItem::SetChunkItem(ChunkItem *value) { | |
205 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 49436030 times.
|
49436030 | assert(value != NULL); |
206 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 49436030 times.
|
49436030 | assert(chunk_item_ == NULL); |
207 | 49436030 | chunk_item_ = value; | |
208 | 49436030 | } | |
209 | |||
210 | |||
211 | 70965314 | void BlockItem::SetFileItem(FileItem *value) { | |
212 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 70965314 times.
|
70965314 | assert(value != NULL); |
213 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 70965314 times.
|
70965314 | assert(file_item_ == NULL); |
214 | 70965314 | file_item_ = value; | |
215 | 70965314 | } | |
216 | |||
217 | |||
218 | 23552 | uint32_t BlockItem::Write(void *buf, uint32_t count) { | |
219 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 23552 times.
|
23552 | assert(type_ == kBlockData); |
220 | |||
221 | 23552 | const uint32_t remaining = capacity_ - size_; | |
222 | 23552 | const uint32_t nbytes = std::min(remaining, count); | |
223 | 23552 | memcpy(data_ + size_, buf, nbytes); | |
224 | 23552 | size_ += nbytes; | |
225 | 23552 | return nbytes; | |
226 | } | ||
227 |