compress.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>
#include "mutt/lib.h"
#include "config/lib.h"
#include "core/lib.h"
#include "gui/lib.h"
#include "lib.h"
#include "format_flags.h"
#include "globals.h"
#include "hook.h"
#include "muttlib.h"
#include "mx.h"
#include "protos.h"
 
struct Email;
 
static const struct Command CompCommands[] = {
  // clang-format off
  { "append-hook", mutt_parse_hook, MUTT_APPEND_HOOK },
  { "close-hook",  mutt_parse_hook, MUTT_CLOSE_HOOK },
  { "open-hook",   mutt_parse_hook, MUTT_OPEN_HOOK },
  // clang-format on
};
 
void mutt_comp_init(void)
{
  commands_register(CompCommands, mutt_array_size(CompCommands));
}
 
static bool lock_realpath(struct Mailbox *m, bool excl)
{
  if (!m || !m->compress_info)
    return false;
 
  struct CompressInfo *ci = m->compress_info;
 
  if (ci->locked)
    return true;
 
  if (excl)
    ci->fp_lock = fopen(m->realpath, "a");
  else
    ci->fp_lock = fopen(m->realpath, "r");
  if (!ci->fp_lock)
  {
    mutt_perror(m->realpath);
    return false;
  }
 
  int r = mutt_file_lock(fileno(ci->fp_lock), excl, true);
  if (r == 0)
    ci->locked = true;
  else if (excl)
  {
    mutt_file_fclose(&ci->fp_lock);
    m->readonly = true;
    return true;
  }
 
  return r == 0;
}
 
static void unlock_realpath(struct Mailbox *m)
{
  if (!m || !m->compress_info)
    return;
 
  struct CompressInfo *ci = m->compress_info;
 
  if (!ci->locked)
    return;
 
  mutt_file_unlock(fileno(ci->fp_lock));
 
  ci->locked = false;
  mutt_file_fclose(&ci->fp_lock);
}
 
static int setup_paths(struct Mailbox *m)
{
  if (!m)
    return -1;
 
  /* Setup the right paths */
  mutt_str_replace(&m->realpath, mailbox_path(m));
 
  /* We will uncompress to TMPDIR */
  struct Buffer *buf = mutt_buffer_pool_get();
  mutt_buffer_mktemp(buf);
  mutt_buffer_copy(&m->pathbuf, buf);
  mutt_buffer_pool_release(&buf);
 
  FILE *fp = mutt_file_fopen(mailbox_path(m), "w");
  if (!fp)
    return -1;
 
  mutt_file_fclose(&fp);
  return 0;
}
 
static void store_size(const struct Mailbox *m)
{
  if (!m || !m->compress_info)
    return;
 
  struct CompressInfo *ci = m->compress_info;
 
  ci->size = mutt_file_get_size(m->realpath);
}
 
static struct CompressInfo *set_compress_info(struct Mailbox *m)
{
  if (!m)
    return NULL;
 
  if (m->compress_info)
    return m->compress_info;
 
  /* Open is compulsory */
  const char *o = mutt_find_hook(MUTT_OPEN_HOOK, mailbox_path(m));
  if (!o)
    return NULL;
 
  const char *c = mutt_find_hook(MUTT_CLOSE_HOOK, mailbox_path(m));
  const char *a = mutt_find_hook(MUTT_APPEND_HOOK, mailbox_path(m));
 
  struct CompressInfo *ci = mutt_mem_calloc(1, sizeof(struct CompressInfo));
  m->compress_info = ci;
 
  ci->cmd_open = mutt_str_dup(o);
  ci->cmd_close = mutt_str_dup(c);
  ci->cmd_append = mutt_str_dup(a);
 
  return ci;
}
 
static void compress_info_free(struct Mailbox *m)
{
  if (!m || !m->compress_info)
    return;
 
  struct CompressInfo *ci = m->compress_info;
  FREE(&ci->cmd_open);
  FREE(&ci->cmd_close);
  FREE(&ci->cmd_append);
 
  unlock_realpath(m);
 
  FREE(&m->compress_info);
}
 
static const char *compress_format_str(char *buf, size_t buflen, size_t col, int cols,
                                       char op, const char *src, const char *prec,
                                       const char *if_str, const char *else_str,
                                       intptr_t data, MuttFormatFlags flags)
{
  if (!buf || (data == 0))
    return src;
 
  struct Mailbox *m = (struct Mailbox *) data;
 
  /* NOTE the compressed file config vars expect %f and %t to be
   * surrounded by '' (unlike other NeoMutt config vars, which add the
   * outer quotes for the user).  This is why we use the
   * mutt_buffer_quote_filename() form with add_outer of false. */
  struct Buffer *quoted = mutt_buffer_pool_get();
  switch (op)
  {
    case 'f':
      /* Compressed file */
      mutt_buffer_quote_filename(quoted, m->realpath, false);
      snprintf(buf, buflen, "%s", mutt_buffer_string(quoted));
      break;
    case 't':
      /* Plaintext, temporary file */
      mutt_buffer_quote_filename(quoted, mailbox_path(m), false);
      snprintf(buf, buflen, "%s", mutt_buffer_string(quoted));
      break;
  }
 
  mutt_buffer_pool_release(&quoted);
  return src;
}
 
