common/msg.c

   1 /*-
   2  * Copyright (c) 1991, 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  * Copyright (c) 1991, 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: msg.c,v 10.60 2002/03/27 10:06:56 skimo Exp $ (Berkeley) $Date: 2002/03/27 10:06:56 $";
  14 #endif /* not lint */
  15
  16 #include <sys/param.h>
  17 #include <sys/types.h>          /* XXX: param.h may not have included types.h */
  18 #include <sys/queue.h>
  19 #include <sys/stat.h>
  20 #include <sys/time.h>
  21
  22 #include <bitstring.h>
  23 #include <ctype.h>
  24 #include <errno.h>
  25 #include <fcntl.h>
  26 #include <limits.h>
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <unistd.h>
  31
  32 #ifdef __STDC__
  33 #include <stdarg.h>
  34 #else
  35 #include <varargs.h>
  36 #endif
  37
  38 #include "common.h"
  39 #include "../vi/vi.h"
  40
  41 /*
  42  * msgq --
  43  *      Display a message.
  44  *
  45  * PUBLIC: void msgq __P((SCR *, mtype_t, const char *, ...));
  46  */
  47 void
  48 #ifdef __STDC__
  49 msgq(SCR *sp, mtype_t mt, const char *fmt, ...)
  50 #else
  51 msgq(sp, mt, fmt, va_alist)
  52         SCR *sp;
  53         mtype_t mt;
  54         const char *fmt;
  55         va_dcl
  56 #endif
  57 {
  58 #ifndef NL_ARGMAX
  59 #define __NL_ARGMAX     20              /* Set to 9 by System V. */
  60         struct {
  61                 const char *str;        /* String pointer. */
  62                 size_t   arg;           /* Argument number. */
  63                 size_t   prefix;        /* Prefix string length. */
  64                 size_t   skip;          /* Skipped string length. */
  65                 size_t   suffix;        /* Suffix string length. */
  66         } str[__NL_ARGMAX];
  67 #endif
  68         static int reenter;             /* STATIC: Re-entrancy check. */
  69         CHAR_T ch;
  70         GS *gp;
  71         WIN *wp;
  72         size_t blen, cnt1, cnt2, len, mlen, nlen, soff;
  73         const char *p, *t, *u;
  74         char *bp, *mp, *rbp, *s_rbp;
  75         va_list ap;
  76
  77         /*
  78          * !!!
  79          * It's possible to enter msg when there's no screen to hold the
  80          * message.  If sp is NULL, ignore the special cases and put the
  81          * message out to stderr.
  82          */
  83         if (sp == NULL) {
  84                 gp = NULL;
  85                 if (mt == M_BERR)
  86                         mt = M_ERR;
  87                 else if (mt == M_VINFO)
  88                         mt = M_INFO;
  89         } else {
  90                 gp = sp->gp;
  91                 wp = sp->wp;
  92                 switch (mt) {
  93                 case M_BERR:
  94                         if (F_ISSET(sp, SC_VI) && !O_ISSET(sp, O_VERBOSE)) {
  95                                 F_SET(gp, G_BELLSCHED);
  96                                 return;
  97                         }
  98                         mt = M_ERR;
  99                         break;
 100                 case M_VINFO:
 101                         if (!O_ISSET(sp, O_VERBOSE))
 102                                 return;
 103                         mt = M_INFO;
 104                         /* FALLTHROUGH */
 105                 case M_INFO:
 106                         if (F_ISSET(sp, SC_EX_SILENT))
 107                                 return;
 108                         break;
 109                 case M_ERR:
 110                 case M_SYSERR:
 111                 case M_DBERR:
 112                         break;
 113                 default:
 114                         abort();
 115                 }
 116         }
 117
 118         /*
 119          * It's possible to reenter msg when it allocates space.  We're
 120          * probably dead anyway, but there's no reason to drop core.
 121          *
 122          * XXX
 123          * Yes, there's a race, but it should only be two instructions.
 124          */
 125         if (reenter++)
 126                 return;
 127
 128         /* Get space for the message. */
 129         nlen = 1024;
 130         if (0) {
 131 retry:          FREE_SPACE(sp, bp, blen);
 132                 nlen *= 2;
 133         }
 134         bp = NULL;
 135         blen = 0;
 136         GET_SPACE_GOTO(sp, bp, blen, nlen);
 137
 138         /*
 139          * Error prefix.
 140          *
 141          * mp:   pointer to the current next character to be written
 142          * mlen: length of the already written characters
 143          * blen: total length of the buffer
 144          */
 145 #define REM     (blen - mlen)
 146         mp = bp;
 147         mlen = 0;
 148         if (mt == M_SYSERR || mt == M_DBERR) {
 149                 p = msg_cat(sp, "020|Error: ", &len);
 150                 if (REM < len)
 151                         goto retry;
 152                 memcpy(mp, p, len);
 153                 mp += len;
 154                 mlen += len;
 155         }
 156
 157         /*
 158          * If we're running an ex command that the user didn't enter, display
 159          * the file name and line number prefix.
 160          */
 161         if ((mt == M_ERR || mt == M_SYSERR) &&
 162             sp != NULL && wp != NULL && wp->if_name != NULL) {
 163                 for (p = wp->if_name; *p != '\0'; ++p) {
 164                         len = snprintf(mp, REM, "%s", KEY_NAME(sp, *p));
 165                         mp += len;
 166                         if ((mlen += len) > blen)
 167                                 goto retry;
 168                 }
 169                 len = snprintf(mp, REM, ", %d: ", wp->if_lno);
 170                 mp += len;
 171                 if ((mlen += len) > blen)
 172                         goto retry;
 173         }
 174
 175         /* If nothing to format, we're done. */
 176         if (fmt == NULL)
 177                 goto nofmt;
 178         fmt = msg_cat(sp, fmt, NULL);
 179
 180 #ifndef NL_ARGMAX
 181         /*
 182          * Nvi should run on machines that don't support the numbered argument
 183          * specifications (%[digit]*$).  We do this by reformatting the string
 184          * so that we can hand it to vsprintf(3) and it will use the arguments
 185          * in the right order.  When vsprintf returns, we put the string back
 186          * into the right order.  It's undefined, according to SVID III, to mix
 187          * numbered argument specifications with the standard style arguments,
 188          * so this should be safe.
 189          *
 190          * In addition, we also need a character that is known to not occur in
 191          * any vi message, for separating the parts of the string.  As callers
 192          * of msgq are responsible for making sure that all the non-printable
 193          * characters are formatted for printing before calling msgq, we use a
 194          * random non-printable character selected at terminal initialization
 195          * time.  This code isn't fast by any means, but as messages should be
 196          * relatively short and normally have only a few arguments, it won't be
 197          * too bad.  Regardless, nobody has come up with any other solution.
 198          *
 199          * The result of this loop is an array of pointers into the message
 200          * string, with associated lengths and argument numbers.  The array
 201          * is in the "correct" order, and the arg field contains the argument
 202          * order.
 203          */
 204         for (p = fmt, soff = 0; soff < __NL_ARGMAX;) {
 205                 for (t = p; *p != '\0' && *p != '%'; ++p);
 206                 if (*p == '\0')
 207                         break;
 208                 ++p;
 209                 if (!isdigit(*p)) {
 210                         if (*p == '%')
 211                                 ++p;
 212                         continue;
 213                 }
 214                 for (u = p; *++p != '\0' && isdigit(*p););
 215                 if (*p != '$')
 216                         continue;
 217
 218                 /* Up to, and including the % character. */
 219                 str[soff].str = t;
 220                 str[soff].prefix = u - t;
 221
 222                 /* Up to, and including the $ character. */
 223                 str[soff].arg = atoi(u);
 224                 str[soff].skip = (p - u) + 1;
 225                 if (str[soff].arg >= __NL_ARGMAX)
 226                         goto ret;
 227
 228                 /* Up to, and including the conversion character. */
 229                 for (u = p; (ch = *++p) != '\0';)
 230                         if (isalpha(ch) &&
 231                             strchr("diouxXfeEgGcspn", ch) != NULL)
 232                                 break;
 233                 str[soff].suffix = p - u;
 234                 if (ch != '\0')
 235                         ++p;
 236                 ++soff;
 237         }
 238
 239         /* If no magic strings, we're done. */
 240         if (soff == 0)
 241                 goto format;
 242
 243          /* Get space for the reordered strings. */
 244         if ((rbp = malloc(nlen)) == NULL)
 245                 goto ret;
 246         s_rbp = rbp;
 247
 248         /*
 249          * Reorder the strings into the message string based on argument
 250          * order.
 251          *
 252          * !!!
 253          * We ignore arguments that are out of order, i.e. if we don't find
 254          * an argument, we continue.  Assume (almost certainly incorrectly)
 255          * that whoever created the string knew what they were doing.
 256          *
 257          * !!!
 258          * Brute force "sort", but since we don't expect more than one or two
 259          * arguments in a string, the setup cost of a fast sort will be more
 260          * expensive than the loop.
 261          */
 262         for (cnt1 = 1; cnt1 <= soff; ++cnt1)
 263                 for (cnt2 = 0; cnt2 < soff; ++cnt2)
 264                         if (cnt1 == str[cnt2].arg) {
 265                                 memmove(s_rbp, str[cnt2].str, str[cnt2].prefix);
 266                                 memmove(s_rbp + str[cnt2].prefix,
 267                                     str[cnt2].str + str[cnt2].prefix +
 268                                     str[cnt2].skip, str[cnt2].suffix);
 269                                 s_rbp += str[cnt2].prefix + str[cnt2].suffix;
 270                                 *s_rbp++ =
 271                                     gp == NULL ? DEFAULT_NOPRINT : gp->noprint;
 272                                 break;
 273                         }
 274         *s_rbp = '\0';
 275         fmt = rbp;
 276 #endif
 277
 278 format: /* Format the arguments into the string. */
 279 #ifdef __STDC__
 280         va_start(ap, fmt);
 281 #else
 282         va_start(ap);
 283 #endif
 284         len = vsnprintf(mp, REM, fmt, ap);
 285         va_end(ap);
 286         if (len >= nlen)
 287                 goto retry;
 288
 289 #ifndef NL_ARGMAX
 290         if (soff == 0)
 291                 goto nofmt;
 292
 293         /*
 294          * Go through the resulting string, and, for each separator character
 295          * separated string, enter its new starting position and length in the
 296          * array.
 297          */
 298         for (p = t = mp, cnt1 = 1,
 299             ch = gp == NULL ? DEFAULT_NOPRINT : gp->noprint; *p != '\0'; ++p)
 300                 if (*p == ch) {
 301                         for (cnt2 = 0; cnt2 < soff; ++cnt2)
 302                                 if (str[cnt2].arg == cnt1)
 303                                         break;
 304                         str[cnt2].str = t;
 305                         str[cnt2].prefix = p - t;
 306                         t = p + 1;
 307                         ++cnt1;
 308                 }
 309
 310         /*
 311          * Reorder the strings once again, putting them back into the
 312          * message buffer.
 313          *
 314          * !!!
 315          * Note, the length of the message gets decremented once for
 316          * each substring, when we discard the separator character.
 317          */
 318         for (s_rbp = rbp, cnt1 = 0; cnt1 < soff; ++cnt1) {
 319                 memmove(rbp, str[cnt1].str, str[cnt1].prefix);
 320                 rbp += str[cnt1].prefix;
 321                 --len;
 322         }
 323         memmove(mp, s_rbp, rbp - s_rbp);
 324
 325         /* Free the reordered string memory. */
 326         free(s_rbp);
 327 #endif
 328
 329 nofmt:  mp += len;
 330         if ((mlen += len) > blen)
 331                 goto retry;
 332         if (mt == M_SYSERR) {
 333                 len = snprintf(mp, REM, ": %s", strerror(errno));
 334                 mp += len;
 335                 if ((mlen += len) > blen)
 336                         goto retry;
 337                 mt = M_ERR;
 338         }
 339         if (mt == M_DBERR) {
 340                 len = snprintf(mp, REM, ": %s", db_strerror(sp->db_error));
 341                 mp += len;
 342                 if ((mlen += len) > blen)
 343                         goto retry;
 344                 mt = M_ERR;
 345         }
 346
 347         /* Add trailing newline. */
 348         if ((mlen += 1) > blen)
 349                 goto retry;
 350         *mp = '\n';
 351
 352         if (sp != NULL)
 353                 (void)ex_fflush(sp);
 354         if (wp != NULL)
 355                 wp->scr_msg(sp, mt, bp, mlen);
 356         else
 357                 (void)fprintf(stderr, "%.*s", (int)mlen, bp);
 358
 359         /* Cleanup. */
 360 ret:    FREE_SPACE(sp, bp, blen);
 361 alloc_err:
 362         reenter = 0;
 363 }
 364
 365 /*
 366  * msgq_str --
 367  *      Display a message with an embedded string.
 368  *
 369  * PUBLIC: void msgq_wstr __P((SCR *, mtype_t, CHAR_T *, char *));
 370  */
 371 void
 372 msgq_wstr(SCR *sp, mtype_t mtype, CHAR_T *str, char *fmt)
 373 {
 374         size_t nlen;
 375         char *nstr;
 376
 377         if (str == NULL) {
 378                 msgq(sp, mtype, fmt);
 379                 return;
 380         }
 381         INT2CHAR(sp, str, STRLEN(str) + 1, nstr, nlen);
 382         msgq_str(sp, mtype, nstr, fmt);
 383 }
 384
 385 /*
 386  * msgq_str --
 387  *      Display a message with an embedded string.
 388  *
 389  * PUBLIC: void msgq_str __P((SCR *, mtype_t, char *, char *));
 390  */
 391 void
 392 msgq_str(SCR *sp, mtype_t mtype, char *str, char *fmt)
 393 {
 394         int nf, sv_errno;
 395         char *p;
 396
 397         if (str == NULL) {
 398                 msgq(sp, mtype, fmt);
 399                 return;
 400         }
 401
 402         sv_errno = errno;
 403         p = msg_print(sp, str, &nf);
 404         errno = sv_errno;
 405         msgq(sp, mtype, fmt, p);
 406         if (nf)
 407                 FREE_SPACE(sp, p, 0);
 408 }
 409
 410 /*
 411  * mod_rpt --
 412  *      Report on the lines that changed.
 413  *
 414  * !!!
 415  * Historic vi documentation (USD:15-8) claimed that "The editor will also
 416  * always tell you when a change you make affects text which you cannot see."
 417  * This wasn't true -- edit a large file and do "100d|1".  We don't implement
 418  * this semantic since it requires tracking each line that changes during a
 419  * command instead of just keeping count.
 420  *
 421  * Line counts weren't right in historic vi, either.  For example, given the
 422  * file:
 423  *      abc
 424  *      def
 425  * the command 2d}, from the 'b' would report that two lines were deleted,
 426  * not one.
 427  *
 428  * PUBLIC: void mod_rpt __P((SCR *));
 429  */
 430 void
 431 mod_rpt(SCR *sp)
 432 {
 433         static char * const action[] = {
 434                 "293|added",
 435                 "294|changed",
 436                 "295|deleted",
 437                 "296|joined",
 438                 "297|moved",
 439                 "298|shifted",
 440                 "299|yanked",
 441         };
 442         static char * const lines[] = {
 443                 "300|line",
 444                 "301|lines",
 445         };
 446         db_recno_t total;
 447         u_long rptval;
 448         int first, cnt;
 449         size_t blen, len, tlen;
 450         const char *t;
 451         char * const *ap;
 452         char *bp, *p;
 453
 454         /* Change reports are turned off in batch mode. */
 455         if (F_ISSET(sp, SC_EX_SILENT))
 456                 return;
 457
 458         /* Reset changing line number. */
 459         sp->rptlchange = OOBLNO;
 460
 461         /*
 462          * Don't build a message if not enough changed.
 463          *
 464          * !!!
 465          * And now, a vi clone test.  Historically, vi reported if the number
 466          * of changed lines was > than the value, not >=, unless it was a yank
 467          * command, which used >=.  No lie.  Furthermore, an action was never
 468          * reported for a single line action.  This is consistent for actions
 469          * other than yank, but yank didn't report single line actions even if
 470          * the report edit option was set to 1.  In addition, setting report to
 471          * 0 in the 4BSD historic vi was equivalent to setting it to 1, for an
 472          * unknown reason (this bug was fixed in System III/V at some point).
 473          * I got complaints, so nvi conforms to System III/V historic practice
 474          * except that we report a yank of 1 line if report is set to 1.
 475          */
 476 #define ARSIZE(a)       sizeof(a) / sizeof (*a)
 477 #define MAXNUM          25
 478         rptval = O_VAL(sp, O_REPORT);
 479         for (cnt = 0, total = 0; cnt < ARSIZE(action); ++cnt)
 480                 total += sp->rptlines[cnt];
 481         if (total == 0)
 482                 return;
 483         if (total <= rptval && sp->rptlines[L_YANKED] < rptval) {
 484                 for (cnt = 0; cnt < ARSIZE(action); ++cnt)
 485                         sp->rptlines[cnt] = 0;
 486                 return;
 487         }
 488
 489         /* Build and display the message. */
 490         GET_SPACE_GOTO(sp, bp, blen, sizeof(action) * MAXNUM + 1);
 491         for (p = bp, first = 1, tlen = 0,
 492             ap = action, cnt = 0; cnt < ARSIZE(action); ++ap, ++cnt)
 493                 if (sp->rptlines[cnt] != 0) {
 494                         if (first)
 495                                 first = 0;
 496                         else {
 497                                 *p++ = ';';
 498                                 *p++ = ' ';
 499                                 tlen += 2;
 500                         }
 501                         len = snprintf(p, MAXNUM, "%lu ",
 502                                 (unsigned long) sp->rptlines[cnt]);
 503                         p += len;
 504                         tlen += len;
 505                         t = msg_cat(sp,
 506                             lines[sp->rptlines[cnt] == 1 ? 0 : 1], &len);
 507                         memcpy(p, t, len);
 508                         p += len;
 509                         tlen += len;
 510                         *p++ = ' ';
 511                         ++tlen;
 512                         t = msg_cat(sp, *ap, &len);
 513                         memcpy(p, t, len);
 514                         p += len;
 515                         tlen += len;
 516                         sp->rptlines[cnt] = 0;
 517                 }
 518
 519         /* Add trailing newline. */
 520         *p = '\n';
 521         ++tlen;
 522
 523         (void)ex_fflush(sp);
 524         sp->wp->scr_msg(sp, M_INFO, bp, tlen);
 525
 526         FREE_SPACE(sp, bp, blen);
 527 alloc_err:
 528         return;
 529
 530 #undef ARSIZE
 531 #undef MAXNUM
 532 }
 533
 534 /*
 535  * msgq_status --
 536  *      Report on the file's status.
 537  *
 538  * PUBLIC: void msgq_status __P((SCR *, db_recno_t, u_int));
 539  */
 540 void
 541 msgq_status(SCR *sp, db_recno_t lno, u_int flags)
 542 {
 543         db_recno_t last;
 544         size_t blen, len;
 545         int cnt, needsep;
 546         const char *t;
 547         char **ap, *bp, *np, *p, *s;
 548
 549         /* Get sufficient memory. */
 550         len = strlen(sp->frp->name);
 551         GET_SPACE_GOTO(sp, bp, blen, len * MAX_CHARACTER_COLUMNS + 128);
 552         p = bp;
 553
 554         /* Copy in the filename. */
 555         for (p = bp, t = sp->frp->name; *t != '\0'; ++t) {
 556                 len = KEY_LEN(sp, *t);
 557                 memcpy(p, KEY_NAME(sp, *t), len);
 558                 p += len;
 559         }
 560         np = p;
 561         *p++ = ':';
 562         *p++ = ' ';
 563
 564         /* Copy in the argument count. */
 565         if (F_ISSET(sp, SC_STATUS_CNT) && sp->argv != NULL) {
 566                 for (cnt = 0, ap = sp->argv; *ap != NULL; ++ap, ++cnt);
 567                 if (cnt > 1) {
 568                         (void)sprintf(p,
 569                             msg_cat(sp, "317|%d files to edit", NULL), cnt);
 570                         p += strlen(p);
 571                         *p++ = ':';
 572                         *p++ = ' ';
 573                 }
 574                 F_CLR(sp, SC_STATUS_CNT);
 575         }
 576
 577         /*
 578          * See nvi/exf.c:file_init() for a description of how and when the
 579          * read-only bit is set.
 580          *
 581          * !!!
 582          * The historic display for "name changed" was "[Not edited]".
 583          */
 584         needsep = 0;
 585         if (F_ISSET(sp->frp, FR_NEWFILE)) {
 586                 F_CLR(sp->frp, FR_NEWFILE);
 587                 t = msg_cat(sp, "021|new file", &len);
 588                 memcpy(p, t, len);
 589                 p += len;
 590                 needsep = 1;
 591         } else {
 592                 if (F_ISSET(sp->frp, FR_NAMECHANGE)) {
 593                         t = msg_cat(sp, "022|name changed", &len);
 594                         memcpy(p, t, len);
 595                         p += len;
 596                         needsep = 1;
 597                 }
 598                 if (needsep) {
 599                         *p++ = ',';
 600                         *p++ = ' ';
 601                 }
 602                 if (F_ISSET(sp->ep, F_MODIFIED))
 603                         t = msg_cat(sp, "023|modified", &len);
 604                 else
 605                         t = msg_cat(sp, "024|unmodified", &len);
 606                 memcpy(p, t, len);
 607                 p += len;
 608                 needsep = 1;
 609         }
 610         if (F_ISSET(sp->frp, FR_UNLOCKED)) {
 611                 if (needsep) {
 612                         *p++ = ',';
 613                         *p++ = ' ';
 614                 }
 615                 t = msg_cat(sp, "025|UNLOCKED", &len);
 616                 memcpy(p, t, len);
 617                 p += len;
 618                 needsep = 1;
 619         }
 620         if (O_ISSET(sp, O_READONLY)) {
 621                 if (needsep) {
 622                         *p++ = ',';
 623                         *p++ = ' ';
 624                 }
 625                 t = msg_cat(sp, "026|readonly", &len);
 626                 memcpy(p, t, len);
 627                 p += len;
 628                 needsep = 1;
 629         }
 630         if (needsep) {
 631                 *p++ = ':';
 632                 *p++ = ' ';
 633         }
 634         if (LF_ISSET(MSTAT_SHOWLAST)) {
 635                 if (db_last(sp, &last))
 636                         return;
 637                 if (last == 0) {
 638                         t = msg_cat(sp, "028|empty file", &len);
 639                         memcpy(p, t, len);
 640                         p += len;
 641                 } else {
 642                         t = msg_cat(sp, "027|line %lu of %lu [%ld%%]", &len);
 643                         (void)sprintf(p, t, lno, last, (lno * 100) / last);
 644                         p += strlen(p);
 645                 }
 646         } else {
 647                 t = msg_cat(sp, "029|line %lu", &len);
 648                 (void)sprintf(p, t, lno);
 649                 p += strlen(p);
 650         }
 651 #ifdef DEBUG
 652         (void)sprintf(p, " (pid %lu)", (u_long)getpid());
 653         p += strlen(p);
 654 #endif
 655         *p++ = '\n';
 656         len = p - bp;
 657
 658         /*
 659          * There's a nasty problem with long path names.  Cscope and tags files
 660          * can result in long paths and vi will request a continuation key from
 661          * the user as soon as it starts the screen.  Unfortunately, the user
 662          * has already typed ahead, and chaos results.  If we assume that the
 663          * characters in the filenames and informational messages only take a
 664          * single screen column each, we can trim the filename.
 665          *
 666          * XXX
 667          * Status lines get put up at fairly awkward times.  For example, when
 668          * you do a filter read (e.g., :read ! echo foo) in the top screen of a
 669          * split screen, we have to repaint the status lines for all the screens
 670          * below the top screen.  We don't want users having to enter continue
 671          * characters for those screens.  Make it really hard to screw this up.
 672          */
 673         s = bp;
 674         if (LF_ISSET(MSTAT_TRUNCATE) && len > sp->cols) {
 675                 for (; s < np && (*s != '/' || (p - s) > sp->cols - 3); ++s);
 676                 if (s == np) {
 677                         s = p - (sp->cols - 5);
 678                         *--s = ' ';
 679                 }
 680                 *--s = '.';
 681                 *--s = '.';
 682                 *--s = '.';
 683                 len = p - s;
 684         }
 685
 686         /* Flush any waiting ex messages. */
 687         (void)ex_fflush(sp);
 688
 689         sp->wp->scr_msg(sp, M_INFO, s, len);
 690
 691         FREE_SPACE(sp, bp, blen);
 692 alloc_err:
 693         return;
 694 }
 695
 696 /*
 697  * msg_open --
 698  *      Open the message catalogs.
 699  *
 700  * PUBLIC: int msg_open __P((SCR *, char *));
 701  */
 702 int
 703 msg_open(SCR *sp, char *file)
 704 {
 705         /*
 706          * !!!
 707          * Assume that the first file opened is the system default, and that
 708          * all subsequent ones user defined.  Only display error messages
 709          * if we can't open the user defined ones -- it's useful to know if
 710          * the system one wasn't there, but if nvi is being shipped with an
 711          * installed system, the file will be there, if it's not, then the
 712          * message will be repeated every time nvi is started up.
 713          */
 714         static int first = 1;
 715         DB *db;
 716         DBT data, key;
 717         db_recno_t msgno;
 718         char *p, *t, buf[MAXPATHLEN];
 719
 720         if ((p = strrchr(file, '/')) != NULL && p[1] == '\0' &&
 721             (((t = getenv("LC_MESSAGES")) != NULL && t[0] != '\0') ||
 722             ((t = getenv("LANG")) != NULL && t[0] != '\0'))) {
 723                 (void)snprintf(buf, sizeof(buf), "%s%s", file, t);
 724                 p = buf;
 725         } else
 726                 p = file;
 727         if ((sp->db_error = db_create(&db, 0, 0)) != 0 ||
 728             (sp->db_error = db->set_re_source(db, p)) != 0 ||
 729             (sp->db_error = db->open(db, NULL, NULL, DB_RECNO, 0, 0)) != 0) {
 730                 if (first) {
 731                         first = 0;
 732                         return (1);
 733                 }
 734                 msgq_str(sp, M_DBERR, p, "%s");
 735                 return (1);
 736         }
 737
 738         /*
 739          * Test record 1 for the magic string.  The msgq call is here so
 740          * the message catalog build finds it.
 741          */
 742 #define VMC     "VI_MESSAGE_CATALOG"
 743         memset(&key, 0, sizeof(key));
 744         key.data = &msgno;
 745         key.size = sizeof(db_recno_t);
 746         memset(&data, 0, sizeof(data));
 747         msgno = 1;
 748         if ((sp->db_error = db->get(db, NULL, &key, &data, 0)) != 0 ||
 749             data.size != sizeof(VMC) - 1 ||
 750             memcmp(data.data, VMC, sizeof(VMC) - 1)) {
 751                 (void)db->close(db, DB_NOSYNC);
 752                 if (first) {
 753                         first = 0;
 754                         return (1);
 755                 }
 756                 msgq_str(sp, M_DBERR, p,
 757                     "030|The file %s is not a message catalog");
 758                 return (1);
 759         }
 760         first = 0;
 761
 762         if (sp->gp->msg != NULL)
 763                 (void)sp->gp->msg->close(sp->gp->msg, DB_NOSYNC);
 764         sp->gp->msg = db;
 765         return (0);
 766 }
 767
 768 /*
 769  * msg_close --
 770  *      Close the message catalogs.
 771  *
 772  * PUBLIC: void msg_close __P((GS *));
 773  */
 774 void
 775 msg_close(GS *gp)
 776 {
 777         if (gp->msg != NULL) {
 778                 (void)gp->msg->close(gp->msg, 0);
 779                 gp->msg = NULL;
 780         }
 781 }
 782
 783 /*
 784  * msg_cont --
 785  *      Return common continuation messages.
 786  *
 787  * PUBLIC: const char *msg_cmsg __P((SCR *, cmsg_t, size_t *));
 788  */
 789 const char *
 790 msg_cmsg(SCR *sp, cmsg_t which, size_t *lenp)
 791 {
 792         switch (which) {
 793         case CMSG_CONF:
 794                 return (msg_cat(sp, "268|confirm? [ynq]", lenp));
 795         case CMSG_CONT:
 796                 return (msg_cat(sp, "269|Press any key to continue: ", lenp));
 797         case CMSG_CONT_EX:
 798                 return (msg_cat(sp,
 799             "270|Press any key to continue [: to enter more ex commands]: ",
 800                     lenp));
 801         case CMSG_CONT_R:
 802                 return (msg_cat(sp, "161|Press Enter to continue: ", lenp));
 803         case CMSG_CONT_S:
 804                 return (msg_cat(sp, "275| cont?", lenp));
 805         case CMSG_CONT_Q:
 806                 return (msg_cat(sp,
 807                     "271|Press any key to continue [q to quit]: ", lenp));
 808         default:
 809                 abort();
 810         }
 811         /* NOTREACHED */
 812 }
 813
 814 /*
 815  * msg_cat --
 816  *      Return a single message from the catalog, plus its length.
 817  *
 818  * !!!
 819  * Only a single catalog message can be accessed at a time, if multiple
 820  * ones are needed, they must be copied into local memory.
 821  *
 822  * PUBLIC: const char *msg_cat __P((SCR *, const char *, size_t *));
 823  */
 824 const char *
 825 msg_cat(SCR *sp, const char *str, size_t *lenp)
 826 {
 827         GS *gp;
 828         DBT data, key;
 829         db_recno_t msgno;
 830
 831         /*
 832          * If it's not a catalog message, i.e. has doesn't have a leading
 833          * number and '|' symbol, we're done.
 834          */
 835         if (isdigit(str[0]) &&
 836             isdigit(str[1]) && isdigit(str[2]) && str[3] == '|') {
 837                 memset(&key, 0, sizeof(key));
 838                 key.data = &msgno;
 839                 key.size = sizeof(db_recno_t);
 840                 memset(&data, 0, sizeof(data));
 841                 msgno = atoi(str);
 842
 843                 /*
 844                  * XXX
 845                  * Really sleazy hack -- we put an extra character on the
 846                  * end of the format string, and then we change it to be
 847                  * the nul termination of the string.  There ought to be
 848                  * a better way.  Once we can allocate multiple temporary
 849                  * memory buffers, maybe we can use one of them instead.
 850                  */
 851                 gp = sp == NULL ? NULL : sp->gp;
 852                 if (gp != NULL && gp->msg != NULL &&
 853                     gp->msg->get(gp->msg, NULL, &key, &data, 0) == 0 &&
 854                     data.size != 0) {
 855                         if (lenp != NULL)
 856                                 *lenp = data.size - 1;
 857                         ((char *)data.data)[data.size - 1] = '\0';
 858                         return (data.data);
 859                 }
 860                 str = &str[4];
 861         }
 862         if (lenp != NULL)
 863                 *lenp = strlen(str);
 864         return (str);
 865 }
 866
 867 /*
 868  * msg_print --
 869  *      Return a printable version of a string, in allocated memory.
 870  *
 871  * PUBLIC: char *msg_print __P((SCR *, const char *, int *));
 872  */
 873 char *
 874 msg_print(SCR *sp, const char *s, int *needfree)
 875 {
 876         size_t blen, nlen;
 877         const char *cp;
 878         char *bp, *ep, *p, *t;
 879
 880         *needfree = 0;
 881
 882         for (cp = s; *cp != '\0'; ++cp)
 883                 if (!isprint(*cp))
 884                         break;
 885         if (*cp == '\0')
 886                 return ((char *)s);     /* SAFE: needfree set to 0. */
 887
 888         nlen = 0;
 889         if (0) {
 890 retry:          if (sp == NULL)
 891                         free(bp);
 892                 else
 893                         FREE_SPACE(sp, bp, blen);
 894                 *needfree = 0;
 895         }
 896         nlen += 256;
 897         if (sp == NULL) {
 898                 if ((bp = malloc(nlen)) == NULL)
 899                         goto alloc_err;
 900         } else
 901                 GET_SPACE_GOTO(sp, bp, blen, nlen);
 902         if (0) {
 903 alloc_err:      return ("");
 904         }
 905         *needfree = 1;
 906
 907         for (p = bp, ep = (bp + blen) - 1, cp = s; *cp != '\0' && p < ep; ++cp)
 908                 for (t = KEY_NAME(sp, *cp); *t != '\0' && p < ep; *p++ = *t++);
 909         if (p == ep)
 910                 goto retry;
 911         *p = '\0';
 912         return (bp);
 913 }