Implement environment variables

kernel/syscall/syscall.c

@@ -827,6 +827,80 @@ DEFINE_SYSCALL (sys_create_volume) {
   return SYSRESULT (vfs_create_volume (proc, rctx, key, type, device, false));
 }
 
+/* int env_set (char* key, void* buffer, size_t size) */
+DEFINE_SYSCALL (sys_env_set) {
+  uint64_t fp, fpg;
+
+  uintptr_t uvaddr_key = a1;
+  uintptr_t uvaddr_buffer = a2;
+  size_t size = (size_t)a3;
+
+  struct limine_hhdm_response* hhdm = limine_hhdm_request.response;
+
+  uintptr_t out_paddr;
+
+  spin_lock (&proc->lock, &fp);
+  struct procgroup* procgroup = proc->procgroup;
+  spin_unlock (&proc->lock, fp);
+
+  spin_lock (&procgroup->lock, &fpg);
+
+  out_paddr = mm_v2p (&procgroup->pd, uvaddr_key);
+
+  if (out_paddr == 0) {
+    spin_unlock (&procgroup->lock, fpg);
+    return SYSRESULT (-ST_BAD_ADDRESS_SPACE);
+  }
+
+  const char* key = (const char*)((uintptr_t)hhdm->offset + out_paddr);
+
+  spin_unlock (&procgroup->lock, fpg);
+
+  void* buffer = sys_get_user_buffer (procgroup, uvaddr_buffer, size);
+
+  if (buffer == NULL)
+    return SYSRESULT (-ST_BAD_ADDRESS_SPACE);
+
+  return SYSRESULT (proc_env_set (proc, key, buffer, size));
+}
+
+/* int env_get (char* key, void* buffer, size_t size) */
+DEFINE_SYSCALL (sys_env_get) {
+  uint64_t fp, fpg;
+
+  uintptr_t uvaddr_key = a1;
+  uintptr_t uvaddr_buffer = a2;
+  size_t size = (size_t)a3;
+
+  struct limine_hhdm_response* hhdm = limine_hhdm_request.response;
+
+  uintptr_t out_paddr;
+
+  spin_lock (&proc->lock, &fp);
+  struct procgroup* procgroup = proc->procgroup;
+  spin_unlock (&proc->lock, fp);
+
+  spin_lock (&procgroup->lock, &fpg);
+
+  out_paddr = mm_v2p (&procgroup->pd, uvaddr_key);
+
+  if (out_paddr == 0) {
+    spin_unlock (&procgroup->lock, fpg);
+    return SYSRESULT (-ST_BAD_ADDRESS_SPACE);
+  }
+
+  const char* key = (const char*)((uintptr_t)hhdm->offset + out_paddr);
+
+  spin_unlock (&procgroup->lock, fpg);
+
+  void* buffer = sys_get_user_buffer (procgroup, uvaddr_buffer, size);
+
+  if (buffer == NULL)
+    return SYSRESULT (-ST_BAD_ADDRESS_SPACE);
+
+  return SYSRESULT (proc_env_set (proc, key, buffer, size));
+}
+
 static syscall_handler_func_t handler_table[] = {
     [SYS_QUIT] = &sys_quit,
     [SYS_TEST] = &sys_test,
@@ -857,6 +931,8 @@ static syscall_handler_func_t handler_table[] = {
     [SYS_CREATE_DIR] = &sys_create_dir,
     [SYS_REMOVE] = &sys_remove,
     [SYS_CREATE_VOLUME] = &sys_create_volume,
+    [SYS_ENV_SET] = &sys_env_set,
+    [SYS_ENV_GET] = &sys_env_get,
 };
 
 syscall_handler_func_t syscall_find_handler (int syscall_num) {