From ed9504845b22c34986a100245165c44a7c013b25 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=B0=B8=E5=BF=A0?= Date: Thu, 23 Jan 2025 13:33:00 +0000 Subject: [PATCH] ADD IsDlpFileBySuffix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 王永忠 --- .../inner_api/dlp_parse/include/dlp_file_kits.h | 1 + .../inner_api/dlp_parse/src/dlp_file_kits.cpp | 13 ++++++++++++- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/interfaces/inner_api/dlp_parse/include/dlp_file_kits.h b/interfaces/inner_api/dlp_parse/include/dlp_file_kits.h index 2eb4ca4..b88d75d 100644 --- a/interfaces/inner_api/dlp_parse/include/dlp_file_kits.h +++ b/interfaces/inner_api/dlp_parse/include/dlp_file_kits.h @@ -44,6 +44,7 @@ public: static bool IsDlpFile(int32_t dlpFd); static void ConvertAbilityInfoWithSupportDlp(const AAFwk::Want& want, std::vector &abilityInfos); + static bool IsDlpFileBySuffix(const std::string &fileSuffix); }; } // namespace DlpPermission } // namespace Security diff --git a/interfaces/inner_api/dlp_parse/src/dlp_file_kits.cpp b/interfaces/inner_api/dlp_parse/src/dlp_file_kits.cpp index 50461cc..7ac677a 100644 --- a/interfaces/inner_api/dlp_parse/src/dlp_file_kits.cpp +++ b/interfaces/inner_api/dlp_parse/src/dlp_file_kits.cpp @@ -78,7 +78,8 @@ static bool IsDlpFileName(const std::string& dlpFileName) return false; } - if (dlpFileName.substr(fileNameLen - dlpSuffixLen, dlpSuffixLen) != DLP_FILE_SUFFIX) { + std::string fileSuffix = dlpFileName.substr(fileNameLen - dlpSuffixLen, dlpSuffixLen); + if (DlpUtils::ToLowerString(fileSuffix) != DLP_FILE_SUFFIX) { return false; } return true; @@ -281,6 +282,16 @@ void DlpFileKits::ConvertAbilityInfoWithSupportDlp(const AAFwk::Want &want, DLP_LOG_ERROR(LABEL, "Query ability info with bundleName error."); } } + +bool DlpFileKits::IsDlpFileBySuffix(const std::string &fileSuffix) +{ + std::string lowerFileSuffix = DlpUtils::ToLowerString(fileSuffix); + if (lowerFileSuffix != DLP_FILE_SUFFIX) { + DLP_LOG_DEBUG(LABEL, "%{public}s is not dlp file suffix", lowerFileSuffix.c_str()); + return false; + } + return true; +} } // namespace DlpPermission } // namespace Security } // namespace OHOS -- Gitee