static void expand_command_str(const struct Mailbox *m, const char *cmd, char *buf, int buflen)
{
  if (!m || !cmd || !buf)
    return;
 
  mutt_expando_format(buf, buflen, 0, buflen, cmd, compress_format_str,
                      (intptr_t) m, MUTT_FORMAT_NO_FLAGS);
}
 
static int execute_command(struct Mailbox *m, const char *command, const char *progress)
{
  if (!m || !command || !progress)
    return 0;
 
  if (m->verbose)
    mutt_message(progress, m->realpath);
 
  int rc = 1;
  char sys_cmd[STR_COMMAND] = { 0 };
 
  mutt_sig_block();
  endwin();
  fflush(stdout);
 
  expand_command_str(m, command, sys_cmd, sizeof(sys_cmd));
 
  if (mutt_system(sys_cmd) != 0)
  {
    rc = 0;
    mutt_any_key_to_continue(NULL);
    mutt_error(_("Error running \"%s\""), sys_cmd);
  }
 
  mutt_sig_unblock();
 
  return rc;
}
 
bool mutt_comp_can_append(struct Mailbox *m)
{
  if (!m)
    return false;
 
  /* If this succeeds, we know there's an open-hook */
  struct CompressInfo *ci = set_compress_info(m);
  if (!ci)
    return false;
 
  /* We have an open-hook, so to append we need an append-hook,
   * or a close-hook. */
  if (ci->cmd_append || ci->cmd_close)
    return true;
 
  mutt_error(_("Can't append without an append-hook or close-hook : %s"), mailbox_path(m));
  return false;
}
 
bool mutt_comp_can_read(const char *path)
{
  if (!path)
    return false;
 
  if (mutt_find_hook(MUTT_OPEN_HOOK, path))
    return true;
 
  return false;
}
 
int mutt_comp_valid_command(const char *cmd)
{
  if (!cmd)
    return 0;
 
  return strstr(cmd, "%f") && strstr(cmd, "%t");
}
 
static bool comp_ac_owns_path(struct Account *a, const char *path)
{
  return false;
}
 
static bool comp_ac_add(struct Account *a, struct Mailbox *m)
{
  return true;
}
 
static enum MxOpenReturns comp_mbox_open(struct Mailbox *m)
{
  struct CompressInfo *ci = set_compress_info(m);
  if (!ci)
    return MX_OPEN_ERROR;
 
  /* If there's no close-hook, or the file isn't writable */
  if (!ci->cmd_close || (access(mailbox_path(m), W_OK) != 0))
    m->readonly = true;
 
  if (setup_paths(m) != 0)
    goto cmo_fail;
  store_size(m);
 
  if (!lock_realpath(m, false))
  {
    mutt_error(_("Unable to lock mailbox"));
    goto cmo_fail;
  }
 
  int rc = execute_command(m, ci->cmd_open, _("Decompressing %s"));
  if (rc == 0)
    goto cmo_fail;
 
  unlock_realpath(m);
 
  m->type = mx_path_probe(mailbox_path(m));
  if (m->type == MUTT_UNKNOWN)
  {
    mutt_error(_("Can't identify the contents of the compressed file"));
    goto cmo_fail;
  }
 
  ci->child_ops = mx_get_ops(m->type);
  if (!ci->child_ops)
  {
    mutt_error(_("Can't find mailbox ops for mailbox type %d"), m->type);
    goto cmo_fail;
  }
 
  m->account->type = m->type;
  return ci->child_ops->mbox_open(m);
 
cmo_fail:
  /* remove the partial uncompressed file */
  (void) remove(mailbox_path(m));
  compress_info_free(m);
  return MX_OPEN_ERROR;
}
 
