src/Core/Utils.cxx

   1 // Copyright (C) 2007-2015  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 string Utils::createAndOpenTemporaryFile(const string & prefix, ofstream & outputStream)
 101 {
 102   if (outputStream.is_open())
 103     outputStream.close();
 104
 105   // Find directory for temporary files
 106   const char * tmpDirName = getenv("TEMP");
 107   if (tmpDirName == NULL) tmpDirName = getenv("TMP");
 108   if (tmpDirName == NULL) tmpDirName = getenv("TEMPDIR");
 109   if (tmpDirName == NULL) tmpDirName = getenv("TMPDIR");
 110   if (tmpDirName == NULL) tmpDirName = "/tmp";
 111
 112   string fileName = (string)tmpDirName + "/libbatch-" + prefix + "-XXXXXX";
 113
 114 #ifdef WIN32
 115
 116   char randstr[7];
 117   srand(time(NULL));
 118
 119   do {
 120     sprintf(randstr, "%06d", rand() % 1000000);
 121     fileName.replace(fileName.size()-6, 6, randstr);
 122   } while (_access_s(fileName.c_str(), 0) == 0);
 123
 124   // Open the file as binary to avoid problems with Windows newlines
 125   outputStream.open(fileName.c_str(), ios_base::binary | ios_base::out);
 126
 127 #else
 128
 129   char * buf = new char[fileName.size()+1];
 130   fileName.copy(buf, fileName.size());
 131   buf[fileName.size()] = '\0';
 132
 133   int fd = mkstemp(buf);
 134   if (fd == -1) {
 135     delete[] buf;
 136     throw RunTimeException(string("Can't create temporary file ") + fileName);
 137   }
 138   fileName = buf;
 139   delete[] buf;
 140
 141   outputStream.open(fileName.c_str());
 142   close(fd);  // Close the file descriptor so that the file is not opened twice
 143
 144 #endif
 145
 146   if (outputStream.fail())
 147     throw RunTimeException(string("Can't open temporary file ") + fileName);
 148
 149   return fileName;
 150 }
 151
 152 int Utils::chmod(const char *path, int mode)
 153 {
 154 #ifdef WIN32
 155   return _chmod(path, mode);
 156 #else
 157   return ::chmod(path, mode);
 158 #endif
 159 }
 160
 161 void Utils::sleep(unsigned int seconds)
 162 {
 163 #ifdef WIN32
 164   Sleep((seconds)*1000);
 165 #else
 166   ::sleep(seconds);
 167 #endif
 168 }
 169
 170 string Utils::fixPath(const string & path)
 171 {
 172   string fixedPath = path;
 173 #ifdef WIN32
 174   for (unsigned int i=0 ; i<fixedPath.size() ; i++) {
 175     if (fixedPath[i] == '/') fixedPath[i] = '\\';
 176   }
 177 #endif
 178   return fixedPath;
 179 }
 180
 181 }