4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore/*
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore * Copyright 2013 Garrett D'Amore <garrett@damore.org>
6b5e5868e7ebf1aff3a5abd7d0c4ef0e5fbf3648Garrett D'Amore * Copyright 2010 Nexenta Systems, Inc. All rights reserved.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * Copyright (c) 2002-2004 Tim J. Robbins.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * All rights reserved.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * Redistribution and use in source and binary forms, with or without
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * modification, are permitted provided that the following conditions
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * are met:
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * 1. Redistributions of source code must retain the above copyright
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * notice, this list of conditions and the following disclaimer.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * 2. Redistributions in binary form must reproduce the above copyright
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * notice, this list of conditions and the following disclaimer in the
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * documentation and/or other materials provided with the distribution.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore * SUCH DAMAGE.
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore */
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include "lint.h"
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include <errno.h>
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include <limits.h>
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include <stdlib.h>
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include <wchar.h>
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore#include "mblocal.h"
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore#include "localeimpl.h"
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore#include "lctype.h"
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amoresize_t
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amorembsnrtowcs_l(wchar_t *_RESTRICT_KYWD dst, const char **_RESTRICT_KYWD src,
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore size_t nms, size_t len, mbstate_t *_RESTRICT_KYWD ps, locale_t loc)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore{
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore static mbstate_t mbs;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore if (ps == NULL)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore ps = &mbs;
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore return (loc->ctype->lc_mbsnrtowcs(dst, src, nms, len, ps));
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore}
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amoresize_t
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amorembsnrtowcs(wchar_t *_RESTRICT_KYWD dst, const char **_RESTRICT_KYWD src,
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore size_t nms, size_t len, mbstate_t *_RESTRICT_KYWD ps)
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore{
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore return (mbsnrtowcs_l(dst, src, nms, len, ps, uselocale(NULL)));
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore}
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amoresize_t
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore__mbsnrtowcs_std(wchar_t *_RESTRICT_KYWD dst, const char **_RESTRICT_KYWD src,
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore size_t nms, size_t len, mbstate_t *_RESTRICT_KYWD ps,
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore mbrtowc_pfn_t pmbrtowc)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore{
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore const char *s;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore size_t nchr;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore wchar_t wc;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore size_t nb;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore s = *src;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore nchr = 0;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore if (dst == NULL) {
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore for (;;) {
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore if ((nb = pmbrtowc(&wc, s, nms, ps)) == (size_t)-1)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore /* Invalid sequence - mbrtowc() sets errno. */
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return ((size_t)-1);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore else if (nb == 0 || nb == (size_t)-2)
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return (nchr);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore s += nb;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore nms -= nb;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore nchr++;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore }
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore /*NOTREACHED*/
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore }
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore while (len-- > 0) {
2d08521bd15501c8370ba2153b9cca4f094979d0Garrett D'Amore if ((nb = pmbrtowc(dst, s, nms, ps)) == (size_t)-1) {
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *src = s;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return ((size_t)-1);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore } else if (nb == (size_t)-2) {
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *src = s + nms;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return (nchr);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore } else if (nb == 0) {
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *src = NULL;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return (nchr);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore }
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore s += nb;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore nms -= nb;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore nchr++;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore dst++;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore }
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore *src = s;
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore return (nchr);
4297a3b0d0a35d80f86fff155e288e885a100e6dGarrett D'Amore}