| Directory: | cvmfs/ |
|---|---|
| File: | cvmfs/shortstring.h |
| Date: | 2026-09-20 02:39:58 |
| Exec | Total | Coverage | |
|---|---|---|---|
| Lines: | 96 | 109 | 88.1% |
| Branches: | 40 | 50 | 80.0% |
| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /** | ||
| 2 | * This file is part of the CernVM File System. | ||
| 3 | * | ||
| 4 | * Implements a string class that stores short strings on the stack and | ||
| 5 | * malloc's a std::string on the heap on overflow. Used for file names and | ||
| 6 | * path names that are usually small. | ||
| 7 | */ | ||
| 8 | |||
| 9 | #ifndef CVMFS_SHORTSTRING_H_ | ||
| 10 | #define CVMFS_SHORTSTRING_H_ | ||
| 11 | |||
| 12 | #include <algorithm> | ||
| 13 | #include <cstring> | ||
| 14 | #include <string> | ||
| 15 | |||
| 16 | #include "util/atomic.h" | ||
| 17 | |||
| 18 | #ifdef CVMFS_NAMESPACE_GUARD | ||
| 19 | namespace CVMFS_NAMESPACE_GUARD { | ||
| 20 | #endif | ||
| 21 | |||
| 22 | const unsigned char kDefaultMaxName = 25; | ||
| 23 | const unsigned char kDefaultMaxLink = 25; | ||
| 24 | const unsigned char kDefaultMaxPath = 200; | ||
| 25 | |||
| 26 | template<unsigned char StackSize, char Type> | ||
| 27 | class ShortString { | ||
| 28 | public: | ||
| 29 | 217712402 | ShortString() : long_string_(NULL), length_(0) { | |
| 30 | #ifdef DEBUGMSG | ||
| 31 | 217712402 | atomic_inc64(&num_instances_); | |
| 32 | #endif | ||
| 33 | 217712402 | } | |
| 34 | 2907174 | ShortString(const ShortString &other) : long_string_(NULL) { | |
| 35 | #ifdef DEBUGMSG | ||
| 36 | 5814348 | atomic_inc64(&num_instances_); | |
| 37 | #endif | ||
| 38 | 5821214 | Assign(other); | |
| 39 | 5815250 | } | |
| 40 | 203381 | ShortString(const char *chars, const unsigned length) : long_string_(NULL) { | |
| 41 | #ifdef DEBUGMSG | ||
| 42 | 203381 | atomic_inc64(&num_instances_); | |
| 43 | #endif | ||
| 44 | 205005 | Assign(chars, length); | |
| 45 | 204106 | } | |
| 46 | 21797816 | explicit ShortString(const std::string &std_string) : long_string_(NULL) { | |
| 47 | #ifdef DEBUGMSG | ||
| 48 | 21797816 | atomic_inc64(&num_instances_); | |
| 49 | #endif | ||
| 50 | 22762414 | Assign(std_string.data(), std_string.length()); | |
| 51 | 22126386 | } | |
| 52 | |||
| 53 | 21854076 | ShortString &operator=(const ShortString &other) { | |
| 54 |
1/2✓ Branch 0 taken 10935941 times.
✗ Branch 1 not taken.
|
21854076 | if (this != &other) |
| 55 | 21871882 | Assign(other); | |
| 56 | 21346170 | return *this; | |
| 57 | } | ||
| 58 | |||
| 59 |
2/2✓ Branch 0 taken 48 times.
✓ Branch 1 taken 122540563 times.
|
245081222 | ~ShortString() { delete long_string_; } |
| 60 | |||
| 61 | 49760466 | void Assign(const char *chars, const unsigned length) { | |
| 62 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 24880233 times.
|
49760466 | delete long_string_; |
| 63 | 49407828 | long_string_ = NULL; | |
| 64 | 49407828 | this->length_ = length; | |
| 65 |
2/2✓ Branch 0 taken 48 times.
✓ Branch 1 taken 24703866 times.
|
49407828 | if (length > StackSize) { |
| 66 | #ifdef DEBUGMSG | ||
| 67 | 96 | atomic_inc64(&num_overflows_); | |
| 68 | #endif | ||
| 69 |
2/4✓ Branch 2 taken 48 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 48 times.
✗ Branch 6 not taken.
|
96 | long_string_ = new std::string(chars, length); |
| 70 | } else { | ||
| 71 |
2/2✓ Branch 0 taken 24338615 times.
✓ Branch 1 taken 365251 times.
|
49407732 | if (length) |
| 72 | 48677230 | memcpy(stack_, chars, length); | |
| 73 | } | ||
| 74 | 49407828 | } | |
| 75 | |||
| 76 | 27616400 | void Assign(const ShortString &other) { | |
| 77 | 27616400 | Assign(other.GetChars(), other.GetLength()); | |
| 78 | 27183322 | } | |
| 79 | |||
| 80 | 214738 | void Append(const char *chars, const unsigned length) { | |
| 81 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 214738 times.
|
214738 | if (long_string_) { |
| 82 | ✗ | long_string_->append(chars, length); | |
| 83 | ✗ | return; | |
| 84 | } | ||
| 85 | |||
| 86 | 214738 | const unsigned new_length = this->length_ + length; | |
| 87 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 214738 times.
|
214738 | if (new_length > StackSize) { |
| 88 | #ifdef DEBUGMSG | ||
| 89 | ✗ | atomic_inc64(&num_overflows_); | |
| 90 | #endif | ||
| 91 | ✗ | long_string_ = new std::string(); | |
| 92 | ✗ | long_string_->reserve(new_length); | |
| 93 | ✗ | long_string_->assign(stack_, length_); | |
| 94 | ✗ | long_string_->append(chars, length); | |
| 95 | ✗ | return; | |
| 96 | } | ||
| 97 |
2/2✓ Branch 0 taken 212484 times.
✓ Branch 1 taken 2254 times.
|
214738 | if (length > 0) |
| 98 | 212484 | memcpy(&stack_[this->length_], chars, length); | |
| 99 | 214738 | this->length_ = new_length; | |
| 100 | } | ||
| 101 | |||
| 102 | /** | ||
| 103 | * Truncates the current string to be of size smaller or equal to current size | ||
| 104 | * | ||
| 105 | * Note: Can lead to a heap allocated string that is shorter than | ||
| 106 | * the reserved stack space. | ||
| 107 | */ | ||
| 108 | 3016 | void Truncate(unsigned new_length) { | |
| 109 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 3016 times.
|
3016 | assert(new_length <= this->GetLength()); |
| 110 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3016 times.
|
3016 | if (long_string_) { |
| 111 | ✗ | long_string_->erase(new_length); | |
| 112 | ✗ | return; | |
| 113 | } | ||
| 114 | 3016 | this->length_ = new_length; | |
| 115 | } | ||
| 116 | |||
| 117 | void Clear() { | ||
| 118 | delete long_string_; | ||
| 119 | long_string_ = NULL; | ||
| 120 | length_ = 0; | ||
| 121 | } | ||
| 122 | |||
| 123 | 51892768 | const char *GetChars() const { | |
| 124 |
2/2✓ Branch 0 taken 72 times.
✓ Branch 1 taken 25946312 times.
|
51892768 | if (long_string_) { |
| 125 | 144 | return long_string_->data(); | |
| 126 | } else { | ||
| 127 | 51892624 | return stack_; | |
| 128 | } | ||
| 129 | } | ||
| 130 | |||
| 131 | 52894126 | unsigned GetLength() const { | |
| 132 |
2/2✓ Branch 0 taken 72 times.
✓ Branch 1 taken 26446991 times.
|
52894126 | if (long_string_) |
| 133 | 144 | return long_string_->length(); | |
| 134 | 52893982 | return length_; | |
| 135 | } | ||
| 136 | |||
| 137 | 51984 | bool IsEmpty() const { return GetLength() == 0; } | |
| 138 | |||
| 139 | 11704220 | std::string ToString() const { | |
| 140 |
1/2✓ Branch 4 taken 11702983 times.
✗ Branch 5 not taken.
|
11704220 | return std::string(this->GetChars(), this->GetLength()); |
| 141 | } | ||
| 142 | |||
| 143 | 155233 | const char *c_str() const { | |
| 144 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 101183 times.
|
155233 | if (long_string_) |
| 145 | ✗ | return long_string_->c_str(); | |
| 146 | |||
| 147 | 155233 | char *c = const_cast<char *>(stack_) + length_; | |
| 148 | 155233 | *c = '\0'; | |
| 149 | 155233 | return stack_; | |
| 150 | } | ||
| 151 | |||
| 152 | 42569 | bool operator==(const ShortString &other) const { | |
| 153 | 42569 | const unsigned this_length = this->GetLength(); | |
| 154 | 42569 | const unsigned other_length = other.GetLength(); | |
| 155 |
2/2✓ Branch 0 taken 25639 times.
✓ Branch 1 taken 14125 times.
|
42569 | if (this_length != other_length) |
| 156 | 25824 | return false; | |
| 157 |
2/2✓ Branch 0 taken 8070 times.
✓ Branch 1 taken 6055 times.
|
16745 | if (this_length == 0) |
| 158 | 10179 | return true; | |
| 159 | |||
| 160 | 6566 | return memcmp(this->GetChars(), other.GetChars(), this_length) == 0; | |
| 161 | } | ||
| 162 | |||
| 163 | 28799 | bool operator!=(const ShortString &other) const { return !(*this == other); } | |
| 164 | |||
| 165 | 47254 | bool operator<(const ShortString &other) const { | |
| 166 | 47254 | const unsigned this_length = this->GetLength(); | |
| 167 | 47254 | const unsigned other_length = other.GetLength(); | |
| 168 | |||
| 169 |
2/2✓ Branch 0 taken 11944 times.
✓ Branch 1 taken 35310 times.
|
47254 | if (this_length < other_length) |
| 170 | 11944 | return true; | |
| 171 |
2/2✓ Branch 0 taken 8334 times.
✓ Branch 1 taken 26976 times.
|
35310 | if (this_length > other_length) |
| 172 | 8334 | return false; | |
| 173 | |||
| 174 | 26976 | const char *this_chars = this->GetChars(); | |
| 175 | 26976 | const char *other_chars = other.GetChars(); | |
| 176 |
2/2✓ Branch 0 taken 276152 times.
✓ Branch 1 taken 23482 times.
|
299634 | for (unsigned i = 0; i < this_length; ++i) { |
| 177 |
2/2✓ Branch 0 taken 2696 times.
✓ Branch 1 taken 273456 times.
|
276152 | if (this_chars[i] < other_chars[i]) |
| 178 | 2696 | return true; | |
| 179 |
2/2✓ Branch 0 taken 798 times.
✓ Branch 1 taken 272658 times.
|
273456 | if (this_chars[i] > other_chars[i]) |
| 180 | 798 | return false; | |
| 181 | } | ||
| 182 | 23482 | return false; | |
| 183 | } | ||
| 184 | |||
| 185 | 45005 | bool StartsWith(const ShortString &other) const { | |
| 186 | 45005 | const unsigned this_length = this->GetLength(); | |
| 187 | 45005 | const unsigned other_length = other.GetLength(); | |
| 188 |
2/2✓ Branch 0 taken 2951 times.
✓ Branch 1 taken 42054 times.
|
45005 | if (this_length < other_length) |
| 189 | 2951 | return false; | |
| 190 | |||
| 191 | 42054 | return memcmp(this->GetChars(), other.GetChars(), other_length) == 0; | |
| 192 | } | ||
| 193 | |||
| 194 | 19045 | ShortString Suffix(const unsigned start_at) const { | |
| 195 | 19045 | const unsigned length = this->GetLength(); | |
| 196 |
2/2✓ Branch 0 taken 108 times.
✓ Branch 1 taken 18937 times.
|
19045 | if (start_at >= length) |
| 197 | 108 | return ShortString("", 0); | |
| 198 | |||
| 199 | 18937 | return ShortString(this->GetChars() + start_at, length - start_at); | |
| 200 | } | ||
| 201 | |||
| 202 | ✗ | static uint64_t num_instances() { return atomic_read64(&num_instances_); } | |
| 203 | ✗ | static uint64_t num_overflows() { return atomic_read64(&num_overflows_); } | |
| 204 | |||
| 205 | 192 | operator bool() const { return not IsEmpty(); } | |
| 206 | |||
| 207 | private: | ||
| 208 | std::string *long_string_; | ||
| 209 | char stack_[StackSize + 1]; // +1 to add a final '\0' if necessary | ||
| 210 | unsigned char length_; | ||
| 211 | static atomic_int64 num_overflows_; | ||
| 212 | static atomic_int64 num_instances_; | ||
| 213 | }; // class ShortString | ||
| 214 | |||
| 215 | typedef ShortString<kDefaultMaxPath, 0> PathString; | ||
| 216 | typedef ShortString<kDefaultMaxName, 1> NameString; | ||
| 217 | typedef ShortString<kDefaultMaxLink, 2> LinkString; | ||
| 218 | |||
| 219 | template<unsigned char StackSize, char Type> | ||
| 220 | atomic_int64 ShortString<StackSize, Type>::num_overflows_ = 0; | ||
| 221 | template<unsigned char StackSize, char Type> | ||
| 222 | atomic_int64 ShortString<StackSize, Type>::num_instances_ = 0; | ||
| 223 | |||
| 224 | // See posix.cc for the std::string counterparts | ||
| 225 | PathString GetParentPath(const PathString &path); | ||
| 226 | NameString GetFileName(const PathString &path); | ||
| 227 | |||
| 228 | bool IsSubPath(const PathString &parent, const PathString &path); | ||
| 229 | |||
| 230 | |||
| 231 | #ifdef CVMFS_NAMESPACE_GUARD | ||
| 232 | } // namespace CVMFS_NAMESPACE_GUARD | ||
| 233 | #endif | ||
| 234 | |||
| 235 | #endif // CVMFS_SHORTSTRING_H_ | ||
| 236 | |||
| 237 |