Put local variables footer in all c, h files.

[idzebra-moved-to-github.git] / util / tstflock.c

diff --git a/util/tstflock.c b/util/tstflock.c
index 0d0be35..ec26477 100644 (file)
--- a/util/tstflock.c
+++ b/util/tstflock.c
@@ -2,21 +2,37 @@
  * Copyright (C) 1995-2005, Index Data ApS
  * See the file LICENSE for details.
  *
- * $Id: tstflock.c,v 1.1 2006-03-23 09:15:25 adam Exp $
+ * $Id: tstflock.c,v 1.6 2006-05-10 08:13:46 adam Exp $
  */
 
 #include <yaz/test.h>
 #if YAZ_POSIX_THREADS
 #include <pthread.h>
 #endif
+#ifdef WIN32
+#include <windows.h>
+#include <process.h>
+#endif
 
 #include <idzebra/flock.h>
+#if HAVE_UNISTD_H
 #include <unistd.h>
+#endif
 #include <string.h>
 
 static char seq[20];
 static char *seqp = seq;
 
+#define NUM_THREADS 2
+
+static void small_sleep()
+{
+#ifdef WIN32
+    Sleep(50);
+#else
+    sleep(1);
+#endif
+}
 void *run_func(void *arg)
 {
     int i;
@@ -26,7 +42,7 @@ void *run_func(void *arg)
         zebra_lock_w(lh);
      
         *seqp++ = 'L';
-        sleep(1);
+        small_sleep();
         *seqp++ = 'U';
         
         zebra_unlock(lh);
@@ -35,29 +51,69 @@ void *run_func(void *arg)
     return 0;
 }
 
-static void tst1()
+#ifdef WIN32
+DWORD WINAPI ThreadProc(void *p)
 {
-    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);
-    *seqp++ = '\0';
+    run_func(p);
+    return 0;
 }
 
+static void tst_win32()
+{
+    HANDLE handles[NUM_THREADS];
+    DWORD dwThreadId[NUM_THREADS];
+    int i, id[NUM_THREADS];
+    
+    for (i = 0; i<NUM_THREADS; 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 */
+    }
+    /* join */
+    WaitForMultipleObjects(NUM_THREADS, handles, TRUE, INFINITE);
+}
+#endif
+
+#if YAZ_POSIX_THREADS
+static void tst_pthread()
+{
+    pthread_t child_thread[NUM_THREADS];
+    int i, id[NUM_THREADS];
+    for (i = 0; i<NUM_THREADS; i++)
+        pthread_create(&child_thread[i], 0 /* attr */, run_func, &id[i]);
+
+    for (i = 0; i<NUM_THREADS; i++)
+        pthread_join(child_thread[i], 0);
+}
+#endif
+
 int main(int argc, char **argv)
 {
     YAZ_CHECK_INIT(argc, argv);
-    tst1();
 
+#ifdef WIN32
+    tst_win32();
+#endif
+#if YAZ_POSIX_THREADS
+    tst_pthread();
+#endif
+
+    *seqp++ = '\0';
+    printf("seq=%s\n", seq);
 #if 0
     /* does not pass.. for bug 529 */
     YAZ_CHECK(strcmp(seq, "LULULULU") == 0);
 #endif
     YAZ_CHECK_TERM;
 }
+
+
 /*
  * Local variables:
  * c-basic-offset: 4