static bool comp_mbox_open_append(struct Mailbox *m, OpenMailboxFlags flags)
{
  /* If this succeeds, we know there's an open-hook */
  struct CompressInfo *ci = set_compress_info(m);
  if (!ci)
    return false;
 
  /* To append we need an append-hook or a close-hook */
  if (!ci->cmd_append && !ci->cmd_close)
  {
    mutt_error(_("Can't append without an append-hook or close-hook : %s"),
               mailbox_path(m));
    goto cmoa_fail1;
  }
 
  if (setup_paths(m) != 0)
    goto cmoa_fail2;
 
  /* Lock the realpath for the duration of the append.
   * It will be unlocked in the close */
  if (!lock_realpath(m, true))
  {
    mutt_error(_("Unable to lock mailbox"));
    goto cmoa_fail2;
  }
 
  /* Open the existing mailbox, unless we are appending */
  if (!ci->cmd_append && (mutt_file_get_size(m->realpath) > 0))
  {
    int rc = execute_command(m, ci->cmd_open, _("Decompressing %s"));
    if (rc == 0)
    {
      mutt_error(_("Compress command failed: %s"), ci->cmd_open);
      goto cmoa_fail2;
    }
    m->type = mx_path_probe(mailbox_path(m));
  }
  else
  {
    m->type = cs_subset_enum(NeoMutt->sub, "mbox_type");
  }
 
  /* We can only deal with mbox and mmdf mailboxes */
  if ((m->type != MUTT_MBOX) && (m->type != MUTT_MMDF))
  {
    mutt_error(_("Unsupported mailbox type for appending"));
    goto cmoa_fail2;
  }
 
  ci->child_ops = mx_get_ops(m->type);
  if (!ci->child_ops)
  {
    mutt_error(_("Can't find mailbox ops for mailbox type %d"), m->type);
    goto cmoa_fail2;
  }
 
  if (!ci->child_ops->mbox_open_append(m, flags))
    goto cmoa_fail2;
 
  return true;
 
cmoa_fail2:
  /* remove the partial uncompressed file */
  (void) remove(mailbox_path(m));
cmoa_fail1:
  /* Free the compress_info to prevent close from trying to recompress */
  compress_info_free(m);
 
  return false;
}
 
static enum MxStatus comp_mbox_check(struct Mailbox *m)
{
  if (!m->compress_info)
    return MX_STATUS_ERROR;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return MX_STATUS_ERROR;
 
  int size = mutt_file_get_size(m->realpath);
  if (size == ci->size)
    return MX_STATUS_OK;
 
  if (!lock_realpath(m, false))
  {
    mutt_error(_("Unable to lock mailbox"));
    return MX_STATUS_ERROR;
  }
 
  int rc = execute_command(m, ci->cmd_open, _("Decompressing %s"));
  store_size(m);
  unlock_realpath(m);
  if (rc == 0)
    return MX_STATUS_ERROR;
 
  return ops->mbox_check(m);
}
 
static enum MxStatus comp_mbox_sync(struct Mailbox *m)
{
  if (!m->compress_info)
    return MX_STATUS_ERROR;
 
  struct CompressInfo *ci = m->compress_info;
 
  if (!ci->cmd_close)
  {
    mutt_error(_("Can't sync a compressed file without a close-hook"));
    return MX_STATUS_ERROR;
  }
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return MX_STATUS_ERROR;
 
  if (!lock_realpath(m, true))
  {
    mutt_error(_("Unable to lock mailbox"));
    return MX_STATUS_ERROR;
  }
 
  enum MxStatus check = comp_mbox_check(m);
  if (check != MX_STATUS_OK)
    goto sync_cleanup;
 
  check = ops->mbox_sync(m);
  if (check != MX_STATUS_OK)
    goto sync_cleanup;
 
  int rc = execute_command(m, ci->cmd_close, _("Compressing %s"));
  if (rc == 0)
  {
    check = MX_STATUS_ERROR;
    goto sync_cleanup;
  }
 
  check = MX_STATUS_OK;
 
sync_cleanup:
  store_size(m);
  unlock_realpath(m);
  return check;
}
 
static enum MxStatus comp_mbox_close(struct Mailbox *m)
{
  if (!m->compress_info)
    return MX_STATUS_ERROR;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
  {
    compress_info_free(m);
    return MX_STATUS_ERROR;
  }
 
  ops->mbox_close(m);
 
  /* sync has already been called, so we only need to delete some files */
  if (m->append)
  {
    const char *append = NULL;
    const char *msg = NULL;
 
    /* The file exists and we can append */
    if ((access(m->realpath, F_OK) == 0) && ci->cmd_append)
    {
      append = ci->cmd_append;
      msg = _("Compressed-appending to %s...");
    }
    else
    {
      append = ci->cmd_close;
      msg = _("Compressing %s");
    }
 
    int rc = execute_command(m, append, msg);
    if (rc == 0)
    {
      mutt_any_key_to_continue(NULL);
      mutt_error(_("Error. Preserving temporary file: %s"), mailbox_path(m));
    }
    else
      remove(mailbox_path(m));
 
    unlock_realpath(m);
  }
  else
  {
    /* If the file was removed, remove the compressed folder too */
    if (access(mailbox_path(m), F_OK) != 0)
    {
      const bool c_save_empty = cs_subset_bool(NeoMutt->sub, "save_empty");
      if (!c_save_empty)
      {
        remove(m->realpath);
      }
    }
    else
    {
      remove(mailbox_path(m));
    }
  }
 
