// Copyright (c) 2023 Yuki Kimoto
// MIT License
#include "spvm_native.h"
#include <unistd.h>
#include <sys/types.h>
#include <errno.h>
#include <stdlib.h>
#include <assert.h>
#if defined(_WIN32)
// None
#else
#include <sys/types.h>
#include <sys/resource.h>
#include <sys/wait.h>
#endif
#if defined(_WIN32)
#include "spvm_sys_windows.h"
#endif
static const char* FILE_NAME = "Sys/Process.c";
int32_t SPVM__Sys__Process__fork(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#fork method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t status = fork();
if (status == -1) {
env->die(env, stack, "[System Error]fork() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = status;
return 0;
#endif
}
int32_t SPVM__Sys__Process__getpriority(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#getpriority method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t which = stack[0].ival;
int32_t who = stack[1].ival;
errno = 0;
int32_t nice = getpriority(which, who);
if (errno != 0) {
env->die(env, stack, "[System Error]getpriority() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = nice;
return 0;
#endif
}
int32_t SPVM__Sys__Process__setpriority(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#setpriority method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t which = stack[0].ival;
int32_t who = stack[1].ival;
int32_t prio = stack[2].ival;
int32_t status = setpriority(which, who, prio);
if (status == -1) {
env->die(env, stack, "[System Error]setpriority() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = status;
return 0;
#endif
}
int32_t SPVM__Sys__Process__sleep(SPVM_ENV* env, SPVM_VALUE* stack) {
int32_t seconds = stack[0].ival;
int32_t rest_time = sleep(seconds);
stack[0].ival = rest_time;
return 0;
}
int32_t SPVM__Sys__Process__usleep(SPVM_ENV* env, SPVM_VALUE* stack) {
int64_t usec = stack[0].lval;
int32_t status = usleep(usec);
if (status == -1) {
env->die(env, stack, "[System Error]usleep() failed.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = status;
return 0;
}
int32_t SPVM__Sys__Process__wait(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#wait method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t* wstatus_ref = stack[0].iref;
if (!wstatus_ref) {
return env->die(env, stack, "The reference of the output wait status $wstatus_ref must be defined.", __func__, FILE_NAME, __LINE__);
}
int wstatus_int;
int32_t process_id = wait(&wstatus_int);
*wstatus_ref = wstatus_int;
if (process_id == -1) {
env->die(env, stack, "[System Error]wait() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = process_id;
return 0;
#endif
}
int32_t SPVM__Sys__Process__waitpid(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#waitpid method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t pid = stack[0].ival;
int32_t* wstatus_ref = stack[1].iref;
int32_t options = stack[2].ival;
if (!wstatus_ref) {
return env->die(env, stack, "The reference of the output wait status $wstatus_ref must be defined.", __func__, FILE_NAME, __LINE__);
}
int wstatus_int;
int32_t process_id = waitpid(pid, &wstatus_int, options);
*wstatus_ref = wstatus_int;
if (process_id == -1) {
env->die(env, stack, "[System Error]waitpid() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = process_id;
return 0;
#endif
}
int32_t SPVM__Sys__Process__system(SPVM_ENV* env, SPVM_VALUE* stack) {
void* obj_command = stack[0].oval;
const char* command = NULL;
if (obj_command) {
command = env->get_chars(env, stack, obj_command);
}
int32_t wstatus = system(command);
if (wstatus == -1) {
env->die(env, stack, "[System Error]system() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = wstatus;
return 0;
}
int32_t SPVM__Sys__Process__exit(SPVM_ENV* env, SPVM_VALUE* stack) {
int32_t stauts = stack[0].ival;
exit(stauts);
return 0;
}
int32_t SPVM__Sys__Process__pipe(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#pipe method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
void* obj_pipefds = stack[0].oval;
if (!obj_pipefds) {
return env->die(env, stack, "The pipe file descriptors $pipefds must be defined.", __func__, FILE_NAME, __LINE__);
}
int32_t pipefds_length = env->length(env, stack, obj_pipefds);
if (!(pipefds_length == 2)) {
return env->die(env, stack, "The length of the file descriptors $pipefds must 2.", __func__, FILE_NAME, __LINE__);
}
int32_t* pipefds = env->get_elems_int(env, stack, obj_pipefds);
int pipefds_int[2] = {pipefds[0], pipefds[1]};
int32_t status = pipe(pipefds_int);
if (status == -1) {
env->die(env, stack, "[System Error]pipe() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
pipefds[0] = pipefds_int[0];
pipefds[1] = pipefds_int[1];
stack[0].ival = status;
return 0;
#endif
}
int32_t SPVM__Sys__Process___pipe(SPVM_ENV* env, SPVM_VALUE* stack) {
#if !defined(_WIN32)
env->die(env, stack, "Sys::Process#_pipe method is not supported in this system(!defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
void* obj_pipefds = stack[0].oval;
unsigned int psize = (uint32_t)stack[1].ival;
if (!obj_pipefds) {
return env->die(env, stack, "The file descriptors $pipefds must be defined.", __func__, FILE_NAME, __LINE__);
}
int32_t pipefds_length = env->length(env, stack, obj_pipefds);
if (!(pipefds_length == 2)) {
return env->die(env, stack, "The length of the file descriptors $pipefds must 2.", __func__, FILE_NAME, __LINE__);
}
int32_t* pipefds = env->get_elems_int(env, stack, obj_pipefds);
int textmode = stack[2].ival;
int pipefds_int[2] = {pipefds[0], pipefds[1]};
int32_t status = _pipe(pipefds_int, psize, textmode);
if (status == -1) {
env->die(env, stack, "[System Error]_pipe() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
pipefds[0] = pipefds_int[0];
pipefds[1] = pipefds_int[1];
stack[0].ival = status;
return 0;
#endif
}
int32_t SPVM__Sys__Process__getpgid(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#getpgid method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t pid = stack[0].ival;
int32_t process_group_id = getpgid(pid);
if (process_group_id == -1) {
env->die(env, stack, "[System Error]getpgid() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = process_group_id;
return 0;
#endif
}
int32_t SPVM__Sys__Process__setpgid(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#setpgid method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t pid = stack[0].ival;
int32_t pgid = stack[1].ival;
int32_t status = setpgid(pid, pgid);
if (status == -1) {
env->die(env, stack, "[System Error]setpgid() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = status;
return 0;
#endif
}
int32_t SPVM__Sys__Process__getpid(SPVM_ENV* env, SPVM_VALUE* stack) {
int32_t process_id = getpid();
stack[0].ival = process_id;
return 0;
}
int32_t SPVM__Sys__Process__getppid(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#getppid method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t parent_process_id = getppid();
stack[0].ival = parent_process_id;
return 0;
#endif
}
int32_t SPVM__Sys__Process__execv(SPVM_ENV* env, SPVM_VALUE* stack) {
int32_t error_id = 0;
void* obj_path = stack[0].oval;
void* obj_args = stack[1].oval;
if (!obj_path) {
return env->die(env, stack, "The command path $path must be defined.", __func__, FILE_NAME, __LINE__);
}
const char* path = env->get_chars(env, stack, obj_path);
if (!obj_args) {
return env->die(env, stack, "The command arguments $args must be defined.", __func__, FILE_NAME, __LINE__);
}
int32_t args_length = env->length(env, stack, obj_args);
char** argv = env->new_memory_block(env, stack, sizeof(char*) * (args_length + 1));
for (int32_t i = 0; i < args_length; i++) {
void* obj_arg = env->get_elem_object(env, stack, obj_args, i);
if (!obj_arg) {
return env->die(env, stack, "The %dth element of the command arguments $args must be defined.", __func__, FILE_NAME, __LINE__, i);
}
char* arg = (char*)env->get_chars(env, stack, obj_arg);
argv[i] = arg;
}
assert(argv[args_length] == NULL);
#if defined(_WIN32)
WCHAR* path_w = spvm_sys_windows_utf8_to_win_wchar(env, stack, path, &error_id, __func__, FILE_NAME, __LINE__);
if (error_id) {
return error_id;
}
WCHAR** argv_w = env->new_memory_block(env, stack, sizeof(WCHAR*) * (args_length + 1));
for (int32_t i = 0; i < args_length; i++) {
char* arg = argv[i];
WCHAR* arg_w = spvm_sys_windows_utf8_to_win_wchar(env, stack, arg, &error_id, __func__, FILE_NAME, __LINE__);
if (error_id) {
return error_id;
}
argv_w[i] = arg_w;
}
int32_t status = _wexecv(path_w, (const WCHAR *const *)argv_w);
#else
int32_t status = execv(path, argv);
#endif
env->free_memory_block(env, stack, argv);
if (status == -1) {
env->die(env, stack, "[System Error]execv() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = status;
return 0;
}
int32_t SPVM__Sys__Process__WIFEXITED(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WCONTINUED
stack[0].ival = WIFEXITED(stack[0].ival);
return 0;
#else
env->die(env, stack, "WIFEXITED is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WEXITSTATUS(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WEXITSTATUS
stack[0].ival = WEXITSTATUS(stack[0].ival);
return 0;
#else
env->die(env, stack, "WEXITSTATUS is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WIFSIGNALED(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WIFSIGNALED
stack[0].ival = WIFSIGNALED(stack[0].ival);
return 0;
#else
env->die(env, stack, "WIFSIGNALED is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WTERMSIG(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WTERMSIG
stack[0].ival = WTERMSIG(stack[0].ival);
return 0;
#else
env->die(env, stack, "WTERMSIG is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WCOREDUMP(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WCOREDUMP
stack[0].ival = WCOREDUMP(stack[0].ival);
return 0;
#else
env->die(env, stack, "WCOREDUMP is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WIFSTOPPED(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WIFSTOPPED
stack[0].ival = WIFSTOPPED(stack[0].ival);
return 0;
#else
env->die(env, stack, "WIFSTOPPED is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WSTOPSIG(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WSTOPSIG
stack[0].ival = WSTOPSIG(stack[0].ival);
return 0;
#else
env->die(env, stack, "WSTOPSIG is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__WIFCONTINUED(SPVM_ENV* env, SPVM_VALUE* stack) {
#ifdef WIFCONTINUED
stack[0].ival = WIFCONTINUED(stack[0].ival);
return 0;
#else
env->die(env, stack, "WIFCONTINUED is not defined in this system.", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#endif
}
int32_t SPVM__Sys__Process__setsid(SPVM_ENV* env, SPVM_VALUE* stack) {
#if defined(_WIN32)
env->die(env, stack, "Sys::Process#setsid method is not supported in this system(defined(_WIN32)).", __func__, FILE_NAME, __LINE__);
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_NOT_SUPPORTED_CLASS;
#else
int32_t session_id = setsid();
if (session_id == -1) {
env->die(env, stack, "[System Error]setsid() failed(%d: %s).", __func__, FILE_NAME, __LINE__, errno, env->strerror_nolen(env, stack, errno));
return SPVM_NATIVE_C_BASIC_TYPE_ID_ERROR_SYSTEM_CLASS;
}
stack[0].ival = session_id;
return 0;
#endif
}