1/* $NetBSD: fgetwln.c,v 1.8 2016/08/27 13:15:48 christos Exp $ */ 2 3/*- 4 * Copyright (c) 2002-2004 Tim J. Robbins. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26 * SUCH DAMAGE. 27 */ 28 29#include <sys/cdefs.h> 30#if defined(LIBC_SCCS) && !defined(lint) 31#if 0 32__FBSDID("$FreeBSD: src/lib/libc/stdio/fgetwln.c,v 1.2 2004/08/06 17:00:09 tjr Exp $"); 33#else 34__RCSID("$NetBSD: fgetwln.c,v 1.8 2016/08/27 13:15:48 christos Exp $"); 35#endif 36#endif /* LIBC_SCCS and not lint */ 37 38#include "namespace.h" 39#include <assert.h> 40#include <errno.h> 41#include <limits.h> 42#include <stdio.h> 43#include <stdlib.h> 44#include <wchar.h> 45#include "reentrant.h" 46#include "local.h" 47 48#ifdef __weak_alias 49__weak_alias(fgetwln,_fgetwln) 50#endif 51 52/* 53 * Expand the line buffer. Return -1 on error. 54#ifdef notdef 55 * The `new size' does not account for a terminating '\0', 56 * so we add 1 here. 57#endif 58 */ 59static int 60__slbexpand(FILE *fp, size_t newsize) 61{ 62 void *p; 63 64#ifdef notdef 65 ++newsize; 66#endif 67 _DIAGASSERT(fp != NULL); 68 69 if (_EXT(fp)->_fgetstr_len >= newsize) 70 return 0; 71 if ((p = realloc(_EXT(fp)->_fgetstr_buf, newsize)) == NULL) 72 return -1; 73 _EXT(fp)->_fgetstr_buf = p; 74 _EXT(fp)->_fgetstr_len = newsize; 75 return 0; 76} 77 78wchar_t * 79fgetwln(FILE * __restrict fp, size_t *lenp) 80{ 81 wint_t wc; 82 size_t len; 83 84 FLOCKFILE(fp); 85 _SET_ORIENTATION(fp, 1); 86 87 len = 0; 88 while ((wc = __fgetwc_unlock(fp)) != WEOF) { 89#define GROW 512 90 if (len * sizeof(wchar_t) >= _EXT(fp)->_fgetstr_len && 91 __slbexpand(fp, (len + GROW) * sizeof(wchar_t))) { 92 fp->_flags |= __SERR; 93 goto error; 94 } 95 *((wchar_t *)(void *)_EXT(fp)->_fgetstr_buf + len++) = wc; 96 if (wc == L'\n') 97 break; 98 } 99 100 /* 101 * The following test assumes that fgetwc() fails when 102 * feof() is already set, and that fgetwc() will never 103 * set feof() in the same call where it also sets ferror() 104 * or returns non-WEOF. 105 * Testing ferror() would not be better because fgetwc() 106 * may succeed even when ferror() is already set. 107 */ 108 109 if (len == 0 || (wc == WEOF && !__sfeof(fp))) 110 goto error; 111 112 FUNLOCKFILE(fp); 113 *lenp = len; 114 return (wchar_t *)(void *)_EXT(fp)->_fgetstr_buf; 115 116error: 117 FUNLOCKFILE(fp); 118 *lenp = 0; 119 return NULL; 120} 121