Home | History | Annotate | Line # | Download | only in sort
init.c revision 1.17.16.1
      1 /*	$NetBSD: init.c,v 1.17.16.1 2008/05/18 12:36:10 yamt Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2000-2003 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Ben Harris and Jaromir Dolecek.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 /*-
     33  * Copyright (c) 1993
     34  *	The Regents of the University of California.  All rights reserved.
     35  *
     36  * This code is derived from software contributed to Berkeley by
     37  * Peter McIlroy.
     38  *
     39  * Redistribution and use in source and binary forms, with or without
     40  * modification, are permitted provided that the following conditions
     41  * are met:
     42  * 1. Redistributions of source code must retain the above copyright
     43  *    notice, this list of conditions and the following disclaimer.
     44  * 2. Redistributions in binary form must reproduce the above copyright
     45  *    notice, this list of conditions and the following disclaimer in the
     46  *    documentation and/or other materials provided with the distribution.
     47  * 3. Neither the name of the University nor the names of its contributors
     48  *    may be used to endorse or promote products derived from this software
     49  *    without specific prior written permission.
     50  *
     51  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     52  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     53  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     54  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     55  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     56  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     57  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     58  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     59  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     60  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     61  * SUCH DAMAGE.
     62  */
     63 
     64 #include "sort.h"
     65 
     66 #ifndef lint
     67 __RCSID("$NetBSD: init.c,v 1.17.16.1 2008/05/18 12:36:10 yamt Exp $");
     68 __SCCSID("@(#)init.c	8.1 (Berkeley) 6/6/93");
     69 #endif /* not lint */
     70 
     71 #include <ctype.h>
     72 #include <string.h>
     73 
     74 static void insertcol __P((struct field *));
     75 static const char *setcolumn __P((const char *, struct field *, int));
     76 
     77 u_char gweights[NBINS];
     78 
     79 /*
     80  * masks of ignored characters.  Alltable is 256 ones.
     81  */
     82 static u_char alltable[NBINS], dtable[NBINS], itable[NBINS];
     83 
     84 /*
     85  * parsed key options
     86  */
     87 struct coldesc *clist = NULL;
     88 int ncols = 0;
     89 
     90 /*
     91  * clist (list of columns which correspond to one or more icol or tcol)
     92  * is in increasing order of columns.
     93  * Fields are kept in increasing order of fields.
     94  */
     95 
     96 /*
     97  * keep clist in order--inserts a column in a sorted array
     98  */
     99 static void
    100 insertcol(field)
    101 	struct field *field;
    102 {
    103 	int i;
    104 	struct coldesc *p;
    105 
    106 	/* Make space for new item */
    107 	p = realloc(clist, (ncols + 2) * sizeof(*clist));
    108 	if (!p)
    109 		err(1, "realloc");
    110 	clist = p;
    111 	memset(&clist[ncols], 0, sizeof(clist[ncols]));
    112 
    113 	for (i = 0; i < ncols; i++)
    114 		if (field->icol.num <= clist[i].num)
    115 			break;
    116 	if (field->icol.num != clist[i].num) {
    117 		memmove(clist+i+1, clist+i, sizeof(COLDESC)*(ncols-i));
    118 		clist[i].num = field->icol.num;
    119 		ncols++;
    120 	}
    121 	if (field->tcol.num && field->tcol.num != field->icol.num) {
    122 		for (i = 0; i < ncols; i++)
    123 			if (field->tcol.num <= clist[i].num)
    124 				break;
    125 		if (field->tcol.num != clist[i].num) {
    126 			memmove(clist+i+1, clist+i,sizeof(COLDESC)*(ncols-i));
    127 			clist[i].num = field->tcol.num;
    128 			ncols++;
    129 		}
    130 	}
    131 }
    132 
    133 /*
    134  * matches fields with the appropriate columns--n^2 but who cares?
    135  */
    136 void
    137 fldreset(fldtab)
    138 	struct field *fldtab;
    139 {
    140 	int i;
    141 
    142 	fldtab[0].tcol.p = clist + ncols - 1;
    143 	for (++fldtab; fldtab->icol.num; ++fldtab) {
    144 		for (i = 0; fldtab->icol.num != clist[i].num; i++)
    145 			;
    146 		fldtab->icol.p = clist + i;
    147 		if (!fldtab->tcol.num)
    148 			continue;
    149 		for (i = 0; fldtab->tcol.num != clist[i].num; i++)
    150 			;
    151 		fldtab->tcol.p = clist + i;
    152 	}
    153 }
    154 
    155 /*
    156  * interprets a column in a -k field
    157  */
    158 static const char *
    159 setcolumn(pos, cur_fld, gflag)
    160 	const char *pos;
    161 	struct field *cur_fld;
    162 	int gflag;
    163 {
    164 	struct column *col;
    165 	char *npos;
    166 	int tmp;
    167 	col = cur_fld->icol.num ? (&cur_fld->tcol) : (&cur_fld->icol);
    168 	col->num = (int) strtol(pos, &npos, 10);
    169 	pos = npos;
    170 	if (col->num <= 0 && !(col->num == 0 && col == &(cur_fld->tcol)))
    171 		errx(2, "field numbers must be positive");
    172 	if (*pos == '.') {
    173 		if (!col->num)
    174 			errx(2, "cannot indent end of line");
    175 		++pos;
    176 		col->indent = (int) strtol(pos, &npos, 10);
    177 		pos = npos;
    178 		if (&cur_fld->icol == col)
    179 			col->indent--;
    180 		if (col->indent < 0)
    181 			errx(2, "illegal offset");
    182 	}
    183 	for(; (tmp = optval(*pos, cur_fld->tcol.num)); pos++)
    184 		cur_fld->flags |= tmp;
    185 	if (cur_fld->icol.num == 0)
    186 		cur_fld->icol.num = 1;
    187 	return (pos);
    188 }
    189 
    190 int
    191 setfield(pos, cur_fld, gflag)
    192 	const char *pos;
    193 	struct field *cur_fld;
    194 	int gflag;
    195 {
    196 	int tmp;
    197 
    198 	cur_fld->weights = ascii;
    199 	cur_fld->mask = alltable;
    200 
    201 	pos = setcolumn(pos, cur_fld, gflag);
    202 	if (*pos == '\0')			/* key extends to EOL. */
    203 		cur_fld->tcol.num = 0;
    204 	else {
    205 		if (*pos != ',')
    206 			errx(2, "illegal field descriptor");
    207 		setcolumn((++pos), cur_fld, gflag);
    208 	}
    209 	if (!cur_fld->flags)
    210 		cur_fld->flags = gflag;
    211 	tmp = cur_fld->flags;
    212 
    213 	/*
    214 	 * Assign appropriate mask table and weight table.
    215 	 * If the global weights are reversed, the local field
    216 	 * must be "re-reversed".
    217 	 */
    218 	if (((tmp & R) ^ (gflag & R)) && (tmp & F))
    219 		cur_fld->weights = RFtable;
    220 	else if (tmp & F)
    221 		cur_fld->weights = Ftable;
    222 	else if ((tmp & R) ^ (gflag & R))
    223 		cur_fld->weights = Rascii;
    224 
    225 	if (tmp & I)
    226 		cur_fld->mask = itable;
    227 	else if (tmp & D)
    228 		cur_fld->mask = dtable;
    229 
    230 	cur_fld->flags |= (gflag & (BI | BT));
    231 	if (!cur_fld->tcol.indent)	/* BT has no meaning at end of field */
    232 		cur_fld->flags &= ~BT;
    233 
    234 	if (cur_fld->tcol.num && !(!(cur_fld->flags & BI)
    235 	    && cur_fld->flags & BT) && (cur_fld->tcol.num <= cur_fld->icol.num
    236 	    && cur_fld->tcol.indent != 0 /* == 0 -> end of field, i.e. okay */
    237 	    && cur_fld->tcol.indent < cur_fld->icol.indent))
    238 		errx(2, "fields out of order");
    239 	insertcol(cur_fld);
    240 	return (cur_fld->tcol.num);
    241 }
    242 
    243 int
    244 optval(desc, tcolflag)
    245 	int desc, tcolflag;
    246 {
    247 	switch(desc) {
    248 		case 'b':
    249 			if (!tcolflag)
    250 				return (BI);
    251 			else
    252 				return (BT);
    253 		case 'd': return (D);
    254 		case 'f': return (F);
    255 		case 'i': return (I);
    256 		case 'n': return (N);
    257 		case 'r': return (R);
    258 		default:  return (0);
    259 	}
    260 }
    261 
    262 /*
    263  * Replace historic +SPEC arguments with appropriate -kSPEC.
    264  */
    265 void
    266 fixit(argc, argv)
    267 	int *argc;
    268 	char **argv;
    269 {
    270 	int i, j, fplus=0;
    271 	char *vpos, *tpos, spec[20];
    272 	int col, indent;
    273 	size_t sz;
    274 
    275 	for (i = 1; i < *argc; i++) {
    276 		if (argv[i][0] != '+' && !fplus)
    277 			continue;
    278 
    279 		if (fplus && (argv[i][0] != '-' || !isdigit((unsigned char)argv[i][1]))) {
    280 			fplus = 0;
    281 			if (argv[i][0] != '+') {
    282 				/* not a -POS argument, skip */
    283 				continue;
    284 			}
    285 		}
    286 
    287 		/* parse spec */
    288 		tpos = argv[i]+1;
    289 		col = (int)strtol(tpos, &tpos, 10);
    290 		if (*tpos == '.') {
    291 			++tpos;
    292 			indent = (int) strtol(tpos, &tpos, 10);
    293 		} else
    294 			indent = 0;
    295 		/* tpos points to optional flags now */
    296 
    297 		/*
    298 		 * For x.y, the obsolescent variant assumed 0 == beginning
    299 		 * of line, while the new form uses 0 == end of line.
    300 		 * Convert accordingly.
    301 		 */
    302 		if (!fplus) {
    303 			/* +POS */
    304 			col += 1;
    305 			indent += 1;
    306 		} else {
    307 			/* -POS */
    308 			if (indent > 0)
    309 				col += 1;
    310 		}
    311 
    312 		/* new style spec */
    313 		sz = snprintf(spec, sizeof(spec), "%d.%d%s", col, indent,
    314 		    tpos);
    315 
    316 		if (!fplus) {
    317 			/* Replace the +POS argument with new-style -kSPEC */
    318 			asprintf(&vpos, "-k%s", spec);
    319 			argv[i] = vpos;
    320 			fplus = 1;
    321 		} else {
    322 			/*
    323 			 * Append the spec to one previously generated from
    324 			 * +POS argument, and remove the argv element.
    325 			 */
    326 			asprintf(&vpos, "%s,%s", argv[i-1], spec);
    327 			free(argv[i-1]);
    328 			argv[i-1] = vpos;
    329 			for (j=i; j < *argc; j++)
    330 				argv[j] = argv[j+1];
    331 			*argc -= 1;
    332 			i--;
    333 		}
    334 	}
    335 }
    336 
    337 /*
    338  * ascii, Rascii, Ftable, and RFtable map
    339  * REC_D -> REC_D;  {not REC_D} -> {not REC_D}.
    340  * gweights maps REC_D -> (0 or 255); {not REC_D} -> {not gweights[REC_D]}.
    341  * Note: when sorting in forward order, to encode character zero in a key,
    342  * use \001\001; character 1 becomes \001\002.  In this case, character 0
    343  * is reserved for the field delimiter.  Analagously for -r (fld_d = 255).
    344  * Note: this is only good for ASCII sorting.  For different LC 's,
    345  * all bets are off.  See also num_init in number.c
    346  */
    347 void
    348 settables(gflags)
    349 	int gflags;
    350 {
    351 	u_char *wts;
    352 	int i, incr;
    353 	for (i=0; i < 256; i++) {
    354 		ascii[i] = i;
    355 		if (i > REC_D && i < 255 - REC_D+1)
    356 			Rascii[i] = 255 - i + 1;
    357 		else
    358 			Rascii[i] = 255 - i;
    359 		if (islower(i)) {
    360 			Ftable[i] = Ftable[toupper(i)];
    361 			RFtable[i] = RFtable[toupper(i)];
    362 		} else if (REC_D>= 'A' && REC_D < 'Z' && i < 'a' && i > REC_D) {
    363 			Ftable[i] = i + 1;
    364 			RFtable[i] = Rascii[i] - 1;
    365 		} else {
    366 			Ftable[i] = i;
    367 			RFtable[i] = Rascii[i];
    368 		}
    369 		alltable[i] = 1;
    370 
    371 		if (i == '\n' || isprint(i))
    372 			itable[i] = 1;
    373 		else
    374 			itable[i] = 0;
    375 
    376 		if (i == '\n' || i == '\t' || i == ' ' || isalnum(i))
    377 			dtable[i] = 1;
    378 		else
    379 			dtable[i] = 0;
    380 	}
    381 
    382 	Rascii[REC_D] = RFtable[REC_D] = REC_D;
    383 	if (isupper(REC_D))
    384 		Ftable[tolower(REC_D)]++;
    385 
    386 	if ((gflags & R) && !((gflags & F) && SINGL_FLD))
    387 		wts = Rascii;
    388 	else if (!((gflags & F) && SINGL_FLD))
    389 		wts = ascii;
    390 	else if (gflags & R)
    391 		wts = RFtable;
    392 	else
    393 		wts = Ftable;
    394 
    395 	memmove(gweights, wts, sizeof(gweights));
    396 	incr = (gflags & R) ? -1 : 1;
    397 	for (i = 0; i < REC_D; i++)
    398 		gweights[i] += incr;
    399 	gweights[REC_D] = ((gflags & R) ? 255 : 0);
    400 	if (SINGL_FLD && (gflags & F)) {
    401 		for (i = 0; i < REC_D; i++) {
    402 			ascii[i] += incr;
    403 			Rascii[i] += incr;
    404 		}
    405 		ascii[REC_D] = Rascii[REC_D] = gweights[REC_D];
    406 	}
    407 }
    408