Revert "Win32: use native Win32 functions directly..." (fixes #5252)
[vlc.git] / src / win32 / thread.c
1 /*****************************************************************************
2  * thread.c : Win32 back-end for LibVLC
3  *****************************************************************************
4  * Copyright (C) 1999-2009 the VideoLAN team
5  *
6  * Authors: Jean-Marc Dressler <polux@via.ecp.fr>
7  *          Samuel Hocevar <sam@zoy.org>
8  *          Gildas Bazin <gbazin@netcourrier.com>
9  *          Clément Sténac
10  *          Rémi Denis-Courmont
11  *          Pierre Ynard
12  *
13  * This program is free software; you can redistribute it and/or modify
14  * it under the terms of the GNU General Public License as published by
15  * the Free Software Foundation; either version 2 of the License, or
16  * (at your option) any later version.
17  *
18  * This program is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
26  *****************************************************************************/
27
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <vlc_common.h>
33
34 #include "libvlc.h"
35 #include <stdarg.h>
36 #include <assert.h>
37 #include <limits.h>
38 #include <errno.h>
39 #ifdef UNDER_CE
40 # include <mmsystem.h>
41 #endif
42
43 static vlc_threadvar_t thread_key;
44
45 /**
46  * Per-thread data
47  */
48 struct vlc_thread
49 {
50     HANDLE         id;
51 #ifdef UNDER_CE
52     HANDLE         cancel_event;
53 #endif
54
55     bool           detached;
56     bool           killable;
57     bool           killed;
58     vlc_cleanup_t *cleaners;
59
60     void        *(*entry) (void *);
61     void          *data;
62 };
63
64 static vlc_mutex_t super_mutex;
65 static vlc_cond_t  super_variable;
66 extern vlc_rwlock_t config_lock, msg_lock;
67
68 BOOL WINAPI DllMain (HINSTANCE, DWORD, LPVOID);
69
70 BOOL WINAPI DllMain (HINSTANCE hinstDll, DWORD fdwReason, LPVOID lpvReserved)
71 {
72     (void) hinstDll;
73     (void) lpvReserved;
74
75     switch (fdwReason)
76     {
77         case DLL_PROCESS_ATTACH:
78             vlc_mutex_init (&super_mutex);
79             vlc_cond_init (&super_variable);
80             vlc_threadvar_create (&thread_key, NULL);
81             vlc_rwlock_init (&config_lock);
82             vlc_rwlock_init (&msg_lock);
83             vlc_CPU_init ();
84             break;
85
86         case DLL_PROCESS_DETACH:
87             vlc_rwlock_destroy (&msg_lock);
88             vlc_rwlock_destroy (&config_lock);
89             vlc_threadvar_delete (&thread_key);
90             vlc_cond_destroy (&super_variable);
91             vlc_mutex_destroy (&super_mutex);
92             break;
93     }
94     return TRUE;
95 }
96
97 static void CALLBACK vlc_cancel_self (ULONG_PTR);
98
99 static DWORD vlc_WaitForMultipleObjects (DWORD count, const HANDLE *handles,
100                                          DWORD delay)
101 {
102     DWORD ret;
103 #ifdef UNDER_CE
104     HANDLE buf[count + 1];
105
106     struct vlc_thread *th = vlc_threadvar_get (thread_key);
107     if (th != NULL)
108     {
109         memcpy (buf, handles, count * sizeof(HANDLE));
110         buf[count++] = th->cancel_event;
111         handles = buf;
112     }
113
114     if (count == 0)
115     {
116          Sleep (delay);
117          ret = WAIT_TIMEOUT;
118     }
119     else
120         ret = WaitForMultipleObjects (count, handles, FALSE, delay);
121
122     if ((th != NULL) && (ret == WAIT_OBJECT_0 + count - 1))
123     {
124         vlc_cancel_self ((uintptr_t)th);
125         ret = WAIT_IO_COMPLETION;
126     }
127 #else
128     if (count == 0)
129     {
130         ret = SleepEx (delay, TRUE);
131         if (ret == 0)
132             ret = WAIT_TIMEOUT;
133     }
134     else
135         ret = WaitForMultipleObjectsEx (count, handles, FALSE, delay, TRUE);
136 #endif
137     /* We do not abandon objects... this would be a bug */
138     assert (ret < WAIT_ABANDONED_0 || WAIT_ABANDONED_0 + count - 1 < ret);
139
140     if (unlikely(ret == WAIT_FAILED))
141         abort (); /* We are screwed! */
142     return ret;
143 }
144
145 static DWORD vlc_WaitForSingleObject (HANDLE handle, DWORD delay)
146 {
147     return vlc_WaitForMultipleObjects (1, &handle, delay);
148 }
149
150 static DWORD vlc_Sleep (DWORD delay)
151 {
152     DWORD ret = vlc_WaitForMultipleObjects (0, NULL, delay);
153     return (ret != WAIT_TIMEOUT) ? ret : 0;
154 }
155
156
157 /*** Mutexes ***/
158 void vlc_mutex_init( vlc_mutex_t *p_mutex )
159 {
160     /* This creates a recursive mutex. This is OK as fast mutexes have
161      * no defined behavior in case of recursive locking. */
162     InitializeCriticalSection (&p_mutex->mutex);
163     p_mutex->dynamic = true;
164 }
165
166 void vlc_mutex_init_recursive( vlc_mutex_t *p_mutex )
167 {
168     InitializeCriticalSection( &p_mutex->mutex );
169     p_mutex->dynamic = true;
170 }
171
172
173 void vlc_mutex_destroy (vlc_mutex_t *p_mutex)
174 {
175     assert (p_mutex->dynamic);
176     DeleteCriticalSection (&p_mutex->mutex);
177 }
178
179 void vlc_mutex_lock (vlc_mutex_t *p_mutex)
180 {
181     if (!p_mutex->dynamic)
182     {   /* static mutexes */
183         int canc = vlc_savecancel ();
184         assert (p_mutex != &super_mutex); /* this one cannot be static */
185
186         vlc_mutex_lock (&super_mutex);
187         while (p_mutex->locked)
188         {
189             p_mutex->contention++;
190             vlc_cond_wait (&super_variable, &super_mutex);
191             p_mutex->contention--;
192         }
193         p_mutex->locked = true;
194         vlc_mutex_unlock (&super_mutex);
195         vlc_restorecancel (canc);
196         return;
197     }
198
199     EnterCriticalSection (&p_mutex->mutex);
200 }
201
202 int vlc_mutex_trylock (vlc_mutex_t *p_mutex)
203 {
204     if (!p_mutex->dynamic)
205     {   /* static mutexes */
206         int ret = EBUSY;
207
208         assert (p_mutex != &super_mutex); /* this one cannot be static */
209         vlc_mutex_lock (&super_mutex);
210         if (!p_mutex->locked)
211         {
212             p_mutex->locked = true;
213             ret = 0;
214         }
215         vlc_mutex_unlock (&super_mutex);
216         return ret;
217     }
218
219     return TryEnterCriticalSection (&p_mutex->mutex) ? 0 : EBUSY;
220 }
221
222 void vlc_mutex_unlock (vlc_mutex_t *p_mutex)
223 {
224     if (!p_mutex->dynamic)
225     {   /* static mutexes */
226         assert (p_mutex != &super_mutex); /* this one cannot be static */
227
228         vlc_mutex_lock (&super_mutex);
229         assert (p_mutex->locked);
230         p_mutex->locked = false;
231         if (p_mutex->contention)
232             vlc_cond_broadcast (&super_variable);
233         vlc_mutex_unlock (&super_mutex);
234         return;
235     }
236
237     LeaveCriticalSection (&p_mutex->mutex);
238 }
239
240 /*** Condition variables ***/
241 enum
242 {
243     CLOCK_REALTIME=0, /* must be zero for VLC_STATIC_COND */
244     CLOCK_MONOTONIC,
245 };
246
247 static void vlc_cond_init_common (vlc_cond_t *p_condvar, unsigned clock)
248 {
249     /* Create a manual-reset event (manual reset is needed for broadcast). */
250     p_condvar->handle = CreateEvent (NULL, TRUE, FALSE, NULL);
251     if (!p_condvar->handle)
252         abort();
253     p_condvar->clock = clock;
254 }
255
256 void vlc_cond_init (vlc_cond_t *p_condvar)
257 {
258     vlc_cond_init_common (p_condvar, CLOCK_MONOTONIC);
259 }
260
261 void vlc_cond_init_daytime (vlc_cond_t *p_condvar)
262 {
263     vlc_cond_init_common (p_condvar, CLOCK_REALTIME);
264 }
265
266 void vlc_cond_destroy (vlc_cond_t *p_condvar)
267 {
268     CloseHandle (p_condvar->handle);
269 }
270
271 void vlc_cond_signal (vlc_cond_t *p_condvar)
272 {
273     if (!p_condvar->handle)
274         return;
275
276     /* This is suboptimal but works. */
277     vlc_cond_broadcast (p_condvar);
278 }
279
280 void vlc_cond_broadcast (vlc_cond_t *p_condvar)
281 {
282     if (!p_condvar->handle)
283         return;
284
285     /* Wake all threads up (as the event HANDLE has manual reset) */
286     SetEvent (p_condvar->handle);
287 }
288
289 void vlc_cond_wait (vlc_cond_t *p_condvar, vlc_mutex_t *p_mutex)
290 {
291     DWORD result;
292
293     if (!p_condvar->handle)
294     {   /* FIXME FIXME FIXME */
295         msleep (50000);
296         return;
297     }
298
299     do
300     {
301         vlc_testcancel ();
302         vlc_mutex_unlock (p_mutex);
303         result = vlc_WaitForSingleObject (p_condvar->handle, INFINITE);
304         vlc_mutex_lock (p_mutex);
305     }
306     while (result == WAIT_IO_COMPLETION);
307
308     ResetEvent (p_condvar->handle);
309 }
310
311 int vlc_cond_timedwait (vlc_cond_t *p_condvar, vlc_mutex_t *p_mutex,
312                         mtime_t deadline)
313 {
314     DWORD result;
315
316     if (!p_condvar->handle)
317     {   /* FIXME FIXME FIXME */
318         msleep (50000);
319         return 0;
320     }
321
322     do
323     {
324         vlc_testcancel ();
325
326         mtime_t total;
327         switch (p_condvar->clock)
328         {
329             case CLOCK_REALTIME: /* FIXME? sub-second precision */
330                 total = CLOCK_FREQ * time (NULL);
331                 break;
332             default:
333                 assert (p_condvar->clock == CLOCK_MONOTONIC);
334                 total = mdate();
335                 break;
336         }
337         total = (deadline - total) / 1000;
338         if( total < 0 )
339             total = 0;
340
341         DWORD delay = (total > 0x7fffffff) ? 0x7fffffff : total;
342         vlc_mutex_unlock (p_mutex);
343         result = vlc_WaitForSingleObject (p_condvar->handle, delay);
344         vlc_mutex_lock (p_mutex);
345     }
346     while (result == WAIT_IO_COMPLETION);
347
348     ResetEvent (p_condvar->handle);
349
350     return (result == WAIT_OBJECT_0) ? 0 : ETIMEDOUT;
351 }
352
353 /*** Semaphore ***/
354 void vlc_sem_init (vlc_sem_t *sem, unsigned value)
355 {
356     *sem = CreateSemaphore (NULL, value, 0x7fffffff, NULL);
357     if (*sem == NULL)
358         abort ();
359 }
360
361 void vlc_sem_destroy (vlc_sem_t *sem)
362 {
363     CloseHandle (*sem);
364 }
365
366 int vlc_sem_post (vlc_sem_t *sem)
367 {
368     ReleaseSemaphore (*sem, 1, NULL);
369     return 0; /* FIXME */
370 }
371
372 void vlc_sem_wait (vlc_sem_t *sem)
373 {
374     DWORD result;
375
376     do
377     {
378         vlc_testcancel ();
379         result = vlc_WaitForSingleObject (*sem, INFINITE);
380     }
381     while (result == WAIT_IO_COMPLETION);
382 }
383
384 /*** Read/write locks */
385 /* SRW (Slim Read Write) locks are available in Vista+ only */
386 void vlc_rwlock_init (vlc_rwlock_t *lock)
387 {
388     vlc_mutex_init (&lock->mutex);
389     vlc_cond_init (&lock->wait);
390     lock->readers = 0; /* active readers */
391     lock->writers = 0; /* waiting writers */
392     lock->writer = 0; /* ID of active writer */
393 }
394
395 void vlc_rwlock_destroy (vlc_rwlock_t *lock)
396 {
397     vlc_cond_destroy (&lock->wait);
398     vlc_mutex_destroy (&lock->mutex);
399 }
400
401 void vlc_rwlock_rdlock (vlc_rwlock_t *lock)
402 {
403     vlc_mutex_lock (&lock->mutex);
404     /* Recursive read-locking is allowed. With the infos available:
405      *  - the loosest possible condition (no active writer) is:
406      *     (lock->writer != 0)
407      *  - the strictest possible condition is:
408      *     (lock->writer != 0 || (lock->readers == 0 && lock->writers > 0))
409      *  or (lock->readers == 0 && (lock->writer != 0 || lock->writers > 0))
410      */
411     while (lock->writer != 0)
412     {
413         assert (lock->readers == 0);
414         vlc_cond_wait (&lock->wait, &lock->mutex);
415     }
416     if (unlikely(lock->readers == ULONG_MAX))
417         abort ();
418     lock->readers++;
419     vlc_mutex_unlock (&lock->mutex);
420 }
421
422 static void vlc_rwlock_rdunlock (vlc_rwlock_t *lock)
423 {
424     vlc_mutex_lock (&lock->mutex);
425     assert (lock->readers > 0);
426
427     /* If there are no readers left, wake up a writer. */
428     if (--lock->readers == 0 && lock->writers > 0)
429         vlc_cond_signal (&lock->wait);
430     vlc_mutex_unlock (&lock->mutex);
431 }
432
433 void vlc_rwlock_wrlock (vlc_rwlock_t *lock)
434 {
435     vlc_mutex_lock (&lock->mutex);
436     if (unlikely(lock->writers == ULONG_MAX))
437         abort ();
438     lock->writers++;
439     /* Wait until nobody owns the lock in either way. */
440     while ((lock->readers > 0) || (lock->writer != 0))
441         vlc_cond_wait (&lock->wait, &lock->mutex);
442     lock->writers--;
443     assert (lock->writer == 0);
444     lock->writer = GetCurrentThreadId ();
445     vlc_mutex_unlock (&lock->mutex);
446 }
447
448 static void vlc_rwlock_wrunlock (vlc_rwlock_t *lock)
449 {
450     vlc_mutex_lock (&lock->mutex);
451     assert (lock->writer == GetCurrentThreadId ());
452     assert (lock->readers == 0);
453     lock->writer = 0; /* Write unlock */
454
455     /* Let reader and writer compete. Scheduler decides who wins. */
456     vlc_cond_broadcast (&lock->wait);
457     vlc_mutex_unlock (&lock->mutex);
458 }
459
460 void vlc_rwlock_unlock (vlc_rwlock_t *lock)
461 {
462     /* Note: If the lock is held for reading, lock->writer is nul.
463      * If the lock is held for writing, only this thread can store a value to
464      * lock->writer. Either way, lock->writer is safe to fetch here. */
465     if (lock->writer != 0)
466         vlc_rwlock_wrunlock (lock);
467     else
468         vlc_rwlock_rdunlock (lock);
469 }
470
471 /*** Thread-specific variables (TLS) ***/
472 struct vlc_threadvar
473 {
474     DWORD                 id;
475     void                (*destroy) (void *);
476     struct vlc_threadvar *prev;
477     struct vlc_threadvar *next;
478 } *vlc_threadvar_last = NULL;
479
480 int vlc_threadvar_create (vlc_threadvar_t *p_tls, void (*destr) (void *))
481 {
482     struct vlc_threadvar *var = malloc (sizeof (*var));
483     if (unlikely(var == NULL))
484         return errno;
485
486     var->id = TlsAlloc();
487     if (var->id == TLS_OUT_OF_INDEXES)
488     {
489         free (var);
490         return EAGAIN;
491     }
492     var->destroy = destr;
493     var->next = NULL;
494     *p_tls = var;
495
496     vlc_mutex_lock (&super_mutex);
497     var->prev = vlc_threadvar_last;
498     vlc_threadvar_last = var;
499     vlc_mutex_unlock (&super_mutex);
500     return 0;
501 }
502
503 void vlc_threadvar_delete (vlc_threadvar_t *p_tls)
504 {
505     struct vlc_threadvar *var = *p_tls;
506
507     vlc_mutex_lock (&super_mutex);
508     if (var->prev != NULL)
509         var->prev->next = var->next;
510     else
511         vlc_threadvar_last = var->next;
512     if (var->next != NULL)
513         var->next->prev = var->prev;
514     vlc_mutex_unlock (&super_mutex);
515
516     TlsFree (var->id);
517     free (var);
518 }
519
520 int vlc_threadvar_set (vlc_threadvar_t key, void *value)
521 {
522     return TlsSetValue (key->id, value) ? ENOMEM : 0;
523 }
524
525 void *vlc_threadvar_get (vlc_threadvar_t key)
526 {
527     return TlsGetValue (key->id);
528 }
529
530 /*** Threads ***/
531 void vlc_threads_setup (libvlc_int_t *p_libvlc)
532 {
533     (void) p_libvlc;
534 }
535
536 static void vlc_thread_cleanup (struct vlc_thread *th)
537 {
538     vlc_threadvar_t key;
539
540 retry:
541     /* TODO: use RW lock or something similar */
542     vlc_mutex_lock (&super_mutex);
543     for (key = vlc_threadvar_last; key != NULL; key = key->prev)
544     {
545         void *value = vlc_threadvar_get (key);
546         if (value != NULL && key->destroy != NULL)
547         {
548             vlc_mutex_unlock (&super_mutex);
549             vlc_threadvar_set (key, NULL);
550             key->destroy (value);
551             goto retry;
552         }
553     }
554     vlc_mutex_unlock (&super_mutex);
555
556     if (th->detached)
557     {
558         CloseHandle (th->id);
559 #ifdef UNDER_CE
560         CloseHandle (th->cancel_event);
561 #endif
562         free (th);
563     }
564 }
565
566 static unsigned __stdcall vlc_entry (void *p)
567 {
568     struct vlc_thread *th = p;
569
570     vlc_threadvar_set (thread_key, th);
571     th->killable = true;
572     th->data = th->entry (th->data);
573     vlc_thread_cleanup (th);
574     return 0;
575 }
576
577 static int vlc_clone_attr (vlc_thread_t *p_handle, bool detached,
578                            void *(*entry) (void *), void *data, int priority)
579 {
580     struct vlc_thread *th = malloc (sizeof (*th));
581     if (unlikely(th == NULL))
582         return ENOMEM;
583     th->entry = entry;
584     th->data = data;
585     th->detached = detached;
586     th->killable = false; /* not until vlc_entry() ! */
587     th->killed = false;
588     th->cleaners = NULL;
589
590     HANDLE hThread;
591 #ifndef UNDER_CE
592     /* When using the MSVCRT C library you have to use the _beginthreadex
593      * function instead of CreateThread, otherwise you'll end up with
594      * memory leaks and the signal functions not working (see Microsoft
595      * Knowledge Base, article 104641) */
596     uintptr_t h;
597
598     h = _beginthreadex (NULL, 0, vlc_entry, th, CREATE_SUSPENDED, NULL);
599     if (h == 0)
600     {
601         int err = errno;
602         free (th);
603         return err;
604     }
605     hThread = (HANDLE)h;
606
607 #else
608     th->cancel_event = CreateEvent (NULL, FALSE, FALSE, NULL);
609     if (th->cancel_event == NULL)
610     {
611         free (th);
612         return ENOMEM;
613     }
614
615     /* Not sure if CREATE_SUSPENDED + ResumeThread() is any useful on WinCE.
616      * Thread handles act up, too. */
617     hThread = CreateThread (NULL, 128*1024, vlc_entry, th,
618                             CREATE_SUSPENDED, NULL);
619     if (hThread == NULL)
620     {
621         CloseHandle (th->cancel_event);
622         free (th);
623         return ENOMEM;
624     }
625
626 #endif
627     /* Thread is suspended, so we can safely set th->id */
628     th->id = hThread;
629     if (p_handle != NULL)
630         *p_handle = th;
631
632     if (priority)
633         SetThreadPriority (hThread, priority);
634
635     ResumeThread (hThread);
636
637     return 0;
638 }
639
640 int vlc_clone (vlc_thread_t *p_handle, void *(*entry) (void *),
641                 void *data, int priority)
642 {
643     return vlc_clone_attr (p_handle, false, entry, data, priority);
644 }
645
646 void vlc_join (vlc_thread_t th, void **result)
647 {
648     do
649         vlc_testcancel ();
650     while (vlc_WaitForSingleObject (th->id, INFINITE)
651                                                         == WAIT_IO_COMPLETION);
652
653     if (result != NULL)
654         *result = th->data;
655     CloseHandle (th->id);
656 #ifdef UNDER_CE
657     CloseHandle (th->cancel_event);
658 #endif
659     free (th);
660 }
661
662 int vlc_clone_detach (vlc_thread_t *p_handle, void *(*entry) (void *),
663                       void *data, int priority)
664 {
665     vlc_thread_t th;
666     if (p_handle == NULL)
667         p_handle = &th;
668
669     return vlc_clone_attr (p_handle, true, entry, data, priority);
670 }
671
672 int vlc_set_priority (vlc_thread_t th, int priority)
673 {
674     if (!SetThreadPriority (th->id, priority))
675         return VLC_EGENERIC;
676     return VLC_SUCCESS;
677 }
678
679 /*** Thread cancellation ***/
680
681 /* APC procedure for thread cancellation */
682 static void CALLBACK vlc_cancel_self (ULONG_PTR self)
683 {
684     struct vlc_thread *th = (void *)self;
685
686     if (likely(th != NULL))
687         th->killed = true;
688 }
689
690 void vlc_cancel (vlc_thread_t th)
691 {
692 #ifndef UNDER_CE
693     QueueUserAPC (vlc_cancel_self, th->id, (uintptr_t)th);
694 #else
695     SetEvent (th->cancel_event);
696 #endif
697 }
698
699 int vlc_savecancel (void)
700 {
701     struct vlc_thread *th = vlc_threadvar_get (thread_key);
702     if (th == NULL)
703         return false; /* Main thread - cannot be cancelled anyway */
704
705     int state = th->killable;
706     th->killable = false;
707     return state;
708 }
709
710 void vlc_restorecancel (int state)
711 {
712     struct vlc_thread *th = vlc_threadvar_get (thread_key);
713     assert (state == false || state == true);
714
715     if (th == NULL)
716         return; /* Main thread - cannot be cancelled anyway */
717
718     assert (!th->killable);
719     th->killable = state != 0;
720 }
721
722 void vlc_testcancel (void)
723 {
724     struct vlc_thread *th = vlc_threadvar_get (thread_key);
725     if (th == NULL)
726         return; /* Main thread - cannot be cancelled anyway */
727
728     if (th->killable && th->killed)
729     {
730         for (vlc_cleanup_t *p = th->cleaners; p != NULL; p = p->next)
731              p->proc (p->data);
732
733         th->data = NULL; /* TODO: special value? */
734         vlc_thread_cleanup (th);
735 #ifndef UNDER_CE
736         _endthreadex(0);
737 #else
738         ExitThread(0);
739 #endif
740     }
741 }
742
743 void vlc_control_cancel (int cmd, ...)
744 {
745     /* NOTE: This function only modifies thread-specific data, so there is no
746      * need to lock anything. */
747     va_list ap;
748
749     struct vlc_thread *th = vlc_threadvar_get (thread_key);
750     if (th == NULL)
751         return; /* Main thread - cannot be cancelled anyway */
752
753     va_start (ap, cmd);
754     switch (cmd)
755     {
756         case VLC_CLEANUP_PUSH:
757         {
758             /* cleaner is a pointer to the caller stack, no need to allocate
759              * and copy anything. As a nice side effect, this cannot fail. */
760             vlc_cleanup_t *cleaner = va_arg (ap, vlc_cleanup_t *);
761             cleaner->next = th->cleaners;
762             th->cleaners = cleaner;
763             break;
764         }
765
766         case VLC_CLEANUP_POP:
767         {
768             th->cleaners = th->cleaners->next;
769             break;
770         }
771     }
772     va_end (ap);
773 }
774
775
776 /*** Clock ***/
777 mtime_t mdate (void)
778 {
779     /* We don't need the real date, just the value of a high precision timer */
780     LARGE_INTEGER counter, freq;
781     if (!QueryPerformanceCounter (&counter)
782      || !QueryPerformanceFrequency (&freq))
783         abort();
784
785     /* Convert to from (1/freq) to microsecond resolution */
786     /* We need to split the division to avoid 63-bits overflow */
787     lldiv_t d = lldiv (counter.QuadPart, freq.QuadPart);
788
789     return (d.quot * 1000000) + ((d.rem * 1000000) / freq.QuadPart);
790 }
791
792 #undef mwait
793 void mwait (mtime_t deadline)
794 {
795     mtime_t delay;
796
797     vlc_testcancel();
798     while ((delay = (deadline - mdate())) > 0)
799     {
800         delay /= 1000;
801         if (unlikely(delay > 0x7fffffff))
802             delay = 0x7fffffff;
803         vlc_Sleep (delay);
804         vlc_testcancel();
805     }
806 }
807
808 #undef msleep
809 void msleep (mtime_t delay)
810 {
811     mwait (mdate () + delay);
812 }
813
814
815 /*** Timers ***/
816 struct vlc_timer
817 {
818 #ifndef UNDER_CE
819     HANDLE handle;
820 #else
821     unsigned id;
822     unsigned interval;
823 #endif
824     void (*func) (void *);
825     void *data;
826 };
827
828 #ifndef UNDER_CE
829 static void CALLBACK vlc_timer_do (void *val, BOOLEAN timeout)
830 {
831     struct vlc_timer *timer = val;
832
833     assert (timeout);
834     timer->func (timer->data);
835 }
836 #else
837 static void CALLBACK vlc_timer_do (unsigned timer_id, unsigned msg,
838                                    DWORD_PTR user, DWORD_PTR unused1,
839                                    DWORD_PTR unused2)
840 {
841     struct vlc_timer *timer = (struct vlc_timer *) user;
842     assert (timer_id == timer->id);
843     (void) msg;
844     (void) unused1;
845     (void) unused2;
846
847     timer->func (timer->data);
848
849     if (timer->interval)
850     {
851         mtime_t interval = timer->interval * 1000;
852         vlc_timer_schedule (timer, false, interval, interval);
853     }
854 }
855 #endif
856
857 int vlc_timer_create (vlc_timer_t *id, void (*func) (void *), void *data)
858 {
859     struct vlc_timer *timer = malloc (sizeof (*timer));
860
861     if (timer == NULL)
862         return ENOMEM;
863     timer->func = func;
864     timer->data = data;
865 #ifndef UNDER_CE
866     timer->handle = INVALID_HANDLE_VALUE;
867 #else
868     timer->id = 0;
869     timer->interval = 0;
870 #endif
871     *id = timer;
872     return 0;
873 }
874
875 void vlc_timer_destroy (vlc_timer_t timer)
876 {
877 #ifndef UNDER_CE
878     if (timer->handle != INVALID_HANDLE_VALUE)
879         DeleteTimerQueueTimer (NULL, timer->handle, INVALID_HANDLE_VALUE);
880 #else
881     if (timer->id)
882         timeKillEvent (timer->id);
883     /* FIXME: timers that have not yet completed will trigger use-after-free */
884 #endif
885     free (timer);
886 }
887
888 void vlc_timer_schedule (vlc_timer_t timer, bool absolute,
889                          mtime_t value, mtime_t interval)
890 {
891 #ifndef UNDER_CE
892     if (timer->handle != INVALID_HANDLE_VALUE)
893     {
894         DeleteTimerQueueTimer (NULL, timer->handle, NULL);
895         timer->handle = INVALID_HANDLE_VALUE;
896     }
897 #else
898     if (timer->id)
899     {
900         timeKillEvent (timer->id);
901         timer->id = 0;
902         timer->interval = 0;
903     }
904 #endif
905     if (value == 0)
906         return; /* Disarm */
907
908     if (absolute)
909         value -= mdate ();
910     value = (value + 999) / 1000;
911     interval = (interval + 999) / 1000;
912
913 #ifndef UNDER_CE
914     if (!CreateTimerQueueTimer (&timer->handle, NULL, vlc_timer_do, timer,
915                                 value, interval, WT_EXECUTEDEFAULT))
916 #else
917     TIMECAPS caps;
918     timeGetDevCaps (&caps, sizeof(caps));
919
920     unsigned delay = value;
921     delay = __MAX(delay, caps.wPeriodMin);
922     delay = __MIN(delay, caps.wPeriodMax);
923
924     unsigned event = TIME_ONESHOT;
925
926     if (interval == delay)
927         event = TIME_PERIODIC;
928     else if (interval)
929         timer->interval = interval;
930
931     timer->id = timeSetEvent (delay, delay / 20, vlc_timer_do, (DWORD) timer,
932                               event);
933     if (!timer->id)
934 #endif
935         abort ();
936 }
937
938 unsigned vlc_timer_getoverrun (vlc_timer_t timer)
939 {
940     (void)timer;
941     return 0;
942 }
943
944
945 /*** CPU ***/
946 unsigned vlc_GetCPUCount (void)
947 {
948 #ifndef UNDER_CE
949     DWORD process;
950     DWORD system;
951
952     if (GetProcessAffinityMask (GetCurrentProcess(), &process, &system))
953         return popcount (system);
954 #endif
955      return 1;
956 }