src/LifeCycleCORBA/SALOME_FileTransferCORBA.cxx

   1 // Copyright (C) 2007-2023  CEA, EDF, 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, or (at your option) any later version.
  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
  23 //  File   : SALOME_FileTransferCORBA.cxx
  24 //  Author : Paul RASCLE, EDF
  25 //  Module : SALOME
  26 //  $Header$
  27 //
  28 #include "SALOME_FileTransferCORBA.hxx"
  29 #include "SALOME_LifeCycleCORBA.hxx"
  30 #include "utilities.h"
  31 #include "Basics_Utils.hxx"
  32 #include <cstdio>
  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  absolute file path on refMachine
  70  *  \param containerName default container name used (FactoryServer) if empty
  71  */
  72 //=============================================================================
  73
  74 SALOME_FileTransferCORBA::SALOME_FileTransferCORBA(std::string refMachine,
  75                                                    std::string origFileName,
  76                                                    std::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 std::string SALOME_FileTransferCORBA::getLocalFile(std::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::ContainerParameters params;
 129       LCC.preSet(params);
 130       params.container_name = _containerName.c_str();
 131       params.resource_params.hostname = _refMachine.c_str();
 132       params.mode = CORBA::string_dup("findorstart");
 133       container = contManager->GiveContainer(params);
 134       if (CORBA::is_nil(container))
 135         {
 136           INFOS("machine " << _refMachine << " unreachable");
 137           return "";
 138         }
 139
 140       _theFileRef = container->createFileRef(_origFileName.c_str());
 141       if (CORBA::is_nil(_theFileRef))
 142         {
 143           INFOS("impossible to create fileRef on " << _refMachine);
 144           return "";
 145         }
 146     }
 147
 148   container = _theFileRef->getContainer();
 149   ASSERT(! CORBA::is_nil(container));
 150
 151   std::string myMachine = Kernel_Utils::GetHostname();
 152   std::string localCopy = _theFileRef->getRef(myMachine.c_str());
 153
 154   if (localCopy.empty()) // no existing copy available
 155     {
 156       if (localFile.empty()) // no name provided for local copy
 157         {
 158           char bufName[256];
 159           localCopy = tmpnam(bufName);
 160           localFile = bufName;
 161           SCRUTE(localFile);
 162         }
 163
 164       FILE* fp;
 165       if ((fp = fopen(localFile.c_str(),"wb")) == NULL)
 166         {
 167           INFOS("file " << localFile << " cannot be open for writing");
 168           return "";
 169         }
 170
 171       Engines::fileTransfer_var fileTransfer = container->getFileTransfer();
 172       ASSERT(! CORBA::is_nil(fileTransfer));
 173
 174       CORBA::Long fileId = fileTransfer->open(_origFileName.c_str());
 175       if (fileId > 0)
 176         {
 177           Engines::fileBlock* aBlock;
 178           int toFollow = 1;
 179           int ctr=0;
 180           while (toFollow)
 181             {
 182               ctr++;
 183               SCRUTE(ctr);
 184               aBlock = fileTransfer->getBlock(fileId);
 185               toFollow = aBlock->length();
 186               SCRUTE(toFollow);
 187               CORBA::Octet *buf = aBlock->get_buffer();
 188 #if defined(_DEBUG_) || defined(_DEBUG)
 189               int nbWri = fwrite(buf, sizeof(CORBA::Octet), toFollow, fp);
 190               ASSERT(nbWri == toFollow);
 191 #else
 192               fwrite(buf, sizeof(CORBA::Octet), toFollow, fp);
 193 #endif
 194               delete aBlock;
 195             }
 196           fclose(fp);
 197           MESSAGE("end of transfer");
 198           fileTransfer->close(fileId);
 199           _theFileRef->addRef(myMachine.c_str(), localFile.c_str());
 200           localCopy = localFile;
 201         }
 202       else
 203         {
 204           INFOS("open reference file for copy impossible");
 205           return "";
 206         }
 207
 208     }
 209   SCRUTE(localCopy);
 210   return localCopy;
 211 }