]> git.cameronkatri.com Git - mandoc.git/blobdiff - mansearch.c
modernize style: "return" is not a function
[mandoc.git] / mansearch.c
index 938d48c4900dfbec8d4ad7a7734dae3d7cfa8097..e019b4877f0eeb2e3c55194c7f5a4570814bb483 100644 (file)
@@ -1,15 +1,15 @@
-/*     $Id: mansearch.c,v 1.49 2014/11/11 19:04:55 schwarze Exp $ */
+/*     $Id: mansearch.c,v 1.58 2015/10/06 18:32:19 schwarze Exp $ */
 /*
  * Copyright (c) 2012 Kristaps Dzonsons <kristaps@bsd.lv>
- * Copyright (c) 2013, 2014 Ingo Schwarze <schwarze@openbsd.org>
+ * Copyright (c) 2013, 2014, 2015 Ingo Schwarze <schwarze@openbsd.org>
  *
  * 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
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHORS DISCLAIM ALL WARRANTIES
  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHORS 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
 #include <sys/types.h>
 
 #include <assert.h>
+#include <errno.h>
 #include <fcntl.h>
 #include <getopt.h>
+#include <glob.h>
 #include <limits.h>
 #include <regex.h>
 #include <stdio.h>
@@ -44,7 +46,7 @@
 
 #include "mandoc.h"
 #include "mandoc_aux.h"
-#include "manpath.h"
+#include "manconf.h"
 #include "mansearch.h"
 
 extern int mansearch_keymax;
@@ -79,11 +81,13 @@ struct      expr {
 
 struct match {
        uint64_t         pageid; /* identifier in database */
+       uint64_t         bits; /* name type mask */
        char            *desc; /* manual page description */
        int              form; /* bit field: formatted, zipped? */
 };
 
