diff --git a/test/fuzztest/distributedinputkit_fuzzer/distributed_input_kit_fuzzer.cpp b/test/fuzztest/distributedinputkit_fuzzer/distributed_input_kit_fuzzer.cpp index 1b7d578..fafd9e4 100644 --- a/test/fuzztest/distributedinputkit_fuzzer/distributed_input_kit_fuzzer.cpp +++ b/test/fuzztest/distributedinputkit_fuzzer/distributed_input_kit_fuzzer.cpp @@ -126,10 +126,11 @@ void StartRemoteInputFuzzTest(const uint8_t* data, size_t size) std::vector dhIds= {}; OHOS::sptr startCb = new (std::nothrow) TestStartDInputCallback(); OHOS::sptr stopCb = new (std::nothrow) TestStopDInputCallback(); + DistributedInput::DistributedInputKit::StartRemoteInput(sinkId, inputTypes, startCb); DistributedInput::DistributedInputKit::StartRemoteInput(srcId, sinkId, inputTypes, startCb); - DistributedInput::DistributedInputKit::StopRemoteInput(srcId, sinkId, inputTypes, stopCb); OHOS::sptr callback = new (std::nothrow) TestIStartStopDInputsCallback(); DistributedInput::DistributedInputKit::StartRemoteInput(sinkId, dhIds, callback); + DistributedInput::DistributedInputKit::StartRemoteInput(srcId, sinkId, dhIds, callback); } void IsNeedFilterOutFuzzTest(const uint8_t* data, size_t size) @@ -165,6 +166,7 @@ void StopRemoteInputFuzzTest(const uint8_t* data, size_t size) DistributedInput::DistributedInputKit::StopRemoteInput(srcId, sinkId, inputTypes, stopCb); OHOS::sptr callback = new (std::nothrow) TestIStartStopDInputsCallback(); DistributedInput::DistributedInputKit::StopRemoteInput(sinkId, dhIds, callback); + DistributedInput::DistributedInputKit::StopRemoteInput(srcId, sinkId, dhIds, callback); } void IsTouchEventNeedFilterOutFuzzTest(const uint8_t* data, size_t size) diff --git a/test/fuzztest/distributedinputsinktransport_fuzzer/distributed_input_sink_transport_fuzzer.cpp b/test/fuzztest/distributedinputsinktransport_fuzzer/distributed_input_sink_transport_fuzzer.cpp index 4382bb9..14658ab 100644 --- a/test/fuzztest/distributedinputsinktransport_fuzzer/distributed_input_sink_transport_fuzzer.cpp +++ b/test/fuzztest/distributedinputsinktransport_fuzzer/distributed_input_sink_transport_fuzzer.cpp @@ -64,6 +64,18 @@ void OnSessionOpenedFuzzTest(const uint8_t* data, size_t size) DistributedInput::DistributedInputSinkTransport::GetInstance().OnSessionOpened(sessionId, result); DistributedInput::DistributedInputSinkTransport::GetInstance().OnSessionClosed(sessionId); } + +void OnBytesReceivedFuzzTest(const uint8_t* data, size_t size) +{ + if ((data == nullptr) || (size <= 0)) { + return; + } + + int32_t sessionId = *(reinterpret_cast(data)); + const char *msg = reinterpret_cast(data); + uint16_t dataLen = *(reinterpret_cast(data)); + DistributedInput::DistributedInputSinkTransport::GetInstance().OnBytesReceived(sessionId, msg, dataLen); +} } // namespace DistributedHardware } // namespace OHOS @@ -74,5 +86,6 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) OHOS::DistributedHardware::RespPrepareRemoteInputFuzzTest(data, size); OHOS::DistributedHardware::RespStartRemoteInputFuzzTest(data, size); OHOS::DistributedHardware::OnSessionOpenedFuzzTest(data, size); + OHOS::DistributedHardware::OnBytesReceivedFuzzTest(data, size); return 0; } \ No newline at end of file diff --git a/test/fuzztest/distributedinputsourcetransport_fuzzer/distributed_input_source_transport_fuzzer.cpp b/test/fuzztest/distributedinputsourcetransport_fuzzer/distributed_input_source_transport_fuzzer.cpp index e70694f..9df1f31 100644 --- a/test/fuzztest/distributedinputsourcetransport_fuzzer/distributed_input_source_transport_fuzzer.cpp +++ b/test/fuzztest/distributedinputsourcetransport_fuzzer/distributed_input_source_transport_fuzzer.cpp @@ -43,6 +43,8 @@ void OpenInputSoftbusFuzzTest(const uint8_t* data, size_t size) usleep(sleepTimeUs); DistributedInput::DistributedInputSourceTransport::GetInstance().OpenInputSoftbus(remoteDevId); DistributedInput::DistributedInputSourceTransport::GetInstance().CloseInputSoftbus(sessionId); + DistributedInput::DistributedInputSourceTransport::GetInstance().OpenInputSoftbusForRelay(remoteDevId); + DistributedInput::DistributedInputSourceTransport::GetInstance().CloseInputSoftbus(sessionId); } void OnSessionOpenedFuzzTest(const uint8_t* data, size_t size) @@ -57,6 +59,18 @@ void OnSessionOpenedFuzzTest(const uint8_t* data, size_t size) DistributedInput::DistributedInputSourceTransport::GetInstance().OnSessionOpened(sessionId, result); DistributedInput::DistributedInputSourceTransport::GetInstance().OnSessionClosed(sessionId); } + +void OnBytesReceivedFuzzTest(const uint8_t* data, size_t size) +{ + if ((data == nullptr) || (size <= 0)) { + return; + } + + int32_t sessionId = *(reinterpret_cast(data)); + const char *msg = reinterpret_cast(data); + uint16_t dataLen = *(reinterpret_cast(data)); + DistributedInput::DistributedInputSourceTransport::GetInstance().OnBytesReceived(sessionId, msg, dataLen); +} } // namespace DistributedHardware } // namespace OHOS @@ -67,5 +81,6 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) /* Run your code on data */ OHOS::DistributedHardware::OpenInputSoftbusFuzzTest(data, size); OHOS::DistributedHardware::OnSessionOpenedFuzzTest(data, size); + OHOS::DistributedHardware::OnBytesReceivedFuzzTest(data, size); return 0; } \ No newline at end of file