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 #include <cstdio>
  32
  33 using namespace std;
  34
  35 /*! \class SALOME_FileTransferCORBA
  36     \brief A class to manage file transfer in SALOME (CORBA context)
  37
  38 */
  39
  40 //=============================================================================
  41 /*!
  42  *  Default constructor, not for use.
  43  */
  44 //=============================================================================
  45
  46 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA()
  47 {
  48   ASSERT(0);
  49 }
  50
  51 //=============================================================================
  52 /*! \brief Constructor to use when we get a fileRef CORBA object from a component
  53  *
  54  *  \param aFileRef file reference CORBA object
  55  */
  56 //=============================================================================
  57
  58 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(Engines::fileRef_ptr
  59                                                    aFileRef)
  60 {
  61   MESSAGE("SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(aFileRef)");
  62   _theFileRef = aFileRef;
  63 }
  64
  65 //=============================================================================
  66 /*! \brief Constructor to use when we have the file name and machine from which to
  67  *         copy, plus an optional Container name on the machine.
  68  *
  69  *  \param refMachine    the machine on which is the file to transfer
  70  *  \param origFileName  abolute file path on refMachine
  71  *  \param containerName default container name used (FactoryServer) if empty
  72  */
  73 //=============================================================================
  74
  75 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(string refMachine,
  76                                                    string origFileName,
  77                                                    string containerName)
  78 {
  79   MESSAGE("SALOME_FileTransferCORBA::SALOME_FileTransferCORBA"
  80           << refMachine << " " << origFileName  << " " << containerName);
  81   _refMachine = refMachine;
  82   _origFileName = origFileName;
  83   _containerName = containerName;
  84   if (_refMachine.empty() || _origFileName.empty())
  85     {
  86       INFOS("bad parameters: machine and file name must be given");
  87     }
  88 }
  89
  90 //=============================================================================
  91 /*!
  92  *  Destructor
  93  */
  94 //=============================================================================
  95
  96 SALOME_FileTransferCORBA::~SALOME_FileTransferCORBA()
  97 {
  98   MESSAGE("SALOME_FileTransferCORBA::~SALOME_FileTransferCORBA");
  99 }
 100
 101 //=============================================================================
 102 /*! \brief Get a local copy of the reference file
 103  *
 104  *  CORBA method
 105  *  \param  localFile optional absolute path to store the copy
 106  *  \return the file name (absolute path) of the copy, may be different from
 107  *          localFile parameter if the copy was already done before the call
 108  */
 109 //=============================================================================
 110
 111 string SALOME_FileTransferCORBA::getLocalFile(string localFile)
 112 {
 113   MESSAGE("SALOME_FileTransferCORBA::getLocalFile " << localFile);
 114
 115   Engines::Container_var container;
 116
 117   if (CORBA::is_nil(_theFileRef))
 118     {
 119       if (_refMachine.empty() || _origFileName.empty())
 120         {
 121           INFOS("not enough parameters: machine and file name must be given");
 122           return "";
 123         }
 124
 125       SALOME_LifeCycleCORBA LCC;
 126       Engines::ContainerManager_var contManager = LCC.getContainerManager();
 127       Engines::ResourcesManager_var resManager = LCC.getResourcesManager();
 128
 129       Engines::MachineParameters params;
 130       LCC.preSet(params);
 131       params.container_name = _containerName.c_str();
 132       params.hostname = _refMachine.c_str();
 133
 134       Engines::ContainerParameters new_params;
 135       LCC.convert(params, new_params);
 136       new_params.mode = CORBA::string_dup("findorstart");
 137       container = contManager->GiveContainer(new_params);
 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 }