1e74101e4STim J. Robbins /*- 27d7287dcSTim J. Robbins * Copyright (c) 2002-2004 Tim J. Robbins. 3e74101e4STim J. Robbins * All rights reserved. 4e74101e4STim J. Robbins * 5e74101e4STim J. Robbins * Redistribution and use in source and binary forms, with or without 6e74101e4STim J. Robbins * modification, are permitted provided that the following conditions 7e74101e4STim J. Robbins * are met: 8e74101e4STim J. Robbins * 1. Redistributions of source code must retain the above copyright 9e74101e4STim J. Robbins * notice, this list of conditions and the following disclaimer. 10e74101e4STim J. Robbins * 2. Redistributions in binary form must reproduce the above copyright 11e74101e4STim J. Robbins * notice, this list of conditions and the following disclaimer in the 12e74101e4STim J. Robbins * documentation and/or other materials provided with the distribution. 13e74101e4STim J. Robbins * 14e74101e4STim J. Robbins * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15e74101e4STim J. Robbins * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16e74101e4STim J. Robbins * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17e74101e4STim J. Robbins * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18e74101e4STim J. Robbins * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19e74101e4STim J. Robbins * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20e74101e4STim J. Robbins * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21e74101e4STim J. Robbins * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22e74101e4STim J. Robbins * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23e74101e4STim J. Robbins * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24e74101e4STim J. Robbins * SUCH DAMAGE. 25e74101e4STim J. Robbins */ 26e74101e4STim J. Robbins 27e74101e4STim J. Robbins #include <sys/cdefs.h> 28e74101e4STim J. Robbins __FBSDID("$FreeBSD$"); 29e74101e4STim J. Robbins 30e74101e4STim J. Robbins #include "namespace.h" 31e74101e4STim J. Robbins #include <errno.h> 32e74101e4STim J. Robbins #include <stdio.h> 337d7287dcSTim J. Robbins #include <string.h> 34e74101e4STim J. Robbins #include <wchar.h> 35e74101e4STim J. Robbins #include "un-namespace.h" 36e74101e4STim J. Robbins #include "libc_private.h" 37e74101e4STim J. Robbins #include "local.h" 387d7287dcSTim J. Robbins #include "mblocal.h" 39e74101e4STim J. Robbins 40e74101e4STim J. Robbins wchar_t * 41e74101e4STim J. Robbins fgetws(wchar_t * __restrict ws, int n, FILE * __restrict fp) 42e74101e4STim J. Robbins { 43e74101e4STim J. Robbins wchar_t *wsp; 447d7287dcSTim J. Robbins size_t nconv; 457d7287dcSTim J. Robbins const char *src; 467d7287dcSTim J. Robbins unsigned char *nl; 47e74101e4STim J. Robbins 487591ae56STim J. Robbins FLOCKFILE(fp); 497591ae56STim J. Robbins ORIENT(fp, 1); 50e74101e4STim J. Robbins 517591ae56STim J. Robbins if (n <= 0) { 527591ae56STim J. Robbins errno = EINVAL; 537591ae56STim J. Robbins goto error; 547591ae56STim J. Robbins } 55e74101e4STim J. Robbins 567d7287dcSTim J. Robbins if (fp->_r <= 0 && __srefill(fp)) 577d7287dcSTim J. Robbins /* EOF */ 587d7287dcSTim J. Robbins goto error; 59e74101e4STim J. Robbins wsp = ws; 607d7287dcSTim J. Robbins do { 617d7287dcSTim J. Robbins src = fp->_p; 627d7287dcSTim J. Robbins nl = memchr(fp->_p, '\n', fp->_r); 637d7287dcSTim J. Robbins nconv = __mbsnrtowcs(wsp, &src, 647d7287dcSTim J. Robbins nl != NULL ? (nl - fp->_p + 1) : fp->_r, 657d7287dcSTim J. Robbins n - 1, &fp->_extra->mbstate); 667d7287dcSTim J. Robbins if (nconv == (size_t)-1) 677d7287dcSTim J. Robbins /* Conversion error */ 687591ae56STim J. Robbins goto error; 697d7287dcSTim J. Robbins if (src == NULL) { 707d7287dcSTim J. Robbins /* 717d7287dcSTim J. Robbins * We hit a null byte. Increment the character count, 727d7287dcSTim J. Robbins * since mbsnrtowcs()'s return value doesn't include 737d7287dcSTim J. Robbins * the terminating null, then resume conversion 747d7287dcSTim J. Robbins * after the null. 757d7287dcSTim J. Robbins */ 767d7287dcSTim J. Robbins nconv++; 774c86f66fSStefan Farfeleder src = memchr(fp->_p, '\0', fp->_r); 784c86f66fSStefan Farfeleder src++; 797d7287dcSTim J. Robbins } 807d7287dcSTim J. Robbins fp->_r -= (unsigned char *)src - fp->_p; 817d7287dcSTim J. Robbins fp->_p = (unsigned char *)src; 827d7287dcSTim J. Robbins n -= nconv; 837d7287dcSTim J. Robbins wsp += nconv; 847d7287dcSTim J. Robbins } while (wsp[-1] != L'\n' && n > 1 && (fp->_r > 0 || 857d7287dcSTim J. Robbins __srefill(fp) == 0)); 86e74101e4STim J. Robbins if (wsp == ws) 877d7287dcSTim J. Robbins /* EOF */ 887591ae56STim J. Robbins goto error; 897d7287dcSTim J. Robbins if (!__mbsinit(&fp->_extra->mbstate)) 907d7287dcSTim J. Robbins /* Incomplete character */ 917d7287dcSTim J. Robbins goto error; 92e74101e4STim J. Robbins *wsp++ = L'\0'; 937591ae56STim J. Robbins FUNLOCKFILE(fp); 94e74101e4STim J. Robbins 95e74101e4STim J. Robbins return (ws); 967591ae56STim J. Robbins 977591ae56STim J. Robbins error: 987591ae56STim J. Robbins FUNLOCKFILE(fp); 997591ae56STim J. Robbins return (NULL); 100e74101e4STim J. Robbins } 101