1 // Copyright 2012 the V8 project authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 // emulates google3/base/once.h
6 //
7 // This header is intended to be included only by v8's internal code. Users
8 // should not use this directly.
9 //
10 // This is basically a portable version of pthread_once().
11 //
12 // This header declares:
13 // * A type called OnceType.
14 // * A macro V8_DECLARE_ONCE() which declares a (global) variable of type
15 //   OnceType.
16 // * A function CallOnce(OnceType* once, void (*init_func)()).
17 //   This function, when invoked multiple times given the same OnceType object,
18 //   will invoke init_func on the first call only, and will make sure none of
19 //   the calls return before that first call to init_func has finished.
20 //
21 // Additionally, the following features are supported:
22 // * A macro V8_ONCE_INIT which is expanded into the expression used to
23 //   initialize a OnceType. This is only useful when clients embed a OnceType
24 //   into a structure of their own and want to initialize it statically.
25 // * The user can provide a parameter which CallOnce() forwards to the
26 //   user-provided function when it is called. Usage example:
27 //     CallOnce(&my_once, &MyFunctionExpectingIntArgument, 10);
28 // * This implementation guarantees that OnceType is a POD (i.e. no static
29 //   initializer generated).
30 //
31 // This implements a way to perform lazy initialization.  It's more efficient
32 // than using mutexes as no lock is needed if initialization has already
33 // happened.
34 //
35 // Example usage:
36 //   void Init();
37 //   V8_DECLARE_ONCE(once_init);
38 //
39 //   // Calls Init() exactly once.
40 //   void InitOnce() {
41 //     CallOnce(&once_init, &Init);
42 //   }
43 //
44 // Note that if CallOnce() is called before main() has begun, it must
45 // only be called by the thread that will eventually call main() -- that is,
46 // the thread that performs dynamic initialization.  In general this is a safe
47 // assumption since people don't usually construct threads before main() starts,
48 // but it is technically not guaranteed.  Unfortunately, Win32 provides no way
49 // whatsoever to statically-initialize its synchronization primitives, so our
50 // only choice is to assume that dynamic initialization is single-threaded.
51 
52 #ifndef V8_BASE_ONCE_H_
53 #define V8_BASE_ONCE_H_
54 
55 #include <stddef.h>
56 
57 #include "src/base/atomicops.h"
58 
59 namespace v8 {
60 namespace base {
61 
62 typedef AtomicWord OnceType;
63 
64 #define V8_ONCE_INIT 0
65 
66 #define V8_DECLARE_ONCE(NAME) ::v8::base::OnceType NAME
67 
68 enum {
69   ONCE_STATE_UNINITIALIZED = 0,
70   ONCE_STATE_EXECUTING_FUNCTION = 1,
71   ONCE_STATE_DONE = 2
72 };
73 
74 typedef void (*NoArgFunction)();
75 typedef void (*PointerArgFunction)(void* arg);
76 
77 template <typename T>
78 struct OneArgFunction {
79   typedef void (*type)(T);
80 };
81 
82 void CallOnceImpl(OnceType* once, PointerArgFunction init_func, void* arg);
83 
CallOnce(OnceType * once,NoArgFunction init_func)84 inline void CallOnce(OnceType* once, NoArgFunction init_func) {
85   if (Acquire_Load(once) != ONCE_STATE_DONE) {
86     CallOnceImpl(once, reinterpret_cast<PointerArgFunction>(init_func), NULL);
87   }
88 }
89 
90 
91 template <typename Arg>
CallOnce(OnceType * once,typename OneArgFunction<Arg * >::type init_func,Arg * arg)92 inline void CallOnce(OnceType* once,
93     typename OneArgFunction<Arg*>::type init_func, Arg* arg) {
94   if (Acquire_Load(once) != ONCE_STATE_DONE) {
95     CallOnceImpl(once, reinterpret_cast<PointerArgFunction>(init_func),
96         static_cast<void*>(arg));
97   }
98 }
99 
100 }  // namespace base
101 }  // namespace v8
102 
103 #endif  // V8_BASE_ONCE_H_
104