1 files changed, 157 insertions, 0 deletions
diff --git a/usr/src/lib/libkmsagent/common/AutoMutex.h b/usr/src/lib/libkmsagent/common/AutoMutex.h
new file mode 100644
index 0000000000..31943e8559
--- /dev/null
+++ b/usr/src/lib/libkmsagent/common/AutoMutex.h
@@ -0,0 +1,157 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+ * or http://www.opensolaris.org/os/licensing.
+ * See the License for the specific language governing permissions
+ * and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
+ */
+
+/*------------------------------------------------------------------------------
+ * Module           : AutoMutex.h
+ -----------------------------------------------------------------------------*/
+
+#ifndef AutoMutex_h
+#define AutoMutex_h
+
+#include "SYSCommon.h"
+
+class CAutoMutex 
+{
+public:
+
+    /*---------------------------------------------------------------------------
+     * Constructor:
+     *  Locks the given mutex handle.
+     *
+     * Input
+     * -----
+     *    i_hMutex        Mutex handle
+     *
+     * Output
+     * ------
+     *    (none)
+     *
+     * Return value       (none)
+     *
+     *--------------------------------------------------------------------------*/
+
+    CAutoMutex( K_MUTEX_HANDLE i_hMutex ) 
+        : m_hMutex( 0 ),
+          m_bLocked( false )
+    {
+        if ( i_hMutex )
+        {
+            Lock( i_hMutex );
+        }
+    }
+
+
+    /*---------------------------------------------------------------------------
+     * Destructor:
+     *  Unlocks this mutex.
+     *
+     * Input
+     * -----
+     *    (none)
+     *
+     * Output
+     * ------
+     *    (none)
+     *
+     * Return value       (none)
+     *
+     *--------------------------------------------------------------------------*/
+
+    virtual ~CAutoMutex() 
+    { 
+        if ( m_bLocked )
+        {
+            Unlock();
+        }
+    }
+
+    /*---------------------------------------------------------------------------
+     * Function: Lock
+     *
+     * Description:
+     *  Locks this mutex handle.  If i_hMutex is null, the handle passed to the
+     *  constructor will be used.  Fatals if there is no valid handle.
+     *
+     * Input
+     * -----
+     *    i_hMutex        Mutex handle to lock
+     *
+     * Output
+     * ------
+     *    (none)
+     *
+     * Return value       (none)
+     *
+     *--------------------------------------------------------------------------*/
+
+    void Lock( K_MUTEX_HANDLE i_hMutex = 0 )
+    {
+        FATAL_ASSERT( !m_bLocked );
+
+        if ( i_hMutex )
+        {
+            m_hMutex = i_hMutex;
+        }
+
+        FATAL_ASSERT( m_hMutex );        
+        K_LockMutex( m_hMutex );
+        m_bLocked = true;
+    }
+
+
+    /*---------------------------------------------------------------------------
+     * Function: Unlock
+     *
+     * Description:
+     *  Unlocks the mutex handle passed to the constructor or to a previous 
+     *  Lock call.  Fatals if the mutex is not locked.
+     *
+     * Input
+     * -----
+     *    (none)
+     *
+     * Output
+     * ------
+     *    (none)
+     *
+     * Return value       (none)
+     *
+     *--------------------------------------------------------------------------*/
+
+    void Unlock()
+    {
+        FATAL_ASSERT( m_bLocked );
+        FATAL_ASSERT( m_hMutex );
+        K_UnlockMutex( m_hMutex );
+        m_bLocked = false;
+    }
+
+private:
+    K_MUTEX_HANDLE m_hMutex;
+    bool m_bLocked;
+};
+
+
+#endif // AutoMutex_h