filter.c

/**
 * @file
 * Pass files through external commands (filters)
 *
 * @authors
 * Copyright (C) 1996-2000 Michael R. Elkins.
 *
 * @copyright
 * 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, see <http://www.gnu.org/licenses/>.
 */

/**
 * @page filter Pass files through external commands (filters)
 *
 * Pass files through external commands (filters)
 */

#include "config.h"
#include <stdlib.h>
#include <sys/wait.h>
#include <unistd.h>
#include "mutt/mutt.h"
#include "mutt.h"
#include "filter.h"
#include "mutt_window.h"

/**
 * mutt_create_filter_fd - Run a command on a pipe (optionally connect stdin/stdout)
 * @param cmd   Command line to invoke using `sh -c`
 * @param in    File stream pointing to stdin for the command process, can be NULL
 * @param out   File stream pointing to stdout for the command process, can be NULL
 * @param err   File stream pointing to stderr for the command process, can be NULL
 * @param fdin  If `in` is NULL and fdin is not -1 then fdin will be used as stdin for the command process
 * @param fdout If `out` is NULL and fdout is not -1 then fdout will be used as stdout for the command process
 * @param fderr If `error` is NULL and fderr is not -1 then fderr will be used as stderr for the command process
 * @retval num PID of the created process
 * @retval -1  Error creating pipes or forking
 *
 * This function provides multiple mechanisms to handle IO sharing for the
 * command process. File streams are prioritized over file descriptors if
 * present.
 *
 * @code{.c}
 *    mutt_create_filter_fd(commandline, NULL, NULL, NULL, -1, -1, -1);
 * @endcode
 *
 * Additionally, in, out, and err will point to FILE* streams representing the
 * processes stdin, stdout, and stderr.
 */
pid_t mutt_create_filter_fd(const char *cmd, FILE **in, FILE **out, FILE **err,
                            int fdin, int fdout, int fderr)
{
  int pin[2], pout[2], perr[2], thepid;

  if (in)
  {
    *in = 0;
    if (pipe(pin) == -1)
      return -1;
  }

  if (out)
  {
    *out = 0;
    if (pipe(pout) == -1)
    {
      if (in)
      {
        close(pin[0]);
        close(pin[1]);
      }
      return -1;
    }
  }

  if (err)
  {
    *err = 0;
    if (pipe(perr) == -1)
    {
      if (in)
      {
        close(pin[0]);
        close(pin[1]);
      }
      if (out)
      {
        close(pout[0]);
        close(pout[1]);
      }
      return -1;
    }
  }

  mutt_sig_block_system();

  thepid = fork();
  if (thepid == 0)
  {
    mutt_sig_unblock_system(false);

    if (in)
    {
      close(pin[1]);
      dup2(pin[0], 0);
      close(pin[0]);
    }
    else if (fdin != -1)
    {
      dup2(fdin, 0);
      close(fdin);
    }

    if (out)
    {
      close(pout[0]);
      dup2(pout[1], 1);
      close(pout[1]);
    }
    else if (fdout != -1)
    {
      dup2(fdout, 1);
      close(fdout);
    }

    if (err)
    {
      close(perr[0]);
      dup2(perr[1], 2);
      close(perr[1]);
    }
    else if (fderr != -1)
    {
      dup2(fderr, 2);
      close(fderr);
    }

    if (MuttIndexWindow && (MuttIndexWindow->cols > 0))
    {
      char columns[11];
      snprintf(columns, sizeof(columns), "%d", MuttIndexWindow->cols);
      mutt_envlist_set("COLUMNS", columns, 1);
    }

    execle(EXECSHELL, "sh", "-c", cmd, NULL, mutt_envlist_getlist());
    _exit(127);
  }
  else if (thepid == -1)
  {
    mutt_sig_unblock_system(true);

    if (in)
    {
      close(pin[0]);
      close(pin[1]);
    }

    if (out)
    {
      close(pout[0]);
      close(pout[1]);
    }

    if (err)
    {
      close(perr[0]);
      close(perr[1]);
    }

    return -1;
  }

  if (out)
  {
    close(pout[1]);
    *out = fdopen(pout[0], "r");
  }

  if (in)
  {
    close(pin[0]);
    *in = fdopen(pin[1], "w");
  }

  if (err)
  {
    close(perr[1]);
    *err = fdopen(perr[0], "r");
  }

  return thepid;
}

/**
 * mutt_create_filter - Set up filter program
 * @param[in]  s   Command string
 * @param[out] in  FILE pointer of stdin
 * @param[out] out FILE pointer of stdout
 * @param[out] err FILE pointer of stderr
 * @retval num PID of filter
 */
pid_t mutt_create_filter(const char *s, FILE **in, FILE **out, FILE **err)
{
  return mutt_create_filter_fd(s, in, out, err, -1, -1, -1);
}

/**
 * mutt_wait_filter - Wait for the exit of a process and return its status
 * @param pid Process id of the process to wait for
 * @retval num Exit status of the process identified by pid
 * @retval -1  Error
 */
int mutt_wait_filter(pid_t pid)
{
  int rc;

  waitpid(pid, &rc, 0);
  mutt_sig_unblock_system(true);
  rc = WIFEXITED(rc) ? WEXITSTATUS(rc) : -1;

  return rc;
}