1 /*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "acodecconfigure_fuzzer.h"
17 #include <iostream>
18 #include "string_ex.h"
19 #include "media_errors.h"
20 #include "directory_ex.h"
21 #include "ui/rs_surface_node.h"
22 #include "window_option.h"
23 #include "test_params_config.h"
24 #include "aw_common.h"
25
26 using namespace std;
27 using namespace OHOS;
28 using namespace OHOS::Media;
29 using namespace OHOS::Media::ACodecTestParam;
30
ACodecConfigureFuzzer()31 ACodecConfigureFuzzer::ACodecConfigureFuzzer()
32 {
33 }
34
~ACodecConfigureFuzzer()35 ACodecConfigureFuzzer::~ACodecConfigureFuzzer()
36 {
37 }
38
FuzzAudioConfigure(uint8_t * data,size_t size)39 bool ACodecConfigureFuzzer::FuzzAudioConfigure(uint8_t *data, size_t size)
40 {
41 constexpr int32_t WAITTING_TIME = 2;
42 std::shared_ptr<ACodecSignal> acodecSignal = std::make_shared<ACodecSignal>();
43 adecCallback_ = std::make_shared<ADecCallbackTest>(acodecSignal);
44 CHECK_INSTANCE_AND_RETURN_RET(adecCallback_, false);
45 aencCallback_ = std::make_shared<AEncCallbackTest>(acodecSignal);
46 CHECK_INSTANCE_AND_RETURN_RET(aencCallback_, false);
47 audioCodec_ = std::make_shared<ACodecMock>(acodecSignal);
48 CHECK_INSTANCE_AND_RETURN_RET(audioCodec_, false);
49 CHECK_BOOL_AND_RETURN_RET(audioCodec_->CreateAudioEncMockByMime("audio/mp4a-latm"), false);
50 CHECK_STATE_AND_RETURN_RET(audioCodec_->SetCallbackEnc(aencCallback_), false);
51 CHECK_BOOL_AND_RETURN_RET(audioCodec_->CreateAudioDecMockByMime("audio/mp4a-latm"), false);
52 CHECK_STATE_AND_RETURN_RET(audioCodec_->SetCallbackDec(adecCallback_), false);
53
54 defaultFormat_ = AVCodecMockFactory::CreateFormat();
55 CHECK_INSTANCE_AND_RETURN_RET(defaultFormat_, false);
56 int32_t data_ = *reinterpret_cast<int32_t *>(data);
57 (void)defaultFormat_->PutIntValue("channel_count", 2); // 2 common channel count
58 (void)defaultFormat_->PutIntValue("sample_rate", data_); // fuzz sample rate
59 (void)defaultFormat_->PutIntValue("audio_sample_format", 1); // 1 AudioStandard::SAMPLE_S16LE
60
61 audioCodec_->SetOutPath("/data/test/media/aac_configurefuzz_out.es");
62 audioCodec_->ConfigureEnc(defaultFormat_);
63 audioCodec_->ConfigureDec(defaultFormat_);
64 audioCodec_->PrepareDec();
65 audioCodec_->PrepareEnc();
66 audioCodec_->StartDec();
67 audioCodec_->StartEnc();
68 sleep(WAITTING_TIME);
69 audioCodec_->FlushDec();
70 audioCodec_->FlushEnc();
71 audioCodec_->StopDec();
72 audioCodec_->StopEnc();
73 audioCodec_->ResetDec();
74 audioCodec_->ResetEnc();
75 if (audioCodec_ != nullptr) {
76 CHECK_STATE_AND_RETURN_RET(audioCodec_->ReleaseDec(), false);
77 CHECK_STATE_AND_RETURN_RET(audioCodec_->ReleaseEnc(), false);
78 }
79 return true;
80 }
81
FuzzACodecConfigure(uint8_t * data,size_t size)82 bool OHOS::Media::FuzzACodecConfigure(uint8_t *data, size_t size)
83 {
84 auto codecfuzzer = std::make_unique<ACodecConfigureFuzzer>();
85 if (codecfuzzer == nullptr) {
86 cout << "codecfuzzer is null" << endl;
87 return 0;
88 }
89 if (size < sizeof(int32_t)) {
90 return 0;
91 }
92 return codecfuzzer->FuzzAudioConfigure(data, size);
93 }
94
95 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(uint8_t * data,size_t size)96 extern "C" int LLVMFuzzerTestOneInput(uint8_t *data, size_t size)
97 {
98 /* Run your code on data */
99 OHOS::Media::FuzzACodecConfigure(data, size);
100 return 0;
101 }
102
103