Rename the _S macro to OSL_ERROR.

[osl.git] / fsck.c

diff --git a/fsck.c b/fsck.c
index dea7ad6e07079bb1672546aea47744e91564007d..29c4e2c7dc9d72d940dbb7c4298f24d84f3dcfc8 100644 (file)
--- a/fsck.c
+++ b/fsck.c
@@ -15,7 +15,7 @@
 #include "log.h"
 #include "osl.h"
 #include "error.h"
-#include "fd.h"
+#include "util.h"
 #include "osl_core.h"
 #include "fsck.cmdline.h"
 
@@ -130,7 +130,7 @@ static int _write_all(int fd, const char *buf, size_t len)
  *
  * \sa malloc(3).
  */
-__must_check __malloc static void *para_malloc(size_t size)
+__must_check __malloc static void *fsck_malloc(size_t size)
 {
        assert(size);
        void *p = malloc(size);
@@ -155,9 +155,9 @@ __must_check __malloc static void *para_malloc(size_t size)
  *
  * \sa calloc(3)
  */
-__must_check __malloc static void *para_calloc(size_t size)
+__must_check __malloc static void *fsck_calloc(size_t size)
 {
-       void *ret = para_malloc(size);
+       void *ret = fsck_malloc(size);
 
        memset(ret, 0, size);
        return ret;
@@ -176,7 +176,7 @@ __must_check __malloc static void *para_calloc(size_t size)
  *
  * \sa strdup(3)
  */
-__must_check __malloc static char *para_strdup(const char *s)
+__must_check __malloc static char *fsck_strdup(const char *s)
 {
        char *ret;
 
@@ -208,6 +208,86 @@ static int uint32_compare(const struct osl_object *obj1, const struct osl_object
        return 0;
 }
 
+/**
+ * A wrapper for fchdir().
+ *
+ * \param fd An open file descriptor.
+ *
+ * \return Standard.
+ */
+static inline int __fchdir(int fd)
+{
+       if (fchdir(fd) < 0)
+               return -ERRNO_TO_ERROR(errno);
+       return 1;
+}
+
+/**
+ * Wrapper for chdir(2).
+ *
+ * \param path The specified directory.
+ *
+ * \return Standard.
+ */
+_static_inline_ int __chdir(const char *path)
+{
+       if (chdir(path) >= 0)
+               return 1;
+       return -ERRNO_TO_ERROR(errno);
+}
+
+/**
+ * Save the cwd and open a given directory.
+ *
+ * \param dirname Path to the directory to open.
+ * \param dir Result pointer.
+ * \param cwd File descriptor of the current working directory.
+ *
+ * \return Standard.
+ *
+ * Opening the current directory (".") and calling fchdir() to return is
+ * usually faster and more reliable than saving cwd in some buffer and calling
+ * chdir() afterwards.
+ *
+ * If \a cwd is not \p NULL "." is opened and the resulting file descriptor is
+ * stored in \a cwd. If the function returns success, and \a cwd is not \p
+ * NULL, the caller must close this file descriptor (probably after calling
+ * fchdir(*cwd)).
+ *
+ * On errors, the function undos everything, so the caller needs neither close
+ * any files, nor change back to the original working directory.
+ *
+ * \sa getcwd(3).
+ *
+ */
+static int fsck_opendir(const char *dirname, DIR **dir, int *cwd)
+{
+       int ret;
+
+       if (cwd) {
+               ret = osl_open(".", O_RDONLY, 0);
+               if (ret < 0)
+                       return ret;
+               *cwd = ret;
+       }
+       ret = __chdir(dirname);
+       if (ret < 0)
+               goto close_cwd;
+       *dir = opendir(".");
+       if (*dir)
+               return 1;
+       ret = -ERRNO_TO_ERROR(errno);
+/* Ignore return value of fchdir() and close(). We're busted anyway. */
+       if (cwd)
+               fchdir(*cwd);
+close_cwd:
+       if (cwd)
+               close(*cwd);
+       return ret;
+}
+
+
+
 /**
  * Traverse the given directory recursively.
  *
@@ -227,7 +307,7 @@ static int for_each_file_in_dir(const char *dirname,
 {
        DIR *dir;
        struct dirent *entry;
-       int cwd_fd, ret2, ret = para_opendir(dirname, &dir, &cwd_fd);
+       int cwd_fd, ret2, ret = fsck_opendir(dirname, &dir, &cwd_fd);
 
        if (ret < 0)
                return ret == -ERRNO_TO_ERROR(EACCES)? 1 : ret;
@@ -263,7 +343,7 @@ static int for_each_file_in_dir(const char *dirname,
        ret = 1;
 out:
        closedir(dir);
-       ret2 = para_fchdir(cwd_fd);
+       ret2 = __fchdir(cwd_fd);
        if (ret2 < 0 && ret >= 0)
                ret = ret2;
        close(cwd_fd);
@@ -412,7 +492,7 @@ static int prune_invalid_rows_from_index(struct osl_table *t)
        DEBUG_LOG("unmapping index\n");
        osl_munmap(t->index_map.data, t->index_map.size);
        filename = index_filename(t->desc);
-       ret = para_truncate(filename, t->row_index_size
+       ret = truncate_file(filename, t->row_index_size
                * t->num_invalid_rows);
        free(filename);
        if (ret < 0)
@@ -428,7 +508,7 @@ static int check_for_invalid_objects(struct osl_table *t, uint32_t **lost_bytes)
 {
        int i, j, ret;
        const struct osl_column_description *cd;
-       uint32_t *loss = para_malloc(sizeof(uint32_t) * t->desc->num_columns);
+       uint32_t *loss = fsck_malloc(sizeof(uint32_t) * t->desc->num_columns);
 
        INFO_LOG("looking for mapped objects not contained in index\n");
        /* first count used bytes */
@@ -496,8 +576,8 @@ static int prune_objects(struct osl_table *t, uint32_t *lost_bytes)
 {
        int i, ret;
        const struct osl_column_description *cd;
-       char **col_filenames = para_calloc(t->desc->num_columns * sizeof(char *));
-       char **new_col_filenames = para_calloc(t->desc->num_columns * sizeof(char *));
+       char **col_filenames = fsck_calloc(t->desc->num_columns * sizeof(char *));
+       char **new_col_filenames = fsck_calloc(t->desc->num_columns * sizeof(char *));
        char *idx_filename = index_filename(t->desc);
        char *old_idx_filename = make_message("%s.bak", idx_filename);
        int fd;
@@ -630,7 +710,7 @@ static int check_disk_storage_presence(struct osl_table *t)
 
        if (!t->num_rows)
                return 1;
-       hashes = para_malloc(t->num_rows * HASH_SIZE);
+       hashes = fsck_malloc(t->num_rows * HASH_SIZE);
        INFO_LOG("looking for missing disk storage objects\n");
        for (i = 0; i < t->num_rows; i++) {
                if (row_is_invalid(t, i))
@@ -849,7 +929,7 @@ const char *get_asc_storage_type(enum osl_storage_type st)
 
 #define APPEND_ASC_SF(sf, flag, str) do { if (sf & flag) { \
        if (str) str = make_message("%s%s", str, " | " # flag); \
-       else str = para_strdup(#flag); }} while (0)
+       else str = fsck_strdup(#flag); }} while (0)
 
 
 char *get_asc_storage_flags(enum osl_storage_type sf)
@@ -1105,7 +1185,7 @@ static int check_all_tables(char *db_dir)
 {
        DIR *dir;
        struct dirent *entry;
-       int cwd_fd, ret2, ret = para_opendir(db_dir, &dir, &cwd_fd);
+       int cwd_fd, ret2, ret = fsck_opendir(db_dir, &dir, &cwd_fd);
 
        if (ret < 0)
                return ret;
@@ -1126,7 +1206,7 @@ static int check_all_tables(char *db_dir)
                        break;
        }
        closedir(dir);
-       ret2 = para_fchdir(cwd_fd);
+       ret2 = __fchdir(cwd_fd);
        if (ret2 < 0 && ret >= 0)
                ret = ret2;
        close(cwd_fd);