/*      $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;
}