1 /*

     2  * Copyright (C) 2006, 2007 Apple Inc.  All rights reserved.

     3  * Copyright (C) 2008 Collabora, Ltd.  All rights reserved.

     4  * Copyright (C) 2008-2009 Torch Mobile, Inc.  All rights reserved.

     5  *

     6  * Redistribution and use in source and binary forms, with or without

     7  * modification, are permitted provided that the following conditions

     8  * are met:

     9  * 1. Redistributions of source code must retain the above copyright

    10  *    notice, this list of conditions and the following disclaimer.

    11  * 2. Redistributions in binary form must reproduce the above copyright

    12  *    notice, this list of conditions and the following disclaimer in the

    13  *    documentation and/or other materials provided with the distribution.

    14  *

    15  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

    16  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    18  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

    19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    22  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

    23  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

    24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

    25  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

    26  */

    27 

    28 #include "config.h"

    29 #include "PluginDatabase.h"

    30 

    31 #include "Frame.h"

    32 #include "KURL.h"

    33 #include "PluginPackage.h"

    34 #include <windows.h>

    35 #include <shlwapi.h>

    36 

    37 #if COMPILER(MINGW)

    38 #define _countof(x) (sizeof(x)/sizeof(x[0]))

    39 #endif

    40 

    41 #if OS(WINCE)

    42 // WINCE doesn't support Registry Key Access Rights. The parameter should always be 0

    43 #define KEY_ENUMERATE_SUB_KEYS 0

    44 

    45 DWORD SHGetValue(HKEY hkey, LPCWSTR pszSubKey, LPCWSTR pszValue, LPDWORD pdwType, LPVOID pvData, LPDWORD pcbData)

    46 {

    47     HKEY key;

    48     if (RegOpenKeyEx(hkey, pszSubKey, 0, 0, &key) == ERROR_SUCCESS) {

    49         DWORD result = RegQueryValueEx(key, pszValue, 0, pdwType, (LPBYTE)pvData, pcbData);

    50         RegCloseKey(key);

    51         return result;

    52     }

    53     return ERROR_INVALID_NAME;

    54 }

    55 

    56 BOOL PathRemoveFileSpec(LPWSTR moduleFileNameStr)

    57 {

    58     if (!*moduleFileNameStr)

    59         return FALSE;

    60 

    61     LPWSTR lastPos = 0;

    62     LPWSTR curPos = moduleFileNameStr;

    63     do {

    64         if (*curPos == L'/' || *curPos == L'\\')

    65             lastPos = curPos;

    66     } while (*++curPos);

    67 

    68     if (lastPos == curPos - 1)

    69         return FALSE;

    70 

    71     if (lastPos)

    72         *lastPos = 0;

    73     else {

    74         moduleFileNameStr[0] = L'\\';

    75         moduleFileNameStr[1] = 0;

    76     }

    77 

    78     return TRUE;

    79 }

    80 #endif

    81 

    82 namespace WebCore {

    83 

    84 static inline void addPluginPathsFromRegistry(HKEY rootKey, HashSet<String>& paths)

    85 {

    86     HKEY key;

    87     HRESULT result = RegOpenKeyExW(rootKey, L"Software\\MozillaPlugins", 0, KEY_ENUMERATE_SUB_KEYS, &key);

    88 

    89     if (result != ERROR_SUCCESS)

    90         return;

    91 

    92     wchar_t name[128];

    93     FILETIME lastModified;

    94 

    95     // Enumerate subkeys

    96     for (int i = 0;; i++) {

    97         DWORD nameLen = _countof(name);

    98         result = RegEnumKeyExW(key, i, name, &nameLen, 0, 0, 0, &lastModified);

    99 

   100         if (result != ERROR_SUCCESS)

   101             break;

   102 

   103         WCHAR pathStr[_MAX_PATH];

   104         DWORD pathStrSize = sizeof(pathStr);

   105         DWORD type;

   106 

   107         result = SHGetValue(key, name, TEXT("Path"), &type, (LPBYTE)pathStr, &pathStrSize);

   108         if (result != ERROR_SUCCESS || type != REG_SZ)

   109             continue;

   110 

   111         paths.add(String(pathStr, pathStrSize / sizeof(WCHAR) - 1));

   112     }

   113 

   114     RegCloseKey(key);

   115 }

   116 

   117 void PluginDatabase::getPluginPathsInDirectories(HashSet<String>& paths) const

   118 {

   119     // FIXME: This should be a case insensitive set.

   120     HashSet<String> uniqueFilenames;

   121 

   122     HANDLE hFind = INVALID_HANDLE_VALUE;

   123     WIN32_FIND_DATAW findFileData;

   124 

   125     String oldWMPPluginPath;

   126     String newWMPPluginPath;

   127 

   128     Vector<String>::const_iterator end = m_pluginDirectories.end();

   129     for (Vector<String>::const_iterator it = m_pluginDirectories.begin(); it != end; ++it) {

   130         String pattern = *it + "\\*";

   131 

   132         hFind = FindFirstFileW(pattern.charactersWithNullTermination(), &findFileData);

   133 

   134         if (hFind == INVALID_HANDLE_VALUE)

   135             continue;

   136 

   137         do {

   138             if (findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)

   139                 continue;

   140 

   141             String filename = String(findFileData.cFileName, wcslen(findFileData.cFileName));

   142             if ((!filename.startsWith("np", false) || !filename.endsWith("dll", false)) &&

   143                 (!equalIgnoringCase(filename, "Plugin.dll") || !it->endsWith("Shockwave 10", false)))

   144                 continue;

   145 

   146             String fullPath = *it + "\\" + filename;

   147             if (!uniqueFilenames.add(fullPath).second)

   148                 continue;

   149 

   150             paths.add(fullPath);

   151 

   152             if (equalIgnoringCase(filename, "npdsplay.dll"))

   153                 oldWMPPluginPath = fullPath;

   154             else if (equalIgnoringCase(filename, "np-mswmp.dll"))

   155                 newWMPPluginPath = fullPath;

   156 

   157         } while (FindNextFileW(hFind, &findFileData) != 0);

   158 

   159         FindClose(hFind);

   160     }

   161 

   162     addPluginPathsFromRegistry(HKEY_LOCAL_MACHINE, paths);

   163     addPluginPathsFromRegistry(HKEY_CURRENT_USER, paths);

   164 

   165     // If both the old and new WMP plugin are present in the plugins set, 

   166     // we remove the old one so we don't end up choosing the old one.

   167     if (!oldWMPPluginPath.isEmpty() && !newWMPPluginPath.isEmpty())

   168         paths.remove(oldWMPPluginPath);

   169 }

   170 

   171 static inline Vector<int> parseVersionString(const String& versionString)

   172 {

   173     Vector<int> version;

   174 

   175     unsigned startPos = 0;

   176     unsigned endPos;

   177     

   178     while (startPos < versionString.length()) {

   179         for (endPos = startPos; endPos < versionString.length(); ++endPos)

   180             if (versionString[endPos] == '.' || versionString[endPos] == '_')

   181                 break;

   182 

   183         int versionComponent = versionString.substring(startPos, endPos - startPos).toInt();

   184         version.append(versionComponent);

   185 

   186         startPos = endPos + 1;

   187     }

   188 

   189     return version;

   190 }

   191 

   192 // This returns whether versionA is higher than versionB

   193 static inline bool compareVersions(const Vector<int>& versionA, const Vector<int>& versionB)

   194 {

   195     for (unsigned i = 0; i < versionA.size(); i++) {

   196         if (i >= versionB.size())

   197             return true;

   198 

   199         if (versionA[i] > versionB[i])

   200             return true;

   201         else if (versionA[i] < versionB[i])

   202             return false;

   203     }

   204 

   205     // If we come here, the versions are either the same or versionB has an extra component, just return false

   206     return false;

   207 }

   208 

   209 static inline void addMozillaPluginDirectories(Vector<String>& directories)

   210 {

   211     // Enumerate all Mozilla plugin directories in the registry

   212     HKEY key;

   213     LONG result;

   214     

   215     result = RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("Software\\Mozilla"), 0, KEY_READ, &key);

   216     if (result == ERROR_SUCCESS) {

   217         WCHAR name[128];

   218         FILETIME lastModified;

   219 

   220         // Enumerate subkeys

   221         for (int i = 0;; i++) {

   222             DWORD nameLen = sizeof(name) / sizeof(WCHAR);

   223             result = RegEnumKeyExW(key, i, name, &nameLen, 0, 0, 0, &lastModified);

   224 

   225             if (result != ERROR_SUCCESS)

   226                 break;

   227 

   228             String extensionsPath = String(name, nameLen) + "\\Extensions";

   229             HKEY extensionsKey;

   230 

   231             // Try opening the key

   232             result = RegOpenKeyEx(key, extensionsPath.charactersWithNullTermination(), 0, KEY_READ, &extensionsKey);

   233 

   234             if (result == ERROR_SUCCESS) {

   235                 // Now get the plugins directory

   236                 WCHAR pluginsDirectoryStr[_MAX_PATH];

   237                 DWORD pluginsDirectorySize = sizeof(pluginsDirectoryStr);

   238                 DWORD type;

   239 

   240                 result = RegQueryValueEx(extensionsKey, TEXT("Plugins"), 0, &type, (LPBYTE)&pluginsDirectoryStr, &pluginsDirectorySize);

   241 

   242                 if (result == ERROR_SUCCESS && type == REG_SZ)

   243                     directories.append(String(pluginsDirectoryStr, pluginsDirectorySize / sizeof(WCHAR) - 1));

   244 

   245                 RegCloseKey(extensionsKey);

   246             }

   247         }

   248         

   249         RegCloseKey(key);

   250     }

   251 }

   252 

   253 static inline void addWindowsMediaPlayerPluginDirectory(Vector<String>& directories)

   254 {

   255 #if !OS(WINCE)

   256     // The new WMP Firefox plugin is installed in \PFiles\Plugins if it can't find any Firefox installs

   257     WCHAR pluginDirectoryStr[_MAX_PATH + 1];

   258     DWORD pluginDirectorySize = ::ExpandEnvironmentStringsW(TEXT("%SYSTEMDRIVE%\\PFiles\\Plugins"), pluginDirectoryStr, _countof(pluginDirectoryStr));

   259 

   260     if (pluginDirectorySize > 0 && pluginDirectorySize <= _countof(pluginDirectoryStr))

   261         directories.append(String(pluginDirectoryStr, pluginDirectorySize - 1));

   262 #endif

   263 

   264     DWORD type;

   265     WCHAR installationDirectoryStr[_MAX_PATH];

   266     DWORD installationDirectorySize = sizeof(installationDirectoryStr);

   267 

   268     HRESULT result = SHGetValue(HKEY_LOCAL_MACHINE, TEXT("Software\\Microsoft\\MediaPlayer"), TEXT("Installation Directory"), &type, (LPBYTE)&installationDirectoryStr, &installationDirectorySize);

   269 

   270     if (result == ERROR_SUCCESS && type == REG_SZ)

   271         directories.append(String(installationDirectoryStr, installationDirectorySize / sizeof(WCHAR) - 1));

   272 }

   273 

   274 static inline void addQuickTimePluginDirectory(Vector<String>& directories)

   275 {

   276     DWORD type;

   277     WCHAR installationDirectoryStr[_MAX_PATH];

   278     DWORD installationDirectorySize = sizeof(installationDirectoryStr);

   279 

   280     HRESULT result = SHGetValue(HKEY_LOCAL_MACHINE, TEXT("Software\\Apple Computer, Inc.\\QuickTime"), TEXT("InstallDir"), &type, (LPBYTE)&installationDirectoryStr, &installationDirectorySize);

   281 

   282     if (result == ERROR_SUCCESS && type == REG_SZ) {

   283         String pluginDir = String(installationDirectoryStr, installationDirectorySize / sizeof(WCHAR) - 1) + "\\plugins";

   284         directories.append(pluginDir);

   285     }

   286 }

   287 

   288 static inline void addAdobeAcrobatPluginDirectory(Vector<String>& directories)

   289 {

   290     HKEY key;

   291     HRESULT result = RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("Software\\Adobe\\Acrobat Reader"), 0, KEY_READ, &key);

   292     if (result != ERROR_SUCCESS)

   293         return;

   294 

   295     WCHAR name[128];

   296     FILETIME lastModified;

   297 

   298     Vector<int> latestAcrobatVersion;

   299     String latestAcrobatVersionString;

   300 

   301     // Enumerate subkeys

   302     for (int i = 0;; i++) {

   303         DWORD nameLen = sizeof(name) / sizeof(WCHAR);

   304         result = RegEnumKeyExW(key, i, name, &nameLen, 0, 0, 0, &lastModified);

   305 

   306         if (result != ERROR_SUCCESS)

   307             break;

   308 

   309         Vector<int> acrobatVersion = parseVersionString(String(name, nameLen));

   310         if (compareVersions(acrobatVersion, latestAcrobatVersion)) {

   311             latestAcrobatVersion = acrobatVersion;

   312             latestAcrobatVersionString = String(name, nameLen);

   313         }

   314     }

   315 

   316     if (!latestAcrobatVersionString.isNull()) {

   317         DWORD type;

   318         WCHAR acrobatInstallPathStr[_MAX_PATH];

   319         DWORD acrobatInstallPathSize = sizeof(acrobatInstallPathStr);

   320 

   321         String acrobatPluginKeyPath = "Software\\Adobe\\Acrobat Reader\\" + latestAcrobatVersionString + "\\InstallPath";

   322         result = SHGetValue(HKEY_LOCAL_MACHINE, acrobatPluginKeyPath.charactersWithNullTermination(), 0, &type, (LPBYTE)acrobatInstallPathStr, &acrobatInstallPathSize);

   323 

   324         if (result == ERROR_SUCCESS) {

   325             String acrobatPluginDirectory = String(acrobatInstallPathStr, acrobatInstallPathSize / sizeof(WCHAR) - 1) + "\\browser";

   326             directories.append(acrobatPluginDirectory);

   327         }

   328     }

   329 

   330     RegCloseKey(key);

   331 }

   332 

   333 static inline String safariPluginsDirectory()

   334 {

   335     WCHAR moduleFileNameStr[_MAX_PATH];

   336     static String pluginsDirectory;

   337     static bool cachedPluginDirectory = false;

   338 

   339     if (!cachedPluginDirectory) {

   340         cachedPluginDirectory = true;

   341 

   342         int moduleFileNameLen = GetModuleFileName(0, moduleFileNameStr, _MAX_PATH);

   343 

   344         if (!moduleFileNameLen || moduleFileNameLen == _MAX_PATH)

   345             goto exit;

   346 

   347         if (!PathRemoveFileSpec(moduleFileNameStr))

   348             goto exit;

   349 

   350         pluginsDirectory = String(moduleFileNameStr) + "\\Plugins";

   351     }

   352 exit:

   353     return pluginsDirectory;

   354 }

   355 

   356 static inline void addMacromediaPluginDirectories(Vector<String>& directories)

   357 {

   358 #if !OS(WINCE)

   359     WCHAR systemDirectoryStr[MAX_PATH];

   360 

   361     if (GetSystemDirectory(systemDirectoryStr, _countof(systemDirectoryStr)) == 0)

   362         return;

   363 

   364     WCHAR macromediaDirectoryStr[MAX_PATH];

   365 

   366     PathCombine(macromediaDirectoryStr, systemDirectoryStr, TEXT("macromed\\Flash"));

   367     directories.append(macromediaDirectoryStr);

   368 

   369     PathCombine(macromediaDirectoryStr, systemDirectoryStr, TEXT("macromed\\Shockwave 10"));

   370     directories.append(macromediaDirectoryStr);

   371 #endif

   372 }

   373 

   374 Vector<String> PluginDatabase::defaultPluginDirectories()

   375 {

   376     Vector<String> directories;

   377     String ourDirectory = safariPluginsDirectory();

   378 

   379     if (!ourDirectory.isNull())

   380         directories.append(ourDirectory);

   381     addQuickTimePluginDirectory(directories);

   382     addAdobeAcrobatPluginDirectory(directories);

   383     addMozillaPluginDirectories(directories);

   384     addWindowsMediaPlayerPluginDirectory(directories);

   385     addMacromediaPluginDirectories(directories);

   386 

   387     return directories;

   388 }

   389 

   390 bool PluginDatabase::isPreferredPluginDirectory(const String& directory)

   391 {

   392     String ourDirectory = safariPluginsDirectory();

   393 

   394     if (!ourDirectory.isNull() && !directory.isNull())

   395         return ourDirectory == directory;

   396 

   397     return false;

   398 }

   399 

   400 }