--- /dev/null
+/**
+ * Copyright (c) 2012-2015, Christopher Jeffrey (MIT License)
+ * Copyright (c) 2017, Daniel Imms (MIT License)
+ *
+ * pty.cc:
+ * This file is responsible for starting processes
+ * with pseudo-terminal file descriptors.
+ *
+ * See:
+ * man pty
+ * man tty_ioctl
+ * man termios
+ * man forkpty
+ */
+
+/**
+ * Includes
+ */
+
+#include <nan.h>
+#include <errno.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/ioctl.h>
+#include <sys/wait.h>
+#include <fcntl.h>
+
+/* forkpty */
+/* http://www.gnu.org/software/gnulib/manual/html_node/forkpty.html */
+#if defined(__GLIBC__) || defined(__CYGWIN__)
+#include <pty.h>
+#elif defined(__APPLE__) || defined(__OpenBSD__) || defined(__NetBSD__)
+#include <util.h>
+#elif defined(__FreeBSD__)
+#include <libutil.h>
+#elif defined(__sun)
+#include <stropts.h> /* for I_PUSH */
+#else
+#include <pty.h>
+#endif
+
+#include <termios.h> /* tcgetattr, tty_ioctl */
+
+/* Some platforms name VWERASE and VDISCARD differently */
+#if !defined(VWERASE) && defined(VWERSE)
+#define VWERASE VWERSE
+#endif
+#if !defined(VDISCARD) && defined(VDISCRD)
+#define VDISCARD VDISCRD
+#endif
+
+/* environ for execvpe */
+/* node/src/node_child_process.cc */
+#if defined(__APPLE__) && !TARGET_OS_IPHONE
+#include <crt_externs.h>
+#define environ (*_NSGetEnviron())
+#else
+extern char **environ;
+#endif
+
+/* for pty_getproc */
+#if defined(__linux__)
+#include <stdio.h>
+#include <stdint.h>
+#elif defined(__APPLE__)
+#include <sys/sysctl.h>
+#include <libproc.h>
+#endif
+
+/**
+ * Structs
+ */
+
+struct pty_baton {
+ Nan::Persistent<v8::Function> cb;
+ int exit_code;
+ int signal_code;
+ pid_t pid;
+ uv_async_t async;
+ uv_thread_t tid;
+};
+
+/**
+ * Methods
+ */
+
+NAN_METHOD(PtyFork);
+NAN_METHOD(PtyOpen);
+NAN_METHOD(PtyResize);
+NAN_METHOD(PtyGetProc);
+
+/**
+ * Functions
+ */
+
+static int
+pty_execvpe(const char *, char **, char **);
+
+static int
+pty_nonblock(int);
+
+static char *
+pty_getproc(int, char *);
+
+static int
+pty_openpty(int *, int *, char *,
+ const struct termios *,
+ const struct winsize *);
+
+static pid_t
+pty_forkpty(int *, char *,
+ const struct termios *,
+ const struct winsize *);
+
+static void
+pty_waitpid(void *);
+
+static void
+pty_after_waitpid(uv_async_t *);
+
+static void
+pty_after_close(uv_handle_t *);
+
+NAN_METHOD(PtyFork) {
+ Nan::HandleScope scope;
+
+ if (info.Length() != 10 ||
+ !info[0]->IsString() ||
+ !info[1]->IsArray() ||
+ !info[2]->IsArray() ||
+ !info[3]->IsString() ||
+ !info[4]->IsNumber() ||
+ !info[5]->IsNumber() ||
+ !info[6]->IsNumber() ||
+ !info[7]->IsNumber() ||
+ !info[8]->IsBoolean() ||
+ !info[9]->IsFunction()) {
+ return Nan::ThrowError(
+ "Usage: pty.fork(file, args, env, cwd, cols, rows, uid, gid, utf8, onexit)");
+ }
+
+ // Make sure the process still listens to SIGINT
+ signal(SIGINT, SIG_DFL);
+
+ // file
+ Nan::Utf8String file(info[0]);
+
+ // args
+ int i = 0;
+ v8::Local<v8::Array> argv_ = v8::Local<v8::Array>::Cast(info[1]);
+ int argc = argv_->Length();
+ int argl = argc + 1 + 1;
+ char **argv = new char*[argl];
+ argv[0] = strdup(*file);
+ argv[argl-1] = NULL;
+ for (; i < argc; i++) {
+ Nan::Utf8String arg(Nan::Get(argv_, i).ToLocalChecked());
+ argv[i+1] = strdup(*arg);
+ }
+
+ // env
+ i = 0;
+ v8::Local<v8::Array> env_ = v8::Local<v8::Array>::Cast(info[2]);
+ int envc = env_->Length();
+ char **env = new char*[envc+1];
+ env[envc] = NULL;
+ for (; i < envc; i++) {
+ Nan::Utf8String pair(Nan::Get(env_, i).ToLocalChecked());
+ env[i] = strdup(*pair);
+ }
+
+ // cwd
+ Nan::Utf8String cwd_(info[3]);
+ char *cwd = strdup(*cwd_);
+
+ // size
+ struct winsize winp;
+ winp.ws_col = info[4]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_row = info[5]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_xpixel = 0;
+ winp.ws_ypixel = 0;
+
+ // termios
+ struct termios t = termios();
+ struct termios *term = &t;
+ term->c_iflag = ICRNL | IXON | IXANY | IMAXBEL | BRKINT;
+ if (Nan::To<bool>(info[8]).FromJust()) {
+#if defined(IUTF8)
+ term->c_iflag |= IUTF8;
+#endif
+ }
+ term->c_oflag = OPOST | ONLCR;
+ term->c_cflag = CREAD | CS8 | HUPCL;
+ term->c_lflag = ICANON | ISIG | IEXTEN | ECHO | ECHOE | ECHOK | ECHOKE | ECHOCTL;
+
+ term->c_cc[VEOF] = 4;
+ term->c_cc[VEOL] = -1;
+ term->c_cc[VEOL2] = -1;
+ term->c_cc[VERASE] = 0x7f;
+ term->c_cc[VWERASE] = 23;
+ term->c_cc[VKILL] = 21;
+ term->c_cc[VREPRINT] = 18;
+ term->c_cc[VINTR] = 3;
+ term->c_cc[VQUIT] = 0x1c;
+ term->c_cc[VSUSP] = 26;
+ term->c_cc[VSTART] = 17;
+ term->c_cc[VSTOP] = 19;
+ term->c_cc[VLNEXT] = 22;
+ term->c_cc[VDISCARD] = 15;
+ term->c_cc[VMIN] = 1;
+ term->c_cc[VTIME] = 0;
+
+ #if (__APPLE__)
+ term->c_cc[VDSUSP] = 25;
+ term->c_cc[VSTATUS] = 20;
+ #endif
+
+ cfsetispeed(term, B38400);
+ cfsetospeed(term, B38400);
+
+ // uid / gid
+ int uid = info[6]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ int gid = info[7]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+
+ // fork the pty
+ int master = -1;
+ pid_t pid = pty_forkpty(&master, nullptr, term, &winp);
+
+ if (pid) {
+ for (i = 0; i < argl; i++) free(argv[i]);
+ delete[] argv;
+ for (i = 0; i < envc; i++) free(env[i]);
+ delete[] env;
+ free(cwd);
+ }
+
+ switch (pid) {
+ case -1:
+ return Nan::ThrowError("forkpty(3) failed.");
+ case 0:
+ if (strlen(cwd)) {
+ if (chdir(cwd) == -1) {
+ perror("chdir(2) failed.");
+ _exit(1);
+ }
+ }
+
+ if (uid != -1 && gid != -1) {
+ if (setgid(gid) == -1) {
+ perror("setgid(2) failed.");
+ _exit(1);
+ }
+ if (setuid(uid) == -1) {
+ perror("setuid(2) failed.");
+ _exit(1);
+ }
+ }
+
+ pty_execvpe(argv[0], argv, env);
+
+ perror("execvp(3) failed.");
+ _exit(1);
+ default:
+ if (pty_nonblock(master) == -1) {
+ return Nan::ThrowError("Could not set master fd to nonblocking.");
+ }
+
+ v8::Local<v8::Object> obj = Nan::New<v8::Object>();
+ Nan::Set(obj,
+ Nan::New<v8::String>("fd").ToLocalChecked(),
+ Nan::New<v8::Number>(master));
+ Nan::Set(obj,
+ Nan::New<v8::String>("pid").ToLocalChecked(),
+ Nan::New<v8::Number>(pid));
+ Nan::Set(obj,
+ Nan::New<v8::String>("pty").ToLocalChecked(),
+ Nan::New<v8::String>(ptsname(master)).ToLocalChecked());
+
+ pty_baton *baton = new pty_baton();
+ baton->exit_code = 0;
+ baton->signal_code = 0;
+ baton->cb.Reset(v8::Local<v8::Function>::Cast(info[9]));
+ baton->pid = pid;
+ baton->async.data = baton;
+
+ uv_async_init(uv_default_loop(), &baton->async, pty_after_waitpid);
+
+ uv_thread_create(&baton->tid, pty_waitpid, static_cast<void*>(baton));
+
+ return info.GetReturnValue().Set(obj);
+ }
+
+ return info.GetReturnValue().SetUndefined();
+}
+
+NAN_METHOD(PtyOpen) {
+ Nan::HandleScope scope;
+
+ if (info.Length() != 2 ||
+ !info[0]->IsNumber() ||
+ !info[1]->IsNumber()) {
+ return Nan::ThrowError("Usage: pty.open(cols, rows)");
+ }
+
+ // size
+ struct winsize winp;
+ winp.ws_col = info[0]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_row = info[1]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_xpixel = 0;
+ winp.ws_ypixel = 0;
+
+ // pty
+ int master, slave;
+ int ret = pty_openpty(&master, &slave, nullptr, NULL, &winp);
+
+ if (ret == -1) {
+ return Nan::ThrowError("openpty(3) failed.");
+ }
+
+ if (pty_nonblock(master) == -1) {
+ return Nan::ThrowError("Could not set master fd to nonblocking.");
+ }
+
+ if (pty_nonblock(slave) == -1) {
+ return Nan::ThrowError("Could not set slave fd to nonblocking.");
+ }
+
+ v8::Local<v8::Object> obj = Nan::New<v8::Object>();
+ Nan::Set(obj,
+ Nan::New<v8::String>("master").ToLocalChecked(),
+ Nan::New<v8::Number>(master));
+ Nan::Set(obj,
+ Nan::New<v8::String>("slave").ToLocalChecked(),
+ Nan::New<v8::Number>(slave));
+ Nan::Set(obj,
+ Nan::New<v8::String>("pty").ToLocalChecked(),
+ Nan::New<v8::String>(ptsname(master)).ToLocalChecked());
+
+ return info.GetReturnValue().Set(obj);
+}
+
+NAN_METHOD(PtyResize) {
+ Nan::HandleScope scope;
+
+ if (info.Length() != 3 ||
+ !info[0]->IsNumber() ||
+ !info[1]->IsNumber() ||
+ !info[2]->IsNumber()) {
+ return Nan::ThrowError("Usage: pty.resize(fd, cols, rows)");
+ }
+
+ int fd = info[0]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+
+ struct winsize winp;
+ winp.ws_col = info[1]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_row = info[2]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+ winp.ws_xpixel = 0;
+ winp.ws_ypixel = 0;
+
+ if (ioctl(fd, TIOCSWINSZ, &winp) == -1) {
+ switch (errno) {
+ case EBADF: return Nan::ThrowError("ioctl(2) failed, EBADF");
+ case EFAULT: return Nan::ThrowError("ioctl(2) failed, EFAULT");
+ case EINVAL: return Nan::ThrowError("ioctl(2) failed, EINVAL");
+ case ENOTTY: return Nan::ThrowError("ioctl(2) failed, ENOTTY");
+ }
+ return Nan::ThrowError("ioctl(2) failed");
+ }
+
+ return info.GetReturnValue().SetUndefined();
+}
+
+/**
+ * Foreground Process Name
+ */
+NAN_METHOD(PtyGetProc) {
+ Nan::HandleScope scope;
+
+ if (info.Length() != 2 ||
+ !info[0]->IsNumber() ||
+ !info[1]->IsString()) {
+ return Nan::ThrowError("Usage: pty.process(fd, tty)");
+ }
+
+ int fd = info[0]->IntegerValue(Nan::GetCurrentContext()).FromJust();
+
+ Nan::Utf8String tty_(info[1]);
+ char *tty = strdup(*tty_);
+ char *name = pty_getproc(fd, tty);
+ free(tty);
+
+ if (name == NULL) {
+ return info.GetReturnValue().SetUndefined();
+ }
+
+ v8::Local<v8::String> name_ = Nan::New<v8::String>(name).ToLocalChecked();
+ free(name);
+ return info.GetReturnValue().Set(name_);
+}
+
+/**
+ * execvpe
+ */
+
+// execvpe(3) is not portable.
+// http://www.gnu.org/software/gnulib/manual/html_node/execvpe.html
+static int
+pty_execvpe(const char *file, char **argv, char **envp) {
+ char **old = environ;
+ environ = envp;
+ int ret = execvp(file, argv);
+ environ = old;
+ return ret;
+}
+
+/**
+ * Nonblocking FD
+ */
+
+static int
+pty_nonblock(int fd) {
+ int flags = fcntl(fd, F_GETFL, 0);
+ if (flags == -1) return -1;
+ return fcntl(fd, F_SETFL, flags | O_NONBLOCK);
+}
+
+/**
+ * pty_waitpid
+ * Wait for SIGCHLD to read exit status.
+ */
+
+static void
+pty_waitpid(void *data) {
+ int ret;
+ int stat_loc;
+
+ pty_baton *baton = static_cast<pty_baton*>(data);
+
+ errno = 0;
+
+ if ((ret = waitpid(baton->pid, &stat_loc, 0)) != baton->pid) {
+ if (ret == -1 && errno == EINTR) {
+ return pty_waitpid(baton);
+ }
+ if (ret == -1 && errno == ECHILD) {
+ // XXX node v0.8.x seems to have this problem.
+ // waitpid is already handled elsewhere.
+ ;
+ } else {
+ assert(false);
+ }
+ }
+
+ if (WIFEXITED(stat_loc)) {
+ baton->exit_code = WEXITSTATUS(stat_loc); // errno?
+ }
+
+ if (WIFSIGNALED(stat_loc)) {
+ baton->signal_code = WTERMSIG(stat_loc);
+ }
+
+ uv_async_send(&baton->async);
+}
+
+/**
+ * pty_after_waitpid
+ * Callback after exit status has been read.
+ */
+
+static void
+pty_after_waitpid(uv_async_t *async) {
+ Nan::HandleScope scope;
+ pty_baton *baton = static_cast<pty_baton*>(async->data);
+
+ v8::Local<v8::Value> argv[] = {
+ Nan::New<v8::Integer>(baton->exit_code),
+ Nan::New<v8::Integer>(baton->signal_code),
+ };
+
+ v8::Local<v8::Function> cb = Nan::New<v8::Function>(baton->cb);
+ baton->cb.Reset();
+ memset(&baton->cb, -1, sizeof(baton->cb));
+ Nan::AsyncResource resource("pty_after_waitpid");
+ resource.runInAsyncScope(Nan::GetCurrentContext()->Global(), cb, 2, argv);
+
+ uv_close((uv_handle_t *)async, pty_after_close);
+}
+
+/**
+ * pty_after_close
+ * uv_close() callback - free handle data
+ */
+
+static void
+pty_after_close(uv_handle_t *handle) {
+ uv_async_t *async = (uv_async_t *)handle;
+ pty_baton *baton = static_cast<pty_baton*>(async->data);
+ delete baton;
+}
+
+/**
+ * pty_getproc
+ * Taken from tmux.
+ */
+
+// Taken from: tmux (http://tmux.sourceforge.net/)
+// Copyright (c) 2009 Nicholas Marriott <nicm@users.sourceforge.net>
+// Copyright (c) 2009 Joshua Elsasser <josh@elsasser.org>
+// Copyright (c) 2009 Todd Carson <toc@daybefore.net>
+//
+// Permission to use, copy, modify, and distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+// WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
+// IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
+// OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+#if defined(__linux__)
+
+static char *
+pty_getproc(int fd, char *tty) {
+ FILE *f;
+ char *path, *buf;
+ size_t len;
+ int ch;
+ pid_t pgrp;
+ int r;
+
+ if ((pgrp = tcgetpgrp(fd)) == -1) {
+ return NULL;
+ }
+
+ r = asprintf(&path, "/proc/%lld/cmdline", (long long)pgrp);
+ if (r == -1 || path == NULL) return NULL;
+
+ if ((f = fopen(path, "r")) == NULL) {
+ free(path);
+ return NULL;
+ }
+
+ free(path);
+
+ len = 0;
+ buf = NULL;
+ while ((ch = fgetc(f)) != EOF) {
+ if (ch == '\0') break;
+ buf = (char *)realloc(buf, len + 2);
+ if (buf == NULL) return NULL;
+ buf[len++] = ch;
+ }
+
+ if (buf != NULL) {
+ buf[len] = '\0';
+ }
+
+ fclose(f);
+ return buf;
+}
+
+#elif defined(__APPLE__)
+
+static char *
+pty_getproc(int fd, char *tty) {
+ int mib[4] = { CTL_KERN, KERN_PROC, KERN_PROC_PID, 0 };
+ size_t size;
+ struct kinfo_proc kp;
+
+ if ((mib[3] = tcgetpgrp(fd)) == -1) {
+ return NULL;
+ }
+
+ size = sizeof kp;
+ if (sysctl(mib, 4, &kp, &size, NULL, 0) == -1) {
+ return NULL;
+ }
+
+ if (*kp.kp_proc.p_comm == '\0') {
+ return NULL;
+ }
+
+ return strdup(kp.kp_proc.p_comm);
+}
+
+#else
+
+static char *
+pty_getproc(int fd, char *tty) {
+ return NULL;
+}
+
+#endif
+
+/**
+ * openpty(3) / forkpty(3)
+ */
+
+static int
+pty_openpty(int *amaster,
+ int *aslave,
+ char *name,
+ const struct termios *termp,
+ const struct winsize *winp) {
+#if defined(__sun)
+ char *slave_name;
+ int slave;
+ int master = open("/dev/ptmx", O_RDWR | O_NOCTTY);
+ if (master == -1) return -1;
+ if (amaster) *amaster = master;
+
+ if (grantpt(master) == -1) goto err;
+ if (unlockpt(master) == -1) goto err;
+
+ slave_name = ptsname(master);
+ if (slave_name == NULL) goto err;
+ if (name) strcpy(name, slave_name);
+
+ slave = open(slave_name, O_RDWR | O_NOCTTY);
+ if (slave == -1) goto err;
+ if (aslave) *aslave = slave;
+
+ ioctl(slave, I_PUSH, "ptem");
+ ioctl(slave, I_PUSH, "ldterm");
+ ioctl(slave, I_PUSH, "ttcompat");
+
+ if (termp) tcsetattr(slave, TCSAFLUSH, termp);
+ if (winp) ioctl(slave, TIOCSWINSZ, winp);
+
+ return 0;
+
+err:
+ close(master);
+ return -1;
+#else
+ return openpty(amaster, aslave, name, (termios *)termp, (winsize *)winp);
+#endif
+}
+
+static pid_t
+pty_forkpty(int *amaster,
+ char *name,
+ const struct termios *termp,
+ const struct winsize *winp) {
+#if defined(__sun)
+ int master, slave;
+
+ int ret = pty_openpty(&master, &slave, name, termp, winp);
+ if (ret == -1) return -1;
+ if (amaster) *amaster = master;
+
+ pid_t pid = fork();
+
+ switch (pid) {
+ case -1:
+ close(master);
+ close(slave);
+ return -1;
+ case 0:
+ close(master);
+
+ setsid();
+
+#if defined(TIOCSCTTY)
+ // glibc does this
+ if (ioctl(slave, TIOCSCTTY, NULL) == -1) {
+ _exit(1);
+ }
+#endif
+
+ dup2(slave, 0);
+ dup2(slave, 1);
+ dup2(slave, 2);
+
+ if (slave > 2) close(slave);
+
+ return 0;
+ default:
+ close(slave);
+ return pid;
+ }
+
+ return -1;
+#else
+ return forkpty(amaster, name, (termios *)termp, (winsize *)winp);
+#endif
+}
+
+/**
+ * Init
+ */
+
+NAN_MODULE_INIT(init) {
+ Nan::HandleScope scope;
+ Nan::Export(target, "fork", PtyFork);
+ Nan::Export(target, "open", PtyOpen);
+ Nan::Export(target, "resize", PtyResize);
+ Nan::Export(target, "process", PtyGetProc);
+}
+
+NODE_MODULE(pty, init)
projects / VSoRC / .git / blobdiff