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 * 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 * 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 * 98 __fgetws_xpg5(wchar_t *ws, int n, FILE *fp) 99 { 100 return (_fgetws_impl(ws, n, fp, 1)); 101 } 102 103 wchar_t * 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