/* Copyright (c) 2002-2018 Dovecot authors, see the included COPYING file */

#include "lib.h"

#include "buffer.h"

#include "str.h"

#include "charset-utf8-private.h"

#include <ctype.h>

const struct charset_utf8_vfuncs *charset_utf8_vfuncs = &charset_iconv;

bool charset_is_utf8(const char *charset)

{

  return strcasecmp(charset, "us-ascii") == 0 ||

    strcasecmp(charset, "ascii") == 0 ||

    strcasecmp(charset, "UTF-8") == 0 ||

    strcasecmp(charset, "UTF8") == 0;

}

int charset_to_utf8_str(const char *charset, normalizer_func_t *normalizer,

      const char *input, string_t *output,

      enum charset_result *result_r)

{

  struct charset_translation *t;

  size_t len = strlen(input);

  if (charset_to_utf8_begin(charset, normalizer, &t) < 0)

    return -1;

  *result_r = charset_to_utf8(t, (const unsigned char *)input,

            &len, output);

  charset_to_utf8_end(&t);

  return 0;

}

struct charset_translation *

charset_utf8_to_utf8_begin(normalizer_func_t *normalizer)

{

  struct charset_translation *trans;

  if (charset_to_utf8_begin("UTF-8", normalizer, &trans) < 0)

    i_unreached();

  return trans;

}

enum charset_result

charset_utf8_to_utf8(normalizer_func_t *normalizer,

         const unsigned char *src, size_t *src_size, buffer_t *dest)

{

  enum charset_result res = CHARSET_RET_OK;

  size_t pos;

  uni_utf8_partial_strlen_n(src, *src_size, &pos);

  if (pos < *src_size) {

    i_assert(*src_size - pos <= CHARSET_MAX_PENDING_BUF_SIZE);

    *src_size = pos;

    res = CHARSET_RET_INCOMPLETE_INPUT;

  }

  if (normalizer != NULL) {

    if (normalizer(src, *src_size, dest) < 0)

      return CHARSET_RET_INVALID_INPUT;

  } else if (!uni_utf8_get_valid_data(src, *src_size, dest)) {

    return CHARSET_RET_INVALID_INPUT;

  } else {

    buffer_append(dest, src, *src_size);

  }

  return res;

}

int charset_to_utf8_begin(const char *charset, normalizer_func_t *normalizer,

        struct charset_translation **t_r)

{

  return charset_utf8_vfuncs->to_utf8_begin(charset, normalizer, t_r);

}

void charset_to_utf8_end(struct charset_translation **_t)

{

  struct charset_translation *t = *_t;

  *_t = NULL;

  charset_utf8_vfuncs->to_utf8_end(t);

}

void charset_to_utf8_reset(struct charset_translation *t)

{

  charset_utf8_vfuncs->to_utf8_reset(t);

}

enum charset_result

charset_to_utf8(struct charset_translation *t,

    const unsigned char *src, size_t *src_size, buffer_t *dest)

{

  return charset_utf8_vfuncs->to_utf8(t, src, src_size, dest);

}