/************************************************ process.c - $Author$ $Date$ created at: Tue Aug 10 14:30:50 JST 1993 Copyright (C) 1993-1999 Yukihiro Matsumoto ************************************************/ #include "ruby.h" #include "rubysig.h" #include #include #include #ifdef HAVE_UNISTD_H #include #endif #ifndef NT #ifdef HAVE_SYS_TIME_H # include #else struct timeval { long tv_sec; /* seconds */ long tv_usec; /* and microseconds */ }; #endif #endif /* NT */ #include struct timeval rb_time_interval _((VALUE)); #ifdef HAVE_SYS_WAIT_H # include #endif #ifdef HAVE_GETPRIORITY # include #endif #ifdef HAVE_VFORK_H #include #endif #include "st.h" #ifdef USE_CWGUSI # include # include "macruby_missing.h" #endif #ifdef __EMX__ #undef HAVE_GETPGRP #endif static VALUE get_pid() { return INT2FIX(getpid()); } static VALUE get_ppid() { #ifdef NT return INT2FIX(0); #else return INT2FIX(getppid()); #endif } VALUE rb_last_status = Qnil; #if !defined(HAVE_WAITPID) && !defined(HAVE_WAIT4) #define NO_WAITPID static st_table *pid_tbl; #endif static int rb_waitpid(pid, flags, st) int pid; int flags; int *st; { int result; #ifndef NO_WAITPID int oflags = flags; if (!rb_thread_alone()) { /* there're other threads to run */ flags |= WNOHANG; } retry: #ifdef HAVE_WAITPID result = waitpid(pid, st, flags); #else /* HAVE_WAIT4 */ result = wait4(pid, st, flags, NULL); #endif if (result < 0) { if (errno == EINTR) { rb_thread_schedule(); goto retry; } return -1; } if (result == 0) { if (oflags & WNOHANG) return 0; rb_thread_schedule(); if (rb_thread_alone()) flags = oflags; goto retry; } #else /* NO_WAITPID */ if (pid_tbl && st_lookup(pid_tbl, pid, st)) { rb_last_status = INT2FIX(*st); st_delete(pid_tbl, &pid, NULL); return pid; } if (flags) { rb_raise(rb_eArgError, "can't do waitpid with flags"); } for (;;) { result = wait(st); if (result < 0) { if (errno == EINTR) { rb_thread_schedule(); continue; } return -1; } if (result == pid) { break; } if (!pid_tbl) pid_tbl = st_init_numtable(); st_insert(pid_tbl, pid, st); if (!rb_thread_alone()) rb_thread_schedule(); } #endif rb_last_status = INT2FIX(*st); return result; } #ifdef NO_WAITPID struct wait_data { int pid; int status; }; static int wait_each(key, value, data) int key, value; struct wait_data *data; { if (data->status != -1) return ST_STOP; data->pid = key; data->status = value; return ST_DELETE; } #endif static VALUE rb_f_wait() { int pid, state; #ifdef NO_WAITPID struct wait_data data; data.status = -1; st_foreach(pid_tbl, wait_each, &data); if (data.status != -1) { rb_last_status = data.status; return INT2FIX(data.pid); } while ((pid = wait(&state)) < 0) { if (errno == EINTR) { rb_thread_schedule(); continue; } rb_sys_fail(0); } rb_last_status = INT2FIX(state); #else if ((pid = rb_waitpid(-1, 0, &state)) < 0) rb_sys_fail(0); #endif return INT2FIX(pid); } static VALUE rb_f_waitpid(obj, vpid, vflags) VALUE obj, vpid, vflags; { int pid, flags, status; if (NIL_P(vflags)) flags = 0; else flags = FIX2UINT(vflags); if ((pid = rb_waitpid(FIX2UINT(vpid), flags, &status)) < 0) rb_sys_fail(0); return INT2FIX(pid); } char *strtok(); #ifdef HAVE_SETITIMER #define before_exec() rb_thread_stop_timer() #define after_exec() rb_thread_start_timer() #else #define before_exec() #define after_exec() #endif extern char *dln_find_exe(); static void security(str) char *str; { if (rb_safe_level() > 0) { #ifndef USE_CWGUSI if (rb_env_path_tainted()) { rb_raise(rb_eSecurityError, "Insecure PATH - %s", str); } #endif } } static int proc_exec_v(argv, prog) char **argv; char *prog; { #ifndef USE_CWGUSI if (prog) { security(prog); } else { security(argv[0]); prog = dln_find_exe(argv[0], 0); if (!prog) { errno = ENOENT; return -1; } } #if (defined(MSDOS) && !defined(DJGPP)) || defined(__human68k__) || defined(__EMX__) || defined(OS2) { #if defined(__human68k__) #define COMMAND "command.x" #endif #if defined(__EMX__) || defined(OS2) /* OS/2 emx */ #define COMMAND "cmd.exe" #endif #if (defined(MSDOS) && !defined(DJGPP)) #define COMMAND "command.com" #endif char *extension; if ((extension = strrchr(prog, '.')) != NULL && strcasecmp(extension, ".bat") == 0) { char **new_argv; char *p; int n; for (n = 0; argv[n]; n++) /* no-op */; new_argv = ALLOCA_N(char *, n + 2); for (; n > 0; n--) new_argv[n + 1] = argv[n]; new_argv[1] = strcpy(ALLOCA_N(char, strlen(argv[0]) + 1), argv[0]); for (p = new_argv[1]; *p != '\0'; p++) if (*p == '/') *p = '\\'; new_argv[0] = COMMAND; argv = new_argv; prog = dln_find_exe(argv[0], 0); if (!prog) { errno = ENOENT; return -1; } } } #endif /* MSDOS or __human68k__ or __EMX__ */ before_exec(); execv(prog, argv); after_exec(); return -1; #else /* USE_CWGUSI */ rb_notimplement(); #endif /* USE_CWGUSI */ } static int proc_exec_n(argc, argv, progv) int argc; VALUE *argv; VALUE progv; { char *prog = 0; char **args; int i; if (progv) { prog = RSTRING(progv)->ptr; } args = ALLOCA_N(char*, argc+1); for (i=0; iptr; } args[i] = 0; if (args[0]) { return proc_exec_v(args, prog); } return -1; } int rb_proc_exec(str) const char *str; { #ifndef USE_CWGUSI const char *s = str; char *ss, *t; char **argv, **a; security(str); for (s=str; *s; s++) { if (*s != ' ' && !ISALPHA(*s) && strchr("*?{}[]<>()~&|\\$;'`\"\n",*s)) { #if defined(MSDOS) int state; before_exec(); state = system(str); after_exec(); if (state != -1) exit(state); #else #if defined(__human68k__) || defined(__CYGWIN32__) || defined(__EMX__) char *shell = dln_find_exe("sh", 0); int state = -1; before_exec(); if (shell) execl(shell, "sh", "-c", str, (char *) NULL); else state = system(str); after_exec(); if (state != -1) exit(state); #else before_exec(); execl("/bin/sh", "sh", "-c", str, (char *)NULL); after_exec(); #endif #endif return -1; } } a = argv = ALLOCA_N(char*, (s-str)/2+2); ss = ALLOCA_N(char, s-str+1); strcpy(ss, str); if (*a++ = strtok(ss, " \t")) { while (t = strtok(NULL, " \t")) { *a++ = t; } *a = NULL; } if (argv[0]) { return proc_exec_v(argv, 0); } errno = ENOENT; return -1; #else /* USE_CWGUSI */ rb_notimplement(); #endif /* USE_CWGUSI */ } #if defined(__human68k__) static int proc_spawn_v(argv, prog) char **argv; char *prog; { char *extension; int state; if (prog) { security(prog); } else { security(argv[0]); prog = dln_find_exe(argv[0], 0); if (!prog) return -1; } if ((extension = strrchr(prog, '.')) != NULL && strcasecmp(extension, ".bat") == 0) { char **new_argv; char *p; int n; for (n = 0; argv[n]; n++) /* no-op */; new_argv = ALLOCA_N(char *, n + 2); for (; n > 0; n--) new_argv[n + 1] = argv[n]; new_argv[1] = strcpy(ALLOCA_N(char, strlen(argv[0]) + 1), argv[0]); for (p = new_argv[1]; *p != '\0'; p++) if (*p == '/') *p = '\\'; new_argv[0] = COMMAND; argv = new_argv; prog = dln_find_exe(argv[0], 0); if (!prog) { errno = ENOENT; return -1; } } before_exec(); state = spawnv(P_WAIT, prog, argv); after_exec(); return state; } static int proc_spawn_n(argc, argv, prog) int argc; VALUE *argv; VALUE prog; { char **args; int i; args = ALLOCA_N(char *, argc + 1); for (i = 0; i < argc; i++) { Check_SafeStr(argv[i]); args[i] = RSTRING(argv[i])->ptr; } Check_SafeStr(prog); args[i] = (char *) 0; if (args[0]) return proc_spawn_v(args, RSTRING(prog)->ptr); return -1; } static int proc_spawn(sv) VALUE sv; { char *str; char *s, *t; char **argv, **a; int state; Check_SafeStr(sv); str = s = RSTRING(sv)->ptr; for (s = str; *s; s++) { if (*s != ' ' && !ISALPHA(*s) && strchr("*?{}[]<>()~&|\\$;'`\"\n",*s)) { char *shell = dln_find_exe("sh", 0); before_exec(); state = shell?spawnl(P_WAIT,shell,"sh","-c",str,(char*)NULL):system(str); after_exec(); return state; } } a = argv = ALLOCA_N(char *, (s - str) / 2 + 2); s = ALLOCA_N(char, s - str + 1); strcpy(s, str); if (*a++ = strtok(s, " \t")) { while (t = strtok(NULL, " \t")) *a++ = t; *a = NULL; } return argv[0] ? proc_spawn_v(argv, 0) : -1; } #endif /* __human68k__ */ static VALUE rb_f_exec(argc, argv) int argc; VALUE *argv; { VALUE prog = 0; int i; if (argc == 0) { rb_raise(rb_eArgError, "wrong # of arguments"); } if (TYPE(argv[0]) == T_ARRAY) { if (RARRAY(argv[0])->len != 2) { rb_raise(rb_eArgError, "wrong first argument"); } prog = RARRAY(argv[0])->ptr[0]; argv[0] = RARRAY(argv[0])->ptr[1]; } if (prog) { Check_SafeStr(prog); } for (i = 0; i < argc; i++) { Check_SafeStr(argv[i]); } if (argc == 1 && prog == 0) { rb_proc_exec(RSTRING(argv[0])->ptr); } else { proc_exec_n(argc, argv, prog); } rb_sys_fail(RSTRING(argv[0])->ptr); } static VALUE rb_f_fork(obj) VALUE obj; { #if !defined(__human68k__) && !defined(NT) && !defined(__MACOS__) && !defined(__EMX__) int pid; rb_secure(2); switch (pid = fork()) { case 0: #ifdef linux after_exec(); #endif if (rb_iterator_p()) { rb_yield(Qnil); _exit(0); } return Qnil; case -1: rb_sys_fail("fork(2)"); return Qnil; default: return INT2FIX(pid); } #else rb_notimplement(); #endif } static VALUE rb_f_exit_bang(argc, argv, obj) int argc; VALUE *argv; VALUE obj; { VALUE status; int istatus; rb_secure(4); if (rb_scan_args(argc, argv, "01", &status) == 1) { istatus = NUM2INT(status); } else { istatus = -1; } #ifdef USE_CWGUSI exit(istatus); #else _exit(istatus); #endif return Qnil; /* not reached */ } void rb_syswait(pid) int pid; { RETSIGTYPE (*hfunc)(), (*qfunc)(), (*ifunc)(); int status; int i; #ifdef SIGHUP hfunc = signal(SIGHUP, SIG_IGN); #endif #ifdef SIGQUIT qfunc = signal(SIGQUIT, SIG_IGN); #endif ifunc = signal(SIGINT, SIG_IGN); do { i = rb_waitpid(pid, 0, &status); } while (i == -1 && errno == EINTR); #ifdef SIGHUP signal(SIGHUP, hfunc); #endif #ifdef SIGQUIT signal(SIGQUIT, qfunc); #endif signal(SIGINT, ifunc); } static VALUE rb_f_system(argc, argv) int argc; VALUE *argv; { #if defined(NT) || defined(__EMX__) VALUE cmd; int state; if (TYPE(argv[0]) == T_ARRAY) { if (RARRAY(argv[0])->len != 2) { rb_raise(rb_eArgError, "wrong first argument"); } argv[0] = RARRAY(argv[0])->ptr[0]; } cmd = rb_ary_join(rb_ary_new4(argc, argv), rb_str_new2(" ")); Check_SafeStr(cmd); state = do_spawn(RSTRING(cmd)->ptr); rb_last_status = INT2FIX(state); if (state == 0) return Qtrue; return Qfalse; #else #ifdef DJGPP VALUE cmd; int state; if (TYPE(argv[0]) == T_ARRAY) { if (RARRAY(argv[0])->len != 2) { rb_raise(rb_eArgError, "wrong first argument"); } argv[0] = RARRAY(argv[0])->ptr[0]; } cmd = rb_ary_join(rb_ary_new4(argc, argv), rb_str_new2(" ")); Check_SafeStr(cmd); state = system(RSTRING(cmd)->ptr); rb_last_status = INT2FIX(state); if (state == 0) return Qtrue; return Qfalse; #else #if defined(__human68k__) VALUE prog = 0; int i; int state; fflush(stdin); fflush(stdout); fflush(stderr); if (argc == 0) { rb_last_status = INT2FIX(0); return INT2FIX(0); } if (TYPE(argv[0]) == T_ARRAY) { if (RARRAY(argv[0])->len != 2) { rb_raise(rb_eArgError, "wrong first argument"); } prog = RARRAY(argv[0])->ptr[0]; argv[0] = RARRAY(argv[0])->ptr[1]; } if (argc == 1 && prog == 0) { state = proc_spawn(argv[0]); } else { state = proc_spawn_n(argc, argv, prog); } rb_last_status = state == -1 ? INT2FIX(127) : INT2FIX(state); return state == 0 ? Qtrue : Qfalse; #else #if defined(USE_CWGUSI) rb_notimplement(); #else volatile VALUE prog = 0; int pid; int i; fflush(stdout); fflush(stderr); if (argc == 0) { rb_last_status = INT2FIX(0); return INT2FIX(0); } if (TYPE(argv[0]) == T_ARRAY) { if (RARRAY(argv[0])->len != 2) { rb_raise(rb_eArgError, "wrong first argument"); } prog = RARRAY(argv[0])->ptr[0]; argv[0] = RARRAY(argv[0])->ptr[1]; } if (prog) { Check_SafeStr(prog); } for (i = 0; i < argc; i++) { Check_SafeStr(argv[i]); } retry: switch (pid = vfork()) { case 0: if (argc == 1 && prog == 0) { rb_proc_exec(RSTRING(argv[0])->ptr); } else { proc_exec_n(argc, argv, prog); } _exit(127); break; /* not reached */ case -1: if (errno == EAGAIN) { rb_thread_sleep(1); goto retry; } rb_sys_fail(0); break; default: rb_syswait(pid); } if (rb_last_status == INT2FIX(0)) return Qtrue; return Qfalse; #endif /* USE_CWGUSI */ #endif /* __human68k__ */ #endif /* DJGPP */ #endif /* NT */ } static VALUE rb_f_sleep(argc, argv) int argc; VALUE *argv; { int beg, end; beg = time(0); if (argc == 0) { rb_thread_sleep_forever(); } else if (argc == 1) { rb_thread_wait_for(rb_time_interval(argv[0])); } else { rb_raise(rb_eArgError, "wrong # of arguments"); } end = time(0) - beg; return INT2FIX(end); } static VALUE proc_getpgrp(argc, argv) int argc; VALUE *argv; { #ifdef HAVE_GETPGRP int pgrp; #ifndef GETPGRP_VOID VALUE vpid; int pid; rb_scan_args(argc, argv, "01", &vpid); pid = NIL_P(vpid)?0:NUM2INT(vpid); pgrp = getpgrp(pid); #else rb_scan_args(argc, argv, "0"); pgrp = getpgrp(); #endif if (pgrp < 0) rb_sys_fail(0); return INT2FIX(pgrp); #else rb_notimplement(); #endif } static VALUE proc_setpgrp(argc, argv) int argc; VALUE *argv; { #ifdef HAVE_SETPGRP #ifndef SETPGRP_VOID VALUE pid, pgrp; int ipid, ipgrp; rb_scan_args(argc, argv, "02", &pid, &pgrp); ipid = NIL_P(pid)?0:NUM2INT(pid); ipgrp = NIL_P(pgrp)?0:NUM2INT(pgrp); if (setpgrp(ipid, ipgrp) < 0) rb_sys_fail(0); #else rb_scan_args(argc, argv, "0"); if (setpgrp() < 0) rb_sys_fail(0); #endif return Qnil; #else rb_notimplement(); #endif } static VALUE proc_getpgid(obj, pid) VALUE obj, pid; { #ifdef HAVE_GETPGID int i; i = getpgid(NUM2INT(pid)); return INT2NUM(i); #else rb_notimplement(); #endif } static VALUE proc_setpgid(obj, pid, pgrp) VALUE obj, pid, pgrp; { #ifdef HAVE_SETPGID int ipid, ipgrp; ipid = NUM2INT(pid); ipgrp = NUM2INT(pgrp); if (setpgid(ipid, ipgrp) < 0) rb_sys_fail(0); return Qnil; #else rb_notimplement(); #endif } static VALUE proc_setsid() { #ifdef HAVE_SETSID int pid = setsid(); if (pid < 0) rb_sys_fail(0); return INT2FIX(pid); #else rb_notimplement(); #endif } static VALUE proc_getpriority(obj, which, who) VALUE obj, which, who; { #ifdef HAVE_GETPRIORITY int prio, iwhich, iwho; iwhich = NUM2INT(which); iwho = NUM2INT(who); prio = getpriority(iwhich, iwho); if (prio < 0) rb_sys_fail(0); return INT2FIX(prio); #else rb_notimplement(); #endif } static VALUE proc_setpriority(obj, which, who, prio) VALUE obj, which, who, prio; { #ifdef HAVE_GETPRIORITY int iwhich, iwho, iprio; iwhich = NUM2INT(which); iwho = NUM2INT(who); iprio = NUM2INT(prio); if (setpriority(iwhich, iwho, iprio) < 0) rb_sys_fail(0); return INT2FIX(0); #else rb_notimplement(); #endif } static VALUE proc_getuid(obj) VALUE obj; { int uid = getuid(); return INT2FIX(uid); } static VALUE proc_setuid(obj, id) VALUE obj, id; { int uid; uid = NUM2INT(id); #ifdef HAVE_SETREUID setreuid(uid, -1); #else #ifdef HAVE_SETRUID setruid(uid); #else { if (geteuid() == uid) setuid(uid); else rb_notimplement(); } #endif #endif return INT2FIX(uid); } static VALUE proc_getgid(obj) VALUE obj; { int gid = getgid(); return INT2FIX(gid); } static VALUE proc_setgid(obj, id) VALUE obj, id; { int gid; gid = NUM2INT(id); #ifdef HAS_SETRGID setrgid((GIDTYPE)gid); #else #ifdef HAVE_SETREGID setregid(gid, -1); #else { if (getegid() == gid) setgid(gid); else rb_notimplement(); } #endif #endif return INT2FIX(gid); } static VALUE proc_geteuid(obj) VALUE obj; { int euid = geteuid(); return INT2FIX(euid); } static VALUE proc_seteuid(obj, euid) VALUE obj, euid; { #ifdef HAVE_SETEUID if (seteuid(NUM2INT(euid)) < 0) rb_sys_fail(0); #else #ifdef HAVE_SETREUID if (setreuid(-1, NUM2INT(euid)) < 0) rb_sys_fail(0); #else euid = NUM2INT(euid); if (euid == getuid()) setuid(euid); else rb_notimplement(); #endif #endif return euid; } static VALUE proc_getegid(obj) VALUE obj; { int egid = getegid(); return INT2FIX(egid); } static VALUE proc_setegid(obj, egid) VALUE obj, egid; { #ifdef HAVE_SETEGID if (setegid(NUM2INT(egid)) < 0) rb_sys_fail(0); #else #ifdef HAVE_SETREGID if (setregid(-1, NUM2INT(egid)) < 0) rb_sys_fail(0); #else egid = NUM2INT(egid); if (egid == getgid()) setgid(egid); else rb_notimplement(); #endif #endif return egid; } VALUE rb_mProcess; void Init_process() { #ifndef USE_CWGUSI rb_define_virtual_variable("$$", get_pid, 0); #endif rb_define_readonly_variable("$?", &rb_last_status); #ifndef USE_CWGUSI rb_define_global_function("exec", rb_f_exec, -1); #endif rb_define_global_function("fork", rb_f_fork, 0); rb_define_global_function("exit!", rb_f_exit_bang, -1); rb_define_global_function("system", rb_f_system, -1); rb_define_global_function("sleep", rb_f_sleep, -1); rb_mProcess = rb_define_module("Process"); #if !defined(NT) && !defined(DJGPP) #ifdef WNOHANG rb_define_const(rb_mProcess, "WNOHANG", INT2FIX(WNOHANG)); #else rb_define_const(rb_mProcess, "WNOHANG", INT2FIX(0)); #endif #ifdef WUNTRACED rb_define_const(rb_mProcess, "WUNTRACED", INT2FIX(WUNTRACED)); #else rb_define_const(rb_mProcess, "WUNTRACED", INT2FIX(0)); #endif #endif #if !defined(NT) rb_define_singleton_method(rb_mProcess, "fork", rb_f_fork, 0); #endif rb_define_singleton_method(rb_mProcess, "exit!", rb_f_exit_bang, -1); #ifndef USE_CWGUSI rb_define_module_function(rb_mProcess, "kill", rb_f_kill, -1); #endif #ifndef NT rb_define_module_function(rb_mProcess, "wait", rb_f_wait, 0); rb_define_module_function(rb_mProcess, "waitpid", rb_f_waitpid, 2); #ifndef USE_CWGUSI rb_define_module_function(rb_mProcess, "pid", get_pid, 0); rb_define_module_function(rb_mProcess, "ppid", get_ppid, 0); #endif /* ifndef USE_CWGUSI */ #endif /* ifndef NT */ rb_define_module_function(rb_mProcess, "getpgrp", proc_getpgrp, -1); rb_define_module_function(rb_mProcess, "setpgrp", proc_setpgrp, -1); rb_define_module_function(rb_mProcess, "getpgid", proc_getpgid, 1); rb_define_module_function(rb_mProcess, "setpgid", proc_setpgid, 2); rb_define_module_function(rb_mProcess, "setsid", proc_setsid, 0); rb_define_module_function(rb_mProcess, "getpriority", proc_getpriority, 2); rb_define_module_function(rb_mProcess, "setpriority", proc_setpriority, 3); #ifdef HAVE_GETPRIORITY rb_define_const(rb_mProcess, "PRIO_PROCESS", INT2FIX(PRIO_PROCESS)); rb_define_const(rb_mProcess, "PRIO_PGRP", INT2FIX(PRIO_PGRP)); rb_define_const(rb_mProcess, "PRIO_USER", INT2FIX(PRIO_USER)); #endif rb_define_module_function(rb_mProcess, "uid", proc_getuid, 0); rb_define_module_function(rb_mProcess, "uid=", proc_setuid, 1); rb_define_module_function(rb_mProcess, "gid", proc_getgid, 0); rb_define_module_function(rb_mProcess, "gid=", proc_setgid, 1); rb_define_module_function(rb_mProcess, "euid", proc_geteuid, 0); rb_define_module_function(rb_mProcess, "euid=", proc_seteuid, 1); rb_define_module_function(rb_mProcess, "egid", proc_getegid, 0); rb_define_module_function(rb_mProcess, "egid=", proc_setegid, 1); }