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 = cm->get_factory (100);
  15     }
  16   catch (const Compression::UnknownCompressorId&)
  17     {
  18       succeed = true;
  19     }
  20   catch (const CORBA::Exception&)
  21     {
  22     }
  23
  24   if (!succeed)
  25   {
  26     ACE_ERROR ((LM_ERROR,
  27                 "(%t) ERROR, get invalid compression factory failed\n"));
  28   }
  29
  30   return succeed;
  31 }
  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 (::Compression::COMPRESSORID_ZLIB, 6);
 106
 107   CORBA::OctetSeq myout;
 108   myout.length ((CORBA::ULong)(mytest.length() * 1.1));
 109
 110   compressor->compress (mytest, myout);
 111
 112   CORBA::OctetSeq decompress;
 113   decompress.length (nelements);
 114
 115   compressor->decompress (myout, decompress);
 116
 117   if (decompress != mytest)
 118     {
 119       ACE_ERROR ((LM_ERROR, "Error, decompress not working\n"));
 120     }
 121   else
 122     {
 123       succeed = true;
 124       ACE_DEBUG ((LM_DEBUG, "Compression worked with zlib, original "
 125                             "size %d, compressed size %d\n",
 126                             mytest.length(), myout.length ()));
 127     }
 128   return succeed;
 129 }
 130
 131 int
 132 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 133 {
 134   int retval = 0;
 135   try
 136     {
 137       CORBA::ORB_var orb =
 138         CORBA::ORB_init (argc, argv);
 139
 140       CORBA::Object_var compression_manager =
 141         orb->resolve_initial_references("CompressionManager");
 142
 143       Compression::CompressionManager_var manager =
 144         Compression::CompressionManager::_narrow (compression_manager.in ());
 145
 146       if (CORBA::is_nil(manager.in ()))
 147         ACE_ERROR_RETURN ((LM_ERROR,
 148                            " (%P|%t) Panic: nil compression manager\n"),
 149                           1);
 150
 151       Compression::CompressorFactory_ptr compressor_factory;
 152
 153       ACE_NEW_RETURN (compressor_factory, TAO::Zlib_CompressorFactory (), 1);
 154
 155       Compression::CompressorFactory_var compr_fact = compressor_factory;
 156       manager->register_factory(compr_fact.in ());
 157
 158       if (!test_duplicate_compression_factory (manager.in (), compr_fact.in ()))
 159         retval = 1;
 160
 161       if (!test_register_nil_compression_factory (manager.in ()))
 162         retval = 1;
 163
 164       if (!test_compression (1024, manager.in ()))
 165         retval = 1;
 166
 167       if (!test_compression (5, manager.in ()))
 168         retval = 1;
 169
 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 }