Browse Source

simplify and optimize FILE lock handling

Rich Felker 14 years ago
parent
commit
a37452430f
5 changed files with 14 additions and 14 deletions
  1. 2 3
      src/internal/stdio_impl.h
  2. 7 6
      src/stdio/__lockfile.c
  3. 3 3
      src/stdio/ftrylockfile.c
  4. 1 1
      src/stdio/vsnprintf.c
  5. 1 1
      src/stdio/vswprintf.c

+ 2 - 3
src/internal/stdio_impl.h

@@ -23,8 +23,8 @@
 
 #define UNGET 4
 
-#define FLOCK(f) ((libc.lockfile && (f)->owner>=0) ? (libc.lockfile((f)),0) : 0)
-#define FUNLOCK(f) ((f)->lockcount && (--(f)->lockcount || ((f)->owner=(f)->lock=0)))
+#define FLOCK(f) ((libc.lockfile && (f)->lock>=0) ? (libc.lockfile((f)),0) : 0)
+#define FUNLOCK(f) ((f)->lockcount && (--(f)->lockcount || ((f)->lock=0)))
 
 #define F_PERM 1
 #define F_NORD 4
@@ -59,7 +59,6 @@ struct __FILE_s {
 	off_t (*seek)(FILE *, off_t, int);
 	int mode;
 	int (*close)(FILE *);
-	int owner;
 };
 
 size_t __stdio_read(FILE *, unsigned char *, size_t);

+ 7 - 6
src/stdio/__lockfile.c

@@ -3,17 +3,18 @@
 
 void __lockfile(FILE *f)
 {
-	int spins;
-	if (f->owner < 0) return;
-	if (f->owner && f->owner == __pthread_self()->tid) {
+	int spins=100000;
+	int tid;
+
+	if (f->lock < 0) return;
+	tid = __pthread_self()->tid;
+	if (f->lock == tid) {
 		while (f->lockcount == INT_MAX);
 		f->lockcount++;
 		return;
 	}
-	spins = 100000;
-	while (a_swap(&f->lock, 1))
+	while (f->lock || a_cas(&f->lock, 0, tid))
 		if (spins) spins--, a_spin();
 		else syscall(SYS_sched_yield);
-	f->owner = __pthread_self()->tid;
 	f->lockcount = 1;
 }

+ 3 - 3
src/stdio/ftrylockfile.c

@@ -3,16 +3,16 @@
 
 int ftrylockfile(FILE *f)
 {
+	int tid = pthread_self()->tid;
 	if (!libc.lockfile) libc.lockfile = __lockfile;
-	if (f->owner && f->owner == pthread_self()->tid) {
+	if (f->lock == tid) {
 		if (f->lockcount == INT_MAX)
 			return -1;
 		f->lockcount++;
 		return 0;
 	}
-	if (a_swap(&f->lock, 1))
+	if (f->lock || a_cas(&f->lock, 0, tid))
 		return -1;
-	f->owner = pthread_self()->tid;
 	f->lockcount = 1;
 	return 0;
 }

+ 1 - 1
src/stdio/vsnprintf.c

@@ -17,7 +17,7 @@ int vsnprintf(char *s, size_t n, const char *fmt, va_list ap)
 	f.write = sn_write;
 	f.buf_size = 1;
 	f.buf = buf;
-	f.owner = -1;
+	f.lock = -1;
 	if (n > INT_MAX) {
 		errno = EOVERFLOW;
 		return -1;

+ 1 - 1
src/stdio/vswprintf.c

@@ -32,7 +32,7 @@ int vswprintf(wchar_t *s, size_t n, const wchar_t *fmt, va_list ap)
 	f.write = sw_write;
 	f.buf_size = sizeof buf;
 	f.buf = buf;
-	f.owner = -1;
+	f.lock = -1;
 	f.cookie = &c;
 	if (!n) {
 		return -1;