/* ----------------------------------------------------------------------- *
 *   
 *   Copyright 1996-2009 The NASM Authors - All Rights Reserved
 *   See the file AUTHORS included with the NASM distribution for
 *   the specific copyright holders.
 *
 *   Redistribution and use in source and binary forms, with or without
 *   modification, are permitted provided that the following
 *   conditions are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * 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 COPYRIGHT HOLDERS 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 COPYRIGHT OWNER 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.
 *
 * ----------------------------------------------------------------------- */

/*
 * strfunc.c
 *
 * String transformation functions
 */

#include "nasmlib.h"
#include "nasm.h"

/*
 * Convert a string in UTF-8 format to UTF-16LE
 */
static size_t utf8_to_16le(uint8_t *str, size_t len, char *op)
{
#define EMIT(x) do { if (op) { WRITESHORT(op,x); } outlen++; } while(0)

    size_t outlen = 0;
    int expect = 0;
    uint8_t c;
    uint32_t v = 0, vmin = 0;

    while (len--) {
	c = *str++;

	if (expect) {
	    if ((c & 0xc0) != 0x80) {
		expect = 0;
		return -1;
	    } else {
		v = (v << 6) | (c & 0x3f);
		if (!--expect) {
		    if (v < vmin || v > 0x10ffff ||
			(v >= 0xd800 && v <= 0xdfff)) {
			return -1;
		    } else if (v > 0xffff) {
			v -= 0x10000;
			EMIT(0xd800 | (v >> 10));
			EMIT(0xdc00 | (v & 0x3ff));
		    } else {
			EMIT(v);
		    }
		}
		continue;
	    }
	}

	if (c < 0x80) {
	    EMIT(c);
	} else if (c < 0xc0 || c >= 0xfe) {
	    /* Invalid UTF-8 */
	    return -1;
	} else if (c < 0xe0) {
	    v = c & 0x1f;
	    expect = 1;
	    vmin = 0x80;
	} else if (c < 0xf0) {
	    v = c & 0x0f;
	    expect = 2;
	    vmin = 0x800;
	} else if (c < 0xf8) {
	    v = c & 0x07;
	    expect = 3;
	    vmin = 0x10000;
	} else if (c < 0xfc) {
	    v = c & 0x03;
	    expect = 4;
	    vmin = 0x200000;
	} else {
	    v = c & 0x01;
	    expect = 5;
	    vmin = 0x4000000;
	}
    }

    return expect ? (size_t)-1 : outlen << 1;

#undef EMIT
}

/*
 * Convert a string in UTF-8 format to UTF-32LE
 */
static size_t utf8_to_32le(uint8_t *str, size_t len, char *op)
{
#define EMIT(x) do { if (op) { WRITELONG(op,x); } outlen++; } while(0)

    size_t outlen = 0;
    int expect = 0;
    uint8_t c;
    uint32_t v = 0, vmin = 0;

    while (len--) {
	c = *str++;

	if (expect) {
	    if ((c & 0xc0) != 0x80) {
		return -1;
	    } else {
		v = (v << 6) | (c & 0x3f);
		if (!--expect) {
		    if (v < vmin || (v >= 0xd800 && v <= 0xdfff)) {
			return -1;
		    } else {
			EMIT(v);
		    }
		}
		continue;
	    }
	}

	if (c < 0x80) {
	    EMIT(c);
	} else if (c < 0xc0 || c >= 0xfe) {
	    /* Invalid UTF-8 */
	    return -1;
	} else if (c < 0xe0) {
	    v = c & 0x1f;
	    expect = 1;
	    vmin = 0x80;
	} else if (c < 0xf0) {
	    v = c & 0x0f;
	    expect = 2;
	    vmin = 0x800;
	} else if (c < 0xf8) {
	    v = c & 0x07;
	    expect = 3;
	    vmin = 0x10000;
	} else if (c < 0xfc) {
	    v = c & 0x03;
	    expect = 4;
	    vmin = 0x200000;
	} else {
	    v = c & 0x01;
	    expect = 5;
	    vmin = 0x4000000;
	}
    }

    return expect ? (size_t)-1 : outlen << 2;

#undef EMIT
}

typedef size_t (*transform_func)(uint8_t *, size_t, char *);

/*
 * Apply a specific string transform and return it in a nasm_malloc'd
 * buffer, returning the length.  On error, returns (size_t)-1 and no
 * buffer is allocated.
 */
size_t string_transform(char *str, size_t len, char **out, enum strfunc func)
{
    /* This should match enum strfunc in nasm.h */
    static const transform_func str_transforms[] = {
	utf8_to_16le,
	utf8_to_32le,
    };
    transform_func transform = str_transforms[func];
    size_t outlen;
    uint8_t *s = (uint8_t *)str;
    char *buf;

    outlen = transform(s, len, NULL);
    if (outlen == (size_t)-1)
	return -1;

    *out = buf = nasm_malloc(outlen+1);
    buf[outlen] = '\0';		/* Forcibly null-terminate the buffer */
    return transform(s, len, buf);
}