Commit | Line | Data |
---|---|---|
20f08463 | 1 | /*- |
ed554bc5 | 2 | * Copyright (c) 1990, 1993, 1994 |
ad787160 | 3 | * The Regents of the University of California. All rights reserved. |
20f08463 KB |
4 | * |
5 | * This code is derived from software contributed to Berkeley by | |
6 | * Hugh Smith at The University of Guelph. | |
7 | * | |
ad787160 C |
8 | * Redistribution and use in source and binary forms, with or without |
9 | * modification, are permitted provided that the following conditions | |
10 | * are met: | |
11 | * 1. Redistributions of source code must retain the above copyright | |
12 | * notice, this list of conditions and the following disclaimer. | |
13 | * 2. Redistributions in binary form must reproduce the above copyright | |
14 | * notice, this list of conditions and the following disclaimer in the | |
15 | * documentation and/or other materials provided with the distribution. | |
16 | * 3. All advertising materials mentioning features or use of this software | |
17 | * must display the following acknowledgement: | |
18 | * This product includes software developed by the University of | |
19 | * California, Berkeley and its contributors. | |
20 | * 4. Neither the name of the University nor the names of its contributors | |
21 | * may be used to endorse or promote products derived from this software | |
22 | * without specific prior written permission. | |
23 | * | |
24 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | |
25 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
26 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
27 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | |
28 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
29 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
30 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
31 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
32 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
33 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
34 | * SUCH DAMAGE. | |
20f08463 KB |
35 | */ |
36 | ||
37 | #ifndef lint | |
fd88f5c5 | 38 | static char sccsid[] = "@(#)replace.c 8.4 (Berkeley) 4/27/95"; |
20f08463 KB |
39 | #endif /* not lint */ |
40 | ||
41 | #include <sys/param.h> | |
42 | #include <sys/stat.h> | |
706c0358 | 43 | |
20f08463 | 44 | #include <ar.h> |
706c0358 JSP |
45 | #include <dirent.h> |
46 | #include <err.h> | |
47 | #include <fcntl.h> | |
20f08463 | 48 | #include <stdio.h> |
1d7fb98b | 49 | #include <string.h> |
706c0358 JSP |
50 | #include <unistd.h> |
51 | ||
20f08463 | 52 | #include "archive.h" |
1d7fb98b | 53 | #include "extern.h" |
20f08463 | 54 | |
20f08463 KB |
55 | /* |
56 | * replace -- | |
57 | * Replace or add named members to archive. Entries already in the | |
58 | * archive are swapped in place. Others are added before or after | |
59 | * the key entry, based on the a, b and i options. If the u option | |
60 | * is specified, modification dates select for replacement. | |
61 | */ | |
706c0358 | 62 | int |
20f08463 KB |
63 | replace(argv) |
64 | char **argv; | |
65 | { | |
706c0358 JSP |
66 | char *file; |
67 | int afd, curfd, errflg, exists, mods, sfd, tfd1, tfd2; | |
20f08463 KB |
68 | struct stat sb; |
69 | CF cf; | |
70 | off_t size, tsize; | |
20f08463 | 71 | |
706c0358 | 72 | errflg = 0; |
20f08463 KB |
73 | /* |
74 | * If doesn't exist, simply append to the archive. There's | |
75 | * a race here, but it's pretty short, and not worth fixing. | |
76 | */ | |
77 | exists = !stat(archive, &sb); | |
78 | afd = open_archive(O_CREAT|O_RDWR); | |
79 | ||
80 | if (!exists) { | |
81 | tfd1 = -1; | |
82 | tfd2 = tmp(); | |
83 | goto append; | |
84 | } | |
85 | ||
86 | tfd1 = tmp(); /* Files before key file. */ | |
87 | tfd2 = tmp(); /* Files after key file. */ | |
20f08463 KB |
88 | |
89 | /* | |
90 | * Break archive into two parts -- entries before and after the key | |
91 | * entry. If positioning before the key, place the key at the | |
92 | * beginning of the after key entries and if positioning after the | |
93 | * key, place the key at the end of the before key entries. Put it | |
94 | * all back together at the end. | |
95 | */ | |
bf53959e | 96 | mods = (options & (AR_A|AR_B)); |
ad4425ce | 97 | for (curfd = tfd1; get_arobj(afd);) { |
1d7fb98b | 98 | if (*argv && (file = files(argv))) { |
20f08463 | 99 | if ((sfd = open(file, O_RDONLY)) < 0) { |
706c0358 JSP |
100 | errflg = 1; |
101 | warn("%s", file); | |
20f08463 KB |
102 | goto useold; |
103 | } | |
104 | (void)fstat(sfd, &sb); | |
317d4892 KB |
105 | if (options & AR_U && sb.st_mtime <= chdr.date) { |
106 | (void)close(sfd); | |
20f08463 | 107 | goto useold; |
317d4892 | 108 | } |
20f08463 KB |
109 | |
110 | if (options & AR_V) | |
af52bb8a | 111 | (void)printf("r - %s\n", file); |
20f08463 | 112 | |
abe76123 | 113 | /* Read from disk, write to an archive; pad on write */ |
20f08463 | 114 | SETCF(sfd, file, curfd, tname, WPAD); |
ce478bb7 | 115 | put_arobj(&cf, &sb); |
20f08463 | 116 | (void)close(sfd); |
ce478bb7 | 117 | skip_arobj(afd); |
20f08463 KB |
118 | continue; |
119 | } | |
120 | ||
121 | if (mods && compare(posname)) { | |
122 | mods = 0; | |
123 | if (options & AR_B) | |
124 | curfd = tfd2; | |
abe76123 | 125 | /* Read and write to an archive; pad on both. */ |
20f08463 | 126 | SETCF(afd, archive, curfd, tname, RPAD|WPAD); |
ce478bb7 | 127 | put_arobj(&cf, (struct stat *)NULL); |
20f08463 KB |
128 | if (options & AR_A) |
129 | curfd = tfd2; | |
130 | } else { | |
abe76123 | 131 | /* Read and write to an archive; pad on both. */ |
20f08463 | 132 | useold: SETCF(afd, archive, curfd, tname, RPAD|WPAD); |
ce478bb7 | 133 | put_arobj(&cf, (struct stat *)NULL); |
20f08463 KB |
134 | } |
135 | } | |
136 | ||
137 | if (mods) { | |
706c0358 | 138 | warnx("%s: archive member not found", posarg); |
20f08463 | 139 | close_archive(afd); |
706c0358 | 140 | return (1); |
20f08463 KB |
141 | } |
142 | ||
143 | /* Append any left-over arguments to the end of the after file. */ | |
144 | append: while (file = *argv++) { | |
145 | if (options & AR_V) | |
af52bb8a | 146 | (void)printf("a - %s\n", file); |
20f08463 | 147 | if ((sfd = open(file, O_RDONLY)) < 0) { |
706c0358 JSP |
148 | errflg = 1; |
149 | warn("%s", file); | |
20f08463 KB |
150 | continue; |
151 | } | |
152 | (void)fstat(sfd, &sb); | |
abe76123 | 153 | /* Read from disk, write to an archive; pad on write. */ |
bf53959e KB |
154 | SETCF(sfd, file, |
155 | options & (AR_A|AR_B) ? tfd1 : tfd2, tname, WPAD); | |
ce478bb7 | 156 | put_arobj(&cf, &sb); |
20f08463 KB |
157 | (void)close(sfd); |
158 | } | |
159 | ||
160 | (void)lseek(afd, (off_t)SARMAG, SEEK_SET); | |
161 | ||
abe76123 | 162 | SETCF(tfd1, tname, afd, archive, NOPAD); |
20f08463 KB |
163 | if (tfd1 != -1) { |
164 | tsize = size = lseek(tfd1, (off_t)0, SEEK_CUR); | |
165 | (void)lseek(tfd1, (off_t)0, SEEK_SET); | |
ce478bb7 | 166 | copy_ar(&cf, size); |
20f08463 KB |
167 | } else |
168 | tsize = 0; | |
169 | ||
170 | tsize += size = lseek(tfd2, (off_t)0, SEEK_CUR); | |
171 | (void)lseek(tfd2, (off_t)0, SEEK_SET); | |
172 | cf.rfd = tfd2; | |
ce478bb7 | 173 | copy_ar(&cf, size); |
20f08463 KB |
174 | |
175 | (void)ftruncate(afd, tsize + SARMAG); | |
176 | close_archive(afd); | |
706c0358 | 177 | return (errflg); |
20f08463 | 178 | } |