> On 19 Dec 2017, at 11:51, 刘歧 <l...@chinaffmpeg.org> wrote: > > > >> On 19 Dec 2017, at 11:42, Karthick Jeyapal <kjeya...@akamai.com> wrote: >> >> >> >> On 12/18/17 2:17 PM, Steven Liu wrote: >>> fix ticket id: 6888 >>> >>> Signed-off-by: Steven Liu <l...@chinaffmpeg.org> >>> --- >>> libavformat/hlsenc.c | 72 >>> ++++++++++++++++++++++++++++++++++++++++++++-------- >>> 1 file changed, 62 insertions(+), 10 deletions(-) >>> >>> diff --git a/libavformat/hlsenc.c b/libavformat/hlsenc.c >>> index e3442c368f..f51fec1030 100644 >>> --- a/libavformat/hlsenc.c >>> +++ b/libavformat/hlsenc.c >>> @@ -352,6 +352,29 @@ static void write_styp(AVIOContext *pb) >>> ffio_wfourcc(pb, "msix"); >>> } >>> +static int flush_dynbuf(VariantStream *vs, int *range_length) >>> +{ >>> + AVFormatContext *ctx = vs->avf; >>> + uint8_t *buffer; >>> + >>> + if (!ctx->pb) { >>> + return AVERROR(EINVAL); >>> + } >>> + >>> + // flush >>> + av_write_frame(ctx, NULL); >>> + avio_flush(ctx->pb); >>> + >>> + // write out to file >>> + *range_length = avio_close_dyn_buf(ctx->pb, &buffer); >>> + ctx->pb = NULL; >>> + avio_write(vs->out, buffer, *range_length); >>> + av_free(buffer); >>> + >>> + // re-open buffer >>> + return avio_open_dyn_buf(&ctx->pb); >>> +} >>> + >>> static int hls_delete_old_segments(AVFormatContext *s, HLSContext *hls, >>> VariantStream *vs) { >>> @@ -677,7 +700,9 @@ static int hls_mux_init(AVFormatContext *s, >>> VariantStream *vs) >>> if ((ret = avio_open_dyn_buf(&oc->pb)) < 0) >>> return ret; >>> - if ((ret = s->io_open(s, &vs->out, vs->base_output_dirname, >>> AVIO_FLAG_WRITE, &options)) < 0) { >>> + ret = hlsenc_io_open(s, &vs->out, vs->base_output_dirname, >>> &options); >>> + av_dict_free(&options); >>> + if (ret < 0) { >>> av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", >>> vs->fmp4_init_filename); >>> return ret; >>> } >>> @@ -1404,9 +1429,10 @@ static int hls_start(AVFormatContext *s, >>> VariantStream *vs) >>> av_dict_free(&options); >>> if (err < 0) >>> return err; >>> - } else >>> + } else if (c->segment_type != SEGMENT_TYPE_FMP4) { >>> if ((err = hlsenc_io_open(s, &oc->pb, oc->filename, &options)) < 0) >>> goto fail; >>> + } >>> if (vs->vtt_basename) { >>> set_http_options(s, &options, c); >>> if ((err = hlsenc_io_open(s, &vtt_oc->pb, vtt_oc->filename, >>> &options)) < 0) >>> @@ -1414,9 +1440,7 @@ static int hls_start(AVFormatContext *s, >>> VariantStream *vs) >>> } >>> av_dict_free(&options); >>> - if (c->segment_type == SEGMENT_TYPE_FMP4 && !(c->flags & >>> HLS_SINGLE_FILE)) { >>> - write_styp(oc->pb); >>> - } else { >>> + if (c->segment_type != SEGMENT_TYPE_FMP4) { >>> /* We only require one PAT/PMT per segment. */ >>> if (oc->oformat->priv_class && oc->priv_data) { >>> char period[21]; >>> @@ -1780,7 +1804,8 @@ static int hls_write_packet(AVFormatContext *s, >>> AVPacket *pkt) >>> vs->size = new_start_pos - vs->start_pos; >>> if (!byterange_mode) { >>> - if (hls->segment_type == SEGMENT_TYPE_FMP4 && >>> !vs->init_range_length) { >>> + if (hls->segment_type == SEGMENT_TYPE_FMP4) { >>> + if (!vs->init_range_length) { >>> avio_flush(oc->pb); >>> range_length = avio_close_dyn_buf(oc->pb, &buffer); >>> avio_write(vs->out, buffer, range_length); >>> @@ -1789,6 +1814,7 @@ static int hls_write_packet(AVFormatContext *s, >>> AVPacket *pkt) >>> vs->packets_written = 0; >>> ff_format_io_close(s, &vs->out); >>> hlsenc_io_close(s, &vs->out, vs->base_output_dirname); >>> + } >>> } else { >>> hlsenc_io_close(s, &oc->pb, oc->filename); >>> } >>> @@ -1807,7 +1833,20 @@ static int hls_write_packet(AVFormatContext *s, >>> AVPacket *pkt) >>> vs->number--; >>> } >>> - if (!vs->fmp4_init_mode || byterange_mode) >>> + if (hls->segment_type == SEGMENT_TYPE_FMP4) { >>> + ret = hlsenc_io_open(s, &vs->out, vs->avf->filename, NULL); >> Will it be better if you call "set_http_options()" here itself, instead of a >> separate patch 4/4? In that way this patch would be self contained. > set_http_options is add option to http operation, so separate two patches. > And this patch just fix ticket >>> + if (ret < 0) { >>> + av_log(NULL, AV_LOG_ERROR, "Failed to open file '%s'\n", >>> + vs->avf->filename); >>> + return ret; >>> + } >>> + write_styp(vs->out); >>> + ret = flush_dynbuf(vs, &range_length); >>> + if (ret < 0) { >>> + return ret; >>> + } >>> + ff_format_io_close(s, &vs->out); >> Again is it better to call hlsenc_io_close() here itself instead of patch >> 3/4. Again just from self-containment perspective. > Replace API to hlsenc_io_close is a new operation, so separate two patch > And this patch just fix ticket >>> + } >>> ret = hls_append_segment(s, hls, vs, vs->duration, >>> vs->start_pos, vs->size); >>> vs->start_pos = new_start_pos; >>> if (ret < 0) { >>> @@ -1861,6 +1900,7 @@ static int hls_write_trailer(struct AVFormatContext >>> *s) >>> AVFormatContext *vtt_oc = NULL; >>> char *old_filename = NULL; >>> int i; >>> + int ret = 0; >>> VariantStream *vs = NULL; >>> for (i = 0; i < hls->nb_varstreams; i++) { >>> @@ -1873,11 +1913,25 @@ static int hls_write_trailer(struct AVFormatContext >>> *s) >>> if (!old_filename) { >>> return AVERROR(ENOMEM); >>> } >>> - >>> + if ( hls->segment_type == SEGMENT_TYPE_FMP4) { >>> + int range_length = 0; >>> + ret = hlsenc_io_open(s, &vs->out, vs->avf->filename, NULL); >>> + if (ret < 0) { >>> + av_log(NULL, AV_LOG_ERROR, "Failed to open file '%s'\n", >>> vs->avf->filename); >>> + return AVERROR(ENOENT); >>> + } >>> + write_styp(vs->out); >>> + ret = flush_dynbuf(vs, &range_length); >>> + if (ret < 0) { >>> + return ret; >>> + } >>> + ff_format_io_close(s, &vs->out); >>> + } >> Will it make sense it make this code as a separate function in order to >> avoid code repetition? > That is optimize operation, this patch just fix ticket, so i will send a new > patch to merge two workflow to on function. > > Thanks > > Steven >>> av_write_trailer(oc); >>> if (oc->pb) { >>> vs->size = avio_tell(vs->avf->pb) - vs->start_pos; >>> + if (hls->segment_type != SEGMENT_TYPE_FMP4) >>> ff_format_io_close(s, &oc->pb); >>> if ((hls->flags & HLS_TEMP_FILE) && oc->filename[0]) { >>> @@ -2238,10 +2292,8 @@ static int hls_init(AVFormatContext *s) >>> } >>> } >>> - if (hls->segment_type != SEGMENT_TYPE_FMP4 || hls->flags & >>> HLS_SINGLE_FILE) { >>> if ((ret = hls_start(s, vs)) < 0) >>> goto fail; >>> - } >>> } >>> fail: >> Thanks, >> Karthick >> _______________________________________________ >> ffmpeg-devel mailing list >> ffmpeg-devel@ffmpeg.org >> http://ffmpeg.org/mailman/listinfo/ffmpeg-devel
pushed Thanks _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org http://ffmpeg.org/mailman/listinfo/ffmpeg-devel