/*      $NetBSD: sys_bsd.c,v 1.40 2018/12/14 06:28:49 maya Exp $        */

/*
 * Copyright (c) 1988, 1990, 1993
 *      The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#ifndef lint
#if 0
from: static char sccsid[] = "@(#)sys_bsd.c     8.4 (Berkeley) 5/30/95";
#else
__RCSID("$NetBSD: sys_bsd.c,v 1.40 2018/12/14 06:28:49 maya Exp $");
#endif
#endif /* not lint */

/*
 * The following routines try to encapsulate what is system dependent
 * (at least between 4.x and dos) which is used in telnet.c.
 */


#include <fcntl.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <signal.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <poll.h>
#include <arpa/telnet.h>

#include "ring.h"
#include "defines.h"
#include "externs.h"
#include "types.h"

void susp(int);
void ayt(int);

void intr(int);
void intr2(int);
void sendwin(int);


int
        tout,                   /* Output file descriptor */
        tin,                    /* Input file descriptor */
        net;

struct  termios old_tc = { .c_iflag = 0 };
extern struct termios new_tc;

void
init_sys(void)
{
    tout = fileno(stdout);
    tin = fileno(stdin);

    errno = 0;
}


int
TerminalWrite(char *buf, int  n)
{
    return write(tout, buf, n);
}

int
TerminalRead(unsigned char *buf, int  n)
{
    return read(tin, buf, n);
}

/*
 *
 */

int
TerminalAutoFlush(void)
{
    return 1;
}

#ifdef  KLUDGELINEMODE
extern int kludgelinemode;
#endif
/*
 * TerminalSpecialChars()
 *
 * Look at an input character to see if it is a special character
 * and decide what to do.
 *
 * Output:
 *
 *      0       Don't add this character.
 *      1       Do add this character
 */

int
TerminalSpecialChars(int c)
{
    if (c == termIntChar) {
        intp();
        return 0;
    } else if (c == termQuitChar) {
#ifdef  KLUDGELINEMODE
        if (kludgelinemode)
            sendbrk();
        else
#endif
            sendabort();
        return 0;
    } else if (c == termEofChar) {
        if (my_want_state_is_will(TELOPT_LINEMODE)) {
            sendeof();
            return 0;
        }
        return 1;
    } else if (c == termSuspChar) {
        sendsusp();
        return(0);
    } else if (c == termFlushChar) {
        xmitAO();               /* Transmit Abort Output */
        return 0;
    } else if (!MODE_LOCAL_CHARS(globalmode)) {
        if (c == termKillChar) {
            xmitEL();
            return 0;
        } else if (c == termEraseChar) {
            xmitEC();           /* Transmit Erase Character */
            return 0;
        }
    }
    return 1;
}


/*
 * Flush output to the terminal
 */

void
TerminalFlushOutput(void)
{
    int com = 0;
    (void) ioctl(fileno(stdout), TIOCFLUSH, &com);
}

void
TerminalSaveState(void)
{
    tcgetattr(0, &old_tc);

    new_tc = old_tc;
}

cc_t *
tcval(int func)
{
    switch(func) {
    case SLC_IP:        return(&termIntChar);
    case SLC_ABORT:     return(&termQuitChar);
    case SLC_EOF:       return(&termEofChar);
    case SLC_EC:        return(&termEraseChar);
    case SLC_EL:        return(&termKillChar);
    case SLC_XON:       return(&termStartChar);
    case SLC_XOFF:      return(&termStopChar);
    case SLC_FORW1:     return(&termForw1Char);
    case SLC_FORW2:     return(&termForw2Char);
    case SLC_AO:        return(&termFlushChar);
    case SLC_SUSP:      return(&termSuspChar);
    case SLC_EW:        return(&termWerasChar);
    case SLC_RP:        return(&termRprntChar);
    case SLC_LNEXT:     return(&termLiteralNextChar);
    case SLC_AYT:       return(&termAytChar);

    case SLC_SYNCH:
    case SLC_BRK:
    case SLC_EOR:
    default:
        return((cc_t *)0);
    }
}

void
TerminalDefaultChars(void)
{
    memmove(new_tc.c_cc, old_tc.c_cc, sizeof(old_tc.c_cc));
}

