* Copyright (c) 2025 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 "util/white_list.h"
#include "internal_inc/osal.h"
#include "util/ffrt_facade.h"
namespace {
constexpr char CONF_FILEPATH[] = "/etc/ffrt/ffrt_whitelist.conf";
constexpr int INDENT_SPACE_NUM = 4;
constexpr int PROCESS_NAME_BUFFER_LENGTH = 1024;
}
namespace ffrt {
WhiteList::WhiteList()
{
whiteListArray_.fill(-1);
LoadFromFile();
}
WhiteList& WhiteList::GetInstance()
{
static WhiteList instance;
return instance;
}
bool WhiteList::IsEnabled(WhiteListKey key, bool defaultWhenAbnormal)
{
if (TryRefreshWhiteListOnInit()) {
if (key != WhiteListKey::UNKNOWN &&
whiteListArray_[static_cast<size_t>(key)] != -1) {
return whiteListArray_[static_cast<size_t>(key)];
}
} else {
std::unique_lock lock(whitelistMutex_);
if (key != WhiteListKey::UNKNOWN &&
whiteListArray_[static_cast<size_t>(key)] != -1) {
return whiteListArray_[static_cast<size_t>(key)];
}
}
return defaultWhenAbnormal;
}
void WhiteList::LoadFromFile()
{
whiteListArray_.fill(0);
char processNameChar[PROCESS_NAME_BUFFER_LENGTH] {};
GetProcessName(processNameChar, PROCESS_NAME_BUFFER_LENGTH);
if (strlen(processNameChar) == 0) {
FFRT_LOGW("Get process name failed.");
return;
}
std::string processName = std::string(processNameChar);
#ifdef OHOS_STANDARD_SYSTEM
std::string whiteProcess;
std::ifstream file(CONF_FILEPATH);
size_t sectionIndex = 0;
if (file.is_open()) {
while (std::getline(file, whiteProcess)) {
if (whiteProcess.find("{") != std::string::npos) {
whiteListArray_[sectionIndex] = false;
continue;
}
if ((whiteProcess != "}" && whiteProcess != "") && sectionIndex < KEY_COUNT &&
processName.find(whiteProcess.substr(INDENT_SPACE_NUM)) != std::string::npos) {
whiteListArray_[sectionIndex] = true;
} else if (whiteProcess == "}") {
sectionIndex++;
}
}
} else {
FFRT_LOGW("white_list.conf does not exist or file permission denied");
}
#else
whiteListArray_[static_cast<size_t>(WhiteListKey::IsInSFFRTList)] = false;
if (processName.find("zygote") != std::string::npos) {
whiteListArray_[static_cast<size_t>(WhiteListKey::IsInSFFRTList)] = true;
}
if (processName.find("CameraDaemon") != std::string::npos) {
whiteListArray_[static_cast<size_t>(WhiteListKey::SetThreadAttr)] = true;
}
#endif
}
bool WhiteList::TryRefreshWhiteListOnInit()
{
if (FFRT_LIKELY(whitelistInited_)) {
return true;
}
if (!ffrt::GetInitFlag()) {
return false;
}
std::lock_guard lock(whitelistMutex_);
if (!whitelistInited_) {
LoadFromFile();
whitelistInited_ = true;
}
return true;
}
}
