X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/47bf6339eb158ba9884befc0c8dab46acb1c0100..63621a81d16ab743de6b57d47578a9a2c670ad22:/uip/scansbr.c diff --git a/uip/scansbr.c b/uip/scansbr.c index bdf0de33..457379cd 100644 --- a/uip/scansbr.c +++ b/uip/scansbr.c @@ -1,6 +1,4 @@ - -/* - * scansbr.c -- routines to help scan along... +/* scansbr.c -- routines to help scan along... * * This code is Copyright (c) 2002, by the authors of nmh. See the * COPYRIGHT file in the root directory of the nmh distribution for @@ -13,19 +11,7 @@ #include #include #include - -m_getfld_state_t gstate; - -#define MAXSCANL 256 /* longest possible scan line */ - -/* - * Buffer size for content part of header fields. We want this - * to be large enough so that we don't do a lot of extra FLDPLUS - * calls on m_getfld but small enough so that we don't snarf - * the entire message body when we're only going to display 30 - * characters of it. - */ -#define SBUFSIZ 512 +#include "sbr/terminal.h" static struct format *fmt; static struct comp *datecomp; /* pntr to "date" comp */ @@ -37,33 +23,36 @@ static struct comp **used_buf = 0; /* stack for comp that use buffers */ static int dat[5]; /* aux. data for format routine */ -char *scanl = 0; /* text of most recent scanline */ +static m_getfld_state_t gstate; /* for accessor functions below */ #define DIEWRERR() adios (scnmsg, "write error on") +#define PUTC(c) \ + if (putc((c), scnout) == EOF) \ + DIEWRERR(); + #define FPUTS(buf) {\ - if (mh_fputs(buf,scnout) == EOF)\ + if (fputs(buf,scnout) == EOF)\ DIEWRERR();\ } -/* - * prototypes - */ -static int mh_fputs(char *, FILE *); - -#ifdef MULTIBYTE_SUPPORT -#define SCAN_CHARWIDTH MB_CUR_MAX -#else -#define SCAN_CHARWIDTH 1 -#endif +/* outnum determines how the input from inb is copied. If positive then + * it is the number of the message to create, e.g. for inc(1), and all + * of the email is copied into that message, with some tweaks. If 0, + * e.g. `scan 42', then reading inb can dubiously stop after a whole + * buffer of body, even though this might not be enough to fulfill the + * scan format and width. Or if -1 then no copy is being created, but + * all of inb must be read because the next message must be found, e.g. + * `scan -file foo.mbox'. */ int scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, - int unseen, char *folder, long size, int noisy) + int unseen, char *folder, long size, int noisy, charstring_t *scanl) { + static bool deja_vu; + static int tty_width; int i, compnum, encrypted, state; - unsigned char *cp, *tmpbuf, *startbody; - char **nxtbuf; + char *cp, *tmpbuf, *startbody, **nxtbuf; char *saved_c_text = NULL; struct comp *cptr; struct comp **savecomp; @@ -72,21 +61,24 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, char name[NAMESZ]; int bufsz; static int rlwidth, slwidth; - static size_t scanl_size; - - /* first-time only initialization */ - if (!scanl) { - if (width == 0) { - if ((width = sc_width ()) < WIDTH/2) - width = WIDTH/2; - else if (width > MAXSCANL) - width = MAXSCANL; + + /* first-time only initialization, which will always happen the + way the code is now, with callers initializing *scanl to NULL. + scanl used to be a global. */ + if (! *scanl) { + if (width == -1) { + if (!deja_vu) { + deja_vu = true; + tty_width = sc_width(); + } + + width = max(tty_width, WIDTH / 2); + } else if (width == 0) { + /* Unlimited width. */ + width = INT_MAX; } dat[3] = slwidth = width; - /* Arbitrarily allocate 20 * slwidth to provide room for lots - of escape sequences. */ - scanl_size = SCAN_CHARWIDTH * (20 * slwidth + 2); - scanl = (char *) mh_xmalloc (scanl_size); + *scanl = charstring_create (min(width, NMH_BUFSIZ)); if (outnum) umask(~m_gmprot()); @@ -97,7 +89,7 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, datecomp = fmt_findcomp("date"); cptr = fmt_findcomp("folder"); if (cptr && folder) - cptr->c_text = getcpy(folder); + cptr->c_text = mh_xstrdup(folder); if (fmt_addcompentry("encrypted")) { ncomps++; } @@ -137,15 +129,10 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, * and it's our responsibility to free it. */ - nxtbuf = compbuffers = (char **) calloc((size_t) ncomps, sizeof(char *)); - if (nxtbuf == NULL) - adios (NULL, "unable to allocate component buffers"); - used_buf = (struct comp **) calloc((size_t) (ncomps+1), - sizeof(struct comp *)); - if (used_buf == NULL) - adios (NULL, "unable to allocate component buffer stack"); + nxtbuf = compbuffers = mh_xcalloc(ncomps, sizeof *nxtbuf); + used_buf = mh_xcalloc(ncomps + 1, sizeof *used_buf); used_buf += ncomps+1; *--used_buf = 0; - rlwidth = bodycomp && (width > SBUFSIZ) ? width : SBUFSIZ; + rlwidth = NMH_BUFSIZ; for (i = ncomps; i--; ) *nxtbuf++ = mh_xmalloc(rlwidth); } @@ -171,21 +158,16 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, if (ferror(inb)) { advise("read", "unable to"); /* "read error" */ return SCNFAT; - } else { - return SCNEOF; } + return SCNEOF; } - if (outnum) { - if (outnum > 0) { - scnmsg = m_name (outnum); - if (*scnmsg == '?') /* msg num out of range */ - return SCNNUM; - } else { - scnmsg = "/dev/null"; - } - if ((scnout = fopen (scnmsg, "w")) == NULL) - adios (scnmsg, "unable to write"); + if (outnum > 0) { + scnmsg = m_name (outnum); + if (*scnmsg == '?') /* msg num out of range */ + return SCNNUM; + if ((scnout = fopen (scnmsg, "w")) == NULL) + adios (scnmsg, "unable to write"); } /* scan - main loop */ @@ -195,9 +177,9 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, case FLD: case FLDPLUS: compnum++; - if (outnum) { + if (scnout) { FPUTS (name); - if ( putc (':', scnout) == EOF) DIEWRERR(); + PUTC(':'); FPUTS (tmpbuf); } /* @@ -211,7 +193,7 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, cptr->c_text = tmpbuf; for (cp = tmpbuf + strlen (tmpbuf) - 1; cp >= tmpbuf; cp--) - if (isspace (*cp)) + if (isspace ((unsigned char) *cp)) *cp = 0; else break; @@ -223,13 +205,12 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, while (state == FLDPLUS) { bufsz = rlwidth; state = m_getfld (&gstate, name, tmpbuf, &bufsz, inb); - if (outnum) + if (scnout) FPUTS (tmpbuf); } break; case BODY: - compnum = -1; /* * A slight hack ... if we have less than rlwidth characters * in the buffer, call m_getfld again. @@ -240,19 +221,21 @@ scan (FILE *inb, int innum, int outnum, char *nfs, int width, int curflg, state = m_getfld (&gstate, name, tmpbuf + i, &bufsz, inb); } - if (! outnum) { + if (outnum == 0) { state = FILEEOF; /* stop now if scan cmd */ if (bodycomp && startbody == NULL) startbody = tmpbuf; goto finished; } - if (putc ('\n', scnout) == EOF) DIEWRERR(); - FPUTS (tmpbuf); + if (scnout) { + PUTC('\n'); + FPUTS (tmpbuf); + } /* * The previous code here used to call m_getfld() using * pointers to the underlying output stdio buffers to * avoid the extra copy. Tests by Markus Schnalke show - * no noticable performance loss on larger mailboxes + * no noticeable performance loss on larger mailboxes * if we incur an extra copy, and messing around with * internal stdio buffers is becoming more and more * unportable as times go on. So from now on just deal @@ -273,26 +256,29 @@ body:; while (state == BODY) { bufsz = rlwidth; state = m_getfld (&gstate, name, tmpbuf, &bufsz, inb); - FPUTS(tmpbuf); + if (scnout) + FPUTS(tmpbuf); } goto finished; case LENERR: case FMTERR: - fprintf (stderr, - innum ? "??Format error (message %d) in " - : "??Format error in ", - outnum ? outnum : innum); + if (innum) + fprintf (stderr, "??Format error (message %d) in ", + outnum ? outnum : innum); + else + fprintf (stderr, "??Format error in "); + fprintf (stderr, "component %d\n", compnum); - if (outnum) { + if (scnout) { FPUTS ("\n\nBAD MSG:\n"); FPUTS (name); - if (putc ('\n', scnout) == EOF) DIEWRERR(); + PUTC('\n'); state = BODY; goto body; } - /* fall through */ + goto finished; case FILEEOF: goto finished; @@ -319,7 +305,7 @@ finished: if (size) dat[2] = size; - else if (outnum > 0) + else if (scnout) { dat[2] = ftell(scnout); if (dat[2] == EOF) DIEWRERR(); @@ -334,10 +320,7 @@ finished: if (datecomp) { if (! datecomp->c_text) { if (datecomp->c_tws == NULL) - datecomp->c_tws = (struct tws *) - calloc((size_t) 1, sizeof(*datecomp->c_tws)); - if (datecomp->c_tws == NULL) - adios (NULL, "unable to allocate tws buffer"); + NEW0(datecomp->c_tws); *datecomp->c_tws = *dlocaltime ((time_t *) &st.st_mtime); datecomp->c_flags |= CF_DATEFAB|CF_TRUE; } else { @@ -346,13 +329,13 @@ finished: } } - fmt_scan (fmt, scanl, scanl_size, slwidth, dat); + fmt_scan (fmt, *scanl, slwidth, dat, NULL); if (bodycomp) bodycomp->c_text = saved_c_text; if (noisy) - fputs (scanl, stdout); + fputs (charstring_buffer (*scanl), stdout); cptr = fmt_findcomp ("encrypted"); encrypted = cptr && cptr->c_text; @@ -362,21 +345,20 @@ finished: cptr->c_text = NULL; } - if (outnum && (ferror(scnout) || fclose (scnout) == EOF)) + if (scnout && (ferror(scnout) || fclose (scnout) == EOF)) DIEWRERR(); return (state != FILEEOF ? SCNERR : encrypted ? SCNENC : SCNMSG); } -static int -mh_fputs(char *s, FILE *stream) -{ - char c; - - while ((c = *s++)) - if (putc (c,stream) == EOF ) - return(EOF); - return (0); +/* The following two functions allow access to the global gstate above. */ +void +scan_finished(void) { + m_getfld_state_destroy (&gstate); } +void +scan_detect_mbox_style (FILE *f) { + m_unknown (&gstate, f); +}