X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=util%2Ftstflock.c;h=a9ca1d72277b30f351ae2ba5678ddef2e46fc038;hb=64dc2c2a92458f6157657807bbe8697ff82f2980;hp=3da7f7a47df4bf5dfe499f121e842b05434459b2;hpb=ce6df1eaab07d33544975ffe67a4b7a231cf795a;p=idzebra-moved-to-github.git

diff --git a/util/tstflock.c b/util/tstflock.c
index 3da7f7a..a9ca1d7 100644
--- a/util/tstflock.c
+++ b/util/tstflock.c
@@ -1,11 +1,14 @@
 /*
- * Copyright (C) 1995-2005, Index Data ApS
+ * Copyright (C) 1995-2006, Index Data ApS
  * See the file LICENSE for details.
  *
- * $Id: tstflock.c,v 1.2 2006-03-23 12:07:12 adam Exp $
+ * $Id: tstflock.c,v 1.10 2006-06-30 13:02:20 adam Exp $
  */
 
+#include <assert.h>
+#include <stdlib.h>
 #include <yaz/test.h>
+#include <yaz/log.h>
 #if YAZ_POSIX_THREADS
 #include <pthread.h>
 #endif
@@ -20,10 +23,10 @@
 #endif
 #include <string.h>
 
-static char seq[20];
-static char *seqp = seq;
+static char seq[1000];
+static char *seqp = 0;
 
-#define NUM_THREADS 2
+#define NUM_THREADS 100
 
 static void small_sleep()
 {
@@ -33,21 +36,45 @@ static void small_sleep()
     sleep(1);
 #endif
 }
+
 void *run_func(void *arg)
 {
     int i;
+    int *pdata = (int*) arg;
+    int use_write_lock = *pdata;
     ZebraLockHandle lh = zebra_lock_create(0, "my.LCK");
     for (i = 0; i<2; i++)
     {
-        zebra_lock_w(lh);
-     
-        *seqp++ = 'L';
-        small_sleep();
-        *seqp++ = 'U';
-        
-        zebra_unlock(lh);
+        int write_lock;
+
+        if (use_write_lock == 2)
+            write_lock = (rand() & 3) == 3 ? 1 : 0;
+        else
+            write_lock = use_write_lock;
+            
+        if (write_lock)
+        {
+            zebra_lock_w(lh);
+            
+            *seqp++ = 'L';
+            small_sleep();
+            *seqp++ = 'U';
+            
+            zebra_unlock(lh);
+        }
+        else
+        {
+            zebra_lock_r(lh);
+            
+            *seqp++ = 'l';
+            small_sleep();
+            *seqp++ = 'u';
+            
+            zebra_unlock(lh);
+        }
     }
     zebra_lock_destroy(lh);
+    *pdata = 123;
     return 0;
 }
 
@@ -57,55 +84,95 @@ DWORD WINAPI ThreadProc(void *p)
     run_func(p);
     return 0;
 }
+#endif
 
-static void tst1()
+static void tst_thread(int num, int write_flag)
 {
+#ifdef WIN32
     HANDLE handles[NUM_THREADS];
     DWORD dwThreadId[NUM_THREADS];
+#endif
+#if YAZ_POSIX_THREADS
+    pthread_t child_thread[NUM_THREADS];
+#endif
     int i, id[NUM_THREADS];
-    
-    for (i = 0; i<NUM_THREADS; i++)
+
+    seqp = seq;
+    assert (num <= NUM_THREADS);
+    for (i = 0; i < num; i++)
     {
-        void *pData = &id[i];
-        handles[i] = CreateThread(
-            NULL,              // default security attributes
-            0,                 // use default stack size  
-            ThreadProc,        // thread function 
-            pData,             // argument to thread function 
-            0,                 // use default creation flags 
-            &dwThreadId[i]);      // returns the thread identifier 
-    }
-    WaitForMultipleObjects(NUM_THREADS, handles, TRUE, INFINITE);
-    /* join */
-    *seqp++ = '\0';
-}
+        id[i] = write_flag;
+#if YAZ_POSIX_THREADS
+        pthread_create(&child_thread[i], 0 /* attr */, run_func, &id[i]);
 #endif
+#ifdef WIN32
+        if (1)
+        {
+            void *pData = &id[i];
+            handles[i] = CreateThread(
+                NULL,              /* default security attributes */
+                0,                 /* use default stack size */
+                ThreadProc,        /* thread function */
+                pData,             /* argument to thread function */
+                0,                 /* use default creation flags */
+                &dwThreadId[i]);   /* returns the thread identifier */
+        }
 
+#endif
+    }
 #if YAZ_POSIX_THREADS
-static void tst1()
-{
-    pthread_t child_thread[2];
-    int id1 = 1;
-    int id2 = 2;
-    pthread_create(&child_thread[0], 0 /* attr */, run_func, &id1);
-    pthread_create(&child_thread[1], 0 /* attr */, run_func, &id2);
-    pthread_join(child_thread[0], 0);
-    pthread_join(child_thread[1], 0);
+    for (i = 0; i<num; i++)
+        pthread_join(child_thread[i], 0);
+#endif
+#ifdef WIN32
+    WaitForMultipleObjects(num, handles, TRUE, INFINITE);
+#endif
+    for (i = 0; i < num; i++)
+        YAZ_CHECK(id[i] == 123);
     *seqp++ = '\0';
 }
-#endif
 
-int main(int argc, char **argv)
+static void tst()
 {
-    YAZ_CHECK_INIT(argc, argv);
-    tst1();
+    tst_thread(4, 1); /* write locks */
+#if 0
     printf("seq=%s\n", seq);
+#endif
+    if (1)
+    {
+        int i = 0;
+        while (seq[i])
+        {
+            YAZ_CHECK_EQ(seq[i], 'L');
+            YAZ_CHECK_EQ(seq[i+1], 'U');
+            i = i + 2;
+        }
+    }
+
 #if 0
-    /* does not pass.. for bug 529 */
-    YAZ_CHECK(strcmp(seq, "LULULULU") == 0);
+    tst_thread(6, 0);  /* read locks */
+    printf("seq=%s\n", seq);
 #endif
+#if 0
+    tst_thread(20, 2); /* random locks */
+    printf("seq=%s\n", seq);
+#endif
+}
+
+int main(int argc, char **argv)
+{
+    YAZ_CHECK_INIT(argc, argv);
+
+    yaz_log_time_format("%s:%!");
+
+    zebra_flock_init();
+
+    tst();
+
     YAZ_CHECK_TERM;
 }
+
+
 /*
  * Local variables:
  * c-basic-offset: 4