  compress_info_free(m);
 
  return MX_STATUS_OK;
}
 
static bool comp_msg_open(struct Mailbox *m, struct Message *msg, int msgno)
{
  if (!m->compress_info)
    return false;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return false;
 
  /* Delegate */
  return ops->msg_open(m, msg, msgno);
}
 
static bool comp_msg_open_new(struct Mailbox *m, struct Message *msg, const struct Email *e)
{
  if (!m->compress_info)
    return false;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return false;
 
  /* Delegate */
  return ops->msg_open_new(m, msg, e);
}
 
static int comp_msg_commit(struct Mailbox *m, struct Message *msg)
{
  if (!m->compress_info)
    return -1;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return -1;
 
  /* Delegate */
  return ops->msg_commit(m, msg);
}
 
static int comp_msg_close(struct Mailbox *m, struct Message *msg)
{
  if (!m->compress_info)
    return -1;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops)
    return -1;
 
  /* Delegate */
  return ops->msg_close(m, msg);
}
 
static int comp_msg_padding_size(struct Mailbox *m)
{
  if (!m->compress_info)
    return 0;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops || !ops->msg_padding_size)
    return 0;
 
  return ops->msg_padding_size(m);
}
 
static int comp_msg_save_hcache(struct Mailbox *m, struct Email *e)
{
  if (!m->compress_info)
    return 0;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops || !ops->msg_save_hcache)
    return 0;
 
  return ops->msg_save_hcache(m, e);
}
 
static int comp_tags_edit(struct Mailbox *m, const char *tags, struct Buffer *buf)
{
  if (!m->compress_info)
    return 0;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops || !ops->tags_edit)
    return 0;
 
  return ops->tags_edit(m, tags, buf);
}
 
static int comp_tags_commit(struct Mailbox *m, struct Email *e, const char *buf)
{
  if (!m->compress_info)
    return 0;
 
  struct CompressInfo *ci = m->compress_info;
 
  const struct MxOps *ops = ci->child_ops;
  if (!ops || !ops->tags_commit)
    return 0;
 
  return ops->tags_commit(m, e, buf);
}
 
static enum MailboxType comp_path_probe(const char *path, const struct stat *st)
{
  if (!st || !S_ISREG(st->st_mode))
    return MUTT_UNKNOWN;
 
  if (mutt_comp_can_read(path))
    return MUTT_COMPRESSED;
 
  return MUTT_UNKNOWN;
}
 
static int comp_path_canon(char *buf, size_t buflen)
{
  mutt_path_canon(buf, buflen, HomeDir, false);
  return 0;
}
 
static int comp_path_pretty(char *buf, size_t buflen, const char *folder)
{
  if (mutt_path_abbr_folder(buf, folder))
    return 0;
 
  if (mutt_path_pretty(buf, buflen, HomeDir, false))
    return 0;
 
  return -1;
}
 
static int comp_path_parent(char *buf, size_t buflen)
{
  if (mutt_path_parent(buf))
    return 0;
 
  if (buf[0] == '~')
    mutt_path_canon(buf, buflen, HomeDir, false);
 
  if (mutt_path_parent(buf))
    return 0;
 
  return -1;
}
 
struct MxOps MxCompOps = {
  // clang-format off
  .type            = MUTT_COMPRESSED,
  .name             = "compressed",
  .is_local         = true,
  .ac_owns_path     = comp_ac_owns_path,
  .ac_add           = comp_ac_add,
  .mbox_open        = comp_mbox_open,
  .mbox_open_append = comp_mbox_open_append,
  .mbox_check       = comp_mbox_check,
  .mbox_check_stats = NULL,
  .mbox_sync        = comp_mbox_sync,
  .mbox_close       = comp_mbox_close,
  .msg_open         = comp_msg_open,
  .msg_open_new     = comp_msg_open_new,
  .msg_commit       = comp_msg_commit,
  .msg_close        = comp_msg_close,
  .msg_padding_size = comp_msg_padding_size,
  .msg_save_hcache  = comp_msg_save_hcache,
  .tags_edit        = comp_tags_edit,
  .tags_commit      = comp_tags_commit,
  .path_probe       = comp_path_probe,
  .path_canon       = comp_path_canon,
  .path_pretty      = comp_path_pretty,
  .path_parent      = comp_path_parent,
  .path_is_empty    = NULL,
  // clang-format on
};