/** @file linux-serial.c
 *  @brief  Simulate serial I/O for linux EZSP UART
 *          applications.
 *
 * <!-- Copyright 2007-2011 by Ember Corporation. All rights reserved.  *80*-->
 */

#define _GNU_SOURCE 1  // This turns on certain non-standard GNU extensions
// to the C library (i.e. strnlen()).
// Include before PLATFORM_HEADER since that also
// includes 'string.h'

#include PLATFORM_HEADER //compiler/micro specifics, types

#include "stack/include/ember-types.h"
#include "stack/include/error.h"
#include <stdio.h>      /* printf */
#include <stdarg.h>     /* va_list, va_start, va_arg, va_end */

#include "hal/hal.h"
#include "plugin/serial/serial.h"
#include "plugin/serial/ember-printf.h"
#include "command-interpreter2.h"
#include "cli.h"

#include <sys/types.h>         // for fstat()
#include <sys/stat.h>          // ""
#include <fcntl.h>             // for fcntl()
#include <stdlib.h>
#include <unistd.h>            // for pipe(), fork()

#if defined NO_READLINE
  #define READLINE_SUPPORT 0
#else
  #define READLINE_SUPPORT 1
  #include <readline/readline.h> // for readline()
  #include <readline/history.h>  // ""

#endif

#include <signal.h>            // for trapping SIGTERM
#include <errno.h>             // for strerror() and errno
#include <stdarg.h>            // for vfprintf()
#include <sys/wait.h>          // for wait()

#if defined(AF_USB_HOST)
  #include "app/util/ezsp/ezsp-protocol.h"
  #include "app/ezsp-host/ezsp-host-io.h"
#elif defined(EZSP_ASH)
// All needed For ashSerialGetFd()
  #include "app/util/ezsp/ezsp-protocol.h"
  #include "app/ezsp-host/ash/ash-host.h"
  #include "app/ezsp-host/ezsp-host-io.h"
  #include "app/ezsp-host/ash/ash-host-ui.h"
#endif

#if defined(EMBER_AF_PLUGIN_GATEWAY)
  #include "app/framework/plugin-host/gateway/gateway-support.h"
#else
  #include "app/util/gateway/backchannel.h"
  #include "app/util/gateway/gateway.h"
#endif

#include "linux-serial.h"
#include "plugin/serial/ember-printf-convert.h"

// Don't like readline and the GPL requirements?  Use 'libedit'.
// It is a call-for-call compatible with the readline library but is
// released under the BSD license.
//   http://www.thrysoee.dk/editline/

//------------------------------------------------------------------------------
// Globals

#define NUM_PORTS              2
#define INVALID_FD             -1
#define INVALID_PID            0
#define MAX_PROMPT_LENGTH      20
#define MAX_NUMBER_OF_COMMANDS 500
#define LINE_FEED              0x0A
#define EOF_CHAR               0x04
#define MAX_STRING_LENGTH      250  // arbitrary limit

static int STDIN = 0;

// The first index is the port, the second
// is the pipe number.  The first two are the data pipes,
// the second two are the control pipes.
static int pipeFileDescriptors[NUM_PORTS][4] = {
	{ INVALID_FD, INVALID_FD, INVALID_FD, INVALID_FD },
	{ INVALID_FD, INVALID_FD, INVALID_FD, INVALID_FD }
};
#define DATA_READER(__port) pipeFileDescriptors[(__port)][0]
#define DATA_WRITER(__port) pipeFileDescriptors[(__port)][1]
#define CONTROL_READER(__port) pipeFileDescriptors[(__port)][2]
#define CONTROL_WRITER(__port) pipeFileDescriptors[(__port)][3]

static bool amParent = true;
static bool useControlChannel = true;
static bool debugOn = true;
static bool promptSet = false;
static char prompt[MAX_PROMPT_LENGTH];
static pid_t childPid[NUM_PORTS] = { INVALID_PID, INVALID_PID };

static bool usingCommandInterpreter = false;

#if READLINE_SUPPORT

// for readline() command completion
static const EmberCommandEntry* allCommands = NULL;

#ifdef __APPLE__
#define strnlen(string, n) strlen((string))
#define rl_completion_matches(a, b) completion_matches((a), (b))
#define strsignal(x) NULL   // This is a _GNU_SOURCE extension.
#endif

// This will be prefixed with $HOME
static const char readlineHistoryFilename[] = ".linux-serial.history";
static char readlineHistoryPath[MAX_STRING_LENGTH];

#endif // READLINE_SUPPORT

static int childProcessPort = -1;
static bool debugChildProcess = 0;

//------------------------------------------------------------------------------
// Forward Declarations

static EmberStatus serialInitInternal(uint8_t port);
static void childRun(uint8_t port);
static void setNonBlockingFD(int fd);
static void debugPrint(const char* formatString, ...);
static void processSerialInput(uint8_t port);

#if READLINE_SUPPORT
char** commandCompletion(const char *text, int start, int end);
char* singleCommandCompletion(const char *text, int start);
char* filenameCompletion(const char* text, int state);
static void initializeHistory(void);
static void writeHistory(void);
static char* duplicateString(const char* source);
#else  // use stubs
  #define initializeHistory()
  #define writeHistory()
  #define add_history(x)
