lib/libc/stdio/local.h

   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)