xref: /freebsd/bin/pax/sel_subs.c (revision ea825d02749f382c3f7e17f28247f20a48733eab)
1 /*-
2  * Copyright (c) 1992 Keith Muller.
3  * Copyright (c) 1992, 1993
4  *	The Regents of the University of California.  All rights reserved.
5  *
6  * This code is derived from software contributed to Berkeley by
7  * Keith Muller of the University of California, San Diego.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 
34 #ifndef lint
35 #if 0
36 static char sccsid[] = "@(#)sel_subs.c	8.1 (Berkeley) 5/31/93";
37 #endif
38 #endif /* not lint */
39 #include <sys/cdefs.h>
40 __FBSDID("$FreeBSD$");
41 
42 #include <sys/types.h>
43 #include <sys/time.h>
44 #include <sys/stat.h>
45 #include <pwd.h>
46 #include <grp.h>
47 #include <stdio.h>
48 #include <string.h>
49 #include <strings.h>
50 #include <stdlib.h>
51 #include "pax.h"
52 #include "sel_subs.h"
53 #include "extern.h"
54 
55 static int str_sec(char *, time_t *);
56 static int usr_match(ARCHD *);
57 static int grp_match(ARCHD *);
58 static int trng_match(ARCHD *);
59 
60 static TIME_RNG *trhead = NULL;		/* time range list head */
61 static TIME_RNG *trtail = NULL;		/* time range list tail */
62 static USRT **usrtb = NULL;		/* user selection table */
63 static GRPT **grptb = NULL;		/* group selection table */
64 
65 /*
66  * Routines for selection of archive members
67  */
68 
69 /*
70  * sel_chk()
71  *	check if this file matches a specified uid, gid or time range
72  * Return:
73  *	0 if this archive member should be processed, 1 if it should be skipped
74  */
75 
76 int
77 sel_chk(ARCHD *arcn)
78 {
79 	if (((usrtb != NULL) && usr_match(arcn)) ||
80 	    ((grptb != NULL) && grp_match(arcn)) ||
81 	    ((trhead != NULL) && trng_match(arcn)))
82 		return(1);
83 	return(0);
84 }
85 
86 /*
87  * User/group selection routines
88  *
89  * Routines to handle user selection of files based on the file uid/gid. To
90  * add an entry, the user supplies either then name or the uid/gid starting with
91  * a # on the command line. A \# will escape the #.
92  */
93 
94 /*
95  * usr_add()
96  *	add a user match to the user match hash table
97  * Return:
98  *	0 if added ok, -1 otherwise;
99  */
100 
101 int
102 usr_add(char *str)
103 {
104 	u_int indx;
105 	USRT *pt;
106 	struct passwd *pw;
107 	uid_t uid;
108 
109 	/*
110 	 * create the table if it doesn't exist
111 	 */
112 	if ((str == NULL) || (*str == '\0'))
113 		return(-1);
114 	if ((usrtb == NULL) &&
115  	    ((usrtb = (USRT **)calloc(USR_TB_SZ, sizeof(USRT *))) == NULL)) {
116 		paxwarn(1, "Unable to allocate memory for user selection table");
117 		return(-1);
118 	}
119 
120 	/*
121 	 * figure out user spec
122 	 */
123 	if (str[0] != '#') {
124 		/*
125 		 * it is a user name, \# escapes # as first char in user name
126 		 */
127 		if ((str[0] == '\\') && (str[1] == '#'))
128 			++str;
129 		if ((pw = getpwnam(str)) == NULL) {
130 			paxwarn(1, "Unable to find uid for user: %s", str);
131 			return(-1);
132 		}
133 		uid = (uid_t)pw->pw_uid;
134 	} else
135 #		ifdef NET2_STAT
136 		uid = (uid_t)atoi(str+1);
137 #		else
138 		uid = (uid_t)strtoul(str+1, NULL, 10);
139 #		endif
140 	endpwent();
141 
142 	/*
143 	 * hash it and go down the hash chain (if any) looking for it
144 	 */
145 	indx = ((unsigned)uid) % USR_TB_SZ;
146 	if ((pt = usrtb[indx]) != NULL) {
147 		while (pt != NULL) {
148 			if (pt->uid == uid)
149 				return(0);
150 			pt = pt->fow;
151 		}
152 	}
153 
154 	/*
155 	 * uid is not yet in the table, add it to the front of the chain
156 	 */
157 	if ((pt = (USRT *)malloc(sizeof(USRT))) != NULL) {
158 		pt->uid = uid;
159 		pt->fow = usrtb[indx];
160 		usrtb[indx] = pt;
161 		return(0);
162 	}
163 	paxwarn(1, "User selection table out of memory");
164 	return(-1);
165 }
166 
167 /*
168  * usr_match()
169  *	check if this files uid matches a selected uid.
170  * Return:
171  *	0 if this archive member should be processed, 1 if it should be skipped
172  */
173 
174 static int
175 usr_match(ARCHD *arcn)
176 {
177 	USRT *pt;
178 
179 	/*
180 	 * hash and look for it in the table
181 	 */
182 	pt = usrtb[((unsigned)arcn->sb.st_uid) % USR_TB_SZ];
183 	while (pt != NULL) {
184 		if (pt->uid == arcn->sb.st_uid)
185 			return(0);
186 		pt = pt->fow;
187 	}
188 
189 	/*
190 	 * not found
191 	 */
192 	return(1);
193 }
194 
195 /*
196  * grp_add()
197  *	add a group match to the group match hash table
198  * Return:
199  *	0 if added ok, -1 otherwise;
200  */
201 
202 int
203 grp_add(char *str)
204 {
205 	u_int indx;
206 	GRPT *pt;
207 	struct group *gr;
208 	gid_t gid;
209 
210 	/*
211 	 * create the table if it doesn't exist
212 	 */
213 	if ((str == NULL) || (*str == '\0'))
214 		return(-1);
215 	if ((grptb == NULL) &&
216  	    ((grptb = (GRPT **)calloc(GRP_TB_SZ, sizeof(GRPT *))) == NULL)) {
217 		paxwarn(1, "Unable to allocate memory fo group selection table");
218 		return(-1);
219 	}
220 
221 	/*
222 	 * figure out user spec
223 	 */
224 	if (str[0] != '#') {
225 		/*
226 		 * it is a group name, \# escapes # as first char in group name
227 		 */
228 		if ((str[0] == '\\') && (str[1] == '#'))
229 			++str;
230 		if ((gr = getgrnam(str)) == NULL) {
231 			paxwarn(1,"Cannot determine gid for group name: %s", str);
232 			return(-1);
233 		}
234 		gid = gr->gr_gid;
235 	} else
236 #		ifdef NET2_STAT
237 		gid = (gid_t)atoi(str+1);
238 #		else
239 		gid = (gid_t)strtoul(str+1, NULL, 10);
240 #		endif
241 	endgrent();
242 
243 	/*
244 	 * hash it and go down the hash chain (if any) looking for it
245 	 */
246 	indx = ((unsigned)gid) % GRP_TB_SZ;
247 	if ((pt = grptb[indx]) != NULL) {
248 		while (pt != NULL) {
249 			if (pt->gid == gid)
250 				return(0);
251 			pt = pt->fow;
252 		}
253 	}
254 
255 	/*
256 	 * gid not in the table, add it to the front of the chain
257 	 */
258 	if ((pt = (GRPT *)malloc(sizeof(GRPT))) != NULL) {
259 		pt->gid = gid;
260 		pt->fow = grptb[indx];
261 		grptb[indx] = pt;
262 		return(0);
263 	}
264 	paxwarn(1, "Group selection table out of memory");
265 	return(-1);
266 }
267 
268 /*
269  * grp_match()
270  *	check if this files gid matches a selected gid.
271  * Return:
272  *	0 if this archive member should be processed, 1 if it should be skipped
273  */
274 
275 static int
276 grp_match(ARCHD *arcn)
277 {
278 	GRPT *pt;
279 
280 	/*
281 	 * hash and look for it in the table
282 	 */
283 	pt = grptb[((unsigned)arcn->sb.st_gid) % GRP_TB_SZ];
284 	while (pt != NULL) {
285 		if (pt->gid == arcn->sb.st_gid)
286 			return(0);
287 		pt = pt->fow;
288 	}
289 
290 	/*
291 	 * not found
292 	 */
293 	return(1);
294 }
295 
296 /*
297  * Time range selection routines
298  *
299  * Routines to handle user selection of files based on the modification and/or
300  * inode change time falling within a specified time range (the non-standard
301  * -T flag). The user may specify any number of different file time ranges.
302  * Time ranges are checked one at a time until a match is found (if at all).
303  * If the file has a mtime (and/or ctime) which lies within one of the time
304  * ranges, the file is selected. Time ranges may have a lower and/or an upper
305  * value. These ranges are inclusive. When no time ranges are supplied to pax
306  * with the -T option, all members in the archive will be selected by the time
307  * range routines. When only a lower range is supplied, only files with a
308  * mtime (and/or ctime) equal to or younger are selected. When only an upper
309  * range is supplied, only files with a mtime (and/or ctime) equal to or older
310  * are selected. When the lower time range is equal to the upper time range,
311  * only files with a mtime (or ctime) of exactly that time are selected.
312  */
313 
314 /*
315  * trng_add()
316  *	add a time range match to the time range list.
317  *	This is a non-standard pax option. Lower and upper ranges are in the
318  *	format: [yy[mm[dd[hh]]]]mm[.ss] and are comma separated.
319  *	Time ranges are based on current time, so 1234 would specify a time of
320  *	12:34 today.
321  * Return:
322  *	0 if the time range was added to the list, -1 otherwise
323  */
324 
325 int
326 trng_add(char *str)
327 {
328 	TIME_RNG *pt;
329 	char *up_pt = NULL;
330 	char *stpt;
331 	char *flgpt;
332 	int dot = 0;
333 
334 	/*
335 	 * throw out the badly formed time ranges
336 	 */
337 	if ((str == NULL) || (*str == '\0')) {
338 		paxwarn(1, "Empty time range string");
339 		return(-1);
340 	}
341 
342 	/*
343 	 * locate optional flags suffix /{cm}.
344 	 */
345 	if ((flgpt = strrchr(str, '/')) != NULL)
346 		*flgpt++ = '\0';
347 
348 	for (stpt = str; *stpt != '\0'; ++stpt) {
349 		if ((*stpt >= '0') && (*stpt <= '9'))
350 			continue;
351 		if ((*stpt == ',') && (up_pt == NULL)) {
352 			*stpt = '\0';
353 			up_pt = stpt + 1;
354 			dot = 0;
355 			continue;
356 		}
357 
358 		/*
359 		 * allow only one dot per range (secs)
360 		 */
361 		if ((*stpt == '.') && (!dot)) {
362 			++dot;
363 			continue;
364 		}
365 		paxwarn(1, "Improperly specified time range: %s", str);
366 		goto out;
367 	}
368 
369 	/*
370 	 * allocate space for the time range and store the limits
371 	 */
372 	if ((pt = (TIME_RNG *)malloc(sizeof(TIME_RNG))) == NULL) {
373 		paxwarn(1, "Unable to allocate memory for time range");
374 		return(-1);
375 	}
376 
377 	/*
378 	 * by default we only will check file mtime, but the user can specify
379 	 * mtime, ctime (inode change time) or both.
380 	 */
381 	if ((flgpt == NULL) || (*flgpt == '\0'))
382 		pt->flgs = CMPMTME;
383 	else {
384 		pt->flgs = 0;
385 		while (*flgpt != '\0') {
386 			switch(*flgpt) {
387 			case 'M':
388 			case 'm':
389 				pt->flgs |= CMPMTME;
390 				break;
391 			case 'C':
392 			case 'c':
393 				pt->flgs |= CMPCTME;
394 				break;
395 			default:
396 				paxwarn(1, "Bad option %c with time range %s",
397 				    *flgpt, str);
398 				free(pt);
399 				goto out;
400 			}
401 			++flgpt;
402 		}
403 	}
404 
405 	/*
406 	 * start off with the current time
407 	 */
408 	pt->low_time = pt->high_time = time(NULL);
409 	if (*str != '\0') {
410 		/*
411 		 * add lower limit
412 		 */
413 		if (str_sec(str, &(pt->low_time)) < 0) {
414 			paxwarn(1, "Illegal lower time range %s", str);
415 			free(pt);
416 			goto out;
417 		}
418 		pt->flgs |= HASLOW;
419 	}
420 
421 	if ((up_pt != NULL) && (*up_pt != '\0')) {
422 		/*
423 		 * add upper limit
424 		 */
425 		if (str_sec(up_pt, &(pt->high_time)) < 0) {
426 			paxwarn(1, "Illegal upper time range %s", up_pt);
427 			free(pt);
428 			goto out;
429 		}
430 		pt->flgs |= HASHIGH;
431 
432 		/*
433 		 * check that the upper and lower do not overlap
434 		 */
435 		if (pt->flgs & HASLOW) {
436 			if (pt->low_time > pt->high_time) {
437 				paxwarn(1, "Upper %s and lower %s time overlap",
438 					up_pt, str);
439 				free(pt);
440 				return(-1);
441 			}
442 		}
443 	}
444 
445 	pt->fow = NULL;
446 	if (trhead == NULL) {
447 		trtail = trhead = pt;
448 		return(0);
449 	}
450 	trtail->fow = pt;
451 	trtail = pt;
452 	return(0);
453 
454     out:
455 	paxwarn(1, "Time range format is: [yy[mm[dd[hh]]]]mm[.ss][/[c][m]]");
456 	return(-1);
457 }
458 
459 /*
460  * trng_match()
461  *	check if this files mtime/ctime falls within any supplied time range.
462  * Return:
463  *	0 if this archive member should be processed, 1 if it should be skipped
464  */
465 
466 static int
467 trng_match(ARCHD *arcn)
468 {
469 	TIME_RNG *pt;
470 
471 	/*
472 	 * have to search down the list one at a time looking for a match.
473 	 * remember time range limits are inclusive.
474 	 */
475 	pt = trhead;
476 	while (pt != NULL) {
477 		switch(pt->flgs & CMPBOTH) {
478 		case CMPBOTH:
479 			/*
480 			 * user wants both mtime and ctime checked for this
481 			 * time range
482 			 */
483 			if (((pt->flgs & HASLOW) &&
484 			    (arcn->sb.st_mtime < pt->low_time) &&
485 			    (arcn->sb.st_ctime < pt->low_time)) ||
486 			    ((pt->flgs & HASHIGH) &&
487 			    (arcn->sb.st_mtime > pt->high_time) &&
488 			    (arcn->sb.st_ctime > pt->high_time))) {
489 				pt = pt->fow;
490 				continue;
491 			}
492 			break;
493 		case CMPCTME:
494 			/*
495 			 * user wants only ctime checked for this time range
496 			 */
497 			if (((pt->flgs & HASLOW) &&
498 			    (arcn->sb.st_ctime < pt->low_time)) ||
499 			    ((pt->flgs & HASHIGH) &&
500 			    (arcn->sb.st_ctime > pt->high_time))) {
501 				pt = pt->fow;
502 				continue;
503 			}
504 			break;
505 		case CMPMTME:
506 		default:
507 			/*
508 			 * user wants only mtime checked for this time range
509 			 */
510 			if (((pt->flgs & HASLOW) &&
511 			    (arcn->sb.st_mtime < pt->low_time)) ||
512 			    ((pt->flgs & HASHIGH) &&
513 			    (arcn->sb.st_mtime > pt->high_time))) {
514 				pt = pt->fow;
515 				continue;
516 			}
517 			break;
518 		}
519 		break;
520 	}
521 
522 	if (pt == NULL)
523 		return(1);
524 	return(0);
525 }
526 
527 /*
528  * str_sec()
529  *	Convert a time string in the format of [yy[mm[dd[hh]]]]mm[.ss] to gmt
530  *	seconds. Tval already has current time loaded into it at entry.
531  * Return:
532  *	0 if converted ok, -1 otherwise
533  */
534 
535 static int
536 str_sec(char *str, time_t *tval)
537 {
538 	struct tm *lt;
539 	char *dot = NULL;
540 
541 	lt = localtime(tval);
542 	if ((dot = strchr(str, '.')) != NULL) {
543 		/*
544 		 * seconds (.ss)
545 		 */
546 		*dot++ = '\0';
547 		if (strlen(dot) != 2)
548 			return(-1);
549 		if ((lt->tm_sec = ATOI2(dot)) > 61)
550 			return(-1);
551 	} else
552 		lt->tm_sec = 0;
553 
554 	switch (strlen(str)) {
555 	case 10:
556 		/*
557 		 * year (yy)
558 		 * watch out for year 2000
559 		 */
560 		if ((lt->tm_year = ATOI2(str)) < 69)
561 			lt->tm_year += 100;
562 		str += 2;
563 		/* FALLTHROUGH */
564 	case 8:
565 		/*
566 		 * month (mm)
567 		 * watch out months are from 0 - 11 internally
568 		 */
569 		if ((lt->tm_mon = ATOI2(str)) > 12)
570 			return(-1);
571 		--lt->tm_mon;
572 		str += 2;
573 		/* FALLTHROUGH */
574 	case 6:
575 		/*
576 		 * day (dd)
577 		 */
578 		if ((lt->tm_mday = ATOI2(str)) > 31)
579 			return(-1);
580 		str += 2;
581 		/* FALLTHROUGH */
582 	case 4:
583 		/*
584 		 * hour (hh)
585 		 */
586 		if ((lt->tm_hour = ATOI2(str)) > 23)
587 			return(-1);
588 		str += 2;
589 		/* FALLTHROUGH */
590 	case 2:
591 		/*
592 		 * minute (mm)
593 		 */
594 		if ((lt->tm_min = ATOI2(str)) > 59)
595 			return(-1);
596 		break;
597 	default:
598 		return(-1);
599 	}
600 	/*
601 	 * convert broken-down time to GMT clock time seconds
602 	 */
603 	if ((*tval = mktime(lt)) == -1)
604 		return(-1);
605 	return(0);
606 }
607