/* Copyright (c) Pigeonhole authors, see top-level COPYING file */

#include "lib.h"

#include "mempool.h"

#include "array.h"

#include "settings.h"

#include "rfc2822.h"

#include "sieve-common.h"

#include "sieve-error.h"

#include "sieve-extensions.h"

#include "ext-editheader-limits.h"

#include "ext-editheader-settings.h"

#include "ext-editheader-common.h"

/*

 * Extension configuration

 */

struct ext_editheader_header {

  const char *name;

  bool forbid_add:1;

  bool forbid_delete:1;

};

struct ext_editheader_context {

  pool_t pool;

  const struct ext_editheader_settings *set;

  ARRAY(struct ext_editheader_header) headers;

  size_t max_header_size;

};

static const struct ext_editheader_header *

ext_editheader_header_find(struct ext_editheader_context *extctx,

         const char *hname)

{

  const struct ext_editheader_header *header;

  if (extctx == NULL)

    return NULL;

  array_foreach(&extctx->headers, header) {

    if (strcasecmp(hname, header->name) == 0)

      return header;

  }

  return NULL;

}

static int

ext_editheader_header_add(struct sieve_instance *svinst,

        struct ext_editheader_context *extctx,

        const char *hname)

{

  struct ext_editheader_header *header;

  const struct ext_editheader_header_settings *set;

  const char *error;

  if (settings_get_filter(svinst->event, "sieve_editheader_header", hname,

        &ext_editheader_header_setting_parser_info, 0,

        &set, &error) < 0) {

    e_error(svinst->event, "%s", error);

    return -1;

  }

  i_assert(ext_editheader_header_find(extctx, hname) == NULL);

  header = array_append_space(&extctx->headers);

  header->name = p_strdup(extctx->pool, hname);

  header->forbid_add = set->forbid_add;

  header->forbid_delete = set->forbid_delete;

  settings_free(set);

  return 0;

}

static int

ext_editheader_config_headers(struct sieve_instance *svinst,

            struct ext_editheader_context *extctx)

{

  const char *hname;

  if (!array_is_created(&extctx->set->headers))

    return 0;

  array_foreach_elem(&extctx->set->headers, hname) {

    if (ext_editheader_header_add(svinst, extctx, hname) < 0)

      return -1;

  }

  return 0;

}

int ext_editheader_load(const struct sieve_extension *ext, void **context_r)

{

  struct sieve_instance *svinst = ext->svinst;

  const struct ext_editheader_settings *set;

  struct ext_editheader_context *extctx;

  const char *error;

  pool_t pool;

  if (settings_get(svinst->event, &ext_editheader_setting_parser_info, 0,

       &set, &error) < 0) {

    e_error(svinst->event, "%s", error);

    return -1;

  }

  pool = pool_alloconly_create("editheader_config", 1024);

  extctx = p_new(pool, struct ext_editheader_context, 1);

  extctx->pool = pool;

  extctx->set = set;

  p_array_init(&extctx->headers, pool, 16);

  if (ext_editheader_config_headers(svinst, extctx) < 0) {

    settings_free(set);

    pool_unref(&pool);

    return -1;

  }

  *context_r = extctx;

  return 0;

}

void ext_editheader_unload(const struct sieve_extension *ext)

{

  struct ext_editheader_context *extctx = ext->context;

  if (extctx == NULL)

    return;

  settings_free(extctx->set);

  pool_unref(&extctx->pool);

}

/*

 * Protected headers

 */

bool ext_editheader_header_allow_add(const struct sieve_extension *ext,

             const char *hname)

{

  struct ext_editheader_context *extctx = ext->context;

  const struct ext_editheader_header *header;

  if (strcasecmp(hname, "subject") == 0)

    return TRUE;

  if (strcasecmp(hname, "x-sieve-redirected-from") == 0)

    return FALSE;

  header = ext_editheader_header_find(extctx, hname);

  if (header == NULL)

    return TRUE;

  return !header->forbid_add;

}

bool ext_editheader_header_allow_delete(const struct sieve_extension *ext,

          const char *hname)

{

  struct ext_editheader_context *extctx = ext->context;

  const struct ext_editheader_header *header;

  if (strcasecmp(hname, "received") == 0 ||

      strcasecmp(hname, "auto-submitted") == 0)

    return FALSE;

  if (strcasecmp(hname, "x-sieve-redirected-from") == 0)

    return FALSE;

  if (strcasecmp(hname, "subject") == 0)

    return TRUE;

  header = ext_editheader_header_find(extctx, hname);

  if (header == NULL)

    return TRUE;

  return !header->forbid_delete;

}

/*

 * Limits

 */

bool ext_editheader_header_too_large(const struct sieve_extension *ext,

             size_t size)

{

  struct ext_editheader_context *extctx = ext->context;

  if (extctx == NULL)

    return size > EXT_EDITHEADER_DEFAULT_MAX_HEADER_SIZE;

  return (size > extctx->set->max_header_size);

}