#include "movenccenc.h"
#include "libavcodec/av1_parse.h"
#include "libavcodec/bytestream.h"
#include "libavcodec/cbs_av1.h"
#include "libavutil/attributes.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/mem.h"
#include "avio_internal.h"
#include "movenc.h"
#include "avc.h"
#include "nal.h"

Functions
static int	auxiliary_info_alloc_size (MOVMuxCencContext *ctx, int size)

static int	auxiliary_info_write (MOVMuxCencContext ctx, const uint8_t buf_in, int size)

static int	auxiliary_info_add_subsample (MOVMuxCencContext *ctx, uint16_t clear_bytes, uint32_t encrypted_bytes)

static void	mov_cenc_write_encrypted (MOVMuxCencContext ctx, AVIOContext pb, const uint8_t *buf_in, int size)
	Encrypt the input buffer and write using avio_write. More...

static int	mov_cenc_start_packet (MOVMuxCencContext *ctx)
	Start writing a packet. More...

static int	mov_cenc_end_packet (MOVMuxCencContext *ctx)
	Finalize a packet. More...

int	ff_mov_cenc_write_packet (MOVMuxCencContext ctx, AVIOContext pb, const uint8_t *buf_in, int size)
	Write a fully encrypted packet. More...

int	ff_mov_cenc_avc_parse_nal_units (MOVMuxCencContext ctx, AVIOContext pb, const uint8_t *buf_in, int size)
	Parse AVC NAL units from annex B format, the nal size and type are written in the clear while the body is encrypted. More...

int	ff_mov_cenc_avc_write_nal_units (AVFormatContext s, MOVMuxCencContext ctx, int nal_length_size, AVIOContext pb, const uint8_t buf_in, int size)
	Write AVC NAL units that are in MP4 format, the nal size and type are written in the clear while the body is encrypted. More...

static int	write_tiles (AVFormatContext s, MOVMuxCencContext ctx, AVIOContext pb, AV1_OBU_Type type, const AV1RawFrameHeader frame_header, const uint8_t fh_data, size_t fh_data_size, const AV1RawTileGroup tile_group)

int	ff_mov_cenc_av1_write_obus (AVFormatContext s, MOVMuxCencContext ctx, AVIOContext pb, const AVPacket pkt)

static int64_t	update_size (AVIOContext *pb, int64_t pos)

static int	mov_cenc_write_senc_tag (MOVMuxCencContext ctx, AVIOContext pb, int64_t *auxiliary_info_offset)

static int	mov_cenc_write_saio_tag (AVIOContext *pb, int64_t auxiliary_info_offset)

static int	mov_cenc_write_saiz_tag (MOVMuxCencContext ctx, AVIOContext pb)

void	ff_mov_cenc_write_stbl_atoms (MOVMuxCencContext ctx, AVIOContext pb, int64_t moof_offset)
	Write the cenc atoms that should reside inside stbl. More...

static int	mov_cenc_write_schi_tag (AVIOContext pb, uint8_t kid)

int	ff_mov_cenc_write_sinf_tag (MOVTrack track, AVIOContext pb, uint8_t *kid)
	Write the sinf atom, contained inside stsd. More...

int	ff_mov_cenc_init (MOVMuxCencContext ctx, uint8_t encryption_key, int use_subsamples, enum AVCodecID codec_id, int bitexact)
	Initialize a CENC context. More...

void	ff_mov_cenc_flush (MOVMuxCencContext *ctx)
	Clear subsample data. More...

void	ff_mov_cenc_free (MOVMuxCencContext *ctx)
	Free a CENC context. More...

Variables
static const CodedBitstreamUnitType	decompose_unit_types []

Function Documentation

◆ auxiliary_info_alloc_size()

static int auxiliary_info_alloc_size	(	MOVMuxCencContext *	ctx,
		int	size
	)

static

Definition at line 33 of file movenccenc.c.

Referenced by auxiliary_info_add_subsample(), and auxiliary_info_write().

◆ auxiliary_info_write()

static int auxiliary_info_write	(	MOVMuxCencContext *	ctx,
		const uint8_t *	buf_in,
		int	size
	)

static

Definition at line 49 of file movenccenc.c.

