diff --git a/wifi/test/fuzztest/frameworks/native/c_adapter/wifichotspot_fuzzer/wifichotspot_fuzzer.cpp b/wifi/test/fuzztest/frameworks/native/c_adapter/wifichotspot_fuzzer/wifichotspot_fuzzer.cpp index 3760b5844fa85edbab6323b3f27de8fdc789b2fc..275eed2e98aa9143e0a9f971115991299eff90b0 100644 --- a/wifi/test/fuzztest/frameworks/native/c_adapter/wifichotspot_fuzzer/wifichotspot_fuzzer.cpp +++ b/wifi/test/fuzztest/frameworks/native/c_adapter/wifichotspot_fuzzer/wifichotspot_fuzzer.cpp @@ -19,6 +19,7 @@ #include "wifichotspot_fuzzer.h" #include "wifi_fuzz_common_func.h" #include "kits/c/wifi_hotspot.h" +#include static void SetHotspotConfigTest(const uint8_t* data, size_t size) { @@ -93,6 +94,9 @@ static void GetHotspotModeTest(const uint8_t* data, size_t size) { int mode = 3; // 3: HotspotMode::LOCAL_ONLY_SOFTAP (void)GetHotspotMode(&mode); + FuzzedDataProvider fdp(data, size); + mode = fdp.ConsumeIntegral(); + (void)GetHotspotMode(&mode); } static void GetLocalOnlyHotspotConfigTest(const uint8_t* data, size_t size) diff --git a/wifi/test/fuzztest/frameworks/native/src/wifip2pimpl_fuzzer/wifip2pimpl_fuzzer.cpp b/wifi/test/fuzztest/frameworks/native/src/wifip2pimpl_fuzzer/wifip2pimpl_fuzzer.cpp index d4f1c86d75e35a3abbe95a5982579508db2a75b4..1e5d7cb1e8e8d01747650aa618dec57150257c21 100644 --- a/wifi/test/fuzztest/frameworks/native/src/wifip2pimpl_fuzzer/wifip2pimpl_fuzzer.cpp +++ b/wifi/test/fuzztest/frameworks/native/src/wifip2pimpl_fuzzer/wifip2pimpl_fuzzer.cpp @@ -20,6 +20,7 @@ #include "wifi_fuzz_common_func.h" #include "src/wifi_p2p_impl.h" #include "wifi_p2p.h" +#include namespace OHOS { namespace Wifi { @@ -96,6 +97,13 @@ namespace Wifi { { std::vector services; WifiP2pPtr->QueryP2pServices(services); + FuzzedDataProvider fdp(data, size); + int vectorLength = fdp.ConsumeIntegral(); + for (int i = 0; i < vectorLength; i++) { + WifiP2pServiceInfo servicetmp; + services.push_back(servicetmp); + } + WifiP2pPtr->QueryP2pServices(services); } void GetSupportedFeaturesTest(const uint8_t* data, size_t size)