1 files changed, 115 insertions, 0 deletions
diff --git a/Src/Plugins/Input/in_mkv/MKVDuration.cpp b/Src/Plugins/Input/in_mkv/MKVDuration.cpp
new file mode 100644
index 00000000..0f84cb62
--- /dev/null
+++ b/Src/Plugins/Input/in_mkv/MKVDuration.cpp
@@ -0,0 +1,115 @@
+#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;
+}
+\ No newline at end of file