//-----------------------------------------------------------------------

// Copyright 2011 Ciaran McHale.

//

// Permission is hereby granted, free of charge, to any person obtaining

// a copy of this software and associated documentation files (the

// "Software"), to deal in the Software without restriction, including

// without limitation the rights to use, copy, modify, merge, publish,

// distribute, sublicense, and/or sell copies of the Software, and to

// permit persons to whom the Software is furnished to do so, subject to

// the following conditions.

//

// The above copyright notice and this permission notice shall be

// included in all copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF

// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS

// BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN

// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN

// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

// SOFTWARE.

//----------------------------------------------------------------------

#include "platform.h"

#include <assert.h>

#include <stdlib.h>

#include <config4cpp/StringBuffer.h>

#ifdef P_STDIO_HAS_LIMITED_FDS

# ifdef WIN32

//--------

// Windows does NOT suffer from this problem. However, we can

// pretend it does so we can compile and test both versions of

// the BufferedFileReader class on Windows.

//--------

# include <io.h>

# include <fcntl.h>

# else

# include <sys/types.h>

# include <sys/stat.h>

# include <fcntl.h>

# endif

#endif

namespace CONFIG4CPP_NAMESPACE {

bool

execCmd(const char * cmd, StringBuffer & output)

{

StringBuffer modifiedCmd;

FILE * pipe;

int ch;

int len;

int pcloseStatus;

output.empty();

//--------

// Execute the command with its stderr and stdout merged together.

// TO DO: the "modifiedCmd" below works on Linux and in a cmd

// shell on Windows, but sometimes it does not work in a Cygwin

// shell.

//--------

modifiedCmd << cmd << " 2>&1";

//modifiedCmd << cmd;

pipe = CONFIG4CPP_POPEN(modifiedCmd.c_str(), "r");

if (!pipe) {

output << "cannot execute '" << cmd << "': popen() failed";

return false;

}

//--------

// Read from the pipe and delete the final '\n', if any.

//--------

while ((ch = fgetc(pipe)) != EOF) {

if (ch != '\r') {

output.append((char)ch);

}

}

len = output.length();

if (len > 0 && output[len-1] == '\n') {

output.deleteLastChar();

}

//--------

// We're done. Return success (true) if the exit status of

// the command was 0.

//--------

pcloseStatus = CONFIG4CPP_PCLOSE(pipe);

return (pcloseStatus == 0);

}

#ifdef WIN32

//--------

// Windows version.

//--------

#include <windows.h>

bool

isCmdInDir(const char * cmd, const char * dir)

{

StringBuffer fileName;

static const char * extensions[] = { "", ".exe", ".bat", 0 };

int i;

DWORD fileAttr;

for (i = 0; extensions[i] != 0; i++) {

fileName = "";

fileName << dir << "\\" << cmd << extensions[i];

fileAttr = GetFileAttributes(fileName.c_str());

if (fileAttr != 0xFFFFFFFF) {

return true;

}

}

return false;

}

#else

//--------

// UNIX version.

//--------

#include <sys/types.h>

#include <sys/stat.h>

#include <unistd.h>

bool

isCmdInDir(const char * cmd, const char * dir)

{

StringBuffer fileName;

struct stat sb;

fileName << dir << "/" << cmd;

if (stat(fileName.c_str(), &sb) == 0) {

return true;

}

return false;

}

#endif

#ifdef P_STDIO_HAS_LIMITED_FDS

BufferedFileReader::BufferedFileReader()

{

m_fd = -1;

}

BufferedFileReader::~BufferedFileReader()

{

if (m_fd != -1) {

close();

}

}

bool

BufferedFileReader::open(const char * fileName)

{

assert(m_fd == -1);

m_fd = ::open(fileName, O_RDONLY);

m_bufIndex = 0;

m_bufLen = 0;

return (m_fd != -1);

}

bool

BufferedFileReader::close()

{

int result;

assert(m_fd != -1);

result = ::close(m_fd);

if (result != -1) {

m_fd = -1;

}

return (result != -1);

}

int

BufferedFileReader::getChar()

{

int size;

int result;

assert(m_fd != -1);

if (m_bufIndex == m_bufLen) {

size = ::read(m_fd, m_buf, BUFFERED_FILE_READER_BUF_SIZE);

m_bufIndex = 0;

m_bufLen = size;

if (size <= 0) {

m_bufLen = 0;

return EOF;

}

}

assert(m_bufIndex < m_bufLen);

result = m_buf[m_bufIndex];

m_bufIndex++;

return result;

}

#else

BufferedFileReader::BufferedFileReader()

{

m_file = 0;

}

BufferedFileReader::~BufferedFileReader()

{

if (m_file != 0) {

fclose(m_file);

}

}

bool

BufferedFileReader::open(const char * fileName)

{

assert(m_file == 0);

m_file = fopen(fileName, "r");

return (m_file != 0);

}

bool

BufferedFileReader::close()

{

int status;

status = fclose(m_file);

return (status != -1);

}

int

BufferedFileReader::getChar()

{

assert(m_file != 0);

return fgetc(m_file);

}

#endif

}; // namespace CONFIG4CPP_NAMESPACE