mbsnrtowcs.inc 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. /*-
  2. * As noted in the source, some portions of this implementation are copied from
  3. * FreeBSD libc. These are covered by the following copyright:
  4. *
  5. * Copyright (c) 2002-2004 Tim J. Robbins.
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions
  9. * are met:
  10. * 1. Redistributions of source code must retain the above copyright
  11. * notice, this list of conditions and the following disclaimer.
  12. * 2. Redistributions in binary form must reproduce the above copyright
  13. * notice, this list of conditions and the following disclaimer in the
  14. * documentation and/or other materials provided with the distribution.
  15. *
  16. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19. * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26. * SUCH DAMAGE.
  27. */
  28. size_t
  29. mbsnrtowcs_l(wchar_t * __restrict dst, const char ** __restrict src,
  30. size_t nms, size_t len, mbstate_t * __restrict ps, locale_t loc)
  31. {
  32. const char *s;
  33. size_t nchr;
  34. wchar_t wc;
  35. size_t nb;
  36. FIX_LOCALE(loc);
  37. s = *src;
  38. nchr = 0;
  39. if (dst == NULL) {
  40. for (;;) {
  41. if ((nb = mbrtowc_l(&wc, s, nms, ps, loc)) == (size_t)-1)
  42. /* Invalid sequence - mbrtowc() sets errno. */
  43. return ((size_t)-1);
  44. else if (nb == 0 || nb == (size_t)-2)
  45. return (nchr);
  46. s += nb;
  47. nms -= nb;
  48. nchr++;
  49. }
  50. /*NOTREACHED*/
  51. }
  52. while (len-- > 0) {
  53. if ((nb = mbrtowc_l(dst, s, nms, ps, loc)) == (size_t)-1) {
  54. *src = s;
  55. return ((size_t)-1);
  56. } else if (nb == (size_t)-2) {
  57. *src = s + nms;
  58. return (nchr);
  59. } else if (nb == 0) {
  60. *src = NULL;
  61. return (nchr);
  62. }
  63. s += nb;
  64. nms -= nb;
  65. nchr++;
  66. dst++;
  67. }
  68. *src = s;
  69. return (nchr);
  70. }