diff options
Diffstat (limited to 'test/scudo/realloc.cpp')
-rw-r--r-- | test/scudo/realloc.cpp | 90 |
1 files changed, 48 insertions, 42 deletions
diff --git a/test/scudo/realloc.cpp b/test/scudo/realloc.cpp index 2a7d5b69f5f2..cc44595001f4 100644 --- a/test/scudo/realloc.cpp +++ b/test/scudo/realloc.cpp @@ -14,54 +14,60 @@ #include <malloc.h> #include <string.h> +#include <vector> + int main(int argc, char **argv) { void *p, *old_p; - size_t size = 32; + // Those sizes will exercise both allocators (Primary & Secondary). + std::vector<size_t> sizes{1, 16, 1024, 32768, 1 << 16, 1 << 17, 1 << 20}; assert(argc == 2); - if (!strcmp(argv[1], "pointers")) { - old_p = p = realloc(nullptr, size); - if (!p) - return 1; - size = malloc_usable_size(p); - // Our realloc implementation will return the same pointer if the size - // requested is lower or equal to the usable size of the associated chunk. - p = realloc(p, size - 1); - if (p != old_p) - return 1; - p = realloc(p, size); - if (p != old_p) - return 1; - // And a new one if the size is greater. - p = realloc(p, size + 1); - if (p == old_p) - return 1; - // A size of 0 will free the chunk and return nullptr. - p = realloc(p, 0); - if (p) - return 1; - old_p = nullptr; - } - if (!strcmp(argv[1], "contents")) { - p = realloc(nullptr, size); - if (!p) - return 1; - for (int i = 0; i < size; i++) - reinterpret_cast<char *>(p)[i] = 'A'; - p = realloc(p, size + 1); - // The contents of the reallocated chunk must match the original one. - for (int i = 0; i < size; i++) - if (reinterpret_cast<char *>(p)[i] != 'A') + for (size_t size : sizes) { + if (!strcmp(argv[1], "pointers")) { + old_p = p = realloc(nullptr, size); + if (!p) return 1; - } - if (!strcmp(argv[1], "memalign")) { - // A chunk coming from memalign cannot be reallocated. - p = memalign(16, size); - if (!p) - return 1; - p = realloc(p, size); - free(p); + size = malloc_usable_size(p); + // Our realloc implementation will return the same pointer if the size + // requested is lower than or equal to the usable size of the associated + // chunk. + p = realloc(p, size - 1); + if (p != old_p) + return 1; + p = realloc(p, size); + if (p != old_p) + return 1; + // And a new one if the size is greater. + p = realloc(p, size + 1); + if (p == old_p) + return 1; + // A size of 0 will free the chunk and return nullptr. + p = realloc(p, 0); + if (p) + return 1; + old_p = nullptr; + } + if (!strcmp(argv[1], "contents")) { + p = realloc(nullptr, size); + if (!p) + return 1; + for (int i = 0; i < size; i++) + reinterpret_cast<char *>(p)[i] = 'A'; + p = realloc(p, size + 1); + // The contents of the reallocated chunk must match the original one. + for (int i = 0; i < size; i++) + if (reinterpret_cast<char *>(p)[i] != 'A') + return 1; + } + if (!strcmp(argv[1], "memalign")) { + // A chunk coming from memalign cannot be reallocated. + p = memalign(16, size); + if (!p) + return 1; + p = realloc(p, size); + free(p); + } } return 0; } |