Fix T62852: crash reading corrupt DDS file.

This commit is contained in:
Brecht Van Lommel
2019-04-16 03:39:29 +02:00
parent 6a50a3d140
commit db80d5c560
4 changed files with 26 additions and 8 deletions

View File

@@ -1148,6 +1148,12 @@ void *DirectDrawSurface::readData(uint &rsize)
stream.seek(header_size);
mem_read(stream, data, size);
if (stream.failed) {
free(data);
data = NULL;
rsize = 0;
}
// Maybe check if size == rsize? assert() isn't in this scope...
return data;

View File

@@ -30,7 +30,7 @@ static const char *msg_error_read = "DDS: trying to read beyond end of stream (c
unsigned int Stream::seek(unsigned int p)
{
if (p > size) {
puts(msg_error_seek);
set_failed(msg_error_seek);
}
else {
pos = p;
@@ -42,7 +42,7 @@ unsigned int Stream::seek(unsigned int p)
unsigned int mem_read(Stream & mem, unsigned long long & i)
{
if (mem.pos + 8 > mem.size) {
puts(msg_error_seek);
mem.set_failed(msg_error_seek);
return(0);
}
memcpy(&i, mem.mem + mem.pos, 8); // @@ todo: make sure little endian
@@ -53,7 +53,7 @@ unsigned int mem_read(Stream & mem, unsigned long long & i)
unsigned int mem_read(Stream & mem, unsigned int & i)
{
if (mem.pos + 4 > mem.size) {
puts(msg_error_read);
mem.set_failed(msg_error_read);
return(0);
}
memcpy(&i, mem.mem + mem.pos, 4); // @@ todo: make sure little endian
@@ -64,7 +64,7 @@ unsigned int mem_read(Stream & mem, unsigned int & i)
unsigned int mem_read(Stream & mem, unsigned short & i)
{
if (mem.pos + 2 > mem.size) {
puts(msg_error_read);
mem.set_failed(msg_error_read);
return(0);
}
memcpy(&i, mem.mem + mem.pos, 2); // @@ todo: make sure little endian
@@ -75,7 +75,7 @@ unsigned int mem_read(Stream & mem, unsigned short & i)
unsigned int mem_read(Stream & mem, unsigned char & i)
{
if (mem.pos + 1 > mem.size) {
puts(msg_error_read);
mem.set_failed(msg_error_read);
return(0);
}
i = (mem.mem + mem.pos)[0];
@@ -86,10 +86,18 @@ unsigned int mem_read(Stream & mem, unsigned char & i)
unsigned int mem_read(Stream & mem, unsigned char *i, unsigned int cnt)
{
if (mem.pos + cnt > mem.size) {
puts(msg_error_read);
mem.set_failed(msg_error_read);
return(0);
}
memcpy(i, mem.mem + mem.pos, cnt);
mem.pos += cnt;
return(cnt);
}
void Stream::set_failed(const char *msg)
{
if (!failed) {
puts(msg_error_seek);
failed = true;
}
}

View File

@@ -28,8 +28,10 @@ struct Stream {
unsigned char *mem; // location in memory
unsigned int size; // size
unsigned int pos; // current position
Stream(unsigned char *m, unsigned int s) : mem(m), size(s), pos(0) {}
bool failed; // error occured when seeking
Stream(unsigned char *m, unsigned int s) : mem(m), size(s), pos(0), failed(false) {}
unsigned int seek(unsigned int p);
void set_failed(const char *msg);
};
unsigned int mem_read(Stream & mem, unsigned long long & i);

View File

@@ -165,7 +165,9 @@ struct ImBuf *imb_load_dds(const unsigned char *mem, size_t size, int flags, cha
ibuf->dds_data.data = (unsigned char *)dds.readData(ibuf->dds_data.size);
/* flip compressed texture */
FlipDXTCImage(dds.width(), dds.height(), dds.mipmapCount(), dds.fourCC(), ibuf->dds_data.data);
if (ibuf->dds_data.data) {
FlipDXTCImage(dds.width(), dds.height(), dds.mipmapCount(), dds.fourCC(), ibuf->dds_data.data);
}
}
else {
ibuf->dds_data.data = NULL;