godotengine
/
godot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure COWData does not reallocate on push back, fixes #22561

This commit is contained in:
Juan Linietsky 2020-03-27 21:52:42 -03:00
parent fcfffd7297
commit 0c24a844ec
1 changed files with 24 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
core/cowdata.h
View File

@ -253,7 +253,9 @@ Error CowData<T>::resize(int p_size) {
ERR_FAIL_COND_V(p_size < 0, ERR_INVALID_PARAMETER); ERR_FAIL_COND_V(p_size < 0, ERR_INVALID_PARAMETER);
if (p_size == size()) int current_size = size();
if (p_size == current_size)
return OK; return OK;
if (p_size == 0) { if (p_size == 0) {
@ -266,24 +268,27 @@ Error CowData<T>::resize(int p_size) {
// possibly changing size, copy on write // possibly changing size, copy on write
_copy_on_write(); _copy_on_write();
size_t current_alloc_size = _get_alloc_size(current_size);
size_t alloc_size; size_t alloc_size;
ERR_FAIL_COND_V(!_get_alloc_size_checked(p_size, &alloc_size), ERR_OUT_OF_MEMORY); ERR_FAIL_COND_V(!_get_alloc_size_checked(p_size, &alloc_size), ERR_OUT_OF_MEMORY);
if (p_size > size()) { if (p_size > current_size) {
if (size() == 0) { if (alloc_size != current_alloc_size) {
// alloc from scratch if (current_size == 0) {
uint32_t *ptr = (uint32_t *)Memory::alloc_static(alloc_size, true); // alloc from scratch
ERR_FAIL_COND_V(!ptr, ERR_OUT_OF_MEMORY); uint32_t *ptr = (uint32_t *)Memory::alloc_static(alloc_size, true);
*(ptr - 1) = 0; //size, currently none ERR_FAIL_COND_V(!ptr, ERR_OUT_OF_MEMORY);
*(ptr - 2) = 1; //refcount *(ptr - 1) = 0; //size, currently none
*(ptr - 2) = 1; //refcount
_ptr = (T *)ptr; _ptr = (T *)ptr;
} else { } else {
void *_ptrnew = (T *)Memory::realloc_static(_ptr, alloc_size, true); void *_ptrnew = (T *)Memory::realloc_static(_ptr, alloc_size, true);
ERR_FAIL_COND_V(!_ptrnew, ERR_OUT_OF_MEMORY); ERR_FAIL_COND_V(!_ptrnew, ERR_OUT_OF_MEMORY);
_ptr = (T *)(_ptrnew); _ptr = (T *)(_ptrnew);
}
} }
// construct the newly created elements // construct the newly created elements
@ -298,7 +303,7 @@ Error CowData<T>::resize(int p_size) {
*_get_size() = p_size; *_get_size() = p_size;
} else if (p_size < size()) { } else if (p_size < current_size) {
if (!__has_trivial_destructor(T)) { if (!__has_trivial_destructor(T)) {
// deinitialize no longer needed elements // deinitialize no longer needed elements
@ -308,10 +313,12 @@ Error CowData<T>::resize(int p_size) {
} }
} }
void *_ptrnew = (T *)Memory::realloc_static(_ptr, alloc_size, true); if (alloc_size != current_alloc_size) {
ERR_FAIL_COND_V(!_ptrnew, ERR_OUT_OF_MEMORY); void *_ptrnew = (T *)Memory::realloc_static(_ptr, alloc_size, true);
ERR_FAIL_COND_V(!_ptrnew, ERR_OUT_OF_MEMORY);
_ptr = (T *)(_ptrnew); _ptr = (T *)(_ptrnew);
}
*_get_size() = p_size; *_get_size() = p_size;
} }