GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/util/file_backed_buffer.cc
Date: 2026-04-12 02:41:08
Exec Total Coverage
Lines: 101 110 91.8%
Branches: 52 78 66.7%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 */
4
5 #include "file_backed_buffer.h"
6
7 #include <cassert>
8 #include <cstdint>
9 #include <cstdio>
10 #include <cstdlib>
11 #include <cstring>
12 #include <string>
13 #include <unistd.h>
14
15 #include "util/exception.h"
16 #include "util/logging.h"
17 #include "util/posix.h"
18 #include "util/smalloc.h"
19
20 27956 FileBackedBuffer *FileBackedBuffer::Create(uint64_t in_memory_threshold,
21 const std::string &tmp_dir) {
22
1/2
✓ Branch 2 taken 27956 times.
✗ Branch 3 not taken.
 27956 return new FileBackedBuffer(in_memory_threshold, tmp_dir);
23 }
24
25 27956 FileBackedBuffer::FileBackedBuffer(uint64_t in_memory_threshold,
26 27956 const std::string &tmp_dir)
27 27956 : in_memory_threshold_(in_memory_threshold)
28 27956 , tmp_dir_(tmp_dir)
29 27956 , state_(kWriteState)
30 27956 , mode_(kMemoryMode)
31 27956 , size_(0)
32 27956 , buf_(NULL)
33 27956 , pos_(0)
34 27956 , fp_(NULL)
35
1/2
✓ Branch 2 taken 27956 times.
✗ Branch 3 not taken.
 27956 , file_path_("")
36 55912 , mmapped_(NULL) { }
37
38 55912 FileBackedBuffer::~FileBackedBuffer() {
39 27956 free(buf_);
40
2/2
✓ Branch 0 taken 16481 times.
✓ Branch 1 taken 11475 times.
 27956 if (mode_ != kFileMode)
41 16481 return;
42
43
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 if (state_ == kWriteState) {
44 ✗ const int retval = fclose(fp_);
45 ✗ if (retval != 0)
46 ✗ PANIC(kLogStderr, "could not close temporary file %s: error %d",
47 file_path_.c_str(), retval);
48 } else {
49 11475 mmapped_->Unmap();
50
1/2
✓ Branch 0 taken 11475 times.
✗ Branch 1 not taken.
 11475 delete mmapped_;
51 }
52 11475 const int retval = unlink(file_path_.c_str());
53
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 if (retval != 0)
54 ✗ PANIC(kLogStderr, "could not delete temporary file %s: error %d",
55 file_path_.c_str(), retval);
56
4/4
✓ Branch 1 taken 11475 times.
✓ Branch 2 taken 16481 times.
✓ Branch 4 taken 11475 times.
✓ Branch 5 taken 16481 times.
 44437 }
