AutoHandleRooter.h

#ifndef _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_
#define _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_

#include <IHandleSys.h>

class AutoHandleRooter
{
public:
        AutoHandleRooter(Handle_t hndl)
        {
                if (hndl != BAD_HANDLE)
                        this->hndl = handlesys->FastCloneHandle(hndl);
                else
                        this->hndl = BAD_HANDLE;
        }

        ~AutoHandleRooter()
        {
                if (hndl != BAD_HANDLE)
                {
                        HandleSecurity sec(g_pCoreIdent, g_pCoreIdent);
                        handlesys->FreeHandle(hndl, &sec);
                }
        }

        Handle_t getHandle()
        {
                return hndl;
        }
private:
        Handle_t hndl;
};

class AutoHandleCloner
{
        Handle_t original;
        HandleSecurity sec;
        AutoHandleRooter ahr;
public:
        AutoHandleCloner(Handle_t original, HandleSecurity sec)
          : original(original), sec(sec), ahr(original)
        {
        }

        ~AutoHandleCloner()
        {
                if (original != BAD_HANDLE)
                {
                        handlesys->FreeHandle(original, &sec);
                }
        }

        Handle_t getClone()
        {
                return ahr.getHandle();
        }
};

#endif /* _INCLUDE_SOURCEMOD_AUTO_HANDLE_ROOTER_H_ */