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

Diff for /mandoc/mandocdb.c between version 1.101 and 1.113

version 1.101, 2014/01/05 04:48:40 version 1.113, 2014/01/19 22:48:16
Line 48 
Line 48 
 #include "manpath.h"  #include "manpath.h"
 #include "mansearch.h"  #include "mansearch.h"
   
   extern int mansearch_keymax;
   extern const char *const mansearch_keynames[];
   
 #define SQL_EXEC(_v) \  #define SQL_EXEC(_v) \
         if (SQLITE_OK != sqlite3_exec(db, (_v), NULL, NULL, NULL)) \          if (SQLITE_OK != sqlite3_exec(db, (_v), NULL, NULL, NULL)) \
                 fprintf(stderr, "%s\n", sqlite3_errmsg(db))                  fprintf(stderr, "%s\n", sqlite3_errmsg(db))
Line 130  struct mdoc_handler {
Line 133  struct mdoc_handler {
 };  };
   
 static  void     dbclose(int);  static  void     dbclose(int);
 static  void     dbindex(const struct mpage *, struct mchars *);  static  void     dbadd(const struct mpage *, struct mchars *);
 static  int      dbopen(int);  static  int      dbopen(int);
 static  void     dbprune(void);  static  void     dbprune(void);
 static  void     filescan(const char *);  static  void     filescan(const char *);
