add wifi fuzzer testcases

Signed-off-by: z00588131 <hw.zhangfeng@huawei.com>

wifi/bundle.json

@@ -105,6 +105,7 @@
                 }
             ],
             "test": [
+                "//foundation/communication/wifi/wifi/test/fuzztest/wifi_sta:fuzztest"
             ]
         },
         "hisysevent_config": [

wifi/test/fuzztest/fuzz_common_func/wifi_fuzz_common_func.h

@@ -0,0 +1,35 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+#ifndef WIFI_FUZZ_COMMON_FUNC_H_
+#define WIFI_FUZZ_COMMON_FUNC_H_
+
+#include <cstdint>
+
+namespace OHOS {
+namespace Wifi {
+inline uint16_t U16_AT(const uint8_t* data)
+{
+    return (data[0] << 8) | data[1];
+}
+
+inline uint32_t U32_AT(const uint8_t* data)
+{
+    return (data[0] << 24) | (data[1] << 16) | (data[2] << 8) | data[3];
+}
+}  // namespace Wifi
+}  // namespace OHOS
+#endif
+

wifi/test/fuzztest/wifi_sta/BUILD.gn

@@ -0,0 +1,23 @@
+# Copyright (C) 2022 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.
+
+import("//build/test.gni")
+group("fuzztest") {
+  testonly = true
+  deps = []
+
+  deps += [
+    "adddeviceconfig_fuzzer:AddDeviceConfigFuzzTest",
+    "connecttodevice_fuzzer:ConnectToDeviceFuzzTest",
+  ]
+}

wifi/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer/BUILD.gn

@@ -0,0 +1,50 @@
+# Copyright (C) 2022 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.
+
+#####################hydra-fuzz###################
+import("//build/config/features.gni")
+import("//build/test.gni")
+import("//foundation/communication/wifi/wifi/wifi.gni")
+module_output_path = "wifi/wifi_sta"
+
+##############################fuzztest##########################################
+ohos_fuzztest("AddDeviceConfigFuzzTest") {
+  module_out_path = module_output_path
+  fuzz_config_file =
+      "$WIFI_ROOT_DIR/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer"
+
+  include_dirs = [
+    "$WIFI_ROOT_DIR/frameworks/native/include",
+    "$WIFI_ROOT_DIR/frameworks/native/interfaces",
+    "$WIFI_ROOT_DIR/services/wifi_standard/wifi_framework/common/net_helper",
+    "$WIFI_ROOT_DIR/test/fuzztest/fuzz_common_func",
+  ]
+
+  cflags = [
+    "-g",
+    "-O0",
+    "-Wno-unused-variable",
+    "-fno-omit-frame-pointer",
+  ]
+  sources = [ "adddeviceconfig_fuzzer.cpp" ]
+
+  deps = [
+    "$WIFI_ROOT_DIR/frameworks/native:wifi_sdk",
+    "//utils/native/base:utils",
+  ]
+
+  external_deps = [ "ipc:ipc_core" ]
+
+  part_name = "wifi"
+  subsystem_name = "communication"
+}

wifi/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer/adddeviceconfig_fuzzer.cpp

@@ -0,0 +1,45 @@
+/*
+ * Copyright (C) 2022 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 "adddeviceconfig_fuzzer.h"
+#include "wifi_device.h"
+
+namespace OHOS {
+namespace Wifi {
+    bool AddDeviceConfigFuzzerTest(const uint8_t* data, size_t size)
+    {
+        std::unique_ptr<WifiDevice> devicePtr = WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID);
+        if (devicePtr == nullptr) {
+            return false;
+        }
+
+        WifiDeviceConfig config;
+        int addResult;
+        config.ssid = std::string(reinterpret_cast<const char*>(data), size);
+        config.bssid = std::string(reinterpret_cast<const char*>(data), size);
+        config.preSharedKey = std::string(reinterpret_cast<const char*>(data), size);
+        config.keyMgmt = std::string(reinterpret_cast<const char*>(data), size);
+        devicePtr->AddDeviceConfig(config, addResult);
+        return true;
+    }
+}  // namespace Wifi
+}  // namespace OHOS
+
+/* Fuzzer entry point */
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
+{
+    OHOS::Wifi::AddDeviceConfigFuzzerTest(data, size);
+    return 0;
+}

wifi/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer/adddeviceconfig_fuzzer.h

@@ -0,0 +1,21 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+#ifndef WIFI_FUZZ_ADD_DEVICE_CONFIG_H_
+#define WIFI_FUZZ_ADD_DEVICE_CONFIG_H_
+
+#define FUZZ_PROJECT_NAME "adddeviceconfig_fuzzer"
+
+#endif

wifi/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer/corpus/init

@@ -0,0 +1,16 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+FUZZ

wifi/test/fuzztest/wifi_sta/adddeviceconfig_fuzzer/project.xml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Copyright (C) 2022 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.
+-->
+<fuzz_config>
+  <fuzztest>
+    <!-- maximum length of a test input -->
+    <max_len>1000</max_len>
+    <!-- maximum total time in seconds to run the fuzzer -->
+    <max_total_time>300</max_total_time>
+    <!-- memory usage limit in Mb -->
+    <rss_limit_mb>4096</rss_limit_mb>
+  </fuzztest>
+</fuzz_config>

wifi/test/fuzztest/wifi_sta/connecttodevice_fuzzer/BUILD.gn

@@ -0,0 +1,50 @@
+# Copyright (C) 2022 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.
+
+#####################hydra-fuzz###################
+import("//build/config/features.gni")
+import("//build/test.gni")
+import("//foundation/communication/wifi/wifi/wifi.gni")
+module_output_path = "wifi/wifi_sta"
+
+##############################fuzztest##########################################
+ohos_fuzztest("ConnectToDeviceFuzzTest") {
+  module_out_path = module_output_path
+  fuzz_config_file =
+      "$WIFI_ROOT_DIR/test/fuzztest/wifi_sta/connecttodevice_fuzzer"
+
+  include_dirs = [
+    "$WIFI_ROOT_DIR/frameworks/native/include",
+    "$WIFI_ROOT_DIR/frameworks/native/interfaces",
+    "$WIFI_ROOT_DIR/services/wifi_standard/wifi_framework/common/net_helper",
+    "$WIFI_ROOT_DIR/test/fuzztest/fuzz_common_func",
+  ]
+
+  cflags = [
+    "-g",
+    "-O0",
+    "-Wno-unused-variable",
+    "-fno-omit-frame-pointer",
+  ]
+  sources = [ "connecttodevice_fuzzer.cpp" ]
+
+  deps = [
+    "$WIFI_ROOT_DIR/frameworks/native:wifi_sdk",
+    "//utils/native/base:utils",
+  ]
+
+  external_deps = [ "ipc:ipc_core" ]
+
+  part_name = "wifi"
+  subsystem_name = "communication"
+}

wifi/test/fuzztest/wifi_sta/connecttodevice_fuzzer/connecttodevice_fuzzer.cpp

@@ -0,0 +1,45 @@
+/*
+ * Copyright (C) 2022 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 "connecttodevice_fuzzer.h"
+#include "wifi_device.h"
+
+namespace OHOS {
+namespace Wifi {
+    bool ConnectToDeviceFuzzerTest(const uint8_t* data, size_t size)
+    {
+        std::unique_ptr<WifiDevice> devicePtr = WifiDevice::GetInstance(WIFI_DEVICE_ABILITY_ID);
+        if (devicePtr == nullptr) {
+            return false;
+        }
+
+        WifiDeviceConfig config;
+        config.ssid = std::string(reinterpret_cast<const char*>(data), size);
+        config.bssid = std::string(reinterpret_cast<const char*>(data), size);
+        config.preSharedKey = std::string(reinterpret_cast<const char*>(data), size);
+        config.keyMgmt = std::string(reinterpret_cast<const char*>(data), size);
+        devicePtr->ConnectToDevice(config);
+        return true;
+    }
+}  // namespace Wifi
+}  // namespace OHOS
+
+/* Fuzzer entry point */
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
+{
+    OHOS::Wifi::ConnectToDeviceFuzzerTest(data, size);
+    return 0;
+}
+

wifi/test/fuzztest/wifi_sta/connecttodevice_fuzzer/connecttodevice_fuzzer.h

@@ -0,0 +1,21 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+#ifndef WIFI_FUZZ_CONNECT_TO_DEVICE_H_
+#define WIFI_FUZZ_CONNECT_TO_DEVICE_H_
+
+#define FUZZ_PROJECT_NAME "connecttodevice_fuzzer"
+
+#endif

wifi/test/fuzztest/wifi_sta/connecttodevice_fuzzer/corpus/init

@@ -0,0 +1,16 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+FUZZ

wifi/test/fuzztest/wifi_sta/connecttodevice_fuzzer/project.xml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Copyright (C) 2022 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.
+-->
+<fuzz_config>
+  <fuzztest>
+    <!-- maximum length of a test input -->
+    <max_len>1000</max_len>
+    <!-- maximum total time in seconds to run the fuzzer -->
+    <max_total_time>300</max_total_time>
+    <!-- memory usage limit in Mb -->
+    <rss_limit_mb>4096</rss_limit_mb>
+  </fuzztest>
+</fuzz_config>