[BACK]Return to mansearch.c CVS log [TXT][DIR] Up to [cvsweb.bsd.lv] / mandoc

Diff for /mandoc/mansearch.c between version 1.1 and 1.6

version 1.1, 2012/06/08 10:36:23 version 1.6, 2013/06/05 02:00:26
Line 18 
Line 18 
 #include "config.h"  #include "config.h"
 #endif  #endif
   
 #include <sys/param.h>  
   
 #include <assert.h>  #include <assert.h>
 #include <fcntl.h>  #include <fcntl.h>
 #include <getopt.h>  #include <getopt.h>
   #include <limits.h>
 #include <stdio.h>  #include <stdio.h>
 #include <stdint.h>  #include <stdint.h>
 #include <stddef.h>  #include <stddef.h>
Line 30 
Line 29 
 #include <string.h>  #include <string.h>
 #include <unistd.h>  #include <unistd.h>
   
   #ifdef HAVE_OHASH
 #include <ohash.h>  #include <ohash.h>
   #else
   #include "compat_ohash.h"
   #endif
 #include <sqlite3.h>  #include <sqlite3.h>
   
 #include "mandoc.h"  #include "mandoc.h"
 #include "manpath.h"  #include "manpath.h"
 #include "mandocdb.h"  
 #include "mansearch.h"  #include "mansearch.h"
   
   #define SQL_BIND_TEXT(_db, _s, _i, _v) \
           if (SQLITE_OK != sqlite3_bind_text \
                   ((_s), (_i)++, (_v), -1, SQLITE_STATIC)) \
                   fprintf(stderr, "%s\n", sqlite3_errmsg((_db)))
   #define SQL_BIND_INT64(_db, _s, _i, _v) \
           if (SQLITE_OK != sqlite3_bind_int64 \
                   ((_s), (_i)++, (_v))) \
                   fprintf(stderr, "%s\n", sqlite3_errmsg((_db)))
   
 struct  expr {  struct  expr {
         int              glob; /* is glob? */          int              glob; /* is glob? */
         uint64_t         bits; /* type-mask */          uint64_t         bits; /* type-mask */
Line 104  static const struct type types[] = {
Line 115  static const struct type types[] = {
 static  void            *hash_alloc(size_t, void *);  static  void            *hash_alloc(size_t, void *);
 static  void             hash_free(void *, size_t, void *);  static  void             hash_free(void *, size_t, void *);
 static  void            *hash_halloc(size_t, void *);  static  void            *hash_halloc(size_t, void *);
 static  struct expr     *exprcomp(int, char *[]);  static  struct expr     *exprcomp(const struct mansearch *,
                                   int, char *[]);
 static  void             exprfree(struct expr *);  static  void             exprfree(struct expr *);
 static  struct expr     *exprterm(char *);  static  struct expr     *exprterm(const struct mansearch *, char *);
 static  char            *sql_statement(const struct expr *,  static  char            *sql_statement(const struct expr *,
                                 const char *, const char *);                                  const char *, const char *);
   
 int  int
 mansearch(const struct manpaths *paths,  mansearch(const struct mansearch *search,
                 const char *arch, const char *sec,                  const struct manpaths *paths,
                 int argc, char *argv[],                  int argc, char *argv[],
                 struct manpage **res, size_t *sz)                  struct manpage **res, size_t *sz)
 {  {
         int              fd, rc;          int              fd, rc, c;
         int64_t          id;          int64_t          id;
         char             buf[MAXPATHLEN];          char             buf[PATH_MAX];
         char            *sql;          char            *sql;
         struct expr     *e, *ep;          struct expr     *e, *ep;
         sqlite3         *db;          sqlite3         *db;
Line 136  mansearch(const struct manpaths *paths, 
Line 148  mansearch(const struct manpaths *paths, 
         info.hfree = hash_free;          info.hfree = hash_free;
         info.key_offset = offsetof(struct match, id);          info.key_offset = offsetof(struct match, id);
   
         *sz = 0;          *sz = cur = maxres = 0;
         sql = NULL;          sql = NULL;
         *res = NULL;          *res = NULL;
         fd = -1;          fd = -1;
         e = NULL;          e = NULL;
         cur = maxres = 0;          rc = 0;
   
         if (0 == argc)          if (0 == argc)
                 goto out;                  goto out;
         if (NULL == (e = exprcomp(argc, argv)))          if (NULL == (e = exprcomp(search, argc, argv)))
                 goto out;                  goto out;
   
         /*          /*
Line 155  mansearch(const struct manpaths *paths, 
Line 167  mansearch(const struct manpaths *paths, 
          * on our current directory from which to start the chdir().           * on our current directory from which to start the chdir().
          */           */
   
         if (NULL == getcwd(buf, MAXPATHLEN)) {          if (NULL == getcwd(buf, PATH_MAX)) {
                 perror(NULL);                  perror(NULL);
                 goto out;                  goto out;
         } else if (-1 == (fd = open(buf, O_RDONLY, 0))) {          } else if (-1 == (fd = open(buf, O_RDONLY, 0))) {
Line 163  mansearch(const struct manpaths *paths, 
Line 175  mansearch(const struct manpaths *paths, 
                 goto out;                  goto out;
         }          }
   
         sql = sql_statement(e, arch, sec);          sql = sql_statement(e, search->arch, search->sec);
   
         /*          /*
          * Loop over the directories (containing databases) for us to           * Loop over the directories (containing databases) for us to
Line 175  mansearch(const struct manpaths *paths, 
Line 187  mansearch(const struct manpaths *paths, 
   
         for (i = 0; i < paths->sz; i++) {          for (i = 0; i < paths->sz; i++) {
                 if (-1 == fchdir(fd)) {                  if (-1 == fchdir(fd)) {
                         /* FIXME: will return success */  
                         perror(buf);                          perror(buf);
                         free(*res);                          free(*res);
                         break;                          break;
Line 184  mansearch(const struct manpaths *paths, 
Line 195  mansearch(const struct manpaths *paths, 
                         continue;                          continue;
                 }                  }
   
                 rc =  sqlite3_open_v2                  c =  sqlite3_open_v2
                         (MANDOC_DB, &db, SQLITE_OPEN_READONLY, NULL);                          (MANDOC_DB, &db,
                            SQLITE_OPEN_READONLY, NULL);
   
                 if (SQLITE_OK != rc) {                  if (SQLITE_OK != c) {
                         perror(MANDOC_DB);                          perror(MANDOC_DB);
                         sqlite3_close(db);                          sqlite3_close(db);
                         continue;                          continue;
                 }                  }
   
                 j = 1;                  j = 1;
                 sqlite3_prepare_v2(db, sql, -1, &s, NULL);                  c = sqlite3_prepare_v2(db, sql, -1, &s, NULL);
                   if (SQLITE_OK != c)
                           fprintf(stderr, "%s\n", sqlite3_errmsg(db));
   
                 if (NULL != arch)                  if (NULL != search->arch)
                         sqlite3_bind_text                          SQL_BIND_TEXT(db, s, j, search->arch);
                                 (s, j++, arch, -1, SQLITE_STATIC);                  if (NULL != search->sec)
                 if (NULL != sec)                          SQL_BIND_TEXT(db, s, j, search->sec);
                         sqlite3_bind_text  
                                 (s, j++, sec, -1, SQLITE_STATIC);  
   
                 for (ep = e; NULL != ep; ep = ep->next) {                  for (ep = e; NULL != ep; ep = ep->next) {
                         sqlite3_bind_text                          SQL_BIND_TEXT(db, s, j, ep->v);
                                 (s, j++, ep->v, -1, SQLITE_STATIC);                          SQL_BIND_INT64(db, s, j, ep->bits);
                         sqlite3_bind_int64  
                                 (s, j++, ep->bits);  
                 }                  }
   
                 memset(&htab, 0, sizeof(struct ohash));                  memset(&htab, 0, sizeof(struct ohash));
Line 221  mansearch(const struct manpaths *paths, 
Line 231  mansearch(const struct manpaths *paths, 
                  * This gives good performance and preserves the                   * This gives good performance and preserves the
                  * distribution of buckets in the table.                   * distribution of buckets in the table.
                  */                   */
                 while (SQLITE_ROW == sqlite3_step(s)) {                  while (SQLITE_ROW == (c = sqlite3_step(s))) {
                         id = sqlite3_column_int64(s, 0);                          id = sqlite3_column_int64(s, 0);
                         idx = ohash_lookup_memory                          idx = ohash_lookup_memory
                                 (&htab, (char *)&id,                                  (&htab, (char *)&id,
Line 240  mansearch(const struct manpaths *paths, 
Line 250  mansearch(const struct manpaths *paths, 
                         ohash_insert(&htab, idx, mp);                          ohash_insert(&htab, idx, mp);
                 }                  }
   
                   if (SQLITE_DONE != c)
                           fprintf(stderr, "%s\n", sqlite3_errmsg(db));
   
                 sqlite3_finalize(s);                  sqlite3_finalize(s);
                 sqlite3_close(db);                  sqlite3_close(db);
   
Line 252  mansearch(const struct manpaths *paths, 
Line 265  mansearch(const struct manpaths *paths, 
                                         (*res, maxres * sizeof(struct manpage));                                          (*res, maxres * sizeof(struct manpage));
                         }                          }
                         strlcpy((*res)[cur].file,                          strlcpy((*res)[cur].file,
                                 paths->paths[i], MAXPATHLEN);                                  paths->paths[i], PATH_MAX);
                         strlcat((*res)[cur].file, "/", MAXPATHLEN);                          strlcat((*res)[cur].file, "/", PATH_MAX);
                         strlcat((*res)[cur].file, mp->file, MAXPATHLEN);                          strlcat((*res)[cur].file, mp->file, PATH_MAX);
                         (*res)[cur].desc = mp->desc;                          (*res)[cur].desc = mp->desc;
                         (*res)[cur].form = mp->form;                          (*res)[cur].form = mp->form;
                         free(mp->file);                          free(mp->file);
Line 263  mansearch(const struct manpaths *paths, 
Line 276  mansearch(const struct manpaths *paths, 
                 }                  }
                 ohash_delete(&htab);                  ohash_delete(&htab);
         }          }
           rc = 1;
 out:  out:
         exprfree(e);          exprfree(e);
         if (-1 != fd)          if (-1 != fd)
                 close(fd);                  close(fd);
         free(sql);          free(sql);
         *sz = cur;          *sz = cur;
         return(1);          return(rc);
 }  }
   
 /*  /*
Line 285  sql_statement(const struct expr *e, const char *arch, 
Line 299  sql_statement(const struct expr *e, const char *arch, 
         const char      *eq = "(key = ? AND bits & ?)";          const char      *eq = "(key = ? AND bits & ?)";
         const char      *andarch = "arch = ? AND ";          const char      *andarch = "arch = ? AND ";
         const char      *andsec = "sec = ? AND ";          const char      *andsec = "sec = ? AND ";
         const size_t     globsz = 27;          size_t           globsz;
         const size_t     eqsz = 22;          size_t           eqsz;
         size_t           sz;          size_t           sz;
   
         sql = mandoc_strdup          sql = mandoc_strdup
Line 295  sql_statement(const struct expr *e, const char *arch, 
Line 309  sql_statement(const struct expr *e, const char *arch, 
                  "INNER JOIN docs ON docs.id=keys.docid "                   "INNER JOIN docs ON docs.id=keys.docid "
                  "WHERE ");                   "WHERE ");
         sz = strlen(sql);          sz = strlen(sql);
           globsz = strlen(glob);
           eqsz = strlen(eq);
   
         if (NULL != arch) {          if (NULL != arch) {
                 sz += strlen(andarch) + 1;                  sz += strlen(andarch) + 1;
                 sql = mandoc_realloc(sql, sz);                  sql = mandoc_realloc(sql, sz);
                 strlcat(sql, andarch, sz);                  strlcat(sql, andarch, sz);
         }          }
   
         if (NULL != sec) {          if (NULL != sec) {
                 sz += strlen(andsec) + 1;                  sz += strlen(andsec) + 1;
                 sql = mandoc_realloc(sql, sz);                  sql = mandoc_realloc(sql, sz);
Line 328  sql_statement(const struct expr *e, const char *arch, 
Line 345  sql_statement(const struct expr *e, const char *arch, 
  * "(", "foo=bar", etc.).   * "(", "foo=bar", etc.).
  */   */
 static struct expr *  static struct expr *
 exprcomp(int argc, char *argv[])  exprcomp(const struct mansearch *search, int argc, char *argv[])
 {  {
         int              i;          int              i;
         struct expr     *first, *next, *cur;          struct expr     *first, *next, *cur;
Line 336  exprcomp(int argc, char *argv[])
Line 353  exprcomp(int argc, char *argv[])
         first = cur = NULL;          first = cur = NULL;
   
         for (i = 0; i < argc; i++) {          for (i = 0; i < argc; i++) {
                 next = exprterm(argv[i]);                  next = exprterm(search, argv[i]);
                 if (NULL == next) {                  if (NULL == next) {
                         exprfree(first);                          exprfree(first);
                         return(NULL);                          return(NULL);
Line 352  exprcomp(int argc, char *argv[])
Line 369  exprcomp(int argc, char *argv[])
 }  }
   
 static struct expr *  static struct expr *
 exprterm(char *buf)  exprterm(const struct mansearch *search, char *buf)
 {  {
         struct expr     *e;          struct expr     *e;
         char            *key, *v;          char            *key, *v;
Line 363  exprterm(char *buf)
Line 380  exprterm(char *buf)
   
         e = mandoc_calloc(1, sizeof(struct expr));          e = mandoc_calloc(1, sizeof(struct expr));
   
           /*"whatis" mode uses an opaque string and default fields. */
   
           if (MANSEARCH_WHATIS & search->flags) {
                   e->v = buf;
                   e->bits = search->deftype;
                   return(e);
           }
   
         /*          /*
          * If no =~ is specified, search with equality over names and           * If no =~ is specified, search with equality over names and
          * descriptions.           * descriptions.
Line 371  exprterm(char *buf)
Line 396  exprterm(char *buf)
   
         if (NULL == (v = strpbrk(buf, "=~"))) {          if (NULL == (v = strpbrk(buf, "=~"))) {
                 e->v = buf;                  e->v = buf;
                 e->bits = TYPE_Nm | TYPE_Nd;                  e->bits = search->deftype;
                 return(e);                  return(e);
         } else if (v == buf)          } else if (v == buf)
                 e->bits = TYPE_Nm | TYPE_Nd;                  e->bits = search->deftype;
   
         e->glob = '~' == *v;          e->glob = '~' == *v;
         *v++ = '\0';          *v++ = '\0';
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.6
CVSweb