57
58 8984115 void FileBackedBuffer::Append(const void *source, uint64_t len) {
59
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8984115 times.
 8984115 assert(source != NULL);
60
61 // Cannot write after Commit()
62
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8984115 times.
 8984115 assert(state_ == kWriteState);
63
64
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 8984070 times.
 8984115 if (len == 0)
65 45 return;
66
67 // check the size and eventually save to file
68
4/4
✓ Branch 0 taken 1064880 times.
✓ Branch 1 taken 7919190 times.
✓ Branch 2 taken 11475 times.
✓ Branch 3 taken 1053405 times.
 8984070 if (mode_ == kMemoryMode && pos_ + len > in_memory_threshold_) {
69 // changes mode_ to kFileMode
70 11475 SaveToFile();
71 }
72
73
2/2
✓ Branch 0 taken 1053405 times.
✓ Branch 1 taken 7930710 times.
 8984115 if (mode_ == kMemoryMode) {
74
2/2
✓ Branch 0 taken 19125 times.
✓ Branch 1 taken 1034280 times.
 1053405 if (buf_ == NULL) {
75
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 19125 times.
 19125 assert(size_ == 0);
76
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 19125 times.
 19125 assert(pos_ == 0);
77 19125 buf_ = reinterpret_cast<unsigned char *>(smalloc(len));
78 19125 size_ = len;
79
2/2
✓ Branch 0 taken 87435 times.
✓ Branch 1 taken 946845 times.
 1034280 } else if (size_ < pos_ + len) {
80 87435 const uint64_t newsize = (size_ * 2 < pos_ + len) ? (pos_ + len)
81 : (size_ * 2);
82 87435 buf_ = reinterpret_cast<unsigned char *>(srealloc(buf_, newsize));
83 87435 size_ = newsize;
84 }
85
86 1053405 memcpy(buf_ + pos_, source, len);
87 1053405 pos_ += len;
88 } else {
89
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7930710 times.
 7930710 assert(fp_ != NULL);
90 7930710 const uint64_t bytes_written = fwrite(source, 1, len, fp_);
91
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7930710 times.
 7930710 if (bytes_written != len) {
92 ✗ PANIC(kLogStderr,
93 "could not append to temporary file %s: length %lu, "
94 "actually written %lu, error %d",
95 file_path_.c_str(), len, bytes_written, ferror(fp_));
96 }
97 7930710 pos_ += len;
98 7930710 size_ += len;
99 }
100 }
101
102 27720 void FileBackedBuffer::Commit() {
103
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27720 times.
 27720 assert(state_ == kWriteState);
104
105
2/2
✓ Branch 0 taken 11475 times.
✓ Branch 1 taken 16245 times.
 27720 if (mode_ == kFileMode) {
106 11475 const int retval = fclose(fp_);
107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 if (retval != 0)
108 ✗ PANIC(kLogStderr, "could not close file after writing finished: %s",
109 file_path_.c_str());
110 11475 fp_ = NULL;
111
112
1/2
✓ Branch 2 taken 11475 times.
✗ Branch 3 not taken.
 11475 mmapped_ = new MemoryMappedFile(file_path_);
113
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 11475 times.
 11475 if (!mmapped_->Map())
114 ✗ PANIC(kLogStderr, "could not memory-map file %s", file_path_.c_str());
115 } else {
116 // Trim memory chunk size to actual data size
117 16245 buf_ = reinterpret_cast<unsigned char *>(srealloc(buf_, pos_));
118 16245 size_ = pos_;
119 }
120
121 27720 pos_ = 0;
122 27720 state_ = kReadState;
123 27720 }
124
125 733185 int64_t FileBackedBuffer::Data(void **ptr, int64_t len, uint64_t pos) {
126
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 733185 times.
 733185 assert(state_ == kReadState);
127
128 1466370 const int64_t actual_len = (pos + len <= size_)
129
2/2
✓ Branch 0 taken 12825 times.
✓ Branch 1 taken 720360 times.
 733185 ? len
130 12825 : static_cast<int64_t>(size_)
131 12825 - static_cast<int64_t>(pos);
132
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 733185 times.
 733185 assert(actual_len >= 0);
133
134
2/2
✓ Branch 0 taken 27180 times.
✓ Branch 1 taken 706005 times.
 733185 if (mode_ == kMemoryMode) {
135 27180 *ptr = buf_ + pos;
136 } else {
137 706005 *ptr = mmapped_->buffer() + pos;
138 }
139 733185 return actual_len;
140 }
141
142 705510 int64_t FileBackedBuffer::Read(void *ptr, int64_t len) {
143 705510 const int64_t bytes_read = ReadP(ptr, len, pos_);
144 705510 pos_ += bytes_read;
145 705510 return bytes_read;
146 }
147
148 705510 int64_t FileBackedBuffer::ReadP(void *ptr, int64_t len, uint64_t pos) {
149 void *source;
150
1/2
✓ Branch 1 taken 705510 times.
✗ Branch 2 not taken.
 705510 const int64_t bytes_read = Data(&source, len, pos);
151 705510 memcpy(ptr, source, bytes_read);
152 705510 return bytes_read;
153 }
154
155 27360 void FileBackedBuffer::Rewind() {
156
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27360 times.
 27360 assert(state_ == kReadState);
157 27360 pos_ = 0;
158 27360 }
159
160 142200 uint64_t FileBackedBuffer::GetSize() const {
161
4/4
✓ Branch 0 taken 360 times.
✓ Branch 1 taken 141840 times.
✓ Branch 2 taken 315 times.
✓ Branch 3 taken 45 times.
 142200 if (state_ == kWriteState && mode_ == kMemoryMode)
162 315 return pos_;
163 141885 return size_;
164 }
165
166
167 11475 void FileBackedBuffer::SaveToFile() {
168
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 assert(state_ == kWriteState);
169
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 assert(mode_ == kMemoryMode);
170
171
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 assert(fp_ == NULL);
172 11475 fp_ = CreateTempFile(tmp_dir_, 0644, "w", &file_path_);
173
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 if (fp_ == NULL)
174 ✗ PANIC(kLogStderr, "could not create temporary file");
175
176 11475 const uint64_t bytes_written = fwrite(buf_, 1, pos_, fp_);
177
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11475 times.
 11475 if (bytes_written != pos_) {
178 ✗ PANIC(kLogStderr,
179 "could not write to temporary file %s: length %lu, "
180 "actually written %lu, error %d",
181 file_path_.c_str(), pos_, bytes_written, ferror(fp_));
182 }
183
184 11475 free(buf_);
185 11475 buf_ = NULL;
186 11475 size_ = pos_;
187 11475 mode_ = kFileMode;
188 11475 }
189