#endif

static void parentCleanupAfterChildDied(void);
static void installSignalHandler(void);
static void childCleanupAndExit(void);
static void handleBackchannelConnection(uint8_t port);

static bool sendGoAhead = true;

//------------------------------------------------------------------------------
// Initialization Functions

// In order to handle input more effeciently and cleanly,
// we use readline().  This is a blocking call, so we create a second
// process to handle the serial input while the main one can simply
// loop executing ezspTick() and other functionality.

EmberStatus emberSerialInit(uint8_t port,
                            SerialBaudRate rate,
                            SerialParity parity,
                            uint8_t stopBits)
{
  static bool emberSerialInitCalled = false;

  debugPrint("emberSerialInit()\n");
  if (port > 1) {
    return EMBER_SERIAL_INVALID_PORT;
  }

  if (childPid[port] != INVALID_PID) {
    debugPrint("Serial port %d already initialized.\n", port);
    return EMBER_SUCCESS;
  }

  // Without the backchannel, there is only one serial port available (STDIN).
  if (!backchannelEnable
      && emberSerialInitCalled) {
    return EMBER_SERIAL_INVALID_PORT;
  }

  if (backchannelEnable) {
    // For the CLI, wait here until a new client connects for the first time.
    BackchannelState state =
      backchannelCheckConnection(port,
                                 (port                     // waitForConnection?
                                  == SERIAL_PORT_CLI));
    if (port == SERIAL_PORT_CLI && state != NEW_CONNECTION) {
      debugPrint("Failed to get new backchannel connection.\n");
      return EMBER_ERR_FATAL;
    } else if ( !(state == NEW_CONNECTION || state == CONNECTION_EXISTS) ) {
      // We will defer initializing the RAW serial port (spawning the child
      // and that jazz) until we actually have a new client connection.
      return EMBER_SUCCESS;
    }
  }

  EmberStatus status = serialInitInternal(port);

  if (status == EMBER_SUCCESS) {
    installSignalHandler();
    emberSerialInitCalled = true;
  }

  return status;
}

static EmberStatus serialInitInternal(uint8_t port)
{
	// Create two pipes, control and data.
	// Fork
	// Parent closes the write end of the data pipe and read end of the control,
	//   child closes the read end of the data pipe and write end of the control.
	// Child sits and waits for input via readline() (or read())
	//   When it receives input it writes it to the data pipe.  It then blocks
	//   on a read call to the control pipe (if it is running the CLI).
	// Parent loops normally, calling emberSerialReadByte() or emberSeriaReadLine()
	//   to get the input.  These calls map to reads of the data pipe, the former
	//   would NOT be blocking while the latter would be.
	//   Parent calls indirectly readyForSerialInput() after reading an entire
	//   line of data (via the next emberSerialReadByte()).

  int status;
  int i;
	pid_t pid;

	for (i = 0; i < 2; i++) {
		char* string = (i == 0
						 ? "data"
						 : "control");
		int* pipePtr = (i == 0
						 ? &(pipeFileDescriptors[port][0])
						 : &(pipeFileDescriptors[port][2]));
		status = pipe(pipePtr);
		if (status != 0) {
			fprintf(stderr,
					"FATAL: Could not create %s pipe (%d): %s",
					string,
					status,
					strerror(status));
			assert(false);
		}
		debugPrint("Serial Port %d, Created %s Pipe, Reader: %d, Writer %d\n",
					port,
					string,
					pipePtr[0],
					pipePtr[1]);
	}

	// Change the data reader to non-blocking so that we will
	// continue to be able to execute ezspTick().
	setNonBlockingFD(DATA_READER(port));

	pid = fork();
	if (pid == 0) {  // Child
		amParent = false;
		while (debugChildProcess) {
		// block indefinitely waiting for debugger to attach
		sleep(1);
		}
	  childRun(port);
		return EMBER_ERR_FATAL;  // should never get here
	} else if (pid == -1) {
		fprintf(stderr, "FATAL: Could not fork! (%d): %s\n",
				errno,
				strerror(errno));
		assert(false);
	}
	// Parent
	debugPrint("fork(): My child's pid is %d\n", pid);

	// BugzId:12928 Parent retains allCommands for other children it might spawn

	close(DATA_WRITER(port));
	close(CONTROL_READER(port));
	DATA_WRITER(port) = INVALID_FD;
	CONTROL_READER(port) = INVALID_FD;

	// BugzId:12928 Leave STDIN open for subsequent forked children to use

	childPid[port] = pid;

	setMicroRebootHandler(&emberSerialCleanup);
	return EMBER_SUCCESS;
}

