134 lines
2.9 KiB
C++
134 lines
2.9 KiB
C++
#include "memory/MemoryManager.h"
|
|
#include "sys/Syscall.h"
|
|
#include "thread/Scheduler.h"
|
|
#include <bits/atfile.h>
|
|
|
|
Result<u64> sys_getpid(Registers*, SyscallArgs)
|
|
{
|
|
return Scheduler::current()->id;
|
|
}
|
|
|
|
Result<u64> sys_getppid(Registers*, SyscallArgs)
|
|
{
|
|
auto* parent = Scheduler::current()->parent;
|
|
|
|
return parent ? parent->id : 0;
|
|
}
|
|
|
|
Result<u64> sys_getuid(Registers*, SyscallArgs)
|
|
{
|
|
return Scheduler::current()->auth.uid;
|
|
}
|
|
|
|
Result<u64> sys_geteuid(Registers*, SyscallArgs)
|
|
{
|
|
return Scheduler::current()->auth.euid;
|
|
}
|
|
|
|
Result<u64> sys_getgid(Registers*, SyscallArgs)
|
|
{
|
|
return Scheduler::current()->auth.gid;
|
|
}
|
|
|
|
Result<u64> sys_getegid(Registers*, SyscallArgs)
|
|
{
|
|
return Scheduler::current()->auth.egid;
|
|
}
|
|
|
|
Result<u64> sys_setuid(Registers*, SyscallArgs args)
|
|
{
|
|
u32 uid = (u32)args[0];
|
|
|
|
Credentials& auth = Scheduler::current()->auth;
|
|
|
|
if (auth.euid == 0)
|
|
{
|
|
auth.uid = auth.euid = auth.suid = uid;
|
|
return 0;
|
|
}
|
|
|
|
if (uid != auth.uid && uid != auth.suid) return err(EPERM);
|
|
auth.euid = uid;
|
|
|
|
return 0;
|
|
}
|
|
|
|
Result<u64> sys_seteuid(Registers*, SyscallArgs args)
|
|
{
|
|
u32 uid = (u32)args[0];
|
|
|
|
Credentials& auth = Scheduler::current()->auth;
|
|
|
|
if (auth.euid != 0 && uid != auth.uid && uid != auth.suid) return err(EPERM);
|
|
auth.euid = uid;
|
|
|
|
return 0;
|
|
}
|
|
|
|
Result<u64> sys_setgid(Registers*, SyscallArgs args)
|
|
{
|
|
u32 gid = (u32)args[0];
|
|
|
|
Credentials& auth = Scheduler::current()->auth;
|
|
|
|
if (auth.euid == 0)
|
|
{
|
|
auth.gid = auth.egid = auth.sgid = gid;
|
|
return 0;
|
|
}
|
|
|
|
if (gid != auth.gid && gid != auth.sgid) return err(EPERM);
|
|
auth.egid = gid;
|
|
|
|
return 0;
|
|
}
|
|
|
|
Result<u64> sys_setegid(Registers*, SyscallArgs args)
|
|
{
|
|
u32 gid = (u32)args[0];
|
|
|
|
Credentials& auth = Scheduler::current()->auth;
|
|
|
|
if (auth.euid != 0 && gid != auth.gid && gid != auth.sgid) return err(EPERM);
|
|
auth.egid = gid;
|
|
|
|
return 0;
|
|
}
|
|
|
|
Result<u64> sys_fchmodat(Registers*, SyscallArgs args)
|
|
{
|
|
int dirfd = (int)args[0];
|
|
auto path = TRY(MemoryManager::strdup_from_user(args[1]));
|
|
mode_t mode = (mode_t)args[2];
|
|
int flags = (int)args[3];
|
|
|
|
auto* current = Scheduler::current();
|
|
|
|
auto inode = TRY(current->resolve_atfile(dirfd, path, flags & AT_EMPTY_PATH, !(flags & AT_SYMLINK_NOFOLLOW)));
|
|
|
|
if (current->auth.euid != 0 && current->auth.euid != inode->uid()) return err(EPERM);
|
|
|
|
TRY(inode->chmod(mode));
|
|
|
|
return 0;
|
|
}
|
|
|
|
Result<u64> sys_fchownat(Registers*, SyscallArgs args)
|
|
{
|
|
int dirfd = (int)args[0];
|
|
auto path = TRY(MemoryManager::strdup_from_user(args[1]));
|
|
u32 uid = (u32)args[2];
|
|
u32 gid = (u32)args[3];
|
|
int flags = (int)args[4];
|
|
|
|
auto* current = Scheduler::current();
|
|
|
|
auto inode = TRY(current->resolve_atfile(dirfd, path, flags & AT_EMPTY_PATH, !(flags & AT_SYMLINK_NOFOLLOW)));
|
|
|
|
if (current->auth.euid != 0) return err(EPERM);
|
|
|
|
TRY(inode->chown(uid == (u32)-1 ? inode->uid() : uid, gid == (u32)-1 ? inode->gid() : gid));
|
|
|
|
return 0;
|
|
}
|