fgetws.c (9199c09a159c4e3e98c212d4eec1edc5252d9e33) fgetws.c (e74101e4eff767325553039def89de70b70f36d3)
1/*-
1/*-
2 * Copyright (c) 2002-2004 Tim J. Robbins.
2 * Copyright (c) 2002 Tim J. Robbins.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright

--- 13 unchanged lines hidden (view full) ---

24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD$");
29
30#include "namespace.h"
31#include <errno.h>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright

--- 13 unchanged lines hidden (view full) ---

24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD$");
29
30#include "namespace.h"
31#include <errno.h>
32#include <rune.h>
32#include <stdio.h>
33#include <stdio.h>
33#include <string.h>
34#include <wchar.h>
35#include "un-namespace.h"
36#include "libc_private.h"
37#include "local.h"
34#include <wchar.h>
35#include "un-namespace.h"
36#include "libc_private.h"
37#include "local.h"
38#include "mblocal.h"
39
40wchar_t *
38
39wchar_t *
41fgetws(wchar_t * __restrict ws, int n, FILE * __restrict fp)
40fgetws(wchar_t *__restrict ws, int n, FILE *__restrict fp)
42{
43 wchar_t *wsp;
41{
42 wchar_t *wsp;
44 size_t nconv;
45 const char *src;
46 unsigned char *nl;
43 long r;
47
44
48 FLOCKFILE(fp);
49 ORIENT(fp, 1);
45 ORIENTLOCK(fp, 1);
50
46
51 if (n <= 0) {
52 errno = EINVAL;
53 goto error;
54 }
47 if (n <= 0)
48 return (NULL);
55
49
56 if (fp->_r <= 0 && __srefill(fp))
57 /* EOF */
58 goto error;
59 wsp = ws;
50 wsp = ws;
60 do {
61 src = fp->_p;
62 nl = memchr(fp->_p, '\n', fp->_r);
63 nconv = __mbsnrtowcs(wsp, &src,
64 nl != NULL ? (nl - fp->_p + 1) : fp->_r,
65 n - 1, &fp->_mbstate);
66 if (nconv == (size_t)-1)
67 /* Conversion error */
68 goto error;
69 if (src == NULL) {
70 /*
71 * We hit a null byte. Increment the character count,
72 * since mbsnrtowcs()'s return value doesn't include
73 * the terminating null, then resume conversion
74 * after the null.
75 */
76 nconv++;
77 src = memchr(fp->_p, '\0', fp->_r);
78 src++;
51 while (n-- > 1) {
52 /* XXX Inefficient */
53 if ((r = fgetrune(fp)) == _INVALID_RUNE) {
54 errno = EILSEQ;
55 return (NULL);
79 }
56 }
80 fp->_r -= (unsigned char *)src - fp->_p;
81 fp->_p = (unsigned char *)src;
82 n -= nconv;
83 wsp += nconv;
84 } while (wsp[-1] != L'\n' && n > 1 && (fp->_r > 0 ||
85 __srefill(fp) == 0));
86 if (wsp == ws)
87 /* EOF */
88 goto error;
89 if (!__mbsinit(&fp->_mbstate))
90 /* Incomplete character */
91 goto error;
92 *wsp = L'\0';
93 FUNLOCKFILE(fp);
57 if (r == EOF) {
58 if (wsp == ws)
59 /* EOF/error, no characters read yet. */
60 return (NULL);
61 break;
62 }
63 *wsp++ = (wchar_t)r;
64 if (r == L'\n')
65 break;
66 }
67 *wsp++ = L'\0';
94
95 return (ws);
68
69 return (ws);
96
97error:
98 FUNLOCKFILE(fp);
99 return (NULL);
100}
70}