[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.13

1.13    ! schwarze    1: /*     $OpenBSD: colrm.c,v 1.12 2016/01/18 20:31:36 schwarze 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:
1.13    ! schwarze   58:        setlocale(LC_CTYPE, "");
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: }