Some simple set_basedir() cleanup; more to come.

[mandoc.git] / mandocdb.c

diff --git a/mandocdb.c b/mandocdb.c
index 85f0cca079a39b8e5e4285042de3f999c4493dec..152fe1c645681b007231b553a32591da708335c4 100644 (file)
--- a/mandocdb.c
+++ b/mandocdb.c
@@ -1,4 +1,4 @@
-/*     $Id: mandocdb.c,v 1.147 2014/05/07 15:10:36 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>
 /*
  * 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) {
 
                /*
        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.
                 */
                 * 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)) {
                        goto out;
 
                if (dbopen(1)) {


@@ -525,7 +525,6 @@ main(int argc, char *argv[])
                }
        }
 out:
                }
        }
 out:

-       set_basedir(NULL);

        manpath_free(&dirs);
        mchars_free(mc);
        mparse_free(mp);
        manpath_free(&dirs);
        mchars_free(mc);
        mparse_free(mp);


@@ -800,10 +799,10 @@ filescan(const char *file)
                return;
        }
 
                return;
        }
 

-       if (strstr(buf, basedir) == buf)
-               start = buf + strlen(basedir) + 1;
-       else if (OP_TEST == op)
+       if (OP_TEST == op)

                start = buf;
                start = buf;

+       else if (strstr(buf, basedir) == buf)
+               start = buf + strlen(basedir);

        else {
                exitcode = (int)MANDOCLEVEL_BADARG;
                say("", "%s: outside base directory", buf);
        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;
                }
                        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));
        }
 
        mlink = mandoc_calloc(1, sizeof(struct mlink));


@@ -2036,13 +2036,22 @@ dbadd(struct mpage *mpage, struct mchars *mc)
        if (debug)
                say(mlink->file, "Adding to database");
 
        if (debug)
                say(mlink->file, "Adding to database");
 

+       i = strlen(mpage->desc) + 1;
+       key = mandoc_calloc(1, sizeof(struct str) + i);
+       memcpy(key->key, mpage->desc, i);
+       render_key(mc, key);
+

        i = 1;
        i = 1;

-       SQL_BIND_TEXT(stmts[STMT_INSERT_PAGE], i, mpage->desc);
+       SQL_BIND_TEXT(stmts[STMT_INSERT_PAGE], i, key->rendered);

        SQL_BIND_INT(stmts[STMT_INSERT_PAGE], i, FORM_SRC == mpage->form);
        SQL_STEP(stmts[STMT_INSERT_PAGE]);
        mpage->pageid = sqlite3_last_insert_rowid(db);
        sqlite3_reset(stmts[STMT_INSERT_PAGE]);
 
        SQL_BIND_INT(stmts[STMT_INSERT_PAGE], i, FORM_SRC == mpage->form);
        SQL_STEP(stmts[STMT_INSERT_PAGE]);
        mpage->pageid = sqlite3_last_insert_rowid(db);
        sqlite3_reset(stmts[STMT_INSERT_PAGE]);
 

+       if (key->rendered != key->key)
+               free(key->rendered);
+       free(key);
+

        while (NULL != mlink) {
                dbadd_mlink(mlink);
                mlink = mlink->next;
        while (NULL != mlink) {
                dbadd_mlink(mlink);
                mlink = mlink->next;


@@ -2221,7 +2230,8 @@ dbopen(int real)
                rc = sqlite3_open_v2(MANDOC_DB, &db, ofl, NULL);
                if (SQLITE_OK != rc) {
                        exitcode = (int)MANDOCLEVEL_SYSERR;
                rc = sqlite3_open_v2(MANDOC_DB, &db, ofl, NULL);
                if (SQLITE_OK != rc) {
                        exitcode = (int)MANDOCLEVEL_SYSERR;

-                       say(MANDOC_DB, "%s", sqlite3_errstr(rc));
+                       if (SQLITE_CANTOPEN != rc)
+                               say(MANDOC_DB, "%s", sqlite3_errstr(rc));

                        return(0);
                }
                goto prepare_statements;
                        return(0);
                }
                goto prepare_statements;


@@ -2365,6 +2375,7 @@ set_basedir(const char *targetdir)
 {
        static char      startdir[PATH_MAX];
        static int       fd;
 {
        static char      startdir[PATH_MAX];
        static int       fd;

+       char            *cp;

 
        /*
         * Remember where we started by keeping a fd open to the origin
 
        /*
         * Remember where we started by keeping a fd open to the origin


@@ -2375,8 +2386,7 @@ set_basedir(const char *targetdir)
        if ('\0' == *startdir) {
                if (NULL == getcwd(startdir, PATH_MAX)) {
                        exitcode = (int)MANDOCLEVEL_SYSERR;
        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))) {
                        return(0);
                }
                if (-1 == (fd = open(startdir, O_RDONLY, 0))) {


@@ -2384,8 +2394,6 @@ set_basedir(const char *targetdir)
                        say("", "&open %s", startdir);
                        return(0);
                }
                        say("", "&open %s", startdir);
                        return(0);
                }

-               if (NULL == targetdir)
-                       targetdir = startdir;

        } else {
                if (-1 == fd)
                        return(0);
        } else {
                if (-1 == fd)
                        return(0);


@@ -2396,10 +2404,6 @@ set_basedir(const char *targetdir)
                        say("", "&chdir %s", startdir);
                        return(0);
                }
                        say("", "&chdir %s", startdir);
                        return(0);
                }

-               if (NULL == targetdir) {
-                       close(fd);
-                       return(1);
-               }

        }
        if (NULL == realpath(targetdir, basedir)) {
                basedir[0] = '\0';
        }
        if (NULL == realpath(targetdir, basedir)) {
                basedir[0] = '\0';


@@ -2411,6 +2415,16 @@ set_basedir(const char *targetdir)
                say("", "&chdir");
                return(0);
        }
                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);
 }
 
        return(1);
 }