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

Annotation of src/usr.bin/tr/str.c, Revision 1.8

1.8     ! millert     1: /*     $OpenBSD: str.c,v 1.7 2002/02/16 21:27:55 millert Exp $ */
1.1       deraadt     2: /*     $NetBSD: str.c,v 1.7 1995/08/31 22:13:47 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.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: #ifndef lint
                     34: #if 0
                     35: static char sccsid[] = "@(#)str.c      8.2 (Berkeley) 4/28/95";
                     36: #endif
1.8     ! millert    37: static char rcsid[] = "$OpenBSD: str.c,v 1.7 2002/02/16 21:27:55 millert Exp $";
1.1       deraadt    38: #endif /* not lint */
                     39:
                     40: #include <sys/cdefs.h>
                     41: #include <sys/types.h>
                     42:
                     43: #include <errno.h>
                     44: #include <stddef.h>
                     45: #include <stdio.h>
                     46: #include <stdlib.h>
                     47: #include <string.h>
                     48: #include <ctype.h>
1.4       mickey     49: #include <err.h>
1.1       deraadt    50:
                     51: #include "extern.h"
                     52:
1.7       millert    53: static int     backslash(STR *);
                     54: static int     bracket(STR *);
                     55: static int     c_class(const void *, const void *);
                     56: static void    genclass(STR *);
                     57: static void    genequiv(STR *);
                     58: static int     genrange(STR *);
                     59: static void    genseq(STR *);
1.1       deraadt    60:
                     61: int
                     62: next(s)
1.6       mpech      63:        STR *s;
1.1       deraadt    64: {
1.6       mpech      65:        int ch;
1.1       deraadt    66:
                     67:        switch (s->state) {
                     68:        case EOS:
                     69:                return (0);
                     70:        case INFINITE:
                     71:                return (1);
                     72:        case NORMAL:
                     73:                switch (ch = *s->str) {
                     74:                case '\0':
                     75:                        s->state = EOS;
                     76:                        return (0);
                     77:                case '\\':
                     78:                        s->lastch = backslash(s);
                     79:                        break;
                     80:                case '[':
                     81:                        if (bracket(s))
                     82:                                return (next(s));
                     83:                        /* FALLTHROUGH */
                     84:                default:
                     85:                        ++s->str;
                     86:                        s->lastch = ch;
                     87:                        break;
                     88:                }
                     89:
                     90:                /* We can start a range at any time. */
                     91:                if (s->str[0] == '-' && genrange(s))
                     92:                        return (next(s));
                     93:                return (1);
                     94:        case RANGE:
                     95:                if (s->cnt-- == 0) {
                     96:                        s->state = NORMAL;
                     97:                        return (next(s));
                     98:                }
                     99:                ++s->lastch;
                    100:                return (1);
                    101:        case SEQUENCE:
                    102:                if (s->cnt-- == 0) {
                    103:                        s->state = NORMAL;
                    104:                        return (next(s));
                    105:                }
                    106:                return (1);
                    107:        case SET:
                    108:                if ((s->lastch = s->set[s->cnt++]) == OOBCH) {
                    109:                        s->state = NORMAL;
                    110:                        return (next(s));
                    111:                }
                    112:                return (1);
1.4       mickey    113:        default:
                    114:                return 0;
1.1       deraadt   115:        }
                    116:        /* NOTREACHED */
                    117: }
                    118:
                    119: static int
                    120: bracket(s)
