xpcom/ds/nsCppSharedAllocator.h
author Matthew Gregan <kinetik@flim.org>
Fri, 02 Mar 2012 13:59:32 +1300
changeset 35275 c23ed8a949b2b27cb5df8a5621b8e60cef01a539
parent 30635 e5f0916b32d3ecacbb14486b4e1885cc24b6ceb5
permissions -rw-r--r--
Bug 730425 - Perform an additional range checks in stream read functions. r=roc a=lsblakk

#ifndef nsCppSharedAllocator_h__
#define nsCppSharedAllocator_h__

#include "nsMemory.h"     // for |nsMemory|
#include NEW_H					// to allow placement |new|


  // under MSVC shut off copious warnings about unused in-lines
#ifdef _MSC_VER
  #pragma warning( disable: 4514 )
#endif

#ifdef HAVE_CPP_NUMERIC_LIMITS
#include <limits>
#else
#include <limits.h>
#endif


template <class T>
class nsCppSharedAllocator
    /*
      ...allows Standard Library containers, et al, to use our global shared
      (XP)COM-aware allocator.
    */
  {
    public:
      typedef T          value_type;
      typedef size_t     size_type;
      typedef ptrdiff_t  difference_type;

      typedef T*         pointer;
      typedef const T*   const_pointer;

      typedef T&         reference;
      typedef const T&   const_reference;



      nsCppSharedAllocator() { }

#ifdef HAVE_CPP_MEMBER_TEMPLATES
      template <class U>
      nsCppSharedAllocator( const nsCppSharedAllocator<U>& ) { }
#endif

     ~nsCppSharedAllocator() { }


      pointer
      address( reference r ) const
        {
          return &r;
        }

      const_pointer
      address( const_reference r ) const
        {
          return &r;
        }

      pointer
      allocate( size_type n, const void* /*hint*/=0 )
        {
          return reinterpret_cast<pointer>(nsMemory::Alloc(static_cast<PRUint32>(n*sizeof(T))));
        }

      void
      deallocate( pointer p, size_type /*n*/ )
        {
          nsMemory::Free(p);
        }

      void
      construct( pointer p, const T& val )
        {
          new (p) T(val);
        }
      
      void
      destroy( pointer p )
        {
          p->~T();
        }

      size_type
      max_size() const
        {
#ifdef HAVE_CPP_NUMERIC_LIMITS
          return numeric_limits<size_type>::max() / sizeof(T);
#else
          return ULONG_MAX / sizeof(T);
#endif
        }

#ifdef HAVE_CPP_MEMBER_TEMPLATES
      template <class U>
      struct rebind
        {
          typedef nsCppSharedAllocator<U> other;
        };
#endif
  };


template <class T>
PRBool
operator==( const nsCppSharedAllocator<T>&, const nsCppSharedAllocator<T>& )
  {
    return PR_TRUE;
  }

template <class T>
PRBool
operator!=( const nsCppSharedAllocator<T>&, const nsCppSharedAllocator<T>& )
  {
    return PR_FALSE;
  }

#endif /* !defined(nsCppSharedAllocator_h__) */