/*
* Copyright (c) 2006-2023, RT-Thread Development Team
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2023-11-13 Shell init ver.
*/
#include "../bsd_porting.h"
#include "../tty_config.h"
#include "../tty_internal.h"
#include "../terminal.h"
#include <rtdef.h>
/*-
* SPDX-License-Identifier: BSD-2-Clause
*
* Copyright (c) 2008 Ed Schouten <ed@FreeBSD.org>
* All rights reserved.
*
* Portions of this software were developed under sponsorship from Snow
* B.V., the Netherlands.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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.
*/
/*
* Standard TTYDISC `termios' line discipline.
*/
/* Statistics. */
static rt_atomic_t tty_nin = 0;
#ifdef USING_BSD_SYSCTL
SYSCTL_ULONG(_kern, OID_AUTO, tty_nin, CTLFLAG_RD, &tty_nin, 0,
"Total amount of bytes received");
#endif
static rt_atomic_t tty_nout = 0;
#ifdef USING_BSD_SYSCTL
SYSCTL_ULONG(_kern, OID_AUTO, tty_nout, CTLFLAG_RD, &tty_nout, 0,
"Total amount of bytes transmitted");
#endif
/* termios comparison macro's. */
#define CMP_CC(v, c) \
(tp->t_termios.c_cc[v] != _POSIX_VDISABLE && tp->t_termios.c_cc[v] == (c))
#define CMP_FLAG(field, opt) (tp->t_termios.c_##field##flag & (opt))
/* Characters that cannot be modified through c_cc. */
#define CTAB '\t'
#define CNL '\n'
#define CCR '\r'
/* Character is a control character. */
#define CTL_VALID(c) ((c) == 0x7f || (unsigned char)(c) < 0x20)
/* Control character should be processed on echo. */
#define CTL_ECHO(c, q) \
(!(q) && ((c) == CERASE2 || (c) == CTAB || (c) == CNL || (c) == CCR))
/* Control character should be printed using ^X notation. */
#define CTL_PRINT(c, q) \
((c) == 0x7f || \
((unsigned char)(c) < 0x20 && ((q) || ((c) != CTAB && (c) != CNL))))
/* Character is whitespace. */
#define CTL_WHITE(c) ((c) == ' ' || (c) == CTAB)
/* Character is alphanumeric. */
#define CTL_ALNUM(c) \
(((c) >= '0' && (c) <= '9') || ((c) >= 'a' && (c) <= 'z') || \
((c) >= 'A' && (c) <= 'Z'))
/* Character is UTF8-encoded. */
#define CTL_UTF8(c) (!!((c)&0x80))
/* Character is a UTF8 continuation byte. */
#define CTL_UTF8_CONT(c) (((c)&0xc0) == 0x80)
#define TTY_STACKBUF 256
#define UTF8_STACKBUF 4
void ttydisc_open(struct lwp_tty *tp)
{
ttydisc_optimize(tp);
}
void ttydisc_close(struct lwp_tty *tp)
{
/* Clean up our flags when leaving the discipline. */
tp->t_flags &= ~(TF_STOPPED | TF_HIWAT | TF_ZOMBIE);
tp->t_termios.c_lflag &= ~FLUSHO;
/*
* POSIX states that we must drain output and flush input on
* last close. Draining has already been done if possible.
*/
tty_flush(tp, FREAD | FWRITE);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, close))
ttyhook_close(tp);
#endif
}
static int ttydisc_read_canonical(struct lwp_tty *tp, struct uio *uio,
int ioflag)
{
char breakc[4] = {CNL}; /* enough to hold \n, VEOF and VEOL. */
int error;
size_t clen, flen = 0, n = 1;
unsigned char lastc = _POSIX_VDISABLE;
#define BREAK_ADD(c) \
do \
{ \
if (tp->t_termios.c_cc[c] != _POSIX_VDISABLE) \
breakc[n++] = tp->t_termios.c_cc[c]; \
} while (0)
/* Determine which characters we should trigger on. */
BREAK_ADD(VEOF);
BREAK_ADD(VEOL);
#undef BREAK_ADD
breakc[n] = '\0';
do
{
error = tty_wait_background(tp, curthread, SIGTTIN);
if (error)
return error;
/*
* Quite a tricky case: unlike the old TTY
* implementation, this implementation copies data back
* to userspace in large chunks. Unfortunately, we can't
* calculate the line length on beforehand if it crosses
* ttyinq_block boundaries, because multiple reads could
* then make this code read beyond the newline.
*
* This is why we limit the read to:
* - The size the user has requested
* - The blocksize (done in tty_inq.c)
* - The amount of bytes until the newline
*
* This causes the line length to be recalculated after
* each block has been copied to userspace. This will
* cause the TTY layer to return data in chunks using
* the blocksize (except the first and last blocks).
*/
clen =
ttyinq_findchar(&tp->t_inq, breakc, uio->uio_resid, (char *)&lastc);
/* No more data. */
if (clen == 0)
{
if (tp->t_flags & TF_ZOMBIE)
return 0;
else if (ioflag & IO_NDELAY)
return -EWOULDBLOCK;
error = tty_wait(tp, &tp->t_inwait);
if (error)
return error;
continue;
}
/* Don't send the EOF char back to userspace. */
if (CMP_CC(VEOF, lastc))
flen = 1;
MPASS(flen <= clen);
/* Read and throw away the EOF character. */
error = ttyinq_read_uio(&tp->t_inq, tp, uio, clen, flen);
if (error)
return error;
/** Notes: proceed when read buf frees and not reaching breaks */
} while (uio->uio_resid > 0 && lastc == _POSIX_VDISABLE);
return 0;
}
static int ttydisc_read_raw_no_timer(struct lwp_tty *tp, struct uio *uio,
int ioflag)
{
size_t vmin = tp->t_termios.c_cc[VMIN];
ssize_t oresid = uio->uio_resid;
int error;
MPASS(tp->t_termios.c_cc[VTIME] == 0);
/*
* This routine implements the easy cases of read()s while in
* non-canonical mode, namely case B and D, where we don't have
* any timers at all.
*/
for (;;)
{
error = tty_wait_background(tp, curthread, SIGTTIN);
if (error)
return error;
error = ttyinq_read_uio(&tp->t_inq, tp, uio, uio->uio_resid, 0);
if (error)
return error;
if (uio->uio_resid == 0 || (oresid - uio->uio_resid) >= vmin)
return 0;
/* We have to wait for more. */
if (tp->t_flags & TF_ZOMBIE)
return 0;
else if (ioflag & IO_NDELAY)
return -EWOULDBLOCK;
error = tty_wait(tp, &tp->t_inwait);
if (error)
return error;
}
}
static int ttydisc_read_raw_read_timer(struct lwp_tty *tp, struct uio *uio,
int ioflag, int oresid)
{
size_t vmin = MAX(tp->t_termios.c_cc[VMIN], 1);
unsigned int vtime = tp->t_termios.c_cc[VTIME];
struct timeval end, now, left;
int error, hz;
MPASS(tp->t_termios.c_cc[VTIME] != 0);
/* Determine when the read should be expired. */
end.tv_sec = vtime / 10;
end.tv_usec = (vtime % 10) * 100000;
getmicrotime(&now);
timevaladd(&end, &now);
for (;;)
{
error = tty_wait_background(tp, curthread, SIGTTIN);
if (error)
return error;
error = ttyinq_read_uio(&tp->t_inq, tp, uio, uio->uio_resid, 0);
if (error)
return error;
if (uio->uio_resid == 0 || (oresid - uio->uio_resid) >= vmin)
return 0;
/* Calculate how long we should wait. */
getmicrotime(&now);
if (timevalcmp(&now, &end, >))
return 0;
left = end;
timevalsub(&left, &now);
hz = tvtohz(&left);
/*
* We have to wait for more. If the timer expires, we
* should return a 0-byte read.
*/
if (tp->t_flags & TF_ZOMBIE)
return 0;
else if (ioflag & IO_NDELAY)
return -EWOULDBLOCK;
error = tty_timedwait(tp, &tp->t_inwait, hz);
if (error)
return (error == -EWOULDBLOCK ? 0 : error);
}
return 0;
}
static int ttydisc_read_raw_interbyte_timer(struct lwp_tty *tp, struct uio *uio,
int ioflag)
{
size_t vmin = tp->t_termios.c_cc[VMIN];
ssize_t oresid = uio->uio_resid;
int error;
MPASS(tp->t_termios.c_cc[VMIN] != 0);
MPASS(tp->t_termios.c_cc[VTIME] != 0);
/*
* When using the interbyte timer, the timer should be started
* after the first byte has been received. We just call into the
* generic read timer code after we've received the first byte.
*/
for (;;)
{
error = tty_wait_background(tp, curthread, SIGTTIN);
if (error)
return error;
error = ttyinq_read_uio(&tp->t_inq, tp, uio, uio->uio_resid, 0);
if (error)
return error;
if (uio->uio_resid == 0 || (oresid - uio->uio_resid) >= vmin)
return 0;
/*
* Not enough data, but we did receive some, which means
* we'll now start using the interbyte timer.
*/
if (oresid != uio->uio_resid)
break;
/* We have to wait for more. */
if (tp->t_flags & TF_ZOMBIE)
return 0;
else if (ioflag & IO_NDELAY)
return -EWOULDBLOCK;
error = tty_wait(tp, &tp->t_inwait);
if (error)
return error;
}
return ttydisc_read_raw_read_timer(tp, uio, ioflag, oresid);
}
int ttydisc_read(struct lwp_tty *tp, struct uio *uio, int ioflag)
{
int error;
tty_assert_locked(tp);
if (uio->uio_resid == 0)
return 0;
if (CMP_FLAG(l, ICANON))
error = ttydisc_read_canonical(tp, uio, ioflag);
else if (tp->t_termios.c_cc[VTIME] == 0)
error = ttydisc_read_raw_no_timer(tp, uio, ioflag);
else if (tp->t_termios.c_cc[VMIN] == 0)
error = ttydisc_read_raw_read_timer(tp, uio, ioflag, uio->uio_resid);
else
error = ttydisc_read_raw_interbyte_timer(tp, uio, ioflag);
if (ttyinq_bytesleft(&tp->t_inq) >= tp->t_inlow ||
ttyinq_bytescanonicalized(&tp->t_inq) == 0)
{
/* Unset the input watermark when we've got enough space. */
tty_hiwat_in_unblock(tp);
}
return error;
}
/* return the offset to special character from string start */
rt_inline unsigned int ttydisc_findchar(const char *obstart, unsigned int oblen)
{
const char *c = obstart;
while (oblen--)
{
if (CTL_VALID(*c))
break;
c++;
}
return (c - obstart);
}
static int ttydisc_write_oproc(struct lwp_tty *tp, char c)
{
unsigned int scnt, error;
MPASS(CMP_FLAG(o, OPOST));
MPASS(CTL_VALID(c));
#define PRINT_NORMAL() ttyoutq_write_nofrag(&tp->t_outq, &c, 1)
switch (c)
{
case CEOF:
/* End-of-text dropping. */
if (CMP_FLAG(o, ONOEOT))
return 0;
return PRINT_NORMAL();
case CERASE2:
/* Handle backspace to fix tab expansion. */
if (PRINT_NORMAL() != 0)
return (-1);
if (tp->t_column > 0)
tp->t_column--;
return 0;
case CTAB:
/* Tab expansion. */
scnt = 8 - (tp->t_column & 7);
if (CMP_FLAG(o, TAB3))
{
error = ttyoutq_write_nofrag(&tp->t_outq, " ", scnt);
}
else
{
error = PRINT_NORMAL();
}
if (error)
return (-1);
tp->t_column += scnt;
MPASS((tp->t_column % 8) == 0);
return 0;
case CNL:
/* Newline conversion. */
if (CMP_FLAG(o, ONLCR))
{
/* Convert \n to \r\n. */
error = ttyoutq_write_nofrag(&tp->t_outq, "\r\n", 2);
}
else
{
error = PRINT_NORMAL();
}
if (error)
return (-1);
if (CMP_FLAG(o, ONLCR | ONLRET))
{
tp->t_column = tp->t_writepos = 0;
ttyinq_reprintpos_set(&tp->t_inq);
}
return 0;
case CCR:
/* Carriage return to newline conversion. */
if (CMP_FLAG(o, OCRNL))
c = CNL;
/* Omit carriage returns on column 0. */
if (CMP_FLAG(o, ONOCR) && tp->t_column == 0)
return 0;
if (PRINT_NORMAL() != 0)
return (-1);
tp->t_column = tp->t_writepos = 0;
ttyinq_reprintpos_set(&tp->t_inq);
return 0;
}
/*
* Invisible control character. Print it, but don't
* increase the column count.
*/
return PRINT_NORMAL();
#undef PRINT_NORMAL
}
/*
* Just like the old TTY implementation, we need to copy data in chunks
* into a temporary buffer. One of the reasons why we need to do this,
* is because output processing (only TAB3 though) may allow the buffer
* to grow eight times.
*/
int ttydisc_write(struct lwp_tty *tp, struct uio *uio, int ioflag)
{
char ob[TTY_STACKBUF];
char *obstart;
int error = 0;
unsigned int oblen = 0;
tty_assert_locked(tp);
if (tp->t_flags & TF_ZOMBIE)
return -EIO;
/*
* We don't need to check whether the process is the foreground
* process group or if we have a carrier. This is already done
* in ttydev_write().
*/
while (uio->uio_resid > 0)
{
unsigned int nlen;
MPASS(oblen == 0);
if (CMP_FLAG(l, FLUSHO))
{
uio->uio_offset += uio->uio_resid;
uio->uio_resid = 0;
return 0;
}
/* Step 1: read data. */
obstart = ob;
nlen = MIN(uio->uio_resid, sizeof ob);
tty_unlock(tp);
error = uiomove(ob, nlen, uio);
tty_lock(tp);
if (error != 0)
break;
oblen = nlen;
if (tty_gone(tp))
{
error = -ENXIO;
break;
}
MPASS(oblen > 0);
/* Step 2: process data. */
do
{
unsigned int plen, wlen;
if (CMP_FLAG(l, FLUSHO))
{
uio->uio_offset += uio->uio_resid;
uio->uio_resid = 0;
return 0;
}
/* Search for special characters for post processing. */
if (CMP_FLAG(o, OPOST))
{
plen = ttydisc_findchar(obstart, oblen);
}
else
{
plen = oblen;
}
if (plen == 0)
{
/*
* We're going to process a character
* that needs processing
*/
if (ttydisc_write_oproc(tp, *obstart) == 0)
{
obstart++;
oblen--;
tp->t_writepos = tp->t_column;
ttyinq_reprintpos_set(&tp->t_inq);
continue;
}
}
else
{
/* We're going to write regular data. */
wlen = ttyoutq_write(&tp->t_outq, obstart, plen);
obstart += wlen;
oblen -= wlen;
tp->t_column += wlen;
tp->t_writepos = tp->t_column;
ttyinq_reprintpos_set(&tp->t_inq);
if (wlen == plen)
continue;
}
/* Watermark reached. Try to sleep. */
tp->t_flags |= TF_HIWAT_OUT;
if (ioflag & IO_NDELAY)
{
error = EWOULDBLOCK;
goto done;
}
/*
* The driver may write back the data
* synchronously. Be sure to check the high
* water mark before going to sleep.
*/
ttydevsw_outwakeup(tp);
if ((tp->t_flags & TF_HIWAT_OUT) == 0)
continue;
error = tty_wait(tp, &tp->t_outwait);
if (error)
goto done;
if (tp->t_flags & TF_ZOMBIE)
{
error = EIO;
goto done;
}
} while (oblen > 0);
}
done:
if (!tty_gone(tp))
ttydevsw_outwakeup(tp);
/*
* Add the amount of bytes that we didn't process back to the
* uio counters. We need to do this to make sure write() doesn't
* count the bytes we didn't store in the queue.
*/
uio->uio_resid += oblen;
return error;
}
/* set optimize flags if possible */
void ttydisc_optimize(struct lwp_tty *tp)
{
tty_assert_locked(tp);
if (ttyhook_hashook(tp, rint_bypass))
{
tp->t_flags |= TF_BYPASS;
}
else if (ttyhook_hashook(tp, rint))
{
tp->t_flags &= ~TF_BYPASS;
}
else if (!CMP_FLAG(i, ICRNL | IGNCR | IMAXBEL | INLCR | ISTRIP | IXON) &&
(!CMP_FLAG(i, BRKINT) || CMP_FLAG(i, IGNBRK)) &&
(!CMP_FLAG(i, PARMRK) ||
CMP_FLAG(i, IGNPAR | IGNBRK) == (IGNPAR | IGNBRK)) &&
!CMP_FLAG(l, ECHO | ICANON | IEXTEN | ISIG | PENDIN))
{
tp->t_flags |= TF_BYPASS;
}
else
{
tp->t_flags &= ~TF_BYPASS;
}
}
void ttydisc_modem(struct lwp_tty *tp, int open)
{
tty_assert_locked(tp);
if (open)
cv_broadcast(&tp->t_dcdwait);
/*
* Ignore modem status lines when CLOCAL is turned on, but don't
* enter the zombie state when the TTY isn't opened, because
* that would cause the TTY to be in zombie state after being
* opened.
*/
if (!tty_opened(tp) || CMP_FLAG(c, CLOCAL))
return;
if (open == 0)
{
/*
* Lost carrier.
*/
tp->t_flags |= TF_ZOMBIE;
lwp_tty_signal_sessleader(tp, SIGHUP);
tty_flush(tp, FREAD | FWRITE);
}
else
{
/*
* Carrier is back again.
*/
/* XXX: what should we do here? */
}
}
static int ttydisc_echo_force(struct lwp_tty *tp, char c, int quote)
{
if (CMP_FLAG(l, FLUSHO))
return 0;
if (CMP_FLAG(o, OPOST) && CTL_ECHO(c, quote))
{
/*
* Only perform postprocessing when OPOST is turned on
* and the character is an unquoted BS/TB/NL/CR.
*/
return ttydisc_write_oproc(tp, c);
}
else if (CMP_FLAG(l, ECHOCTL) && CTL_PRINT(c, quote))
{
/*
* Only use ^X notation when ECHOCTL is turned on and
* we've got an quoted control character.
*
* Print backspaces when echoing an end-of-file.
*/
char ob[4] = "^?\b\b";
/* Print ^X notation. */
if (c != 0x7f)
ob[1] = c + 'A' - 1;
if (!quote && CMP_CC(VEOF, c))
{
return ttyoutq_write_nofrag(&tp->t_outq, ob, 4);
}
else
{
tp->t_column += 2;
return ttyoutq_write_nofrag(&tp->t_outq, ob, 2);
}
}
else
{
/* Can just be printed. */
tp->t_column++;
return ttyoutq_write_nofrag(&tp->t_outq, &c, 1);
}
}
static int ttydisc_echo(struct lwp_tty *tp, char c, int quote)
{
/*
* Only echo characters when ECHO is turned on, or ECHONL when
* the character is an unquoted newline.
*/
if (!CMP_FLAG(l, ECHO) && (!CMP_FLAG(l, ECHONL) || c != CNL || quote))
return 0;
return ttydisc_echo_force(tp, c, quote);
}
static void ttydisc_reprint_char(void *d, char c, int quote)
{
struct lwp_tty *tp = d;
ttydisc_echo(tp, c, quote);
}
static void ttydisc_reprint(struct lwp_tty *tp)
{
cc_t c;
/* Print ^R\n, followed by the line. */
c = tp->t_termios.c_cc[VREPRINT];
if (c != _POSIX_VDISABLE)
ttydisc_echo(tp, c, 0);
ttydisc_echo(tp, CNL, 0);
ttyinq_reprintpos_reset(&tp->t_inq);
ttyinq_line_iterate_from_linestart(&tp->t_inq, ttydisc_reprint_char, tp);
}
struct ttydisc_recalc_length
{
struct lwp_tty *tp;
unsigned int curlen;
};
static void ttydisc_recalc_charlength(void *d, char c, int quote)
{
struct ttydisc_recalc_length *data = d;
struct lwp_tty *tp = data->tp;
if (CTL_PRINT(c, quote))
{
if (CMP_FLAG(l, ECHOCTL))
data->curlen += 2;
}
else if (c == CTAB)
{
data->curlen += 8 - (data->curlen & 7);
}
else
{
data->curlen++;
}
}
static unsigned int ttydisc_recalc_linelength(struct lwp_tty *tp)
{
struct ttydisc_recalc_length data = {tp, tp->t_writepos};
ttyinq_line_iterate_from_reprintpos(&tp->t_inq, ttydisc_recalc_charlength,
&data);
return (data.curlen);
}
static int ttydisc_rubchar(struct lwp_tty *tp)
{
char c;
int quote;
unsigned int prevpos, tablen;
if (ttyinq_peekchar(&tp->t_inq, &c, "e) != 0)
return (-1);
ttyinq_unputchar(&tp->t_inq);
if (CMP_FLAG(l, ECHO))
{
/*
* Remove the character from the screen. This is even
* safe for characters that span multiple characters
* (tabs, quoted, etc).
*/
if (tp->t_writepos >= tp->t_column)
{
/* Retype the sentence. */
ttydisc_reprint(tp);
}
else if (CMP_FLAG(l, ECHOE))
{
if (CTL_PRINT(c, quote))
{
/* Remove ^X formatted chars. */
if (CMP_FLAG(l, ECHOCTL))
{
tp->t_column -= 2;
ttyoutq_write_nofrag(&tp->t_outq, "\b\b \b\b", 6);
}
}
else if (c == ' ')
{
/* Space character needs no rubbing. */
tp->t_column -= 1;
ttyoutq_write_nofrag(&tp->t_outq, "\b", 1);
}
else if (c == CTAB)
{
/*
* Making backspace work with tabs is
* quite hard. Recalculate the length of
* this character and remove it.
*
* Because terminal settings could be
* changed while the line is being
* inserted, the calculations don't have
* to be correct. Make sure we keep the
* tab length within proper bounds.
*/
prevpos = ttydisc_recalc_linelength(tp);
if (prevpos >= tp->t_column)
tablen = 1;
else
tablen = tp->t_column - prevpos;
if (tablen > 8)
tablen = 8;
tp->t_column = prevpos;
ttyoutq_write_nofrag(&tp->t_outq, "\b\b\b\b\b\b\b\b", tablen);
return 0;
}
#ifdef USING_BSD_UTF8
else if ((tp->t_termios.c_iflag & IUTF8) != 0 && CTL_UTF8(c))
{
uint8_t bytes[UTF8_STACKBUF] = {0};
int curidx = UTF8_STACKBUF - 1, cwidth = 1, nb = 0;
teken_char_t codepoint;
/* Save current byte. */
bytes[curidx] = c;
curidx--;
nb++;
/* Loop back through inq until we hit the
* leading byte. */
while (CTL_UTF8_CONT(c) && nb < UTF8_STACKBUF)
{
/*
* Check if we've reached the beginning
* of the line.
*/
if (ttyinq_peekchar(&tp->t_inq, &c, "e) != 0)
break;
ttyinq_unputchar(&tp->t_inq);
bytes[curidx] = c;
curidx--;
nb++;
}
/*
* Shift array so that the leading
* byte ends up at idx 0.
*/
if (nb < UTF8_STACKBUF)
memmove(&bytes[0], &bytes[curidx + 1],
nb * sizeof(uint8_t));
/* Check for malformed UTF8 characters. */
if (nb == UTF8_STACKBUF && CTL_UTF8_CONT(bytes[0]))
{
/*
* Place all bytes back into the inq and
* delete the last byte only.
*/
ttyinq_write(&tp->t_inq, bytes, UTF8_STACKBUF, 0);
ttyinq_unputchar(&tp->t_inq);
}
else
{
/* Find codepoint and width. */
codepoint = teken_utf8_bytes_to_codepoint(bytes, nb);
if (codepoint == TEKEN_UTF8_INVALID_CODEPOINT ||
(cwidth = teken_wcwidth(codepoint)) == -1)
{
/*
* Place all bytes back into the
* inq and fall back to
* default behaviour.
*/
cwidth = 1;
ttyinq_write(&tp->t_inq, bytes, nb, 0);
ttyinq_unputchar(&tp->t_inq);
}
}
tp->t_column -= cwidth;
/*
* Delete character by punching
* 'cwidth' spaces over it.
*/
if (cwidth == 1)
ttyoutq_write_nofrag(&tp->t_outq, "\b \b", 3);
else if (cwidth == 2)
ttyoutq_write_nofrag(&tp->t_outq, "\b\b \b\b", 6);
}
#endif
else
{
/*
* Remove a regular character by
* punching a space over it.
*/
tp->t_column -= 1;
ttyoutq_write_nofrag(&tp->t_outq, "\b \b", 3);
}
}
else
{
/* Don't print spaces. */
ttydisc_echo(tp, tp->t_termios.c_cc[VERASE], 0);
}
}
return 0;
}
static void ttydisc_rubword(struct lwp_tty *tp)
{
char c;
int quote;
#ifdef ALTWERASE
int alnum;
#endif
/* Strip whitespace first. */
for (;;)
{
if (ttyinq_peekchar(&tp->t_inq, &c, "e) != 0)
return;
if (!CTL_WHITE(c))
break;
ttydisc_rubchar(tp);
}
/*
* Record whether the last character from the previous iteration
* was alphanumeric or not. We need this to implement ALTWERASE.
*/
#ifdef ALTWERASE
alnum = CTL_ALNUM(c);
#endif
for (;;)
{
ttydisc_rubchar(tp);
if (ttyinq_peekchar(&tp->t_inq, &c, "e) != 0)
return;
if (CTL_WHITE(c))
return;
#ifdef ALTWERASE
if (CMP_FLAG(l, ALTWERASE) && CTL_ALNUM(c) != alnum)
return;
#endif
}
}
int ttydisc_rint(struct lwp_tty *tp, char c, int flags)
{
int signal, quote = 0;
char ob[3] = {0xff, 0x00};
size_t ol;
tty_assert_locked(tp);
rt_atomic_add(&tty_nin, 1);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, rint))
return ttyhook_rint(tp, c, flags);
#endif
if (tp->t_flags & TF_BYPASS)
goto processed;
if (flags)
{
if (flags & TRE_BREAK)
{
if (CMP_FLAG(i, IGNBRK))
{
/* Ignore break characters. */
return 0;
}
else if (CMP_FLAG(i, BRKINT))
{
/* Generate SIGINT on break. */
tty_flush(tp, FREAD | FWRITE);
lwp_tty_signal_pgrp(tp, SIGINT);
return 0;
}
else
{
/* Just print it. */
goto parmrk;
}
}
else if (flags & TRE_FRAMING ||
(flags & TRE_PARITY && CMP_FLAG(i, INPCK)))
{
if (CMP_FLAG(i, IGNPAR))
{
/* Ignore bad characters. */
return 0;
}
else
{
/* Just print it. */
goto parmrk;
}
}
}
/* Allow any character to perform a wakeup. */
if (CMP_FLAG(i, IXANY))
{
tp->t_flags &= ~TF_STOPPED;
tp->t_termios.c_lflag &= ~FLUSHO; /* FLUSHO: Output is being flushed */
}
/* Remove the top bit. */
if (CMP_FLAG(i, ISTRIP))
c &= ~0x80;
/* Skip input processing when we want to print it literally. */
if (tp->t_flags & TF_LITERAL)
{
tp->t_flags &= ~TF_LITERAL;
quote = 1;
goto processed;
}
/* Special control characters that are implementation dependent. */
if (CMP_FLAG(l, IEXTEN))
{
/* Accept the next character as literal. */
if (CMP_CC(VLNEXT, c))
{
if (CMP_FLAG(l, ECHO))
{
if (CMP_FLAG(l, ECHOE))
ttyoutq_write_nofrag(&tp->t_outq, "^\b", 2);
else
ttydisc_echo(tp, c, 0);
}
tp->t_flags |= TF_LITERAL;
return 0;
}
/* Discard processing */
if (CMP_CC(VDISCARD, c))
{
if (CMP_FLAG(l, FLUSHO))
{
tp->t_termios.c_lflag &= ~FLUSHO;
}
else
{
tty_flush(tp, FWRITE);
ttydisc_echo(tp, c, 0);
if (tp->t_inq.ti_end > 0)
ttydisc_reprint(tp);
tp->t_termios.c_lflag |= FLUSHO;
}
}
}
/*
* Handle signal processing.
*/
if (CMP_FLAG(l, ISIG))
{
#ifdef USING_BSD_SIGINFO
if (CMP_FLAG(l, ICANON | IEXTEN) == (ICANON | IEXTEN))
{
if (CMP_CC(VSTATUS, c))
{
tty_signal_pgrp(tp, SIGINFO);
return 0;
}
}
#endif /* USING_BSD_SIGINFO */
/*
* When compared to the old implementation, this
* implementation also flushes the output queue. POSIX
* is really brief about this, but does makes us assume
* we have to do so.
*/
signal = 0;
if (CMP_CC(VINTR, c))
{
signal = SIGINT;
}
else if (CMP_CC(VQUIT, c))
{
signal = SIGQUIT;
}
else if (CMP_CC(VSUSP, c))
{
signal = SIGTSTP;
}
if (signal != 0)
{
/*
* Echo the character before signalling the
* processes.
*/
if (!CMP_FLAG(l, NOFLSH))
tty_flush(tp, FREAD | FWRITE);
ttydisc_echo(tp, c, 0);
lwp_tty_signal_pgrp(tp, signal);
return 0;
}
}
/*
* Handle start/stop characters.
*/
if (CMP_FLAG(i, IXON))
{
if (CMP_CC(VSTOP, c))
{
/* Stop it if we aren't stopped yet. */
if ((tp->t_flags & TF_STOPPED) == 0)
{
tp->t_flags |= TF_STOPPED;
return 0;
}
/*
* Fallthrough:
* When VSTART == VSTOP, we should make this key
* toggle it.
*/
if (!CMP_CC(VSTART, c))
return 0;
}
if (CMP_CC(VSTART, c))
{
tp->t_flags &= ~TF_STOPPED;
return 0;
}
}
/* Conversion of CR and NL. */
switch (c)
{
case CCR:
if (CMP_FLAG(i, IGNCR))
return 0;
if (CMP_FLAG(i, ICRNL))
c = CNL;
break;
case CNL:
if (CMP_FLAG(i, INLCR))
c = CCR;
break;
}
/* Canonical line editing. */
if (CMP_FLAG(l, ICANON))
{
if (CMP_CC(VERASE, c)
#ifdef VERASE2
|| CMP_CC(VERASE2, c)
#endif
)
{
ttydisc_rubchar(tp);
return 0;
}
else if (CMP_CC(VKILL, c))
{
while (ttydisc_rubchar(tp) == 0)
;
return 0;
}
else if (CMP_FLAG(l, IEXTEN))
{
if (CMP_CC(VWERASE, c))
{
ttydisc_rubword(tp);
return 0;
}
else if (CMP_CC(VREPRINT, c))
{
ttydisc_reprint(tp);
return 0;
}
}
}
processed:
if (CMP_FLAG(i, PARMRK) && (unsigned char)c == 0xff)
{
/* Print 0xff 0xff. */
ob[1] = 0xff;
ol = 2;
quote = 1;
}
else
{
ob[0] = c;
ol = 1;
}
goto print;
parmrk:
if (CMP_FLAG(i, PARMRK))
{
/* Prepend 0xff 0x00 0x.. */
ob[2] = c;
ol = 3;
quote = 1;
}
else
{
ob[0] = c;
ol = 1;
}
print:
/* See if we can store this on the input queue. */
if (ttyinq_write_nofrag(&tp->t_inq, ob, ol, quote) != 0)
{
if (CMP_FLAG(i, IMAXBEL))
ttyoutq_write_nofrag(&tp->t_outq, "\a", 1);
/*
* Prevent a deadlock here. It may be possible that a
* user has entered so much data, there is no data
* available to read(), but the buffers are full anyway.
*
* Only enter the high watermark if the device driver
* can actually transmit something.
*/
if (ttyinq_bytescanonicalized(&tp->t_inq) == 0)
return 0;
tty_hiwat_in_block(tp);
return (-1);
}
/*
* In raw mode, we canonicalize after receiving a single
* character. Otherwise, we canonicalize when we receive a
* newline, VEOL or VEOF, but only when it isn't quoted.
*/
if (!CMP_FLAG(l, ICANON) ||
(!quote && (c == CNL || CMP_CC(VEOL, c) || CMP_CC(VEOF, c))))
{
ttyinq_canonicalize(&tp->t_inq);
}
ttydisc_echo(tp, c, quote);
return 0;
}
size_t ttydisc_rint_simple(struct lwp_tty *tp, const void *buf, size_t len)
{
const char *cbuf;
if (ttydisc_can_bypass(tp))
return (ttydisc_rint_bypass(tp, buf, len));
for (cbuf = buf; len-- > 0; cbuf++)
{
if (ttydisc_rint(tp, *cbuf, 0) != 0)
break;
}
return (cbuf - (const char *)buf);
}
size_t ttydisc_rint_bypass(struct lwp_tty *tp, const void *buf, size_t len)
{
size_t ret;
tty_assert_locked(tp);
MPASS(tp->t_flags & TF_BYPASS);
rt_atomic_add(&tty_nin, len);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, rint_bypass))
return ttyhook_rint_bypass(tp, buf, len);
#endif
ret = ttyinq_write(&tp->t_inq, buf, len, 0);
ttyinq_canonicalize(&tp->t_inq);
if (ret < len)
tty_hiwat_in_block(tp);
return (ret);
}
void ttydisc_rint_done(struct lwp_tty *tp)
{
tty_assert_locked(tp);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, rint_done))
ttyhook_rint_done(tp);
#endif
/* Wake up readers. */
tty_wakeup(tp, FREAD);
/* Wake up driver for echo. */
ttydevsw_outwakeup(tp);
}
size_t ttydisc_rint_poll(struct lwp_tty *tp)
{
size_t l;
tty_assert_locked(tp);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, rint_poll))
return ttyhook_rint_poll(tp);
#endif
/*
* XXX: Still allow character input when there's no space in the
* buffers, but we haven't entered the high watermark. This is
* to allow backspace characters to be inserted when in
* canonical mode.
*/
l = ttyinq_bytesleft(&tp->t_inq);
if (l == 0 && (tp->t_flags & TF_HIWAT_IN) == 0)
return (1);
return (l);
}
static void ttydisc_wakeup_watermark(struct lwp_tty *tp)
{
size_t c;
c = ttyoutq_bytesleft(&tp->t_outq);
if (tp->t_flags & TF_HIWAT_OUT)
{
/* Only allow us to run when we're below the watermark. */
if (c < tp->t_outlow)
return;
/* Reset the watermark. */
tp->t_flags &= ~TF_HIWAT_OUT;
}
else
{
/* Only run when we have data at all. */
if (c == 0)
return;
}
tty_wakeup(tp, FWRITE);
}
size_t ttydisc_getc(struct lwp_tty *tp, void *buf, size_t len)
{
tty_assert_locked(tp);
if (tp->t_flags & TF_STOPPED)
return 0;
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, getc_inject))
return ttyhook_getc_inject(tp, buf, len);
#endif
len = ttyoutq_read(&tp->t_outq, buf, len);
#ifdef USING_BSD_HOOK
if (ttyhook_hashook(tp, getc_capture))
ttyhook_getc_capture(tp, buf, len);
#endif
ttydisc_wakeup_watermark(tp);
rt_atomic_add(&tty_nout, len);
return (len);
}
int ttydisc_getc_uio(struct lwp_tty *tp, struct uio *uio)
{
int error = 0;
ssize_t obytes = uio->uio_resid;
size_t len;
char buf[TTY_STACKBUF];
tty_assert_locked(tp);
if (tp->t_flags & TF_STOPPED)
return 0;
/*
* When a TTY hook is attached, we cannot perform unbuffered
* copying to userspace. Just call ttydisc_getc() and
* temporarily store data in a shadow buffer.
*/
if (ttyhook_hashook(tp, getc_capture) || ttyhook_hashook(tp, getc_inject))
{
while (uio->uio_resid > 0)
{
/* Read to shadow buffer. */
len = ttydisc_getc(tp, buf, MIN(uio->uio_resid, sizeof buf));
if (len == 0)
break;
/* Copy to userspace. */
tty_unlock(tp);
error = uiomove(buf, len, uio);
tty_lock(tp);
if (error != 0)
break;
}
}
else
{
error = ttyoutq_read_uio(&tp->t_outq, tp, uio);
ttydisc_wakeup_watermark(tp);
rt_atomic_add(&tty_nout, obytes - uio->uio_resid);
}
return error;
}
size_t ttydisc_getc_poll(struct lwp_tty *tp)
{
tty_assert_locked(tp);
if (tp->t_flags & TF_STOPPED)
return 0;
#if USING_BSD_HOOK
if (ttyhook_hashook(tp, getc_poll))
return ttyhook_getc_poll(tp);
#endif
return ttyoutq_bytesused(&tp->t_outq);
}