xref: /freebsd/bin/pax/pax.c (revision f0d6447c0b771cec9a8b47b1f829ec78f7e58ee4)
14b88c807SRodney W. Grimes /*-
28a16b7a1SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
38a16b7a1SPedro F. Giffuni  *
44b88c807SRodney W. Grimes  * Copyright (c) 1992 Keith Muller.
54b88c807SRodney W. Grimes  * Copyright (c) 1992, 1993
64b88c807SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
74b88c807SRodney W. Grimes  *
84b88c807SRodney W. Grimes  * This code is derived from software contributed to Berkeley by
94b88c807SRodney W. Grimes  * Keith Muller of the University of California, San Diego.
104b88c807SRodney W. Grimes  *
114b88c807SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
124b88c807SRodney W. Grimes  * modification, are permitted provided that the following conditions
134b88c807SRodney W. Grimes  * are met:
144b88c807SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
154b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
164b88c807SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
174b88c807SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
184b88c807SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
19fbbd9655SWarner Losh  * 3. Neither the name of the University nor the names of its contributors
204b88c807SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
214b88c807SRodney W. Grimes  *    without specific prior written permission.
224b88c807SRodney W. Grimes  *
234b88c807SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
244b88c807SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
254b88c807SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
264b88c807SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
274b88c807SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
284b88c807SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
294b88c807SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
304b88c807SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
314b88c807SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
324b88c807SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
334b88c807SRodney W. Grimes  * SUCH DAMAGE.
344b88c807SRodney W. Grimes  */
354b88c807SRodney W. Grimes 
364b88c807SRodney W. Grimes #include <sys/types.h>
374b88c807SRodney W. Grimes #include <sys/stat.h>
384b88c807SRodney W. Grimes #include <sys/time.h>
394b88c807SRodney W. Grimes #include <sys/resource.h>
401192d531SKris Kennaway #include <err.h>
414b88c807SRodney W. Grimes #include <errno.h>
42b1787decSKris Kennaway #include <fcntl.h>
43c2ad0566SAndrey A. Chernov #include <locale.h>
44ffbef1cdSKris Kennaway #include <paths.h>
45007d3350SEivind Eklund #include <signal.h>
46007d3350SEivind Eklund #include <stdio.h>
47007d3350SEivind Eklund #include <stdlib.h>
48cafefe8cSDima Dorfman #include <string.h>
494b88c807SRodney W. Grimes #include "pax.h"
504b88c807SRodney W. Grimes #include "extern.h"
5146251ddeSWarner Losh static int gen_init(void);
524b88c807SRodney W. Grimes 
534b88c807SRodney W. Grimes /*
544b88c807SRodney W. Grimes  * PAX main routines, general globals and some simple start up routines
554b88c807SRodney W. Grimes  */
564b88c807SRodney W. Grimes 
574b88c807SRodney W. Grimes /*
584b88c807SRodney W. Grimes  * Variables that can be accessed by any routine within pax
594b88c807SRodney W. Grimes  */
604b88c807SRodney W. Grimes int	act = DEFOP;		/* read/write/append/copy */
614b88c807SRodney W. Grimes FSUB	*frmt = NULL;		/* archive format type */
624b88c807SRodney W. Grimes int	cflag;			/* match all EXCEPT pattern/file */
63b1787decSKris Kennaway int	cwdfd;			/* starting cwd */
644b88c807SRodney W. Grimes int	dflag;			/* directory member match only  */
654b88c807SRodney W. Grimes int	iflag;			/* interactive file/archive rename */
664b88c807SRodney W. Grimes int	kflag;			/* do not overwrite existing files */
674b88c807SRodney W. Grimes int	lflag;			/* use hard links when possible */
684b88c807SRodney W. Grimes int	nflag;			/* select first archive member match */
694b88c807SRodney W. Grimes int	tflag;			/* restore access time after read */
704b88c807SRodney W. Grimes int	uflag;			/* ignore older modification time files */
714b88c807SRodney W. Grimes int	vflag;			/* produce verbose output */
72*f0d6447cSZhan-Wei int	Dflag;			/* same as uflag except for inode change time */
734b88c807SRodney W. Grimes int	Hflag;			/* follow command line symlinks (write only) */
744b88c807SRodney W. Grimes int	Lflag;			/* follow symlinks when writing */
75d6e1f8d7SEitan Adler int	Oflag;			/* limit to single volume */
764b88c807SRodney W. Grimes int	Xflag;			/* archive files with same device id only */
774b88c807SRodney W. Grimes int	Yflag;			/* same as Dflg except after name mode */
784b88c807SRodney W. Grimes int	Zflag;			/* same as uflg except after name mode */
794b88c807SRodney W. Grimes int	vfpart;			/* is partial verbose output in progress */
804b88c807SRodney W. Grimes int	patime = 1;		/* preserve file access time */
814b88c807SRodney W. Grimes int	pmtime = 1;		/* preserve file modification times */
82b1787decSKris Kennaway int	nodirs;			/* do not create directories as needed */
834b88c807SRodney W. Grimes int	pmode;			/* preserve file mode bits */
844b88c807SRodney W. Grimes int	pids;			/* preserve file uid/gid */
85b1787decSKris Kennaway int	rmleadslash = 0;	/* remove leading '/' from pathnames */
864b88c807SRodney W. Grimes int	exit_val;		/* exit value */
874b88c807SRodney W. Grimes int	docrc;			/* check/create file crc */
884b88c807SRodney W. Grimes char	*dirptr;		/* destination dir in a copy */
8940feca3aSMark Murray const	char *argv0;		/* root of argv[0] */
90b1787decSKris Kennaway sigset_t s_mask;		/* signal mask for cleanup critical sect */
91ab01f433SPeter Wemm FILE	*listf;			/* file pointer to print file list to */
92ffbef1cdSKris Kennaway char	*tempfile;		/* tempfile to use for mkstemp(3) */
93ffbef1cdSKris Kennaway char	*tempbase;		/* basename of tempfile to use for mkstemp(3) */
944b88c807SRodney W. Grimes 
954b88c807SRodney W. Grimes /*
964b88c807SRodney W. Grimes  *	PAX - Portable Archive Interchange
974b88c807SRodney W. Grimes  *
984b88c807SRodney W. Grimes  *	A utility to read, write, and write lists of the members of archive
994b88c807SRodney W. Grimes  *	files and copy directory hierarchies. A variety of archive formats
1004b88c807SRodney W. Grimes  *	are supported (some are described in POSIX 1003.1 10.1):
1014b88c807SRodney W. Grimes  *
1024b88c807SRodney W. Grimes  *		ustar - 10.1.1 extended tar interchange format
1034b88c807SRodney W. Grimes  *		cpio  - 10.1.2 extended cpio interchange format
1044b88c807SRodney W. Grimes  *		tar - old BSD 4.3 tar format
1054b88c807SRodney W. Grimes  *		binary cpio - old cpio with binary header format
1064b88c807SRodney W. Grimes  *		sysVR4 cpio -  with and without CRC
1074b88c807SRodney W. Grimes  *
1084b88c807SRodney W. Grimes  * This version is a superset of IEEE Std 1003.2b-d3
1094b88c807SRodney W. Grimes  *
1104b88c807SRodney W. Grimes  * Summary of Extensions to the IEEE Standard:
1114b88c807SRodney W. Grimes  *
1124b88c807SRodney W. Grimes  * 1	READ ENHANCEMENTS
1134b88c807SRodney W. Grimes  * 1.1	Operations which read archives will continue to operate even when
1144b88c807SRodney W. Grimes  *	processing archives which may be damaged, truncated, or fail to meet
1154b88c807SRodney W. Grimes  *	format specs in several different ways. Damaged sections of archives
1164b88c807SRodney W. Grimes  *	are detected and avoided if possible. Attempts will be made to resync
1174b88c807SRodney W. Grimes  *	archive read operations even with badly damaged media.
1184b88c807SRodney W. Grimes  * 1.2	Blocksize requirements are not strictly enforced on archive read.
1194b88c807SRodney W. Grimes  *	Tapes which have variable sized records can be read without errors.
1204b88c807SRodney W. Grimes  * 1.3	The user can specify via the non-standard option flag -E if error
1214b88c807SRodney W. Grimes  *	resync operation should stop on a media error, try a specified number
1224b88c807SRodney W. Grimes  *	of times to correct, or try to correct forever.
1234b88c807SRodney W. Grimes  * 1.4	Sparse files (lseek holes) stored on the archive (but stored with blocks
1244b88c807SRodney W. Grimes  *	of all zeros will be restored with holes appropriate for the target
1254b88c807SRodney W. Grimes  *	file system
1264b88c807SRodney W. Grimes  * 1.5	The user is notified whenever something is found during archive
1274b88c807SRodney W. Grimes  *	read operations which violates spec (but the read will continue).
1284b88c807SRodney W. Grimes  * 1.6	Multiple archive volumes can be read and may span over different
1294b88c807SRodney W. Grimes  *	archive devices
1304b88c807SRodney W. Grimes  * 1.7	Rigidly restores all file attributes exactly as they are stored on the
1314b88c807SRodney W. Grimes  *	archive.
1324b88c807SRodney W. Grimes  * 1.8	Modification change time ranges can be specified via multiple -T
1334b88c807SRodney W. Grimes  *	options. These allow a user to select files whose modification time
1344b88c807SRodney W. Grimes  *	lies within a specific time range.
1354b88c807SRodney W. Grimes  * 1.9	Files can be selected based on owner (user name or uid) via one or more
1364b88c807SRodney W. Grimes  *	-U options.
1374b88c807SRodney W. Grimes  * 1.10	Files can be selected based on group (group name or gid) via one o
1384b88c807SRodney W. Grimes  *	more -G options.
13946be34b9SKris Kennaway  * 1.11	File modification time can be checked against existing file after
1404b88c807SRodney W. Grimes  *	name modification (-Z)
1414b88c807SRodney W. Grimes  *
1424b88c807SRodney W. Grimes  * 2	WRITE ENHANCEMENTS
1434b88c807SRodney W. Grimes  * 2.1	Write operation will stop instead of allowing a user to create a flawed
1444b88c807SRodney W. Grimes  *	flawed archive (due to any problem).
14546be34b9SKris Kennaway  * 2.2	Archives written by pax are forced to strictly conform to both the
14646be34b9SKris Kennaway  *	archive and pax the specific format specifications.
1474b88c807SRodney W. Grimes  * 2.3	Blocking size and format is rigidly enforced on writes.
1484b88c807SRodney W. Grimes  * 2.4	Formats which may exhibit header overflow problems (they have fields
1494b88c807SRodney W. Grimes  *	too small for large file systems, such as inode number storage), use
1504b88c807SRodney W. Grimes  *	routines designed to repair this problem. These techniques still
1514b88c807SRodney W. Grimes  *	conform to both pax and format specifications, but no longer truncate
1524b88c807SRodney W. Grimes  *	these fields. This removes any restrictions on using these archive
1534b88c807SRodney W. Grimes  *	formats on large file systems.
1544b88c807SRodney W. Grimes  * 2.5	Multiple archive volumes can be written and may span over different
1554b88c807SRodney W. Grimes  *	archive devices
1564b88c807SRodney W. Grimes  * 2.6	A archive volume record limit allows the user to specify the number
1574b88c807SRodney W. Grimes  *	of bytes stored on an archive volume. When reached the user is
1584b88c807SRodney W. Grimes  *	prompted for the next archive volume. This is specified with the
15946be34b9SKris Kennaway  *	non-standard -B flag. The limit is rounded up to the next blocksize.
1604b88c807SRodney W. Grimes  * 2.7	All archive padding during write use zero filled sections. This makes
1614b88c807SRodney W. Grimes  *	it much easier to pull data out of flawed archive during read
1624b88c807SRodney W. Grimes  *	operations.
1634b88c807SRodney W. Grimes  * 2.8	Access time reset with the -t applies to all file nodes (including
1644b88c807SRodney W. Grimes  *	directories).
1654b88c807SRodney W. Grimes  * 2.9	Symbolic links can be followed with -L (optional in the spec).
1664b88c807SRodney W. Grimes  * 2.10	Modification or inode change time ranges can be specified via
1674b88c807SRodney W. Grimes  *	multiple -T options. These allow a user to select files whose
1684b88c807SRodney W. Grimes  *	modification or inode change time lies within a specific time range.
1694b88c807SRodney W. Grimes  * 2.11	Files can be selected based on owner (user name or uid) via one or more
1704b88c807SRodney W. Grimes  *	-U options.
1714b88c807SRodney W. Grimes  * 2.12	Files can be selected based on group (group name or gid) via one o
1724b88c807SRodney W. Grimes  *	more -G options.
1734b88c807SRodney W. Grimes  * 2.13	Symlinks which appear on the command line can be followed (without
1744b88c807SRodney W. Grimes  *	following other symlinks; -H flag)
1754b88c807SRodney W. Grimes  *
1764b88c807SRodney W. Grimes  * 3	COPY ENHANCEMENTS
1774b88c807SRodney W. Grimes  * 3.1	Sparse files (lseek holes) can be copied without expanding the holes
1784b88c807SRodney W. Grimes  *	into zero filled blocks. The file copy is created with holes which are
1794b88c807SRodney W. Grimes  *	appropriate for the target file system
1804b88c807SRodney W. Grimes  * 3.2	Access time as well as modification time on copied file trees can be
1814b88c807SRodney W. Grimes  *	preserved with the appropriate -p options.
1824b88c807SRodney W. Grimes  * 3.3	Access time reset with the -t applies to all file nodes (including
1834b88c807SRodney W. Grimes  *	directories).
1844b88c807SRodney W. Grimes  * 3.4	Symbolic links can be followed with -L (optional in the spec).
1854b88c807SRodney W. Grimes  * 3.5	Modification or inode change time ranges can be specified via
1864b88c807SRodney W. Grimes  *	multiple -T options. These allow a user to select files whose
1874b88c807SRodney W. Grimes  *	modification or inode change time lies within a specific time range.
1884b88c807SRodney W. Grimes  * 3.6	Files can be selected based on owner (user name or uid) via one or more
1894b88c807SRodney W. Grimes  *	-U options.
1904b88c807SRodney W. Grimes  * 3.7	Files can be selected based on group (group name or gid) via one o
1914b88c807SRodney W. Grimes  *	more -G options.
1924b88c807SRodney W. Grimes  * 3.8	Symlinks which appear on the command line can be followed (without
1934b88c807SRodney W. Grimes  *	following other symlinks; -H flag)
19446be34b9SKris Kennaway  * 3.9  File inode change time can be checked against existing file before
1954b88c807SRodney W. Grimes  *	name modification (-D)
19646be34b9SKris Kennaway  * 3.10 File inode change time can be checked against existing file after
1974b88c807SRodney W. Grimes  *	name modification (-Y)
19846be34b9SKris Kennaway  * 3.11	File modification time can be checked against existing file after
1994b88c807SRodney W. Grimes  *	name modification (-Z)
2004b88c807SRodney W. Grimes  *
2014b88c807SRodney W. Grimes  * 4	GENERAL ENHANCEMENTS
2024b88c807SRodney W. Grimes  * 4.1	Internal structure is designed to isolate format dependent and
2034b88c807SRodney W. Grimes  *	independent functions. Formats are selected via a format driver table.
2044b88c807SRodney W. Grimes  *	This encourages the addition of new archive formats by only having to
2054b88c807SRodney W. Grimes  *	write those routines which id, read and write the archive header.
2064b88c807SRodney W. Grimes  */
2074b88c807SRodney W. Grimes 
2084b88c807SRodney W. Grimes /*
2094b88c807SRodney W. Grimes  * main()
2104b88c807SRodney W. Grimes  *	parse options, set up and operate as specified by the user.
2114b88c807SRodney W. Grimes  *	any operational flaw will set exit_val to non-zero
2124b88c807SRodney W. Grimes  * Return: 0 if ok, 1 otherwise
2134b88c807SRodney W. Grimes  */
2144b88c807SRodney W. Grimes 
2154b88c807SRodney W. Grimes int
main(int argc,char * argv[])21646251ddeSWarner Losh main(int argc, char *argv[])
2174b88c807SRodney W. Grimes {
21840feca3aSMark Murray 	const char *tmpdir;
219ffbef1cdSKris Kennaway 	size_t tdlen;
220ffbef1cdSKris Kennaway 
2216e0d255eSAndrey A. Chernov 	(void) setlocale(LC_ALL, "");
222ab01f433SPeter Wemm 	listf = stderr;
223b1787decSKris Kennaway 	/*
224b1787decSKris Kennaway 	 * Keep a reference to cwd, so we can always come back home.
225b1787decSKris Kennaway 	 */
2267aee7c62SPedro F. Giffuni 	cwdfd = open(".", O_RDONLY | O_CLOEXEC);
227b1787decSKris Kennaway 	if (cwdfd < 0) {
228b1787decSKris Kennaway 		syswarn(0, errno, "Can't open current working directory.");
229b1787decSKris Kennaway 		return(exit_val);
230b1787decSKris Kennaway 	}
231ffbef1cdSKris Kennaway 
232ffbef1cdSKris Kennaway 	/*
233ffbef1cdSKris Kennaway 	 * Where should we put temporary files?
234ffbef1cdSKris Kennaway 	 */
235ffbef1cdSKris Kennaway 	if ((tmpdir = getenv("TMPDIR")) == NULL || *tmpdir == '\0')
236ffbef1cdSKris Kennaway 		tmpdir = _PATH_TMP;
237ffbef1cdSKris Kennaway 	tdlen = strlen(tmpdir);
238ffbef1cdSKris Kennaway 	while (tdlen > 0 && tmpdir[tdlen - 1] == '/')
239ffbef1cdSKris Kennaway 		tdlen--;
240ffbef1cdSKris Kennaway 	tempfile = malloc(tdlen + 1 + sizeof(_TFILE_BASE));
241ffbef1cdSKris Kennaway 	if (tempfile == NULL) {
242778766feSKris Kennaway 		paxwarn(1, "Cannot allocate memory for temp file name.");
243ffbef1cdSKris Kennaway 		return(exit_val);
244ffbef1cdSKris Kennaway 	}
245ffbef1cdSKris Kennaway 	if (tdlen)
246ffbef1cdSKris Kennaway 		memcpy(tempfile, tmpdir, tdlen);
247ffbef1cdSKris Kennaway 	tempbase = tempfile + tdlen;
248ffbef1cdSKris Kennaway 	*tempbase++ = '/';
249ffbef1cdSKris Kennaway 
2504b88c807SRodney W. Grimes 	/*
2514b88c807SRodney W. Grimes 	 * parse options, determine operational mode, general init
2524b88c807SRodney W. Grimes 	 */
2534b88c807SRodney W. Grimes 	options(argc, argv);
2544b88c807SRodney W. Grimes 	if ((gen_init() < 0) || (tty_init() < 0))
2554b88c807SRodney W. Grimes 		return(exit_val);
2564b88c807SRodney W. Grimes 
2574b88c807SRodney W. Grimes 	/*
2584b88c807SRodney W. Grimes 	 * select a primary operation mode
2594b88c807SRodney W. Grimes 	 */
2604b88c807SRodney W. Grimes 	switch (act) {
2614b88c807SRodney W. Grimes 	case EXTRACT:
2624b88c807SRodney W. Grimes 		extract();
2634b88c807SRodney W. Grimes 		break;
2644b88c807SRodney W. Grimes 	case ARCHIVE:
2654b88c807SRodney W. Grimes 		archive();
2664b88c807SRodney W. Grimes 		break;
2674b88c807SRodney W. Grimes 	case APPND:
2681192d531SKris Kennaway 		if (gzip_program != NULL)
2691192d531SKris Kennaway 			err(1, "can not gzip while appending");
2704b88c807SRodney W. Grimes 		append();
2714b88c807SRodney W. Grimes 		break;
2724b88c807SRodney W. Grimes 	case COPY:
2734b88c807SRodney W. Grimes 		copy();
2744b88c807SRodney W. Grimes 		break;
2754b88c807SRodney W. Grimes 	default:
2764b88c807SRodney W. Grimes 	case LIST:
2774b88c807SRodney W. Grimes 		list();
2784b88c807SRodney W. Grimes 		break;
2794b88c807SRodney W. Grimes 	}
2804b88c807SRodney W. Grimes 	return(exit_val);
2814b88c807SRodney W. Grimes }
2824b88c807SRodney W. Grimes 
2834b88c807SRodney W. Grimes /*
2844b88c807SRodney W. Grimes  * sig_cleanup()
2854b88c807SRodney W. Grimes  *	when interrupted we try to do whatever delayed processing we can.
2864b88c807SRodney W. Grimes  *	This is not critical, but we really ought to limit our damage when we
2874b88c807SRodney W. Grimes  *	are aborted by the user.
2884b88c807SRodney W. Grimes  * Return:
2894b88c807SRodney W. Grimes  *	never....
2904b88c807SRodney W. Grimes  */
2914b88c807SRodney W. Grimes 
2924b88c807SRodney W. Grimes void
sig_cleanup(int which_sig)2934b88c807SRodney W. Grimes sig_cleanup(int which_sig)
2944b88c807SRodney W. Grimes {
2954b88c807SRodney W. Grimes 	/*
2964b88c807SRodney W. Grimes 	 * restore modes and times for any dirs we may have created
2974b88c807SRodney W. Grimes 	 * or any dirs we may have read. Set vflag and vfpart so the user
2984b88c807SRodney W. Grimes 	 * will clearly see the message on a line by itself.
2994b88c807SRodney W. Grimes 	 */
3004b88c807SRodney W. Grimes 	vflag = vfpart = 1;
3014b88c807SRodney W. Grimes 	if (which_sig == SIGXCPU)
302778766feSKris Kennaway 		paxwarn(0, "Cpu time limit reached, cleaning up.");
3034b88c807SRodney W. Grimes 	else
304778766feSKris Kennaway 		paxwarn(0, "Signal caught, cleaning up.");
3054b88c807SRodney W. Grimes 
3064b88c807SRodney W. Grimes 	ar_close();
3074b88c807SRodney W. Grimes 	proc_dir();
3084b88c807SRodney W. Grimes 	if (tflag)
3094b88c807SRodney W. Grimes 		atdir_end();
3104b88c807SRodney W. Grimes 	exit(1);
3114b88c807SRodney W. Grimes }
3124b88c807SRodney W. Grimes 
3134b88c807SRodney W. Grimes /*
3147aee7c62SPedro F. Giffuni  * setup_sig()
3157aee7c62SPedro F. Giffuni  *	set a signal to be caught, but only if it isn't being ignored already
3167aee7c62SPedro F. Giffuni  */
3177aee7c62SPedro F. Giffuni 
3187aee7c62SPedro F. Giffuni static int
setup_sig(int sig,const struct sigaction * n_hand)3197aee7c62SPedro F. Giffuni setup_sig(int sig, const struct sigaction *n_hand)
3207aee7c62SPedro F. Giffuni {
3217aee7c62SPedro F. Giffuni 	struct sigaction o_hand;
3227aee7c62SPedro F. Giffuni 
3237aee7c62SPedro F. Giffuni 	if (sigaction(sig, NULL, &o_hand) < 0)
3247aee7c62SPedro F. Giffuni 		return (-1);
3257aee7c62SPedro F. Giffuni 
3267aee7c62SPedro F. Giffuni 	if (o_hand.sa_handler == SIG_IGN)
3277aee7c62SPedro F. Giffuni 		return (0);
3287aee7c62SPedro F. Giffuni 
3297aee7c62SPedro F. Giffuni 	return (sigaction(sig, n_hand, NULL));
3307aee7c62SPedro F. Giffuni }
3317aee7c62SPedro F. Giffuni 
3327aee7c62SPedro F. Giffuni /*
3334b88c807SRodney W. Grimes  * gen_init()
3344b88c807SRodney W. Grimes  *	general setup routines. Not all are required, but they really help
3354b88c807SRodney W. Grimes  *	when dealing with a medium to large sized archives.
3364b88c807SRodney W. Grimes  */
3374b88c807SRodney W. Grimes 
3384b88c807SRodney W. Grimes static int
gen_init(void)3394b88c807SRodney W. Grimes gen_init(void)
3404b88c807SRodney W. Grimes {
3414b88c807SRodney W. Grimes 	struct rlimit reslimit;
3424b88c807SRodney W. Grimes 	struct sigaction n_hand;
3434b88c807SRodney W. Grimes 
3444b88c807SRodney W. Grimes 	/*
3454b88c807SRodney W. Grimes 	 * Really needed to handle large archives. We can run out of memory for
3464b88c807SRodney W. Grimes 	 * internal tables really fast when we have a whole lot of files...
3474b88c807SRodney W. Grimes 	 */
3484b88c807SRodney W. Grimes 	if (getrlimit(RLIMIT_DATA , &reslimit) == 0){
3494b88c807SRodney W. Grimes 		reslimit.rlim_cur = reslimit.rlim_max;
3504b88c807SRodney W. Grimes 		(void)setrlimit(RLIMIT_DATA , &reslimit);
3514b88c807SRodney W. Grimes 	}
3524b88c807SRodney W. Grimes 
3534b88c807SRodney W. Grimes 	/*
3544b88c807SRodney W. Grimes 	 * should file size limits be waived? if the os limits us, this is
3554b88c807SRodney W. Grimes 	 * needed if we want to write a large archive
3564b88c807SRodney W. Grimes 	 */
3574b88c807SRodney W. Grimes 	if (getrlimit(RLIMIT_FSIZE , &reslimit) == 0){
3584b88c807SRodney W. Grimes 		reslimit.rlim_cur = reslimit.rlim_max;
3594b88c807SRodney W. Grimes 		(void)setrlimit(RLIMIT_FSIZE , &reslimit);
3604b88c807SRodney W. Grimes 	}
3614b88c807SRodney W. Grimes 
3624b88c807SRodney W. Grimes 	/*
3634b88c807SRodney W. Grimes 	 * increase the size the stack can grow to
3644b88c807SRodney W. Grimes 	 */
3654b88c807SRodney W. Grimes 	if (getrlimit(RLIMIT_STACK , &reslimit) == 0){
3664b88c807SRodney W. Grimes 		reslimit.rlim_cur = reslimit.rlim_max;
3674b88c807SRodney W. Grimes 		(void)setrlimit(RLIMIT_STACK , &reslimit);
3684b88c807SRodney W. Grimes 	}
3694b88c807SRodney W. Grimes 
3704b88c807SRodney W. Grimes 	/*
3714b88c807SRodney W. Grimes 	 * not really needed, but doesn't hurt
3724b88c807SRodney W. Grimes 	 */
3734b88c807SRodney W. Grimes 	if (getrlimit(RLIMIT_RSS , &reslimit) == 0){
3744b88c807SRodney W. Grimes 		reslimit.rlim_cur = reslimit.rlim_max;
3754b88c807SRodney W. Grimes 		(void)setrlimit(RLIMIT_RSS , &reslimit);
3764b88c807SRodney W. Grimes 	}
3774b88c807SRodney W. Grimes 
3784b88c807SRodney W. Grimes 	/*
3794b88c807SRodney W. Grimes 	 * signal handling to reset stored directory times and modes. Since
3804b88c807SRodney W. Grimes 	 * we deal with broken pipes via failed writes we ignore it. We also
3810266a5d6SDag-Erling Smørgrav 	 * deal with any file size limit through failed writes. Cpu time
3824b88c807SRodney W. Grimes 	 * limits are caught and a cleanup is forced.
3834b88c807SRodney W. Grimes 	 */
3844b88c807SRodney W. Grimes 	if ((sigemptyset(&s_mask) < 0) || (sigaddset(&s_mask, SIGTERM) < 0) ||
3854b88c807SRodney W. Grimes 	    (sigaddset(&s_mask,SIGINT) < 0)||(sigaddset(&s_mask,SIGHUP) < 0) ||
3864b88c807SRodney W. Grimes 	    (sigaddset(&s_mask,SIGPIPE) < 0)||(sigaddset(&s_mask,SIGQUIT)<0) ||
3874b88c807SRodney W. Grimes 	    (sigaddset(&s_mask,SIGXCPU) < 0)||(sigaddset(&s_mask,SIGXFSZ)<0)) {
388778766feSKris Kennaway 		paxwarn(1, "Unable to set up signal mask");
3894b88c807SRodney W. Grimes 		return(-1);
3904b88c807SRodney W. Grimes 	}
391b1787decSKris Kennaway 	memset(&n_hand, 0, sizeof n_hand);
3924b88c807SRodney W. Grimes 	n_hand.sa_mask = s_mask;
3934b88c807SRodney W. Grimes 	n_hand.sa_flags = 0;
3944b88c807SRodney W. Grimes 	n_hand.sa_handler = sig_cleanup;
3954b88c807SRodney W. Grimes 
3967aee7c62SPedro F. Giffuni 	if (setup_sig(SIGHUP,  &n_hand) ||
3977aee7c62SPedro F. Giffuni 	   setup_sig(SIGTERM, &n_hand) ||
3987aee7c62SPedro F. Giffuni 	   setup_sig(SIGINT,  &n_hand) ||
3997aee7c62SPedro F. Giffuni 	   setup_sig(SIGQUIT, &n_hand) ||
4007aee7c62SPedro F. Giffuni 	   setup_sig(SIGXCPU, &n_hand))
4014b88c807SRodney W. Grimes 		goto out;
4024b88c807SRodney W. Grimes 
4034b88c807SRodney W. Grimes 	n_hand.sa_handler = SIG_IGN;
4047aee7c62SPedro F. Giffuni 	if ((sigaction(SIGPIPE, &n_hand, NULL) < 0) ||
4057aee7c62SPedro F. Giffuni 	    (sigaction(SIGXFSZ, &n_hand, NULL) < 0))
4064b88c807SRodney W. Grimes 		goto out;
4074b88c807SRodney W. Grimes 	return(0);
4084b88c807SRodney W. Grimes 
4094b88c807SRodney W. Grimes     out:
410778766feSKris Kennaway 	syswarn(1, errno, "Unable to set up signal handler");
4114b88c807SRodney W. Grimes 	return(-1);
4124b88c807SRodney W. Grimes }
413