// Checks to see if there is a remote connection in place, or if a new
// one has come in.  When a new one comes in, spawn a child process to
// deal with it.
static bool handleRemoteConnection(uint8_t port)
{
  BackchannelState state =
    backchannelCheckConnection(port,
                               false); // don't wait for new connection
  if (state == CONNECTION_ERROR
      || state == NO_CONNECTION) {
    // BugzId:12928 If child still exists, return true to suck pipe 'til empty
    return (childPid[port] != INVALID_PID);
  } else if (state == CONNECTION_EXISTS) {
    return true;
  } // else
    //   state == NEW_CONNECTION

  return (EMBER_SUCCESS == serialInitInternal(port));
}

static void handleBackchannelConnection(uint8_t port)
{
  // BugzId:12928 Close out sockets used by other children but not this one
  int i;
  for (i = 0; i < NUM_PORTS; i++) {
    if (i != port) {
      backchannelCloseConnection(i);
    }
  }

  if (EMBER_SUCCESS
      ==  backchannelMapStandardInputOutputToRemoteConnection(port)) {
    if (port == SERIAL_PORT_CLI) {
      fprintf(stdout, "Connected.\r\n");
      fflush(stdout);
    } // else
      //   Raw port, don't print anything.
  } else {
    childCleanupAndExit();
  }
}

static void childRun(uint8_t port)
{
  childProcessPort = port;

  close(DATA_READER(port));
  close(CONTROL_WRITER(port));

  if (backchannelEnable) {
    handleBackchannelConnection(port);
  } // else
    //   don't do anything

  // This fixes a bug on Cygwin where a process that
  // holds onto an open file descriptor after a fork()
  // can prevent the other parent process from performing a
  // close() and re-open() on it.  Specifically this only seems
  // to affect serial port FDs, which is what ASH uses.

#ifdef EZSP_ASH
  if (INVALID_FD != ezspSerialGetFd()) {
    int port = ezspSerialGetFd();
    if (0 != close(port)) {
      fprintf(stderr, "FATAL: Could not close EZSP file descriptor %d: %s\n",
              port,
              strerror(errno));
      assert(false);
    }
  }
#endif

  installSignalHandler();
  processSerialInput(port);

  // Normally the above function NEVER returns
  // If we get here it is an error.
  assert(0);
}

void emberSerialSetPrompt(const char* thePrompt)
{
  if (thePrompt == NULL) {
    promptSet = false;
    return;
  }

  // Substract one for the '>'
  snprintf(prompt,
           MAX_PROMPT_LENGTH - 1,
           "%s>",
           thePrompt);
}

static void setNonBlockingFD(int fd)
{
	int flags = fcntl(fd, F_GETFL);
	int status = fcntl(fd, F_SETFD, flags | O_NONBLOCK);
	if (status != 0) {
		fprintf(stderr,
				"FATAL: Could not set pipe reader to non-blocking (%d): %s\n",
				errno,
				strerror(errno));
		assert(false);
	}
}

// It is expected this is only called within the parent process.
void emberSerialCleanup(void)
{
	uint8_t port;
	for (port = 0; port < 2; port++) {
		if (childPid[port] != INVALID_PID) {
			int status;
			close(DATA_READER(port));
			close(CONTROL_WRITER(port));
			DATA_READER(port) = INVALID_FD;
			CONTROL_WRITER(port) = INVALID_FD;
			debugPrint("Waiting for child on port %d to terminate.\n", port);
		  kill(childPid[port], SIGTERM);
		  wait(&status);
			debugPrint("Child on port %d exited with status: %d\n", port, status);
			childPid[port] = INVALID_PID;
		}
	}
  gatewayBackchannelStop();
}

static void parentCleanupAfterChildDied(void)
{
  // First we need to determine which child died.
  // We assume that a child has really died, otherwise the
  // wait() call will block.
  int status;
  pid_t pid = wait(&status);
  int i;
  int childPort = -1;
  for (i = 0; i < NUM_PORTS; i++) {
    if (childPid[i] == pid) {
      childPort = i;
      break;
    }
  }
  if (childPort == -1) {
    // A child process died that we didn't know about!
    assert(0);
  }
  childPid[childPort] = INVALID_PID;
  close(DATA_READER(childPort));
  close(CONTROL_WRITER(childPort));
  DATA_READER(childPort) = INVALID_FD;
  CONTROL_WRITER(childPort) = INVALID_FD;
  // BugzId:12928 Parent needs to close its socket too
  if (backchannelEnable) {
    backchannelCloseConnection(childPort);
  }
}

static void childCleanupAndExit(void)
{
  if (childProcessPort == SERIAL_PORT_CLI) {
    writeHistory();
  }
  if (childProcessPort != -1) {
    if (backchannelEnable) {
      backchannelStopServer(childProcessPort);
    }
    close(DATA_WRITER(childProcessPort));
    close(CONTROL_READER(childProcessPort));
  }

  exit(0);
}

// This works only for the command interpreter.
// Loop and get pointers to all the strings of the available commands.
void emberSerialCommandCompletionInit(EmberCommandEntry listOfCommands[])
{
#if READLINE_SUPPORT
  allCommands = listOfCommands;
  rl_attempted_completion_function = commandCompletion;
  rl_completion_entry_function = filenameCompletion;
#endif
  usingCommandInterpreter = true;
}

