king
/
v811_spc009
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '8f4c154c05'
${ noResults }
v811_spc009/external/libxaac/fuzzer/xaac_dec_fuzzer.cpp

875 lines
31 KiB
Raw Blame History

/******************************************************************************
*
* Copyright (C) 2019 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*****************************************************************************
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
*/
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <vector>
#include "ixheaacd_type_def.h"
#include "ixheaacd_error_standards.h"
#include "ixheaacd_error_handler.h"
#include "ixheaacd_apicmd_standards.h"
#include "ixheaacd_memory_standards.h"
#include "ixheaacd_aac_config.h"
#include "impd_apicmd_standards.h"
#include "impd_drc_config_params.h"
/* 64*-0.25dB = -16 dB below full scale for mobile conf */
#define DRC_DEFAULT_MOBILE_REF_LEVEL 64
/* maximum compression of dynamic range for mobile conf */
#define DRC_DEFAULT_MOBILE_DRC_CUT 127
/* maximum compression of dynamic range for mobile conf */
#define DRC_DEFAULT_MOBILE_DRC_BOOST 127
/* switch for heavy compression for mobile conf */
#define DRC_DEFAULT_MOBILE_DRC_HEAVY 1
/* encoder target level; -1 => the value is unknown, otherwise dB \
step value (e.g. 64 for -16 dB) */
#define DRC_DEFAULT_MOBILE_ENC_LEVEL (-1)
#define MAX_CHANNEL_COUNT 8
#define MAX_MEM_ALLOCS 100
class Codec {
public:
IA_ERRORCODE initDecoder(const uint8_t* data, size_t size, bool isADTS);
IA_ERRORCODE initXAACDecoder(bool isADTS);
IA_ERRORCODE initXAACDrc(const uint8_t* data, size_t size);
IA_ERRORCODE deInitXAACDecoder();
IA_ERRORCODE deInitMPEGDDDrc();
IA_ERRORCODE configXAACDecoder(uint8_t* inBuffer, uint32_t inBufferLength,
int32_t* bytesConsumed);
IA_ERRORCODE initMPEGDDDrc();
int configMPEGDDrc();
IA_ERRORCODE decodeXAACStream(uint8_t* inBuffer, uint32_t inBufferLength,
int32_t* bytesConsumed, int32_t* outBytes);
IA_ERRORCODE getXAACStreamInfo();
IA_ERRORCODE setXAACDRCInfo(int32_t drcCut, int32_t drcBoost,
int32_t drcRefLevel, int32_t drcHeavyCompression,
int32_t drEffectType);
private:
void* mXheaacCodecHandle;
void* mMpegDDrcHandle;
uint32_t mInputBufferSize;
uint32_t mOutputFrameLength;
int8_t* mInputBuffer;
int8_t* mOutputBuffer;
int32_t mSampFreq;
int32_t mNumChannels;
int32_t mPcmWdSz;
int32_t mChannelMask;
bool mIsCodecInitialized;
bool mIsCodecConfigFlushRequired;
int8_t* mDrcInBuf;
int8_t* mDrcOutBuf;
int32_t mMpegDDRCPresent;
int32_t mDRCFlag;
std::vector<void*> mMemoryVec;
std::vector<void*> mDrcMemoryVec;
};
extern "C" IA_ERRORCODE ixheaacd_dec_api(pVOID p_ia_module_obj, WORD32 i_cmd,
WORD32 i_idx, pVOID pv_value);
extern "C" IA_ERRORCODE ia_drc_dec_api(pVOID p_ia_module_obj, WORD32 i_cmd,
WORD32 i_idx, pVOID pv_value);
extern "C" IA_ERRORCODE ixheaacd_get_config_param(pVOID p_ia_process_api_obj,
pWORD32 pi_samp_freq,
pWORD32 pi_num_chan,
pWORD32 pi_pcm_wd_sz,
pWORD32 pi_channel_mask);
IA_ERRORCODE Codec::initXAACDecoder(bool isADTS) {
/* First part */
/* Error Handler Init */
/* Get Library Name, Library Version and API Version */
/* Initialize API structure + Default config set */
/* Set config params from user */
/* Initialize memory tables */
/* Get memory information and allocate memory */
mInputBufferSize = 0;
mInputBuffer = nullptr;
mOutputBuffer = nullptr;
/* Process struct initing end */
/* ******************************************************************/
/* Initialize API structure and set config params to default */
/* ******************************************************************/
/* API size */
uint32_t pui_api_size;
/* Get the API size */
IA_ERRORCODE err_code =
ixheaacd_dec_api(nullptr, IA_API_CMD_GET_API_SIZE, 0, &pui_api_size);
/* Allocate memory for API */
mXheaacCodecHandle = malloc(pui_api_size);
if (!mXheaacCodecHandle) {
return IA_FATAL_ERROR;
}
mMemoryVec.push_back(mXheaacCodecHandle);
/* Set the config params to default values */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_API_PRE_CONFIG_PARAMS, nullptr);
/* Get the API size */
err_code = ia_drc_dec_api(nullptr, IA_API_CMD_GET_API_SIZE, 0, &pui_api_size);
/* Allocate memory for API */
mMpegDDrcHandle = malloc(pui_api_size);
if (!mMpegDDrcHandle) {
return IA_FATAL_ERROR;
}
mMemoryVec.push_back(mMpegDDrcHandle);
/* Set the config params to default values */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_API_PRE_CONFIG_PARAMS, nullptr);
/* ******************************************************************/
/* Set config parameters */
/* ******************************************************************/
uint32_t ui_mp4_flag = isADTS ? 0 : 1;
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_ISMP4, &ui_mp4_flag);
/* ******************************************************************/
/* Initialize Memory info tables */
/* ******************************************************************/
uint32_t ui_proc_mem_tabs_size;
pVOID pv_alloc_ptr;
/* Get memory info tables size */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_MEMTABS_SIZE,
0, &ui_proc_mem_tabs_size);
pv_alloc_ptr = malloc(ui_proc_mem_tabs_size);
if (!pv_alloc_ptr) {
return IA_FATAL_ERROR;
}
mMemoryVec.push_back(pv_alloc_ptr);
/* Set pointer for process memory tables */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_MEMTABS_PTR, 0,
pv_alloc_ptr);
/* initialize the API, post config, fill memory tables */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_API_POST_CONFIG_PARAMS, nullptr);
/* ******************************************************************/
/* Allocate Memory with info from library */
/* ******************************************************************/
/* There are four different types of memories, that needs to be allocated */
/* persistent,scratch,input and output */
for (int i = 0; i < 4; i++) {
int ui_size = 0, ui_alignment = 0, ui_type = 0;
/* Get memory size */
err_code = ixheaacd_dec_api(mXheaacCodecHandle,
IA_API_CMD_GET_MEM_INFO_SIZE, i, &ui_size);
/* Get memory alignment */
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_MEM_INFO_ALIGNMENT,
i, &ui_alignment);
/* Get memory type */
err_code = ixheaacd_dec_api(mXheaacCodecHandle,
IA_API_CMD_GET_MEM_INFO_TYPE, i, &ui_type);
pv_alloc_ptr = NULL;
ui_alignment = (ui_alignment + sizeof(void *) - 1) / sizeof(void *);
ui_alignment = ui_alignment * sizeof(void *);
if (0 != posix_memalign(&pv_alloc_ptr, ui_alignment, ui_size)) {
return IA_FATAL_ERROR;
}
if (!pv_alloc_ptr) {
return IA_FATAL_ERROR;
}
mMemoryVec.push_back(pv_alloc_ptr);
/* Set the buffer pointer */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_MEM_PTR, i,
pv_alloc_ptr);
if (ui_type == IA_MEMTYPE_INPUT) {
mInputBuffer = (pWORD8)pv_alloc_ptr;
mInputBufferSize = ui_size;
}
if (ui_type == IA_MEMTYPE_OUTPUT) mOutputBuffer = (pWORD8)pv_alloc_ptr;
}
/* End first part */
return IA_NO_ERROR;
}
enum {
DRC_TARGET_LEVEL_OFFSET = 6,
DRC_ATTENUATION_OFFSET,
DRC_BOOST_OFFSET,
DRC_COMPRESS_OFFSET,
DRC_EFFECT_OFFSET
};
IA_ERRORCODE Codec::initXAACDrc(const uint8_t* data, size_t size) {
IA_ERRORCODE err_code = IA_NO_ERROR;
unsigned int ui_drc_val;
// DRC_PRES_MODE_WRAP_DESIRED_TARGET
size_t targetLevelOffset =
std::min((size_t)DRC_TARGET_LEVEL_OFFSET, size - 1);
int32_t targetRefLevel = data[targetLevelOffset];
ui_drc_val = (unsigned int)targetRefLevel;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_TARGET_LEVEL,
&ui_drc_val);
/* Use ui_drc_val from PROP_DRC_OVERRIDE_REF_LEVEL or
* DRC_DEFAULT_MOBILE_REF_LEVEL
* for IA_ENHAACPLUS_DEC_DRC_TARGET_LOUDNESS too */
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_DRC_TARGET_LOUDNESS, &ui_drc_val);
size_t attenuationOffset = std::min((size_t)DRC_ATTENUATION_OFFSET, size - 1);
int32_t attenuationFactor = data[attenuationOffset];
ui_drc_val = (unsigned int)attenuationFactor;
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_CUT, &ui_drc_val);
// DRC_PRES_MODE_WRAP_DESIRED_BOOST_FACTOR
size_t boostOffset = std::min((size_t)DRC_BOOST_OFFSET, size - 1);
int32_t boostFactor = data[boostOffset];
ui_drc_val = (unsigned int)boostFactor;
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_BOOST, &ui_drc_val);
// DRC_PRES_MODE_WRAP_DESIRED_HEAVY
size_t compressOffset = std::min((size_t)DRC_COMPRESS_OFFSET, size - 1);
int32_t compressMode = data[compressOffset];
ui_drc_val = (unsigned int)compressMode;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_HEAVY_COMP,
&ui_drc_val);
// AAC_UNIDRC_SET_EFFECT
size_t effectOffset = std::min((size_t)DRC_EFFECT_OFFSET, size - 1);
int32_t effectType = data[effectOffset];
ui_drc_val = (unsigned int)effectType;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_DRC_EFFECT_TYPE, &ui_drc_val);
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::deInitXAACDecoder() {
/* Error code */
IA_ERRORCODE err_code = IA_NO_ERROR;
if (mXheaacCodecHandle) {
/* Tell that the input is over in this buffer */
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INPUT_OVER, 0, nullptr);
}
/* Irrespective of error returned in IA_API_CMD_INPUT_OVER, free allocated
* memory */
for (void* buf : mMemoryVec) {
if (buf) free(buf);
}
mMemoryVec.clear();
mXheaacCodecHandle = nullptr;
return err_code;
}
IA_ERRORCODE Codec::deInitMPEGDDDrc() {
for (void* buf : mDrcMemoryVec) {
if (buf) free(buf);
}
mDrcMemoryVec.clear();
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::configXAACDecoder(uint8_t* inBuffer,
uint32_t inBufferLength,
int32_t* bytesConsumed) {
if (mInputBufferSize < inBufferLength) {
inBufferLength = mInputBufferSize;
}
/* Copy the buffer passed by Android plugin to codec input buffer */
memcpy(mInputBuffer, inBuffer, inBufferLength);
/* Set number of bytes to be processed */
IA_ERRORCODE err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_SET_INPUT_BYTES, 0, &inBufferLength);
if (mIsCodecConfigFlushRequired) {
/* If codec is already initialized, then GA header is passed again */
/* Need to call the Flush API instead of INIT_PROCESS */
mIsCodecInitialized =
false; /* Codec needs to be Reinitialized after flush */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_GA_HDR, nullptr);
} else {
/* Initialize the process */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_PROCESS, nullptr);
}
uint32_t ui_init_done;
/* Checking for end of initialization */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_DONE_QUERY, &ui_init_done);
/* How much buffer is used in input buffers */
err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_GET_CURIDX_INPUT_BUF, 0, bytesConsumed);
if (ui_init_done) {
err_code = getXAACStreamInfo();
mIsCodecInitialized = true;
err_code = configMPEGDDrc();
}
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::initMPEGDDDrc() {
IA_ERRORCODE err_code = IA_NO_ERROR;
for (int i = 0; i < (WORD32)2; i++) {
WORD32 ui_size, ui_alignment, ui_type;
pVOID pv_alloc_ptr;
/* Get memory size */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_GET_MEM_INFO_SIZE, i,
&ui_size);
/* Get memory alignment */
err_code = ia_drc_dec_api(
mMpegDDrcHandle, IA_API_CMD_GET_MEM_INFO_ALIGNMENT, i, &ui_alignment);
/* Get memory type */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_GET_MEM_INFO_TYPE, i,
&ui_type);
pv_alloc_ptr = malloc(ui_size);
if (pv_alloc_ptr == nullptr) {
return IA_FATAL_ERROR;
}
mDrcMemoryVec.push_back(pv_alloc_ptr);
/* Set the buffer pointer */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_MEM_PTR, i,
pv_alloc_ptr);
}
WORD32 ui_size;
ui_size = 8192 * 2;
mDrcInBuf = (int8_t*)malloc(ui_size);
if (mDrcInBuf == nullptr) {
return IA_FATAL_ERROR;
}
mDrcMemoryVec.push_back(mDrcInBuf);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_MEM_PTR, 2, mDrcInBuf);
mDrcOutBuf = (int8_t*)malloc(ui_size);
if (mDrcOutBuf == nullptr) {
return IA_FATAL_ERROR;
}
mDrcMemoryVec.push_back(mDrcOutBuf);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_MEM_PTR, 3, mDrcOutBuf);
return IA_NO_ERROR;
}
int Codec::configMPEGDDrc() {
IA_ERRORCODE err_code = IA_NO_ERROR;
int i_effect_type;
int i_loud_norm;
int i_target_loudness;
unsigned int i_sbr_mode;
uint32_t ui_proc_mem_tabs_size = 0;
pVOID pv_alloc_ptr = NULL;
/* Sampling Frequency */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_SAMP_FREQ, &mSampFreq);
/* Total Number of Channels */
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_NUM_CHANNELS, &mNumChannels);
/* PCM word size */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_PCM_WDSZ, &mPcmWdSz);
/*Set Effect Type*/
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_EFFECT_TYPE,
&i_effect_type);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_EFFECT_TYPE, &i_effect_type);
/*Set target loudness */
err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_TARGET_LOUDNESS, &i_target_loudness);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_TARGET_LOUDNESS, &i_target_loudness);
/*Set loud_norm_flag*/
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_LOUD_NORM,
&i_loud_norm);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_LOUD_NORM, &i_loud_norm);
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_SBR_MODE, &i_sbr_mode);
/* Get memory info tables size */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_GET_MEMTABS_SIZE, 0,
&ui_proc_mem_tabs_size);
pv_alloc_ptr = malloc(ui_proc_mem_tabs_size);
if (pv_alloc_ptr == NULL) {
return IA_FATAL_ERROR;
}
memset(pv_alloc_ptr, 0, ui_proc_mem_tabs_size);
mMemoryVec.push_back(pv_alloc_ptr);
/* Set pointer for process memory tables */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_MEMTABS_PTR, 0,
pv_alloc_ptr);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_API_POST_CONFIG_PARAMS, nullptr);
/* Free any memory that is allocated for MPEG D Drc so far */
deInitMPEGDDDrc();
err_code = initMPEGDDDrc();
if (err_code != IA_NO_ERROR) {
deInitMPEGDDDrc();
return err_code;
}
/* DRC buffers
buf[0] - contains extension element pay load loudness related
buf[1] - contains extension element pay load*/
{
VOID* p_array[2][16];
WORD32 ii;
WORD32 buf_sizes[2][16];
WORD32 num_elements;
WORD32 num_config_ext;
WORD32 bit_str_fmt = 1;
WORD32 uo_num_chan;
memset(buf_sizes, 0, 32 * sizeof(WORD32));
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_EXT_ELE_BUF_SIZES,
&buf_sizes[0][0]);
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_EXT_ELE_PTR, &p_array);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_SET_BUFF_PTR, nullptr);
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_NUM_ELE, &num_elements);
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_NUM_CONFIG_EXT,
&num_config_ext);
for (ii = 0; ii < num_config_ext; ii++) {
/*copy loudness bitstream*/
if (buf_sizes[0][ii] > 0) {
memcpy(mDrcInBuf, p_array[0][ii], buf_sizes[0][ii]);
/*Set bitstream_split_format */
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_BITS_FORMAT, &bit_str_fmt);
/* Set number of bytes to be processed */
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_INPUT_BYTES_IL_BS, 0,
&buf_sizes[0][ii]);
/* Execute process */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_CPY_IL_BSF_BUFF, nullptr);
mDRCFlag = 1;
}
}
for (ii = 0; ii < num_elements; ii++) {
/*copy config bitstream*/
if (buf_sizes[1][ii] > 0) {
memcpy(mDrcInBuf, p_array[1][ii], buf_sizes[1][ii]);
/* Set number of bytes to be processed */
/*Set bitstream_split_format */
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_BITS_FORMAT, &bit_str_fmt);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_INPUT_BYTES_IC_BS, 0,
&buf_sizes[1][ii]);
/* Execute process */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_CPY_IC_BSF_BUFF, nullptr);
mDRCFlag = 1;
}
}
if (mDRCFlag == 1) {
mMpegDDRCPresent = 1;
} else {
mMpegDDRCPresent = 0;
}
/*Read interface buffer config file bitstream*/
if (mMpegDDRCPresent == 1) {
WORD32 interface_is_present = 1;
if (i_sbr_mode != 0) {
if (i_sbr_mode == 1) {
mOutputFrameLength = 2048;
} else if (i_sbr_mode == 3) {
mOutputFrameLength = 4096;
} else {
mOutputFrameLength = 1024;
}
} else {
mOutputFrameLength = 4096;
}
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_FRAME_SIZE,
(WORD32*)&mOutputFrameLength);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_INT_PRESENT,
&interface_is_present);
/* Execute process */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_CPY_IN_BSF_BUFF, nullptr);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_PROCESS, nullptr);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_NUM_CHANNELS, &uo_num_chan);
}
}
return err_code;
}
IA_ERRORCODE Codec::initDecoder(const uint8_t* data, size_t size, bool isADTS) {
IA_ERRORCODE err_code = IA_NO_ERROR;
err_code = initXAACDecoder(isADTS);
if (err_code != IA_NO_ERROR) {
/* Call deInit to free any allocated memory */
deInitXAACDecoder();
return IA_FATAL_ERROR;
}
err_code = initXAACDrc(data, size);
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::decodeXAACStream(uint8_t* inBuffer, uint32_t inBufferLength,
int32_t* bytesConsumed,
int32_t* outBytes) {
if (mInputBufferSize < inBufferLength) {
inBufferLength = mInputBufferSize;
}
/* If codec is not initialized, call configXAACDecoder decoder again */
if (!mIsCodecInitialized) {
configXAACDecoder(inBuffer, inBufferLength, bytesConsumed);
}
/* Copy the buffer passed by Android plugin to codec input buffer */
memcpy(mInputBuffer, inBuffer, inBufferLength);
/* Set number of bytes to be processed */
IA_ERRORCODE err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_SET_INPUT_BYTES, 0, &inBufferLength);
/* Execute process */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_EXECUTE,
IA_CMD_TYPE_DO_EXECUTE, nullptr);
/* Checking for end of processing */
uint32_t ui_exec_done;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_EXECUTE,
IA_CMD_TYPE_DONE_QUERY, &ui_exec_done);
if (ui_exec_done != 1) {
VOID* p_array; // ITTIAM:buffer to handle gain payload
WORD32 buf_size = 0; // ITTIAM:gain payload length
WORD32 bit_str_fmt = 1;
WORD32 gain_stream_flag = 1;
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_GAIN_PAYLOAD_LEN, &buf_size);
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_GAIN_PAYLOAD_BUF, &p_array);
if (buf_size > 0) {
/*Set bitstream_split_format */
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_PARAM_BITS_FORMAT, &bit_str_fmt);
memcpy(mDrcInBuf, p_array, buf_size);
/* Set number of bytes to be processed */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_INPUT_BYTES_BS,
0, &buf_size);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_GAIN_STREAM_FLAG, &gain_stream_flag);
/* Execute process */
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_INIT,
IA_CMD_TYPE_INIT_CPY_BSF_BUFF, nullptr);
mMpegDDRCPresent = 1;
}
}
/* How much buffer is used in input buffers */
err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_GET_CURIDX_INPUT_BUF, 0, bytesConsumed);
/* Get the output bytes */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_OUTPUT_BYTES,
0, outBytes);
if (mMpegDDRCPresent == 1) {
memcpy(mDrcInBuf, mOutputBuffer, *outBytes);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_INPUT_BYTES, 0,
outBytes);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_EXECUTE,
IA_CMD_TYPE_DO_EXECUTE, nullptr);
memcpy(mOutputBuffer, mDrcOutBuf, *outBytes);
}
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::getXAACStreamInfo() {
IA_ERRORCODE err_code = IA_NO_ERROR;
/* Sampling frequency */
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_SAMP_FREQ, &mSampFreq);
/* Total Number of Channels */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_NUM_CHANNELS,
&mNumChannels);
if (mNumChannels > MAX_CHANNEL_COUNT) {
return IA_FATAL_ERROR;
}
/* PCM word size */
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_PCM_WDSZ, &mPcmWdSz);
if ((mPcmWdSz / 8) != 2) {
return IA_FATAL_ERROR;
}
/* channel mask to tell the arrangement of channels in bit stream */
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_CHANNEL_MASK,
&mChannelMask);
/* Channel mode to tell MONO/STEREO/DUAL-MONO/NONE_OF_THESE */
uint32_t ui_channel_mode;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_CHANNEL_MODE,
&ui_channel_mode);
/* Channel mode to tell SBR PRESENT/NOT_PRESENT */
uint32_t ui_sbr_mode;
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_SBR_MODE, &ui_sbr_mode);
/* mOutputFrameLength = 1024 * (1 + SBR_MODE) for AAC */
/* For USAC it could be 1024 * 3 , support to query */
/* not yet added in codec */
mOutputFrameLength = 1024 * (1 + ui_sbr_mode);
return IA_NO_ERROR;
}
IA_ERRORCODE Codec::setXAACDRCInfo(int32_t drcCut, int32_t drcBoost,
int32_t drcRefLevel,
int32_t drcHeavyCompression,
int32_t drEffectType) {
IA_ERRORCODE err_code = IA_NO_ERROR;
int32_t ui_drc_enable = 1;
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_ENABLE,
&ui_drc_enable);
if (drcCut != -1) {
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_CUT, &drcCut);
}
if (drcBoost != -1) {
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_BOOST, &drcBoost);
}
if (drcRefLevel != -1) {
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_TARGET_LEVEL,
&drcRefLevel);
}
if (drcRefLevel != -1) {
err_code =
ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_DRC_TARGET_LOUDNESS, &drcRefLevel);
}
if (drcHeavyCompression != -1) {
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_HEAVY_COMP,
&drcHeavyCompression);
}
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_DRC_EFFECT_TYPE, &drEffectType);
int32_t i_effect_type, i_target_loudness, i_loud_norm;
/*Set Effect Type*/
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_EFFECT_TYPE,
&i_effect_type);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_EFFECT_TYPE, &i_effect_type);
/*Set target loudness */
err_code = ixheaacd_dec_api(
mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_TARGET_LOUDNESS, &i_target_loudness);
err_code =
ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_TARGET_LOUDNESS, &i_target_loudness);
/*Set loud_norm_flag*/
err_code = ixheaacd_dec_api(mXheaacCodecHandle, IA_API_CMD_GET_CONFIG_PARAM,
IA_ENHAACPLUS_DEC_CONFIG_PARAM_DRC_LOUD_NORM,
&i_loud_norm);
err_code = ia_drc_dec_api(mMpegDDrcHandle, IA_API_CMD_SET_CONFIG_PARAM,
IA_DRC_DEC_CONFIG_DRC_LOUD_NORM, &i_loud_norm);
return IA_NO_ERROR;
}
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
int status;
if (size < 1) return 0;
Codec* codec = new Codec();
bool isADTS = false;
if (size >= 2) {
if ((data[0] == 0xFF) && ((data[1] & 0xF0) == 0xF0)) {
isADTS = true;
}
}
status = codec->initDecoder(data, size, isADTS);
if (0 == status) {
int32_t bytesConsumed = 0;
status = codec->configXAACDecoder((uint8_t*)data, size, &bytesConsumed);
while ((int32_t)size > bytesConsumed) {
int32_t numOutBytes;
size -= bytesConsumed;
data += bytesConsumed;
status = codec->decodeXAACStream((uint8_t*)data, size, &bytesConsumed,
&numOutBytes);
/* If decoder doesn't consume any bytes, advance by 4 bytes */
if (0 == bytesConsumed) bytesConsumed = 4;
}
}
status = codec->deInitXAACDecoder();
status = codec->deInitMPEGDDDrc();
delete codec;
return 0;
}
Reference in new issue View Git Blame Copy Permalink