253 lines
8.0 KiB
C++
253 lines
8.0 KiB
C++
|
/*
|
||
|
|
* Copyright (c) 2023 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 <iostream>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <dirent.h>
|
||
|
|
|
||
|
|
#include "init_eng.h"
|
||
|
|
#include "init_utils.h"
|
||
|
|
#include "param_stub.h"
|
||
|
|
#include "bootstage.h"
|
||
|
|
#include "securec.h"
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
using namespace testing::ext;
|
||
|
|
|
||
|
|
namespace init_ut {
|
||
|
|
static const std::string SRC_FILE_PATH = STARTUP_INIT_UT_PATH"/eng/source/test.txt";
|
||
|
|
static const std::string TARGET_PATH = STARTUP_INIT_UT_PATH"/eng/link_name";
|
||
|
|
static const std::string ENG_ROOT_PATH = STARTUP_INIT_UT_PATH"/eng/";
|
||
|
|
|
||
|
|
static bool RemoveDir(const std::string &path)
|
||
|
|
{
|
||
|
|
if (path.empty()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
std::string strPath = path;
|
||
|
|
if (strPath.at(strPath.length() - 1) != '/') {
|
||
|
|
strPath.append("/");
|
||
|
|
}
|
||
|
|
DIR *d = opendir(strPath.c_str());
|
||
|
|
if (d != nullptr) {
|
||
|
|
struct dirent *dt = nullptr;
|
||
|
|
dt = readdir(d);
|
||
|
|
while (dt != nullptr) {
|
||
|
|
if (strcmp(dt->d_name, "..") == 0 || strcmp(dt->d_name, ".") == 0) {
|
||
|
|
dt = readdir(d);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
struct stat st {};
|
||
|
|
auto file_name = strPath + std::string(dt->d_name);
|
||
|
|
stat(file_name.c_str(), &st);
|
||
|
|
if (S_ISDIR(st.st_mode)) {
|
||
|
|
RemoveDir(file_name);
|
||
|
|
} else {
|
||
|
|
remove(file_name.c_str());
|
||
|
|
}
|
||
|
|
dt = readdir(d);
|
||
|
|
}
|
||
|
|
closedir(d);
|
||
|
|
}
|
||
|
|
return rmdir(strPath.c_str()) == 0 ? true : false;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool IsFileExist(const std::string &path)
|
||
|
|
{
|
||
|
|
if (path.empty()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
struct stat st {};
|
||
|
|
if (stat(path.c_str(), &st) == 0 && S_ISREG(st.st_mode)) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool IsDirExist(const std::string &path)
|
||
|
|
{
|
||
|
|
if (path.empty()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
struct stat st {};
|
||
|
|
if (stat(path.c_str(), &st) == 0 && S_ISDIR(st.st_mode)) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
class EngUnitTest : public testing::Test {
|
||
|
|
public:
|
||
|
|
static void SetUpTestCase(void) {};
|
||
|
|
static void TearDownTestCase(void) {};
|
||
|
|
void SetUp() {};
|
||
|
|
void TearDown() {};
|
||
|
|
};
|
||
|
|
|
||
|
|
HWTEST_F(EngUnitTest, TestFilesOverlay, TestSize.Level1)
|
||
|
|
{
|
||
|
|
bool isDel = false;
|
||
|
|
bool isExist = IsDirExist(ENG_ROOT_PATH.c_str());
|
||
|
|
if (isExist) {
|
||
|
|
isDel = RemoveDir(ENG_ROOT_PATH.c_str());
|
||
|
|
EXPECT_EQ(isDel, true);
|
||
|
|
}
|
||
|
|
isExist = IsDirExist(TARGET_PATH.c_str());
|
||
|
|
if (isExist) {
|
||
|
|
isDel = RemoveDir(TARGET_PATH.c_str());
|
||
|
|
EXPECT_EQ(isDel, true);
|
||
|
|
}
|
||
|
|
DebugFilesOverlay(ENG_ROOT_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
|
||
|
|
CreateTestFile(SRC_FILE_PATH.c_str(), "test");
|
||
|
|
isExist = IsFileExist(SRC_FILE_PATH.c_str());
|
||
|
|
EXPECT_EQ(isExist, true);
|
||
|
|
|
||
|
|
DebugFilesOverlay(SRC_FILE_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
isExist = IsFileExistWithType(SRC_FILE_PATH.c_str(), TYPE_REG);
|
||
|
|
EXPECT_EQ(isExist, true);
|
||
|
|
|
||
|
|
|
||
|
|
if (IsFileExistWithType(TARGET_PATH.c_str(), TYPE_LINK)) {
|
||
|
|
if (unlink(TARGET_PATH.c_str()) < 0) {
|
||
|
|
EXPECT_TRUE(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
int ret = symlink(ENG_ROOT_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
EXPECT_EQ(ret, 0);
|
||
|
|
isExist = IsFileExistWithType(TARGET_PATH.c_str(), TYPE_LINK);
|
||
|
|
EXPECT_EQ(isExist, true);
|
||
|
|
DebugFilesOverlay(TARGET_PATH.c_str(), ENG_ROOT_PATH.c_str());
|
||
|
|
EXPECT_EQ(ret, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
HWTEST_F(EngUnitTest, TestBindMountFile, TestSize.Level1)
|
||
|
|
{
|
||
|
|
BindMountFile("data/init_ut", "");
|
||
|
|
BindMountFile("data", "target");
|
||
|
|
BindMountFile("/data/init_ut//", "/");
|
||
|
|
BindMountFile("/data/init_ut", "/");
|
||
|
|
BindMountFile("/data", "/");
|
||
|
|
BindMountFile("/data/", "/");
|
||
|
|
|
||
|
|
bool isExist = false;
|
||
|
|
if (!IsFileExist(SRC_FILE_PATH.c_str())) {
|
||
|
|
CreateTestFile(SRC_FILE_PATH.c_str(), "test reg file mount");
|
||
|
|
isExist = IsFileExist(SRC_FILE_PATH.c_str());
|
||
|
|
EXPECT_EQ(isExist, true);
|
||
|
|
BindMountFile(SRC_FILE_PATH.c_str(), "/");
|
||
|
|
}
|
||
|
|
BindMountFile(SRC_FILE_PATH.c_str(), "/");
|
||
|
|
|
||
|
|
if (IsFileExist(SRC_FILE_PATH.c_str())) {
|
||
|
|
RemoveDir(STARTUP_INIT_UT_PATH"/eng/source");
|
||
|
|
isExist = IsFileExist(SRC_FILE_PATH.c_str());
|
||
|
|
EXPECT_EQ(isExist, false);
|
||
|
|
}
|
||
|
|
if (IsFileExistWithType(TARGET_PATH.c_str(), TYPE_LINK)) {
|
||
|
|
if (unlink(TARGET_PATH.c_str()) < 0) {
|
||
|
|
EXPECT_TRUE(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool isLinkFile = IsFileExist(TARGET_PATH.c_str());
|
||
|
|
EXPECT_EQ(isLinkFile, false);
|
||
|
|
BindMountFile(SRC_FILE_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
|
||
|
|
int ret = symlink(SRC_FILE_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
EXPECT_EQ(ret, 0);
|
||
|
|
isLinkFile = IsFileExistWithType(TARGET_PATH.c_str(), TYPE_LINK);
|
||
|
|
EXPECT_EQ(isLinkFile, true);
|
||
|
|
BindMountFile(SRC_FILE_PATH.c_str(), TARGET_PATH.c_str());
|
||
|
|
BindMountFile(TARGET_PATH.c_str(), SRC_FILE_PATH.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
HWTEST_F(EngUnitTest, TestMountCmd, TestSize.Level1)
|
||
|
|
{
|
||
|
|
char mountCmd[MOUNT_CMD_MAX_LEN] = {};
|
||
|
|
MountEngPartitions();
|
||
|
|
BuildMountCmd(mountCmd, MOUNT_CMD_MAX_LEN, "/eng/source", "/eng/target", "ext4");
|
||
|
|
BuildMountCmd(mountCmd, 0, "/eng/source", "/eng/target", "ext4");
|
||
|
|
}
|
||
|
|
|
||
|
|
HWTEST_F(EngUnitTest, TestFileType, TestSize.Level1)
|
||
|
|
{
|
||
|
|
std::string targetFile = "/data/init_ut/eng/target_file";
|
||
|
|
std::string linkName = "/data/init_ut/eng/link_name_test";
|
||
|
|
bool isExist = false;
|
||
|
|
|
||
|
|
if (!IsFileExist(SRC_FILE_PATH.c_str())) {
|
||
|
|
CreateTestFile(SRC_FILE_PATH.c_str(), "test");
|
||
|
|
isExist = IsFileExist(SRC_FILE_PATH.c_str());
|
||
|
|
EXPECT_EQ(isExist, true);
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPECT_EQ(IsFileExistWithType(SRC_FILE_PATH.c_str(), TYPE_REG), true);
|
||
|
|
EXPECT_EQ(IsFileExistWithType(STARTUP_INIT_UT_PATH"/eng", TYPE_DIR), true);
|
||
|
|
|
||
|
|
EXPECT_EQ(IsFileExistWithType(STARTUP_INIT_UT_PATH"/eng", TYPE_LINK), false);
|
||
|
|
EXPECT_EQ(IsFileExistWithType(STARTUP_INIT_UT_PATH"/eng", TYPE_REG), false);
|
||
|
|
EXPECT_EQ(IsFileExistWithType(STARTUP_INIT_UT_PATH"/eng", TYPE_ANY), true);
|
||
|
|
EXPECT_EQ(IsFileExistWithType(SRC_FILE_PATH.c_str(), TYPE_DIR), false);
|
||
|
|
|
||
|
|
if (IsFileExist(targetFile)) {
|
||
|
|
if (unlink(targetFile.c_str()) < 0) {
|
||
|
|
std::cout << "Failed to unlink file " << targetFile << " err = " << errno << std::endl;
|
||
|
|
EXPECT_TRUE(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
int fd = open(targetFile.c_str(), O_CREAT | O_CLOEXEC | O_WRONLY, 0644);
|
||
|
|
if (fd < 0) {
|
||
|
|
std::cout << "Failed to create file " << targetFile << " err = " << errno << std::endl;
|
||
|
|
EXPECT_TRUE(false);
|
||
|
|
} else {
|
||
|
|
std::string buffer = "hello";
|
||
|
|
write(fd, buffer.c_str(), buffer.size());
|
||
|
|
close(fd); // avoid leak
|
||
|
|
}
|
||
|
|
|
||
|
|
if (IsFileExist(linkName)) {
|
||
|
|
if (unlink(linkName.c_str()) < 0) {
|
||
|
|
std::cout << "Failed to unlink file " << linkName << " err = " << errno << std::endl;
|
||
|
|
EXPECT_TRUE(false);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int ret = symlink(targetFile.c_str(), linkName.c_str());
|
||
|
|
EXPECT_EQ(ret, 0);
|
||
|
|
bool isFileExist = IsFileExistWithType(linkName.c_str(), TYPE_LINK);
|
||
|
|
EXPECT_EQ(isFileExist, true);
|
||
|
|
|
||
|
|
isFileExist = IsFileExistWithType("/eng/target", TYPE_LINK);
|
||
|
|
EXPECT_EQ(isFileExist, false);
|
||
|
|
|
||
|
|
isFileExist = IsFileExistWithType("/eng/target", TYPE_REG);
|
||
|
|
EXPECT_EQ(isFileExist, false);
|
||
|
|
}
|
||
|
|
|
||
|
|
HWTEST_F(EngUnitTest, TestHook, TestSize.Level1)
|
||
|
|
{
|
||
|
|
HookMgrExecute(GetBootStageHookMgr(), INIT_GLOBAL_INIT, nullptr, nullptr);
|
||
|
|
PrepareCmdLineData();
|
||
|
|
HookMgrExecute(GetBootStageHookMgr(), INIT_GLOBAL_INIT, nullptr, nullptr);
|
||
|
|
const char *cmdLine = "ohos.boot.root_package=off ";
|
||
|
|
CreateTestFile(BOOT_CMD_LINE, cmdLine);
|
||
|
|
HookMgrExecute(GetBootStageHookMgr(), INIT_GLOBAL_INIT, nullptr, nullptr);
|
||
|
|
}
|
||
|
|
} // namespace init_ut
|