/**
 *    @date 2007.10.29
 *  @author PeakGao <peakgao163@163.com>
 */
#ifndef OM_MEMPOOLALLOCATOR_H
#define OM_MEMPOOLALLOCATOR_H

#include <ace/pre.h>

//#include <ace/ACE_export.h>

#if !defined (ACE_LACKS_PRAGMA_ONCE)
# pragma once
#endif /* ACE_LACKS_PRAGMA_ONCE */

#include <ace/Malloc_Base.h>

#if defined (ACE_HAS_MALLOC_STATS)
#if defined (ACE_HAS_THREADS)
#include "ace/Process_Mutex.h"
#define ACE_PROCESS_MUTEX ACE_Process_Mutex
#else
#include "ace/SV_Semaphore_Simple.h"
#define ACE_PROCESS_MUTEX ACE_SV_Semaphore_Simple
#endif /* ACE_HAS_THREADS */

#endif /* ACE_HAS_MALLOC_STATS */

#include "MemPoolT.h"


namespace om{

class My_Allocator : public ACE_Allocator, public CachePool
{
public:
  /// These methods are defined.
  virtual void *malloc (size_t nbytes);
  virtual void *calloc (size_t nbytes, char initial_value = '\0');
  virtual void *calloc (size_t n_elem, size_t elem_size, char initial_value = '\0');
  virtual void free (void *ptr);

  /// These methods are no-ops.
  virtual int remove (void);
  virtual int bind (const char *name, void *pointer, int duplicates = 0);
  virtual int trybind (const char *name, void *&pointer);
  virtual int find (const char *name, void *&pointer);
  virtual int find (const char *name);
  virtual int unbind (const char *name);
  virtual int unbind (const char *name, void *&pointer);
  virtual int sync (ssize_t len = -1, int flags = MS_SYNC);
  virtual int sync (void *addr, size_t len, int flags = MS_SYNC);
  virtual int protect (ssize_t len = -1, int prot = PROT_RDWR);
  virtual int protect (void *addr, size_t len, int prot = PROT_RDWR);
#if defined (ACE_HAS_MALLOC_STATS)
  virtual void print_stats (void) const;
#endif /* ACE_HAS_MALLOC_STATS */
  virtual void dump (void) const;

private:
  // DO NOT ADD ANY STATE (DATA MEMBERS) TO THIS CLASS!!!!  See the
  // <ACE_Allocator::instance> implementation for explanation.
};


#include /**/ <ace/post.h>

} // namespace om

#endif // OM_MEMPOOLALLOCATOR_H

/**
 *    @date 2007.10.29
 *  @author PeakGao <peakgao163@163.com>
 */
 
#include "MemPoolAllocator.h"
#include <ace/OS_NS_string.h>

namespace om{

    void *
    My_Allocator::malloc (size_t nbytes)
    {
        if (nbytes > 0 && nbytes <= CachePool::getBlockSize())
          return CachePool::alloc();

      return NULL;
    }

    void *
    My_Allocator::calloc (size_t nbytes,
                               char initial_value)
    {
      void *ptr = malloc(nbytes);
      if (!ptr)
          return NULL;

      ACE_OS::memset (ptr, initial_value, nbytes);
      return (void *) ptr;
    }

    void *
    My_Allocator::calloc (size_t n_elem, size_t elem_size, char initial_value)
    {
      return My_Allocator::calloc (n_elem * elem_size, initial_value);
    }

    void
    My_Allocator::free (void *ptr)
    {
        CachePool::free(ptr);
    }

    int
    My_Allocator::remove (void)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::bind (const char *, void *, int)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::trybind (const char *, void *&)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::find (const char *, void *&)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::find (const char *)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::unbind (const char *)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::unbind (const char *, void *&)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::sync (ssize_t, int)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::sync (void *, size_t, int)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::protect (ssize_t, int)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    int
    My_Allocator::protect (void *, size_t, int)
    {
      ACE_NOTSUP_RETURN (-1);
    }

    #if defined (ACE_HAS_MALLOC_STATS)
    void
    My_Allocator::print_stats (void) const
    {
    }
    #endif /* ACE_HAS_MALLOC_STATS */

    void
    My_Allocator::dump (void) const
    {
    #if defined (ACE_HAS_DUMP)
    #endif /* ACE_HAS_DUMP */
    }


} // namespace om