xref: /dragonfly/lib/libc/stdio/local.h (revision 3ea159d2)
1 /*-
2  * Copyright (c) 1990, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Chris Torek.
7  *
8  * Copyright (c) 2011 The FreeBSD Foundation
9  * All rights reserved.
10  * Portions of this software were developed by David Chisnall
11  * under sponsorship from the FreeBSD Foundation.
12  *
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 3. Neither the name of the University nor the names of its contributors
22  *    may be used to endorse or promote products derived from this software
23  *    without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35  * SUCH DAMAGE.
36  *
37  *	@(#)local.h	8.3 (Berkeley) 7/3/94
38  * $FreeBSD: head/lib/libc/stdio/local.h 234799 2012-04-29 16:28:39Z das $
39  */
40 
41 #include <sys/types.h>	/* for off_t */
42 #include <string.h>
43 #include <wchar.h>
44 #include <locale.h>
45 #include <xlocale_private.h>
46 
47 #include "priv_stdio.h"
48 #include "wcio.h"
49 
50 /*
51  * Information local to this implementation of stdio,
52  * in particular, macros and private variables.
53  */
54 
55 extern int	_sread(FILE *, char *, int);
56 extern int	_swrite(FILE *, char const *, int);
57 extern fpos_t	_sseek(FILE *, fpos_t, int);
58 extern int	_ftello(FILE *, fpos_t *);
59 extern int	_fseeko(FILE *, off_t, int, int);
60 extern int	__fflush(FILE *fp);
61 extern void	__fcloseall(void);
62 extern wint_t	__fgetwc_mbs(FILE *, mbstate_t *, int *, locale_t);
63 extern wint_t	__fputwc(wchar_t, FILE *, locale_t);
64 extern int	__sflush(FILE *);
65 extern FILE	*__sfp(void);
66 extern int	__slbexpand(FILE *, size_t);
67 extern int	__srefill(FILE *);
68 extern int	__sread(void *, char *, int);
69 extern int	__swrite(void *, char const *, int);
70 extern fpos_t	__sseek(void *, fpos_t, int);
71 extern int	__sclose(void *);
72 extern void	__sinit(void);
73 extern void	_cleanup(void);
74 extern void	__smakebuf(FILE *);
75 extern int	__swhatbuf(FILE *, size_t *, int *);
76 extern int	_fwalk(int (*)(FILE *));
77 extern int	__svfscanf(FILE *, locale_t, const char *, __va_list);
78 extern int	__swsetup(FILE *);
79 extern int	__sflags(const char *, int *);
80 extern int	__ungetc(int, FILE *);
81 extern wint_t	__ungetwc(wint_t, FILE *, locale_t);
82 extern int	__vfprintf(FILE *, locale_t, const char *, __va_list);
83 extern int	__vfscanf(FILE *, const char *, __va_list);
84 extern int	__vfwprintf(FILE *, locale_t, const wchar_t *, __va_list);
85 extern int	__vfwscanf(FILE * __restrict, locale_t, const wchar_t * __restrict,
86 		    __va_list);
87 extern size_t	__fread(void * __restrict buf, size_t size, size_t count,
88 		FILE * __restrict fp);
89 extern int	__sdidinit;
90 
91 static inline wint_t
92 __fgetwc(FILE *fp, locale_t locale)
93 {
94 	int nread;
95 	struct wchar_io_data *wcio;
96 	mbstate_t *st;
97 
98 	_SET_ORIENTATION(fp,1);
99 	wcio = WCIO_GET(fp);
100 
101 	if (wcio == NULL) {
102 		return WEOF;
103 	}
104 	wcio->wcio_ungetwc_inbuf = 0;
105 	st = &wcio->wcio_mbstate_out;
106 
107 	return (__fgetwc_mbs(fp, st, &nread, locale));
108 }
109 
110 /*
111  * Prepare the given FILE for writing, and return 0 iff it
112  * can be written now.  Otherwise, return EOF and set errno.
113  */
114 #define	prepwrite(fp) \
115 	((((fp)->pub._flags & __SWR) == 0 || \
116 	    ((fp)->_bf._base == NULL && ((fp)->pub._flags & __SSTR) == 0)) && \
117 	 __swsetup(fp))
118 
119 /*
120  * Test whether the given stdio file has an active ungetc buffer;
121  * release such a buffer, without restoring ordinary unread data.
122  */
123 #define	HASUB(fp) ((fp)->_ub._base != NULL)
124 #define	FREEUB(fp) { \
125 	if ((fp)->_ub._base != (fp)->_ubuf) \
126 		free((char *)(fp)->_ub._base); \
127 	(fp)->_ub._base = NULL; \
128 }
129 
130 /*
131  * test for an fgetln() buffer.
132  */
133 #define	HASLB(fp) ((fp)->_lb._base != NULL)
134 #define	FREELB(fp) { \
135 	free((char *)(fp)->_lb._base); \
136 	(fp)->_lb._base = NULL; \
137 }
138 
139 /*
140  * Structure initializations for 'fake' FILE objects.
141  */
142 #define	FAKE_FILE {				\
143 	.pub._lbfsize = 0,			\
144 	.pub._fileno = -1,			\
145 	._fl_mutex = PTHREAD_MUTEX_INITIALIZER,	\
146 }
147 
148 /*
149  * Set the orientation for a stream. If o > 0, the stream has wide-
150  * orientation. If o < 0, the stream has byte-orientation.
151  */
152 #define	ORIENT(fp, o)	_SET_ORIENTATION(fp, o)
153