public/AutoHandleRooter.h
author Nicholas Hastings <skamonkey@gmail.com>
Thu Jul 04 11:28:32 2013 -0400 (2013-07-04)
changeset 3931 576fa42454dc
parent 3872 4c93f74cae51
permissions -rw-r--r--
Fixed build for real.
     1 /**
     2  * vim: set ts=4 sw=4 tw=99 noet :
     3  * =============================================================================
     4  * SourceMod
     5  * Copyright (C) 2004-2009 AlliedModders LLC.  All rights reserved.
     6  * =============================================================================
     7  *
     8  * This program is free software; you can redistribute it and/or modify it under
     9  * the terms of the GNU General Public License, version 3.0, as published by the
    10  * Free Software Foundation.
    11  * 
    12  * This program is distributed in the hope that it will be useful, but WITHOUT
    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
    15  * details.
    16  *
    17  * You should have received a copy of the GNU General Public License along with
    18  * this program.  If not, see <http://www.gnu.org/licenses/>.
    19  *
    20  * As a special exception, AlliedModders LLC gives you permission to link the
    21  * code of this program (as well as its derivative works) to "Half-Life 2," the
    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
    23  * by the Valve Corporation.  You must obey the GNU General Public License in
    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants
    25  * this exception to all derivative works.  AlliedModders LLC defines further
    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
    27  * or <http://www.sourcemod.net/license.php>.
    28  *
    29  * Version: $Id$
    30  */
    31 
    32 #ifndef _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_
    33 #define _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_
    34 
    35 #include <IHandleSys.h>
    36 
    37 class AutoHandleRooter
    38 {
    39 public:
    40 	AutoHandleRooter(Handle_t hndl)
    41 	{
    42 		if (hndl != BAD_HANDLE)
    43 			this->hndl = handlesys->FastCloneHandle(hndl);
    44 		else
    45 			this->hndl = BAD_HANDLE;
    46 	}
    47 
    48 	~AutoHandleRooter()
    49 	{
    50 		if (hndl != BAD_HANDLE)
    51 		{
    52 			HandleSecurity sec(g_pCoreIdent, g_pCoreIdent);
    53 			handlesys->FreeHandle(hndl, &sec);
    54 		}
    55 	}
    56 
    57 	Handle_t getHandle()
    58 	{
    59 		return hndl;
    60 	}
    61 private:
    62 	Handle_t hndl;
    63 };
    64 
    65 class AutoHandleCloner
    66 {
    67 	Handle_t original;
    68 	HandleSecurity sec;
    69 	AutoHandleRooter ahr;
    70 public:
    71 	AutoHandleCloner(Handle_t original, HandleSecurity sec)
    72 	  : original(original), sec(sec), ahr(original)
    73 	{
    74 	}
    75 
    76 	~AutoHandleCloner()
    77 	{
    78 		if (original != BAD_HANDLE)
    79 		{
    80 			handlesys->FreeHandle(original, &sec);
    81 		}
    82 	}
    83 
    84 	Handle_t getClone()
    85 	{
    86 		return ahr.getHandle();
    87 	}
    88 };
    89 
    90 #endif /* _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_ */
    91