gecko-dev/image/build/nsImageModule.cpp
Jon Bauman 3b93e0e5f8 Bug 1635391 - Firefox does not download WebP and AVIF when their support flags are off. r=mattwoodrow
Dynamically toggle support for displaying image/webp and image/avif according
to the values of the image.webp.enabled and image.avif.enabled prefs.

Differential Revision: https://phabricator.services.mozilla.com/D76977
2020-05-27 18:51:25 +00:00

86 lines
2.5 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "nsImageModule.h"
#include "mozilla/ModuleUtils.h"
#include "mozilla/StaticPrefs_image.h"
#include "DecodePool.h"
#include "ImageFactory.h"
#include "nsICategoryManager.h"
#include "ShutdownTracker.h"
#include "SurfaceCache.h"
#include "imgLoader.h"
using namespace mozilla::image;
struct ImageEnablementCookie {
bool (*mIsEnabled)();
const nsLiteralCString mMimeType;
};
static void UpdateContentViewerRegistration(const char* aPref, void* aData) {
auto* cookie = static_cast<ImageEnablementCookie*>(aData);
nsCOMPtr<nsICategoryManager> catMan =
do_GetService(NS_CATEGORYMANAGER_CONTRACTID);
if (!catMan) {
return;
}
static nsLiteralCString kCategory =
NS_LITERAL_CSTRING("Gecko-Content-Viewers");
static nsLiteralCString kContractId = NS_LITERAL_CSTRING(
"@mozilla.org/content/plugin/document-loader-factory;1");
if (cookie->mIsEnabled()) {
catMan->AddCategoryEntry(kCategory, cookie->mMimeType, kContractId,
false /* aPersist */, true /* aReplace */);
} else {
catMan->DeleteCategoryEntry(
kCategory, cookie->mMimeType, false /* aPersist */
);
}
}
static bool sInitialized = false;
nsresult mozilla::image::EnsureModuleInitialized() {
MOZ_ASSERT(NS_IsMainThread());
if (sInitialized) {
return NS_OK;
}
static ImageEnablementCookie kAVIFCookie = {
mozilla::StaticPrefs::image_avif_enabled,
NS_LITERAL_CSTRING("image/avif")};
static ImageEnablementCookie kWebPCookie = {
mozilla::StaticPrefs::image_webp_enabled,
NS_LITERAL_CSTRING("image/webp")};
Preferences::RegisterCallbackAndCall(UpdateContentViewerRegistration,
"image.avif.enabled", &kAVIFCookie);
Preferences::RegisterCallbackAndCall(UpdateContentViewerRegistration,
"image.webp.enabled", &kWebPCookie);
mozilla::image::ShutdownTracker::Initialize();
mozilla::image::ImageFactory::Initialize();
mozilla::image::DecodePool::Initialize();
mozilla::image::SurfaceCache::Initialize();
imgLoader::GlobalInit();
sInitialized = true;
return NS_OK;
}
void mozilla::image::ShutdownModule() {
if (!sInitialized) {
return;
}
imgLoader::Shutdown();
mozilla::image::SurfaceCache::Shutdown();
sInitialized = false;
}