1.6       mpech     121:        STR *s;
1.1       deraadt   122: {
1.6       mpech     123:        char *p;
1.1       deraadt   124:
                    125:        switch (s->str[1]) {
                    126:        case ':':                               /* "[:class:]" */
                    127:                if ((p = strstr(s->str + 2, ":]")) == NULL)
                    128:                        return (0);
                    129:                *p = '\0';
                    130:                s->str += 2;
                    131:                genclass(s);
                    132:                s->str = p + 2;
                    133:                return (1);
                    134:        case '=':                               /* "[=equiv=]" */
                    135:                if ((p = strstr(s->str + 2, "=]")) == NULL)
                    136:                        return (0);
                    137:                s->str += 2;
                    138:                genequiv(s);
                    139:                return (1);
                    140:        default:                                /* "[\###*n]" or "[#*n]" */
                    141:                if ((p = strpbrk(s->str + 2, "*]")) == NULL)
                    142:                        return (0);
1.3       millert   143:                if (p[0] != '*' || strchr(p, ']') == NULL)
1.1       deraadt   144:                        return (0);
                    145:                s->str += 1;
                    146:                genseq(s);
                    147:                return (1);
                    148:        }
                    149:        /* NOTREACHED */
                    150: }
                    151:
                    152: typedef struct {
                    153:        char *name;
1.7       millert   154:        int (*func)(int);
1.1       deraadt   155:        int *set;
                    156: } CLASS;
                    157:
                    158: static CLASS classes[] = {
                    159:        { "alnum",  isalnum,  },
                    160:        { "alpha",  isalpha,  },
                    161:        { "blank",  isblank,  },
                    162:        { "cntrl",  iscntrl,  },
                    163:        { "digit",  isdigit,  },
                    164:        { "graph",  isgraph,  },
                    165:        { "lower",  islower,  },
1.5       deraadt   166:        { "print",  isprint,  },
1.1       deraadt   167:        { "punct",  ispunct,  },
                    168:        { "space",  isspace,  },
                    169:        { "upper",  isupper,  },
                    170:        { "xdigit", isxdigit, },
                    171: };
                    172:
                    173: static void
                    174: genclass(s)
                    175:        STR *s;
                    176: {
1.7       millert   177:        int cnt, (*func)(int);
1.1       deraadt   178:        CLASS *cp, tmp;
                    179:        int *p;
                    180:
                    181:        tmp.name = s->str;
                    182:        if ((cp = (CLASS *)bsearch(&tmp, classes, sizeof(classes) /
                    183:            sizeof(CLASS), sizeof(CLASS), c_class)) == NULL)
1.4       mickey    184:                errx(1, "unknown class %s", s->str);
1.1       deraadt   185:
                    186:        if ((cp->set = p = malloc((NCHARS + 1) * sizeof(int))) == NULL)
1.4       mickey    187:                errx(1, "no memory for a class");
1.1       deraadt   188:        bzero(p, (NCHARS + 1) * sizeof(int));
                    189:        for (cnt = 0, func = cp->func; cnt < NCHARS; ++cnt)
                    190:                if ((func)(cnt))
                    191:                        *p++ = cnt;
                    192:        *p = OOBCH;
                    193:
                    194:        s->cnt = 0;
                    195:        s->state = SET;
                    196:        s->set = cp->set;
                    197: }
                    198:
                    199: static int
                    200: c_class(a, b)
                    201:        const void *a, *b;
                    202: {
                    203:        return (strcmp(((CLASS *)a)->name, ((CLASS *)b)->name));
                    204: }
                    205:
                    206: /*
                    207:  * English doesn't have any equivalence classes, so for now
                    208:  * we just syntax check and grab the character.
                    209:  */
                    210: static void
                    211: genequiv(s)
                    212:        STR *s;
                    213: {
                    214:        if (*s->str == '\\') {
                    215:                s->equiv[0] = backslash(s);
                    216:                if (*s->str != '=')
1.4       mickey    217:                        errx(1, "misplaced equivalence equals sign");
1.1       deraadt   218:        } else {
                    219:                s->equiv[0] = s->str[0];
                    220:                if (s->str[1] != '=')
1.4       mickey    221:                        errx(1, "misplaced equivalence equals sign");
1.1       deraadt   222:        }
                    223:        s->str += 2;
                    224:        s->cnt = 0;
                    225:        s->state = SET;
                    226:        s->set = s->equiv;
                    227: }
                    228:
                    229: static int
                    230: genrange(s)
                    231:        STR *s;
                    232: {
                    233:        int stopval;
                    234:        char *savestart;
                    235:
                    236:        savestart = s->str;
                    237:        stopval = *++s->str == '\\' ? backslash(s) : *s->str++;
                    238:        if (stopval < (u_char)s->lastch) {
                    239:                s->str = savestart;
                    240:                return (0);
                    241:        }
                    242:        s->cnt = stopval - s->lastch + 1;
                    243:        s->state = RANGE;
                    244:        --s->lastch;
                    245:        return (1);
                    246: }
                    247:
                    248: static void
                    249: genseq(s)
                    250:        STR *s;
                    251: {
                    252:        char *ep;
                    253:
                    254:        if (s->which == STRING1)
1.4       mickey    255:                errx(1, "sequences only valid in string2");
1.1       deraadt   256:
                    257:        if (*s->str == '\\')
                    258:                s->lastch = backslash(s);
                    259:        else
                    260:                s->lastch = *s->str++;
                    261:        if (*s->str != '*')
1.4       mickey    262:                errx(1, "misplaced sequence asterisk");
1.1       deraadt   263:
                    264:        switch (*++s->str) {
                    265:        case '\\':
                    266:                s->cnt = backslash(s);
                    267:                break;
                    268:        case ']':
                    269:                s->cnt = 0;
                    270:                ++s->str;
                    271:                break;
                    272:        default:
                    273:                if (isdigit(*s->str)) {
                    274:                        s->cnt = strtol(s->str, &ep, 0);
                    275:                        if (*ep == ']') {
                    276:                                s->str = ep + 1;
                    277:                                break;
                    278:                        }
                    279:                }
1.4       mickey    280:                errx(1, "illegal sequence count");
1.1       deraadt   281:                /* NOTREACHED */
                    282:        }
                    283:
                    284:        s->state = s->cnt ? SEQUENCE : INFINITE;
                    285: }
                    286:
                    287: /*
                    288:  * Translate \??? into a character.  Up to 3 octal digits, if no digits either
                    289:  * an escape code or a literal character.
                    290:  */
                    291: static int
                    292: backslash(s)
1.6       mpech     293:        STR *s;
1.1       deraadt   294: {
1.6       mpech     295:        int ch, cnt, val;
1.1       deraadt   296:
                    297:        for (cnt = val = 0;;) {
                    298:                ch = *++s->str;
                    299:                if (!isascii(ch) || !isdigit(ch))
                    300:                        break;
                    301:                val = val * 8 + ch - '0';
                    302:                if (++cnt == 3) {
                    303:                        ++s->str;
                    304:                        break;
                    305:                }
                    306:        }
                    307:        if (cnt)
                    308:                return (val);
                    309:        if (ch != '\0')
                    310:                ++s->str;
                    311:        switch (ch) {
                    312:                case 'a':                       /* escape characters */
                    313:                        return ('\7');
                    314:                case 'b':
                    315:                        return ('\b');
                    316:                case 'f':
                    317:                        return ('\f');
                    318:                case 'n':
                    319:                        return ('\n');
                    320:                case 'r':
                    321:                        return ('\r');
                    322:                case 't':
                    323:                        return ('\t');
                    324:                case 'v':
                    325:                        return ('\13');
                    326:                case '\0':                      /*  \" -> \ */
                    327:                        s->state = EOS;
                    328:                        return ('\\');
                    329:                default:                        /* \x" -> x */
                    330:                        return (ch);
                    331:        }
                    332: }