TAO/tests/Compression/bzip2server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/OS_NS_stdio.h"
   3 #include "tao/ORB.h"
   4 #include "tao/Compression/Compression.h"
   5 #include "tao/Compression/bzip2/Bzip2Compressor_Factory.h"
   6
   7 bool
   8 test_invalid_compression_factory (Compression::CompressionManager_ptr cm)
   9 {
  10   bool succeed = false;
  11   try
  12     {
  13       // Get an invalid compression factory
  14       Compression::CompressorFactory_var factory =
  15         cm->get_factory (100);
  16     }
  17   catch (const Compression::UnknownCompressorId&)
  18     {
  19       succeed = true;
  20     }
  21   catch (const CORBA::Exception&)
  22     {
  23     }
  24
  25   if (!succeed)
  26   {
  27     ACE_ERROR ((LM_ERROR,
  28                 "(%t) ERROR, get invalid compression factory failed\n"));
  29   }
  30
  31   return succeed;
  32 }
  33
  34 bool
  35 test_duplicate_compression_factory (
  36   Compression::CompressionManager_ptr cm,
  37   Compression::CompressorFactory_ptr cf)
  38 {
  39   bool succeed = false;
  40   try
  41     {
  42       // Register duplicate
  43       cm->register_factory (cf);
  44     }
  45   catch (const Compression::FactoryAlreadyRegistered&)
  46     {
  47       succeed = true;
  48     }
  49   catch (const CORBA::Exception&)
  50     {
  51     }
  52
  53   if (!succeed)
  54   {
  55     ACE_ERROR ((LM_ERROR,
  56                 "(%t) ERROR, register duplicate factory failed\n"));
  57   }
  58
  59   return succeed;
  60 }
  61
  62 bool
  63 test_register_nil_compression_factory (
  64   Compression::CompressionManager_ptr cm)
  65 {
  66   bool succeed = false;
  67   try
  68     {
  69       // Register nil factory
  70       cm->register_factory (Compression::CompressorFactory::_nil());
  71     }
  72   catch (const CORBA::BAD_PARAM& ex)
  73     {
  74       if ((ex.minor() & 0xFFFU) == 44)
  75         {
  76           succeed = true;
  77         }
  78     }
  79   catch (const CORBA::Exception&)
  80     {
  81     }
  82
  83   if (!succeed)
  84   {
  85     ACE_ERROR ((LM_ERROR,
  86                 "(%t) ERROR, register nill factory failed\n"));
  87   }
  88
  89   return succeed;
  90 }
  91
  92 bool
  93 test_compression (CORBA::ULong nelements,
  94               Compression::CompressionManager_ptr cm)
  95 {
  96   bool succeed = false;
  97
  98   CORBA::OctetSeq mytest;
  99   mytest.length (nelements);
 100   for (CORBA::ULong j = 0; j != nelements; ++j)
 101     {
 102       mytest[j] = 'a';
 103     }
 104
 105   Compression::Compressor_var compressor = cm->get_compressor (
 106     ::Compression::COMPRESSORID_BZIP2, 6);
 107
 108   CORBA::OctetSeq myout;
 109   myout.length ((CORBA::ULong)(mytest.length() * 1.1));
 110
 111   compressor->compress (mytest, myout);
 112
 113   CORBA::OctetSeq decompress;
 114   decompress.length (nelements);
 115
 116   compressor->decompress (myout, decompress);
 117
 118   if (decompress != mytest)
 119     {
 120       ACE_ERROR ((LM_ERROR, "Error, decompress not working\n"));
 121     }
 122   else
 123     {
 124       succeed = true;
 125       ACE_DEBUG ((LM_DEBUG, "Compression worked with bzip2, original "
 126                             "size %d, compressed size %d\n",
 127                             mytest.length(), myout.length ()));
 128     }
 129   return succeed;
 130 }
 131
 132 int
 133 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 134 {
 135   int retval = 0;
 136   try
 137     {
 138       CORBA::ORB_var orb =
 139         CORBA::ORB_init (argc, argv);
 140
 141       CORBA::Object_var compression_manager =
 142         orb->resolve_initial_references("CompressionManager");
 143
 144       Compression::CompressionManager_var manager =
 145         Compression::CompressionManager::_narrow (compression_manager.in ());
 146
 147       if (CORBA::is_nil(manager.in ()))
 148         ACE_ERROR_RETURN ((LM_ERROR,
 149                            " (%P|%t) Panic: nil compression manager\n"),
 150                           1);
 151
 152       Compression::CompressorFactory_ptr compressor_factory;
 153
 154       ACE_NEW_RETURN (compressor_factory, TAO::Bzip2_CompressorFactory (), 1);
 155
 156       Compression::CompressorFactory_var compr_fact = compressor_factory;
 157       manager->register_factory(compr_fact.in ());
 158
 159       if (!test_duplicate_compression_factory (manager.in (), compr_fact.in ()))
 160         retval = 1;
 161
 162       if (!test_register_nil_compression_factory (manager.in ()))
 163         retval = 1;
 164
 165       if (!test_compression (1024, manager.in ()))
 166         retval = 1;
 167
 168       if (!test_compression (5, manager.in ()))
 169         retval = 1;
 170
 171       if (!test_invalid_compression_factory (manager.in ()))
 172         retval = 1;
 173
 174       orb->destroy ();
 175     }
 176   catch (const CORBA::Exception& ex)
 177     {
 178       ex._tao_print_exception ("Exception caught:");
 179       retval = 1;
 180     }
 181
 182   return retval;
 183 }