madtty.c

   1 /*
   2     LICENSE INFORMATION:
   3     This program is free software; you can redistribute it and/or
   4     modify it under the terms of the GNU Lesser General Public
   5     License (LGPL) as published by the Free Software Foundation.
   6
   7     Please refer to the COPYING file for more information.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12     General Public License for more details.
  13
  14     You should have received a copy of the GNU Lesser General Public
  15     License along with this program; if not, write to the Free Software
  16     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17
  18     Copyright © 2004 Bruno T. C. de Oliveira
  19     Copyright © 2006 Pierre Habouzit
  20  */
  21
  22 #define _GNU_SOURCE
  23 #include <assert.h>
  24 #include <ctype.h>
  25 #include <errno.h>
  26 #include <fcntl.h>
  27 #include <locale.h>
  28 #include <pty.h>
  29 #include <signal.h>
  30 #include <stdio.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <sys/types.h>
  34 #include <wchar.h>
  35
  36 #include "madtty.h"
  37
  38 #define IS_CONTROL(ch) !((ch) & 0xffffff60UL)
  39
  40 enum {
  41     C0_NUL = 0x00,
  42             C0_SOH, C0_STX, C0_ETX, C0_EOT, C0_ENQ, C0_ACK, C0_BEL,
  43     C0_BS , C0_HT , C0_LF , C0_VT , C0_FF , C0_CR , C0_SO , C0_SI ,
  44     C0_DLE, C0_DC1, C0_DC2, D0_DC3, C0_DC4, C0_NAK, C0_SYN, C0_ETB,
  45     C0_CAN, C0_EM , C0_SUB, C0_ESC, C0_IS4, C0_IS3, C0_IS2, C0_IS1,
  46 };
  47
  48 enum {
  49     C1_40 = 0x40,
  50             C1_41 , C1_BPH, C1_NBH, C1_44 , C1_NEL, C1_SSA, C1_ESA,
  51     C1_HTS, C1_HTJ, C1_VTS, C1_PLD, C1_PLU, C1_RI , C1_SS2, C1_SS3,
  52     C1_DCS, C1_PU1, C1_PU2, C1_STS, C1_CCH, C1_MW , C1_SPA, C1_EPA,
  53     C1_SOS, C1_59 , C1_SCI, C1_CSI, CS_ST , C1_OSC, C1_PM , C1_APC,
  54 };
  55
  56 enum {
  57     CSI_ICH = 0x40,
  58              CSI_CUU, CSI_CUD, CSI_CUF, CSI_CUB, CSI_CNL, CSI_CPL, CSI_CHA,
  59     CSI_CUP, CSI_CHT, CSI_ED , CSI_EL , CSI_IL , CSI_DL , CSI_EF , CSI_EA ,
  60     CSI_DCH, CSI_SEE, CSI_CPR, CSI_SU , CSI_SD , CSI_NP , CSI_PP , CSI_CTC,
  61     CSI_ECH, CSI_CVT, CSI_CBT, CSI_SRS, CSI_PTX, CSI_SDS, CSI_SIMD, CSI_5F,
  62     CSI_HPA, CSI_HPR, CSI_REP, CSI_DA , CSI_VPA, CSI_VPR, CSI_HVP, CSI_TBC,
  63     CSI_SM , CSI_MC , CSI_HPB, CSI_VPB, CSI_RM , CSI_SGR, CSI_DSR, CSI_DAQ,
  64     CSI_70 , CSI_71 , CSI_72 , CSI_73 , CSI_74 , CSI_75 , CSI_76 , CSI_77 ,
  65     CSI_78 , CSI_79 , CSI_7A , CSI_7B , CSI_7C , CSI_7D , CSI_7E , CSI_7F
  66 };
  67
  68 struct mtty_row_t {
  69     wchar_t  *text;
  70     uint16_t *attr;
  71     unsigned dirty : 1;
  72 };
  73
  74 static char const * const keytable[KEY_MAX+1] = {
  75     ['\n']          = "\r",
  76     [KEY_UP]        = "\e[A",
  77     [KEY_DOWN]      = "\e[B",
  78     [KEY_RIGHT]     = "\e[C",
  79     [KEY_LEFT]      = "\e[D",
  80     [KEY_BACKSPACE] = "\177",
  81     [KEY_HOME]      = "\e[1~",
  82     [KEY_IC]        = "\e[2~",
  83     [KEY_DC]        = "\e[3~",
  84     [KEY_END]       = "\e[4~",
  85     [KEY_PPAGE]     = "\e[5~",
  86     [KEY_NPAGE]     = "\e[6~",
  87     [KEY_SUSPEND]   = "\x1A",  /* Ctrl+Z gets mapped to this */
  88     [KEY_F(1)]      = "\e[[A",
  89     [KEY_F(2)]      = "\e[[B",
  90     [KEY_F(3)]      = "\e[[C",
  91     [KEY_F(4)]      = "\e[[D",
  92     [KEY_F(5)]      = "\e[[E",
  93     [KEY_F(6)]      = "\e[17~",
  94     [KEY_F(7)]      = "\e[18~",
  95     [KEY_F(8)]      = "\e[19~",
  96     [KEY_F(9)]      = "\e[20~",
  97     [KEY_F(10)]     = "\e[21~",
  98 };
  99
 100 static void mtty_row_set(mtty_row_t *row, int start, int len, uint16_t attr)
 101 {
 102     row->dirty = true;
 103     wmemset(row->text + start, 0, len);
 104     for (int i = start; i < len + start; i++) {
 105         row->attr[i] = attr;
 106     }
 107 }
 108
 109 static void mtty_row_roll(mtty_row_t *start, mtty_row_t *end, int count)
 110 {
 111     int n = end - start;
 112
 113     count %= n;
 114     if (count < 0)
 115         count += n;
 116
 117     if (count) {
 118         mtty_row_t *buf = alloca(count * sizeof(mtty_row_t));
 119
 120         memcpy(buf, start, count * sizeof(mtty_row_t));
 121         memmove(start, start + count, (n - count) * sizeof(mtty_row_t));
 122         memcpy(end - count, buf, count * sizeof(mtty_row_t));
 123         for (mtty_row_t *row = start; row < end; row++) {
 124             row->dirty = true;
 125         }
 126     }
 127 }
 128
 129 static void clamp_cursor_to_bounds(madtty_t *rt)
 130 {
 131     if (rt->curs_row < rt->lines) {
 132         rt->curs_row = rt->lines;
 133     }
 134     if (rt->curs_row >= rt->lines + rt->rows) {
 135         rt->curs_row = rt->lines + rt->rows - 1;
 136     }
 137
 138     if (rt->curs_col < 0) {
 139         rt->curs_col = 0;
 140     }
 141     if (rt->curs_col >= rt->cols) {
 142         rt->curs_col = rt->cols - 1;
 143     }
 144 }
 145
 146 static void cursor_line_down(madtty_t *rt)
 147 {
 148     rt->curs_row++;
 149     if (rt->curs_row < rt->scroll_bot)
 150         return;
 151
 152     rt->curs_row = rt->scroll_bot - 1;
 153     mtty_row_roll(rt->scroll_top, rt->scroll_bot, 1);
 154     mtty_row_set(rt->curs_row, 0, rt->cols, 0);
 155 }
 156
 157 __attribute__((const))
 158 static uint16_t build_attrs(unsigned curattrs)
 159 {
 160     return ((curattrs & ~A_COLOR) | COLOR_PAIR(curattrs & 0xff))
 161         >> NCURSES_ATTR_SHIFT;
 162 }
 163
 164 static void new_escape_sequence(madtty_t *rt)
 165 {
 166     rt->escaped = true;
 167     rt->elen    = 0;
 168     rt->ebuf[0] = '\0';
 169 }
 170
 171 static void cancel_escape_sequence(madtty_t *rt)
 172 {
 173     rt->escaped = false;
 174     rt->elen    = 0;
 175     rt->ebuf[0] = '\0';
 176 }
 177
 178 static bool is_valid_csi_ender(int c)
 179 {
 180     return (c >= 'a' && c <= 'z')
 181         || (c >= 'A' && c <= 'Z')
 182         || (c == '@' || c == '`');
 183 }
 184
 185 /* interprets a 'set attribute' (SGR) CSI escape sequence */
 186 static void interpret_csi_SGR(madtty_t *rt, int param[], int pcount)
 187 {
 188     int i;
 189
 190     if (pcount == 0) {
 191         /* special case: reset attributes */
 192         rt->curattrs = A_NORMAL;
 193         return;
 194     }
 195
 196     for (i = 0; i < pcount; i++) {
 197         switch (param[i]) {
 198 #define CASE(x, op)  case x: op; break
 199             CASE(0,  rt->curattrs = A_NORMAL);
 200             CASE(1,  rt->curattrs |= A_BOLD);
 201             CASE(4,  rt->curattrs |= A_UNDERLINE);
 202             CASE(5,  rt->curattrs |= A_BLINK);
 203             CASE(6,  rt->curattrs |= A_BLINK);
 204             CASE(7,  rt->curattrs |= A_REVERSE);
 205             CASE(8,  rt->curattrs |= A_INVIS);
 206             CASE(22, rt->curattrs &= ~A_BOLD);
 207             CASE(24, rt->curattrs &= ~A_UNDERLINE);
 208             CASE(25, rt->curattrs &= ~A_BLINK);
 209             CASE(27, rt->curattrs &= ~A_REVERSE);
 210             CASE(28, rt->curattrs &= ~A_INVIS);
 211
 212           case 30 ... 37:
 213             rt->curattrs &= ~0xf0;
 214             rt->curattrs |= (param[i] - 29) << 4;
 215             break;
 216
 217           case 39:
 218             rt->curattrs &= ~0xf0;
 219             break;
 220
 221           case 40 ... 47:
 222             rt->curattrs &= ~0x0f;
 223             rt->curattrs |= (param[i] - 39);
 224             break;
 225
 226           case 49:
 227             rt->curattrs &= ~0x0f;
 228             break;
 229
 230           default:
 231             break;
 232         }
 233     }
 234 }
 235
 236 /* interprets an 'erase display' (ED) escape sequence */
 237 static void interpret_csi_ED(madtty_t *rt, int param[], int pcount)
 238 {
 239     mtty_row_t *row, *start, *end;
 240     attr_t attr = build_attrs(rt->curattrs);
 241
 242     /* decide range */
 243     if (pcount && param[0] == 2) {
 244         start = rt->lines;
 245         end   = rt->lines + rt->rows;
 246     } else
 247     if (pcount && param[0] == 1) {
 248         start = rt->lines;
 249         end   = rt->curs_row;
 250         mtty_row_set(rt->curs_row, 0, rt->curs_col + 1, attr);
 251     } else {
 252         mtty_row_set(rt->curs_row, rt->curs_col,
 253                      rt->cols - rt->curs_col, attr);
 254         start = rt->curs_row + 1;
 255         end   = rt->lines + rt->rows;
 256     }
 257
 258     for (row = start; row < end; row++) {
 259         mtty_row_set(row, 0, rt->cols, attr);
 260     }
 261 }
 262
 263 /* interprets a 'move cursor' (CUP) escape sequence */
 264 static void interpret_csi_CUP(madtty_t *rt, int param[], int pcount)
 265 {
 266     if (pcount == 0) {
 267         /* special case */
 268         rt->curs_row = rt->lines;
 269         rt->curs_col = 0;
 270         return;
 271     } else
 272     if (pcount < 2) {
 273         return;  /* malformed */
 274     }
 275
 276     rt->curs_row = rt->lines + param[0] - 1;  /* convert from 1-based to 0-based */
 277     rt->curs_col = param[1] - 1;  /* convert from 1-based to 0-based */
 278
 279     clamp_cursor_to_bounds(rt);
 280 }
 281
 282 /* Interpret the 'relative mode' sequences: CUU, CUD, CUF, CUB, CNL,
 283  * CPL, CHA, HPR, VPA, VPR, HPA */
 284 static void interpret_csi_C(madtty_t *rt, char verb, int param[], int pcount)
 285 {
 286     int n = (pcount && param[0] > 0) ? param[0] : 1;
 287
 288     switch (verb) {
 289       case 'A':           rt->curs_row -= n; break;
 290       case 'B': case 'e': rt->curs_row += n; break;
 291       case 'C': case 'a': rt->curs_col += n; break;
 292       case 'D':           rt->curs_col -= n; break;
 293       case 'E':           rt->curs_row += n; rt->curs_col = 0; break;
 294       case 'F':           rt->curs_row -= n; rt->curs_col = 0; break;
 295       case 'G': case '`': rt->curs_col  = param[0] - 1; break;
 296       case 'd':           rt->curs_row  = rt->lines + param[0] - 1; break;
 297     }
 298
 299     clamp_cursor_to_bounds(rt);
 300 }
 301
 302 /* Interpret the 'erase line' escape sequence */
 303 static void interpret_csi_EL(madtty_t *rt, int param[], int pcount)
 304 {
 305     attr_t attr = build_attrs(rt->curattrs);
 306
 307     switch (pcount ? param[0] : 0) {
 308       case 1:
 309         mtty_row_set(rt->curs_row, 0, rt->curs_col + 1, attr);
 310         break;
 311       case 2:
 312         mtty_row_set(rt->curs_row, 0, rt->cols, attr);
 313         break;
 314       default:
 315         mtty_row_set(rt->curs_row, rt->curs_col, rt->cols - rt->curs_col,
 316                      attr);
 317         break;
 318     }
 319 }
 320
 321 /* Interpret the 'insert blanks' sequence (ICH) */
 322 static void interpret_csi_ICH(madtty_t *rt, int param[], int pcount)
 323 {
 324     mtty_row_t *row = rt->curs_row;
 325     int n = (pcount && param[0] > 0) ? param[0] : 1;
 326     int i;
 327
 328     if (rt->curs_col + n > rt->cols) {
 329         n = rt->cols - rt->curs_col;
 330     }
 331
 332     for (i = rt->cols - 1; i >= rt->curs_col + n; i--) {
 333         row->text[i] = row->text[i - n];
 334         row->attr[i] = row->attr[i - n];
 335     }
 336
 337     mtty_row_set(row, rt->curs_col, n, build_attrs(rt->curattrs));
 338 }
 339
 340 /* Interpret the 'delete chars' sequence (DCH) */
 341 static void interpret_csi_DCH(madtty_t *rt, int param[], int pcount)
 342 {
 343     mtty_row_t *row = rt->curs_row;
 344     int n = (pcount && param[0] > 0) ? param[0] : 1;
 345     int i;
 346
 347     if (rt->curs_col + n > rt->cols) {
 348         n = rt->cols - rt->curs_col;
 349     }
 350
 351     for (i = rt->curs_col; i < rt->cols - n; i++) {
 352         row->text[i] = row->text[i + n];
 353         row->attr[i] = row->attr[i + n];
 354     }
 355
 356     mtty_row_set(row, rt->cols - n, n, build_attrs(rt->curattrs));
 357 }
 358
 359 /* Interpret a 'scroll reverse' (SR) */
 360 static void interpret_csi_SR(madtty_t *rt)
 361 {
 362     mtty_row_roll(rt->scroll_top, rt->scroll_bot, -1);
 363     mtty_row_set(rt->scroll_top, 0, rt->cols, build_attrs(rt->curattrs));
 364 }
 365
 366 /* Interpret an 'insert line' sequence (IL) */
 367 static void interpret_csi_IL(madtty_t *rt, int param[], int pcount)
 368 {
 369     int n = (pcount && param[0] > 0) ? param[0] : 1;
 370
 371     if (rt->curs_row + n >= rt->scroll_bot) {
 372         for (mtty_row_t *row = rt->curs_row; row < rt->scroll_bot; row++) {
 373             mtty_row_set(row, 0, rt->cols, build_attrs(rt->curattrs));
 374         }
 375     } else {
 376         mtty_row_roll(rt->curs_row, rt->scroll_bot, -n);
 377         for (mtty_row_t *row = rt->curs_row; row < rt->curs_row + n; row++) {
 378             mtty_row_set(row, 0, rt->cols, build_attrs(rt->curattrs));
 379         }
 380     }
 381 }
 382
 383 /* Interpret a 'delete line' sequence (DL) */
 384 static void interpret_csi_DL(madtty_t *rt, int param[], int pcount)
 385 {
 386     int n = (pcount && param[0] > 0) ? param[0] : 1;
 387
 388     if (rt->curs_row + n >= rt->scroll_bot) {
 389         for (mtty_row_t *row = rt->curs_row; row < rt->scroll_bot; row++) {
 390             mtty_row_set(row, 0, rt->cols, build_attrs(rt->curattrs));
 391         }
 392     } else {
 393         mtty_row_roll(rt->curs_row, rt->scroll_bot, n);
 394         for (mtty_row_t *row = rt->scroll_bot - n; row < rt->scroll_bot; row++) {
 395             mtty_row_set(row, 0, rt->cols, build_attrs(rt->curattrs));
 396         }
 397     }
 398 }
 399
 400 /* Interpret an 'erase characters' (ECH) sequence */
 401 static void interpret_csi_ECH(madtty_t *rt, int param[], int pcount)
 402 {
 403     int n = (pcount && param[0] > 0) ? param[0] : 1;
 404
 405     if (rt->curs_col + n < rt->cols) {
 406         n = rt->cols - rt->curs_col;
 407     }
 408     mtty_row_set(rt->curs_row, rt->curs_col, n, build_attrs(rt->curattrs));
 409 }
 410
 411 /* Interpret a 'set scrolling region' (DECSTBM) sequence */
 412 static void interpret_csi_DECSTBM(madtty_t *rt, int param[], int pcount)
 413 {
 414     int new_top, new_bot;
 415
 416     switch (pcount) {
 417       case 0:
 418         rt->scroll_top = rt->lines;
 419         rt->scroll_bot = rt->lines + rt->rows;
 420         break;
 421       default:
 422         return; /* malformed */
 423
 424       case 2:
 425         new_top = param[0] - 1;
 426         new_bot = param[1];
 427
 428         /* clamp to bounds */
 429         if (new_top < 0)
 430             new_top = 0;
 431         if (new_top >= rt->rows)
 432             new_top = rt->rows - 1;
 433         if (new_bot < 0)
 434             new_bot = 0;
 435         if (new_bot >= rt->rows)
 436             new_bot = rt->rows;
 437
 438         /* check for range validity */
 439         if (new_top < new_bot) {
 440             rt->scroll_top = rt->lines + new_top;
 441             rt->scroll_bot = rt->lines + new_bot;
 442         }
 443         break;
 444     }
 445 }
 446
 447 static void es_interpret_csi(madtty_t *rt)
 448 {
 449     static int csiparam[MAX_CSI_ES_PARAMS];
 450     int param_count = 0;
 451     const char *p = rt->ebuf + 1;
 452     char verb = rt->ebuf[rt->elen - 1];
 453
 454     p += rt->ebuf[1] == '?'; /* CSI private mode */
 455
 456     /* parse numeric parameters */
 457     while (isdigit((unsigned char)*p) || *p == ';') {
 458         if (*p == ';') {
 459             if (param_count >= MAX_CSI_ES_PARAMS) return; /* too long! */
 460             csiparam[param_count++] = 0;
 461         } else {
 462             if (param_count == 0) csiparam[param_count++] = 0;
 463             csiparam[param_count - 1] *= 10;
 464             csiparam[param_count - 1] += *p - '0';
 465         }
 466
 467         p++;
 468     }
 469
 470     if (rt->ebuf[1] == '?') {
 471         switch (verb) {
 472           case 'l':
 473             if (csiparam[0] == 25)
 474                 rt->curshid = true;
 475             break;
 476
 477           case 'h':
 478             if (csiparam[0] == 25)
 479                 rt->curshid = false;
 480             break;
 481         }
 482     }
 483
 484     /* delegate handling depending on command character (verb) */
 485     switch (verb) {
 486       case 'h':
 487         if (param_count == 1 && csiparam[0] == 4) /* insert mode */
 488             rt->insert = true;
 489         break;
 490       case 'l':
 491         if (param_count == 1 && csiparam[0] == 4) /* replace mode */
 492             rt->insert = false;
 493         break;
 494       case 'm': /* it's a 'set attribute' sequence */
 495         interpret_csi_SGR(rt, csiparam, param_count); break;
 496       case 'J': /* it's an 'erase display' sequence */
 497         interpret_csi_ED(rt, csiparam, param_count); break;
 498       case 'H': case 'f': /* it's a 'move cursor' sequence */
 499         interpret_csi_CUP(rt, csiparam, param_count); break;
 500       case 'A': case 'B': case 'C': case 'D': case 'E': case 'F': case 'G':
 501       case 'e': case 'a': case 'd': case '`':
 502         /* it is a 'relative move' */
 503         interpret_csi_C(rt, verb, csiparam, param_count); break;
 504       case 'K': /* erase line */
 505         interpret_csi_EL(rt, csiparam, param_count); break;
 506       case '@': /* insert characters */
 507         interpret_csi_ICH(rt, csiparam, param_count); break;
 508       case 'P': /* delete characters */
 509         interpret_csi_DCH(rt, csiparam, param_count); break;
 510       case 'L': /* insert lines */
 511         interpret_csi_IL(rt, csiparam, param_count); break;
 512       case 'M': /* delete lines */
 513         interpret_csi_DL(rt, csiparam, param_count); break;
 514       case 'X': /* erase chars */
 515         interpret_csi_ECH(rt, csiparam, param_count); break;
 516       case 'r': /* set scrolling region */
 517         interpret_csi_DECSTBM(rt, csiparam, param_count); break;
 518       case 's': /* save cursor location */
 519         rt->curs_srow = rt->curs_row - rt->lines;
 520         rt->curs_scol = rt->curs_col;
 521         break;
 522       case 'u': /* restore cursor location */
 523         rt->curs_row = rt->lines + rt->curs_srow;
 524         rt->curs_col = rt->curs_scol;
 525         clamp_cursor_to_bounds(rt);
 526         break;
 527       default:
 528         break;
 529     }
 530 }
 531
 532 static void try_interpret_escape_seq(madtty_t *rt)
 533 {
 534     char lastchar  = rt->ebuf[rt->elen-1];
 535
 536     switch (*rt->ebuf) {
 537       case '\0':
 538         return;
 539
 540       case 'M':
 541         interpret_csi_SR(rt);
 542         cancel_escape_sequence(rt);
 543         return;
 544
 545       case '(':
 546       case ')':
 547         if (rt->elen == 2)
 548             goto cancel;
 549         break;
 550
 551       case ']': /* xterm thing */
 552         if (lastchar == '\a')
 553             goto cancel;
 554         break;
 555
 556       default:
 557         goto cancel;
 558
 559       case '[':
 560         if (is_valid_csi_ender(lastchar)) {
 561             es_interpret_csi(rt);
 562             cancel_escape_sequence(rt);
 563             return;
 564         }
 565         break;
 566     }
 567
 568     if (rt->elen + 1 >= (int)sizeof(rt->ebuf)) {
 569 cancel:
 570 #if 0
 571         int i;
 572         fprintf(stderr, "cancelled: \\033");
 573         for (i = 0; i < rt->elen; i++) {
 574             int c = rt->ebuf[i];
 575             if (isprint(c) && c >= ' ') {
 576                 fputc(c, stderr);
 577             } else {
 578                 fprintf(stderr, "\\%03o", c);
 579             }
 580         }
 581         fputc('\n', stderr);
 582 #endif
 583         cancel_escape_sequence(rt);
 584     }
 585 }
 586
 587 static void madtty_process_nonprinting(madtty_t *rt, wchar_t wc)
 588 {
 589     switch (wc) {
 590       case C0_ESC:
 591         new_escape_sequence(rt);
 592         break;
 593
 594       case C0_BEL:
 595         /* do nothing for now... maybe a visual bell would be nice? */
 596         break;
 597
 598       case C0_BS:
 599         if (rt->curs_col > 0)
 600             rt->curs_col--;
 601         break;
 602
 603       case C0_HT: /* tab */
 604         rt->curs_col = (rt->curs_col + 8) & ~7;
 605         if (rt->curs_col >= rt->cols)
 606             rt->curs_col = rt->cols - 1;
 607         break;
 608
 609       case C0_CR:
 610         rt->curs_col = 0;
 611         break;
 612
 613       case C0_VT:
 614       case C0_FF:
 615       case C0_LF:
 616         cursor_line_down(rt);
 617         break;
 618
 619       case C0_SO:               /* shift out - acs */
 620         rt->graphmode = true;
 621         break;
 622       case C0_SI:               /* shift in - acs */
 623         rt->graphmode = false;
 624         break;
 625     }
 626 }
 627
 628 void madtty_putc(madtty_t *rt, wchar_t wc)
 629 {
 630     int width = 0;
 631
 632     if (!rt->seen_input) {
 633         rt->seen_input = 1;
 634         kill(-rt->childpid, SIGWINCH);
 635     }
 636
 637 #if 0
 638     if (wc == '\n' || (wc >= ' ' && isprint(wc))) {
 639         fputc(wc, stderr);
 640     } else {
 641         fprintf(stderr, "\\%03o", wc);
 642     }
 643 #endif
 644
 645     if (rt->escaped) {
 646         assert (rt->elen + 1 < (int)sizeof(rt->ebuf));
 647         rt->ebuf[rt->elen]   = wc;
 648         rt->ebuf[++rt->elen] = '\0';
 649         try_interpret_escape_seq(rt);
 650     } else if (IS_CONTROL(wc)) {
 651         madtty_process_nonprinting(rt, wc);
 652     } else {
 653         mtty_row_t *tmp;
 654
 655         if (rt->graphmode) {
 656             // vt100 special graphics and line drawing
 657             // 5f-7e standard vt100
 658             // 40-5e rxvt extension for extra curses acs chars
 659             static uint16_t vt100_0[62] = { // 41 .. 7e
 660                         0x2191, 0x2193, 0x2192, 0x2190, 0x2588, 0x259a, 0x2603, // 41-47 hi mr. snowman!
 661                      0,      0,      0,      0,      0,      0,      0,      0, // 48-4f
 662                      0,      0,      0,      0,      0,      0,      0,      0, // 50-57
 663                      0,      0,      0,      0,      0,      0,      0, 0x0020, // 58-5f
 664                 0x25c6, 0x2592, 0x2409, 0x240c, 0x240d, 0x240a, 0x00b0, 0x00b1, // 60-67
 665                 0x2424, 0x240b, 0x2518, 0x2510, 0x250c, 0x2514, 0x253c, 0x23ba, // 68-6f
 666                 0x23bb, 0x2500, 0x23bc, 0x23bd, 0x251c, 0x2524, 0x2534, 0x252c, // 70-77
 667                 0x2502, 0x2264, 0x2265, 0x03c0, 0x2260, 0x00a3, 0x00b7,         // 78-7e
 668             };
 669
 670             if (wc >= 0x41 && wc <= 0x7e && vt100_0[wc - 0x41]) {
 671                 wc = vt100_0[wc - 0x41];
 672             }
 673             width = 1;
 674         } else {
 675             width = wcwidth(wc) ?: 1;
 676         }
 677
 678         if (width == 2 && rt->curs_col == rt->cols - 1) {
 679             tmp = rt->curs_row;
 680             tmp->dirty = true;
 681             tmp->text[rt->curs_col] = 0;
 682             tmp->attr[rt->curs_col] = build_attrs(rt->curattrs);
 683             rt->curs_col++;
 684         }
 685
 686         if (rt->curs_col >= rt->cols) {
 687             rt->curs_col = 0;
 688             cursor_line_down(rt);
 689         }
 690
 691         tmp = rt->curs_row;
 692         tmp->dirty = true;
 693
 694         if (rt->insert) {
 695             wmemmove(tmp->text + rt->curs_col + width, tmp->text + rt->curs_col,
 696                      (rt->cols - rt->curs_col - width));
 697             memmove(tmp->attr + rt->curs_col + width, tmp->attr + rt->curs_col,
 698                     (rt->cols - rt->curs_col - width) * sizeof(tmp->attr[0]));
 699         }
 700
 701         tmp->text[rt->curs_col] = wc;
 702         tmp->attr[rt->curs_col] = build_attrs(rt->curattrs);
 703         rt->curs_col++;
 704         if (width == 2) {
 705             tmp->text[rt->curs_col] = 0;
 706             tmp->attr[rt->curs_col] = build_attrs(rt->curattrs);
 707             rt->curs_col++;
 708         }
 709     }
 710 }
 711
 712 int madtty_process(madtty_t *rt)
 713 {
 714     int res, pos = 0;
 715
 716     if (rt->pty < 0) {
 717         errno = EINVAL;
 718         return -1;
 719     }
 720
 721     res = read(rt->pty, rt->rbuf + rt->rlen, sizeof(rt->rbuf) - rt->rlen);
 722     if (res < 0)
 723         return -1;
 724
 725     rt->rlen += res;
 726     while (pos < rt->rlen) {
 727         wchar_t wc;
 728         ssize_t len;
 729
 730         len = (ssize_t)mbrtowc(&wc, rt->rbuf + pos, rt->rlen - pos, &rt->ps);
 731         if (len == -2) {
 732             rt->rlen -= pos;
 733             memmove(rt->rbuf, rt->rbuf + pos, rt->rlen);
 734             return 0;
 735         }
 736
 737         if (len == -1) {
 738             len = 1;
 739             wc  = rt->rbuf[pos];
 740         }
 741
 742         pos += len ? len : 1;
 743         madtty_putc(rt, wc);
 744     }
 745
 746     rt->rlen -= pos;
 747     memmove(rt->rbuf, rt->rbuf + pos, rt->rlen);
 748     return 0;
 749 }
 750
 751 madtty_t *madtty_create(int rows, int cols)
 752 {
 753     madtty_t *rt;
 754     int i;
 755
 756     if (rows <= 0 || cols <= 0)
 757         return NULL;
 758
 759     rt = (madtty_t*)calloc(sizeof(madtty_t), 1);
 760     if (!rt)
 761         return NULL;
 762
 763     /* record dimensions */
 764     rt->rows = rows;
 765     rt->cols = cols;
 766
 767     /* default mode is replace */
 768     rt->insert = false;
 769
 770     /* create the cell matrix */
 771     rt->lines = (mtty_row_t*)calloc(sizeof(mtty_row_t), rt->rows);
 772     for (i = 0; i < rt->rows; i++) {
 773         rt->lines[i].text = (wchar_t *)calloc(sizeof(wchar_t), rt->cols);
 774         rt->lines[i].attr = (uint16_t *)calloc(sizeof(uint16_t), rt->cols);
 775     }
 776
 777     rt->pty = -1;  /* no pty for now */
 778
 779     /* initialization of other public fields */
 780     rt->curs_row = rt->lines;
 781     rt->curs_col = 0;
 782     rt->curattrs = A_NORMAL;  /* white text over black background */
 783
 784     /* initial scrolling area is the whole window */
 785     rt->scroll_top = rt->lines;
 786     rt->scroll_bot = rt->lines + rt->rows;
 787
 788     return rt;
 789 }
 790
 791 void madtty_destroy(madtty_t *rt)
 792 {
 793     int i;
 794     if (!rt)
 795         return;
 796
 797     for (i = 0; i < rt->rows; i++) {
 798         free(rt->lines[i].text);
 799         free(rt->lines[i].attr);
 800     }
 801     free(rt->lines);
 802     free(rt);
 803 }
 804
 805 void madtty_draw(madtty_t *rt, WINDOW *win, int srow, int scol)
 806 {
 807     curs_set(0);
 808     for (int i = 0; i < rt->rows; i++) {
 809         mtty_row_t *row = rt->lines + i;
 810
 811
 812         if (!row->dirty)
 813             continue;
 814
 815         wmove(win, srow + i, scol);
 816         for (int j = 0; j < rt->cols; j++) {
 817             if (!j || row->attr[j] != row->attr[j - 1])
 818                 wattrset(win, (attr_t)row->attr[j] << NCURSES_ATTR_SHIFT);
 819             if (row->text[j] >= 128) {
 820                 char buf[MB_CUR_MAX + 1];
 821                 int len;
 822
 823                 len = wcrtomb(buf, row->text[j], NULL);
 824                 waddnstr(win, buf, len);
 825                 if (wcwidth(row->text[j]) > 1)
 826                     j++;
 827             } else {
 828                 waddch(win, row->text[j] > ' ' ? row->text[j] : ' ');
 829             }
 830         }
 831         row->dirty = false;
 832     }
 833
 834     wmove(win, srow + rt->curs_row - rt->lines, scol + rt->curs_col);
 835     curs_set(!rt->curshid);
 836 }
 837
 838 /******************************************************/
 839
 840 pid_t madtty_forkpty(madtty_t *rt, const char *path, const char *argv[])
 841 {
 842     struct winsize ws;
 843     pid_t pid;
 844
 845     ws.ws_row    = rt->rows;
 846     ws.ws_col    = rt->cols;
 847     ws.ws_xpixel = ws.ws_ypixel = 0;
 848
 849     pid = forkpty(&rt->pty, NULL, NULL, &ws);
 850     if (pid < 0)
 851         return -1;
 852
 853     if (pid == 0) {
 854         setsid();
 855         setenv("TERM", "rxvt", 1);
 856         execv(path, (char *const*)argv);
 857         fprintf(stderr, "\nexecv() failed.\nCommand: '%s'\n", argv[0]);
 858         exit(1);
 859     }
 860
 861     return rt->childpid = pid;
 862 }
 863
 864 void madtty_keypress(madtty_t *rt, int keycode)
 865 {
 866     char c = (char)keycode;
 867     const char *buf;
 868     int len;
 869
 870 #if 0
 871     if (keycode == KEY_F(1)) {
 872 #define MIN(a, b) ((a < (b)) ? a : (b))
 873         kill(-rt->childpid, SIGWINCH);
 874         rt->scroll_bot = MIN(rt->scroll_bot, rt->lines + rt->rows);
 875         rt->curs_row = MIN(rt->curs_row, rt->lines + rt->rows);
 876         printf(stderr, "%d\n", rt->rows);
 877         return;
 878     }
 879 #endif
 880     if (keycode >= 0 && keycode < KEY_MAX && keytable[keycode]) {
 881         buf = keytable[keycode];
 882         len = strlen(keytable[keycode]);
 883     } else {
 884         buf = &c;
 885         len = 1;
 886     }
 887
 888     while (len > 0) {
 889         int res = write(rt->pty, buf, len);
 890         if (res < 0 && errno != EAGAIN && errno != EINTR)
 891             return;
 892
 893         buf += res;
 894         len -= res;
 895     }
 896 }
 897
 898 void madtty_initialize(void)
 899 {
 900     setlocale(LC_ALL, "");
 901     initscr();
 902     start_color();
 903     use_default_colors();
 904     noecho();
 905     raw();
 906     nodelay(stdscr, TRUE);
 907     keypad(stdscr, TRUE);
 908     ESCDELAY=50;
 909
 910     for (int i = -1; i < 8; i++) {
 911         for (int j = -1; j < 8; j++) {
 912             init_pair((i + 1) * 16 + j + 1, i, j);
 913         }
 914     }
 915 }