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