mirror of
https://github.com/openharmony/frame_aware_sched.git
synced 2026-07-01 22:04:00 -04:00
ce17b6640e
Signed-off-by: rongkunshi <shirongkun@huawei.com>
492 lines
13 KiB
C++
492 lines
13 KiB
C++
/*
|
|
* Copyright (c) 2022 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 "rtg_interface.h"
|
|
#include <linux/types.h>
|
|
#include <sys/ioctl.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <string>
|
|
#include <iostream>
|
|
|
|
#include "rme_log_domain.h"
|
|
|
|
namespace OHOS {
|
|
namespace RME {
|
|
|
|
DEFINE_RMELOG_INTELLISENSE("rtg_interface");
|
|
|
|
#define MAX_LENGTH 100
|
|
#define MAX_STR_LEN 100
|
|
#define MAX_PATH_LEN 100
|
|
|
|
#define RTG_SCHED_IPC_MAGIC 0xAB
|
|
|
|
#define CMD_ID_SET_ENABLE \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_ENABLE, struct rtg_enable_data)
|
|
#define CMD_ID_SET_RTG \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_RTG, struct rtg_str_data)
|
|
#define CMD_ID_SET_CONFIG \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_CONFIG, struct rtg_str_data)
|
|
#define CMD_ID_SET_RTG_ATTR \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_RTG_ATTR, struct rtg_str_data)
|
|
#define CMD_ID_BEGIN_FRAME_FREQ \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, BEGIN_FRAME_FREQ, struct proc_state_data)
|
|
#define CMD_ID_END_FRAME_FREQ \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, END_FRAME_FREQ, struct proc_state_data)
|
|
#define CMD_ID_END_SCENE \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, END_SCENE, struct proc_state_data)
|
|
#define CMD_ID_SET_MIN_UTIL \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_MIN_UTIL, struct proc_state_data)
|
|
#define CMD_ID_SET_MARGIN \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SET_MARGIN, struct proc_state_data)
|
|
#define CMD_ID_LIST_RTG \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, LIST_RTG, struct rtg_info)
|
|
#define CMD_ID_LIST_RTG_THREAD \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, LIST_RTG_THREAD, struct rtg_grp_data)
|
|
#define CMD_ID_SEARCH_RTG \
|
|
_IOWR(RTG_SCHED_IPC_MAGIC, SEARCH_RTG, struct proc_state_data)
|
|
|
|
int BasicOpenRtgNode(int pid)
|
|
{
|
|
char fileName[MAX_PATH_LEN] = {0};
|
|
snprintf(fileName, MAX_PATH_LEN, "/proc/%d/rtg", pid);
|
|
int fd = open(fileName, O_RDWR);
|
|
if (fd < 0) {
|
|
RME_LOGE("Open fail, errno = %{public}d(%{public}s), dev = %{public}s", errno, strerror(errno), fileName);
|
|
}
|
|
return fd;
|
|
}
|
|
|
|
enum rtg_type {
|
|
VIP = 0,
|
|
TOP_TASK_KEY,
|
|
TOP_TASK,
|
|
NORMAL_TASK,
|
|
RTG_TYPE_MAX,
|
|
};
|
|
|
|
int EnableRtg(bool flag)
|
|
{
|
|
struct rtg_enable_data enableData;
|
|
char configStr[] = "load_freq_switch:1;sched_cycle:1";
|
|
int pid = getpid();
|
|
|
|
enableData.enable = flag;
|
|
enableData.len = sizeof(configStr);
|
|
enableData.data = configStr;
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
if (ioctl(fd, CMD_ID_SET_ENABLE, &enableData)) {
|
|
RME_LOGE("set rtg config enable failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set rtg config enable success.");
|
|
}
|
|
close(fd);
|
|
return 0;
|
|
};
|
|
|
|
int CreateNewRtgGrp(int prioType, int rtNum)
|
|
{
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
if ((prioType > 0) && (prioType < RTG_TYPE_MAX)) {
|
|
grp_data.prio_type = prioType;
|
|
}
|
|
if (rtNum > 0) {
|
|
grp_data.rt_cnt = rtNum;
|
|
}
|
|
grp_data.rtg_cmd = CMD_CREATE_RTG_GRP;
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("create rtg grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("create rtg grp success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int AddThreadToRtg(int tid, int grpId)
|
|
{
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int fd = BasicOpenRtgNode(tid);
|
|
if (fd < 0) {
|
|
RME_LOGE("open node failed.");
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
grp_data.tid_num = 1;
|
|
grp_data.tids[0] = tid;
|
|
grp_data.grp_id = grpId;
|
|
grp_data.rtg_cmd = CMD_ADD_RTG_THREAD;
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("add rtg grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("add rtg grp success");
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int AddThreadsToRtg(vector<int> tids, int grpId)
|
|
{
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
int num = tids.size();
|
|
if (num > MAX_TID_NUM) {
|
|
return -1;
|
|
}
|
|
grp_data.tid_num = num;
|
|
grp_data.grp_id = grpId;
|
|
grp_data.rtg_cmd = CMD_ADD_RTG_THREAD;
|
|
for (int i = 0; i < num; i++) {
|
|
if (tids[i] < 0) {
|
|
return -1;
|
|
}
|
|
grp_data.tids[i] = tids[i];
|
|
}
|
|
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("add rtg grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("add rtg grp success");
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
};
|
|
|
|
int RemoveRtgThread(int tid){
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int fd = BasicOpenRtgNode(tid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
grp_data.tid_num = 1;
|
|
grp_data.tids[0] = tid;
|
|
grp_data.rtg_cmd = CMD_REMOVE_RTG_THREAD;
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("remove grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("remove grp success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
};
|
|
|
|
int ClearRtgGrp(int GrpId){
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
grp_data.rtg_cmd = CMD_CLEAR_RTG_GRP;
|
|
grp_data.grp_id = GrpId;
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("clear rtg grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("clear rtg grp success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
};
|
|
|
|
int DestroyRtgGrp(int GrpId){
|
|
struct rtg_grp_data grp_data;
|
|
int ret;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
grp_data.rtg_cmd = CMD_DESTROY_RTG_GRP;
|
|
grp_data.grp_id = GrpId;
|
|
ret = ioctl(fd, CMD_ID_SET_RTG, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("create rtg grp failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("create rtg grp success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
};
|
|
|
|
|
|
int SetMaxVipRtgs(int rtframe){
|
|
int ret = 0;
|
|
char str_data[MAX_STR_LEN] = {};
|
|
snprintf(str_data, MAX_STR_LEN, "rtframe:%d", rtframe);
|
|
struct rtg_str_data strData;
|
|
strData.len = strlen(str_data);
|
|
strData.data = str_data;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_SET_CONFIG, &strData);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set single config failed, ret = %{public}d", ret);
|
|
} else {
|
|
RME_LOGI("set single config success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int SetFrameRateAndPrioType(int rtgId, int rate, int rtgType){
|
|
int ret = 0;
|
|
char str_data[MAX_LENGTH] = {};
|
|
snprintf(str_data, MAX_LENGTH, "rtgId:%d;rate:%d;type:%d", rtgId, rate, rtgType);
|
|
struct rtg_str_data strData;
|
|
strData.len = strlen(str_data);
|
|
strData.data = str_data;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_SET_RTG_ATTR, &strData);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set rtg attr failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set rtg attr success, get rtg id %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int BeginFrameFreq(int grpId, int stateParam){
|
|
int ret = 0;
|
|
struct proc_state_data state_data;
|
|
state_data.grp_id = grpId;
|
|
state_data.state_param = stateParam;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_BEGIN_FRAME_FREQ, &state_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set BeginFrameFreq failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set BeginFrameFreq success, get ret %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int EndFrameFreq(int grpId, int stateParam){
|
|
int ret = 0;
|
|
struct proc_state_data state_data;
|
|
state_data.grp_id = grpId;
|
|
state_data.state_param = stateParam;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_END_FRAME_FREQ, &state_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set EndFrameFreq failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set EndFrameFreq success, get ret %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int EndScene(int rtgId){
|
|
int ret = 0;
|
|
struct proc_state_data state_data;
|
|
state_data.grp_id = rtgId;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_END_SCENE, &state_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set EndScene failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set EndScene success, get ret %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int SetMinUtil(int grpId, int stateParam){
|
|
int ret = 0;
|
|
struct proc_state_data state_data;
|
|
state_data.grp_id = grpId;
|
|
state_data.state_param = stateParam;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_SET_MIN_UTIL, &state_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set min util failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set min util success, get ret %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int SetMargin(int grpId, int stateParam) {
|
|
int ret = 0;
|
|
struct proc_state_data state_data;
|
|
state_data.grp_id = grpId;
|
|
state_data.state_param = stateParam;
|
|
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
ret = ioctl(fd, CMD_ID_SET_MARGIN, &state_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("set margin failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("set margin success, get ret %{public}d.", ret);
|
|
}
|
|
close(fd);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ListRtgGroup(vector<int> *rs)
|
|
{
|
|
int ret = 0;
|
|
struct rtg_info rtg_info;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
if (!rs) {
|
|
return -1;
|
|
}
|
|
memset(&rtg_info, 0, sizeof(struct rtg_info));
|
|
ret = ioctl(fd, CMD_ID_LIST_RTG, &rtg_info);
|
|
if ( ret < 0) {
|
|
RME_LOGE("list rtg group failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("list rtg group success with num %{public}d", rtg_info.rtg_num);
|
|
rs->clear();
|
|
for (int i = 0; i < rtg_info.rtg_num; i++) {
|
|
rs->push_back(rtg_info.rtgs[i]);
|
|
}
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int ListRtgThread(int grpId, vector<int> *rs)
|
|
{
|
|
int ret = 0;
|
|
struct rtg_grp_data grp_data;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
if (!rs) {
|
|
return -1;
|
|
}
|
|
memset(&grp_data, 0, sizeof(struct rtg_grp_data));
|
|
grp_data.grp_id = grpId;
|
|
ret = ioctl(fd, CMD_ID_LIST_RTG_THREAD, &grp_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("list rtg thread failed, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("list rtg thread success with tid num %{public}d", grp_data.tid_num);
|
|
rs->clear();
|
|
for (int i = 0; i < grp_data.tid_num; i++) {
|
|
rs->push_back(grp_data.tids[i]);
|
|
}
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
int SearchRtgForTid(int stateParam)
|
|
{
|
|
int ret = 0;
|
|
struct proc_state_data search_data;
|
|
int pid = getpid();
|
|
int fd = BasicOpenRtgNode(pid);
|
|
if (fd < 0) {
|
|
return fd;
|
|
}
|
|
if (stateParam <= 0) {
|
|
RME_LOGI("Search tid err: invalid tid.");
|
|
return -1;
|
|
}
|
|
memset(&search_data, 0, sizeof(struct proc_state_data));
|
|
search_data.state_param = stateParam;
|
|
ret = ioctl(fd, CMD_ID_SEARCH_RTG, &search_data);
|
|
if ( ret < 0) {
|
|
RME_LOGE("Search tid fail, errno = %{public}d (%{public}s)", errno, strerror(errno));
|
|
} else {
|
|
RME_LOGI("Search tid %{public}d success with rtg_grp %{public}d", stateParam, ret);
|
|
}
|
|
close(fd);
|
|
return ret;
|
|
}
|
|
|
|
} // namespace RME
|
|
} // namespace OHOS
|