/*-
 * Copyright (C) 2009 Gabor Kovesdan <gabor@FreeBSD.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/endian.h>

#include <err.h>
#include <errno.h>
#include <iconv.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>

/*
 * iconv_open must return (iconv_t)-1 on non-existing encoding
 * and set errno to EINVAL.
 */
static int
open_1(void)
{
        iconv_t cd;

        errno = 0;
        cd = iconv_open("nonexisting", "foobar");

        if ((cd == (iconv_t)-1) && (errno == EINVAL))
                return (0);
        else {
                iconv_close(cd);
                return (1);
        }
}

/*
 * iconv_open must return (iconv_t)-1 if too much files are open
 * and set errno to ENFILE.
 */
#define MAX_LIMIT       1025
static int
open_2(void)
{
        iconv_t cd[MAX_LIMIT];
        size_t i;
        int ret;

        errno = 0;
        ret = 1;
        for (i = 0; i < MAX_LIMIT; i++) {
                cd[i] = iconv_open("ASCII", "UTF8");
                if (cd[i] == (iconv_t)-1) {
                        if (errno == ENFILE || errno == EMFILE)
                                ret = 0;
                        cd[i] = NULL;
                        break;
                }
        }

        for (i = MIN(i, nitems(cd) - 1); i > 0; i--)
                iconv_close(cd[i]);
        return (ret);
}

/*
 * iconv_close must return (iconv_t)-1 if conversion descriptor is
 * invalid and set errno to EBADF.
 */
static int
close_1(void)
{
        iconv_t cd = (iconv_t)-1;

        return ((iconv_close(cd) == -1) && (errno = EBADF) ? 0 : 1);
}

static int
conv_ebadf(void)
{
        iconv_t cd = (iconv_t)-1;

        errno = 0;
        return ((iconv(cd, NULL, 0, NULL, 0) == (size_t)-1 && errno == EBADF) ? 0 : 1);
}

static int
conv_ret(void)
{
        iconv_t cd;
        size_t inbytesleft, outbytesleft;
        char *inptr;
        char *outptr;
        uint32_t outbuf[4];
        uint32_t inbuf[2] = { 0x00000151, 0x00000171 };

        if ((cd = iconv_open("ASCII", "UTF-32LE")) == (iconv_t)-1)
                return (1);

        inptr = (char *)inbuf;
        outptr = (char *)outbuf;
        inbytesleft = 8;
        outbytesleft = 16;

        return (iconv(cd, &inptr, &inbytesleft, &outptr, &outbytesleft) == 2 ? 0 : 1);
}

static int
conv_2big(void)
{
        iconv_t cd;
        size_t inbytesleft, outbytesleft;
        char *inptr;
        char *outptr;
        uint32_t inbuf[4];
        uint32_t outbuf[2];
        int ret;

        if ((cd = iconv_open("ASCII", "ASCII")) == (iconv_t)-1)
                return (1);

        inptr = (char *)inbuf;
        outptr = (char *)outbuf;
        inbytesleft = 16;
        outbytesleft = 8;

        errno = 0;
        ret = iconv(cd, &inptr, &inbytesleft, &outptr, &outbytesleft);

#ifdef VERBOSE
        printf("inptr - inbuf = %d\n", (const uint8_t *)inptr - (uint8_t *)inbuf);
        printf("inbytesleft = %d\n", inbytesleft);
        printf("outbytesleft = %d\n", outbytesleft);
        printf("outptr - outbuf = %d\n", (uint8_t *)outptr - (uint8_t *)outbuf);
        printf("errno = %d\n", errno);
        printf("ret = %d\n", (int)ret);
#endif

        if (((const uint8_t *)inptr - (uint8_t *)inbuf == 8) && (inbytesleft == 8)  &&
            (outbytesleft == 0) && ((uint8_t *)outptr - (uint8_t *)outbuf == 8) &&
            (errno == E2BIG) && ((size_t)ret == (size_t)-1))
                return (0);
        else
                return (1);
}

static int
conv_einval(void)
{
        iconv_t  cd;
        size_t inbytesleft, outbytesleft;
        char *inptr;
        char *outptr;
        uint32_t outbuf[4];
        uint16_t inbuf[1] = { 0xEA42 };
        int ret;

        if ((cd = iconv_open("UTF-32", "BIG5")) == (iconv_t)-1)
                return (1);

        inptr = (char *)inbuf;
        outptr = (char *)outbuf;
        inbytesleft = 2;
        outbytesleft = 16;

        errno = 0;
        ret = iconv(cd, &inptr, &inbytesleft, &outptr, &outbytesleft);

#ifdef VERBOSE
        printf("inptr - inbuf = %d\n", (const uint8_t *)inptr - (uint8_t *)inbuf);
        printf("inbytesleft = %d\n", inbytesleft);
        printf("outbytesleft = %d\n", outbytesleft);
        printf("outptr - outbuf = %d\n", (uint8_t *)outptr - (uint8_t *)outbuf);
        printf("errno = %d\n", errno);
        printf("ret = %d\n", (int)ret);
#endif

        if (((const uint8_t *)inptr - (uint8_t *)inbuf == 1) && (inbytesleft == 1)  &&
            (outbytesleft == 8) && ((uint8_t *)outptr - (uint8_t *)outbuf == 8) &&
            (errno == EINVAL) && ((size_t)ret == (size_t)-1))
                return (0);
        else
                return (1);
}

static int
conv_eilseq(void)
{
        iconv_t cd;
        size_t inbytesleft, outbytesleft;
        char *inptr;
        char *outptr;
        uint32_t outbuf[4];
        uint16_t inbuf[1] = { 0x8AC0 };
        int ret;

        if ((cd = iconv_open("Latin2", "UTF-16LE")) == (iconv_t)-1)
                return (1);

        inptr = (char *)inbuf;
        outptr = (char *)outbuf;
        inbytesleft = 4;
        outbytesleft = 16;

        errno = 0;
        ret = iconv(cd, &inptr, &inbytesleft, &outptr, &outbytesleft);

#ifdef VERBOSE
        printf("inptr - inbuf = %d\n", (const uint8_t *)inptr - (uint8_t *)inbuf);
        printf("inbytesleft = %d\n", inbytesleft);
        printf("outbytesleft = %d\n", outbytesleft);
        printf("outptr - outbuf = %d\n", (uint8_t *)outptr - (uint8_t *)outbuf);
        printf("errno = %d\n", errno);
        printf("ret = %d\n", (int)ret);
#endif

        if (((const uint8_t *)inptr - (uint8_t *)inbuf == 0) && (inbytesleft == 4)  &&
            (outbytesleft == 16) && ((uint8_t *)outptr - (uint8_t *)outbuf == 0) &&
            (errno == EILSEQ) && ((size_t)ret == (size_t)-1))
                return (0);
        else
                return (1);
}

static void
test(int (tester) (void), const char * label)
{
        int ret;

        if ((ret = tester()))
                printf("%s failed (%d)\n", label, ret);
        else
                printf("%s succeeded\n", label);
}

int
main(void)
{

        test(open_1, "open_1");
        test(open_2, "open_2");
        test(close_1, "close_1");
        test(conv_ret, "conv_ret");
        test(conv_ebadf, "conv_ebadf");
        test(conv_2big, "conv_2big");
        test(conv_einval, "conv_einval");
        test(conv_eilseq, "conv_eilseq");
}