mirror of
https://github.com/CCExtractor/ccextractor.git
synced 2026-04-05 21:51:23 +00:00
395f9b3213
* fix: MKV subtitle track .(null) extension for KATE and unknown codec IDs The matroska_track_text_subtitle_id_extensions array had 7 entries for an 8-value enum, leaving MATROSKA_TRACK_SUBTITLE_CODEC_ID_KATE (index 7) out of bounds. On most platforms this read NULL, which then caused strlen(NULL) UB and snprintf to emit .(null) in the output filename. Two fixes: - Add "kate" at index 7 in the extensions array so KATE tracks produce correct .kate output filenames - Add a NULL guard in generate_filename_from_track() so any future unknown codec ID safely falls back to .bin instead of crashing or producing .(null) Fixes #972 * fix: MKV subtitle track .(null) extension for KATE and unknown codec IDs The matroska_track_text_subtitle_id_extensions array had 7 entries for an 8-value enum, leaving MATROSKA_TRACK_SUBTITLE_CODEC_ID_KATE (index 7) out of bounds. On most platforms this read NULL, which then caused strlen(NULL) UB and snprintf to emit .(null) in the output filename. Two fixes: - Add "kate" at index 7 in the extensions array so KATE tracks produce correct .kate output filenames - Add a NULL guard in generate_filename_from_track() so any future unknown codec ID safely falls back to .bin instead of crashing or producing .(null) Fixes #972 * fix: MKV subtitle track .(null) extension for KATE and unknown codec IDs The matroska_track_text_subtitle_id_extensions array had 7 entries for an 8-value enum, leaving MATROSKA_TRACK_SUBTITLE_CODEC_ID_KATE (index 7) out of bounds. On most platforms this read NULL, which then caused strlen(NULL) UB and snprintf to emit .(null) in the output filename. Two fixes: - Add "kate" at index 7 in the extensions array so KATE tracks produce correct .kate output filenames - Add a NULL guard in generate_filename_from_track() so any future unknown codec ID safely falls back to .bin instead of crashing or producing .(null) Fixes #972 --------- Co-authored-by: Dhanush Varma <your@email.com>
305 lines
10 KiB
C
305 lines
10 KiB
C
#ifndef MATROSKA_H
|
|
#define MATROSKA_H
|
|
|
|
/* Output specificators */
|
|
#if (defined(WIN32) || defined(_WIN32_WCE)) && (defined(__MINGW32__) || !defined(__GNUC__))
|
|
#define LLD_M "I64d"
|
|
#define LLU_M "I64u"
|
|
#define LLX_M "I64x"
|
|
#define LLD "%I64d"
|
|
#define LLU "%I64u"
|
|
#elif defined(__SYMBIAN32__)
|
|
#define LLD_M "d"
|
|
#define LLU_M "u"
|
|
#define LLX_M "x"
|
|
#define LLD "%d"
|
|
#define LLU "%u"
|
|
#elif defined(__DARWIN__) || defined(__APPLE__)
|
|
#define LLD_M "lld"
|
|
#define LLU_M "llu"
|
|
#define LLX_M "llx"
|
|
#define LLD "%lld"
|
|
#define LLU "%llu"
|
|
#elif defined(_LP64) /* Unix 64 bits */
|
|
#define LLD_M "ld"
|
|
#define LLU_M "lu"
|
|
#define LLX_M "lx"
|
|
#define LLD "%ld"
|
|
#define LLU "%lu"
|
|
#else /* Unix 32 bits */
|
|
#define LLD_M "lld"
|
|
#define LLU_M "llu"
|
|
#define LLX_M "llx"
|
|
#define LLD "%lld"
|
|
#define LLU "%llu"
|
|
#endif
|
|
|
|
/* EBML header ids */
|
|
#define MATROSKA_EBML_HEADER 0x1A45DFA3
|
|
#define MATROSKA_EBML_VERSION 0x4286
|
|
#define MATROSKA_EBML_READ_VERSION 0x42F7
|
|
#define MATROSKA_EBML_MAX_ID_LENGTH 0x42F2
|
|
#define MATROSKA_EBML_MAX_SIZE_LENGTH 0x42F3
|
|
#define MATROSKA_EBML_DOC_TYPE 0x4282
|
|
#define MATROSKA_EBML_DOC_TYPE_VERSION 0x4287
|
|
#define MATROSKA_EBML_DOC_TYPE_READ_VERSION 0x4285
|
|
|
|
/* Segment ids */
|
|
#define MATROSKA_SEGMENT_HEADER 0x18538067
|
|
#define MATROSKA_SEGMENT_SEEK_HEAD 0x114D9B74
|
|
#define MATROSKA_SEGMENT_INFO 0x1549A966
|
|
#define MATROSKA_SEGMENT_CLUSTER 0x1F43B675
|
|
#define MATROSKA_SEGMENT_TRACKS 0x1654AE6B
|
|
#define MATROSKA_SEGMENT_CUES 0x1C53BB6B
|
|
#define MATROSKA_SEGMENT_ATTACHMENTS 0x1941A469
|
|
#define MATROSKA_SEGMENT_CHAPTERS 0x1043A770
|
|
#define MATROSKA_SEGMENT_TAGS 0x1254C367
|
|
|
|
/* Segment info ids */
|
|
#define MATROSKA_SEGMENT_INFO_SEGMENT_UID 0x73A4
|
|
#define MATROSKA_SEGMENT_INFO_SEGMENT_FILENAME 0x7384
|
|
#define MATROSKA_SEGMENT_INFO_PREV_UID 0x3CB923
|
|
#define MATROSKA_SEGMENT_INFO_PREV_FILENAME 0x3C83AB
|
|
#define MATROSKA_SEGMENT_INFO_NEXT_UID 0x3EB923
|
|
#define MATROSKA_SEGMENT_INFO_NEXT_FILENAME 0x3E83BB
|
|
#define MATROSKA_SEGMENT_INFO_SEGMENT_FAMILY 0x4444
|
|
#define MATROSKA_SEGMENT_INFO_CHAPTER_TRANSLATE 0x6924
|
|
#define MATROSKA_SEGMENT_INFO_TIMECODE_SCALE 0x2AD7B1
|
|
#define MATROSKA_SEGMENT_INFO_DURATION 0x4489
|
|
#define MATROSKA_SEGMENT_INFO_DATE_UTC 0x4461
|
|
#define MATROSKA_SEGMENT_INFO_TITLE 0x7BA9
|
|
#define MATROSKA_SEGMENT_MUXING_APP 0x4D80
|
|
#define MATROSKA_SEGMENT_WRITING_APP 0x5741
|
|
|
|
/* Segment cluster ids */
|
|
#define MATROSKA_SEGMENT_CLUSTER_TIMECODE 0xE7
|
|
#define MATROSKA_SEGMENT_CLUSTER_SILENT_TRACKS 0x5854
|
|
#define MATROSKA_SEGMENT_CLUSTER_POSITION 0xA7
|
|
#define MATROSKA_SEGMENT_CLUSTER_PREV_SIZE 0xAB
|
|
#define MATROSKA_SEGMENT_CLUSTER_SIMPLE_BLOCK 0xA3
|
|
#define MATROSKA_SEGMENT_CLUSTER_ENCRYPTED_BLOCK 0xAF
|
|
|
|
/* Segment cluster block group ids */
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP 0xA0
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_BLOCK 0xA1
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_BLOCK_VIRTUAL 0xA2
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_BLOCK_ADDITIONS 0x75A1
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_BLOCK_DURATION 0x9B
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_REFERENCE_PRIORITY 0xFA
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_REFERENCE_BLOCK 0xFB
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_CODEC_STATE 0xA4
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_DISCARD_PADDING 0x75A2
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_SLICES 0x8E
|
|
#define MATROSKA_SEGMENT_CLUSTER_BLOCK_GROUP_REFERENCE_FRAME 0xC8
|
|
|
|
/* Segment tracks ids */
|
|
#define MATROSKA_SEGMENT_TRACK_ENTRY 0xAE
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_NUMBER 0xD7
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_UID 0x73C5
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_TYPE 0x83
|
|
#define MATROSKA_SEGMENT_TRACK_FLAG_ENABLED 0xB9
|
|
#define MATROSKA_SEGMENT_TRACK_FLAG_DEFAULT 0x88
|
|
#define MATROSKA_SEGMENT_TRACK_FLAG_FORCED 0x55AA
|
|
#define MATROSKA_SEGMENT_TRACK_FLAG_LACING 0x9C
|
|
#define MATROSKA_SEGMENT_TRACK_MIN_CACHE 0x6DE7
|
|
#define MATROSKA_SEGMENT_TRACK_MAX_CACHE 0x6DF8
|
|
#define MATROSKA_SEGMENT_TRACK_DEFAULT_DURATION 0x23E383
|
|
#define MATROSKA_SEGMENT_TRACK_DEFAULT_DECODED_FIELD_DURATION 0x234E7A
|
|
#define MATROSKA_SEGMENT_TRACK_MAX_BLOCK_ADDITION_ID 0x55EE
|
|
#define MATROSKA_SEGMENT_TRACK_NAME 0x536E
|
|
#define MATROSKA_SEGMENT_TRACK_LANGUAGE 0x22B59C
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_ID 0x86
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_PRIVATE 0x63A2
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_NAME 0x258688
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_ATTACHMENT_LINK 0x7446
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_DECODE_ALL 0xAA
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_OVERLAY 0x6FAB
|
|
#define MATROSKA_SEGMENT_TRACK_CODEC_DELAY 0x56AA
|
|
#define MATROSKA_SEGMENT_TRACK_SEEK_PRE_ROLL 0x56BB
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_TRANSLATE 0x6624
|
|
#define MATROSKA_SEGMENT_TRACK_VIDEO 0xE0
|
|
#define MATROSKA_SEGMENT_TRACK_AUDIO 0xE1
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_OPERATION 0xE2
|
|
#define MATROSKA_SEGMENT_TRACK_CONTENT_ENCODINGS 0x6D80
|
|
|
|
/* Misc ids */
|
|
#define MATROSKA_VOID 0xEC
|
|
#define MATROSKA_CRC32 0xBF
|
|
#define MATROSKA_SEGMENT_TRACK_LANGUAGE_IETF 0x22B59D
|
|
|
|
/* DEFENCE FROM THE FOOL - deprecated IDs */
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_TIMECODE_SCALE 0x23314F
|
|
#define MATROSKA_SEGMENT_TRACK_TRACK_OFFSET 0x537F
|
|
|
|
/* DivX trick track extenstions (in track entry) */
|
|
#define MATROSKA_SEGMENT_TRACK_TRICK_TRACK_UID 0xC0
|
|
#define MATROSKA_SEGMENT_TRACK_TRICK_TRACK_SEGMENT_UID 0xC1
|
|
#define MATROSKA_SEGMENT_TRACK_TRICK_TRACK_FLAG 0xC6
|
|
#define MATROSKA_SEGMENT_TRACK_TRICK_MASTER_TRACK_UID 0xC7
|
|
#define MATROSKA_SEGMENT_TRACK_TRICK_MASTER_TRACK_SEGMENT_UID 0xC4
|
|
|
|
/* Other defines */
|
|
#define MATROSKA_MAX_ID_LENGTH 4
|
|
#define MAX_FILE_NAME_SIZE 260
|
|
|
|
/* Enums */
|
|
enum matroska_track_entry_type
|
|
{
|
|
MATROSKA_TRACK_TYPE_VIDEO = 1,
|
|
MATROSKA_TRACK_TYPE_AUDIO = 2,
|
|
MATROSKA_TRACK_TYPE_COMPLEX = 3,
|
|
MATROSKA_TRACK_TYPE_LOGO = 0x10,
|
|
MATROSKA_TRACK_TYPE_SUBTITLE = 0x11,
|
|
MATROSKA_TRACK_TYPE_BUTTONS = 0x12,
|
|
MATROSKA_TRACK_TYPE_CONTROL = 0x20,
|
|
};
|
|
|
|
enum matroska_track_subtitle_codec_id
|
|
{
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_UTF8 = 0,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_SSA,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_ASS,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_USF,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_WEBVTT,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_BITMAP,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_VOBSUB,
|
|
MATROSKA_TRACK_SUBTITLE_CODEC_ID_KATE
|
|
};
|
|
|
|
char *matroska_track_text_subtitle_id_strings[] = {
|
|
"S_TEXT/UTF8",
|
|
"S_TEXT/SSA",
|
|
"S_TEXT/ASS",
|
|
"S_TEXT/USF",
|
|
"S_TEXT/WEBVTT",
|
|
"S_IMAGE/BMP",
|
|
"S_VOBSUB",
|
|
"S_KATE"};
|
|
|
|
char *matroska_track_text_subtitle_id_extensions[] = {
|
|
"srt", "ssa", "ass",
|
|
"usf", "vtt", "bmp",
|
|
"idx", "kate", NULL // Unknown/sentinel
|
|
};
|
|
|
|
char *avc_codec_id = "V_MPEG4/ISO/AVC";
|
|
char *hevc_codec_id = "V_MPEGH/ISO/HEVC";
|
|
char *mpeg2_codec_id = "V_MPEG2";
|
|
char *dvb_codec_id = "S_DVBSUB";
|
|
|
|
/* Messages */
|
|
#define MATROSKA_INFO "\nMatroska parser info: "
|
|
#define MATROSKA_WARNING "\nMatroska parser warning: "
|
|
#define MATROSKA_ERROR "\nMatroska parser error: "
|
|
|
|
/* Boilerplate code */
|
|
#define MATROSKA_SWITCH_BREAK(a, b) \
|
|
(a) = 0; \
|
|
(b) = 0; \
|
|
break
|
|
|
|
/* Structures */
|
|
|
|
struct block_addition
|
|
{
|
|
char *cue_settings_list;
|
|
ULLONG cue_settings_list_size;
|
|
char *cue_identifier;
|
|
ULLONG cue_identifier_size;
|
|
char *comment;
|
|
ULLONG comment_size;
|
|
};
|
|
|
|
struct matroska_sub_sentence
|
|
{
|
|
char *text;
|
|
ULLONG text_size;
|
|
ULLONG time_start;
|
|
ULLONG time_end;
|
|
struct block_addition *blockaddition;
|
|
};
|
|
|
|
struct matroska_avc_frame
|
|
{
|
|
UBYTE *data;
|
|
ULLONG len;
|
|
ULLONG FTS;
|
|
};
|
|
|
|
struct matroska_sub_track
|
|
{
|
|
char *header; // Style header for ASS/SSA (and other) subtitles
|
|
char *lang;
|
|
char *lang_ietf; // IETF language tag (BCP47)
|
|
ULLONG track_number;
|
|
ULLONG lang_index;
|
|
enum matroska_track_subtitle_codec_id codec_id;
|
|
char *codec_id_string;
|
|
ULLONG last_timestamp;
|
|
|
|
int sentence_count;
|
|
struct matroska_sub_sentence **sentences;
|
|
};
|
|
|
|
struct matroska_ctx
|
|
{
|
|
struct matroska_sub_track **sub_tracks;
|
|
struct lib_ccx_ctx *ctx;
|
|
struct cc_subtitle dec_sub;
|
|
int avc_track_number; // ID of AVC track. -1 if there is none
|
|
int hevc_track_number; // ID of HEVC track. -1 if there is none
|
|
int mpeg2_track_number; // ID of MPEG2 track. -1 if there is none
|
|
int sub_tracks_count;
|
|
int block_index;
|
|
int sentence_count;
|
|
char *filename;
|
|
ULLONG current_second;
|
|
FILE *file;
|
|
};
|
|
|
|
/* Bytestream and parser functions */
|
|
void skip_bytes(FILE *file, ULLONG n);
|
|
void set_bytes(FILE *file, ULLONG n);
|
|
ULLONG get_current_byte(FILE *file);
|
|
UBYTE *read_byte_block(FILE *file, ULLONG n);
|
|
char *read_bytes_signed(FILE *file, ULLONG n);
|
|
UBYTE mkv_read_byte(FILE *file);
|
|
|
|
ULLONG read_vint_length(FILE *file);
|
|
UBYTE *read_vint_block(FILE *file);
|
|
char *read_vint_block_signed(FILE *file);
|
|
ULLONG read_vint_block_int(FILE *file);
|
|
char *read_vint_block_string(FILE *file);
|
|
void read_vint_block_skip(FILE *file);
|
|
|
|
void parse_ebml(FILE *file);
|
|
void parse_segment_info(FILE *file);
|
|
struct matroska_sub_sentence *parse_segment_cluster_block_group_block(struct matroska_ctx *mkv_ctx, ULLONG cluster_timecode);
|
|
void parse_segment_cluster_block_group(struct matroska_ctx *mkv_ctx, ULLONG cluster_timecode);
|
|
void parse_segment_cluster(struct matroska_ctx *mkv_ctx);
|
|
void parse_simple_block(struct matroska_ctx *mkv_ctx, ULLONG frame_timestamp);
|
|
int process_avc_frame_mkv(struct matroska_ctx *mkv_ctx, struct matroska_avc_frame frame);
|
|
int process_hevc_frame_mkv(struct matroska_ctx *mkv_ctx, struct matroska_avc_frame frame);
|
|
int process_mpeg2_frame_mkv(struct matroska_ctx *mkv_ctx, struct matroska_avc_frame frame);
|
|
void parse_segment_track_entry(struct matroska_ctx *mkv_ctx);
|
|
void parse_private_codec_data(struct matroska_ctx *mkv_ctx, char *codec_id_string, ULLONG track_number, char *lang);
|
|
void parse_segment_tracks(struct matroska_ctx *mkv_ctx);
|
|
void parse_segment(struct matroska_ctx *mkv_ctx);
|
|
|
|
/* Writing and helper functions */
|
|
char *generate_timestamp_utf8(ULLONG milliseconds);
|
|
char *generate_timestamp_ass_ssa(ULLONG milliseconds);
|
|
int find_sub_track_index(struct matroska_ctx *mkv_ctx, ULLONG track_number);
|
|
char *get_track_entry_type_description(enum matroska_track_entry_type type);
|
|
enum matroska_track_subtitle_codec_id get_track_subtitle_codec_id(char *codec_id);
|
|
char *generate_filename_from_track(struct matroska_ctx *mkv_ctx, struct matroska_sub_track *track);
|
|
char *ass_ssa_sentence_erase_read_order(char *text);
|
|
void save_sub_track(struct matroska_ctx *mkv_ctx, struct matroska_sub_track *track);
|
|
void free_sub_track(struct matroska_sub_track *track);
|
|
void matroska_save_all(struct matroska_ctx *mkv_ctx, char *lang);
|
|
void matroska_free_all(struct matroska_ctx *mkv_ctx);
|
|
void matroska_parse(struct matroska_ctx *mkv_ctx);
|
|
FILE *create_file(struct lib_ccx_ctx *ctx);
|
|
|
|
#endif // MATROSKA_H
|