src/LifeCycleCORBA/SALOME_FileTransferCORBA.cxx

   1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 //  File   : SALOME_FileTransferCORBA.cxx
  23 //  Author : Paul RASCLE, EDF
  24 //  Module : SALOME
  25 //  $Header$
  26 //
  27 #include "SALOME_FileTransferCORBA.hxx"
  28 #include "SALOME_LifeCycleCORBA.hxx"
  29 #include "utilities.h"
  30 #include "Basics_Utils.hxx"
  31
  32 using namespace std;
  33
  34 /*! \class SALOME_FileTransferCORBA
  35     \brief A class to manage file transfer in SALOME (CORBA context)
  36
  37 */
  38
  39 //=============================================================================
  40 /*!
  41  *  Default constructor, not for use.
  42  */
  43 //=============================================================================
  44
  45 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA()
  46 {
  47   ASSERT(0);
  48 }
  49
  50 //=============================================================================
  51 /*! \brief Constructor to use when we get a fileRef CORBA object from a component
  52  *
  53  *  \param aFileRef file reference CORBA object
  54  */
  55 //=============================================================================
  56
  57 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(Engines::fileRef_ptr
  58                                                    aFileRef)
  59 {
  60   MESSAGE("SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(aFileRef)");
  61   _theFileRef = aFileRef;
  62 }
  63
  64 //=============================================================================
  65 /*! \brief Constructor to use when we have the file name and machine from which to
  66  *         copy, plus an optional Container name on the machine.
  67  *
  68  *  \param refMachine    the machine on which is the file to transfer
  69  *  \param origFileName  abolute file path on refMachine
  70  *  \param containerName default container name used (FactoryServer) if empty
  71  */
  72 //=============================================================================
  73
  74 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(string refMachine,
  75                                                    string origFileName,
  76                                                    string containerName)
  77 {
  78   MESSAGE("SALOME_FileTransferCORBA::SALOME_FileTransferCORBA"
  79           << refMachine << " " << origFileName  << " " << containerName);
  80   _refMachine = refMachine;
  81   _origFileName = origFileName;
  82   _containerName = containerName;
  83   if (_refMachine.empty() || _origFileName.empty())
  84     {
  85       INFOS("bad parameters: machine and file name must be given");
  86     }
  87 }
  88
  89 //=============================================================================
  90 /*!
  91  *  Destructor
  92  */
  93 //=============================================================================
  94
  95 SALOME_FileTransferCORBA::~SALOME_FileTransferCORBA()
  96 {
  97   MESSAGE("SALOME_FileTransferCORBA::~SALOME_FileTransferCORBA");
  98 }
  99
 100 //=============================================================================
 101 /*! \brief Get a local copy of the reference file
 102  *
 103  *  CORBA method
 104  *  \param  localFile optional absolute path to store the copy
 105  *  \return the file name (absolute path) of the copy, may be different from
 106  *          localFile parameter if the copy was already done before the call
 107  */
 108 //=============================================================================
 109
 110 string SALOME_FileTransferCORBA::getLocalFile(string localFile)
 111 {
 112   MESSAGE("SALOME_FileTransferCORBA::getLocalFile " << localFile);
 113
 114   Engines::Container_var container;
 115
 116   if (CORBA::is_nil(_theFileRef))
 117     {
 118       if (_refMachine.empty() || _origFileName.empty())
 119         {
 120           INFOS("not enough parameters: machine and file name must be given");
 121           return "";
 122         }
 123
 124       SALOME_LifeCycleCORBA LCC;
 125       Engines::ContainerManager_var contManager = LCC.getContainerManager();
 126       Engines::ResourcesManager_var resManager = LCC.getResourcesManager();
 127
 128       Engines::MachineParameters params;
 129       LCC.preSet(params);
 130       params.container_name = _containerName.c_str();
 131       params.hostname = _refMachine.c_str();
 132
 133       Engines::CompoList clist;
 134       Engines::MachineList_var listOfMachines =
 135         resManager->GetFittingResources(params, clist);
 136
 137       container = contManager->FindOrStartContainer(params, listOfMachines);
 138       if (CORBA::is_nil(container))
 139         {
 140           INFOS("machine " << _refMachine << " unreachable");
 141           return "";
 142         }
 143
 144       _theFileRef = container->createFileRef(_origFileName.c_str());
 145       if (CORBA::is_nil(_theFileRef))
 146         {
 147           INFOS("impossible to create fileRef on " << _refMachine);
 148           return "";
 149         }
 150     }
 151
 152   container = _theFileRef->getContainer();
 153   ASSERT(! CORBA::is_nil(container));
 154
 155   string myMachine = Kernel_Utils::GetHostname();
 156   string localCopy = _theFileRef->getRef(myMachine.c_str());
 157
 158   if (localCopy.empty()) // no existing copy available
 159     {
 160       if (localFile.empty()) // no name provided for local copy
 161         {
 162           char bufName[256];
 163           localCopy = tmpnam(bufName);
 164           localFile = bufName;
 165           SCRUTE(localFile);
 166         }
 167
 168       FILE* fp;
 169       if ((fp = fopen(localFile.c_str(),"wb")) == NULL)
 170         {
 171           INFOS("file " << localFile << " cannot be open for writing");
 172           return "";
 173         }
 174
 175       Engines::fileTransfer_var fileTransfer = container->getFileTransfer();
 176       ASSERT(! CORBA::is_nil(fileTransfer));
 177
 178       CORBA::Long fileId = fileTransfer->open(_origFileName.c_str());
 179       if (fileId > 0)
 180         {
 181           Engines::fileBlock* aBlock;
 182           int toFollow = 1;
 183           int ctr=0;
 184           while (toFollow)
 185             {
 186               ctr++;
 187               SCRUTE(ctr);
 188               aBlock = fileTransfer->getBlock(fileId);
 189               toFollow = aBlock->length();
 190               SCRUTE(toFollow);
 191               CORBA::Octet *buf = aBlock->get_buffer();
 192 #if defined(_DEBUG_) || defined(_DEBUG)
 193               int nbWri = fwrite(buf, sizeof(CORBA::Octet), toFollow, fp);
 194               ASSERT(nbWri == toFollow);
 195 #else
 196               fwrite(buf, sizeof(CORBA::Octet), toFollow, fp);
 197 #endif
 198               delete aBlock;
 199             }
 200           fclose(fp);
 201           MESSAGE("end of transfer");
 202           fileTransfer->close(fileId);
 203           _theFileRef->addRef(myMachine.c_str(), localFile.c_str());
 204           localCopy = localFile;
 205         }
 206       else
 207         {
 208           INFOS("open reference file for copy impossible");
 209           return "";
 210         }
 211
 212     }
 213   SCRUTE(localCopy);
 214   return localCopy;
 215 }