Libc-1439.100.3.tar.gz

[apple/libc.git] / stdio / FreeBSD / findfp.c

diff --git a/stdio/FreeBSD/findfp.c b/stdio/FreeBSD/findfp.c
index f370d7abc6982e138d63e2a23fedddb231cc80cb..233ca39d01147719c6782cd8ed4cb1afad74c914 100644 (file)
--- a/stdio/FreeBSD/findfp.c
+++ b/stdio/FreeBSD/findfp.c
@@ -30,28 +30,40 @@
  * SUCH DAMAGE.
  */
 
  * SUCH DAMAGE.
  */
 

+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Wcomma"
+

 #if defined(LIBC_SCCS) && !defined(lint)
 static char sccsid[] = "@(#)findfp.c   8.2 (Berkeley) 1/4/94";
 #endif /* LIBC_SCCS and not lint */
 #include <sys/cdefs.h>
 __FBSDID("$FreeBSD: src/lib/libc/stdio/findfp.c,v 1.34 2009/12/05 19:31:38 ed Exp $");
 
 #if defined(LIBC_SCCS) && !defined(lint)
 static char sccsid[] = "@(#)findfp.c   8.2 (Berkeley) 1/4/94";
 #endif /* LIBC_SCCS and not lint */
 #include <sys/cdefs.h>
 __FBSDID("$FreeBSD: src/lib/libc/stdio/findfp.c,v 1.34 2009/12/05 19:31:38 ed Exp $");
 

+#include <TargetConditionals.h>
+

 #include <sys/param.h>
 #include <machine/atomic.h>
 #include <unistd.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/param.h>
 #include <machine/atomic.h>
 #include <unistd.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>

+#include <libkern/OSAtomic.h>
+#include <errno.h>

 
 

+#include <pthread.h>

 #include <spinlock.h>
 
 #include "libc_private.h"
 #include "local.h"
 #include "glue.h"
 
 #include <spinlock.h>
 
 #include "libc_private.h"
 #include "local.h"
 #include "glue.h"
 

-int    __sdidinit;
+pthread_once_t __sdidinit;

 
 

+#if !TARGET_OS_IPHONE

 #define        NDYNAMIC 10             /* add ten more whenever necessary */
 #define        NDYNAMIC 10             /* add ten more whenever necessary */

+#else
+#define        NDYNAMIC 1              /* add one at a time on embedded */
+#endif

 
 #define        std(flags, file) {              \
        ._flags = (flags),              \
 
 #define        std(flags, file) {              \
        ._flags = (flags),              \


@@ -61,12 +73,31 @@ int __sdidinit;
        ._read = __sread,               \
        ._seek = __sseek,               \
        ._write = __swrite,             \
        ._read = __sread,               \
        ._seek = __sseek,               \
        ._write = __swrite,             \

+       ._extra = __sFX + file,         \

 }
 }

+  /* set counted */
+#define __sFXInit3      {.fl_mutex = PTHREAD_RECURSIVE_MUTEX_INITIALIZER, .counted = 1}
+
+static int64_t __scounted;             /* streams counted against STREAM_MAX */
+static int64_t __stream_max;
+
+#if !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
+/* usual and usual_extra are data pigs. See 7929728. For embedded we should
+ * always allocate dynamically, and probably should for desktop too. */

                                /* the usual - (stdin + stdout + stderr) */
 static FILE usual[FOPEN_MAX - 3];
                                /* the usual - (stdin + stdout + stderr) */
 static FILE usual[FOPEN_MAX - 3];

+static struct __sFILEX usual_extra[FOPEN_MAX - 3];

 static struct glue uglue = { NULL, FOPEN_MAX - 3, usual };
 static struct glue uglue = { NULL, FOPEN_MAX - 3, usual };

+#endif /* !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR) */
+
+static struct __sFILEX __sFX[3] = {__sFXInit3, __sFXInit3, __sFXInit3};

 
 

