| 155 | } |
| 156 | |
| 157 | bool String::reserve(size_t size) |
| 158 | { |
| 159 | // Can we use SSO here to avoid allocation? |
| 160 | if(size <= SSO_CAPACITY) { |
| 161 | // If already using SSO then no further action required |
| 162 | if(sso.set) { |
| 163 | return true; |
| 164 | } |
| 165 | |
| 166 | // If heap hasn't been used yet then switch to SSO mode |
| 167 | if(ptr.buffer == nullptr) { |
| 168 | sso.set = true; |
| 169 | } else { |
| 170 | // Otherwise continue with existing heap allocation |
| 171 | assert(ptr.capacity >= size); |
| 172 | } |
| 173 | |
| 174 | return true; |
| 175 | } |
| 176 | |
| 177 | // Reallocation required? |
| 178 | if(!sso.set && ptr.buffer != nullptr && ptr.capacity >= size) { |
| 179 | return true; // Nope :-) |
| 180 | } |
| 181 | |
| 182 | // Need to handle resizing an existing heap buffer and moving from SSO to heap |
| 183 | char* newbuffer = (char*)realloc(sso.set ? nullptr : ptr.buffer, size + 1); |
| 184 | if(newbuffer == nullptr) { |
| 185 | // allocation failed - leave existing buffer arrangement unchanged |
| 186 | return false; |
| 187 | } |
| 188 | |
| 189 | if(sso.set) { |
| 190 | // Move content out of SSO |
| 191 | memcpy(newbuffer, sso.buffer, sso.len); |
| 192 | ptr.len = sso.len; |
| 193 | sso.set = false; |
| 194 | } |
| 195 | ptr.buffer = newbuffer; |
| 196 | ptr.capacity = size; |
| 197 | return true; |
| 198 | } |
| 199 | |
| 200 | String::Buffer String::getBuffer() |
| 201 | { |