Files
liuxiaowei42 9f911dc894 add fuzz
Signed-off-by: liuxiaowei42 <liuxiaowei42@huawei.com>
2025-06-26 16:26:37 +08:00

90 lines
3.2 KiB
C++

/*
* Copyright (c) 2023-2025 Huawei Device Co., Ltd.
* 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.
*/
#include "dinputinitsink_fuzzer.h"
#include <cstddef>
#include <cstdint>
#include <functional>
#include <iostream>
#include <thread>
#include <unistd.h>
#include <refbase.h>
#include "constants_dinput.h"
#include "distributed_input_handler.h"
#include "distributed_input_kit.h"
#include "distributed_input_sink_handler.h"
#include "i_distributed_sink_input.h"
namespace OHOS {
namespace DistributedHardware {
void InitSinkFuzzTest(const uint8_t *data, size_t size)
{
if ((data == nullptr) || (size == 0)) {
return;
}
std::string params(reinterpret_cast<const char*>(data), size);
DistributedInput::DistributedInputSinkHandler::GetInstance().InitSink(params);
const uint32_t sleepTimeUs = 50 * 1000;
usleep(sleepTimeUs);
}
void OnLoadSystemAbilitySuccessFuzzTest(const uint8_t *data, size_t size)
{
if ((data == nullptr) || (size < sizeof(int32_t))) {
return;
}
int32_t saId = *(reinterpret_cast<const int32_t*>(data));
DistributedInput::DistributedInputSinkHandler::SALoadSinkCb saLoadSinkCb;
saLoadSinkCb.OnLoadSystemAbilitySuccess(saId, nullptr);
}
void OnLoadSystemAbilityFailFuzzTest(const uint8_t *data, size_t size)
{
if ((data == nullptr) || (size < sizeof(int32_t))) {
return;
}
int32_t saId = *(reinterpret_cast<const int32_t*>(data));
DistributedInput::DistributedInputSinkHandler::SALoadSinkCb saLoadSinkCb;
saLoadSinkCb.OnLoadSystemAbilityFail(saId);
}
void OnRemoteSinkSvrDiedFuzzTest(const uint8_t *data, size_t size)
{
if ((data == nullptr) || (size == 0)) {
return;
}
std::string networkId(reinterpret_cast<const char*>(data), size);
DistributedInput::DistributedInputSinkHandler::GetInstance().OnRemoteSinkSvrDied(nullptr);
DistributedInput::DistributedInputSinkHandler::GetInstance().RegisterPrivacyResources(nullptr);
DistributedInput::DistributedInputSinkHandler::GetInstance().PauseDistributedHardware(networkId);
DistributedInput::DistributedInputSinkHandler::GetInstance().ResumeDistributedHardware(networkId);
DistributedInput::DistributedInputSinkHandler::GetInstance().StopDistributedHardware(networkId);
}
} // namespace DistributedHardware
} // namespace OHOS
/* Fuzzer entry point */
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
/* Run your code on data */
OHOS::DistributedHardware::InitSinkFuzzTest(data, size);
OHOS::DistributedHardware::OnLoadSystemAbilitySuccessFuzzTest(data, size);
OHOS::DistributedHardware::OnLoadSystemAbilityFailFuzzTest(data, size);
OHOS::DistributedHardware::OnRemoteSinkSvrDiedFuzzTest(data, size);
return 0;
}