1*6897da5cSDirk Helbig /******************************************************************************
2*6897da5cSDirk Helbig *
3*6897da5cSDirk Helbig * Copyright 2022 Google LLC
4*6897da5cSDirk Helbig *
5*6897da5cSDirk Helbig * Licensed under the Apache License, Version 2.0 (the "License");
6*6897da5cSDirk Helbig * you may not use this file except in compliance with the License.
7*6897da5cSDirk Helbig * You may obtain a copy of the License at:
8*6897da5cSDirk Helbig *
9*6897da5cSDirk Helbig * http://www.apache.org/licenses/LICENSE-2.0
10*6897da5cSDirk Helbig *
11*6897da5cSDirk Helbig * Unless required by applicable law or agreed to in writing, software
12*6897da5cSDirk Helbig * distributed under the License is distributed on an "AS IS" BASIS,
13*6897da5cSDirk Helbig * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*6897da5cSDirk Helbig * See the License for the specific language governing permissions and
15*6897da5cSDirk Helbig * limitations under the License.
16*6897da5cSDirk Helbig *
17*6897da5cSDirk Helbig ******************************************************************************/
18*6897da5cSDirk Helbig
19*6897da5cSDirk Helbig #include <lc3_cpp.h>
20*6897da5cSDirk Helbig #include <fuzzer/FuzzedDataProvider.h>
21*6897da5cSDirk Helbig
22*6897da5cSDirk Helbig using namespace lc3;
23*6897da5cSDirk Helbig
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)24*6897da5cSDirk Helbig extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
25*6897da5cSDirk Helbig {
26*6897da5cSDirk Helbig const int dt_list[] = { 2500, 5000, 7500, 10000 };
27*6897da5cSDirk Helbig const int sr_list[] = { 8000, 16000, 24000, 32000, 48000, 96000 };
28*6897da5cSDirk Helbig
29*6897da5cSDirk Helbig FuzzedDataProvider fdp(data, size);
30*6897da5cSDirk Helbig
31*6897da5cSDirk Helbig int dt_us = fdp.PickValueInArray(dt_list);
32*6897da5cSDirk Helbig int sr_hz = fdp.PickValueInArray(sr_list);
33*6897da5cSDirk Helbig int nchannels =fdp.PickValueInArray({1, 2});
34*6897da5cSDirk Helbig bool hrmode = fdp.ConsumeBool();
35*6897da5cSDirk Helbig
36*6897da5cSDirk Helbig int sr_pcm_hz = fdp.PickValueInArray(sr_list);
37*6897da5cSDirk Helbig if (sr_pcm_hz < sr_hz)
38*6897da5cSDirk Helbig sr_pcm_hz = 0;
39*6897da5cSDirk Helbig
40*6897da5cSDirk Helbig Decoder dec(dt_us, sr_hz, sr_pcm_hz, nchannels, hrmode);
41*6897da5cSDirk Helbig
42*6897da5cSDirk Helbig int frame_size = fdp.ConsumeIntegralInRange(
43*6897da5cSDirk Helbig LC3_MIN_FRAME_BYTES, LC3_MAX_FRAME_BYTES);
44*6897da5cSDirk Helbig
45*6897da5cSDirk Helbig PcmFormat fmt = fdp.PickValueInArray(
46*6897da5cSDirk Helbig { PcmFormat::kS16, PcmFormat::kS24,
47*6897da5cSDirk Helbig PcmFormat::kS24In3Le, PcmFormat::kF32 });
48*6897da5cSDirk Helbig
49*6897da5cSDirk Helbig int frame_samples = dec.GetFrameSamples();
50*6897da5cSDirk Helbig if (frame_samples < 0)
51*6897da5cSDirk Helbig return -1;
52*6897da5cSDirk Helbig
53*6897da5cSDirk Helbig int sample_bytes =
54*6897da5cSDirk Helbig fmt == PcmFormat::kS16 ? sizeof(int16_t) :
55*6897da5cSDirk Helbig fmt == PcmFormat::kS24 ? sizeof(int32_t) :
56*6897da5cSDirk Helbig fmt == PcmFormat::kS24In3Le ? sizeof(uint8_t) * 3 :
57*6897da5cSDirk Helbig fmt == PcmFormat::kF32 ? sizeof(float) : 0;
58*6897da5cSDirk Helbig
59*6897da5cSDirk Helbig if (fdp.remaining_bytes() < frame_size * nchannels)
60*6897da5cSDirk Helbig return -1;
61*6897da5cSDirk Helbig
62*6897da5cSDirk Helbig dec.Decode(
63*6897da5cSDirk Helbig fdp.ConsumeBytes<uint8_t>(nchannels * frame_size).data(), frame_size,
64*6897da5cSDirk Helbig fmt, std::vector<uint8_t>(nchannels * frame_samples * sample_bytes).data());
65*6897da5cSDirk Helbig
66*6897da5cSDirk Helbig return 0;
67*6897da5cSDirk Helbig }
68