TAO/tests/POA/Default_Servant2/File_i.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     File_i.cpp
   5  *
   6  *    Implementation of the File IDL module and the interfaces
   7  *    Descriptor and System in it.
   8  *
   9  *  @author  Irfan Pyarali
  10  */
  11 //=============================================================================
  12
  13
  14 #include "File_i.h"
  15 #include "ace/OS_NS_stdio.h"
  16 #include "ace/OS_NS_fcntl.h"
  17 #include "ace/OS_NS_unistd.h"
  18 #include "ace/Truncate.h"
  19 #include "tao/PortableServer/PortableServer.h"
  20
  21 // IDL File::System constructor
  22 FileImpl::System::System (CORBA::ORB_ptr orb, PortableServer::POA_ptr poa)
  23   : orb_ (CORBA::ORB::_duplicate (orb)),
  24     poa_ (PortableServer::POA::_duplicate (poa)),
  25     // Create the Default Descriptor Servant
  26     fd_servant_ (poa)
  27 {
  28   // set the default servant of the POA
  29   poa->set_servant (&this->fd_servant_);
  30 }
  31
  32 FileImpl::System::~System ()
  33 {
  34 }
  35
  36 PortableServer::POA_ptr
  37 FileImpl::System::_default_POA ()
  38 {
  39   return PortableServer::POA::_duplicate (this->poa_.in ());
  40 }
  41
  42 File::Descriptor_ptr
  43 FileImpl::System::open (const char *file_name,
  44                         CORBA::Long flags)
  45 {
  46   // Do an ACE_OS::open
  47   ACE_HANDLE file_descriptor = ACE_OS::open (file_name,
  48                                              flags,
  49                                              ACE_DEFAULT_FILE_PERMS);
  50
  51   if (file_descriptor == ACE_INVALID_HANDLE)
  52     {
  53       throw File::IOError ();
  54     }
  55
  56   char file_descriptor_buffer[BUFSIZ];
  57
  58   // convert ACE_HANDLE to a string
  59   ACE_OS::sprintf (file_descriptor_buffer,
  60                    "%ld",
  61                    ACE_Utils::truncate_cast<long int> ((intptr_t)file_descriptor));
  62
  63   //Create an objectID from the ACE_HANDLE string
  64   PortableServer::ObjectId_var oid =
  65     PortableServer::string_to_ObjectId (file_descriptor_buffer);
  66
  67   // create an object reference with the specified ObjectID got
  68   // from ACE_HANDLE string
  69   CORBA::Object_var obj =
  70     this->poa_->create_reference_with_id (oid.in (),
  71                                           "IDL:File/Descriptor:1.0");
  72
  73   // Narrow the object reference to a File Descriptor
  74   File::Descriptor_var fd =
  75     File::Descriptor::_narrow (obj.in ());
  76
  77
  78   return fd._retn ();
  79 }
  80
  81 void
  82 FileImpl::System::shutdown ()
  83 {
  84   this->orb_->shutdown (false);
  85 }
  86
  87 // IDL File::Descriptor constructor
  88 FileImpl::Descriptor::Descriptor (PortableServer::POA_ptr poa)
  89   : poa_ (PortableServer::POA::_duplicate (poa))
  90 {
  91 }
  92
  93 FileImpl::Descriptor::~Descriptor ()
  94 {
  95 }
  96
  97 PortableServer::POA_ptr
  98 FileImpl::Descriptor::_default_POA ()
  99 {
 100   return PortableServer::POA::_duplicate (this->poa_.in ());
 101 }
 102
 103 //Extracts the ACE_HANDLE from the passed object reference
 104 ACE_HANDLE
 105 FileImpl::Descriptor::fd ()
 106 {
 107   //
 108   // One way of getting our id.
 109   //
 110
 111   // Get a reference to myself
 112   File::Descriptor_var me = this->_this ();
 113
 114   // Get the ObjectId from the reference
 115   PortableServer::ObjectId_var oid1 =
 116     this->poa_->reference_to_id (me.in ());
 117
 118   //
 119   // Another way of getting our id.
 120   //
 121
 122   PortableServer::ObjectId_var oid2 =
 123     this->poa_->servant_to_id (this);
 124
 125   //
 126   // Yet another way of getting our id.
 127   //
 128
 129   int argc = 0;
 130   ACE_TCHAR **argv = 0;
 131   CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 132
 133   // Get the POA Current object reference
 134   CORBA::Object_var obj =
 135     orb->resolve_initial_references ("POACurrent");
 136
 137   // Narrow the object reference to a POA Current reference
 138   PortableServer::Current_var poa_current =
 139     PortableServer::Current::_narrow (obj.in ());
 140
 141   PortableServer::ObjectId_var oid3 =
 142     poa_current->get_object_id ();
 143
 144   ACE_ASSERT (oid1.in () == oid2.in ());
 145   ACE_ASSERT (oid2.in () == oid3.in ());
 146
 147   // Convert the ObjectId to a string
 148   CORBA::String_var s =
 149     PortableServer::ObjectId_to_string (oid1.in ());
 150
 151   // Get the ACE_HANDLE from the string
 152   return (ACE_HANDLE)(intptr_t)ACE_OS::atol (s.in ());
 153 }
 154
 155 CORBA::Long
 156 FileImpl::Descriptor::write (const File::Descriptor::DataBuffer &buffer)
 157 {
 158   ACE_HANDLE file_descriptor = this->fd ();
 159
 160   const CORBA::Octet *data = &buffer[0];
 161
 162   ssize_t len = ACE_OS::write (file_descriptor,
 163                                data,
 164                                buffer.length ());
 165   if (len > 0)
 166     return len;
 167
 168   throw File::IOError ();
 169 }
 170
 171 File::Descriptor::DataBuffer *
 172 FileImpl::Descriptor::read (CORBA::Long num_bytes)
 173 {
 174   ACE_HANDLE file_descriptor = this->fd ();
 175
 176   CORBA::Octet *buffer = File::Descriptor::DataBuffer::allocbuf (num_bytes);
 177   int length = ACE_OS::read (file_descriptor, buffer, num_bytes);
 178
 179   if (length > 0)
 180     return new File::Descriptor::DataBuffer (length,
 181                                              length,
 182                                              buffer,
 183                                              1);
 184
 185   File::Descriptor::DataBuffer::freebuf (buffer);
 186   throw File::IOError ();
 187 }
 188
 189 CORBA::ULong
 190 FileImpl::Descriptor::lseek (CORBA::ULong offset,
 191                              CORBA::Long whence)
 192 {
 193   ACE_HANDLE file_descriptor = this->fd ();
 194
 195   CORBA::Long result = (CORBA::Long) ACE_OS::lseek (file_descriptor,
 196                                                     offset,
 197                                                     whence);
 198   if (result == -1)
 199     throw File::IOError ();
 200
 201   return (CORBA::ULong) result;
 202 }
 203
 204 void
 205 FileImpl::Descriptor::destroy ()
 206 {
 207   // Get the ACE_HANDLE for this object reference
 208   ACE_HANDLE file_descriptor = this->fd ();
 209
 210   // Close the file corresponding to this object reference.
 211   ACE_OS::close (file_descriptor);
 212 }