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