-static void             buildnames(struct manpage *, sqlite3 *,
+static void             buildnames(const struct mansearch *,
+                               struct manpage *, sqlite3 *,
                                sqlite3_stmt *, uint64_t,
                                const char *, int form);
 static char            *buildoutput(sqlite3 *, sqlite3_stmt *,
@@ -94,8 +98,6 @@ static        void            *hash_calloc(size_t, size_t, void *);
 static struct expr     *exprcomp(const struct mansearch *,
                                int, char *[]);
 static void             exprfree(struct expr *);
-static struct expr     *exprspec(struct expr *, uint64_t,
-                                const char *, const char *);
 static struct expr     *exprterm(const struct mansearch *, char *, int);
 static int              manpage_compare(const void *, const void *);
 static void             sql_append(char **sql, size_t *sz,
@@ -119,7 +121,7 @@ mansearch_setup(int start)
        if (start) {
                if (NULL != pagecache) {
                        fprintf(stderr, "pagecache already enabled\n");
-                       return((int)MANDOCLEVEL_BADARG);
+                       return (int)MANDOCLEVEL_BADARG;
                }
 
                pagecache = mmap(NULL, PC_PAGESIZE * PC_NUMPAGES,
@@ -129,30 +131,30 @@ mansearch_setup(int start)
                if (MAP_FAILED == pagecache) {
                        perror("mmap");
                        pagecache = NULL;
-                       return((int)MANDOCLEVEL_SYSERR);
+                       return (int)MANDOCLEVEL_SYSERR;
                }
 
                c = sqlite3_config(SQLITE_CONFIG_PAGECACHE,
                    pagecache, PC_PAGESIZE, PC_NUMPAGES);
 
                if (SQLITE_OK == c)
-                       return((int)MANDOCLEVEL_OK);
+                       return (int)MANDOCLEVEL_OK;
 
                fprintf(stderr, "pagecache: %s\n", sqlite3_errstr(c));
 
        } else if (NULL == pagecache) {
                fprintf(stderr, "pagecache missing\n");
-               return((int)MANDOCLEVEL_BADARG);
+               return (int)MANDOCLEVEL_BADARG;
        }
 
        if (-1 == munmap(pagecache, PC_PAGESIZE * PC_NUMPAGES)) {
                perror("munmap");
                pagecache = NULL;
-               return((int)MANDOCLEVEL_SYSERR);
+               return (int)MANDOCLEVEL_SYSERR;
        }
 
        pagecache = NULL;
-       return((int)MANDOCLEVEL_OK);
+       return (int)MANDOCLEVEL_OK;
 }
 
 int
@@ -161,7 +163,6 @@ mansearch(const struct mansearch *search,
                int argc, char *argv[],
                struct manpage **res, size_t *sz)
 {
-       int              fd, rc, c, indexbit;
        int64_t          pageid;
        uint64_t         outbit, iterbit;
        char             buf[PATH_MAX];
@@ -175,26 +176,23 @@ mansearch(const struct mansearch *search,
        struct ohash     htab;
        unsigned int     idx;
        size_t           i, j, cur, maxres;
+       int              c, chdir_status, getcwd_status, indexbit;
+
+       if (argc == 0 || (e = exprcomp(search, argc, argv)) == NULL) {
+               *sz = 0;
+               return 0;
+       }
 
        info.calloc = hash_calloc;
        info.alloc = hash_alloc;
        info.free = hash_free;
        info.key_offset = offsetof(struct match, pageid);
 
-       *sz = cur = maxres = 0;
-       sql = NULL;
+       cur = maxres = 0;
        *res = NULL;
-       fd = -1;
-       e = NULL;
-       rc = 0;
 
-       if (0 == argc)
-               goto out;
-       if (NULL == (e = exprcomp(search, argc, argv)))
-               goto out;
-
-       outbit = 0;
        if (NULL != search->outkey) {
+               outbit = TYPE_Nd;
                for (indexbit = 0, iterbit = 1;
                     indexbit < mansearch_keymax;
                     indexbit++, iterbit <<= 1) {
@@ -204,22 +202,22 @@ mansearch(const struct mansearch *search,
                                break;
                        }
                }
-       }
+       } else
+               outbit = 0;
 
        /*
-        * Save a descriptor to the current working directory.
-        * Since pathnames in the "paths" variable might be relative,
-        * and we'll be chdir()ing into them, we need to keep a handle
-        * on our current directory from which to start the chdir().
+        * Remember the original working directory, if possible.
+        * This will be needed if the second or a later directory
+        * is given as a relative path.
+        * Do not error out if the current directory is not
+        * searchable: Maybe it won't be needed after all.
         */
 
-       if (NULL == getcwd(buf, PATH_MAX)) {
-               perror("getcwd");
-               goto out;
-       } else if (-1 == (fd = open(buf, O_RDONLY, 0))) {
-               perror(buf);
-               goto out;
-       }
+       if (getcwd(buf, PATH_MAX) == NULL) {
+               getcwd_status = 0;
+               (void)strlcpy(buf, strerror(errno), sizeof(buf));
+       } else
+               getcwd_status = 1;
 
        sql = sql_statement(e);
 
@@ -231,21 +229,30 @@ mansearch(const struct mansearch *search,
         * scan it for our match expression.
         */
 
+       chdir_status = 0;
        for (i = 0; i < paths->sz; i++) {
-               if (-1 == fchdir(fd)) {
-                       perror(buf);
-                       free(*res);
-                       break;
-               } else if (-1 == chdir(paths->paths[i])) {
+               if (chdir_status && paths->paths[i][0] != '/') {
+                       if ( ! getcwd_status) {
+                               fprintf(stderr, "%s: getcwd: %s\n",
+                                   paths->paths[i], buf);
+                               continue;
+                       } else if (chdir(buf) == -1) {
+                               perror(buf);
+                               continue;
+                       }
+               }
+               if (chdir(paths->paths[i]) == -1) {
                        perror(paths->paths[i]);
                        continue;
                }
+               chdir_status = 1;
 
                c = sqlite3_open_v2(MANDOC_DB, &db,
                    SQLITE_OPEN_READONLY, NULL);
 
                if (SQLITE_OK != c) {
-                       perror(MANDOC_DB);
+                       fprintf(stderr, "%s/%s: %s\n",
+                           paths->paths[i], MANDOC_DB, strerror(errno));
                        sqlite3_close(db);
                        continue;
                }
@@ -301,6 +308,7 @@ mansearch(const struct mansearch *search,
                        mp = mandoc_calloc(1, sizeof(struct match));
                        mp->pageid = pageid;
                        mp->form = sqlite3_column_int(s, 1);
+                       mp->bits = sqlite3_column_int64(s, 3);
                        if (TYPE_Nd == outbit)
                                mp->desc = mandoc_strdup((const char *)
                                    sqlite3_column_text(s, 0));
@@ -336,16 +344,19 @@ mansearch(const struct mansearch *search,
                        }
                        mpage = *res + cur;
                        mpage->ipath = i;
+                       mpage->bits = mp->bits;
                        mpage->sec = 10;
                        mpage->form = mp->form;
-                       buildnames(mpage, db, s, mp->pageid,
+                       buildnames(search, mpage, db, s, mp->pageid,
                            paths->paths[i], mp->form);
-                       mpage->output = TYPE_Nd & outbit ?
-                           mp->desc : outbit ?
-                           buildoutput(db, s2, mp->pageid, outbit) : NULL;
-
+                       if (mpage->names != NULL) {
+                               mpage->output = TYPE_Nd & outbit ?
+                                   mp->desc : outbit ?
+                                   buildoutput(db, s2, mp->pageid, outbit) :
+                                   NULL;
+                               cur++;
+                       }
                        free(mp);
-                       cur++;
                }
 
                sqlite3_finalize(s);
@@ -362,17 +373,12 @@ mansearch(const struct mansearch *search,
                        break;
        }
        qsort(*res, cur, sizeof(struct manpage), manpage_compare);
-       rc = 1;
-out:
-       if (-1 != fd) {
-               if (-1 == fchdir(fd))
-                       perror(buf);
-               close(fd);
-       }
+       if (chdir_status && getcwd_status && chdir(buf) == -1)
+               perror(buf);
        exprfree(e);
        free(sql);
        *sz = cur;
-       return(rc);
+       return 1;
 }
 
 void
@@ -396,23 +402,25 @@ manpage_compare(const void *vp1, const void *vp2)
 
        mp1 = vp1;
        mp2 = vp2;
-       diff = mp1->sec - mp2->sec;
-       return(diff ? diff : strcasecmp(mp1->names, mp2->names));
+       return (diff = mp2->bits - mp1->bits) ? diff :
+           (diff = mp1->sec - mp2->sec) ? diff :
+           strcasecmp(mp1->names, mp2->names);
 }
 
 static void
-buildnames(struct manpage *mpage, sqlite3 *db, sqlite3_stmt *s,
+buildnames(const struct mansearch *search, struct manpage *mpage,
+               sqlite3 *db, sqlite3_stmt *s,
                uint64_t pageid, const char *path, int form)
 {
-       char            *newnames, *prevsec, *prevarch;
+       glob_t           globinfo;
+       char            *firstname, *newnames, *prevsec, *prevarch;
        const char      *oldnames, *sep1, *name, *sec, *sep2, *arch, *fsec;
-       const char      *gzip;
        size_t           i;
-       int              c;
+       int              c, globres;
 
        mpage->file = NULL;
        mpage->names = NULL;
-       prevsec = prevarch = NULL;
+       firstname = prevsec = prevarch = NULL;
        i = 1;
        SQL_BIND_INT64(db, s, i, pageid);
        while (SQLITE_ROW == (c = sqlite3_step(s))) {
@@ -427,10 +435,15 @@ buildnames(struct manpage *mpage, sqlite3 *db, sqlite3_stmt *s,
                        sep1 = ", ";
                }
 
-               /* Fetch the next name. */
+               /* Fetch the next name, rejecting sec/arch mismatches. */
 
                sec = (const char *)sqlite3_column_text(s, 0);
+               if (search->sec != NULL && strcasecmp(sec, search->sec))
+                       continue;
                arch = (const char *)sqlite3_column_text(s, 1);
+               if (search->arch != NULL && *arch != '\0' &&
+                   strcasecmp(arch, search->arch))
+                       continue;
                name = (const char *)sqlite3_column_text(s, 2);
 
                /* Remember the first section found. */
@@ -469,7 +482,7 @@ buildnames(struct manpage *mpage, sqlite3 *db, sqlite3_stmt *s,
 
                /* Also save the first file name encountered. */
 
-               if (NULL != mpage->file)
+               if (mpage->file != NULL)
                        continue;
 
                if (form & FORM_SRC) {
@@ -479,22 +492,41 @@ buildnames(struct manpage *mpage, sqlite3 *db, sqlite3_stmt *s,
                        sep1 = "cat";
                        fsec = "0";
                }
-               if (form & FORM_GZ)
-                       gzip = ".gz";
+               sep2 = *arch == '\0' ? "" : "/";
+               mandoc_asprintf(&mpage->file, "%s/%s%s%s%s/%s.%s",
+                   path, sep1, sec, sep2, arch, name, fsec);
+               if (access(mpage->file, R_OK) != -1)
+                       continue;
+
+               /* Handle unusual file name extensions. */
+
+               if (firstname == NULL)
+                       firstname = mpage->file;
                else
-                       gzip = "";
-               sep2 = '\0' == *arch ? "" : "/";
-               mandoc_asprintf(&mpage->file, "%s/%s%s%s%s/%s.%s%s",
-                   path, sep1, sec, sep2, arch, name, fsec, gzip);
+                       free(mpage->file);
+               mandoc_asprintf(&mpage->file, "%s/%s%s%s%s/%s.*",
+                   path, sep1, sec, sep2, arch, name);
+               globres = glob(mpage->file, 0, NULL, &globinfo);
+               free(mpage->file);
+               mpage->file = globres ? NULL :
+                   mandoc_strdup(*globinfo.gl_pathv);
+               globfree(&globinfo);
        }
-       if (SQLITE_DONE != c)
+       if (c != SQLITE_DONE)
                fprintf(stderr, "%s\n", sqlite3_errmsg(db));
        sqlite3_reset(s);
 
+       /* If none of the files is usable, use the first name. */
+
+       if (mpage->file == NULL)
+               mpage->file = firstname;
+       else if (mpage->file != firstname)
+               free(firstname);
+
        /* Append one final section to the names. */
 
-       if (NULL != prevsec) {
-               sep2 = '\0' == *prevarch ? "" : "/";
+       if (prevsec != NULL) {
+               sep2 = *prevarch == '\0' ? "" : "/";
                mandoc_asprintf(&newnames, "%s(%s%s%s)",
                    mpage->names, prevsec, sep2, prevarch);
                free(mpage->names);
@@ -533,7 +565,7 @@ buildoutput(sqlite3 *db, sqlite3_stmt *s, uint64_t pageid, uint64_t outbit)
        if (SQLITE_DONE != c)
                fprintf(stderr, "%s\n", sqlite3_errmsg(db));
        sqlite3_reset(s);
-       return(output);
+       return output;
 }
 
 /*
@@ -592,8 +624,10 @@ sql_statement(const struct expr *e)
        size_t           sz;
        int              needop;
 
-       sql = mandoc_strdup(
-           "SELECT desc, form, pageid FROM mpages WHERE ");
+       sql = mandoc_strdup(e->equal ?
+           "SELECT desc, form, pageid, bits "
+               "FROM mpages NATURAL JOIN names WHERE " :
+           "SELECT desc, form, pageid, 0 FROM mpages WHERE ");
        sz = strlen(sql);
 
        for (needop = 0; NULL != e; e = e->next) {
@@ -613,8 +647,7 @@ sql_statement(const struct expr *e)
                        ? "pageid IN (SELECT pageid FROM names "
                          "WHERE name REGEXP ?)"
                        : e->equal
-                       ? "pageid IN (SELECT pageid FROM names "
-                         "WHERE name = ?)"
+                       ? "name = ? "
                        : "pageid IN (SELECT pageid FROM names "
                          "WHERE name MATCH ?)")
                    : (NULL == e->substr
@@ -627,7 +660,7 @@ sql_statement(const struct expr *e)
                needop = 1;
        }
 
-       return(sql);
+       return sql;
 }
 
 /*
@@ -643,8 +676,7 @@ exprcomp(const struct mansearch *search, int argc, char *argv[])
        struct expr     *first, *prev, *cur, *next;
 
        first = cur = NULL;
-       logic = igncase = toclose = 0;
-       toopen = NULL != search->sec || NULL != search->arch;
+       logic = igncase = toopen = toclose = 0;
 
        for (i = 0; i < argc; i++) {
                if (0 == strcmp("(", argv[i])) {
@@ -710,45 +742,13 @@ exprcomp(const struct mansearch *search, int argc, char *argv[])
 
                toopen = logic = igncase = 0;
        }
-       if (toopen || logic || igncase || toclose)
-               goto fail;
-
-       if (NULL != search->sec || NULL != search->arch)
-               cur->close++;
-       if (NULL != search->arch)
-               cur = exprspec(cur, TYPE_arch, search->arch, "^(%s|any)$");
-       if (NULL != search->sec)
-               exprspec(cur, TYPE_sec, search->sec, "^%s$");
-
-       return(first);
+       if ( ! (toopen || logic || igncase || toclose))
+               return first;
 
 fail:
        if (NULL != first)
                exprfree(first);
-       return(NULL);
-}
-
-static struct expr *
-exprspec(struct expr *cur, uint64_t key, const char *value,
-               const char *format)
-{
-       char     errbuf[BUFSIZ];
-       char    *cp;
-       int      irc;
-
-       mandoc_asprintf(&cp, format, value);
-       cur->next = mandoc_calloc(1, sizeof(struct expr));
-       cur = cur->next;
-       cur->and = 1;
-       cur->bits = key;
-       if (0 != (irc = regcomp(&cur->regexp, cp,
-           REG_EXTENDED | REG_NOSUB | REG_ICASE))) {
-               regerror(irc, &cur->regexp, errbuf, sizeof(errbuf));
-               fprintf(stderr, "regcomp: %s\n", errbuf);
-               cur->substr = value;
-       }
-       free(cp);
-       return(cur);
+       return NULL;
 }
 
 static struct expr *
@@ -761,7 +761,7 @@ exprterm(const struct mansearch *search, char *buf, int cs)
        int              i, irc;
 
        if ('\0' == *buf)
-               return(NULL);
+               return NULL;
 
        e = mandoc_calloc(1, sizeof(struct expr));
 
@@ -769,7 +769,7 @@ exprterm(const struct mansearch *search, char *buf, int cs)
                e->bits = TYPE_Nm;
                e->substr = buf;
                e->equal = 1;
-               return(e);
+               return e;
        }
 
        /*
@@ -807,12 +807,12 @@ exprterm(const struct mansearch *search, char *buf, int cs)
                        regerror(irc, &e->regexp, errbuf, sizeof(errbuf));
                        fprintf(stderr, "regcomp: %s\n", errbuf);
                        free(e);
-                       return(NULL);
+                       return NULL;
                }
        }
 
        if (e->bits)
-               return(e);
+               return e;
 
        /*
         * Parse out all possible fields.
@@ -834,13 +834,13 @@ exprterm(const struct mansearch *search, char *buf, int cs)
                if (i == mansearch_keymax) {
                        if (strcasecmp(key, "any")) {
                                free(e);
-                               return(NULL);
+                               return NULL;
                        }
                        e->bits |= ~0ULL;
                }
        }
 
-       return(e);
+       return e;
 }
 
 static void
@@ -859,14 +859,14 @@ static void *
 hash_calloc(size_t nmemb, size_t sz, void *arg)
 {
 
-       return(mandoc_calloc(nmemb, sz));
+       return mandoc_calloc(nmemb, sz);
 }
 
 static void *
 hash_alloc(size_t sz, void *arg)
 {
 
-       return(mandoc_malloc(sz));
+       return mandoc_malloc(sz);
 }
 
 static void