X-Git-Url: https://git.wh0rd.org/?p=dump.git;a=blobdiff_plain;f=restore%2Futilities.c;h=f17bd9ce458a7214ab50050f787c794d79a033a1;hp=e0db9afa83b5c68e9ed70d73e81450901f4a2123;hb=f32a70b091d3ee6524e760ec2cb21080f67fe2ee;hpb=3e8961b70386c298ef95b85171ec9bb5dea9649a

diff --git a/restore/utilities.c b/restore/utilities.c
index e0db9af..f17bd9c 100644
--- a/restore/utilities.c
+++ b/restore/utilities.c
@@ -37,7 +37,7 @@
 
 #ifndef lint
 static const char rcsid[] =
-	"$Id: utilities.c,v 1.24 2003/11/22 16:52:16 stelian Exp $";
+	"$Id: utilities.c,v 1.28 2005/03/30 13:21:45 stelian Exp $";
 #endif /* not lint */
 
 #include <config.h>
@@ -187,8 +187,13 @@ removenode(struct entry *ep)
 
 	if (ep->e_type != NODE)
 		badentry(ep, "removenode: not a node");
-	if (ep->e_entries != NULL)
-		badentry(ep, "removenode: non-empty directory");
+	if (ep->e_entries != NULL) {
+		int i;
+		for (i = 0; i < DIRHASH_SIZE; i++) {
+			if (ep->e_entries[i] != NULL)
+				badentry(ep, "removenode: non-empty directory");
+		}
+	}
 	ep->e_flags |= REMOVED;
 	ep->e_flags &= ~TMPNAME;
 	cp = myname(ep);
@@ -264,10 +269,10 @@ linkit(char *existing, char *new, int type)
 			 */
 #ifdef sunos
 #else
-			if (fgetflags (existing, &s) != -1 &&
-			    fsetflags (existing, 0) != -1) {
+			if (lgetflags (existing, &s) != -1 &&
+			    lsetflags (existing, 0) != -1) {
 			    	ret = link(existing, new);
-				fsetflags(existing, s);
+				lsetflags(existing, s);
 			}
 #endif
 #endif
@@ -371,8 +376,15 @@ badentry(struct entry *ep, const char *msg)
 	fprintf(stderr, "parent name %s\n", myname(ep->e_parent));
 	if (ep->e_sibling != NULL)
 		fprintf(stderr, "sibling name: %s\n", myname(ep->e_sibling));
-	if (ep->e_entries != NULL)
-		fprintf(stderr, "next entry name: %s\n", myname(ep->e_entries));
+	if (ep->e_entries != NULL) {
+		int i;
+		for (i = 0; i < DIRHASH_SIZE; i++) {
+			if (ep->e_entries[i] != NULL) {
+				fprintf(stderr, "next entry name: %s\n", myname(ep->e_entries[i]));
+				break;
+			}
+		}
+	}
 	if (ep->e_links != NULL)
 		fprintf(stderr, "next link name: %s\n", myname(ep->e_links));
 	if (ep->e_next != NULL)
@@ -475,8 +487,10 @@ panic(fmt, va_alist)
 	if (yflag)
 		return;
 	if (reply("abort") == GOOD) {
-		if (reply("dump core") == GOOD)
+		if (reply("dump core") == GOOD) {
+			fchdir(wdfd);
 			abort();
+		}
 		exit(1);
 	}
 }
@@ -712,9 +726,46 @@ CreateAppleDoubleFileRes(char *oFile, FndrFileInfo *finderinfo, mode_t mode, int
 	(void)fchown(fdout, uid, gid);
 	(void)fchmod(fdout, mode);
 	close(fdout);
-	(void)fsetflags(oFile, flags);
+	(void)lsetflags(oFile, flags);
 	utimes(oFile, timep);
 	free(pp);
 	return err;
 }
 #endif /* DUMP_MACOSX */
+
+int
+lgetflags(const char *path, unsigned long *flags)
+{
+	int err;
+	struct STAT sb;
+
+	err = LSTAT(path, &sb);
+	if (err < 0)
+		return err;
+
+	if (S_ISLNK(sb.st_mode) || S_ISFIFO(sb.st_mode)) {
+		// no way to get/set flags on a symlink
+		*flags = 0;
+		return 0;
+	}
+	else
+		return fgetflags(path, flags);
+}
+
+int
+lsetflags(const char *path, unsigned long flags)
+{
+	int err;
+	struct STAT sb;
+
+	err = LSTAT(path, &sb);
+	if (err < 0)
+		return err;
+
+	if (S_ISLNK(sb.st_mode) || S_ISFIFO(sb.st_mode)) {
+		// no way to get/set flags on a symlink
+		return 0;
+	}
+	else
+		return fsetflags(path, flags);
+}