This "fixes" PR64535 by changing the fixed object size emergency pool to a variable EH object size (but fixed arena size) allocator. Via combining the dependent and non-dependent EH arenas this should allow around 600 bad_alloc throws in OOM situations on x86_64-linux compared to the current 64 which should provide some headroom to the poor souls using EH to communicate OOM in a heavily threaded enviroment.
Bootstrapped and tested on x86_64-unknown-linux-gnu (with the #if 1 as in the patch below, forcing the use of the allocator). Comments? Ok with only the #else path retained? What about the buffer size - we're now free to choose sth that doesn't depend on the size of INT_MAX (previously required for old allocator bitmap)? With the cost of some more members I can make the allocator more generic (use a constructor with a arena and a arena size parameter) and we may move it somewhere public under __gnu_cxx? But eventually boost has something like this anyway. Thanks, Richard. 2015-01-12 Richard Biener <rguent...@suse.de> PR libstdc++/64535 * libsupc++/eh_alloc.cc: Include new. (bitmask_type): Remove. (one_buffer): Likewise. (emergency_buffer): Likewise. (emergency_used): Likewise. (dependents_buffer): Likewise. (dependents_used): Likewise. (class pool): New custom fixed-size arena, variable size object allocator. (emergency_pool): New global. (__cxxabiv1::__cxa_allocate_exception): Use new emergency_pool. (__cxxabiv1::__cxa_free_exception): Likewise. (__cxxabiv1::__cxa_allocate_dependent_exception): Likewise. (__cxxabiv1::__cxa_free_dependent_exception): Likewise. Index: libstdc++-v3/libsupc++/eh_alloc.cc =================================================================== --- libstdc++-v3/libsupc++/eh_alloc.cc (revision 216303) +++ libstdc++-v3/libsupc++/eh_alloc.cc (working copy) @@ -34,6 +34,7 @@ #include <exception> #include "unwind-cxx.h" #include <ext/concurrence.h> +#include <new> #if _GLIBCXX_HOSTED using std::free; @@ -72,62 +73,176 @@ using namespace __cxxabiv1; # define EMERGENCY_OBJ_COUNT 4 #endif -#if INT_MAX == 32767 || EMERGENCY_OBJ_COUNT <= 32 -typedef unsigned int bitmask_type; -#else -#if defined (_GLIBCXX_LLP64) -typedef unsigned long long bitmask_type; -#else -typedef unsigned long bitmask_type; -#endif -#endif - - -typedef char one_buffer[EMERGENCY_OBJ_SIZE] __attribute__((aligned)); -static one_buffer emergency_buffer[EMERGENCY_OBJ_COUNT]; -static bitmask_type emergency_used; - -static __cxa_dependent_exception dependents_buffer[EMERGENCY_OBJ_COUNT]; -static bitmask_type dependents_used; namespace { // A single mutex controlling emergency allocations. __gnu_cxx::__mutex emergency_mutex; -} -extern "C" void * -__cxxabiv1::__cxa_allocate_exception(std::size_t thrown_size) _GLIBCXX_NOTHROW -{ - void *ret; + // A fixed-size heap, variable size object allocator + class pool + { + public: + pool(); - thrown_size += sizeof (__cxa_refcounted_exception); - ret = malloc (thrown_size); + void *allocate (size_t); + void free (void *); + + bool in_pool (void *); + + private: + struct free_entry { + size_t size; + free_entry *next; + }; + struct allocated_entry { + size_t size; + char data[]; + }; + free_entry *first_free_entry; + char arena[EMERGENCY_OBJ_SIZE * EMERGENCY_OBJ_COUNT + + EMERGENCY_OBJ_COUNT * sizeof (__cxa_dependent_exception)] + __attribute__((aligned(__alignof__(free_entry)))); + }; - if (! ret) + pool::pool() { - __gnu_cxx::__scoped_lock sentry(emergency_mutex); + first_free_entry = reinterpret_cast <free_entry *> (arena); + new (first_free_entry) free_entry; + first_free_entry->size = EMERGENCY_OBJ_SIZE * EMERGENCY_OBJ_COUNT; + first_free_entry->next = NULL; + } - bitmask_type used = emergency_used; - unsigned int which = 0; + void *pool::allocate (size_t size) + { + __gnu_cxx::__scoped_lock sentry(emergency_mutex); + /* We need an additional size_t member. */ + size += sizeof (size_t); + /* And we need to at least hand out objects of the size of + a freelist entry. */ + if (size < sizeof (free_entry)) + size = sizeof (free_entry); + /* And we need to align objects we hand out to the required + alignment of a freelist entry (this really aligns the + tail which will become a new freelist entry). */ + size = ((size + __alignof__(free_entry) - 1) + & ~(__alignof__(free_entry) - 1)); + /* Search for an entry of proper size on the freelist. */ + free_entry **e; + for (e = &first_free_entry; + *e && (*e)->size < size; + e = &(*e)->next) + ; + if (!*e) + return NULL; + allocated_entry *x; + if ((*e)->size - size >= sizeof (free_entry)) + { + /* Slit block if it is too large. */ + free_entry *f = reinterpret_cast <free_entry *> + (reinterpret_cast <char *> (*e) + size); + size_t sz = (*e)->size; + free_entry *next = (*e)->next; + new (f) free_entry; + f->next = next; + f->size = sz - size; + x = reinterpret_cast <allocated_entry *> (*e); + new (x) allocated_entry; + x->size = size; + *e = f; + } + else + { + /* Exact size match or too small overhead for a free entry. */ + size_t sz = (*e)->size; + free_entry *next = (*e)->next; + x = reinterpret_cast <allocated_entry *> (*e); + new (x) allocated_entry; + x->size = sz; + *e = next; + } + return &x->data; + } - if (thrown_size > EMERGENCY_OBJ_SIZE) - goto failed; - while (used & 1) + void pool::free (void *data) + { + __gnu_cxx::__scoped_lock sentry(emergency_mutex); + allocated_entry *e = reinterpret_cast <allocated_entry *> + (reinterpret_cast <char *> (data) - sizeof (size_t)); + size_t sz = e->size; + if (!first_free_entry) + { + /* If the free list is empty just put the entry there. */ + free_entry *f = reinterpret_cast <free_entry *> (e); + new (f) free_entry; + f->size = sz; + f->next = NULL; + first_free_entry = f; + } + else if (reinterpret_cast <char *> (e) + sz + == reinterpret_cast <char *> (first_free_entry)) { - used >>= 1; - if (++which >= EMERGENCY_OBJ_COUNT) - goto failed; + /* Check if we can merge with the first free entry being right + after us. */ + free_entry *f = reinterpret_cast <free_entry *> (e); + new (f) free_entry; + f->size = sz + first_free_entry->size; + f->next = first_free_entry->next; + first_free_entry = f; } + else + { + /* Else search for a free item we can merge with at its end. */ + free_entry **fe; + for (fe = &first_free_entry; + (*fe)->next + && (reinterpret_cast <char *> ((*fe)->next) + > reinterpret_cast <char *> (e) + sz); + fe = &(*fe)->next) + ; + if (reinterpret_cast <char *> (*fe) + (*fe)->size + == reinterpret_cast <char *> (e)) + /* Merge with the freelist entry. */ + (*fe)->size += sz; + else + { + /* Else put it after it which keeps the freelist sorted. */ + free_entry *f = reinterpret_cast <free_entry *> (e); + new (f) free_entry; + f->size = sz; + f->next = (*fe)->next; + (*fe)->next = f; + } + } + } + + bool pool::in_pool (void *ptr) + { + char *p = reinterpret_cast <char *> (ptr); + return (p > arena + && p < arena + sizeof (arena)); + } - emergency_used |= (bitmask_type)1 << which; - ret = &emergency_buffer[which][0]; + pool emergency_pool; +} - failed:; +extern "C" void * +__cxxabiv1::__cxa_allocate_exception(std::size_t thrown_size) _GLIBCXX_NOTHROW +{ + void *ret; - if (!ret) - std::terminate (); - } + thrown_size += sizeof (__cxa_refcounted_exception); +#if 1 + ret = emergency_pool.allocate (thrown_size); +#else + ret = malloc (thrown_size); + + if (!ret) + ret = emergency_pool.allocate (thrown_size); +#endif + + if (!ret) + std::terminate (); memset (ret, 0, sizeof (__cxa_refcounted_exception)); @@ -138,19 +253,11 @@ __cxxabiv1::__cxa_allocate_exception(std extern "C" void __cxxabiv1::__cxa_free_exception(void *vptr) _GLIBCXX_NOTHROW { - char *base = (char *) emergency_buffer; - char *ptr = (char *) vptr; - if (ptr >= base - && ptr < base + sizeof (emergency_buffer)) - { - const unsigned int which - = (unsigned) (ptr - base) / EMERGENCY_OBJ_SIZE; - - __gnu_cxx::__scoped_lock sentry(emergency_mutex); - emergency_used &= ~((bitmask_type)1 << which); - } + char *ptr = (char *) vptr - sizeof (__cxa_refcounted_exception); + if (emergency_pool.in_pool (ptr)) + emergency_pool.free (ptr); else - free (ptr - sizeof (__cxa_refcounted_exception)); + free (ptr); } @@ -163,27 +270,11 @@ __cxxabiv1::__cxa_allocate_dependent_exc (malloc (sizeof (__cxa_dependent_exception))); if (!ret) - { - __gnu_cxx::__scoped_lock sentry(emergency_mutex); - - bitmask_type used = dependents_used; - unsigned int which = 0; - - while (used & 1) - { - used >>= 1; - if (++which >= EMERGENCY_OBJ_COUNT) - goto failed; - } + ret = static_cast <__cxa_dependent_exception*> + (emergency_pool.allocate (sizeof (__cxa_dependent_exception))); - dependents_used |= (bitmask_type)1 << which; - ret = &dependents_buffer[which]; - - failed:; - - if (!ret) - std::terminate (); - } + if (!ret) + std::terminate (); memset (ret, 0, sizeof (__cxa_dependent_exception)); @@ -195,17 +286,8 @@ extern "C" void __cxxabiv1::__cxa_free_dependent_exception (__cxa_dependent_exception *vptr) _GLIBCXX_NOTHROW { - char *base = (char *) dependents_buffer; - char *ptr = (char *) vptr; - if (ptr >= base - && ptr < base + sizeof (dependents_buffer)) - { - const unsigned int which - = (unsigned) (ptr - base) / sizeof (__cxa_dependent_exception); - - __gnu_cxx::__scoped_lock sentry(emergency_mutex); - dependents_used &= ~((bitmask_type)1 << which); - } + if (emergency_pool.in_pool (vptr)) + emergency_pool.free (vptr); else free (vptr); }