// \file child.cpp
//
// Copyright (C) 2014 MicroNeil Research Corporation.
//
// This program is part of the MicroNeil Research Open Library Project. For
// more information go to http://www.microneil.com/OpenLibrary/index.html
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2 of the License, or (at your
// option) any later version.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
// more details.
//
// You should have received a copy of the GNU General Public License along with
// this program; if not, write to the Free Software Foundation, Inc., 59 Temple
// Place, Suite 330, Boston, MA 02111-1307 USA
//==============================================================================

#ifndef _WIN32
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <sys/select.h>

#include <cstring>
#include <cerrno>
#endif

#include <stdexcept>
#include <utility>

#include "CodeDweller/timing.hpp"
#include "CodeDweller/child.hpp"

namespace CodeDweller {

  ChildStream::ChildStream(std::vector<std::string> const &args,
                           size_t bufSize) :
    std::iostream(&childStreambuf),
    childStreambuf(bufSize),
    cmdArgs(args) {

    init();
    run();

  }

  ChildStream::ChildStream(std::string const &childpath, size_t bufSize) :
    std::iostream(&childStreambuf),
    childStreambuf(bufSize) {

    cmdArgs.push_back(childpath);
    init();
    run();
  }

  ChildStream::ChildStream(size_t bufSize) :
    std::iostream(&childStreambuf),
    childStreambuf(bufSize) {

    init();
  }

  ChildStream::~ChildStream() {
    try {
      close();
    } catch (std::exception &e) {
      ;
    }
  }

  void ChildStream::init() {

    childStarted = false;
    childExited = false;
    exitCodeObtainedFlag = false;
    exitCode = 0;
  }

  void ChildStream::open(std::vector<std::string> const &args) {
    cmdArgs = args;
    init();
    run();
  }

  void ChildStream::open(std::string const &childpath) {
    cmdArgs.clear();
    cmdArgs.push_back(childpath);
    init();
    run();
  }

  size_t ChildStream::numBytesAvailable() const {

    return childStreambuf.numBytesAvailable();

  }

  bool ChildStream::isRunning() const {
    return childStarted && !childExited;
  }

  void ChildStream::run() {

    if (childStarted) {
      throw std::logic_error("Child process was active when "
			     "run() was called");
    }

    if (cmdArgs.empty()) {
      throw std::invalid_argument("A child executable must be specified.");
    }

#ifdef _WIN32
    // Set the bInheritHandle flag so pipe handles are inherited. 
    SECURITY_ATTRIBUTES securityAttributes; 

    securityAttributes.nLength = sizeof(SECURITY_ATTRIBUTES); 
    securityAttributes.bInheritHandle = true; 
    securityAttributes.lpSecurityDescriptor = NULL; 

    // Create a pipe for the child process's STDOUT.
    HANDLE childStdOutAtChild;
    HANDLE childStdOutAtParent;
    HANDLE childStdInAtChild;
    HANDLE childStdInAtParent;
    int bufferSize = 0;

    if (!CreatePipe(&childStdOutAtParent,
		    &childStdOutAtChild,
		    &securityAttributes,
		    bufferSize)) {
      throw std::runtime_error("Error from CreatePipe for stdout:  " +
			       getErrorText());
    }

    // Ensure the read handle to the pipe for STDOUT is not inherited.
    int inheritFlag = 0;
    if (!SetHandleInformation(childStdOutAtParent,
			      HANDLE_FLAG_INHERIT,
			      inheritFlag) ) {
      throw std::runtime_error("Error from GetHandleInformation for stdout:  " +
			       getErrorText());
    }

    // Create a pipe for the child process's STDIN. 
    if (! CreatePipe(&childStdInAtChild,
		     &childStdInAtParent,
		     &securityAttributes,
		     bufferSize)) {
      throw std::runtime_error("Error from CreatePipe for stdin:  " +
			       getErrorText());
    }

    // Ensure the write handle to the pipe for STDIN is not inherited. 
    if (!SetHandleInformation(childStdInAtParent,
			      HANDLE_FLAG_INHERIT,
			      inheritFlag)) {
      throw std::runtime_error("Error from GetHandleInformation for stdin:  " +
			       getErrorText());
    }
 
    // Set up members of the PROCESS_INFORMATION structure. 
    PROCESS_INFORMATION processInfo; 

    std::fill((char *) &processInfo,
	      ((char *) &processInfo) + sizeof(PROCESS_INFORMATION),
	      0);
 
    // Set up members of the STARTUPINFO structure.  This structure
    // specifies the STDIN and STDOUT handles for redirection.
    STARTUPINFO startInfo;

    std::fill((char *) &startInfo,
	      ((char *) &startInfo) + sizeof(STARTUPINFO),
	      0);
    startInfo.cb = sizeof(STARTUPINFO); 
    startInfo.hStdError = childStdOutAtChild;
    startInfo.hStdOutput = childStdOutAtChild;
    startInfo.hStdInput = childStdInAtChild;
    startInfo.dwFlags |= STARTF_USESTDHANDLES;
 
    // Assemble the command line.
    std::string cmdline;

    if (cmdArgs.size() == 1) {
      cmdline = cmdArgs[0];
    } else {

      // Append all but last command-line arguments.
      for (size_t i = 0; i < cmdArgs.size() - 1; i++) {
	cmdline += cmdArgs[i] + " ";
      }
      cmdline += cmdArgs.back(); // Append last command-line argument.
    }

    // Create the child process. 
    bool status;
 
    status = CreateProcess(NULL, 
			   (char *) cmdline.c_str(),
			   NULL,          // process security attributes 
			   NULL,          // primary thread security attributes 
			   true,          // handles are inherited 
			   0,             // creation flags 
			   NULL,          // use parent's environment 
			   NULL,          // use parent's current directory 
			   &startInfo,    // STARTUPINFO pointer 
			   &processInfo); // receives PROCESS_INFORMATION 

    // If an error occurs, exit the application. 
    if (!status ) {
      throw std::runtime_error("Error from CreateProcess with "
			       "command line \"" + cmdline + "\":  " +
			       getErrorText());
    }

    // Provide the stream buffers with the handles for communicating
    // with the child process.
    childStreambuf.setInputHandle(childStdOutAtParent);
    childStreambuf.setOutputHandle(childStdInAtParent);

    // Save the handles to the child process and its primary thread.
    childProcess = processInfo.hProcess;
    childThread = processInfo.hThread;

    // Close the child's end of the pipes.
    if (!CloseHandle(childStdOutAtChild)) {
      throw std::runtime_error("Error closing the child process "
			       "stdout handle:  " + getErrorText());
    }
     
    if (!CloseHandle(childStdInAtChild)) {
      throw std::runtime_error("Error closing the child process "
			       "stdin handle:  " + getErrorText());
    }

#else

    // Create the pipes for the stdin and stdout.
    int childStdInPipe[2];
    int childStdOutPipe[2];

    if (pipe(childStdInPipe) != 0) {
      throw std::runtime_error("Error creating pipe for stdin:  " +
			       getErrorText());
    }
    if (pipe(childStdOutPipe) != 0) {
      ::close(childStdInPipe[0]);
      ::close(childStdInPipe[1]);
      throw std::runtime_error("Error creating pipe for stdout:  " +
			       getErrorText());
    }

    // Create the child process. 
    childPid = fork();
    if (-1 == childPid) {
      for (int i = 0; i < 2; i++) {
	::close(childStdInPipe[i]);
	::close(childStdOutPipe[i]);
      }
      throw std::runtime_error("Error creating child process:  " +
			       getErrorText());
    }
 
    if (0 == childPid) {

      // The child executes this.  Redirect stdin.
      if (dup2(childStdInPipe[0], STDIN_FILENO) == -1) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error redirecting stdin in the child:  " + getErrorText();
	::write(childStdOutPipe[1], errMsg.data(), errMsg.size());
	exit(-1);
      }

      // Redirect stdout.
      if (dup2(childStdOutPipe[1], STDOUT_FILENO) == -1) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error redirecting stdout in the child:  " + getErrorText();
	::write(childStdOutPipe[1], errMsg.data(), errMsg.size());
	exit(-1);
      }

      // Close pipes.
      if ( (::close(childStdInPipe[0]) != 0) ||
	   (::close(childStdInPipe[1]) != 0) ||
	   (::close(childStdOutPipe[0]) != 0) ||
	   (::close(childStdOutPipe[1]) != 0) ) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error closing the pipes in the child:  " + getErrorText();
	::write(STDOUT_FILENO, errMsg.data(), errMsg.size());
	exit(-1);

      }

      // Prepare the arguments.
      std::vector<const char *> execvArgv;

      for (auto &arg : cmdArgs) {
	execvArgv.push_back(arg.c_str());
      }
      execvArgv.push_back((char *) NULL);

      // Run the child process image.
      (void) execv(execvArgv[0], (char * const *) &(execvArgv[0]));

      // Error from exec.
      std::string errMsg;

      // Send message to parent.
      errMsg = "Error from exec:  " + getErrorText();
      ::write(STDOUT_FILENO, errMsg.data(), errMsg.size());
      exit(-1);

    }

    // Provide the stream buffers with the file descriptors for
    // communicating with the child process.
    childStreambuf.setInputFileDescriptor(childStdOutPipe[0]);
    childStreambuf.setOutputFileDescriptor(childStdInPipe[1]);

    // Close the child's end of the pipes.
    if ( (::close(childStdInPipe[0]) != 0) ||
	 (::close(childStdOutPipe[1]) != 0) ) {
      std::string errMsg;

      throw std::runtime_error("Error closing child's end of pipes in "
			       "the parent:  " + getErrorText());
    }