// This is for QA's cli
// We no longer support command completion in this case.
void emberSerialCommandCompletionInitCli(cliSerialCmdEntry cliCmdList[],
                                         int cliCmdListLength)
{
#if READLINE_SUPPORT
  allCommands = NULL;
#endif
}

//------------------------------------------------------------------------------
// Serial Input

int emberSerialGetInputFd(uint8_t port)
{
	if (port > (NUM_PORTS - 1)) {
		return INVALID_FD;
	}
	return DATA_READER(port);
}

static bool readyForSerialInput(uint8_t port)
{
  if (useControlChannel) {
    char data = '1';
    if (1 == write(CONTROL_WRITER(port), &data, 1)) {
      //debugPrint("Sent reader process 'go-ahead' signal.\n");
    } else {
      // We assume this function is only used for CLI input.
      // If the CLI process died, then the parent should also
      // go away.
      // BugzId:12928: Except in the case of backchannel CLI
      // client, where parent cleans up and awaits a new client.
      // For the RAW input, which is only accessible with the
      // backchannel enabled, that child process may come and go
      // and we don't need to worry about it.
      if (backchannelEnable) {
        return true; // Let caller know we didn't deliver go-ahead
      }
      emberSerialCleanup();
      exit(-1);
    }
  }
  return false; // go-ahead delivered ok
}

// returns # bytes available for reading
uint16_t emberSerialReadAvailable(uint8_t port)
{
  // I am not aware of a good way to get the exact number
  // of bytes available for reading from an anonymous pipe.
  // We use select() to see if there is any data available for
  // reading and that means there is at least one byte present
  // or EOF (pipe closed) has happened.
  fd_set readSet;
  struct timeval timeout = { 0, 0 }; // return immediately
  int fdsWithData;

  if (DATA_READER(port) == INVALID_FD) {
    return 0;
  }

  FD_ZERO(&readSet);
  FD_SET(DATA_READER(port), &readSet);

  fdsWithData = select(DATA_READER(port) + 1,   // per the man page
                       &readSet,
                       NULL,
                       NULL,
                       &timeout);
  if (fdsWithData < 0) {
    fprintf(stderr,
            "Fatal: select() returned error: %s\n",
            strerror(errno));
    assert(false);
  }
  return (fdsWithData > 0 ? 1 : 0);
}

void emberSerialSendReadyToRead(uint8_t port)
{
  if (port == SERIAL_PORT_CLI && sendGoAhead) {
    sendGoAhead = readyForSerialInput(port);
  }
}

// This should only be called by the parent process (i.e. the main app)
EmberStatus emberSerialReadByte(uint8_t port, uint8_t *dataByte)
{
  static bool waitForEol = false;
  size_t bytes;

  if (backchannelEnable
      && !handleRemoteConnection(port)) {
    if (port == SERIAL_PORT_CLI) {
      // BugzId:12928 While waiting for new client, pretend serial port empty
      // and set sendGoAhead to force a go-ahead when new client connects
      sendGoAhead = true;
      return EMBER_SERIAL_RX_EMPTY;
    }
    return EMBER_SERIAL_INVALID_PORT;
  }

  // The command interpreter reads bytes until it gets an EOL.
  // The CLI reads bytes until it gets a "\r\n".
  // The latter is easily supported since the data is sitting
  // in our data pipe.  The former requires a little extra work.

  // For the command interpreter We don't want to send the 'go-ahead' until
  // the parent reads a byte of data after it has read the EOL.
  if (waitForEol) {
    waitForEol = false;
    return EMBER_SERIAL_RX_EMPTY;
  }

  emberSerialSendReadyToRead(port);

  if (0 == emberSerialReadAvailable(port)) {
    return EMBER_SERIAL_RX_EMPTY;
  }

   // debugPrint("Parent: Data is ready.\n");
  bytes = read(DATA_READER(port), (void*)dataByte, 1);

  // We have read the entire line of input, the serial input process will
  // be blocking until we tell it to go ahead and read more input.
  // BugzId:12928 Treat EOF as an EOL
  if (port == SERIAL_PORT_CLI && (bytes == 0 || *dataByte == '\n')) {
    sendGoAhead = true;
    if (usingCommandInterpreter) {
      waitForEol = true;
    }
  }

  if (bytes == 1) {
//    debugPrint("emberSerialReadByte(): %c\n", (char)*dataByte);
    return EMBER_SUCCESS;
  } 
  else if (bytes == 0) {
    // BugzId:12928 clean up for new client
    // read() only returns 0 on EOF; a non-blocking
    // read() returns -1/EAGAIN on no data but that shouldn't
    // happen per the emberSerialReadAvailable() check above.
    kill(childPid[port], SIGTERM); // Should be superfluous, but make sure
    parentCleanupAfterChildDied();
    *dataByte = '\n';
    return EMBER_SUCCESS;
  } else {
    return EMBER_ERR_FATAL;
  }
}

