asgardius/virtualx-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add buffer size check to Image.load_tga_from_buffer(). Fixes #67985

Browse source
This commit is contained in:
Mark Riedesel 2022-10-30 11:20:50 -05:00
parent ad3f2a2340
commit 5cb07486db
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
modules/tga/image_loader_tga.cpp
View file

@ -284,14 +284,21 @@ Error ImageLoaderTGA::load_image(Ref<Image> p_image, Ref<FileAccess> f, BitField
err = FAILED; err = FAILED;
} }
uint64_t color_map_size;
if (has_color_map) { if (has_color_map) {
if (tga_header.color_map_length > 256 || (tga_header.color_map_depth != 24) || tga_header.color_map_type != 1) { if (tga_header.color_map_length > 256 || (tga_header.color_map_depth != 24) || tga_header.color_map_type != 1) {
err = FAILED; err = FAILED;
} }
color_map_size = tga_header.color_map_length * (tga_header.color_map_depth >> 3);
} else { } else {
if (tga_header.color_map_type) { if (tga_header.color_map_type) {
err = FAILED; err = FAILED;
} }
color_map_size = 0;
}
if ((src_image_len - f->get_position()) < (tga_header.id_length + color_map_size)) {
err = FAILED; // TGA data appears to be truncated (fewer bytes than expected).
} }
if (tga_header.image_width <= 0 || tga_header.image_height <= 0) { if (tga_header.image_width <= 0 || tga_header.image_height <= 0) {
@ -308,7 +315,6 @@ Error ImageLoaderTGA::load_image(Ref<Image> p_image, Ref<FileAccess> f, BitField
Vector<uint8_t> palette; Vector<uint8_t> palette;
if (has_color_map) { if (has_color_map) {
size_t color_map_size = tga_header.color_map_length * (tga_header.color_map_depth >> 3);
err = palette.resize(color_map_size); err = palette.resize(color_map_size);
if (err == OK) { if (err == OK) {
uint8_t *palette_w = palette.ptrw(); uint8_t *palette_w = palette.ptrw();