#endif

    childStarted = true;

  }

  void ChildStream::close() {

    if (!childStarted) {
      throw std::logic_error("Child process was not started "
			     "when close() was called");
    }

    // Terminate the child if it's running.
    if (isRunning()) {

      std::string errorString;
      bool errorOccurred = false;

#ifdef _WIN32
      // Ignore errors.  Reason: Terminating a proces that doesn't
      // exist (e.g. has exited) gives an error.
      (void) TerminateProcess(childProcess, terminateExitCode);
#else
      if (kill(childPid, SIGKILL) != 0) {
        errorString += (errorOccurred ? "\n" : "");
        errorString += "Error terminating the child process:  " +
          getErrorText();
        errorOccurred = true;
      } else if (waitpid(childPid, NULL, 0) != childPid) {
        errorString += (errorOccurred ? "\n" : "");
        errorString += "Error waiting for the child process:  " +
          getErrorText();
        errorOccurred = true;
      }
#endif

      if (errorOccurred) {
        throw std::runtime_error(errorString);
      }

    }

    // Reset.
    init();

  }

  bool ChildStream::isDone() {

    if (childExited) {

      return true;

    }

    if (!childStarted) {
      throw std::logic_error("Child process was not started "
			     "when isDone() was called");
    }

    int result;

#ifdef _WIN32
    if (!GetExitCodeProcess(childProcess, (LPDWORD) &result)) {
      throw std::runtime_error("Error checking status of child process:  " +
			       getErrorText());
    }

    if (STILL_ACTIVE == result) {

      return false;

    }

    // Child process has exited.  Save the exit code.
    exitCode = result;
    exitCodeObtainedFlag = true;

#else
    int status = 0;

    result = waitpid(childPid, &status, WNOHANG);
    if (-1 == result) {
      throw std::runtime_error("Error checking status of child process:  " +
			       getErrorText());
    } else if (0 == result) {

      // Child is still running.
      return false;

    }

    if (WIFEXITED(status)) {

      // Child exited normally.
      exitCode = WEXITSTATUS(status);
      exitCodeObtainedFlag = true;

    }

#endif

    childExited = true;

    return true;

  }

  int32_t ChildStream::result() {

    if (exitCodeObtainedFlag) {

      return exitCode;

    }

    // Check whether the process is running, and get the exit code.
    if (!isDone()) {
      throw std::logic_error("Child process was still running "
			     "when result() was called");
    }

    // Child process has exited.
    if (!exitCodeObtainedFlag) {

      // Exit code is not available.
      throw std::runtime_error("Child process has exited but the exit "
			       "code is not available");

    }

    return exitCode;

  }

  std::string ChildStream::getErrorText() {

#ifdef _WIN32
    LPVOID winMsgBuf;
    DWORD lastError = GetLastError(); 

    FormatMessage(
		  FORMAT_MESSAGE_ALLOCATE_BUFFER | 
		  FORMAT_MESSAGE_FROM_SYSTEM |
		  FORMAT_MESSAGE_IGNORE_INSERTS,
		  NULL,
		  lastError,
		  MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		  (char *) &winMsgBuf,
		  0, NULL );

    std::string errMsg((char *) winMsgBuf);

    LocalFree(winMsgBuf);
    return errMsg;
#else
    return strerror(errno);
#endif
  }

  ChildStream::ChildStreambuf::ChildStreambuf(std::size_t bufSize) :
