Changeset 8645 in vbox for trunk/src/VBox/Runtime/common/misc

Timestamp:

May 7, 2008 11:01:00 AM (17 years ago)

Author:

vboxsync

Message:

Added lock counts to the threads.

File:

• trunk/src/VBox/Runtime/common/misc/thread.cpp (modified) (7 diffs)

trunk/src/VBox/Runtime/common/misc/thread.cpp

@@ -199 +199 @@
 #ifdef IN_RING3
 
-inline void rtThreadLockRW(void)
+DECLINLINE(void) rtThreadLockRW(void)
 {
     if (g_ThreadRWSem == NIL_RTSEMRW)
@@ -208 +208 @@
 
 
-inline void rtThreadLockRD(void)
+DECLINLINE(void) rtThreadLockRD(void)
 {
     if (g_ThreadRWSem == NIL_RTSEMRW)
@@ -217 +217 @@
 
 
-inline void rtThreadUnLockRW(void)
+DECLINLINE(void) rtThreadUnLockRW(void)
 {
     int rc = RTSemRWReleaseWrite(g_ThreadRWSem);
@@ -224 +224 @@
 
 
-inline void rtThreadUnLockRD(void)
+DECLINLINE(void) rtThreadUnLockRD(void)
 {
     int rc = RTSemRWReleaseRead(g_ThreadRWSem);
@@ -244 +244 @@
     /*
      * Allocate and insert the thread.
+     * (It is vital that rtThreadNativeAdopt updates the TLS before
+     * we try inserting the thread because of locking.)
      */
     int rc = VERR_NO_MEMORY;
@@ -1054 +1056 @@
 
 /**
+ * Gets the number of write locks and critical sections the specified
+ * thread owns.
+ *
+ * This number does not include any nested lock/critect entries.
+ *
+ * Note that it probably will return 0 for non-strict builds since
+ * release builds doesn't do unnecessary diagnostic counting like this.
+ *
+ * @returns Number of locks on success (0+) and VERR_INVALID_HANDLER on failure
+ * @param   Thread          The thread we're inquiring about.
+ */
+RTDECL(int32_t) RTThreadGetWriteLockCount(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    if (!pThread)
+        return VERR_INVALID_HANDLE;
+    int32_t cWriteLocks = ASMAtomicReadS32(&pThread->cWriteLocks);
+    rtThreadRelease(pThread);
+    return cWriteLocks;
+}
+
+
+/**
+ * Works the THREADINT::cWriteLocks member, mostly internal.
+ *
+ * @param   Thread      The current thread.
+ */
+RTDECL(void) RTThreadWriteLockInc(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    Assert(pThread);
+    ASMAtomicIncS32(&pThread->cWriteLocks);
+    rtThreadRelease(pThread);
+}
+
+
+/**
+ * Works the THREADINT::cWriteLocks member, mostly internal.
+ *
+ * @param   Thread      The current thread.
+ */
+RTDECL(void) RTThreadWriteLockDec(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    Assert(pThread);
+    ASMAtomicDecS32(&pThread->cWriteLocks);
+    rtThreadRelease(pThread);
+}
+
+
+/**
+ * Gets the number of read locks the specified thread owns.
+ *
+ * Note that nesting read lock entry will be included in the
+ * total sum. And that it probably will return 0 for non-strict
+ * builds since release builds doesn't do unnecessary diagnostic
+ * counting like this.
+ *
+ * @returns Number of read locks on success (0+) and VERR_INVALID_HANDLER on failure
+ * @param   Thread          The thread we're inquiring about.
+ */
+RTDECL(int32_t) RTThreadGetReadLockCount(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    if (!pThread)
+        return VERR_INVALID_HANDLE;
+    int32_t cReadLocks = ASMAtomicReadS32(&pThread->cReadLocks);
+    rtThreadRelease(pThread);
+    return cReadLocks;
+}
+
+
+/**
+ * Works the THREADINT::cReadLocks member.
+ *
+ * @param   Thread      The current thread.
+ */
+RTDECL(void) RTThreadReadLockInc(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    Assert(pThread);
+    ASMAtomicIncS32(&pThread->cReadLocks);
+    rtThreadRelease(pThread);
+}
+
+
+/**
+ * Works the THREADINT::cReadLocks member.
+ *
+ * @param   Thread      The current thread.
+ */
+RTDECL(void) RTThreadReadLockDec(RTTHREAD Thread)
+{
+    PRTTHREADINT pThread = rtThreadGet(Thread);
+    Assert(pThread);
+    ASMAtomicDecS32(&pThread->cReadLocks);
+    rtThreadRelease(pThread);
+}
+
+
+
+
+
+/**
  * Recalculates scheduling attributes for the the default process
  * priority using the specified priority type for the calling thread.
@@ -1062 +1168 @@
  *
  * @returns iprt status code.
+ * @remarks Will only work for strict builds.
  */
 int rtThreadDoCalcDefaultPriority(RTTHREADTYPE enmType)