forked from suyu/suyu
service: acc: Ensure proper profile size
This commit is contained in:
parent
940618a64d
commit
5323d9f6b3
2 changed files with 50 additions and 12 deletions
|
@ -3,11 +3,13 @@
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <array>
|
#include <array>
|
||||||
|
|
||||||
#include "common/common_types.h"
|
#include "common/common_types.h"
|
||||||
#include "common/fs/file.h"
|
#include "common/fs/file.h"
|
||||||
#include "common/fs/path_util.h"
|
#include "common/fs/path_util.h"
|
||||||
#include "common/logging/log.h"
|
#include "common/logging/log.h"
|
||||||
#include "common/polyfill_ranges.h"
|
#include "common/polyfill_ranges.h"
|
||||||
|
#include "common/stb.h"
|
||||||
#include "common/string_util.h"
|
#include "common/string_util.h"
|
||||||
#include "common/swap.h"
|
#include "common/swap.h"
|
||||||
#include "core/constants.h"
|
#include "core/constants.h"
|
||||||
|
@ -38,9 +40,36 @@ static std::filesystem::path GetImagePath(const Common::UUID& uuid) {
|
||||||
fmt::format("system/save/8000000000000010/su/avators/{}.jpg", uuid.FormattedString());
|
fmt::format("system/save/8000000000000010/su/avators/{}.jpg", uuid.FormattedString());
|
||||||
}
|
}
|
||||||
|
|
||||||
static constexpr u32 SanitizeJPEGSize(std::size_t size) {
|
static void JPGToMemory(void* context, void* data, int len) {
|
||||||
|
std::vector<u8>* jpg_image = static_cast<std::vector<u8>*>(context);
|
||||||
|
unsigned char* jpg = static_cast<unsigned char*>(data);
|
||||||
|
jpg_image->insert(jpg_image->end(), jpg, jpg + len);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void SanitizeJPEGImageSize(std::vector<u8>& image) {
|
||||||
constexpr std::size_t max_jpeg_image_size = 0x20000;
|
constexpr std::size_t max_jpeg_image_size = 0x20000;
|
||||||
return static_cast<u32>(std::min(size, max_jpeg_image_size));
|
constexpr int profile_dimensions = 256;
|
||||||
|
int original_width, original_height, color_channels;
|
||||||
|
|
||||||
|
const auto plain_image =
|
||||||
|
stbi_load_from_memory(image.data(), static_cast<int>(image.size()), &original_width,
|
||||||
|
&original_height, &color_channels, STBI_rgb);
|
||||||
|
|
||||||
|
// Resize image to match 256*256
|
||||||
|
if (original_width != profile_dimensions || original_height != profile_dimensions) {
|
||||||
|
// Use vector instead of array to avoid overflowing the stack
|
||||||
|
std::vector<u8> out_image(profile_dimensions * profile_dimensions * STBI_rgb);
|
||||||
|
stbir_resize_uint8_srgb(plain_image, original_width, original_height, 0, out_image.data(),
|
||||||
|
profile_dimensions, profile_dimensions, 0, STBI_rgb, 0,
|
||||||
|
STBIR_FILTER_BOX);
|
||||||
|
image.clear();
|
||||||
|
if (!stbi_write_jpg_to_func(JPGToMemory, &image, profile_dimensions, profile_dimensions,
|
||||||
|
STBI_rgb, out_image.data(), 0)) {
|
||||||
|
LOG_ERROR(Service_ACC, "Failed to resize the user provided image.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
image.resize(std::min(image.size(), max_jpeg_image_size));
|
||||||
}
|
}
|
||||||
|
|
||||||
class IManagerForSystemService final : public ServiceFramework<IManagerForSystemService> {
|
class IManagerForSystemService final : public ServiceFramework<IManagerForSystemService> {
|
||||||
|
@ -339,19 +368,20 @@ protected:
|
||||||
LOG_WARNING(Service_ACC,
|
LOG_WARNING(Service_ACC,
|
||||||
"Failed to load user provided image! Falling back to built-in backup...");
|
"Failed to load user provided image! Falling back to built-in backup...");
|
||||||
ctx.WriteBuffer(Core::Constants::ACCOUNT_BACKUP_JPEG);
|
ctx.WriteBuffer(Core::Constants::ACCOUNT_BACKUP_JPEG);
|
||||||
rb.Push(SanitizeJPEGSize(Core::Constants::ACCOUNT_BACKUP_JPEG.size()));
|
rb.Push(static_cast<u32>(Core::Constants::ACCOUNT_BACKUP_JPEG.size()));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const u32 size = SanitizeJPEGSize(image.GetSize());
|
std::vector<u8> buffer(image.GetSize());
|
||||||
std::vector<u8> buffer(size);
|
|
||||||
|
|
||||||
if (image.Read(buffer) != buffer.size()) {
|
if (image.Read(buffer) != buffer.size()) {
|
||||||
LOG_ERROR(Service_ACC, "Failed to read all the bytes in the user provided image.");
|
LOG_ERROR(Service_ACC, "Failed to read all the bytes in the user provided image.");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
SanitizeJPEGImageSize(buffer);
|
||||||
|
|
||||||
ctx.WriteBuffer(buffer);
|
ctx.WriteBuffer(buffer);
|
||||||
rb.Push<u32>(size);
|
rb.Push(static_cast<u32>(buffer.size()));
|
||||||
}
|
}
|
||||||
|
|
||||||
void GetImageSize(HLERequestContext& ctx) {
|
void GetImageSize(HLERequestContext& ctx) {
|
||||||
|
@ -365,10 +395,18 @@ protected:
|
||||||
if (!image.IsOpen()) {
|
if (!image.IsOpen()) {
|
||||||
LOG_WARNING(Service_ACC,
|
LOG_WARNING(Service_ACC,
|
||||||
"Failed to load user provided image! Falling back to built-in backup...");
|
"Failed to load user provided image! Falling back to built-in backup...");
|
||||||
rb.Push(SanitizeJPEGSize(Core::Constants::ACCOUNT_BACKUP_JPEG.size()));
|
rb.Push(static_cast<u32>(Core::Constants::ACCOUNT_BACKUP_JPEG.size()));
|
||||||
} else {
|
return;
|
||||||
rb.Push(SanitizeJPEGSize(image.GetSize()));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::vector<u8> buffer(image.GetSize());
|
||||||
|
|
||||||
|
if (image.Read(buffer) != buffer.size()) {
|
||||||
|
LOG_ERROR(Service_ACC, "Failed to read all the bytes in the user provided image.");
|
||||||
|
}
|
||||||
|
|
||||||
|
SanitizeJPEGImageSize(buffer);
|
||||||
|
rb.Push(static_cast<u32>(buffer.size()));
|
||||||
}
|
}
|
||||||
|
|
||||||
void Store(HLERequestContext& ctx) {
|
void Store(HLERequestContext& ctx) {
|
||||||
|
|
|
@ -306,10 +306,10 @@ void ConfigureProfileManager::SetUserImage() {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Some games crash when the profile image is too big. Resize any image bigger than 256x256
|
// Profile image must be 256x256
|
||||||
QImage image(image_path);
|
QImage image(image_path);
|
||||||
if (image.width() > 256 || image.height() > 256) {
|
if (image.width() != 256 || image.height() != 256) {
|
||||||
image = image.scaled(256, 256, Qt::KeepAspectRatio);
|
image = image.scaled(256, 256, Qt::KeepAspectRatioByExpanding, Qt::SmoothTransformation);
|
||||||
if (!image.save(image_path)) {
|
if (!image.save(image_path)) {
|
||||||
QMessageBox::warning(this, tr("Error resizing user image"),
|
QMessageBox::warning(this, tr("Error resizing user image"),
|
||||||
tr("Unable to resize image"));
|
tr("Unable to resize image"));
|
||||||
|
|
Loading…
Reference in a new issue