#ifdef _WIN32
    inputHandle(0),
    outputHandle(0),
#else
    inputFileDescriptor(-1),
    outputFileDescriptor(-1),
#endif
    bufferSize(bufSize),
    readBuffer(bufferSize + 1),
    writeBuffer(bufferSize + 1) {

    // Read buffer initialization.
    char *end = &(readBuffer.front()) + readBuffer.size();

    // Indicate to underflow that underflow has not been called.
    setg(end, end, end);

    // Write buffer initialization.
    char *base = &(writeBuffer.front());

    // Indicate to overflow that overflow has not been called.
    setp(base, base + writeBuffer.size() - 1);

  }

#ifdef _WIN32
  void ChildStream::ChildStreambuf::setInputHandle(HANDLE inHandle) {

    inputHandle = inHandle;

  }
#else
  void ChildStream::ChildStreambuf::setInputFileDescriptor(int inFd) {

    inputFileDescriptor = inFd;

  }
#endif

  size_t ChildStream::ChildStreambuf::numBytesAvailable() const {

    size_t nBytesAvailable =  egptr() - gptr();

#ifdef _WIN32
    DWORD lpTotalBytesAvail;

    if (!PeekNamedPipe(inputHandle,
		       NULL,
		       0,
		       NULL,
		       &lpTotalBytesAvail,
		       NULL)) {
      throw std::runtime_error("Error from PeekNamedPipe:  " +
			       getErrorText());
    }

    if (lpTotalBytesAvail > 0) {
      nBytesAvailable++;
    }
#else
    fd_set readFd;
    int retVal;
    struct timeval timeout = {0, 0};

    FD_ZERO(&readFd);
    FD_SET(inputFileDescriptor, &readFd);

    // Check if input is available.
    retVal = select(inputFileDescriptor + 1, &readFd, NULL, NULL, &timeout);

    if (-1 == retVal) {
      throw std::runtime_error("Error from select(): " + getErrorText());
    } else if (retVal > 0) {
	nBytesAvailable++;
    }
#endif

    return nBytesAvailable;

  }

  std::streambuf::int_type ChildStream::ChildStreambuf::underflow() {

    // Check for empty buffer.
    if (gptr() < egptr()) {

      // Not empty.
      return traits_type::to_int_type(*gptr());

    }

    // Need to fill the buffer.
    char *base = &(readBuffer.front());
    char *start = base;

    // Check whether this is the first fill.
    if (eback() == base) {

      // Not the first fill.  Copy one putback character.
      *(eback()) = *(egptr() - 1);
      start++;
    }

    // start points to the start of the buffer.  Fill buffer.
#ifdef _WIN32
    DWORD nBytesRead;

    if (!ReadFile(inputHandle,
		  start,
		  readBuffer.size() - (start - base),
		  &nBytesRead,
		  NULL)) {
      return traits_type::eof();
    }
#else
    ssize_t nBytesRead;

    nBytesRead = ::read(inputFileDescriptor,
                        start,
                        readBuffer.size() - (start - base));
    if (-1 == nBytesRead) {
      return traits_type::eof();
    }
#endif

    // Check for EOF.
    if (0 == nBytesRead) {
      return traits_type::eof();
    }

    // Update buffer pointers.
    setg(base, start, start + nBytesRead);

    return traits_type::to_int_type(*gptr());

  }

