winamp/Src/nsmkv/main.cpp
2024-09-24 14:54:57 +02:00

227 lines
4.8 KiB
C++

#include "vint.h"
#include "header.h"
#include "ebml_float.h"
#include "segment.h"
#include "ebml_unsigned.h"
#include "ebml_signed.h"
#include <stdio.h>
#include <assert.h>
#include "SeekTable.h"
#include <time.h>
#include "parser.h"
#include "SegmentInfo.h"
#include "global_elements.h"
#include "Tracks.h"
#include "Cluster.h"
#include "Cues.h"
#include "Chapters.h"
#include "Tags.h"
#include "read.h"
#include "Attachments.h"
using namespace nsmkv;
static nsmkv::SeekTable seekTable;
static nsmkv::Header header;
static SegmentInfo segment_info;
static Tracks tracks;
static Cues cues;
static Attachments attachments;
static Tags tags;
uint64_t max_id_length = header.ebml_max_id_length;
uint64_t max_size_length = header.ebml_max_size_length;
uint32_t num_seekhead_elements_found = 0;
uint32_t num_seek_elements_found = 0;
bool ebml_segment_found = false;
uint64_t segment_data_offset = 0;
// returns bytes read. 0 means EOF
uint64_t read_vsint(FILE *f, int64_t *val)
{
uint8_t data[9];
size_t bytes_read = fread(data, 1, 1, f);
if (bytes_read != 1)
return 0;
uint8_t length = vint_get_number_bytes(data[0]);
bytes_read = fread(data+1, 1, length, f);
if (bytes_read != length)
return 0;
*val = vsint_read_ptr_len(length+1, data);
return bytes_read+1;
}
char *read_utf8(FILE *f, size_t size)
{
char *doctype = (char *)malloc(size + 1);
if (doctype)
{
doctype[size]=0;
if (fread(doctype, 1, size, f) == size)
return doctype;
}
free(doctype);
return 0;
}
// returns bytes read. 0 means EOF
uint64_t ReadSegment(FILE *f, uint64_t size)
{
uint64_t total_bytes_read=0;
// store the segment element data offset for later use
segment_data_offset = ftell(f);
#ifdef WA_VALIDATE
printf("[%I64u] Segment element data offset\n", segment_data_offset);
#endif
while (size)
{
uint64_t this_position = ftell(f);
#ifdef WA_VALIDATE
printf("[%I64u] ", this_position);
#endif
ebml_node node;
uint64_t bytes_read = read_ebml_node((nsmkv::MKVReader*)f, &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_attachments:
{
printf(" Attachments\n");
ReadAttachment((nsmkv::MKVReader*)f, node.size, attachments);
}
break;
case mkv_metaseek_seekhead:
{
printf(" SeekHead\n");
ReadSeekHead((nsmkv::MKVReader*)f, node.size, seekTable);
}
break;
case mkv_segment_segmentinfo:
{
printf(" SegmentInfo\n");
ReadSegmentInfo((nsmkv::MKVReader*)f, node.size, segment_info);
}
break;
case mkv_segment_tracks:
{
printf(" Tracks\n");
ReadTracks((nsmkv::MKVReader*)f, node.size, tracks);
}
break;
case mkv_segment_cues:
{
printf(" Cues\n");
ReadCues((nsmkv::MKVReader*)f, node.size, cues);
}
break;
case mkv_segment_cluster:
{
printf(" Clusters\n");
nsmkv::Cluster cluster;
ReadCluster((nsmkv::MKVReader*)f, node.size, cluster);
}
break;
case mkv_segment_chapters:
{
printf(" Chapters\n");
}
break;
case mkv_segment_tags:
{
printf(" Tags\n");
nsmkv::ReadTags(f, node.size, tags);
}
break;
default:
ReadGlobal((nsmkv::MKVReader*)f, node.id, node.size);
}
}
return total_bytes_read;
}
int main()
{
// char *file_in = "\\\\o2d2\\ftp\\usr\\nullsoft\\test media\\mkv\\Ratatouille.2007.nHD.720.x264.NhaNc3.mkv";
//char *file_in = "\\\\o2d2\\ftp\\usr\\nullsoft\\test media\\mkv\\cham_mp4v_aac.mkv";
char *file_in = "c:/users/benski/desktop/metadata.mkv";
FILE *f = fopen(file_in, "rb");
if (f == NULL)
{
printf("****Error attempting to open file: %s\n",file_in);
return -1;
}
else
{
printf("Starting Processing of File: %s\n",file_in);
}
ebml_node node;
while (read_ebml_node((nsmkv::MKVReader*)f, &node))
{
switch(node.id)
{
case mkv_header:
if (header.ebml_header_found == false)
{
#ifdef WA_VALIDATE
printf("MKV header found, processing...\n");
#endif
header.ebml_header_found = true;
nsmkv::ReadHeader((nsmkv::MKVReader*)f, node.size, header);
}
else
{
#ifdef WA_VALIDATE
printf("Extra MKV header found, ignoring...\n");
#endif
nsmkv::Header extraHeader;
nsmkv::ReadHeader((nsmkv::MKVReader*)f, node.size, extraHeader);
}
break;
case mkv_segment:
printf("MKV Segment element found, processing\n");
#ifdef WA_VALIDATE
ebml_segment_found = true;
#endif
ReadSegment(f, node.size);
break;
default:
ReadGlobal((nsmkv::MKVReader*)f, node.id, node.size);
}
}
// seekTable.Dump();
fclose(f);
printf("Number of SeekHead elements found: %I32u\n",num_seekhead_elements_found);
printf("Number of Seek elements found: %I32u\n",num_seek_elements_found);
}