musl/src/stdio/ftrylockfile.c
Rich Felker c21f750727 fix stdio lock dependency on read-after-free not faulting
instead of using a waiters count, add a bit to the lock field
indicating that the lock may have waiters. threads which obtain the
lock after contending for it will perform a potentially-spurious wake
when they release the lock.
2018-04-18 14:22:49 -04:00

44 lines
985 B
C

#include "stdio_impl.h"
#include "pthread_impl.h"
#include <limits.h>
#define MAYBE_WAITERS 0x40000000
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)
{
pthread_t self = __pthread_self();
int tid = self->tid;
int owner = f->lock;
if ((owner & ~MAYBE_WAITERS) == tid) {
if (f->lockcount == LONG_MAX)
return -1;
f->lockcount++;
return 0;
}
if (owner < 0) f->lock = owner = 0;
if (owner || a_cas(&f->lock, 0, tid))
return -1;
f->lockcount = 1;
f->prev_locked = 0;
f->next_locked = self->stdio_locks;
if (f->next_locked) f->next_locked->prev_locked = f;
self->stdio_locks = f;
return 0;
}