12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121 |
- /*
- *
- * Copyright (c) 1996,1997
- * Silicon Graphics Computer Systems, Inc.
- *
- * Copyright (c) 1997
- * Moscow Center for SPARC Technology
- *
- * Copyright (c) 1999
- * Boris Fomitchev
- *
- * This material is provided "as is", with absolutely no warranty expressed
- * or implied. Any use is at your own risk.
- *
- * Permission to use or copy this software for any purpose is hereby granted
- * without fee, provided the above notices are retained on all copies.
- * Permission to modify the code and to distribute modified code is granted,
- * provided the above notices are retained, and a notice that the code was
- * modified is included with the above copyright notice.
- *
- */
- #include "stlport_prefix.h"
- #include <memory>
- #if defined (__GNUC__) && (defined (__CYGWIN__) || defined (__MINGW32__))
- # include <malloc.h>
- #endif
- #if defined (_STLP_PTHREADS) && !defined (_STLP_NO_THREADS)
- # include <pthread_alloc>
- # include <cerrno>
- #endif
- #include <stl/_threads.h>
- #include "lock_free_slist.h"
- #if defined (__WATCOMC__)
- # pragma warning 13 9
- # pragma warning 367 9
- # pragma warning 368 9
- #endif
- #if defined (_STLP_SGI_THREADS)
- // We test whether threads are in use before locking.
- // Perhaps this should be moved into stl_threads.h, but that
- // probably makes it harder to avoid the procedure call when
- // it isn't needed.
- extern "C" {
- extern int __us_rsthread_malloc;
- }
- #endif
- // Specialised debug form of new operator which does not provide "false"
- // memory leaks when run with debug CRT libraries.
- #if defined (_STLP_MSVC) && (_STLP_MSVC >= 1020 && defined (_STLP_DEBUG_ALLOC)) && !defined (_STLP_WCE)
- # include <crtdbg.h>
- inline char* __stlp_new_chunk(size_t __bytes) {
- void *__chunk = _STLP_CHECK_NULL_ALLOC(::operator new(__bytes, __FILE__, __LINE__));
- return __STATIC_CAST(char*, __chunk);
- }
- inline void __stlp_delete_chunck(void* __p) { ::operator delete(__p, __FILE__, __LINE__); }
- #else
- # ifdef _STLP_NODE_ALLOC_USE_MALLOC
- # include <cstdlib>
- inline char* __stlp_new_chunk(size_t __bytes) {
- // do not use _STLP_CHECK_NULL_ALLOC, this macro is dedicated to new operator.
- void *__chunk = _STLP_VENDOR_CSTD::malloc(__bytes);
- if (__chunk == 0) {
- _STLP_THROW_BAD_ALLOC;
- }
- return __STATIC_CAST(char*, __chunk);
- }
- inline void __stlp_delete_chunck(void* __p) { _STLP_VENDOR_CSTD::free(__p); }
- # else
- inline char* __stlp_new_chunk(size_t __bytes)
- { return __STATIC_CAST(char*, _STLP_STD::__stl_new(__bytes)); }
- inline void __stlp_delete_chunck(void* __p) { _STLP_STD::__stl_delete(__p); }
- # endif
- #endif
- /* This is an additional atomic operations to the ones already defined in
- * stl/_threads.h, platform should try to support it to improve performance.
- * __add_atomic_t _STLP_ATOMIC_ADD(volatile __add_atomic_t* __target, __add_atomic_t __val) :
- * does *__target = *__target + __val and returns the old *__target value */
- typedef long __add_atomic_t;
- typedef unsigned long __uadd_atomic_t;
- #if defined (__GNUC__) && defined (__i386__)
- inline long _STLP_atomic_add_gcc_x86(long volatile* p, long addend) {
- long result;
- __asm__ __volatile__
- ("lock; xaddl %1, %0;"
- :"=m" (*p), "=r" (result)
- :"m" (*p), "1" (addend)
- :"cc");
- return result + addend;
- }
- # define _STLP_ATOMIC_ADD(__dst, __val) _STLP_atomic_add_gcc_x86(__dst, __val)
- #elif defined (_STLP_WIN32THREADS)
- // The Win32 API function InterlockedExchangeAdd is not available on Windows 95.
- # if !defined (_STLP_WIN95_LIKE)
- # if defined (_STLP_NEW_PLATFORM_SDK)
- # define _STLP_ATOMIC_ADD(__dst, __val) InterlockedExchangeAdd(__dst, __val)
- # else
- # define _STLP_ATOMIC_ADD(__dst, __val) InterlockedExchangeAdd(__CONST_CAST(__add_atomic_t*, __dst), __val)
- # endif
- # endif
- #endif
- #if defined (__OS400__)
- // dums 02/05/2007: is it really necessary ?
- enum { _ALIGN = 16, _ALIGN_SHIFT = 4 };
- #else
- enum { _ALIGN = 2 * sizeof(void*), _ALIGN_SHIFT = 2 + sizeof(void*) / 4 };
- #endif
- #define _S_FREELIST_INDEX(__bytes) ((__bytes - size_t(1)) >> (int)_ALIGN_SHIFT)
- _STLP_BEGIN_NAMESPACE
- // malloc_alloc out-of-memory handling
- static __oom_handler_type __oom_handler = __STATIC_CAST(__oom_handler_type, 0);
- #ifdef _STLP_THREADS
- _STLP_mutex __oom_handler_lock;
- #endif
- void* _STLP_CALL __malloc_alloc::allocate(size_t __n)
- {
- void *__result = malloc(__n);
- if ( 0 == __result ) {
- __oom_handler_type __my_malloc_handler;
- for (;;) {
- {
- #ifdef _STLP_THREADS
- _STLP_auto_lock _l( __oom_handler_lock );
- #endif
- __my_malloc_handler = __oom_handler;
- }
- if ( 0 == __my_malloc_handler) {
- _STLP_THROW_BAD_ALLOC;
- }
- (*__my_malloc_handler)();
- __result = malloc(__n);
- if ( __result )
- return __result;
- }
- }
- return __result;
- }
- __oom_handler_type _STLP_CALL __malloc_alloc::set_malloc_handler(__oom_handler_type __f)
- {
- #ifdef _STLP_THREADS
- _STLP_auto_lock _l( __oom_handler_lock );
- #endif
- __oom_handler_type __old = __oom_handler;
- __oom_handler = __f;
- return __old;
- }
- // *******************************************************
- // Default node allocator.
- // With a reasonable compiler, this should be roughly as fast as the
- // original STL class-specific allocators, but with less fragmentation.
- //
- // Important implementation properties:
- // 1. If the client request an object of size > _MAX_BYTES, the resulting
- // object will be obtained directly from malloc.
- // 2. In all other cases, we allocate an object of size exactly
- // _S_round_up(requested_size). Thus the client has enough size
- // information that we can return the object to the proper free list
- // without permanently losing part of the object.
- //
- #define _STLP_NFREELISTS 16
- #if defined (_STLP_LEAKS_PEDANTIC) && defined (_STLP_USE_DYNAMIC_LIB)
- /*
- * We can only do cleanup of the node allocator memory pool if we are
- * sure that the STLport library is used as a shared one as it guaranties
- * the unicity of the node allocator instance. Without that guaranty node
- * allocator instances might exchange memory blocks making the implementation
- * of a cleaning process much more complicated.
- */
- # define _STLP_DO_CLEAN_NODE_ALLOC
- #endif
- /* When STLport is used without multi threaded safety we use the node allocator
- * implementation with locks as locks becomes no-op. The lock free implementation
- * always use system specific atomic operations which are slower than 'normal'
- * ones.
- */
- #if defined (_STLP_THREADS) && \
- defined (_STLP_HAS_ATOMIC_FREELIST) && defined (_STLP_ATOMIC_ADD)
- /*
- * We have an implementation of the atomic freelist (_STLP_atomic_freelist)
- * for this architecture and compiler. That means we can use the non-blocking
- * implementation of the node-allocation engine.*/
- # define _STLP_USE_LOCK_FREE_IMPLEMENTATION
- #endif
- #if !defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- # if defined (_STLP_THREADS)
- class _Node_Alloc_Lock {
- static _STLP_STATIC_MUTEX& _S_Mutex() {
- static _STLP_STATIC_MUTEX mutex _STLP_MUTEX_INITIALIZER;
- return mutex;
- }
- public:
- _Node_Alloc_Lock() {
- # if defined (_STLP_SGI_THREADS)
- if (__us_rsthread_malloc)
- # endif
- _S_Mutex()._M_acquire_lock();
- }
- ~_Node_Alloc_Lock() {
- # if defined (_STLP_SGI_THREADS)
- if (__us_rsthread_malloc)
- # endif
- _S_Mutex()._M_release_lock();
- }
- };
- # else
- class _Node_Alloc_Lock {
- public:
- _Node_Alloc_Lock() { }
- ~_Node_Alloc_Lock() { }
- };
- # endif
- struct _Node_alloc_obj {
- _Node_alloc_obj * _M_next;
- };
- #endif
- class __node_alloc_impl {
- static inline size_t _STLP_CALL _S_round_up(size_t __bytes)
- { return (((__bytes) + (size_t)_ALIGN-1) & ~((size_t)_ALIGN - 1)); }
- #if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- typedef _STLP_atomic_freelist::item _Obj;
- typedef _STLP_atomic_freelist _Freelist;
- typedef _STLP_atomic_freelist _ChunkList;
- // Header of blocks of memory that have been allocated as part of
- // a larger chunk but have not yet been chopped up into nodes.
- struct _FreeBlockHeader : public _STLP_atomic_freelist::item {
- char* _M_end; // pointer to end of free memory
- };
- #else
- typedef _Node_alloc_obj _Obj;
- typedef _Obj* _STLP_VOLATILE _Freelist;
- typedef _Obj* _ChunkList;
- #endif
- private:
- // Returns an object of size __n, and optionally adds to size __n free list.
- static _Obj* _S_refill(size_t __n);
- // Allocates a chunk for nobjs of size __p_size. nobjs may be reduced
- // if it is inconvenient to allocate the requested number.
- static char* _S_chunk_alloc(size_t __p_size, int& __nobjs);
- // Chunk allocation state.
- static _Freelist _S_free_list[_STLP_NFREELISTS];
- // Amount of total allocated memory
- #if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- static _STLP_VOLATILE __add_atomic_t _S_heap_size;
- #else
- static size_t _S_heap_size;
- #endif
- #if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- // List of blocks of free memory
- static _STLP_atomic_freelist _S_free_mem_blocks;
- #else
- // Start of the current free memory buffer
- static char* _S_start_free;
- // End of the current free memory buffer
- static char* _S_end_free;
- #endif
- #if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- public:
- // Methods to report alloc/dealloc calls to the counter system.
- # if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- typedef _STLP_VOLATILE __stl_atomic_t _AllocCounter;
- # else
- typedef __stl_atomic_t _AllocCounter;
- # endif
- static _AllocCounter& _STLP_CALL _S_alloc_counter();
- static void _S_alloc_call();
- static void _S_dealloc_call();
- private:
- // Free all the allocated chuncks of memory
- static void _S_chunk_dealloc();
- // Beginning of the linked list of allocated chunks of memory
- static _ChunkList _S_chunks;
- #endif /* _STLP_DO_CLEAN_NODE_ALLOC */
- public:
- /* __n must be > 0 */
- static void* _M_allocate(size_t& __n);
- /* __p may not be 0 */
- static void _M_deallocate(void *__p, size_t __n);
- };
- #if !defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- void* __node_alloc_impl::_M_allocate(size_t& __n) {
- __n = _S_round_up(__n);
- _Obj * _STLP_VOLATILE * __my_free_list = _S_free_list + _S_FREELIST_INDEX(__n);
- _Obj *__r;
- // Acquire the lock here with a constructor call.
- // This ensures that it is released in exit or during stack
- // unwinding.
- _Node_Alloc_Lock __lock_instance;
- if ( (__r = *__my_free_list) != 0 ) {
- *__my_free_list = __r->_M_next;
- } else {
- __r = _S_refill(__n);
- }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- _S_alloc_call();
- # endif
- // lock is released here
- return __r;
- }
- void __node_alloc_impl::_M_deallocate(void *__p, size_t __n) {
- _Obj * _STLP_VOLATILE * __my_free_list = _S_free_list + _S_FREELIST_INDEX(__n);
- _Obj * __pobj = __STATIC_CAST(_Obj*, __p);
- // acquire lock
- _Node_Alloc_Lock __lock_instance;
- __pobj->_M_next = *__my_free_list;
- *__my_free_list = __pobj;
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- _S_dealloc_call();
- # endif
- // lock is released here
- }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- # define _STLP_OFFSET sizeof(_Obj)
- # else
- # define _STLP_OFFSET 0
- # endif
- /* We allocate memory in large chunks in order to avoid fragmenting */
- /* the malloc heap too much. */
- /* We assume that size is properly aligned. */
- /* We hold the allocation lock. */
- char* __node_alloc_impl::_S_chunk_alloc(size_t _p_size, int& __nobjs) {
- char* __result;
- size_t __total_bytes = _p_size * __nobjs;
- size_t __bytes_left = _S_end_free - _S_start_free;
- if (__bytes_left > 0) {
- if (__bytes_left >= __total_bytes) {
- __result = _S_start_free;
- _S_start_free += __total_bytes;
- return __result;
- }
- if (__bytes_left >= _p_size) {
- __nobjs = (int)(__bytes_left / _p_size);
- __total_bytes = _p_size * __nobjs;
- __result = _S_start_free;
- _S_start_free += __total_bytes;
- return __result;
- }
- // Try to make use of the left-over piece.
- _Obj* _STLP_VOLATILE* __my_free_list = _S_free_list + _S_FREELIST_INDEX(__bytes_left);
- __REINTERPRET_CAST(_Obj*, _S_start_free)->_M_next = *__my_free_list;
- *__my_free_list = __REINTERPRET_CAST(_Obj*, _S_start_free);
- _S_start_free = _S_end_free = 0;
- }
- size_t __bytes_to_get = 2 * __total_bytes + _S_round_up(_S_heap_size) + _STLP_OFFSET;
- _STLP_TRY {
- _S_start_free = __stlp_new_chunk(__bytes_to_get);
- }
- #if defined (_STLP_USE_EXCEPTIONS)
- catch (const _STLP_STD::bad_alloc&) {
- _Obj* _STLP_VOLATILE* __my_free_list;
- _Obj* __p;
- // Try to do with what we have. That can't hurt.
- // We do not try smaller requests, since that tends
- // to result in disaster on multi-process machines.
- for (size_t __i = _p_size; __i <= (size_t)_MAX_BYTES; __i += (size_t)_ALIGN) {
- __my_free_list = _S_free_list + _S_FREELIST_INDEX(__i);
- __p = *__my_free_list;
- if (0 != __p) {
- *__my_free_list = __p -> _M_next;
- _S_start_free = __REINTERPRET_CAST(char*, __p);
- _S_end_free = _S_start_free + __i;
- return _S_chunk_alloc(_p_size, __nobjs);
- // Any leftover piece will eventually make it to the
- // right free list.
- }
- }
- __bytes_to_get = __total_bytes + _STLP_OFFSET;
- _S_start_free = __stlp_new_chunk(__bytes_to_get);
- }
- #endif
- _S_heap_size += __bytes_to_get >> 4;
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- __REINTERPRET_CAST(_Obj*, _S_start_free)->_M_next = _S_chunks;
- _S_chunks = __REINTERPRET_CAST(_Obj*, _S_start_free);
- # endif
- _S_end_free = _S_start_free + __bytes_to_get;
- _S_start_free += _STLP_OFFSET;
- return _S_chunk_alloc(_p_size, __nobjs);
- }
- /* Returns an object of size __n, and optionally adds to size __n free list.*/
- /* We assume that __n is properly aligned. */
- /* We hold the allocation lock. */
- _Node_alloc_obj* __node_alloc_impl::_S_refill(size_t __n) {
- int __nobjs = 20;
- char* __chunk = _S_chunk_alloc(__n, __nobjs);
- if (1 == __nobjs) return __REINTERPRET_CAST(_Obj*, __chunk);
- _Obj* _STLP_VOLATILE* __my_free_list = _S_free_list + _S_FREELIST_INDEX(__n);
- _Obj* __result;
- _Obj* __current_obj;
- _Obj* __next_obj;
- /* Build free list in chunk */
- __result = __REINTERPRET_CAST(_Obj*, __chunk);
- *__my_free_list = __next_obj = __REINTERPRET_CAST(_Obj*, __chunk + __n);
- for (--__nobjs; --__nobjs; ) {
- __current_obj = __next_obj;
- __next_obj = __REINTERPRET_CAST(_Obj*, __REINTERPRET_CAST(char*, __next_obj) + __n);
- __current_obj->_M_next = __next_obj;
- }
- __next_obj->_M_next = 0;
- return __result;
- }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- void __node_alloc_impl::_S_alloc_call()
- { ++_S_alloc_counter(); }
- void __node_alloc_impl::_S_dealloc_call() {
- __stl_atomic_t &counter = _S_alloc_counter();
- if (--counter == 0)
- { _S_chunk_dealloc(); }
- }
- /* We deallocate all the memory chunks */
- void __node_alloc_impl::_S_chunk_dealloc() {
- _Obj *__pcur = _S_chunks, *__pnext;
- while (__pcur != 0) {
- __pnext = __pcur->_M_next;
- __stlp_delete_chunck(__pcur);
- __pcur = __pnext;
- }
- _S_chunks = 0;
- _S_start_free = _S_end_free = 0;
- _S_heap_size = 0;
- memset(__REINTERPRET_CAST(char*, __CONST_CAST(_Obj**, &_S_free_list[0])), 0, _STLP_NFREELISTS * sizeof(_Obj*));
- }
- # endif
- #else
- void* __node_alloc_impl::_M_allocate(size_t& __n) {
- __n = _S_round_up(__n);
- _Obj* __r = _S_free_list[_S_FREELIST_INDEX(__n)].pop();
- if (__r == 0)
- { __r = _S_refill(__n); }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- _S_alloc_call();
- # endif
- return __r;
- }
- void __node_alloc_impl::_M_deallocate(void *__p, size_t __n) {
- _S_free_list[_S_FREELIST_INDEX(__n)].push(__STATIC_CAST(_Obj*, __p));
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- _S_dealloc_call();
- # endif
- }
- /* Returns an object of size __n, and optionally adds additional ones to */
- /* freelist of objects of size __n. */
- /* We assume that __n is properly aligned. */
- __node_alloc_impl::_Obj* __node_alloc_impl::_S_refill(size_t __n) {
- int __nobjs = 20;
- char* __chunk = _S_chunk_alloc(__n, __nobjs);
- if (__nobjs <= 1)
- return __REINTERPRET_CAST(_Obj*, __chunk);
- // Push all new nodes (minus first one) onto freelist
- _Obj* __result = __REINTERPRET_CAST(_Obj*, __chunk);
- _Obj* __cur_item = __result;
- _Freelist* __my_freelist = _S_free_list + _S_FREELIST_INDEX(__n);
- for (--__nobjs; __nobjs != 0; --__nobjs) {
- __cur_item = __REINTERPRET_CAST(_Obj*, __REINTERPRET_CAST(char*, __cur_item) + __n);
- __my_freelist->push(__cur_item);
- }
- return __result;
- }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- # define _STLP_OFFSET _ALIGN
- # else
- # define _STLP_OFFSET 0
- # endif
- /* We allocate memory in large chunks in order to avoid fragmenting */
- /* the malloc heap too much. */
- /* We assume that size is properly aligned. */
- char* __node_alloc_impl::_S_chunk_alloc(size_t _p_size, int& __nobjs) {
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- //We are going to add a small memory block to keep all the allocated blocks
- //address, we need to do so respecting the memory alignment. The following
- //static assert checks that the reserved block is big enough to store a pointer.
- _STLP_STATIC_ASSERT(sizeof(_Obj) <= _ALIGN)
- # endif
- char* __result = 0;
- __add_atomic_t __total_bytes = __STATIC_CAST(__add_atomic_t, _p_size) * __nobjs;
- _FreeBlockHeader* __block = __STATIC_CAST(_FreeBlockHeader*, _S_free_mem_blocks.pop());
- if (__block != 0) {
- // We checked a block out and can now mess with it with impugnity.
- // We'll put the remainder back into the list if we're done with it below.
- char* __buf_start = __REINTERPRET_CAST(char*, __block);
- __add_atomic_t __bytes_left = __block->_M_end - __buf_start;
- if ((__bytes_left < __total_bytes) && (__bytes_left >= __STATIC_CAST(__add_atomic_t, _p_size))) {
- // There's enough left for at least one object, but not as much as we wanted
- __result = __buf_start;
- __nobjs = (int)(__bytes_left/_p_size);
- __total_bytes = __STATIC_CAST(__add_atomic_t, _p_size) * __nobjs;
- __bytes_left -= __total_bytes;
- __buf_start += __total_bytes;
- }
- else if (__bytes_left >= __total_bytes) {
- // The block has enough left to satisfy all that was asked for
- __result = __buf_start;
- __bytes_left -= __total_bytes;
- __buf_start += __total_bytes;
- }
- if (__bytes_left != 0) {
- // There is still some memory left over in block after we satisfied our request.
- if ((__result != 0) && (__bytes_left >= (__add_atomic_t)sizeof(_FreeBlockHeader))) {
- // We were able to allocate at least one object and there is still enough
- // left to put remainder back into list.
- _FreeBlockHeader* __newblock = __REINTERPRET_CAST(_FreeBlockHeader*, __buf_start);
- __newblock->_M_end = __block->_M_end;
- _S_free_mem_blocks.push(__newblock);
- }
- else {
- // We were not able to allocate enough for at least one object.
- // Shove into freelist of nearest (rounded-down!) size.
- size_t __rounded_down = _S_round_up(__bytes_left + 1) - (size_t)_ALIGN;
- if (__rounded_down > 0)
- _S_free_list[_S_FREELIST_INDEX(__rounded_down)].push((_Obj*)__buf_start);
- }
- }
- if (__result != 0)
- return __result;
- }
- // We couldn't satisfy it from the list of free blocks, get new memory.
- __add_atomic_t __bytes_to_get = 2 * __total_bytes +
- __STATIC_CAST(__add_atomic_t,
- _S_round_up(__STATIC_CAST(__uadd_atomic_t, _STLP_ATOMIC_ADD(&_S_heap_size, 0)))) +
- _STLP_OFFSET;
- _STLP_TRY {
- __result = __stlp_new_chunk(__bytes_to_get);
- }
- #if defined (_STLP_USE_EXCEPTIONS)
- catch (const bad_alloc&) {
- // Allocation failed; try to canibalize from freelist of a larger object size.
- for (size_t __i = _p_size; __i <= (size_t)_MAX_BYTES; __i += (size_t)_ALIGN) {
- _Obj* __p = _S_free_list[_S_FREELIST_INDEX(__i)].pop();
- if (0 != __p) {
- if (__i < sizeof(_FreeBlockHeader)) {
- // Not enough to put into list of free blocks, divvy it up here.
- // Use as much as possible for this request and shove remainder into freelist.
- __nobjs = (int)(__i/_p_size);
- __total_bytes = __nobjs * __STATIC_CAST(__add_atomic_t, _p_size);
- size_t __bytes_left = __i - __total_bytes;
- size_t __rounded_down = _S_round_up(__bytes_left+1) - (size_t)_ALIGN;
- if (__rounded_down > 0) {
- _S_free_list[_S_FREELIST_INDEX(__rounded_down)].push(__REINTERPRET_CAST(_Obj*, __REINTERPRET_CAST(char*, __p) + __total_bytes));
- }
- return __REINTERPRET_CAST(char*, __p);
- }
- else {
- // Add node to list of available blocks and recursively allocate from it.
- _FreeBlockHeader* __newblock = (_FreeBlockHeader*)__p;
- __newblock->_M_end = __REINTERPRET_CAST(char*, __p) + __i;
- _S_free_mem_blocks.push(__newblock);
- return _S_chunk_alloc(_p_size, __nobjs);
- }
- }
- }
- // We were not able to find something in a freelist, try to allocate a smaller amount.
- __bytes_to_get = __total_bytes + _STLP_OFFSET;
- __result = __stlp_new_chunk(__bytes_to_get);
- // This should either throw an exception or remedy the situation.
- // Thus we assume it succeeded.
- }
- #endif
- // Alignment check
- _STLP_VERBOSE_ASSERT(((__REINTERPRET_CAST(size_t, __result) & __STATIC_CAST(size_t, _ALIGN - 1)) == 0),
- _StlMsg_DBA_DELETED_TWICE)
- _STLP_ATOMIC_ADD(&_S_heap_size, __bytes_to_get >> 4);
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- // We have to track the allocated memory chunks for release on exit.
- _S_chunks.push(__REINTERPRET_CAST(_Obj*, __result));
- __result += _ALIGN;
- __bytes_to_get -= _ALIGN;
- # endif
- if (__bytes_to_get > __total_bytes) {
- // Push excess memory allocated in this chunk into list of free memory blocks
- _FreeBlockHeader* __freeblock = __REINTERPRET_CAST(_FreeBlockHeader*, __result + __total_bytes);
- __freeblock->_M_end = __result + __bytes_to_get;
- _S_free_mem_blocks.push(__freeblock);
- }
- return __result;
- }
- # if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- void __node_alloc_impl::_S_alloc_call()
- { _STLP_ATOMIC_INCREMENT(&_S_alloc_counter()); }
- void __node_alloc_impl::_S_dealloc_call() {
- _STLP_VOLATILE __stl_atomic_t *pcounter = &_S_alloc_counter();
- if (_STLP_ATOMIC_DECREMENT(pcounter) == 0)
- _S_chunk_dealloc();
- }
- /* We deallocate all the memory chunks */
- void __node_alloc_impl::_S_chunk_dealloc() {
- // Note: The _Node_alloc_helper class ensures that this function
- // will only be called when the (shared) library is unloaded or the
- // process is shutdown. It's thus not possible that another thread
- // is currently trying to allocate a node (we're not thread-safe here).
- //
- // Clear the free blocks and all freelistst. This makes sure that if
- // for some reason more memory is allocated again during shutdown
- // (it'd also be really nasty to leave references to deallocated memory).
- _S_free_mem_blocks.clear();
- _S_heap_size = 0;
- for (size_t __i = 0; __i < _STLP_NFREELISTS; ++__i) {
- _S_free_list[__i].clear();
- }
- // Detach list of chunks and free them all
- _Obj* __chunk = _S_chunks.clear();
- while (__chunk != 0) {
- _Obj* __next = __chunk->_M_next;
- __stlp_delete_chunck(__chunk);
- __chunk = __next;
- }
- }
- # endif
- #endif
- #if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- struct __node_alloc_cleaner {
- ~__node_alloc_cleaner()
- { __node_alloc_impl::_S_dealloc_call(); }
- };
- # if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- _STLP_VOLATILE __stl_atomic_t& _STLP_CALL
- # else
- __stl_atomic_t& _STLP_CALL
- # endif
- __node_alloc_impl::_S_alloc_counter() {
- static _AllocCounter _S_counter = 1;
- static __node_alloc_cleaner _S_node_alloc_cleaner;
- return _S_counter;
- }
- #endif
- #if !defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- _Node_alloc_obj * _STLP_VOLATILE
- __node_alloc_impl::_S_free_list[_STLP_NFREELISTS]
- = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
- // The 16 zeros are necessary to make version 4.1 of the SunPro
- // compiler happy. Otherwise it appears to allocate too little
- // space for the array.
- #else
- _STLP_atomic_freelist __node_alloc_impl::_S_free_list[_STLP_NFREELISTS];
- _STLP_atomic_freelist __node_alloc_impl::_S_free_mem_blocks;
- #endif
- #if !defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- char *__node_alloc_impl::_S_start_free = 0;
- char *__node_alloc_impl::_S_end_free = 0;
- #endif
- #if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- _STLP_VOLATILE __add_atomic_t
- #else
- size_t
- #endif
- __node_alloc_impl::_S_heap_size = 0;
- #if defined (_STLP_DO_CLEAN_NODE_ALLOC)
- # if defined (_STLP_USE_LOCK_FREE_IMPLEMENTATION)
- _STLP_atomic_freelist __node_alloc_impl::_S_chunks;
- # else
- _Node_alloc_obj* __node_alloc_impl::_S_chunks = 0;
- # endif
- #endif
- void * _STLP_CALL __node_alloc::_M_allocate(size_t& __n)
- { return __node_alloc_impl::_M_allocate(__n); }
- void _STLP_CALL __node_alloc::_M_deallocate(void *__p, size_t __n)
- { __node_alloc_impl::_M_deallocate(__p, __n); }
- #if defined (_STLP_PTHREADS) && !defined (_STLP_NO_THREADS)
- # define _STLP_DATA_ALIGNMENT 8
- _STLP_MOVE_TO_PRIV_NAMESPACE
- // *******************************************************
- // __perthread_alloc implementation
- union _Pthread_alloc_obj {
- union _Pthread_alloc_obj * __free_list_link;
- char __client_data[_STLP_DATA_ALIGNMENT]; /* The client sees this. */
- };
- // Pthread allocators don't appear to the client to have meaningful
- // instances. We do in fact need to associate some state with each
- // thread. That state is represented by _Pthread_alloc_per_thread_state.
- struct _Pthread_alloc_per_thread_state {
- typedef _Pthread_alloc_obj __obj;
- enum { _S_NFREELISTS = _MAX_BYTES / _STLP_DATA_ALIGNMENT };
- // Free list link for list of available per thread structures.
- // When one of these becomes available for reuse due to thread
- // termination, any objects in its free list remain associated
- // with it. The whole structure may then be used by a newly
- // created thread.
- _Pthread_alloc_per_thread_state() : __next(0)
- { memset((void *)__CONST_CAST(_Pthread_alloc_obj**, __free_list), 0, (size_t)_S_NFREELISTS * sizeof(__obj *)); }
- // Returns an object of size __n, and possibly adds to size n free list.
- void *_M_refill(size_t __n);
- _Pthread_alloc_obj* volatile __free_list[_S_NFREELISTS];
- _Pthread_alloc_per_thread_state *__next;
- // this data member is only to be used by per_thread_allocator, which returns memory to the originating thread.
- _STLP_mutex _M_lock;
- };
- // Pthread-specific allocator.
- class _Pthread_alloc_impl {
- public: // but only for internal use:
- typedef _Pthread_alloc_per_thread_state __state_type;
- typedef char value_type;
- // Allocates a chunk for nobjs of size size. nobjs may be reduced
- // if it is inconvenient to allocate the requested number.
- static char *_S_chunk_alloc(size_t __size, size_t &__nobjs, __state_type*);
- enum {_S_ALIGN = _STLP_DATA_ALIGNMENT};
- static size_t _S_round_up(size_t __bytes)
- { return (((__bytes) + (int)_S_ALIGN - 1) & ~((int)_S_ALIGN - 1)); }
- static size_t _S_freelist_index(size_t __bytes)
- { return (((__bytes) + (int)_S_ALIGN - 1) / (int)_S_ALIGN - 1); }
- private:
- // Chunk allocation state. And other shared state.
- // Protected by _S_chunk_allocator_lock.
- static _STLP_STATIC_MUTEX _S_chunk_allocator_lock;
- static char *_S_start_free;
- static char *_S_end_free;
- static size_t _S_heap_size;
- static __state_type *_S_free_per_thread_states;
- static pthread_key_t _S_key;
- static bool _S_key_initialized;
- // Pthread key under which per thread state is stored.
- // Allocator instances that are currently unclaimed by any thread.
- static void _S_destructor(void *instance);
- // Function to be called on thread exit to reclaim per thread
- // state.
- static __state_type *_S_new_per_thread_state();
- public:
- // Return a recycled or new per thread state.
- static __state_type *_S_get_per_thread_state();
- private:
- // ensure that the current thread has an associated
- // per thread state.
- class _M_lock;
- friend class _M_lock;
- class _M_lock {
- public:
- _M_lock () { _S_chunk_allocator_lock._M_acquire_lock(); }
- ~_M_lock () { _S_chunk_allocator_lock._M_release_lock(); }
- };
- public:
- /* n must be > 0 */
- static void * allocate(size_t& __n);
- /* p may not be 0 */
- static void deallocate(void *__p, size_t __n);
- // boris : versions for per_thread_allocator
- /* n must be > 0 */
- static void * allocate(size_t& __n, __state_type* __a);
- /* p may not be 0 */
- static void deallocate(void *__p, size_t __n, __state_type* __a);
- static void * reallocate(void *__p, size_t __old_sz, size_t& __new_sz);
- };
- /* Returns an object of size n, and optionally adds to size n free list.*/
- /* We assume that n is properly aligned. */
- /* We hold the allocation lock. */
- void *_Pthread_alloc_per_thread_state::_M_refill(size_t __n) {
- typedef _Pthread_alloc_obj __obj;
- size_t __nobjs = 128;
- char * __chunk = _Pthread_alloc_impl::_S_chunk_alloc(__n, __nobjs, this);
- __obj * volatile * __my_free_list;
- __obj * __result;
- __obj * __current_obj, * __next_obj;
- size_t __i;
- if (1 == __nobjs) {
- return __chunk;
- }
- __my_free_list = __free_list + _Pthread_alloc_impl::_S_freelist_index(__n);
- /* Build free list in chunk */
- __result = (__obj *)__chunk;
- *__my_free_list = __next_obj = (__obj *)(__chunk + __n);
- for (__i = 1; ; ++__i) {
- __current_obj = __next_obj;
- __next_obj = (__obj *)((char *)__next_obj + __n);
- if (__nobjs - 1 == __i) {
- __current_obj -> __free_list_link = 0;
- break;
- } else {
- __current_obj -> __free_list_link = __next_obj;
- }
- }
- return __result;
- }
- void _Pthread_alloc_impl::_S_destructor(void *__instance) {
- _M_lock __lock_instance; // Need to acquire lock here.
- _Pthread_alloc_per_thread_state* __s = (_Pthread_alloc_per_thread_state*)__instance;
- __s -> __next = _S_free_per_thread_states;
- _S_free_per_thread_states = __s;
- }
- _Pthread_alloc_per_thread_state* _Pthread_alloc_impl::_S_new_per_thread_state() {
- /* lock already held here. */
- if (0 != _S_free_per_thread_states) {
- _Pthread_alloc_per_thread_state *__result = _S_free_per_thread_states;
- _S_free_per_thread_states = _S_free_per_thread_states -> __next;
- return __result;
- }
- else {
- return new _Pthread_alloc_per_thread_state;
- }
- }
- _Pthread_alloc_per_thread_state* _Pthread_alloc_impl::_S_get_per_thread_state() {
- int __ret_code;
- __state_type* __result;
- if (_S_key_initialized && (__result = (__state_type*) pthread_getspecific(_S_key)))
- return __result;
- /*REFERENCED*/
- _M_lock __lock_instance; // Need to acquire lock here.
- if (!_S_key_initialized) {
- if (pthread_key_create(&_S_key, _S_destructor)) {
- _STLP_THROW_BAD_ALLOC; // failed
- }
- _S_key_initialized = true;
- }
- __result = _S_new_per_thread_state();
- __ret_code = pthread_setspecific(_S_key, __result);
- if (__ret_code) {
- if (__ret_code == ENOMEM) {
- _STLP_THROW_BAD_ALLOC;
- } else {
- // EINVAL
- _STLP_ABORT();
- }
- }
- return __result;
- }
- /* We allocate memory in large chunks in order to avoid fragmenting */
- /* the malloc heap too much. */
- /* We assume that size is properly aligned. */
- char *_Pthread_alloc_impl::_S_chunk_alloc(size_t __p_size, size_t &__nobjs, _Pthread_alloc_per_thread_state *__a) {
- typedef _Pthread_alloc_obj __obj;
- {
- char * __result;
- size_t __total_bytes;
- size_t __bytes_left;
- /*REFERENCED*/
- _M_lock __lock_instance; // Acquire lock for this routine
- __total_bytes = __p_size * __nobjs;
- __bytes_left = _S_end_free - _S_start_free;
- if (__bytes_left >= __total_bytes) {
- __result = _S_start_free;
- _S_start_free += __total_bytes;
- return __result;
- } else if (__bytes_left >= __p_size) {
- __nobjs = __bytes_left/__p_size;
- __total_bytes = __p_size * __nobjs;
- __result = _S_start_free;
- _S_start_free += __total_bytes;
- return __result;
- } else {
- size_t __bytes_to_get = 2 * __total_bytes + _S_round_up(_S_heap_size);
- // Try to make use of the left-over piece.
- if (__bytes_left > 0) {
- __obj * volatile * __my_free_list = __a->__free_list + _S_freelist_index(__bytes_left);
- ((__obj *)_S_start_free) -> __free_list_link = *__my_free_list;
- *__my_free_list = (__obj *)_S_start_free;
- }
- # ifdef _SGI_SOURCE
- // Try to get memory that's aligned on something like a
- // cache line boundary, so as to avoid parceling out
- // parts of the same line to different threads and thus
- // possibly different processors.
- {
- const int __cache_line_size = 128; // probable upper bound
- __bytes_to_get &= ~(__cache_line_size-1);
- _S_start_free = (char *)memalign(__cache_line_size, __bytes_to_get);
- if (0 == _S_start_free) {
- _S_start_free = (char *)__malloc_alloc::allocate(__bytes_to_get);
- }
- }
- # else /* !SGI_SOURCE */
- _S_start_free = (char *)__malloc_alloc::allocate(__bytes_to_get);
- # endif
- _S_heap_size += __bytes_to_get >> 4;
- _S_end_free = _S_start_free + __bytes_to_get;
- }
- }
- // lock is released here
- return _S_chunk_alloc(__p_size, __nobjs, __a);
- }
- /* n must be > 0 */
- void *_Pthread_alloc_impl::allocate(size_t& __n) {
- typedef _Pthread_alloc_obj __obj;
- __obj * volatile * __my_free_list;
- __obj * __result;
- __state_type* __a;
- if (__n > _MAX_BYTES) {
- return __malloc_alloc::allocate(__n);
- }
- __n = _S_round_up(__n);
- __a = _S_get_per_thread_state();
- __my_free_list = __a->__free_list + _S_freelist_index(__n);
- __result = *__my_free_list;
- if (__result == 0) {
- void *__r = __a->_M_refill(__n);
- return __r;
- }
- *__my_free_list = __result->__free_list_link;
- return __result;
- };
- /* p may not be 0 */
- void _Pthread_alloc_impl::deallocate(void *__p, size_t __n) {
- typedef _Pthread_alloc_obj __obj;
- __obj *__q = (__obj *)__p;
- __obj * volatile * __my_free_list;
- __state_type* __a;
- if (__n > _MAX_BYTES) {
- __malloc_alloc::deallocate(__p, __n);
- return;
- }
- __a = _S_get_per_thread_state();
- __my_free_list = __a->__free_list + _S_freelist_index(__n);
- __q -> __free_list_link = *__my_free_list;
- *__my_free_list = __q;
- }
- // boris : versions for per_thread_allocator
- /* n must be > 0 */
- void *_Pthread_alloc_impl::allocate(size_t& __n, __state_type* __a) {
- typedef _Pthread_alloc_obj __obj;
- __obj * volatile * __my_free_list;
- __obj * __result;
- if (__n > _MAX_BYTES) {
- return __malloc_alloc::allocate(__n);
- }
- __n = _S_round_up(__n);
- // boris : here, we have to lock per thread state, as we may be getting memory from
- // different thread pool.
- _STLP_auto_lock __lock(__a->_M_lock);
- __my_free_list = __a->__free_list + _S_freelist_index(__n);
- __result = *__my_free_list;
- if (__result == 0) {
- void *__r = __a->_M_refill(__n);
- return __r;
- }
- *__my_free_list = __result->__free_list_link;
- return __result;
- };
- /* p may not be 0 */
- void _Pthread_alloc_impl::deallocate(void *__p, size_t __n, __state_type* __a) {
- typedef _Pthread_alloc_obj __obj;
- __obj *__q = (__obj *)__p;
- __obj * volatile * __my_free_list;
- if (__n > _MAX_BYTES) {
- __malloc_alloc::deallocate(__p, __n);
- return;
- }
- // boris : here, we have to lock per thread state, as we may be returning memory from
- // different thread.
- _STLP_auto_lock __lock(__a->_M_lock);
- __my_free_list = __a->__free_list + _S_freelist_index(__n);
- __q -> __free_list_link = *__my_free_list;
- *__my_free_list = __q;
- }
- void *_Pthread_alloc_impl::reallocate(void *__p, size_t __old_sz, size_t& __new_sz) {
- void * __result;
- size_t __copy_sz;
- if (__old_sz > _MAX_BYTES && __new_sz > _MAX_BYTES) {
- return realloc(__p, __new_sz);
- }
- if (_S_round_up(__old_sz) == _S_round_up(__new_sz)) return __p;
- __result = allocate(__new_sz);
- __copy_sz = __new_sz > __old_sz? __old_sz : __new_sz;
- memcpy(__result, __p, __copy_sz);
- deallocate(__p, __old_sz);
- return __result;
- }
- _Pthread_alloc_per_thread_state* _Pthread_alloc_impl::_S_free_per_thread_states = 0;
- pthread_key_t _Pthread_alloc_impl::_S_key = 0;
- _STLP_STATIC_MUTEX _Pthread_alloc_impl::_S_chunk_allocator_lock _STLP_MUTEX_INITIALIZER;
- bool _Pthread_alloc_impl::_S_key_initialized = false;
- char *_Pthread_alloc_impl::_S_start_free = 0;
- char *_Pthread_alloc_impl::_S_end_free = 0;
- size_t _Pthread_alloc_impl::_S_heap_size = 0;
- void * _STLP_CALL _Pthread_alloc::allocate(size_t& __n)
- { return _Pthread_alloc_impl::allocate(__n); }
- void _STLP_CALL _Pthread_alloc::deallocate(void *__p, size_t __n)
- { _Pthread_alloc_impl::deallocate(__p, __n); }
- void * _STLP_CALL _Pthread_alloc::allocate(size_t& __n, __state_type* __a)
- { return _Pthread_alloc_impl::allocate(__n, __a); }
- void _STLP_CALL _Pthread_alloc::deallocate(void *__p, size_t __n, __state_type* __a)
- { _Pthread_alloc_impl::deallocate(__p, __n, __a); }
- void * _STLP_CALL _Pthread_alloc::reallocate(void *__p, size_t __old_sz, size_t& __new_sz)
- { return _Pthread_alloc_impl::reallocate(__p, __old_sz, __new_sz); }
- _Pthread_alloc_per_thread_state* _STLP_CALL _Pthread_alloc::_S_get_per_thread_state()
- { return _Pthread_alloc_impl::_S_get_per_thread_state(); }
- _STLP_MOVE_TO_STD_NAMESPACE
- #endif
- _STLP_END_NAMESPACE
- #undef _S_FREELIST_INDEX
|