diff options
Diffstat (limited to 'src/stdio')
-rw-r--r-- | src/stdio/fclose.c | 6 | ||||
-rw-r--r-- | src/stdio/ftrylockfile.c | 22 | ||||
-rw-r--r-- | src/stdio/funlockfile.c | 10 |
3 files changed, 36 insertions, 2 deletions
diff --git a/src/stdio/fclose.c b/src/stdio/fclose.c index 38e8a1e3..317b3c90 100644 --- a/src/stdio/fclose.c +++ b/src/stdio/fclose.c @@ -1,4 +1,8 @@ #include "stdio_impl.h" +#include "libc.h" + +static void dummy(FILE *f) { } +weak_alias(dummy, __unlist_locked_file); int fclose(FILE *f) { @@ -7,6 +11,8 @@ int fclose(FILE *f) FFINALLOCK(f); + __unlist_locked_file(f); + if (!(perm = f->flags & F_PERM)) { OFLLOCK(); if (f->prev) f->prev->next = f->next; diff --git a/src/stdio/ftrylockfile.c b/src/stdio/ftrylockfile.c index 56cccafd..6f9a4b88 100644 --- a/src/stdio/ftrylockfile.c +++ b/src/stdio/ftrylockfile.c @@ -2,9 +2,26 @@ #include "pthread_impl.h" #include <limits.h> +void __do_orphaned_stdio_locks() +{ + FILE *f; + for (f=__pthread_self()->stdio_locks; f; f=f->next_locked) + a_store(&f->lock, 0x40000000); +} + +void __unlist_locked_file(FILE *f) +{ + if (f->lockcount) { + if (f->next_locked) f->next_locked->prev_locked = f->prev_locked; + if (f->prev_locked) f->prev_locked->next_locked = f->next_locked; + else __pthread_self()->stdio_locks = f->next_locked; + } +} + int ftrylockfile(FILE *f) { - int tid = __pthread_self()->tid; + pthread_t self = __pthread_self(); + int tid = self->tid; if (f->lock == tid) { if (f->lockcount == LONG_MAX) return -1; @@ -15,5 +32,8 @@ int ftrylockfile(FILE *f) if (f->lock || a_cas(&f->lock, 0, tid)) return -1; f->lockcount = 1; + f->prev_locked = 0; + f->next_locked = self->stdio_locks; + self->stdio_locks = f; return 0; } diff --git a/src/stdio/funlockfile.c b/src/stdio/funlockfile.c index f8a2a071..30a07ef4 100644 --- a/src/stdio/funlockfile.c +++ b/src/stdio/funlockfile.c @@ -1,7 +1,15 @@ #include "stdio_impl.h" #include "pthread_impl.h" +void __unlist_locked_file(FILE *); + void funlockfile(FILE *f) { - if (!--f->lockcount) __unlockfile(f); + if (f->lockcount == 1) { + __unlist_locked_file(f); + f->lockcount = 0; + __unlockfile(f); + } else { + f->lockcount--; + } } |