1 /*
2 * Copyright 2011 Nexenta Systems, Inc. All rights reserved.
3 * Copyright (c) 2002-2004 Tim J. Robbins.
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28 #include "lint.h"
29 #include "mse_int.h"
30 #include "file64.h"
31 #include <errno.h>
32 #include <stdio.h>
33 #include <string.h>
34 #include <wchar.h>
35 #include "mblocal.h"
36 #include "mse.h"
37 #include "stdiom.h"
38 #include "libc.h"
39
40 /*
41 * FreeBSD implementation here included a full version that tried to be more
42 * efficient with memory strings. However, for simplicity's sake, we are
43 * going to just use fgetwc(). We also do the stream orientation thing for
44 * XPG5 if we need to.
45 */
46
47 wchar_t *
_fgetws_impl(wchar_t * _RESTRICT_KYWD ws,int n,FILE * _RESTRICT_KYWD fp,int orient)48 _fgetws_impl(wchar_t *_RESTRICT_KYWD ws, int n, FILE *_RESTRICT_KYWD fp,
49 int orient)
50 {
51 wint_t wc;
52 wchar_t *wsp;
53 rmutex_t *lk;
54
55 FLOCKFILE(lk, fp);
56 if (orient && GET_NO_MODE(fp))
57 _setorientation(fp, _WC_MODE);
58
59 if (n <= 0) {
60 errno = EINVAL;
61 FUNLOCKFILE(lk);
62 return (NULL);
63 }
64
65 wsp = ws;
66 while (--n) {
67 wc = _fgetwc_unlocked(fp);
68 if (wc == EOF) {
69 /*
70 * This can happen because of an EOF on
71 * the stream, or because of a decoding error.
72 * Its up to the caller to check errno.
73 */
74 if (wsp == ws) {
75 /* EOF with no data read */
76 FUNLOCKFILE(lk);
77 return (NULL);
78 }
79 break;
80 }
81 *wsp++ = wc;
82
83 if (wc == L'\n')
84 break;
85 }
86 *wsp = 0;
87 FUNLOCKFILE(lk);
88 return (ws);
89 }
90
91 wchar_t *
fgetws(wchar_t * _RESTRICT_KYWD ws,int n,FILE * _RESTRICT_KYWD fp)92 fgetws(wchar_t *_RESTRICT_KYWD ws, int n, FILE *_RESTRICT_KYWD fp)
93 {
94 return (_fgetws_impl(ws, n, fp, 0));
95 }
96
97 wchar_t *
__fgetws_xpg5(wchar_t * ws,int n,FILE * fp)98 __fgetws_xpg5(wchar_t *ws, int n, FILE *fp)
99 {
100 return (_fgetws_impl(ws, n, fp, 1));
101 }
102
103 wchar_t *
getws(wchar_t * ws)104 getws(wchar_t *ws)
105 {
106 wint_t wc;
107 wchar_t *wsp;
108 rmutex_t *lk;
109
110 FLOCKFILE(lk, stdin);
111
112 wsp = ws;
113 for (;;) {
114 wc = _fgetwc_unlocked(stdin);
115 if (wc == EOF) {
116 /*
117 * This can happen because of an EOF on
118 * the stream, or because of a decoding error.
119 * Its up to the caller to check errno.
120 */
121 if (wsp == ws) {
122 /* EOF with no data read */
123 FUNLOCKFILE(lk);
124 return (NULL);
125 }
126 break;
127 }
128 *wsp++ = wc;
129
130 if (wc == L'\n')
131 break;
132 }
133 *wsp = 0;
134 FUNLOCKFILE(lk);
135 return (ws);
136 }
137