| 1 | /*- |
| 2 | * Copyright (c) 1990, 1993 |
| 3 | * The Regents of the University of California. All rights reserved. |
| 4 | * |
| 5 | * This code is derived from software contributed to Berkeley by |
| 6 | * Hugh Smith at The University of Guelph. |
| 7 | * |
| 8 | * %sccs.include.redist.c% |
| 9 | */ |
| 10 | |
| 11 | #ifndef lint |
| 12 | static char sccsid[] = "@(#)archive.c 8.2 (Berkeley) %G%"; |
| 13 | #endif /* not lint */ |
| 14 | |
| 15 | #include <sys/param.h> |
| 16 | #include <sys/stat.h> |
| 17 | |
| 18 | #include <ar.h> |
| 19 | #include <dirent.h> |
| 20 | #include <err.h> |
| 21 | #include <errno.h> |
| 22 | #include <fcntl.h> |
| 23 | #include <stdio.h> |
| 24 | #include <stdlib.h> |
| 25 | #include <string.h> |
| 26 | #include <unistd.h> |
| 27 | |
| 28 | #include "archive.h" |
| 29 | #include "extern.h" |
| 30 | |
| 31 | typedef struct ar_hdr HDR; |
| 32 | static char hb[sizeof(HDR) + 1]; /* real header */ |
| 33 | |
| 34 | int |
| 35 | open_archive(mode) |
| 36 | int mode; |
| 37 | { |
| 38 | int created, fd, nr; |
| 39 | char buf[SARMAG]; |
| 40 | |
| 41 | created = 0; |
| 42 | if (mode & O_CREAT) { |
| 43 | mode |= O_EXCL; |
| 44 | if ((fd = open(archive, mode, DEFFILEMODE)) >= 0) { |
| 45 | /* POSIX.2 puts create message on stderr. */ |
| 46 | if (!(options & AR_C)) |
| 47 | warnx("creating archive %s", archive); |
| 48 | created = 1; |
| 49 | goto opened; |
| 50 | } |
| 51 | if (errno != EEXIST) |
| 52 | error(archive); |
| 53 | mode &= ~O_EXCL; |
| 54 | } |
| 55 | if ((fd = open(archive, mode, DEFFILEMODE)) < 0) |
| 56 | error(archive); |
| 57 | |
| 58 | /* |
| 59 | * Attempt to place a lock on the opened file - if we get an |
| 60 | * error then someone is already working on this library (or |
| 61 | * it's going across NFS). |
| 62 | */ |
| 63 | opened: if (flock(fd, LOCK_EX|LOCK_NB) && errno != EOPNOTSUPP) |
| 64 | error(archive); |
| 65 | |
| 66 | /* |
| 67 | * If not created, O_RDONLY|O_RDWR indicates that it has to be |
| 68 | * in archive format. |
| 69 | */ |
| 70 | if (!created && |
| 71 | ((mode & O_ACCMODE) == O_RDONLY || (mode & O_ACCMODE) == O_RDWR)) { |
| 72 | if ((nr = read(fd, buf, SARMAG) != SARMAG)) { |
| 73 | if (nr >= 0) |
| 74 | badfmt(); |
| 75 | error(archive); |
| 76 | } else if (bcmp(buf, ARMAG, SARMAG)) |
| 77 | badfmt(); |
| 78 | } else if (write(fd, ARMAG, SARMAG) != SARMAG) |
| 79 | error(archive); |
| 80 | return (fd); |
| 81 | } |
| 82 | |
| 83 | void |
| 84 | close_archive(fd) |
| 85 | int fd; |
| 86 | { |
| 87 | |
| 88 | (void)close(fd); /* Implicit unlock. */ |
| 89 | } |
| 90 | |
| 91 | /* Convert ar header field to an integer. */ |
| 92 | #define AR_ATOI(from, to, len, base) { \ |
| 93 | memmove(buf, from, len); \ |
| 94 | buf[len] = '\0'; \ |
| 95 | to = strtol(buf, (char **)NULL, base); \ |
| 96 | } |
| 97 | |
| 98 | /* |
| 99 | * get_arobj -- |
| 100 | * read the archive header for this member |
| 101 | */ |
| 102 | int |
| 103 | get_arobj(fd) |
| 104 | int fd; |
| 105 | { |
| 106 | struct ar_hdr *hdr; |
| 107 | int len, nr; |
| 108 | char *p, buf[20]; |
| 109 | |
| 110 | nr = read(fd, hb, sizeof(HDR)); |
| 111 | if (nr != sizeof(HDR)) { |
| 112 | if (!nr) |
| 113 | return (0); |
| 114 | if (nr < 0) |
| 115 | error(archive); |
| 116 | badfmt(); |
| 117 | } |
| 118 | |
| 119 | hdr = (struct ar_hdr *)hb; |
| 120 | if (strncmp(hdr->ar_fmag, ARFMAG, sizeof(ARFMAG) - 1)) |
| 121 | badfmt(); |
| 122 | |
| 123 | /* Convert the header into the internal format. */ |
| 124 | #define DECIMAL 10 |
| 125 | #define OCTAL 8 |
| 126 | |
| 127 | AR_ATOI(hdr->ar_date, chdr.date, sizeof(hdr->ar_date), DECIMAL); |
| 128 | AR_ATOI(hdr->ar_uid, chdr.uid, sizeof(hdr->ar_uid), DECIMAL); |
| 129 | AR_ATOI(hdr->ar_gid, chdr.gid, sizeof(hdr->ar_gid), DECIMAL); |
| 130 | AR_ATOI(hdr->ar_mode, chdr.mode, sizeof(hdr->ar_mode), OCTAL); |
| 131 | AR_ATOI(hdr->ar_size, chdr.size, sizeof(hdr->ar_size), DECIMAL); |
| 132 | |
| 133 | /* Leading spaces should never happen. */ |
| 134 | if (hdr->ar_name[0] == ' ') |
| 135 | badfmt(); |
| 136 | |
| 137 | /* |
| 138 | * Long name support. Set the "real" size of the file, and the |
| 139 | * long name flag/size. |
| 140 | */ |
| 141 | if (!bcmp(hdr->ar_name, AR_EFMT1, sizeof(AR_EFMT1) - 1)) { |
| 142 | chdr.lname = len = atoi(hdr->ar_name + sizeof(AR_EFMT1) - 1); |
| 143 | if (len <= 0 || len > MAXNAMLEN) |
| 144 | badfmt(); |
| 145 | nr = read(fd, chdr.name, len); |
| 146 | if (nr != len) { |
| 147 | if (nr < 0) |
| 148 | error(archive); |
| 149 | badfmt(); |
| 150 | } |
| 151 | chdr.name[len] = 0; |
| 152 | chdr.size -= len; |
| 153 | } else { |
| 154 | chdr.lname = 0; |
| 155 | memmove(chdr.name, hdr->ar_name, sizeof(hdr->ar_name)); |
| 156 | |
| 157 | /* Strip trailing spaces, null terminate. */ |
| 158 | for (p = chdr.name + sizeof(hdr->ar_name) - 1; *p == ' '; --p); |
| 159 | *++p = '\0'; |
| 160 | } |
| 161 | return (1); |
| 162 | } |
| 163 | |
| 164 | static int already_written; |
| 165 | |
| 166 | /* |
| 167 | * put_arobj -- |
| 168 | * Write an archive member to a file. |
| 169 | */ |
| 170 | void |
| 171 | put_arobj(cfp, sb) |
| 172 | CF *cfp; |
| 173 | struct stat *sb; |
| 174 | { |
| 175 | int lname; |
| 176 | char *name; |
| 177 | struct ar_hdr *hdr; |
| 178 | off_t size; |
| 179 | |
| 180 | /* |
| 181 | * If passed an sb structure, reading a file from disk. Get stat(2) |
| 182 | * information, build a name and construct a header. (Files are named |
| 183 | * by their last component in the archive.) If not, then just write |
| 184 | * the last header read. |
| 185 | */ |
| 186 | if (sb) { |
| 187 | name = rname(cfp->rname); |
| 188 | (void)fstat(cfp->rfd, sb); |
| 189 | |
| 190 | /* |
| 191 | * If not truncating names and the name is too long or contains |
| 192 | * a space, use extended format 1. |
| 193 | */ |
| 194 | lname = strlen(name); |
| 195 | if (options & AR_TR) { |
| 196 | if (lname > OLDARMAXNAME) { |
| 197 | (void)fflush(stdout); |
| 198 | warnx("warning: %s truncated to %.*s\n", |
| 199 | name, OLDARMAXNAME, name); |
| 200 | (void)fflush(stderr); |
| 201 | } |
| 202 | (void)sprintf(hb, HDR3, name, sb->st_mtimespec.ts_sec, |
| 203 | sb->st_uid, sb->st_gid, sb->st_mode, sb->st_size, |
| 204 | ARFMAG); |
| 205 | lname = 0; |
| 206 | } else if (lname > sizeof(hdr->ar_name) || strchr(name, ' ')) |
| 207 | (void)sprintf(hb, HDR1, AR_EFMT1, lname, |
| 208 | sb->st_mtimespec.ts_sec, sb->st_uid, sb->st_gid, |
| 209 | sb->st_mode, sb->st_size + lname, ARFMAG); |
| 210 | else { |
| 211 | lname = 0; |
| 212 | (void)sprintf(hb, HDR2, name, sb->st_mtimespec.ts_sec, |
| 213 | sb->st_uid, sb->st_gid, sb->st_mode, sb->st_size, |
| 214 | ARFMAG); |
| 215 | } |
| 216 | size = sb->st_size; |
| 217 | } else { |
| 218 | lname = chdr.lname; |
| 219 | name = chdr.name; |
| 220 | size = chdr.size; |
| 221 | } |
| 222 | |
| 223 | if (write(cfp->wfd, hb, sizeof(HDR)) != sizeof(HDR)) |
| 224 | error(cfp->wname); |
| 225 | if (lname) { |
| 226 | if (write(cfp->wfd, name, lname) != lname) |
| 227 | error(cfp->wname); |
| 228 | already_written = lname; |
| 229 | } |
| 230 | copy_ar(cfp, size); |
| 231 | already_written = 0; |
| 232 | } |
| 233 | |
| 234 | /* |
| 235 | * copy_ar -- |
| 236 | * Copy size bytes from one file to another - taking care to handle the |
| 237 | * extra byte (for odd size files) when reading archives and writing an |
| 238 | * extra byte if necessary when adding files to archive. The length of |
| 239 | * the object is the long name plus the object itself; the variable |
| 240 | * already_written gets set if a long name was written. |
| 241 | * |
| 242 | * The padding is really unnecessary, and is almost certainly a remnant |
| 243 | * of early archive formats where the header included binary data which |
| 244 | * a PDP-11 required to start on an even byte boundary. (Or, perhaps, |
| 245 | * because 16-bit word addressed copies were faster?) Anyhow, it should |
| 246 | * have been ripped out long ago. |
| 247 | */ |
| 248 | void |
| 249 | copy_ar(cfp, size) |
| 250 | CF *cfp; |
| 251 | off_t size; |
| 252 | { |
| 253 | static char pad = '\n'; |
| 254 | off_t sz; |
| 255 | int from, nr, nw, off, to; |
| 256 | char buf[8*1024]; |
| 257 | |
| 258 | if (!(sz = size)) |
| 259 | return; |
| 260 | |
| 261 | from = cfp->rfd; |
| 262 | to = cfp->wfd; |
| 263 | sz = size; |
| 264 | while (sz && (nr = read(from, buf, MIN(sz, sizeof(buf)))) > 0) { |
| 265 | sz -= nr; |
| 266 | for (off = 0; off < nr; nr -= off, off += nw) |
| 267 | if ((nw = write(to, buf + off, nr)) < 0) |
| 268 | error(cfp->wname); |
| 269 | } |
| 270 | if (sz) { |
| 271 | if (nr == 0) |
| 272 | badfmt(); |
| 273 | error(cfp->rname); |
| 274 | } |
| 275 | |
| 276 | if (cfp->flags & RPAD && size & 1 && (nr = read(from, buf, 1)) != 1) { |
| 277 | if (nr == 0) |
| 278 | badfmt(); |
| 279 | error(cfp->rname); |
| 280 | } |
| 281 | if (cfp->flags & WPAD && (size + already_written) & 1 && |
| 282 | write(to, &pad, 1) != 1) |
| 283 | error(cfp->wname); |
| 284 | } |
| 285 | |
| 286 | /* |
| 287 | * skip_arobj - |
| 288 | * Skip over an object -- taking care to skip the pad bytes. |
| 289 | */ |
| 290 | void |
| 291 | skip_arobj(fd) |
| 292 | int fd; |
| 293 | { |
| 294 | off_t len; |
| 295 | |
| 296 | len = chdr.size + (chdr.size + chdr.lname & 1); |
| 297 | if (lseek(fd, len, SEEK_CUR) == (off_t)-1) |
| 298 | error(archive); |
| 299 | } |