From 367ad9d23ccd56b0394d213c78ed2da5fe2fe592 Mon Sep 17 00:00:00 2001
From: 0xf00sec <159052166+0xf00sec@users.noreply.github.com>
Date: Sun, 22 Jun 2025 16:01:15 +0300
Subject: UserMgr.cc

rt_copy_memory calls were replaced with rt_copy_memory_safe, and #include <NeKit/Utils.h>
---
 dev/kernel/src/UserMgr.cc | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'dev/kernel/src/UserMgr.cc')

diff --git a/dev/kernel/src/UserMgr.cc b/dev/kernel/src/UserMgr.cc
index 8eade85e..78a306f7 100644
--- a/dev/kernel/src/UserMgr.cc
+++ b/dev/kernel/src/UserMgr.cc
@@ -15,6 +15,9 @@
 #include <KernelKit/KPC.h>
 #include <KernelKit/UserMgr.h>
 #include <NeKit/KernelPanic.h>
+#include <NeKit/KString.h>
+#include <NeKit/ThreadLocalStorage.h>
+#include <NeKit/Utils.h>
 
 #define kStdUserType (0xEE)
 #define kSuperUserType (0xEF)
@@ -56,14 +59,14 @@ namespace Detail {
 ////////////////////////////////////////////////////////////
 User::User(const Int32& sel, const Char* user_name) : mUserRing((UserRingKind) sel) {
   MUST_PASS(sel >= 0);
-  rt_copy_memory((VoidPtr) user_name, this->mUserName, rt_string_len(user_name));
+  rt_copy_memory_safe((VoidPtr) user_name, this->mUserName, rt_string_len(user_name), kMaxUserNameLen);
 }
 
 ////////////////////////////////////////////////////////////
 /// @brief User ring constructor.
 ////////////////////////////////////////////////////////////
 User::User(const UserRingKind& ring_kind, const Char* user_name) : mUserRing(ring_kind) {
-  rt_copy_memory((VoidPtr) user_name, this->mUserName, rt_string_len(user_name));
+  rt_copy_memory_safe((VoidPtr) user_name, this->mUserName, rt_string_len(user_name), kMaxUserNameLen);
 }
 
 ////////////////////////////////////////////////////////////
-- 
cgit v1.2.3