1 | // Licensed to the .NET Foundation under one or more agreements. |
2 | // The .NET Foundation licenses this file to you under the MIT license. |
3 | // See the LICENSE file in the project root for more information. |
4 | // |
5 | // CRST.H |
6 | // |
7 | |
8 | // |
9 | // Debug-instrumented hierarchical critical sections. |
10 | // |
11 | // |
12 | // The hierarchy: |
13 | // -------------- |
14 | // The EE divides critical sections into numbered groups or "levels." |
15 | // Crsts that guard the lowest level data structures that don't |
16 | // use other services are grouped into the lowest-numbered levels. |
17 | // The higher-numbered levels are reserved for high-level crsts |
18 | // that guard broad swatches of code. Multiple groups can share the |
19 | // same number to indicate that they're disjoint (their locks will never |
20 | // nest.) |
21 | // |
22 | // The fundamental rule of the hierarchy that threads can only request |
23 | // a crst whose level is lower than any crst currently held by the thread. |
24 | // E.g. if a thread current holds a level-3 crst, he can try to enter |
25 | // a level-2 crst, but not a level-4 crst, nor a different level-3 |
26 | // crst. This prevents the cyclic dependencies that lead to deadlock. |
27 | // |
28 | // For debugging purposes Crsts are all also grouped by a type (e.g. |
29 | // CrstRemoting, the type of Crst used to synchronize certain remoting |
30 | // operations). Each type maps to one level (though a level may map to |
31 | // multiple types). The idea here is for the programmer to express Crst types |
32 | // and their dependencies (e.g. a CrstClassInit instance may be acquired |
33 | // while a CrstRemoting instance is already held) in a high level manner |
34 | // while an external script handles the mechanical process of assigning |
35 | // numerical levels to each type. See file:..\inc\CrstTypes.def for these high level |
36 | // type definitions. |
37 | // |
38 | // |
39 | // To create a crst: |
40 | // |
41 | // Crst *pcrst = new Crst(type); |
42 | // |
43 | // where "type" is one of the enums created in the auto-generated |
44 | // file:..\inc\CrstTypes.h header file (matching the definition in |
45 | // file:..\inc\CrstTypes.def). |
46 | // |
47 | // By default, crsts don't support nested enters by the same thread. If |
48 | // you need reentrancy, use the alternate form: |
49 | // |
50 | // Crst *pcrst = new Crst(type, TRUE); |
51 | // |
52 | // Since reentrancies never block the caller, they're allowed to |
53 | // "violate" the level ordering rule. |
54 | // |
55 | // |
56 | // To enter/leave a crst: |
57 | // ---------------------- |
58 | // |
59 | // |
60 | // pcrst->Enter(); |
61 | // pcrst->Leave(); |
62 | // |
63 | // An assertion will fire on Enter() if a thread attempts to take locks |
64 | // in the wrong order. |
65 | // |
66 | // Finally, a few DEBUG-only methods: |
67 | // |
68 | // To assert taking a crst won't violate level order: |
69 | // -------------------------------------------------- |
70 | // |
71 | // _ASSERTE(pcrst->IsSafeToTake()); |
72 | // |
73 | // This is a good line to put at the start of any function that |
74 | // enters a crst in some circumstances but not others. If it |
75 | // always enters the crst, it's not necessary to call IsSafeToTake() |
76 | // since Enter() does this for you. |
77 | // |
78 | // To assert that the current thread owns a crst: |
79 | // -------------------------------------------------- |
80 | // |
81 | // _ASSERTE(pcrst->OwnedByCurrentThread()); |
82 | |
83 | |
84 | |
85 | #ifndef __crst_h__ |
86 | #define __crst_h__ |
87 | |
88 | #include "util.hpp" |
89 | #include "debugmacros.h" |
90 | #include "log.h" |
91 | |
92 | #define ShutDown_Start 0x00000001 |
93 | #define ShutDown_Finalize1 0x00000002 |
94 | #define ShutDown_Finalize2 0x00000004 |
95 | #define ShutDown_Profiler 0x00000008 |
96 | #define ShutDown_COM 0x00000010 |
97 | #define ShutDown_SyncBlock 0x00000020 |
98 | #define ShutDown_IUnknown 0x00000040 |
99 | #define ShutDown_Phase2 0x00000080 |
100 | |
101 | #ifndef DACCESS_COMPILE |
102 | extern bool g_fProcessDetach; |
103 | extern DWORD g_fEEShutDown; |
104 | #endif |
105 | // Total count of Crst lock of the type (Shutdown) that are currently in use |
106 | extern Volatile<LONG> g_ShutdownCrstUsageCount; |
107 | extern Volatile<LONG> g_fForbidEnterEE; |
108 | |
109 | // The CRST. |
110 | class CrstBase |
111 | { |
112 | // The following classes and methods violate the requirement that Crst usage be |
113 | // exception-safe, or they satisfy that requirement using techniques other than |
114 | // Holder objects: |
115 | friend class Thread; |
116 | friend class ThreadStore; |
117 | friend class ThreadSuspend; |
118 | template <typename ELEMENT> |
119 | friend class ListLockBase; |
120 | template <typename ELEMENT> |
121 | friend class ListLockEntryBase; |
122 | //friend class CExecutionEngine; |
123 | friend struct SavedExceptionInfo; |
124 | friend void EEEnterCriticalSection(CRITSEC_COOKIE cookie); |
125 | friend void EELeaveCriticalSection(CRITSEC_COOKIE cookie); |
126 | friend class CodeVersionManager; |
127 | |
128 | friend class Debugger; |
129 | friend class Crst; |
130 | |
131 | #ifdef FEATURE_DBGIPC_TRANSPORT_VM |
132 | // The debugger transport code uses a holder for its Crst, but it needs to share the holder implementation |
133 | // with its right side code as well (which can't see the Crst implementation and actually uses a |
134 | // CRITICAL_SECTION as the base lock). So make DbgTransportSession a friend here so we can use Enter() and |
135 | // Leave() in order to build a shared holder class. |
136 | friend class DbgTransportLock; |
137 | #endif // FEATURE_DBGIPC_TRANSPORT_VM |
138 | |
139 | // PendingTypeLoadEntry acquires the lock during construction before anybody has a chance to see it to avoid |
140 | // level violations. |
141 | friend class PendingTypeLoadEntry; |
142 | |
143 | public: |
144 | #ifdef _DEBUG |
145 | enum NoLevelCheckFlag |
146 | { |
147 | CRST_NO_LEVEL_CHECK = 1, |
148 | CRST_LEVEL_CHECK = 0, |
149 | }; |
150 | #endif |
151 | |
152 | private: |
153 | // Some Crsts have a "shutdown" mode. |
154 | // A Crst in shutdown mode can only be taken / released by special |
155 | // (the helper / finalizer / shutdown) threads. Any other thread that tries to take |
156 | // the a "shutdown" crst will immediately release the Crst and instead just block forever. |
157 | // |
158 | // This prevents random threads from blocking the special threads from doing finalization on shutdown. |
159 | // |
160 | // Unfortunately, each Crst needs its own "shutdown" flag because we can't convert all the locks |
161 | // into shutdown locks at once. For eg, the TSL needs to suspend the runtime before |
162 | // converting to a shutdown lock. But it can't suspend the runtime while holding |
163 | // a UNSAFE_ANYMODE lock (such as the debugger-lock). So at least the debugger-lock |
164 | // and TSL need to be set separately. |
165 | // |
166 | // So for such Crsts, it's the caller's responsibility to detect if the crst is in |
167 | // shutdown mode, and if so, call this function after enter. |
168 | void ReleaseAndBlockForShutdownIfNotSpecialThread(); |
169 | |
170 | // Enter & Leave are deliberately private to force callers to use the |
171 | // Holder class. If you bypass the Holder class and access these members |
172 | // directly, your lock is not exception-safe. |
173 | // |
174 | // noLevelCheckFlag parameter lets you disable the crst level checking. This is |
175 | // very dangerous so it is only used when the constructor is the one performing |
176 | // the Enter (that attempt cannot possibly block since the current thread is |
177 | // the only one with a pointer to the crst.) |
178 | // |
179 | // For obvious reasons, this parameter must never be made public. |
180 | void Enter(INDEBUG(NoLevelCheckFlag noLevelCheckFlag = CRST_LEVEL_CHECK)); |
181 | void Leave(); |
182 | |
183 | void SpinEnter(); |
184 | |
185 | #ifndef DACCESS_COMPILE |
186 | DEBUG_NOINLINE static void AcquireLock(CrstBase *c) PUB { |
187 | WRAPPER_NO_CONTRACT; |
188 | ANNOTATION_SPECIAL_HOLDER_CALLER_NEEDS_DYNAMIC_CONTRACT; |
189 | c->Enter(); |
190 | } |
191 | |
192 | DEBUG_NOINLINE static void ReleaseLock(CrstBase *c) PUB { |
193 | WRAPPER_NO_CONTRACT; |
194 | ANNOTATION_SPECIAL_HOLDER_CALLER_NEEDS_DYNAMIC_CONTRACT; |
195 | c->Leave(); |
196 | } |
197 | |
198 | #else // DACCESS_COMPILE |
199 | |
200 | // in DAC builds, we don't actually acquire the lock, we just determine whether the LS |
201 | // already holds it. If so, we assume the data is inconsistent and throw an exception. |
202 | // Argument: |
203 | // input: c - the lock to be checked. |
204 | // Note: Throws |
205 | static void AcquireLock(CrstBase * c) PUB |
206 | { |
207 | SUPPORTS_DAC; |
208 | if (c->GetEnterCount() != 0) |
209 | { |
210 | ThrowHR(CORDBG_E_PROCESS_NOT_SYNCHRONIZED); |
211 | } |
212 | }; |
213 | |
214 | static void ReleaseLock(CrstBase *c) PUB |
215 | { |
216 | SUPPORTS_DAC; |
217 | }; |
218 | #endif // DACCESS_COMPILE |
219 | |
220 | public: |
221 | //----------------------------------------------------------------- |
222 | // Clean up critical section |
223 | // Safe to call multiple times or on non-initialized critical section |
224 | //----------------------------------------------------------------- |
225 | void Destroy(); |
226 | |
227 | #ifdef _DEBUG |
228 | //----------------------------------------------------------------- |
229 | // Check if attempting to take the lock would violate level order. |
230 | //----------------------------------------------------------------- |
231 | BOOL IsSafeToTake(); |
232 | // Checks that the lock can be taken |
233 | BOOL Debug_CanTake() |
234 | { |
235 | WRAPPER_NO_CONTRACT; |
236 | // Actually take the lock and release it immediatelly, that will do all the necessary checks |
237 | Enter(); |
238 | Leave(); |
239 | return TRUE; |
240 | } |
241 | void SetCantLeave(BOOL bSet) |
242 | { |
243 | LIMITED_METHOD_CONTRACT; |
244 | if (bSet) |
245 | FastInterlockIncrement(&m_cannotLeave); |
246 | else |
247 | { |
248 | _ASSERTE(m_cannotLeave); |
249 | FastInterlockDecrement(&m_cannotLeave); |
250 | } |
251 | }; |
252 | //----------------------------------------------------------------- |
253 | // Is the current thread the owner? |
254 | //----------------------------------------------------------------- |
255 | BOOL OwnedByCurrentThread() |
256 | { |
257 | WRAPPER_NO_CONTRACT; |
258 | #ifdef CROSSGEN_COMPILE |
259 | return TRUE; |
260 | #else |
261 | return m_holderthreadid.IsCurrentThread(); |
262 | #endif |
263 | } |
264 | |
265 | CrstBase *GetThreadsOwnedCrsts(); |
266 | void SetThreadsOwnedCrsts(CrstBase *pCrst); |
267 | |
268 | __declspec(noinline) EEThreadId GetHolderThreadId() |
269 | { |
270 | LIMITED_METHOD_CONTRACT; |
271 | return m_holderthreadid; |
272 | } |
273 | |
274 | #endif //_DEBUG |
275 | |
276 | //----------------------------------------------------------------- |
277 | // For clients who want to assert whether they are in or out of the |
278 | // region. |
279 | //----------------------------------------------------------------- |
280 | UINT GetEnterCount() |
281 | { |
282 | LIMITED_METHOD_DAC_CONTRACT; |
283 | #ifdef _DEBUG |
284 | return m_entercount; |
285 | #else |
286 | return 0; |
287 | #endif //_DEBUG |
288 | } |
289 | |
290 | protected: |
291 | |
292 | VOID InitWorker(INDEBUG_COMMA(CrstType crstType) CrstFlags flags); |
293 | |
294 | #ifdef _DEBUG |
295 | void DebugInit(CrstType crstType, CrstFlags flags); |
296 | void DebugDestroy(); |
297 | #endif |
298 | |
299 | union { |
300 | CRITICAL_SECTION m_criticalsection; |
301 | }; |
302 | |
303 | typedef enum |
304 | { |
305 | // Mask to indicate reserved flags |
306 | CRST_RESERVED_FLAGS_MASK = 0xC0000000, |
307 | // private flag to indicate initialized Crsts |
308 | CRST_INITIALIZED = 0x80000000, |
309 | // private flag to indicate Crst is OS Critical Section |
310 | CRST_OS_CRIT_SEC = 0x40000000, |
311 | // rest of the flags are CrstFlags |
312 | } CrstReservedFlags; |
313 | DWORD m_dwFlags; // Re-entrancy and same level |
314 | #ifdef _DEBUG |
315 | UINT m_entercount; // # of unmatched Enters. |
316 | CrstType m_crstType; // Type enum (should have a descriptive name for debugging) |
317 | const char *m_tag; // Stringized form of the tag for easy debugging |
318 | int m_crstlevel; // what level is the crst in? |
319 | EEThreadId m_holderthreadid; // current holder (or NULL) |
320 | CrstBase *m_next; // link for global linked list |
321 | CrstBase *m_prev; // link for global linked list |
322 | Volatile<LONG> m_cannotLeave; |
323 | |
324 | // Check for dead lock situation. |
325 | ULONG m_countNoTriggerGC; |
326 | |
327 | void PostEnter (); |
328 | void PreEnter (); |
329 | void PreLeave (); |
330 | #endif //_DEBUG |
331 | |
332 | private: |
333 | |
334 | void SetOSCritSec () |
335 | { |
336 | m_dwFlags |= CRST_OS_CRIT_SEC; |
337 | } |
338 | void ResetOSCritSec () |
339 | { |
340 | m_dwFlags &= ~CRST_OS_CRIT_SEC; |
341 | } |
342 | BOOL IsOSCritSec () |
343 | { |
344 | return m_dwFlags & CRST_OS_CRIT_SEC; |
345 | } |
346 | void SetCrstInitialized() |
347 | { |
348 | m_dwFlags |= CRST_INITIALIZED; |
349 | } |
350 | |
351 | BOOL IsCrstInitialized() |
352 | { |
353 | return m_dwFlags & CRST_INITIALIZED; |
354 | } |
355 | |
356 | BOOL CanBeTakenDuringShutdown() |
357 | { |
358 | return m_dwFlags & CRST_TAKEN_DURING_SHUTDOWN; |
359 | } |
360 | |
361 | void SetFlags(CrstFlags f) |
362 | { |
363 | WRAPPER_NO_CONTRACT; |
364 | _ASSERTE(((CrstFlags)(f & ~CRST_RESERVED_FLAGS_MASK)) == f); |
365 | m_dwFlags = (f & ~CRST_RESERVED_FLAGS_MASK) | (m_dwFlags & CRST_RESERVED_FLAGS_MASK); |
366 | } |
367 | |
368 | void ResetFlags() // resets the reserved and the CrstFlags |
369 | { |
370 | m_dwFlags = 0; |
371 | } |
372 | // ------------------------------- Holders ------------------------------ |
373 | public: |
374 | // |
375 | // CrstHolder is optimized for the common use that takes the lock in constructor |
376 | // and releases it in destructor. Users that require all Holder features |
377 | // can use CrstHolderWithState. |
378 | // |
379 | class CrstHolder |
380 | { |
381 | CrstBase * m_pCrst; |
382 | |
383 | public: |
384 | inline CrstHolder(CrstBase * pCrst) |
385 | : m_pCrst(pCrst) |
386 | { |
387 | WRAPPER_NO_CONTRACT; |
388 | AcquireLock(pCrst); |
389 | } |
390 | |
391 | inline ~CrstHolder() |
392 | { |
393 | WRAPPER_NO_CONTRACT; |
394 | |
395 | VALIDATE_HOLDER_STACK_CONSUMPTION_FOR_TYPE(HSV_ValidateMinimumStackReq); |
396 | ReleaseLock(m_pCrst); |
397 | } |
398 | }; |
399 | |
400 | // Note that the holders for CRSTs are used in extremely low stack conditions. Because of this, they |
401 | // aren't allowed to use more than HOLDER_CODE_MINIMUM_STACK_LIMIT pages of stack. |
402 | typedef DacHolder<CrstBase *, CrstBase::AcquireLock, CrstBase::ReleaseLock, 0, CompareDefault, HSV_ValidateMinimumStackReq> CrstHolderWithState; |
403 | |
404 | // We have some situations where we're already holding a lock, and we need to release and reacquire the lock across a window. |
405 | // This is a dangerous construct because the backout code can block. |
406 | // Generally, it's better to use a regular CrstHolder, and then use the Release() / Acquire() methods on it. |
407 | // This just exists to convert legacy OS Critical Section patterns over to holders. |
408 | typedef DacHolder<CrstBase *, CrstBase::ReleaseLock, CrstBase::AcquireLock, 0, CompareDefault, HSV_ValidateMinimumStackReq> UnsafeCrstInverseHolder; |
409 | }; |
410 | |
411 | typedef CrstBase::CrstHolder CrstHolder; |
412 | typedef CrstBase::CrstHolderWithState CrstHolderWithState; |
413 | |
414 | |
415 | // The CRST. |
416 | class Crst : public CrstBase |
417 | { |
418 | public: |
419 | void *operator new(size_t size) |
420 | { |
421 | WRAPPER_NO_CONTRACT; |
422 | return new BYTE[size]; |
423 | } |
424 | |
425 | private: |
426 | // Do not use inplace operator new on Crst. A wrong destructor would be called if the constructor fails. |
427 | // Use CrstStatic or CrstExplicitInit instead of the inplace operator new. |
428 | void *operator new(size_t size, void *pInPlace); |
429 | |
430 | public: |
431 | |
432 | #ifndef DACCESS_COMPILE |
433 | |
434 | //----------------------------------------------------------------- |
435 | // Constructor. |
436 | //----------------------------------------------------------------- |
437 | Crst(CrstType crstType, CrstFlags flags = CRST_DEFAULT) |
438 | { |
439 | WRAPPER_NO_CONTRACT; |
440 | |
441 | // throw away the debug-only parameter in retail |
442 | InitWorker(INDEBUG_COMMA(crstType) flags); |
443 | } |
444 | |
445 | //----------------------------------------------------------------- |
446 | // Destructor. |
447 | //----------------------------------------------------------------- |
448 | ~Crst() |
449 | { |
450 | WRAPPER_NO_CONTRACT; |
451 | |
452 | Destroy(); |
453 | }; |
454 | |
455 | #else |
456 | |
457 | Crst(CrstType crstType, CrstFlags flags = CRST_DEFAULT) { |
458 | LIMITED_METHOD_CONTRACT; |
459 | }; |
460 | |
461 | #endif |
462 | |
463 | Crst() { |
464 | LIMITED_METHOD_CONTRACT; |
465 | } |
466 | }; |
467 | |
468 | typedef DPTR(Crst) PTR_Crst; |
469 | |
470 | /* to be used as static variable - no constructor/destructor, assumes zero |
471 | initialized memory */ |
472 | class CrstStatic : public CrstBase |
473 | { |
474 | public: |
475 | VOID Init(CrstType crstType, CrstFlags flags = CRST_DEFAULT) |
476 | { |
477 | WRAPPER_NO_CONTRACT; |
478 | |
479 | _ASSERTE((flags & CRST_INITIALIZED) == 0); |
480 | |
481 | // throw away the debug-only parameter in retail |
482 | InitWorker(INDEBUG_COMMA(crstType) flags); |
483 | } |
484 | |
485 | bool InitNoThrow(CrstType crstType, CrstFlags flags = CRST_DEFAULT) |
486 | { |
487 | CONTRACTL { |
488 | NOTHROW; |
489 | } CONTRACTL_END; |
490 | |
491 | _ASSERTE((flags & CRST_INITIALIZED) == 0); |
492 | |
493 | bool fSuccess = false; |
494 | |
495 | EX_TRY |
496 | { |
497 | // throw away the debug-only parameter in retail |
498 | InitWorker(INDEBUG_COMMA(crstType) flags); |
499 | fSuccess = true; |
500 | } |
501 | EX_CATCH |
502 | { |
503 | } |
504 | EX_END_CATCH(SwallowAllExceptions) |
505 | |
506 | return fSuccess; |
507 | } |
508 | }; |
509 | |
510 | /* to be used as regular variable when a explicit call to Init method is needed */ |
511 | class CrstExplicitInit : public CrstStatic |
512 | { |
513 | public: |
514 | CrstExplicitInit() { |
515 | m_dwFlags = 0; |
516 | } |
517 | ~CrstExplicitInit() { |
518 | #ifndef DACCESS_COMPILE |
519 | Destroy(); |
520 | #endif |
521 | } |
522 | }; |
523 | |
524 | __inline BOOL IsOwnerOfCrst(LPVOID lock) |
525 | { |
526 | WRAPPER_NO_CONTRACT; |
527 | |
528 | #ifdef _DEBUG |
529 | return ((Crst*)lock)->OwnedByCurrentThread(); |
530 | #else |
531 | // This function should not be called on free build. |
532 | DebugBreak(); |
533 | return TRUE; |
534 | #endif |
535 | } |
536 | |
537 | #ifdef TEST_DATA_CONSISTENCY |
538 | // used for test purposes. Determines if a crst is held. |
539 | void DebugTryCrst(CrstBase * pLock); |
540 | #endif |
541 | #endif // __crst_h__ |
542 | |
543 | |
544 | |