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
  20 class A
  21 {
  22 public:
  23   A ();
  24
  25   void *operator new (size_t n);
  26
  27 #if defined (ACE_HAS_NEW_NOTHROW)
  28   void *operator new (size_t n, const ACE_nothrow_t&) throw();
  29 #if !defined (ACE_LACKS_PLACEMENT_OPERATOR_DELETE)
  30   void operator delete (void *p, const ACE_nothrow_t&) throw ();
  31 #endif /* ACE_LACKS_PLACEMENT_OPERATOR_DELETE */
  32 #endif
  33
  34   void * operator new (size_t n, void *p);
  35
  36   void operator delete (void *);
  37
  38 #if !defined (ACE_LACKS_PLACEMENT_OPERATOR_DELETE)
  39   void operator delete (void *, void *);
  40 #endif /* ACE_LACKS_PLACEMENT_OPERATOR_DELETE */
  41
  42   /// Have we been dynamically created?
  43   bool dynamic_;
  44 };
  45
  46 void*
  47 A::operator new (size_t n)
  48 {
  49   ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
  50
  51   if (dynamic_instance == 0)
  52     {
  53       // If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
  54       // keys.  Try using ACE_HAS_TSS_EMULATION, or increasing
  55       // ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
  56       ACE_TEST_ASSERT (dynamic_instance != 0);
  57
  58       ACE_throw_bad_alloc;
  59     }
  60   else
  61     {
  62       // Allocate the memory and store it (usually in thread-specific
  63       // storage, depending on config flags).
  64       dynamic_instance->set ();
  65
  66       return ::new char[n];
  67     }
  68 }
  69
  70 #if defined (ACE_HAS_NEW_NOTHROW)
  71 void*
  72 A::operator new (size_t n, const ACE_nothrow_t&) throw()
  73 {
  74   ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
  75
  76   if (dynamic_instance == 0)
  77     {
  78       // If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
  79       // keys.  Try using ACE_HAS_TSS_EMULATION, or increasing
  80       // ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
  81       ACE_TEST_ASSERT (dynamic_instance != 0);
  82
  83       return 0;
  84     }
  85   else
  86     {
  87       // Allocate the memory and store it (usually in thread-specific
  88       // storage, depending on config flags).
  89       dynamic_instance->set ();
  90
  91       return ::new(ACE_nothrow) char[n];
  92     }
  93 }
  94
  95 #if !defined (ACE_LACKS_PLACEMENT_OPERATOR_DELETE)
  96 void
  97 A::operator delete (void *p, const ACE_nothrow_t&) throw()
  98 {
  99   ::delete [] static_cast <char *> (p);
 100 }
 101 #endif /* ACE_LACKS_PLACEMENT_OPERATOR_DELETE */
 102
 103 #endif /* ACE_HAS_NEW_NOTHROW */
 104
 105 void
 106 A::operator delete (void *obj)
 107 {
 108   ::delete [] static_cast <char *> (obj);
 109 }
 110
 111 A::A()
 112 {
 113   this->dynamic_ = ACE_Dynamic::instance ()->is_dynamic ();
 114
 115   if (this->dynamic_)
 116     // Make sure to reset the flag.
 117     ACE_Dynamic::instance ()->reset ();
 118 }
 119
 120 int
 121 run_main (int, ACE_TCHAR *[])
 122 {
 123   ACE_START_TEST (ACE_TEXT ("Dynamic_Test"));
 124   A from_stack;
 125   A* heap = 0;
 126   ACE_NEW_RETURN (heap, A, 1);
 127   if (from_stack.dynamic_)
 128     {
 129       ACE_ERROR_RETURN ((LM_ERROR,
 130                           ACE_TEXT ("dynamic_ is true for an object on the stack\n")),
 131                         1);
 132     }
 133
 134   if (!heap->dynamic_)
 135     {
 136       ACE_ERROR_RETURN ((LM_ERROR,
 137                           ACE_TEXT ("dynamic_ is false for an object from the heap\n")),
 138                         1);
 139     }
 140
 141   delete heap;
 142   ACE_END_TEST;
 143   return 0;
 144 }
 145