[BACK]Return to lam.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / lam

Annotation of src/usr.bin/lam/lam.c, Revision 1.20

1.20    ! schwarze    1: /*     $OpenBSD: lam.c,v 1.19 2015/10/09 01:37:08 deraadt Exp $        */
1.1       deraadt     2: /*     $NetBSD: lam.c,v 1.2 1994/11/14 20:27:42 jtc Exp $      */
                      3:
                      4: /*-
                      5:  * Copyright (c) 1993
                      6:  *     The Regents of the University of California.  All rights reserved.
                      7:  *
                      8:  * Redistribution and use in source and binary forms, with or without
                      9:  * modification, are permitted provided that the following conditions
                     10:  * are met:
                     11:  * 1. Redistributions of source code must retain the above copyright
                     12:  *    notice, this list of conditions and the following disclaimer.
                     13:  * 2. Redistributions in binary form must reproduce the above copyright
                     14:  *    notice, this list of conditions and the following disclaimer in the
                     15:  *    documentation and/or other materials provided with the distribution.
1.8       millert    16:  * 3. Neither the name of the University nor the names of its contributors
1.1       deraadt    17:  *    may be used to endorse or promote products derived from this software
                     18:  *    without specific prior written permission.
                     19:  *
                     20:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     21:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     22:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     23:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     24:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     25:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     26:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     27:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     28:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     29:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     30:  * SUCH DAMAGE.
                     31:  */
                     32:
                     33: /*
                     34:  *     lam - laminate files
                     35:  *     Author:  John Kunze, UCB
                     36:  */
                     37:
1.17      deraadt    38: #include <sys/param.h> /* NOFILE_MAX */
1.13      ray        39:
1.11      millert    40: #include <ctype.h>
                     41: #include <err.h>
1.1       deraadt    42: #include <stdio.h>
                     43: #include <stdlib.h>
                     44: #include <string.h>
1.15      millert    45: #include <unistd.h>
1.1       deraadt    46:
                     47: #define        BIGBUFSIZ       5 * BUFSIZ
                     48:
                     49: struct openfile {              /* open file structure */
                     50:        FILE    *fp;            /* file pointer */
                     51:        short   eof;            /* eof flag */
                     52:        short   pad;            /* pad flag for missing columns */
                     53:        char    eol;            /* end of line character */
                     54:        char    *sepstring;     /* string to print before each line */
                     55:        char    *format;        /* printf(3) style string spec. */
1.13      ray        56: }      input[NOFILE_MAX + 1];  /* last one is for the last -s arg. */
                     57: #define INPUTSIZE sizeof(input) / sizeof(*input)
1.1       deraadt    58:
1.13      ray        59: int    numfiles;               /* number of open files */
1.1       deraadt    60: int    nofinalnl;              /* normally append \n to each output line */
                     61: char   line[BIGBUFSIZ];
                     62: char   *linep;
                     63:
1.11      millert    64: void    usage(void);
1.4       millert    65: char   *gatherline(struct openfile *);
1.11      millert    66: void    getargs(int, char *[]);
1.4       millert    67: char   *pad(struct openfile *);
1.1       deraadt    68:
                     69: int
1.9       deraadt    70: main(int argc, char *argv[])
1.1       deraadt    71: {
1.13      ray        72:        int i;
1.18      deraadt    73:
1.19      deraadt    74:        if (pledge("stdio rpath", NULL) == -1)
                     75:                err(1, "pledge");
1.1       deraadt    76:
1.13      ray        77:        /* Process arguments, set numfiles to file argument count. */
1.11      millert    78:        getargs(argc, argv);
1.13      ray        79:        if (numfiles == 0)
1.11      millert    80:                usage();
1.13      ray        81:        /* Concatenate lines from each file, then print. */
1.1       deraadt    82:        for (;;) {
                     83:                linep = line;
1.13      ray        84:                /*
                     85:                 * For each file that has a line to print, numfile is
                     86:                 * incremented.  Thus if numfiles is 0, we are done.
                     87:                 */
                     88:                numfiles = 0;
                     89:                for (i = 0; i < INPUTSIZE - 1 && input[i].fp != NULL; i++)
                     90:                        linep = gatherline(&input[i]);
                     91:                if (numfiles == 0)
1.1       deraadt    92:                        exit(0);
                     93:                fputs(line, stdout);
1.13      ray        94:                /* Print terminating -s argument. */
                     95:                fputs(input[i].sepstring, stdout);
1.1       deraadt    96:                if (!nofinalnl)
                     97:                        putchar('\n');
                     98:        }
                     99: }
                    100:
                    101: void
