Home | History | Annotate | Line # | Download | only in makemandb
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(&section_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(&section_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 {&gt;, &lt;, &quot;, &amp;}.
    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], "&lt;", 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], "&gt;", 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], "&quot;", 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], "&amp;", 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