#ifdef _WIN32
  void ChildStream::ChildStreambuf::setOutputHandle(HANDLE outHandle) {

    outputHandle = outHandle;

  }
#else
  void ChildStream::ChildStreambuf::setOutputFileDescriptor(int outFd) {

    outputFileDescriptor = outFd;

  }
#endif

  void ChildStream::ChildStreambuf::flushBuffer() {

    // Write.
    std::ptrdiff_t nBytes = pptr() - pbase();

#ifdef _WIN32
    DWORD nBytesWritten;

    if (!WriteFile(outputHandle,
		   pbase(),
		   nBytes,
		   &nBytesWritten,
		   NULL)) {
      // Clear the output buffer.
      pbump(-nBytes);
      throw std::runtime_error("Error writing to child process:  " +
			       getErrorText());
    }
#else
    ssize_t nBytesWritten;

    nBytesWritten = ::write(outputFileDescriptor, pbase(), nBytes);

#endif

    // Clear the output buffer.
    pbump(-nBytes);

    if (nBytes != nBytesWritten) {
      throw std::runtime_error("Not all data was written to to child "
			       "process:  " + getErrorText());
    }

    return;

  }

  std::streambuf::int_type ChildStream::ChildStreambuf::overflow(int_type ch) {

    // Check whether we're writing EOF.
    if (traits_type::eof() != ch) {

      // Not writing EOF.
      *(pptr()) = ch;
      pbump(1);

      // Write.
      flushBuffer();

      // Success.
      return ch;

    }

    return traits_type::eof();

  }

  int ChildStream::ChildStreambuf::sync() {

    flushBuffer(); // Throws exception on failure.

    // Success.
    return 1;

  }

  Child::CircularBuffer::CircularBuffer(size_t maxSize) :
    buffer(maxSize + 1),
    capacity(maxSize) {

      iBegin = 0;
      iEnd = 0;
  }

  bool Child::CircularBuffer::empty() const {
    return (iBegin == iEnd);
  }

  size_t Child::CircularBuffer::nUsed() const {
    return capacity - nFree();
  }

  void Child::CircularBuffer::clear() {
    iBegin = 0;
    iEnd = 0;
  }

  size_t Child::CircularBuffer::nFree() const {

    size_t iLocalBegin = iBegin;

    // Correct an invalid value.
    if (iLocalBegin >= capacity + 1) {
      iLocalBegin = 0;
    }

    if (iLocalBegin <= iEnd) {
      return capacity - (iEnd - iLocalBegin);
    }
    return iLocalBegin - iEnd - 1;

  }

  void Child::CircularBuffer::put(char const *ptr, size_t nBytes) {

    for (size_t i = 0; i < nBytes; i++) {
      buffer[iEnd] = *ptr;
      ptr++;
      nextIndex(iEnd);
    }

  }

  Child::Child(std::vector<std::string> const &args,
               size_t bufSize,
               std::uint16_t nominalAboveMinPollTime_ms,
               std::uint16_t deltaPollTime_ms) :
    bufferCapacity(bufSize),
    readBuffer(bufferCapacity),
    writeBuffer(bufferCapacity),
    nominalPollTime_ms(nominalAboveMinPollTime_ms +
                       CodeDweller::MinimumSleeperTime),
    maximumPollTime_ms(nominalPollTime_ms + deltaPollTime_ms) {

    init();
    open(args);
  }

  Child::Child(std::string const &childpath,
               size_t bufSize,
               std::uint16_t nominalAboveMinPollTime_ms,
               std::uint16_t deltaPollTime_ms) :
    bufferCapacity(bufSize),
    readBuffer(bufferCapacity),
    writeBuffer(bufferCapacity),
    nominalPollTime_ms(nominalAboveMinPollTime_ms +
                       CodeDweller::MinimumSleeperTime),
    maximumPollTime_ms(nominalPollTime_ms + deltaPollTime_ms) {

    init();
    open(childpath);
  }

  Child::Child(size_t bufSize,
               std::uint16_t nominalAboveMinPollTime_ms,
               std::uint16_t deltaPollTime_ms) :
    bufferCapacity(bufSize),
    readBuffer(bufferCapacity),
    writeBuffer(bufferCapacity),
    nominalPollTime_ms(nominalAboveMinPollTime_ms +
                       CodeDweller::MinimumSleeperTime),
    maximumPollTime_ms(nominalPollTime_ms + deltaPollTime_ms) {

      init();
  }

  Child::~Child() {
    try {
      close();
    } catch (std::exception &e) {
      ;
    }
  }

  void Child::open(std::vector<std::string> const &args) {
    cmdArgs = args;

    if (isRunning()) {
      throw std::logic_error("The child process was already active.");
    }

    init();
    run();
  }

  void Child::open(std::string const &childpath) {

    if (isRunning()) {
      throw std::logic_error("The child process was already active.");
    }

    cmdArgs.clear();
    cmdArgs.push_back(childpath);
    init();
    run();
  }

  void Child::init() {

    childStarted = false;
    childExited = false;
    exitCodeObtainedFlag = false;
    exitCode = 0;

    readBuffer.clear();
    nWriteBytes = 0;
    nTransmitBytes = 0;

    readerThreadIsRunning = false;
    writerThreadIsRunning = false;
    stopReaderFlag = true;
    stopWriterFlag = true;
    errorText.clear();

  }

  bool Child::isFinishedWriting() const {
    return ( (0 == nWriteBytes) &&
             (0 == nTransmitBytes) );
  }

  bool Child::closeInput(std::string &errorString) {

#ifdef _WIN32
    // Close if not already closed.
    if ( (NULL != inputHandle) &&
         (CloseHandle(inputHandle) == 0) ) {
      errorString += (errorString.empty() ? "" : "\n");
      errorString += "Error closing input to the child:  " + getErrorText();
      return false;
    }

    // Closed successfully.
    inputHandle = NULL;
    return true;
#else
    // Close if not already closed.
    if ( (-1 != inputFileDescriptor) &&
         (0 != ::close(inputFileDescriptor)) ) {
      errorString += (errorString.empty() ? "" : "\n");
      errorString += "Error closing input to the child:  " + getErrorText();
      return false;
    }

    // Closed successfully.
    inputFileDescriptor = -1;
    return true;
#endif

  }

  bool Child::closeOutput(std::string &errorString) {

#ifdef _WIN32
    // Close if not already closed.
    if ( (NULL != outputHandle) &&
         (CloseHandle(outputHandle) == 0) ) {
      errorString += (errorString.empty() ? "" : "\n");
      errorString += "Error closing output to the child:  " + getErrorText();
      return false;
    }

    // Closed successfully.
    outputHandle = NULL;
    return true;
#else
    // Close if not already closed.
    if ( (-1 != outputFileDescriptor) &&
         (0 != ::close(outputFileDescriptor)) ) {
      errorString += (errorString.empty() ? "" : "\n");
      errorString += "Error closing output to the child:  " + getErrorText();
      return false;
    }

    // Closed successfully.
    outputFileDescriptor = -1;
    return true;
#endif

  }

  void Child::closeStdIn() {

    if (!childStarted) {
      throw std::logic_error("Child process was not started "
			     "when close() was called");
    }

    // Stop writer thread.  Note: None of the error conditions that
    // cause an exception to be thrown by join() can ever occur.
    stopWriterFlag = true;

    std::string errorString;

    if (!closeOutput(errorString)) {
      throw std::runtime_error(errorString);
    }

    if (writerThreadIsRunning) {
      writerThread.join();
      writerThreadIsRunning = false;
    }

  }

  void Child::close() {

    if (!childStarted) {
      throw std::logic_error("Child process was not started "
			     "when close() was called");
    }

    // Stop the reader and writer threads.  Note: None of the error
    // conditions that cause an exception to be thrown by join()
    // can ever occur.
    stopReaderFlag = true;
    stopWriterFlag = true;

    // Terminate the child if it's running.
    if (isRunning()) {

      std::string errorString;
      bool noErrorOccurred = true;

#ifdef _WIN32
      // Ignore errors.  Reason: Terminating a proces that doesn't
      // exist (e.g. has exited) gives an error.
      (void) TerminateProcess(childProcess, terminateExitCode);

      noErrorOccurred = noErrorOccurred && closeInput(errorString);
      noErrorOccurred = noErrorOccurred && closeOutput(errorString);
#else
      noErrorOccurred = noErrorOccurred && closeInput(errorString);
      noErrorOccurred = noErrorOccurred && closeOutput(errorString);

      if (kill(childPid, SIGKILL) != 0) {

        errorString += (errorString.empty() ? "" : "\n");
        errorString += "Error terminating the child process:  " +
          getErrorText();
        noErrorOccurred = false;

      } else if (waitpid(childPid, NULL, 0) != childPid) {

        errorString += (errorString.empty() ? "" : "\n");
        errorString += "Error waiting for the child process:  " +
          getErrorText();
        noErrorOccurred = false;

      }
#endif

      if (!noErrorOccurred) {
        throw std::runtime_error(errorString);
      }

    }

    if (readerThreadIsRunning) {
      readerThread.join();
      readerThreadIsRunning = false;
    }

    if (writerThreadIsRunning) {
      writerThread.join();
      writerThreadIsRunning = false;
    }

    // Reset.
    init();

  }

  bool Child::isRunning() const {
    return childStarted && !childExited;
  }

  bool Child::errorOccurred(std::string &errorDescription) const {
    errorDescription = errorText;
    return !errorDescription.empty();
  }

  void Child::run() {

    if (childStarted) {
      throw std::logic_error("Child process was active when "
			     "run() was called");
    }

    if (cmdArgs.empty()) {
      throw std::invalid_argument("A child executable must be specified.");
    }

#ifdef _WIN32
    // Set the bInheritHandle flag so pipe handles are inherited. 
    SECURITY_ATTRIBUTES securityAttributes; 

    securityAttributes.nLength = sizeof(SECURITY_ATTRIBUTES); 
    securityAttributes.bInheritHandle = true; 
    securityAttributes.lpSecurityDescriptor = NULL; 

    // Create a pipe for the child process's STDOUT.
    HANDLE childStdOutAtChild;
    HANDLE childStdOutAtParent;
    HANDLE childStdInAtChild;
    HANDLE childStdInAtParent;
    int bufferSize = 0;

    if (!CreatePipe(&childStdOutAtParent,
		    &childStdOutAtChild,
		    &securityAttributes,
		    bufferSize)) {
      throw std::runtime_error("Error from CreatePipe for stdout:  " +
			       getErrorText());
    }

    // Ensure the read handle to the pipe for STDOUT is not inherited.
    int inheritFlag = 0;
    if (!SetHandleInformation(childStdOutAtParent,
			      HANDLE_FLAG_INHERIT,
			      inheritFlag) ) {
      throw std::runtime_error("Error from GetHandleInformation for stdout:  " +
			       getErrorText());
    }

    // Create a pipe for the child process's STDIN. 
    if (! CreatePipe(&childStdInAtChild,
		     &childStdInAtParent,
		     &securityAttributes,
		     bufferSize)) {
      throw std::runtime_error("Error from CreatePipe for stdin:  " +
			       getErrorText());
    }

    // Ensure the write handle to the pipe for STDIN is not inherited. 
    if (!SetHandleInformation(childStdInAtParent,
			      HANDLE_FLAG_INHERIT,
			      inheritFlag)) {
      throw std::runtime_error("Error from GetHandleInformation for stdin:  " +
			       getErrorText());
    }
 
    // Set up members of the PROCESS_INFORMATION structure. 
    PROCESS_INFORMATION processInfo; 

    std::fill((char *) &processInfo,
	      ((char *) &processInfo) + sizeof(PROCESS_INFORMATION),
	      0);
 
    // Set up members of the STARTUPINFO structure.  This structure
    // specifies the STDIN and STDOUT handles for redirection.
    STARTUPINFO startInfo;

    std::fill((char *) &startInfo,
	      ((char *) &startInfo) + sizeof(STARTUPINFO),
	      0);
    startInfo.cb = sizeof(STARTUPINFO); 
    startInfo.hStdError = childStdOutAtChild;
    startInfo.hStdOutput = childStdOutAtChild;
    startInfo.hStdInput = childStdInAtChild;
    startInfo.dwFlags |= STARTF_USESTDHANDLES;
 
    // Assemble the command line.
    std::string cmdline;

    if (cmdArgs.size() == 1) {
      cmdline = cmdArgs[0];
    } else {

      // Append all but last command-line arguments.
      for (size_t i = 0; i < cmdArgs.size() - 1; i++) {
	cmdline += cmdArgs[i] + " ";
      }
      cmdline += cmdArgs.back(); // Append last command-line argument.
    }

    // Create the child process. 
    bool status;
 
    status = CreateProcess(NULL, 
			   (char *) cmdline.c_str(),
			   NULL,          // process security attributes 
			   NULL,          // primary thread security attributes 
			   true,          // handles are inherited 
			   0,             // creation flags 
			   NULL,          // use parent's environment 
			   NULL,          // use parent's current directory 
			   &startInfo,    // STARTUPINFO pointer 
			   &processInfo); // receives PROCESS_INFORMATION 

    // If an error occurs, exit the application. 
    if (!status ) {
      throw std::runtime_error("Error from CreateProcess with "
			       "command line \"" + cmdline + "\":  " +
			       getErrorText());
    }

    // Provide the stream buffers with the handles for communicating
    // with the child process.
    inputHandle = childStdOutAtParent;
    outputHandle = childStdInAtParent;

    // Save the handles to the child process and its primary thread.
    childProcess = processInfo.hProcess;
    childThread = processInfo.hThread;

    // Close the child's end of the pipes.
    if (!CloseHandle(childStdOutAtChild)) {
      throw std::runtime_error("Error closing the child process "
			       "stdout handle:  " + getErrorText());
    }
     
    if (!CloseHandle(childStdInAtChild)) {
      throw std::runtime_error("Error closing the child process "
			       "stdin handle:  " + getErrorText());
    }

#else

    // Create the pipes for the stdin and stdout.
    int childStdInPipe[2];
    int childStdOutPipe[2];

    if (pipe(childStdInPipe) != 0) {
      throw std::runtime_error("Error creating pipe for stdin:  " +
			       getErrorText());
    }
    if (pipe(childStdOutPipe) != 0) {
      ::close(childStdInPipe[0]);
      ::close(childStdInPipe[1]);
      throw std::runtime_error("Error creating pipe for stdout:  " +
			       getErrorText());
    }

    // Create the child process. 
    childPid = fork();
    if (-1 == childPid) {
      for (int i = 0; i < 2; i++) {
	::close(childStdInPipe[i]);
	::close(childStdOutPipe[i]);
      }
      throw std::runtime_error("Error creating child process:  " +
			       getErrorText());
    }
 
    if (0 == childPid) {

      // The child executes this.  Redirect stdin.
      if (dup2(childStdInPipe[0], STDIN_FILENO) == -1) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error redirecting stdin in the child:  " + getErrorText();
	::write(childStdOutPipe[1], errMsg.data(), errMsg.size());
	exit(-1);
      }

      // Redirect stdout.
      if (dup2(childStdOutPipe[1], STDOUT_FILENO) == -1) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error redirecting stdout in the child:  " + getErrorText();
	::write(childStdOutPipe[1], errMsg.data(), errMsg.size());
	exit(-1);
      }

      // Close pipes.
      if ( (::close(childStdInPipe[0]) != 0) ||
	   (::close(childStdInPipe[1]) != 0) ||
	   (::close(childStdOutPipe[0]) != 0) ||
	   (::close(childStdOutPipe[1]) != 0) ) {
	std::string errMsg;

	// Send message to parent.
	errMsg = "Error closing the pipes in the child:  " + getErrorText();
	::write(STDOUT_FILENO, errMsg.data(), errMsg.size());
	exit(-1);

      }

      // Prepare the arguments.
      std::vector<const char *> execvArgv;

      for (auto &arg : cmdArgs) {
	execvArgv.push_back(arg.c_str());
      }
      execvArgv.push_back((char *) NULL);

      // Run the child process image.
      (void) execv(execvArgv[0], (char * const *) &(execvArgv[0]));

      // Error from exec.
      std::string errMsg;

      // Send message to parent.
      errMsg = "Error from exec:  " + getErrorText();
      ::write(STDOUT_FILENO, errMsg.data(), errMsg.size());
      exit(-1);

    }

    // Provide the stream buffers with the file descriptors for
    // communicating with the child process.
    inputFileDescriptor = childStdOutPipe[0];
    outputFileDescriptor = childStdInPipe[1];

    // Close the child's end of the pipes.
    if ( (::close(childStdInPipe[0]) != 0) ||
	 (::close(childStdOutPipe[1]) != 0) ) {
      std::string errMsg;

      throw std::runtime_error("Error closing child's end of pipes in "
			       "the parent:  " + getErrorText());
    }

