409 lines
14 KiB
C
409 lines
14 KiB
C
/*-----------------------------------------------------------------------------+
|
||
| FIXREG.C |
|
||
| |
|
||
| Publisher and Video For Windows make evil changes to the registry |
|
||
| when they are installed. Look for these changes. If they are spotted |
|
||
| then put up a message box to warn the user and offer the user the chance to |
|
||
| correct them (i.e. stuff our version back in) |
|
||
| |
|
||
| (C) Copyright Microsoft Corporation 1994. All rights reserved. |
|
||
| |
|
||
| Revision History |
|
||
| 10-Aug-1994 Lauriegr Created. |
|
||
| |
|
||
+-----------------------------------------------------------------------------*/
|
||
|
||
#include <windows.h>
|
||
#include "mplayer.h"
|
||
#include "fixreg.h"
|
||
#include "registry.h"
|
||
|
||
/* The idea is to call CheckRegValues(hinst) on a separate thread
|
||
(sort of backgroundy thing) and have it just die
|
||
quietly if there's no problem. If on the other hand there is a problem
|
||
then we need to get the message box up - and it's a VERY BAD IDEA to
|
||
try to put a message box up on anything other than the thread that's doing
|
||
all the UI (otherwise ScottLu will get you with a weasle word - guaranteed).
|
||
|
||
So the background thread should PostMessage (Post, don't Send - more weasles)
|
||
to the main thread a message to say "WM_BADREG". The main thread should then
|
||
wack up the dialog box by calling FixRegValues.
|
||
|
||
Suggested coding in main thread:
|
||
|
||
BackgroundRegCheck(hwndmain);
|
||
|
||
in window proc for hwndmain:
|
||
case WM_HEYUP:
|
||
// if ( IDOK == MessageBox(hwnd, text, appname, MB_OKCANCEL) )
|
||
if ( IDOK == ErrorResBox(hwnd, NULL, MB_OKCANCEL, IDS_APPNAME, IDS_BADREG) )
|
||
if (!SetRegValues())
|
||
Error(ghwndApp, IDS_FIXREGERROR);
|
||
*/
|
||
|
||
/* This is a reg setting to disable the check. */
|
||
extern SZCODE aszOptionsSection[];
|
||
static SZCODE aszIgnoreRegistryCheck[] = TEXT("Ignore Registry Check");
|
||
|
||
/* These are the things we check up.
|
||
|
||
First define them as static strings, since the compiler's not smart enough
|
||
to spot common strings.
|
||
|
||
NOTE - these values are NOT LOCALISED, except for the ones that are.
|
||
*/
|
||
|
||
#ifdef CHICAGO_PRODUCT
|
||
#define APPNAME TEXT("mplayer.exe")
|
||
#define WINDIR TEXT("%s\\") // To be replaced by Windows directory
|
||
LPTSTR pWindowsDirectory = NULL;
|
||
#else
|
||
#define APPNAME TEXT("mplay32.exe")
|
||
#define WINDIR
|
||
#endif
|
||
TCHAR szMPlayer[] = TEXT("MPlayer");
|
||
TCHAR szMPlayer_CLSID[] = TEXT("MPlayer\\CLSID");
|
||
TCHAR szMPOLE2GUID[] = TEXT("{00022601-0000-0000-C000-000000000046}");
|
||
TCHAR szMPCLSID_OLE1GUID[] = TEXT("CLSID\\{0003000E-0000-0000-C000-000000000046}");
|
||
TCHAR szMPStdExecute_Server[] = TEXT("MPlayer\\protocol\\StdExecute\\server");
|
||
TCHAR szAppName[] = WINDIR APPNAME;
|
||
TCHAR szMPShell_Open_Command[] = TEXT("MPlayer\\shell\\open\\command");
|
||
TCHAR szAppName_Play_Close[] = WINDIR APPNAME TEXT(" /play /close %1");
|
||
TCHAR szMPlayer_insertable[] = TEXT("MPlayer\\insertable");
|
||
TCHAR szEmpty[] = TEXT("");
|
||
TCHAR szMPStdFileEdit_Handler[] = TEXT("MPlayer\\protocol\\StdFileEditing\\handler");
|
||
#ifdef CHICAGO_PRODUCT
|
||
TCHAR szMCIOLE[] = WINDIR TEXT("mciole.dll");
|
||
#else
|
||
TCHAR szMCIOLE16[] = TEXT("mciole16.dll");
|
||
TCHAR szMPStdFileEdit_Hand32[] = TEXT("MPlayer\\protocol\\StdFileEditing\\handler32");
|
||
TCHAR szMCIOLE32[] = TEXT("mciole32.dll");
|
||
#endif
|
||
TCHAR szMPStdFileEdit_Package[] = TEXT("MPlayer\\protocol\\StdFileEditing\\PackageObjects");
|
||
TCHAR szMPStdFileEdit_Server[] = TEXT("MPlayer\\protocol\\StdFileEditing\\server");
|
||
TCHAR szMPStdFileEdit_verb_0[] = TEXT("MPlayer\\protocol\\StdFileEditing\\verb\\0");
|
||
TCHAR szMPStdFileEdit_verb_1[] = TEXT("MPlayer\\protocol\\StdFileEditing\\verb\\1");
|
||
|
||
/* That sleazebag Publisher setup even farts around with these new settings!!
|
||
*/
|
||
TCHAR szAVIStdFileEdit_Server[] = TEXT("AVIFile\\protocol\\StdFileEditing\\server");
|
||
TCHAR szMIDStdFileEdit_Server[] = TEXT("MIDFile\\protocol\\StdFileEditing\\server");
|
||
TCHAR szServerAVI[] = WINDIR APPNAME TEXT(" /avi");
|
||
TCHAR szServerMID[] = WINDIR APPNAME TEXT(" /mid");
|
||
|
||
/* The following ones DO need to be localised.
|
||
|
||
They will be loaded in CheckRegValues.
|
||
*/
|
||
#define RES_STR_LEN 40 /* Should be enough as a maximum resource string. */
|
||
TCHAR szMediaClip[RES_STR_LEN]; // IDS_CLASSROOT in resources
|
||
TCHAR sz_Play[RES_STR_LEN]; // IDS_PLAYVERB in resources
|
||
TCHAR sz_Edit[RES_STR_LEN]; // IDS_EDITVERB in resources
|
||
|
||
TCHAR szAviFile[] = TEXT("AVIFile");
|
||
TCHAR szMidFile[] = TEXT("MIDFile");
|
||
|
||
|
||
/* Array of registry value-data pairs to check:
|
||
*/
|
||
LPTSTR RegValues[] =
|
||
{
|
||
szMPlayer, szMediaClip,
|
||
szMPlayer_CLSID, szMPOLE2GUID,
|
||
szMPCLSID_OLE1GUID, szMediaClip,
|
||
szMPStdExecute_Server, szAppName,
|
||
szMPShell_Open_Command, szAppName_Play_Close,
|
||
szMPlayer_insertable, szEmpty,
|
||
#ifdef CHICAGO_PRODUCT
|
||
szMPStdFileEdit_Handler, szMCIOLE,
|
||
#else
|
||
szMPStdFileEdit_Handler, szMCIOLE16,
|
||
szMPStdFileEdit_Hand32, szMCIOLE32,
|
||
#endif
|
||
szMPStdFileEdit_Package, szEmpty,
|
||
szMPStdFileEdit_Server, szAppName,
|
||
szMPStdFileEdit_verb_0, sz_Play,
|
||
szMPStdFileEdit_verb_1, sz_Edit,
|
||
|
||
aszKeyAVI, szAviFile,
|
||
aszKeyMID, szMidFile,
|
||
aszKeyRMI, szMidFile,
|
||
|
||
szAVIStdFileEdit_Server, szServerAVI,
|
||
szMIDStdFileEdit_Server, szServerMID
|
||
};
|
||
|
||
|
||
#ifdef CHICAGO_PRODUCT
|
||
|
||
/* AllocWindowsDirectory
|
||
*
|
||
* Dynamically allocates a string containing the Windows directory.
|
||
* This may be freed using FreeStr().
|
||
*
|
||
*/
|
||
LPTSTR AllocWindowsDirectory()
|
||
{
|
||
UINT cchWinPath;
|
||
LPTSTR pWindowsDirectory = NULL;
|
||
|
||
cchWinPath = GetWindowsDirectory(NULL, 0);
|
||
|
||
if (cchWinPath > 0)
|
||
{
|
||
if (pWindowsDirectory = AllocMem(cchWinPath * sizeof(TCHAR)))
|
||
{
|
||
cchWinPath = GetWindowsDirectory(pWindowsDirectory, cchWinPath);
|
||
|
||
if (cchWinPath == 0)
|
||
{
|
||
/* Unlikely, but check anyway:
|
||
*/
|
||
DPF0("GetWindowsDiretory failed: Error %d\n", GetLastError());
|
||
|
||
*pWindowsDirectory = TEXT('\0');
|
||
}
|
||
}
|
||
}
|
||
|
||
return pWindowsDirectory;
|
||
}
|
||
|
||
#endif
|
||
|
||
|
||
/* Check that a REG_SZ value in the registry has the value that it should do
|
||
Return TRUE if it does, FALSE if it doesn't.
|
||
*/
|
||
BOOL CheckRegValue(HKEY RootKey, LPTSTR KeyName, LPTSTR ShouldBe)
|
||
{
|
||
DWORD Type;
|
||
TCHAR Data[100];
|
||
DWORD cData = sizeof(Data);
|
||
LONG lRet;
|
||
HKEY hkey;
|
||
|
||
|
||
if (ERROR_SUCCESS!=RegOpenKeyEx( RootKey
|
||
, KeyName
|
||
, 0 /* reserved */
|
||
, KEY_QUERY_VALUE
|
||
, &hkey
|
||
)
|
||
)
|
||
return FALSE; /* couldn't even open the key */
|
||
|
||
|
||
lRet=RegQueryValueEx( hkey
|
||
, NULL /* ValueName */
|
||
, NULL /* reserved */
|
||
, &Type
|
||
, (LPBYTE)Data
|
||
, &cData
|
||
);
|
||
|
||
RegCloseKey(hkey); /* no idea what to do if this fails */
|
||
|
||
if (ERROR_SUCCESS!=lRet) return FALSE; /* couldn't query it */
|
||
|
||
/* Data, cData and Type give the data, length and type */
|
||
if (Type!=REG_SZ) return FALSE;
|
||
lRet = lstrcmpi(Data,ShouldBe); /* capture lRet to make debug easier */
|
||
return 0==lRet;
|
||
|
||
} /* CheckRegValue */
|
||
|
||
|
||
/* check the registry for anything evil. Return TRUE if it's OK else FALSE */
|
||
BOOL CheckRegValues(void)
|
||
{
|
||
HKEY HCL = HKEY_CLASSES_ROOT; /* save typing! */
|
||
|
||
/* Now just check that the OLE2 class ID is correct
|
||
*/
|
||
if( !CheckRegValue( HCL, szMPlayer_CLSID, szMPOLE2GUID ) )
|
||
return FALSE;
|
||
|
||
/* Running the old MPlayer on Chicago also screws up the
|
||
* file-extension associations, so make sure they haven't changed:
|
||
*/
|
||
if( !CheckRegValue( HCL, aszKeyAVI, szAviFile ) )
|
||
return FALSE;
|
||
|
||
return TRUE;
|
||
|
||
} /* CheckRegValues */
|
||
|
||
|
||
/* start this thread to get the registry checked out.
|
||
hwnd is typed as a LPVOID because that's what CreateThread wants.
|
||
*/
|
||
DWORD WINAPI RegCheckThread(LPVOID hwnd)
|
||
{
|
||
if (!CheckRegValues())
|
||
PostMessage((HWND)hwnd, WM_BADREG, 0, 0);
|
||
|
||
return 0; /* end of thread! */
|
||
}
|
||
|
||
|
||
/* Call this with the hwnd that you want a WM_BADREG message posted to
|
||
It will check the registry. No news is good news.
|
||
It does the work on a separate thread, so this should return quickly.
|
||
*/
|
||
void BackgroundRegCheck(HWND hwnd)
|
||
{
|
||
HANDLE hThread;
|
||
DWORD thid;
|
||
hThread = CreateThread( NULL /* no special security */
|
||
, 0 /* default stack size */
|
||
, RegCheckThread
|
||
, (LPVOID)hwnd
|
||
, 0 /* start running at once */
|
||
, &thid
|
||
);
|
||
if (hThread!=NULL) CloseHandle(hThread); /* we don't need this any more */
|
||
|
||
/* Else we're in some sort of trouble - dunno what to do.
|
||
Can't think of an intelligible message to give to the user.
|
||
Too bad. Creep home quietly.
|
||
*/
|
||
|
||
} /* BackgroundRegCheck */
|
||
|
||
|
||
/* returns TRUE if it worked. Dunno what to do if it didn't
|
||
|
||
*/
|
||
BOOL SetRegValue(HKEY RootKey, LPTSTR KeyName, LPTSTR ValueName, LPTSTR ShouldBe)
|
||
{
|
||
HKEY hkey;
|
||
|
||
if (ERROR_SUCCESS!=RegOpenKeyEx( RootKey
|
||
, KeyName
|
||
, 0 /* reserved */
|
||
, KEY_SET_VALUE
|
||
, &hkey
|
||
)
|
||
) {
|
||
/* Maybe the key has been DELETED - we've seen that */
|
||
DWORD dwDisp;
|
||
if (ERROR_SUCCESS!=RegCreateKeyEx( RootKey
|
||
, KeyName
|
||
, 0 /* reserved */
|
||
, TEXT("") /* class */
|
||
, REG_OPTION_NON_VOLATILE
|
||
, KEY_SET_VALUE
|
||
, NULL /* SecurityAttributes */
|
||
, &hkey
|
||
, &dwDisp
|
||
)
|
||
) /* well we're really in trouble */
|
||
return FALSE;
|
||
else /* So now it exists, but we now have to open it */
|
||
if (ERROR_SUCCESS!=RegOpenKeyEx( RootKey
|
||
, KeyName
|
||
, 0 /* reserved */
|
||
, KEY_SET_VALUE
|
||
, &hkey
|
||
)
|
||
) /* Give up */
|
||
return FALSE;
|
||
|
||
}
|
||
|
||
|
||
if (ERROR_SUCCESS!=RegSetValueEx( hkey
|
||
, ValueName
|
||
, 0 /* reserved */
|
||
, REG_SZ
|
||
, (LPBYTE)ShouldBe
|
||
, (lstrlen(ShouldBe)+1)*sizeof(TCHAR) /* BYTES */
|
||
)
|
||
)
|
||
return FALSE; /* couldn't set it */
|
||
|
||
if ( ERROR_SUCCESS!=RegCloseKey(hkey) )
|
||
/* no idea what to do!*/ ; /* couldn't set it */
|
||
|
||
/* I'm NOT calling RegFlushKey. They'll get there eventually */
|
||
|
||
return TRUE;
|
||
|
||
} /* SetRegValue */
|
||
|
||
|
||
/* Update the registry with the correct values. Return TRUE if everything succeeds */
|
||
BOOL SetRegValues(void)
|
||
{
|
||
HKEY HCL = HKEY_CLASSES_ROOT; /* save typing! */
|
||
DWORD i;
|
||
#ifdef CHICAGO_PRODUCT
|
||
TCHAR Buffer[MAX_PATH+40];
|
||
#endif
|
||
|
||
if( !( LOADSTRING( IDS_CLASSROOT, szMediaClip )
|
||
&& LOADSTRING( IDS_PLAYVERB, sz_Play )
|
||
&& LOADSTRING( IDS_EDITVERB, sz_Edit ) ) )
|
||
/* If any of the strings fails to load, forget it:
|
||
*/
|
||
return TRUE;
|
||
|
||
#ifdef CHICAGO_PRODUCT
|
||
if (pWindowsDirectory == NULL)
|
||
{
|
||
if ((pWindowsDirectory = AllocWindowsDirectory()) == NULL)
|
||
return TRUE;
|
||
}
|
||
#endif
|
||
|
||
for( i = 0; i < ( sizeof RegValues / sizeof *RegValues ); i+=2 )
|
||
{
|
||
/* Do a check to see whether this one needs changing,
|
||
* to avoid gratuitous changes, and to avoid the slim chance
|
||
* that an unnecessary SetRegValue might fail:
|
||
*/
|
||
#ifdef CHICAGO_PRODUCT
|
||
/* Do substitution of Windows directory, if required.
|
||
* This simply copies the value to the buffer unchanged
|
||
* if it doesn't contain a replacement character.
|
||
*/
|
||
wsprintf(Buffer, RegValues[i+1], pWindowsDirectory);
|
||
|
||
if( !CheckRegValue( HCL, RegValues[i], Buffer ) )
|
||
#else
|
||
if( !CheckRegValue( HCL, RegValues[i], RegValues[i+1] ) )
|
||
#endif
|
||
{
|
||
#ifdef CHICAGO_PRODUCT
|
||
DPF("Fixing the registry: Value - %"DTS"; Data - %"DTS"\n", RegValues[i], Buffer);
|
||
if( !SetRegValue( HCL, RegValues[i], NULL, Buffer ) )
|
||
#else
|
||
DPF("Fixing the registry: Value - %"DTS"; Data - %"DTS"\n", RegValues[i], RegValues[i+1]);
|
||
if( !SetRegValue( HCL, RegValues[i], NULL, RegValues[i+1] ) )
|
||
#endif
|
||
return FALSE;
|
||
}
|
||
}
|
||
|
||
#ifdef CHICAGO_PRODUCT
|
||
FreeStr (pWindowsDirectory);
|
||
#endif
|
||
|
||
return TRUE;
|
||
|
||
} /* SetRegValues */
|
||
|
||
BOOL IgnoreRegCheck()
|
||
{
|
||
DWORD fIgnore = 0L;
|
||
ReadRegistryData(aszOptionsSection
|
||
, aszIgnoreRegistryCheck
|
||
, NULL
|
||
, (LPBYTE)&fIgnore
|
||
, sizeof fIgnore);
|
||
|
||
return (fIgnore != 0L);
|
||
|
||
} /* IgnoreRegCheck */
|
||
|