#if !READLINE_SUPPORT
// Support for those systems without the readline library.
static char* readline(const char* prompt)
{
  uint8_t i = 0;
  char* data = malloc(EMBER_COMMAND_BUFFER_LENGTH + 1);  // add 1 for '\0'
  bool done = false;

  debugPrint("Allocated %d bytes for readline()\n", EMBER_COMMAND_BUFFER_LENGTH + 1);

  if (data == NULL) {
    return NULL;
  }

  fprintf(stdout, "%s", prompt);
  fflush(stdout);

  while (!done && i < EMBER_COMMAND_BUFFER_LENGTH) {
    ssize_t bytes = read(STDIN, &(data[i]), 1);
    if (bytes == -1) {
      if (errno == EINTR) {
        continue;
      } else {
        fprintf(stderr, "Attempt to read from STDIN failed!\n");
        assert(false);
      }
    } else if (bytes == 0) {
      // BugzId:12928 EOF encountered, return NULL like real readline() would
      free(data);
      return(NULL);
    }

//    debugPrint("char: 0x%02x\n", (int)data[i]);
    if (data[i] == LINE_FEED        // line feed
        || data[i] == EOF_CHAR) {
      done = true;
    }
    i++;
  }
  if (data[i - 1] == LINE_FEED) {
    // Don't need the LF delineator, it is implied when this function
    // returns.
    i--;
  }
  data[i] = '\0';
  if (data[0] == EOF_CHAR) {
    // BugzId:12928 EOF encountered, return NULL to caller
    free(data);
    return(NULL);
  }
  return data;
}
#endif

// This should only be called by the parent process (i.e. the main app)
EmberStatus emberSerialReadLine(uint8_t port, char *data, uint8_t max)
{
  uint8_t count = 0;
  EmberStatus status;
  while (count < max) {
    status = emberSerialReadByte(port, (uint8_t*)&(data[count]));
    if (EMBER_SUCCESS == status) {
      count++;
    } else if (EMBER_SERIAL_RX_EMPTY == status) {
      return EMBER_SUCCESS;
    }
  }
  return EMBER_SUCCESS;
}

// Read a line of data and write it to our pipe.
// Wait for the other process to give us the go-ahead.

static void processSerialInput(uint8_t port)
{
  int length;
  char* readData = NULL;
  char newLine[] = "\r\n";  // The CLI code requires \r\n as the final bytes.
  char goAhead;
  char singleByte[2];

  if (port == SERIAL_PORT_CLI) {
    initializeHistory();
  }

  debugPrint("Processing input for port %d.\n", port);

  while (1) {
    if (port == SERIAL_PORT_CLI && useControlChannel) {
      // Wait for the parent to give the "go ahead"
    //  debugPrint("Child process waiting for parent's 'go-ahead'.\n");
      read(CONTROL_READER(port), &goAhead, 1);
    }

    if (port == SERIAL_PORT_CLI) {
      // If we are operating in a backchannel environment, then STDIN has
      // already been re-mapped to the correct socket.

      // This performs a malloc()
      readData = readline(prompt);
    } else {
      // Raw data
      readData = &(singleByte[0]);
      bool success = (backchannelEnable
                      ? (EMBER_SUCCESS == backchannelReceive(port, readData))
                      : (1 != read(STDIN, readData, 1)));

      if (!success) {
        childCleanupAndExit();
      }
      singleByte[1] = '\0';
    }
    if (readData == NULL) {
      // BugzId:12928 On disconnect, let child die; parent will deal with it
      fprintf(stderr,
              "Child process for serial input got EOF. Exiting\n");
      childCleanupAndExit();
    }
    length = (port == SERIAL_PORT_CLI
              ? strnlen(readData, 255)  // 255 is an arbitrary maximum
              : 1);
    assert(-1 < write(DATA_WRITER(port), readData, length));

    if (port == SERIAL_PORT_CLI) {
      //debugPrint("readline() input (%d bytes): %s\n", length, readData);
      assert(2 == write(DATA_WRITER(port), &newLine, 2));
      if (length > 0) {
        add_history(readData);
      }
    }

    if (port == SERIAL_PORT_CLI) {
      free(readData);
    }
  }
}

//------------------------------------------------------------------------------
// Serial Output

EmberStatus emberSerialGuaranteedPrintf(uint8_t port, PGM_P format, ...)
{
  va_list vargs = { 0 };
  EmberStatus stat = EMBER_SUCCESS;

  va_start(vargs, format);
  stat = emberSerialPrintfVarArg(port, format, vargs);
  va_end(vargs);
  return stat;
}

// This is implemented because a few applications count on this internal
// function.  This is essentially a 'sprintf()' where the application wants
// formatted text output but uses its own mechanism to print.  We will ignore
// that use since it doesn't make much sense in the case of a gateway
// application.  If the application really wants to do something special for
// printing formatted output it should call sprintf() on its own.
uint8_t emPrintfInternal(emPrintfFlushHandler handler,
                         uint8_t port,
                         PGM_P buff,
                         va_list list)
{
  return (EmberStatus)emberSerialPrintfVarArg(port, buff, list);
}

