PluginDatabase.cpp [plain text]
#include "config.h"
#include "PluginDatabase.h"
#include "Frame.h"
#include "KURL.h"
#include "PluginPackage.h"
#include <stdlib.h>
namespace WebCore {
PluginDatabase* PluginDatabase::installedPlugins()
{
static PluginDatabase* plugins = 0;
if (!plugins) {
plugins = new PluginDatabase;
plugins->setPluginDirectories(PluginDatabase::defaultPluginDirectories());
plugins->refresh();
}
return plugins;
}
bool PluginDatabase::isMIMETypeRegistered(const String& mimeType)
{
if (mimeType.isNull())
return false;
if (m_registeredMIMETypes.contains(mimeType))
return true;
return (refresh() && m_registeredMIMETypes.contains(mimeType));
}
void PluginDatabase::addExtraPluginDirectory(const String& directory)
{
m_pluginDirectories.append(directory);
refresh();
}
bool PluginDatabase::refresh()
{
bool pluginSetChanged = false;
if (!m_plugins.isEmpty()) {
PluginSet pluginsToUnload;
getDeletedPlugins(pluginsToUnload);
PluginSet::const_iterator end = pluginsToUnload.end();
for (PluginSet::const_iterator it = pluginsToUnload.begin(); it != end; ++it)
remove(it->get());
pluginSetChanged = !pluginsToUnload.isEmpty();
}
HashSet<String> paths;
getPluginPathsInDirectories(paths);
HashMap<String, time_t> pathsWithTimes;
bool shouldSkipUnchangedFiles = !pluginSetChanged;
HashSet<String>::const_iterator pathsEnd = paths.end();
for (HashSet<String>::const_iterator it = paths.begin(); it != pathsEnd; ++it) {
time_t lastModified;
if (!getFileModificationTime(*it, lastModified))
continue;
pathsWithTimes.add(*it, lastModified);
if (shouldSkipUnchangedFiles && m_pluginPathsWithTimes.get(*it) == lastModified)
continue;
if (RefPtr<PluginPackage> oldPackage = m_pluginsByPath.get(*it)) {
ASSERT(!shouldSkipUnchangedFiles || oldPackage->lastModified() != lastModified);
remove(oldPackage.get());
}
RefPtr<PluginPackage> package = PluginPackage::createPackage(*it, lastModified);
if (package && add(package.release()))
pluginSetChanged = true;
}
pathsWithTimes.swap(m_pluginPathsWithTimes);
if (!pluginSetChanged)
return false;
m_registeredMIMETypes.clear();
PluginSet::const_iterator end = m_plugins.end();
for (PluginSet::const_iterator it = m_plugins.begin(); it != end; ++it) {
MIMEToDescriptionsMap::const_iterator map_end = (*it)->mimeToDescriptions().end();
for (MIMEToDescriptionsMap::const_iterator map_it = (*it)->mimeToDescriptions().begin(); map_it != map_end; ++map_it) {
m_registeredMIMETypes.add(map_it->first);
}
}
return true;
}
Vector<PluginPackage*> PluginDatabase::plugins() const
{
Vector<PluginPackage*> result;
PluginSet::const_iterator end = m_plugins.end();
for (PluginSet::const_iterator it = m_plugins.begin(); it != end; ++it)
result.append((*it).get());
return result;
}
int PluginDatabase::preferredPluginCompare(const void* a, const void* b)
{
PluginPackage* pluginA = *static_cast<PluginPackage* const*>(a);
PluginPackage* pluginB = *static_cast<PluginPackage* const*>(b);
return pluginA->compare(*pluginB);
}
PluginPackage* PluginDatabase::pluginForMIMEType(const String& mimeType)
{
if (mimeType.isEmpty())
return 0;
String key = mimeType.lower();
PluginSet::const_iterator end = m_plugins.end();
Vector<PluginPackage*, 2> pluginChoices;
for (PluginSet::const_iterator it = m_plugins.begin(); it != end; ++it) {
if ((*it)->mimeToDescriptions().contains(key))
pluginChoices.append((*it).get());
}
if (pluginChoices.isEmpty())
return 0;
qsort(pluginChoices.data(), pluginChoices.size(), sizeof(PluginPackage*), PluginDatabase::preferredPluginCompare);
return pluginChoices[0];
}
String PluginDatabase::MIMETypeForExtension(const String& extension) const
{
if (extension.isEmpty())
return String();
PluginSet::const_iterator end = m_plugins.end();
String mimeType;
Vector<PluginPackage*, 2> pluginChoices;
HashMap<PluginPackage*, String> mimeTypeForPlugin;
for (PluginSet::const_iterator it = m_plugins.begin(); it != end; ++it) {
MIMEToExtensionsMap::const_iterator mime_end = (*it)->mimeToExtensions().end();
for (MIMEToExtensionsMap::const_iterator mime_it = (*it)->mimeToExtensions().begin(); mime_it != mime_end; ++mime_it) {
const Vector<String>& extensions = mime_it->second;
bool foundMapping = false;
for (unsigned i = 0; i < extensions.size(); i++) {
if (equalIgnoringCase(extensions[i], extension)) {
PluginPackage* plugin = (*it).get();
pluginChoices.append(plugin);
mimeTypeForPlugin.add(plugin, mime_it->first);
foundMapping = true;
break;
}
}
if (foundMapping)
break;
}
}
if (pluginChoices.isEmpty())
return String();
qsort(pluginChoices.data(), pluginChoices.size(), sizeof(PluginPackage*), PluginDatabase::preferredPluginCompare);
return mimeTypeForPlugin.get(pluginChoices[0]);
}
PluginPackage* PluginDatabase::findPlugin(const KURL& url, String& mimeType)
{
PluginPackage* plugin = pluginForMIMEType(mimeType);
String filename = url.string();
if (!plugin) {
String filename = url.lastPathComponent();
if (!filename.endsWith("/")) {
int extensionPos = filename.reverseFind('.');
if (extensionPos != -1) {
String extension = filename.substring(extensionPos + 1);
mimeType = MIMETypeForExtension(extension);
plugin = pluginForMIMEType(mimeType);
}
}
}
return plugin;
}
void PluginDatabase::getDeletedPlugins(PluginSet& plugins) const
{
PluginSet::const_iterator end = m_plugins.end();
for (PluginSet::const_iterator it = m_plugins.begin(); it != end; ++it) {
if (!fileExists((*it)->path()))
plugins.add(*it);
}
}
bool PluginDatabase::add(PassRefPtr<PluginPackage> prpPackage)
{
ASSERT_ARG(prpPackage, prpPackage);
RefPtr<PluginPackage> package = prpPackage;
if (!m_plugins.add(package).second)
return false;
m_pluginsByPath.add(package->path(), package);
return true;
}
void PluginDatabase::remove(PluginPackage* package)
{
m_plugins.remove(package);
m_pluginsByPath.remove(package->path());
}
#if !PLATFORM(WIN_OS) || PLATFORM(WX)
Vector<String> PluginDatabase::defaultPluginDirectories()
{
Vector<String> paths;
#if defined(XP_UNIX)
String userPluginPath = homeDirectoryPath();
userPluginPath.append(String("/.mozilla/plugins"));
paths.append(userPluginPath);
userPluginPath = homeDirectoryPath();
userPluginPath.append(String("/.netscape/plugins"));
paths.append(userPluginPath);
paths.append("/usr/lib/browser/plugins");
paths.append("/usr/local/lib/mozilla/plugins");
paths.append("/usr/lib/firefox/plugins");
paths.append("/usr/lib64/browser-plugins");
paths.append("/usr/lib/browser-plugins");
paths.append("/usr/lib/mozilla/plugins");
paths.append("/usr/local/netscape/plugins");
paths.append("/opt/mozilla/plugins");
paths.append("/opt/mozilla/lib/plugins");
paths.append("/opt/netscape/plugins");
paths.append("/opt/netscape/communicator/plugins");
paths.append("/usr/lib/netscape/plugins");
paths.append("/usr/lib/netscape/plugins-libc5");
paths.append("/usr/lib/netscape/plugins-libc6");
paths.append("/usr/lib64/netscape/plugins");
paths.append("/usr/lib64/mozilla/plugins");
String mozHome(getenv("MOZILLA_HOME"));
mozHome.append("/plugins");
paths.append(mozHome);
Vector<String> mozPaths;
String mozPath(getenv("MOZ_PLUGIN_PATH"));
mozPath.split(UChar(':'), false, mozPaths);
paths.append(mozPaths);
#elif defined(XP_MACOSX)
String userPluginPath = homeDirectoryPath();
userPluginPath.append(String("/Library/Internet Plug-Ins"));
paths.append(userPluginPath);
paths.append("/Library/Internet Plug-Ins");
#elif defined(XP_WIN)
String userPluginPath = homeDirectoryPath();
userPluginPath.append(String("\\Application Data\\Mozilla\\plugins"));
paths.append(userPluginPath);
#endif
#if PLATFORM(QT)
Vector<String> qtPaths;
String qtPath(getenv("QTWEBKIT_PLUGIN_PATH"));
qtPath.split(UChar(':'), false, qtPaths);
paths.append(qtPaths);
#endif
return paths;
}
bool PluginDatabase::isPreferredPluginDirectory(const String& path)
{
String preferredPath = homeDirectoryPath();
#if defined(XP_UNIX)
preferredPath.append(String("/.mozilla/plugins"));
#elif defined(XP_MACOSX)
preferredPath.append(String("/Library/Internet Plug-Ins"));
#elif defined(XP_WIN)
preferredPath.append(String("\\Application Data\\Mozilla\\plugins"));
#endif
return path == preferredPath;
}
void PluginDatabase::getPluginPathsInDirectories(HashSet<String>& paths) const
{
HashSet<String> uniqueFilenames;
#if defined(XP_UNIX)
String fileNameFilter("*.so");
#else
String fileNameFilter("");
#endif
Vector<String>::const_iterator dirsEnd = m_pluginDirectories.end();
for (Vector<String>::const_iterator dIt = m_pluginDirectories.begin(); dIt != dirsEnd; ++dIt) {
Vector<String> pluginPaths = listDirectory(*dIt, fileNameFilter);
Vector<String>::const_iterator pluginsEnd = pluginPaths.end();
for (Vector<String>::const_iterator pIt = pluginPaths.begin(); pIt != pluginsEnd; ++pIt) {
if (!fileExists(*pIt))
continue;
paths.add(*pIt);
}
}
}
#endif // !PLATFORM(WIN_OS)
}