add decompress fuzztest

Signed-off-by: zhaolihui <zhaolihui2@huawei.com>

test/BUILD.gn

@@ -413,10 +413,34 @@ ohos_fuzztest("DaemonUsbReadFuzzTest") {
   ]
 }
 
+ohos_fuzztest("DecompressToLocalFuzzTest") {
+  module_out_path = FUZZ_OUTPUT_PATH
+  fuzz_config_file = "${hdc_path}/test/fuzztest/decompresstolocal_fuzzer"
+  configs = [ ":hdc_test_config" ]
+  cflags = fuzz_cflags
+  sources = [
+    "${hdc_path}/src/common/compress.cpp",
+    "${hdc_path}/src/common/decompress.cpp",
+    "${hdc_path}/src/common/entry.cpp",
+    "${hdc_path}/src/common/header.cpp",
+    "fuzztest/decompresstolocal_fuzzer/DecompressToLocal_fuzzer.cpp",
+  ]
+  deps = [ ":hdc_common" ]
+  external_deps = [
+    "init:libbeget_proxy",
+    "init:libbegetutil",
+  ]
+  public_external_deps = [
+    "bounds_checking_function:libsec_static",
+    "libuv:uv",
+  ]
+}
+
 group("hdc_fuzztest") {
   testonly = true
   deps = [
     ":DaemonUsbReadFuzzTest",
+    ":DecompressToLocalFuzzTest",
     ":ForwardReadStreamFuzzTest",
     ":JdwpReadStreamFuzzTest",
   ]

test/fuzztest/decompresstolocal_fuzzer/DecompressToLocal_fuzzer.cpp

@@ -0,0 +1,146 @@
+/*
+ * Copyright (c) 2024 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 "DecompressToLocal_fuzzer.h"
+
+namespace Hdc {
+
+static int RemoveDir(const string &dir)
+{
+    DIR *pdir = opendir(dir.c_str());
+    if (pdir == nullptr) {
+        WRITE_LOG(LOG_WARN, "opendir failed dir:%s", dir.c_str());
+        return -1;
+    }
+    struct dirent *ent;
+    struct stat st;
+    while ((ent = readdir(pdir)) != nullptr) {
+        if (ent->d_name[0] == '.') {
+            continue;
+        }
+        std::string subpath = dir + Base::GetPathSep() + ent->d_name;
+        if (lstat(subpath.c_str(), &st) == -1) {
+            WRITE_LOG(LOG_WARN, "lstat failed subpath:%s", subpath.c_str());
+            continue;
+        }
+        if (S_ISDIR(st.st_mode)) {
+            if (RemoveDir(subpath) == -1) {
+                closedir(pdir);
+                return -1;
+            }
+            rmdir(subpath.c_str());
+        } else if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
+            unlink(subpath.c_str());
+        } else {
+            WRITE_LOG(LOG_WARN, "lstat st_mode:%07o subpath:%s", st.st_mode, subpath.c_str());
+        }
+    }
+    if (rmdir(dir.c_str()) == -1) {
+        closedir(pdir);
+        return -1;
+    }
+    closedir(pdir);
+    return 0;
+}
+
+static void RemovePath(const string &path)
+{
+    struct stat st;
+    if (lstat(path.c_str(), &st) == -1) {
+        WRITE_LOG(LOG_WARN, "lstat failed path:%s", path.c_str());
+        return;
+    }
+    if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
+        unlink(path.c_str());
+    } else if (S_ISDIR(st.st_mode)) {
+        if (path == "." || path == "..") {
+            return;
+        }
+        int rc = RemoveDir(path);
+        if (rc < 0) {
+            WRITE_LOG(LOG_WARN, "RemoveDir rc:%d path:%s", rc, path.c_str());
+        }
+    }
+}
+
+static string CompressToTar(const char *dir)
+{
+    string sdir = dir;
+    string tarname = Base::GetRandomString(EXPECTED_LEN) + ".tar";
+    string tarpath = Base::GetTmpDir() + tarname;
+    // WRITE_LOG(LOG_DEBUG, "tarpath:%s", tarpath.c_str());
+    Compress c;
+    c.UpdataPrefix(sdir);
+    c.AddPath(sdir);
+    c.SaveToFile(tarpath);
+    return tarpath;
+}
+
+static string DecompressFromTar(const char *path)
+{
+    string dir;
+    string tarpath = path;
+    string::size_type rindex = tarpath.rfind(".tar");
+    if (rindex != string::npos) {
+        dir = tarpath.substr(0, rindex) + Base::GetPathSep();
+        // WRITE_LOG(LOG_DEBUG, "path:%s dir:%s", path, dir.c_str());
+        Decompress dc(tarpath);
+        dc.DecompressToLocal(dir);
+    }
+    return dir;
+}
+
+static string GenerateFile(const uint8_t *data, size_t size)
+{
+    string dir = "./decompresstolocal";
+    if (access(dir.c_str(), 0) != 0) {
+        mkdir(dir.c_str(), S_IRWXU);
+    }
+    string path =  dir + "/" + "abcdefgh.data";
+    FILE *fp = fopen(path.c_str(), "wb");
+    if (!fp) {
+        WRITE_LOG(LOG_WARN,"fopen failed %s", path.c_str());
+        return dir;
+    }
+    size_t count = fwrite(data, sizeof(uint8_t), size, fp);
+    if (count < 0) {
+        WRITE_LOG(LOG_WARN,"fwrite count:%zd error:%d", count, ferror(fp));
+    }
+    fclose(fp);
+    return dir;
+}
+
+bool FuzzDecompressToLocal(const uint8_t *data, size_t size)
+{
+    if (size <= 0) {
+        return true;
+    }
+    string dir = GenerateFile(data, size);
+    string tarpath = CompressToTar(dir.c_str());
+    string tardir = DecompressFromTar(tarpath.c_str());
+    RemovePath(tardir);
+    RemovePath(tarpath);
+    RemovePath(dir);
+    return true;
+}
+} // namespace Hdc
+
+/* Fuzzer entry point */
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
+{
+    /* Run your code on data */
+    Hdc::FuzzDecompressToLocal(data, size);
+    return 0;
+}

test/fuzztest/decompresstolocal_fuzzer/DecompressToLocal_fuzzer.h

@@ -0,0 +1,27 @@
+/*
+ * Copyright (c) 2024 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 DECOMPRESSTOLOCAL_FUZZER_H
+#define DECOMPRESSTOLOCAL_FUZZER_H
+
+#include <unistd.h>
+#include "compress.h"
+#include "decompress.h"
+
+#define FUZZ_PROJECT_NAME "DecompressToLocal_fuzzer"
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size);
+
+#endif // DECOMPRESSTOLOCAL_FUZZER_H

test/fuzztest/decompresstolocal_fuzzer/corpus/init

@@ -0,0 +1,16 @@
+/*
+ * Copyright (c) 2024 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

test/fuzztest/decompresstolocal_fuzzer/project.xml

@@ -0,0 +1,25 @@
+<?xml version="1.0" encoding="utf-8"?>
+<!-- Copyright (c) 2024 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>10</max_total_time>
+    <!-- memory usage limit in Mb -->
+    <rss_limit_mb>4096</rss_limit_mb>
+  </fuzztest>
+</fuzz_config>