// Main printing routine.
// Calls into normal C 'vprintf()'
EmberStatus emberSerialPrintfVarArg(uint8_t port, PGM_P formatString, va_list ap)
{
  EmberStatus stat = EMBER_SERIAL_INVALID_PORT;
  char* newFormatString = transformEmberPrintfToStandardPrintf(formatString,
                                                               !backchannelEnable);
  if (newFormatString == NULL) {
    return EMBER_NO_BUFFERS;
  }
  if (backchannelEnable) {
    if (handleRemoteConnection(port)) {
      stat = backchannelClientVprintf(port, newFormatString, ap);
      if (stat == EMBER_ERR_FATAL) {
        // BugzId:12928 Defer cleanup of disappeared client to emberSerialReadByte()
      }
    }
  } else {
    stat = (0 == vprintf(newFormatString, ap)
            ? EMBER_ERR_FATAL
            : EMBER_SUCCESS);
    fflush(stdout);
  }
  free(newFormatString);
  return stat;
}

EmberStatus emberSerialPrintf(uint8_t port, PGM_P format, ...)
{
  va_list vargs = { 0 };
  EmberStatus stat = EMBER_SUCCESS;

  va_start(vargs, format);
  stat = emberSerialPrintfVarArg(port, format, vargs);
  va_end(vargs);
  return stat;
}

EmberStatus emberSerialPrintfLine(uint8_t port, PGM_P formatString, ...)
{
  EmberStatus stat = EMBER_SUCCESS;
  va_list ap = { 0 };
  va_start(ap, formatString);
  stat = emberSerialPrintfVarArg(port, formatString, ap);
  va_end(ap);
  emberSerialPrintCarriageReturn(port);
  return stat;
}

EmberStatus emberSerialPrintCarriageReturn(uint8_t port)
{
  return emberSerialPrintf(port, "\r\n");
}

EmberStatus emberSerialWriteByte(uint8_t port, uint8_t dataByte)
{
  return emberSerialWriteData(port, &dataByte, 1);
}

EmberStatus emberSerialWriteHex(uint8_t port, uint8_t dataByte)
{
  uint8_t hex[3];
  sprintf((char*)hex, "%2x", dataByte);
  return emberSerialWriteData(port, hex, 2);
}

EmberStatus emberSerialWriteData(uint8_t port, uint8_t *data, uint8_t length)
{
  EmberStatus stat = EMBER_ERR_FATAL;
  if (backchannelEnable) {
    if (handleRemoteConnection(port)) {
      // Ideally we expect the "emberSerialWriteData()" call to be atomic,
      // and not block.  We probably should try to determine if the write()
      // will block before executing it.
      stat = backchannelSend(port, data, length);
      if (stat == EMBER_ERR_FATAL) {
        // BugzId:12928 Defer cleanup of disappeared client to emberSerialReadByte()
      }
    } // else
      //   failure, fall-thru
  } else {
    // Normal IO
    if (fwrite(data, length, 1, stdout) == 1) {
      stat = EMBER_SUCCESS;
    }
  }
  return stat;
}

EmberStatus emberSerialWaitSend(uint8_t port)
{
  if (!backchannelEnable) {
    fflush(stdout);
  }
  return EMBER_SUCCESS;
}

//------------------------------------------------------------------------------
// Serial buffer maintenance

void emberSerialFlushRx(uint8_t port)
{
  char buf;
  while (0 < emberSerialReadAvailable(port)) {
    read(DATA_READER(port), &buf, 1);
  }
}

//------------------------------------------------------------------------------
// Serial Buffer Cleanup Tick

void emberSerialBufferTick(void)
{
}

//------------------------------------------------------------------------------
// Command Completion support

#if READLINE_SUPPORT

// See readline manual for additional details:
//   http://tiswww.case.edu/php/chet/readline/readline.html#SEC45

// We can complete the command and all complex commands (i.e. zcl global read ...)
// Complex commands require us to traverse the EmberCommandEntry tree and
// go into each sub-menu.  We will also display (though not complete)
// the syntax for each command.

// Per readline integration:
// Attempt to complete on the contents of TEXT.  START and END bound the
// region of rl_line_buffer that contains the word to complete.  TEXT is
// the word to complete.  We can use the entire contents of rl_line_buffer
// in case we want to do some simple parsing.  Return the array of matches,
// or NULL if there aren't any.

static EmberCommandEntry* subMenuPtr = NULL;
static const char* commandArgumentPtr = NULL;

static const char* findNextNonSpaceChar(const char* line)
{
  while (*line != '\0' && *line == ' ') {
    line++;
  }
  return line;
}

static const char* findNextSpaceChar(const char* line)
{
  while (*line != '\0' && *line != ' ') {
    line++;
  }
  return line;
}