Referenced by mov_cenc_start_packet().

◆ auxiliary_info_add_subsample()

static int auxiliary_info_add_subsample	(	MOVMuxCencContext *	ctx,
		uint16_t	clear_bytes,
		uint32_t	encrypted_bytes
	)

static

Definition at line 64 of file movenccenc.c.

Referenced by ff_mov_cenc_av1_write_obus(), ff_mov_cenc_avc_parse_nal_units(), ff_mov_cenc_avc_write_nal_units(), ff_mov_cenc_write_packet(), and write_tiles().

◆ mov_cenc_write_encrypted()

static void mov_cenc_write_encrypted	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		const uint8_t *	buf_in,
		int	size
	)

static

Encrypt the input buffer and write using avio_write.

Definition at line 95 of file movenccenc.c.

Referenced by ff_mov_cenc_avc_parse_nal_units(), ff_mov_cenc_avc_write_nal_units(), ff_mov_cenc_write_packet(), and write_tiles().

◆ mov_cenc_start_packet()

static int mov_cenc_start_packet ( MOVMuxCencContext * ctx )

static

Start writing a packet.

Definition at line 115 of file movenccenc.c.

Referenced by ff_mov_cenc_av1_write_obus(), ff_mov_cenc_avc_parse_nal_units(), ff_mov_cenc_avc_write_nal_units(), and ff_mov_cenc_write_packet().

◆ mov_cenc_end_packet()

static int mov_cenc_end_packet ( MOVMuxCencContext * ctx )

static

Finalize a packet.

Definition at line 143 of file movenccenc.c.

Referenced by ff_mov_cenc_av1_write_obus(), ff_mov_cenc_avc_parse_nal_units(), ff_mov_cenc_avc_write_nal_units(), and ff_mov_cenc_write_packet().

◆ ff_mov_cenc_write_packet()

int ff_mov_cenc_write_packet	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		const uint8_t *	buf_in,
		int	size
	)

Write a fully encrypted packet.

Definition at line 173 of file movenccenc.c.

Referenced by ff_mov_write_packet().

◆ ff_mov_cenc_avc_parse_nal_units()

int ff_mov_cenc_avc_parse_nal_units	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		const uint8_t *	buf_in,
		int	size
	)

Parse AVC NAL units from annex B format, the nal size and type are written in the clear while the body is encrypted.

Definition at line 198 of file movenccenc.c.

Referenced by ff_mov_write_packet().

◆ ff_mov_cenc_avc_write_nal_units()

int ff_mov_cenc_avc_write_nal_units	(	AVFormatContext *	s,
		MOVMuxCencContext *	ctx,
		int	nal_length_size,
		AVIOContext *	pb,
		const uint8_t *	buf_in,
		int	size
	)

Write AVC NAL units that are in MP4 format, the nal size and type are written in the clear while the body is encrypted.

Definition at line 238 of file movenccenc.c.

Referenced by ff_mov_write_packet().

◆ write_tiles()

static int write_tiles	(	AVFormatContext *	s,
		MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		AV1_OBU_Type	type,
		const AV1RawFrameHeader *	frame_header,
		const uint8_t *	fh_data,
		size_t	fh_data_size,
		const AV1RawTileGroup *	tile_group
	)

static

Definition at line 287 of file movenccenc.c.

Referenced by ff_mov_cenc_av1_write_obus().

◆ ff_mov_cenc_av1_write_obus()

int ff_mov_cenc_av1_write_obus	(	AVFormatContext *	s,
		MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		const AVPacket *	pkt
	)

Definition at line 387 of file movenccenc.c.

Referenced by ff_mov_write_packet().

◆ update_size()

static int64_t update_size	(	AVIOContext *	pb,
		int64_t	pos
	)

static

Definition at line 485 of file movenccenc.c.

Referenced by ff_mov_cenc_write_sinf_tag(), mov_cenc_write_saio_tag(), mov_cenc_write_saiz_tag(), mov_cenc_write_schi_tag(), and mov_cenc_write_senc_tag().

◆ mov_cenc_write_senc_tag()

static int mov_cenc_write_senc_tag	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		int64_t *	auxiliary_info_offset
	)

static

Definition at line 495 of file movenccenc.c.

