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