src/Core/Utils.cxx

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