lib/libc/locale/ansi.c

   1 /*-
   2  * Copyright (c) 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * This code is derived from software contributed to Berkeley by
   6  * Paul Borman at Krystal Technologies.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  * 3. All advertising materials mentioning features or use of this software
  17  *    must display the following acknowledgement:
  18  *      This product includes software developed by the University of
  19  *      California, Berkeley and its contributors.
  20  * 4. Neither the name of the University nor the names of its contributors
  21  *    may be used to endorse or promote products derived from this software
  22  *    without specific prior written permission.
  23  *
  24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34  * SUCH DAMAGE.
  35  *
  36  * @(#)ansi.c   8.1 (Berkeley) 6/27/93
  37  * $FreeBSD: src/lib/libc/locale/ansi.c,v 1.3.6.1 2003/04/05 08:28:24 tjr Exp $
  38  * $DragonFly: src/lib/libc/locale/Attic/ansi.c,v 1.2 2003/06/17 04:26:43 dillon Exp $
  39  */
  40
  41 #include <errno.h>
  42 #include <stdlib.h>
  43 #include <limits.h>
  44 #include <stddef.h>
  45 #include <rune.h>
  46
  47 int
  48 mblen(s, n)
  49         const char *s;
  50         size_t n;
  51 {
  52         char const *e;
  53
  54         if (s == 0 || *s == 0)
  55                 return (0);     /* No support for state dependent encodings. */
  56
  57         if (sgetrune(s, n, &e) == _INVALID_RUNE)
  58                 return (s - e);
  59         return (e - s);
  60 }
  61
  62 int
  63 mbtowc(pwc, s, n)
  64         wchar_t *pwc;
  65         const char *s;
  66         size_t n;
  67 {
  68         char const *e;
  69         rune_t r;
  70
  71         if (s == 0 || *s == 0)
  72                 return (0);     /* No support for state dependent encodings. */
  73
  74         if ((r = sgetrune(s, n, &e)) == _INVALID_RUNE)
  75                 return (s - e);
  76         if (pwc)
  77                 *pwc = r;
  78         return (e - s);
  79 }
  80
  81 int
  82 wctomb(s, wchar)
  83         char *s;
  84         wchar_t wchar;
  85 {
  86         char *e;
  87
  88         if (s == 0)
  89                 return (0);     /* No support for state dependent encodings. */
  90
  91         if (wchar == 0) {
  92                 *s = 0;
  93                 return (1);
  94         }
  95
  96         sputrune(wchar, s, MB_CUR_MAX, &e);
  97         return (e ? e - s : -1);
  98 }
  99
 100 size_t
 101 mbstowcs(pwcs, s, n)
 102         wchar_t *pwcs;
 103         const char *s;
 104         size_t n;
 105 {
 106         const char *e;
 107         int cnt;
 108         rune_t r;
 109
 110         if (s == NULL) {
 111                 errno = EINVAL;
 112                 return (-1);
 113         }
 114
 115         if (pwcs == NULL) {
 116                 /* Convert and count only, do not store. */
 117                 cnt = 0;
 118                 while ((r = sgetrune(s, MB_LEN_MAX, &e)) != _INVALID_RUNE &&
 119                     r != 0) {
 120                         s = e;
 121                         cnt++;
 122                 }
 123                 if (r == _INVALID_RUNE) {
 124                         errno = EILSEQ;
 125                         return (-1);
 126                 }
 127                 return (cnt);
 128         }
 129
 130         /* Convert, store and count characters. */
 131         cnt = 0;
 132         while (n-- > 0) {
 133                 *pwcs = sgetrune(s, MB_LEN_MAX, &e);
 134                 if (*pwcs == _INVALID_RUNE) {
 135                         errno = EILSEQ;
 136                         return (-1);
 137                 }
 138                 if (*pwcs++ == L'\0')
 139                         break;
 140                 s = e;
 141                 ++cnt;
 142         }
 143
 144         return (cnt);
 145 }
 146
 147 size_t
 148 wcstombs(s, pwcs, n)
 149         char *s;
 150         const wchar_t *pwcs;
 151         size_t n;
 152 {
 153         char buf[MB_LEN_MAX];
 154         char *e;
 155         int cnt, nb;
 156
 157         if (pwcs == NULL || n > INT_MAX) {
 158                 errno = EINVAL;
 159                 return (-1);
 160         }
 161
 162         cnt = 0;
 163
 164         if (s == NULL) {
 165                 /* Convert and count only, do not store. */
 166                 while (*pwcs != L'\0') {
 167                         if (sputrune(*pwcs++, buf, MB_LEN_MAX, &e) == 0) {
 168                                 errno = EILSEQ;
 169                                 return (-1);
 170                         }
 171                         cnt += e - buf;
 172                 }
 173                 return (cnt);
 174         }
 175
 176         /* Convert, store and count characters. */
 177         nb = n;
 178         while (nb > 0) {
 179                 if (*pwcs == L'\0') {
 180                         *s = '\0';
 181                         break;
 182                 }
 183                 if (sputrune(*pwcs++, s, nb, &e) == 0) {
 184                         errno = EILSEQ;
 185                         return (-1);
 186                 }
 187                 if (e == NULL)          /* too long */
 188                         return (cnt);
 189                 cnt += e - s;
 190                 nb -= e - s;
 191                 s = e;
 192         }
 193
 194         return (cnt);
 195 }