bdfae2a5 |
/* |
ec21c215 |
* H.264 MP4 to Annex B byte stream format filter |
074bfa7d |
* Copyright (c) 2007 Benoit Fouet <benoit.fouet@free.fr> |
bdfae2a5 |
*
* 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
*/
|
6a5d31ac |
#include "libavutil/intreadwrite.h" |
bdfae2a5 |
#include "avcodec.h"
typedef struct H264BSFContext {
uint8_t length_size;
uint8_t first_idr; |
d5cc1ed7 |
int extradata_parsed; |
bdfae2a5 |
} H264BSFContext;
|
639c697c |
static int alloc_and_copy(uint8_t **poutbuf, int *poutbuf_size, |
bb5cfc48 |
const uint8_t *sps_pps, uint32_t sps_pps_size,
const uint8_t *in, uint32_t in_size) { |
bdfae2a5 |
uint32_t offset = *poutbuf_size;
uint8_t nal_header_size = offset ? 3 : 4; |
639c697c |
void *tmp; |
bdfae2a5 |
*poutbuf_size += sps_pps_size+in_size+nal_header_size; |
639c697c |
tmp = av_realloc(*poutbuf, *poutbuf_size);
if (!tmp)
return AVERROR(ENOMEM);
*poutbuf = tmp; |
bdfae2a5 |
if (sps_pps)
memcpy(*poutbuf+offset, sps_pps, sps_pps_size);
memcpy(*poutbuf+sps_pps_size+nal_header_size+offset, in, in_size); |
1f7d2f54 |
if (!offset) { |
bdfae2a5 |
AV_WB32(*poutbuf+sps_pps_size, 1); |
1f7d2f54 |
} else { |
48aecf5a |
(*poutbuf+offset+sps_pps_size)[0] = (*poutbuf+offset+sps_pps_size)[1] = 0;
(*poutbuf+offset+sps_pps_size)[2] = 1; |
bdfae2a5 |
} |
639c697c |
return 0; |
bdfae2a5 |
}
static int h264_mp4toannexb_filter(AVBitStreamFilterContext *bsfc,
AVCodecContext *avctx, const char *args,
uint8_t **poutbuf, int *poutbuf_size,
const uint8_t *buf, int buf_size,
int keyframe) {
H264BSFContext *ctx = bsfc->priv_data; |
f1fdd208 |
int i; |
bdfae2a5 |
uint8_t unit_type; |
52486603 |
int32_t nal_size;
uint32_t cumul_size = 0;
const uint8_t *buf_end = buf + buf_size; |
7ae251b4 |
int ret = AVERROR(EINVAL); |
bdfae2a5 |
/* nothing to filter */
if (!avctx->extradata || avctx->extradata_size < 6) {
*poutbuf = (uint8_t*) buf;
*poutbuf_size = buf_size;
return 0;
}
/* retrieve sps and pps NAL units from extradata */ |
d5cc1ed7 |
if (!ctx->extradata_parsed) { |
bdfae2a5 |
uint16_t unit_size; |
d5cc1ed7 |
uint64_t total_size = 0; |
02dd3666 |
uint8_t *out = NULL, unit_nb, sps_done = 0, sps_seen = 0, pps_seen = 0; |
bdfae2a5 |
const uint8_t *extradata = avctx->extradata+4;
static const uint8_t nalu_header[4] = {0, 0, 0, 1};
/* retrieve length coded size */
ctx->length_size = (*extradata++ & 0x3) + 1;
/* retrieve sps and pps unit(s) */
unit_nb = *extradata++ & 0x1f; /* number of sps unit(s) */
if (!unit_nb) { |
9d4cb45d |
goto pps; |
4e84f994 |
} else { |
02dd3666 |
sps_seen = 1; |
bdfae2a5 |
} |
02dd3666 |
|
bdfae2a5 |
while (unit_nb--) { |
639c697c |
void *tmp;
|
bdfae2a5 |
unit_size = AV_RB16(extradata);
total_size += unit_size+4; |
bb5cfc48 |
if (total_size > INT_MAX - FF_INPUT_BUFFER_PADDING_SIZE ||
extradata+2+unit_size > avctx->extradata+avctx->extradata_size) { |
bdfae2a5 |
av_free(out);
return AVERROR(EINVAL);
} |
639c697c |
tmp = av_realloc(out, total_size + FF_INPUT_BUFFER_PADDING_SIZE);
if (!tmp) {
av_free(out); |
bdfae2a5 |
return AVERROR(ENOMEM); |
639c697c |
}
out = tmp; |
bdfae2a5 |
memcpy(out+total_size-unit_size-4, nalu_header, 4);
memcpy(out+total_size-unit_size, extradata+2, unit_size);
extradata += 2+unit_size; |
9d4cb45d |
pps: |
025225d7 |
if (!unit_nb && !sps_done++) { |
bdfae2a5 |
unit_nb = *extradata++; /* number of pps unit(s) */ |
025225d7 |
if (unit_nb)
pps_seen = 1;
} |
bdfae2a5 |
}
|
02dd3666 |
if(out)
memset(out + total_size, 0, FF_INPUT_BUFFER_PADDING_SIZE);
if (!sps_seen)
av_log(avctx, AV_LOG_WARNING, "Warning: SPS NALU missing or invalid. The resulting stream may not play.\n");
if (!pps_seen)
av_log(avctx, AV_LOG_WARNING, "Warning: PPS NALU missing or invalid. The resulting stream may not play.\n");
|
d5cc1ed7 |
av_free(avctx->extradata);
avctx->extradata = out;
avctx->extradata_size = total_size;
ctx->first_idr = 1;
ctx->extradata_parsed = 1; |
bdfae2a5 |
}
*poutbuf_size = 0;
*poutbuf = NULL;
do { |
7ae251b4 |
ret= AVERROR(EINVAL); |
52486603 |
if (buf + ctx->length_size > buf_end)
goto fail;
|
f1fdd208 |
for (nal_size = 0, i = 0; i<ctx->length_size; i++)
nal_size = (nal_size << 8) | buf[i]; |
bdfae2a5 |
buf += ctx->length_size;
unit_type = *buf & 0x1f;
|
52486603 |
if (buf + nal_size > buf_end || nal_size < 0)
goto fail;
|
bdfae2a5 |
/* prepend only to the first type 5 NAL unit of an IDR picture */
if (ctx->first_idr && unit_type == 5) { |
7ae251b4 |
if ((ret=alloc_and_copy(poutbuf, poutbuf_size, |
bb5cfc48 |
avctx->extradata, avctx->extradata_size, |
7ae251b4 |
buf, nal_size)) < 0) |
639c697c |
goto fail; |
bdfae2a5 |
ctx->first_idr = 0; |
1f7d2f54 |
} else { |
7ae251b4 |
if ((ret=alloc_and_copy(poutbuf, poutbuf_size, |
bb5cfc48 |
NULL, 0, |
7ae251b4 |
buf, nal_size)) < 0) |
639c697c |
goto fail; |
bdfae2a5 |
if (!ctx->first_idr && unit_type == 1)
ctx->first_idr = 1;
}
buf += nal_size;
cumul_size += nal_size + ctx->length_size;
} while (cumul_size < buf_size);
return 1; |
52486603 |
fail:
av_freep(poutbuf);
*poutbuf_size = 0; |
7ae251b4 |
return ret; |
bdfae2a5 |
}
|
d36beb3f |
AVBitStreamFilter ff_h264_mp4toannexb_bsf = { |
bdfae2a5 |
"h264_mp4toannexb",
sizeof(H264BSFContext),
h264_mp4toannexb_filter,
}; |