source: trunk/j/src/jpty/jpty.c @ 2

/*
 * jpty.c
 *
 * Copyright (C) 2000-2001 Peter Graves
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifndef __CYGWIN__
#include <fcntl.h>
#include <grp.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <termios.h>
#endif

#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#ifndef __CYGWIN__
static void set_noecho(int fd);
static void loop(int fdm);
static int  open_master_pty(char *name, size_t size);
#endif

int main(int argc, char *argv[]) 
{
#ifndef __CYGWIN__
    char slave_name[20];
    pid_t pid;
    int fdm;
#endif
    int i = 1;

    if (argc < 2) exit(1);

    /* Check for -cd option. */
    if (!strcmp("-cd", argv[i]))
    {
        /* Next arg is directory to change to. */
        if (++i < argc)
        {
            if (chdir(argv[i]) < 0)
                exit(1);
            ++i;
        }
    }

    /* We should not be out of args here! */
    if (i >= argc) exit(1);

#ifdef __CYGWIN__
    setenv("TERM", "dumb", 1);
    execvp(argv[i], &argv[i]);
    exit(1); /* Not reached. */
#else

    fdm = open_master_pty(slave_name, sizeof(slave_name));

    if (fdm < 0) exit(1);

    pid = fork();

    if (pid < 0) exit(1);

    if (pid == 0)
    { 
        /* Child process. */
        int slave;

        close(fdm);

        if (setsid() < 0) exit(1);

        slave = open(slave_name, O_RDWR);

        if (slave < 0) exit(1);

#ifdef __linux__
        if (ioctl(slave, TIOCSCTTY, NULL) < 0) exit(1);
#endif

        if (dup2(slave, STDIN_FILENO) != STDIN_FILENO) exit(1);

        if (dup2(slave, STDOUT_FILENO) != STDOUT_FILENO) exit(1);

        if (dup2(slave, STDERR_FILENO) != STDERR_FILENO) exit(1);

        if (slave > STDERR_FILENO) close(slave);

        set_noecho(STDIN_FILENO);

        putenv("TERM=dumb");

        execvp(argv[i], &argv[i]);

        /* Not reached. */
        exit(1);
    }

    /* Parent process. */

    loop(fdm);

    exit(0);

#endif
}

#ifndef __CYGWIN__
static void set_noecho(int fd)
{
    struct termios t;
    if (tcgetattr(fd, &t) < 0) exit(1);
    t.c_lflag &= ~(ECHO | ECHOCTL | ECHOE | ECHOK | ECHOKE | ECHONL | ECHOPRT);
    t.c_oflag &= ~(ONLCR);
    if (tcsetattr(fd, TCSANOW, &t) < 0) exit(1);
}

/* Copy stdin to fdm, copy fdm to stdout. */
static void loop(int fdm)
{
    char buf[1024];
    fd_set fdset;
    int done = 0;

    while (!done)
    {
        FD_ZERO(&fdset);
        FD_SET(STDIN_FILENO, &fdset);
        FD_SET(fdm, &fdset);

        select(fdm + 1, &fdset, NULL, NULL, NULL);

        if (FD_ISSET(fdm, &fdset))
        {
            int i = read(fdm, buf, sizeof(buf));

            if (i > 0)
                write(STDOUT_FILENO, buf, i);
            else
                done = 1;
        }

        if (FD_ISSET(STDIN_FILENO, &fdset))
        {
            int i = read(STDIN_FILENO, buf, sizeof(buf));

            if (i > 0)
                write(fdm, buf, i);
            else
                done = 1;
        }
    }
}

static int open_master_pty(char *namebuf, size_t bufsize)
{ 
    char pattern[] = "/dev/ptyXX";

    if (bufsize >= sizeof(pattern))
    {
        int i, j;

        strcpy(namebuf, pattern);

        for (i = 0; i < 16; i++)
        {
            struct stat statbuf;

            namebuf[8] = "pqrstuvwxyzabcde"[i];
            namebuf[9] = '0';

            if (stat(namebuf, &statbuf) < 0)
                continue;

            for (j = 0; j < 16; j++)
            {
                int fdm;

                namebuf[9] = "0123456789abcdef"[j];

                fdm = open(namebuf, O_RDWR);

                if (fdm > 0)
                {
                    namebuf[5] = 't';
                    return fdm;
                }
            }
        }
    }

    return -1;
}
#endif