cl/cl_main.c

   1 /*-
   2  * Copyright (c) 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  * Copyright (c) 1993, 1994, 1995, 1996
   5  *      Keith Bostic.  All rights reserved.
   6  *
   7  * See the LICENSE file for redistribution information.
   8  */
   9
  10 #include "config.h"
  11
  12 #ifndef lint
  13 static const char sccsid[] = "$Id: cl_main.c,v 10.54 2001/07/29 19:07:27 skimo Exp $ (Berkeley) $Date: 2001/07/29 19:07:27 $";
  14 #endif /* not lint */
  15
  16 #include <sys/types.h>
  17 #include <sys/queue.h>
  18
  19 #include <bitstring.h>
  20 #include <errno.h>
  21 #include <fcntl.h>
  22 #include <signal.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #ifdef HAVE_TERM_H
  27 #include <term.h>
  28 #endif
  29 #include <termios.h>
  30 #include <unistd.h>
  31
  32 #include "../common/common.h"
  33 #include "../ip/extern.h"
  34 #include "cl.h"
  35 #include "pathnames.h"
  36
  37 GS *__global_list;                              /* GLOBAL: List of screens. */
  38 sigset_t __sigblockset;                         /* GLOBAL: Blocked signals. */
  39
  40 static void        cl_func_std __P((WIN *));
  41 static void        cl_end __P((CL_PRIVATE *));
  42 static CL_PRIVATE *cl_init __P((WIN *));
  43 static void        perr __P((char *, char *));
  44 static int         setsig __P((int, struct sigaction *, void (*)(int)));
  45 static void        sig_end __P((GS *));
  46 static void        term_init __P((char *, char *));
  47
  48 /*
  49  * main --
  50  *      This is the main loop for the standalone curses editor.
  51  */
  52 int
  53 main(int argc, char **argv)
  54 {
  55         static int reenter;
  56         CL_PRIVATE *clp;
  57         GS *gp;
  58         WIN *wp;
  59         size_t rows, cols;
  60         int rval;
  61         char **p_av, **t_av, *ttype;
  62
  63         /* If loaded at 0 and jumping through a NULL pointer, stop. */
  64         if (reenter++)
  65                 abort();
  66
  67         /* Create and initialize the global structure. */
  68         __global_list = gp = gs_init(argv[0]);
  69
  70         /*
  71          * Strip out any arguments that vi isn't going to understand.  There's
  72          * no way to portably call getopt twice, so arguments parsed here must
  73          * be removed from the argument list.
  74          */
  75         for (p_av = t_av = argv;;) {
  76                 if (*t_av == NULL) {
  77                         *p_av = NULL;
  78                         break;
  79                 }
  80                 if (!strcmp(*t_av, "--")) {
  81                         while ((*p_av++ = *t_av++) != NULL);
  82                         break;
  83                 }
  84                 *p_av++ = *t_av++;
  85         }
  86
  87         /* Create new window */
  88         wp = gs_new_win(gp);
  89
  90         /* Create and initialize the CL_PRIVATE structure. */
  91         clp = cl_init(wp);
  92
  93         /*
  94          * Initialize the terminal information.
  95          *
  96          * We have to know what terminal it is from the start, since we may
  97          * have to use termcap/terminfo to find out how big the screen is.
  98          */
  99         if ((ttype = getenv("TERM")) == NULL)
 100                 ttype = "unknown";
 101         term_init(gp->progname, ttype);
 102
 103         /* Add the terminal type to the global structure. */
 104         if ((OG_D_STR(gp, GO_TERM) =
 105             OG_STR(gp, GO_TERM) = strdup(ttype)) == NULL)
 106                 perr(gp->progname, NULL);
 107
 108         /* Figure out how big the screen is. */
 109         if (cl_ssize(NULL, 0, &rows, &cols, NULL))
 110                 exit (1);
 111
 112         /* Add the rows and columns to the global structure. */
 113         OG_VAL(gp, GO_LINES) = OG_D_VAL(gp, GO_LINES) = rows;
 114         OG_VAL(gp, GO_COLUMNS) = OG_D_VAL(gp, GO_COLUMNS) = cols;
 115
 116         /* Ex wants stdout to be buffered. */
 117         (void)setvbuf(stdout, NULL, _IOFBF, 0);
 118
 119         /* Start catching signals. */
 120         if (sig_init(gp, NULL))
 121                 exit (1);
 122
 123         /* Run ex/vi. */
 124         rval = editor(wp, argc, argv);
 125
 126         /* Clean out the global structure. */
 127         gs_end(gp);
 128
 129         /* Clean up signals. */
 130         sig_end(gp);
 131
 132         /* Clean up the terminal. */
 133         (void)cl_quit(gp);
 134
 135         /*
 136          * XXX
 137          * Reset the O_MESG option.
 138          */
 139         if (clp->tgw != TGW_UNKNOWN)
 140                 (void)cl_omesg(NULL, clp, clp->tgw == TGW_SET);
 141
 142         /*
 143          * XXX
 144          * Reset the X11 xterm icon/window name.
 145          */
 146         if (F_ISSET(clp, CL_RENAME))
 147                 cl_setname(gp, clp->oname);
 148
 149         /* If a killer signal arrived, pretend we just got it. */
 150         if (clp->killersig) {
 151                 (void)signal(clp->killersig, SIG_DFL);
 152                 (void)kill(getpid(), clp->killersig);
 153                 /* NOTREACHED */
 154         }
 155
 156         /* Free the global and CL private areas. */
 157 #if defined(DEBUG) || defined(PURIFY) || defined(LIBRARY)
 158         cl_end(clp);
 159         free(gp);
 160 #endif
 161
 162         exit (rval);
 163 }
 164
 165 /*
 166  * cl_init --
 167  *      Create and partially initialize the CL structure.
 168  */
 169 static CL_PRIVATE *
 170 cl_init(WIN *wp)
 171 {
 172         CL_PRIVATE *clp;
 173         int fd;
 174         GS *gp;
 175
 176         gp = wp->gp;
 177
 178         /* Allocate the CL private structure. */
 179         CALLOC_NOMSG(NULL, clp, CL_PRIVATE *, 1, sizeof(CL_PRIVATE));
 180         if (clp == NULL)
 181                 perr(gp->progname, NULL);
 182         gp->cl_private = clp;
 183
 184         /*
 185          * Set the CL_STDIN_TTY flag.  It's purpose is to avoid setting
 186          * and resetting the tty if the input isn't from there.  We also
 187          * use the same test to determine if we're running a script or
 188          * not.
 189          */
 190         if (isatty(STDIN_FILENO))
 191                 F_SET(clp, CL_STDIN_TTY);
 192         else
 193                 F_SET(gp, G_SCRIPTED);
 194
 195         /*
 196          * We expect that if we've lost our controlling terminal that the
 197          * open() (but not the tcgetattr()) will fail.
 198          */
 199         if (F_ISSET(clp, CL_STDIN_TTY)) {
 200                 if (tcgetattr(STDIN_FILENO, &clp->orig) == -1)
 201                         goto tcfail;
 202         } else if ((fd = open(_PATH_TTY, O_RDONLY, 0)) != -1) {
 203                 if (tcgetattr(fd, &clp->orig) == -1) {
 204 tcfail:                 perr(gp->progname, "tcgetattr");
 205                         exit (1);
 206                 }
 207                 (void)close(fd);
 208         }
 209
 210         /* Initialize the list of curses functions. */
 211         cl_func_std(wp);
 212
 213         return (clp);
 214 }
 215
 216 /*
 217  * cl_end --
 218  *      Discard the CL structure.
 219  */
 220 static void
 221 cl_end(CL_PRIVATE *clp)
 222 {
 223         if (clp->oname != NULL)
 224                 free(clp->oname);
 225         free(clp);
 226 }
 227
 228 /*
 229  * term_init --
 230  *      Initialize terminal information.
 231  */
 232 static void
 233 term_init(char *name, char *ttype)
 234 {
 235         int err;
 236
 237         /* Set up the terminal database information. */
 238         setupterm(ttype, STDOUT_FILENO, &err);
 239         switch (err) {
 240         case -1:
 241                 (void)fprintf(stderr,
 242                     "%s: No terminal database found\n", name);
 243                 exit (1);
 244         case 0:
 245                 (void)fprintf(stderr,
 246                     "%s: %s: unknown terminal type\n", name, ttype);
 247                 exit (1);
 248         }
 249 }
 250
 251 #define GLOBAL_CLP \
 252         CL_PRIVATE *clp = GCLP(__global_list);
 253 static void
 254 h_hup(int signo)
 255 {
 256         GLOBAL_CLP;
 257
 258         F_SET(clp, CL_SIGHUP);
 259         clp->killersig = SIGHUP;
 260 }
 261
 262 static void
 263 h_int(int signo)
 264 {
 265         GLOBAL_CLP;
 266
 267         F_SET(clp, CL_SIGINT);
 268 }
 269
 270 static void
 271 h_term(int signo)
 272 {
 273         GLOBAL_CLP;
 274
 275         F_SET(clp, CL_SIGTERM);
 276         clp->killersig = SIGTERM;
 277 }
 278
 279 static void
 280 h_winch(int signo)
 281 {
 282         GLOBAL_CLP;
 283
 284         F_SET(clp, CL_SIGWINCH);
 285 }
 286 #undef  GLOBAL_CLP
 287
 288 /*
 289  * sig_init --
 290  *      Initialize signals.
 291  *
 292  * PUBLIC: int sig_init __P((GS *, SCR *));
 293  */
 294 int
 295 sig_init(GS *gp, SCR *sp)
 296 {
 297         CL_PRIVATE *clp;
 298
 299         clp = GCLP(gp);
 300
 301         if (sp == NULL) {
 302                 (void)sigemptyset(&__sigblockset);
 303                 if (sigaddset(&__sigblockset, SIGHUP) ||
 304                     setsig(SIGHUP, &clp->oact[INDX_HUP], h_hup) ||
 305                     sigaddset(&__sigblockset, SIGINT) ||
 306                     setsig(SIGINT, &clp->oact[INDX_INT], h_int) ||
 307                     sigaddset(&__sigblockset, SIGTERM) ||
 308                     setsig(SIGTERM, &clp->oact[INDX_TERM], h_term)
 309 #ifdef SIGWINCH
 310                     ||
 311                     sigaddset(&__sigblockset, SIGWINCH) ||
 312                     setsig(SIGWINCH, &clp->oact[INDX_WINCH], h_winch)
 313 #endif
 314                     ) {
 315                         perr(gp->progname, NULL);
 316                         return (1);
 317                 }
 318         } else
 319                 if (setsig(SIGHUP, NULL, h_hup) ||
 320                     setsig(SIGINT, NULL, h_int) ||
 321                     setsig(SIGTERM, NULL, h_term)
 322 #ifdef SIGWINCH
 323                     ||
 324                     setsig(SIGWINCH, NULL, h_winch)
 325 #endif
 326                     ) {
 327                         msgq(sp, M_SYSERR, "signal-reset");
 328                 }
 329         return (0);
 330 }
 331
 332 /*
 333  * setsig --
 334  *      Set a signal handler.
 335  */
 336 static int
 337 setsig(int signo, struct sigaction *oactp, void (*handler) (int))
 338 {
 339         struct sigaction act;
 340
 341         /*
 342          * Use sigaction(2), not signal(3), since we don't always want to
 343          * restart system calls.  The example is when waiting for a command
 344          * mode keystroke and SIGWINCH arrives.  Besides, you can't portably
 345          * restart system calls (thanks, POSIX!).  On the other hand, you
 346          * can't portably NOT restart system calls (thanks, Sun!).  SunOS
 347          * used SA_INTERRUPT as their extension to NOT restart read calls.
 348          * We sure hope nobody else used it for anything else.  Mom told me
 349          * there'd be days like this.  She just never told me that there'd
 350          * be so many.
 351          */
 352         act.sa_handler = handler;
 353         sigemptyset(&act.sa_mask);
 354
 355 #ifdef SA_INTERRUPT
 356         act.sa_flags = SA_INTERRUPT;
 357 #else
 358         act.sa_flags = 0;
 359 #endif
 360         return (sigaction(signo, &act, oactp));
 361 }
 362
 363 /*
 364  * sig_end --
 365  *      End signal setup.
 366  */
 367 static void
 368 sig_end(GS *gp)
 369 {
 370         CL_PRIVATE *clp;
 371
 372         clp = GCLP(gp);
 373         (void)sigaction(SIGHUP, NULL, &clp->oact[INDX_HUP]);
 374         (void)sigaction(SIGINT, NULL, &clp->oact[INDX_INT]);
 375         (void)sigaction(SIGTERM, NULL, &clp->oact[INDX_TERM]);
 376 #ifdef SIGWINCH
 377         (void)sigaction(SIGWINCH, NULL, &clp->oact[INDX_WINCH]);
 378 #endif
 379 }
 380
 381 /*
 382  * cl_func_std --
 383  *      Initialize the standard curses functions.
 384  */
 385 static void
 386 cl_func_std(WIN *wp)
 387 {
 388         GS *gp;
 389
 390         gp = wp->gp;
 391
 392         gp->scr_addstr = cl_addstr;
 393         gp->scr_waddstr = cl_waddstr;
 394         gp->scr_attr = cl_attr;
 395         gp->scr_baud = cl_baud;
 396         gp->scr_bell = cl_bell;
 397         gp->scr_busy = NULL;
 398         gp->scr_child = NULL;
 399         gp->scr_clrtoeol = cl_clrtoeol;
 400         gp->scr_cursor = cl_cursor;
 401         gp->scr_deleteln = cl_deleteln;
 402         gp->scr_reply = NULL;
 403         gp->scr_discard = cl_discard;
 404         gp->scr_event = cl_event;
 405         gp->scr_ex_adjust = cl_ex_adjust;
 406         gp->scr_fmap = cl_fmap;
 407         gp->scr_insertln = cl_insertln;
 408         gp->scr_keyval = cl_keyval;
 409         gp->scr_move = cl_move;
 410         wp->scr_msg = NULL;
 411         gp->scr_optchange = cl_optchange;
 412         gp->scr_refresh = cl_refresh;
 413         gp->scr_rename = cl_rename;
 414         gp->scr_screen = cl_screen;
 415         gp->scr_split = cl_split;
 416         gp->scr_suspend = cl_suspend;
 417         gp->scr_usage = cl_usage;
 418 }
 419
 420 /*
 421  * perr --
 422  *      Print system error.
 423  */
 424 static void
 425 perr(char *name, char *msg)
 426 {
 427         (void)fprintf(stderr, "%s:", name);
 428         if (msg != NULL)
 429                 (void)fprintf(stderr, "%s:", msg);
 430         (void)fprintf(stderr, "%s\n", strerror(errno));
 431         exit(1);
 432 }