dlls/msi/handle.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2002-2004 Mike McCormack for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winreg.h"
  26 #include "shlwapi.h"
  27 #include "wine/debug.h"
  28 #include "msi.h"
  29 #include "msiquery.h"
  30 #include "msipriv.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  33
  34 static CRITICAL_SECTION MSI_handle_cs;
  35 static CRITICAL_SECTION_DEBUG MSI_handle_cs_debug =
  36 {
  37     0, 0, &MSI_handle_cs,
  38     { &MSI_handle_cs_debug.ProcessLocksList,
  39       &MSI_handle_cs_debug.ProcessLocksList },
  40       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_handle_cs") }
  41 };
  42 static CRITICAL_SECTION MSI_handle_cs = { &MSI_handle_cs_debug, -1, 0, 0, 0, 0 };
  43
  44 static CRITICAL_SECTION MSI_object_cs;
  45 static CRITICAL_SECTION_DEBUG MSI_object_cs_debug =
  46 {
  47     0, 0, &MSI_object_cs,
  48     { &MSI_object_cs_debug.ProcessLocksList,
  49       &MSI_object_cs_debug.ProcessLocksList },
  50       0, 0, { (DWORD_PTR)(__FILE__ ": MSI_object_cs") }
  51 };
  52 static CRITICAL_SECTION MSI_object_cs = { &MSI_object_cs_debug, -1, 0, 0, 0, 0 };
  53
  54 typedef struct msi_handle_info_t
  55 {
  56     MSIOBJECTHDR *obj;
  57     DWORD dwThreadId;
  58 } msi_handle_info;
  59
  60 static msi_handle_info msihandletable[MSIMAXHANDLES];
  61
  62 MSIHANDLE alloc_msihandle( MSIOBJECTHDR *obj )
  63 {
  64     MSIHANDLE ret = 0;
  65     UINT i;
  66
  67     EnterCriticalSection( &MSI_handle_cs );
  68
  69     /* find a slot */
  70     for(i=0; i<MSIMAXHANDLES; i++)
  71         if( !msihandletable[i].obj )
  72             break;
  73     if( (i>=MSIMAXHANDLES) || msihandletable[i].obj )
  74         goto out;
  75
  76     msiobj_addref( obj );
  77     msihandletable[i].obj = obj;
  78     msihandletable[i].dwThreadId = GetCurrentThreadId();
  79     ret = (MSIHANDLE) (i+1);
  80 out:
  81     TRACE("%p -> %ld\n", obj, ret );
  82
  83     LeaveCriticalSection( &MSI_handle_cs );
  84     return ret;
  85 }
  86
  87 void *msihandle2msiinfo(MSIHANDLE handle, UINT type)
  88 {
  89     MSIOBJECTHDR *ret = NULL;
  90
  91     EnterCriticalSection( &MSI_handle_cs );
  92     handle--;
  93     if( handle<0 )
  94         goto out;
  95     if( handle>=MSIMAXHANDLES )
  96         goto out;
  97     if( !msihandletable[handle].obj )
  98         goto out;
  99     if( msihandletable[handle].obj->magic != MSIHANDLE_MAGIC )
 100         goto out;
 101     if( type && (msihandletable[handle].obj->type != type) )
 102         goto out;
 103     ret = msihandletable[handle].obj;
 104     msiobj_addref( ret );
 105
 106 out:
 107     LeaveCriticalSection( &MSI_handle_cs );
 108
 109     return (void*) ret;
 110 }
 111
 112 void *alloc_msiobject(UINT type, UINT size, msihandledestructor destroy )
 113 {
 114     MSIOBJECTHDR *info;
 115
 116     info = msi_alloc_zero( size );
 117     if( info )
 118     {
 119         info->magic = MSIHANDLE_MAGIC;
 120         info->type = type;
 121         info->refcount = 1;
 122         info->destructor = destroy;
 123     }
 124
 125     return info;
 126 }
 127
 128 void msiobj_addref( MSIOBJECTHDR *info )
 129 {
 130     TRACE("%p\n", info);
 131
 132     if( !info )
 133         return;
 134
 135     if( info->magic != MSIHANDLE_MAGIC )
 136     {
 137         ERR("Invalid handle!\n");
 138         return;
 139     }
 140
 141     InterlockedIncrement(&info->refcount);
 142 }
 143
 144 void msiobj_lock( MSIOBJECTHDR *info )
 145 {
 146     EnterCriticalSection( &MSI_object_cs );
 147 }
 148
 149 void msiobj_unlock( MSIOBJECTHDR *info )
 150 {
 151     LeaveCriticalSection( &MSI_object_cs );
 152 }
 153
 154 int msiobj_release( MSIOBJECTHDR *info )
 155 {
 156     int ret;
 157
 158     TRACE("%p\n",info);
 159
 160     if( !info )
 161         return -1;
 162
 163     if( info->magic != MSIHANDLE_MAGIC )
 164     {
 165         ERR("Invalid handle!\n");
 166         return -1;
 167     }
 168
 169     ret = InterlockedDecrement( &info->refcount );
 170     if( ret==0 )
 171     {
 172         if( info->destructor )
 173             info->destructor( info );
 174         msi_free( info );
 175         TRACE("object %p destroyed\n", info);
 176     }
 177
 178     return ret;
 179 }
 180
 181 /***********************************************************
 182  *   MsiCloseHandle   [MSI.@]
 183  */
 184 UINT WINAPI MsiCloseHandle(MSIHANDLE handle)
 185 {
 186     MSIOBJECTHDR *info;
 187     UINT ret = ERROR_INVALID_HANDLE;
 188
 189     TRACE("%lx\n",handle);
 190
 191     EnterCriticalSection( &MSI_handle_cs );
 192
 193     info = msihandle2msiinfo(handle, 0);
 194     if( !info )
 195         goto out;
 196
 197     if( info->magic != MSIHANDLE_MAGIC )
 198     {
 199         ERR("Invalid handle!\n");
 200         goto out;
 201     }
 202
 203     msiobj_release( info );
 204     msihandletable[handle-1].obj = NULL;
 205     ret = ERROR_SUCCESS;
 206
 207     TRACE("handle %lx Destroyed\n", handle);
 208 out:
 209     LeaveCriticalSection( &MSI_handle_cs );
 210     if( info )
 211         msiobj_release( info );
 212
 213     return ret;
 214 }
 215
 216 /***********************************************************
 217  *   MsiCloseAllHandles   [MSI.@]
 218  *
 219  *  Closes all handles owned by the current thread
 220  *
 221  *  RETURNS:
 222  *   The number of handles closed
 223  */
 224 UINT WINAPI MsiCloseAllHandles(void)
 225 {
 226     UINT i, n=0;
 227
 228     TRACE("\n");
 229
 230     for(i=0; i<MSIMAXHANDLES; i++)
 231     {
 232         if(msihandletable[i].dwThreadId == GetCurrentThreadId())
 233         {
 234             MsiCloseHandle( i+1 );
 235             n++;
 236         }
 237     }
 238
 239     return n;
 240 }