/* * Implementation of the Microsoft Installer (msi.dll) * * Copyright 2004 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ /* * Pages I need * http://msdn.microsoft.com/library/default.asp?url=/library/en-us/msi/setup/installexecutesequence_table.asp http://msdn.microsoft.com/library/default.asp?url=/library/en-us/msi/setup/standard_actions_reference.asp */ #include #include #include "windef.h" #include "winbase.h" #include "winerror.h" #include "winreg.h" #include "wine/debug.h" #include "msi.h" #include "msiquery.h" #include "objbase.h" #include "objidl.h" #include "msipriv.h" #include "winnls.h" #include "winuser.h" #include "shlobj.h" #include "wine/unicode.h" #include "ver.h" #define CUSTOM_ACTION_TYPE_MASK 0x3F WINE_DEFAULT_DEBUG_CHANNEL(msi); typedef struct tagMSIFEATURE { WCHAR Feature[96]; WCHAR Feature_Parent[96]; WCHAR Title[0x100]; WCHAR Description[0x100]; INT Display; INT Level; WCHAR Directory[96]; INT Attributes; INSTALLSTATE State; BOOL Enabled; INT ComponentCount; INT Components[1024]; /* yes hardcoded limit.... I am bad */ INT Cost; } MSIFEATURE; typedef struct tagMSICOMPONENT { WCHAR Component[96]; WCHAR ComponentId[96]; WCHAR Directory[96]; INT Attributes; WCHAR Condition[0x100]; WCHAR KeyPath[96]; INSTALLSTATE State; BOOL FeatureState; BOOL Enabled; INT Cost; } MSICOMPONENT; typedef struct tagMSIFOLDER { WCHAR Directory[96]; WCHAR TargetDefault[96]; WCHAR SourceDefault[96]; WCHAR ResolvedTarget[MAX_PATH]; WCHAR ResolvedSource[MAX_PATH]; WCHAR Property[MAX_PATH]; /* initially set property */ INT ParentIndex; INT State; /* 0 = uninitialized */ /* 1 = existing */ /* 2 = created remove if empty */ /* 3 = created persist if empty */ INT Cost; INT Space; }MSIFOLDER; typedef struct tagMSIFILE { WCHAR File[72]; INT ComponentIndex; WCHAR FileName[MAX_PATH]; INT FileSize; WCHAR Version[72]; WCHAR Language[20]; INT Attributes; INT Sequence; INT State; /* 0 = uninitialize */ /* 1 = not present */ /* 2 = present but replace */ /* 3 = present do not replace */ /* 4 = Installed */ WCHAR SourcePath[MAX_PATH]; WCHAR TargetPath[MAX_PATH]; BOOL Temporary; }MSIFILE; /* * Prototypes */ static UINT ACTION_ProcessExecSequence(MSIPACKAGE *package, BOOL UIran); static UINT ACTION_ProcessUISequence(MSIPACKAGE *package); UINT ACTION_PerformAction(MSIPACKAGE *package, const WCHAR *action); static UINT ACTION_LaunchConditions(MSIPACKAGE *package); static UINT ACTION_CostInitialize(MSIPACKAGE *package); static UINT ACTION_CreateFolders(MSIPACKAGE *package); static UINT ACTION_CostFinalize(MSIPACKAGE *package); static UINT ACTION_FileCost(MSIPACKAGE *package); static UINT ACTION_InstallFiles(MSIPACKAGE *package); static UINT ACTION_DuplicateFiles(MSIPACKAGE *package); static UINT ACTION_WriteRegistryValues(MSIPACKAGE *package); static UINT ACTION_CustomAction(MSIPACKAGE *package,const WCHAR *action); static UINT ACTION_InstallInitialize(MSIPACKAGE *package); static UINT ACTION_InstallValidate(MSIPACKAGE *package); static UINT ACTION_ProcessComponents(MSIPACKAGE *package); static UINT ACTION_RegisterTypeLibraries(MSIPACKAGE *package); static UINT ACTION_RegisterClassInfo(MSIPACKAGE *package); static UINT ACTION_RegisterProgIdInfo(MSIPACKAGE *package); static UINT ACTION_CreateShortcuts(MSIPACKAGE *package); static UINT ACTION_PublishProduct(MSIPACKAGE *package); static UINT HANDLE_CustomType1(MSIPACKAGE *package, const LPWSTR source, const LPWSTR target, const INT type); static UINT HANDLE_CustomType2(MSIPACKAGE *package, const LPWSTR source, const LPWSTR target, const INT type); static DWORD deformat_string(MSIPACKAGE *package, WCHAR* ptr,WCHAR** data); static UINT resolve_folder(MSIPACKAGE *package, LPCWSTR name, LPWSTR path, BOOL source, BOOL set_prop, MSIFOLDER **folder); static UINT track_tempfile(MSIPACKAGE *package, LPCWSTR name, LPCWSTR path); /* * consts and values used */ static const WCHAR cszSourceDir[] = {'S','o','u','r','c','e','D','i','r',0}; static const WCHAR cszRootDrive[] = {'R','O','O','T','D','R','I','V','E',0}; static const WCHAR cszTargetDir[] = {'T','A','R','G','E','T','D','I','R',0}; static const WCHAR cszTempFolder[]= {'T','e','m','p','F','o','l','d','e','r',0}; static const WCHAR cszDatabase[]={'D','A','T','A','B','A','S','E',0}; static const WCHAR c_collen[] = {'C',':','\\',0}; static const WCHAR cszlsb[]={'[',0}; static const WCHAR cszrsb[]={']',0}; static const WCHAR cszbs[]={'\\',0}; const static WCHAR szCreateFolders[] = {'C','r','e','a','t','e','F','o','l','d','e','r','s',0}; const static WCHAR szCostFinalize[] = {'C','o','s','t','F','i','n','a','l','i','z','e',0}; const static WCHAR szInstallFiles[] = {'I','n','s','t','a','l','l','F','i','l','e','s',0}; const static WCHAR szDuplicateFiles[] = {'D','u','p','l','i','c','a','t','e','F','i','l','e','s',0}; const static WCHAR szWriteRegistryValues[] = {'W','r','i','t','e','R','e','g','i','s','t','r','y','V','a','l','u','e','s',0}; const static WCHAR szCostInitialize[] = {'C','o','s','t','I','n','i','t','i','a','l','i','z','e',0}; const static WCHAR szFileCost[] = {'F','i','l','e','C','o','s','t',0}; const static WCHAR szInstallInitialize[] = {'I','n','s','t','a','l','l','I','n','i','t','i','a','l','i','z','e',0}; const static WCHAR szInstallValidate[] = {'I','n','s','t','a','l','l','V','a','l','i','d','a','t','e',0}; const static WCHAR szLaunchConditions[] = {'L','a','u','n','c','h','C','o','n','d','i','t','i','o','n','s',0}; const static WCHAR szProcessComponents[] = {'P','r','o','c','e','s','s','C','o','m','p','o','n','e','n','t','s',0}; const static WCHAR szRegisterTypeLibraries[] = {'R','e','g','i','s','t','e','r','T','y','p','e','L','i','b','r','a','r', 'i','e','s',0}; const static WCHAR szRegisterClassInfo[] = {'R','e','g','i','s','t','e','r','C','l','a','s','s','I','n','f','o',0}; const static WCHAR szRegisterProgIdInfo[] = {'R','e','g','i','s','t','e','r','P','r','o','g','I','d','I','n','f','o',0}; const static WCHAR szCreateShortcuts[] = {'C','r','e','a','t','e','S','h','o','r','t','c','u','t','s',0}; const static WCHAR szPublishProduct[] = {'P','u','b','l','i','s','h','P','r','o','d','u','c','t',0}; /******************************************************** * helper functions to get around current HACKS and such ********************************************************/ inline static void reduce_to_longfilename(WCHAR* filename) { if (strchrW(filename,'|')) { WCHAR newname[MAX_PATH]; strcpyW(newname,strchrW(filename,'|')+1); strcpyW(filename,newname); } } inline static char *strdupWtoA( const WCHAR *str ) { char *ret = NULL; if (str) { DWORD len = WideCharToMultiByte( CP_ACP, 0, str, -1, NULL, 0, NULL, NULL ); if ((ret = HeapAlloc( GetProcessHeap(), 0, len ))) WideCharToMultiByte( CP_ACP, 0, str, -1, ret, len, NULL, NULL ); } return ret; } inline static WCHAR *strdupAtoW( const char *str ) { WCHAR *ret = NULL; if (str) { DWORD len = MultiByteToWideChar( CP_ACP, 0, str, -1, NULL, 0 ); if ((ret = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) MultiByteToWideChar( CP_ACP, 0, str, -1, ret, len ); } return ret; } inline static WCHAR *load_dynamic_stringW(MSIRECORD *row, INT index) { UINT rc; DWORD sz; LPWSTR ret; sz = 0; rc = MSI_RecordGetStringW(row,index,NULL,&sz); if (sz <= 0) return NULL; sz ++; ret = HeapAlloc(GetProcessHeap(),0,sz * sizeof (WCHAR)); rc = MSI_RecordGetStringW(row,index,ret,&sz); return ret; } inline static int get_loaded_component(MSIPACKAGE* package, LPCWSTR Component ) { INT rc = -1; INT i; for (i = 0; i < package->loaded_components; i++) { if (strcmpW(Component,package->components[i].Component)==0) { rc = i; break; } } return rc; } inline static int get_loaded_feature(MSIPACKAGE* package, LPCWSTR Feature ) { INT rc = -1; INT i; for (i = 0; i < package->loaded_features; i++) { if (strcmpW(Feature,package->features[i].Feature)==0) { rc = i; break; } } return rc; } inline static int get_loaded_file(MSIPACKAGE* package, LPCWSTR file) { INT rc = -1; INT i; for (i = 0; i < package->loaded_files; i++) { if (strcmpW(file,package->files[i].File)==0) { rc = i; break; } } return rc; } static UINT track_tempfile(MSIPACKAGE *package, LPCWSTR name, LPCWSTR path) { int i; int index; if (!package) return -2; for (i=0; i < package->loaded_files; i++) if (strcmpW(package->files[i].File,name)==0) return -1; index = package->loaded_files; package->loaded_files++; if (package->loaded_files== 1) package->files = HeapAlloc(GetProcessHeap(),0,sizeof(MSIFILE)); else package->files = HeapReAlloc(GetProcessHeap(),0, package->files , package->loaded_files * sizeof(MSIFILE)); memset(&package->files[index],0,sizeof(MSIFILE)); strcpyW(package->files[index].File,name); strcpyW(package->files[index].TargetPath,path); package->files[index].Temporary = TRUE; TRACE("Tracking tempfile (%s)\n",debugstr_w(package->files[index].File)); return 0; } void ACTION_remove_tracked_tempfiles(MSIPACKAGE* package) { int i; if (!package) return; for (i = 0; i < package->loaded_files; i++) { if (package->files[i].Temporary) DeleteFileW(package->files[i].TargetPath); } } static void ui_progress(MSIPACKAGE *package, int a, int b, int c, int d ) { MSIRECORD * row; row = MSI_CreateRecord(4); MSI_RecordSetInteger(row,1,a); MSI_RecordSetInteger(row,2,b); MSI_RecordSetInteger(row,3,c); MSI_RecordSetInteger(row,4,d); MSI_ProcessMessage(package, INSTALLMESSAGE_PROGRESS, row); msiobj_release(&row->hdr); } static void ui_actiondata(MSIPACKAGE *package, LPCWSTR action, MSIRECORD * record) { static const WCHAR Query_t[] = {'S','E','L','E','C','T',' ','*',' ','f','r','o','m',' ','A','c','t','i','o', 'n','T','e','x','t',' ','w','h','e','r','e',' ','A','c','t','i','o','n',' ','=', ' ','\'','%','s','\'',0}; WCHAR message[1024]; UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static WCHAR *ActionFormat=NULL; static WCHAR LastAction[0x100] = {0}; WCHAR Query[1024]; LPWSTR ptr; if (strcmpW(LastAction,action)!=0) { sprintfW(Query,Query_t,action); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); return; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); return; } if (MSI_RecordIsNull(row,3)) { msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); return; } if (ActionFormat) HeapFree(GetProcessHeap(),0,ActionFormat); ActionFormat = load_dynamic_stringW(row,3); strcpyW(LastAction,action); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); } message[0]=0; ptr = ActionFormat; while (*ptr) { LPWSTR ptr2; LPWSTR data=NULL; WCHAR tmp[1023]; INT field; ptr2 = strchrW(ptr,'['); if (ptr2) { strncpyW(tmp,ptr,ptr2-ptr); tmp[ptr2-ptr]=0; strcatW(message,tmp); ptr2++; field = atoiW(ptr2); data = load_dynamic_stringW(record,field); if (data) { strcatW(message,data); HeapFree(GetProcessHeap(),0,data); } ptr=strchrW(ptr2,']'); ptr++; } else { strcatW(message,ptr); break; } } row = MSI_CreateRecord(1); MSI_RecordSetStringW(row,1,message); MSI_ProcessMessage(package, INSTALLMESSAGE_ACTIONDATA, row); msiobj_release(&row->hdr); } static void ui_actionstart(MSIPACKAGE *package, LPCWSTR action) { static const WCHAR template_s[]= {'A','c','t','i','o','n',' ','%','s',':',' ','%','s','.',' ','%','s','.',0}; static const WCHAR format[] = {'H','H','\'',':','\'','m','m','\'',':','\'','s','s',0}; static const WCHAR Query_t[] = {'S','E','L','E','C','T',' ','*',' ','f','r','o','m',' ','A','c','t','i','o', 'n','T','e','x','t',' ','w','h','e','r','e',' ','A','c','t','i','o','n',' ','=', ' ','\'','%','s','\'',0}; WCHAR message[1024]; WCHAR timet[0x100]; UINT rc; MSIQUERY * view; MSIRECORD * row = 0; WCHAR *ActionText=NULL; WCHAR Query[1024]; GetTimeFormatW(LOCALE_USER_DEFAULT, 0, NULL, format, timet, 0x100); sprintfW(Query,Query_t,action); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return; } ActionText = load_dynamic_stringW(row,2); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); sprintfW(message,template_s,timet,action,ActionText); row = MSI_CreateRecord(1); MSI_RecordSetStringW(row,1,message); MSI_ProcessMessage(package, INSTALLMESSAGE_ACTIONSTART, row); msiobj_release(&row->hdr); HeapFree(GetProcessHeap(),0,ActionText); } static void ui_actioninfo(MSIPACKAGE *package, LPCWSTR action, BOOL start, UINT rc) { MSIRECORD * row; static const WCHAR template_s[]= {'A','c','t','i','o','n',' ','s','t','a','r','t',' ','%','s',':',' ','%','s', '.',0}; static const WCHAR template_e[]= {'A','c','t','i','o','n',' ','e','n','d','e','d',' ','%','s',':',' ','%','s', '.',' ','R','e','t','u','r','n',' ','v','a','l','u','e',' ','%','i','.',0}; static const WCHAR format[] = {'H','H','\'',':','\'','m','m','\'',':','\'','s','s',0}; WCHAR message[1024]; WCHAR timet[0x100]; GetTimeFormatW(LOCALE_USER_DEFAULT, 0, NULL, format, timet, 0x100); if (start) sprintfW(message,template_s,timet,action); else sprintfW(message,template_e,timet,action,rc); row = MSI_CreateRecord(1); MSI_RecordSetStringW(row,1,message); MSI_ProcessMessage(package, INSTALLMESSAGE_INFO, row); msiobj_release(&row->hdr); } /**************************************************** * TOP level entry points *****************************************************/ UINT ACTION_DoTopLevelINSTALL(MSIPACKAGE *package, LPCWSTR szPackagePath, LPCWSTR szCommandLine) { DWORD sz; WCHAR buffer[10]; UINT rc; static const WCHAR szUILevel[] = {'U','I','L','e','v','e','l',0}; if (szPackagePath) { LPWSTR p; WCHAR check[MAX_PATH]; WCHAR pth[MAX_PATH]; DWORD size; strcpyW(pth,szPackagePath); p = strrchrW(pth,'\\'); if (p) { p++; *p=0; } size = MAX_PATH; if (MSI_GetPropertyW(package,cszSourceDir,check,&size) != ERROR_SUCCESS ) MSI_SetPropertyW(package, cszSourceDir, pth); } if (szCommandLine) { LPWSTR ptr,ptr2; ptr = (LPWSTR)szCommandLine; while (*ptr) { WCHAR prop[0x100]; WCHAR val[0x100]; TRACE("Looking at %s\n",debugstr_w(ptr)); ptr2 = strchrW(ptr,'='); if (ptr2) { BOOL quote=FALSE; DWORD len = 0; while (*ptr == ' ') ptr++; strncpyW(prop,ptr,ptr2-ptr); prop[ptr2-ptr]=0; ptr2++; ptr = ptr2; while (*ptr && (quote || (!quote && *ptr!=' '))) { if (*ptr == '"') quote = !quote; ptr++; len++; } if (*ptr2=='"') { ptr2++; len -= 2; } strncpyW(val,ptr2,len); val[len]=0; if (strlenW(prop) > 0) { TRACE("Found commandline property (%s) = (%s)\n", debugstr_w(prop), debugstr_w(val)); MSI_SetPropertyW(package,prop,val); } } ptr++; } } sz = 10; if (MSI_GetPropertyW(package,szUILevel,buffer,&sz) == ERROR_SUCCESS) { if (atoiW(buffer) >= INSTALLUILEVEL_REDUCED) { rc = ACTION_ProcessUISequence(package); if (rc == ERROR_SUCCESS) rc = ACTION_ProcessExecSequence(package,TRUE); } else rc = ACTION_ProcessExecSequence(package,FALSE); } else rc = ACTION_ProcessExecSequence(package,FALSE); return rc; } static UINT ACTION_ProcessExecSequence(MSIPACKAGE *package, BOOL UIran) { MSIQUERY * view; UINT rc; static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','*',' ', 'f','r','o','m',' ', 'I','n','s','t','a','l','l','E','x','e','c','u','t','e', 'S','e','q','u','e','n','c','e',' ', 'w','h','e','r','e',' ','S','e','q','u','e','n','c','e',' ', '>',' ','%','i',' ','o','r','d','e','r',' ', 'b','y',' ','S','e','q','u','e','n','c','e',0 }; WCHAR Query[1024]; MSIRECORD * row = 0; static const WCHAR IVQuery[] = { 's','e','l','e','c','t',' ','S','e','q','u','e','n','c','e',' ', 'f','r','o','m',' ','I','n','s','t','a','l','l', 'E','x','e','c','u','t','e','S','e','q','u','e','n','c','e',' ', 'w','h','e','r','e',' ','A','c','t','i','o','n',' ','=',' ', '`','I','n','s','t','a','l','l','V','a','l','i','d','a','t','e','`', 0}; if (UIran) { INT seq = 0; rc = MSI_DatabaseOpenViewW(package->db, IVQuery, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } seq = MSI_RecordGetInteger(row,1); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); sprintfW(Query,ExecSeqQuery,seq); } else sprintfW(Query,ExecSeqQuery,0); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc == ERROR_SUCCESS) { rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); goto end; } TRACE("Running the actions \n"); while (1) { WCHAR buffer[0x100]; DWORD sz = 0x100; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } /* check conditions */ if (!MSI_RecordIsNull(row,2)) { LPWSTR cond = NULL; cond = load_dynamic_stringW(row,2); if (cond) { /* this is a hack to skip errors in the condition code */ if (MSI_EvaluateConditionW(package, cond) == MSICONDITION_FALSE) { HeapFree(GetProcessHeap(),0,cond); msiobj_release(&row->hdr); continue; } else HeapFree(GetProcessHeap(),0,cond); } } sz=0x100; rc = MSI_RecordGetStringW(row,1,buffer,&sz); if (rc != ERROR_SUCCESS) { ERR("Error is %x\n",rc); msiobj_release(&row->hdr); break; } rc = ACTION_PerformAction(package,buffer); if (rc != ERROR_SUCCESS) { ERR("Execution halted due to error (%i)\n",rc); msiobj_release(&row->hdr); break; } msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); } end: return rc; } static UINT ACTION_ProcessUISequence(MSIPACKAGE *package) { MSIQUERY * view; UINT rc; static const WCHAR ExecSeqQuery [] = { 's','e','l','e','c','t',' ','*',' ', 'f','r','o','m',' ','I','n','s','t','a','l','l', 'U','I','S','e','q','u','e','n','c','e',' ', 'w','h','e','r','e',' ','S','e','q','u','e','n','c','e',' ', '>',' ','0',' ', 'o','r','d','e','r',' ','b','y',' ','S','e','q','u','e','n','c','e',0}; rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc == ERROR_SUCCESS) { rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); goto end; } TRACE("Running the actions \n"); while (1) { WCHAR buffer[0x100]; DWORD sz = 0x100; MSIRECORD * row = 0; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } /* check conditions */ if (!MSI_RecordIsNull(row,2)) { LPWSTR cond = NULL; cond = load_dynamic_stringW(row,2); if (cond) { /* this is a hack to skip errors in the condition code */ if (MSI_EvaluateConditionW(package, cond) == MSICONDITION_FALSE) { HeapFree(GetProcessHeap(),0,cond); msiobj_release(&row->hdr); continue; } else HeapFree(GetProcessHeap(),0,cond); } } sz=0x100; rc = MSI_RecordGetStringW(row,1,buffer,&sz); if (rc != ERROR_SUCCESS) { ERR("Error is %x\n",rc); msiobj_release(&row->hdr); break; } rc = ACTION_PerformAction(package,buffer); if (rc != ERROR_SUCCESS) { ERR("Execution halted due to error (%i)\n",rc); msiobj_release(&row->hdr); break; } msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); } end: return rc; } /******************************************************** * ACTION helper functions and functions that perform the actions *******************************************************/ /* * Alot of actions are really important even if they don't do anything * explicit.. Lots of properties are set at the beginning of the installation * CostFinalize does a bunch of work to translated the directories and such * * But until I get write access to the database that is hard, so I am going to * hack it to see if I can get something to run. */ UINT ACTION_PerformAction(MSIPACKAGE *package, const WCHAR *action) { UINT rc = ERROR_SUCCESS; TRACE("Performing action (%s)\n",debugstr_w(action)); ui_actioninfo(package, action, TRUE, 0); ui_actionstart(package, action); ui_progress(package,2,1,0,0); /* pre install, setup and configuration block */ if (strcmpW(action,szLaunchConditions)==0) rc = ACTION_LaunchConditions(package); else if (strcmpW(action,szCostInitialize)==0) rc = ACTION_CostInitialize(package); else if (strcmpW(action,szFileCost)==0) rc = ACTION_FileCost(package); else if (strcmpW(action,szCostFinalize)==0) rc = ACTION_CostFinalize(package); else if (strcmpW(action,szInstallValidate)==0) rc = ACTION_InstallValidate(package); /* install block */ else if (strcmpW(action,szProcessComponents)==0) rc = ACTION_ProcessComponents(package); else if (strcmpW(action,szInstallInitialize)==0) rc = ACTION_InstallInitialize(package); else if (strcmpW(action,szCreateFolders)==0) rc = ACTION_CreateFolders(package); else if (strcmpW(action,szInstallFiles)==0) rc = ACTION_InstallFiles(package); else if (strcmpW(action,szDuplicateFiles)==0) rc = ACTION_DuplicateFiles(package); else if (strcmpW(action,szWriteRegistryValues)==0) rc = ACTION_WriteRegistryValues(package); else if (strcmpW(action,szRegisterTypeLibraries)==0) rc = ACTION_RegisterTypeLibraries(package); else if (strcmpW(action,szRegisterClassInfo)==0) rc = ACTION_RegisterClassInfo(package); else if (strcmpW(action,szRegisterProgIdInfo)==0) rc = ACTION_RegisterProgIdInfo(package); else if (strcmpW(action,szCreateShortcuts)==0) rc = ACTION_CreateShortcuts(package); else if (strcmpW(action,szPublishProduct)==0) rc = ACTION_PublishProduct(package); /* Called during iTunes but unimplemented and seem important ResolveSource (sets SourceDir) RegisterProduct InstallFinalize */ else if ((rc = ACTION_CustomAction(package,action)) != ERROR_SUCCESS) { FIXME("UNHANDLED MSI ACTION %s\n",debugstr_w(action)); rc = ERROR_SUCCESS; } ui_actioninfo(package, action, FALSE, rc); return rc; } static UINT ACTION_CustomAction(MSIPACKAGE *package,const WCHAR *action) { UINT rc = ERROR_SUCCESS; MSIQUERY * view; MSIRECORD * row = 0; WCHAR ExecSeqQuery[1024] = {'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','`',' ','=',' ','`',0}; static const WCHAR end[]={'`',0}; UINT type; LPWSTR source; LPWSTR target; WCHAR *deformated=NULL; strcatW(ExecSeqQuery,action); strcatW(ExecSeqQuery,end); rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } type = MSI_RecordGetInteger(row,2); source = load_dynamic_stringW(row,3); target = load_dynamic_stringW(row,4); TRACE("Handling custom action %s (%x %s %s)\n",debugstr_w(action),type, debugstr_w(source), debugstr_w(target)); /* we are ignoring ALOT of flags and important synchronization stuff */ switch (type & CUSTOM_ACTION_TYPE_MASK) { case 1: /* DLL file stored in a Binary table stream */ rc = HANDLE_CustomType1(package,source,target,type); break; case 2: /* EXE file stored in a Binary table strem */ rc = HANDLE_CustomType2(package,source,target,type); break; case 35: /* Directory set with formatted text. */ case 51: /* Property set with formatted text. */ deformat_string(package,target,&deformated); rc = MSI_SetPropertyW(package,source,deformated); HeapFree(GetProcessHeap(),0,deformated); break; default: FIXME("UNHANDLED ACTION TYPE %i (%s %s)\n", type & CUSTOM_ACTION_TYPE_MASK, debugstr_w(source), debugstr_w(target)); } HeapFree(GetProcessHeap(),0,source); HeapFree(GetProcessHeap(),0,target); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static UINT store_binary_to_temp(MSIPACKAGE *package, const LPWSTR source, LPWSTR tmp_file) { DWORD sz=MAX_PATH; if (MSI_GetPropertyW(package, cszTempFolder, tmp_file, &sz) != ERROR_SUCCESS) GetTempPathW(MAX_PATH,tmp_file); strcatW(tmp_file,source); if (GetFileAttributesW(tmp_file) != INVALID_FILE_ATTRIBUTES) { TRACE("File already exists\n"); return ERROR_SUCCESS; } else { /* write out the file */ UINT rc; MSIQUERY * view; MSIRECORD * row = 0; WCHAR Query[1024] = {'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','=','`',0}; static const WCHAR end[]={'`',0}; HANDLE the_file; CHAR buffer[1024]; if (track_tempfile(package, source, tmp_file)!=0) FIXME("File Name in temp tracking collision\n"); the_file = CreateFileW(tmp_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (the_file == INVALID_HANDLE_VALUE) return ERROR_FUNCTION_FAILED; strcatW(Query,source); strcatW(Query,end); rc = MSI_DatabaseOpenViewW( package->db, Query, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } do { DWORD write; sz = 1024; rc = MSI_RecordReadStream(row,2,buffer,&sz); if (rc != ERROR_SUCCESS) { ERR("Failed to get stream\n"); CloseHandle(the_file); DeleteFileW(tmp_file); break; } WriteFile(the_file,buffer,sz,&write,NULL); } while (sz == 1024); CloseHandle(the_file); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); } return ERROR_SUCCESS; } typedef UINT __stdcall CustomEntry(MSIHANDLE); typedef struct { MSIPACKAGE *package; WCHAR target[MAX_PATH]; WCHAR source[MAX_PATH]; } thread_struct; #if 0 static DWORD WINAPI DllThread(LPVOID info) { HANDLE DLL; LPSTR proc; thread_struct *stuff; CustomEntry *fn; stuff = (thread_struct*)info; TRACE("Asynchronous start (%s, %s) \n", debugstr_w(stuff->source), debugstr_w(stuff->target)); DLL = LoadLibraryW(stuff->source); if (DLL) { proc = strdupWtoA( stuff->target ); fn = (CustomEntry*)GetProcAddress(DLL,proc); if (fn) { MSIHANDLE hPackage; MSIPACKAGE *package = stuff->package; TRACE("Calling function\n"); hPackage = msiobj_findhandle( &package->hdr ); if( !hPackage ) ERR("Handle for object %p not found\n", package ); fn(hPackage); msiobj_release( &package->hdr ); } else ERR("Cannot load functon\n"); HeapFree(GetProcessHeap(),0,proc); FreeLibrary(DLL); } else ERR("Unable to load library\n"); msiobj_release( &stuff->package->hdr ); HeapFree( GetProcessHeap(), 0, info ); return 0; } #endif static UINT HANDLE_CustomType1(MSIPACKAGE *package, const LPWSTR source, const LPWSTR target, const INT type) { WCHAR tmp_file[MAX_PATH]; CustomEntry *fn; HANDLE DLL; LPSTR proc; store_binary_to_temp(package, source, tmp_file); 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); } if (type & 0xc0) { /* DWORD ThreadId; */ thread_struct *info = HeapAlloc( GetProcessHeap(), 0, sizeof(*info) ); /* msiobj_addref( &package->hdr ); */ info->package = package; strcpyW(info->target,target); strcpyW(info->source,tmp_file); TRACE("Start Asynchronous execution\n"); FIXME("DATABASE NOT THREADSAFE... not starting\n"); /* CreateThread(NULL,0,DllThread,(LPVOID)&info,0,&ThreadId); */ /* FIXME: release the package if the CreateThread fails */ HeapFree( GetProcessHeap(), 0, info ); return ERROR_SUCCESS; } DLL = LoadLibraryW(tmp_file); if (DLL) { proc = strdupWtoA( target ); fn = (CustomEntry*)GetProcAddress(DLL,proc); if (fn) { MSIHANDLE hPackage; TRACE("Calling function\n"); hPackage = msiobj_findhandle( &package->hdr ); if( !hPackage ) ERR("Handle for object %p not found\n", package ); fn(hPackage); msiobj_release( &package->hdr ); } else ERR("Cannot load functon\n"); HeapFree(GetProcessHeap(),0,proc); FreeLibrary(DLL); } else ERR("Unable to load library\n"); return ERROR_SUCCESS; } static UINT HANDLE_CustomType2(MSIPACKAGE *package, const LPWSTR source, const LPWSTR target, const INT type) { WCHAR tmp_file[MAX_PATH*2]; STARTUPINFOW si; PROCESS_INFORMATION info; BOOL rc; WCHAR *deformated; static const WCHAR spc[] = {' ',0}; memset(&si,0,sizeof(STARTUPINFOW)); memset(&info,0,sizeof(PROCESS_INFORMATION)); store_binary_to_temp(package, source, tmp_file); strcatW(tmp_file,spc); deformat_string(package,target,&deformated); strcatW(tmp_file,deformated); HeapFree(GetProcessHeap(),0,deformated); TRACE("executing exe %s \n",debugstr_w(tmp_file)); rc = CreateProcessW(NULL, tmp_file, NULL, NULL, FALSE, 0, NULL, c_collen, &si, &info); if ( !rc ) { ERR("Unable to execute command\n"); return ERROR_SUCCESS; } if (!(type & 0xc0)) WaitForSingleObject(info.hProcess,INFINITE); return ERROR_SUCCESS; } /*********************************************************************** * create_full_pathW * * Recursively create all directories in the path. * * shamelessly stolen from setupapi/queue.c */ static BOOL create_full_pathW(const WCHAR *path) { BOOL ret = TRUE; int len; WCHAR *new_path; new_path = HeapAlloc(GetProcessHeap(), 0, (strlenW(path) + 1) * sizeof(WCHAR)); strcpyW(new_path, path); while((len = strlenW(new_path)) && new_path[len - 1] == '\\') new_path[len - 1] = 0; while(!CreateDirectoryW(new_path, NULL)) { WCHAR *slash; DWORD last_error = GetLastError(); if(last_error == ERROR_ALREADY_EXISTS) break; if(last_error != ERROR_PATH_NOT_FOUND) { ret = FALSE; break; } if(!(slash = strrchrW(new_path, '\\'))) { ret = FALSE; break; } len = slash - new_path; new_path[len] = 0; if(!create_full_pathW(new_path)) { ret = FALSE; break; } new_path[len] = '\\'; } HeapFree(GetProcessHeap(), 0, new_path); return ret; } /* * Also we cannot enable/disable components either, so for now I am just going * to do all the directories for all the components. */ static UINT ACTION_CreateFolders(MSIPACKAGE *package) { static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','D','i','r','e','c','t','o','r','y','_',' ', 'f','r','o','m',' ','C','r','e','a','t','e','F','o','l','d','e','r',0 }; UINT rc; MSIQUERY *view; MSIFOLDER *folder; rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view ); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR dir[0x100]; WCHAR full_path[MAX_PATH]; DWORD sz; MSIRECORD *row = NULL, *uirow; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz=0x100; rc = MSI_RecordGetStringW(row,1,dir,&sz); if (rc!= ERROR_SUCCESS) { ERR("Unable to get folder id \n"); msiobj_release(&row->hdr); continue; } sz = MAX_PATH; rc = resolve_folder(package,dir,full_path,FALSE,FALSE,&folder); if (rc != ERROR_SUCCESS) { ERR("Unable to resolve folder id %s\n",debugstr_w(dir)); msiobj_release(&row->hdr); continue; } TRACE("Folder is %s\n",debugstr_w(full_path)); /* UI stuff */ uirow = MSI_CreateRecord(1); MSI_RecordSetStringW(uirow,1,full_path); ui_actiondata(package,szCreateFolders,uirow); msiobj_release( &uirow->hdr ); if (folder->State == 0) create_full_pathW(full_path); folder->State = 3; msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static int load_component(MSIPACKAGE* package, MSIRECORD * row) { int index = package->loaded_components; DWORD sz; /* fill in the data */ package->loaded_components++; if (package->loaded_components == 1) package->components = HeapAlloc(GetProcessHeap(),0, sizeof(MSICOMPONENT)); else package->components = HeapReAlloc(GetProcessHeap(),0, package->components, package->loaded_components * sizeof(MSICOMPONENT)); memset(&package->components[index],0,sizeof(MSICOMPONENT)); sz = 96; MSI_RecordGetStringW(row,1,package->components[index].Component,&sz); TRACE("Loading Component %s\n", debugstr_w(package->components[index].Component)); sz = 0x100; if (!MSI_RecordIsNull(row,2)) MSI_RecordGetStringW(row,2,package->components[index].ComponentId,&sz); sz = 96; MSI_RecordGetStringW(row,3,package->components[index].Directory,&sz); package->components[index].Attributes = MSI_RecordGetInteger(row,4); sz = 0x100; MSI_RecordGetStringW(row,5,package->components[index].Condition,&sz); sz = 96; MSI_RecordGetStringW(row,6,package->components[index].KeyPath,&sz); package->components[index].State = INSTALLSTATE_UNKNOWN; package->components[index].Enabled = TRUE; package->components[index].FeatureState= FALSE; return index; } static void load_feature(MSIPACKAGE* package, MSIRECORD * row) { int index = package->loaded_features; DWORD sz; static const WCHAR Query1[] = {'S','E','L','E','C','T',' ','C','o','m','p', 'o','n','e','n','t','_',' ','F','R','O','M',' ','F','e','a','t','u','r','e', 'C','o','m','p','o','n','e','n','t','s',' ','W','H','E','R','E',' ','F','e', 'a','t','u','r','e','_','=','\'','%','s','\'',0}; static const WCHAR Query2[] = {'S','E','L','E','C','T',' ','*',' ','F','R', 'O','M',' ','C','o','m','p','o','n','e','n','t',' ','W','H','E','R','E',' ','C', 'o','m','p','o','n','e','n','t','=','\'','%','s','\'',0}; WCHAR Query[1024]; MSIQUERY * view; MSIQUERY * view2; MSIRECORD * row2; MSIRECORD * row3; UINT rc; /* fill in the data */ package->loaded_features ++; if (package->loaded_features == 1) package->features = HeapAlloc(GetProcessHeap(),0,sizeof(MSIFEATURE)); else package->features = HeapReAlloc(GetProcessHeap(),0,package->features, package->loaded_features * sizeof(MSIFEATURE)); memset(&package->features[index],0,sizeof(MSIFEATURE)); sz = 96; MSI_RecordGetStringW(row,1,package->features[index].Feature,&sz); TRACE("Loading feature %s\n",debugstr_w(package->features[index].Feature)); sz = 96; if (!MSI_RecordIsNull(row,2)) MSI_RecordGetStringW(row,2,package->features[index].Feature_Parent,&sz); sz = 0x100; if (!MSI_RecordIsNull(row,3)) MSI_RecordGetStringW(row,3,package->features[index].Title,&sz); sz = 0x100; if (!MSI_RecordIsNull(row,4)) MSI_RecordGetStringW(row,4,package->features[index].Description,&sz); if (!MSI_RecordIsNull(row,5)) package->features[index].Display = MSI_RecordGetInteger(row,5); package->features[index].Level= MSI_RecordGetInteger(row,6); sz = 96; if (!MSI_RecordIsNull(row,7)) MSI_RecordGetStringW(row,7,package->features[index].Directory,&sz); package->features[index].Attributes= MSI_RecordGetInteger(row,8); package->features[index].State = INSTALLSTATE_UNKNOWN; /* load feature components */ sprintfW(Query,Query1,package->features[index].Feature); rc = MSI_DatabaseOpenViewW(package->db,Query,&view); if (rc != ERROR_SUCCESS) return; rc = MSI_ViewExecute(view,0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return; } while (1) { DWORD sz = 0x100; WCHAR buffer[0x100]; DWORD rc; INT c_indx; INT cnt = package->features[index].ComponentCount; rc = MSI_ViewFetch(view,&row2); if (rc != ERROR_SUCCESS) break; sz = 0x100; MSI_RecordGetStringW(row2,1,buffer,&sz); /* check to see if the component is already loaded */ c_indx = get_loaded_component(package,buffer); if (c_indx != -1) { TRACE("Component %s already loaded at %i\n", debugstr_w(buffer), c_indx); package->features[index].Components[cnt] = c_indx; package->features[index].ComponentCount ++; } sprintfW(Query,Query2,buffer); rc = MSI_DatabaseOpenViewW(package->db,Query,&view2); if (rc != ERROR_SUCCESS) { msiobj_release( &row2->hdr ); continue; } rc = MSI_ViewExecute(view2,0); if (rc != ERROR_SUCCESS) { msiobj_release( &row2->hdr ); MSI_ViewClose(view2); msiobj_release( &view2->hdr ); continue; } while (1) { DWORD rc; rc = MSI_ViewFetch(view2,&row3); if (rc != ERROR_SUCCESS) break; c_indx = load_component(package,row3); msiobj_release( &row3->hdr ); package->features[index].Components[cnt] = c_indx; package->features[index].ComponentCount ++; } MSI_ViewClose(view2); msiobj_release( &view2->hdr ); msiobj_release( &row2->hdr ); } MSI_ViewClose(view); msiobj_release(&view->hdr); } /* * I am not doing any of the costing functionality yet. * Mostly looking at doing the Component and Feature loading * * The native MSI does ALOT of modification to tables here. Mostly adding alot * of temporary columns to the Feature and Component tables. * * note: native msi also tracks the short filename. but I am only going to * track the long ones. Also looking at this directory table * it appears that the directory table does not get the parents * resolved base on property only based on their entrys in the * directory table. */ static UINT ACTION_CostInitialize(MSIPACKAGE *package) { MSIQUERY * view; MSIRECORD * row; DWORD sz; UINT rc; static const WCHAR Query_all[] = { 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ','F','e','a','t','u','r','e',0}; static const WCHAR szCosting[] = { 'C','o','s','t','i','n','g','C','o','m','p','l','e','t','e',0 }; static const WCHAR szZero[] = { '0', 0 }; MSI_SetPropertyW(package, szCosting, szZero); MSI_SetPropertyW(package, cszRootDrive , c_collen); sz = 0x100; rc = MSI_DatabaseOpenViewW(package->db,Query_all,&view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view,0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { DWORD rc; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) break; load_feature(package,row); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return ERROR_SUCCESS; } static int load_file(MSIPACKAGE* package, MSIRECORD * row) { int index = package->loaded_files; int i; WCHAR buffer[0x100]; DWORD sz; /* fill in the data */ package->loaded_files++; if (package->loaded_files== 1) package->files = HeapAlloc(GetProcessHeap(),0,sizeof(MSIFILE)); else package->files = HeapReAlloc(GetProcessHeap(),0, package->files , package->loaded_files * sizeof(MSIFILE)); memset(&package->files[index],0,sizeof(MSIFILE)); sz = 72; MSI_RecordGetStringW(row,1,package->files[index].File,&sz); sz = 0x100; MSI_RecordGetStringW(row,2,buffer,&sz); package->files[index].ComponentIndex = -1; for (i = 0; i < package->loaded_components; i++) if (strcmpW(package->components[i].Component,buffer)==0) { package->files[index].ComponentIndex = i; break; } if (package->files[index].ComponentIndex == -1) ERR("Unfound Component %s\n",debugstr_w(buffer)); sz = MAX_PATH; MSI_RecordGetStringW(row,3,package->files[index].FileName,&sz); reduce_to_longfilename(package->files[index].FileName); package->files[index].FileSize = MSI_RecordGetInteger(row,4); sz = 72; if (!MSI_RecordIsNull(row,5)) MSI_RecordGetStringW(row,5,package->files[index].Version,&sz); sz = 20; if (!MSI_RecordIsNull(row,6)) MSI_RecordGetStringW(row,6,package->files[index].Language,&sz); if (!MSI_RecordIsNull(row,7)) package->files[index].Attributes= MSI_RecordGetInteger(row,7); package->files[index].Sequence= MSI_RecordGetInteger(row,8); package->files[index].Temporary = FALSE; package->files[index].State = 0; TRACE("File Loaded (%s)\n",debugstr_w(package->files[index].File)); return ERROR_SUCCESS; } static UINT ACTION_FileCost(MSIPACKAGE *package) { MSIQUERY * view; MSIRECORD * row; UINT rc; static const WCHAR Query[] = { 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ','F','i','l','e',' ', 'O','r','d','e','r',' ','b','y',' ','S','e','q','u','e','n','c','e', 0}; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return ERROR_SUCCESS; } while (1) { rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } load_file(package,row); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return ERROR_SUCCESS; } static INT load_folder(MSIPACKAGE *package, const WCHAR* dir) { WCHAR Query[1024] = {'s','e','l','e','c','t',' ','*',' ','f','r','o','m',' ','D','i','r','e','c', 't','o','r','y',' ','w','h','e','r','e',' ','`','D','i','r','e','c','t', 'o','r','y','`',' ','=',' ','`',0}; static const WCHAR end[]={'`',0}; UINT rc; MSIQUERY * view; WCHAR targetbuffer[0x100]; WCHAR *srcdir = NULL; WCHAR *targetdir = NULL; WCHAR parent[0x100]; DWORD sz=0x100; MSIRECORD * row = 0; INT i,index = -1; TRACE("Looking for dir %s\n",debugstr_w(dir)); for (i = 0; i < package->loaded_folders; i++) { if (strcmpW(package->folders[i].Directory,dir)==0) { TRACE(" %s retuning on index %i\n",debugstr_w(dir),i); return i; } } TRACE("Working to load %s\n",debugstr_w(dir)); index = package->loaded_folders; package->loaded_folders++; if (package->loaded_folders== 1) package->folders = HeapAlloc(GetProcessHeap(),0, sizeof(MSIFOLDER)); else package->folders= HeapReAlloc(GetProcessHeap(),0, package->folders, package->loaded_folders* sizeof(MSIFOLDER)); memset(&package->folders[index],0,sizeof(MSIFOLDER)); strcpyW(package->folders[index].Directory,dir); strcatW(Query,dir); strcatW(Query,end); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return -1; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return -1; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return -1; } sz=0x100; MSI_RecordGetStringW(row,3,targetbuffer,&sz); targetdir=targetbuffer; /* split src and target dir */ if (strchrW(targetdir,':')) { srcdir=strchrW(targetdir,':'); *srcdir=0; srcdir ++; } else srcdir=NULL; /* for now only pick long filename versions */ if (strchrW(targetdir,'|')) { targetdir = strchrW(targetdir,'|'); *targetdir = 0; targetdir ++; } if (srcdir && strchrW(srcdir,'|')) { srcdir= strchrW(srcdir,'|'); *srcdir= 0; srcdir ++; } /* now check for root dirs */ if (targetdir[0] == '.' && targetdir[1] == 0) targetdir = NULL; if (srcdir && srcdir[0] == '.' && srcdir[1] == 0) srcdir = NULL; if (targetdir) strcpyW(package->folders[index].TargetDefault,targetdir); if (srcdir) strcpyW(package->folders[index].SourceDefault,srcdir); else if (targetdir) strcpyW(package->folders[index].SourceDefault,targetdir); if (MSI_RecordIsNull(row,2)) parent[0]=0; else { sz=0x100; MSI_RecordGetStringW(row,2,parent,&sz); } if (parent[0]) { i = load_folder(package,parent); package->folders[index].ParentIndex = i; TRACE("Parent is index %i... %s %s\n", package->folders[index].ParentIndex, debugstr_w(package->folders[package->folders[index].ParentIndex].Directory), debugstr_w(parent)); } else package->folders[index].ParentIndex = -2; sz = MAX_PATH; rc = MSI_GetPropertyW(package, dir, package->folders[index].Property, &sz); if (rc != ERROR_SUCCESS) package->folders[index].Property[0]=0; msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); TRACE(" %s retuning on index %i\n",debugstr_w(dir),index); return index; } static UINT resolve_folder(MSIPACKAGE *package, LPCWSTR name, LPWSTR path, BOOL source, BOOL set_prop, MSIFOLDER **folder) { INT i; UINT rc = ERROR_SUCCESS; DWORD sz; TRACE("Working to resolve %s\n",debugstr_w(name)); if (!path) return rc; /* special resolving for Target and Source root dir */ if (strcmpW(name,cszTargetDir)==0 || strcmpW(name,cszSourceDir)==0) { if (!source) { sz = MAX_PATH; rc = MSI_GetPropertyW(package,cszTargetDir,path,&sz); if (rc != ERROR_SUCCESS) { sz = MAX_PATH; rc = MSI_GetPropertyW(package,cszRootDrive,path,&sz); if (set_prop) MSI_SetPropertyW(package,cszTargetDir,path); } if (folder) *folder = &(package->folders[0]); return rc; } else { sz = MAX_PATH; rc = MSI_GetPropertyW(package,cszSourceDir,path,&sz); if (rc != ERROR_SUCCESS) { sz = MAX_PATH; rc = MSI_GetPropertyW(package,cszDatabase,path,&sz); if (rc == ERROR_SUCCESS) { LPWSTR ptr = strrchrW(path,'\\'); if (ptr) { ptr++; *ptr = 0; } } } if (folder) *folder = &(package->folders[0]); return rc; } } for (i = 0; i < package->loaded_folders; i++) { if (strcmpW(package->folders[i].Directory,name)==0) break; } if (i >= package->loaded_folders) return ERROR_FUNCTION_FAILED; if (folder) *folder = &(package->folders[i]); if (!source && package->folders[i].ResolvedTarget[0]) { strcpyW(path,package->folders[i].ResolvedTarget); TRACE(" already resolved to %s\n",debugstr_w(path)); return ERROR_SUCCESS; } else if (source && package->folders[i].ResolvedSource[0]) { strcpyW(path,package->folders[i].ResolvedSource); return ERROR_SUCCESS; } else if (!source && package->folders[i].Property[0]) { strcpyW(path,package->folders[i].Property); TRACE(" internally set to %s\n",debugstr_w(path)); if (set_prop) MSI_SetPropertyW(package,name,path); return ERROR_SUCCESS; } if (package->folders[i].ParentIndex >= 0) { TRACE(" ! Parent is %s\n", debugstr_w(package->folders[ package->folders[i].ParentIndex].Directory)); resolve_folder(package, package->folders[ package->folders[i].ParentIndex].Directory, path,source, set_prop, NULL); if (!source) { if (package->folders[i].TargetDefault[0]) { strcatW(path,package->folders[i].TargetDefault); strcatW(path,cszbs); } strcpyW(package->folders[i].ResolvedTarget,path); TRACE(" resolved into %s\n",debugstr_w(path)); if (set_prop) MSI_SetPropertyW(package,name,path); } else { if (package->folders[i].SourceDefault[0]) { strcatW(path,package->folders[i].SourceDefault); strcatW(path,cszbs); } strcpyW(package->folders[i].ResolvedSource,path); } } return rc; } /* * Alot is done in this function aside from just the costing. * The costing needs to be implemented at some point but for now I am going * to focus on the directory building * */ static UINT ACTION_CostFinalize(MSIPACKAGE *package) { static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','*',' ','f','r','o','m',' ', 'D','i','r','e','c','t','o','r','y',0}; static const WCHAR ConditionQuery[] = { 's','e','l','e','c','t',' ','*',' ','f','r','o','m',' ', 'C','o','n','d','i','t','i','o','n',0}; static const WCHAR szCosting[] = { 'C','o','s','t','i','n','g','C','o','m','p','l','e','t','e',0 }; static const WCHAR szOne[] = { '1', 0 }; UINT rc; MSIQUERY * view; INT i; TRACE("Building Directory properties\n"); rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc == ERROR_SUCCESS) { rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR name[0x100]; WCHAR path[MAX_PATH]; MSIRECORD * row = 0; DWORD sz; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz=0x100; MSI_RecordGetStringW(row,1,name,&sz); /* This helper function now does ALL the work */ TRACE("Dir %s ...\n",debugstr_w(name)); load_folder(package,name); resolve_folder(package,name,path,FALSE,TRUE,NULL); TRACE("resolves to %s\n",debugstr_w(path)); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); } TRACE("File calculations %i files\n",package->loaded_files); for (i = 0; i < package->loaded_files; i++) { MSICOMPONENT* comp = NULL; MSIFILE* file= NULL; file = &package->files[i]; if (file->ComponentIndex >= 0) comp = &package->components[file->ComponentIndex]; if (comp) { /* calculate target */ resolve_folder(package, comp->Directory, file->TargetPath, FALSE, FALSE, NULL); strcatW(file->TargetPath,file->FileName); TRACE("file %s resolves to %s\n", debugstr_w(file->File),debugstr_w(file->TargetPath)); if (GetFileAttributesW(file->TargetPath) == INVALID_FILE_ATTRIBUTES) { file->State = 1; comp->Cost += file->FileSize; } else { if (file->Version[0]) { DWORD handle; DWORD versize; UINT sz; LPVOID version; WCHAR filever[0x100]; static const WCHAR name[] = {'\\','V','a','r','F','i','l','e','I','n','f','o', '\\','F','i','l','e','V','e','r','s','i','o','n',0}; FIXME("Version comparison.. Untried Untested and most " "likely very very wrong\n"); versize = GetFileVersionInfoSizeW(file->TargetPath,&handle); version = HeapAlloc(GetProcessHeap(),0,versize); GetFileVersionInfoW(file->TargetPath, 0, versize, version); sz = 0x100; VerQueryValueW(version,name,(LPVOID)filever,&sz); HeapFree(GetProcessHeap(),0,version); if (strcmpW(version,file->Version)<0) { file->State = 2; FIXME("cost should be diff in size\n"); comp->Cost += file->FileSize; } else file->State = 3; } else file->State = 3; } } } TRACE("Evaluating Condition Table\n"); rc = MSI_DatabaseOpenViewW(package->db, ConditionQuery, &view); if (rc == ERROR_SUCCESS) { rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR Feature[0x100]; MSIRECORD * row = 0; DWORD sz; int feature_index; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz = 0x100; MSI_RecordGetStringW(row,1,Feature,&sz); feature_index = get_loaded_feature(package,Feature); if (feature_index < 0) ERR("FAILED to find loaded feature %s\n",debugstr_w(Feature)); else { LPWSTR Condition; Condition = load_dynamic_stringW(row,3); if (MSI_EvaluateConditionW(package,Condition) == MSICONDITION_TRUE) { int level = MSI_RecordGetInteger(row,2); TRACE("Reseting feature %s to level %i\n", debugstr_w(Feature), level); package->features[feature_index].Level = level; } HeapFree(GetProcessHeap(),0,Condition); } msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); } TRACE("Enabling or Disabling Components\n"); for (i = 0; i < package->loaded_components; i++) { if (package->components[i].Condition[0]) { if (MSI_EvaluateConditionW(package, package->components[i].Condition) == MSICONDITION_FALSE) { TRACE("Disabling component %s\n", debugstr_w(package->components[i].Component)); package->components[i].Enabled = FALSE; } } } MSI_SetPropertyW(package,szCosting,szOne); return ERROR_SUCCESS; } /* * This is a helper function for handling embedded cabinet media */ static UINT writeout_cabinet_stream(MSIPACKAGE *package, WCHAR* stream_name, WCHAR* source) { UINT rc; USHORT* data; UINT size; DWORD write; HANDLE the_file; WCHAR tmp[MAX_PATH]; rc = read_raw_stream_data(package->db,stream_name,&data,&size); if (rc != ERROR_SUCCESS) return rc; write = MAX_PATH; if (MSI_GetPropertyW(package, cszTempFolder, tmp, &write)) GetTempPathW(MAX_PATH,tmp); GetTempFileNameW(tmp,stream_name,0,source); track_tempfile(package,strrchrW(source,'\\'), source); the_file = CreateFileW(source, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (the_file == INVALID_HANDLE_VALUE) { rc = ERROR_FUNCTION_FAILED; goto end; } WriteFile(the_file,data,size,&write,NULL); CloseHandle(the_file); TRACE("wrote %li bytes to %s\n",write,debugstr_w(source)); end: HeapFree(GetProcessHeap(),0,data); return rc; } /*********************************************************************** * extract_cabinet_file * * Extract files from a cab file. */ static void (WINAPI *pExtractFiles)( LPSTR, LPSTR, DWORD, DWORD, DWORD, DWORD ); static BOOL extract_cabinet_file_advpack( const WCHAR *cabinet, const WCHAR *root) { static HMODULE advpack; char *cab_path, *cab_file; if (!pExtractFiles) { if (!advpack && !(advpack = LoadLibraryA( "advpack.dll" ))) { ERR( "could not load advpack.dll\n" ); return FALSE; } if (!(pExtractFiles = (void *)GetProcAddress( advpack, "ExtractFiles" ))) { ERR( "could not find ExtractFiles in advpack.dll\n" ); return FALSE; } } if (!(cab_file = strdupWtoA( cabinet ))) return FALSE; if (!(cab_path = strdupWtoA( root ))) return FALSE; FIXME( "awful hack: extracting cabinet %s\n", debugstr_a(cab_file) ); pExtractFiles( cab_file, cab_path, 0, 0, 0, 0 ); HeapFree( GetProcessHeap(), 0, cab_file ); HeapFree( GetProcessHeap(), 0, cab_path ); return TRUE; } static BOOL extract_cabinet_file_cabinet( const WCHAR *cabinet, const WCHAR *root) { /* from cabinet.h */ struct ExtractFileList { LPSTR filename; struct ExtractFileList *next; BOOL unknown; /* always 1L */ } ; typedef struct { long result1; /* 0x000 */ long unknown1[3]; /* 0x004 */ struct ExtractFileList* filelist; /* 0x010 */ long filecount; /* 0x014 */ long unknown2; /* 0x018 */ char directory[0x104]; /* 0x01c */ char lastfile[0x20c]; /* 0x120 */ } EXTRACTdest; HRESULT WINAPI Extract(EXTRACTdest *dest, LPCSTR what); char *cab_path, *src_path; EXTRACTdest exd; struct ExtractFileList fl; if (!(cab_path = strdupWtoA( cabinet ))) return FALSE; if (!(src_path = strdupWtoA( root ))) return FALSE; memset(&exd,0,sizeof(exd)); strcpy(exd.directory,src_path); exd.unknown2 = 0x1; fl.filename = cab_path; fl.next = NULL; fl.unknown = 1; exd.filelist = &fl; FIXME( "more aweful hack: extracting cabinet %s\n", debugstr_a(cab_path) ); Extract(&exd,cab_path); HeapFree( GetProcessHeap(), 0, cab_path ); HeapFree( GetProcessHeap(), 0, src_path ); return TRUE; } static BOOL extract_cabinet_file(const WCHAR* source, const WCHAR* path) { TRACE("Extracting %s to %s\n",debugstr_w(source), debugstr_w(path)); if (!extract_cabinet_file_advpack(source,path)) return extract_cabinet_file_cabinet(source,path); return TRUE; } static UINT ready_media_for_file(MSIPACKAGE *package, UINT sequence, WCHAR* path) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; WCHAR source[MAX_PATH]; static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','*',' ', 'f','r','o','m',' ','M','e','d','i','a',' ', 'w','h','e','r','e',' ','L','a','s','t','S','e','q','u','e','n','c','e',' ','>','=',' ','%','i',' ', 'o','r','d','e','r',' ','b','y',' ','L','a','s','t','S','e','q','u','e','n','c','e',0}; WCHAR Query[1024]; WCHAR cab[0x100]; DWORD sz=0x100; INT seq; static INT last_sequence = 0; if (sequence <= last_sequence) { TRACE("Media already ready (%i, %i)\n",sequence,last_sequence); return ERROR_SUCCESS; } sprintfW(Query,ExecSeqQuery,sequence); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } seq = MSI_RecordGetInteger(row,2); last_sequence = seq; if (!MSI_RecordIsNull(row,4)) { sz=0x100; MSI_RecordGetStringW(row,4,cab,&sz); TRACE("Source is CAB %s\n",debugstr_w(cab)); /* the stream does not contain the # character */ if (cab[0]=='#') { writeout_cabinet_stream(package,&cab[1],source); strcpyW(path,source); *(strrchrW(path,'\\')+1)=0; } else { sz = MAX_PATH; if (MSI_GetPropertyW(package, cszSourceDir, source, &sz)) { ERR("No Source dir defined \n"); rc = ERROR_FUNCTION_FAILED; } else { strcpyW(path,source); strcatW(source,cab); /* extract the cab file into a folder in the temp folder */ sz = MAX_PATH; if (MSI_GetPropertyW(package, cszTempFolder,path, &sz) != ERROR_SUCCESS) GetTempPathW(MAX_PATH,path); } } rc = !extract_cabinet_file(source,path); } msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } inline static UINT create_component_directory ( MSIPACKAGE* package, INT component) { UINT rc; MSIFOLDER *folder; WCHAR install_path[MAX_PATH]; rc = resolve_folder(package, package->components[component].Directory, install_path, FALSE, FALSE, &folder); if (rc != ERROR_SUCCESS) return rc; /* create the path */ if (folder->State == 0) { create_full_pathW(install_path); folder->State = 2; } return rc; } static UINT ACTION_InstallFiles(MSIPACKAGE *package) { UINT rc = ERROR_SUCCESS; INT index; MSIRECORD * uirow; WCHAR uipath[MAX_PATH]; if (!package) return ERROR_INVALID_HANDLE; /* increment progress bar each time action data is sent */ ui_progress(package,1,1,1,0); for (index = 0; index < package->loaded_files; index++) { WCHAR path_to_source[MAX_PATH]; MSIFILE *file; file = &package->files[index]; if (file->Temporary) continue; if (!package->components[file->ComponentIndex].Enabled || !package->components[file->ComponentIndex].FeatureState) { TRACE("File %s is not scheduled for install\n", debugstr_w(file->File)); continue; } if ((file->State == 1) || (file->State == 2)) { TRACE("Installing %s\n",debugstr_w(file->File)); rc = ready_media_for_file(package,file->Sequence,path_to_source); /* * WARNING! * our file table could change here because a new temp file * may have been created */ file = &package->files[index]; if (rc != ERROR_SUCCESS) { ERR("Unable to ready media\n"); rc = ERROR_FUNCTION_FAILED; break; } create_component_directory( package, file->ComponentIndex); strcpyW(file->SourcePath, path_to_source); strcatW(file->SourcePath, file->File); TRACE("file paths %s to %s\n",debugstr_w(file->SourcePath), debugstr_w(file->TargetPath)); /* the UI chunk */ uirow=MSI_CreateRecord(9); MSI_RecordSetStringW(uirow,1,file->File); strcpyW(uipath,file->TargetPath); *(strrchrW(uipath,'\\')+1)=0; MSI_RecordSetStringW(uirow,9,uipath); MSI_RecordSetInteger(uirow,6,file->FileSize); ui_actiondata(package,szInstallFiles,uirow); msiobj_release( &uirow->hdr ); rc = !MoveFileW(file->SourcePath,file->TargetPath); ui_progress(package,2,0,0,0); if (rc) { ERR("Unable to move file (error %li)\n",GetLastError()); rc = ERROR_SUCCESS; } else file->State = 4; } } return rc; } inline static UINT get_file_target(MSIPACKAGE *package, LPCWSTR file_key, LPWSTR file_source) { INT index; if (!package) return ERROR_INVALID_HANDLE; for (index = 0; index < package->loaded_files; index ++) { if (strcmpW(file_key,package->files[index].File)==0) { if (package->files[index].State >= 3) { strcpyW(file_source,package->files[index].TargetPath); return ERROR_SUCCESS; } else return ERROR_FILE_NOT_FOUND; } } return ERROR_FUNCTION_FAILED; } static UINT ACTION_DuplicateFiles(MSIPACKAGE *package) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','*',' ','f','r','o','m',' ', 'D','u','p','l','i','c','a','t','e','F','i','l','e',0}; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR file_key[0x100]; WCHAR file_source[MAX_PATH]; WCHAR dest_name[0x100]; WCHAR dest_path[MAX_PATH]; WCHAR component[0x100]; INT component_index; DWORD sz=0x100; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz=0x100; rc = MSI_RecordGetStringW(row,2,component,&sz); if (rc != ERROR_SUCCESS) { ERR("Unable to get component\n"); msiobj_release(&row->hdr); break; } component_index = get_loaded_component(package,component); if (!package->components[component_index].Enabled || !package->components[component_index].FeatureState) { TRACE("Skipping copy due to disabled component\n"); msiobj_release(&row->hdr); continue; } sz=0x100; rc = MSI_RecordGetStringW(row,3,file_key,&sz); if (rc != ERROR_SUCCESS) { ERR("Unable to get file key\n"); msiobj_release(&row->hdr); break; } rc = get_file_target(package,file_key,file_source); if (rc != ERROR_SUCCESS) { ERR("Original file unknown %s\n",debugstr_w(file_key)); msiobj_release(&row->hdr); break; } if (MSI_RecordIsNull(row,4)) { strcpyW(dest_name,strrchrW(file_source,'\\')+1); } else { sz=0x100; MSI_RecordGetStringW(row,4,dest_name,&sz); reduce_to_longfilename(dest_name); } if (MSI_RecordIsNull(row,5)) { strcpyW(dest_path,file_source); *strrchrW(dest_path,'\\')=0; } else { WCHAR destkey[0x100]; sz=0x100; MSI_RecordGetStringW(row,5,destkey,&sz); sz = 0x100; rc = resolve_folder(package, destkey, dest_path,FALSE,FALSE,NULL); if (rc != ERROR_SUCCESS) { ERR("Unable to get destination folder\n"); msiobj_release(&row->hdr); break; } } strcatW(dest_path,dest_name); TRACE("Duplicating file %s to %s\n",debugstr_w(file_source), debugstr_w(dest_path)); if (strcmpW(file_source,dest_path)) rc = !CopyFileW(file_source,dest_path,TRUE); else rc = ERROR_SUCCESS; if (rc != ERROR_SUCCESS) ERR("Failed to copy file\n"); FIXME("We should track these duplicate files as well\n"); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } /* OK this value is "interpretted" and then formatted based on the first few characters */ static LPSTR parse_value(MSIPACKAGE *package, WCHAR *value, DWORD *type, DWORD *size) { LPSTR data = NULL; if (value[0]=='#' && value[1]!='#' && value[1]!='%') { if (value[1]=='x') { LPWSTR ptr; CHAR byte[5]; LPWSTR deformated; int count; deformat_string(package, &value[2], &deformated); /* binary value type */ ptr = deformated; *type=REG_BINARY; *size = strlenW(ptr)/2; data = HeapAlloc(GetProcessHeap(),0,*size); byte[0] = '0'; byte[1] = 'x'; byte[4] = 0; count = 0; while (*ptr) { byte[2]= *ptr; ptr++; byte[3]= *ptr; ptr++; data[count] = (BYTE)strtol(byte,NULL,0); count ++; } HeapFree(GetProcessHeap(),0,deformated); TRACE("Data %li bytes(%i)\n",*size,count); } else { LPWSTR deformated; deformat_string(package, &value[1], &deformated); *type=REG_DWORD; *size = sizeof(DWORD); data = HeapAlloc(GetProcessHeap(),0,*size); *(LPDWORD)data = atoiW(deformated); TRACE("DWORD %i\n",*data); HeapFree(GetProcessHeap(),0,deformated); } } else { WCHAR *ptr; *type=REG_SZ; if (value[0]=='#') { if (value[1]=='%') { ptr = &value[2]; *type=REG_EXPAND_SZ; } else ptr = &value[1]; } else ptr=value; *size = deformat_string(package, ptr,(LPWSTR*)&data); } return data; } static UINT ACTION_WriteRegistryValues(MSIPACKAGE *package) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = { 's','e','l','e','c','t',' ','*',' ', 'f','r','o','m',' ','R','e','g','i','s','t','r','y',0 }; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } /* increment progress bar each time action data is sent */ ui_progress(package,1,1,1,0); while (1) { static const WCHAR szHCR[] = {'H','K','E','Y','_','C','L','A','S','S','E','S','_','R','O','O','T','\\',0}; static const WCHAR szHCU[] = {'H','K','E','Y','_','C','U','R','R','E','N','T','_','U','S','E','R','\\',0}; static const WCHAR szHLM[] = {'H','K','E','Y','_','L','O','C','A','L','_','M','A','C','H','I','N','E', '\\',0}; static const WCHAR szHU[] = {'H','K','E','Y','_','U','S','E','R','S','\\',0}; WCHAR key[0x100]; WCHAR name[0x100]; LPWSTR value; LPSTR value_data = NULL; HKEY root_key, hkey; DWORD type,size; WCHAR component[0x100]; INT component_index; MSIRECORD * uirow; WCHAR uikey[0x110]; INT root; DWORD sz=0x100; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz= 0x100; MSI_RecordGetStringW(row,6,component,&sz); component_index = get_loaded_component(package,component); if (!package->components[component_index].Enabled || !package->components[component_index].FeatureState) { TRACE("Skipping write due to disabled component\n"); msiobj_release(&row->hdr); continue; } /* null values have special meanings during uninstalls and such */ if(MSI_RecordIsNull(row,5)) { msiobj_release(&row->hdr); continue; } root = MSI_RecordGetInteger(row,2); sz = 0x100; MSI_RecordGetStringW(row,3,key,&sz); sz = 0x100; if (MSI_RecordIsNull(row,4)) name[0]=0; else MSI_RecordGetStringW(row,4,name,&sz); /* get the root key */ switch (root) { case 0: root_key = HKEY_CLASSES_ROOT; strcpyW(uikey,szHCR); break; case 1: root_key = HKEY_CURRENT_USER; strcpyW(uikey,szHCU); break; case 2: root_key = HKEY_LOCAL_MACHINE; strcpyW(uikey,szHLM); break; case 3: root_key = HKEY_USERS; strcpyW(uikey,szHU); break; default: ERR("Unknown root %i\n",root); root_key=NULL; break; } if (!root_key) { msiobj_release(&row->hdr); continue; } strcatW(uikey,key); if (RegCreateKeyW( root_key, key, &hkey)) { ERR("Could not create key %s\n",debugstr_w(key)); msiobj_release(&row->hdr); continue; } value = load_dynamic_stringW(row,5); value_data = parse_value(package, value, &type, &size); if (value_data) { TRACE("Setting value %s\n",debugstr_w(name)); RegSetValueExW(hkey, name, 0, type, value_data, size); uirow = MSI_CreateRecord(3); MSI_RecordSetStringW(uirow,2,name); MSI_RecordSetStringW(uirow,1,uikey); if (type == REG_SZ) MSI_RecordSetStringW(uirow,3,(LPWSTR)value_data); else MSI_RecordSetStringW(uirow,3,value); ui_actiondata(package,szWriteRegistryValues,uirow); ui_progress(package,2,0,0,0); msiobj_release( &uirow->hdr ); HeapFree(GetProcessHeap(),0,value_data); } HeapFree(GetProcessHeap(),0,value); msiobj_release(&row->hdr); RegCloseKey(hkey); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } /* * This helper function should probably go alot of places * * Thinking about this, maybe this should become yet another Bison file */ static DWORD deformat_string(MSIPACKAGE *package, WCHAR* ptr,WCHAR** data) { WCHAR* mark=NULL; DWORD size=0; DWORD chunk=0; WCHAR key[0x100]; LPWSTR value; DWORD sz; UINT rc; /* scan for special characters */ if (!strchrW(ptr,'[') || (strchrW(ptr,'[') && !strchrW(ptr,']'))) { /* not formatted */ size = (strlenW(ptr)+1) * sizeof(WCHAR); *data = HeapAlloc(GetProcessHeap(),0,size); strcpyW(*data,ptr); return size; } /* formatted string located */ mark = strchrW(ptr,'['); if (mark != ptr) { INT cnt = (mark - ptr); TRACE("%i (%i) characters before marker\n",cnt,(mark-ptr)); size = cnt * sizeof(WCHAR); size += sizeof(WCHAR); *data = HeapAlloc(GetProcessHeap(),0,size); strncpyW(*data,ptr,cnt); (*data)[cnt]=0; } else { size = sizeof(WCHAR); *data = HeapAlloc(GetProcessHeap(),0,size); (*data)[0]=0; } mark++; strcpyW(key,mark); *strchrW(key,']')=0; mark = strchrW(mark,']'); mark++; TRACE("Current %s .. %s\n",debugstr_w(*data),debugstr_w(mark)); sz = 0; rc = MSI_GetPropertyW(package, key, NULL, &sz); if ((rc == ERROR_SUCCESS) || (rc == ERROR_MORE_DATA)) { LPWSTR newdata; sz++; value = HeapAlloc(GetProcessHeap(),0,sz * sizeof(WCHAR)); MSI_GetPropertyW(package, key, value, &sz); chunk = (strlenW(value)+1) * sizeof(WCHAR); size+=chunk; newdata = HeapReAlloc(GetProcessHeap(),0,*data,size); *data = newdata; strcatW(*data,value); } TRACE("Current %s .. %s\n",debugstr_w(*data),debugstr_w(mark)); if (*mark!=0) { LPWSTR newdata; chunk = (strlenW(mark)+1) * sizeof(WCHAR); size+=chunk; newdata = HeapReAlloc(GetProcessHeap(),0,*data,size); *data = newdata; strcatW(*data,mark); } (*data)[strlenW(*data)]=0; TRACE("Current %s .. %s\n",debugstr_w(*data),debugstr_w(mark)); /* recursively do this to clean up */ mark = HeapAlloc(GetProcessHeap(),0,size); strcpyW(mark,*data); TRACE("String at this point %s\n",debugstr_w(mark)); size = deformat_string(package,mark,data); HeapFree(GetProcessHeap(),0,mark); return size; } static UINT ACTION_InstallInitialize(MSIPACKAGE *package) { WCHAR level[10000]; INT install_level; DWORD sz; INT i,j; DWORD rc; LPWSTR override = NULL; static const WCHAR addlocal[]={'A','D','D','L','O','C','A','L',0}; static const WCHAR all[]={'A','L','L',0}; static const WCHAR szlevel[] = { 'I','N','S','T','A','L','L','L','E','V','E','L',0}; static const WCHAR szAddLocal[] = { 'A','D','D','L','O','C','A','L',0}; /* I do not know if this is where it should happen.. but */ TRACE("Checking Install Level\n"); sz = 10000; if (MSI_GetPropertyW(package,szlevel,level,&sz)==ERROR_SUCCESS) install_level = atoiW(level); else install_level = 1; sz = 0; rc = MSI_GetPropertyW(package,szAddLocal,NULL,&sz); if (rc == ERROR_SUCCESS || rc == ERROR_MORE_DATA) { sz++; override = HeapAlloc(GetProcessHeap(),0,sz*sizeof(WCHAR)); MSI_GetPropertyW(package, addlocal,override,&sz); } /* * Components FeatureState defaults to FALSE. The idea is we want to * enable the component is ANY feature that uses it is enabled to install */ for(i = 0; i < package->loaded_features; i++) { BOOL feature_state= ((package->features[i].Level > 0) && (package->features[i].Level <= install_level)); if (override && (strcmpiW(override,all)==0 || strstrW(override,package->features[i].Feature))) { TRACE("Override of install level found\n"); feature_state = TRUE; package->features[i].Enabled = feature_state; } TRACE("Feature %s has a state of %i\n", debugstr_w(package->features[i].Feature), feature_state); for( j = 0; j < package->features[i].ComponentCount; j++) { package->components[package->features[i].Components[j]].FeatureState |= feature_state; } } if (override != NULL) HeapFree(GetProcessHeap(),0,override); /* * So basically we ONLY want to install a component if its Enabled AND * FeatureState are both TRUE */ return ERROR_SUCCESS; } static UINT ACTION_InstallValidate(MSIPACKAGE *package) { DWORD progress = 0; static const WCHAR q1[]={ 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ','R','e','g','i','s','t','r','y',0}; UINT rc; MSIQUERY * view; MSIRECORD * row = 0; TRACE(" InstallValidate \n"); rc = MSI_DatabaseOpenViewW(package->db, q1, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } progress +=1; msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); ui_progress(package,0,progress+package->loaded_files,0,0); return ERROR_SUCCESS; } static UINT ACTION_LaunchConditions(MSIPACKAGE *package) { UINT rc; MSIQUERY * view = NULL; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = { 'S','E','L','E','C','T',' ','*',' ', 'f','r','o','m',' ','L','a','u','n','c','h','C','o','n','d','i','t','i','o','n',0}; static const WCHAR title[]= {'I','n','s','t','a','l','l',' ','F','a', 'i','l','e','d',0}; TRACE("Checking launch conditions\n"); rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = ERROR_SUCCESS; while (rc == ERROR_SUCCESS) { LPWSTR cond = NULL; LPWSTR message = NULL; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } cond = load_dynamic_stringW(row,1); if (MSI_EvaluateConditionW(package,cond) != MSICONDITION_TRUE) { message = load_dynamic_stringW(row,2); MessageBoxW(NULL,message,title,MB_OK); HeapFree(GetProcessHeap(),0,message); rc = ERROR_FUNCTION_FAILED; } HeapFree(GetProcessHeap(),0,cond); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static void resolve_keypath( MSIPACKAGE* package, INT component_index, WCHAR *keypath) { MSICOMPONENT* cmp = &package->components[component_index]; if (cmp->KeyPath[0]==0) { resolve_folder(package,cmp->Directory,keypath,FALSE,FALSE,NULL); return; } if ((cmp->Attributes & 0x4) || (cmp->Attributes & 0x20)) { FIXME("UNIMPLEMENTED keypath as Registry or ODBC Source\n"); keypath[0]=0; } else { int j; j = get_loaded_file(package,cmp->KeyPath); if (j>=0) strcpyW(keypath,package->files[j].TargetPath); } } /* * Ok further analysis makes me think that this work is * actually done in the PublishComponents and PublishFeatures * step, and not here. It appears like the keypath and all that is * resolved in this step, however actually written in the Publish steps. * But we will leave it here for now because it is unclear */ static UINT ACTION_ProcessComponents(MSIPACKAGE *package) { WCHAR productcode[0x100]; WCHAR squished_pc[0x100]; WCHAR squished_cc[0x100]; DWORD sz; UINT rc; INT i; HKEY hkey=0,hkey2=0,hkey3=0; static const WCHAR szProductCode[]= {'P','r','o','d','u','c','t','C','o','d','e',0}; static const WCHAR szInstaller[] = { 'S','o','f','t','w','a','r','e','\\', 'M','i','c','r','o','s','o','f','t','\\', 'W','i','n','d','o','w','s','\\', 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\', 'I','n','s','t','a','l','l','e','r',0 }; static const WCHAR szFeatures[] = { 'F','e','a','t','u','r','e','s',0 }; static const WCHAR szComponents[] = { 'C','o','m','p','o','n','e','n','t','s',0 }; if (!package) return ERROR_INVALID_HANDLE; /* writes the Component and Features values to the registry */ sz = 0x100; rc = MSI_GetPropertyW(package,szProductCode,productcode,&sz); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; squash_guid(productcode,squished_pc); rc = RegCreateKeyW(HKEY_LOCAL_MACHINE,szInstaller,&hkey); if (rc != ERROR_SUCCESS) goto end; rc = RegCreateKeyW(hkey,szFeatures,&hkey2); if (rc != ERROR_SUCCESS) goto end; rc = RegCreateKeyW(hkey2,squished_pc,&hkey3); if (rc != ERROR_SUCCESS) goto end; /* here the guids are base 85 encoded */ for (i = 0; i < package->loaded_features; i++) { LPWSTR data = NULL; GUID clsid; int j; INT size; size = package->features[i].ComponentCount*21*sizeof(WCHAR); data = HeapAlloc(GetProcessHeap(), 0, size); data[0] = 0; for (j = 0; j < package->features[i].ComponentCount; j++) { WCHAR buf[21]; TRACE("From %s\n",debugstr_w(package->components [package->features[i].Components[j]].ComponentId)); CLSIDFromString(package->components [package->features[i].Components[j]].ComponentId, &clsid); encode_base85_guid(&clsid,buf); TRACE("to %s\n",debugstr_w(buf)); strcatW(data,buf); } size = strlenW(data)*sizeof(WCHAR); RegSetValueExW(hkey3,package->features[i].Feature,0,REG_SZ, (LPSTR)data,size); HeapFree(GetProcessHeap(),0,data); } RegCloseKey(hkey3); RegCloseKey(hkey2); rc = RegCreateKeyW(hkey,szComponents,&hkey2); if (rc != ERROR_SUCCESS) goto end; for (i = 0; i < package->loaded_components; i++) { if (package->components[i].ComponentId[0]!=0) { WCHAR keypath[0x1000]; MSIRECORD * uirow; squash_guid(package->components[i].ComponentId,squished_cc); rc = RegCreateKeyW(hkey2,squished_cc,&hkey3); if (rc != ERROR_SUCCESS) continue; resolve_keypath(package,i,keypath); RegSetValueExW(hkey3,squished_pc,0,REG_SZ,(LPVOID)keypath, (strlenW(keypath)+1)*sizeof(WCHAR)); RegCloseKey(hkey3); /* UI stuff */ uirow = MSI_CreateRecord(3); MSI_RecordSetStringW(uirow,1,productcode); MSI_RecordSetStringW(uirow,2,package->components[i].ComponentId); MSI_RecordSetStringW(uirow,3,keypath); ui_actiondata(package,szProcessComponents,uirow); msiobj_release( &uirow->hdr ); } } end: RegCloseKey(hkey2); RegCloseKey(hkey); return rc; } static UINT ACTION_RegisterTypeLibraries(MSIPACKAGE *package) { /* * OK this is a bit confusing.. I am given a _Component key and I believe * that the file that is being registered as a type library is the "key file * of that component" which I interpret to mean "The file in the KeyPath of * that component". */ UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR Query[] = { 'S','E','L','E','C','T',' ','*',' ', 'f','r','o','m',' ','T','y','p','e','L','i','b',0}; ITypeLib *ptLib; HRESULT res; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR component[0x100]; DWORD sz; INT index; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz = 0x100; MSI_RecordGetStringW(row,3,component,&sz); index = get_loaded_component(package,component); if (index < 0) { msiobj_release(&row->hdr); continue; } if (!package->components[index].Enabled || !package->components[index].FeatureState) { TRACE("Skipping typelib reg due to disabled component\n"); msiobj_release(&row->hdr); continue; } index = get_loaded_file(package,package->components[index].KeyPath); if (index < 0) { msiobj_release(&row->hdr); continue; } res = LoadTypeLib(package->files[index].TargetPath,&ptLib); if (SUCCEEDED(res)) { WCHAR help[MAX_PATH]; WCHAR helpid[0x100]; sz = 0x100; MSI_RecordGetStringW(row,6,helpid,&sz); resolve_folder(package,helpid,help,FALSE,FALSE,NULL); res = RegisterTypeLib(ptLib,package->files[index].TargetPath,help); if (!SUCCEEDED(res)) ERR("Failed to register type library %s\n", debugstr_w(package->files[index].TargetPath)); else { /* Yes the row has more fields than I need, but #1 is correct and the only one I need. Why make a new row? */ ui_actiondata(package,szRegisterTypeLibraries,row); TRACE("Registered %s\n", debugstr_w(package->files[index].TargetPath)); } if (ptLib) ITypeLib_Release(ptLib); } else ERR("Failed to load type library %s\n", debugstr_w(package->files[index].TargetPath)); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static UINT register_appid(MSIPACKAGE *package, LPCWSTR clsid, LPCWSTR app ) { static const WCHAR szAppID[] = { 'A','p','p','I','D',0 }; UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = {'S','E','L','E','C','T',' ','*',' ','f','r','o','m',' ','A','p','p','I' ,'d',' ','w','h','e','r','e',' ','A','p','p','I','d','=','`','%','s','`',0}; WCHAR Query[0x1000]; HKEY hkey2,hkey3; LPWSTR buffer=0; if (!package) return ERROR_INVALID_HANDLE; sprintfW(Query,ExecSeqQuery,clsid); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } RegCreateKeyW(HKEY_CLASSES_ROOT,szAppID,&hkey2); RegCreateKeyW(hkey2,clsid,&hkey3); RegSetValueExW(hkey3,NULL,0,REG_SZ,(LPVOID)app, (strlenW(app)+1)*sizeof(WCHAR)); rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } if (!MSI_RecordIsNull(row,2)) { LPWSTR deformated=0; UINT size; static const WCHAR szRemoteServerName[] = {'R','e','m','o','t','e','S','e','r','v','e','r','N','a','m','e',0}; buffer = load_dynamic_stringW(row,2); size = deformat_string(package,buffer,&deformated); RegSetValueExW(hkey3,szRemoteServerName,0,REG_SZ,(LPVOID)deformated, size); HeapFree(GetProcessHeap(),0,deformated); HeapFree(GetProcessHeap(),0,buffer); } if (!MSI_RecordIsNull(row,3)) { static const WCHAR szLocalService[] = {'L','o','c','a','l','S','e','r','v','i','c','e',0}; UINT size; buffer = load_dynamic_stringW(row,3); size = (strlenW(buffer)+1) * sizeof(WCHAR); RegSetValueExW(hkey3,szLocalService,0,REG_SZ,(LPVOID)buffer,size); HeapFree(GetProcessHeap(),0,buffer); } if (!MSI_RecordIsNull(row,4)) { static const WCHAR szService[] = {'S','e','r','v','i','c','e','P','a','r','a','m','e','t','e','r','s',0}; UINT size; buffer = load_dynamic_stringW(row,4); size = (strlenW(buffer)+1) * sizeof(WCHAR); RegSetValueExW(hkey3,szService,0,REG_SZ,(LPVOID)buffer,size); HeapFree(GetProcessHeap(),0,buffer); } if (!MSI_RecordIsNull(row,5)) { static const WCHAR szDLL[] = {'D','l','l','S','u','r','r','o','g','a','t','e',0}; UINT size; buffer = load_dynamic_stringW(row,5); size = (strlenW(buffer)+1) * sizeof(WCHAR); RegSetValueExW(hkey3,szDLL,0,REG_SZ,(LPVOID)buffer,size); HeapFree(GetProcessHeap(),0,buffer); } if (!MSI_RecordIsNull(row,6)) { static const WCHAR szActivate[] = {'A','c','t','i','v','a','t','e','A','s','S','t','o','r','a','g','e',0}; static const WCHAR szY[] = {'Y',0}; if (MSI_RecordGetInteger(row,6)) RegSetValueExW(hkey3,szActivate,0,REG_SZ,(LPVOID)szY,4); } if (!MSI_RecordIsNull(row,7)) { static const WCHAR szRunAs[] = {'R','u','n','A','s',0}; static const WCHAR szUser[] = {'I','n','t','e','r','a','c','t','i','v','e',' ','U','s','e','r',0}; if (MSI_RecordGetInteger(row,7)) RegSetValueExW(hkey3,szRunAs,0,REG_SZ,(LPVOID)szUser,34); } msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); RegCloseKey(hkey3); RegCloseKey(hkey2); return rc; } static UINT ACTION_RegisterClassInfo(MSIPACKAGE *package) { /* * Again I am assuming the words, "Whose key file represents" when referring * to a Component as to meaning that Components KeyPath file * * Also there is a very strong connection between ClassInfo and ProgID * that I am mostly glossing over. * What would be more propper is to load the ClassInfo and the ProgID info * into memory data structures and then be able to enable and disable them * based on component. */ UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = { 'S','E','L','E','C','T',' ','*',' ', 'f','r','o','m',' ','C','l','a','s','s',0}; static const WCHAR szCLSID[] = { 'C','L','S','I','D',0 }; static const WCHAR szProgID[] = { 'P','r','o','g','I','D',0 }; static const WCHAR szAppID[] = { 'A','p','p','I','D',0 }; HKEY hkey,hkey2,hkey3; if (!package) return ERROR_INVALID_HANDLE; rc = RegCreateKeyW(HKEY_CLASSES_ROOT,szCLSID,&hkey); if (rc != ERROR_SUCCESS) return ERROR_FUNCTION_FAILED; rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; goto end; } rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); goto end; } while (1) { WCHAR clsid[0x100]; WCHAR buffer[0x100]; WCHAR desc[0x100]; DWORD sz; INT index; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz=0x100; MSI_RecordGetStringW(row,3,buffer,&sz); index = get_loaded_component(package,buffer); if (index < 0) { msiobj_release(&row->hdr); continue; } if (!package->components[index].Enabled || !package->components[index].FeatureState) { TRACE("Skipping class reg due to disabled component\n"); msiobj_release(&row->hdr); continue; } sz=0x100; MSI_RecordGetStringW(row,1,clsid,&sz); RegCreateKeyW(hkey,clsid,&hkey2); if (!MSI_RecordIsNull(row,5)) { sz=0x100; MSI_RecordGetStringW(row,5,desc,&sz); RegSetValueExW(hkey2,NULL,0,REG_SZ,(LPVOID)desc, (strlenW(desc)+1)*sizeof(WCHAR)); } else desc[0]=0; sz=0x100; MSI_RecordGetStringW(row,2,buffer,&sz); RegCreateKeyW(hkey2,buffer,&hkey3); index = get_loaded_file(package,package->components[index].KeyPath); RegSetValueExW(hkey3,NULL,0,REG_SZ, (LPVOID)package->files[index].TargetPath, (strlenW(package->files[index].TargetPath)+1) *sizeof(WCHAR)); RegCloseKey(hkey3); if (!MSI_RecordIsNull(row,4)) { sz=0x100; MSI_RecordGetStringW(row,4,buffer,&sz); RegCreateKeyW(hkey2,szProgID,&hkey3); RegSetValueExW(hkey3,NULL,0,REG_SZ,(LPVOID)buffer, (strlenW(buffer)+1)*sizeof(WCHAR)); RegCloseKey(hkey3); } if (!MSI_RecordIsNull(row,6)) { sz=0x100; MSI_RecordGetStringW(row,6,buffer,&sz); RegSetValueExW(hkey2,szAppID,0,REG_SZ,(LPVOID)buffer, (strlenW(buffer)+1)*sizeof(WCHAR)); register_appid(package,buffer,desc); } RegCloseKey(hkey2); FIXME("Process the rest of the fields >7\n"); ui_actiondata(package,szRegisterClassInfo,row); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); end: RegCloseKey(hkey); return rc; } static UINT register_progid_base(MSIRECORD * row, LPWSTR clsid) { static const WCHAR szCLSID[] = { 'C','L','S','I','D',0 }; HKEY hkey,hkey2; WCHAR buffer[0x100]; DWORD sz; sz = 0x100; MSI_RecordGetStringW(row,1,buffer,&sz); RegCreateKeyW(HKEY_CLASSES_ROOT,buffer,&hkey); if (!MSI_RecordIsNull(row,4)) { sz = 0x100; MSI_RecordGetStringW(row,4,buffer,&sz); RegSetValueExW(hkey,NULL,0,REG_SZ,(LPVOID)buffer, (strlenW(buffer)+1) * sizeof(WCHAR)); } if (!MSI_RecordIsNull(row,3)) { sz = 0x100; MSI_RecordGetStringW(row,3,buffer,&sz); RegCreateKeyW(hkey,szCLSID,&hkey2); RegSetValueExW(hkey2,NULL,0,REG_SZ,(LPVOID)buffer, (strlenW(buffer)+1) * sizeof(WCHAR)); if (clsid) strcpyW(clsid,buffer); RegCloseKey(hkey2); } else { FIXME("UNHANDLED case, Parent progid but classid is NULL\n"); return ERROR_FUNCTION_FAILED; } if (!MSI_RecordIsNull(row,5)) FIXME ("UNHANDLED icon in Progid\n"); return ERROR_SUCCESS; } static UINT register_progid(MSIPACKAGE *package, MSIRECORD * row, LPWSTR clsid); static UINT register_parent_progid(MSIPACKAGE *package, LPCWSTR parent, LPWSTR clsid) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR Query_t[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','P','r','o','g' ,'I','d',' ','w','h','e','r','e',' ','P','r','o','g','I','d',' ','=',' ','`' ,'%','s','`',0}; WCHAR Query[0x1000]; if (!package) return ERROR_INVALID_HANDLE; sprintfW(Query,Query_t,parent); rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return rc; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } register_progid(package,row,clsid); msiobj_release(&row->hdr); MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static UINT register_progid(MSIPACKAGE *package, MSIRECORD * row, LPWSTR clsid) { UINT rc = ERROR_SUCCESS; if (MSI_RecordIsNull(row,2)) rc = register_progid_base(row,clsid); else { WCHAR buffer[0x1000]; DWORD sz; HKEY hkey,hkey2; static const WCHAR szCLSID[] = { 'C','L','S','I','D',0 }; sz = 0x100; MSI_RecordGetStringW(row,2,buffer,&sz); rc = register_parent_progid(package,buffer,clsid); sz = 0x100; MSI_RecordGetStringW(row,1,buffer,&sz); RegCreateKeyW(HKEY_CLASSES_ROOT,buffer,&hkey); /* clasid is same as parent */ RegCreateKeyW(hkey,szCLSID,&hkey2); RegSetValueExW(hkey2,NULL,0,REG_SZ,(LPVOID)clsid, (strlenW(clsid)+1) * sizeof(WCHAR)); RegCloseKey(hkey2); if (!MSI_RecordIsNull(row,4)) { sz = 0x100; MSI_RecordGetStringW(row,4,buffer,&sz); RegSetValueExW(hkey,NULL,0,REG_SZ,(LPVOID)buffer, (strlenW(buffer)+1) * sizeof(WCHAR)); } if (!MSI_RecordIsNull(row,5)) FIXME ("UNHANDLED icon in Progid\n"); RegCloseKey(hkey); } return rc; } static UINT ACTION_RegisterProgIdInfo(MSIPACKAGE *package) { /* * Sigh, here I am just brute force registering all progids * this needs to be linked to the Classes that have been registered * but the easiest way to do that is to load all these stuff into * memory for easy checking. * * Gives me something to continue to work toward. */ UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR Query[] = { 'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ','P','r','o','g','I','d',0}; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR clsid[0x1000]; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } register_progid(package,row,clsid); ui_actiondata(package,szRegisterProgIdInfo,row); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } static UINT build_icon_path(MSIPACKAGE *package, LPCWSTR icon_name, LPWSTR FilePath) { WCHAR ProductCode[0x100]; WCHAR SystemFolder[MAX_PATH]; DWORD sz; static const WCHAR szInstaller[] = {'I','n','s','t','a','l','l','e','r','\\',0}; static const WCHAR szProductCode[] = {'P','r','o','d','u','c','t','C','o','d','e',0}; static const WCHAR szFolder[] = {'W','i','n','d','o','w','s','F','o','l','d','e','r',0}; sz = 0x100; MSI_GetPropertyW(package,szProductCode,ProductCode,&sz); if (strlenW(ProductCode)==0) return ERROR_FUNCTION_FAILED; sz = MAX_PATH; MSI_GetPropertyW(package,szFolder,SystemFolder,&sz); strcatW(SystemFolder,szInstaller); strcatW(SystemFolder,ProductCode); create_full_pathW(SystemFolder); strcpyW(FilePath,SystemFolder); strcatW(FilePath,cszbs); strcatW(FilePath,icon_name); return ERROR_SUCCESS; } static UINT ACTION_CreateShortcuts(MSIPACKAGE *package) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR Query[] = { 'S','E','L','E','C','T',' ','*',' ','f','r','o','m',' ', 'S','h','o','r','t','c','u','t',0}; IShellLinkW *sl; IPersistFile *pf; HRESULT res; if (!package) return ERROR_INVALID_HANDLE; res = CoInitialize( NULL ); if (FAILED (res)) { ERR("CoInitialize failed\n"); return ERROR_FUNCTION_FAILED; } rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { WCHAR target_file[MAX_PATH]; WCHAR buffer[0x100]; DWORD sz; DWORD index; static const WCHAR szlnk[]={'.','l','n','k',0}; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz = 0x100; MSI_RecordGetStringW(row,4,buffer,&sz); index = get_loaded_component(package,buffer); if (index < 0) { msiobj_release(&row->hdr); continue; } if (!package->components[index].Enabled || !package->components[index].FeatureState) { TRACE("Skipping shortcut creation due to disabled component\n"); msiobj_release(&row->hdr); continue; } ui_actiondata(package,szCreateShortcuts,row); res = CoCreateInstance( &CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, &IID_IShellLinkW, (LPVOID *) &sl ); if (FAILED(res)) { ERR("Is IID_IShellLink\n"); msiobj_release(&row->hdr); continue; } res = IShellLinkW_QueryInterface( sl, &IID_IPersistFile,(LPVOID*) &pf ); if( FAILED( res ) ) { ERR("Is IID_IPersistFile\n"); msiobj_release(&row->hdr); continue; } sz = 0x100; MSI_RecordGetStringW(row,2,buffer,&sz); resolve_folder(package, buffer,target_file,FALSE,FALSE,NULL); sz = 0x100; MSI_RecordGetStringW(row,3,buffer,&sz); reduce_to_longfilename(buffer); strcatW(target_file,buffer); if (!strchrW(target_file,'.')) strcatW(target_file,szlnk); sz = 0x100; MSI_RecordGetStringW(row,5,buffer,&sz); if (strchrW(buffer,'[')) { LPWSTR deformated; deformat_string(package,buffer,&deformated); IShellLinkW_SetPath(sl,deformated); HeapFree(GetProcessHeap(),0,deformated); } else { FIXME("UNHANDLED shortcut format, advertised shortcut\n"); IPersistFile_Release( pf ); IShellLinkW_Release( sl ); msiobj_release(&row->hdr); continue; } if (!MSI_RecordIsNull(row,6)) { LPWSTR deformated; sz = 0x100; MSI_RecordGetStringW(row,6,buffer,&sz); deformat_string(package,buffer,&deformated); IShellLinkW_SetArguments(sl,deformated); HeapFree(GetProcessHeap(),0,deformated); } if (!MSI_RecordIsNull(row,7)) { LPWSTR deformated; deformated = load_dynamic_stringW(row,7); IShellLinkW_SetDescription(sl,deformated); HeapFree(GetProcessHeap(),0,deformated); } if (!MSI_RecordIsNull(row,8)) IShellLinkW_SetHotkey(sl,MSI_RecordGetInteger(row,8)); if (!MSI_RecordIsNull(row,9)) { WCHAR Path[MAX_PATH]; INT index; sz = 0x100; MSI_RecordGetStringW(row,9,buffer,&sz); build_icon_path(package,buffer,Path); index = MSI_RecordGetInteger(row,10); IShellLinkW_SetIconLocation(sl,Path,index); } if (!MSI_RecordIsNull(row,11)) IShellLinkW_SetShowCmd(sl,MSI_RecordGetInteger(row,11)); if (!MSI_RecordIsNull(row,12)) { WCHAR Path[MAX_PATH]; sz = 0x100; MSI_RecordGetStringW(row,12,buffer,&sz); resolve_folder(package, buffer, Path, FALSE, FALSE, NULL); IShellLinkW_SetWorkingDirectory(sl,Path); } TRACE("Writing shortcut to %s\n",debugstr_w(target_file)); IPersistFile_Save(pf,target_file,FALSE); IPersistFile_Release( pf ); IShellLinkW_Release( sl ); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); CoUninitialize(); return rc; } /* * 99% of the work done here is only done for * advertised installs. However this is where the * Icon table is processed and written out * so that is what I am going to do here. */ static UINT ACTION_PublishProduct(MSIPACKAGE *package) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR Query[]={ 'S','E','L','E','C','T',' ','*',' ', 'f','r','o','m',' ','I','c','o','n',0}; DWORD sz; if (!package) return ERROR_INVALID_HANDLE; rc = MSI_DatabaseOpenViewW(package->db, Query, &view); if (rc != ERROR_SUCCESS) return ERROR_SUCCESS; rc = MSI_ViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { HANDLE the_file; WCHAR FilePath[MAX_PATH]; WCHAR FileName[MAX_PATH]; CHAR buffer[1024]; rc = MSI_ViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } sz = MAX_PATH; MSI_RecordGetStringW(row,1,FileName,&sz); if (sz == 0) { ERR("Unable to get FileName\n"); msiobj_release(&row->hdr); continue; } build_icon_path(package,FileName,FilePath); TRACE("Creating icon file at %s\n",debugstr_w(FilePath)); the_file = CreateFileW(FilePath, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); if (the_file == INVALID_HANDLE_VALUE) { ERR("Unable to create file %s\n",debugstr_w(FilePath)); msiobj_release(&row->hdr); continue; } do { DWORD write; sz = 1024; rc = MSI_RecordReadStream(row,2,buffer,&sz); if (rc != ERROR_SUCCESS) { ERR("Failed to get stream\n"); CloseHandle(the_file); DeleteFileW(FilePath); break; } WriteFile(the_file,buffer,sz,&write,NULL); } while (sz == 1024); CloseHandle(the_file); msiobj_release(&row->hdr); } MSI_ViewClose(view); msiobj_release(&view->hdr); return rc; } /* Msi functions that seem appropriate here */ UINT WINAPI MsiDoActionA( MSIHANDLE hInstall, LPCSTR szAction ) { LPWSTR szwAction; UINT rc; TRACE(" exteral attempt at action %s\n",szAction); if (!szAction) return ERROR_FUNCTION_FAILED; if (hInstall == 0) return ERROR_FUNCTION_FAILED; szwAction = strdupAtoW(szAction); if (!szwAction) return ERROR_FUNCTION_FAILED; rc = MsiDoActionW(hInstall, szwAction); HeapFree(GetProcessHeap(),0,szwAction); return rc; } UINT WINAPI MsiDoActionW( MSIHANDLE hInstall, LPCWSTR szAction ) { MSIPACKAGE *package; UINT ret = ERROR_INVALID_HANDLE; TRACE(" external attempt at action %s \n",debugstr_w(szAction)); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if( package ) { ret = ACTION_PerformAction(package,szAction); msiobj_release( &package->hdr ); } return ret; } UINT WINAPI MsiGetTargetPathA( MSIHANDLE hInstall, LPCSTR szFolder, LPSTR szPathBuf, DWORD* pcchPathBuf) { LPWSTR szwFolder; LPWSTR szwPathBuf; UINT rc; TRACE("getting folder %s %p %li\n",szFolder,szPathBuf, *pcchPathBuf); if (!szFolder) return ERROR_FUNCTION_FAILED; if (hInstall == 0) return ERROR_FUNCTION_FAILED; szwFolder = strdupAtoW(szFolder); if (!szwFolder) return ERROR_FUNCTION_FAILED; szwPathBuf = HeapAlloc( GetProcessHeap(), 0 , *pcchPathBuf * sizeof(WCHAR)); rc = MsiGetTargetPathW(hInstall, szwFolder, szwPathBuf,pcchPathBuf); WideCharToMultiByte( CP_ACP, 0, szwPathBuf, *pcchPathBuf, szPathBuf, *pcchPathBuf, NULL, NULL ); HeapFree(GetProcessHeap(),0,szwFolder); HeapFree(GetProcessHeap(),0,szwPathBuf); return rc; } UINT WINAPI MsiGetTargetPathW( MSIHANDLE hInstall, LPCWSTR szFolder, LPWSTR szPathBuf, DWORD* pcchPathBuf) { WCHAR path[MAX_PATH]; UINT rc; MSIPACKAGE *package; TRACE("(%s %p %li)\n",debugstr_w(szFolder),szPathBuf,*pcchPathBuf); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if( !package ) return ERROR_INVALID_HANDLE; rc = resolve_folder(package, szFolder, path, FALSE, FALSE, NULL); msiobj_release( &package->hdr ); if (rc == ERROR_SUCCESS && strlenW(path) > *pcchPathBuf) { *pcchPathBuf = strlenW(path)+1; return ERROR_MORE_DATA; } else if (rc == ERROR_SUCCESS) { *pcchPathBuf = strlenW(path)+1; strcpyW(szPathBuf,path); TRACE("Returning Path %s\n",debugstr_w(path)); } return rc; } UINT WINAPI MsiGetSourcePathA( MSIHANDLE hInstall, LPCSTR szFolder, LPSTR szPathBuf, DWORD* pcchPathBuf) { LPWSTR szwFolder; LPWSTR szwPathBuf; UINT rc; TRACE("getting source %s %p %li\n",szFolder,szPathBuf, *pcchPathBuf); if (!szFolder) return ERROR_FUNCTION_FAILED; if (hInstall == 0) return ERROR_FUNCTION_FAILED; szwFolder = strdupAtoW(szFolder); if (!szwFolder) return ERROR_FUNCTION_FAILED; szwPathBuf = HeapAlloc( GetProcessHeap(), 0 , *pcchPathBuf * sizeof(WCHAR)); rc = MsiGetSourcePathW(hInstall, szwFolder, szwPathBuf,pcchPathBuf); WideCharToMultiByte( CP_ACP, 0, szwPathBuf, *pcchPathBuf, szPathBuf, *pcchPathBuf, NULL, NULL ); HeapFree(GetProcessHeap(),0,szwFolder); HeapFree(GetProcessHeap(),0,szwPathBuf); return rc; } UINT WINAPI MsiGetSourcePathW( MSIHANDLE hInstall, LPCWSTR szFolder, LPWSTR szPathBuf, DWORD* pcchPathBuf) { WCHAR path[MAX_PATH]; UINT rc; MSIPACKAGE *package; TRACE("(%s %p %li)\n",debugstr_w(szFolder),szPathBuf,*pcchPathBuf); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if( !package ) return ERROR_INVALID_HANDLE; rc = resolve_folder(package, szFolder, path, TRUE, FALSE, NULL); msiobj_release( &package->hdr ); if (rc == ERROR_SUCCESS && strlenW(path) > *pcchPathBuf) { *pcchPathBuf = strlenW(path)+1; return ERROR_MORE_DATA; } else if (rc == ERROR_SUCCESS) { *pcchPathBuf = strlenW(path)+1; strcpyW(szPathBuf,path); TRACE("Returning Path %s\n",debugstr_w(path)); } return rc; } UINT WINAPI MsiSetTargetPathA(MSIHANDLE hInstall, LPCSTR szFolder, LPCSTR szFolderPath) { LPWSTR szwFolder; LPWSTR szwFolderPath; UINT rc; if (!szFolder) return ERROR_FUNCTION_FAILED; if (hInstall == 0) return ERROR_FUNCTION_FAILED; szwFolder = strdupAtoW(szFolder); if (!szwFolder) return ERROR_FUNCTION_FAILED; szwFolderPath = strdupAtoW(szFolderPath); if (!szwFolderPath) { HeapFree(GetProcessHeap(),0,szwFolder); return ERROR_FUNCTION_FAILED; } rc = MsiSetTargetPathW(hInstall, szwFolder, szwFolderPath); HeapFree(GetProcessHeap(),0,szwFolder); HeapFree(GetProcessHeap(),0,szwFolderPath); return rc; } UINT MSI_SetTargetPathW(MSIPACKAGE *package, LPCWSTR szFolder, LPCWSTR szFolderPath) { INT i; WCHAR path[MAX_PATH]; MSIFOLDER *folder; TRACE("(%p %s %s)\n",package, debugstr_w(szFolder),debugstr_w(szFolderPath)); if (package==NULL) return ERROR_INVALID_HANDLE; if (szFolderPath[0]==0) return ERROR_FUNCTION_FAILED; if (GetFileAttributesW(szFolderPath) == INVALID_FILE_ATTRIBUTES) return ERROR_FUNCTION_FAILED; resolve_folder(package,szFolder,path,FALSE,FALSE,&folder); if (!folder) return ERROR_INVALID_PARAMETER; strcpyW(folder->Property,szFolderPath); for (i = 0; i < package->loaded_folders; i++) package->folders[i].ResolvedTarget[0]=0; for (i = 0; i < package->loaded_folders; i++) resolve_folder(package, package->folders[i].Directory, path, FALSE, TRUE, NULL); return ERROR_SUCCESS; } UINT WINAPI MsiSetTargetPathW(MSIHANDLE hInstall, LPCWSTR szFolder, LPCWSTR szFolderPath) { MSIPACKAGE *package; UINT ret; TRACE("(%s %s)\n",debugstr_w(szFolder),debugstr_w(szFolderPath)); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); ret = MSI_SetTargetPathW( package, szFolder, szFolderPath ); msiobj_release( &package->hdr ); return ret; } BOOL WINAPI MsiGetMode(MSIHANDLE hInstall, DWORD iRunMode) { FIXME("STUB (%li)\n",iRunMode); return FALSE; } /* * according to the docs when this is called it immediently recalculates all the * components states as well */ UINT WINAPI MsiSetFeatureStateA(MSIHANDLE hInstall, LPCSTR szFeature, INSTALLSTATE iState) { LPWSTR szwFeature = NULL; UINT rc; szwFeature = strdupAtoW(szFeature); if (!szwFeature) return ERROR_FUNCTION_FAILED; rc = MsiSetFeatureStateW(hInstall,szwFeature, iState); HeapFree(GetProcessHeap(),0,szwFeature); return rc; } UINT WINAPI MsiSetFeatureStateW(MSIHANDLE hInstall, LPCWSTR szFeature, INSTALLSTATE iState) { MSIPACKAGE* package; INT index; TRACE(" %s to %i\n",debugstr_w(szFeature), iState); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if (!package) return ERROR_INVALID_HANDLE; index = get_loaded_feature(package,szFeature); if (index < 0) return ERROR_UNKNOWN_FEATURE; package->features[index].State = iState; return ERROR_SUCCESS; } UINT WINAPI MsiGetFeatureStateA(MSIHANDLE hInstall, LPSTR szFeature, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { LPWSTR szwFeature = NULL; UINT rc; szwFeature = strdupAtoW(szFeature); rc = MsiGetFeatureStateW(hInstall,szwFeature,piInstalled, piAction); HeapFree( GetProcessHeap(), 0 , szwFeature); return rc; } UINT MSI_GetFeatureStateW(MSIPACKAGE *package, LPWSTR szFeature, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { INT index; index = get_loaded_feature(package,szFeature); if (index < 0) return ERROR_UNKNOWN_FEATURE; if (piInstalled) *piInstalled = package->features[index].State; if (piAction) { if (package->features[index].Enabled) *piAction = INSTALLSTATE_LOCAL; else *piAction = INSTALLSTATE_UNKNOWN; } return ERROR_SUCCESS; } UINT WINAPI MsiGetFeatureStateW(MSIHANDLE hInstall, LPWSTR szFeature, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { MSIPACKAGE* package; UINT ret; TRACE("%ld %s %p %p\n", hInstall, debugstr_w(szFeature), piInstalled, piAction); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if (!package) return ERROR_INVALID_HANDLE; ret = MSI_GetFeatureStateW(package, szFeature, piInstalled, piAction); msiobj_release( &package->hdr ); return ret; } UINT WINAPI MsiGetComponentStateA(MSIHANDLE hInstall, LPSTR szComponent, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { LPWSTR szwComponent= NULL; UINT rc; szwComponent= strdupAtoW(szComponent); rc = MsiGetComponentStateW(hInstall,szwComponent,piInstalled, piAction); HeapFree( GetProcessHeap(), 0 , szwComponent); return rc; } UINT MSI_GetComponentStateW(MSIPACKAGE *package, LPWSTR szComponent, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { INT index; TRACE("%p %s %p %p\n", package, debugstr_w(szComponent), piInstalled, piAction); index = get_loaded_component(package,szComponent); if (index < 0) return ERROR_UNKNOWN_COMPONENT; if (piInstalled) *piInstalled = package->components[index].State; if (piAction) { if (package->components[index].Enabled && package->components[index].FeatureState) *piAction = INSTALLSTATE_LOCAL; else *piAction = INSTALLSTATE_UNKNOWN; } return ERROR_SUCCESS; } UINT WINAPI MsiGetComponentStateW(MSIHANDLE hInstall, LPWSTR szComponent, INSTALLSTATE *piInstalled, INSTALLSTATE *piAction) { MSIPACKAGE* package; UINT ret; TRACE("%ld %s %p %p\n", hInstall, debugstr_w(szComponent), piInstalled, piAction); package = msihandle2msiinfo(hInstall, MSIHANDLETYPE_PACKAGE); if (!package) return ERROR_INVALID_HANDLE; ret = MSI_GetComponentStateW( package, szComponent, piInstalled, piAction); msiobj_release( &package->hdr ); return ret; } #if 0 static UINT ACTION_Template(MSIPACKAGE *package) { UINT rc; MSIQUERY * view; MSIRECORD * row = 0; static const WCHAR ExecSeqQuery[] = {0}; rc = MsiDatabaseOpenViewW(package->db, ExecSeqQuery, &view); if (rc != ERROR_SUCCESS) return rc; rc = MsiViewExecute(view, 0); if (rc != ERROR_SUCCESS) { MsiViewClose(view); msiobj_release(&view->hdr); return rc; } while (1) { rc = MsiViewFetch(view,&row); if (rc != ERROR_SUCCESS) { rc = ERROR_SUCCESS; break; } msiobj_release(&row->hdr); } MsiViewClose(view); msiobj_release(&view->hdr); return rc; } #endif