ACE/tests/Dynamic_Test.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    Dynamic_Test.cpp
   5  *
   6  *  This tests the ACE_Dynamic class
   7  *
   8  *  @author Johnny Willemsen  <jwillemsen@remedy.nl>
   9  */
  10 //=============================================================================
  11
  12
  13 #include "test_config.h"
  14 #include "ace/OS_NS_string.h"
  15 #include "ace/Dynamic.h"
  16 #include "ace/OS_Memory.h"
  17
  18
  19 class A
  20 {
  21 public:
  22   A ();
  23
  24   void *operator new (size_t n);
  25
  26   void *operator new (size_t n, const std::nothrow_t&) throw();
  27   void operator delete (void *p, const std::nothrow_t&) throw ();
  28   void * operator new (size_t n, void *p);
  29
  30   void operator delete (void *);
  31
  32   void operator delete (void *, void *);
  33
  34   /// Have we been dynamically created?
  35   bool dynamic_;
  36 };
  37
  38 void*
  39 A::operator new (size_t n)
  40 {
  41   ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
  42
  43   if (dynamic_instance == 0)
  44     {
  45       // If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
  46       // keys.  Try using ACE_HAS_TSS_EMULATION, or increasing
  47       // ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
  48       ACE_TEST_ASSERT (dynamic_instance != 0);
  49
  50       ACE_throw_bad_alloc;
  51     }
  52   else
  53     {
  54       // Allocate the memory and store it (usually in thread-specific
  55       // storage, depending on config flags).
  56       dynamic_instance->set ();
  57
  58       return ::new char[n];
  59     }
  60 }
  61
  62 void*
  63 A::operator new (size_t n, const std::nothrow_t&) throw()
  64 {
  65   ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
  66
  67   if (dynamic_instance == 0)
  68     {
  69       // If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
  70       // keys.  Try using ACE_HAS_TSS_EMULATION, or increasing
  71       // ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
  72       ACE_TEST_ASSERT (dynamic_instance != 0);
  73
  74       return 0;
  75     }
  76   else
  77     {
  78       // Allocate the memory and store it (usually in thread-specific
  79       // storage, depending on config flags).
  80       dynamic_instance->set ();
  81
  82       return ::new(std::nothrow) char[n];
  83     }
  84 }
  85
  86 void
  87 A::operator delete (void *p, const std::nothrow_t&) throw()
  88 {
  89   ::delete [] static_cast <char *> (p);
  90 }
  91
  92 void
  93 A::operator delete (void *obj)
  94 {
  95   ::delete [] static_cast <char *> (obj);
  96 }
  97
  98 A::A()
  99 {
 100   this->dynamic_ = ACE_Dynamic::instance ()->is_dynamic ();
 101
 102   if (this->dynamic_)
 103     // Make sure to reset the flag.
 104     ACE_Dynamic::instance ()->reset ();
 105 }
 106
 107 int
 108 run_main (int, ACE_TCHAR *[])
 109 {
 110   ACE_START_TEST (ACE_TEXT ("Dynamic_Test"));
 111   A from_stack;
 112   A* heap = 0;
 113   ACE_NEW_RETURN (heap, A, 1);
 114   if (from_stack.dynamic_)
 115     {
 116       ACE_ERROR_RETURN ((LM_ERROR,
 117                           ACE_TEXT ("dynamic_ is true for an object on the stack\n")),
 118                         1);
 119     }
 120
 121   if (!heap->dynamic_)
 122     {
 123       ACE_ERROR_RETURN ((LM_ERROR,
 124                           ACE_TEXT ("dynamic_ is false for an object from the heap\n")),
 125                         1);
 126     }
 127
 128   delete heap;
 129   ACE_END_TEST;
 130   return 0;
 131 }
 132