Referenced by ff_mov_cenc_write_stbl_atoms().

◆ mov_cenc_write_saio_tag()

static int mov_cenc_write_saio_tag	(	AVIOContext *	pb,
		int64_t	auxiliary_info_offset
	)

static

Definition at line 509 of file movenccenc.c.

Referenced by ff_mov_cenc_write_stbl_atoms().

◆ mov_cenc_write_saiz_tag()

static int mov_cenc_write_saiz_tag	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb
	)

static

Definition at line 528 of file movenccenc.c.

Referenced by ff_mov_cenc_write_stbl_atoms().

◆ ff_mov_cenc_write_stbl_atoms()

void ff_mov_cenc_write_stbl_atoms	(	MOVMuxCencContext *	ctx,
		AVIOContext *	pb,
		int64_t	moof_offset
	)

Write the cenc atoms that should reside inside stbl.

Definition at line 542 of file movenccenc.c.

Referenced by mov_write_stbl_tag(), and mov_write_traf_tag().

◆ mov_cenc_write_schi_tag()

static int mov_cenc_write_schi_tag	(	AVIOContext *	pb,
		uint8_t *	kid
	)

static

Definition at line 552 of file movenccenc.c.

Referenced by ff_mov_cenc_write_sinf_tag().

◆ ff_mov_cenc_write_sinf_tag()

int ff_mov_cenc_write_sinf_tag	(	MOVTrack *	track,
		AVIOContext *	pb,
		uint8_t *	kid
	)

Write the sinf atom, contained inside stsd.

Definition at line 568 of file movenccenc.c.

Referenced by mov_write_audio_tag(), and mov_write_video_tag().

◆ ff_mov_cenc_init()

int ff_mov_cenc_init	(	MOVMuxCencContext *	ctx,
		uint8_t *	encryption_key,
		int	use_subsamples,
		enum AVCodecID	codec_id,
		int	bitexact
	)

Initialize a CENC context.

Parameters

key	encryption key, must have a length of AES_CTR_KEY_SIZE
use_subsamples	when enabled parts of a packet can be encrypted, otherwise the whole packet is encrypted

Definition at line 600 of file movenccenc.c.

Referenced by mov_init().

◆ ff_mov_cenc_flush()

void ff_mov_cenc_flush ( MOVMuxCencContext * ctx )

Clear subsample data.

To be called for fragmented output.

Definition at line 633 of file movenccenc.c.

Referenced by mov_flush_fragment().

◆ ff_mov_cenc_free()

void ff_mov_cenc_free ( MOVMuxCencContext * ctx )

Free a CENC context.

Definition at line 639 of file movenccenc.c.

Referenced by mov_free().

Variable Documentation

◆ decompose_unit_types

const CodedBitstreamUnitType decompose_unit_types[]

static

Initial value:

= {
    AV1_OBU_TEMPORAL_DELIMITER,
    AV1_OBU_SEQUENCE_HEADER,
    AV1_OBU_FRAME_HEADER,
    AV1_OBU_TILE_GROUP,
    AV1_OBU_FRAME,
}

Definition at line 592 of file movenccenc.c.

Referenced by ff_mov_cenc_init().

Functions

Variables

Function Documentation

◆ auxiliary_info_alloc_size()

◆ auxiliary_info_write()

◆ auxiliary_info_add_subsample()

◆ mov_cenc_write_encrypted()

◆ mov_cenc_start_packet()

◆ mov_cenc_end_packet()

◆ ff_mov_cenc_write_packet()

◆ ff_mov_cenc_avc_parse_nal_units()

◆ ff_mov_cenc_avc_write_nal_units()

◆ write_tiles()

◆ ff_mov_cenc_av1_write_obus()

◆ update_size()

◆ mov_cenc_write_senc_tag()

◆ mov_cenc_write_saio_tag()

◆ mov_cenc_write_saiz_tag()

◆ ff_mov_cenc_write_stbl_atoms()

◆ mov_cenc_write_schi_tag()

◆ ff_mov_cenc_write_sinf_tag()

◆ ff_mov_cenc_init()

◆ ff_mov_cenc_flush()

◆ ff_mov_cenc_free()

Variable Documentation

◆ decompose_unit_types