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

Diff for /mandoc/mansearch.c between version 1.7 and 1.8

version 1.7, 2013/10/19 20:43:13 version 1.8, 2013/10/20 00:03:05
Line 23 
Line 23 
 #include <fcntl.h>  #include <fcntl.h>
 #include <getopt.h>  #include <getopt.h>
 #include <limits.h>  #include <limits.h>
   #include <regex.h>
 #include <stdio.h>  #include <stdio.h>
 #include <stdint.h>  #include <stdint.h>
 #include <stddef.h>  #include <stddef.h>
Line 42 
Line 43 
 #include "mansearch.h"  #include "mansearch.h"
   
 #define SQL_BIND_TEXT(_db, _s, _i, _v) \  #define SQL_BIND_TEXT(_db, _s, _i, _v) \
         if (SQLITE_OK != sqlite3_bind_text \          do { if (SQLITE_OK != sqlite3_bind_text \
                 ((_s), (_i)++, (_v), -1, SQLITE_STATIC)) \                  ((_s), (_i)++, (_v), -1, SQLITE_STATIC)) \
                 fprintf(stderr, "%s\n", sqlite3_errmsg((_db)))                  fprintf(stderr, "%s\n", sqlite3_errmsg((_db))); \
           } while (0)
 #define SQL_BIND_INT64(_db, _s, _i, _v) \  #define SQL_BIND_INT64(_db, _s, _i, _v) \
         if (SQLITE_OK != sqlite3_bind_int64 \          do { if (SQLITE_OK != sqlite3_bind_int64 \
                 ((_s), (_i)++, (_v))) \                  ((_s), (_i)++, (_v))) \
                 fprintf(stderr, "%s\n", sqlite3_errmsg((_db)))                  fprintf(stderr, "%s\n", sqlite3_errmsg((_db))); \
           } while (0)
   #define SQL_BIND_BLOB(_db, _s, _i, _v) \
           do { if (SQLITE_OK != sqlite3_bind_blob \
                   ((_s), (_i)++, (&_v), sizeof(_v), SQLITE_STATIC)) \
                   fprintf(stderr, "%s\n", sqlite3_errmsg((_db))); \
           } while (0)
   
 struct  expr {  struct  expr {
         int              glob; /* is glob? */          uint64_t         bits;    /* type-mask */
         uint64_t         bits; /* type-mask */          const char      *substr;  /* to search for, if applicable */
         const char      *v; /* search value */          regex_t          regexp;  /* compiled regexp, if applicable */
         struct expr     *next; /* next in sequence */          struct expr     *next;    /* next in sequence */
 };  };
   
 struct  match {  struct  match {
Line 119  static void  *hash_halloc(size_t, void *);
Line 127  static void  *hash_halloc(size_t, void *);
 static  struct expr     *exprcomp(const struct mansearch *,  static  struct expr     *exprcomp(const struct mansearch *,
                                 int, char *[]);                                  int, char *[]);
 static  void             exprfree(struct expr *);  static  void             exprfree(struct expr *);
 static  struct expr     *exprterm(const struct mansearch *, char *);  static  struct expr     *exprterm(const struct mansearch *, char *, int);
 static  void             sql_match(sqlite3_context *context,  static  void             sql_match(sqlite3_context *context,
                                 int argc, sqlite3_value **argv);                                  int argc, sqlite3_value **argv);
   static  void             sql_regexp(sqlite3_context *context,
                                   int argc, sqlite3_value **argv);
 static  char            *sql_statement(const struct expr *,  static  char            *sql_statement(const struct expr *,
                                 const char *, const char *);                                  const char *, const char *);
   
Line 208  mansearch(const struct mansearch *search,
Line 218  mansearch(const struct mansearch *search,
                         continue;                          continue;
                 }                  }
   
                 /* Define the SQL function for substring matching. */                  /*
                    * Define the SQL functions for substring
                    * and regular expression matching.
                    */
   
                 c = sqlite3_create_function(db, "match", 2,                  c = sqlite3_create_function(db, "match", 2,
                     SQLITE_ANY, NULL, sql_match, NULL, NULL);                      SQLITE_ANY, NULL, sql_match, NULL, NULL);
                 if (SQLITE_OK != c) {                  assert(SQLITE_OK == c);
                         fprintf(stderr, "%s\n", sqlite3_errmsg(db));                  c = sqlite3_create_function(db, "regexp", 2,
                         break;                      SQLITE_ANY, NULL, sql_regexp, NULL, NULL);
                 }                  assert(SQLITE_OK == c);
   
                 j = 1;                  j = 1;
                 c = sqlite3_prepare_v2(db, sql, -1, &s, NULL);                  c = sqlite3_prepare_v2(db, sql, -1, &s, NULL);
Line 228  mansearch(const struct mansearch *search,
Line 241  mansearch(const struct mansearch *search,
                         SQL_BIND_TEXT(db, s, j, search->sec);                          SQL_BIND_TEXT(db, s, j, search->sec);
   
                 for (ep = e; NULL != ep; ep = ep->next) {                  for (ep = e; NULL != ep; ep = ep->next) {
                         SQL_BIND_TEXT(db, s, j, ep->v);                          if (NULL == ep->substr) {
                                   SQL_BIND_BLOB(db, s, j, ep->regexp);
                           } else
                                   SQL_BIND_TEXT(db, s, j, ep->substr);
                         SQL_BIND_INT64(db, s, j, ep->bits);                          SQL_BIND_INT64(db, s, j, ep->bits);
                 }                  }
   
Line 315  sql_match(sqlite3_context *context, int argc, sqlite3_
Line 331  sql_match(sqlite3_context *context, int argc, sqlite3_
 }  }
   
 /*  /*
    * Implement regular expression match
    * as an application-defined SQL function.
    */
   static void
   sql_regexp(sqlite3_context *context, int argc, sqlite3_value **argv)
   {
   
           assert(2 == argc);
           sqlite3_result_int(context, !regexec(
               (regex_t *)sqlite3_value_blob(argv[0]),
               (const char *)sqlite3_value_text(argv[1]),
               0, NULL, 0));
   }
   
   /*
  * Prepare the search SQL statement.   * Prepare the search SQL statement.
  * We search for any of the words specified in our match expression.   * We search for any of the words specified in our match expression.
  * We filter the per-doc AND expressions when collecting results.   * We filter the per-doc AND expressions when collecting results.
Line 324  sql_statement(const struct expr *e, const char *arch, 
Line 355  sql_statement(const struct expr *e, const char *arch, 
 {  {
         char            *sql;          char            *sql;
         const char      *substr = "(key MATCH ? AND bits & ?)";          const char      *substr = "(key MATCH ? AND bits & ?)";
         const char      *glob = "(key GLOB ? AND bits & ?)";          const char      *regexp = "(key REGEXP ? AND bits & ?)";
         const char      *andarch = "arch = ? AND ";          const char      *andarch = "arch = ? AND ";
         const char      *andsec = "sec = ? AND ";          const char      *andsec = "sec = ? AND ";
         size_t           substrsz;          size_t           substrsz;
         size_t           globsz;          size_t           regexpsz;
         size_t           sz;          size_t           sz;
   
         sql = mandoc_strdup          sql = mandoc_strdup
Line 338  sql_statement(const struct expr *e, const char *arch, 
Line 369  sql_statement(const struct expr *e, const char *arch, 
                  "WHERE ");                   "WHERE ");
         sz = strlen(sql);          sz = strlen(sql);
         substrsz = strlen(substr);          substrsz = strlen(substr);
         globsz = strlen(glob);          regexpsz = strlen(regexp);
   
         if (NULL != arch) {          if (NULL != arch) {
                 sz += strlen(andarch) + 1;                  sz += strlen(andarch) + 1;
Line 357  sql_statement(const struct expr *e, const char *arch, 
Line 388  sql_statement(const struct expr *e, const char *arch, 
         strlcat(sql, "(", sz);          strlcat(sql, "(", sz);
   
         for ( ; NULL != e; e = e->next) {          for ( ; NULL != e; e = e->next) {
                 sz += (e->glob ? globsz : substrsz) +                  sz += (NULL == e->substr ? regexpsz : substrsz) +
                         (NULL == e->next ? 3 : 5);                          (NULL == e->next ? 3 : 5);
                 sql = mandoc_realloc(sql, sz);                  sql = mandoc_realloc(sql, sz);
                 strlcat(sql, e->glob ? glob : substr, sz);                  strlcat(sql, NULL == e->substr ? regexp : substr, sz);
                 strlcat(sql, NULL == e->next ? ");" : " OR ", sz);                  strlcat(sql, NULL == e->next ? ");" : " OR ", sz);
         }          }
   
Line 375  sql_statement(const struct expr *e, const char *arch, 
Line 406  sql_statement(const struct expr *e, const char *arch, 
 static struct expr *  static struct expr *
 exprcomp(const struct mansearch *search, int argc, char *argv[])  exprcomp(const struct mansearch *search, int argc, char *argv[])
 {  {
         int              i;          int              i, cs;
         struct expr     *first, *next, *cur;          struct expr     *first, *next, *cur;
   
         first = cur = NULL;          first = cur = NULL;
   
         for (i = 0; i < argc; i++) {          for (i = 0; i < argc; i++) {
                 next = exprterm(search, argv[i]);                  if (0 == strcmp("-i", argv[i])) {
                           if (++i >= argc)
                                   return(NULL);
                           cs = 0;
                   } else
                           cs = 1;
                   next = exprterm(search, argv[i], cs);
                 if (NULL == next) {                  if (NULL == next) {
                         exprfree(first);                          exprfree(first);
                         return(NULL);                          return(NULL);
Line 397  exprcomp(const struct mansearch *search, int argc, cha
Line 434  exprcomp(const struct mansearch *search, int argc, cha
 }  }
   
 static struct expr *  static struct expr *
 exprterm(const struct mansearch *search, char *buf)  exprterm(const struct mansearch *search, char *buf, int cs)
 {  {
         struct expr     *e;          struct expr     *e;
         char            *key, *v;          char            *key, *v;
Line 411  exprterm(const struct mansearch *search, char *buf)
Line 448  exprterm(const struct mansearch *search, char *buf)
         /*"whatis" mode uses an opaque string and default fields. */          /*"whatis" mode uses an opaque string and default fields. */
   
         if (MANSEARCH_WHATIS & search->flags) {          if (MANSEARCH_WHATIS & search->flags) {
                 e->v = buf;                  e->substr = buf;
                 e->bits = search->deftype;                  e->bits = search->deftype;
                 return(e);                  return(e);
         }          }
Line 423  exprterm(const struct mansearch *search, char *buf)
Line 460  exprterm(const struct mansearch *search, char *buf)
          */           */
   
         if (NULL == (v = strpbrk(buf, "=~"))) {          if (NULL == (v = strpbrk(buf, "=~"))) {
                 e->v = buf;                  e->substr = buf;
                 e->bits = search->deftype;                  e->bits = search->deftype;
                 return(e);                  return(e);
         } else if (v == buf)          } else if (v == buf)
                 e->bits = search->deftype;                  e->bits = search->deftype;
   
         e->glob = '~' == *v;          if ('~' == *v++) {
         *v++ = '\0';                  if (regcomp(&e->regexp, v,
         e->v = v;                      REG_EXTENDED | REG_NOSUB | (cs ? 0 : REG_ICASE))) {
                           free(e);
                           return(NULL);
                   }
           } else
                   e->substr = v;
           v[-1] = '\0';
   
         /*          /*
          * Parse out all possible fields.           * Parse out all possible fields.

Legend:
Removed from v.1.7  
changed lines
  Added in v.1.8

CVSweb