Line 154  static int  parse_mdoc_Nd(struct mpage *, const struct
Line 157  static int  parse_mdoc_Nd(struct mpage *, const struct
 static  int      parse_mdoc_Nm(struct mpage *, const struct mdoc_node *);  static  int      parse_mdoc_Nm(struct mpage *, const struct mdoc_node *);
 static  int      parse_mdoc_Sh(struct mpage *, const struct mdoc_node *);  static  int      parse_mdoc_Sh(struct mpage *, const struct mdoc_node *);
 static  int      parse_mdoc_Xr(struct mpage *, const struct mdoc_node *);  static  int      parse_mdoc_Xr(struct mpage *, const struct mdoc_node *);
 static  void     putkey(const struct mpage *,  static  void     putkey(const struct mpage *, char *, uint64_t);
                         const char *, uint64_t);  
 static  void     putkeys(const struct mpage *,  static  void     putkeys(const struct mpage *,
                         const char *, size_t, uint64_t);                          const char *, size_t, uint64_t);
 static  void     putmdockey(const struct mpage *,  static  void     putmdockey(const struct mpage *,
Line 167  static int  treescan(void);
Line 169  static int  treescan(void);
 static  size_t   utf8(unsigned int, char [7]);  static  size_t   utf8(unsigned int, char [7]);
   
 static  char            *progname;  static  char            *progname;
 static  int              use_all; /* use all found files */  
 static  int              nodb; /* no database changes */  static  int              nodb; /* no database changes */
   static  int              quick; /* abort the parse early */
   static  int              use_all; /* use all found files */
 static  int              verb; /* print what we're doing */  static  int              verb; /* print what we're doing */
 static  int              warnings; /* warn about crap */  static  int              warnings; /* warn about crap */
 static  int              write_utf8; /* write UTF-8 output; else ASCII */  static  int              write_utf8; /* write UTF-8 output; else ASCII */
Line 347  main(int argc, char *argv[])
Line 350  main(int argc, char *argv[])
         path_arg = NULL;          path_arg = NULL;
         op = OP_DEFAULT;          op = OP_DEFAULT;
   
         while (-1 != (ch = getopt(argc, argv, "aC:d:nT:tu:vW")))          while (-1 != (ch = getopt(argc, argv, "aC:d:nQT:tu:vW")))
                 switch (ch) {                  switch (ch) {
                 case ('a'):                  case ('a'):
                         use_all = 1;                          use_all = 1;
Line 365  main(int argc, char *argv[])
Line 368  main(int argc, char *argv[])
                 case ('n'):                  case ('n'):
                         nodb = 1;                          nodb = 1;
                         break;                          break;
                   case ('Q'):
                           quick = 1;
                           break;
                 case ('T'):                  case ('T'):
                         if (strcmp(optarg, "utf8")) {                          if (strcmp(optarg, "utf8")) {
                                 fprintf(stderr, "-T%s: Unsupported "                                  fprintf(stderr, "-T%s: Unsupported "
Line 404  main(int argc, char *argv[])
Line 410  main(int argc, char *argv[])
   
         exitcode = (int)MANDOCLEVEL_OK;          exitcode = (int)MANDOCLEVEL_OK;
         mp = mparse_alloc(MPARSE_AUTO,          mp = mparse_alloc(MPARSE_AUTO,
                 MANDOCLEVEL_FATAL, NULL, NULL, NULL);                  MANDOCLEVEL_FATAL, NULL, NULL, quick);
         mc = mchars_alloc();          mc = mchars_alloc();
   
         ohash_init(&mpages, 6, &mpages_info);          ohash_init(&mpages, 6, &mpages_info);
Line 494  out:
Line 500  out:
         ohash_delete(&mlinks);          ohash_delete(&mlinks);
         return(exitcode);          return(exitcode);
 usage:  usage:
         fprintf(stderr, "usage: %s [-anvW] [-C file] [-Tutf8]\n"          fprintf(stderr, "usage: %s [-anQvW] [-C file] [-Tutf8]\n"
                         "       %s [-anvW] [-Tutf8] dir ...\n"                          "       %s [-anQvW] [-Tutf8] dir ...\n"
                         "       %s [-nvW] [-Tutf8] -d dir [file ...]\n"                          "       %s [-nQvW] [-Tutf8] -d dir [file ...]\n"
                         "       %s [-nvW] -u dir [file ...]\n"                          "       %s [-nvW] -u dir [file ...]\n"
                         "       %s -t file ...\n",                          "       %s [-Q] -t file ...\n",
                        progname, progname, progname,                         progname, progname, progname,
                        progname, progname);                         progname, progname);
   
Line 696  filescan(const char *file)
Line 702  filescan(const char *file)
                 exitcode = (int)MANDOCLEVEL_BADARG;                  exitcode = (int)MANDOCLEVEL_BADARG;
                 say(file, NULL);                  say(file, NULL);
                 return;                  return;
         } else if (OP_TEST != op && strstr(buf, basedir) != buf) {          }
   
           if (strstr(buf, basedir) == buf)
                   start = buf + strlen(basedir) + 1;
           else if (OP_TEST == op)
                   start = buf;
           else {
                 exitcode = (int)MANDOCLEVEL_BADARG;                  exitcode = (int)MANDOCLEVEL_BADARG;
                 say("", "%s: outside base directory", buf);                  say("", "%s: outside base directory", buf);
                 return;                  return;
         } else if (-1 == stat(buf, &st)) {          }
   
           if (-1 == stat(buf, &st)) {
                 exitcode = (int)MANDOCLEVEL_BADARG;                  exitcode = (int)MANDOCLEVEL_BADARG;
                 say(file, NULL);                  say(file, NULL);
                 return;                  return;
Line 709  filescan(const char *file)
Line 723  filescan(const char *file)
                 say(file, "Not a regular file");                  say(file, "Not a regular file");
                 return;                  return;
         }          }
         start = buf + strlen(basedir);  
         mlink = mandoc_calloc(1, sizeof(struct mlink));          mlink = mandoc_calloc(1, sizeof(struct mlink));
         strlcpy(mlink->file, start, sizeof(mlink->file));          strlcpy(mlink->file, start, sizeof(mlink->file));
   
Line 942  mlink_check(struct mpage *mpage, struct mlink *mlink)
Line 956  mlink_check(struct mpage *mpage, struct mlink *mlink)
 static void  static void
 mpages_merge(struct mchars *mc, struct mparse *mp)  mpages_merge(struct mchars *mc, struct mparse *mp)
 {  {
           char                     any[] = "any";
         struct ohash_info        str_info;          struct ohash_info        str_info;
         struct mpage            *mpage;          struct mpage            *mpage;
         struct mlink            *mlink;          struct mlink            *mlink;
         struct mdoc             *mdoc;          struct mdoc             *mdoc;
         struct man              *man;          struct man              *man;
         const char              *cp;          char                    *cp;
         int                      match;          int                      match;
         unsigned int             pslot;          unsigned int             pslot;
         enum mandoclevel         lvl;          enum mandoclevel         lvl;
Line 957  mpages_merge(struct mchars *mc, struct mparse *mp)
Line 972  mpages_merge(struct mchars *mc, struct mparse *mp)
         str_info.hfree = hash_free;          str_info.hfree = hash_free;
         str_info.key_offset = offsetof(struct str, key);          str_info.key_offset = offsetof(struct str, key);
   
           if (0 == nodb)
                   SQL_EXEC("BEGIN TRANSACTION");
   
         mpage = ohash_first(&mpages, &pslot);          mpage = ohash_first(&mpages, &pslot);
         while (NULL != mpage) {          while (NULL != mpage) {
                 mlinks_undupe(mpage);                  mlinks_undupe(mpage);
Line 1010  mpages_merge(struct mchars *mc, struct mparse *mp)
Line 1028  mpages_merge(struct mchars *mc, struct mparse *mp)
                 }                  }
                 putkey(mpage, mpage->sec, TYPE_sec);                  putkey(mpage, mpage->sec, TYPE_sec);
                 putkey(mpage, '\0' == *mpage->arch ?                  putkey(mpage, '\0' == *mpage->arch ?
                     "any" : mpage->arch, TYPE_arch);                      any : mpage->arch, TYPE_arch);
   
                 for (mlink = mpage->mlinks; mlink; mlink = mlink->next) {                  for (mlink = mpage->mlinks; mlink; mlink = mlink->next) {
                         if ('\0' != *mlink->dsec)                          if ('\0' != *mlink->dsec)
Line 1018  mpages_merge(struct mchars *mc, struct mparse *mp)
Line 1036  mpages_merge(struct mchars *mc, struct mparse *mp)
                         if ('\0' != *mlink->fsec)                          if ('\0' != *mlink->fsec)
                                 putkey(mpage, mlink->fsec, TYPE_sec);                                  putkey(mpage, mlink->fsec, TYPE_sec);
                         putkey(mpage, '\0' == *mlink->arch ?                          putkey(mpage, '\0' == *mlink->arch ?
                             "any" : mlink->arch, TYPE_arch);                              any : mlink->arch, TYPE_arch);
                         putkey(mpage, mlink->name, TYPE_Nm);                          putkey(mpage, mlink->name, TYPE_Nm);
                 }                  }
   
Line 1043  mpages_merge(struct mchars *mc, struct mparse *mp)
Line 1061  mpages_merge(struct mchars *mc, struct mparse *mp)
                 else                  else
                         parse_cat(mpage);                          parse_cat(mpage);
   
                 dbindex(mpage, mc);                  dbadd(mpage, mc);
                 ohash_delete(&strings);                  ohash_delete(&strings);
                 mpage = ohash_next(&mpages, &pslot);                  mpage = ohash_next(&mpages, &pslot);
         }          }
   
           if (0 == nodb)
                   SQL_EXEC("END TRANSACTION");
 }  }
   
 static void  static void
Line 1165  parse_cat(struct mpage *mpage)
Line 1186  parse_cat(struct mpage *mpage)
  * Put a type/word pair into the word database for this particular file.   * Put a type/word pair into the word database for this particular file.
  */   */
 static void  static void
 putkey(const struct mpage *mpage, const char *value, uint64_t type)  putkey(const struct mpage *mpage, char *value, uint64_t type)
 {  {
           char     *cp;
   
         assert(NULL != value);          assert(NULL != value);
           if (TYPE_arch == type)
                   for (cp = value; *cp; cp++)
                           if (isupper((unsigned char)*cp))
                                   *cp = _tolower((unsigned char)*cp);
         putkeys(mpage, value, strlen(value), type);          putkeys(mpage, value, strlen(value), type);
 }  }
   
Line 1284  parse_man(struct mpage *mpage, const struct man_node *
Line 1310  parse_man(struct mpage *mpage, const struct man_node *
                                 byte = start[sz];                                  byte = start[sz];
                                 start[sz] = '\0';                                  start[sz] = '\0';
   
                                   /*
                                    * Assume a stray trailing comma in the
                                    * name list if a name begins with a dash.
                                    */
   
                                   if ('-' == start[0] ||
                                       ('\\' == start[0] && '-' == start[1]))
                                           break;
   
                                 putkey(mpage, start, TYPE_Nm);                                  putkey(mpage, start, TYPE_Nm);
   
                                 if (' ' == byte) {                                  if (' ' == byte) {
Line 1413  parse_mdoc_Fd(struct mpage *mpage, const struct mdoc_n
Line 1448  parse_mdoc_Fd(struct mpage *mpage, const struct mdoc_n
 static int  static int
 parse_mdoc_Fn(struct mpage *mpage, const struct mdoc_node *n)  parse_mdoc_Fn(struct mpage *mpage, const struct mdoc_node *n)
 {  {
         const char      *cp;          char    *cp;
   
         if (NULL == (n = n->child) || MDOC_TEXT != n->type)          if (NULL == (n = n->child) || MDOC_TEXT != n->type)
                 return(0);                  return(0);
Line 1535  putkeys(const struct mpage *mpage,
Line 1570  putkeys(const struct mpage *mpage,
         const char *cp, size_t sz, uint64_t v)          const char *cp, size_t sz, uint64_t v)
 {  {
         struct str      *s;          struct str      *s;
         unsigned int     slot;  
         const char      *end;          const char      *end;
           uint64_t         mask;
           unsigned int     slot;
           int              i;
   
         if (0 == sz)          if (0 == sz)
                 return;                  return;
   
           if (verb > 1) {
                   for (i = 0, mask = 1;
                        i < mansearch_keymax;
                        i++, mask <<= 1)
                           if (mask & v)
                                   break;
                   say(mpage->mlinks->file, "Adding key %s=%*s",
                       mansearch_keynames[i], sz, cp);
           }
   
         end = cp + sz;          end = cp + sz;
         slot = ohash_qlookupi(&strings, cp, &end);          slot = ohash_qlookupi(&strings, cp, &end);
         s = ohash_find(&strings, slot);          s = ohash_find(&strings, slot);
Line 1728  render_key(struct mchars *mc, struct str *key)
Line 1775  render_key(struct mchars *mc, struct str *key)
  * Also, handle escape sequences at the last possible moment.   * Also, handle escape sequences at the last possible moment.
  */   */
 static void  static void
 dbindex(const struct mpage *mpage, struct mchars *mc)  dbadd(const struct mpage *mpage, struct mchars *mc)
 {  {
         struct mlink    *mlink;          struct mlink    *mlink;
         struct str      *key;          struct str      *key;
         const char      *desc;  
         int64_t          recno;          int64_t          recno;
         size_t           i;          size_t           i;
         unsigned int     slot;          unsigned int     slot;
   
         if (verb)          if (verb)
                 say(mpage->mlinks->file, "Adding to index");                  say(mpage->mlinks->file, "Adding to database");
   
         if (nodb)          if (nodb)
                 return;                  return;
   
         desc = "";  
         if (NULL != mpage->desc && '\0' != *mpage->desc) {  
                 key = ohash_find(&strings,  
                         ohash_qlookup(&strings, mpage->desc));  
                 assert(NULL != key);  
                 if (NULL == key->rendered)  
                         render_key(mc, key);  
                 desc = key->rendered;  
         }  
   
         SQL_EXEC("BEGIN TRANSACTION");  
   
         i = 1;          i = 1;
         SQL_BIND_TEXT(stmts[STMT_INSERT_PAGE], i, desc);  
         SQL_BIND_INT(stmts[STMT_INSERT_PAGE], i, FORM_SRC == mpage->form);          SQL_BIND_INT(stmts[STMT_INSERT_PAGE], i, FORM_SRC == mpage->form);
         SQL_STEP(stmts[STMT_INSERT_PAGE]);          SQL_STEP(stmts[STMT_INSERT_PAGE]);
         recno = sqlite3_last_insert_rowid(db);          recno = sqlite3_last_insert_rowid(db);
Line 1764  dbindex(const struct mpage *mpage, struct mchars *mc)
Line 1797  dbindex(const struct mpage *mpage, struct mchars *mc)
   
         for (mlink = mpage->mlinks; mlink; mlink = mlink->next) {          for (mlink = mpage->mlinks; mlink; mlink = mlink->next) {
                 i = 1;                  i = 1;
                 SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->file);  
                 SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->dsec);                  SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->dsec);
                 SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->arch);                  SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->arch);
                 SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->name);                  SQL_BIND_TEXT(stmts[STMT_INSERT_LINK], i, mlink->name);
Line 1788  dbindex(const struct mpage *mpage, struct mchars *mc)
Line 1820  dbindex(const struct mpage *mpage, struct mchars *mc)
                         free(key->rendered);                          free(key->rendered);
                 free(key);                  free(key);
         }          }
   
         SQL_EXEC("END TRANSACTION");  
 }  }
   
 static void  static void
Line 1800  dbprune(void)
Line 1830  dbprune(void)
         size_t           i;          size_t           i;
         unsigned int     slot;          unsigned int     slot;
   
         if (nodb)          if (0 == nodb)
                 return;                  SQL_EXEC("BEGIN TRANSACTION");
   
         mpage = ohash_first(&mpages, &slot);          for (mpage = ohash_first(&mpages, &slot); NULL != mpage;
         while (NULL != mpage) {               mpage = ohash_next(&mpages, &slot)) {
                 mlink = mpage->mlinks;                  mlink = mpage->mlinks;
                 i = 1;  
                 SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE], i, mlink->file);  
                 SQL_STEP(stmts[STMT_DELETE_PAGE]);  
                 sqlite3_reset(stmts[STMT_DELETE_PAGE]);  
                 if (verb)                  if (verb)
                         say(mlink->file, "Deleted from index");                          say(mlink->file, "Deleting from database");
                 mpage = ohash_next(&mpages, &slot);                  if (nodb)
                           continue;
                   for ( ; NULL != mlink; mlink = mlink->next) {
                           i = 1;
                           SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
                               i, mlink->dsec);
                           SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
                               i, mlink->arch);
                           SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
                               i, mlink->name);
                           SQL_STEP(stmts[STMT_DELETE_PAGE]);
                           sqlite3_reset(stmts[STMT_DELETE_PAGE]);
                   }
         }          }
   
           if (0 == nodb)
                   SQL_EXEC("END TRANSACTION");
 }  }
   
 /*  /*
Line 1893  dbopen(int real)
Line 1934  dbopen(int real)
         }          }
   
         sql = "CREATE TABLE \"mpages\" (\n"          sql = "CREATE TABLE \"mpages\" (\n"
               " \"desc\" TEXT NOT NULL,\n"  
               " \"form\" INTEGER NOT NULL,\n"                " \"form\" INTEGER NOT NULL,\n"
               " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"                " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"
               ");\n"                ");\n"
               "\n"                "\n"
               "CREATE TABLE \"mlinks\" (\n"                "CREATE TABLE \"mlinks\" (\n"
               " \"file\" TEXT NOT NULL,\n"  
               " \"sec\" TEXT NOT NULL,\n"                " \"sec\" TEXT NOT NULL,\n"
               " \"arch\" TEXT NOT NULL,\n"                " \"arch\" TEXT NOT NULL,\n"
               " \"name\" TEXT NOT NULL,\n"                " \"name\" TEXT NOT NULL,\n"
               " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "                " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "
                 "ON DELETE CASCADE,\n"                  "ON DELETE CASCADE\n"
               " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"  
               ");\n"                ");\n"
               "\n"                "\n"
               "CREATE TABLE \"keys\" (\n"                "CREATE TABLE \"keys\" (\n"
               " \"bits\" INTEGER NOT NULL,\n"                " \"bits\" INTEGER NOT NULL,\n"
               " \"key\" TEXT NOT NULL,\n"                " \"key\" TEXT NOT NULL,\n"
               " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "                " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "
                 "ON DELETE CASCADE,\n"                  "ON DELETE CASCADE\n"
               " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"                ");\n";
               ");\n"  
               "\n"  
               "CREATE INDEX \"key_index\" ON keys (key);\n";  
   
         if (SQLITE_OK != sqlite3_exec(db, sql, NULL, NULL, NULL)) {          if (SQLITE_OK != sqlite3_exec(db, sql, NULL, NULL, NULL)) {
                 exitcode = (int)MANDOCLEVEL_SYSERR;                  exitcode = (int)MANDOCLEVEL_SYSERR;
Line 1926  dbopen(int real)
Line 1961  dbopen(int real)
   
 prepare_statements:  prepare_statements:
         SQL_EXEC("PRAGMA foreign_keys = ON");          SQL_EXEC("PRAGMA foreign_keys = ON");
         sql = "DELETE FROM mpages where file=?";          sql = "DELETE FROM mpages WHERE id IN "
                   "(SELECT pageid FROM mlinks WHERE "
                   "sec=? AND arch=? AND name=?)";
         sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_DELETE_PAGE], NULL);          sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_DELETE_PAGE], NULL);
         sql = "INSERT INTO mpages "          sql = "INSERT INTO mpages "
                 "(desc,form) VALUES (?,?)";                  "(form) VALUES (?)";
         sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_INSERT_PAGE], NULL);          sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_INSERT_PAGE], NULL);
         sql = "INSERT INTO mlinks "          sql = "INSERT INTO mlinks "
                 "(file,sec,arch,name,pageid) VALUES (?,?,?,?,?)";                  "(sec,arch,name,pageid) VALUES (?,?,?,?)";
         sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_INSERT_LINK], NULL);          sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_INSERT_LINK], NULL);
         sql = "INSERT INTO keys "          sql = "INSERT INTO keys "
                 "(bits,key,pageid) VALUES (?,?,?)";                  "(bits,key,pageid) VALUES (?,?,?)";

Legend:
Removed from v.1.101  
changed lines
  Added in v.1.113

CVSweb