/*
 * TerminalNewMode - set up terminal to a specific mode.
 *      MODE_ECHO: do local terminal echo
 *      MODE_FLOW: do local flow control
 *      MODE_TRAPSIG: do local mapping to TELNET IAC sequences
 *      MODE_EDIT: do local line editing
 *
 *      Command mode:
 *              MODE_ECHO|MODE_EDIT|MODE_FLOW|MODE_TRAPSIG
 *              local echo
 *              local editing
 *              local xon/xoff
 *              local signal mapping
 *
 *      Linemode:
 *              local/no editing
 *      Both Linemode and Single Character mode:
 *              local/remote echo
 *              local/no xon/xoff
 *              local/no signal mapping
 */


void
TerminalNewMode(int f)
{
    static int prevmode = 0;
    struct termios tmp_tc;
    int onoff;
    int old;
    cc_t esc;

    globalmode = f&~MODE_FORCE;
    if (prevmode == f)
        return;

    /*
     * Write any outstanding data before switching modes
     * ttyflush() returns 0 only when there is no more data
     * left to write out, it returns -1 if it couldn't do
     * anything at all, otherwise it returns 1 + the number
     * of characters left to write.
#ifndef USE_TERMIO
     * We would really like to ask the kernel to wait for the output
     * to drain, like we can do with the TCSADRAIN, but we don't have
     * that option.  The only ioctl that waits for the output to
     * drain, TIOCSETP, also flushes the input queue, which is NOT
     * what we want (TIOCSETP is like TCSADFLUSH).
#endif
     */
    old = ttyflush(SYNCHing|flushout);
    if (old < 0 || old > 1) {
        tcgetattr(tin, &tmp_tc);
        do {
            /*
             * Wait for data to drain, then flush again.
             */
            tcsetattr(tin, TCSADRAIN, &tmp_tc);
            old = ttyflush(SYNCHing|flushout);
        } while (old < 0 || old > 1);
    }

    old = prevmode;
    prevmode = f&~MODE_FORCE;
    tmp_tc = new_tc;

    if (f&MODE_ECHO) {
        tmp_tc.c_lflag |= ECHO;
        tmp_tc.c_oflag |= ONLCR;
        if (crlf)
                tmp_tc.c_iflag |= ICRNL;
    } else {
        tmp_tc.c_lflag &= ~ECHO;
        tmp_tc.c_oflag &= ~ONLCR;
# ifdef notdef
        if (crlf)
                tmp_tc.c_iflag &= ~ICRNL;
# endif
    }

    if ((f&MODE_FLOW) == 0) {
        tmp_tc.c_iflag &= ~(IXOFF|IXON);        /* Leave the IXANY bit alone */
    } else {
        if (restartany < 0) {
                tmp_tc.c_iflag |= IXOFF|IXON;   /* Leave the IXANY bit alone */
        } else if (restartany > 0) {
                tmp_tc.c_iflag |= IXOFF|IXON|IXANY;
        } else {
                tmp_tc.c_iflag |= IXOFF|IXON;
                tmp_tc.c_iflag &= ~IXANY;
        }
    }

    if ((f&MODE_TRAPSIG) == 0) {
        tmp_tc.c_lflag &= ~ISIG;
        localchars = 0;
    } else {
        tmp_tc.c_lflag |= ISIG;
        localchars = 1;
    }

    if (f&MODE_EDIT) {
        tmp_tc.c_lflag |= ICANON;
    } else {
        tmp_tc.c_lflag &= ~ICANON;
        tmp_tc.c_iflag &= ~ICRNL;
        tmp_tc.c_cc[VMIN] = 1;
        tmp_tc.c_cc[VTIME] = 0;
    }

    if ((f&(MODE_EDIT|MODE_TRAPSIG)) == 0) {
        tmp_tc.c_lflag &= ~IEXTEN;
    }

    if (f&MODE_SOFT_TAB) {
# ifdef OXTABS
        tmp_tc.c_oflag |= OXTABS;
# endif
# ifdef TABDLY
        tmp_tc.c_oflag &= ~TABDLY;
        tmp_tc.c_oflag |= TAB3;
# endif
    } else {
# ifdef OXTABS
        tmp_tc.c_oflag &= ~OXTABS;
# endif
# ifdef TABDLY
        tmp_tc.c_oflag &= ~TABDLY;
# endif
    }

    if (f&MODE_LIT_ECHO) {
# ifdef ECHOCTL
        tmp_tc.c_lflag &= ~ECHOCTL;
# endif
    } else {
# ifdef ECHOCTL
        tmp_tc.c_lflag |= ECHOCTL;
# endif
    }

    if (f == -1) {
        onoff = 0;
    } else {
        if (f & MODE_INBIN)
                tmp_tc.c_iflag &= ~ISTRIP;
        else
                tmp_tc.c_iflag |= ISTRIP;
        if (f & MODE_OUTBIN) {
                tmp_tc.c_cflag &= ~(CSIZE|PARENB);
                tmp_tc.c_cflag |= CS8;
                tmp_tc.c_oflag &= ~OPOST;
        } else {
                tmp_tc.c_cflag &= ~(CSIZE|PARENB);
                tmp_tc.c_cflag |= old_tc.c_cflag & (CSIZE|PARENB);
                tmp_tc.c_oflag |= OPOST;
        }
        onoff = 1;
    }

    if (f != -1) {
        (void) signal(SIGTSTP, susp);
        (void) signal(SIGINFO, ayt);
#if     defined(USE_TERMIO) && defined(NOKERNINFO)
        tmp_tc.c_lflag |= NOKERNINFO;
#endif
        /*
         * We don't want to process ^Y here.  It's just another
         * character that we'll pass on to the back end.  It has
         * to process it because it will be processed when the
         * user attempts to read it, not when we send it.
         */
# ifdef VDSUSP
        tmp_tc.c_cc[VDSUSP] = (cc_t)(_POSIX_VDISABLE);
# endif
        /*
         * If the VEOL character is already set, then use VEOL2,
         * otherwise use VEOL.
         */
        esc = (rlogin != _POSIX_VDISABLE) ? rlogin : escape;
        if ((tmp_tc.c_cc[VEOL] != esc)
# ifdef VEOL2
            && (tmp_tc.c_cc[VEOL2] != esc)
# endif
            ) {
                if (tmp_tc.c_cc[VEOL] == (cc_t)(_POSIX_VDISABLE))
                    tmp_tc.c_cc[VEOL] = esc;
# ifdef VEOL2
                else if (tmp_tc.c_cc[VEOL2] == (cc_t)(_POSIX_VDISABLE))
                    tmp_tc.c_cc[VEOL2] = esc;
# endif
        }
    } else {
        (void) signal(SIGINFO, (void (*)(int)) ayt_status);
        (void) signal(SIGTSTP, SIG_DFL);
        (void) sigsetmask(sigblock(0) & ~(1<<(SIGTSTP-1)));
        tmp_tc = old_tc;
    }
    if (tcsetattr(tin, TCSADRAIN, &tmp_tc) < 0)
        tcsetattr(tin, TCSANOW, &tmp_tc);

    ioctl(tin, FIONBIO, &onoff);
    ioctl(tout, FIONBIO, &onoff);

}