-static FILE __sF[3] = {
+/*
+ * We can't make this 'static' due to binary compatibility concerns.
+ * This also means we cannot change the sizeof(FILE) and must continue to
+ * use the __sFILEX stuff to add to FILE.
+ */
+FILE __sF[3] = {

        std(__SRD, STDIN_FILENO),
        std(__SWR, STDOUT_FILENO),
        std(__SWR|__SNBF, STDERR_FILENO)
        std(__SRD, STDIN_FILENO),
        std(__SWR, STDOUT_FILENO),
        std(__SWR|__SNBF, STDERR_FILENO)


@@ -76,38 +107,45 @@ FILE *__stdinp = &__sF[0];
 FILE *__stdoutp = &__sF[1];
 FILE *__stderrp = &__sF[2];
 
 FILE *__stdoutp = &__sF[1];
 FILE *__stderrp = &__sF[2];
 

+#if !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)

 struct glue __sglue = { &uglue, 3, __sF };
 static struct glue *lastglue = &uglue;
 struct glue __sglue = { &uglue, 3, __sF };
 static struct glue *lastglue = &uglue;

+#else
+struct glue __sglue = { NULL, 3, __sF };
+static struct glue *lastglue = &__sglue;
+#endif

 
 static struct glue *   moreglue(int);
 
 
 static struct glue *   moreglue(int);
 

-static spinlock_t thread_lock = _SPINLOCK_INITIALIZER;
-#define THREAD_LOCK()  if (__isthreaded) _SPINLOCK(&thread_lock)
-#define THREAD_UNLOCK()        if (__isthreaded) _SPINUNLOCK(&thread_lock)
-
-#if NOT_YET
-#define        SET_GLUE_PTR(ptr, val)  atomic_set_rel_ptr(&(ptr), (uintptr_t)(val))
-#else
-#define        SET_GLUE_PTR(ptr, val)  ptr = val
-#endif
+static pthread_mutex_t filelist_lock = PTHREAD_MUTEX_INITIALIZER;
+#define FILELIST_LOCK()        do { pthread_mutex_lock(&filelist_lock); } while(0)
+#define FILELIST_UNLOCK()      do { pthread_mutex_unlock(&filelist_lock); } while(0)

 
 static struct glue *
 
 static struct glue *

-moreglue(n)
-       int n;
+moreglue(int n)

 {
        struct glue *g;
 {
        struct glue *g;

-       static FILE empty;

        FILE *p;
        FILE *p;

+       struct __sFILEX *fx;
+       size_t align;

 
 

-       g = (struct glue *)malloc(sizeof(*g) + ALIGNBYTES + n * sizeof(FILE));
+       align = __alignof__( (struct { FILE f; struct __sFILEX s; }){} );
+       g = (struct glue *)malloc(sizeof(*g) + align + n * sizeof(FILE) +
+           n * sizeof(struct __sFILEX));

        if (g == NULL)
                return (NULL);
        if (g == NULL)
                return (NULL);

-       p = (FILE *)ALIGN(g + 1);
+       p = (FILE *)roundup((uintptr_t)(g + 1), align);
+       fx = (struct __sFILEX *)&p[n];

        g->next = NULL;
        g->niobs = n;
        g->iobs = p;
        g->next = NULL;
        g->niobs = n;
        g->iobs = p;

-       while (--n >= 0)
-               *p++ = empty;
+        
+       while (--n >= 0) {
+               bzero(p, sizeof(*p));
+               p->_extra = fx;
+               INITEXTRA(p);
+               p++, fx++;
+       }

        return (g);
 }
 
        return (g);
 }
 


@@ -115,33 +153,51 @@ moreglue(n)
  * Find a free FILE for fopen et al.
  */
 FILE *
  * Find a free FILE for fopen et al.
  */
 FILE *

-__sfp()
+__sfp(int count)

 {
        FILE    *fp;
        int     n;
        struct glue *g;
 
 {
        FILE    *fp;
        int     n;
        struct glue *g;
 

-       if (!__sdidinit)
-               __sinit();
+       pthread_once(&__sdidinit, __sinit);
+
+       if (count) {
+               int64_t new = OSAtomicIncrement64(&__scounted);
+               if (new > __stream_max) {
+                       /* Greater than the saved limit, check again with getrlimit. */
+                       if (new > (__stream_max = sysconf(_SC_STREAM_MAX))){
+                               OSAtomicDecrement64(&__scounted);
+                               errno = EMFILE;
+                               return NULL;
+                       }
+               }
+               /* Overflowing #streams beyond RLIMIT_INFINITY */
+               if (new < 0) {
+                       errno = EOVERFLOW;
+                       return NULL;
+               }
+       }

        /*
         * The list must be locked because a FILE may be updated.
         */
        /*
         * The list must be locked because a FILE may be updated.
         */

-       THREAD_LOCK();
+       FILELIST_LOCK();

        for (g = &__sglue; g != NULL; g = g->next) {
                for (fp = g->iobs, n = g->niobs; --n >= 0; fp++)
                        if (fp->_flags == 0)
                                goto found;
        }
        for (g = &__sglue; g != NULL; g = g->next) {
                for (fp = g->iobs, n = g->niobs; --n >= 0; fp++)
                        if (fp->_flags == 0)
                                goto found;
        }

-       THREAD_UNLOCK();        /* don't hold lock while malloc()ing. */
+       FILELIST_UNLOCK();      /* don't hold lock while malloc()ing. */

        if ((g = moreglue(NDYNAMIC)) == NULL)
                return (NULL);
        if ((g = moreglue(NDYNAMIC)) == NULL)
                return (NULL);

-       THREAD_LOCK();          /* reacquire the lock */
-       SET_GLUE_PTR(lastglue->next, g); /* atomically append glue to list */
+       FILELIST_LOCK();                /* reacquire the lock */
+       lastglue->next = g; /* atomically append glue to list */

        lastglue = g;           /* not atomic; only accessed when locked */
        fp = g->iobs;
 found:
        fp->_flags = 1;         /* reserve this slot; caller sets real flags */
        lastglue = g;           /* not atomic; only accessed when locked */
        fp = g->iobs;
 found:
        fp->_flags = 1;         /* reserve this slot; caller sets real flags */

-       THREAD_UNLOCK();
+       FILELIST_UNLOCK();
+       
+       /* _flags = 1 means the FILE* is in use, and this thread owns the object while it is being initialized */

        fp->_p = NULL;          /* no current pointer */
        fp->_w = 0;             /* nothing to read or write */
        fp->_r = 0;
        fp->_p = NULL;          /* no current pointer */
        fp->_w = 0;             /* nothing to read or write */
        fp->_r = 0;


@@ -155,11 +211,30 @@ found:
        fp->_lb._base = NULL;   /* no line buffer */
        fp->_lb._size = 0;
 /*     fp->_lock = NULL; */    /* once set always set (reused) */
        fp->_lb._base = NULL;   /* no line buffer */
        fp->_lb._size = 0;
 /*     fp->_lock = NULL; */    /* once set always set (reused) */

-       fp->_orientation = 0;
-       memset(&fp->_mbstate, 0, sizeof(mbstate_t));
+       INITEXTRA(fp);
+       fp->_extra->counted = count ? 1 : 0;

        return (fp);
 }
 
        return (fp);
 }
 

+/*
+ * Mark as free and update count as needed
+ */
+__private_extern__ void
+__sfprelease(FILE *fp)
+{
+       if (fp->_counted) {
+               OSAtomicDecrement64(&__scounted);
+               fp->_counted = 0;
+       }
+       
+       pthread_mutex_destroy(&fp->_extra->fl_mutex);
+       
+       /* Make sure nobody else is enumerating the list while we clear the "in use" _flags field. */
+       FILELIST_LOCK();
+       fp->_flags = 0;
+       FILELIST_UNLOCK();
+}
+

 /*
  * XXX.  Force immediate allocation of internal memory.  Not used by stdio,
  * but documented historically for certain applications.  Bad applications.
 /*
  * XXX.  Force immediate allocation of internal memory.  Not used by stdio,
  * but documented historically for certain applications.  Bad applications.


@@ -182,10 +257,10 @@ f_prealloc(void)
        for (g = &__sglue; (n -= g->niobs) > 0 && g->next; g = g->next)
                /* void */;
        if ((n > 0) && ((g = moreglue(n)) != NULL)) {
        for (g = &__sglue; (n -= g->niobs) > 0 && g->next; g = g->next)
                /* void */;
        if ((n > 0) && ((g = moreglue(n)) != NULL)) {

-               THREAD_LOCK();
-               SET_GLUE_PTR(lastglue->next, g);
+               FILELIST_LOCK();
+               lastglue->next = g;

                lastglue = g;
                lastglue = g;

-               THREAD_UNLOCK();
+               FILELIST_UNLOCK();

        }
 }
 
        }
 }
 


