apropos-utils.c revision 1.4.2.2 1 1.4.2.2 yamt /* $NetBSD: apropos-utils.c,v 1.4.2.2 2012/04/17 00:09:49 yamt Exp $ */
2 1.4.2.2 yamt /*-
3 1.4.2.2 yamt * Copyright (c) 2011 Abhinav Upadhyay <er.abhinav.upadhyay (at) gmail.com>
4 1.4.2.2 yamt * All rights reserved.
5 1.4.2.2 yamt *
6 1.4.2.2 yamt * This code was developed as part of Google's Summer of Code 2011 program.
7 1.4.2.2 yamt *
8 1.4.2.2 yamt * Redistribution and use in source and binary forms, with or without
9 1.4.2.2 yamt * modification, are permitted provided that the following conditions
10 1.4.2.2 yamt * are met:
11 1.4.2.2 yamt *
12 1.4.2.2 yamt * 1. Redistributions of source code must retain the above copyright
13 1.4.2.2 yamt * notice, this list of conditions and the following disclaimer.
14 1.4.2.2 yamt * 2. Redistributions in binary form must reproduce the above copyright
15 1.4.2.2 yamt * notice, this list of conditions and the following disclaimer in
16 1.4.2.2 yamt * the documentation and/or other materials provided with the
17 1.4.2.2 yamt * distribution.
18 1.4.2.2 yamt *
19 1.4.2.2 yamt * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 1.4.2.2 yamt * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 1.4.2.2 yamt * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22 1.4.2.2 yamt * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
23 1.4.2.2 yamt * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
24 1.4.2.2 yamt * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
25 1.4.2.2 yamt * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26 1.4.2.2 yamt * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
27 1.4.2.2 yamt * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 1.4.2.2 yamt * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
29 1.4.2.2 yamt * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 1.4.2.2 yamt * SUCH DAMAGE.
31 1.4.2.2 yamt */
32 1.4.2.2 yamt
33 1.4.2.2 yamt #include <sys/cdefs.h>
34 1.4.2.2 yamt __RCSID("$NetBSD: apropos-utils.c,v 1.4.2.2 2012/04/17 00:09:49 yamt Exp $");
35 1.4.2.2 yamt
36 1.4.2.2 yamt #include <sys/stat.h>
37 1.4.2.2 yamt
38 1.4.2.2 yamt #include <assert.h>
39 1.4.2.2 yamt #include <ctype.h>
40 1.4.2.2 yamt #include <err.h>
41 1.4.2.2 yamt #include <math.h>
42 1.4.2.2 yamt #include <stdio.h>
43 1.4.2.2 yamt #include <stdlib.h>
44 1.4.2.2 yamt #include <string.h>
45 1.4.2.2 yamt #include <util.h>
46 1.4.2.2 yamt #include <zlib.h>
47 1.4.2.2 yamt
48 1.4.2.2 yamt #include "apropos-utils.h"
49 1.4.2.2 yamt #include "mandoc.h"
50 1.4.2.2 yamt #include "sqlite3.h"
51 1.4.2.2 yamt
52 1.4.2.2 yamt typedef struct orig_callback_data {
53 1.4.2.2 yamt void *data;
54 1.4.2.2 yamt int (*callback) (void *, const char *, const char *, const char *,
55 1.4.2.2 yamt const char *, size_t);
56 1.4.2.2 yamt } orig_callback_data;
57 1.4.2.2 yamt
58 1.4.2.2 yamt typedef struct inverse_document_frequency {
59 1.4.2.2 yamt double value;
60 1.4.2.2 yamt int status;
61 1.4.2.2 yamt } inverse_document_frequency;
62 1.4.2.2 yamt
63 1.4.2.2 yamt /* weights for individual columns */
64 1.4.2.2 yamt static const double col_weights[] = {
65 1.4.2.2 yamt 2.0, // NAME
66 1.4.2.2 yamt 2.00, // Name-description
67 1.4.2.2 yamt 0.55, // DESCRIPTION
68 1.4.2.2 yamt 0.10, // LIBRARY
69 1.4.2.2 yamt 0.001, //RETURN VALUES
70 1.4.2.2 yamt 0.20, //ENVIRONMENT
71 1.4.2.2 yamt 0.01, //FILES
72 1.4.2.2 yamt 0.001, //EXIT STATUS
73 1.4.2.2 yamt 2.00, //DIAGNOSTICS
74 1.4.2.2 yamt 0.05, //ERRORS
75 1.4.2.2 yamt 0.00, //md5_hash
76 1.4.2.2 yamt 1.00 //machine
77 1.4.2.2 yamt };
78 1.4.2.2 yamt
79 1.4.2.2 yamt /*
80 1.4.2.2 yamt * lower --
81 1.4.2.2 yamt * Converts the string str to lower case
82 1.4.2.2 yamt */
83 1.4.2.2 yamt char *
84 1.4.2.2 yamt lower(char *str)
85 1.4.2.2 yamt {
86 1.4.2.2 yamt assert(str);
87 1.4.2.2 yamt int i = 0;
88 1.4.2.2 yamt char c;
89 1.4.2.2 yamt while (str[i] != '\0') {
90 1.4.2.2 yamt c = tolower((unsigned char) str[i]);
91 1.4.2.2 yamt str[i++] = c;
92 1.4.2.2 yamt }
93 1.4.2.2 yamt return str;
94 1.4.2.2 yamt }
95 1.4.2.2 yamt
96 1.4.2.2 yamt /*
97 1.4.2.2 yamt * concat--
98 1.4.2.2 yamt * Utility function. Concatenates together: dst, a space character and src.
99 1.4.2.2 yamt * dst + " " + src
100 1.4.2.2 yamt */
101 1.4.2.2 yamt void
102 1.4.2.2 yamt concat(char **dst, const char *src)
103 1.4.2.2 yamt {
104 1.4.2.2 yamt concat2(dst, src, strlen(src));
105 1.4.2.2 yamt }
106 1.4.2.2 yamt
107 1.4.2.2 yamt void
108 1.4.2.2 yamt concat2(char **dst, const char *src, size_t srclen)
109 1.4.2.2 yamt {
110 1.4.2.2 yamt size_t total_len, dst_len;
111 1.4.2.2 yamt assert(src != NULL);
112 1.4.2.2 yamt
113 1.4.2.2 yamt /* If destination buffer dst is NULL, then simply strdup the source buffer */
114 1.4.2.2 yamt if (*dst == NULL) {
115 1.4.2.2 yamt *dst = estrdup(src);
116 1.4.2.2 yamt return;
117 1.4.2.2 yamt }
118 1.4.2.2 yamt
119 1.4.2.2 yamt dst_len = strlen(*dst);
120 1.4.2.2 yamt /*
121 1.4.2.2 yamt * NUL Byte and separator space
122 1.4.2.2 yamt */
123 1.4.2.2 yamt total_len = dst_len + srclen + 2;
124 1.4.2.2 yamt
125 1.4.2.2 yamt *dst = erealloc(*dst, total_len);
126 1.4.2.2 yamt
127 1.4.2.2 yamt /* Append a space at the end of dst */
128 1.4.2.2 yamt (*dst)[dst_len++] = ' ';
129 1.4.2.2 yamt
130 1.4.2.2 yamt /* Now, copy src at the end of dst */
131 1.4.2.2 yamt memcpy(*dst + dst_len, src, srclen + 1);
132 1.4.2.2 yamt }
133 1.4.2.2 yamt
134 1.4.2.2 yamt void
135 1.4.2.2 yamt close_db(sqlite3 *db)
136 1.4.2.2 yamt {
137 1.4.2.2 yamt sqlite3_close(db);
138 1.4.2.2 yamt sqlite3_shutdown();
139 1.4.2.2 yamt }
140 1.4.2.2 yamt
141 1.4.2.2 yamt /*
142 1.4.2.2 yamt * create_db --
143 1.4.2.2 yamt * Creates the database schema.
144 1.4.2.2 yamt */
145 1.4.2.2 yamt static int
146 1.4.2.2 yamt create_db(sqlite3 *db)
147 1.4.2.2 yamt {
148 1.4.2.2 yamt const char *sqlstr = NULL;
149 1.4.2.2 yamt char *schemasql;
150 1.4.2.2 yamt char *errmsg = NULL;
151 1.4.2.2 yamt
152 1.4.2.2 yamt /*------------------------ Create the tables------------------------------*/
153 1.4.2.2 yamt
154 1.4.2.2 yamt #if NOTYET
155 1.4.2.2 yamt sqlite3_exec(db, "PRAGMA journal_mode = WAL", NULL, NULL, NULL);
156 1.4.2.2 yamt #else
157 1.4.2.2 yamt sqlite3_exec(db, "PRAGMA journal_mode = DELETE", NULL, NULL, NULL);
158 1.4.2.2 yamt #endif
159 1.4.2.2 yamt
160 1.4.2.2 yamt schemasql = sqlite3_mprintf("PRAGMA user_version = %d",
161 1.4.2.2 yamt APROPOS_SCHEMA_VERSION);
162 1.4.2.2 yamt sqlite3_exec(db, schemasql, NULL, NULL, &errmsg);
163 1.4.2.2 yamt if (errmsg != NULL)
164 1.4.2.2 yamt goto out;
165 1.4.2.2 yamt sqlite3_free(schemasql);
166 1.4.2.2 yamt
167 1.4.2.2 yamt sqlstr = "CREATE VIRTUAL TABLE mandb USING fts4(section, name, "
168 1.4.2.2 yamt "name_desc, desc, lib, return_vals, env, files, "
169 1.4.2.2 yamt "exit_status, diagnostics, errors, md5_hash UNIQUE, machine, "
170 1.4.2.2 yamt "compress=zip, uncompress=unzip, tokenize=porter); " //mandb
171 1.4.2.2 yamt "CREATE TABLE IF NOT EXISTS mandb_meta(device, inode, mtime, "
172 1.4.2.2 yamt "file UNIQUE, md5_hash UNIQUE, id INTEGER PRIMARY KEY); "
173 1.4.2.2 yamt //mandb_meta
174 1.4.2.2 yamt "CREATE TABLE IF NOT EXISTS mandb_links(link, target, section, "
175 1.4.2.2 yamt "machine); "; //mandb_links
176 1.4.2.2 yamt
177 1.4.2.2 yamt sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg);
178 1.4.2.2 yamt if (errmsg != NULL)
179 1.4.2.2 yamt goto out;
180 1.4.2.2 yamt
181 1.4.2.2 yamt sqlstr = "CREATE INDEX IF NOT EXISTS index_mandb_links ON mandb_links "
182 1.4.2.2 yamt "(link); "
183 1.4.2.2 yamt "CREATE INDEX IF NOT EXISTS index_mandb_meta_dev ON mandb_meta "
184 1.4.2.2 yamt "(device, inode)";
185 1.4.2.2 yamt sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg);
186 1.4.2.2 yamt if (errmsg != NULL)
187 1.4.2.2 yamt goto out;
188 1.4.2.2 yamt return 0;
189 1.4.2.2 yamt
190 1.4.2.2 yamt out:
191 1.4.2.2 yamt warnx("%s", errmsg);
192 1.4.2.2 yamt free(errmsg);
193 1.4.2.2 yamt sqlite3_close(db);
194 1.4.2.2 yamt sqlite3_shutdown();
195 1.4.2.2 yamt return -1;
196 1.4.2.2 yamt }
197 1.4.2.2 yamt
198 1.4.2.2 yamt /*
199 1.4.2.2 yamt * zip --
200 1.4.2.2 yamt * User defined Sqlite function to compress the FTS table
201 1.4.2.2 yamt */
202 1.4.2.2 yamt static void
203 1.4.2.2 yamt zip(sqlite3_context *pctx, int nval, sqlite3_value **apval)
204 1.4.2.2 yamt {
205 1.4.2.2 yamt int nin;
206 1.4.2.2 yamt long int nout;
207 1.4.2.2 yamt const unsigned char * inbuf;
208 1.4.2.2 yamt unsigned char *outbuf;
209 1.4.2.2 yamt
210 1.4.2.2 yamt assert(nval == 1);
211 1.4.2.2 yamt nin = sqlite3_value_bytes(apval[0]);
212 1.4.2.2 yamt inbuf = (const unsigned char *) sqlite3_value_blob(apval[0]);
213 1.4.2.2 yamt nout = nin + 13 + (nin + 999) / 1000;
214 1.4.2.2 yamt outbuf = emalloc(nout);
215 1.4.2.2 yamt compress(outbuf, (unsigned long *) &nout, inbuf, nin);
216 1.4.2.2 yamt sqlite3_result_blob(pctx, outbuf, nout, free);
217 1.4.2.2 yamt }
218 1.4.2.2 yamt
219 1.4.2.2 yamt /*
220 1.4.2.2 yamt * unzip --
221 1.4.2.2 yamt * User defined Sqlite function to uncompress the FTS table.
222 1.4.2.2 yamt */
223 1.4.2.2 yamt static void
224 1.4.2.2 yamt unzip(sqlite3_context *pctx, int nval, sqlite3_value **apval)
225 1.4.2.2 yamt {
226 1.4.2.2 yamt unsigned int rc;
227 1.4.2.2 yamt unsigned char *outbuf;
228 1.4.2.2 yamt z_stream stream;
229 1.4.2.2 yamt
230 1.4.2.2 yamt assert(nval == 1);
231 1.4.2.2 yamt stream.next_in = __UNCONST(sqlite3_value_blob(apval[0]));
232 1.4.2.2 yamt stream.avail_in = sqlite3_value_bytes(apval[0]);
233 1.4.2.2 yamt stream.avail_out = stream.avail_in * 2 + 100;
234 1.4.2.2 yamt stream.next_out = outbuf = emalloc(stream.avail_out);
235 1.4.2.2 yamt stream.zalloc = NULL;
236 1.4.2.2 yamt stream.zfree = NULL;
237 1.4.2.2 yamt
238 1.4.2.2 yamt if (inflateInit(&stream) != Z_OK) {
239 1.4.2.2 yamt free(outbuf);
240 1.4.2.2 yamt return;
241 1.4.2.2 yamt }
242 1.4.2.2 yamt
243 1.4.2.2 yamt while ((rc = inflate(&stream, Z_SYNC_FLUSH)) != Z_STREAM_END) {
244 1.4.2.2 yamt if (rc != Z_OK ||
245 1.4.2.2 yamt (stream.avail_out != 0 && stream.avail_in == 0)) {
246 1.4.2.2 yamt free(outbuf);
247 1.4.2.2 yamt return;
248 1.4.2.2 yamt }
249 1.4.2.2 yamt outbuf = erealloc(outbuf, stream.total_out * 2);
250 1.4.2.2 yamt stream.next_out = outbuf + stream.total_out;
251 1.4.2.2 yamt stream.avail_out = stream.total_out;
252 1.4.2.2 yamt }
253 1.4.2.2 yamt if (inflateEnd(&stream) != Z_OK) {
254 1.4.2.2 yamt free(outbuf);
255 1.4.2.2 yamt return;
256 1.4.2.2 yamt }
257 1.4.2.2 yamt outbuf = erealloc(outbuf, stream.total_out);
258 1.4.2.2 yamt sqlite3_result_text(pctx, (const char *) outbuf, stream.total_out, free);
259 1.4.2.2 yamt }
260 1.4.2.2 yamt
261 1.4.2.2 yamt /* init_db --
262 1.4.2.2 yamt * Prepare the database. Register the compress/uncompress functions and the
263 1.4.2.2 yamt * stopword tokenizer.
264 1.4.2.2 yamt * db_flag specifies the mode in which to open the database. 3 options are
265 1.4.2.2 yamt * available:
266 1.4.2.2 yamt * 1. DB_READONLY: Open in READONLY mode. An error if db does not exist.
267 1.4.2.2 yamt * 2. DB_READWRITE: Open in read-write mode. An error if db does not exist.
268 1.4.2.2 yamt * 3. DB_CREATE: Open in read-write mode. It will try to create the db if
269 1.4.2.2 yamt * it does not exist already.
270 1.4.2.2 yamt * RETURN VALUES:
271 1.4.2.2 yamt * The function will return NULL in case the db does not exist and DB_CREATE
272 1.4.2.2 yamt * was not specified. And in case DB_CREATE was specified and yet NULL is
273 1.4.2.2 yamt * returned, then there was some other error.
274 1.4.2.2 yamt * In normal cases the function should return a handle to the db.
275 1.4.2.2 yamt */
276 1.4.2.2 yamt sqlite3 *
277 1.4.2.2 yamt init_db(int db_flag)
278 1.4.2.2 yamt {
279 1.4.2.2 yamt sqlite3 *db = NULL;
280 1.4.2.2 yamt sqlite3_stmt *stmt;
281 1.4.2.2 yamt struct stat sb;
282 1.4.2.2 yamt int rc;
283 1.4.2.2 yamt int create_db_flag = 0;
284 1.4.2.2 yamt
285 1.4.2.2 yamt /* Check if the database exists or not */
286 1.4.2.2 yamt if (!(stat(DBPATH, &sb) == 0 && S_ISREG(sb.st_mode))) {
287 1.4.2.2 yamt /* Database does not exist, check if DB_CREATE was specified, and set
288 1.4.2.2 yamt * flag to create the database schema
289 1.4.2.2 yamt */
290 1.4.2.2 yamt if (db_flag != (MANDB_CREATE)) {
291 1.4.2.2 yamt warnx("Missing apropos database. "
292 1.4.2.2 yamt "Please run makemandb to create it.");
293 1.4.2.2 yamt return NULL;
294 1.4.2.2 yamt }
295 1.4.2.2 yamt create_db_flag = 1;
296 1.4.2.2 yamt }
297 1.4.2.2 yamt
298 1.4.2.2 yamt /* Now initialize the database connection */
299 1.4.2.2 yamt sqlite3_initialize();
300 1.4.2.2 yamt rc = sqlite3_open_v2(DBPATH, &db, db_flag, NULL);
301 1.4.2.2 yamt
302 1.4.2.2 yamt if (rc != SQLITE_OK) {
303 1.4.2.2 yamt warnx("%s", sqlite3_errmsg(db));
304 1.4.2.2 yamt sqlite3_shutdown();
305 1.4.2.2 yamt return NULL;
306 1.4.2.2 yamt }
307 1.4.2.2 yamt
308 1.4.2.2 yamt if (create_db_flag && create_db(db) < 0) {
309 1.4.2.2 yamt warnx("%s", "Unable to create database schema");
310 1.4.2.2 yamt goto error;
311 1.4.2.2 yamt }
312 1.4.2.2 yamt
313 1.4.2.2 yamt rc = sqlite3_prepare_v2(db, "PRAGMA user_version", -1, &stmt, NULL);
314 1.4.2.2 yamt if (rc != SQLITE_OK) {
315 1.4.2.2 yamt warnx("Unable to query schema version: %s",
316 1.4.2.2 yamt sqlite3_errmsg(db));
317 1.4.2.2 yamt goto error;
318 1.4.2.2 yamt }
319 1.4.2.2 yamt if (sqlite3_step(stmt) != SQLITE_ROW) {
320 1.4.2.2 yamt sqlite3_finalize(stmt);
321 1.4.2.2 yamt warnx("Unable to query schema version: %s",
322 1.4.2.2 yamt sqlite3_errmsg(db));
323 1.4.2.2 yamt goto error;
324 1.4.2.2 yamt }
325 1.4.2.2 yamt if (sqlite3_column_int(stmt, 0) != APROPOS_SCHEMA_VERSION) {
326 1.4.2.2 yamt sqlite3_finalize(stmt);
327 1.4.2.2 yamt warnx("Incorrect schema version found. "
328 1.4.2.2 yamt "Please run makemandb -f.");
329 1.4.2.2 yamt goto error;
330 1.4.2.2 yamt }
331 1.4.2.2 yamt sqlite3_finalize(stmt);
332 1.4.2.2 yamt
333 1.4.2.2 yamt sqlite3_extended_result_codes(db, 1);
334 1.4.2.2 yamt
335 1.4.2.2 yamt /* Register the zip and unzip functions for FTS compression */
336 1.4.2.2 yamt rc = sqlite3_create_function(db, "zip", 1, SQLITE_ANY, NULL, zip, NULL, NULL);
337 1.4.2.2 yamt if (rc != SQLITE_OK) {
338 1.4.2.2 yamt warnx("Unable to register function: compress: %s",
339 1.4.2.2 yamt sqlite3_errmsg(db));
340 1.4.2.2 yamt goto error;
341 1.4.2.2 yamt }
342 1.4.2.2 yamt
343 1.4.2.2 yamt rc = sqlite3_create_function(db, "unzip", 1, SQLITE_ANY, NULL,
344 1.4.2.2 yamt unzip, NULL, NULL);
345 1.4.2.2 yamt if (rc != SQLITE_OK) {
346 1.4.2.2 yamt warnx("Unable to register function: uncompress: %s",
347 1.4.2.2 yamt sqlite3_errmsg(db));
348 1.4.2.2 yamt goto error;
349 1.4.2.2 yamt }
350 1.4.2.2 yamt return db;
351 1.4.2.2 yamt error:
352 1.4.2.2 yamt sqlite3_close(db);
353 1.4.2.2 yamt sqlite3_shutdown();
354 1.4.2.2 yamt return NULL;
355 1.4.2.2 yamt }
356 1.4.2.2 yamt
357 1.4.2.2 yamt /*
358 1.4.2.2 yamt * rank_func --
359 1.4.2.2 yamt * Sqlite user defined function for ranking the documents.
360 1.4.2.2 yamt * For each phrase of the query, it computes the tf and idf and adds them over.
361 1.4.2.2 yamt * It computes the final rank, by multiplying tf and idf together.
362 1.4.2.2 yamt * Weight of term t for document d = (term frequency of t in d *
363 1.4.2.2 yamt * inverse document frequency of t)
364 1.4.2.2 yamt *
365 1.4.2.2 yamt * Term Frequency of term t in document d = Number of times t occurs in d /
366 1.4.2.2 yamt * Number of times t appears in all
367 1.4.2.2 yamt * documents
368 1.4.2.2 yamt *
369 1.4.2.2 yamt * Inverse document frequency of t = log(Total number of documents /
370 1.4.2.2 yamt * Number of documents in which t occurs)
371 1.4.2.2 yamt */
372 1.4.2.2 yamt static void
373 1.4.2.2 yamt rank_func(sqlite3_context *pctx, int nval, sqlite3_value **apval)
374 1.4.2.2 yamt {
375 1.4.2.2 yamt inverse_document_frequency *idf = sqlite3_user_data(pctx);
376 1.4.2.2 yamt double tf = 0.0;
377 1.4.2.2 yamt const unsigned int *matchinfo;
378 1.4.2.2 yamt int ncol;
379 1.4.2.2 yamt int nphrase;
380 1.4.2.2 yamt int iphrase;
381 1.4.2.2 yamt int ndoc;
382 1.4.2.2 yamt int doclen = 0;
383 1.4.2.2 yamt const double k = 3.75;
384 1.4.2.2 yamt /* Check that the number of arguments passed to this function is correct. */
385 1.4.2.2 yamt assert(nval == 1);
386 1.4.2.2 yamt
387 1.4.2.2 yamt matchinfo = (const unsigned int *) sqlite3_value_blob(apval[0]);
388 1.4.2.2 yamt nphrase = matchinfo[0];
389 1.4.2.2 yamt ncol = matchinfo[1];
390 1.4.2.2 yamt ndoc = matchinfo[2 + 3 * ncol * nphrase + ncol];
391 1.4.2.2 yamt for (iphrase = 0; iphrase < nphrase; iphrase++) {
392 1.4.2.2 yamt int icol;
393 1.4.2.2 yamt const unsigned int *phraseinfo = &matchinfo[2 + ncol+ iphrase * ncol * 3];
394 1.4.2.2 yamt for(icol = 1; icol < ncol; icol++) {
395 1.4.2.2 yamt
396 1.4.2.2 yamt /* nhitcount: number of times the current phrase occurs in the current
397 1.4.2.2 yamt * column in the current document.
398 1.4.2.2 yamt * nglobalhitcount: number of times current phrase occurs in the current
399 1.4.2.2 yamt * column in all documents.
400 1.4.2.2 yamt * ndocshitcount: number of documents in which the current phrase
401 1.4.2.2 yamt * occurs in the current column at least once.
402 1.4.2.2 yamt */
403 1.4.2.2 yamt int nhitcount = phraseinfo[3 * icol];
404 1.4.2.2 yamt int nglobalhitcount = phraseinfo[3 * icol + 1];
405 1.4.2.2 yamt int ndocshitcount = phraseinfo[3 * icol + 2];
406 1.4.2.2 yamt doclen = matchinfo[2 + icol ];
407 1.4.2.2 yamt double weight = col_weights[icol - 1];
408 1.4.2.2 yamt if (idf->status == 0 && ndocshitcount)
409 1.4.2.2 yamt idf->value += log(((double)ndoc / ndocshitcount))* weight;
410 1.4.2.2 yamt
411 1.4.2.2 yamt /* Dividing the tf by document length to normalize the effect of
412 1.4.2.2 yamt * longer documents.
413 1.4.2.2 yamt */
414 1.4.2.2 yamt if (nglobalhitcount > 0 && nhitcount)
415 1.4.2.2 yamt tf += (((double)nhitcount * weight) / (nglobalhitcount * doclen));
416 1.4.2.2 yamt }
417 1.4.2.2 yamt }
418 1.4.2.2 yamt idf->status = 1;
419 1.4.2.2 yamt
420 1.4.2.2 yamt /* Final score = (tf * idf)/ ( k + tf)
421 1.4.2.2 yamt * Dividing by k+ tf further normalizes the weight leading to better
422 1.4.2.2 yamt * results.
423 1.4.2.2 yamt * The value of k is experimental
424 1.4.2.2 yamt */
425 1.4.2.2 yamt double score = (tf * idf->value/ ( k + tf)) ;
426 1.4.2.2 yamt sqlite3_result_double(pctx, score);
427 1.4.2.2 yamt return;
428 1.4.2.2 yamt }
429 1.4.2.2 yamt
430 1.4.2.2 yamt /*
431 1.4.2.2 yamt * run_query --
432 1.4.2.2 yamt * Performs the searches for the keywords entered by the user.
433 1.4.2.2 yamt * The 2nd param: snippet_args is an array of strings providing values for the
434 1.4.2.2 yamt * last three parameters to the snippet function of sqlite. (Look at the docs).
435 1.4.2.2 yamt * The 3rd param: args contains rest of the search parameters. Look at
436 1.4.2.2 yamt * arpopos-utils.h for the description of individual fields.
437 1.4.2.2 yamt *
438 1.4.2.2 yamt */
439 1.4.2.2 yamt int
440 1.4.2.2 yamt run_query(sqlite3 *db, const char *snippet_args[3], query_args *args)
441 1.4.2.2 yamt {
442 1.4.2.2 yamt const char *default_snippet_args[3];
443 1.4.2.2 yamt char *section_clause = NULL;
444 1.4.2.2 yamt char *limit_clause = NULL;
445 1.4.2.2 yamt char *machine_clause = NULL;
446 1.4.2.2 yamt char *query;
447 1.4.2.2 yamt const char *section;
448 1.4.2.2 yamt char *name;
449 1.4.2.2 yamt const char *name_desc;
450 1.4.2.2 yamt const char *machine;
451 1.4.2.2 yamt const char *snippet;
452 1.4.2.2 yamt const char *name_temp;
453 1.4.2.2 yamt char *slash_ptr;
454 1.4.2.2 yamt char *m = NULL;
455 1.4.2.2 yamt int rc;
456 1.4.2.2 yamt inverse_document_frequency idf = {0, 0};
457 1.4.2.2 yamt sqlite3_stmt *stmt;
458 1.4.2.2 yamt
459 1.4.2.2 yamt if (args->machine)
460 1.4.2.2 yamt easprintf(&machine_clause, "AND machine = \'%s\' ", args->machine);
461 1.4.2.2 yamt
462 1.4.2.2 yamt /* Register the rank function */
463 1.4.2.2 yamt rc = sqlite3_create_function(db, "rank_func", 1, SQLITE_ANY, (void *)&idf,
464 1.4.2.2 yamt rank_func, NULL, NULL);
465 1.4.2.2 yamt if (rc != SQLITE_OK) {
466 1.4.2.2 yamt warnx("Unable to register the ranking function: %s",
467 1.4.2.2 yamt sqlite3_errmsg(db));
468 1.4.2.2 yamt sqlite3_close(db);
469 1.4.2.2 yamt sqlite3_shutdown();
470 1.4.2.2 yamt exit(EXIT_FAILURE);
471 1.4.2.2 yamt }
472 1.4.2.2 yamt
473 1.4.2.2 yamt /* We want to build a query of the form: "select x,y,z from mandb where
474 1.4.2.2 yamt * mandb match :query [AND (section LIKE '1' OR section LIKE '2' OR...)]
475 1.4.2.2 yamt * ORDER BY rank DESC..."
476 1.4.2.2 yamt * NOTES: 1. The portion in square brackets is optional, it will be there
477 1.4.2.2 yamt * only if the user has specified an option on the command line to search in
478 1.4.2.2 yamt * one or more specific sections.
479 1.4.2.2 yamt * 2. I am using LIKE operator because '=' or IN operators do not seem to be
480 1.4.2.2 yamt * working with the compression option enabled.
481 1.4.2.2 yamt */
482 1.4.2.2 yamt
483 1.4.2.2 yamt if (args->sec_nums) {
484 1.4.2.2 yamt char *temp;
485 1.4.2.2 yamt int i;
486 1.4.2.2 yamt
487 1.4.2.2 yamt for (i = 0; i < SECMAX; i++) {
488 1.4.2.2 yamt if (args->sec_nums[i] == 0)
489 1.4.2.2 yamt continue;
490 1.4.2.2 yamt easprintf(&temp, " OR section = \'%d\'", i + 1);
491 1.4.2.2 yamt if (section_clause) {
492 1.4.2.2 yamt concat(§ion_clause, temp);
493 1.4.2.2 yamt free(temp);
494 1.4.2.2 yamt } else {
495 1.4.2.2 yamt section_clause = temp;
496 1.4.2.2 yamt }
497 1.4.2.2 yamt }
498 1.4.2.2 yamt if (section_clause) {
499 1.4.2.2 yamt /*
500 1.4.2.2 yamt * At least one section requested, add glue for query.
501 1.4.2.2 yamt */
502 1.4.2.2 yamt temp = section_clause;
503 1.4.2.2 yamt /* Skip " OR " before first term. */
504 1.4.2.2 yamt easprintf(§ion_clause, " AND (%s)", temp + 4);
505 1.4.2.2 yamt free(temp);
506 1.4.2.2 yamt }
507 1.4.2.2 yamt }
508 1.4.2.2 yamt if (args->nrec >= 0) {
509 1.4.2.2 yamt /* Use the provided number of records and offset */
510 1.4.2.2 yamt easprintf(&limit_clause, " LIMIT %d OFFSET %d",
511 1.4.2.2 yamt args->nrec, args->offset);
512 1.4.2.2 yamt }
513 1.4.2.2 yamt
514 1.4.2.2 yamt if (snippet_args == NULL) {
515 1.4.2.2 yamt default_snippet_args[0] = "";
516 1.4.2.2 yamt default_snippet_args[1] = "";
517 1.4.2.2 yamt default_snippet_args[2] = "...";
518 1.4.2.2 yamt snippet_args = default_snippet_args;
519 1.4.2.2 yamt }
520 1.4.2.2 yamt query = sqlite3_mprintf("SELECT section, name, name_desc, machine,"
521 1.4.2.2 yamt " snippet(mandb, %Q, %Q, %Q, -1, 40 ),"
522 1.4.2.2 yamt " rank_func(matchinfo(mandb, \"pclxn\")) AS rank"
523 1.4.2.2 yamt " FROM mandb"
524 1.4.2.2 yamt " WHERE mandb MATCH %Q %s "
525 1.4.2.2 yamt "%s"
526 1.4.2.2 yamt " ORDER BY rank DESC"
527 1.4.2.2 yamt "%s",
528 1.4.2.2 yamt snippet_args[0], snippet_args[1], snippet_args[2], args->search_str,
529 1.4.2.2 yamt machine_clause ? machine_clause : "",
530 1.4.2.2 yamt section_clause ? section_clause : "",
531 1.4.2.2 yamt limit_clause ? limit_clause : "");
532 1.4.2.2 yamt
533 1.4.2.2 yamt free(machine_clause);
534 1.4.2.2 yamt free(section_clause);
535 1.4.2.2 yamt free(limit_clause);
536 1.4.2.2 yamt
537 1.4.2.2 yamt if (query == NULL) {
538 1.4.2.2 yamt *args->errmsg = estrdup("malloc failed");
539 1.4.2.2 yamt return -1;
540 1.4.2.2 yamt }
541 1.4.2.2 yamt rc = sqlite3_prepare_v2(db, query, -1, &stmt, NULL);
542 1.4.2.2 yamt if (rc == SQLITE_IOERR) {
543 1.4.2.2 yamt warnx("Corrupt database. Please rerun makemandb");
544 1.4.2.2 yamt sqlite3_free(query);
545 1.4.2.2 yamt return -1;
546 1.4.2.2 yamt } else if (rc != SQLITE_OK) {
547 1.4.2.2 yamt warnx("%s", sqlite3_errmsg(db));
548 1.4.2.2 yamt sqlite3_free(query);
549 1.4.2.2 yamt return -1;
550 1.4.2.2 yamt }
551 1.4.2.2 yamt
552 1.4.2.2 yamt while (sqlite3_step(stmt) == SQLITE_ROW) {
553 1.4.2.2 yamt section = (const char *) sqlite3_column_text(stmt, 0);
554 1.4.2.2 yamt name_temp = (const char *) sqlite3_column_text(stmt, 1);
555 1.4.2.2 yamt name_desc = (const char *) sqlite3_column_text(stmt, 2);
556 1.4.2.2 yamt machine = (const char *) sqlite3_column_text(stmt, 3);
557 1.4.2.2 yamt snippet = (const char *) sqlite3_column_text(stmt, 4);
558 1.4.2.2 yamt if ((slash_ptr = strrchr(name_temp, '/')) != NULL)
559 1.4.2.2 yamt name_temp = slash_ptr + 1;
560 1.4.2.2 yamt if (machine && machine[0]) {
561 1.4.2.2 yamt m = estrdup(machine);
562 1.4.2.2 yamt easprintf(&name, "%s/%s", lower(m),
563 1.4.2.2 yamt name_temp);
564 1.4.2.2 yamt free(m);
565 1.4.2.2 yamt } else {
566 1.4.2.2 yamt name = estrdup((const char *) sqlite3_column_text(stmt, 1));
567 1.4.2.2 yamt }
568 1.4.2.2 yamt
569 1.4.2.2 yamt (args->callback)(args->callback_data, section, name, name_desc, snippet,
570 1.4.2.2 yamt strlen(snippet));
571 1.4.2.2 yamt
572 1.4.2.2 yamt free(name);
573 1.4.2.2 yamt }
574 1.4.2.2 yamt
575 1.4.2.2 yamt sqlite3_finalize(stmt);
576 1.4.2.2 yamt sqlite3_free(query);
577 1.4.2.2 yamt return *(args->errmsg) == NULL ? 0 : -1;
578 1.4.2.2 yamt }
579 1.4.2.2 yamt
580 1.4.2.2 yamt /*
581 1.4.2.2 yamt * callback_html --
582 1.4.2.2 yamt * Callback function for run_query_html. It builds the html output and then
583 1.4.2.2 yamt * calls the actual user supplied callback function.
584 1.4.2.2 yamt */
585 1.4.2.2 yamt static int
586 1.4.2.2 yamt callback_html(void *data, const char *section, const char *name,
587 1.4.2.2 yamt const char *name_desc, const char *snippet, size_t snippet_length)
588 1.4.2.2 yamt {
589 1.4.2.2 yamt const char *temp = snippet;
590 1.4.2.2 yamt int i = 0;
591 1.4.2.2 yamt size_t sz = 0;
592 1.4.2.2 yamt int count = 0;
593 1.4.2.2 yamt struct orig_callback_data *orig_data = (struct orig_callback_data *) data;
594 1.4.2.2 yamt int (*callback) (void *, const char *, const char *, const char *,
595 1.4.2.2 yamt const char *, size_t) = orig_data->callback;
596 1.4.2.2 yamt
597 1.4.2.2 yamt /* First scan the snippet to find out the number of occurrences of {'>', '<'
598 1.4.2.2 yamt * '"', '&'}.
599 1.4.2.2 yamt * Then allocate a new buffer with sufficient space to be able to store the
600 1.4.2.2 yamt * quoted versions of the special characters {>, <, ", &}.
601 1.4.2.2 yamt * Copy over the characters from the original snippet to this buffer while
602 1.4.2.2 yamt * replacing the special characters with their quoted versions.
603 1.4.2.2 yamt */
604 1.4.2.2 yamt
605 1.4.2.2 yamt while (*temp) {
606 1.4.2.2 yamt sz = strcspn(temp, "<>\"&\002\003");
607 1.4.2.2 yamt temp += sz + 1;
608 1.4.2.2 yamt count++;
609 1.4.2.2 yamt }
610 1.4.2.2 yamt size_t qsnippet_length = snippet_length + count * 5;
611 1.4.2.2 yamt char *qsnippet = emalloc(qsnippet_length + 1);
612 1.4.2.2 yamt sz = 0;
613 1.4.2.2 yamt while (*snippet) {
614 1.4.2.2 yamt sz = strcspn(snippet, "<>\"&\002\003");
615 1.4.2.2 yamt if (sz) {
616 1.4.2.2 yamt memcpy(&qsnippet[i], snippet, sz);
617 1.4.2.2 yamt snippet += sz;
618 1.4.2.2 yamt i += sz;
619 1.4.2.2 yamt }
620 1.4.2.2 yamt
621 1.4.2.2 yamt switch (*snippet++) {
622 1.4.2.2 yamt case '<':
623 1.4.2.2 yamt memcpy(&qsnippet[i], "<", 4);
624 1.4.2.2 yamt i += 4;
625 1.4.2.2 yamt break;
626 1.4.2.2 yamt case '>':
627 1.4.2.2 yamt memcpy(&qsnippet[i], ">", 4);
628 1.4.2.2 yamt i += 4;
629 1.4.2.2 yamt break;
630 1.4.2.2 yamt case '\"':
631 1.4.2.2 yamt memcpy(&qsnippet[i], """, 6);
632 1.4.2.2 yamt i += 6;
633 1.4.2.2 yamt break;
634 1.4.2.2 yamt case '&':
635 1.4.2.2 yamt /* Don't perform the quoting if this & is part of an mdoc escape
636 1.4.2.2 yamt * sequence, e.g. \&
637 1.4.2.2 yamt */
638 1.4.2.2 yamt if (i && *(snippet - 2) != '\\') {
639 1.4.2.2 yamt memcpy(&qsnippet[i], "&", 5);
640 1.4.2.2 yamt i += 5;
641 1.4.2.2 yamt } else {
642 1.4.2.2 yamt qsnippet[i++] = '&';
643 1.4.2.2 yamt }
644 1.4.2.2 yamt break;
645 1.4.2.2 yamt case '\002':
646 1.4.2.2 yamt memcpy(&qsnippet[i], "<b>", 3);
647 1.4.2.2 yamt i += 3;
648 1.4.2.2 yamt break;
649 1.4.2.2 yamt case '\003':
650 1.4.2.2 yamt memcpy(&qsnippet[i], "</b>", 4);
651 1.4.2.2 yamt i += 4;
652 1.4.2.2 yamt break;
653 1.4.2.2 yamt default:
654 1.4.2.2 yamt break;
655 1.4.2.2 yamt }
656 1.4.2.2 yamt }
657 1.4.2.2 yamt qsnippet[++i] = 0;
658 1.4.2.2 yamt (*callback)(orig_data->data, section, name, name_desc,
659 1.4.2.2 yamt (const char *)qsnippet, qsnippet_length);
660 1.4.2.2 yamt free(qsnippet);
661 1.4.2.2 yamt return 0;
662 1.4.2.2 yamt }
663 1.4.2.2 yamt
664 1.4.2.2 yamt /*
665 1.4.2.2 yamt * run_query_html --
666 1.4.2.2 yamt * Utility function to output query result in HTML format.
667 1.4.2.2 yamt * It internally calls run_query only, but it first passes the output to it's
668 1.4.2.2 yamt * own custom callback function, which preprocess the snippet for quoting
669 1.4.2.2 yamt * inline HTML fragments.
670 1.4.2.2 yamt * After that it delegates the call the actual user supplied callback function.
671 1.4.2.2 yamt */
672 1.4.2.2 yamt int
673 1.4.2.2 yamt run_query_html(sqlite3 *db, query_args *args)
674 1.4.2.2 yamt {
675 1.4.2.2 yamt struct orig_callback_data orig_data;
676 1.4.2.2 yamt orig_data.callback = args->callback;
677 1.4.2.2 yamt orig_data.data = args->callback_data;
678 1.4.2.2 yamt const char *snippet_args[] = {"\002", "\003", "..."};
679 1.4.2.2 yamt args->callback = &callback_html;
680 1.4.2.2 yamt args->callback_data = (void *) &orig_data;
681 1.4.2.2 yamt return run_query(db, snippet_args, args);
682 1.4.2.2 yamt }
683 1.4.2.2 yamt
684 1.4.2.2 yamt /*
685 1.4.2.2 yamt * callback_pager --
686 1.4.2.2 yamt * A callback similar to callback_html. It overstrikes the matching text in
687 1.4.2.2 yamt * the snippet so that it appears emboldened when viewed using a pager like
688 1.4.2.2 yamt * more or less.
689 1.4.2.2 yamt */
690 1.4.2.2 yamt static int
691 1.4.2.2 yamt callback_pager(void *data, const char *section, const char *name,
692 1.4.2.2 yamt const char *name_desc, const char *snippet, size_t snippet_length)
693 1.4.2.2 yamt {
694 1.4.2.2 yamt struct orig_callback_data *orig_data = (struct orig_callback_data *) data;
695 1.4.2.2 yamt char *psnippet;
696 1.4.2.2 yamt const char *temp = snippet;
697 1.4.2.2 yamt int count = 0;
698 1.4.2.2 yamt int i = 0;
699 1.4.2.2 yamt size_t sz = 0;
700 1.4.2.2 yamt size_t psnippet_length;
701 1.4.2.2 yamt
702 1.4.2.2 yamt /* Count the number of bytes of matching text. For each of these bytes we
703 1.4.2.2 yamt * will use 2 extra bytes to overstrike it so that it appears bold when
704 1.4.2.2 yamt * viewed using a pager.
705 1.4.2.2 yamt */
706 1.4.2.2 yamt while (*temp) {
707 1.4.2.2 yamt sz = strcspn(temp, "\002\003");
708 1.4.2.2 yamt temp += sz;
709 1.4.2.2 yamt if (*temp == '\003') {
710 1.4.2.2 yamt count += 2 * (sz);
711 1.4.2.2 yamt }
712 1.4.2.2 yamt temp++;
713 1.4.2.2 yamt }
714 1.4.2.2 yamt
715 1.4.2.2 yamt psnippet_length = snippet_length + count;
716 1.4.2.2 yamt psnippet = emalloc(psnippet_length + 1);
717 1.4.2.2 yamt
718 1.4.2.2 yamt /* Copy the bytes from snippet to psnippet:
719 1.4.2.2 yamt * 1. Copy the bytes before \002 as it is.
720 1.4.2.2 yamt * 2. The bytes after \002 need to be overstriked till we encounter \003.
721 1.4.2.2 yamt * 3. To overstrike a byte 'A' we need to write 'A\bA'
722 1.4.2.2 yamt */
723 1.4.2.2 yamt while (*snippet) {
724 1.4.2.2 yamt sz = strcspn(snippet, "\002");
725 1.4.2.2 yamt memcpy(&psnippet[i], snippet, sz);
726 1.4.2.2 yamt snippet += sz;
727 1.4.2.2 yamt i += sz;
728 1.4.2.2 yamt
729 1.4.2.2 yamt /* Don't change this. Advancing the pointer without reading the byte
730 1.4.2.2 yamt * is causing strange behavior.
731 1.4.2.2 yamt */
732 1.4.2.2 yamt if (*snippet == '\002')
733 1.4.2.2 yamt snippet++;
734 1.4.2.2 yamt while (*snippet && *snippet != '\003') {
735 1.4.2.2 yamt psnippet[i++] = *snippet;
736 1.4.2.2 yamt psnippet[i++] = '\b';
737 1.4.2.2 yamt psnippet[i++] = *snippet++;
738 1.4.2.2 yamt }
739 1.4.2.2 yamt if (*snippet)
740 1.4.2.2 yamt snippet++;
741 1.4.2.2 yamt }
742 1.4.2.2 yamt
743 1.4.2.2 yamt psnippet[i] = 0;
744 1.4.2.2 yamt (orig_data->callback)(orig_data->data, section, name, name_desc, psnippet,
745 1.4.2.2 yamt psnippet_length);
746 1.4.2.2 yamt free(psnippet);
747 1.4.2.2 yamt return 0;
748 1.4.2.2 yamt }
749 1.4.2.2 yamt
750 1.4.2.2 yamt /*
751 1.4.2.2 yamt * run_query_pager --
752 1.4.2.2 yamt * Utility function similar to run_query_html. This function tries to
753 1.4.2.2 yamt * pre-process the result assuming it will be piped to a pager.
754 1.4.2.2 yamt * For this purpose it first calls it's own callback function callback_pager
755 1.4.2.2 yamt * which then delegates the call to the user supplied callback.
756 1.4.2.2 yamt */
757 1.4.2.2 yamt int run_query_pager(sqlite3 *db, query_args *args)
758 1.4.2.2 yamt {
759 1.4.2.2 yamt struct orig_callback_data orig_data;
760 1.4.2.2 yamt orig_data.callback = args->callback;
761 1.4.2.2 yamt orig_data.data = args->callback_data;
762 1.4.2.2 yamt const char *snippet_args[] = {"\002", "\003", "..."};
763 1.4.2.2 yamt args->callback = &callback_pager;
764 1.4.2.2 yamt args->callback_data = (void *) &orig_data;
765 1.4.2.2 yamt return run_query(db, snippet_args, args);
766 1.4.2.2 yamt }
767