xref: /onnv-gate/usr/src/lib/libkmsagent/common/AutoMutex.h (revision 12720:3db6e0082404)

/*
 * 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