115 lines
2.1 KiB
C++
115 lines
2.1 KiB
C++
#include "MKVDuration.h"
|
|
#include "../nsmkv/global_elements.h"
|
|
#include "../nsmkv/read.h"
|
|
#include "../nsmkv/segment.h"
|
|
#include "../nsmkv/file_mkv_reader.h"
|
|
|
|
MKVDuration::MKVDuration()
|
|
{
|
|
segment_info_found = false;
|
|
content_length = 0;
|
|
}
|
|
|
|
bool MKVDuration::Open(const wchar_t *filename)
|
|
{
|
|
FILE *f = _wfopen(filename, L"rb");
|
|
if (!f)
|
|
return false;
|
|
|
|
MKVReaderFILE reader(f);
|
|
|
|
content_length = reader.GetContentLength();
|
|
|
|
ebml_node node;
|
|
while (!segment_info_found)
|
|
{
|
|
if (read_ebml_node(&reader, &node) == 0)
|
|
break;
|
|
|
|
switch(node.id)
|
|
{
|
|
case mkv_header:
|
|
if (nsmkv::ReadHeader(&reader, node.size, header) == 0)
|
|
{
|
|
return false;
|
|
}
|
|
break;
|
|
case mkv_segment:
|
|
if (ReadSegment(&reader, node.size) == 0)
|
|
{
|
|
return false;
|
|
}
|
|
break;
|
|
default:
|
|
nsmkv::SkipNode(&reader, node.id, node.size);
|
|
|
|
}
|
|
}
|
|
|
|
return segment_info_found;
|
|
}
|
|
|
|
uint64_t MKVDuration::ReadSegment(nsmkv::MKVReader *reader, uint64_t size)
|
|
{
|
|
uint64_t total_bytes_read=0;
|
|
while (size && !segment_info_found)
|
|
{
|
|
ebml_node node;
|
|
uint64_t bytes_read = read_ebml_node(reader, &node);
|
|
|
|
if (bytes_read == 0)
|
|
return 0;
|
|
|
|
// benski> checking bytes_read and node.size separately prevents possible integer overflow attack
|
|
if (bytes_read > size)
|
|
return 0;
|
|
total_bytes_read+=bytes_read;
|
|
size-=bytes_read;
|
|
|
|
if (node.size > size)
|
|
return 0;
|
|
total_bytes_read+=node.size;
|
|
size-=node.size;
|
|
|
|
switch(node.id)
|
|
{
|
|
case mkv_segment_segmentinfo:
|
|
{
|
|
printf("SegmentInfo\n");
|
|
if (ReadSegmentInfo(reader, node.size, segment_info) == 0)
|
|
return 0;
|
|
segment_info_found=true;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
if (nsmkv::SkipNode(reader, node.id, node.size) == 0)
|
|
return 0;
|
|
}
|
|
}
|
|
return total_bytes_read;
|
|
}
|
|
|
|
int MKVDuration::GetLengthMilliseconds()
|
|
{
|
|
if (!segment_info_found)
|
|
return -1000;
|
|
else
|
|
return segment_info.GetDurationMilliseconds();
|
|
}
|
|
|
|
const char *MKVDuration::GetTitle()
|
|
{
|
|
return segment_info.title;
|
|
}
|
|
|
|
int MKVDuration::GetBitrate()
|
|
{
|
|
if (segment_info_found)
|
|
{
|
|
int time_ms = segment_info.GetDurationMilliseconds();
|
|
if (time_ms)
|
|
return (int) (8ULL * content_length / (uint64_t)time_ms);
|
|
}
|
|
return 0;
|
|
} |