mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-18 15:55:36 +00:00
Bug #354895 --> add vista registry support to thunderbird's shell service
r=rob strong
This commit is contained in:
parent
a9c9356be8
commit
1b9da026a1
@ -68,7 +68,7 @@ CPPSRCS = nsModule.cpp \
|
|||||||
$(NULL)
|
$(NULL)
|
||||||
|
|
||||||
ifeq ($(OS_ARCH),WINNT)
|
ifeq ($(OS_ARCH),WINNT)
|
||||||
OS_LIBS += $(call EXPAND_LIBNAME,shell32)
|
OS_LIBS += $(call EXPAND_LIBNAME,ole32 shell32)
|
||||||
endif
|
endif
|
||||||
|
|
||||||
LOCAL_INCLUDES = \
|
LOCAL_INCLUDES = \
|
||||||
|
@ -62,7 +62,7 @@ NS_GENERIC_FACTORY_CONSTRUCTOR(nsOEProfileMigrator)
|
|||||||
NS_GENERIC_FACTORY_CONSTRUCTOR(nsOutlookProfileMigrator)
|
NS_GENERIC_FACTORY_CONSTRUCTOR(nsOutlookProfileMigrator)
|
||||||
|
|
||||||
#include "nsMailWinIntegration.h"
|
#include "nsMailWinIntegration.h"
|
||||||
NS_GENERIC_FACTORY_CONSTRUCTOR(nsWindowsShellService)
|
NS_GENERIC_FACTORY_CONSTRUCTOR_INIT(nsWindowsShellService, Init)
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(XP_WIN32) || defined(XP_MACOSX)
|
#if defined(XP_WIN32) || defined(XP_MACOSX)
|
||||||
|
@ -46,12 +46,12 @@
|
|||||||
#include "nsIMapiSupport.h"
|
#include "nsIMapiSupport.h"
|
||||||
#endif
|
#endif
|
||||||
#include "shlobj.h"
|
#include "shlobj.h"
|
||||||
|
#include "windows.h"
|
||||||
|
#include "shellapi.h"
|
||||||
|
#include "nsILocalFile.h"
|
||||||
|
|
||||||
#include <mbstring.h>
|
#include <mbstring.h>
|
||||||
|
|
||||||
#define MOZ_CLIENT_MAIL_KEY "Software\\Clients\\Mail"
|
|
||||||
#define MOZ_CLIENT_NEWS_KEY "Software\\Clients\\News"
|
|
||||||
|
|
||||||
#ifndef MAX_BUF
|
#ifndef MAX_BUF
|
||||||
#define MAX_BUF 4096
|
#define MAX_BUF 4096
|
||||||
#endif
|
#endif
|
||||||
@ -66,7 +66,8 @@ OpenUserKeyForReading(HKEY aStartKey, const char* aKeyName, HKEY* aKey)
|
|||||||
{
|
{
|
||||||
DWORD result = ::RegOpenKeyEx(aStartKey, aKeyName, 0, KEY_READ, aKey);
|
DWORD result = ::RegOpenKeyEx(aStartKey, aKeyName, 0, KEY_READ, aKey);
|
||||||
|
|
||||||
switch (result) {
|
switch (result)
|
||||||
|
{
|
||||||
case ERROR_SUCCESS:
|
case ERROR_SUCCESS:
|
||||||
break;
|
break;
|
||||||
case ERROR_ACCESS_DENIED:
|
case ERROR_ACCESS_DENIED:
|
||||||
@ -83,26 +84,41 @@ OpenUserKeyForReading(HKEY aStartKey, const char* aKeyName, HKEY* aKey)
|
|||||||
return NS_OK;
|
return NS_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
static nsresult
|
// Sets the default mail registry keys for Windows versions prior to Vista.
|
||||||
OpenKeyForWriting(const char* aKeyName, HKEY* aKey, PRBool aForAllUsers, PRBool aCreate)
|
// Try to open / create the key in HKLM and if that fails try to do the same
|
||||||
|
// in HKCU. Though this is not strictly the behavior I would expect it is the
|
||||||
|
// same behavior that Firefox and IE has when setting the default browser previous to Vista.
|
||||||
|
static nsresult OpenKeyForWriting(HKEY aStartKey, const char* aKeyName, HKEY* aKey, PRBool aHKLMOnly)
|
||||||
{
|
{
|
||||||
nsresult rv = NS_OK;
|
DWORD dwDisp = 0;
|
||||||
|
DWORD rv = ::RegCreateKeyEx(aStartKey, aKeyName, 0, NULL, 0,
|
||||||
|
KEY_READ | KEY_WRITE, NULL, aKey, &dwDisp);
|
||||||
|
|
||||||
HKEY rootKey = aForAllUsers ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
|
switch (rv)
|
||||||
DWORD result = ::RegOpenKeyEx(rootKey, aKeyName, 0, KEY_READ | KEY_WRITE, aKey);
|
{
|
||||||
|
|
||||||
switch (result) {
|
|
||||||
case ERROR_SUCCESS:
|
case ERROR_SUCCESS:
|
||||||
break;
|
break;
|
||||||
case ERROR_ACCESS_DENIED:
|
case ERROR_ACCESS_DENIED:
|
||||||
return NS_ERROR_FILE_ACCESS_DENIED;
|
if (aHKLMOnly || aStartKey == HKEY_CURRENT_USER)
|
||||||
|
return NS_ERROR_FILE_ACCESS_DENIED;
|
||||||
|
// fallback to HKCU immediately on access denied since we won't be able
|
||||||
|
// to create the key.
|
||||||
|
return OpenKeyForWriting(HKEY_CURRENT_USER, aKeyName, aKey, aHKLMOnly);
|
||||||
case ERROR_FILE_NOT_FOUND:
|
case ERROR_FILE_NOT_FOUND:
|
||||||
if (aCreate)
|
rv = ::RegCreateKey(aStartKey, aKeyName, aKey);
|
||||||
result = ::RegCreateKey(HKEY_LOCAL_MACHINE, aKeyName, aKey);
|
if (rv != ERROR_SUCCESS)
|
||||||
rv = NS_ERROR_FILE_NOT_FOUND;
|
{
|
||||||
break;
|
if (aHKLMOnly || aStartKey == HKEY_CURRENT_USER)
|
||||||
|
{
|
||||||
|
// prevent infinite recursion on the second pass through here if
|
||||||
|
// ::RegCreateKey fails in the current user case.
|
||||||
|
return NS_ERROR_FILE_ACCESS_DENIED;
|
||||||
|
}
|
||||||
|
return OpenKeyForWriting(HKEY_CURRENT_USER, aKeyName, aKey, aHKLMOnly);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return rv;
|
|
||||||
|
return NS_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
///////////////////////////////////////////////////////////////////////////////
|
///////////////////////////////////////////////////////////////////////////////
|
||||||
@ -110,23 +126,39 @@ OpenKeyForWriting(const char* aKeyName, HKEY* aKey, PRBool aForAllUsers, PRBool
|
|||||||
///////////////////////////////////////////////////////////////////////////////
|
///////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
typedef enum { NO_SUBSTITUTION = 0x00,
|
typedef enum { NO_SUBSTITUTION = 0x00,
|
||||||
PATH_SUBSTITUTION = 0x01,
|
APP_PATH_SUBSTITUTION = 0x01,
|
||||||
APPNAME_SUBSTITUTION = 0x02,
|
APPNAME_SUBSTITUTION = 0x02,
|
||||||
MAPIDLLPATH_SUBSTITUTION = 0x04,
|
UNINST_PATH_SUBSTITUTION = 0x04,
|
||||||
USE_FOR_DEFAULT_TEST = 0x08} SettingFlags;
|
MAPIDLL_PATH_SUBSTITUTION = 0x08,
|
||||||
|
HKLM_ONLY = 0x10,
|
||||||
|
USE_FOR_DEFAULT_TEST = 0x20} SettingFlags;
|
||||||
|
|
||||||
#define CLS "SOFTWARE\\Classes\\"
|
#define CLS "SOFTWARE\\Classes\\"
|
||||||
#define MAILCLIENTS "SOFTWARE\\Clients\\Mail\\"
|
#define MAILCLIENTS "SOFTWARE\\Clients\\Mail\\"
|
||||||
#define NEWSCLIENTS "SOFTWARE\\Clients\\News\\"
|
#define NEWSCLIENTS "SOFTWARE\\Clients\\News\\"
|
||||||
|
#define MOZ_CLIENT_MAIL_KEY "Software\\Clients\\Mail"
|
||||||
|
#define MOZ_CLIENT_NEWS_KEY "Software\\Clients\\News"
|
||||||
#define DI "\\DefaultIcon"
|
#define DI "\\DefaultIcon"
|
||||||
|
#define II "\\InstallInfo"
|
||||||
|
|
||||||
|
// APP_REG_NAME_MAIL and APP_REG_NAME_NEWS should be kept in synch with
|
||||||
|
// AppRegNameMail and AppRegNameNews in the installer file: defines.nsi.in
|
||||||
|
#define APP_REG_NAME_MAIL L"Thunderbird"
|
||||||
|
#define APP_REG_NAME_NEWS L"Thunderbird (News)"
|
||||||
#define CLS_EML "ThunderbirdEML"
|
#define CLS_EML "ThunderbirdEML"
|
||||||
|
#define CLS_MAILTOURL "Thunderbird.Url.mailto"
|
||||||
|
#define CLS_NEWSURL "Thunderbird.Url.news"
|
||||||
|
#define CLS_FEEDURL "Thunderbird.Url.feed"
|
||||||
#define SOP "\\shell\\open\\command"
|
#define SOP "\\shell\\open\\command"
|
||||||
|
|
||||||
|
// For the InstallInfo HideIconsCommand, ShowIconsCommand, and ReinstallCommand
|
||||||
|
// registry keys. This must be kept in sync with the uninstaller.
|
||||||
|
#define UNINSTALL_EXE "\\uninstall\\helper.exe"
|
||||||
#define EXE "thunderbird.exe"
|
#define EXE "thunderbird.exe"
|
||||||
|
|
||||||
#define VAL_URL_ICON "%APPPATH%,0"
|
#define VAL_URL_ICON "%APPPATH%,0"
|
||||||
#define VAL_FILE_ICON "%APPPATH%,1"
|
#define VAL_FILE_ICON "%APPPATH%,0"
|
||||||
#define VAL_OPEN "%APPPATH% \"%1\""
|
#define VAL_OPEN "\"%APPPATH%\" \"%1\""
|
||||||
#define VAL_OPEN_WITH_URL "%APPPATH% -url \"%1\""
|
|
||||||
|
|
||||||
#define MAKE_KEY_NAME1(PREFIX, MID) \
|
#define MAKE_KEY_NAME1(PREFIX, MID) \
|
||||||
PREFIX MID
|
PREFIX MID
|
||||||
@ -135,114 +167,151 @@ typedef enum { NO_SUBSTITUTION = 0x00,
|
|||||||
PREFIX MID SUFFIX
|
PREFIX MID SUFFIX
|
||||||
|
|
||||||
static SETTING gMailSettings[] = {
|
static SETTING gMailSettings[] = {
|
||||||
// File Extensions
|
// File Extension Aliases
|
||||||
{ MAKE_KEY_NAME1(CLS, ".eml"), "", "", NO_SUBSTITUTION },
|
{ MAKE_KEY_NAME1(CLS, ".eml"), "", CLS_EML, NO_SUBSTITUTION },
|
||||||
{ MAKE_KEY_NAME2(CLS, CLS_EML, DI), "", VAL_FILE_ICON, PATH_SUBSTITUTION },
|
|
||||||
{ MAKE_KEY_NAME2(CLS, CLS_EML, SOP), "", VAL_OPEN, PATH_SUBSTITUTION},
|
|
||||||
|
|
||||||
|
// File Extension Class
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_EML, DI), "", VAL_FILE_ICON, APP_PATH_SUBSTITUTION },
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_EML, SOP), "", VAL_OPEN, APP_PATH_SUBSTITUTION},
|
||||||
|
|
||||||
|
// Protocol Handler Class - for Vista and above
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_MAILTOURL, DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION },
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_MAILTOURL, SOP), "", "\"%APPPATH%\" -compose \"%1\"", APP_PATH_SUBSTITUTION },
|
||||||
|
|
||||||
// Protocol Handlers
|
// Protocol Handlers
|
||||||
{ MAKE_KEY_NAME2(CLS, "mailto", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION},
|
{ MAKE_KEY_NAME2(CLS, "mailto", DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION},
|
||||||
// we use the following key for our default mail app test so don't set the NON_ESSENTIAL flag
|
{ MAKE_KEY_NAME2(CLS, "mailto", SOP), "", "\"%APPPATH%\" -compose \"%1\"", APP_PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
||||||
{ MAKE_KEY_NAME2(CLS, "mailto", SOP), "", "%APPPATH% -compose \"%1\"", PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
|
||||||
|
|
||||||
// Windows XP Start Menu
|
// Mail Client Keys
|
||||||
{ MAKE_KEY_NAME1(MAILCLIENTS, "%APPNAME%"),
|
{ MAKE_KEY_NAME1(MAILCLIENTS, "%APPNAME%"),
|
||||||
"DLLPath",
|
"DLLPath",
|
||||||
"%MAPIDLLPATH%",
|
"%MAPIDLLPATH%",
|
||||||
MAPIDLLPATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
MAPIDLL_PATH_SUBSTITUTION | HKLM_ONLY | APPNAME_SUBSTITUTION },
|
||||||
|
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", II),
|
||||||
|
"HideIconsCommand",
|
||||||
|
"\"%UNINSTPATH%\" /HideShortcuts",
|
||||||
|
UNINST_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
|
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", II),
|
||||||
|
"ReinstallCommand",
|
||||||
|
"\"%UNINSTPATH%\" /SetAsDefaultAppGlobal",
|
||||||
|
UNINST_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
|
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", II),
|
||||||
|
"ShowIconsCommand",
|
||||||
|
"\"%UNINSTPATH%\" /ShowShortcuts",
|
||||||
|
UNINST_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", DI),
|
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", DI),
|
||||||
"",
|
"",
|
||||||
"%APPPATH%,0",
|
"%APPPATH%,0",
|
||||||
PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
APP_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", SOP),
|
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%", SOP),
|
||||||
"",
|
"",
|
||||||
"%APPPATH% -mail",
|
"\"%APPPATH%\" -mail",
|
||||||
PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
APP_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME1(MAILCLIENTS, "%APPNAME%\\shell\\properties\\command"),
|
{ MAKE_KEY_NAME1(MAILCLIENTS, "%APPNAME%\\shell\\properties\\command"),
|
||||||
"",
|
"",
|
||||||
"%APPPATH% -options",
|
"\"%APPPATH%\" -options",
|
||||||
PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
APP_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%\\Protocols\\mailto", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%\\Protocols\\mailto", SOP), "", "%APPPATH% -compose \"%1\"", PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%\\.eml", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(MAILCLIENTS, "%APPNAME%\\.eml", SOP), "", VAL_OPEN, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION }
|
|
||||||
|
|
||||||
// These values must be set by hand, since they contain localized strings.
|
|
||||||
// shell\properties (default) REG_SZ Firefox &Options
|
|
||||||
};
|
};
|
||||||
|
|
||||||
static SETTING gNewsSettings[] = {
|
static SETTING gNewsSettings[] = {
|
||||||
// Protocol Handlers
|
// Protocol Handler Class - for Vista and above
|
||||||
{ MAKE_KEY_NAME2(CLS, "news", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION},
|
{ MAKE_KEY_NAME2(CLS, CLS_NEWSURL, DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION },
|
||||||
{ MAKE_KEY_NAME2(CLS, "news", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
{ MAKE_KEY_NAME2(CLS, CLS_NEWSURL, SOP), "", "\"%APPPATH%\" -mail \"%1\"", APP_PATH_SUBSTITUTION },
|
||||||
{ MAKE_KEY_NAME2(CLS, "nntp", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(CLS, "nntp", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
|
||||||
{ MAKE_KEY_NAME2(CLS, "snews", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(CLS, "snews", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION},
|
|
||||||
|
|
||||||
// Client Keys
|
// Protocol Handlers
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "news", DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION},
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "news", SOP), "", "\"%APPPATH%\" -mail \"%1\"", APP_PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "nntp", DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION},
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "nntp", SOP), "", "\"%APPPATH%\" -mail \"%1\"", APP_PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "snews", DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION},
|
||||||
|
{ MAKE_KEY_NAME2(CLS, "snews", SOP), "", "\"%APPPATH%\"-mail \"%1\"", APP_PATH_SUBSTITUTION},
|
||||||
|
|
||||||
|
// News Client Keys
|
||||||
{ MAKE_KEY_NAME1(NEWSCLIENTS, "%APPNAME%"),
|
{ MAKE_KEY_NAME1(NEWSCLIENTS, "%APPNAME%"),
|
||||||
"DLLPath",
|
"DLLPath",
|
||||||
"%MAPIDLLPATH%",
|
"%MAPIDLLPATH%",
|
||||||
MAPIDLLPATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
MAPIDLL_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%", DI),
|
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%", DI),
|
||||||
"",
|
"",
|
||||||
"%APPPATH%,0",
|
"%APPPATH%,0",
|
||||||
PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
APP_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%", SOP),
|
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%", SOP),
|
||||||
"",
|
"",
|
||||||
"%APPPATH% -mail",
|
"\"%APPPATH%\" -mail",
|
||||||
PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
APP_PATH_SUBSTITUTION | APPNAME_SUBSTITUTION | HKLM_ONLY },
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\news", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\news", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\nntp", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\nntp", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | APPNAME_SUBSTITUTION },
|
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\snews", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION | APPNAME_SUBSTITUTION},
|
|
||||||
{ MAKE_KEY_NAME2(NEWSCLIENTS, "%APPNAME%\\Protocols\\snews", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | APPNAME_SUBSTITUTION }
|
|
||||||
};
|
};
|
||||||
|
|
||||||
static SETTING gFeedSettings[] = {
|
static SETTING gFeedSettings[] = {
|
||||||
|
// Protocol Handler Class - for Vista and above
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_FEEDURL, DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION },
|
||||||
|
{ MAKE_KEY_NAME2(CLS, CLS_FEEDURL, SOP), "", VAL_OPEN, APP_PATH_SUBSTITUTION },
|
||||||
|
|
||||||
// Protocol Handlers
|
// Protocol Handlers
|
||||||
{ MAKE_KEY_NAME2(CLS, "feed", DI), "", VAL_URL_ICON, PATH_SUBSTITUTION},
|
{ MAKE_KEY_NAME2(CLS, "feed", DI), "", VAL_URL_ICON, APP_PATH_SUBSTITUTION},
|
||||||
{ MAKE_KEY_NAME2(CLS, "feed", SOP), "", "%APPPATH% -mail \"%1\"", PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
{ MAKE_KEY_NAME2(CLS, "feed", SOP), "", "%APPPATH% -mail \"%1\"", APP_PATH_SUBSTITUTION | USE_FOR_DEFAULT_TEST},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
nsresult nsWindowsShellService::Init()
|
||||||
|
{
|
||||||
|
nsresult rv;
|
||||||
|
|
||||||
|
nsCOMPtr<nsIStringBundleService> bundleService(do_GetService("@mozilla.org/intl/stringbundle;1", &rv));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
nsCOMPtr<nsIStringBundle> bundle, brandBundle;
|
||||||
|
rv = bundleService->CreateBundle("chrome://branding/locale/brand.properties", getter_AddRefs(brandBundle));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
brandBundle->GetStringFromName(NS_LITERAL_STRING("brandFullName").get(),
|
||||||
|
getter_Copies(mBrandFullName));
|
||||||
|
brandBundle->GetStringFromName(NS_LITERAL_STRING("brandShortName").get(),
|
||||||
|
getter_Copies(mBrandShortName));
|
||||||
|
|
||||||
|
char appPath[MAX_BUF];
|
||||||
|
if (!::GetModuleFileName(0, appPath, MAX_BUF))
|
||||||
|
return NS_ERROR_FAILURE;
|
||||||
|
|
||||||
|
mAppLongPath = appPath;
|
||||||
|
|
||||||
|
nsCOMPtr<nsILocalFile> lf;
|
||||||
|
rv = NS_NewNativeLocalFile(mAppLongPath, PR_TRUE,
|
||||||
|
getter_AddRefs(lf));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
nsCOMPtr<nsIFile> appDir;
|
||||||
|
rv = lf->GetParent(getter_AddRefs(appDir));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
appDir->GetNativePath(mUninstallPath);
|
||||||
|
mUninstallPath.Append(UNINSTALL_EXE);
|
||||||
|
|
||||||
|
// Support short path to the exe so if it is already set the user is not
|
||||||
|
// prompted to set the default mail client again.
|
||||||
|
if (!::GetShortPathName(appPath, appPath, MAX_BUF))
|
||||||
|
return NS_ERROR_FAILURE;
|
||||||
|
|
||||||
|
ToUpperCase(mAppShortPath = appPath);
|
||||||
|
|
||||||
|
rv = NS_NewNativeLocalFile(mAppLongPath, PR_TRUE, getter_AddRefs(lf));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
rv = lf->SetNativeLeafName(nsDependentCString("mozMapi32.dll"));
|
||||||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||||||
|
|
||||||
|
return lf->GetNativePath(mMapiDLLPath);
|
||||||
|
}
|
||||||
|
|
||||||
nsWindowsShellService::nsWindowsShellService()
|
nsWindowsShellService::nsWindowsShellService()
|
||||||
:mCheckedThisSession(PR_FALSE)
|
:mCheckedThisSession(PR_FALSE)
|
||||||
{
|
{
|
||||||
nsresult rv;
|
|
||||||
// fill in mAppPath
|
|
||||||
char buf[MAX_BUF];
|
|
||||||
::GetModuleFileName(NULL, buf, sizeof(buf));
|
|
||||||
::GetShortPathName(buf, buf, sizeof(buf));
|
|
||||||
ToUpperCase(mAppPath = buf);
|
|
||||||
|
|
||||||
mMapiDLLPath = mAppPath;
|
|
||||||
// remove the process name from the string (thunderbird.exe)
|
|
||||||
char* pathSep = (char *) _mbsrchr((const unsigned char *) mAppPath.get(), '\\');
|
|
||||||
if (pathSep)
|
|
||||||
mMapiDLLPath.Truncate(pathSep - mAppPath.get() + 1);
|
|
||||||
// now append mozMapi32.dll
|
|
||||||
mMapiDLLPath += "mozMapi32.dll";
|
|
||||||
|
|
||||||
nsCOMPtr<nsIStringBundleService> bundleService(do_GetService("@mozilla.org/intl/stringbundle;1", &rv));
|
|
||||||
if (NS_SUCCEEDED(rv))
|
|
||||||
{
|
|
||||||
nsCOMPtr<nsIStringBundle> bundle, brandBundle;
|
|
||||||
rv = bundleService->CreateBundle("chrome://branding/locale/brand.properties", getter_AddRefs(brandBundle));
|
|
||||||
if (NS_SUCCEEDED(rv))
|
|
||||||
{
|
|
||||||
brandBundle->GetStringFromName(NS_LITERAL_STRING("brandFullName").get(),
|
|
||||||
getter_Copies(mBrandFullName));
|
|
||||||
brandBundle->GetStringFromName(NS_LITERAL_STRING("brandShortName").get(),
|
|
||||||
getter_Copies(mBrandShortName));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
NS_IMETHODIMP
|
NS_IMETHODIMP
|
||||||
nsWindowsShellService::IsDefaultClient(PRBool aStartupCheck, PRUint16 aApps, PRBool *aIsDefaultClient)
|
nsWindowsShellService::IsDefaultClient(PRBool aStartupCheck, PRUint16 aApps, PRBool *aIsDefaultClient)
|
||||||
{
|
{
|
||||||
|
if (IsDefaultClientVista(aStartupCheck, aApps, aIsDefaultClient))
|
||||||
|
return NS_OK;
|
||||||
|
|
||||||
*aIsDefaultClient = PR_TRUE;
|
*aIsDefaultClient = PR_TRUE;
|
||||||
|
|
||||||
// for each type,
|
// for each type,
|
||||||
@ -262,17 +331,61 @@ nsWindowsShellService::IsDefaultClient(PRBool aStartupCheck, PRUint16 aApps, PRB
|
|||||||
return NS_OK;
|
return NS_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
DWORD
|
||||||
|
nsWindowsShellService::DeleteRegKeyDefaultValue(HKEY baseKey, const char *keyName)
|
||||||
|
{
|
||||||
|
HKEY key;
|
||||||
|
DWORD rc = ::RegOpenKeyEx(baseKey, keyName, 0, KEY_WRITE, &key);
|
||||||
|
if (rc == ERROR_SUCCESS) {
|
||||||
|
rc = ::RegDeleteValue(key, "");
|
||||||
|
::RegCloseKey(key);
|
||||||
|
}
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
NS_IMETHODIMP
|
NS_IMETHODIMP
|
||||||
nsWindowsShellService::SetDefaultClient(PRBool aForAllUsers, PRUint16 aApps)
|
nsWindowsShellService::SetDefaultClient(PRBool aForAllUsers, PRUint16 aApps)
|
||||||
{
|
{
|
||||||
|
// Delete the protocol and file handlers under HKCU if they exist. This way
|
||||||
|
// the HKCU registry is cleaned up when HKLM is writeable or if it isn't
|
||||||
|
// the values will then be added under HKCU.
|
||||||
|
if (aApps & nsIShellService::MAIL)
|
||||||
|
{
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\ThunderbirdEML");
|
||||||
|
(void)DeleteRegKeyDefaultValue(HKEY_CURRENT_USER, "Software\\Classes\\.eml");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\mailto\\shell\\open");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\mailto\\DefaultIcon");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (aApps & nsIShellService::NEWS)
|
||||||
|
{
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\news\\shell\\open");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\news\\DefaultIcon");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\snews\\shell\\open");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\snews\\DefaultIcon");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\nntp\\shell\\open");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\nntp\\DefaultIcon");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (aApps & nsIShellService::RSS)
|
||||||
|
{
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\feed\\shell\\open");
|
||||||
|
(void)DeleteRegKey(HKEY_CURRENT_USER, "Software\\Classes\\feed\\DefaultIcon");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (SetDefaultClientVista(aApps))
|
||||||
|
return NS_OK;
|
||||||
|
|
||||||
nsresult rv = NS_OK;
|
nsresult rv = NS_OK;
|
||||||
if (aApps & nsIShellService::MAIL)
|
if (aApps & nsIShellService::MAIL)
|
||||||
rv |= setDefaultMail(aForAllUsers);
|
rv |= setDefaultMail();
|
||||||
|
|
||||||
if (aApps & nsIShellService::NEWS)
|
if (aApps & nsIShellService::NEWS)
|
||||||
rv |= setDefaultNews(aForAllUsers);
|
rv |= setDefaultNews();
|
||||||
|
|
||||||
if (aApps & nsIShellService::RSS)
|
if (aApps & nsIShellService::RSS)
|
||||||
setKeysForSettings(gFeedSettings, sizeof(gFeedSettings)/sizeof(SETTING),
|
setKeysForSettings(gFeedSettings, sizeof(gFeedSettings)/sizeof(SETTING),
|
||||||
NS_ConvertUTF16toUTF8(mBrandFullName).get(), aForAllUsers);
|
NS_ConvertUTF16toUTF8(mBrandFullName).get());
|
||||||
|
|
||||||
// Refresh the Shell
|
// Refresh the Shell
|
||||||
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
|
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
|
||||||
@ -300,17 +413,15 @@ nsWindowsShellService::SetShouldCheckDefaultClient(PRBool aShouldCheck)
|
|||||||
}
|
}
|
||||||
|
|
||||||
nsresult
|
nsresult
|
||||||
nsWindowsShellService::setDefaultMail(PRBool aForAllUsers)
|
nsWindowsShellService::setDefaultMail()
|
||||||
{
|
{
|
||||||
nsresult rv;
|
nsresult rv;
|
||||||
NS_ConvertUTF16toUTF8 appName(mBrandFullName);
|
NS_ConvertUTF16toUTF8 appName(mBrandFullName);
|
||||||
setKeysForSettings(gMailSettings, sizeof(gMailSettings)/sizeof(SETTING), appName.get(), aForAllUsers);
|
setKeysForSettings(gMailSettings, sizeof(gMailSettings)/sizeof(SETTING), appName.get());
|
||||||
|
|
||||||
// at least for now, this key needs to be written to HKLM instead of HKCU
|
// at least for now, this key needs to be written to HKLM instead of HKCU
|
||||||
// which is where the windows operating system looks (at least on Win XP and earlier)
|
// which is where the windows operating system looks (at least on Win XP and earlier)
|
||||||
// for Tools / Internet Options / Programs) so pass in
|
SetRegKey(NS_LITERAL_CSTRING(MOZ_CLIENT_MAIL_KEY).get(), "", appName.get(), PR_TRUE);
|
||||||
// true for all users.
|
|
||||||
SetRegKey(NS_LITERAL_CSTRING(MOZ_CLIENT_MAIL_KEY).get(), "", appName.get(), PR_TRUE, /* aForAllUsers */ PR_TRUE);
|
|
||||||
|
|
||||||
nsCAutoString nativeFullName;
|
nsCAutoString nativeFullName;
|
||||||
// For now, we use 'A' APIs (see bug 240272, 239279)
|
// For now, we use 'A' APIs (see bug 240272, 239279)
|
||||||
@ -319,7 +430,7 @@ nsWindowsShellService::setDefaultMail(PRBool aForAllUsers)
|
|||||||
nsCAutoString key1(NS_LITERAL_CSTRING(MAILCLIENTS));
|
nsCAutoString key1(NS_LITERAL_CSTRING(MAILCLIENTS));
|
||||||
key1.Append(appName);
|
key1.Append(appName);
|
||||||
key1.Append("\\");
|
key1.Append("\\");
|
||||||
SetRegKey(key1.get(), "", nativeFullName.get(), PR_TRUE, aForAllUsers);
|
SetRegKey(key1.get(), "", nativeFullName.get(), PR_TRUE);
|
||||||
|
|
||||||
// Set the Options and Safe Mode start menu context menu item labels
|
// Set the Options and Safe Mode start menu context menu item labels
|
||||||
nsCOMPtr<nsIStringBundle> bundle;
|
nsCOMPtr<nsIStringBundle> bundle;
|
||||||
@ -341,7 +452,7 @@ nsWindowsShellService::setDefaultMail(PRBool aForAllUsers)
|
|||||||
nsCAutoString nativeTitle;
|
nsCAutoString nativeTitle;
|
||||||
// For the now, we use 'A' APIs (see bug 240272, 239279)
|
// For the now, we use 'A' APIs (see bug 240272, 239279)
|
||||||
NS_CopyUnicodeToNative(optionsTitle, nativeTitle);
|
NS_CopyUnicodeToNative(optionsTitle, nativeTitle);
|
||||||
SetRegKey(optionsKey.get(), "", nativeTitle.get(), PR_TRUE, aForAllUsers);
|
SetRegKey(optionsKey.get(), "", nativeTitle.get(), PR_TRUE);
|
||||||
#ifndef __MINGW32__
|
#ifndef __MINGW32__
|
||||||
// Tell the MAPI Service to register the mapi proxy dll now that we are the default mail application
|
// Tell the MAPI Service to register the mapi proxy dll now that we are the default mail application
|
||||||
nsCOMPtr<nsIMapiSupport> mapiService (do_GetService(NS_IMAPISUPPORT_CONTRACTID, &rv));
|
nsCOMPtr<nsIMapiSupport> mapiService (do_GetService(NS_IMAPISUPPORT_CONTRACTID, &rv));
|
||||||
@ -353,16 +464,14 @@ nsWindowsShellService::setDefaultMail(PRBool aForAllUsers)
|
|||||||
}
|
}
|
||||||
|
|
||||||
nsresult
|
nsresult
|
||||||
nsWindowsShellService::setDefaultNews(PRBool aForAllUsers)
|
nsWindowsShellService::setDefaultNews()
|
||||||
{
|
{
|
||||||
NS_ConvertUTF16toUTF8 appName(mBrandFullName);
|
NS_ConvertUTF16toUTF8 appName(mBrandFullName);
|
||||||
setKeysForSettings(gNewsSettings, sizeof(gNewsSettings)/sizeof(SETTING), appName.get(), aForAllUsers);
|
setKeysForSettings(gNewsSettings, sizeof(gNewsSettings)/sizeof(SETTING), appName.get());
|
||||||
|
|
||||||
// at least for now, this key needs to be written to HKLM instead of HKCU
|
// at least for now, this key needs to be written to HKLM instead of HKCU
|
||||||
// which is where the windows operating system looks (at least on Win XP and earlier)
|
// which is where the windows operating system looks (at least on Win XP and earlier)
|
||||||
// for Tools / Internet Options / Programs) so pass in
|
SetRegKey(NS_LITERAL_CSTRING(MOZ_CLIENT_NEWS_KEY).get(), "", appName.get(), PR_TRUE);
|
||||||
// true for all users.
|
|
||||||
SetRegKey(NS_LITERAL_CSTRING(MOZ_CLIENT_NEWS_KEY).get(), "", appName.get(), PR_TRUE, /* aForAllUsers */ PR_TRUE);
|
|
||||||
|
|
||||||
nsCAutoString nativeFullName;
|
nsCAutoString nativeFullName;
|
||||||
// For now, we use 'A' APIs (see bug 240272, 239279)
|
// For now, we use 'A' APIs (see bug 240272, 239279)
|
||||||
@ -370,25 +479,60 @@ nsWindowsShellService::setDefaultNews(PRBool aForAllUsers)
|
|||||||
nsCAutoString key1(NS_LITERAL_CSTRING(NEWSCLIENTS));
|
nsCAutoString key1(NS_LITERAL_CSTRING(NEWSCLIENTS));
|
||||||
key1.Append(appName);
|
key1.Append(appName);
|
||||||
key1.Append("\\");
|
key1.Append("\\");
|
||||||
SetRegKey(key1.get(), "", nativeFullName.get(), PR_TRUE, aForAllUsers);
|
SetRegKey(key1.get(), "", nativeFullName.get(), PR_TRUE);
|
||||||
return NS_OK;
|
return NS_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Utility function to delete a registry subkey.
|
||||||
|
DWORD
|
||||||
|
nsWindowsShellService::DeleteRegKey(HKEY baseKey, const char *keyName)
|
||||||
|
{
|
||||||
|
// Make sure input subkey isn't null.
|
||||||
|
if (!keyName || !::strlen(keyName))
|
||||||
|
return ERROR_BADKEY;
|
||||||
|
|
||||||
|
DWORD rc;
|
||||||
|
// Open subkey.
|
||||||
|
HKEY key;
|
||||||
|
rc = ::RegOpenKeyEx(baseKey, keyName, 0, KEY_ENUMERATE_SUB_KEYS | DELETE, &key);
|
||||||
|
|
||||||
|
// Continue till we get an error or are done.
|
||||||
|
while (rc == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
char subkeyName[_MAX_PATH];
|
||||||
|
DWORD len = sizeof subkeyName;
|
||||||
|
// Get first subkey name. Note that we always get the
|
||||||
|
// first one, then delete it. So we need to get
|
||||||
|
// the first one next time, also.
|
||||||
|
rc = ::RegEnumKeyEx(key, 0, subkeyName, &len, 0, 0, 0, 0);
|
||||||
|
if (rc == ERROR_NO_MORE_ITEMS)
|
||||||
|
{
|
||||||
|
// No more subkeys. Delete the main one.
|
||||||
|
rc = ::RegDeleteKey(baseKey, keyName);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (rc == ERROR_SUCCESS)
|
||||||
|
{
|
||||||
|
// Another subkey, delete it, recursively.
|
||||||
|
rc = DeleteRegKey(key, subkeyName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close the key we opened.
|
||||||
|
::RegCloseKey(key);
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
void
|
void
|
||||||
nsWindowsShellService::SetRegKey(const char* aKeyName, const char* aValueName,
|
nsWindowsShellService::SetRegKey(const char* aKeyName, const char* aValueName,
|
||||||
const char* aValue, PRBool aReplaceExisting,
|
const char* aValue, PRBool aHKLMOnly)
|
||||||
PRBool aForAllUsers)
|
|
||||||
{
|
{
|
||||||
char buf[MAX_BUF];
|
char buf[MAX_BUF];
|
||||||
DWORD len = sizeof buf;
|
DWORD len = sizeof buf;
|
||||||
|
|
||||||
HKEY theKey;
|
HKEY theKey;
|
||||||
nsresult rv = OpenKeyForWriting(aKeyName, &theKey, aForAllUsers, PR_TRUE);
|
nsresult rv = OpenKeyForWriting(HKEY_LOCAL_MACHINE, aKeyName, &theKey, aHKLMOnly);
|
||||||
if (NS_FAILED(rv) && rv != NS_ERROR_FILE_NOT_FOUND) return;
|
if (NS_FAILED(rv))
|
||||||
|
|
||||||
// If we're not allowed to replace an existing key, and one exists (i.e. the
|
|
||||||
// result isn't ERROR_FILE_NOT_FOUND, then just return now.
|
|
||||||
if (!aReplaceExisting && rv != NS_ERROR_FILE_NOT_FOUND)
|
|
||||||
return;
|
return;
|
||||||
|
|
||||||
// Get the old value
|
// Get the old value
|
||||||
@ -418,13 +562,22 @@ nsWindowsShellService::TestForDefault(SETTING aSettings[], PRInt32 aSize)
|
|||||||
{
|
{
|
||||||
if (settings->flags & USE_FOR_DEFAULT_TEST)
|
if (settings->flags & USE_FOR_DEFAULT_TEST)
|
||||||
{
|
{
|
||||||
nsCAutoString data(settings->valueData);
|
nsCAutoString dataLongPath(settings->valueData);
|
||||||
nsCAutoString key(settings->keyName);
|
nsCAutoString dataShortPath(settings->valueData);
|
||||||
if (settings->flags & PATH_SUBSTITUTION) {
|
if (settings->flags & APP_PATH_SUBSTITUTION) {
|
||||||
PRInt32 offset = data.Find("%APPPATH%");
|
PRInt32 offset = dataLongPath.Find("%APPPATH%");
|
||||||
data.Replace(offset, 9, mAppPath);
|
dataLongPath.Replace(offset, 9, mAppLongPath);
|
||||||
|
// Remove the quotes around %APPPATH% in VAL_OPEN for short paths
|
||||||
|
PRInt32 offsetQuoted = dataShortPath.Find("\"%APPPATH%\"");
|
||||||
|
if (offsetQuoted != -1)
|
||||||
|
dataShortPath.Replace(offsetQuoted, 11, mAppShortPath);
|
||||||
|
else
|
||||||
|
dataShortPath.Replace(offset, 9, mAppShortPath);
|
||||||
}
|
}
|
||||||
if (settings->flags & APPNAME_SUBSTITUTION) {
|
|
||||||
|
nsCAutoString key(settings->keyName);
|
||||||
|
if (settings->flags & APPNAME_SUBSTITUTION)
|
||||||
|
{
|
||||||
PRInt32 offset = key.Find("%APPNAME%");
|
PRInt32 offset = key.Find("%APPNAME%");
|
||||||
key.Replace(offset, 9, appName);
|
key.Replace(offset, 9, appName);
|
||||||
}
|
}
|
||||||
@ -432,13 +585,15 @@ nsWindowsShellService::TestForDefault(SETTING aSettings[], PRInt32 aSize)
|
|||||||
::ZeroMemory(currValue, sizeof(currValue));
|
::ZeroMemory(currValue, sizeof(currValue));
|
||||||
HKEY theKey;
|
HKEY theKey;
|
||||||
nsresult rv = OpenUserKeyForReading(HKEY_CURRENT_USER, key.get(), &theKey);
|
nsresult rv = OpenUserKeyForReading(HKEY_CURRENT_USER, key.get(), &theKey);
|
||||||
if (NS_SUCCEEDED(rv)) {
|
if (NS_SUCCEEDED(rv))
|
||||||
|
{
|
||||||
DWORD len = sizeof currValue;
|
DWORD len = sizeof currValue;
|
||||||
DWORD result = ::RegQueryValueEx(theKey, settings->valueName, NULL, NULL, (LPBYTE)currValue, &len);
|
DWORD result = ::RegQueryValueEx(theKey, settings->valueName, NULL, NULL, (LPBYTE)currValue, &len);
|
||||||
// Close the key we opened.
|
// Close the key we opened.
|
||||||
::RegCloseKey(theKey);
|
::RegCloseKey(theKey);
|
||||||
if (REG_FAILED(result) || strcmp(data.get(), currValue) != 0) {
|
if (REG_FAILED(result) || !dataLongPath.EqualsIgnoreCase(currValue) && !dataShortPath.EqualsIgnoreCase(currValue))
|
||||||
// Key wasn't set, or was set to something else (something else became the default browser)
|
{
|
||||||
|
// Key wasn't set, or was set to something else (something else became the default client)
|
||||||
isDefault = PR_FALSE;
|
isDefault = PR_FALSE;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@ -455,28 +610,142 @@ nsWindowsShellService::TestForDefault(SETTING aSettings[], PRInt32 aSize)
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
void
|
void
|
||||||
nsWindowsShellService::setKeysForSettings(SETTING aSettings[], PRInt32 aSize, const char * aAppName, PRBool aForAllUsers)
|
nsWindowsShellService::setKeysForSettings(SETTING aSettings[], PRInt32 aSize, const char * aAppName)
|
||||||
{
|
{
|
||||||
SETTING* settings;
|
SETTING* settings;
|
||||||
SETTING* end = aSettings + aSize;
|
SETTING* end = aSettings + aSize;
|
||||||
|
PRInt32 offset;
|
||||||
|
|
||||||
for (settings = aSettings; settings < end; ++settings) {
|
for (settings = aSettings; settings < end; ++settings)
|
||||||
|
{
|
||||||
nsCAutoString data(settings->valueData);
|
nsCAutoString data(settings->valueData);
|
||||||
nsCAutoString key(settings->keyName);
|
nsCAutoString key(settings->keyName);
|
||||||
if (settings->flags & PATH_SUBSTITUTION) {
|
if (settings->flags & APP_PATH_SUBSTITUTION)
|
||||||
PRInt32 offset = data.Find("%APPPATH%");
|
{
|
||||||
data.Replace(offset, 9, mAppPath);
|
offset = data.Find("%APPPATH%");
|
||||||
|
data.Replace(offset, 9, mAppLongPath);
|
||||||
}
|
}
|
||||||
if (settings->flags & MAPIDLLPATH_SUBSTITUTION) {
|
if (settings->flags & MAPIDLL_PATH_SUBSTITUTION)
|
||||||
PRInt32 offset = data.Find("%MAPIDLLPATH%");
|
{
|
||||||
|
offset = data.Find("%MAPIDLLPATH%");
|
||||||
data.Replace(offset, 13, mMapiDLLPath);
|
data.Replace(offset, 13, mMapiDLLPath);
|
||||||
}
|
}
|
||||||
if (settings->flags & APPNAME_SUBSTITUTION) {
|
if (settings->flags & APPNAME_SUBSTITUTION)
|
||||||
PRInt32 offset = key.Find("%APPNAME%");
|
{
|
||||||
|
offset = key.Find("%APPNAME%");
|
||||||
key.Replace(offset, 9, aAppName);
|
key.Replace(offset, 9, aAppName);
|
||||||
}
|
}
|
||||||
|
if (settings->flags & UNINST_PATH_SUBSTITUTION)
|
||||||
|
{
|
||||||
|
offset = data.Find("%UNINSTPATH%");
|
||||||
|
data.Replace(offset, 12, mUninstallPath);
|
||||||
|
}
|
||||||
|
|
||||||
SetRegKey(key.get(), settings->valueName, data.get(),
|
SetRegKey(key.get(), settings->valueName, data.get(), settings->flags & HKLM_ONLY);
|
||||||
PR_TRUE, aForAllUsers);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Support for versions of shlobj.h that don't include the Vista API's
|
||||||
|
#if !defined(IApplicationAssociationRegistration)
|
||||||
|
|
||||||
|
typedef enum tagASSOCIATIONLEVEL
|
||||||
|
{
|
||||||
|
AL_MACHINE,
|
||||||
|
AL_EFFECTIVE,
|
||||||
|
AL_USER
|
||||||
|
} ASSOCIATIONLEVEL;
|
||||||
|
|
||||||
|
typedef enum tagASSOCIATIONTYPE
|
||||||
|
{
|
||||||
|
AT_FILEEXTENSION,
|
||||||
|
AT_URLPROTOCOL,
|
||||||
|
AT_STARTMENUCLIENT,
|
||||||
|
AT_MIMETYPE
|
||||||
|
} ASSOCIATIONTYPE;
|
||||||
|
|
||||||
|
MIDL_INTERFACE("4e530b0a-e611-4c77-a3ac-9031d022281b")
|
||||||
|
IApplicationAssociationRegistration : public IUnknown
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE QueryCurrentDefault(LPCWSTR pszQuery,
|
||||||
|
ASSOCIATIONTYPE atQueryType,
|
||||||
|
ASSOCIATIONLEVEL alQueryLevel,
|
||||||
|
LPWSTR *ppszAssociation) = 0;
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE QueryAppIsDefault(LPCWSTR pszQuery,
|
||||||
|
ASSOCIATIONTYPE atQueryType,
|
||||||
|
ASSOCIATIONLEVEL alQueryLevel,
|
||||||
|
LPCWSTR pszAppRegistryName,
|
||||||
|
BOOL *pfDefault) = 0;
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE QueryAppIsDefaultAll(ASSOCIATIONLEVEL alQueryLevel,
|
||||||
|
LPCWSTR pszAppRegistryName,
|
||||||
|
BOOL *pfDefault) = 0;
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE SetAppAsDefault(LPCWSTR pszAppRegistryName,
|
||||||
|
LPCWSTR pszSet,
|
||||||
|
ASSOCIATIONTYPE atSetType) = 0;
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE SetAppAsDefaultAll(LPCWSTR pszAppRegistryName) = 0;
|
||||||
|
virtual HRESULT STDMETHODCALLTYPE ClearUserAssociations( void) = 0;
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
|
static const CLSID CLSID_ApplicationAssociationReg = {0x591209C7,0x767B,0x42B2,{0x9F,0xBA,0x44,0xEE,0x46,0x15,0xF2,0xC7}};
|
||||||
|
static const IID IID_IApplicationAssociationReg = {0x4e530b0a,0xe611,0x4c77,{0xa3,0xac,0x90,0x31,0xd0,0x22,0x28,0x1b}};
|
||||||
|
|
||||||
|
PRBool
|
||||||
|
nsWindowsShellService::IsDefaultClientVista(PRBool aStartupCheck, PRUint16 aApps, PRBool* aIsDefaultClient)
|
||||||
|
{
|
||||||
|
IApplicationAssociationRegistration* pAAR;
|
||||||
|
|
||||||
|
HRESULT hr = CoCreateInstance (CLSID_ApplicationAssociationReg,
|
||||||
|
NULL,
|
||||||
|
CLSCTX_INPROC,
|
||||||
|
IID_IApplicationAssociationReg,
|
||||||
|
(void**)&pAAR);
|
||||||
|
|
||||||
|
if (SUCCEEDED(hr))
|
||||||
|
{
|
||||||
|
PRBool isDefaultMail = PR_TRUE;
|
||||||
|
PRBool isDefaultNews = PR_TRUE;
|
||||||
|
if (aApps & nsIShellService::MAIL)
|
||||||
|
pAAR->QueryAppIsDefaultAll(AL_EFFECTIVE, APP_REG_NAME_MAIL, &isDefaultMail);
|
||||||
|
if (aApps & nsIShellService::NEWS)
|
||||||
|
pAAR->QueryAppIsDefaultAll(AL_EFFECTIVE, APP_REG_NAME_NEWS, &isDefaultNews);
|
||||||
|
|
||||||
|
*aIsDefaultClient = isDefaultNews && isDefaultMail;
|
||||||
|
|
||||||
|
// If this is the first mail window, maintain internal state that we've
|
||||||
|
// checked this session (so that subsequent window opens don't show the
|
||||||
|
// default browser dialog).
|
||||||
|
if (aStartupCheck)
|
||||||
|
mCheckedThisSession = PR_TRUE;
|
||||||
|
|
||||||
|
pAAR->Release();
|
||||||
|
return PR_TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
return PR_FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
PRBool
|
||||||
|
nsWindowsShellService::SetDefaultClientVista(PRUint16 aApps)
|
||||||
|
{
|
||||||
|
IApplicationAssociationRegistration* pAAR;
|
||||||
|
|
||||||
|
HRESULT hr = CoCreateInstance (CLSID_ApplicationAssociationReg,
|
||||||
|
NULL,
|
||||||
|
CLSCTX_INPROC,
|
||||||
|
IID_IApplicationAssociationReg,
|
||||||
|
(void**)&pAAR);
|
||||||
|
|
||||||
|
if (SUCCEEDED(hr))
|
||||||
|
{
|
||||||
|
if (aApps & nsIShellService::MAIL)
|
||||||
|
hr = pAAR->SetAppAsDefaultAll(APP_REG_NAME_MAIL);
|
||||||
|
if (aApps & nsIShellService::NEWS)
|
||||||
|
hr = pAAR->SetAppAsDefaultAll(APP_REG_NAME_NEWS);
|
||||||
|
|
||||||
|
pAAR->Release();
|
||||||
|
return PR_TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
return PR_FALSE;
|
||||||
|
}
|
||||||
|
@ -43,6 +43,7 @@
|
|||||||
#include "nsIGenericFactory.h"
|
#include "nsIGenericFactory.h"
|
||||||
#include "nsString.h"
|
#include "nsString.h"
|
||||||
|
|
||||||
|
#include <ole2.h>
|
||||||
#include <windows.h>
|
#include <windows.h>
|
||||||
|
|
||||||
#define NS_MAILWININTEGRATION_CID \
|
#define NS_MAILWININTEGRATION_CID \
|
||||||
@ -61,24 +62,32 @@ class nsWindowsShellService : public nsIShellService
|
|||||||
public:
|
public:
|
||||||
nsWindowsShellService();
|
nsWindowsShellService();
|
||||||
virtual ~nsWindowsShellService() {};
|
virtual ~nsWindowsShellService() {};
|
||||||
|
NS_HIDDEN_(nsresult) Init();
|
||||||
|
|
||||||
NS_DECL_ISUPPORTS
|
NS_DECL_ISUPPORTS
|
||||||
NS_DECL_NSISHELLSERVICE
|
NS_DECL_NSISHELLSERVICE
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void SetRegKey(const char* aKeyName, const char* aValueName,
|
void SetRegKey(const char* aKeyName, const char* aValueName,
|
||||||
const char* aValue,
|
const char* aValue, PRBool aHKLMOnly);
|
||||||
PRBool aReplaceExisting, PRBool aForAllUsers);
|
DWORD DeleteRegKey(HKEY baseKey, const char *keyName);
|
||||||
|
DWORD DeleteRegKeyDefaultValue(HKEY baseKey, const char *keyName);
|
||||||
|
|
||||||
PRBool TestForDefault(SETTING aSettings[], PRInt32 aSize);
|
PRBool TestForDefault(SETTING aSettings[], PRInt32 aSize);
|
||||||
void setKeysForSettings(SETTING aSettings[], PRInt32 aSize,
|
void setKeysForSettings(SETTING aSettings[], PRInt32 aSize,
|
||||||
const char * aAppname, PRBool aForAllUsers);
|
const char * aAppname);
|
||||||
nsresult setDefaultMail(PRBool aForAllUsers);
|
nsresult setDefaultMail();
|
||||||
nsresult setDefaultNews(PRBool aForAllUsers);
|
nsresult setDefaultNews();
|
||||||
|
|
||||||
|
PRBool IsDefaultClientVista(PRBool aStartupCheck, PRUint16 aApps, PRBool* aIsDefaultClient);
|
||||||
|
PRBool SetDefaultClientVista(PRUint16 aApps);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
PRBool mCheckedThisSession;
|
PRBool mCheckedThisSession;
|
||||||
nsCString mAppPath;
|
nsCString mAppLongPath;
|
||||||
|
nsCString mAppShortPath;
|
||||||
nsCString mMapiDLLPath;
|
nsCString mMapiDLLPath;
|
||||||
|
nsCString mUninstallPath;
|
||||||
nsXPIDLString mBrandFullName;
|
nsXPIDLString mBrandFullName;
|
||||||
nsXPIDLString mBrandShortName;
|
nsXPIDLString mBrandShortName;
|
||||||
};
|
};
|
||||||
|
Loading…
Reference in New Issue
Block a user