src/Local/Batch_BatchManager_Local_SSH.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 /*
  23  * BatchManager_Local_SSH.cxx :
  24  *
  25  * Auteur : Ivan DUTKA-MALEN - EDF R&D
  26  * Mail   : mailto:ivan.dutka-malen@der.edf.fr
  27  * Date   : Thu Nov  6 10:17:22 2003
  28  * Projet : Salome 2
  29  *
  30  */
  31
  32 #ifdef HAVE_CONFIG_H
  33 #  include <SALOMEconfig.h>
  34 #endif
  35
  36 #include <iostream>
  37 #include <fstream>
  38 #include <sstream>
  39 #include <cstdlib>
  40 #include <sys/types.h>
  41 #ifndef WIN32
  42 #include <sys/wait.h>
  43 #include <unistd.h>
  44 #endif
  45 #include <ctime>
  46
  47 #include <pthread.h>
  48 #include <signal.h>
  49 #include <errno.h>
  50 #include <string.h>
  51 #include "Batch_IOMutex.hxx"
  52 #include "Batch_BatchManager_Local_SSH.hxx"
  53
  54 #include "Batch_config.h"
  55
  56 #ifndef RM
  57 #error "RM undefined. You must set RM to a valid path to a rm-like command."
  58 #endif
  59
  60 #ifndef SCP
  61 #error "SCP undefined. You must set SCP to a valid path to a scp-like command."
  62 #endif
  63
  64 #ifndef SSH
  65 #error "SSH undefined. You must set SSH to a valid path to a ssh-like command."
  66 #endif
  67
  68 using namespace std;
  69
  70 namespace Batch {
  71
  72
  73   // Constructeur
  74   BatchManager_Local_SSH::BatchManager_Local_SSH(const FactBatchManager * parent, const char * host) throw(InvalidArgumentException,ConnexionFailureException) : BatchManager_Local(parent, host)
  75   {
  76   }
  77
  78   // Destructeur
  79   BatchManager_Local_SSH::~BatchManager_Local_SSH()
  80   {
  81   }
  82
  83
  84   // Methode abstraite qui renvoie la commande de copie du fichier source en destination
  85   string BatchManager_Local_SSH::copy_command(const std::string & user_source,
  86                                               const std::string & host_source,
  87                                               const std::string & source,
  88                                               const std::string & user_destination,
  89                                               const std::string & host_destination,
  90                                               const std::string & destination) const
  91   {
  92     ostringstream fullsource;
  93     if (host_source.size() != 0) {
  94       fullsource << host_source << ":";
  95     }
  96     fullsource << source;
  97
  98     ostringstream fulldestination;
  99     if (host_destination.size() != 0) {
 100       fulldestination << host_destination << ":";
 101     }
 102     fulldestination << destination;
 103
 104     ostringstream copy_cmd;
 105     // Option -p is used to keep the same permissions for the destination file (particularly useful to keep scripts
 106     // executable when copying them)
 107     copy_cmd << SCP << " -p " << fullsource.str() << " " << fulldestination.str();
 108     return copy_cmd.str();
 109   }
 110
 111   // Methode abstraite qui renvoie la commande a executer
 112   string BatchManager_Local_SSH::exec_command(Parametre & param) const
 113   {
 114     ostringstream exec_sub_cmd;
 115     exec_sub_cmd << "cd " << param[WORKDIR] << ";";
 116     exec_sub_cmd << param[EXECUTABLE];
 117
 118     if (param.find(ARGUMENTS) != param.end()) {
 119       Versatile V = param[ARGUMENTS];
 120       for(Versatile::const_iterator it=V.begin(); it!=V.end(); it++) {
 121         StringType argt = * static_cast<StringType *>(*it);
 122         string     arg  = argt;
 123         exec_sub_cmd << " " << arg;
 124       }
 125     }
 126
 127
 128     Versatile new_arguments;
 129     new_arguments.setMaxSize(0);
 130     new_arguments = string(param[EXECUTIONHOST]);
 131
 132
 133     if (param.find(USER) != param.end()) {
 134       new_arguments += "-l";
 135       new_arguments += string(param[USER]);
 136     }
 137
 138     new_arguments += exec_sub_cmd.str();
 139
 140     param[ARGUMENTS] = new_arguments;
 141
 142     // Sous Linux on est oblige de modifier ces deux parametres pour faire fonctionner la commande rsh
 143     param[EXECUTABLE] = SSH;
 144     param.erase(NAME);
 145
 146     return SSH;
 147   }
 148
 149   // Methode qui renvoie la commande d'effacement du fichier
 150   string BatchManager_Local_SSH::remove_command(const std::string & user_destination,
 151                                                 const std::string & host_destination,
 152                                                 const std::string & destination) const
 153   {
 154     string host = (host_destination.size()) ? host_destination : "localhost:";
 155
 156     ostringstream remove_cmd;
 157     remove_cmd << SSH << " " << host << " \"" << RM << " " << destination << "\"";
 158     return remove_cmd.str();
 159   }
 160 }