sync
[bitrig.git] / bin / pax / tables.c
blobcfbf4a55c307796c8d7d089287605ab83312c072
1 /* $OpenBSD: tables.c,v 1.27 2012/12/04 02:24:45 deraadt Exp $ */
2 /* $NetBSD: tables.c,v 1.4 1995/03/21 09:07:45 cgd Exp $ */
4 /*-
5 * Copyright (c) 1992 Keith Muller.
6 * Copyright (c) 1992, 1993
7 * The Regents of the University of California. All rights reserved.
9 * This code is derived from software contributed to Berkeley by
10 * Keith Muller of the University of California, San Diego.
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
14 * are met:
15 * 1. Redistributions of source code must retain the above copyright
16 * notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above copyright
18 * notice, this list of conditions and the following disclaimer in the
19 * documentation and/or other materials provided with the distribution.
20 * 3. 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.
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.
37 #include <sys/types.h>
38 #include <sys/param.h>
39 #include <sys/time.h>
40 #include <sys/stat.h>
41 #include <sys/fcntl.h>
42 #include <stdio.h>
43 #include <string.h>
44 #include <unistd.h>
45 #include <errno.h>
46 #include <stdlib.h>
47 #include "pax.h"
48 #include "tables.h"
49 #include "extern.h"
52 * Routines for controlling the contents of all the different databases pax
53 * keeps. Tables are dynamically created only when they are needed. The
54 * goal was speed and the ability to work with HUGE archives. The databases
55 * were kept simple, but do have complex rules for when the contents change.
56 * As of this writing, the posix library functions were more complex than
57 * needed for this application (pax databases have very short lifetimes and
58 * do not survive after pax is finished). Pax is required to handle very
59 * large archives. These database routines carefully combine memory usage and
60 * temporary file storage in ways which will not significantly impact runtime
61 * performance while allowing the largest possible archives to be handled.
62 * Trying to force the fit to the posix database routines was not considered
63 * time well spent.
66 static HRDLNK **ltab = NULL; /* hard link table for detecting hard links */
67 static FTM **ftab = NULL; /* file time table for updating arch */
68 static NAMT **ntab = NULL; /* interactive rename storage table */
69 static DEVT **dtab = NULL; /* device/inode mapping tables */
70 static ATDIR **atab = NULL; /* file tree directory time reset table */
71 static DIRDATA *dirp = NULL; /* storage for setting created dir time/mode */
72 static size_t dirsize; /* size of dirp table */
73 static long dircnt = 0; /* entries in dir time/mode storage */
74 static int ffd = -1; /* tmp file for file time table name storage */
76 static DEVT *chk_dev(dev_t, int);
79 * hard link table routines
81 * The hard link table tries to detect hard links to files using the device and
82 * inode values. We do this when writing an archive, so we can tell the format
83 * write routine that this file is a hard link to another file. The format
84 * write routine then can store this file in whatever way it wants (as a hard
85 * link if the format supports that like tar, or ignore this info like cpio).
86 * (Actually a field in the format driver table tells us if the format wants
87 * hard link info. if not, we do not waste time looking for them). We also use
88 * the same table when reading an archive. In that situation, this table is
89 * used by the format read routine to detect hard links from stored dev and
90 * inode numbers (like cpio). This will allow pax to create a link when one
91 * can be detected by the archive format.
95 * lnk_start
96 * Creates the hard link table.
97 * Return:
98 * 0 if created, -1 if failure
102 lnk_start(void)
104 if (ltab != NULL)
105 return(0);
106 if ((ltab = (HRDLNK **)calloc(L_TAB_SZ, sizeof(HRDLNK *))) == NULL) {
107 paxwarn(1, "Cannot allocate memory for hard link table");
108 return(-1);
110 return(0);
114 * chk_lnk()
115 * Looks up entry in hard link hash table. If found, it copies the name
116 * of the file it is linked to (we already saw that file) into ln_name.
117 * lnkcnt is decremented and if goes to 1 the node is deleted from the
118 * database. (We have seen all the links to this file). If not found,
119 * we add the file to the database if it has the potential for having
120 * hard links to other files we may process (it has a link count > 1)
121 * Return:
122 * if found returns 1; if not found returns 0; -1 on error
126 chk_lnk(ARCHD *arcn)
128 HRDLNK *pt;
129 HRDLNK **ppt;
130 u_int indx;
132 if (ltab == NULL)
133 return(-1);
135 * ignore those nodes that cannot have hard links
137 if ((arcn->type == PAX_DIR) || (arcn->sb.st_nlink <= 1))
138 return(0);
141 * hash inode number and look for this file
143 indx = ((unsigned)arcn->sb.st_ino) % L_TAB_SZ;
144 if ((pt = ltab[indx]) != NULL) {
146 * its hash chain in not empty, walk down looking for it
148 ppt = &(ltab[indx]);
149 while (pt != NULL) {
150 if ((pt->ino == arcn->sb.st_ino) &&
151 (pt->dev == arcn->sb.st_dev))
152 break;
153 ppt = &(pt->fow);
154 pt = pt->fow;
157 if (pt != NULL) {
159 * found a link. set the node type and copy in the
160 * name of the file it is to link to. we need to
161 * handle hardlinks to regular files differently than
162 * other links.
164 arcn->ln_nlen = strlcpy(arcn->ln_name, pt->name,
165 sizeof(arcn->ln_name));
166 /* XXX truncate? */
167 if (arcn->nlen >= sizeof(arcn->name))
168 arcn->nlen = sizeof(arcn->name) - 1;
169 if (arcn->type == PAX_REG)
170 arcn->type = PAX_HRG;
171 else
172 arcn->type = PAX_HLK;
175 * if we have found all the links to this file, remove
176 * it from the database
178 if (--pt->nlink <= 1) {
179 *ppt = pt->fow;
180 (void)free((char *)pt->name);
181 (void)free((char *)pt);
183 return(1);
188 * we never saw this file before. It has links so we add it to the
189 * front of this hash chain
191 if ((pt = (HRDLNK *)malloc(sizeof(HRDLNK))) != NULL) {
192 if ((pt->name = strdup(arcn->name)) != NULL) {
193 pt->dev = arcn->sb.st_dev;
194 pt->ino = arcn->sb.st_ino;
195 pt->nlink = arcn->sb.st_nlink;
196 pt->fow = ltab[indx];
197 ltab[indx] = pt;
198 return(0);
200 (void)free((char *)pt);
203 paxwarn(1, "Hard link table out of memory");
204 return(-1);
208 * purg_lnk
209 * remove reference for a file that we may have added to the data base as
210 * a potential source for hard links. We ended up not using the file, so
211 * we do not want to accidently point another file at it later on.
214 void
215 purg_lnk(ARCHD *arcn)
217 HRDLNK *pt;
218 HRDLNK **ppt;
219 u_int indx;
221 if (ltab == NULL)
222 return;
224 * do not bother to look if it could not be in the database
226 if ((arcn->sb.st_nlink <= 1) || (arcn->type == PAX_DIR) ||
227 (arcn->type == PAX_HLK) || (arcn->type == PAX_HRG))
228 return;
231 * find the hash chain for this inode value, if empty return
233 indx = ((unsigned)arcn->sb.st_ino) % L_TAB_SZ;
234 if ((pt = ltab[indx]) == NULL)
235 return;
238 * walk down the list looking for the inode/dev pair, unlink and
239 * free if found
241 ppt = &(ltab[indx]);
242 while (pt != NULL) {
243 if ((pt->ino == arcn->sb.st_ino) &&
244 (pt->dev == arcn->sb.st_dev))
245 break;
246 ppt = &(pt->fow);
247 pt = pt->fow;
249 if (pt == NULL)
250 return;
253 * remove and free it
255 *ppt = pt->fow;
256 (void)free((char *)pt->name);
257 (void)free((char *)pt);
261 * lnk_end()
262 * pull apart a existing link table so we can reuse it. We do this between
263 * read and write phases of append with update. (The format may have
264 * used the link table, and we need to start with a fresh table for the
265 * write phase
268 void
269 lnk_end(void)
271 int i;
272 HRDLNK *pt;
273 HRDLNK *ppt;
275 if (ltab == NULL)
276 return;
278 for (i = 0; i < L_TAB_SZ; ++i) {
279 if (ltab[i] == NULL)
280 continue;
281 pt = ltab[i];
282 ltab[i] = NULL;
285 * free up each entry on this chain
287 while (pt != NULL) {
288 ppt = pt;
289 pt = ppt->fow;
290 (void)free((char *)ppt->name);
291 (void)free((char *)ppt);
294 return;
298 * modification time table routines
300 * The modification time table keeps track of last modification times for all
301 * files stored in an archive during a write phase when -u is set. We only
302 * add a file to the archive if it is newer than a file with the same name
303 * already stored on the archive (if there is no other file with the same
304 * name on the archive it is added). This applies to writes and appends.
305 * An append with an -u must read the archive and store the modification time
306 * for every file on that archive before starting the write phase. It is clear
307 * that this is one HUGE database. To save memory space, the actual file names
308 * are stored in a scratch file and indexed by an in-memory hash table. The
309 * hash table is indexed by hashing the file path. The nodes in the table store
310 * the length of the filename and the lseek offset within the scratch file
311 * where the actual name is stored. Since there are never any deletions from
312 * this table, fragmentation of the scratch file is never a issue. Lookups
313 * seem to not exhibit any locality at all (files in the database are rarely
314 * looked up more than once...), so caching is just a waste of memory. The
315 * only limitation is the amount of scratch file space available to store the
316 * path names.
320 * ftime_start()
321 * create the file time hash table and open for read/write the scratch
322 * file. (after created it is unlinked, so when we exit we leave
323 * no witnesses).
324 * Return:
325 * 0 if the table and file was created ok, -1 otherwise
329 ftime_start(void)
332 if (ftab != NULL)
333 return(0);
334 if ((ftab = (FTM **)calloc(F_TAB_SZ, sizeof(FTM *))) == NULL) {
335 paxwarn(1, "Cannot allocate memory for file time table");
336 return(-1);
340 * get random name and create temporary scratch file, unlink name
341 * so it will get removed on exit
343 memcpy(tempbase, _TFILE_BASE, sizeof(_TFILE_BASE));
344 if ((ffd = mkstemp(tempfile)) < 0) {
345 syswarn(1, errno, "Unable to create temporary file: %s",
346 tempfile);
347 return(-1);
349 (void)unlink(tempfile);
351 return(0);
355 * chk_ftime()
356 * looks up entry in file time hash table. If not found, the file is
357 * added to the hash table and the file named stored in the scratch file.
358 * If a file with the same name is found, the file times are compared and
359 * the most recent file time is retained. If the new file was younger (or
360 * was not in the database) the new file is selected for storage.
361 * Return:
362 * 0 if file should be added to the archive, 1 if it should be skipped,
363 * -1 on error
367 chk_ftime(ARCHD *arcn)
369 FTM *pt;
370 int namelen;
371 u_int indx;
372 char ckname[PAXPATHLEN+1];
375 * no info, go ahead and add to archive
377 if (ftab == NULL)
378 return(0);
381 * hash the pathname and look up in table
383 namelen = arcn->nlen;
384 indx = st_hash(arcn->name, namelen, F_TAB_SZ);
385 if ((pt = ftab[indx]) != NULL) {
387 * the hash chain is not empty, walk down looking for match
388 * only read up the path names if the lengths match, speeds
389 * up the search a lot
391 while (pt != NULL) {
392 if (pt->namelen == namelen) {
394 * potential match, have to read the name
395 * from the scratch file.
397 if (lseek(ffd,pt->seek,SEEK_SET) != pt->seek) {
398 syswarn(1, errno,
399 "Failed ftime table seek");
400 return(-1);
402 if (read(ffd, ckname, namelen) != namelen) {
403 syswarn(1, errno,
404 "Failed ftime table read");
405 return(-1);
409 * if the names match, we are done
411 if (!strncmp(ckname, arcn->name, namelen))
412 break;
416 * try the next entry on the chain
418 pt = pt->fow;
421 if (pt != NULL) {
423 * found the file, compare the times, save the newer
425 if (arcn->sb.st_mtime > pt->mtime) {
427 * file is newer
429 pt->mtime = arcn->sb.st_mtime;
430 return(0);
433 * file is older
435 return(1);
440 * not in table, add it
442 if ((pt = (FTM *)malloc(sizeof(FTM))) != NULL) {
444 * add the name at the end of the scratch file, saving the
445 * offset. add the file to the head of the hash chain
447 if ((pt->seek = lseek(ffd, (off_t)0, SEEK_END)) >= 0) {
448 if (write(ffd, arcn->name, namelen) == namelen) {
449 pt->mtime = arcn->sb.st_mtime;
450 pt->namelen = namelen;
451 pt->fow = ftab[indx];
452 ftab[indx] = pt;
453 return(0);
455 syswarn(1, errno, "Failed write to file time table");
456 } else
457 syswarn(1, errno, "Failed seek on file time table");
458 } else
459 paxwarn(1, "File time table ran out of memory");
461 if (pt != NULL)
462 (void)free((char *)pt);
463 return(-1);
467 * Interactive rename table routines
469 * The interactive rename table keeps track of the new names that the user
470 * assigns to files from tty input. Since this map is unique for each file
471 * we must store it in case there is a reference to the file later in archive
472 * (a link). Otherwise we will be unable to find the file we know was
473 * extracted. The remapping of these files is stored in a memory based hash
474 * table (it is assumed since input must come from /dev/tty, it is unlikely to
475 * be a very large table).
479 * name_start()
480 * create the interactive rename table
481 * Return:
482 * 0 if successful, -1 otherwise
486 name_start(void)
488 if (ntab != NULL)
489 return(0);
490 if ((ntab = (NAMT **)calloc(N_TAB_SZ, sizeof(NAMT *))) == NULL) {
491 paxwarn(1, "Cannot allocate memory for interactive rename table");
492 return(-1);
494 return(0);
498 * add_name()
499 * add the new name to old name mapping just created by the user.
500 * If an old name mapping is found (there may be duplicate names on an
501 * archive) only the most recent is kept.
502 * Return:
503 * 0 if added, -1 otherwise
507 add_name(char *oname, int onamelen, char *nname)
509 NAMT *pt;
510 u_int indx;
512 if (ntab == NULL) {
514 * should never happen
516 paxwarn(0, "No interactive rename table, links may fail");
517 return(0);
521 * look to see if we have already mapped this file, if so we
522 * will update it
524 indx = st_hash(oname, onamelen, N_TAB_SZ);
525 if ((pt = ntab[indx]) != NULL) {
527 * look down the has chain for the file
529 while ((pt != NULL) && (strcmp(oname, pt->oname) != 0))
530 pt = pt->fow;
532 if (pt != NULL) {
534 * found an old mapping, replace it with the new one
535 * the user just input (if it is different)
537 if (strcmp(nname, pt->nname) == 0)
538 return(0);
540 (void)free((char *)pt->nname);
541 if ((pt->nname = strdup(nname)) == NULL) {
542 paxwarn(1, "Cannot update rename table");
543 return(-1);
545 return(0);
550 * this is a new mapping, add it to the table
552 if ((pt = (NAMT *)malloc(sizeof(NAMT))) != NULL) {
553 if ((pt->oname = strdup(oname)) != NULL) {
554 if ((pt->nname = strdup(nname)) != NULL) {
555 pt->fow = ntab[indx];
556 ntab[indx] = pt;
557 return(0);
559 (void)free((char *)pt->oname);
561 (void)free((char *)pt);
563 paxwarn(1, "Interactive rename table out of memory");
564 return(-1);
568 * sub_name()
569 * look up a link name to see if it points at a file that has been
570 * remapped by the user. If found, the link is adjusted to contain the
571 * new name (oname is the link to name)
574 void
575 sub_name(char *oname, int *onamelen, size_t onamesize)
577 NAMT *pt;
578 u_int indx;
580 if (ntab == NULL)
581 return;
583 * look the name up in the hash table
585 indx = st_hash(oname, *onamelen, N_TAB_SZ);
586 if ((pt = ntab[indx]) == NULL)
587 return;
589 while (pt != NULL) {
591 * walk down the hash chain looking for a match
593 if (strcmp(oname, pt->oname) == 0) {
595 * found it, replace it with the new name
596 * and return (we know that oname has enough space)
598 *onamelen = strlcpy(oname, pt->nname, onamesize);
599 if (*onamelen >= onamesize)
600 *onamelen = onamesize - 1; /* XXX truncate? */
601 return;
603 pt = pt->fow;
607 * no match, just return
609 return;
613 * device/inode mapping table routines
614 * (used with formats that store device and inodes fields)
616 * device/inode mapping tables remap the device field in a archive header. The
617 * device/inode fields are used to determine when files are hard links to each
618 * other. However these values have very little meaning outside of that. This
619 * database is used to solve one of two different problems.
621 * 1) when files are appended to an archive, while the new files may have hard
622 * links to each other, you cannot determine if they have hard links to any
623 * file already stored on the archive from a prior run of pax. We must assume
624 * that these inode/device pairs are unique only within a SINGLE run of pax
625 * (which adds a set of files to an archive). So we have to make sure the
626 * inode/dev pairs we add each time are always unique. We do this by observing
627 * while the inode field is very dense, the use of the dev field is fairly
628 * sparse. Within each run of pax, we remap any device number of a new archive
629 * member that has a device number used in a prior run and already stored in a
630 * file on the archive. During the read phase of the append, we store the
631 * device numbers used and mark them to not be used by any file during the
632 * write phase. If during write we go to use one of those old device numbers,
633 * we remap it to a new value.
635 * 2) Often the fields in the archive header used to store these values are
636 * too small to store the entire value. The result is an inode or device value
637 * which can be truncated. This really can foul up an archive. With truncation
638 * we end up creating links between files that are really not links (after
639 * truncation the inodes are the same value). We address that by detecting
640 * truncation and forcing a remap of the device field to split truncated
641 * inodes away from each other. Each truncation creates a pattern of bits that
642 * are removed. We use this pattern of truncated bits to partition the inodes
643 * on a single device to many different devices (each one represented by the
644 * truncated bit pattern). All inodes on the same device that have the same
645 * truncation pattern are mapped to the same new device. Two inodes that
646 * truncate to the same value clearly will always have different truncation
647 * bit patterns, so they will be split from away each other. When we spot
648 * device truncation we remap the device number to a non truncated value.
649 * (for more info see table.h for the data structures involved).
653 * dev_start()
654 * create the device mapping table
655 * Return:
656 * 0 if successful, -1 otherwise
660 dev_start(void)
662 if (dtab != NULL)
663 return(0);
664 if ((dtab = (DEVT **)calloc(D_TAB_SZ, sizeof(DEVT *))) == NULL) {
665 paxwarn(1, "Cannot allocate memory for device mapping table");
666 return(-1);
668 return(0);
672 * add_dev()
673 * add a device number to the table. this will force the device to be
674 * remapped to a new value if it be used during a write phase. This
675 * function is called during the read phase of an append to prohibit the
676 * use of any device number already in the archive.
677 * Return:
678 * 0 if added ok, -1 otherwise
682 add_dev(ARCHD *arcn)
684 if (chk_dev(arcn->sb.st_dev, 1) == NULL)
685 return(-1);
686 return(0);
690 * chk_dev()
691 * check for a device value in the device table. If not found and the add
692 * flag is set, it is added. This does NOT assign any mapping values, just
693 * adds the device number as one that need to be remapped. If this device
694 * is already mapped, just return with a pointer to that entry.
695 * Return:
696 * pointer to the entry for this device in the device map table. Null
697 * if the add flag is not set and the device is not in the table (it is
698 * not been seen yet). If add is set and the device cannot be added, null
699 * is returned (indicates an error).
702 static DEVT *
703 chk_dev(dev_t dev, int add)
705 DEVT *pt;
706 u_int indx;
708 if (dtab == NULL)
709 return(NULL);
711 * look to see if this device is already in the table
713 indx = ((unsigned)dev) % D_TAB_SZ;
714 if ((pt = dtab[indx]) != NULL) {
715 while ((pt != NULL) && (pt->dev != dev))
716 pt = pt->fow;
719 * found it, return a pointer to it
721 if (pt != NULL)
722 return(pt);
726 * not in table, we add it only if told to as this may just be a check
727 * to see if a device number is being used.
729 if (add == 0)
730 return(NULL);
733 * allocate a node for this device and add it to the front of the hash
734 * chain. Note we do not assign remaps values here, so the pt->list
735 * list must be NULL.
737 if ((pt = (DEVT *)malloc(sizeof(DEVT))) == NULL) {
738 paxwarn(1, "Device map table out of memory");
739 return(NULL);
741 pt->dev = dev;
742 pt->list = NULL;
743 pt->fow = dtab[indx];
744 dtab[indx] = pt;
745 return(pt);
748 * map_dev()
749 * given an inode and device storage mask (the mask has a 1 for each bit
750 * the archive format is able to store in a header), we check for inode
751 * and device truncation and remap the device as required. Device mapping
752 * can also occur when during the read phase of append a device number was
753 * seen (and was marked as do not use during the write phase). WE ASSUME
754 * that unsigned longs are the same size or bigger than the fields used
755 * for ino_t and dev_t. If not the types will have to be changed.
756 * Return:
757 * 0 if all ok, -1 otherwise.
761 map_dev(ARCHD *arcn, u_long dev_mask, u_long ino_mask)
763 DEVT *pt;
764 DLIST *dpt;
765 static dev_t lastdev = 0; /* next device number to try */
766 int trc_ino = 0;
767 int trc_dev = 0;
768 ino_t trunc_bits = 0;
769 ino_t nino;
771 if (dtab == NULL)
772 return(0);
774 * check for device and inode truncation, and extract the truncated
775 * bit pattern.
777 if ((arcn->sb.st_dev & (dev_t)dev_mask) != arcn->sb.st_dev)
778 ++trc_dev;
779 if ((nino = arcn->sb.st_ino & (ino_t)ino_mask) != arcn->sb.st_ino) {
780 ++trc_ino;
781 trunc_bits = arcn->sb.st_ino & (ino_t)(~ino_mask);
785 * see if this device is already being mapped, look up the device
786 * then find the truncation bit pattern which applies
788 if ((pt = chk_dev(arcn->sb.st_dev, 0)) != NULL) {
790 * this device is already marked to be remapped
792 for (dpt = pt->list; dpt != NULL; dpt = dpt->fow)
793 if (dpt->trunc_bits == trunc_bits)
794 break;
796 if (dpt != NULL) {
798 * we are being remapped for this device and pattern
799 * change the device number to be stored and return
801 arcn->sb.st_dev = dpt->dev;
802 arcn->sb.st_ino = nino;
803 return(0);
805 } else {
807 * this device is not being remapped YET. if we do not have any
808 * form of truncation, we do not need a remap
810 if (!trc_ino && !trc_dev)
811 return(0);
814 * we have truncation, have to add this as a device to remap
816 if ((pt = chk_dev(arcn->sb.st_dev, 1)) == NULL)
817 goto bad;
820 * if we just have a truncated inode, we have to make sure that
821 * all future inodes that do not truncate (they have the
822 * truncation pattern of all 0's) continue to map to the same
823 * device number. We probably have already written inodes with
824 * this device number to the archive with the truncation
825 * pattern of all 0's. So we add the mapping for all 0's to the
826 * same device number.
828 if (!trc_dev && (trunc_bits != 0)) {
829 if ((dpt = (DLIST *)malloc(sizeof(DLIST))) == NULL)
830 goto bad;
831 dpt->trunc_bits = 0;
832 dpt->dev = arcn->sb.st_dev;
833 dpt->fow = pt->list;
834 pt->list = dpt;
839 * look for a device number not being used. We must watch for wrap
840 * around on lastdev (so we do not get stuck looking forever!)
842 while (++lastdev > 0) {
843 if (chk_dev(lastdev, 0) != NULL)
844 continue;
846 * found an unused value. If we have reached truncation point
847 * for this format we are hosed, so we give up. Otherwise we
848 * mark it as being used.
850 if (((lastdev & ((dev_t)dev_mask)) != lastdev) ||
851 (chk_dev(lastdev, 1) == NULL))
852 goto bad;
853 break;
856 if ((lastdev <= 0) || ((dpt = (DLIST *)malloc(sizeof(DLIST))) == NULL))
857 goto bad;
860 * got a new device number, store it under this truncation pattern.
861 * change the device number this file is being stored with.
863 dpt->trunc_bits = trunc_bits;
864 dpt->dev = lastdev;
865 dpt->fow = pt->list;
866 pt->list = dpt;
867 arcn->sb.st_dev = lastdev;
868 arcn->sb.st_ino = nino;
869 return(0);
871 bad:
872 paxwarn(1, "Unable to fix truncated inode/device field when storing %s",
873 arcn->name);
874 paxwarn(0, "Archive may create improper hard links when extracted");
875 return(0);
879 * directory access/mod time reset table routines (for directories READ by pax)
881 * The pax -t flag requires that access times of archive files be the same
882 * before being read by pax. For regular files, access time is restored after
883 * the file has been copied. This database provides the same functionality for
884 * directories read during file tree traversal. Restoring directory access time
885 * is more complex than files since directories may be read several times until
886 * all the descendants in their subtree are visited by fts. Directory access
887 * and modification times are stored during the fts pre-order visit (done
888 * before any descendants in the subtree are visited) and restored after the
889 * fts post-order visit (after all the descendants have been visited). In the
890 * case of premature exit from a subtree (like from the effects of -n), any
891 * directory entries left in this database are reset during final cleanup
892 * operations of pax. Entries are hashed by inode number for fast lookup.
896 * atdir_start()
897 * create the directory access time database for directories READ by pax.
898 * Return:
899 * 0 is created ok, -1 otherwise.
903 atdir_start(void)
905 if (atab != NULL)
906 return(0);
907 if ((atab = (ATDIR **)calloc(A_TAB_SZ, sizeof(ATDIR *))) == NULL) {
908 paxwarn(1,"Cannot allocate space for directory access time table");
909 return(-1);
911 return(0);
916 * atdir_end()
917 * walk through the directory access time table and reset the access time
918 * of any directory who still has an entry left in the database. These
919 * entries are for directories READ by pax
922 void
923 atdir_end(void)
925 ATDIR *pt;
926 int i;
928 if (atab == NULL)
929 return;
931 * for each non-empty hash table entry reset all the directories
932 * chained there.
934 for (i = 0; i < A_TAB_SZ; ++i) {
935 if ((pt = atab[i]) == NULL)
936 continue;
938 * remember to force the times, set_ftime() looks at pmtime
939 * and patime, which only applies to things CREATED by pax,
940 * not read by pax. Read time reset is controlled by -t.
942 for (; pt != NULL; pt = pt->fow)
943 set_ftime(pt->name, pt->mtime, pt->atime, 1);
948 * add_atdir()
949 * add a directory to the directory access time table. Table is hashed
950 * and chained by inode number. This is for directories READ by pax
953 void
954 add_atdir(char *fname, dev_t dev, ino_t ino, time_t mtime, time_t atime)
956 ATDIR *pt;
957 u_int indx;
959 if (atab == NULL)
960 return;
963 * make sure this directory is not already in the table, if so just
964 * return (the older entry always has the correct time). The only
965 * way this will happen is when the same subtree can be traversed by
966 * different args to pax and the -n option is aborting fts out of a
967 * subtree before all the post-order visits have been made.
969 indx = ((unsigned)ino) % A_TAB_SZ;
970 if ((pt = atab[indx]) != NULL) {
971 while (pt != NULL) {
972 if ((pt->ino == ino) && (pt->dev == dev))
973 break;
974 pt = pt->fow;
978 * oops, already there. Leave it alone.
980 if (pt != NULL)
981 return;
985 * add it to the front of the hash chain
987 if ((pt = (ATDIR *)malloc(sizeof(ATDIR))) != NULL) {
988 if ((pt->name = strdup(fname)) != NULL) {
989 pt->dev = dev;
990 pt->ino = ino;
991 pt->mtime = mtime;
992 pt->atime = atime;
993 pt->fow = atab[indx];
994 atab[indx] = pt;
995 return;
997 (void)free((char *)pt);
1000 paxwarn(1, "Directory access time reset table ran out of memory");
1001 return;
1005 * get_atdir()
1006 * look up a directory by inode and device number to obtain the access
1007 * and modification time you want to set to. If found, the modification
1008 * and access time parameters are set and the entry is removed from the
1009 * table (as it is no longer needed). These are for directories READ by
1010 * pax
1011 * Return:
1012 * 0 if found, -1 if not found.
1016 get_atdir(dev_t dev, ino_t ino, time_t *mtime, time_t *atime)
1018 ATDIR *pt;
1019 ATDIR **ppt;
1020 u_int indx;
1022 if (atab == NULL)
1023 return(-1);
1025 * hash by inode and search the chain for an inode and device match
1027 indx = ((unsigned)ino) % A_TAB_SZ;
1028 if ((pt = atab[indx]) == NULL)
1029 return(-1);
1031 ppt = &(atab[indx]);
1032 while (pt != NULL) {
1033 if ((pt->ino == ino) && (pt->dev == dev))
1034 break;
1036 * no match, go to next one
1038 ppt = &(pt->fow);
1039 pt = pt->fow;
1043 * return if we did not find it.
1045 if (pt == NULL)
1046 return(-1);
1049 * found it. return the times and remove the entry from the table.
1051 *ppt = pt->fow;
1052 *mtime = pt->mtime;
1053 *atime = pt->atime;
1054 (void)free((char *)pt->name);
1055 (void)free((char *)pt);
1056 return(0);
1060 * directory access mode and time storage routines (for directories CREATED
1061 * by pax).
1063 * Pax requires that extracted directories, by default, have their access/mod
1064 * times and permissions set to the values specified in the archive. During the
1065 * actions of extracting (and creating the destination subtree during -rw copy)
1066 * directories extracted may be modified after being created. Even worse is
1067 * that these directories may have been created with file permissions which
1068 * prohibits any descendants of these directories from being extracted. When
1069 * directories are created by pax, access rights may be added to permit the
1070 * creation of files in their subtree. Every time pax creates a directory, the
1071 * times and file permissions specified by the archive are stored. After all
1072 * files have been extracted (or copied), these directories have their times
1073 * and file modes reset to the stored values. The directory info is restored in
1074 * reverse order as entries were added to the data file from root to leaf. To
1075 * restore atime properly, we must go backwards. The data file consists of
1076 * records with two parts, the file name followed by a DIRDATA trailer. The
1077 * fixed sized trailer contains the size of the name plus the off_t location in
1078 * the file. To restore we work backwards through the file reading the trailer
1079 * then the file name.
1083 * dir_start()
1084 * set up the directory time and file mode storage for directories CREATED
1085 * by pax.
1086 * Return:
1087 * 0 if ok, -1 otherwise
1091 dir_start(void)
1093 if (dirp != NULL)
1094 return(0);
1096 dirsize = DIRP_SIZE;
1097 if ((dirp = calloc(dirsize, sizeof(DIRDATA))) == NULL) {
1098 paxwarn(1, "Unable to allocate memory for directory times");
1099 return(-1);
1101 return(0);
1105 * add_dir()
1106 * add the mode and times for a newly CREATED directory
1107 * name is name of the directory, psb the stat buffer with the data in it,
1108 * frc_mode is a flag that says whether to force the setting of the mode
1109 * (ignoring the user set values for preserving file mode). Frc_mode is
1110 * for the case where we created a file and found that the resulting
1111 * directory was not writeable and the user asked for file modes to NOT
1112 * be preserved. (we have to preserve what was created by default, so we
1113 * have to force the setting at the end. this is stated explicitly in the
1114 * pax spec)
1117 void
1118 add_dir(char *name, struct stat *psb, int frc_mode)
1120 DIRDATA *dblk;
1121 char realname[MAXPATHLEN], *rp;
1123 if (dirp == NULL)
1124 return;
1126 if (havechd && *name != '/') {
1127 if ((rp = realpath(name, realname)) == NULL) {
1128 paxwarn(1, "Cannot canonicalize %s", name);
1129 return;
1131 name = rp;
1133 if (dircnt == dirsize) {
1134 dblk = realloc(dirp, 2 * dirsize * sizeof(DIRDATA));
1135 if (dblk == NULL) {
1136 paxwarn(1, "Unable to store mode and times for created"
1137 " directory: %s", name);
1138 return;
1140 dirp = dblk;
1141 dirsize *= 2;
1143 dblk = &dirp[dircnt];
1144 if ((dblk->name = strdup(name)) == NULL) {
1145 paxwarn(1, "Unable to store mode and times for created"
1146 " directory: %s", name);
1147 return;
1149 dblk->mode = psb->st_mode & 0xffff;
1150 dblk->mtime = psb->st_mtime;
1151 dblk->atime = psb->st_atime;
1152 dblk->frc_mode = frc_mode;
1153 ++dircnt;
1157 * proc_dir()
1158 * process all file modes and times stored for directories CREATED
1159 * by pax
1162 void
1163 proc_dir(void)
1165 DIRDATA *dblk;
1166 long cnt;
1168 if (dirp == NULL)
1169 return;
1171 * read backwards through the file and process each directory
1173 cnt = dircnt;
1174 while (--cnt >= 0) {
1176 * frc_mode set, make sure we set the file modes even if
1177 * the user didn't ask for it (see file_subs.c for more info)
1179 dblk = &dirp[cnt];
1180 if (pmode || dblk->frc_mode)
1181 set_pmode(dblk->name, dblk->mode);
1182 if (patime || pmtime)
1183 set_ftime(dblk->name, dblk->mtime, dblk->atime, 0);
1184 free(dblk->name);
1187 free(dirp);
1188 dirp = NULL;
1189 dircnt = 0;
1193 * database independent routines
1197 * st_hash()
1198 * hashes filenames to a u_int for hashing into a table. Looks at the tail
1199 * end of file, as this provides far better distribution than any other
1200 * part of the name. For performance reasons we only care about the last
1201 * MAXKEYLEN chars (should be at LEAST large enough to pick off the file
1202 * name). Was tested on 500,000 name file tree traversal from the root
1203 * and gave almost a perfectly uniform distribution of keys when used with
1204 * prime sized tables (MAXKEYLEN was 128 in test). Hashes (sizeof int)
1205 * chars at a time and pads with 0 for last addition.
1206 * Return:
1207 * the hash value of the string MOD (%) the table size.
1210 u_int
1211 st_hash(char *name, int len, int tabsz)
1213 char *pt;
1214 char *dest;
1215 char *end;
1216 int i;
1217 u_int key = 0;
1218 int steps;
1219 int res;
1220 u_int val;
1223 * only look at the tail up to MAXKEYLEN, we do not need to waste
1224 * time here (remember these are pathnames, the tail is what will
1225 * spread out the keys)
1227 if (len > MAXKEYLEN) {
1228 pt = &(name[len - MAXKEYLEN]);
1229 len = MAXKEYLEN;
1230 } else
1231 pt = name;
1234 * calculate the number of u_int size steps in the string and if
1235 * there is a runt to deal with
1237 steps = len/sizeof(u_int);
1238 res = len % sizeof(u_int);
1241 * add up the value of the string in unsigned integer sized pieces
1242 * too bad we cannot have unsigned int aligned strings, then we
1243 * could avoid the expensive copy.
1245 for (i = 0; i < steps; ++i) {
1246 end = pt + sizeof(u_int);
1247 dest = (char *)&val;
1248 while (pt < end)
1249 *dest++ = *pt++;
1250 key += val;
1254 * add in the runt padded with zero to the right
1256 if (res) {
1257 val = 0;
1258 end = pt + res;
1259 dest = (char *)&val;
1260 while (pt < end)
1261 *dest++ = *pt++;
1262 key += val;
1266 * return the result mod the table size
1268 return(key % tabsz);