st.c

   1 /* See LICENSE for licence details. */
   2 #define _XOPEN_SOURCE
   3 #include <ctype.h>
   4 #include <errno.h>
   5 #include <fcntl.h>
   6 #include <locale.h>
   7 #include <stdarg.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <signal.h>
  12 #include <sys/ioctl.h>
  13 #include <sys/select.h>
  14 #include <sys/stat.h>
  15 #include <sys/types.h>
  16 #include <sys/wait.h>
  17 #include <unistd.h>
  18 #include <X11/Xlib.h>
  19 #include <X11/keysym.h>
  20 #include <X11/Xutil.h>
  21
  22 #define TNAME "st"
  23
  24 /* Arbitrary sizes */
  25 #define ESCSIZ 256
  26 #define ESCARG 16
  27
  28 #define SERRNO strerror(errno)
  29 #define MIN(a, b)  ((a) < (b) ? (a) : (b))
  30 #define MAX(a, b)  ((a) < (b) ? (b) : (a))
  31 #define LEN(a)     (sizeof(a) / sizeof(a[0]))
  32 #define DEFAULT(a, b)     (a) = (a) ? (a) : (b)
  33 #define BETWEEN(x, a, b)  ((a) <= (x) && (x) <= (b))
  34 #define LIMIT(x, a, b)    (x) = (x) < (a) ? (a) : (x) > (b) ? (b) : (x)
  35
  36 /* Attribute, Cursor, Character state, Terminal mode, Screen draw mode */
  37 enum { ATnone=0 , ATreverse=1 , ATunderline=2, ATbold=4 };
  38 enum { CSup, CSdown, CSright, CSleft, CShide, CSdraw, CSwrap, CSsave, CSload };
  39 enum { CRset=1, CRupdate=2 };
  40 enum { TMwrap=1, TMinsert=2, TMaltcharset };
  41 enum { SCupdate, SCredraw };
  42
  43 typedef int Color;
  44
  45 typedef struct {
  46         KeySym k;
  47         char s[ESCSIZ];
  48 } Key;
  49
  50 typedef struct {
  51         char c;     /* character code  */
  52         char mode;  /* attribute flags */
  53         Color fg;   /* foreground      */
  54         Color bg;   /* background      */
  55         char state; /* state flag      */
  56 } Glyph;
  57
  58 typedef Glyph* Line;
  59
  60 typedef struct {
  61         Glyph attr;  /* current char attributes */
  62         char hidden;
  63         int x;
  64         int y;
  65 } TCursor;
  66
  67 /* Escape sequence structs */
  68 /* ESC <pre> [[ [<priv>] <arg> [;]] <mode>] */
  69 typedef struct {
  70         char buf[ESCSIZ+1]; /* raw string */
  71         int len;            /* raw string length */
  72         char pre;
  73         char priv;
  74         int arg[ESCARG+1];
  75         int narg;           /* nb of args */
  76         char mode;
  77 } Escseq;
  78
  79 /* Internal representation of the screen */
  80 typedef struct {
  81         int row;    /* nb row */
  82         int col;    /* nb col */
  83         Line* line; /* screen */
  84         TCursor c;  /* cursor */
  85         int top;    /* top    scroll limit */
  86         int bot;    /* bottom scroll limit */
  87         int mode;   /* terminal mode */
  88 } Term;
  89
  90 /* Purely graphic info */
  91 typedef struct {
  92         Display* dis;
  93         Window win;
  94         int scr;
  95         int w;  /* window width  */
  96         int h;  /* window height */
  97         int ch; /* char height */
  98         int cw; /* char width  */
  99 } XWindow;
 100
 101 #include "config.h"
 102
 103 /* Drawing Context */
 104 typedef struct {
 105         unsigned long col[LEN(colorname)];
 106         XFontStruct* font;
 107         GC gc;
 108 } DC;
 109
 110 void die(const char *errstr, ...);
 111 void draw(int);
 112 void execsh(void);
 113 void sigchld(int);
 114 char* kmap(KeySym);
 115 void kpress(XKeyEvent *);
 116 void resize(XEvent *);
 117 void run(void);
 118
 119 int escaddc(char);
 120 int escfinal(char);
 121 void escdump(void);
 122 void eschandle(void);
 123 void escparse(void);
 124 void escreset(void);
 125
 126 void tclearregion(int, int, int, int);
 127 void tcpos(int);
 128 void tcursor(int);
 129 void tdeletechar(int);
 130 void tdeleteline(int);
 131 void tdump(void);
 132 void tinsertblank(int);
 133 void tinsertblankline(int);
 134 void tmoveto(int, int);
 135 void tnew(int, int);
 136 void tnewline(void);
 137 void tputc(char);
 138 void tputs(char*, int);
 139 void tresize(int, int);
 140 void tscroll(void);
 141 void tsetattr(int*, int);
 142 void tsetchar(char);
 143 void tsetscroll(int, int);
 144
 145 void ttynew(void);
 146 void ttyread(void);
 147 void ttyresize(int, int);
 148 void ttywrite(char *, size_t);
 149
 150 unsigned long xgetcol(const char *);
 151 void xclear(int, int, int, int);
 152 void xcursor(int);
 153 void xdrawc(int, int, Glyph);
 154 void xinit(void);
 155 void xscroll(void);
 156
 157
 158 /* Globals */
 159 DC dc;
 160 XWindow xw;
 161 Term term;
 162 Escseq escseq;
 163 int cmdfd;
 164 pid_t pid;
 165 int running;
 166
 167 void
 168 die(const char *errstr, ...) {
 169         va_list ap;
 170
 171         va_start(ap, errstr);
 172         vfprintf(stderr, errstr, ap);
 173         va_end(ap);
 174         exit(EXIT_FAILURE);
 175 }
 176
 177 void
 178 execsh(void) {
 179         char *args[3] = {SHELL, "-i", NULL};
 180         putenv("TERM=" TNAME);
 181         execvp(SHELL, args);
 182 }
 183
 184 void
 185 xbell(void) { /* visual bell */
 186         XRectangle r = { 0, 0, xw.w, xw.h };
 187         XSetForeground(xw.dis, dc.gc, dc.col[BellCol]);
 188         XFillRectangles(xw.dis, xw.win, dc.gc, &r, 1);
 189         /* usleep(30000); */
 190         draw(SCredraw);
 191 }
 192
 193 void
 194 sigchld(int a) {
 195         int stat = 0;
 196         if(waitpid(pid, &stat, 0) < 0)
 197                 die("Waiting for pid %hd failed: %s\n", pid, SERRNO);
 198         if(WIFEXITED(stat))
 199                 exit(WEXITSTATUS(stat));
 200         else
 201                 exit(EXIT_FAILURE);
 202 }
 203
 204
 205 void
 206 ttynew(void) {
 207         int m, s;
 208         char *pts;
 209
 210         if((m = posix_openpt(O_RDWR | O_NOCTTY)) < 0)
 211                 die("openpt failed: %s\n", SERRNO);
 212         if(grantpt(m) < 0)
 213                 die("grandpt failed: %s\n", SERRNO);
 214         if(unlockpt(m) < 0)
 215                 die("unlockpt failed: %s\n", SERRNO);
 216         if(!(pts = ptsname(m)))
 217                 die("ptsname failed: %s\n", SERRNO);
 218         if((s = open(pts, O_RDWR | O_NOCTTY)) < 0)
 219                 die("Couldn't open slave: %s\n", SERRNO);
 220         fcntl(s, F_SETFL, O_NDELAY);
 221         switch(pid = fork()) {
 222         case -1:
 223                 die("fork failed\n");
 224                 break;
 225         case 0:
 226                 setsid(); /* create a new process group */
 227                 dup2(s, STDIN_FILENO);
 228                 dup2(s, STDOUT_FILENO);
 229                 dup2(s, STDERR_FILENO);
 230                 if(ioctl(s, TIOCSCTTY, NULL) < 0)
 231                         die("ioctl TTIOCSTTY failed: %s\n", SERRNO);
 232                 execsh();
 233                 break;
 234         default:
 235                 close(s);
 236                 cmdfd = m;
 237                 signal(SIGCHLD, sigchld);
 238         }
 239 }
 240
 241 void
 242 dump(char c) {
 243         static int col;
 244         fprintf(stderr, " %02x %c ", c, isprint(c)?c:'.');
 245         if(++col % 10 == 0)
 246                 fprintf(stderr, "\n");
 247 }
 248
 249 void
 250 ttyread(void) {
 251         char buf[BUFSIZ] = {0};
 252         int ret;
 253
 254         switch(ret = read(cmdfd, buf, BUFSIZ)) {
 255         case -1:
 256                 die("Couldn't read from shell: %s\n", SERRNO);
 257                 break;
 258         default:
 259                 tputs(buf, ret);
 260         }
 261 }
 262
 263 void
 264 ttywrite(char *s, size_t n) {
 265         if(write(cmdfd, s, n) == -1)
 266                 die("write error on tty: %s\n", SERRNO);
 267 }
 268
 269 void
 270 ttyresize(int x, int y) {
 271         struct winsize w;
 272
 273         w.ws_row = term.row;
 274         w.ws_col = term.col;
 275         w.ws_xpixel = w.ws_ypixel = 0;
 276         if(ioctl(cmdfd, TIOCSWINSZ, &w) < 0)
 277                 fprintf(stderr, "Couldn't set window size: %s\n", SERRNO);
 278 }
 279
 280 int
 281 escfinal(char c) {
 282         if(escseq.len == 1)
 283                 switch(c) {
 284                 case '[':
 285                 case ']':
 286                 case '(':
 287                         return 0;
 288                 case '=':
 289                 case '>':
 290                 default:
 291                         return 1;
 292                 }
 293         else if(BETWEEN(c, 0x40, 0x7E))
 294                 return 1;
 295         return 0;
 296 }
 297
 298 void
 299 tcpos(int mode) {
 300         static int x = 0;
 301         static int y = 0;
 302
 303         if(mode == CSsave)
 304                 x = term.c.x, y = term.c.y;
 305         else if(mode == CSload)
 306                 tmoveto(x, y);
 307 }
 308
 309 void
 310 tnew(int col, int row) {   /* screen size */
 311         term.row = row, term.col = col;
 312         term.top = 0, term.bot = term.row - 1;
 313         /* mode */
 314         term.mode = TMwrap;
 315         /* cursor */
 316         term.c.attr.mode = ATnone;
 317         term.c.attr.fg = DefaultFG;
 318         term.c.attr.bg = DefaultBG;
 319         term.c.x = term.c.y = 0;
 320         term.c.hidden = 0;
 321         /* allocate screen */
 322         term.line = calloc(term.row, sizeof(Line));
 323         for(row = 0 ; row < term.row; row++)
 324                 term.line[row] = calloc(term.col, sizeof(Glyph));
 325 }
 326
 327 void
 328 tscroll(void) {
 329         Line temp = term.line[term.top];
 330         int i;
 331
 332         for(i = term.top; i < term.bot; i++)
 333                 term.line[i] = term.line[i+1];
 334         memset(temp, 0, sizeof(Glyph) * term.col);
 335         term.line[term.bot] = temp;
 336         xscroll();
 337 }
 338
 339 void
 340 tnewline(void) {
 341         int y = term.c.y + 1;
 342
 343         if(y > term.bot) {
 344                 tscroll(), y = term.bot;
 345         }
 346         tmoveto(0, y);
 347 }
 348
 349 int
 350 escaddc(char c) {
 351         escseq.buf[escseq.len++] = c;
 352         if(escfinal(c) || escseq.len >= ESCSIZ) {
 353                 escparse(), eschandle();
 354                 return 0;
 355         }
 356         return 1;
 357 }
 358
 359 void
 360 escparse(void) {
 361         /* int noarg = 1; */
 362         char *p = escseq.buf;
 363
 364         escseq.narg = 0;
 365         switch(escseq.pre = *p++) {
 366         case '[': /* CSI */
 367                 if(*p == '?')
 368                         escseq.priv = 1, p++;
 369
 370                 while(p < escseq.buf+escseq.len) {
 371                         while(isdigit(*p)) {
 372                                 escseq.arg[escseq.narg] *= 10;
 373                                 escseq.arg[escseq.narg] += *(p++) - '0'/*, noarg = 0 */;
 374                         }
 375                         if(*p == ';')
 376                                 escseq.narg++, p++;
 377                         else {
 378                                 escseq.mode = *p;
 379                                 escseq.narg++;
 380                                 return;
 381                         }
 382                 }
 383                 break;
 384         case '(':
 385                 /* XXX: graphic character set */
 386                 break;
 387         }
 388 }
 389
 390 void
 391 tmoveto(int x, int y) {
 392         term.c.x = x < 0 ? 0 : x >= term.col ? term.col-1 : x;
 393         term.c.y = y < 0 ? 0 : y >= term.row ? term.row-1 : y;
 394 }
 395
 396 void
 397 tcursor(int dir) {
 398         int xf = term.c.x, yf = term.c.y;
 399
 400         switch(dir) {
 401         case CSup:
 402                 yf--;
 403                 break;
 404         case CSdown:
 405                 yf++;
 406                 break;
 407         case CSleft:
 408                 xf--;
 409                 if(xf < 0) {
 410                         xf = term.col-1, yf--;
 411                         if(yf < term.top)
 412                                 yf = term.top, xf = 0;
 413                 }
 414                 break;
 415         case CSright:
 416                 xf++;
 417                 if(xf >= term.col) {
 418                         xf = 0, yf++;
 419                         if(yf > term.bot)
 420                                 yf = term.bot, tscroll();
 421                 }
 422                 break;
 423         }
 424         tmoveto(xf, yf);
 425 }
 426
 427 void
 428 tsetchar(char c) {
 429         term.line[term.c.y][term.c.x] = term.c.attr;
 430         term.line[term.c.y][term.c.x].c = c;
 431         term.line[term.c.y][term.c.x].state |= CRset | CRupdate;
 432 }
 433
 434 void
 435 tclearregion(int x1, int y1, int x2, int y2) {
 436         int x, y;
 437
 438         LIMIT(x1, 0, term.col-1);
 439         LIMIT(x2, 0, term.col-1);
 440         LIMIT(y1, 0, term.row-1);
 441         LIMIT(y2, 0, term.row-1);
 442
 443         /* XXX: could be optimized */
 444         for(x = x1; x <= x2; x++)
 445                 for(y = y1; y <= y2; y++)
 446                         memset(&term.line[y][x], 0, sizeof(Glyph));
 447
 448         xclear(x1, y1, x2, y2);
 449 }
 450
 451 void
 452 tdeletechar(int n) {
 453         int src = term.c.x + n;
 454         int dst = term.c.x;
 455         int size = term.col - src;
 456
 457         if(src >= term.col) {
 458                 tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
 459                 return;
 460         }
 461         memmove(&term.line[term.c.y][dst], &term.line[term.c.y][src], size * sizeof(Glyph));
 462         tclearregion(term.col-size, term.c.y, term.col-1, term.c.y);
 463 }
 464
 465 void
 466 tinsertblank(int n) {
 467         int src = term.c.x;
 468         int dst = src + n;
 469         int size = term.col - n - src;
 470
 471         if(dst >= term.col) {
 472                 tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
 473                 return;
 474         }
 475         memmove(&term.line[term.c.y][dst], &term.line[term.c.y][src], size * sizeof(Glyph));
 476         tclearregion(src, term.c.y, dst, term.c.y);
 477 }
 478
 479 void
 480 tinsertblankline (int n) {
 481         int i;
 482         Line blank;
 483         int bot = term.bot;
 484
 485         if(term.c.y > term.bot)
 486                 bot = term.row - 1;
 487         else if(term.c.y < term.top)
 488                 bot = term.top - 1;
 489         if(term.c.y + n >= bot) {
 490                 tclearregion(0, term.c.y, term.col-1, bot);
 491                 return;
 492         }
 493         for(i = bot; i >= term.c.y+n; i--) {
 494                 /* swap deleted line <-> blanked line */
 495                 blank = term.line[i];
 496                 term.line[i] = term.line[i-n];
 497                 term.line[i-n] = blank;
 498                 /* blank it */
 499                 memset(blank, 0, term.col * sizeof(Glyph));
 500         }
 501 }
 502
 503
 504 void
 505 tdeleteline(int n) {
 506         int i;
 507         Line blank;
 508         int bot = term.bot;
 509
 510         if(term.c.y > term.bot)
 511                 bot = term.row - 1;
 512         else if(term.c.y < term.top)
 513                 bot = term.top - 1;
 514         if(term.c.y + n >= bot) {
 515                 tclearregion(0, term.c.y, term.col-1, bot);
 516                 return;
 517         }
 518         for(i = term.c.y; i <= bot-n; i++) {
 519                 /* swap deleted line <-> blanked line */
 520                 blank = term.line[i];
 521                 term.line[i] = term.line[i+n];
 522                 term.line[i+n] = blank;
 523                 /* blank it */
 524                 memset(blank, 0, term.col * sizeof(Glyph));
 525         }
 526 }
 527
 528 void
 529 tsetattr(int *attr, int l) {
 530         int i;
 531
 532                 for(i = 0; i < l; i++) {
 533                         switch(attr[i]) {
 534                         case 0:
 535                                 memset(&term.c.attr, 0, sizeof(term.c.attr));
 536                                 term.c.attr.fg = DefaultFG;
 537                                 term.c.attr.bg = DefaultBG;
 538                                 break;
 539                         case 1:
 540                                 term.c.attr.mode |= ATbold;
 541                                 break;
 542                         case 4:
 543                                 term.c.attr.mode |= ATunderline;
 544                                 break;
 545                         case 7:
 546                                 term.c.attr.mode |= ATreverse;
 547                                 break;
 548                         case 8:
 549                                 term.c.hidden = CShide;
 550                                 break;
 551                         case 22:
 552                                 term.c.attr.mode &= ~ATbold;
 553                                 break;
 554                         case 24:
 555                                 term.c.attr.mode &= ~ATunderline;
 556                                 break;
 557                         case 27:
 558                                 term.c.attr.mode &= ~ATreverse;
 559                                 break;
 560                         case 39:
 561                                 term.c.attr.fg = DefaultFG;
 562                                 break;
 563                         case 49:
 564                                 term.c.attr.fg = DefaultBG;
 565                                 break;
 566                         default:
 567                                 if(BETWEEN(attr[i], 30, 37))
 568                                         term.c.attr.fg = attr[i] - 30;
 569                                 else if(BETWEEN(attr[i], 40, 47))
 570                                         term.c.attr.bg = attr[i] - 40;
 571                                 break;
 572                         }
 573                 }
 574 }
 575
 576 void
 577 tsetscroll(int t, int b) {
 578         int temp;
 579
 580         LIMIT(t, 0, term.row-1);
 581         LIMIT(b, 0, term.row-1);
 582         if(t > b) {
 583                 temp = t;
 584                 t = b;
 585                 b = temp;
 586         }
 587         term.top = t;
 588         term.bot = b;
 589 }
 590
 591
 592 void
 593 eschandle(void) {
 594         switch(escseq.pre) {
 595         default:
 596                 goto unknown_seq;
 597         case '[':
 598                 switch(escseq.mode) {
 599                 default:
 600                 unknown_seq:
 601                         fprintf(stderr, "erresc: unknown sequence\n");
 602                         escdump();
 603                         break;
 604                 case '@': /* Insert <n> blank char */
 605                         DEFAULT(escseq.arg[0], 1);
 606                         tinsertblank(escseq.arg[0]);
 607                         break;
 608                 case 'A': /* Cursor <n> Up */
 609                 case 'e':
 610                         DEFAULT(escseq.arg[0], 1);
 611                         tmoveto(term.c.x, term.c.y-escseq.arg[0]);
 612                         break;
 613                 case 'B': /* Cursor <n> Down */
 614                         DEFAULT(escseq.arg[0], 1);
 615                         tmoveto(term.c.x, term.c.y+escseq.arg[0]);
 616                         break;
 617                 case 'C': /* Cursor <n> Forward */
 618                 case 'a':
 619                         DEFAULT(escseq.arg[0], 1);
 620                         tmoveto(term.c.x+escseq.arg[0], term.c.y);
 621                         break;
 622                 case 'D': /* Cursor <n> Backward */
 623                         DEFAULT(escseq.arg[0], 1);
 624                         tmoveto(term.c.x-escseq.arg[0], term.c.y);
 625                         break;
 626                 case 'E': /* Cursor <n> Down and first col */
 627                         DEFAULT(escseq.arg[0], 1);
 628                         tmoveto(0, term.c.y+escseq.arg[0]);
 629                         break;
 630                 case 'F': /* Cursor <n> Up and first col */
 631                         DEFAULT(escseq.arg[0], 1);
 632                         tmoveto(0, term.c.y-escseq.arg[0]);
 633                         break;
 634                 case 'G': /* Move to <col> */
 635                 case '`':
 636                         DEFAULT(escseq.arg[0], 1);
 637                         tmoveto(escseq.arg[0]-1, term.c.y);
 638                         break;
 639                 case 'H': /* Move to <row> <col> */
 640                 case 'f':
 641                         DEFAULT(escseq.arg[0], 1);
 642                         DEFAULT(escseq.arg[1], 1);
 643                         tmoveto(escseq.arg[1]-1, escseq.arg[0]-1);
 644                         break;
 645                 case 'J': /* Clear screen */
 646                         switch(escseq.arg[0]) {
 647                         case 0: /* below */
 648                                 tclearregion(term.c.x, term.c.y, term.col-1, term.row-1);
 649                                 break;
 650                         case 1: /* above */
 651                                 tclearregion(0, 0, term.c.x, term.c.y);
 652                                 break;
 653                         case 2: /* all */
 654                                 tclearregion(0, 0, term.col-1, term.row-1);
 655                                 break;
 656                         }
 657                         break;
 658                 case 'K': /* Clear line */
 659                         switch(escseq.arg[0]) {
 660                         case 0: /* right */
 661                                 tclearregion(term.c.x, term.c.y, term.col-1, term.c.y);
 662                                 break;
 663                         case 1: /* left */
 664                                 tclearregion(0, term.c.y, term.c.x, term.c.y);
 665                                 break;
 666                         case 2: /* all */
 667                                 tclearregion(0, term.c.y, term.col-1, term.c.y);
 668                                 break;
 669                         }
 670                         break;
 671                 case 'L': /* Insert <n> blank lines */
 672                         DEFAULT(escseq.arg[0], 1);
 673                         tinsertblankline(escseq.arg[0]);
 674                         break;
 675                 case 'l':
 676                         if(escseq.priv && escseq.arg[0] == 25)
 677                                 term.c.hidden = 1;
 678                         break;
 679                 case 'M': /* Delete <n> lines */
 680                         DEFAULT(escseq.arg[0], 1);
 681                         tdeleteline(escseq.arg[0]);
 682                         break;
 683                 case 'P': /* Delete <n> char */
 684                         DEFAULT(escseq.arg[0], 1);
 685                         tdeletechar(escseq.arg[0]);
 686                         break;
 687                 case 'd': /* Move to <row> */
 688                         DEFAULT(escseq.arg[0], 1);
 689                         tmoveto(term.c.x, escseq.arg[0]-1);
 690                         break;
 691                 case 'h': /* Set terminal mode */
 692                         if(escseq.priv && escseq.arg[0] == 25)
 693                                 term.c.hidden = 0;
 694                         break;
 695                 case 'm': /* Terminal attribute (color) */
 696                         tsetattr(escseq.arg, escseq.narg);
 697                         break;
 698                 case 'r':
 699                         if(escseq.priv)
 700                                 ;
 701                         else {
 702                                 DEFAULT(escseq.arg[0], 1);
 703                                 DEFAULT(escseq.arg[1], term.row);
 704                                 tsetscroll(escseq.arg[0]-1, escseq.arg[1]-1);
 705                         }
 706                         break;
 707                 case 's': /* Save cursor position */
 708                         tcpos(CSsave);
 709                         break;
 710                 case 'u': /* Load cursor position */
 711                         tcpos(CSload);
 712                         break;
 713                 }
 714                 break;
 715         }
 716 }
 717
 718 void
 719 escdump(void) {
 720         int i;
 721         printf("rawbuf  : %s\n", escseq.buf);
 722         printf("prechar : %c\n", escseq.pre);
 723         printf("private : %c\n", escseq.priv ? '?' : ' ');
 724         printf("narg    : %d\n", escseq.narg);
 725         if(escseq.narg)
 726                 for(i = 0; i < escseq.narg; i++)
 727                         printf("\targ %d = %d\n", i, escseq.arg[i]);
 728         printf("mode    : %c\n", escseq.mode);
 729 }
 730
 731 void
 732 escreset(void) {
 733         memset(&escseq, 0, sizeof(escseq));
 734 }
 735
 736 void
 737 tputtab(void) {
 738         int space = TAB - term.c.x % TAB;
 739
 740         if(term.c.x + space >= term.col)
 741                 space--;
 742
 743         for(; space > 0; space--)
 744                 tcursor(CSright);
 745 }
 746
 747 void
 748 tputc(char c) {
 749         static int inesc = 0;
 750 #if 0
 751         dump(c);
 752 #endif
 753         /* start of escseq */
 754         if(c == '\033')
 755                 escreset(), inesc = 1;
 756         else if(inesc) {
 757                 inesc = escaddc(c);
 758         } /* normal char */
 759         else switch(c) {
 760                 default:
 761                         tsetchar(c);
 762                         tcursor(CSright);
 763                         break;
 764                 case '\t':
 765                         tputtab();
 766                         break;
 767                 case '\b':
 768                         tcursor(CSleft);
 769                         break;
 770                 case '\r':
 771                         tmoveto(0, term.c.y);
 772                         break;
 773                 case '\n':
 774                         tnewline();
 775                         break;
 776                 case '\a':
 777                         xbell();
 778                         break;
 779         }
 780 }
 781
 782 void
 783 tputs(char *s, int len) {
 784         for(; len > 0; len--)
 785                 tputc(*s++);
 786 }
 787
 788 void
 789 tdump(void) {
 790         int row, col;
 791         Glyph c;
 792
 793         for(row = 0; row < term.row; row++) {
 794                 for(col = 0; col < term.col; col++) {
 795                         if(col == term.c.x && row == term.c.y)
 796                                 putchar('#');
 797                         else {
 798                                 c = term.line[row][col];
 799                                 putchar(c.state & CRset ? c.c : '.');
 800                         }
 801                 }
 802                 putchar('\n');
 803         }
 804 }
 805
 806 void
 807 tresize(int col, int row) {
 808         int i;
 809         Line *line;
 810         int minrow = MIN(row, term.row);
 811         int mincol = MIN(col, term.col);
 812
 813         if(col < 1 || row < 1)
 814                 return;
 815         /* alloc */
 816         line = calloc(row, sizeof(Line));
 817         for(i = 0 ; i < row; i++)
 818                 line[i] = calloc(col, sizeof(Glyph));
 819         /* copy */
 820         for(i = 0 ; i < minrow; i++)
 821                 memcpy(line[i], term.line[i], mincol * sizeof(Glyph));
 822         /* free */
 823         for(i = 0; i < term.row; i++)
 824                 free(term.line[i]);
 825         free(term.line);
 826
 827         LIMIT(term.c.x, 0, col-1);
 828         LIMIT(term.c.y, 0, row-1);
 829         LIMIT(term.top, 0, row-1);
 830         LIMIT(term.bot, 0, row-1);
 831
 832         term.bot = row-1;
 833         term.line = line;
 834         term.col = col, term.row = row;
 835 }
 836
 837 unsigned long
 838 xgetcol(const char *s) {
 839         XColor color;
 840         Colormap cmap = DefaultColormap(xw.dis, xw.scr);
 841
 842         if(!XAllocNamedColor(xw.dis, cmap, s, &color, &color)) {
 843                 color.pixel = WhitePixel(xw.dis, xw.scr);
 844                 fprintf(stderr, "Could not allocate color '%s'\n", s);
 845         }
 846         return color.pixel;
 847 }
 848
 849
 850 void
 851 xclear(int x1, int y1, int x2, int y2) {
 852         XClearArea(xw.dis, xw.win,
 853                         x1 * xw.cw, y1 * xw.ch,
 854                         (x2-x1+1) * xw.cw, (y2-y1+1) * xw.ch,
 855                         False);
 856 }
 857
 858
 859 void
 860 xscroll(void) {
 861         int srcy = (term.top+1) * xw.ch;
 862         int dsty = term.top * xw.ch;
 863         int height = (term.bot-term.top) * xw.ch;
 864
 865         xcursor(CShide);
 866         XCopyArea(xw.dis, xw.win, xw.win, dc.gc, 0, srcy, xw.w, height, 0, dsty);
 867         xclear(0, term.bot, term.col-1, term.bot);
 868 }
 869
 870
 871
 872
 873 void
 874 xinit(void) {
 875         XGCValues values;
 876         unsigned long valuemask;
 877         XClassHint chint;
 878         XWMHints wmhint;
 879         XSizeHints shint;
 880         char *args[] = {NULL};
 881         int i;
 882
 883         xw.dis = XOpenDisplay(NULL);
 884         xw.scr = XDefaultScreen(xw.dis);
 885         if(!xw.dis)
 886                 die("Can't open display\n");
 887
 888         /* font */
 889         if(!(dc.font = XLoadQueryFont(xw.dis, FONT)))
 890                 die("Can't load font %s\n", FONT);
 891
 892         xw.cw = dc.font->max_bounds.rbearing - dc.font->min_bounds.lbearing;
 893         xw.ch = dc.font->ascent + dc.font->descent + LINESPACE;
 894
 895         /* colors */
 896         for(i = 0; i < LEN(colorname); i++)
 897                 dc.col[i] = xgetcol(colorname[i]);
 898
 899         term.c.attr.fg = DefaultFG;
 900         term.c.attr.bg = DefaultBG;
 901         term.c.attr.mode = ATnone;
 902         /* windows */
 903         xw.h = term.row * xw.ch;
 904         xw.w = term.col * xw.cw;
 905         /* XXX: this BORDER is useless after the first resize, handle it in xdraws() */
 906         xw.win = XCreateSimpleWindow(xw.dis, XRootWindow(xw.dis, xw.scr), 0, 0,
 907                         xw.w, xw.h, BORDER,
 908                         dc.col[DefaultBG],
 909                         dc.col[DefaultBG]);
 910         /* gc */
 911         values.foreground = XWhitePixel(xw.dis, xw.scr);
 912         values.font = dc.font->fid;
 913         valuemask = GCForeground | GCFont;
 914         dc.gc = XCreateGC(xw.dis, xw.win, valuemask, &values);
 915         XMapWindow(xw.dis, xw.win);
 916         /* wm stuff */
 917         chint.res_name = TNAME, chint.res_class = TNAME;
 918         wmhint.input = 1, wmhint.flags = InputHint;
 919         shint.height_inc = xw.ch, shint.width_inc = xw.cw;
 920         shint.height = xw.h, shint.width = xw.w;
 921         shint.flags = PSize | PResizeInc;
 922         XSetWMProperties(xw.dis, xw.win, NULL, NULL, &args[0], 0, &shint, &wmhint, &chint);
 923         XStoreName(xw.dis, xw.win, TNAME);
 924         XSync(xw.dis, 0);
 925 }
 926
 927 void
 928 xdrawc(int x, int y, Glyph g) {
 929         XRectangle r = { x * xw.cw, y * xw.ch, xw.cw, xw.ch };
 930         unsigned long xfg, xbg;
 931
 932         /* reverse video */
 933         if(g.mode & ATreverse)
 934                 xfg = dc.col[g.bg], xbg = dc.col[g.fg];
 935         else
 936                 xfg = dc.col[g.fg], xbg = dc.col[g.bg];
 937         /* background */
 938         XSetForeground(xw.dis, dc.gc, xbg);
 939         XFillRectangles(xw.dis, xw.win, dc.gc, &r, 1);
 940         /* string */
 941         XSetForeground(xw.dis, dc.gc, xfg);
 942         XDrawString(xw.dis, xw.win, dc.gc, r.x, r.y+dc.font->ascent, &(g.c), 1);
 943         if(g.mode & ATbold)      /* XXX: bold hack (draw again at x+1) */
 944                 XDrawString(xw.dis, xw.win, dc.gc, r.x+1, r.y+dc.font->ascent, &(g.c), 1);
 945         /* underline */
 946         if(g.mode & ATunderline) {
 947                 r.y += dc.font->ascent + 1;
 948                 XDrawLine(xw.dis, xw.win, dc.gc, r.x, r.y, r.x+r.width-1, r.y);
 949         }
 950 }
 951
 952 void
 953 xcursor(int mode) {
 954         static int oldx = 0;
 955         static int oldy = 0;
 956         Glyph g = {' ', ATnone, DefaultBG, DefaultCS, 0};
 957
 958         LIMIT(oldx, 0, term.col-1);
 959         LIMIT(oldy, 0, term.row-1);
 960
 961         if(term.line[term.c.y][term.c.x].state & CRset)
 962                 g.c = term.line[term.c.y][term.c.x].c;
 963         /* remove the old cursor */
 964         if(term.line[oldy][oldx].state & CRset)
 965                 xdrawc(oldx, oldy, term.line[oldy][oldx]);
 966         else xclear(oldx, oldy, oldx, oldy); /* XXX: maybe a bug */
 967         if(mode == CSdraw && !term.c.hidden) {
 968                 xdrawc(term.c.x, term.c.y, g);
 969                 oldx = term.c.x, oldy = term.c.y;
 970         }
 971 }
 972
 973
 974 void
 975 draw(int redraw_all) {
 976         int x, y;
 977         int changed, set;
 978
 979         if(redraw_all)
 980                 XClearWindow(xw.dis, xw.win);
 981         /* XXX: drawing could be optimised */
 982         for(y = 0; y < term.row; y++) {
 983                 for(x = 0; x < term.col; x++) {
 984                         changed = term.line[y][x].state & CRupdate;
 985                         set = term.line[y][x].state & CRset;
 986                         if((changed && set) || (redraw_all && set)) {
 987                                 term.line[y][x].state &= ~CRupdate;
 988                                 xdrawc(x, y, term.line[y][x]);
 989                         }
 990                 }
 991         }
 992         xcursor(CSdraw);
 993 }
 994
 995 char*
 996 kmap(KeySym k) {
 997         int i;
 998         for(i = 0; i < LEN(key); i++)
 999                 if(key[i].k == k)
1000                         return (char*)key[i].s;
1001         return NULL;
1002 }
1003
1004 void
1005 kpress(XKeyEvent *e) {
1006         KeySym ksym;
1007         char buf[32];
1008         int len;
1009         int meta;
1010         int shift;
1011         char* skmap;
1012
1013         meta  = e->state & Mod1Mask;
1014         shift = e->state & ShiftMask;
1015         len = XLookupString(e, buf, sizeof(buf), &ksym, NULL);
1016         if(skmap = kmap(ksym))
1017                 ttywrite(skmap, strlen(skmap));
1018         else if(len > 0) {
1019                 buf[sizeof(buf)-1] = '\0';
1020                 if(meta && len == 1)
1021                         ttywrite("\033", 1);
1022                 ttywrite(buf, len);
1023         } else
1024                 switch(ksym) {
1025                 case XK_Insert:
1026                         if(shift)
1027                                 /* XXX: paste X clipboard */;
1028                         break;
1029                 default:
1030                         fprintf(stderr, "errkey: %d\n", (int)ksym);
1031                         break;
1032                 }
1033 }
1034
1035 void
1036 resize(XEvent *e) {
1037         int col, row;
1038         col = e->xconfigure.width / xw.cw;
1039         row = e->xconfigure.height / xw.ch;
1040
1041         if(term.col != col || term.row != row) {
1042                 tresize(col, row);
1043                 ttyresize(col, row);
1044                 xw.w = e->xconfigure.width;
1045                 xw.h = e->xconfigure.height;
1046                 draw(SCredraw);
1047         }
1048 }
1049
1050 void
1051 run(void) {
1052         int ret;
1053         XEvent ev;
1054         fd_set rfd;
1055         int xfd = XConnectionNumber(xw.dis);
1056
1057         running = 1;
1058         XSelectInput(xw.dis, xw.win, ExposureMask | KeyPressMask | StructureNotifyMask);
1059         XResizeWindow(xw.dis, xw.win, xw.w , xw.h); /* fix resize bug in wmii (?) */
1060
1061         while(running) {
1062                 FD_ZERO(&rfd);
1063                 FD_SET(cmdfd, &rfd);
1064                 FD_SET(xfd, &rfd);
1065                 XFlush(xw.dis);
1066                 ret = select(MAX(xfd, cmdfd)+1, &rfd, NULL, NULL, NULL);
1067
1068                 if(ret < 0)
1069                         die("select failed: %s\n", SERRNO);
1070
1071                 if(FD_ISSET(xfd, &rfd)) {
1072                         while(XPending(xw.dis)) {
1073                                 XNextEvent(xw.dis, &ev);
1074                                 switch (ev.type) {
1075                                 default:
1076                                         break;
1077                                 case KeyPress:
1078                                         kpress(&ev.xkey);
1079                                         break;
1080                                 case Expose:
1081                                         draw(SCredraw);
1082                                         break;
1083                                 case ConfigureNotify:
1084                                         resize(&ev);
1085                                         break;
1086                                 }
1087                         }
1088                 }
1089                 if(FD_ISSET(cmdfd, &rfd)) {
1090                         ttyread();
1091                         draw(SCupdate);
1092                 }
1093         }
1094 }
1095
1096 int
1097 main(int argc, char *argv[]) {
1098         if(argc == 2 && !strncmp("-v", argv[1], 3))
1099                 die("st-" VERSION ", © 2009 st engineers\n");
1100         else if(argc != 1)
1101                 die("usage: st [-v]\n");
1102         setlocale(LC_CTYPE, "");
1103         tnew(80, 24);
1104         ttynew();
1105         xinit();
1106         run();
1107         return 0;
1108 }