PluginInfoStore.cpp [plain text]
#include "config.h"
#include "PluginInfoStore.h"
#include <WebCore/KURL.h>
#include <WebCore/MIMETypeRegistry.h>
#include <algorithm>
#include <wtf/ListHashSet.h>
#include <wtf/StdLibExtras.h>
using namespace std;
using namespace WebCore;
namespace WebKit {
PluginInfoStore::PluginInfoStore()
: m_pluginListIsUpToDate(false)
{
}
void PluginInfoStore::setAdditionalPluginsDirectories(const Vector<String>& directories)
{
m_additionalPluginsDirectories = directories;
refresh();
}
void PluginInfoStore::refresh()
{
m_pluginListIsUpToDate = false;
}
template <typename T, typename U>
static void addFromVector(T& hashSet, const U& vector)
{
for (size_t i = 0; i < vector.size(); ++i)
hashSet.add(vector[i]);
}
#if OS(WINDOWS)
typedef ListHashSet<String, 32, CaseFoldingHash> PathHashSet;
#else
typedef ListHashSet<String, 32> PathHashSet;
#endif
void PluginInfoStore::loadPluginsIfNecessary()
{
if (m_pluginListIsUpToDate)
return;
PathHashSet uniquePluginPaths;
for (size_t i = 0; i < m_additionalPluginsDirectories.size(); ++i)
addFromVector(uniquePluginPaths, pluginPathsInDirectory(m_additionalPluginsDirectories[i]));
Vector<String> directories = pluginsDirectories();
for (size_t i = 0; i < directories.size(); ++i)
addFromVector(uniquePluginPaths, pluginPathsInDirectory(directories[i]));
addFromVector(uniquePluginPaths, individualPluginPaths());
Vector<Plugin> plugins;
PathHashSet::const_iterator end = uniquePluginPaths.end();
for (PathHashSet::const_iterator it = uniquePluginPaths.begin(); it != end; ++it)
loadPlugin(plugins, *it);
m_plugins.swap(plugins);
m_pluginListIsUpToDate = true;
}
void PluginInfoStore::loadPlugin(Vector<Plugin>& plugins, const String& pluginPath)
{
Plugin plugin;
if (!getPluginInfo(pluginPath, plugin))
return;
if (!shouldUsePlugin(plugins, plugin))
return;
plugins.append(plugin);
}
Vector<PluginInfoStore::Plugin> PluginInfoStore::plugins()
{
loadPluginsIfNecessary();
Vector<Plugin> plugins(m_plugins);
return plugins;
}
PluginInfoStore::Plugin PluginInfoStore::findPluginForMIMEType(const String& mimeType)
{
ASSERT(!mimeType.isNull());
for (size_t i = 0; i < m_plugins.size(); ++i) {
const Plugin& plugin = m_plugins[i];
for (size_t j = 0; j < plugin.info.mimes.size(); ++j) {
const MimeClassInfo& mimeClassInfo = plugin.info.mimes[j];
if (mimeClassInfo.type == mimeType)
return plugin;
}
}
return Plugin();
}
PluginInfoStore::Plugin PluginInfoStore::findPluginForExtension(const String& extension, String& mimeType)
{
ASSERT(!extension.isNull());
for (size_t i = 0; i < m_plugins.size(); ++i) {
const Plugin& plugin = m_plugins[i];
for (size_t j = 0; j < plugin.info.mimes.size(); ++j) {
const MimeClassInfo& mimeClassInfo = plugin.info.mimes[j];
const Vector<String>& extensions = mimeClassInfo.extensions;
if (find(extensions.begin(), extensions.end(), extension) != extensions.end()) {
mimeType = mimeClassInfo.type;
return plugin;
}
}
}
return Plugin();
}
static inline String pathExtension(const KURL& url)
{
String extension;
String filename = url.lastPathComponent();
if (!filename.endsWith("/")) {
int extensionPos = filename.reverseFind('.');
if (extensionPos != -1)
extension = filename.substring(extensionPos + 1);
}
return extension;
}
#if !PLATFORM(MAC)
String PluginInfoStore::getMIMETypeForExtension(const String& extension)
{
return MIMETypeRegistry::getMIMETypeForExtension(extension);
}
#endif
PluginInfoStore::Plugin PluginInfoStore::findPlugin(String& mimeType, const KURL& url)
{
loadPluginsIfNecessary();
if (!mimeType.isNull()) {
Plugin plugin = findPluginForMIMEType(mimeType);
if (!plugin.path.isNull())
return plugin;
}
String extension = pathExtension(url).lower();
if (!extension.isNull() && mimeType.isEmpty()) {
Plugin plugin = findPluginForExtension(extension, mimeType);
if (!plugin.path.isNull())
return plugin;
String extensionMimeType = getMIMETypeForExtension(extension);
if (!extensionMimeType.isNull()) {
Plugin plugin = findPluginForMIMEType(extensionMimeType);
if (!plugin.path.isNull()) {
mimeType = extensionMimeType;
return plugin;
}
}
}
return Plugin();
}
PluginInfoStore::Plugin PluginInfoStore::infoForPluginWithPath(const String& pluginPath)
{
for (size_t i = 0; i < m_plugins.size(); ++i) {
if (m_plugins[i].path == pluginPath)
return m_plugins[i];
}
ASSERT_NOT_REACHED();
return Plugin();
}
}