xref: /illumos-gate/usr/src/lib/libc/port/locale/fgetws.c (revision 45bed668)
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