void
TerminalSpeeds(long *ispeed, long *ospeed)
{
    long in, out;

    out = cfgetospeed(&old_tc);
    in = cfgetispeed(&old_tc);
    if (in == 0) {
        in = out;

        *ispeed = in;
        *ospeed = out;
    }
}

int
TerminalWindowSize(long *rows, long *cols)
{
    struct winsize ws;

    if (ioctl(fileno(stdin), TIOCGWINSZ, (char *)&ws) >= 0) {
        *rows = ws.ws_row;
        *cols = ws.ws_col;
        return 1;
    }
    return 0;
}

int
NetClose(int fd)
{
    return close(fd);
}

/*
 * Various signal handling routines.
 */

/* ARGSUSED */
void
intr(int sig)
{
    if (localchars) {
        intp();
        return;
    }
    setcommandmode();
    longjmp(toplevel, -1);
}

/* ARGSUSED */
void
intr2(int sig)
{
    if (localchars) {
#ifdef  KLUDGELINEMODE
        if (kludgelinemode)
            sendbrk();
        else
#endif
            sendabort();
        return;
    }
}

/* ARGSUSED */
void
susp(int sig)
{
    if ((rlogin != _POSIX_VDISABLE) && rlogin_susp())
        return;
    if (localchars)
        sendsusp();
}