// Recursive function.  Traverses the command-tree
// to find our current sub-menu so we can complete the
// commands for that branch.  If we are at a command-leaf,
// then for completing arguments we figure out where in the arguments
// we are (i.e. the command 'zcl global read <uint16_t> <uint16_t>'
// has two arguments, and we can determine if the line has the first or second
// one in it.
// This function returns true when it has sub-menus/elements that can be parsed.
// It returns false otherwise.
static void setCurrentCommandPtr(const char* nextToken)
{
  const char* wordStart = findNextNonSpaceChar(nextToken == NULL
                                               ? rl_line_buffer
                                               : nextToken);
  const char* wordEnd   = findNextSpaceChar(wordStart);

  uint8_t i = 0;

  if (nextToken == NULL) {
    subMenuPtr = allCommands;
    commandArgumentPtr = NULL;
  }

  if (wordEnd == wordStart) {
    // Nothing to parse.
    return;
  }

  if (commandArgumentPtr && *wordEnd == ' ') {
    if (*commandArgumentPtr != '\0') {
      commandArgumentPtr++;
      setCurrentCommandPtr(wordEnd);
    }
    return;
  }

  while (subMenuPtr != NULL && subMenuPtr[i].name != NULL) {
    // A token is only really complete when there is a space
    // at the end, which is appended by the readline() library
    // code.  In other words, when the user types the first
    // piece of command 'foo' as 'fo' a TAB will complete this
    // as 'foo '.  If the user types the complete 'foo' WITHOUT
    // a space then we want the readline code to validate and
    // complete this by appending a space.
    // This only really matters for complex commands such as 'foo bar'.
    // In order for the 'bar' to be presented as an option when only
    // 'foo' is typed, we need to make sure that a space is appended
    // after that token.
    // This long winded explanation is intended for why we have
    // the "&& *wordEnd == ' '" portion of the if statement.
    if (0 == strncmp(subMenuPtr[i].name,
                     wordStart,
                     strlen(subMenuPtr[i].name))
        && *wordEnd == ' ') {
      if (subMenuPtr[i].action == NULL) {
        // This must be a sub-menu.
        subMenuPtr = (EmberCommandEntry*)subMenuPtr[i].argumentTypes;
        setCurrentCommandPtr(wordEnd);
        return;
      } else {
        // We have parsed all the known sub-menus and are now at an actual
        // command.  We will no longer walk down the command-tree.  But
        // we will continue to parse the command tokens.
        commandArgumentPtr = subMenuPtr[i].argumentTypes;
        subMenuPtr = NULL;
        setCurrentCommandPtr(wordEnd);
      }

      if (subMenuPtr == NULL) {
        return;
      }
    }
    i++;
  }
}

char** commandCompletion(const char* text, int start, int end)
{
  char** matches;

  matches = (char **)NULL;

  debugPrint("commandCompletion(): %s, start: %d, end: %d\n",
             text,
             start,
             end);

  // If this word is at the start of the line, then it is the first
  // token in a command.  Just complete the command.
  // Otherwise we need to find where we are in the current
  // command tree and use that for our search.

  setCurrentCommandPtr(NULL);
  matches = rl_completion_matches(text, singleCommandCompletion);

  return (matches);
}

// Generator function for command completion.  STATE lets us know whether
// to start from scratch; without any state (i.e. STATE == 0), then we
// start at the top of the list.

char* singleCommandCompletion(const char *text, int state)
{
  static int listIndex;
  static int length;

  if (!state) {
    listIndex = 0;
    length = strnlen(text, EMBER_COMMAND_BUFFER_LENGTH + 1);
    // EMBER_COMMAND_BUFFER_LENGTH needs to be adjusted if this assert() is hit.
    assert(length <= EMBER_COMMAND_BUFFER_LENGTH);
  }

  debugPrint("singleCommandCompletion(): listIndex: %d\n", listIndex);

  if (commandArgumentPtr
      && *commandArgumentPtr != '\0'
      && !state) {
    const char* argumentHelp[] = {
      "uint8_t",
      "uint16_t",
      "uint32_t",
      "int8_t",
      "quoted-string-or-hex-array"
    };
    int argumentIndex = -1;

    switch (*commandArgumentPtr) {
      case 'u':
        argumentIndex = 0;
        break;
      case 'v':
        argumentIndex = 1;
        break;
      case 'w':
        argumentIndex = 2;
        break;
      case 's':
        argumentIndex = 3;
        break;
      case 'b':
        argumentIndex = 4;
        break;
    }

    if (argumentIndex >= 0) {
      printf("\n%s <%s>\n",
             rl_line_buffer,
             argumentHelp[argumentIndex]);
      rl_on_new_line();
      rl_redisplay();
    }

    // Always returning NULL prevents command-completion from working,
    // which is what we want.  We can't complete the arbitrary uint16_t
    // and uint8_t types, but we do want to display what's next.
    return (char*)NULL;
  }

  while (subMenuPtr != NULL
         && subMenuPtr[listIndex].name != NULL) {
    const char* command = subMenuPtr[listIndex].name;
    listIndex++;

    // The trick here is that an empty command line has a length of 0,
    // and thus will match in our strncmp()
    if (0 == strncmp(command, text, length)) {
//      debugPrint("singleCommandCompeletion(): found %s\n", command);

      // It is expected that the readline library will free the data.
      return duplicateString(command);
    }
  }

  debugPrint("singleCommandCompeletion(): no more completions\n");
  return (char*)NULL;
}

char* filenameCompletion(const char* text, int state)
{
  // We don't do filename completion.

  // Implementing this function prevents readline() from using its default
  // implementation, which displays filenames from the CWD.  Not very helpful.
  return (char*)NULL;
}

