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 <pthread.h>
  43 #include <string.h>
  44 #include <wchar.h>
  45 #include <locale.h>
  46 #include <xlocale_private.h>
  47
  48 #include "priv_stdio.h"
  49 #include "wcio.h"
  50
  51 /*
  52  * Information local to this implementation of stdio,
  53  * in particular, macros and private variables.
  54  */
  55
  56 extern int      _sread(FILE *, char *, int);
  57 extern int      _swrite(FILE *, char const *, int);
  58 extern fpos_t   _sseek(FILE *, fpos_t, int);
  59 extern int      _ftello(FILE *, fpos_t *);
  60 extern int      _fseeko(FILE *, off_t, int, int);
  61 extern int      __fflush(FILE *fp);
  62 extern void     __fcloseall(void);
  63 extern wint_t   __fgetwc_mbs(FILE *, mbstate_t *, int *, locale_t);
  64 extern wint_t   __fputwc(wchar_t, FILE *, locale_t);
  65 extern int      __sflush(FILE *);
  66 extern FILE     *__sfp(void);
  67 extern int      __slbexpand(FILE *, size_t);
  68 extern int      __srefill(FILE *);
  69 extern int      __sread(void *, char *, int);
  70 extern int      __swrite(void *, char const *, int);
  71 extern fpos_t   __sseek(void *, fpos_t, int);
  72 extern int      __sclose(void *);
  73 extern void     __sinit(void);
  74 extern void     _cleanup(void);
  75 extern void     __smakebuf(FILE *);
  76 extern int      __swhatbuf(FILE *, size_t *, int *);
  77 extern int      _fwalk(int (*)(FILE *));
  78 extern int      __svfscanf(FILE *, locale_t, const char *, __va_list);
  79 extern int      __swsetup(FILE *);
  80 extern int      __sflags(const char *, int *);
  81 extern int      __ungetc(int, FILE *);
  82 extern wint_t   __ungetwc(wint_t, FILE *, locale_t);
  83 extern int      __vfprintf(FILE *, locale_t, const char *, __va_list);
  84 extern int      __vfscanf(FILE *, const char *, __va_list);
  85 extern int      __vfwprintf(FILE *, locale_t, const wchar_t *, __va_list);
  86 extern int      __vfwscanf(FILE * __restrict, locale_t, const wchar_t * __restrict,
  87                     __va_list);
  88 extern size_t   __fread(void * __restrict buf, size_t size, size_t count,
  89                 FILE * __restrict fp);
  90 extern int      __sdidinit;
  91
  92 static inline wint_t
  93 __fgetwc(FILE *fp, locale_t locale)
  94 {
  95         int nread;
  96         struct wchar_io_data *wcio;
  97         mbstate_t *st;
  98
  99         _SET_ORIENTATION(fp,1);
 100         wcio = WCIO_GET(fp);
 101
 102         if (wcio == NULL) {
 103                 return WEOF;
 104         }
 105         wcio->wcio_ungetwc_inbuf = 0;
 106         st = &wcio->wcio_mbstate_out;
 107
 108         return (__fgetwc_mbs(fp, st, &nread, locale));
 109 }
 110
 111 /*
 112  * Prepare the given FILE for writing, and return 0 iff it
 113  * can be written now.  Otherwise, return EOF and set errno.
 114  */
 115 #define prepwrite(fp) \
 116         ((((fp)->pub._flags & __SWR) == 0 || \
 117             ((fp)->_bf._base == NULL && ((fp)->pub._flags & __SSTR) == 0)) && \
 118          __swsetup(fp))
 119
 120 /*
 121  * Test whether the given stdio file has an active ungetc buffer;
 122  * release such a buffer, without restoring ordinary unread data.
 123  */
 124 #define HASUB(fp) ((fp)->_ub._base != NULL)
 125 #define FREEUB(fp) { \
 126         if ((fp)->_ub._base != (fp)->_ubuf) \
 127                 free((char *)(fp)->_ub._base); \
 128         (fp)->_ub._base = NULL; \
 129 }
 130
 131 /*
 132  * test for an fgetln() buffer.
 133  */
 134 #define HASLB(fp) ((fp)->_lb._base != NULL)
 135 #define FREELB(fp) { \
 136         free((char *)(fp)->_lb._base); \
 137         (fp)->_lb._base = NULL; \
 138 }
 139
 140 /*
 141  * Structure initializations for 'fake' FILE objects.
 142  */
 143 #define FAKE_FILE {                             \
 144         .pub._lbfsize = 0,                      \
 145         .pub._fileno = -1,                      \
 146         ._fl_mutex = PTHREAD_MUTEX_INITIALIZER, \
 147 }
 148
 149 /*
 150  * Set the orientation for a stream. If o > 0, the stream has wide-
 151  * orientation. If o < 0, the stream has byte-orientation.
 152  */
 153 #define ORIENT(fp, o)   _SET_ORIENTATION(fp, o)