/* ARGSUSED */
void
sendwin(int sig)
{
    if (connected) {
        sendnaws();
    }
}

/* ARGSUSED */
void
ayt(int sig)
{
    if (connected)
        sendayt();
    else
        ayt_status();
}


void
sys_telnet_init(void)
{
    int one = 1;

    (void) signal(SIGINT, intr);
    (void) signal(SIGQUIT, intr2);
    (void) signal(SIGPIPE, SIG_IGN);
    (void) signal(SIGWINCH, sendwin);
    (void) signal(SIGTSTP, susp);
    (void) signal(SIGINFO, ayt);

    setconnmode(0);

    ioctl(net, FIONBIO, &one);
    if (setsockopt(net, SOL_SOCKET, SO_OOBINLINE, &one, sizeof(one)) == -1) {
        perror("setsockopt");
    }
}

/*
 * Process rings -
 *
 *      This routine tries to fill up/empty our various rings.
 *
 *      The parameter specifies whether this is a poll operation,
 *      or a block-until-something-happens operation.
 *
 *      The return value is 1 if something happened, 0 if not, < 0 if an
 *      error occurred.
 */

int
process_rings(int netin, int netout, int netex, int ttyin, int ttyout,
    int dopoll)         /* If 0, then block until something to do */
{
    struct pollfd set[3];
    int c;
                /* One wants to be a bit careful about setting returnValue
                 * to one, since a one implies we did some useful work,
                 * and therefore probably won't be called to block next
                 * time (TN3270 mode only).
                 */
    int returnValue = 0;

    set[0].fd = net;
    set[0].events = (netout ? POLLOUT : 0) | (netin ? POLLIN : 0) |
        (netex ? POLLPRI : 0);
    set[1].fd = tout;
    set[1].events = ttyout ? POLLOUT : 0;
    set[2].fd = tin;
    set[2].events = ttyin ? POLLIN : 0;

    if ((c = poll(set, 3, dopoll ? 0 : INFTIM)) < 0) {
        if (c == -1) {
                    /*
                     * we can get EINTR if we are in line mode,
                     * and the user does an escape (TSTP), or
                     * some other signal generator.
                     */
            if (errno == EINTR) {
                return 0;
            }
                    /* I don't like this, does it ever happen? */
            printf("sleep(5) from telnet, after poll\r\n");
            sleep(5);
        }
        return 0;
    }

    /*
     * Any urgent data?
     */
    if (set[0].revents & POLLPRI) {
        SYNCHing = 1;
        (void) ttyflush(1);     /* flush already enqueued data */
    }

    /*
     * Something to read from the network...
     */
    if (set[0].revents & POLLIN) {
        int canread;

        canread = ring_empty_consecutive(&netiring);
        c = recv(net, (char *)netiring.supply, canread, 0);
        if (c < 0 && errno == EWOULDBLOCK) {
            c = 0;
        } else if (c <= 0) {
            return -1;
        }
        if (netdata) {
            Dump('<', netiring.supply, c);
        }
        if (c)
            ring_supplied(&netiring, c);
        returnValue = 1;
    }

    /*
     * Something to read from the tty...
     */
    if (set[2].revents & POLLIN) {
        c = TerminalRead(ttyiring.supply, ring_empty_consecutive(&ttyiring));
        if (c < 0 && errno == EIO)
            c = 0;
        if (c < 0 && errno == EWOULDBLOCK) {
            c = 0;
        } else {
            if (c < 0) {
                return -1;
            }
            if (c == 0) {
                /* must be an EOF... */
                if (MODE_LOCAL_CHARS(globalmode) && isatty(tin)) {
                    *ttyiring.supply = termEofChar;
                    c = 1;
                } else {
                    clienteof = 1;
                    shutdown(net, 1);
                    return 0;
                }
            }
            if (termdata) {
                Dump('<', ttyiring.supply, c);
            }
            ring_supplied(&ttyiring, c);
        }
        returnValue = 1;                /* did something useful */
    }

    if (set[0].revents & POLLOUT) {
        returnValue |= netflush();
    }

    if (set[1].revents & (POLLHUP|POLLNVAL))
        return(-1);

    if (set[1].revents & POLLOUT) {
        returnValue |= (ttyflush(SYNCHing|flushout) > 0);
    }

    return returnValue;
}