mirror of
https://github.com/classilla/tenfourfox.git
synced 2025-03-08 09:32:25 +00:00
145 lines
4.2 KiB
C++
145 lines
4.2 KiB
C++
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* 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 <stdlib.h>
|
|
#include "mozilla/dom/ContentChild.h"
|
|
#include "nsMimeTypes.h"
|
|
#include "nsIURL.h"
|
|
#include "nsXULAppAPI.h"
|
|
#include "AndroidBridge.h"
|
|
#include "nsIconChannel.h"
|
|
#include "nsIStringStream.h"
|
|
#include "nsNetUtil.h"
|
|
#include "nsComponentManagerUtils.h"
|
|
#include "nsNullPrincipal.h"
|
|
|
|
NS_IMPL_ISUPPORTS(nsIconChannel,
|
|
nsIRequest,
|
|
nsIChannel)
|
|
|
|
using namespace mozilla;
|
|
using mozilla::dom::ContentChild;
|
|
|
|
static nsresult
|
|
GetIconForExtension(const nsACString& aFileExt, uint32_t aIconSize,
|
|
uint8_t* const aBuf)
|
|
{
|
|
if (!AndroidBridge::Bridge()) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
AndroidBridge::Bridge()->GetIconForExtension(aFileExt, aIconSize, aBuf);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
static nsresult
|
|
CallRemoteGetIconForExtension(const nsACString& aFileExt, uint32_t aIconSize,
|
|
uint8_t* const aBuf)
|
|
{
|
|
NS_ENSURE_TRUE(aBuf != nullptr, NS_ERROR_NULL_POINTER);
|
|
|
|
// An array has to be used to get data from remote process
|
|
InfallibleTArray<uint8_t> bits;
|
|
uint32_t bufSize = aIconSize * aIconSize * 4;
|
|
|
|
if (!ContentChild::GetSingleton()->SendGetIconForExtension(
|
|
PromiseFlatCString(aFileExt), aIconSize, &bits)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
NS_ASSERTION(bits.Length() == bufSize, "Pixels array is incomplete");
|
|
if (bits.Length() != bufSize) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
memcpy(aBuf, bits.Elements(), bufSize);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
static nsresult
|
|
moz_icon_to_channel(nsIURI* aURI, const nsACString& aFileExt,
|
|
uint32_t aIconSize, nsIChannel** aChannel)
|
|
{
|
|
NS_ENSURE_TRUE(aIconSize < 256 && aIconSize > 0, NS_ERROR_UNEXPECTED);
|
|
|
|
int width = aIconSize;
|
|
int height = aIconSize;
|
|
|
|
// moz-icon data should have two bytes for the size,
|
|
// then the ARGB pixel values with pre-multiplied Alpha
|
|
const int channels = 4;
|
|
long int buf_size = 2 + channels * height * width;
|
|
uint8_t* const buf = (uint8_t*)moz_xmalloc(buf_size);
|
|
NS_ENSURE_TRUE(buf, NS_ERROR_OUT_OF_MEMORY);
|
|
uint8_t* out = buf;
|
|
|
|
*(out++) = width;
|
|
*(out++) = height;
|
|
|
|
nsresult rv;
|
|
if (XRE_IsParentProcess()) {
|
|
rv = GetIconForExtension(aFileExt, aIconSize, out);
|
|
} else {
|
|
rv = CallRemoteGetIconForExtension(aFileExt, aIconSize, out);
|
|
}
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Encode the RGBA data
|
|
const uint8_t* in = out;
|
|
for (int y = 0; y < height; ++y) {
|
|
for (int x = 0; x < width; ++x) {
|
|
uint8_t r = *(in++);
|
|
uint8_t g = *(in++);
|
|
uint8_t b = *(in++);
|
|
uint8_t a = *(in++);
|
|
#define DO_PREMULTIPLY(c_) uint8_t(uint16_t(c_) * uint16_t(a) / uint16_t(255))
|
|
*(out++) = DO_PREMULTIPLY(b);
|
|
*(out++) = DO_PREMULTIPLY(g);
|
|
*(out++) = DO_PREMULTIPLY(r);
|
|
*(out++) = a;
|
|
#undef DO_PREMULTIPLY
|
|
}
|
|
}
|
|
|
|
nsCOMPtr<nsIStringInputStream> stream =
|
|
do_CreateInstance("@mozilla.org/io/string-input-stream;1", &rv);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = stream->AdoptData((char*)buf, buf_size);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIPrincipal> nullPrincipal = nsNullPrincipal::Create();
|
|
NS_ENSURE_TRUE(nullPrincipal, NS_ERROR_FAILURE);
|
|
|
|
return NS_NewInputStreamChannel(aChannel,
|
|
aURI,
|
|
stream,
|
|
nullPrincipal,
|
|
nsILoadInfo::SEC_NORMAL,
|
|
nsIContentPolicy::TYPE_OTHER,
|
|
NS_LITERAL_CSTRING(IMAGE_ICON_MS));
|
|
}
|
|
|
|
nsresult
|
|
nsIconChannel::Init(nsIURI* aURI)
|
|
{
|
|
nsCOMPtr<nsIMozIconURI> iconURI = do_QueryInterface(aURI);
|
|
NS_ASSERTION(iconURI, "URI is not an nsIMozIconURI");
|
|
|
|
nsAutoCString stockIcon;
|
|
iconURI->GetStockIcon(stockIcon);
|
|
|
|
uint32_t desiredImageSize;
|
|
iconURI->GetImageSize(&desiredImageSize);
|
|
|
|
nsAutoCString iconFileExt;
|
|
iconURI->GetFileExtension(iconFileExt);
|
|
|
|
return moz_icon_to_channel(iconURI, iconFileExt, desiredImageSize,
|
|
getter_AddRefs(mRealChannel));
|
|
}
|