mirror of
https://github.com/nyanmisaka/ffmpeg-rockchip.git
synced 2025-11-01 04:53:04 +08:00
b6305a7873
Prevents out-of-bound writes when parsing tile row heights. Fixes Coverity issue #1538300. Reviewed-by: James Almer <jamrial@gmail.com> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
225 lines
10 KiB
C
225 lines
10 KiB
C
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
/**
|
|
* @file
|
|
* EVC decoder/parser shared code
|
|
*/
|
|
|
|
#ifndef AVCODEC_EVC_PS_H
|
|
#define AVCODEC_EVC_PS_H
|
|
|
|
#include <stdint.h>
|
|
|
|
#include "evc.h"
|
|
#include "get_bits.h"
|
|
|
|
#define EVC_MAX_QP_TABLE_SIZE 58
|
|
#define NUM_CPB 32
|
|
|
|
// rpl structure
|
|
typedef struct RefPicListStruct {
|
|
uint32_t ref_pic_num;
|
|
uint32_t ref_pics[EVC_MAX_NUM_REF_PICS];
|
|
} RefPicListStruct;
|
|
|
|
// chromaQP table structure to be signalled in SPS
|
|
typedef struct ChromaQpTable {
|
|
uint8_t chroma_qp_table_present_flag; // u(1)
|
|
uint8_t same_qp_table_for_chroma; // u(1)
|
|
uint8_t global_offset_flag; // u(1)
|
|
uint8_t num_points_in_qp_table_minus1[2]; // ue(v)
|
|
uint8_t delta_qp_in_val_minus1[2][EVC_MAX_QP_TABLE_SIZE]; // u(6)
|
|
int delta_qp_out_val[2][EVC_MAX_QP_TABLE_SIZE]; // se(v)
|
|
} ChromaQpTable;
|
|
|
|
// Hypothetical Reference Decoder (HRD) parameters, part of VUI
|
|
typedef struct HRDParameters {
|
|
uint8_t cpb_cnt_minus1; // ue(v)
|
|
uint8_t bit_rate_scale; // u(4)
|
|
uint8_t cpb_size_scale; // u(4)
|
|
uint32_t bit_rate_value_minus1[NUM_CPB]; // ue(v)
|
|
uint32_t cpb_size_value_minus1[NUM_CPB]; // ue(v)
|
|
uint8_t cbr_flag[NUM_CPB]; // u(1)
|
|
uint8_t initial_cpb_removal_delay_length_minus1; // u(5)
|
|
uint8_t cpb_removal_delay_length_minus1; // u(5)
|
|
uint8_t dpb_output_delay_length_minus1; // u(5)
|
|
uint8_t time_offset_length; // u(5)
|
|
} HRDParameters;
|
|
|
|
// video usability information (VUI) part of SPS
|
|
typedef struct VUIParameters {
|
|
uint8_t aspect_ratio_info_present_flag; // u(1)
|
|
uint8_t aspect_ratio_idc; // u(8)
|
|
uint16_t sar_width; // u(16)
|
|
uint16_t sar_height; // u(16)
|
|
uint8_t overscan_info_present_flag; // u(1)
|
|
uint8_t overscan_appropriate_flag; // u(1)
|
|
uint8_t video_signal_type_present_flag; // u(1)
|
|
uint8_t video_format; // u(3)
|
|
uint8_t video_full_range_flag; // u(1)
|
|
uint8_t colour_description_present_flag; // u(1)
|
|
uint8_t colour_primaries; // u(8)
|
|
uint8_t transfer_characteristics; // u(8)
|
|
uint8_t matrix_coefficients; // u(8)
|
|
uint8_t chroma_loc_info_present_flag; // u(1)
|
|
uint8_t chroma_sample_loc_type_top_field; // ue(v)
|
|
uint8_t chroma_sample_loc_type_bottom_field; // ue(v)
|
|
uint8_t neutral_chroma_indication_flag; // u(1)
|
|
uint8_t field_seq_flag; // u(1)
|
|
uint8_t timing_info_present_flag; // u(1)
|
|
uint32_t num_units_in_tick; // u(32)
|
|
uint32_t time_scale; // u(32)
|
|
uint8_t fixed_pic_rate_flag; // u(1)
|
|
uint8_t nal_hrd_parameters_present_flag; // u(1)
|
|
uint8_t vcl_hrd_parameters_present_flag; // u(1)
|
|
uint8_t low_delay_hrd_flag; // u(1)
|
|
uint8_t pic_struct_present_flag; // u(1)
|
|
uint8_t bitstream_restriction_flag; // u(1)
|
|
uint8_t motion_vectors_over_pic_boundaries_flag; // u(1)
|
|
uint8_t max_bytes_per_pic_denom; // ue(v)
|
|
uint8_t max_bits_per_mb_denom; // ue(v)
|
|
uint8_t log2_max_mv_length_horizontal; // ue(v)
|
|
uint8_t log2_max_mv_length_vertical; // ue(v)
|
|
uint32_t num_reorder_pics; // ue(v)
|
|
uint32_t max_dec_pic_buffering; // ue(v)
|
|
|
|
HRDParameters hrd_parameters;
|
|
} VUIParameters;
|
|
|
|
// The sturcture reflects SPS RBSP(raw byte sequence payload) layout
|
|
// @see ISO_IEC_23094-1 section 7.3.2.1
|
|
//
|
|
// The following descriptors specify the parsing process of each element
|
|
// u(n) - unsigned integer using n bits
|
|
// ue(v) - unsigned integer 0-th order Exp_Golomb-coded syntax element with the left bit first
|
|
typedef struct EVCParserSPS {
|
|
uint8_t sps_seq_parameter_set_id; // ue(v)
|
|
uint8_t profile_idc; // u(8)
|
|
uint8_t level_idc; // u(8)
|
|
uint32_t toolset_idc_h; // u(32)
|
|
uint32_t toolset_idc_l; // u(32)
|
|
uint8_t chroma_format_idc; // ue(v)
|
|
uint32_t pic_width_in_luma_samples; // ue(v)
|
|
uint32_t pic_height_in_luma_samples; // ue(v)
|
|
uint8_t bit_depth_luma_minus8; // ue(v)
|
|
uint8_t bit_depth_chroma_minus8; // ue(v)
|
|
|
|
uint8_t sps_btt_flag; // u(1)
|
|
uint32_t log2_ctu_size_minus2; // ue(v)
|
|
uint32_t log2_min_cb_size_minus2; // ue(v)
|
|
uint32_t log2_diff_ctu_max_14_cb_size; // ue(v)
|
|
uint32_t log2_diff_ctu_max_tt_cb_size; // ue(v)
|
|
uint32_t log2_diff_min_cb_min_tt_cb_size_minus2; // ue(v)
|
|
|
|
uint8_t sps_suco_flag; // u(1)
|
|
uint32_t log2_diff_ctu_size_max_suco_cb_size; // ue(v)
|
|
uint32_t log2_diff_max_suco_min_suco_cb_size; // ue(v)
|
|
|
|
uint8_t sps_admvp_flag; // u(1)
|
|
uint8_t sps_affine_flag; // u(1)
|
|
uint8_t sps_amvr_flag; // u(1)
|
|
uint8_t sps_dmvr_flag; // u(1)
|
|
uint8_t sps_mmvd_flag; // u(1)
|
|
uint8_t sps_hmvp_flag; // u(1)
|
|
|
|
uint8_t sps_eipd_flag; // u(1)
|
|
uint8_t sps_ibc_flag; // u(1)
|
|
uint32_t log2_max_ibc_cand_size_minus2; // ue(v)
|
|
|
|
uint8_t sps_cm_init_flag; // u(1)
|
|
uint8_t sps_adcc_flag; // u(1)
|
|
|
|
uint8_t sps_iqt_flag; // u(1)
|
|
uint8_t sps_ats_flag; // u(1)
|
|
|
|
uint8_t sps_addb_flag; // u(1)
|
|
uint8_t sps_alf_flag; // u(1)
|
|
uint8_t sps_htdf_flag; // u(1)
|
|
uint8_t sps_rpl_flag; // u(1)
|
|
uint8_t sps_pocs_flag; // u(1)
|
|
uint8_t sps_dquant_flag; // u(1)
|
|
uint8_t sps_dra_flag; // u(1)
|
|
|
|
uint32_t log2_max_pic_order_cnt_lsb_minus4; // ue(v)
|
|
uint32_t log2_sub_gop_length; // ue(v)
|
|
uint32_t log2_ref_pic_gap_length; // ue(v)
|
|
|
|
uint8_t max_num_tid0_ref_pics; // ue(v)
|
|
|
|
uint32_t sps_max_dec_pic_buffering_minus1; // ue(v)
|
|
uint8_t long_term_ref_pic_flag; // u(1)
|
|
uint8_t rpl1_same_as_rpl0_flag; // u(1)
|
|
uint8_t num_ref_pic_list_in_sps[2]; // ue(v)
|
|
struct RefPicListStruct rpls[2][EVC_MAX_NUM_RPLS];
|
|
|
|
uint8_t picture_cropping_flag; // u(1)
|
|
uint32_t picture_crop_left_offset; // ue(v)
|
|
uint32_t picture_crop_right_offset; // ue(v)
|
|
uint32_t picture_crop_top_offset; // ue(v)
|
|
uint32_t picture_crop_bottom_offset; // ue(v)
|
|
|
|
struct ChromaQpTable chroma_qp_table_struct;
|
|
|
|
uint8_t vui_parameters_present_flag; // u(1)
|
|
|
|
struct VUIParameters vui_parameters;
|
|
|
|
} EVCParserSPS;
|
|
|
|
typedef struct EVCParserPPS {
|
|
uint8_t pps_pic_parameter_set_id; // ue(v)
|
|
uint8_t pps_seq_parameter_set_id; // ue(v)
|
|
uint8_t num_ref_idx_default_active_minus1[2]; // ue(v)
|
|
uint8_t additional_lt_poc_lsb_len; // ue(v)
|
|
uint8_t rpl1_idx_present_flag; // u(1)
|
|
uint8_t single_tile_in_pic_flag; // u(1)
|
|
uint32_t num_tile_columns_minus1; // ue(v)
|
|
uint32_t num_tile_rows_minus1; // ue(v)
|
|
uint8_t uniform_tile_spacing_flag; // u(1)
|
|
uint32_t tile_column_width_minus1[EVC_MAX_TILE_COLUMNS]; // ue(v)
|
|
uint32_t tile_row_height_minus1[EVC_MAX_TILE_ROWS]; // ue(v)
|
|
uint8_t loop_filter_across_tiles_enabled_flag; // u(1)
|
|
uint32_t tile_offset_len_minus1; // ue(v)
|
|
uint8_t tile_id_len_minus1; // ue(v)
|
|
uint8_t explicit_tile_id_flag; // u(1)
|
|
uint32_t tile_id_val[EVC_MAX_TILE_ROWS][EVC_MAX_TILE_COLUMNS]; // u(v)
|
|
uint8_t pic_dra_enabled_flag; // u(1)
|
|
uint8_t pic_dra_aps_id; // u(5)
|
|
uint8_t arbitrary_slice_present_flag; // u(1)
|
|
uint8_t constrained_intra_pred_flag; // u(1)
|
|
uint8_t cu_qp_delta_enabled_flag; // u(1)
|
|
uint32_t log2_cu_qp_delta_area_minus6; // ue(v)
|
|
|
|
} EVCParserPPS;
|
|
|
|
typedef struct EVCParamSets {
|
|
EVCParserSPS *sps[EVC_MAX_SPS_COUNT];
|
|
EVCParserPPS *pps[EVC_MAX_PPS_COUNT];
|
|
} EVCParamSets;
|
|
|
|
// @see ISO_IEC_23094-1 (7.3.2.1 SPS RBSP syntax)
|
|
int ff_evc_parse_sps(GetBitContext *gb, EVCParamSets *ps);
|
|
|
|
// @see ISO_IEC_23094-1 (7.3.2.2 SPS RBSP syntax)
|
|
int ff_evc_parse_pps(GetBitContext *gb, EVCParamSets *ps);
|
|
|
|
void ff_evc_ps_free(EVCParamSets *ps);
|
|
|
|
#endif /* AVCODEC_EVC_PS_H */
|