]>
git.cameronkatri.com Git - mandoc.git/blob - apropos_db.c
4a305e59312b134f90954f8779b3e1bdca459622
1 /* $Id: apropos_db.c,v 1.18 2011/12/01 23:55:58 kristaps Exp $ */
3 * Copyright (c) 2011 Kristaps Dzonsons <kristaps@bsd.lv>
4 * Copyright (c) 2011 Ingo Schwarze <schwarze@openbsd.org>
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
34 #include "apropos_db.h"
38 struct res res
; /* resulting record info */
40 * Maintain a binary tree for checking the uniqueness of `rec'
41 * when adding elements to the results array.
42 * Since the results array is dynamic, use offset in the array
43 * instead of a pointer to the structure.
47 int matched
; /* expression is true */
48 int *matches
; /* partial truth evaluations */
52 int regex
; /* is regex? */
53 int index
; /* index in match array */
54 uint64_t mask
; /* type-mask */
55 int and; /* is rhs of logical AND? */
56 char *v
; /* search value */
57 regex_t re
; /* compiled re, if regex */
58 struct expr
*next
; /* next in sequence */
68 struct rec
*node
; /* record array for dir tree */
69 int len
; /* length of record array */
72 static const struct type types
[] = {
116 static DB
*btree_open(void);
117 static int btree_read(const DBT
*, const DBT
*,
118 const struct mchars
*,
119 struct db_val
*, char **);
120 static int expreval(const struct expr
*, int *);
121 static void exprexec(const struct expr
*,
122 const char *, uint64_t, struct rec
*);
123 static int exprmark(const struct expr
*,
124 const char *, uint64_t, int *);
125 static struct expr
*exprexpr(int, char *[], int *, int *, size_t *);
126 static struct expr
*exprterm(char *, int);
127 static DB
*index_open(void);
128 static int index_read(const DBT
*, const DBT
*, int,
129 const struct mchars
*, struct rec
*);
130 static void norm_string(const char *,
131 const struct mchars
*, char **);
132 static size_t norm_utf8(unsigned int, char[7]);
133 static void recfree(struct rec
*);
134 static int single_search(struct rectree
*, const struct opts
*,
135 const struct expr
*, size_t terms
,
136 struct mchars
*, int);
139 * Open the keyword mandoc-db database.
147 memset(&info
, 0, sizeof(BTREEINFO
));
150 db
= dbopen(MANDOC_DB
, O_RDONLY
, 0, DB_BTREE
, &info
);
158 * Read a keyword from the database and normalise it.
159 * Return 0 if the database is insane, else 1.
162 btree_read(const DBT
*k
, const DBT
*v
,
163 const struct mchars
*mc
,
164 struct db_val
*dbv
, char **buf
)
166 const struct db_val
*vp
;
168 /* Are our sizes sane? */
169 if (k
->size
< 2 || sizeof(struct db_val
) != v
->size
)
172 /* Is our string nil-terminated? */
173 if ('\0' != ((const char *)k
->data
)[(int)k
->size
- 1])
177 norm_string((const char *)k
->data
, mc
, buf
);
178 dbv
->rec
= betoh32(vp
->rec
);
179 dbv
->mask
= betoh64(vp
->mask
);
184 * Take a Unicode codepoint and produce its UTF-8 encoding.
185 * This isn't the best way to do this, but it works.
186 * The magic numbers are from the UTF-8 packaging.
187 * They're not as scary as they seem: read the UTF-8 spec for details.
190 norm_utf8(unsigned int cp
, char out
[7])
196 if (cp
<= 0x0000007F) {
199 } else if (cp
<= 0x000007FF) {
201 out
[0] = (cp
>> 6 & 31) | 192;
202 out
[1] = (cp
& 63) | 128;
203 } else if (cp
<= 0x0000FFFF) {
205 out
[0] = (cp
>> 12 & 15) | 224;
206 out
[1] = (cp
>> 6 & 63) | 128;
207 out
[2] = (cp
& 63) | 128;
208 } else if (cp
<= 0x001FFFFF) {
210 out
[0] = (cp
>> 18 & 7) | 240;
211 out
[1] = (cp
>> 12 & 63) | 128;
212 out
[2] = (cp
>> 6 & 63) | 128;
213 out
[3] = (cp
& 63) | 128;
214 } else if (cp
<= 0x03FFFFFF) {
216 out
[0] = (cp
>> 24 & 3) | 248;
217 out
[1] = (cp
>> 18 & 63) | 128;
218 out
[2] = (cp
>> 12 & 63) | 128;
219 out
[3] = (cp
>> 6 & 63) | 128;
220 out
[4] = (cp
& 63) | 128;
221 } else if (cp
<= 0x7FFFFFFF) {
223 out
[0] = (cp
>> 30 & 1) | 252;
224 out
[1] = (cp
>> 24 & 63) | 128;
225 out
[2] = (cp
>> 18 & 63) | 128;
226 out
[3] = (cp
>> 12 & 63) | 128;
227 out
[4] = (cp
>> 6 & 63) | 128;
228 out
[5] = (cp
& 63) | 128;
237 * Normalise strings from the index and database.
238 * These strings are escaped as defined by mandoc_char(7) along with
239 * other goop in mandoc.h (e.g., soft hyphens).
240 * This function normalises these into a nice UTF-8 string.
241 * Returns 0 if the database is fucked.
244 norm_string(const char *val
, const struct mchars
*mc
, char **buf
)
248 const char *seq
, *cpp
;
251 static const char res
[] = { '\\', '\t',
252 ASCII_NBRSP
, ASCII_HYPH
, '\0' };
254 /* Pre-allocate by the length of the input */
256 bsz
= strlen(val
) + 1;
257 *buf
= mandoc_realloc(*buf
, bsz
);
260 while ('\0' != *val
) {
262 * Halt on the first escape sequence.
263 * This also halts on the end of string, in which case
264 * we just copy, fallthrough, and exit the loop.
266 if ((sz
= strcspn(val
, res
)) > 0) {
267 memcpy(&(*buf
)[pos
], val
, sz
);
272 if (ASCII_HYPH
== *val
) {
276 } else if ('\t' == *val
|| ASCII_NBRSP
== *val
) {
280 } else if ('\\' != *val
)
283 /* Read past the slash. */
289 * Parse the escape sequence and see if it's a
290 * predefined character or special character.
293 esc
= mandoc_escape(&val
, &seq
, &len
);
294 if (ESCAPE_ERROR
== esc
)
298 * XXX - this just does UTF-8, but we need to know
299 * beforehand whether we should do text substitution.
303 case (ESCAPE_SPECIAL
):
304 if (0 != (u
= mchars_spec2cp(mc
, seq
, len
)))
312 * If we have a Unicode codepoint, try to convert that
313 * to a UTF-8 byte string.
317 if (0 == (sz
= norm_utf8(u
, utfbuf
)))
320 /* Copy the rendered glyph into the stream. */
325 *buf
= mandoc_realloc(*buf
, bsz
);
327 memcpy(&(*buf
)[pos
], cpp
, sz
);
335 * Open the filename-index mandoc-db database.
336 * Returns NULL if opening failed.
343 db
= dbopen(MANDOC_IDX
, O_RDONLY
, 0, DB_RECNO
, NULL
);
351 * Safely unpack from an index file record into the structure.
352 * Returns 1 if an entry was unpacked, 0 if the database is insane.
355 index_read(const DBT
*key
, const DBT
*val
, int index
,
356 const struct mchars
*mc
, struct rec
*rec
)
361 #define INDEX_BREAD(_dst) \
363 if (NULL == (np = memchr(cp, '\0', left))) \
365 norm_string(cp, mc, &(_dst)); \
366 left -= (np - cp) + 1; \
368 } while (/* CONSTCOND */ 0)
371 cp
= (char *)val
->data
;
373 rec
->res
.rec
= *(recno_t
*)key
->data
;
374 rec
->res
.volume
= index
;
376 INDEX_BREAD(rec
->res
.type
);
377 INDEX_BREAD(rec
->res
.file
);
378 INDEX_BREAD(rec
->res
.cat
);
379 INDEX_BREAD(rec
->res
.title
);
380 INDEX_BREAD(rec
->res
.arch
);
381 INDEX_BREAD(rec
->res
.desc
);
386 * Search mandocdb databases in paths for expression "expr".
387 * Filter out by "opts".
388 * Call "res" with the results, which may be zero.
389 * Return 0 if there was a database error, else return 1.
392 apropos_search(int pathsz
, char **paths
, const struct opts
*opts
,
393 const struct expr
*expr
, size_t terms
, void *arg
,
394 void (*res
)(struct res
*, size_t, void *))
401 memset(&tree
, 0, sizeof(struct rectree
));
407 * Main loop. Change into the directory containing manpage
408 * databases. Run our expession over each database in the set.
411 for (i
= 0; i
< pathsz
; i
++) {
414 if ( ! single_search(&tree
, opts
, expr
, terms
, mc
, i
))
419 * Count matching files, transfer to a "clean" array, then feed
420 * them to the output handler.
423 for (mlen
= i
= 0; i
< tree
.len
; i
++)
424 if (tree
.node
[i
].matched
)
427 ress
= mandoc_malloc(mlen
* sizeof(struct res
));
429 for (mlen
= i
= 0; i
< tree
.len
; i
++)
430 if (tree
.node
[i
].matched
)
431 memcpy(&ress
[mlen
++], &tree
.node
[i
].res
,
434 (*res
)(ress
, mlen
, arg
);
439 for (i
= 0; i
< tree
.len
; i
++)
440 recfree(&tree
.node
[i
]);
448 single_search(struct rectree
*tree
, const struct opts
*opts
,
449 const struct expr
*expr
, size_t terms
,
450 struct mchars
*mc
, int vol
)
467 memset(&r
, 0, sizeof(struct rec
));
469 if (NULL
== (btree
= btree_open()))
472 if (NULL
== (idx
= index_open())) {
473 (*btree
->close
)(btree
);
477 while (0 == (ch
= (*btree
->seq
)(btree
, &key
, &val
, R_NEXT
))) {
478 if ( ! btree_read(&key
, &val
, mc
, &vb
, &buf
))
482 * See if this keyword record matches any of the
483 * expressions we have stored.
485 if ( ! exprmark(expr
, buf
, vb
.mask
, NULL
))
489 * O(log n) scan for prior records. Since a record
490 * number is unbounded, this has decent performance over
491 * a complex hash function.
494 for (leaf
= root
; leaf
>= 0; )
495 if (vb
.rec
> rs
[leaf
].res
.rec
&&
498 else if (vb
.rec
< rs
[leaf
].res
.rec
&&
505 * If we find a record, see if it has already evaluated
506 * to true. If it has, great, just keep going. If not,
507 * try to evaluate it now and continue anyway.
510 if (leaf
>= 0 && rs
[leaf
].res
.rec
== vb
.rec
) {
511 if (0 == rs
[leaf
].matched
)
512 exprexec(expr
, buf
, vb
.mask
, &rs
[leaf
]);
517 * We have a new file to examine.
518 * Extract the manpage's metadata from the index
519 * database, then begin partial evaluation.
523 key
.size
= sizeof(recno_t
);
525 if (0 != (*idx
->get
)(idx
, &key
, &val
, 0))
529 if ( ! index_read(&key
, &val
, vol
, mc
, &r
))
532 /* XXX: this should be elsewhere, I guess? */
534 if (opts
->cat
&& strcasecmp(opts
->cat
, r
.res
.cat
))
536 if (opts
->arch
&& strcasecmp(opts
->arch
, r
.res
.arch
))
539 tree
->node
= rs
= mandoc_realloc
540 (rs
, (tree
->len
+ 1) * sizeof(struct rec
));
542 memcpy(&rs
[tree
->len
], &r
, sizeof(struct rec
));
543 rs
[tree
->len
].matches
=
544 mandoc_calloc(terms
, sizeof(int));
546 exprexec(expr
, buf
, vb
.mask
, &rs
[tree
->len
]);
548 /* Append to our tree. */
551 if (vb
.rec
> rs
[leaf
].res
.rec
)
552 rs
[leaf
].rhs
= tree
->len
;
554 rs
[leaf
].lhs
= tree
->len
;
558 memset(&r
, 0, sizeof(struct rec
));
562 (*btree
->close
)(btree
);
570 recfree(struct rec
*rec
)
576 free(rec
->res
.title
);
584 * Compile a list of straight-up terms.
585 * The arguments are re-written into ~[[:<:]]term[[:>:]], or "term"
586 * surrounded by word boundaries, then pumped through exprterm().
587 * Terms are case-insensitive.
588 * This emulates whatis(1) behaviour.
591 termcomp(int argc
, char *argv
[], size_t *tt
)
595 struct expr
*e
, *next
;
602 for (pos
= argc
- 1; pos
>= 0; pos
--) {
603 sz
= strlen(argv
[pos
]) + 18;
604 buf
= mandoc_realloc(buf
, sz
);
605 strlcpy(buf
, "Nm~[[:<:]]", sz
);
606 strlcat(buf
, argv
[pos
], sz
);
607 strlcat(buf
, "[[:>:]]", sz
);
608 if (NULL
== (next
= exprterm(buf
, 0))) {
623 * Compile a sequence of logical expressions.
624 * See apropos.1 for a grammar of this sequence.
627 exprcomp(int argc
, char *argv
[], size_t *tt
)
635 e
= exprexpr(argc
, argv
, &pos
, &lvl
, tt
);
637 if (0 == lvl
&& pos
>= argc
)
645 * Compile an array of tokens into an expression.
646 * An informal expression grammar is defined in apropos(1).
647 * Return NULL if we fail doing so. All memory will be cleaned up.
648 * Return the root of the expression sequence if alright.
651 exprexpr(int argc
, char *argv
[], int *pos
, int *lvl
, size_t *tt
)
653 struct expr
*e
, *first
, *next
;
658 for ( ; *pos
< argc
; (*pos
)++) {
662 * Close out a subexpression.
665 if (NULL
!= e
&& 0 == strcmp(")", argv
[*pos
])) {
672 * Small note: if we're just starting, don't let "-a"
673 * and "-o" be considered logical operators: they're
674 * just tokens unless pairwise joining, in which case we
675 * record their existence (or assume "OR").
679 if (NULL
!= e
&& 0 == strcmp("-a", argv
[*pos
]))
681 else if (NULL
!= e
&& 0 == strcmp("-o", argv
[*pos
]))
684 if (log
> 0 && ++(*pos
) >= argc
)
688 * Now we parse the term part. This can begin with
689 * "-i", in which case the expression is case
693 if (0 == strcmp("(", argv
[*pos
])) {
696 next
= mandoc_calloc(1, sizeof(struct expr
));
697 next
->subexpr
= exprexpr(argc
, argv
, pos
, lvl
, tt
);
698 if (NULL
== next
->subexpr
) {
702 } else if (0 == strcmp("-i", argv
[*pos
])) {
703 if (++(*pos
) >= argc
)
705 next
= exprterm(argv
[*pos
], 0);
707 next
= exprterm(argv
[*pos
], 1);
712 next
->and = log
== 1;
713 next
->index
= (int)(*tt
)++;
715 /* Append to our chain of expressions. */
733 * Parse a terminal expression with the grammar as defined in
735 * Return NULL if we fail the parse.
738 exprterm(char *buf
, int cs
)
745 memset(&e
, 0, sizeof(struct expr
));
747 /* Choose regex or substring match. */
749 if (NULL
== (e
.v
= strpbrk(buf
, "=~"))) {
753 e
.regex
= '~' == *e
.v
;
757 /* Determine the record types to search for. */
761 while (NULL
!= (key
= strsep(&buf
, ","))) {
763 while (types
[i
].mask
&&
764 strcmp(types
[i
].name
, key
))
766 e
.mask
|= types
[i
].mask
;
770 e
.mask
= TYPE_Nm
| TYPE_Nd
;
773 i
= REG_EXTENDED
| REG_NOSUB
| (cs
? 0 : REG_ICASE
);
774 if (regcomp(&e
.re
, e
.v
, i
))
778 e
.v
= mandoc_strdup(e
.v
);
780 p
= mandoc_calloc(1, sizeof(struct expr
));
781 memcpy(p
, &e
, sizeof(struct expr
));
786 exprfree(struct expr
*p
)
792 exprfree(p
->subexpr
);
803 exprmark(const struct expr
*p
, const char *cp
,
804 uint64_t mask
, int *ms
)
807 for ( ; p
; p
= p
->next
) {
809 if (exprmark(p
->subexpr
, cp
, mask
, ms
))
812 } else if ( ! (mask
& p
->mask
))
816 if (regexec(&p
->re
, cp
, 0, NULL
, 0))
818 } else if (NULL
== strcasestr(cp
, p
->v
))
831 expreval(const struct expr
*p
, int *ms
)
836 * AND has precedence over OR. Analysis is left-right, though
837 * it doesn't matter because there are no side-effects.
838 * Thus, step through pairwise ANDs and accumulate their Boolean
839 * evaluation. If we encounter a single true AND collection or
840 * standalone term, the whole expression is true (by definition
844 for (match
= 0; p
&& ! match
; p
= p
->next
) {
845 /* Evaluate a subexpression, if applicable. */
846 if (p
->subexpr
&& ! ms
[p
->index
])
847 ms
[p
->index
] = expreval(p
->subexpr
, ms
);
849 match
= ms
[p
->index
];
850 for ( ; p
->next
&& p
->next
->and; p
= p
->next
) {
851 /* Evaluate a subexpression, if applicable. */
852 if (p
->next
->subexpr
&& ! ms
[p
->next
->index
])
854 expreval(p
->next
->subexpr
, ms
);
855 match
= match
&& ms
[p
->next
->index
];
863 * First, update the array of terms for which this expression evaluates
865 * Second, logically evaluate all terms over the updated array of truth
867 * If this evaluates to true, mark the expression as satisfied.
870 exprexec(const struct expr
*e
, const char *cp
,
871 uint64_t mask
, struct rec
*r
)
874 assert(0 == r
->matched
);
875 exprmark(e
, cp
, mask
, r
->matches
);
876 r
->matched
= expreval(e
, r
->matches
);