Some simple set_basedir() cleanup; more to come.

1) Refrain from calling set_basedir() in the -t case,
and do not attempt to strip anything from the file names in that case.
Testing individual files cannot reasonably have any notion of a base dir.

2) Remove the possibility of passing NULL to set_basedir().
It was dangerous because it was not idempotent, and it served no purpose
except closing a file descriptor right before exit(), which is pointless.
Besides, the file descriptor is likely to be removed completely, soon.

3) Make sure that /foobar isn't treated as a subdirectory of /foo;
this fixes a bug reported by espie@.

diff --git a/mandocdb.c b/mandocdb.c
index 2798efcafa0a72d1d7670659ed72ee938fbf00a9..152fe1c645681b007231b553a32591da708335c4 100644 (file)
--- a/mandocdb.c
+++ b/mandocdb.c
@@ -1,4 +1,4 @@
-/*     $Id: mandocdb.c,v 1.149 2014/06/18 19:34:04 schwarze Exp $ */
+/*     $Id: mandocdb.c,v 1.150 2014/06/19 00:45:37 schwarze Exp $ */
 /*
  * Copyright (c) 2011, 2012 Kristaps Dzonsons <kristaps@bsd.lv>
  * Copyright (c) 2011, 2012, 2013, 2014 Ingo Schwarze <schwarze@openbsd.org>
@@ -433,10 +433,10 @@ main(int argc, char *argv[])
        if (OP_UPDATE == op || OP_DELETE == op || OP_TEST == op) {
 
                /*
-                * All of these deal with a specific directory.
+                * Most of these deal with a specific directory.
                 * Jump into that directory first.
                 */
-               if (0 == set_basedir(path_arg))
+               if (OP_TEST != op && 0 == set_basedir(path_arg))
                        goto out;
 
                if (dbopen(1)) {
@@ -525,7 +525,6 @@ main(int argc, char *argv[])
                }
        }
 out:
-       set_basedir(NULL);
        manpath_free(&dirs);
        mchars_free(mc);
        mparse_free(mp);
@@ -800,10 +799,10 @@ filescan(const char *file)
                return;
        }
 
-       if (strstr(buf, basedir) == buf)
-               start = buf + strlen(basedir) + 1;
-       else if (OP_TEST == op)
+       if (OP_TEST == op)
                start = buf;
+       else if (strstr(buf, basedir) == buf)
+               start = buf + strlen(basedir);
        else {
                exitcode = (int)MANDOCLEVEL_BADARG;
                say("", "%s: outside base directory", buf);
@@ -829,8 +828,9 @@ filescan(const char *file)
                        say(file, "Filename too long");
                        return;
                }
-               start = strstr(buf, basedir) == buf ?
-                   buf + strlen(basedir) + 1 : buf;
+               start = buf;
+               if (OP_TEST != op && strstr(buf, basedir) == buf)
+                       start += strlen(basedir);
        }
 
        mlink = mandoc_calloc(1, sizeof(struct mlink));
@@ -2375,6 +2375,7 @@ set_basedir(const char *targetdir)
 {
        static char      startdir[PATH_MAX];
        static int       fd;
+       char            *cp;
 
        /*
         * Remember where we started by keeping a fd open to the origin
@@ -2385,8 +2386,7 @@ set_basedir(const char *targetdir)
        if ('\0' == *startdir) {
                if (NULL == getcwd(startdir, PATH_MAX)) {
                        exitcode = (int)MANDOCLEVEL_SYSERR;
-                       if (NULL != targetdir)
-                               say("", "&getcwd");
+                       say("", "&getcwd");
                        return(0);
                }
                if (-1 == (fd = open(startdir, O_RDONLY, 0))) {
@@ -2394,8 +2394,6 @@ set_basedir(const char *targetdir)
                        say("", "&open %s", startdir);
                        return(0);
                }
-               if (NULL == targetdir)
-                       targetdir = startdir;
        } else {
                if (-1 == fd)
                        return(0);
@@ -2406,10 +2404,6 @@ set_basedir(const char *targetdir)
                        say("", "&chdir %s", startdir);
                        return(0);
                }
-               if (NULL == targetdir) {
-                       close(fd);
-                       return(1);
-               }
        }
        if (NULL == realpath(targetdir, basedir)) {
                basedir[0] = '\0';
@@ -2421,6 +2415,16 @@ set_basedir(const char *targetdir)
                say("", "&chdir");
                return(0);
        }
+       cp = strchr(basedir, '\0');
+       if ('/' != cp[-1]) {
+               if (cp - basedir >= PATH_MAX - 1) {
+                       exitcode = (int)MANDOCLEVEL_SYSERR;
+                       say("", "Filename too long");
+                       return(0);
+               }
+               *cp++ = '/';
+               *cp = '\0';
+       }
        return(1);
 }