From 944bf6e8ef0aa26d75f638321d69f9441b80cff3 Mon Sep 17 00:00:00 2001
From: wTong6 <wangtong94@huawei.com>
Date: Wed, 31 Jan 2024 11:16:34 +0800
Subject: [PATCH] fix:modify dac_override

Signed-off-by: wTong6 <wangtong94@huawei.com>
---
 .../copy_listener/trans_listener.cpp          | 96 +++++++++++++++++++
 .../properties/copy_listener/trans_listener.h |  4 +-
 2 files changed, 99 insertions(+), 1 deletion(-)

diff --git a/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.cpp b/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.cpp
index 62eaea2ef..88c46010f 100644
--- a/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.cpp
+++ b/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.cpp
@@ -15,14 +15,81 @@
 
 #include "trans_listener.h"
 
+#include <dirent.h>
+#include <filesystem>
+
 #include "distributed_file_daemon_manager.h"
+#include "file_uri.h"
 #include "ipc_skeleton.h"
 #include "uri.h"
 
 namespace OHOS {
 namespace FileManagement {
 namespace ModuleFileIO {
+using namespace OHOS::AppFileService;
+using namespace AppFileService::ModuleFileUri;
 const std::string NETWORK_PARA = "?networkid=";
+const std::string FILE_MANAGER_AUTHORITY = "docs";
+const std::string MEDIA_AUTHORITY = "media";
+const std::string DISTRIBUTED_PATH = "/data/storage/el2/distributedfiles";
+
+void TransListener::RmDirectory(const std::string &path)
+{
+    DIR *dir = opendir(path.c_str());
+    if (dir == nullptr) {
+        HILOGE("Open dir failed");
+        return;
+    }
+    dirent *entry = nullptr;
+    while ((entry = readdir(dir)) != nullptr) {
+        if (strcmp(entry->d_name, "..") == 0 || strcmp(entry->d_name, ".") == 0 ||
+            strcmp(entry->d_name, ".remote_share") == 0) {
+            continue;
+        }
+        std::string subPath = path + "/" + entry->d_name;
+        std::filesystem::path pathName(subPath);
+        if (std::filesystem::exists(pathName)) {
+            std::error_code errCode;
+            std::filesystem::remove_all(pathName, errCode);
+            if (errCode.value() != 0) {
+                HILOGE("Failed to remove directory, error code: %{public}d", errCode.value());
+                return;
+            }
+        }
+    }
+    closedir(dir);
+}
+
+void TransListener::CopyDir(const std::string &path, const std::string &sandboxPath)
+{
+    DIR *dir = opendir(path.c_str());
+    if (dir == nullptr) {
+        HILOGE("Open dir failed");
+        return;
+    }
+    dirent *entry = nullptr;
+    while ((entry = readdir(dir)) != nullptr) {
+        if (strcmp(entry->d_name, "..") == 0 || strcmp(entry->d_name, ".") == 0 ||
+            strcmp(entry->d_name, ".remote_share") == 0) {
+            continue;
+        }
+        std::string subPath = path + "/" + entry->d_name;
+        if (std::filesystem::is_directory(subPath)) {
+            auto pos = subPath.find_last_of('/');
+            if (pos == std::string::npos) {
+                closedir(dir);
+                return;
+            }
+            auto dirName = subPath.substr(pos);
+            std::filesystem::create_directories(sandboxPath + dirName);
+            std::filesystem::copy(subPath, sandboxPath + dirName,
+                std::filesystem::copy_options::recursive | std::filesystem::copy_options::update_existing);
+        } else {
+            std::filesystem::copy(subPath, sandboxPath, std::filesystem::copy_options::update_existing);
+        }
+    }
+    closedir(dir);
+}
 
 NError TransListener::CopyFileFromSoftBus(const std::string &srcUri,
                                           const std::string &destUri,
@@ -48,6 +115,35 @@ NError TransListener::CopyFileFromSoftBus(const std::string &srcUri,
     if (transListener->copyEvent_ == FAILED) {
         return NError(EIO);
     }
+
+    Uri uri(srcUri);
+    auto authority = uri.GetAuthority();
+    if (authority == FILE_MANAGER_AUTHORITY || authority == MEDIA_AUTHORITY) {
+        HILOGW("Public or media path not copy");
+        return NError(ERRNO_NOERR);
+    }
+
+    FileUri fileUri(destUri);
+    std::string sandboxPath = fileUri.GetPath();
+    if (std::filesystem::exists(sandboxPath) && std::filesystem::is_directory(sandboxPath)) {
+        HILOGI("Copy dir");
+        CopyDir(DISTRIBUTED_PATH, sandboxPath);
+    } else {
+        auto pos = srcUri.find_last_of('/');
+        if (pos == std::string::npos) {
+            HILOGE("invalid uri");
+            return NError(EIO);
+        }
+        auto fileName = srcUri.substr(pos);
+        auto networkIdPos = fileName.find(NETWORK_PARA);
+        if (networkIdPos == std::string::npos) {
+            HILOGE("Not remote uri");
+            return NError(EIO);
+        }
+        fileName = fileName.substr(0, networkIdPos);
+        std::filesystem::copy(DISTRIBUTED_PATH + fileName, sandboxPath, std::filesystem::copy_options::update_existing);
+    }
+    RmDirectory(DISTRIBUTED_PATH);
     return NError(ERRNO_NOERR);
 }
 
diff --git a/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.h b/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.h
index 5a4aecdab..a31156934 100644
--- a/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.h
+++ b/interfaces/kits/js/src/mod_fs/properties/copy_listener/trans_listener.h
@@ -37,6 +37,8 @@ public:
                                       std::shared_ptr<JsCallbackObject> callback);
     static std::string GetNetworkIdFromUri(const std::string &uri);
     static void CallbackComplete(uv_work_t *work, int stat);
+    static void RmDirectory(const std::string &path);
+    static void CopyDir(const std::string &path, const std::string &sandboxPath);
     std::mutex cvMutex_;
     std::condition_variable cv_;
     int copyEvent_ = NONE;
@@ -47,4 +49,4 @@ public:
 } // namespace FileManagement
 } // namespace OHOS
 
-#endif // FILEMANAGEMENT_FILE_API_TRANS_LISTENER_H
+#endif // FILEMANAGEMENT_FILE_API_TRANS_LISTENER_H
\ No newline at end of file
-- 
Gitee