everything in its own place

libdirutils/dirutils.c

@@ -0,0 +1,192 @@
+/*
+ * Copyright (c) 2012 Kyle Isom <kyle@tyrfingr.is>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
+ * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
+ * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
+ * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA
+ * OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
+ * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+ * PERFORMANCE OF THIS SOFTWARE.
+ * ---------------------------------------------------------------------
+ */
+
+#include <sys/types.h>
+#include <sys/queue.h>
+#include <sys/stat.h>
+
+#include <dirent.h>
+#include <errno.h>
+#include <libgen.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "kst/dirutils.h"
+#include "kst/dirlist.h"
+
+
+static int	  _parent_exists(const char *);
+static int	  _rmdirs(const char *);
+
+/*
+ * Determines whether a directory exists.
+ */
+EXISTS_STATUS
+path_exists(const char *path)
+{
+	struct stat	   st;
+	int		   rv;
+
+	rv = stat(path, &st);
+	if (rv == -1) {
+		switch (errno) {
+		case EACCES:
+			return EXISTS_NOPERM;
+			break;
+		case ENOENT:
+			return EXISTS_NOENT;
+			break;
+		default:
+			return EXISTS_ERROR;
+		}
+	}
+
+	if (st.st_mode & S_IFDIR) {
+		return EXISTS_DIR;
+	}
+	else if (st.st_mode & S_IFREG) {
+		return EXISTS_FILE;
+	}
+	else {
+		return EXISTS_OTHER;
+	}
+}
+
+
+/*
+ * create a path and any directories above it required
+ */
+int
+makedirs(const char *path)
+{
+	struct tq_dirlst	*lst;
+	struct dirlst		     *elm;
+	size_t			      path_sz;
+	char			    *dnam_p, *curpath;
+
+	path_sz = strlen(path);
+	lst = dirlst_create(path, path_sz);
+	if (NULL == lst)
+		return EXIT_FAILURE;
+
+	curpath = strdup(path);
+	while (!_parent_exists(curpath)) {
+		dnam_p = dirname(curpath);
+		curpath = strdup(dnam_p);
+		dirlst_push(lst, curpath, strlen(curpath));
+	}
+	free(curpath);
+
+	while (NULL != (elm = dirlst_pop(lst))) {
+		if (-1 == mkdir(elm->path, 0777)) {
+			free(elm);
+			return EXIT_FAILURE;
+		}
+		free(elm);
+	}
+
+	return dirlst_destroy(&lst);
+}
+
+
+/*
+ * remove a directory and any subdirectories
+ */
+int
+rmdirs(const char *path)
+{
+	return _rmdirs(path);
+}
+
+
+/*
+ * Determine whether the parent directory exists, and return true if it
+ * does.
+ */
+int
+_parent_exists(const char *path)
+{
+	char	    *name_buf;
+	char	    *dnam_p;
+
+	name_buf = strdup(path);
+
+	dnam_p = dirname(name_buf);
+	if (EXISTS_DIR != path_exists(dnam_p)) {
+		return 0;
+	}
+	else {
+		return 1;
+	}
+}
+
+
+/*
+ * remove a directory, all files in it, and all subdirectories.
+ */
+int
+_rmdirs(const char *path)
+{
+	char			 child[FILENAME_MAX + 1];
+	struct dirent		  *dp;
+	DIR		      *dirp;
+	int		       fail = EXIT_FAILURE;
+
+	if (NULL == (dirp = opendir(path))) {
+		return EXIT_FAILURE;
+	}
+	
+	while (NULL != (dp = readdir(dirp))) {
+		if (0 == strncmp("..", dp->d_name, 3)) {
+			continue;
+		}
+		
+		if (0 == strncmp(".", dp->d_name, 2)) {
+			continue;
+		}
+
+		
+		snprintf(child, FILENAME_MAX, "%s/%s", path, dp->d_name);
+		if (DT_DIR == dp->d_type) {
+			fail = _rmdirs(child);
+			if (EXIT_FAILURE == fail) {
+				break;
+			}
+		}
+		else {
+			fail = unlink(child);
+			if (-1 == fail) {
+				fail = EXIT_FAILURE;
+				break;
+			}
+		}
+	}
+	if (-1 == closedir(dirp))
+		return EXIT_FAILURE;
+	if (EXIT_FAILURE == fail) {
+		return EXIT_FAILURE;
+	}
+	else if (-1 == rmdir(path)) {
+		return EXIT_FAILURE;
+	}
+	else {
+		return EXIT_SUCCESS;
+	}
+}