2020-09-08 10:21:39 +08:00
|
|
|
/*
|
2021-03-11 18:43:57 +08:00
|
|
|
* Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved.
|
|
|
|
* Copyright (c) 2020-2021 Huawei Device Co., Ltd. All rights reserved.
|
2020-09-08 10:21:39 +08:00
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
|
|
* are permitted provided that the following conditions are met:
|
|
|
|
*
|
|
|
|
* 1. Redistributions of source code must retain the above copyright notice, this list of
|
|
|
|
* conditions and the following disclaimer.
|
|
|
|
*
|
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice, this list
|
|
|
|
* of conditions and the following disclaimer in the documentation and/or other materials
|
|
|
|
* provided with the distribution.
|
|
|
|
*
|
|
|
|
* 3. Neither the name of the copyright holder nor the names of its contributors may be used
|
|
|
|
* to endorse or promote products derived from this software without specific prior written
|
|
|
|
* permission.
|
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
|
|
|
|
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
|
|
|
|
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
|
|
|
|
* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
|
|
|
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
|
|
|
|
* OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
|
|
|
* ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "los_process_pri.h"
|
|
|
|
#include "los_task_pri.h"
|
2021-03-11 18:43:57 +08:00
|
|
|
#include "los_sched_pri.h"
|
2020-09-08 10:21:39 +08:00
|
|
|
#include "los_hw_pri.h"
|
|
|
|
#include "los_sys_pri.h"
|
|
|
|
#include "los_futex_pri.h"
|
2021-03-11 18:43:57 +08:00
|
|
|
#include "los_mp.h"
|
2020-09-08 10:21:39 +08:00
|
|
|
#include "user_copy.h"
|
|
|
|
#include "time.h"
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
#include "capability_api.h"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
static int OsPermissionToCheck(unsigned int pid, unsigned int who)
|
|
|
|
{
|
|
|
|
int ret = LOS_GetProcessGroupID(pid);
|
|
|
|
if (ret < 0) {
|
|
|
|
return ret;
|
|
|
|
} else if (ret == OS_KERNEL_PROCESS_GROUP) {
|
|
|
|
return -EPERM;
|
|
|
|
} else if ((ret == OS_USER_PRIVILEGE_PROCESS_GROUP) && (pid != who)) {
|
|
|
|
return -EPERM;
|
|
|
|
} else if (pid == OsGetUserInitProcessID()) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int OsUserTaskSchedulerSet(unsigned int tid, unsigned short policy, unsigned short priority, bool policyFlag)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
unsigned int intSave;
|
2021-03-11 18:43:57 +08:00
|
|
|
bool needSched = false;
|
2020-09-08 10:21:39 +08:00
|
|
|
|
|
|
|
if (OS_TID_CHECK_INVALID(tid)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (priority > OS_TASK_PRIORITY_LOWEST) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((policy != LOS_SCHED_FIFO) && (policy != LOS_SCHED_RR)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
2021-03-11 18:43:57 +08:00
|
|
|
LosTaskCB *taskCB = OS_TCB_FROM_TID(tid);
|
2020-09-08 10:21:39 +08:00
|
|
|
ret = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
policy = (policyFlag == true) ? policy : taskCB->policy;
|
|
|
|
needSched = OsSchedModifyTaskSchedParam(taskCB, policy, priority);
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
|
|
|
|
LOS_MpSchedule(OS_MP_CPU_ALL);
|
|
|
|
if (needSched && OS_SCHEDULER_ACTIVE) {
|
|
|
|
LOS_Schedule();
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_OK;
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void SysSchedYield(int type)
|
|
|
|
{
|
2021-03-11 18:43:57 +08:00
|
|
|
(void)LOS_TaskYield();
|
2020-09-08 10:21:39 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedGetScheduler(int id, int flag)
|
|
|
|
{
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
unsigned int intSave;
|
|
|
|
int policy;
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (flag < 0) {
|
|
|
|
if (OS_TID_CHECK_INVALID(id)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OS_TCB_FROM_TID(id);
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return -ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
policy = taskCB->policy;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return policy;
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_GetProcessScheduler(id);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedSetScheduler(int id, int policy, int prio, int flag)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (flag < 0) {
|
2021-03-11 18:43:57 +08:00
|
|
|
return -OsUserTaskSchedulerSet(id, policy, prio, true);
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if (prio < OS_USER_PROCESS_PRIORITY_HIGHEST) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (id == 0) {
|
|
|
|
id = (int)LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = OsPermissionToCheck(id, LOS_GetCurrProcessID());
|
|
|
|
if (ret < 0) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
return OsSetProcessScheduler(LOS_PRIO_PROCESS, id, prio, policy);
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedGetParam(int id, int flag)
|
|
|
|
{
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
int pri;
|
|
|
|
unsigned int intSave;
|
|
|
|
|
|
|
|
if (flag < 0) {
|
|
|
|
if (OS_TID_CHECK_INVALID(id)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OS_TCB_FROM_TID(id);
|
|
|
|
pri = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (pri != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return -pri;
|
|
|
|
}
|
|
|
|
|
|
|
|
pri = taskCB->priority;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return pri;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (id == 0) {
|
|
|
|
id = (int)LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (OS_PID_CHECK_INVALID(id)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return OsGetProcessPriority(LOS_PRIO_PROCESS, id);
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
int SysSetProcessPriority(int which, int who, unsigned int prio)
|
2020-09-08 10:21:39 +08:00
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (prio < OS_USER_PROCESS_PRIORITY_HIGHEST) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
if (who == 0) {
|
|
|
|
who = (int)LOS_GetCurrProcessID();
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
ret = OsPermissionToCheck(who, LOS_GetCurrProcessID());
|
2020-09-08 10:21:39 +08:00
|
|
|
if (ret < 0) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
return OsSetProcessScheduler(which, who, prio, LOS_GetProcessScheduler(who));
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
int SysSchedSetParam(int id, unsigned int prio, int flag)
|
2020-09-08 10:21:39 +08:00
|
|
|
{
|
2021-03-11 18:43:57 +08:00
|
|
|
if (flag < 0) {
|
|
|
|
return -OsUserTaskSchedulerSet(id, LOS_SCHED_RR, prio, false);
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
return SysSetProcessPriority(LOS_PRIO_PROCESS, id, prio);
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetProcessPriority(int which, int who)
|
|
|
|
{
|
|
|
|
if (who == 0) {
|
|
|
|
who = (int)LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
return OsGetProcessPriority(which, who);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedGetPriorityMin(int policy)
|
|
|
|
{
|
2021-03-11 18:43:57 +08:00
|
|
|
if (policy != LOS_SCHED_RR) {
|
2020-09-08 10:21:39 +08:00
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return OS_USER_PROCESS_PRIORITY_HIGHEST;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedGetPriorityMax(int policy)
|
|
|
|
{
|
2021-03-11 18:43:57 +08:00
|
|
|
if (policy != LOS_SCHED_RR) {
|
2020-09-08 10:21:39 +08:00
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return OS_USER_PROCESS_PRIORITY_LOWEST;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSchedRRGetInterval(int pid, struct timespec *tp)
|
|
|
|
{
|
2021-03-11 18:43:57 +08:00
|
|
|
unsigned int intSave;
|
2020-09-08 10:21:39 +08:00
|
|
|
int ret;
|
2021-03-11 18:43:57 +08:00
|
|
|
time_t timeSlice = 0;
|
2020-09-08 10:21:39 +08:00
|
|
|
struct timespec tv;
|
2021-03-11 18:43:57 +08:00
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
LosProcessCB *processCB = NULL;
|
2020-09-08 10:21:39 +08:00
|
|
|
|
|
|
|
if (tp == NULL) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (OS_PID_CHECK_INVALID(pid)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
if (pid == 0) {
|
|
|
|
processCB = OsCurrProcessGet();
|
|
|
|
} else {
|
|
|
|
processCB = OS_PCB_FROM_PID(pid);
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
2020-09-08 10:21:39 +08:00
|
|
|
/* if can not find process by pid return ESRCH */
|
2021-03-11 18:43:57 +08:00
|
|
|
if (OsProcessIsInactive(processCB)) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
2020-09-08 10:21:39 +08:00
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
LOS_DL_LIST_FOR_EACH_ENTRY(taskCB, &processCB->threadSiblingList, LosTaskCB, threadList) {
|
|
|
|
if (!OsTaskIsInactive(taskCB) && (taskCB->policy == LOS_SCHED_RR)) {
|
|
|
|
timeSlice += taskCB->initTimeSlice;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
|
|
|
|
timeSlice = timeSlice * OS_NS_PER_CYCLE;
|
|
|
|
tv.tv_sec = timeSlice / OS_SYS_NS_PER_SECOND;
|
|
|
|
tv.tv_nsec = timeSlice % OS_SYS_NS_PER_SECOND;
|
2020-09-08 10:21:39 +08:00
|
|
|
ret = LOS_ArchCopyToUser(tp, &tv, sizeof(struct timespec));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysWait(int pid, USER int *status, int options, void *rusage)
|
|
|
|
{
|
|
|
|
(void)rusage;
|
|
|
|
|
|
|
|
return LOS_Wait(pid, status, (unsigned int)options, NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysFork(void)
|
|
|
|
{
|
2021-02-27 17:51:32 +08:00
|
|
|
return OsClone(0, 0, 0);
|
2020-09-08 10:21:39 +08:00
|
|
|
}
|
|
|
|
|
2021-03-31 11:05:43 +08:00
|
|
|
int SysVfork(void)
|
|
|
|
{
|
|
|
|
return OsClone(CLONE_VFORK, 0, 0);
|
|
|
|
}
|
|
|
|
|
2020-09-08 10:21:39 +08:00
|
|
|
unsigned int SysGetPPID(void)
|
|
|
|
{
|
|
|
|
return OsCurrProcessGet()->parentProcessID;
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned int SysGetPID(void)
|
|
|
|
{
|
|
|
|
return LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetProcessGroupID(unsigned int pid, unsigned int gid)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (pid == 0) {
|
|
|
|
pid = LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (gid == 0) {
|
|
|
|
gid = pid;
|
|
|
|
} else if (gid <= OS_USER_PRIVILEGE_PROCESS_GROUP) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = OsPermissionToCheck(pid, gid);
|
|
|
|
if (ret < 0) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
return OsSetProcessGroupID(pid, gid);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetProcessGroupID(unsigned int pid)
|
|
|
|
{
|
|
|
|
if (pid == 0) {
|
|
|
|
pid = LOS_GetCurrProcessID();
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_GetProcessGroupID(pid);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetCurrProcessGroupID(void)
|
|
|
|
{
|
|
|
|
return LOS_GetCurrProcessGroupID();
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetUserID(void)
|
|
|
|
{
|
|
|
|
return LOS_GetUserID();
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetEffUserID(void)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
UINT32 intSave;
|
|
|
|
int euid;
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
euid = (int)OsCurrUserGet()->effUserID;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return euid;
|
|
|
|
#else
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetEffGID(void)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
UINT32 intSave;
|
|
|
|
int egid;
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
egid = (int)OsCurrUserGet()->effGid;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return egid;
|
|
|
|
#else
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetRealEffSaveUserID(int *ruid, int *euid, int *suid)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
int realUserID, effUserID, saveUserID;
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
unsigned int intSave;
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
realUserID = OsCurrUserGet()->userID;
|
|
|
|
effUserID = OsCurrUserGet()->effUserID;
|
|
|
|
saveUserID = OsCurrUserGet()->effUserID;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
#else
|
|
|
|
realUserID = 0;
|
|
|
|
effUserID = 0;
|
|
|
|
saveUserID = 0;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(ruid, &realUserID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(euid, &effUserID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(suid, &saveUserID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetUserID(int uid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
int ret = -EPERM;
|
|
|
|
unsigned int intSave;
|
|
|
|
|
|
|
|
if (uid < 0) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
User *user = OsCurrUserGet();
|
|
|
|
if (IsCapPermit(CAP_SETUID)) {
|
|
|
|
user->userID = uid;
|
|
|
|
user->effUserID = uid;
|
|
|
|
/* add process to a user */
|
|
|
|
} else if (user->userID != uid) {
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_OK;
|
|
|
|
/* add process to a user */
|
|
|
|
EXIT:
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
#else
|
|
|
|
if (uid != 0) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
static int SetRealEffSaveUserIDCheck(int ruid, int euid, int suid)
|
|
|
|
{
|
|
|
|
if ((ruid < 0) && (ruid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((euid < 0) && (euid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((suid < 0) && (suid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
int SysSetRealEffSaveUserID(int ruid, int euid, int suid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if ((ruid == -1) && (euid == -1) && (suid == -1)) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = SetRealEffSaveUserIDCheck(ruid, euid, suid);
|
|
|
|
if (ret != 0) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ruid >= 0) {
|
|
|
|
if (((euid != -1) && (euid != ruid)) || ((suid != -1) && (suid != ruid))) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return SysSetUserID(ruid);
|
|
|
|
} else if (euid >= 0) {
|
|
|
|
if ((suid != -1) && (suid != euid)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return SysSetUserID(euid);
|
|
|
|
} else {
|
|
|
|
return SysSetUserID(suid);
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
if ((ruid != 0) || (euid != 0) || (suid != 0)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetRealEffUserID(int ruid, int euid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
return SysSetRealEffSaveUserID(ruid, euid, -1);
|
|
|
|
#else
|
|
|
|
if ((ruid != 0) || (euid != 0)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetGroupID(int gid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
int ret = -EPERM;
|
|
|
|
unsigned int intSave;
|
|
|
|
unsigned int count;
|
|
|
|
unsigned int oldGid;
|
|
|
|
User *user = NULL;
|
|
|
|
|
|
|
|
if (gid < 0) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
user = OsCurrUserGet();
|
|
|
|
if (IsCapPermit(CAP_SETGID)) {
|
|
|
|
oldGid = user->gid;
|
|
|
|
user->gid = gid;
|
|
|
|
user->effGid = gid;
|
|
|
|
for (count = 0; count < user->groupNumber; count++) {
|
|
|
|
if (user->groups[count] == oldGid) {
|
|
|
|
user->groups[count] = gid;
|
|
|
|
ret = LOS_OK;
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else if (user->gid != gid) {
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_OK;
|
|
|
|
/* add process to a user */
|
|
|
|
EXIT:
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
#else
|
|
|
|
if (gid != 0) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetRealEffSaveGroupID(int *rgid, int *egid, int *sgid)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
int realGroupID, effGroupID, saveGroupID;
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
unsigned int intSave;
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
realGroupID = OsCurrUserGet()->gid;
|
|
|
|
effGroupID = OsCurrUserGet()->effGid;
|
|
|
|
saveGroupID = OsCurrUserGet()->effGid;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
#else
|
|
|
|
realGroupID = 0;
|
|
|
|
effGroupID = 0;
|
|
|
|
saveGroupID = 0;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(rgid, &realGroupID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(egid, &effGroupID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(sgid, &saveGroupID, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
static int SetRealEffSaveGroupIDCheck(int rgid, int egid, int sgid)
|
|
|
|
{
|
|
|
|
if ((rgid < 0) && (rgid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((egid < 0) && (egid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((sgid < 0) && (sgid != -1)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
int SysSetRealEffSaveGroupID(int rgid, int egid, int sgid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if ((rgid == -1) && (egid == -1) && (sgid == -1)) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = SetRealEffSaveGroupIDCheck(rgid, egid, sgid);
|
|
|
|
if (ret != 0) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (rgid >= 0) {
|
|
|
|
if (((egid != -1) && (egid != rgid)) || ((sgid != -1) && (sgid != rgid))) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return SysSetGroupID(rgid);
|
|
|
|
} else if (egid >= 0) {
|
|
|
|
if ((sgid != -1) && (sgid != egid)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return SysSetGroupID(egid);
|
|
|
|
} else {
|
|
|
|
return SysSetGroupID(sgid);
|
|
|
|
}
|
|
|
|
|
|
|
|
#else
|
|
|
|
if ((rgid != 0) || (egid != 0) || (sgid != 0)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetRealEffGroupID(int rgid, int egid)
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
return SysSetRealEffSaveGroupID(rgid, egid, -1);
|
|
|
|
#else
|
|
|
|
if ((rgid != 0) || (egid != 0)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysGetGroupID(void)
|
|
|
|
{
|
|
|
|
return LOS_GetGroupID();
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
static int SetGroups(int listSize, const int *safeList, int size)
|
|
|
|
{
|
|
|
|
User *oldUser = NULL;
|
|
|
|
unsigned int intSave;
|
|
|
|
|
|
|
|
User *newUser = LOS_MemAlloc(m_aucSysMem1, sizeof(User) + listSize * sizeof(int));
|
|
|
|
if (newUser == NULL) {
|
|
|
|
return -ENOMEM;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
oldUser = OsCurrUserGet();
|
|
|
|
(VOID)memcpy_s(newUser, sizeof(User), oldUser, sizeof(User));
|
|
|
|
if (safeList != NULL) {
|
|
|
|
(VOID)memcpy_s(newUser->groups, size * sizeof(int), safeList, size * sizeof(int));
|
|
|
|
}
|
|
|
|
if (listSize == size) {
|
|
|
|
newUser->groups[listSize] = oldUser->gid;
|
|
|
|
}
|
|
|
|
|
|
|
|
newUser->groupNumber = listSize + 1;
|
|
|
|
OsCurrProcessGet()->user = newUser;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
|
|
|
|
(void)LOS_MemFree(m_aucSysMem1, oldUser);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int GetGroups(int size, int list[])
|
|
|
|
{
|
|
|
|
unsigned int intSave;
|
|
|
|
int groupCount;
|
|
|
|
int ret;
|
|
|
|
int *safeList = NULL;
|
|
|
|
unsigned int listSize;
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
groupCount = OsCurrUserGet()->groupNumber;
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
|
|
|
|
listSize = groupCount * sizeof(int);
|
|
|
|
if (size == 0) {
|
|
|
|
return groupCount;
|
|
|
|
} else if (list == NULL) {
|
|
|
|
return -EFAULT;
|
|
|
|
} else if (size < groupCount) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
safeList = LOS_MemAlloc(m_aucSysMem1, listSize);
|
|
|
|
if (safeList == NULL) {
|
|
|
|
return -ENOMEM;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
(void)memcpy_s(safeList, listSize, &OsCurrProcessGet()->user->groups[0], listSize);
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(list, safeList, listSize);
|
|
|
|
if (ret != 0) {
|
|
|
|
groupCount = -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
(void)LOS_MemFree(m_aucSysMem1, safeList);
|
|
|
|
return groupCount;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
int SysGetGroups(int size, int list[])
|
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
return GetGroups(size, list);
|
|
|
|
#else
|
|
|
|
int group = 0;
|
|
|
|
int groupCount = 1;
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (size == 0) {
|
|
|
|
return groupCount;
|
|
|
|
} else if (list == NULL) {
|
|
|
|
return -EFAULT;
|
|
|
|
} else if (size < groupCount) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyToUser(list, &group, sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return groupCount;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
int SysSetGroups(int size, const int list[])
|
2020-09-08 10:21:39 +08:00
|
|
|
{
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
int ret;
|
|
|
|
int gid;
|
|
|
|
int listSize = size;
|
|
|
|
unsigned int count;
|
|
|
|
int *safeList = NULL;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if ((size != 0) && (list == NULL)) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((size < 0) || (size > OS_GROUPS_NUMBER_MAX)) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef LOSCFG_SECURITY_CAPABILITY
|
|
|
|
if (!IsCapPermit(CAP_SETGID)) {
|
|
|
|
return -EPERM;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (size != 0) {
|
|
|
|
safeList = LOS_MemAlloc(m_aucSysMem1, size * sizeof(int));
|
|
|
|
if (safeList == NULL) {
|
|
|
|
return -ENOMEM;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyFromUser(safeList, list, size * sizeof(int));
|
|
|
|
if (ret != 0) {
|
|
|
|
ret = -EFAULT;
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
gid = OsCurrUserGet()->gid;
|
|
|
|
for (count = 0; count < size; count++) {
|
|
|
|
if (safeList[count] == gid) {
|
|
|
|
listSize = size - 1;
|
|
|
|
} else if (safeList[count] < 0) {
|
|
|
|
ret = -EINVAL;
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = SetGroups(listSize, safeList, size);
|
|
|
|
EXIT:
|
|
|
|
if (safeList != NULL) {
|
|
|
|
(void)LOS_MemFree(m_aucSysMem1, safeList);
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
#else
|
|
|
|
return 0;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned int SysCreateUserThread(const TSK_ENTRY_FUNC func, const UserTaskParam *userParam, bool joinable)
|
|
|
|
{
|
|
|
|
TSK_INIT_PARAM_S param = { 0 };
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
ret = LOS_ArchCopyFromUser(&(param.userParam), userParam, sizeof(UserTaskParam));
|
|
|
|
if (ret != 0) {
|
|
|
|
return OS_INVALID_VALUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
param.pfnTaskEntry = func;
|
|
|
|
if (joinable == TRUE) {
|
|
|
|
param.uwResved = OS_TASK_FLAG_PTHREAD_JOIN;
|
|
|
|
} else {
|
|
|
|
param.uwResved = OS_TASK_FLAG_DETACHED;
|
|
|
|
}
|
|
|
|
|
|
|
|
return OsCreateUserTask(OS_INVALID_VALUE, ¶m);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysSetThreadArea(const char *area)
|
|
|
|
{
|
|
|
|
unsigned int intSave;
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
LosProcessCB *processCB = NULL;
|
|
|
|
unsigned int ret = LOS_OK;
|
|
|
|
|
|
|
|
if (!LOS_IsUserAddress((unsigned long)(uintptr_t)area)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OsCurrTaskGet();
|
|
|
|
processCB = OS_PCB_FROM_PID(taskCB->processID);
|
|
|
|
if (processCB->processMode != OS_USER_MODE) {
|
|
|
|
ret = EPERM;
|
|
|
|
goto OUT;
|
|
|
|
}
|
|
|
|
|
|
|
|
taskCB->userArea = (unsigned long)(uintptr_t)area;
|
|
|
|
OUT:
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
char *SysGetThreadArea(void)
|
|
|
|
{
|
|
|
|
return (char *)(OsCurrTaskGet()->userArea);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysUserThreadSetDeatch(unsigned int taskID)
|
|
|
|
{
|
|
|
|
unsigned int intSave;
|
|
|
|
int ret;
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
|
|
|
|
if (OS_TID_CHECK_INVALID(taskID)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OS_TCB_FROM_TID(taskID);
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
|
2021-05-08 11:23:08 +08:00
|
|
|
ret = OsTaskSetDetachUnsafe(taskCB);
|
2020-09-08 10:21:39 +08:00
|
|
|
|
|
|
|
EXIT:
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysUserThreadDetach(unsigned int taskID)
|
|
|
|
{
|
|
|
|
unsigned int intSave;
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
unsigned int ret;
|
|
|
|
|
|
|
|
if (OS_TID_CHECK_INVALID(taskID)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OS_TCB_FROM_TID(taskID);
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = OsTaskDeleteUnsafe(taskCB, OS_PRO_EXIT_OK, intSave);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
return ESRCH;
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysThreadJoin(unsigned int taskID)
|
|
|
|
{
|
|
|
|
unsigned int intSave;
|
|
|
|
unsigned int ret;
|
|
|
|
LosTaskCB *taskCB = NULL;
|
|
|
|
|
|
|
|
if (OS_TID_CHECK_INVALID(taskID)) {
|
|
|
|
return EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
taskCB = OS_TCB_FROM_TID(taskID);
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(taskCB);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
goto EXIT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = OsTaskJoinPendUnsafe(OS_TCB_FROM_TID(taskID));
|
|
|
|
|
|
|
|
EXIT:
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
void SysUserExitGroup(int status)
|
|
|
|
{
|
|
|
|
OsTaskExitGroup((unsigned int)status);
|
|
|
|
}
|
|
|
|
|
|
|
|
void SysThreadExit(int status)
|
|
|
|
{
|
|
|
|
OsTaskToExit(OsCurrTaskGet(), (unsigned int)status);
|
|
|
|
}
|
|
|
|
|
|
|
|
int SysFutex(const unsigned int *uAddr, unsigned int flags, int val,
|
|
|
|
unsigned int absTime, const unsigned int *newUserAddr)
|
|
|
|
{
|
|
|
|
if ((flags & FUTEX_MASK) == FUTEX_REQUEUE) {
|
|
|
|
return -OsFutexRequeue(uAddr, flags, val, absTime, newUserAddr);
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((flags & FUTEX_MASK) == FUTEX_WAKE) {
|
|
|
|
return -OsFutexWake(uAddr, flags, val);
|
|
|
|
}
|
|
|
|
|
|
|
|
return -OsFutexWait(uAddr, flags, val, absTime);
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned int SysGetTid(void)
|
|
|
|
{
|
|
|
|
return OsCurrTaskGet()->taskID;
|
|
|
|
}
|
|
|
|
|
2021-03-11 18:43:57 +08:00
|
|
|
/* If flag >= 0, the process mode is used. If flag < 0, the thread mode is used. */
|
|
|
|
static int SchedAffinityParameterPreprocess(int id, int flag, unsigned int *taskID, unsigned int *processID)
|
|
|
|
{
|
|
|
|
if (flag >= 0) {
|
|
|
|
if (OS_PID_CHECK_INVALID(id)) {
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
*taskID = (id == 0) ? (OsCurrTaskGet()->taskID) : (OS_PCB_FROM_PID((UINT32)id)->threadGroupID);
|
|
|
|
*processID = (id == 0) ? (OS_TCB_FROM_TID(*taskID)->processID) : id;
|
|
|
|
} else {
|
|
|
|
if (OS_TID_CHECK_INVALID(id)) {
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
*taskID = id;
|
|
|
|
*processID = OS_INVALID_VALUE;
|
|
|
|
}
|
|
|
|
return LOS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If flag >= 0, the process mode is used. If flag < 0, the thread mode is used. */
|
|
|
|
int SysSchedGetAffinity(int id, unsigned int *cpuset, int flag)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
unsigned int processID;
|
|
|
|
unsigned int taskID;
|
|
|
|
unsigned int intSave;
|
|
|
|
unsigned int cpuAffiMask;
|
|
|
|
|
|
|
|
ret = SchedAffinityParameterPreprocess(id, flag, &taskID, &processID);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
if (flag >= 0) {
|
|
|
|
if (OsProcessIsInactive(OS_PCB_FROM_PID(processID))) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(OS_TCB_FROM_TID(taskID));
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
if (ret == EINVAL) {
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
return -ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#if (LOSCFG_KERNEL_SMP == YES)
|
|
|
|
cpuAffiMask = (unsigned int)OS_TCB_FROM_TID(taskID)->cpuAffiMask;
|
|
|
|
#else
|
|
|
|
cpuAffiMask = 1;
|
|
|
|
#endif /* LOSCFG_KERNEL_SMP */
|
|
|
|
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
ret = LOS_ArchCopyToUser(cpuset, &cpuAffiMask, sizeof(unsigned int));
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
return -EFAULT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* If flag >= 0, the process mode is used. If flag < 0, the thread mode is used. */
|
|
|
|
int SysSchedSetAffinity(int id, const unsigned short cpuset, int flag)
|
|
|
|
{
|
|
|
|
int ret;
|
|
|
|
unsigned int processID;
|
|
|
|
unsigned int taskID;
|
|
|
|
unsigned int intSave;
|
|
|
|
unsigned short currCpuMask;
|
|
|
|
bool needSched = FALSE;
|
|
|
|
|
|
|
|
if (cpuset > LOSCFG_KERNEL_CPU_MASK) {
|
|
|
|
return -EINVAL;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = SchedAffinityParameterPreprocess(id, flag, &taskID, &processID);
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (flag >= 0) {
|
|
|
|
ret = OsPermissionToCheck(processID, LOS_GetCurrProcessID());
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
if (OsProcessIsInactive(OS_PCB_FROM_PID(processID))) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
SCHEDULER_LOCK(intSave);
|
|
|
|
ret = OsUserTaskOperatePermissionsCheck(OS_TCB_FROM_TID(taskID));
|
|
|
|
if (ret != LOS_OK) {
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
if (ret == EINVAL) {
|
|
|
|
return -ESRCH;
|
|
|
|
}
|
|
|
|
return -ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
needSched = OsTaskCpuAffiSetUnsafe(taskID, cpuset, &currCpuMask);
|
|
|
|
SCHEDULER_UNLOCK(intSave);
|
|
|
|
if (needSched && OS_SCHEDULER_ACTIVE) {
|
|
|
|
LOS_MpSchedule(currCpuMask);
|
|
|
|
LOS_Schedule();
|
|
|
|
}
|
|
|
|
|
|
|
|
return LOS_OK;
|
|
|
|
}
|
|
|
|
|