1 /* 2 * Copyright (c) 2015 The WebM project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef VPX_PORTS_VPX_ONCE_H_ 12 #define VPX_PORTS_VPX_ONCE_H_ 13 14 #include "vpx_config.h" 15 16 /* Implement a function wrapper to guarantee initialization 17 * thread-safety for library singletons. 18 * 19 * NOTE: These functions use static locks, and can only be 20 * used with one common argument per compilation unit. So 21 * 22 * file1.c: 23 * vpx_once(foo); 24 * ... 25 * vpx_once(foo); 26 * 27 * file2.c: 28 * vpx_once(bar); 29 * 30 * will ensure foo() and bar() are each called only once, but in 31 * 32 * file1.c: 33 * vpx_once(foo); 34 * vpx_once(bar): 35 * 36 * bar() will never be called because the lock is used up 37 * by the call to foo(). 38 */ 39 40 #if CONFIG_MULTITHREAD && defined(_WIN32) 41 #include <windows.h> 42 #include <stdlib.h> 43 /* Declare a per-compilation-unit state variable to track the progress 44 * of calling func() only once. This must be at global scope because 45 * local initializers are not thread-safe in MSVC prior to Visual 46 * Studio 2015. 47 * 48 * As a static, once_state will be zero-initialized as program start. 49 */ 50 static LONG once_state; 51 static void once(void (*func)(void)) { 52 /* Try to advance once_state from its initial value of 0 to 1. 53 * Only one thread can succeed in doing so. 54 */ 55 if (InterlockedCompareExchange(&once_state, 1, 0) == 0) { 56 /* We're the winning thread, having set once_state to 1. 57 * Call our function. */ 58 func(); 59 /* Now advance once_state to 2, unblocking any other threads. */ 60 InterlockedIncrement(&once_state); 61 return; 62 } 63 64 /* We weren't the winning thread, but we want to block on 65 * the state variable so we don't return before func() 66 * has finished executing elsewhere. 67 * 68 * Try to advance once_state from 2 to 2, which is only possible 69 * after the winning thead advances it from 1 to 2. 70 */ 71 while (InterlockedCompareExchange(&once_state, 2, 2) != 2) { 72 /* State isn't yet 2. Try again. 73 * 74 * We are used for singleton initialization functions, 75 * which should complete quickly. Contention will likewise 76 * be rare, so it's worthwhile to use a simple but cpu- 77 * intensive busy-wait instead of successive backoff, 78 * waiting on a kernel object, or another heavier-weight scheme. 79 * 80 * We can at least yield our timeslice. 81 */ 82 Sleep(0); 83 } 84 85 /* We've seen once_state advance to 2, so we know func() 86 * has been called. And we've left once_state as we found it, 87 * so other threads will have the same experience. 88 * 89 * It's safe to return now. 90 */ 91 return; 92 } 93 94 #elif CONFIG_MULTITHREAD && defined(__OS2__) 95 #define INCL_DOS 96 #include <os2.h> 97 static void once(void (*func)(void)) { 98 static int done; 99 100 /* If the initialization is complete, return early. */ 101 if (done) return; 102 103 /* Causes all other threads in the process to block themselves 104 * and give up their time slice. 105 */ 106 DosEnterCritSec(); 107 108 if (!done) { 109 func(); 110 done = 1; 111 } 112 113 /* Restores normal thread dispatching for the current process. */ 114 DosExitCritSec(); 115 } 116 117 #elif CONFIG_MULTITHREAD && HAVE_PTHREAD_H 118 #include <pthread.h> 119 static void once(void (*func)(void)) { 120 static pthread_once_t lock = PTHREAD_ONCE_INIT; 121 pthread_once(&lock, func); 122 } 123 124 #else 125 /* No-op version that performs no synchronization. *_rtcd() is idempotent, 126 * so as long as your platform provides atomic loads/stores of pointers 127 * no synchronization is strictly necessary. 128 */ 129 130 static void once(void (*func)(void)) { 131 static int done; 132 133 if (!done) { 134 func(); 135 done = 1; 136 } 137 } 138 #endif 139 140 #endif // VPX_PORTS_VPX_ONCE_H_ 141