static char* duplicateString(const char* source)
{
  char* returnData;
  returnData = malloc(strnlen(source, EMBER_COMMAND_BUFFER_LENGTH) + 1);

  assert(returnData != NULL);

  strcpy(returnData, source);
  return returnData;
}

static void initializeHistory(void)
{
  int myErrno;
  char* homeDirectory;

  readlineHistoryPath[0] = '\0';

  using_history();  // initialize readline() history.

  homeDirectory = getenv("HOME");
  if (homeDirectory == NULL) {
    debugPrint("Error: HOME directory env variable is not defined.\n");
    return;
  } else {
    debugPrint("Home Directory: %s\n", homeDirectory);
  }

  snprintf(readlineHistoryPath,
           MAX_STRING_LENGTH - 1,
           "%s/%s",
           homeDirectory,
           readlineHistoryFilename);
  readlineHistoryPath[MAX_STRING_LENGTH - 1] = '\0';

  myErrno = read_history(readlineHistoryPath);
  if (myErrno != 0) {
    debugPrint("Could not open history file '%s': %s\n",
               readlineHistoryPath,
               strerror(myErrno));
  } else {
    debugPrint("%d history entries read from file '%s'\n",
               history_length,         // readline global
               readlineHistoryPath);
  }
}

static void writeHistory(void)
{
  int myErrno;
  if (readlineHistoryPath[0] == '\0') {
    debugPrint("Readline history path is empty, not writing history.\n");
    return;
  }
  myErrno = write_history(readlineHistoryPath);
  if (myErrno != 0) {
    debugPrint("Failed to write history file '%s': %s\n",
               readlineHistoryPath,
               strerror(myErrno));
  } else {
    debugPrint("Wrote %d entries to history file '%s'\n",
               history_length,       // readline global
               readlineHistoryPath);
  }
}

#endif // #if READLINE_SUPPORT

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

/*
   Ideally we want the parent process to catch SIGCHLD and then
   cleanup after the child.  If the CLI terminated, the parent
   process will also quit.

   The question is, how do I determine which child quit?
   Presumably this can be done with wait(), which returns the
   child's pid and its associated status.

   The one tricky part is the issue with handling the signal
   SIGCHLD during normal shutdown.  I presume that a signal
   handler cannot be fired within a signal handel, but I am
   not certain.  I need a way of telling whether this
   is a graceful shutdown or not.  Basically if the parent
   initiated the shutdown the children should just go away
   and the parent would ignore notification that they did that
   (SIGCHLD).  If the child died unexpectedly then the parent
   will not ignore that and should nicely cleanup
   and not necessarily exit (in the case of the CLI we presume
   it would).
 */
static void signalHandler(int signal)
{
  static bool reportSigttin = true;
  const char* signalName = strsignal(signal);

  if (signal == SIGTTOU) {
    debugOn = false;
    return;
  }

  if (signal != SIGTTIN || reportSigttin) {
    debugPrint("%s caught signal %s (%d)\n",
               (amParent ? "Parent" : "Child"),
               (signalName == NULL ? "???" : signalName),
               signal);
  }

  if (signal == SIGPIPE) {
    // Ignore this.
    // In general SIGPIPE is useless because it doesn't
    //   give you any context for understanding what read/write from
    //   a pipe caused the error.
    // Instead the send(), receive(), read(), and write() primitives
    //  will return with an error and set errno to EPIPE indicating
    //  that failure.
    return;
  }

  if (signal == SIGTTIN) {
    reportSigttin = false;

    return;
  }

  /*
     if (signal == SIGCHLD) {
     if (amParent) {
      parentCleanupAfterChildDied();
     }
     return;
     }
   */

  // Assume that this is only called for SIGTERM and SIGINT
  if (amParent) {
    emberSerialCleanup();
    exit(-1);
  } else {
    childCleanupAndExit();
  }
}

static void installSignalHandler(void)
{
	int i = 0;
	int signals[] = { SIGTERM, SIGINT, SIGPIPE, SIGTTIN, SIGTTOU, 0 };

	struct sigaction sigAction;
	MEMSET(&sigAction, 0, sizeof(struct sigaction));
	sigAction.sa_handler = signalHandler;

	while (signals[i] != 0) {
		if (-1 == sigaction(signals[i], &sigAction, NULL)) {
			debugPrint("Could not install signal handler.\n");
		} else {
			const char* signalName = strsignal(signals[i]);
			debugPrint("%s installed signal handler for %s (%d).\n",
						(amParent ? "Parent" : "Child"),
						(signalName == NULL
						 ? "???"
						 : signalName),
						signals[i]);
		}
		i++;
	}
}

//------------------------------------------------------------------------------
// Debug

static void debugPrint(const char* formatString, ...)
{
	if (debugOn) {
		va_list ap = { 0 };
		fprintf(stderr, "[linux-serial debug] ");
		va_start(ap, formatString);
		vfprintf(stderr, formatString, ap);
		va_end(ap);
		fflush(stderr);
	}
}
