mirror of
https://github.com/RGBCube/serenity
synced 2025-05-31 06:48:12 +00:00
2dfd236dcd
Matroska::Reader functions now return DecoderErrorOr instead of values being declared Optional. Useful errors can be handled by the users of the parser, similarly to the VP9 decoder. A lot of the error checking in the reader is a lot cleaner thanks to this change, since all reads can be range checked in Streamer::read_octet() now. Most functions for the Streamer class are now also out-of-line in Reader.cpp now instead of residing in the header.
102 lines
3 KiB
C++
102 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2021, Hunter Salyer <thefalsehonesty@gmail.com>
|
|
* Copyright (c) 2022, Gregory Bertilson <Zaggy1024@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/Debug.h>
|
|
#include <AK/IntegralMath.h>
|
|
#include <AK/NonnullOwnPtrVector.h>
|
|
#include <AK/Optional.h>
|
|
#include <AK/OwnPtr.h>
|
|
#include <LibVideo/DecoderError.h>
|
|
|
|
#include "Document.h"
|
|
|
|
namespace Video::Matroska {
|
|
|
|
class Reader {
|
|
public:
|
|
Reader(u8 const* data, size_t size)
|
|
: m_streamer(data, size)
|
|
{
|
|
}
|
|
|
|
static DecoderErrorOr<NonnullOwnPtr<MatroskaDocument>> parse_matroska_from_file(StringView path);
|
|
static DecoderErrorOr<NonnullOwnPtr<MatroskaDocument>> parse_matroska_from_data(u8 const*, size_t);
|
|
|
|
DecoderErrorOr<NonnullOwnPtr<MatroskaDocument>> parse();
|
|
|
|
private:
|
|
class Streamer {
|
|
public:
|
|
Streamer(u8 const* data, size_t size)
|
|
: m_data_ptr(data)
|
|
, m_size_remaining(size)
|
|
{
|
|
}
|
|
|
|
u8 const* data() { return m_data_ptr; }
|
|
|
|
char const* data_as_chars() { return reinterpret_cast<char const*>(m_data_ptr); }
|
|
|
|
size_t octets_read() { return m_octets_read.last(); }
|
|
|
|
void push_octets_read() { m_octets_read.append(0); }
|
|
|
|
void pop_octets_read()
|
|
{
|
|
auto popped = m_octets_read.take_last();
|
|
if (!m_octets_read.is_empty())
|
|
m_octets_read.last() += popped;
|
|
}
|
|
|
|
ErrorOr<u8> read_octet();
|
|
|
|
ErrorOr<i16> read_i16();
|
|
|
|
ErrorOr<u64> read_variable_size_integer(bool mask_length = true);
|
|
ErrorOr<i64> read_variable_size_signed_integer();
|
|
|
|
ErrorOr<u64> read_u64();
|
|
ErrorOr<double> read_float();
|
|
|
|
ErrorOr<String> read_string();
|
|
|
|
ErrorOr<void> read_unknown_element();
|
|
|
|
ErrorOr<void> drop_octets(size_t num_octets);
|
|
|
|
bool at_end() const { return m_size_remaining == 0; }
|
|
bool has_octet() const { return m_size_remaining >= 1; }
|
|
|
|
size_t remaining() const { return m_size_remaining; }
|
|
void set_remaining(size_t remaining) { m_size_remaining = remaining; }
|
|
|
|
private:
|
|
u8 const* m_data_ptr { nullptr };
|
|
size_t m_size_remaining { 0 };
|
|
Vector<size_t> m_octets_read { 0 };
|
|
};
|
|
|
|
DecoderErrorOr<void> parse_master_element(StringView element_name, Function<DecoderErrorOr<void>(u64 element_id)> element_consumer);
|
|
DecoderErrorOr<EBMLHeader> parse_ebml_header();
|
|
|
|
DecoderErrorOr<void> parse_segment_elements(MatroskaDocument&);
|
|
DecoderErrorOr<NonnullOwnPtr<SegmentInformation>> parse_information();
|
|
|
|
DecoderErrorOr<void> parse_tracks(MatroskaDocument&);
|
|
DecoderErrorOr<NonnullOwnPtr<TrackEntry>> parse_track_entry();
|
|
DecoderErrorOr<TrackEntry::VideoTrack> parse_video_track_information();
|
|
DecoderErrorOr<TrackEntry::ColorFormat> parse_video_color_information();
|
|
DecoderErrorOr<TrackEntry::AudioTrack> parse_audio_track_information();
|
|
DecoderErrorOr<NonnullOwnPtr<Cluster>> parse_cluster();
|
|
DecoderErrorOr<NonnullOwnPtr<Block>> parse_simple_block();
|
|
|
|
Streamer m_streamer;
|
|
};
|
|
|
|
}
|