diff options
Diffstat (limited to 'trunk/ext/pty/pty.c')
-rw-r--r-- | trunk/ext/pty/pty.c | 470 |
1 files changed, 0 insertions, 470 deletions
diff --git a/trunk/ext/pty/pty.c b/trunk/ext/pty/pty.c deleted file mode 100644 index e0d571a581..0000000000 --- a/trunk/ext/pty/pty.c +++ /dev/null @@ -1,470 +0,0 @@ -#include "ruby/config.h" -#ifdef RUBY_EXTCONF_H -#include RUBY_EXTCONF_H -#endif -#include <stdio.h> -#include <sys/types.h> -#include <sys/stat.h> -#include <sys/file.h> -#include <fcntl.h> -#include <errno.h> -#include <pwd.h> -#ifdef HAVE_SYS_IOCTL_H -#include <sys/ioctl.h> -#endif -#ifdef HAVE_LIBUTIL_H -#include <libutil.h> -#endif -#ifdef HAVE_PTY_H -#include <pty.h> -#endif -#ifdef HAVE_SYS_WAIT_H -#include <sys/wait.h> -#else -#define WIFSTOPPED(status) (((status) & 0xff) == 0x7f) -#endif -#include <ctype.h> - -#include "ruby/ruby.h" -#include "ruby/io.h" -#include "ruby/util.h" - -#include <signal.h> -#ifdef HAVE_SYS_STROPTS_H -#include <sys/stropts.h> -#endif - -#ifdef HAVE_UNISTD_H -#include <unistd.h> -#endif - -#define DEVICELEN 16 - -#if !defined(HAVE_OPENPTY) -#if defined(__hpux) -static const -char MasterDevice[] = "/dev/ptym/pty%s", - SlaveDevice[] = "/dev/pty/tty%s", - *const deviceNo[] = { - "p0","p1","p2","p3","p4","p5","p6","p7", - "p8","p9","pa","pb","pc","pd","pe","pf", - "q0","q1","q2","q3","q4","q5","q6","q7", - "q8","q9","qa","qb","qc","qd","qe","qf", - "r0","r1","r2","r3","r4","r5","r6","r7", - "r8","r9","ra","rb","rc","rd","re","rf", - "s0","s1","s2","s3","s4","s5","s6","s7", - "s8","s9","sa","sb","sc","sd","se","sf", - "t0","t1","t2","t3","t4","t5","t6","t7", - "t8","t9","ta","tb","tc","td","te","tf", - "u0","u1","u2","u3","u4","u5","u6","u7", - "u8","u9","ua","ub","uc","ud","ue","uf", - "v0","v1","v2","v3","v4","v5","v6","v7", - "v8","v9","va","vb","vc","vd","ve","vf", - "w0","w1","w2","w3","w4","w5","w6","w7", - "w8","w9","wa","wb","wc","wd","we","wf", - 0, - }; -#elif defined(_IBMESA) /* AIX/ESA */ -static const -char MasterDevice[] = "/dev/ptyp%s", - SlaveDevice[] = "/dev/ttyp%s", - *const deviceNo[] = { -"00","01","02","03","04","05","06","07","08","09","0a","0b","0c","0d","0e","0f", -"10","11","12","13","14","15","16","17","18","19","1a","1b","1c","1d","1e","1f", -"20","21","22","23","24","25","26","27","28","29","2a","2b","2c","2d","2e","2f", -"30","31","32","33","34","35","36","37","38","39","3a","3b","3c","3d","3e","3f", -"40","41","42","43","44","45","46","47","48","49","4a","4b","4c","4d","4e","4f", -"50","51","52","53","54","55","56","57","58","59","5a","5b","5c","5d","5e","5f", -"60","61","62","63","64","65","66","67","68","69","6a","6b","6c","6d","6e","6f", -"70","71","72","73","74","75","76","77","78","79","7a","7b","7c","7d","7e","7f", -"80","81","82","83","84","85","86","87","88","89","8a","8b","8c","8d","8e","8f", -"90","91","92","93","94","95","96","97","98","99","9a","9b","9c","9d","9e","9f", -"a0","a1","a2","a3","a4","a5","a6","a7","a8","a9","aa","ab","ac","ad","ae","af", -"b0","b1","b2","b3","b4","b5","b6","b7","b8","b9","ba","bb","bc","bd","be","bf", -"c0","c1","c2","c3","c4","c5","c6","c7","c8","c9","ca","cb","cc","cd","ce","cf", -"d0","d1","d2","d3","d4","d5","d6","d7","d8","d9","da","db","dc","dd","de","df", -"e0","e1","e2","e3","e4","e5","e6","e7","e8","e9","ea","eb","ec","ed","ee","ef", -"f0","f1","f2","f3","f4","f5","f6","f7","f8","f9","fa","fb","fc","fd","fe","ff", - }; -#elif !defined(HAVE_PTSNAME) -static const -char MasterDevice[] = "/dev/pty%s", - SlaveDevice[] = "/dev/tty%s", - *const deviceNo[] = { - "p0","p1","p2","p3","p4","p5","p6","p7", - "p8","p9","pa","pb","pc","pd","pe","pf", - "q0","q1","q2","q3","q4","q5","q6","q7", - "q8","q9","qa","qb","qc","qd","qe","qf", - "r0","r1","r2","r3","r4","r5","r6","r7", - "r8","r9","ra","rb","rc","rd","re","rf", - "s0","s1","s2","s3","s4","s5","s6","s7", - "s8","s9","sa","sb","sc","sd","se","sf", - 0, - }; -#endif -#endif /* !defined(HAVE_OPENPTY) */ - -#ifndef HAVE_SETEUID -# ifdef HAVE_SETREUID -# define seteuid(e) setreuid(-1, (e)) -# else /* NOT HAVE_SETREUID */ -# ifdef HAVE_SETRESUID -# define seteuid(e) setresuid(-1, (e), -1) -# else /* NOT HAVE_SETRESUID */ - /* I can't set euid. (;_;) */ -# endif /* HAVE_SETRESUID */ -# endif /* HAVE_SETREUID */ -#endif /* NO_SETEUID */ - -static VALUE eChildExited; - -static VALUE -echild_status(VALUE self) -{ - return rb_ivar_get(self, rb_intern("status")); -} - -struct pty_info { - int fd; - rb_pid_t child_pid; - VALUE thread; -}; - -static void -raise_from_wait(const char *state, const struct pty_info *info) -{ - char buf[1024]; - VALUE exc; - - snprintf(buf, sizeof(buf), "pty - %s: %ld", state, (long)info->child_pid); - exc = rb_exc_new2(eChildExited, buf); - rb_iv_set(exc, "status", rb_last_status_get()); - rb_funcall(info->thread, rb_intern("raise"), 1, exc); -} - -static VALUE -pty_syswait(void *arg) -{ - const struct pty_info *const info = arg; - rb_pid_t cpid; - int status; - - for (;;) { - cpid = rb_waitpid(info->child_pid, &status, WUNTRACED); - if (cpid == -1) return Qnil; - -#if defined(WIFSTOPPED) -#elif defined(IF_STOPPED) -#define WIFSTOPPED(status) IF_STOPPED(status) -#else ----->> Either IF_STOPPED or WIFSTOPPED is needed <<---- -#endif /* WIFSTOPPED | IF_STOPPED */ - if (WIFSTOPPED(status)) { /* suspend */ - raise_from_wait("stopped", info); - } - else if (kill(info->child_pid, 0) == 0) { - raise_from_wait("changed", info); - } - else { - raise_from_wait("exited", info); - return Qnil; - } - } -} - -static void getDevice(int*, int*, char [DEVICELEN]); - -struct exec_info { - int argc; - VALUE *argv; -}; - -static VALUE -pty_exec(VALUE v) -{ - struct exec_info *arg = (struct exec_info *)v; - return rb_f_exec(arg->argc, arg->argv); -} - -static void -establishShell(int argc, VALUE *argv, struct pty_info *info, - char SlaveName[DEVICELEN]) -{ - int master,slave; - rb_pid_t pid; - char *p, tmp, *getenv(); - struct passwd *pwent; - VALUE v; - struct exec_info arg; - int status; - - if (argc == 0) { - const char *shellname; - - if ((p = getenv("SHELL")) != NULL) { - shellname = p; - } - else { - pwent = getpwuid(getuid()); - if (pwent && pwent->pw_shell) - shellname = pwent->pw_shell; - else - shellname = "/bin/sh"; - } - v = rb_str_new2(shellname); - argc = 1; - argv = &v; - } - getDevice(&master, &slave, SlaveName); - - info->thread = rb_thread_current(); - if ((pid = fork()) < 0) { - close(master); - close(slave); - rb_sys_fail("fork failed"); - } - - if (pid == 0) { /* child */ - /* - * Set free from process group and controlling terminal - */ -#ifdef HAVE_SETSID - (void) setsid(); -#else /* HAS_SETSID */ -# ifdef HAVE_SETPGRP -# ifdef SETGRP_VOID - if (setpgrp() == -1) - perror("setpgrp()"); -# else /* SETGRP_VOID */ - if (setpgrp(0, getpid()) == -1) - rb_sys_fail("setpgrp()"); - { - int i = open("/dev/tty", O_RDONLY); - if (i < 0) rb_sys_fail("/dev/tty"); - if (ioctl(i, TIOCNOTTY, (char *)0)) - perror("ioctl(TIOCNOTTY)"); - close(i); - } -# endif /* SETGRP_VOID */ -# endif /* HAVE_SETPGRP */ -#endif /* HAS_SETSID */ - - /* - * obtain new controlling terminal - */ -#if defined(TIOCSCTTY) - close(master); - (void) ioctl(slave, TIOCSCTTY, (char *)0); - /* errors ignored for sun */ -#else - close(slave); - slave = open(SlaveName, O_RDWR); - if (slave < 0) { - perror("open: pty slave"); - _exit(1); - } - close(master); -#endif - write(slave, "", 1); - dup2(slave,0); - dup2(slave,1); - dup2(slave,2); - close(slave); -#if defined(HAVE_SETEUID) || defined(HAVE_SETREUID) || defined(HAVE_SETRESUID) - seteuid(getuid()); -#endif - - arg.argc = argc; - arg.argv = argv; - rb_protect(pty_exec, (VALUE)&arg, &status); - sleep(1); - _exit(1); - } - - read(master, &tmp, 1); - close(slave); - - info->child_pid = pid; - info->fd = master; -} - -static VALUE -pty_finalize_syswait(struct pty_info *info) -{ - rb_thread_kill(info->thread); - rb_funcall(info->thread, rb_intern("value"), 0); - rb_detach_process(info->child_pid); - return Qnil; -} - -static int -get_device_once(int *master, int *slave, char SlaveName[DEVICELEN], int fail) -{ -#if defined HAVE_OPENPTY -/* - * Use openpty(3) of 4.3BSD Reno and later, - * or the same interface function. - */ - if (openpty(master, slave, SlaveName, - (struct termios *)0, (struct winsize *)0) == -1) { - if (!fail) return -1; - rb_raise(rb_eRuntimeError, "openpty() failed"); - } - - return 0; -#elif defined HAVE__GETPTY - char *name; - - if (!(name = _getpty(master, O_RDWR, 0622, 0))) { - if (!fail) return -1; - rb_raise(rb_eRuntimeError, "_getpty() failed"); - } - - *slave = open(name, O_RDWR); - strlcpy(SlaveName, name, sizeof SlaveName); - - return 0; -#else /* HAVE__GETPTY */ - int i,j; - -#ifdef HAVE_PTSNAME - char *pn; - void (*s)(); - - extern char *ptsname(int); - extern int unlockpt(int); - extern int grantpt(int); - - if((i = open("/dev/ptmx", O_RDWR, 0)) != -1) { - s = signal(SIGCHLD, SIG_DFL); - if(grantpt(i) != -1) { - signal(SIGCHLD, s); - if(unlockpt(i) != -1) { - if((pn = ptsname(i)) != NULL) { - if((j = open(pn, O_RDWR, 0)) != -1) { -#if defined I_PUSH && !defined linux - if(ioctl(j, I_PUSH, "ptem") != -1) { - if(ioctl(j, I_PUSH, "ldterm") != -1) { - ioctl(j, I_PUSH, "ttcompat"); -#endif - *master = i; - *slave = j; - strlcpy(SlaveName, pn, sizeof SlaveName); - return 0; -#if defined I_PUSH && !defined linux - } - } -#endif - } - } - } - } - close(i); - } - if (!fail) rb_raise(rb_eRuntimeError, "can't get Master/Slave device"); - return -1; -#else - const char *const *p; - char MasterName[DEVICELEN]; - - for (p = deviceNo; *p != NULL; p++) { - snprintf(MasterName, sizeof MasterName, MasterDevice, *p); - if ((i = open(MasterName,O_RDWR,0)) >= 0) { - *master = i; - snprintf(SlaveName, sizeof SlaveName, SlaveDevice, *p); - if ((j = open(SlaveName,O_RDWR,0)) >= 0) { - *slave = j; - chown(SlaveName, getuid(), getgid()); - chmod(SlaveName, 0622); - return 0; - } - close(i); - } - } - if (fail) rb_raise(rb_eRuntimeError, "can't get %s", SlaveName); - return -1; -#endif -#endif -} - -static void -getDevice(int *master, int *slave, char SlaveName[DEVICELEN]) -{ - if (get_device_once(master, slave, SlaveName, 0)) { - rb_gc(); - get_device_once(master, slave, SlaveName, 1); - } -} - -/* ruby function: getpty */ -static VALUE -pty_getpty(int argc, VALUE *argv, VALUE self) -{ - VALUE res; - struct pty_info info; - struct pty_info thinfo; - rb_io_t *wfptr,*rfptr; - VALUE rport = rb_obj_alloc(rb_cFile); - VALUE wport = rb_obj_alloc(rb_cFile); - char SlaveName[DEVICELEN]; - - MakeOpenFile(rport, rfptr); - MakeOpenFile(wport, wfptr); - - establishShell(argc, argv, &info, SlaveName); - - rfptr->mode = rb_io_mode_flags("r"); - rfptr->fd = info.fd; - rfptr->pathv = rb_obj_freeze(rb_str_new_cstr(SlaveName)); - - wfptr->mode = rb_io_mode_flags("w") | FMODE_SYNC; - wfptr->fd = dup(info.fd); - wfptr->pathv = rfptr->pathv; - - res = rb_ary_new2(3); - rb_ary_store(res,0,(VALUE)rport); - rb_ary_store(res,1,(VALUE)wport); - rb_ary_store(res,2,PIDT2NUM(info.child_pid)); - - thinfo.thread = rb_thread_create(pty_syswait, (void*)&info); - thinfo.child_pid = info.child_pid; - rb_thread_schedule(); - - if (rb_block_given_p()) { - rb_ensure(rb_yield, res, pty_finalize_syswait, (VALUE)&thinfo); - return Qnil; - } - return res; -} - -/* ruby function: protect_signal - obsolete */ -static VALUE -pty_protect(VALUE self) -{ - rb_warn("PTY::protect_signal is no longer needed"); - rb_yield(Qnil); - return self; -} - -/* ruby function: reset_signal - obsolete */ -static VALUE -pty_reset_signal(VALUE self) -{ - rb_warn("PTY::reset_signal is no longer needed"); - return self; -} - -static VALUE cPTY; - -void -Init_pty() -{ - cPTY = rb_define_module("PTY"); - rb_define_module_function(cPTY,"getpty",pty_getpty,-1); - rb_define_module_function(cPTY,"spawn",pty_getpty,-1); - rb_define_module_function(cPTY,"protect_signal",pty_protect,0); - rb_define_module_function(cPTY,"reset_signal",pty_reset_signal,0); - - eChildExited = rb_define_class_under(cPTY,"ChildExited",rb_eRuntimeError); - rb_define_method(eChildExited,"status",echild_status,0); -} |