111 lines
3.5 KiB
C++
111 lines
3.5 KiB
C++
|
// Copyright 2019 The Chromium Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <stddef.h>
|
||
|
|
#include "ldacBT.h"
|
||
|
|
|
||
|
|
#include <fuzzer/FuzzedDataProvider.h>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
constexpr LDACBT_SMPL_FMT_T kFormat[] = {
|
||
|
|
LDACBT_SMPL_FMT_S16, LDACBT_SMPL_FMT_S24, LDACBT_SMPL_FMT_S32,
|
||
|
|
LDACBT_SMPL_FMT_F32};
|
||
|
|
constexpr int32_t kEqmidValue[] = {LDACBT_EQMID_HQ, LDACBT_EQMID_SQ,
|
||
|
|
LDACBT_EQMID_MQ};
|
||
|
|
constexpr int32_t kChannel[] = {LDACBT_CHANNEL_MODE_STEREO,
|
||
|
|
LDACBT_CHANNEL_MODE_DUAL_CHANNEL,
|
||
|
|
LDACBT_CHANNEL_MODE_MONO};
|
||
|
|
constexpr int32_t kSamplingFrequency[] = {44100, 48000, 2 * 44100, 2 * 48000};
|
||
|
|
constexpr int32_t kLdacBtRequiredMtu = 679;
|
||
|
|
constexpr int32_t kMaxWlValue = 4;
|
||
|
|
constexpr int32_t kMinChValue = 1;
|
||
|
|
constexpr int32_t kMaxChValue = 2;
|
||
|
|
constexpr int32_t kOutputSize = 1024;
|
||
|
|
|
||
|
|
class Codec {
|
||
|
|
public:
|
||
|
|
Codec(const uint8_t *data, size_t size) : mFdp(data, size){};
|
||
|
|
~Codec() { deInitEncoder(); }
|
||
|
|
bool initEncoder();
|
||
|
|
void deInitEncoder();
|
||
|
|
void encodeFrames(const uint8_t *data, size_t size);
|
||
|
|
void getSamplingFrequeny();
|
||
|
|
void getBitrate();
|
||
|
|
void getVersion();
|
||
|
|
FuzzedDataProvider mFdp;
|
||
|
|
|
||
|
|
private:
|
||
|
|
int32_t mChannel;
|
||
|
|
LDACBT_SMPL_FMT_T mFormat;
|
||
|
|
HANDLE_LDAC_BT mHandleLdacBt;
|
||
|
|
};
|
||
|
|
|
||
|
|
bool Codec::initEncoder() {
|
||
|
|
mHandleLdacBt = ldacBT_get_handle();
|
||
|
|
|
||
|
|
mChannel = mFdp.PickValueInArray(kChannel);
|
||
|
|
mFormat = mFdp.PickValueInArray(kFormat);
|
||
|
|
int32_t status = ldacBT_init_handle_encode(
|
||
|
|
mHandleLdacBt, kLdacBtRequiredMtu, mFdp.PickValueInArray(kEqmidValue),
|
||
|
|
mChannel, mFormat, mFdp.PickValueInArray(kSamplingFrequency));
|
||
|
|
if (!status) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Codec::encodeFrames(const uint8_t *data, size_t size) {
|
||
|
|
int32_t pcmUsed, streamSize, frameNum;
|
||
|
|
unsigned char pStream[kOutputSize];
|
||
|
|
int32_t ch, wl, frameSize;
|
||
|
|
ch = (mChannel == LDAC_CCI_MONO) ? kMinChValue : kMaxChValue;
|
||
|
|
wl = mFormat > kMaxWlValue ? kMaxWlValue : mFormat;
|
||
|
|
frameSize = LDACBT_ENC_LSU * ch * wl;
|
||
|
|
std::vector<uint8_t> tmpData(frameSize);
|
||
|
|
uint8_t *readPointer = const_cast<uint8_t *>(data);
|
||
|
|
while (size > 0) {
|
||
|
|
if (size < frameSize) {
|
||
|
|
memcpy(tmpData.data(), data, size);
|
||
|
|
size = frameSize;
|
||
|
|
readPointer = tmpData.data();
|
||
|
|
}
|
||
|
|
ldacBT_encode(mHandleLdacBt, readPointer, &pcmUsed, pStream, &streamSize,
|
||
|
|
&frameNum);
|
||
|
|
readPointer += frameSize;
|
||
|
|
size -= frameSize;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void Codec::getSamplingFrequeny() { ldacBT_get_sampling_freq(mHandleLdacBt); }
|
||
|
|
|
||
|
|
void Codec::getBitrate() { ldacBT_get_bitrate(mHandleLdacBt); }
|
||
|
|
|
||
|
|
void Codec::getVersion() { ldacBT_get_version(); }
|
||
|
|
|
||
|
|
void Codec::deInitEncoder() {
|
||
|
|
ldacBT_close_handle(mHandleLdacBt);
|
||
|
|
ldacBT_free_handle(mHandleLdacBt);
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *buf, size_t size) {
|
||
|
|
Codec codec(buf, size);
|
||
|
|
/* initEncoder() initializes mHandleLdacBt and is supposed to be called before
|
||
|
|
encoding begins. Hence initEncoder() is not included in the lambda
|
||
|
|
construct. */
|
||
|
|
if (codec.initEncoder()) {
|
||
|
|
while (codec.mFdp.remaining_bytes() > 0) {
|
||
|
|
auto executeFunction =
|
||
|
|
codec.mFdp.PickValueInArray<const std::function<void()>>({
|
||
|
|
[&]() { codec.encodeFrames(buf, size); },
|
||
|
|
[&]() { codec.getSamplingFrequeny(); },
|
||
|
|
[&]() { codec.getBitrate(); },
|
||
|
|
[&]() { codec.getVersion(); },
|
||
|
|
});
|
||
|
|
executeFunction();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|