@@ -197,7 +272,7 @@ f_prealloc(void)
  * The name `_cleanup' is, alas, fairly well known outside stdio.
  */
 void
  * The name `_cleanup' is, alas, fairly well known outside stdio.
  */
 void

-_cleanup()
+_cleanup(void)

 {
        /* (void) _fwalk(fclose); */
        (void) _fwalk(__sflush);                /* `cheating' */
 {
        /* (void) _fwalk(fclose); */
        (void) _fwalk(__sflush);                /* `cheating' */


@@ -207,10 +282,23 @@ _cleanup()
  * __sinit() is called whenever stdio's internal variables must be set up.
  */
 void
  * __sinit() is called whenever stdio's internal variables must be set up.
  */
 void

-__sinit()
+__sinit(void)

 {
 {

+#if !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
+       int i;
+#endif

 
        /* Make sure we clean up on exit. */
        __cleanup = _cleanup;           /* conservative */
 
        /* Make sure we clean up on exit. */
        __cleanup = _cleanup;           /* conservative */

-       __sdidinit = 1;
+       __stream_max = sysconf(_SC_STREAM_MAX);
+       __scounted = 3;                 /* std{in,out,err} already exists */
+
+#if !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
+       /* Set _extra for the usual suspects. */
+       for (i = 0; i < FOPEN_MAX - 3; i++) {
+               usual[i]._extra = &usual_extra[i];
+               INITEXTRA(&usual[i]);
+       }
+#endif

 }
 }

+#pragma clang diagnostic pop