/* * Custom Action processing for the Microsoft Installer (msi.dll) * * Copyright 2005 Aric Stewart for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #define COBJMACROS #include #include "windef.h" #include "winbase.h" #include "winerror.h" #include "msidefs.h" #include "msipriv.h" #include "winuser.h" #include "wine/debug.h" #include "wine/unicode.h" WINE_DEFAULT_DEBUG_CHANNEL(msi); #define CUSTOM_ACTION_TYPE_MASK 0x3F static const WCHAR c_collen[] = {'C',':','\\',0}; static const WCHAR cszTempFolder[]= {'T','e','m','p','F','o','l','d','e','r',0}; typedef struct tagMSIRUNNINGACTION { struct list entry; HANDLE handle; BOOL process; LPWSTR name; } MSIRUNNINGACTION; static UINT HANDLE_CustomType1(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType2(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType17(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType18(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType19(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType50(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static UINT HANDLE_CustomType34(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action); static BOOL check_execution_scheduling_options(MSIPACKAGE *package, LPCWSTR action, UINT options) { if (!package->script) return TRUE; if ((options & msidbCustomActionTypeClientRepeat) == msidbCustomActionTypeClientRepeat) { if (!(package->script->InWhatSequence & SEQUENCE_UI && package->script->InWhatSequence & SEQUENCE_EXEC)) { TRACE("Skipping action due to dbCustomActionTypeClientRepeat option.\n"); return FALSE; } } else if (options & msidbCustomActionTypeFirstSequence) { if (package->script->InWhatSequence & SEQUENCE_UI && package->script->InWhatSequence & SEQUENCE_EXEC ) { TRACE("Skipping action due to msidbCustomActionTypeFirstSequence option.\n"); return FALSE; } } else if (options & msidbCustomActionTypeOncePerProcess) { if (check_unique_action(package,action)) { TRACE("Skipping action due to msidbCustomActionTypeOncePerProcess option.\n"); return FALSE; } else register_unique_action(package,action); } return TRUE; } /* stores the CustomActionData before the action: * [CustomActionData]Action */ static LPWSTR msi_get_deferred_action(LPCWSTR action, LPWSTR actiondata) { LPWSTR deferred; DWORD len; static const WCHAR begin[] = {'[',0}; static const WCHAR end[] = {']',0}; if (!actiondata) return strdupW(action); len = lstrlenW(action) + lstrlenW(actiondata) + 3; deferred = msi_alloc(len * sizeof(WCHAR)); lstrcpyW(deferred, begin); lstrcatW(deferred, actiondata); lstrcatW(deferred, end); lstrcatW(deferred, action); return deferred; } UINT ACTION_CustomAction(MSIPACKAGE *package,LPCWSTR action, BOOL execute) { UINT rc = ERROR_SUCCESS; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ', '`','C','u','s','t','o' ,'m','A','c','t','i','o','n','`', ' ','W','H','E','R','E',' ','`','A','c','t','i' ,'o','n','`',' ', '=',' ','\'','%','s','\'',0}; UINT type; LPCWSTR source, target; LPWSTR ptr, deferred_data = NULL; LPWSTR action_copy = strdupW(action); WCHAR *deformated=NULL; /* deferred action: [CustomActionData]Action */ if ((ptr = strchrW(action_copy, ']'))) { deferred_data = action_copy + 1; *ptr = '\0'; action = ptr + 1; } row = MSI_QueryGetRecord( package->db, ExecSeqQuery, action ); if (!row) { msi_free(action_copy); return ERROR_CALL_NOT_IMPLEMENTED; } type = MSI_RecordGetInteger(row,2); source = MSI_RecordGetString(row,3); target = MSI_RecordGetString(row,4); TRACE("Handling custom action %s (%x %s %s)\n",debugstr_w(action),type, debugstr_w(source), debugstr_w(target)); /* handle some of the deferred actions */ if (type & msidbCustomActionTypeTSAware) FIXME("msidbCustomActionTypeTSAware not handled\n"); if (type & msidbCustomActionTypeInScript) { if (type & msidbCustomActionTypeNoImpersonate) FIXME("msidbCustomActionTypeNoImpersonate not handled\n"); if (type & msidbCustomActionTypeRollback) { FIXME("Rollback only action... rollbacks not supported yet\n"); schedule_action(package, ROLLBACK_SCRIPT, action); rc = ERROR_SUCCESS; goto end; } if (!execute) { LPWSTR actiondata = msi_dup_property(package, action); LPWSTR deferred = msi_get_deferred_action(action, actiondata); if (type & msidbCustomActionTypeCommit) { TRACE("Deferring Commit Action!\n"); schedule_action(package, COMMIT_SCRIPT, deferred); } else { TRACE("Deferring Action!\n"); schedule_action(package, INSTALL_SCRIPT, deferred); } rc = ERROR_SUCCESS; msi_free(deferred); goto end; } else { /*Set ActionData*/ static const WCHAR szActionData[] = { 'C','u','s','t','o','m','A','c','t','i','o','n','D','a','t','a',0}; static const WCHAR szBlank[] = {0}; LPWSTR actiondata = msi_dup_property( package, action ); if (deferred_data) MSI_SetPropertyW(package,szActionData,deferred_data); else if (actiondata) MSI_SetPropertyW(package,szActionData,actiondata); else MSI_SetPropertyW(package,szActionData,szBlank); msi_free(actiondata); } } else if (!check_execution_scheduling_options(package,action,type)) { rc = ERROR_SUCCESS; goto end; } switch (type & CUSTOM_ACTION_TYPE_MASK) { case 1: /* DLL file stored in a Binary table stream */ rc = HANDLE_CustomType1(package,source,target,type,action); break; case 2: /* EXE file stored in a Binary table stream */ rc = HANDLE_CustomType2(package,source,target,type,action); break; case 18: /*EXE file installed with package */ rc = HANDLE_CustomType18(package,source,target,type,action); break; case 19: /* Error that halts install */ rc = HANDLE_CustomType19(package,source,target,type,action); break; case 17: rc = HANDLE_CustomType17(package,source,target,type,action); break; case 50: /*EXE file specified by a property value */ rc = HANDLE_CustomType50(package,source,target,type,action); break; case 34: /*EXE to be run in specified directory */ rc = HANDLE_CustomType34(package,source,target,type,action); break; case 35: /* Directory set with formatted text. */ deformat_string(package,target,&deformated); MSI_SetTargetPathW(package, source, deformated); msi_free(deformated); break; case 51: /* Property set with formatted text. */ deformat_string(package,target,&deformated); rc = MSI_SetPropertyW(package,source,deformated); msi_free(deformated); break; default: FIXME("UNHANDLED ACTION TYPE %i (%s %s)\n", type & CUSTOM_ACTION_TYPE_MASK, debugstr_w(source), debugstr_w(target)); } end: msi_free(action_copy); msiobj_release(&row->hdr); return rc; } static UINT store_binary_to_temp(MSIPACKAGE *package, LPCWSTR source, LPWSTR tmp_file) { static const WCHAR query[] = { 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ', '`','B','i' ,'n','a','r','y','`',' ','W','H','E','R','E',' ', '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0}; MSIRECORD *row = 0; HANDLE file; CHAR buffer[1024]; static const WCHAR f1[] = {'m','s','i',0}; WCHAR fmt[MAX_PATH]; DWORD sz = MAX_PATH; UINT r; if (MSI_GetPropertyW(package, cszTempFolder, fmt, &sz) != ERROR_SUCCESS) GetTempPathW(MAX_PATH, fmt); if (GetTempFileNameW(fmt, f1, 0, tmp_file) == 0) { TRACE("Unable to create file\n"); return ERROR_FUNCTION_FAILED; } track_tempfile(package, tmp_file); row = MSI_QueryGetRecord(package->db, query, source); if (!row) return ERROR_FUNCTION_FAILED; /* write out the file */ file = CreateFileW(tmp_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (file == INVALID_HANDLE_VALUE) r = ERROR_FUNCTION_FAILED; else { do { DWORD write; sz = sizeof buffer; r = MSI_RecordReadStream(row, 2, buffer, &sz); if (r != ERROR_SUCCESS) { ERR("Failed to get stream\n"); break; } WriteFile(file, buffer, sz, &write, NULL); } while (sz == sizeof buffer); CloseHandle(file); } msiobj_release(&row->hdr); return r; } static void file_running_action(MSIPACKAGE* package, HANDLE Handle, BOOL process, LPCWSTR name) { MSIRUNNINGACTION *action; action = msi_alloc( sizeof(MSIRUNNINGACTION) ); action->handle = Handle; action->process = process; action->name = strdupW(name); list_add_tail( &package->RunningActions, &action->entry ); } static UINT custom_get_process_return( HANDLE process ) { DWORD rc = 0; GetExitCodeProcess( process, &rc ); if (rc != 0) return ERROR_FUNCTION_FAILED; return ERROR_SUCCESS; } static UINT custom_get_thread_return( HANDLE thread ) { DWORD rc = 0; GetExitCodeThread( thread, &rc ); switch (rc) { case ERROR_FUNCTION_NOT_CALLED: case ERROR_SUCCESS: case ERROR_INSTALL_USEREXIT: case ERROR_INSTALL_FAILURE: return rc; case ERROR_NO_MORE_ITEMS: return ERROR_SUCCESS; default: ERR("Invalid Return Code %d\n",rc); return ERROR_INSTALL_FAILURE; } } static UINT process_handle(MSIPACKAGE* package, UINT type, HANDLE ThreadHandle, HANDLE ProcessHandle, LPCWSTR Name) { UINT rc = ERROR_SUCCESS; if (!(type & msidbCustomActionTypeAsync)) { /* synchronous */ TRACE("Synchronous Execution of action %s\n",debugstr_w(Name)); if (ProcessHandle) msi_dialog_check_messages(ProcessHandle); else msi_dialog_check_messages(ThreadHandle); if (!(type & msidbCustomActionTypeContinue)) { if (ProcessHandle) rc = custom_get_process_return(ProcessHandle); else rc = custom_get_thread_return(ThreadHandle); } CloseHandle(ThreadHandle); if (ProcessHandle) CloseHandle(ProcessHandle); } else { TRACE("Asynchronous Execution of action %s\n",debugstr_w(Name)); /* asynchronous */ if (!(type & msidbCustomActionTypeContinue)) { if (ProcessHandle) { file_running_action(package, ProcessHandle, TRUE, Name); CloseHandle(ThreadHandle); } else file_running_action(package, ThreadHandle, FALSE, Name); } else { CloseHandle(ThreadHandle); if (ProcessHandle) CloseHandle(ProcessHandle); } } return rc; } typedef UINT __stdcall CustomEntry(MSIHANDLE); typedef struct { MSIPACKAGE *package; WCHAR *target; WCHAR *source; } thread_struct; static DWORD WINAPI ACTION_CallDllFunction(thread_struct *stuff) { HANDLE hModule; LPSTR proc; CustomEntry *fn; DWORD rc = ERROR_SUCCESS; TRACE("calling function (%s, %s)\n", debugstr_w(stuff->source), debugstr_w(stuff->target)); hModule = LoadLibraryW(stuff->source); if (hModule) { proc = strdupWtoA( stuff->target ); fn = (CustomEntry*)GetProcAddress(hModule,proc); if (fn) { MSIHANDLE hPackage; MSIPACKAGE *package = stuff->package; TRACE("Calling function %s\n", proc); hPackage = alloc_msihandle( &package->hdr ); if (hPackage) { rc = fn( hPackage ); MsiCloseHandle( hPackage ); } else ERR("Handle for object %p not found\n", package ); } else ERR("failed to resolve functon %s\n", debugstr_a(proc)); msi_free(proc); FreeLibrary(hModule); } else ERR("failed to load dll %s\n", debugstr_w(stuff->source)); msiobj_release( &stuff->package->hdr ); msi_free(stuff->source); msi_free(stuff->target); msi_free(stuff); return rc; } static DWORD WINAPI DllThread(LPVOID info) { thread_struct *stuff; DWORD rc = 0; TRACE("MSI Thread (%x) started for custom action\n", GetCurrentThreadId()); stuff = (thread_struct*)info; rc = ACTION_CallDllFunction(stuff); TRACE("MSI Thread (%x) finished (rc %i)\n",GetCurrentThreadId(), rc); /* close all handles for this thread */ MsiCloseAllHandles(); return rc; } static HANDLE do_msidbCustomActionTypeDll(MSIPACKAGE *package, LPCWSTR dll, LPCWSTR target) { thread_struct *info; info = msi_alloc( sizeof(*info) ); msiobj_addref( &package->hdr ); info->package = package; info->target = strdupW(target); info->source = strdupW(dll); return CreateThread(NULL, 0, DllThread, info, 0, NULL); } static UINT HANDLE_CustomType1(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { WCHAR tmp_file[MAX_PATH]; UINT r = ERROR_SUCCESS; HANDLE ThreadHandle; r = store_binary_to_temp(package, source, tmp_file); if (r != ERROR_SUCCESS) return r; TRACE("Calling function %s from %s\n",debugstr_w(target), debugstr_w(tmp_file)); if (!strchrW(tmp_file,'.')) { static const WCHAR dot[]={'.',0}; strcatW(tmp_file,dot); } ThreadHandle = do_msidbCustomActionTypeDll( package, tmp_file, target ); r = process_handle(package, type, ThreadHandle, NULL, action); return r; } static UINT HANDLE_CustomType2(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { WCHAR tmp_file[MAX_PATH]; STARTUPINFOW si; PROCESS_INFORMATION info; BOOL rc; INT len; WCHAR *deformated = NULL; WCHAR *cmd; static const WCHAR spc[] = {' ',0}; UINT r = ERROR_SUCCESS; memset(&si,0,sizeof(STARTUPINFOW)); r = store_binary_to_temp(package, source, tmp_file); if (r != ERROR_SUCCESS) return r; deformat_string(package,target,&deformated); len = strlenW(tmp_file)+2; if (deformated) len += strlenW(deformated); cmd = msi_alloc(sizeof(WCHAR)*len); strcpyW(cmd,tmp_file); if (deformated) { strcatW(cmd,spc); strcatW(cmd,deformated); msi_free(deformated); } TRACE("executing exe %s\n", debugstr_w(cmd)); rc = CreateProcessW(NULL, cmd, NULL, NULL, FALSE, 0, NULL, c_collen, &si, &info); msi_free(cmd); if ( !rc ) { ERR("Unable to execute command %s\n", debugstr_w(cmd)); return ERROR_SUCCESS; } r = process_handle(package, type, info.hThread, info.hProcess, action); return r; } static UINT HANDLE_CustomType17(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { HANDLE hThread; MSIFILE *file; TRACE("%s %s\n", debugstr_w(source), debugstr_w(target)); file = get_loaded_file( package, source ); if (!file) { ERR("invalid file key %s\n", debugstr_w( source )); return ERROR_FUNCTION_FAILED; } hThread = do_msidbCustomActionTypeDll( package, file->TargetPath, target ); return process_handle(package, type, hThread, NULL, action); } static UINT HANDLE_CustomType18(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { STARTUPINFOW si; PROCESS_INFORMATION info; BOOL rc; WCHAR *deformated; WCHAR *cmd; INT len; static const WCHAR spc[] = {' ',0}; MSIFILE *file; UINT prc; memset(&si,0,sizeof(STARTUPINFOW)); file = get_loaded_file(package,source); if( !file ) return ERROR_FUNCTION_FAILED; len = lstrlenW( file->TargetPath ); deformat_string(package,target,&deformated); if (deformated) len += strlenW(deformated); len += 2; cmd = msi_alloc(len * sizeof(WCHAR)); lstrcpyW( cmd, file->TargetPath); if (deformated) { strcatW(cmd, spc); strcatW(cmd, deformated); msi_free(deformated); } TRACE("executing exe %s\n", debugstr_w(cmd)); rc = CreateProcessW(NULL, cmd, NULL, NULL, FALSE, 0, NULL, c_collen, &si, &info); if ( !rc ) { ERR("Unable to execute command %s\n", debugstr_w(cmd)); msi_free(cmd); return ERROR_SUCCESS; } msi_free(cmd); prc = process_handle(package, type, info.hThread, info.hProcess, action); return prc; } static UINT HANDLE_CustomType19(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { static const WCHAR query[] = { 'S','E','L','E','C','T',' ','`','M','e','s','s','a','g','e','`',' ', 'F','R','O','M',' ','`','E','r','r','o','r','`',' ', 'W','H','E','R','E',' ','`','E','r','r','o','r','`',' ','=',' ', '%','s',0 }; MSIRECORD *row = 0; LPWSTR deformated = NULL; deformat_string( package, target, &deformated ); /* first try treat the error as a number */ row = MSI_QueryGetRecord( package->db, query, deformated ); if( row ) { LPCWSTR error = MSI_RecordGetString( row, 1 ); MessageBoxW( NULL, error, NULL, MB_OK ); msiobj_release( &row->hdr ); } else MessageBoxW( NULL, deformated, NULL, MB_OK ); msi_free( deformated ); return ERROR_FUNCTION_FAILED; } static UINT HANDLE_CustomType50(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { STARTUPINFOW si; PROCESS_INFORMATION info; WCHAR *prop; BOOL rc; WCHAR *deformated; WCHAR *cmd; INT len; static const WCHAR spc[] = {' ',0}; memset(&si,0,sizeof(STARTUPINFOW)); memset(&info,0,sizeof(PROCESS_INFORMATION)); prop = msi_dup_property( package, source ); if (!prop) return ERROR_SUCCESS; deformat_string(package,target,&deformated); len = strlenW(prop) + 2; if (deformated) len += strlenW(deformated); cmd = msi_alloc(sizeof(WCHAR)*len); strcpyW(cmd,prop); if (deformated) { strcatW(cmd,spc); strcatW(cmd,deformated); msi_free(deformated); } msi_free(prop); TRACE("executing exe %s\n", debugstr_w(cmd)); rc = CreateProcessW(NULL, cmd, NULL, NULL, FALSE, 0, NULL, c_collen, &si, &info); if ( !rc ) { ERR("Unable to execute command %s\n", debugstr_w(cmd)); msi_free(cmd); return ERROR_SUCCESS; } msi_free(cmd); return process_handle(package, type, info.hThread, info.hProcess, action); } static UINT HANDLE_CustomType34(MSIPACKAGE *package, LPCWSTR source, LPCWSTR target, const INT type, LPCWSTR action) { LPWSTR filename, deformated; STARTUPINFOW si; PROCESS_INFORMATION info; BOOL rc; UINT prc; memset(&si,0,sizeof(STARTUPINFOW)); filename = resolve_folder(package, source, FALSE, FALSE, NULL); if (!filename) return ERROR_FUNCTION_FAILED; SetCurrentDirectoryW(filename); msi_free(filename); deformat_string(package,target,&deformated); if (!deformated) return ERROR_FUNCTION_FAILED; TRACE("executing exe %s\n", debugstr_w(deformated)); rc = CreateProcessW(NULL, deformated, NULL, NULL, FALSE, 0, NULL, c_collen, &si, &info); if ( !rc ) { ERR("Unable to execute command %s\n", debugstr_w(deformated)); msi_free(deformated); return ERROR_SUCCESS; } msi_free(deformated); prc = process_handle(package, type, info.hThread, info.hProcess, action); return prc; } void ACTION_FinishCustomActions(MSIPACKAGE* package) { struct list *item, *cursor; DWORD rc; LIST_FOR_EACH_SAFE( item, cursor, &package->RunningActions ) { MSIRUNNINGACTION *action = LIST_ENTRY( item, MSIRUNNINGACTION, entry ); TRACE("Checking on action %s\n", debugstr_w(action->name)); list_remove( &action->entry ); if (action->process) GetExitCodeProcess( action->handle, &rc ); else GetExitCodeThread( action->handle, &rc ); if (rc == STILL_ACTIVE) { TRACE("Waiting on action %s\n", debugstr_w( action->name) ); msi_dialog_check_messages( action->handle ); } CloseHandle( action->handle ); msi_free( action->name ); msi_free( action ); } }