1.11      millert   102: getargs(int argc, char *argv[])
1.1       deraadt   103: {
1.11      millert   104:        struct openfile *ip = input;
1.3       mpech     105:        char *p;
1.11      millert   106:        int ch, P, S, F, T;
                    107:        size_t siz;
1.1       deraadt   108:
                    109:        P = S = F = T = 0;              /* capitalized options */
1.11      millert   110:        while (optind < argc) {
                    111:                switch (ch = getopt(argc, argv, "F:f:P:p:S:s:T:t:")) {
1.20    ! schwarze  112:                case 'P': case 'p':
        !           113:                        P = (ch == 'P');
        !           114:                        ip->pad = 1;
        !           115:                        /* FALLTHROUGH */
1.11      millert   116:                case 'F': case 'f':
                    117:                        F = (ch == 'F');
                    118:                        /* Validate format string argument. */
                    119:                        for (p = optarg; *p != '\0'; p++)
1.16      deraadt   120:                                if (!isdigit((unsigned char)*p) &&
                    121:                                    *p != '.' && *p != '-')
1.11      millert   122:                                        errx(1, "%s: invalid width specified",
                    123:                                             optarg);
                    124:                        /* '%' + width + 's' + '\0' */
                    125:                        siz = p - optarg + 3;
                    126:                        if ((p = realloc(ip->format, siz)) == NULL)
                    127:                                err(1, NULL);
                    128:                        snprintf(p, siz, "%%%ss", optarg);
                    129:                        ip->format = p;
                    130:                        break;
                    131:                case 'S': case 's':
                    132:                        S = (ch == 'S');
                    133:                        ip->sepstring = optarg;
                    134:                        break;
                    135:                case 'T': case 't':
                    136:                        T = (ch == 'T');
                    137:                        if (strlen(optarg) != 1)
                    138:                                usage();
                    139:                        ip->eol = optarg[0];
                    140:                        nofinalnl = 1;
                    141:                        break;
                    142:                case -1:
                    143:                        if (optind >= argc)
                    144:                                break;          /* to support "--" */
1.13      ray       145:                        /* This is a file, not a flag. */
                    146:                        ++numfiles;
                    147:                        if (numfiles >= INPUTSIZE)
                    148:                                errx(1, "too many files");
1.11      millert   149:                        if (strcmp(argv[optind], "-") == 0)
1.1       deraadt   150:                                ip->fp = stdin;
1.11      millert   151:                        else if ((ip->fp = fopen(argv[optind], "r")) == NULL)
                    152:                                err(1, "%s", argv[optind]);
1.1       deraadt   153:                        ip->pad = P;
1.11      millert   154:                        if (ip->sepstring == NULL)
                    155:                                ip->sepstring = S ? (ip-1)->sepstring : "";
                    156:                        if (ip->format == NULL)
                    157:                                ip->format = (P || F) ? (ip-1)->format : "%s";
                    158:                        if (ip->eol == '\0')
                    159:                                ip->eol = T ? (ip-1)->eol : '\n';
1.1       deraadt   160:                        ip++;
1.11      millert   161:                        optind++;
1.1       deraadt   162:                        break;
                    163:                default:
1.11      millert   164:                        usage();
                    165:                        /* NOTREACHED */
1.1       deraadt   166:                }
                    167:        }
                    168:        ip->fp = NULL;
1.11      millert   169:        if (ip->sepstring == NULL)
1.1       deraadt   170:                ip->sepstring = "";
                    171: }
                    172:
                    173: char *
1.9       deraadt   174: pad(struct openfile *ip)
1.1       deraadt   175: {
1.7       millert   176:        size_t n;
1.3       mpech     177:        char *lp = linep;
1.1       deraadt   178:
1.7       millert   179:        n = strlcpy(lp, ip->sepstring,  line + sizeof(line) - lp);
                    180:        lp += (n < line + sizeof(line) - lp) ? n : strlen(lp);
1.1       deraadt   181:        if (ip->pad) {
1.7       millert   182:                n = snprintf(lp, line + sizeof(line) - lp, ip->format, "");
1.12      deraadt   183:                if (n > 0)
                    184:                        lp += (n < line + sizeof(line) - lp) ? n : strlen(lp);
1.1       deraadt   185:        }
                    186:        return (lp);
                    187: }
                    188:
1.13      ray       189: /*
                    190:  * Grab line from file, appending to linep.  Increments numfiles if file
                    191:  * is still open.
                    192:  */
1.1       deraadt   193: char *
1.9       deraadt   194: gatherline(struct openfile *ip)
1.1       deraadt   195: {
1.7       millert   196:        size_t n;
1.1       deraadt   197:        char s[BUFSIZ];
1.3       mpech     198:        char *p;
                    199:        char *lp = linep;
1.11      millert   200:        char *end = s + BUFSIZ - 1;
1.7       millert   201:        int c;
1.1       deraadt   202:
                    203:        if (ip->eof)
                    204:                return (pad(ip));
                    205:        for (p = s; (c = fgetc(ip->fp)) != EOF && p < end; p++)
                    206:                if ((*p = c) == ip->eol)
                    207:                        break;
                    208:        *p = '\0';
                    209:        if (c == EOF) {
                    210:                ip->eof = 1;
                    211:                if (ip->fp == stdin)
                    212:                        fclose(stdin);
                    213:                return (pad(ip));
                    214:        }
1.13      ray       215:        /* Something will be printed. */
                    216:        numfiles++;
1.7       millert   217:        n = strlcpy(lp, ip->sepstring, line + sizeof(line) - lp);
                    218:        lp += (n < line + sizeof(line) - lp) ? n : strlen(lp);
1.11      millert   219:        n = snprintf(lp, line + sizeof(line) - lp, ip->format, s);
1.12      deraadt   220:        if (n > 0)
                    221:                lp += (n < line + sizeof(line) - lp) ? n : strlen(lp);
1.1       deraadt   222:        return (lp);
                    223: }
                    224:
                    225: void
1.11      millert   226: usage(void)
1.1       deraadt   227: {
1.10      mickey    228:        extern char *__progname;
1.11      millert   229:
1.1       deraadt   230:        fprintf(stderr,
1.11      millert   231:            "usage: %s [-f min.max] [-p min.max] [-s sepstring] [-t c] file ...\n",
1.10      mickey    232:            __progname);
1.1       deraadt   233:        exit(1);
                    234: }