[apple/xnu.git] / SETUP / installfile / installfile.c

/*
 * Copyright (c) 2012 Apple, Inc. All rights reserved.
 *
 * @APPLE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this
 * file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_LICENSE_HEADER_END@
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdbool.h>
#include <errno.h>
#include <err.h>
#include <sysexits.h>

#include <sys/stat.h>
#include <sys/fcntl.h>
#include <sys/param.h>
#include <sys/time.h>

#include <copyfile.h>

void usage(void);

int main(int argc, char * argv[])
{
	struct stat sb;
	void *mset;
	mode_t mode;
	bool gotmode = false;
	int ch;
	int ret;
	int srcfd, dstfd;
	const char *src = NULL;
	const char *dst = NULL;
	char dsttmpname[MAXPATHLEN];

	while ((ch = getopt(argc, argv, "cSm:")) != -1) {
		switch(ch) {
			case 'c':
			case 'S':
				/* ignored for compatibility */
				break;
			case 'm':
				gotmode = true;
				mset = setmode(optarg);
				if (!mset)
					errx(EX_USAGE, "Unrecognized mode %s", optarg);

				mode = getmode(mset, 0);
				free(mset);
				break;
			case '?':
			default:
				usage();
		}
	}

	argc -= optind;
	argv += optind;

	if (argc < 2) {
		usage();
	}

	src = argv[0];
	dst = argv[1];

	srcfd = open(src, O_RDONLY | O_SYMLINK, 0);
	if (srcfd < 0)
		err(EX_NOINPUT, "open(%s)", src);

	ret = fstat(srcfd, &sb);
	if (ret < 0)
		err(EX_NOINPUT, "fstat(%s)", src);

	if (!S_ISREG(sb.st_mode))
		err(EX_USAGE, "%s is not a regular file", src);

	snprintf(dsttmpname, sizeof(dsttmpname), "%s.XXXXXX", dst);

	dstfd = mkstemp(dsttmpname);
	if (dstfd < 0)
		err(EX_UNAVAILABLE, "mkstemp(%s)", dsttmpname);

	ret = fcopyfile(srcfd, dstfd, NULL,
					COPYFILE_DATA);
	if (ret < 0)
		err(EX_UNAVAILABLE, "fcopyfile(%s, %s)", src, dsttmpname);

	ret = futimes(dstfd, NULL);
	if (ret < 0)
		err(EX_UNAVAILABLE, "futimes(%s)", dsttmpname);

	if (gotmode) {
		ret = fchmod(dstfd, mode);
		if (ret < 0)
			err(EX_NOINPUT, "fchmod(%s, %ho)", dsttmpname, mode);
	}

	ret = rename(dsttmpname, dst);
	if (ret < 0)
		err(EX_NOINPUT, "rename(%s, %s)", dsttmpname, dst);

	ret = close(dstfd);
	if (ret < 0)
		err(EX_NOINPUT, "close(dst)");

	ret = close(srcfd);
	if (ret < 0)
		err(EX_NOINPUT, "close(src)");

	return 0;
}

void usage(void)
{
	fprintf(stderr, "Usage: %s [-c] [-S] [-m <mode>] <src> <dst>\n",
			getprogname());
	exit(EX_USAGE);
}
Commit	Line	Data
39236c6e A	1	/*
	2	* Copyright (c) 2012 Apple, Inc. All rights reserved.
	3	*
	4	* @APPLE_LICENSE_HEADER_START@
	5	*
	6	* This file contains Original Code and/or Modifications of Original Code
	7	* as defined in and that are subject to the Apple Public Source License
	8	* Version 2.0 (the 'License'). You may not use this file except in
	9	* compliance with the License. Please obtain a copy of the License at
	10	* http://www.opensource.apple.com/apsl/ and read it before using this
	11	* file.
	12	*
	13	* The Original Code and all software distributed under the License are
	14	* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
	15	* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
	16	* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
	17	* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
	18	* Please see the License for the specific language governing rights and
	19	* limitations under the License.
	20	*
	21	* @APPLE_LICENSE_HEADER_END@
	22	*/
	23
	24	#include <stdio.h>
	25	#include <stdlib.h>
	26	#include <unistd.h>
	27	#include <stdbool.h>
	28	#include <errno.h>
	29	#include <err.h>
	30	#include <sysexits.h>
	31
	32	#include <sys/stat.h>
	33	#include <sys/fcntl.h>
	34	#include <sys/param.h>
	35	#include <sys/time.h>
	36
	37	#include <copyfile.h>
	38
	39	void usage(void);
	40
	41	int main(int argc, char * argv[])
	42	{
	43	struct stat sb;
	44	void *mset;
	45	mode_t mode;
	46	bool gotmode = false;
	47	int ch;
	48	int ret;
	49	int srcfd, dstfd;
	50	const char *src = NULL;
	51	const char *dst = NULL;
	52	char dsttmpname[MAXPATHLEN];
	53
	54	while ((ch = getopt(argc, argv, "cSm:")) != -1) {
	55	switch(ch) {
	56	case 'c':
	57	case 'S':
	58	/* ignored for compatibility */
	59	break;
	60	case 'm':
	61	gotmode = true;
	62	mset = setmode(optarg);
	63	if (!mset)
	64	errx(EX_USAGE, "Unrecognized mode %s", optarg);
65
66	mode = getmode(mset, 0);
67	free(mset);
68	break;
69	case '?':
70	default:
71	usage();
72	}
73	}
74
75	argc -= optind;
76	argv += optind;
77
78	if (argc < 2) {
79	usage();
80	}
81
82	src = argv[0];
83	dst = argv[1];
84
85	srcfd = open(src, O_RDONLY \| O_SYMLINK, 0);
86	if (srcfd < 0)
87	err(EX_NOINPUT, "open(%s)", src);
88
89	ret = fstat(srcfd, &sb);
90	if (ret < 0)
91	err(EX_NOINPUT, "fstat(%s)", src);
92
93	if (!S_ISREG(sb.st_mode))
94	err(EX_USAGE, "%s is not a regular file", src);
95
96	snprintf(dsttmpname, sizeof(dsttmpname), "%s.XXXXXX", dst);
97
98	dstfd = mkstemp(dsttmpname);
99	if (dstfd < 0)
100	err(EX_UNAVAILABLE, "mkstemp(%s)", dsttmpname);
101
102	ret = fcopyfile(srcfd, dstfd, NULL,
103	COPYFILE_DATA);
104	if (ret < 0)
105	err(EX_UNAVAILABLE, "fcopyfile(%s, %s)", src, dsttmpname);
106
107	ret = futimes(dstfd, NULL);
108	if (ret < 0)
109	err(EX_UNAVAILABLE, "futimes(%s)", dsttmpname);
110
111	if (gotmode) {
112	ret = fchmod(dstfd, mode);
113	if (ret < 0)
114	err(EX_NOINPUT, "fchmod(%s, %ho)", dsttmpname, mode);
115	}
116
117	ret = rename(dsttmpname, dst);
118	if (ret < 0)
119	err(EX_NOINPUT, "rename(%s, %s)", dsttmpname, dst);
120
121	ret = close(dstfd);
122	if (ret < 0)
123	err(EX_NOINPUT, "close(dst)");
124
125	ret = close(srcfd);
126	if (ret < 0)
127	err(EX_NOINPUT, "close(src)");
128
129	return 0;
130	}
131
132	void usage(void)
133	{
134	fprintf(stderr, "Usage: %s [-c] [-S] [-m <mode>] <src> <dst>\n",
135	getprogname());
136	exit(EX_USAGE);
137	}