#include "windows.h"
#include "msi.h"
#include "msiquery.h"

UINT __stdcall GetInstallDir(MSIHANDLE);
UINT __stdcall SetInstallDir(MSIHANDLE);

void PostLastError(LONG);

UINT __stdcall GetInstallDir(MSIHANDLE hInstall)
{
LONG lResult;
char* lpInstallDir = new char[MAX_PATH];
DWORD pchBuff = MAX_PATH;

lResult = MsiGetProperty(hInstall, "INSTALLDIR", lpInstallDir, &pchBuff);

if (lResult==ERROR_SUCCESS)
    MessageBox(NULL,lpInstallDir,"InstallDir",NULL);
else
    PostLastError(lResult);

return ERROR_SUCCESS;
}


UINT __stdcall SetInstallDir(MSIHANDLE hInstall)
{
LONG lResult;
lResult = MsiSetProperty(hInstall,"INSTALLDIR","C:\\Program Files\\InstallShield");
if (lResult != ERROR_SUCCESS)
  	PostLastError(lResult);
return ERROR_SUCCESS;
}


void PostLastError(LONG lResult)
{
switch(lResult)
{
case ERROR_INVALID_HANDLE:
 	MessageBox(NULL,"Invalid Handle","ERROR",NULL);
case ERROR_INVALID_PARAMETER:
  	MessageBox(NULL,"Invalid Parameter","ERROR",NULL);
case ERROR_MORE_DATA:
	MessageBox(NULL,"More Data","ERROR",NULL);
case ERROR_FUNCTION_FAILED:
  	MessageBox(NULL,"Function Failed","ERROR",NULL);
 }
}