src/Core/Utils.cxx

   1 // Copyright (C) 2007-2021  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, 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  * Utils.cxx
  24  *
  25  *  Created on: 30 jan. 2012
  26  *  Author : Renaud BARATE - EDF R&D
  27  */
  28
  29 #include <cstdlib>
  30 #include <cstdio>
  31 #include <iostream>
  32 #include <fstream>
  33 #include <time.h>
  34
  35 #ifdef WIN32
  36 #  include <Windows.h>
  37 #  include <io.h>
  38 #  define popen _popen
  39 #  define pclose _pclose
  40 #else
  41 #  include <sys/stat.h>
  42 #  include <unistd.h>
  43 #endif
  44
  45 #include <libbatch_config.h>
  46 #include "Utils.hxx"
  47 #include "RunTimeException.hxx"
  48
  49 using namespace std;
  50 namespace Batch {
  51
  52 int Utils::getCommandOutput(const string & command, string & output)
  53 {
  54   // Reinitialize output
  55   output = "";
  56
  57   // Call command
  58   FILE * fp = popen(command.c_str(), "r");
  59   if (fp == NULL) {
  60     return -1;
  61   }
  62
  63   // Read the output and store it
  64   char buf[1024];
  65   while (fgets(buf, sizeof(buf), fp) != NULL) {
  66     output += buf;
  67   }
  68
  69   // close and get status
  70   int status = pclose(fp);
  71   return status;
  72 }
  73
  74 bool Utils::isAbsolutePath(const string & path)
  75 {
  76   if (path.size() == 0)
  77     return false;
  78 #ifdef WIN32
  79   // On Windows, absolute paths may begin with something like "C:"
  80   if (path[1] == ':')
  81     return true;
  82 #endif
  83   return path[0] == '/';
  84 }
  85
  86 std::string Utils::dirname(const std::string & path)
  87 {
  88 #ifdef WIN32
  89   const char separator = '\\';
  90 #else
  91   const char separator = '/';
  92 #endif
  93   std::size_t found = path.rfind(separator);
  94   if(found != std::string::npos)
  95     return path.substr(0, found+1);
  96   else
  97     return std::string(".");
  98 }
  99
 100 bool Utils::isOption(const std::string & val)
 101 {
 102   return val.size() > 0 && val[0] == '-';
 103 }
 104
 105 bool Utils::usesRsyncRelativePath(const std::string & path)
 106 {
 107   return path.find("/./") != std::string::npos;
 108 }
 109
 110 string Utils::createAndOpenTemporaryFile(const string & prefix, ofstream & outputStream)
 111 {
 112   if (outputStream.is_open())
 113     outputStream.close();
 114
 115   // Find directory for temporary files
 116   const char * tmpDirName = getenv("TEMP");
 117   if (tmpDirName == NULL) tmpDirName = getenv("TMP");
 118   if (tmpDirName == NULL) tmpDirName = getenv("TEMPDIR");
 119   if (tmpDirName == NULL) tmpDirName = getenv("TMPDIR");
 120   if (tmpDirName == NULL) tmpDirName = "/tmp";
 121
 122 #ifdef WIN32
 123
 124   string fileName = (string)tmpDirName + "\\libbatch-" + prefix + "-XXXXXX";
 125   char randstr[7];
 126   srand(time(NULL));
 127
 128   do {
 129     sprintf(randstr, "%06d", rand() % 1000000);
 130     fileName.replace(fileName.size()-6, 6, randstr);
 131   } while (_access_s(fileName.c_str(), 0) == 0);
 132
 133   // Open the file as binary to avoid problems with Windows newlines
 134   outputStream.open(fileName.c_str(), ios_base::binary | ios_base::out);
 135
 136 #else
 137   string fileName = (string)tmpDirName + "/libbatch-" + prefix + "-XXXXXX";
 138   char * buf = new char[fileName.size()+1];
 139   fileName.copy(buf, fileName.size());
 140   buf[fileName.size()] = '\0';
 141
 142   int fd = mkstemp(buf);
 143   if (fd == -1) {
 144     delete[] buf;
 145     throw RunTimeException(string("Can't create temporary file ") + fileName);
 146   }
 147   fileName = buf;
 148   delete[] buf;
 149
 150   outputStream.open(fileName.c_str());
 151   close(fd);  // Close the file descriptor so that the file is not opened twice
 152
 153 #endif
 154
 155   if (outputStream.fail())
 156     throw RunTimeException(string("Can't open temporary file ") + fileName);
 157
 158   return fileName;
 159 }
 160
 161 int Utils::chmod(const char *path, int mode)
 162 {
 163 #ifdef WIN32
 164   return _chmod(path, mode);
 165 #else
 166   return ::chmod(path, mode);
 167 #endif
 168 }
 169
 170 void Utils::sleep(unsigned int seconds)
 171 {
 172 #ifdef WIN32
 173   Sleep((seconds)*1000);
 174 #else
 175   ::sleep(seconds);
 176 #endif
 177 }
 178
 179 string Utils::fixPath(const string & path)
 180 {
 181   string fixedPath = path;
 182 #ifdef WIN32
 183   for (unsigned int i=0 ; i<fixedPath.size() ; i++) {
 184     if (fixedPath[i] == '/') fixedPath[i] = '\\';
 185   }
 186 #endif
 187   return fixedPath;
 188 }
 189
 190 }