Download-Tools/mkvtoolnix
1
0
Fork 0
mirror of https://gitlab.com/mbunkus/mkvtoolnix.git synced 2024-12-23 19:31:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Safer handling of closing files. Also accept text files with a length < 4 bytes and don't segfault on them ;)

Browse Source
This commit is contained in:
Moritz Bunkus 2005-04-25 12:24:49 +00:00
parent 1f80ec1867
commit 7aaa57b0e0
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/common/mm_io.cpp
View File

@ -197,8 +197,10 @@ mm_file_io_c::read(void *buffer,
void
mm_file_io_c::close() {
if (file != NULL)
if (NULL != file) {
fclose((FILE *)file);
file = NULL;
}
}
bool
@ -288,8 +290,10 @@ mm_file_io_c::mm_file_io_c(const string &path,
void
mm_file_io_c::close() {
if (file != NULL)
if (NULL != file) {
CloseHandle((HANDLE)file);
file = NULL;
}
file_name = "";
}
@ -965,37 +969,38 @@ mm_mem_io_c::eof() {
mm_text_io_c::mm_text_io_c(mm_io_c *_in,
bool _delete_in):
mm_proxy_io_c(_in, _delete_in) {
mm_proxy_io_c(_in, _delete_in),
byte_order(BO_NONE),
bom_len(0) {
unsigned char buffer[4];
int num_read;
_in->setFilePointer(0, seek_beginning);
if (_in->read(buffer, 4) != 4) {
_in->close();
if (_delete_in)
delete _in;
throw error_c("end-of-file reading the BOM");
num_read = _in->read(buffer, 4);
if (2 > num_read) {
_in->setFilePointer(0, seek_beginning);
return;
}
if ((buffer[0] == 0xef) && (buffer[1] == 0xbb) && (buffer[2] == 0xbf)) {
if ((3 <= num_read) &&
(buffer[0] == 0xef) && (buffer[1] == 0xbb) && (buffer[2] == 0xbf)) {
byte_order = BO_UTF8;
bom_len = 3;
} else if ((buffer[0] == 0xff) && (buffer[1] == 0xfe) &&
} else if ((4 <= num_read) && (buffer[0] == 0xff) && (buffer[1] == 0xfe) &&
(buffer[2] == 0x00) && (buffer[3] == 0x00)) {
byte_order = BO_UTF32_LE;
bom_len = 4;
} else if ((buffer[0] == 0x00) && (buffer[1] == 0x00) &&
} else if ((4 <= num_read) && (buffer[0] == 0x00) && (buffer[1] == 0x00) &&
(buffer[2] == 0xfe) && (buffer[3] == 0xff)) {
byte_order = BO_UTF32_BE;
bom_len = 4;
} else if ((buffer[0] == 0xff) && (buffer[1] == 0xfe)) {
} else if ((2 <= num_read) && (buffer[0] == 0xff) && (buffer[1] == 0xfe)) {
byte_order = BO_UTF16_LE;
bom_len = 2;
} else if ((buffer[0] == 0xfe) && (buffer[1] == 0xff)) {
} else if ((2 <= num_read) && (buffer[0] == 0xfe) && (buffer[1] == 0xff)) {
byte_order = BO_UTF16_BE;
bom_len = 2;
} else {
byte_order = BO_NONE;
bom_len = 0;
}
_in->setFilePointer(bom_len, seek_beginning);