From patchwork Thu Dec 26 10:53:41 2019 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Andreas Rheinhardt X-Patchwork-Id: 16974 Return-Path: X-Original-To: patchwork@ffaux-bg.ffmpeg.org Delivered-To: patchwork@ffaux-bg.ffmpeg.org Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by ffaux.localdomain (Postfix) with ESMTP id 17446449EFF for ; Thu, 26 Dec 2019 12:54:29 +0200 (EET) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id F3A8E68AABF; Thu, 26 Dec 2019 12:54:28 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wm1-f68.google.com (mail-wm1-f68.google.com [209.85.128.68]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 6102468AC2D for ; Thu, 26 Dec 2019 12:54:27 +0200 (EET) Received: by mail-wm1-f68.google.com with SMTP id f129so5817049wmf.2 for ; Thu, 26 Dec 2019 02:54:27 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:cc:subject:date:message-id:in-reply-to:references :mime-version:content-transfer-encoding; bh=g74Eq53+nInC3oOb81XkIIYzsLhcCLmEkUsce0AgEOc=; b=KwVDZ7oi4ZhEYfnig+BeR3JQN6jjzqCWcODpXWuy9S533ILSBpbTZiuBMcGvb9IgBg YDunGj9z0jDt50KzSkqmrcK54RQJf32TPa4AdpX8OcYSNat/2Eeonz/NE9MSEYiMnzZr RjsmkmCZEfaF95MSYGkcM3teXJqpmQL7qUqnQzXBOMkUwIb5vZGjbE2FfyDv5uCIm93X c3ZwaHe0NaZyntgY/fAE2M3JtPrVB4X6llNnaucH2gunZKXLeCZkM+NHalvEhUGlwWMs z8k6tM+NxOmN61U8sVyx+oWip0dqM00dIBE12Q9AryEcVEE470EZ18a6bObkUOSzvWXo Rvrg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:cc:subject:date:message-id:in-reply-to :references:mime-version:content-transfer-encoding; bh=g74Eq53+nInC3oOb81XkIIYzsLhcCLmEkUsce0AgEOc=; b=T2Zf2kL88zMWb4sq+KEKSYArS9HM7ntr14hHwUa6REQK91iFWcqtlKVNYHMZzLoyQI J4Renr4oIFOWrvBCsbBzePU1BtjAQp5k7LchiDjaKZoo0ChKW3zt7vk/jZaXuiWwAkxZ 01cxKfsbGvpEVFAzC0NzsL5UCN8XIl/aApGxnLQConqG/N70JrRgIn0b75u/OMIY1U0b ZnZ3w/nm0amUb9JpzDCi+T6jKM+um9EpyaMYB2elvtG82cybqTRC6hl9ZleCNf1nOSD6 1sksE6FqiUlQDAkepApxsj60/J9wj3UoZlEOlC6yiTWl00ZOvEQRQb3BOMm0cs3B294w vckQ== X-Gm-Message-State: APjAAAUpOL81u5ktDAf3kK5afEMhRazCF3SeuS/COvf4pEsW2LWdO5GS xpnEf2+/burekfkoodUlUrwfchLd X-Google-Smtp-Source: APXvYqw3VMcCcfpntLkenterNJ3m6G702I3n9LnMuKbryZ+Qj59AOSKeTgMaXzh99sSHmpuOsOejvA== X-Received: by 2002:a1c:cc06:: with SMTP id h6mr12764067wmb.118.1577357666163; Thu, 26 Dec 2019 02:54:26 -0800 (PST) Received: from sblaptop.fritz.box (ipbcc10203.dynamic.kabel-deutschland.de. [188.193.2.3]) by smtp.gmail.com with ESMTPSA id c68sm7997618wme.13.2019.12.26.02.54.24 (version=TLS1_3 cipher=TLS_AES_256_GCM_SHA384 bits=256/256); Thu, 26 Dec 2019 02:54:25 -0800 (PST) From: Andreas Rheinhardt To: ffmpeg-devel@ffmpeg.org Date: Thu, 26 Dec 2019 11:53:41 +0100 Message-Id: <20191226105342.11175-16-andreas.rheinhardt@gmail.com> X-Mailer: git-send-email 2.20.1 In-Reply-To: <20191226105342.11175-1-andreas.rheinhardt@gmail.com> References: <20191226105342.11175-1-andreas.rheinhardt@gmail.com> MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH 16/17] avformat/matroskadec: Use av_fast_realloc_array() X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Andreas Rheinhardt Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" instead of av_fast_realloc() for allocating an array. It has the advantage of doing it's own overflow checks and does not overallocate unnecessarily: It allocates exactly one element if one element is desired. This is advantageous for CueTrackPositions: While it is allowed (and supported) to have more than one of them in a CuePoint, there is typically only one of them, so that overallocating is a waste. The memory used for a file with 540000 cues entries therefore decreased from 69624 KB to 52592 KB. Signed-off-by: Andreas Rheinhardt --- libavformat/matroskadec.c | 56 +++++++++++++++++++-------------------- 1 file changed, 27 insertions(+), 29 deletions(-) diff --git a/libavformat/matroskadec.c b/libavformat/matroskadec.c index 72624dc3f1..fe4a444410 100644 --- a/libavformat/matroskadec.c +++ b/libavformat/matroskadec.c @@ -109,8 +109,8 @@ typedef const struct EbmlSyntax { } EbmlSyntax; typedef struct EbmlList { - int nb_elem; - unsigned int alloc_elem_size; + unsigned nb_elem; + unsigned nb_allocated; void *elem; } EbmlList; @@ -1226,13 +1226,10 @@ static int ebml_parse(MatroskaDemuxContext *matroska, data = (char *) data + syntax->data_offset; if (syntax->list_elem_size) { EbmlList *list = data; - void *newelem; - - if ((unsigned)list->nb_elem + 1 >= UINT_MAX / syntax->list_elem_size) - return AVERROR(ENOMEM); - newelem = av_fast_realloc(list->elem, - &list->alloc_elem_size, - (list->nb_elem + 1) * syntax->list_elem_size); + void *newelem = av_fast_realloc_array(list->elem, + &list->nb_allocated, + list->nb_elem + 1, + syntax->list_elem_size); if (!newelem) return AVERROR(ENOMEM); list->elem = newelem; @@ -1464,7 +1461,7 @@ level_check: static void ebml_free(EbmlSyntax *syntax, void *data) { - int i, j; + int i; for (i = 0; syntax[i].id; i++) { void *data_off = (char *) data + syntax[i].data_offset; switch (syntax[i].type) { @@ -1480,12 +1477,12 @@ static void ebml_free(EbmlSyntax *syntax, void *data) if (syntax[i].list_elem_size) { EbmlList *list = data_off; char *ptr = list->elem; - for (j = 0; j < list->nb_elem; + for (unsigned j = 0; j < list->nb_elem; j++, ptr += syntax[i].list_elem_size) ebml_free(syntax[i].def.n, ptr); av_freep(&list->elem); list->nb_elem = 0; - list->alloc_elem_size = 0; + list->nb_allocated = 0; } else ebml_free(syntax[i].def.n, data_off); default: @@ -1548,7 +1545,7 @@ static MatroskaTrack *matroska_find_track_by_num(MatroskaDemuxContext *matroska, int num) { MatroskaTrack *tracks = matroska->tracks.elem; - int i; + unsigned i; for (i = 0; i < matroska->tracks.nb_elem; i++) if (tracks[i].num == num) @@ -1702,7 +1699,7 @@ static void matroska_convert_tag(AVFormatContext *s, EbmlList *list, { MatroskaTag *tags = list->elem; char key[1024]; - int i; + unsigned i; for (i = 0; i < list->nb_elem; i++) { const char *lang = tags[i].lang && @@ -1736,7 +1733,7 @@ static void matroska_convert_tags(AVFormatContext *s) { MatroskaDemuxContext *matroska = s->priv_data; MatroskaTags *tags = matroska->tags.elem; - int i, j; + unsigned i, j; for (i = 0; i < matroska->tags.nb_elem; i++) { if (tags[i].target.attachuid) { @@ -1752,7 +1749,7 @@ static void matroska_convert_tags(AVFormatContext *s) } if (!found) { av_log(NULL, AV_LOG_WARNING, - "The tags at index %d refer to a " + "The tags at index %u refer to a " "non-existent attachment %"PRId64".\n", i, tags[i].target.attachuid); } @@ -1769,7 +1766,7 @@ static void matroska_convert_tags(AVFormatContext *s) } if (!found) { av_log(NULL, AV_LOG_WARNING, - "The tags at index %d refer to a non-existent chapter " + "The tags at index %u refer to a non-existent chapter " "%"PRId64".\n", i, tags[i].target.chapteruid); } @@ -1786,7 +1783,7 @@ static void matroska_convert_tags(AVFormatContext *s) } if (!found) { av_log(NULL, AV_LOG_WARNING, - "The tags at index %d refer to a non-existent track " + "The tags at index %u refer to a non-existent track " "%"PRId64".\n", i, tags[i].target.trackuid); } @@ -1835,7 +1832,7 @@ static int matroska_parse_seekhead_entry(MatroskaDemuxContext *matroska, static void matroska_execute_seekhead(MatroskaDemuxContext *matroska) { EbmlList *seekhead_list = &matroska->seekhead; - int i; + unsigned i; // we should not do any seeking in the streaming case if (!(matroska->ctx->pb->seekable & AVIO_SEEKABLE_NORMAL)) @@ -1871,7 +1868,7 @@ static void matroska_add_index_entries(MatroskaDemuxContext *matroska) EbmlList *index_list; MatroskaIndex *index; uint64_t index_scale = 1; - int i, j; + unsigned i, j; if (matroska->ctx->flags & AVFMT_FLAG_IGNIDX) return; @@ -2265,8 +2262,8 @@ static int matroska_parse_tracks(AVFormatContext *s) MatroskaDemuxContext *matroska = s->priv_data; MatroskaTrack *tracks = matroska->tracks.elem; AVStream *st; - int i, j, ret; - int k; + unsigned i, j, k; + int ret; for (i = 0; i < matroska->tracks.nb_elem; i++) { MatroskaTrack *track = &tracks[i]; @@ -2716,7 +2713,7 @@ static int matroska_parse_tracks(AVFormatContext *s) char buf[32]; if (planes[j].type >= MATROSKA_VIDEO_STEREO_PLANE_COUNT) continue; - snprintf(buf, sizeof(buf), "%s_%d", + snprintf(buf, sizeof(buf), "%s_%u", ff_matroska_video_stereo_plane[planes[j].type], i); for (k=0; k < matroska->tracks.nb_elem; k++) if (planes[j].uid == tracks[k].uid && tracks[k].stream) { @@ -2791,7 +2788,8 @@ static int matroska_read_header(AVFormatContext *s) uint64_t max_start = 0; int64_t pos; Ebml ebml = { 0 }; - int i, j, res; + unsigned i, j; + int res; matroska->ctx = s; matroska->cues_parsing_deferred = 1; @@ -3708,7 +3706,7 @@ static int matroska_read_seek(AVFormatContext *s, int stream_index, MatroskaDemuxContext *matroska = s->priv_data; MatroskaTrack *tracks = NULL; AVStream *st = s->streams[stream_index]; - int i, index; + int index; /* Parse the CUES now since we need the index data to seek. */ if (matroska->cues_parsing_deferred > 0) { @@ -3734,7 +3732,7 @@ static int matroska_read_seek(AVFormatContext *s, int stream_index, goto err; tracks = matroska->tracks.elem; - for (i = 0; i < matroska->tracks.nb_elem; i++) { + for (unsigned i = 0; i < matroska->tracks.nb_elem; i++) { tracks[i].audio.pkt_cnt = 0; tracks[i].audio.sub_packet_cnt = 0; tracks[i].audio.buf_timecode = AV_NOPTS_VALUE; @@ -3770,7 +3768,7 @@ static int matroska_read_close(AVFormatContext *s) { MatroskaDemuxContext *matroska = s->priv_data; MatroskaTrack *tracks = matroska->tracks.elem; - int n; + unsigned n; matroska_clear_queue(matroska); @@ -4043,7 +4041,7 @@ static int webm_dash_manifest_cues(AVFormatContext *s, int64_t init_range) MatroskaSeekhead *seekhead = seekhead_list->elem; char *buf; int64_t cues_start = -1, cues_end = -1, before_pos, bandwidth; - int i; + unsigned i; int end = 0; // determine cues start and end positions @@ -4099,7 +4097,7 @@ static int webm_dash_manifest_cues(AVFormatContext *s, int64_t init_range) buf = av_malloc_array(s->streams[0]->nb_index_entries, 20); if (!buf) return -1; strcpy(buf, ""); - for (i = 0; i < s->streams[0]->nb_index_entries; i++) { + for (int i = 0; i < s->streams[0]->nb_index_entries; i++) { int ret = snprintf(buf + end, 20, "%" PRId64"%s", s->streams[0]->index_entries[i].timestamp, i != s->streams[0]->nb_index_entries - 1 ? "," : "");