#endif

    childStarted = true;

    // Start the reader and writer threads.
    stopReaderFlag = false;
    stopWriterFlag = false;

    try {

      std::thread readerTemp(&Child::readFromChild, this);
      readerThread = std::move(readerTemp);

    } catch (std::exception &e) {

      throw std::runtime_error("Error starting reader thread:  " +
                               getErrorText());

    }

    try {

      std::thread writerTemp(&Child::writeToChild, this);
      writerThread = std::move(writerTemp);

    } catch (std::exception &e) {

      stopReaderFlag = true;
      stopWriterFlag = true;
      readerThread.join();

      throw std::runtime_error("Error starting writer thread:  " +
                               getErrorText());

    }

    readerThreadIsRunning = true;
    writerThreadIsRunning = true;

  }

  bool Child::isDone() {

    if (childExited) {

      return true;

    }

    if (!childStarted) {
      throw std::logic_error("Child process was not started "
			     "when isDone() was called");
    }

    int result;

#ifdef _WIN32
    if (!GetExitCodeProcess(childProcess, (LPDWORD) &result)) {
      throw std::runtime_error("Error checking status of child process:  " +
			       getErrorText());
    }

    if (STILL_ACTIVE == result) {

      return false;

    }

    // Child process has exited.  Save the exit code.
    exitCode = result;
    exitCodeObtainedFlag = true;

#else
    int status = 0;

    result = waitpid(childPid, &status, WNOHANG);
    if (-1 == result) {
      throw std::runtime_error("Error checking status of child process:  " +
			       getErrorText());
    } else if (0 == result) {

      // Child is still running.
      return false;

    }

    if (WIFEXITED(status)) {

      // Child exited normally.
      exitCode = WEXITSTATUS(status);
      exitCodeObtainedFlag = true;

    }

#endif

    childExited = true;

    // Stop threads.
    stopReaderFlag = true;
    stopWriterFlag = true;
    if (readerThreadIsRunning) {
      readerThread.join();
      readerThreadIsRunning = false;
    }
    if (writerThreadIsRunning) {
      writerThread.join();
      writerThreadIsRunning = false;
    }

    return true;

  }

  int32_t Child::result() {

    if (exitCodeObtainedFlag) {

      return exitCode;

    }

    // Check whether the process is running, and get the exit code.
    if (!isDone()) {
      throw std::logic_error("Child process was still running "
			     "when result() was called");
    }

    // Child process has exited.
    if (!exitCodeObtainedFlag) {

      // Exit code is not available.
      throw std::runtime_error("Child process has exited but the exit "
			       "code is not available");

    }

    return exitCode;

  }

  std::string Child::getErrorText() {

#ifdef _WIN32
    LPVOID winMsgBuf;
    DWORD lastError = GetLastError(); 

    FormatMessage(
		  FORMAT_MESSAGE_ALLOCATE_BUFFER | 
		  FORMAT_MESSAGE_FROM_SYSTEM |
		  FORMAT_MESSAGE_IGNORE_INSERTS,
		  NULL,
		  lastError,
		  MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		  (char *) &winMsgBuf,
		  0, NULL );

    std::string errMsg((char *) winMsgBuf);

    LocalFree(winMsgBuf);
    return errMsg;
#else
    return strerror(errno);
#endif
  }

  void Child::readFromChild() {

    std::vector<char> rxBuf(bufferCapacity);

    CodeDweller::PollTimer pollTimer(nominalPollTime_ms, maximumPollTime_ms);

    while (!stopReaderFlag) {

      char *bufferPtr;

      bufferPtr = &(rxBuf[0]);

      // Blocking read from the child.
#ifdef _WIN32
      DWORD nBytesRead;

      if (!ReadFile(inputHandle,
                    bufferPtr,
                    bufferCapacity,
                    &nBytesRead,
                    NULL)) {

        if (stopReaderFlag) {
          break;
        }

	// Broken pipe occurs when the child exits; this is not
	// necessarily an error condition.
	if (GetLastError() != ERROR_BROKEN_PIPE) {

	  errorText = "Error reading from the child process:  ";
	  errorText += getErrorText();

	}

	break;

      }
#else
      ssize_t nBytesRead;

      nBytesRead = ::read(inputFileDescriptor,
                          bufferPtr,
                          bufferCapacity);
      if (-1 == nBytesRead) {

        if (stopReaderFlag) {
          break;
        }

        errorText = "Error reading from the child process:  ";
        errorText += getErrorText();
        break;

      } else if (0 == nBytesRead) {

        // EOF; child exited.
        break;

      }
#endif

      // Copy to the shared read buffer.
      while ((nBytesRead > 0) && !stopReaderFlag) {

        int nBytesToPut, nBytesFree;

        nBytesToPut = nBytesRead;

        // Can be called in the reader thread without locking the
        // mutex.
        nBytesFree = readBuffer.nFree();

        if (nBytesToPut > nBytesFree) {
          nBytesToPut = nBytesFree;
        }

        if (nBytesToPut > 0) {
          std::lock_guard<std::mutex> lock(readBufferMutex);

          readBuffer.put(bufferPtr, nBytesToPut);
          bufferPtr += nBytesToPut;
          nBytesRead -= nBytesToPut;

          pollTimer.reset();

        } else {
          pollTimer.pause();
        }

      }

    }

  }

  void Child::writeToChild() {

    std::vector<char> localWriteBuffer(bufferCapacity);
    size_t nLocalWriteBytes;

    CodeDweller::PollTimer pollTimer(nominalPollTime_ms, maximumPollTime_ms);

    while (!stopWriterFlag) {

      char *bufferPtr;

      // Poll for data in the shared write buffer.
      while ((0 == nWriteBytes) && !stopWriterFlag) {
        pollTimer.pause();
      }

      if (stopWriterFlag) {
        goto exit;
      }

      // Copy from the shared write buffer.
      {
        std::lock_guard<std::mutex> lock(writeBufferMutex);

        localWriteBuffer.swap(writeBuffer);
        nLocalWriteBytes = nWriteBytes;
        nWriteBytes = 0;
      }

      if (stopWriterFlag) {
        goto exit;
      }

      pollTimer.reset();

      // Blocking write to the child.
      bufferPtr = &(localWriteBuffer[0]);

      while (nLocalWriteBytes > 0) {

#ifdef _WIN32
        DWORD nBytesWritten;

        if (!WriteFile(outputHandle,
                       bufferPtr,
                       nLocalWriteBytes,
                       &nBytesWritten,
                       NULL)) {

          if (stopWriterFlag) {
            goto exit;
          }

          errorText = "Error writing to the child process:  ";
          errorText += getErrorText();
          goto exit;

        }

        if (stopWriterFlag) {
          goto exit;
        }
#else
        ssize_t nBytesWritten;

        nBytesWritten = ::write(outputFileDescriptor,
                                bufferPtr,
                                nLocalWriteBytes);

        if (stopWriterFlag) {
          goto exit;
        }

        if (-1 == nBytesWritten) {

          if (ENOSPC != errno) {

            // Some error other than no space.
            errorText = "Error writing to the child process:  ";
            errorText += getErrorText();
            goto exit;

          }

        }
#endif

        nLocalWriteBytes -= nBytesWritten;
        bufferPtr += nBytesWritten;

      }

    }

  exit: return;

  }

}