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

Annotation of src/usr.bin/colrm/colrm.c, Revision 1.12

1.12    ! schwarze    1: /*     $OpenBSD: colrm.c,v 1.11 2015/10/09 01:37:06 deraadt Exp $      */
1.1       deraadt     2: /*     $NetBSD: colrm.c,v 1.4 1995/09/02 05:51:37 jtc Exp $    */
                      3:
                      4: /*-
                      5:  * Copyright (c) 1991, 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.7       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: #include <sys/types.h>
                     34:
                     35: #include <err.h>
                     36: #include <errno.h>
                     37: #include <limits.h>
1.12    ! schwarze   38: #include <locale.h>
1.1       deraadt    39: #include <stdio.h>
                     40: #include <stdlib.h>
                     41: #include <string.h>
                     42: #include <unistd.h>
1.12    ! schwarze   43: #include <wchar.h>
1.1       deraadt    44:
                     45: #define        TAB     8
                     46:
1.6       millert    47: void usage(void);
1.1       deraadt    48:
                     49: int
1.8       deraadt    50: main(int argc, char *argv[])
1.1       deraadt    51: {
1.12    ! schwarze   52:        char     *line, *p;
        !            53:        ssize_t   linesz;
        !            54:        wchar_t   wc;
        !            55:        u_long    column, newcol, start, stop;
        !            56:        int       ch, len, width;
        !            57:
        !            58:        setlocale(LC_ALL, "");
1.10      deraadt    59:
1.11      deraadt    60:        if (pledge("stdio", NULL) == -1)
                     61:                err(1, "pledge");
1.1       deraadt    62:
1.3       millert    63:        while ((ch = getopt(argc, argv, "")) != -1)
1.1       deraadt    64:                switch(ch) {
                     65:                case '?':
                     66:                default:
                     67:                        usage();
                     68:                }
                     69:        argc -= optind;
                     70:        argv += optind;
                     71:
                     72:        start = stop = 0;
                     73:        switch(argc) {
                     74:        case 2:
                     75:                stop = strtol(argv[1], &p, 10);
                     76:                if (stop <= 0 || *p)
                     77:                        errx(1, "illegal column -- %s", argv[1]);
                     78:                /* FALLTHROUGH */
                     79:        case 1:
                     80:                start = strtol(argv[0], &p, 10);
                     81:                if (start <= 0 || *p)
                     82:                        errx(1, "illegal column -- %s", argv[0]);
                     83:                break;
                     84:        case 0:
                     85:                break;
                     86:        default:
                     87:                usage();
                     88:        }
                     89:
                     90:        if (stop && start > stop)
                     91:                err(1, "illegal start and stop columns");
                     92:
1.12    ! schwarze   93:        line = NULL;
        !            94:        while (getline(&line, &linesz, stdin) != -1) {
        !            95:                column = 0;
        !            96:                width = 0;
        !            97:                for (p = line; *p != '\0'; p += len) {
        !            98:                        len = 1;
        !            99:                        switch (*p) {
        !           100:                        case '\n':
        !           101:                                putchar('\n');
        !           102:                                continue;
        !           103:                        case '\b':
        !           104:                                /*
        !           105:                                 * Pass it through if the previous character
        !           106:                                 * was in scope, still represented by the
        !           107:                                 * current value of "column".
        !           108:                                 * Allow an optional second backspace
        !           109:                                 * after a double-width character.
        !           110:                                 */
        !           111:                                if (start == 0 || column < start ||
        !           112:                                    (stop > 0 &&
        !           113:                                     column > stop + (width > 1))) {
        !           114:                                        putchar('\b');
        !           115:                                        if (width > 1 && p[1] == '\b')
        !           116:                                                putchar('\b');
        !           117:                                }
        !           118:                                if (width > 1 && p[1] == '\b')
        !           119:                                        p++;
        !           120:                                column -= width;
        !           121:                                continue;
        !           122:                        case '\t':
        !           123:                                newcol = (column + TAB) & ~(TAB - 1);
        !           124:                                if (start == 0 || newcol < start) {
        !           125:                                        putchar('\t');
        !           126:                                        column = newcol;
        !           127:                                } else
        !           128:                                        /*
        !           129:                                         * Expand tabs that intersect or
        !           130:                                         * follow deleted columns.
        !           131:                                         */
        !           132:                                        while (column < newcol)
        !           133:                                                if (++column < start ||
        !           134:                                                    (stop > 0 &&
        !           135:                                                     column > stop))
        !           136:                                                        putchar(' ');
        !           137:                                continue;
        !           138:                        default:
        !           139:                                break;
        !           140:                        }
        !           141:
        !           142:                        /*
        !           143:                         * Handle the three cases of invalid bytes,
        !           144:                         * non-printable, and printable characters.
        !           145:                         */
        !           146:
        !           147:                        if ((len = mbtowc(&wc, p, MB_CUR_MAX)) == -1) {
        !           148:                                (void)mbtowc(NULL, NULL, MB_CUR_MAX);
        !           149:                                len = 1;
        !           150:                                width = 1;
        !           151:                        } else if ((width = wcwidth(wc)) == -1)
        !           152:                                width = 1;
        !           153:
        !           154:                        /*
        !           155:                         * If the character completely fits before or
        !           156:                         * after the cut, keep it; otherwise, skip it.
        !           157:                         */
        !           158:
        !           159:                        if ((start == 0 || column + width < start ||
        !           160:                            (stop > 0 && column + (width > 0) > stop)))
        !           161:                                fwrite(p, 1, len, stdout);
        !           162:
        !           163:                        /*
        !           164:                         * If the cut cuts the character in half
        !           165:                         * and no backspace follows,
        !           166:                         * print a blank for correct columnation.
        !           167:                         */
        !           168:
        !           169:                        else if (width > 1 && p[len] != '\b' &&
        !           170:                            (start == 0 || column + 1 < start ||
        !           171:                            (stop > 0 && column + width > stop)))
        !           172:                                putchar(' ');
        !           173:
        !           174:                        column += width;
1.1       deraadt   175:                }
                    176:        }
1.12    ! schwarze  177:        if (ferror(stdin))
        !           178:                err(1, "stdin");
        !           179:        if (ferror(stdout))
        !           180:                err(1, "stdout");
        !           181:        return 0;
1.1       deraadt   182: }
                    183:
                    184: void
1.8       deraadt   185: usage(void)
1.1       deraadt   186: {
                    187:        (void)fprintf(stderr, "usage: colrm [start [stop]]\n");
                    188:        exit(1);
                    189: }