]>
Commit | Line | Data |
---|---|---|
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 | |
42 | main(int argc, char * argv[]) | |
43 | { | |
44 | struct stat sb; | |
45 | void *mset; | |
46 | mode_t mode; | |
47 | bool gotmode = false; | |
48 | int ch; | |
49 | int ret; | |
50 | int srcfd, dstfd; | |
51 | const char *src = NULL; | |
52 | const char *dst = NULL; | |
53 | char dsttmpname[MAXPATHLEN]; | |
54 | ||
55 | while ((ch = getopt(argc, argv, "cSm:")) != -1) { | |
56 | switch (ch) { | |
57 | case 'c': | |
58 | case 'S': | |
59 | /* ignored for compatibility */ | |
60 | break; | |
61 | case 'm': | |
62 | gotmode = true; | |
63 | mset = setmode(optarg); | |
64 | if (!mset) { | |
65 | errx(EX_USAGE, "Unrecognized mode %s", optarg); | |
66 | } | |
67 | ||
68 | mode = getmode(mset, 0); | |
69 | free(mset); | |
70 | break; | |
71 | case '?': | |
72 | default: | |
73 | usage(); | |
74 | } | |
75 | } | |
76 | ||
77 | argc -= optind; | |
78 | argv += optind; | |
79 | ||
80 | if (argc < 2) { | |
81 | usage(); | |
82 | } | |
83 | ||
84 | src = argv[0]; | |
85 | dst = argv[1]; | |
86 | ||
87 | srcfd = open(src, O_RDONLY | O_SYMLINK, 0); | |
88 | if (srcfd < 0) { | |
89 | err(EX_NOINPUT, "open(%s)", src); | |
90 | } | |
91 | ||
92 | ret = fstat(srcfd, &sb); | |
93 | if (ret < 0) { | |
94 | err(EX_NOINPUT, "fstat(%s)", src); | |
95 | } | |
96 | ||
97 | if (!S_ISREG(sb.st_mode)) { | |
98 | err(EX_USAGE, "%s is not a regular file", src); | |
99 | } | |
100 | ||
101 | snprintf(dsttmpname, sizeof(dsttmpname), "%s.XXXXXX", dst); | |
102 | ||
103 | dstfd = mkstemp(dsttmpname); | |
104 | if (dstfd < 0) { | |
105 | err(EX_UNAVAILABLE, "mkstemp(%s)", dsttmpname); | |
106 | } | |
107 | ||
108 | ret = fcopyfile(srcfd, dstfd, NULL, | |
109 | COPYFILE_DATA); | |
110 | if (ret < 0) { | |
111 | err(EX_UNAVAILABLE, "fcopyfile(%s, %s)", src, dsttmpname); | |
112 | } | |
113 | ||
114 | ret = futimes(dstfd, NULL); | |
115 | if (ret < 0) { | |
116 | err(EX_UNAVAILABLE, "futimes(%s)", dsttmpname); | |
117 | } | |
118 | ||
119 | if (gotmode) { | |
120 | ret = fchmod(dstfd, mode); | |
121 | if (ret < 0) { | |
122 | err(EX_NOINPUT, "fchmod(%s, %ho)", dsttmpname, mode); | |
123 | } | |
124 | } | |
125 | ||
126 | ret = rename(dsttmpname, dst); | |
127 | if (ret < 0) { | |
128 | err(EX_NOINPUT, "rename(%s, %s)", dsttmpname, dst); | |
129 | } | |
130 | ||
131 | ret = close(dstfd); | |
132 | if (ret < 0) { | |
133 | err(EX_NOINPUT, "close(dst)"); | |
134 | } | |
135 | ||
136 | ret = close(srcfd); | |
137 | if (ret < 0) { | |
138 | err(EX_NOINPUT, "close(src)"); | |
139 | } | |
140 | ||
141 | return 0; | |
142 | } | |
143 | ||
144 | void | |
145 | usage(void) | |
146 | { | |
147 | fprintf(stderr, "Usage: %s [-c] [-S] [-m <mode>] <src> <dst>\n", | |
148 | getprogname()); | |
149 | exit(EX_USAGE); | |
150 | } |