599 lines
21 KiB
C++
599 lines
21 KiB
C++
|
/*
|
|||
|
|
* Copyright (c) 2021 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 <gtest/gtest.h>
|
|||
|
|
|
|||
|
|
#include "init_param.h"
|
|||
|
|
#include "param_message.h"
|
|||
|
|
#include "param_stub.h"
|
|||
|
|
#include "trigger_manager.h"
|
|||
|
|
#include "le_timer.h"
|
|||
|
|
|
|||
|
|
using namespace testing::ext;
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
static int TestTriggerExecute(TriggerNode *trigger, const char *content, uint32_t size)
|
|||
|
|
{
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static int CheckServerParamValue(const char *name, const char *expectValue)
|
|||
|
|
{
|
|||
|
|
char tmp[PARAM_BUFFER_SIZE] = {0};
|
|||
|
|
u_int32_t len = sizeof(tmp);
|
|||
|
|
SystemReadParam(name, tmp, &len);
|
|||
|
|
printf("CheckParamValue name %s value: \'%s\' expectValue:\'%s\' \n", name, tmp, expectValue);
|
|||
|
|
EXPECT_NE((int)strlen(tmp), 0);
|
|||
|
|
if (expectValue != nullptr) {
|
|||
|
|
EXPECT_EQ(strcmp(tmp, expectValue), 0);
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
namespace init_ut {
|
|||
|
|
static ParamTaskPtr g_worker = nullptr;
|
|||
|
|
class ParamServiceUnitTest : public ::testing::Test {
|
|||
|
|
public:
|
|||
|
|
ParamServiceUnitTest() {}
|
|||
|
|
virtual ~ParamServiceUnitTest() {}
|
|||
|
|
|
|||
|
|
static void SetUpTestCase(void)
|
|||
|
|
{
|
|||
|
|
PrepareInitUnitTestEnv();
|
|||
|
|
}
|
|||
|
|
void SetUp()
|
|||
|
|
{
|
|||
|
|
if (GetParamSecurityLabel() != nullptr) {
|
|||
|
|
GetParamSecurityLabel()->cred.uid = 1000; // 1000 test uid
|
|||
|
|
GetParamSecurityLabel()->cred.gid = 1000; // 1000 test gid
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
void TearDown() {}
|
|||
|
|
void TestBody() {}
|
|||
|
|
|
|||
|
|
int TestSetParams(const char *params[][1 + 1], int num)
|
|||
|
|
{
|
|||
|
|
for (int i = 0; i < num; i++) {
|
|||
|
|
SystemWriteParam(params[i][0], params[i][1]);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// check
|
|||
|
|
for (int i = 0; i < num; i++) {
|
|||
|
|
CheckServerParamValue(params[i][0], params[i][1]);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (int i = num - 1; i >= 0; i--) {
|
|||
|
|
CheckServerParamValue(params[i][0], params[i][1]);
|
|||
|
|
}
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestAddSecurityLabel1()
|
|||
|
|
{
|
|||
|
|
GetParamSecurityLabel()->cred.gid = 9999; // 9999 test gid
|
|||
|
|
const char *name = "label1.test.aaa.bbb.ccc.dddd.eee";
|
|||
|
|
const char *value = "2001";
|
|||
|
|
uint32_t labelIndex = 0;
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
// 获取到跟属性
|
|||
|
|
WorkSpace *workspace = GetWorkSpace(WORKSPACE_INDEX_DAC);
|
|||
|
|
(void)FindTrieNode(workspace, name, strlen(name), &labelIndex);
|
|||
|
|
ParamSecurityNode *node = (ParamSecurityNode *)GetTrieNode(workspace, labelIndex);
|
|||
|
|
if (node == nullptr) {
|
|||
|
|
EXPECT_EQ(1, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
EXPECT_EQ(node->gid, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加一个label，最长匹配到这个节点
|
|||
|
|
int TestAddSecurityLabel2()
|
|||
|
|
{
|
|||
|
|
GetParamSecurityLabel()->cred.gid = 9999; // 9999 test gid
|
|||
|
|
const char *name = "label2.test.aaa.bbb.ccc.dddd.eee";
|
|||
|
|
const char *value = "2001";
|
|||
|
|
ParamAuditData auditData = {};
|
|||
|
|
auditData.name = "label2.test.aaa";
|
|||
|
|
auditData.dacData.gid = 202; // 202 test dac gid
|
|||
|
|
auditData.dacData.uid = geteuid();
|
|||
|
|
auditData.dacData.mode = 0666; // 0666 test mode
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
uint32_t labelIndex = 0;
|
|||
|
|
AddSecurityLabel(&auditData);
|
|||
|
|
WorkSpace *workspace = GetWorkSpace(WORKSPACE_INDEX_DAC);
|
|||
|
|
(void)FindTrieNode(workspace, name, strlen(name), &labelIndex);
|
|||
|
|
ParamSecurityNode *node = (ParamSecurityNode *)GetTrieNode(workspace, labelIndex);
|
|||
|
|
if (node == nullptr) {
|
|||
|
|
EXPECT_EQ(1, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
EXPECT_EQ(node->gid, auditData.dacData.gid);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加一个label，最长匹配到最后一个相同节点
|
|||
|
|
int TestAddSecurityLabel3()
|
|||
|
|
{
|
|||
|
|
GetParamSecurityLabel()->cred.gid = 9999; // 9999 test gid
|
|||
|
|
const char *name = "label3.test.aaa.bbb.ccc.dddd.eee";
|
|||
|
|
const char *value = "2001";
|
|||
|
|
ParamAuditData auditData = {};
|
|||
|
|
auditData.name = "label3.test.aaa";
|
|||
|
|
auditData.dacData.gid = 203; // 203 test gid
|
|||
|
|
auditData.dacData.uid = geteuid();
|
|||
|
|
auditData.dacData.mode = 0666; // 0666 test mode
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
AddSecurityLabel(&auditData);
|
|||
|
|
|
|||
|
|
auditData.name = "label3.test.aaa.bbb.ccc.dddd.eee.dddd";
|
|||
|
|
auditData.dacData.gid = 202; // 202 test dac gid
|
|||
|
|
auditData.dacData.uid = geteuid();
|
|||
|
|
auditData.dacData.mode = 0666; // 0666 test mode
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
AddSecurityLabel(&auditData);
|
|||
|
|
|
|||
|
|
uint32_t labelIndex = 0;
|
|||
|
|
WorkSpace *workspace = GetWorkSpace(WORKSPACE_INDEX_DAC);
|
|||
|
|
ParamTrieNode *paramNode = FindTrieNode(workspace, name, strlen(name), &labelIndex);
|
|||
|
|
ParamSecurityNode *node = (ParamSecurityNode *)GetTrieNode(workspace, labelIndex);
|
|||
|
|
if (paramNode == nullptr || node == nullptr) {
|
|||
|
|
EXPECT_EQ(1, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
EXPECT_EQ((int)node->gid, 203); // 203 test gid
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加一个label，完全匹配
|
|||
|
|
int TestAddSecurityLabel4()
|
|||
|
|
{
|
|||
|
|
GetParamSecurityLabel()->cred.gid = 9999; // 9999 test gid
|
|||
|
|
const char *name = "label4.test.aaa.bbb.ccc.dddd.eee";
|
|||
|
|
const char *value = "2001";
|
|||
|
|
ParamAuditData auditData = {};
|
|||
|
|
auditData.name = "label4.test.aaa.bbb.ccc.dddd.eee";
|
|||
|
|
auditData.dacData.gid = 203; // 203 test gid
|
|||
|
|
auditData.dacData.uid = geteuid();
|
|||
|
|
auditData.dacData.mode = 0666; // 0666 test mode
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
uint32_t labelIndex = 0;
|
|||
|
|
AddSecurityLabel(&auditData);
|
|||
|
|
WorkSpace *workspace = GetWorkSpace(WORKSPACE_INDEX_DAC);
|
|||
|
|
ParamTrieNode *paramNode = FindTrieNode(workspace, name, strlen(name), &labelIndex);
|
|||
|
|
ParamSecurityNode *node = (ParamSecurityNode *)GetTrieNode(workspace, labelIndex);
|
|||
|
|
if (paramNode == nullptr || node == nullptr) {
|
|||
|
|
EXPECT_EQ(1, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
EXPECT_EQ(node->gid, auditData.dacData.gid);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void TestBufferValue(char *buffer, uint32_t len)
|
|||
|
|
{
|
|||
|
|
const int printBase = 10;
|
|||
|
|
for (uint32_t index = 0; index <= len; index++) {
|
|||
|
|
buffer[index] = '0' + index % printBase;
|
|||
|
|
if (index != 0 && index % printBase == 0) {
|
|||
|
|
buffer[index] = '.';
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
buffer[len] = '\0';
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestNameIsValid()
|
|||
|
|
{
|
|||
|
|
char buffer[PARAM_BUFFER_SIZE];
|
|||
|
|
// set name length = 127
|
|||
|
|
TestBufferValue(buffer, PARAM_NAME_LEN_MAX - 1);
|
|||
|
|
int ret = SystemWriteParam(buffer, "1111");
|
|||
|
|
EXPECT_EQ(ret, 0);
|
|||
|
|
TestBufferValue(buffer, PARAM_NAME_LEN_MAX);
|
|||
|
|
ret = SystemWriteParam(buffer, "1111");
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
TestBufferValue(buffer, PARAM_NAME_LEN_MAX + 1);
|
|||
|
|
ret = SystemWriteParam(buffer, "1111");
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
|
|||
|
|
ret = SystemWriteParam(" ", "1111");
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = SystemWriteParam("__.__..", "1111");
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = SystemWriteParam(".__..", "1111");
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = SystemWriteParam("test.-@:_", "1111");
|
|||
|
|
EXPECT_EQ(ret, 0);
|
|||
|
|
|
|||
|
|
// 保存一个只读的属性，大于最大值
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX - 1);
|
|||
|
|
ret = SystemWriteParam("const.test_readonly.dddddddddddddddddd.fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_EQ(ret, 0);
|
|||
|
|
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX + 1);
|
|||
|
|
ret = SystemWriteParam("const.test_readonly.aaaaaaaaaaaaaaaaaa.fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_EQ(ret, 0);
|
|||
|
|
|
|||
|
|
// 更新只读项目
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX - 1);
|
|||
|
|
ret = SystemWriteParam("const.test_readonly.dddddddddddddddddd.fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
|
|||
|
|
// 写普通属性
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX - 1);
|
|||
|
|
ret = SystemWriteParam("test.bigvalue.dddddddddddddddddd.fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_EQ(ret, 0);
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX + 1);
|
|||
|
|
ret = SystemWriteParam("test.bigvalue.dddddddddddddddddd.fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
|
|||
|
|
// invalid name
|
|||
|
|
TestBufferValue(buffer, PARAM_VALUE_LEN_MAX - 1);
|
|||
|
|
ret = SystemWriteParam("test.invalidname..fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = SystemWriteParam("test.invalidname.%fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = SystemWriteParam("test.invalidname.$fffffffffffffffffff", buffer);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestParamTraversal()
|
|||
|
|
{
|
|||
|
|
char value[PARAM_BUFFER_SIZE + PARAM_BUFFER_SIZE] = {0};
|
|||
|
|
SystemTraversalParameter(
|
|||
|
|
"",
|
|||
|
|
[](ParamHandle handle, void *cookie) {
|
|||
|
|
SystemGetParameterName(handle, (char *)cookie, PARAM_BUFFER_SIZE);
|
|||
|
|
u_int32_t len = PARAM_BUFFER_SIZE;
|
|||
|
|
SystemGetParameterValue(handle, ((char *)cookie) + PARAM_BUFFER_SIZE, &len);
|
|||
|
|
printf("$$$$$$$$Param %s=%s \n", (char *)cookie, ((char *)cookie) + PARAM_BUFFER_SIZE);
|
|||
|
|
},
|
|||
|
|
(void *)value);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestUpdateParam(const char *name, const char *value)
|
|||
|
|
{
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
CheckServerParamValue(name, value);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestPersistParam()
|
|||
|
|
{
|
|||
|
|
RegisterPersistParamOps(nullptr);
|
|||
|
|
LoadPersistParams();
|
|||
|
|
SystemWriteParam("persist.111.ffff.bbbb.cccc.dddd.eeee", "1101");
|
|||
|
|
SystemWriteParam("persist.111.aaaa.bbbb.cccc.dddd.eeee", "1102");
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.eeee", "1103");
|
|||
|
|
CheckServerParamValue("persist.111.bbbb.cccc.dddd.eeee", "1103");
|
|||
|
|
SystemWriteParam("persist.111.cccc.bbbb.cccc.dddd.eeee", "1104");
|
|||
|
|
SystemWriteParam("persist.111.eeee.bbbb.cccc.dddd.eeee", "1105");
|
|||
|
|
CheckServerParamValue("persist.111.ffff.bbbb.cccc.dddd.eeee", "1101");
|
|||
|
|
SystemWriteParam("persist.111.ffff.bbbb.cccc.dddd.eeee", "1106");
|
|||
|
|
CheckServerParamValue("persist.111.ffff.bbbb.cccc.dddd.eeee", "1106");
|
|||
|
|
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.1107", "1107");
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.1108", "1108");
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.1109", "1108");
|
|||
|
|
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.1110", "1108");
|
|||
|
|
SystemWriteParam("persist.111.bbbb.cccc.dddd.1111", "1108");
|
|||
|
|
TimerCallbackForSave(nullptr, nullptr);
|
|||
|
|
LoadPersistParams();
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ParamTaskPtr CreateAndGetStreamTask()
|
|||
|
|
{
|
|||
|
|
OnIncomingConnect(LE_GetDefaultLoop(), GetParamService()->serverTask);
|
|||
|
|
EXPECT_NE(GetParamService()->watcherTask, nullptr);
|
|||
|
|
return GetParamService()->watcherTask;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestServiceProcessMessage(const char *name, const char *value, int userLabel)
|
|||
|
|
{
|
|||
|
|
if (g_worker == nullptr) {
|
|||
|
|
g_worker = CreateAndGetStreamTask();
|
|||
|
|
}
|
|||
|
|
if (g_worker == nullptr) {
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
ParamSecurityOps *paramSecurityOps = GetParamSecurityOps(0);
|
|||
|
|
if (userLabel && paramSecurityOps != nullptr) {
|
|||
|
|
paramSecurityOps->securityFreeLabel = TestFreeLocalSecurityLabel;
|
|||
|
|
paramSecurityOps->securityCheckParamPermission = TestCheckParamPermission;
|
|||
|
|
}
|
|||
|
|
uint32_t msgSize = sizeof(ParamMessage) + sizeof(ParamMsgContent) + PARAM_ALIGN(strlen(value) + 1);
|
|||
|
|
ParamMessage *request = (ParamMessage *)CreateParamMessage(MSG_SET_PARAM, name, msgSize);
|
|||
|
|
PARAM_CHECK(request != nullptr, return -1, "Failed to malloc for connect");
|
|||
|
|
do {
|
|||
|
|
request->type = MSG_SET_PARAM;
|
|||
|
|
uint32_t offset = 0;
|
|||
|
|
int ret = FillParamMsgContent(request, &offset, PARAM_VALUE, value, strlen(value));
|
|||
|
|
PARAM_CHECK(ret == 0, break, "Failed to fill value");
|
|||
|
|
ProcessMessage((const ParamTaskPtr)g_worker, (const ParamMessage *)request);
|
|||
|
|
} while (0);
|
|||
|
|
free(request);
|
|||
|
|
RegisterSecurityOps(1);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int AddWatch(int type, const char *name, const char *value)
|
|||
|
|
{
|
|||
|
|
if (g_worker == nullptr) {
|
|||
|
|
g_worker = CreateAndGetStreamTask();
|
|||
|
|
}
|
|||
|
|
if (g_worker == nullptr) {
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
uint32_t msgSize = sizeof(ParamMessage) + sizeof(ParamMsgContent) + sizeof(ParamMsgContent)
|
|||
|
|
+ PARAM_ALIGN(strlen(value) + 1) + sizeof(uint32_t);
|
|||
|
|
ParamMessage *request = (ParamMessage *)(ParamMessage *)CreateParamMessage(type, name, msgSize);
|
|||
|
|
PARAM_CHECK(request != nullptr, return -1, "Failed to malloc for connect");
|
|||
|
|
do {
|
|||
|
|
uint32_t offset = 0;
|
|||
|
|
int ret = FillParamMsgContent(request, &offset, PARAM_VALUE, value, strlen(value));
|
|||
|
|
PARAM_CHECK(ret == 0, break, "Failed to fill value");
|
|||
|
|
|
|||
|
|
ParamMsgContent *content = (ParamMsgContent *)(request->data + offset);
|
|||
|
|
content->type = PARAM_WAIT_TIMEOUT;
|
|||
|
|
content->contentSize = sizeof(uint32_t);
|
|||
|
|
*((uint32_t *)(content->content)) = 100; // 100 is test value
|
|||
|
|
offset += sizeof(ParamMsgContent) + sizeof(uint32_t);
|
|||
|
|
request->msgSize = offset + sizeof(ParamMessage);
|
|||
|
|
|
|||
|
|
ProcessMessage((const ParamTaskPtr)g_worker, (const ParamMessage *)request);
|
|||
|
|
} while (0);
|
|||
|
|
free(request);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 无匹配
|
|||
|
|
int TestAddParamWait1()
|
|||
|
|
{
|
|||
|
|
const char *name = "wait.aaa.bbb.ccc.111";
|
|||
|
|
const char *value = "wait1";
|
|||
|
|
AddWatch(MSG_WAIT_PARAM, name, value);
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
// execute trigger
|
|||
|
|
LE_DoAsyncEvent(LE_GetDefaultLoop(), GetTriggerWorkSpace()->eventHandle);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模糊匹配
|
|||
|
|
int TestAddParamWait2()
|
|||
|
|
{
|
|||
|
|
const char *name = "wait.aaa.bbb.ccc.222";
|
|||
|
|
const char *value = "wait2";
|
|||
|
|
AddWatch(MSG_WAIT_PARAM, name, "*");
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
// execute trigger
|
|||
|
|
LE_DoAsyncEvent(LE_GetDefaultLoop(), GetTriggerWorkSpace()->eventHandle);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 属性存在
|
|||
|
|
int TestAddParamWait3()
|
|||
|
|
{
|
|||
|
|
const char *name = "wait.aaa.bbb.ccc.333";
|
|||
|
|
const char *value = "wait3";
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
AddWatch(MSG_WAIT_PARAM, name, value);
|
|||
|
|
// execute trigger
|
|||
|
|
LE_DoAsyncEvent(LE_GetDefaultLoop(), GetTriggerWorkSpace()->eventHandle);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestAddParamWatch1()
|
|||
|
|
{
|
|||
|
|
const char *name = "watch.aaa.bbb.ccc.111";
|
|||
|
|
const char *value = "watch1";
|
|||
|
|
AddWatch(MSG_ADD_WATCHER, name, value);
|
|||
|
|
std::string newName = name;
|
|||
|
|
newName += ".test.test.test";
|
|||
|
|
SystemWriteParam(newName.c_str(), value);
|
|||
|
|
// execute trigger
|
|||
|
|
LE_DoAsyncEvent(LE_GetDefaultLoop(), GetTriggerWorkSpace()->eventHandle);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestAddParamWatch2()
|
|||
|
|
{
|
|||
|
|
const char *name = "watch.aaa.bbb.ccc.222";
|
|||
|
|
const char *value = "watch2";
|
|||
|
|
AddWatch(MSG_ADD_WATCHER, name, "*");
|
|||
|
|
SystemWriteParam(name, value);
|
|||
|
|
// execute trigger
|
|||
|
|
LE_DoAsyncEvent(LE_GetDefaultLoop(), GetTriggerWorkSpace()->eventHandle);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestAddParamWatch3()
|
|||
|
|
{
|
|||
|
|
const char *name = "watch.aaa.bbb.ccc.333";
|
|||
|
|
const char *value = "watch3";
|
|||
|
|
std::string newName = name;
|
|||
|
|
newName += ".test.test.test";
|
|||
|
|
SystemWriteParam(newName.c_str(), value);
|
|||
|
|
AddWatch(MSG_ADD_WATCHER, name, value);
|
|||
|
|
char buffer[] = "testbuff";
|
|||
|
|
CheckTrigger(GetTriggerWorkSpace(), TRIGGER_PARAM_WATCH, buffer, strlen(buffer), TestTriggerExecute);
|
|||
|
|
AddWatch(MSG_DEL_WATCHER, name, value);
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestCloseTriggerWatch()
|
|||
|
|
{
|
|||
|
|
#ifdef PARAM_SUPPORT_TRIGGER
|
|||
|
|
SystemDumpTriggers(1, nullptr);
|
|||
|
|
#endif
|
|||
|
|
ParamWatcher *watcher = (ParamWatcher *)ParamGetTaskUserData(g_worker);
|
|||
|
|
ClearWatchTrigger(watcher, TRIGGER_PARAM_WAIT);
|
|||
|
|
ParamTaskClose(g_worker);
|
|||
|
|
g_worker = nullptr;
|
|||
|
|
ParamTaskClose(g_worker);
|
|||
|
|
SystemWriteParam("init.svc.param_watcher", "stopped");
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestServiceCtrl(const char *serviceName, uint16_t mode)
|
|||
|
|
{
|
|||
|
|
// service forbid
|
|||
|
|
TestSetParamCheckResult("ohos.servicectrl.", mode, 1);
|
|||
|
|
return SystemWriteParam("ohos.ctl.start", serviceName);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int TestPowerCtrl(const char *reboot, uint16_t mode)
|
|||
|
|
{
|
|||
|
|
// service forbid
|
|||
|
|
TestSetParamCheckResult("ohos.servicectrl.reboot", mode, 1);
|
|||
|
|
return SystemWriteParam("ohos.startup.powerctrl", reboot);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestPersistParam, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestPersistParam();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestReadCmdline, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
const char *snData = "222222222222222222222";
|
|||
|
|
CreateTestFile(STARTUP_INIT_UT_PATH "/system/etc/sn", snData);
|
|||
|
|
// for cmdline
|
|||
|
|
std::string cmdLineForSn = "bootgroup=device.charge.group earlycon=uart8250,mmio32,0xfe660000 "
|
|||
|
|
"root=PARTUUID=614e0000-0000 rw rootwait rootfstype=ext4 console=ttyFIQ0 hardware=rk3568"
|
|||
|
|
" BOOT_IMAGE=/kernel init=/init";
|
|||
|
|
std::string snWithPath = cmdLineForSn + "ohos.boot.sn=" + STARTUP_INIT_UT_PATH "/system/etc/sn";
|
|||
|
|
CreateTestFile(BOOT_CMD_LINE, snWithPath.c_str());
|
|||
|
|
LoadParamFromCmdLine();
|
|||
|
|
std::string snWithSN = cmdLineForSn + "ohos.boot.sn=" + snData;
|
|||
|
|
CreateTestFile(BOOT_CMD_LINE, snWithSN.c_str());
|
|||
|
|
LoadParamFromCmdLine();
|
|||
|
|
CreateTestFile(BOOT_CMD_LINE, cmdLineForSn.c_str());
|
|||
|
|
LoadParamFromCmdLine();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestServiceProcessMessage, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestServiceProcessMessage("wertt.qqqq.wwww.rrrr", "wwww.eeeee", 1);
|
|||
|
|
test.TestServiceProcessMessage("wertt.2222.wwww.3333", "wwww.eeeee", 0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWait1, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWait1();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWait2, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWait2();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWait3, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWait3();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWatch1, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWatch1();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWatch2, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWatch2();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#ifndef OHOS_LITE
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestAddParamWatch3, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestAddParamWatch3();
|
|||
|
|
if (GetParamService()->timer != nullptr) {
|
|||
|
|
((TimerTask *)GetParamService()->timer)->processTimer(nullptr, nullptr);
|
|||
|
|
}
|
|||
|
|
int hashCode = CheckWatchTriggerTimeout();
|
|||
|
|
EXPECT_EQ(hashCode, 0);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestCloseTriggerWatch, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
test.TestCloseTriggerWatch();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestServiceCtrl, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
int ret = test.TestServiceCtrl("server1", 0770);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#ifdef PARAM_SUPPORT_SELINUX
|
|||
|
|
// selinux forbid
|
|||
|
|
ret = test.TestServiceCtrl("server2", 0772);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#endif
|
|||
|
|
ret = 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
HWTEST_F(ParamServiceUnitTest, TestPowerCtrl, TestSize.Level0)
|
|||
|
|
{
|
|||
|
|
ParamServiceUnitTest test;
|
|||
|
|
int ret = test.TestPowerCtrl("reboot,shutdown", 0770);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = test.TestPowerCtrl("reboot,shutdown", 0772);
|
|||
|
|
#ifdef PARAM_SUPPORT_SELINUX
|
|||
|
|
// selinux forbid
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#endif
|
|||
|
|
ret = test.TestPowerCtrl("reboot,updater", 0770);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = test.TestPowerCtrl("reboot,updater", 0772);
|
|||
|
|
#ifdef PARAM_SUPPORT_SELINUX
|
|||
|
|
// selinux forbid
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#endif
|
|||
|
|
ret = test.TestPowerCtrl("reboot,flashd", 0770);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = test.TestPowerCtrl("reboot,flashd", 0772);
|
|||
|
|
#ifdef PARAM_SUPPORT_SELINUX
|
|||
|
|
// selinux forbid
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
ret = test.TestPowerCtrl("reboot", 0770);
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
ret = test.TestPowerCtrl("reboot", 0772);
|
|||
|
|
#ifdef PARAM_SUPPORT_SELINUX
|
|||
|
|
// selinux forbid
|
|||
|
|
EXPECT_NE(ret, 0);
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
ParamServiceStop();
|
|||
|
|
}
|
|||
|
|
#endif
|
|||
|
|
} // namespace init_ut
|