makemandb.c revision 1.36 1 1.36 christos /* $NetBSD: makemandb.c,v 1.36 2016/04/13 01:41:18 christos Exp $ */
2 1.1 joerg /*
3 1.1 joerg * Copyright (c) 2011 Abhinav Upadhyay <er.abhinav.upadhyay (at) gmail.com>
4 1.1 joerg * Copyright (c) 2011 Kristaps Dzonsons <kristaps (at) bsd.lv>
5 1.1 joerg *
6 1.1 joerg * Permission to use, copy, modify, and distribute this software for any
7 1.1 joerg * purpose with or without fee is hereby granted, provided that the above
8 1.1 joerg * copyright notice and this permission notice appear in all copies.
9 1.1 joerg *
10 1.1 joerg * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 1.1 joerg * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 1.1 joerg * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 1.1 joerg * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 1.1 joerg * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 1.1 joerg * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 1.1 joerg * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 1.1 joerg */
18 1.1 joerg
19 1.1 joerg #include <sys/cdefs.h>
20 1.36 christos __RCSID("$NetBSD: makemandb.c,v 1.36 2016/04/13 01:41:18 christos Exp $");
21 1.1 joerg
22 1.1 joerg #include <sys/stat.h>
23 1.1 joerg #include <sys/types.h>
24 1.1 joerg
25 1.1 joerg #include <assert.h>
26 1.1 joerg #include <dirent.h>
27 1.1 joerg #include <err.h>
28 1.5 joerg #include <archive.h>
29 1.3 joerg #include <libgen.h>
30 1.1 joerg #include <md5.h>
31 1.1 joerg #include <stdio.h>
32 1.1 joerg #include <stdlib.h>
33 1.1 joerg #include <string.h>
34 1.1 joerg #include <unistd.h>
35 1.1 joerg #include <util.h>
36 1.1 joerg
37 1.1 joerg #include "apropos-utils.h"
38 1.17 christos #include "dist/man.h"
39 1.17 christos #include "dist/mandoc.h"
40 1.17 christos #include "dist/mdoc.h"
41 1.1 joerg
42 1.1 joerg #define BUFLEN 1024
43 1.1 joerg #define MDOC 0 //If the page is of mdoc(7) type
44 1.1 joerg #define MAN 1 //If the page is of man(7) type
45 1.1 joerg
46 1.1 joerg /*
47 1.1 joerg * A data structure for holding section specific data.
48 1.1 joerg */
49 1.1 joerg typedef struct secbuff {
50 1.1 joerg char *data;
51 1.1 joerg size_t buflen; //Total length of buffer allocated initially
52 1.1 joerg size_t offset; // Current offset in the buffer.
53 1.1 joerg } secbuff;
54 1.1 joerg
55 1.1 joerg typedef struct makemandb_flags {
56 1.1 joerg int optimize;
57 1.1 joerg int limit; // limit the indexing to only NAME section
58 1.1 joerg int recreate; // Database was created from scratch
59 1.1 joerg int verbosity; // 0: quiet, 1: default, 2: verbose
60 1.1 joerg } makemandb_flags;
61 1.1 joerg
62 1.1 joerg typedef struct mandb_rec {
63 1.1 joerg /* Fields for mandb table */
64 1.1 joerg char *name; // for storing the name of the man page
65 1.1 joerg char *name_desc; // for storing the one line description (.Nd)
66 1.1 joerg secbuff desc; // for storing the DESCRIPTION section
67 1.1 joerg secbuff lib; // for the LIBRARY section
68 1.1 joerg secbuff return_vals; // RETURN VALUES
69 1.1 joerg secbuff env; // ENVIRONMENT
70 1.1 joerg secbuff files; // FILES
71 1.1 joerg secbuff exit_status; // EXIT STATUS
72 1.1 joerg secbuff diagnostics; // DIAGNOSTICS
73 1.1 joerg secbuff errors; // ERRORS
74 1.1 joerg char section[2];
75 1.1 joerg
76 1.29 plunky int xr_found; // To track whether a .Xr was seen when parsing a section
77 1.1 joerg
78 1.1 joerg /* Fields for mandb_meta table */
79 1.1 joerg char *md5_hash;
80 1.1 joerg dev_t device;
81 1.1 joerg ino_t inode;
82 1.1 joerg time_t mtime;
83 1.1 joerg
84 1.1 joerg /* Fields for mandb_links table */
85 1.1 joerg char *machine;
86 1.1 joerg char *links; //all the links to a page in a space separated form
87 1.1 joerg char *file_path;
88 1.1 joerg
89 1.1 joerg /* Non-db fields */
90 1.1 joerg int page_type; //Indicates the type of page: mdoc or man
91 1.1 joerg } mandb_rec;
92 1.1 joerg
93 1.1 joerg static void append(secbuff *sbuff, const char *src);
94 1.1 joerg static void init_secbuffs(mandb_rec *);
95 1.1 joerg static void free_secbuffs(mandb_rec *);
96 1.5 joerg static int check_md5(const char *, sqlite3 *, const char *, char **, void *, size_t);
97 1.1 joerg static void cleanup(mandb_rec *);
98 1.1 joerg static void set_section(const struct mdoc *, const struct man *, mandb_rec *);
99 1.1 joerg static void set_machine(const struct mdoc *, mandb_rec *);
100 1.1 joerg static int insert_into_db(sqlite3 *, mandb_rec *);
101 1.5 joerg static void begin_parse(const char *, struct mparse *, mandb_rec *,
102 1.5 joerg const void *, size_t len);
103 1.1 joerg static void pmdoc_node(const struct mdoc_node *, mandb_rec *);
104 1.1 joerg static void pmdoc_Nm(const struct mdoc_node *, mandb_rec *);
105 1.1 joerg static void pmdoc_Nd(const struct mdoc_node *, mandb_rec *);
106 1.1 joerg static void pmdoc_Sh(const struct mdoc_node *, mandb_rec *);
107 1.1 joerg static void pmdoc_Xr(const struct mdoc_node *, mandb_rec *);
108 1.1 joerg static void pmdoc_Pp(const struct mdoc_node *, mandb_rec *);
109 1.18 christos static void pmdoc_macro_handler(const struct mdoc_node *, mandb_rec *,
110 1.1 joerg enum mdoct);
111 1.1 joerg static void pman_node(const struct man_node *n, mandb_rec *);
112 1.1 joerg static void pman_parse_node(const struct man_node *, secbuff *);
113 1.1 joerg static void pman_parse_name(const struct man_node *, mandb_rec *);
114 1.1 joerg static void pman_sh(const struct man_node *, mandb_rec *);
115 1.1 joerg static void pman_block(const struct man_node *, mandb_rec *);
116 1.6 joerg static void traversedir(const char *, const char *, sqlite3 *, struct mparse *);
117 1.1 joerg static void mdoc_parse_section(enum mdoc_sec, const char *, mandb_rec *);
118 1.1 joerg static void man_parse_section(enum man_sec, const struct man_node *, mandb_rec *);
119 1.6 joerg static void build_file_cache(sqlite3 *, const char *, const char *,
120 1.6 joerg struct stat *);
121 1.1 joerg static void update_db(sqlite3 *, struct mparse *, mandb_rec *);
122 1.1 joerg __dead static void usage(void);
123 1.1 joerg static void optimize(sqlite3 *);
124 1.1 joerg static char *parse_escape(const char *);
125 1.23 wiz static void replace_hyph(char *);
126 1.1 joerg static makemandb_flags mflags = { .verbosity = 1 };
127 1.1 joerg
128 1.1 joerg typedef void (*pman_nf)(const struct man_node *n, mandb_rec *);
129 1.1 joerg typedef void (*pmdoc_nf)(const struct mdoc_node *n, mandb_rec *);
130 1.30 christos static const pmdoc_nf mdocs[MDOC_MAX + 1] = {
131 1.1 joerg NULL, /* Ap */
132 1.1 joerg NULL, /* Dd */
133 1.1 joerg NULL, /* Dt */
134 1.1 joerg NULL, /* Os */
135 1.30 christos
136 1.1 joerg pmdoc_Sh, /* Sh */
137 1.1 joerg NULL, /* Ss */
138 1.1 joerg pmdoc_Pp, /* Pp */
139 1.1 joerg NULL, /* D1 */
140 1.30 christos
141 1.1 joerg NULL, /* Dl */
142 1.1 joerg NULL, /* Bd */
143 1.1 joerg NULL, /* Ed */
144 1.1 joerg NULL, /* Bl */
145 1.30 christos
146 1.1 joerg NULL, /* El */
147 1.1 joerg NULL, /* It */
148 1.1 joerg NULL, /* Ad */
149 1.1 joerg NULL, /* An */
150 1.30 christos
151 1.1 joerg NULL, /* Ar */
152 1.1 joerg NULL, /* Cd */
153 1.1 joerg NULL, /* Cm */
154 1.1 joerg NULL, /* Dv */
155 1.30 christos
156 1.1 joerg NULL, /* Er */
157 1.1 joerg NULL, /* Ev */
158 1.1 joerg NULL, /* Ex */
159 1.1 joerg NULL, /* Fa */
160 1.30 christos
161 1.1 joerg NULL, /* Fd */
162 1.1 joerg NULL, /* Fl */
163 1.1 joerg NULL, /* Fn */
164 1.1 joerg NULL, /* Ft */
165 1.30 christos
166 1.1 joerg NULL, /* Ic */
167 1.1 joerg NULL, /* In */
168 1.1 joerg NULL, /* Li */
169 1.1 joerg pmdoc_Nd, /* Nd */
170 1.30 christos
171 1.1 joerg pmdoc_Nm, /* Nm */
172 1.1 joerg NULL, /* Op */
173 1.1 joerg NULL, /* Ot */
174 1.1 joerg NULL, /* Pa */
175 1.30 christos
176 1.1 joerg NULL, /* Rv */
177 1.1 joerg NULL, /* St */
178 1.1 joerg NULL, /* Va */
179 1.1 joerg NULL, /* Vt */
180 1.30 christos
181 1.1 joerg pmdoc_Xr, /* Xr */
182 1.1 joerg NULL, /* %A */
183 1.1 joerg NULL, /* %B */
184 1.1 joerg NULL, /* %D */
185 1.30 christos
186 1.1 joerg NULL, /* %I */
187 1.1 joerg NULL, /* %J */
188 1.1 joerg NULL, /* %N */
189 1.1 joerg NULL, /* %O */
190 1.30 christos
191 1.1 joerg NULL, /* %P */
192 1.1 joerg NULL, /* %R */
193 1.1 joerg NULL, /* %T */
194 1.1 joerg NULL, /* %V */
195 1.30 christos
196 1.1 joerg NULL, /* Ac */
197 1.1 joerg NULL, /* Ao */
198 1.1 joerg NULL, /* Aq */
199 1.1 joerg NULL, /* At */
200 1.30 christos
201 1.1 joerg NULL, /* Bc */
202 1.1 joerg NULL, /* Bf */
203 1.1 joerg NULL, /* Bo */
204 1.1 joerg NULL, /* Bq */
205 1.30 christos
206 1.1 joerg NULL, /* Bsx */
207 1.1 joerg NULL, /* Bx */
208 1.1 joerg NULL, /* Db */
209 1.1 joerg NULL, /* Dc */
210 1.30 christos
211 1.1 joerg NULL, /* Do */
212 1.1 joerg NULL, /* Dq */
213 1.1 joerg NULL, /* Ec */
214 1.1 joerg NULL, /* Ef */
215 1.30 christos
216 1.1 joerg NULL, /* Em */
217 1.1 joerg NULL, /* Eo */
218 1.1 joerg NULL, /* Fx */
219 1.1 joerg NULL, /* Ms */
220 1.30 christos
221 1.1 joerg NULL, /* No */
222 1.1 joerg NULL, /* Ns */
223 1.1 joerg NULL, /* Nx */
224 1.1 joerg NULL, /* Ox */
225 1.30 christos
226 1.1 joerg NULL, /* Pc */
227 1.1 joerg NULL, /* Pf */
228 1.1 joerg NULL, /* Po */
229 1.1 joerg NULL, /* Pq */
230 1.30 christos
231 1.1 joerg NULL, /* Qc */
232 1.1 joerg NULL, /* Ql */
233 1.1 joerg NULL, /* Qo */
234 1.1 joerg NULL, /* Qq */
235 1.30 christos
236 1.1 joerg NULL, /* Re */
237 1.1 joerg NULL, /* Rs */
238 1.1 joerg NULL, /* Sc */
239 1.1 joerg NULL, /* So */
240 1.30 christos
241 1.1 joerg NULL, /* Sq */
242 1.1 joerg NULL, /* Sm */
243 1.1 joerg NULL, /* Sx */
244 1.1 joerg NULL, /* Sy */
245 1.30 christos
246 1.1 joerg NULL, /* Tn */
247 1.1 joerg NULL, /* Ux */
248 1.1 joerg NULL, /* Xc */
249 1.1 joerg NULL, /* Xo */
250 1.30 christos
251 1.1 joerg NULL, /* Fo */
252 1.1 joerg NULL, /* Fc */
253 1.1 joerg NULL, /* Oo */
254 1.1 joerg NULL, /* Oc */
255 1.30 christos
256 1.1 joerg NULL, /* Bk */
257 1.1 joerg NULL, /* Ek */
258 1.1 joerg NULL, /* Bt */
259 1.1 joerg NULL, /* Hf */
260 1.30 christos
261 1.1 joerg NULL, /* Fr */
262 1.1 joerg NULL, /* Ud */
263 1.1 joerg NULL, /* Lb */
264 1.1 joerg NULL, /* Lp */
265 1.30 christos
266 1.1 joerg NULL, /* Lk */
267 1.1 joerg NULL, /* Mt */
268 1.1 joerg NULL, /* Brq */
269 1.1 joerg NULL, /* Bro */
270 1.30 christos
271 1.1 joerg NULL, /* Brc */
272 1.1 joerg NULL, /* %C */
273 1.1 joerg NULL, /* Es */
274 1.1 joerg NULL, /* En */
275 1.30 christos
276 1.1 joerg NULL, /* Dx */
277 1.1 joerg NULL, /* %Q */
278 1.1 joerg NULL, /* br */
279 1.1 joerg NULL, /* sp */
280 1.30 christos
281 1.1 joerg NULL, /* %U */
282 1.1 joerg NULL, /* Ta */
283 1.30 christos NULL, /* ll */
284 1.30 christos NULL, /* text */
285 1.1 joerg };
286 1.1 joerg
287 1.1 joerg static const pman_nf mans[MAN_MAX] = {
288 1.1 joerg NULL, //br
289 1.1 joerg NULL, //TH
290 1.1 joerg pman_sh, //SH
291 1.1 joerg NULL, //SS
292 1.1 joerg NULL, //TP
293 1.1 joerg NULL, //LP
294 1.1 joerg NULL, //PP
295 1.1 joerg NULL, //P
296 1.1 joerg NULL, //IP
297 1.1 joerg NULL, //HP
298 1.1 joerg NULL, //SM
299 1.1 joerg NULL, //SB
300 1.1 joerg NULL, //BI
301 1.1 joerg NULL, //IB
302 1.1 joerg NULL, //BR
303 1.1 joerg NULL, //RB
304 1.1 joerg NULL, //R
305 1.1 joerg pman_block, //B
306 1.1 joerg NULL, //I
307 1.1 joerg NULL, //IR
308 1.1 joerg NULL, //RI
309 1.1 joerg NULL, //sp
310 1.1 joerg NULL, //nf
311 1.1 joerg NULL, //fi
312 1.1 joerg NULL, //RE
313 1.1 joerg NULL, //RS
314 1.1 joerg NULL, //DT
315 1.1 joerg NULL, //UC
316 1.1 joerg NULL, //PD
317 1.1 joerg NULL, //AT
318 1.1 joerg NULL, //in
319 1.1 joerg NULL, //ft
320 1.30 christos NULL, //OP
321 1.30 christos NULL, //EX
322 1.30 christos NULL, //EE
323 1.30 christos NULL, //UR
324 1.30 christos NULL, //UE
325 1.30 christos NULL, //ll
326 1.1 joerg };
327 1.1 joerg
328 1.1 joerg
329 1.1 joerg int
330 1.1 joerg main(int argc, char *argv[])
331 1.1 joerg {
332 1.1 joerg FILE *file;
333 1.30 christos struct mchars *mchars;
334 1.1 joerg const char *sqlstr, *manconf = NULL;
335 1.3 joerg char *line, *command, *parent;
336 1.1 joerg char *errmsg;
337 1.1 joerg int ch;
338 1.1 joerg struct mparse *mp;
339 1.1 joerg sqlite3 *db;
340 1.1 joerg ssize_t len;
341 1.1 joerg size_t linesize;
342 1.1 joerg struct mandb_rec rec;
343 1.1 joerg
344 1.13 wiz while ((ch = getopt(argc, argv, "C:floQqv")) != -1) {
345 1.1 joerg switch (ch) {
346 1.1 joerg case 'C':
347 1.1 joerg manconf = optarg;
348 1.1 joerg break;
349 1.1 joerg case 'f':
350 1.1 joerg mflags.recreate = 1;
351 1.1 joerg break;
352 1.1 joerg case 'l':
353 1.1 joerg mflags.limit = 1;
354 1.1 joerg break;
355 1.1 joerg case 'o':
356 1.1 joerg mflags.optimize = 1;
357 1.1 joerg break;
358 1.13 wiz case 'Q':
359 1.13 wiz mflags.verbosity = 0;
360 1.13 wiz break;
361 1.1 joerg case 'q':
362 1.13 wiz mflags.verbosity = 1;
363 1.1 joerg break;
364 1.1 joerg case 'v':
365 1.1 joerg mflags.verbosity = 2;
366 1.1 joerg break;
367 1.1 joerg default:
368 1.1 joerg usage();
369 1.1 joerg }
370 1.1 joerg }
371 1.1 joerg
372 1.1 joerg memset(&rec, 0, sizeof(rec));
373 1.1 joerg
374 1.1 joerg init_secbuffs(&rec);
375 1.30 christos mchars = mchars_alloc();
376 1.30 christos if (mchars == NULL)
377 1.30 christos errx(EXIT_FAILURE, "Can't allocate mchars");
378 1.30 christos mp = mparse_alloc(0, MANDOCLEVEL_BADARG, NULL, mchars, NULL);
379 1.1 joerg
380 1.15 wiz if (manconf) {
381 1.15 wiz char *arg;
382 1.15 wiz size_t command_len = shquote(manconf, NULL, 0) + 1;
383 1.15 wiz arg = emalloc(command_len);
384 1.15 wiz shquote(manconf, arg, command_len);
385 1.15 wiz easprintf(&command, "man -p -C %s", arg);
386 1.15 wiz free(arg);
387 1.15 wiz } else {
388 1.15 wiz command = estrdup("man -p");
389 1.15 wiz manconf = MANCONF;
390 1.15 wiz }
391 1.15 wiz
392 1.19 christos if (mflags.recreate) {
393 1.19 christos char *dbp = get_dbpath(manconf);
394 1.19 christos /* No error here, it will fail in init_db in the same call */
395 1.19 christos if (dbp != NULL)
396 1.19 christos remove(dbp);
397 1.19 christos }
398 1.15 wiz
399 1.15 wiz if ((db = init_db(MANDB_CREATE, manconf)) == NULL)
400 1.1 joerg exit(EXIT_FAILURE);
401 1.1 joerg
402 1.1 joerg sqlite3_exec(db, "PRAGMA synchronous = 0", NULL, NULL, &errmsg);
403 1.1 joerg if (errmsg != NULL) {
404 1.1 joerg warnx("%s", errmsg);
405 1.1 joerg free(errmsg);
406 1.1 joerg close_db(db);
407 1.1 joerg exit(EXIT_FAILURE);
408 1.1 joerg }
409 1.1 joerg
410 1.1 joerg sqlite3_exec(db, "ATTACH DATABASE \':memory:\' AS metadb", NULL, NULL,
411 1.1 joerg &errmsg);
412 1.1 joerg if (errmsg != NULL) {
413 1.1 joerg warnx("%s", errmsg);
414 1.1 joerg free(errmsg);
415 1.1 joerg close_db(db);
416 1.1 joerg exit(EXIT_FAILURE);
417 1.1 joerg }
418 1.1 joerg
419 1.1 joerg
420 1.1 joerg /* Call man -p to get the list of man page dirs */
421 1.1 joerg if ((file = popen(command, "r")) == NULL) {
422 1.1 joerg close_db(db);
423 1.1 joerg err(EXIT_FAILURE, "fopen failed");
424 1.1 joerg }
425 1.1 joerg free(command);
426 1.1 joerg
427 1.1 joerg /* Begin the transaction for indexing the pages */
428 1.1 joerg sqlite3_exec(db, "BEGIN", NULL, NULL, &errmsg);
429 1.1 joerg if (errmsg != NULL) {
430 1.1 joerg warnx("%s", errmsg);
431 1.1 joerg free(errmsg);
432 1.34 christos close_db(db);
433 1.1 joerg exit(EXIT_FAILURE);
434 1.1 joerg }
435 1.18 christos
436 1.6 joerg sqlstr = "CREATE TABLE metadb.file_cache(device, inode, mtime, parent,"
437 1.6 joerg " file PRIMARY KEY);"
438 1.6 joerg "CREATE UNIQUE INDEX metadb.index_file_cache_dev"
439 1.1 joerg " ON file_cache (device, inode)";
440 1.1 joerg
441 1.1 joerg sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg);
442 1.1 joerg if (errmsg != NULL) {
443 1.1 joerg warnx("%s", errmsg);
444 1.1 joerg free(errmsg);
445 1.1 joerg close_db(db);
446 1.1 joerg exit(EXIT_FAILURE);
447 1.1 joerg }
448 1.1 joerg
449 1.1 joerg if (mflags.verbosity)
450 1.1 joerg printf("Building temporary file cache\n");
451 1.1 joerg line = NULL;
452 1.1 joerg linesize = 0;
453 1.1 joerg while ((len = getline(&line, &linesize, file)) != -1) {
454 1.1 joerg /* Replace the new line character at the end of string with '\0' */
455 1.1 joerg line[len - 1] = '\0';
456 1.3 joerg parent = estrdup(line);
457 1.6 joerg char *pdir = estrdup(dirname(parent));
458 1.3 joerg free(parent);
459 1.1 joerg /* Traverse the man page directories and parse the pages */
460 1.6 joerg traversedir(pdir, line, db, mp);
461 1.6 joerg free(pdir);
462 1.1 joerg }
463 1.1 joerg free(line);
464 1.1 joerg
465 1.1 joerg if (pclose(file) == -1) {
466 1.1 joerg close_db(db);
467 1.1 joerg cleanup(&rec);
468 1.1 joerg free_secbuffs(&rec);
469 1.1 joerg err(EXIT_FAILURE, "pclose error");
470 1.1 joerg }
471 1.1 joerg
472 1.13 wiz if (mflags.verbosity)
473 1.13 wiz printf("Performing index update\n");
474 1.1 joerg update_db(db, mp, &rec);
475 1.1 joerg mparse_free(mp);
476 1.30 christos mchars_free(mchars);
477 1.1 joerg free_secbuffs(&rec);
478 1.1 joerg
479 1.1 joerg /* Commit the transaction */
480 1.1 joerg sqlite3_exec(db, "COMMIT", NULL, NULL, &errmsg);
481 1.1 joerg if (errmsg != NULL) {
482 1.1 joerg warnx("%s", errmsg);
483 1.1 joerg free(errmsg);
484 1.33 christos close_db(db);
485 1.1 joerg exit(EXIT_FAILURE);
486 1.1 joerg }
487 1.1 joerg
488 1.1 joerg if (mflags.optimize)
489 1.1 joerg optimize(db);
490 1.1 joerg
491 1.1 joerg close_db(db);
492 1.1 joerg return 0;
493 1.1 joerg }
494 1.1 joerg
495 1.1 joerg /*
496 1.1 joerg * traversedir --
497 1.1 joerg * Traverses the given directory recursively and passes all the man page files
498 1.1 joerg * in the way to build_file_cache()
499 1.1 joerg */
500 1.1 joerg static void
501 1.6 joerg traversedir(const char *parent, const char *file, sqlite3 *db,
502 1.6 joerg struct mparse *mp)
503 1.1 joerg {
504 1.1 joerg struct stat sb;
505 1.1 joerg struct dirent *dirp;
506 1.1 joerg DIR *dp;
507 1.1 joerg char *buf;
508 1.1 joerg
509 1.1 joerg if (stat(file, &sb) < 0) {
510 1.13 wiz if (mflags.verbosity)
511 1.13 wiz warn("stat failed: %s", file);
512 1.1 joerg return;
513 1.1 joerg }
514 1.18 christos
515 1.1 joerg /* If it is a directory, traverse it recursively */
516 1.1 joerg if (S_ISDIR(sb.st_mode)) {
517 1.1 joerg if ((dp = opendir(file)) == NULL) {
518 1.13 wiz if (mflags.verbosity)
519 1.13 wiz warn("opendir error: %s", file);
520 1.1 joerg return;
521 1.1 joerg }
522 1.18 christos
523 1.1 joerg while ((dirp = readdir(dp)) != NULL) {
524 1.1 joerg /* Avoid . and .. entries in a directory */
525 1.1 joerg if (strncmp(dirp->d_name, ".", 1)) {
526 1.1 joerg easprintf(&buf, "%s/%s", file, dirp->d_name);
527 1.6 joerg traversedir(parent, buf, db, mp);
528 1.1 joerg free(buf);
529 1.1 joerg }
530 1.1 joerg }
531 1.1 joerg closedir(dp);
532 1.1 joerg }
533 1.20 wiz
534 1.20 wiz if (!S_ISREG(sb.st_mode) && !S_ISLNK(sb.st_mode))
535 1.20 wiz return;
536 1.20 wiz
537 1.20 wiz if (sb.st_size == 0) {
538 1.20 wiz if (mflags.verbosity)
539 1.20 wiz warnx("Empty file: %s", file);
540 1.20 wiz return;
541 1.20 wiz }
542 1.20 wiz build_file_cache(db, parent, file, &sb);
543 1.1 joerg }
544 1.1 joerg
545 1.1 joerg /* build_file_cache --
546 1.25 snj * This function generates an md5 hash of the file passed as its 2nd parameter
547 1.1 joerg * and stores it in a temporary table file_cache along with the full file path.
548 1.1 joerg * This is done to support incremental updation of the database.
549 1.1 joerg * The temporary table file_cache is dropped thereafter in the function
550 1.1 joerg * update_db(), once the database has been updated.
551 1.1 joerg */
552 1.1 joerg static void
553 1.6 joerg build_file_cache(sqlite3 *db, const char *parent, const char *file,
554 1.6 joerg struct stat *sb)
555 1.1 joerg {
556 1.1 joerg const char *sqlstr;
557 1.1 joerg sqlite3_stmt *stmt = NULL;
558 1.1 joerg int rc, idx;
559 1.1 joerg assert(file != NULL);
560 1.1 joerg dev_t device_cache = sb->st_dev;
561 1.1 joerg ino_t inode_cache = sb->st_ino;
562 1.1 joerg time_t mtime_cache = sb->st_mtime;
563 1.1 joerg
564 1.1 joerg sqlstr = "INSERT INTO metadb.file_cache VALUES (:device, :inode,"
565 1.6 joerg " :mtime, :parent, :file)";
566 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
567 1.1 joerg if (rc != SQLITE_OK) {
568 1.13 wiz if (mflags.verbosity)
569 1.13 wiz warnx("%s", sqlite3_errmsg(db));
570 1.1 joerg return;
571 1.1 joerg }
572 1.1 joerg
573 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":device");
574 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, device_cache);
575 1.1 joerg if (rc != SQLITE_OK) {
576 1.13 wiz if (mflags.verbosity)
577 1.13 wiz warnx("%s", sqlite3_errmsg(db));
578 1.1 joerg sqlite3_finalize(stmt);
579 1.1 joerg return;
580 1.1 joerg }
581 1.1 joerg
582 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":inode");
583 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, inode_cache);
584 1.1 joerg if (rc != SQLITE_OK) {
585 1.13 wiz if (mflags.verbosity)
586 1.13 wiz warnx("%s", sqlite3_errmsg(db));
587 1.1 joerg sqlite3_finalize(stmt);
588 1.1 joerg return;
589 1.1 joerg }
590 1.1 joerg
591 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":mtime");
592 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, mtime_cache);
593 1.1 joerg if (rc != SQLITE_OK) {
594 1.13 wiz if (mflags.verbosity)
595 1.13 wiz warnx("%s", sqlite3_errmsg(db));
596 1.1 joerg sqlite3_finalize(stmt);
597 1.1 joerg return;
598 1.1 joerg }
599 1.1 joerg
600 1.6 joerg idx = sqlite3_bind_parameter_index(stmt, ":parent");
601 1.6 joerg rc = sqlite3_bind_text(stmt, idx, parent, -1, NULL);
602 1.6 joerg if (rc != SQLITE_OK) {
603 1.13 wiz if (mflags.verbosity)
604 1.13 wiz warnx("%s", sqlite3_errmsg(db));
605 1.6 joerg sqlite3_finalize(stmt);
606 1.6 joerg return;
607 1.6 joerg }
608 1.6 joerg
609 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":file");
610 1.1 joerg rc = sqlite3_bind_text(stmt, idx, file, -1, NULL);
611 1.1 joerg if (rc != SQLITE_OK) {
612 1.13 wiz if (mflags.verbosity)
613 1.13 wiz warnx("%s", sqlite3_errmsg(db));
614 1.1 joerg sqlite3_finalize(stmt);
615 1.1 joerg return;
616 1.1 joerg }
617 1.1 joerg
618 1.1 joerg sqlite3_step(stmt);
619 1.1 joerg sqlite3_finalize(stmt);
620 1.1 joerg }
621 1.1 joerg
622 1.1 joerg static void
623 1.1 joerg update_existing_entry(sqlite3 *db, const char *file, const char *hash,
624 1.1 joerg mandb_rec *rec, int *new_count, int *link_count, int *err_count)
625 1.1 joerg {
626 1.1 joerg int update_count, rc, idx;
627 1.1 joerg const char *inner_sqlstr;
628 1.1 joerg sqlite3_stmt *inner_stmt;
629 1.1 joerg
630 1.1 joerg update_count = sqlite3_total_changes(db);
631 1.1 joerg inner_sqlstr = "UPDATE mandb_meta SET device = :device,"
632 1.1 joerg " inode = :inode, mtime = :mtime WHERE"
633 1.1 joerg " md5_hash = :md5 AND file = :file AND"
634 1.1 joerg " (device <> :device2 OR inode <> "
635 1.1 joerg " :inode2 OR mtime <> :mtime2)";
636 1.1 joerg rc = sqlite3_prepare_v2(db, inner_sqlstr, -1, &inner_stmt, NULL);
637 1.1 joerg if (rc != SQLITE_OK) {
638 1.13 wiz if (mflags.verbosity)
639 1.13 wiz warnx("%s", sqlite3_errmsg(db));
640 1.1 joerg return;
641 1.1 joerg }
642 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":device");
643 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->device);
644 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":inode");
645 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->inode);
646 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":mtime");
647 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->mtime);
648 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":md5");
649 1.1 joerg sqlite3_bind_text(inner_stmt, idx, hash, -1, NULL);
650 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":file");
651 1.1 joerg sqlite3_bind_text(inner_stmt, idx, file, -1, NULL);
652 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":device2");
653 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->device);
654 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":inode2");
655 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->inode);
656 1.1 joerg idx = sqlite3_bind_parameter_index(inner_stmt, ":mtime2");
657 1.1 joerg sqlite3_bind_int64(inner_stmt, idx, rec->mtime);
658 1.1 joerg
659 1.1 joerg rc = sqlite3_step(inner_stmt);
660 1.1 joerg if (rc == SQLITE_DONE) {
661 1.1 joerg /* Check if an update has been performed. */
662 1.1 joerg if (update_count != sqlite3_total_changes(db)) {
663 1.13 wiz if (mflags.verbosity == 2)
664 1.1 joerg printf("Updated %s\n", file);
665 1.1 joerg (*new_count)++;
666 1.1 joerg } else {
667 1.1 joerg /* Otherwise it was a hardlink. */
668 1.1 joerg (*link_count)++;
669 1.1 joerg }
670 1.1 joerg } else {
671 1.13 wiz if (mflags.verbosity == 2)
672 1.13 wiz warnx("Could not update the meta data for %s", file);
673 1.1 joerg (*err_count)++;
674 1.1 joerg }
675 1.1 joerg sqlite3_finalize(inner_stmt);
676 1.1 joerg }
677 1.1 joerg
678 1.5 joerg /* read_and_decompress --
679 1.22 chs * Reads the given file into memory. If it is compressed, decompress
680 1.5 joerg * it before returning to the caller.
681 1.5 joerg */
682 1.5 joerg static int
683 1.22 chs read_and_decompress(const char *file, void **bufp, size_t *len)
684 1.5 joerg {
685 1.5 joerg size_t off;
686 1.5 joerg ssize_t r;
687 1.5 joerg struct archive *a;
688 1.5 joerg struct archive_entry *ae;
689 1.22 chs char *buf;
690 1.5 joerg
691 1.5 joerg if ((a = archive_read_new()) == NULL)
692 1.5 joerg errx(EXIT_FAILURE, "memory allocation failed");
693 1.5 joerg
694 1.22 chs *bufp = NULL;
695 1.5 joerg if (archive_read_support_compression_all(a) != ARCHIVE_OK ||
696 1.5 joerg archive_read_support_format_raw(a) != ARCHIVE_OK ||
697 1.5 joerg archive_read_open_filename(a, file, 65536) != ARCHIVE_OK ||
698 1.5 joerg archive_read_next_header(a, &ae) != ARCHIVE_OK)
699 1.5 joerg goto archive_error;
700 1.5 joerg *len = 65536;
701 1.22 chs buf = emalloc(*len);
702 1.5 joerg off = 0;
703 1.5 joerg for (;;) {
704 1.22 chs r = archive_read_data(a, buf + off, *len - off);
705 1.5 joerg if (r == ARCHIVE_OK) {
706 1.36 christos archive_read_finish(a);
707 1.22 chs *bufp = buf;
708 1.5 joerg *len = off;
709 1.5 joerg return 0;
710 1.5 joerg }
711 1.5 joerg if (r <= 0) {
712 1.22 chs free(buf);
713 1.5 joerg break;
714 1.5 joerg }
715 1.5 joerg off += r;
716 1.5 joerg if (off == *len) {
717 1.5 joerg *len *= 2;
718 1.5 joerg if (*len < off) {
719 1.13 wiz if (mflags.verbosity)
720 1.13 wiz warnx("File too large: %s", file);
721 1.22 chs free(buf);
722 1.36 christos archive_read_finish(a);
723 1.5 joerg return -1;
724 1.5 joerg }
725 1.22 chs buf = erealloc(buf, *len);
726 1.5 joerg }
727 1.5 joerg }
728 1.5 joerg
729 1.5 joerg archive_error:
730 1.5 joerg warnx("Error while reading `%s': %s", file, archive_error_string(a));
731 1.36 christos archive_read_finish(a);
732 1.5 joerg return -1;
733 1.5 joerg }
734 1.5 joerg
735 1.1 joerg /* update_db --
736 1.1 joerg * Does an incremental updation of the database by checking the file_cache.
737 1.1 joerg * It parses and adds the pages which are present in file_cache,
738 1.1 joerg * but not in the database.
739 1.1 joerg * It also removes the pages which are present in the databse,
740 1.1 joerg * but not in the file_cache.
741 1.1 joerg */
742 1.1 joerg static void
743 1.1 joerg update_db(sqlite3 *db, struct mparse *mp, mandb_rec *rec)
744 1.1 joerg {
745 1.1 joerg const char *sqlstr;
746 1.1 joerg sqlite3_stmt *stmt = NULL;
747 1.22 chs char *file;
748 1.22 chs char *parent;
749 1.1 joerg char *errmsg = NULL;
750 1.5 joerg char *md5sum;
751 1.5 joerg void *buf;
752 1.5 joerg size_t buflen;
753 1.22 chs struct sql_row {
754 1.22 chs struct sql_row *next;
755 1.22 chs dev_t device;
756 1.22 chs ino_t inode;
757 1.22 chs time_t mtime;
758 1.22 chs char *parent;
759 1.22 chs char *file;
760 1.22 chs } *rows, *row;
761 1.1 joerg int new_count = 0; /* Counter for newly indexed/updated pages */
762 1.1 joerg int total_count = 0; /* Counter for total number of pages */
763 1.1 joerg int err_count = 0; /* Counter for number of failed pages */
764 1.1 joerg int link_count = 0; /* Counter for number of hard/sym links */
765 1.1 joerg int md5_status;
766 1.1 joerg int rc;
767 1.1 joerg
768 1.6 joerg sqlstr = "SELECT device, inode, mtime, parent, file"
769 1.6 joerg " FROM metadb.file_cache fc"
770 1.6 joerg " WHERE NOT EXISTS(SELECT 1 FROM mandb_meta WHERE"
771 1.6 joerg " device = fc.device AND inode = fc.inode AND "
772 1.6 joerg " mtime = fc.mtime AND file = fc.file)";
773 1.1 joerg
774 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
775 1.1 joerg if (rc != SQLITE_OK) {
776 1.13 wiz if (mflags.verbosity)
777 1.1 joerg warnx("%s", sqlite3_errmsg(db));
778 1.1 joerg close_db(db);
779 1.1 joerg errx(EXIT_FAILURE, "Could not query file cache");
780 1.1 joerg }
781 1.1 joerg
782 1.5 joerg buf = NULL;
783 1.22 chs rows = NULL;
784 1.1 joerg while (sqlite3_step(stmt) == SQLITE_ROW) {
785 1.22 chs row = emalloc(sizeof(struct sql_row));
786 1.22 chs row->device = sqlite3_column_int64(stmt, 0);
787 1.22 chs row->inode = sqlite3_column_int64(stmt, 1);
788 1.22 chs row->mtime = sqlite3_column_int64(stmt, 2);
789 1.22 chs row->parent = estrdup((const char *) sqlite3_column_text(stmt, 3));
790 1.22 chs row->file = estrdup((const char *) sqlite3_column_text(stmt, 4));
791 1.22 chs row->next = rows;
792 1.22 chs rows = row;
793 1.1 joerg total_count++;
794 1.22 chs }
795 1.22 chs sqlite3_finalize(stmt);
796 1.22 chs
797 1.22 chs for ( ; rows != NULL; free(parent), free(file), free(buf)) {
798 1.22 chs row = rows;
799 1.22 chs rows = rows->next;
800 1.22 chs
801 1.22 chs rec->device = row->device;
802 1.22 chs rec->inode = row->inode;
803 1.22 chs rec->mtime = row->mtime;
804 1.22 chs parent = row->parent;
805 1.22 chs file = row->file;
806 1.22 chs free(row);
807 1.22 chs
808 1.5 joerg if (read_and_decompress(file, &buf, &buflen)) {
809 1.5 joerg err_count++;
810 1.5 joerg continue;
811 1.5 joerg }
812 1.5 joerg md5_status = check_md5(file, db, "mandb_meta", &md5sum, buf, buflen);
813 1.5 joerg assert(md5sum != NULL);
814 1.1 joerg if (md5_status == -1) {
815 1.13 wiz if (mflags.verbosity)
816 1.13 wiz warnx("An error occurred in checking md5 value"
817 1.1 joerg " for file %s", file);
818 1.1 joerg err_count++;
819 1.1 joerg continue;
820 1.1 joerg }
821 1.1 joerg
822 1.1 joerg if (md5_status == 0) {
823 1.1 joerg /*
824 1.1 joerg * The MD5 hash is already present in the database,
825 1.1 joerg * so simply update the metadata, ignoring symlinks.
826 1.1 joerg */
827 1.1 joerg struct stat sb;
828 1.1 joerg stat(file, &sb);
829 1.1 joerg if (S_ISLNK(sb.st_mode)) {
830 1.5 joerg free(md5sum);
831 1.1 joerg link_count++;
832 1.1 joerg continue;
833 1.1 joerg }
834 1.5 joerg update_existing_entry(db, file, md5sum, rec,
835 1.1 joerg &new_count, &link_count, &err_count);
836 1.5 joerg free(md5sum);
837 1.1 joerg continue;
838 1.1 joerg }
839 1.1 joerg
840 1.1 joerg if (md5_status == 1) {
841 1.1 joerg /*
842 1.1 joerg * The MD5 hash was not present in the database.
843 1.1 joerg * This means is either a new file or an updated file.
844 1.1 joerg * We should go ahead with parsing.
845 1.1 joerg */
846 1.35 christos if (chdir(parent) == -1) {
847 1.35 christos if (mflags.verbosity)
848 1.35 christos warn("chdir failed for `%s', could "
849 1.35 christos "not index `%s'", parent, file);
850 1.35 christos err_count++;
851 1.35 christos free(md5sum);
852 1.35 christos continue;
853 1.35 christos }
854 1.35 christos
855 1.13 wiz if (mflags.verbosity == 2)
856 1.1 joerg printf("Parsing: %s\n", file);
857 1.5 joerg rec->md5_hash = md5sum;
858 1.1 joerg rec->file_path = estrdup(file);
859 1.1 joerg // file_path is freed by insert_into_db itself.
860 1.5 joerg begin_parse(file, mp, rec, buf, buflen);
861 1.1 joerg if (insert_into_db(db, rec) < 0) {
862 1.13 wiz if (mflags.verbosity)
863 1.35 christos warnx("Error in indexing `%s'", file);
864 1.1 joerg err_count++;
865 1.1 joerg } else {
866 1.1 joerg new_count++;
867 1.1 joerg }
868 1.1 joerg }
869 1.1 joerg }
870 1.18 christos
871 1.13 wiz if (mflags.verbosity == 2) {
872 1.10 uwe printf("Total Number of new or updated pages encountered = %d\n"
873 1.11 wiz "Total number of (hard or symbolic) links found = %d\n"
874 1.1 joerg "Total number of pages that were successfully"
875 1.1 joerg " indexed/updated = %d\n"
876 1.1 joerg "Total number of pages that could not be indexed"
877 1.1 joerg " due to errors = %d\n",
878 1.11 wiz total_count - link_count, link_count, new_count, err_count);
879 1.1 joerg }
880 1.1 joerg
881 1.7 joerg if (mflags.recreate)
882 1.1 joerg return;
883 1.1 joerg
884 1.13 wiz if (mflags.verbosity == 2)
885 1.1 joerg printf("Deleting stale index entries\n");
886 1.1 joerg
887 1.1 joerg sqlstr = "DELETE FROM mandb_meta WHERE file NOT IN"
888 1.1 joerg " (SELECT file FROM metadb.file_cache);"
889 1.9 wiz "DELETE FROM mandb_links WHERE md5_hash NOT IN"
890 1.9 wiz " (SELECT md5_hash from mandb_meta);"
891 1.1 joerg "DROP TABLE metadb.file_cache;"
892 1.1 joerg "DELETE FROM mandb WHERE rowid NOT IN"
893 1.1 joerg " (SELECT id FROM mandb_meta);";
894 1.1 joerg
895 1.1 joerg sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg);
896 1.1 joerg if (errmsg != NULL) {
897 1.1 joerg warnx("Removing old entries failed: %s", errmsg);
898 1.1 joerg warnx("Please rebuild database from scratch with -f.");
899 1.1 joerg free(errmsg);
900 1.1 joerg return;
901 1.1 joerg }
902 1.1 joerg }
903 1.1 joerg
904 1.1 joerg /*
905 1.1 joerg * begin_parse --
906 1.1 joerg * parses the man page using libmandoc
907 1.1 joerg */
908 1.1 joerg static void
909 1.5 joerg begin_parse(const char *file, struct mparse *mp, mandb_rec *rec,
910 1.5 joerg const void *buf, size_t len)
911 1.1 joerg {
912 1.1 joerg struct mdoc *mdoc;
913 1.1 joerg struct man *man;
914 1.1 joerg mparse_reset(mp);
915 1.1 joerg
916 1.1 joerg rec->xr_found = 0;
917 1.1 joerg
918 1.30 christos if (mparse_readmem(mp, buf, len, file) >= MANDOCLEVEL_BADARG) {
919 1.13 wiz /* Printing this warning at verbosity level 2
920 1.13 wiz * because some packages from pkgsrc might trigger several
921 1.13 wiz * of such warnings.
922 1.13 wiz */
923 1.13 wiz if (mflags.verbosity == 2)
924 1.13 wiz warnx("%s: Parse failure", file);
925 1.1 joerg return;
926 1.1 joerg }
927 1.1 joerg
928 1.30 christos mparse_result(mp, &mdoc, &man, NULL);
929 1.1 joerg if (mdoc == NULL && man == NULL) {
930 1.13 wiz if (mflags.verbosity == 2)
931 1.13 wiz warnx("Not a man(7) or mdoc(7) page");
932 1.1 joerg return;
933 1.1 joerg }
934 1.1 joerg
935 1.1 joerg set_machine(mdoc, rec);
936 1.1 joerg set_section(mdoc, man, rec);
937 1.1 joerg if (mdoc) {
938 1.1 joerg rec->page_type = MDOC;
939 1.1 joerg pmdoc_node(mdoc_node(mdoc), rec);
940 1.1 joerg } else {
941 1.1 joerg rec->page_type = MAN;
942 1.1 joerg pman_node(man_node(man), rec);
943 1.1 joerg }
944 1.1 joerg }
945 1.1 joerg
946 1.1 joerg /*
947 1.1 joerg * set_section --
948 1.1 joerg * Extracts the section number and normalizes it to only the numeric part
949 1.1 joerg * (Which should be the first character of the string).
950 1.1 joerg */
951 1.1 joerg static void
952 1.1 joerg set_section(const struct mdoc *md, const struct man *m, mandb_rec *rec)
953 1.1 joerg {
954 1.1 joerg if (md) {
955 1.1 joerg const struct mdoc_meta *md_meta = mdoc_meta(md);
956 1.31 christos if (md_meta->msec == NULL) {
957 1.31 christos rec->section[0] = '?';
958 1.31 christos } else
959 1.31 christos rec->section[0] = md_meta->msec[0];
960 1.1 joerg } else if (m) {
961 1.1 joerg const struct man_meta *m_meta = man_meta(m);
962 1.31 christos if (m_meta->msec == NULL)
963 1.31 christos rec->section[0] = '?';
964 1.31 christos else
965 1.31 christos rec->section[0] = m_meta->msec[0];
966 1.31 christos } else
967 1.31 christos return;
968 1.31 christos
969 1.31 christos if (rec->section[0] == '?' && mflags.verbosity == 2)
970 1.31 christos warnx("%s: Missing section number", rec->file_path);
971 1.1 joerg }
972 1.1 joerg
973 1.1 joerg /*
974 1.1 joerg * get_machine --
975 1.1 joerg * Extracts the machine architecture information if available.
976 1.1 joerg */
977 1.1 joerg static void
978 1.1 joerg set_machine(const struct mdoc *md, mandb_rec *rec)
979 1.1 joerg {
980 1.1 joerg if (md == NULL)
981 1.1 joerg return;
982 1.1 joerg const struct mdoc_meta *md_meta = mdoc_meta(md);
983 1.1 joerg if (md_meta->arch)
984 1.1 joerg rec->machine = estrdup(md_meta->arch);
985 1.1 joerg }
986 1.1 joerg
987 1.1 joerg static void
988 1.1 joerg pmdoc_node(const struct mdoc_node *n, mandb_rec *rec)
989 1.1 joerg {
990 1.1 joerg
991 1.1 joerg if (n == NULL)
992 1.1 joerg return;
993 1.1 joerg
994 1.1 joerg switch (n->type) {
995 1.1 joerg case (MDOC_BODY):
996 1.1 joerg /* FALLTHROUGH */
997 1.1 joerg case (MDOC_TAIL):
998 1.1 joerg /* FALLTHROUGH */
999 1.1 joerg case (MDOC_ELEM):
1000 1.1 joerg if (mdocs[n->tok] == NULL)
1001 1.1 joerg break;
1002 1.1 joerg (*mdocs[n->tok])(n, rec);
1003 1.1 joerg break;
1004 1.1 joerg default:
1005 1.1 joerg break;
1006 1.1 joerg }
1007 1.1 joerg
1008 1.1 joerg pmdoc_node(n->child, rec);
1009 1.1 joerg pmdoc_node(n->next, rec);
1010 1.1 joerg }
1011 1.1 joerg
1012 1.1 joerg /*
1013 1.1 joerg * pmdoc_Nm --
1014 1.1 joerg * Extracts the Name of the manual page from the .Nm macro
1015 1.1 joerg */
1016 1.1 joerg static void
1017 1.1 joerg pmdoc_Nm(const struct mdoc_node *n, mandb_rec *rec)
1018 1.1 joerg {
1019 1.1 joerg if (n->sec != SEC_NAME)
1020 1.1 joerg return;
1021 1.1 joerg
1022 1.1 joerg for (n = n->child; n; n = n->next) {
1023 1.1 joerg if (n->type == MDOC_TEXT) {
1024 1.32 christos char *escaped_name = parse_escape(n->string);
1025 1.32 christos concat(&rec->name, escaped_name);
1026 1.32 christos free(escaped_name);
1027 1.1 joerg }
1028 1.1 joerg }
1029 1.1 joerg }
1030 1.1 joerg
1031 1.1 joerg /*
1032 1.1 joerg * pmdoc_Nd --
1033 1.1 joerg * Extracts the one line description of the man page from the .Nd macro
1034 1.1 joerg */
1035 1.1 joerg static void
1036 1.1 joerg pmdoc_Nd(const struct mdoc_node *n, mandb_rec *rec)
1037 1.1 joerg {
1038 1.1 joerg char *buf = NULL;
1039 1.29 plunky char *name;
1040 1.23 wiz char *nd_text;
1041 1.1 joerg
1042 1.28 joerg if (n == NULL || (n->type != MDOC_TEXT && n->tok == MDOC_MAX))
1043 1.1 joerg return;
1044 1.1 joerg
1045 1.1 joerg if (n->type == MDOC_TEXT) {
1046 1.1 joerg if (rec->xr_found && n->next) {
1047 1.1 joerg /*
1048 1.1 joerg * An Xr macro was seen previously, so parse this
1049 1.29 plunky * and the next node, as "Name(Section)".
1050 1.1 joerg */
1051 1.29 plunky name = n->string;
1052 1.1 joerg n = n->next;
1053 1.29 plunky assert(n->type == MDOC_TEXT);
1054 1.29 plunky easprintf(&buf, "%s(%s)", name, n->string);
1055 1.1 joerg concat(&rec->name_desc, buf);
1056 1.1 joerg free(buf);
1057 1.1 joerg } else {
1058 1.32 christos nd_text = parse_escape(n->string);
1059 1.23 wiz concat(&rec->name_desc, nd_text);
1060 1.23 wiz free(nd_text);
1061 1.1 joerg }
1062 1.1 joerg rec->xr_found = 0;
1063 1.1 joerg } else if (mdocs[n->tok] == pmdoc_Xr) {
1064 1.1 joerg /* Remember that we have encountered an Xr macro */
1065 1.1 joerg rec->xr_found = 1;
1066 1.1 joerg }
1067 1.1 joerg
1068 1.1 joerg if (n->child)
1069 1.1 joerg pmdoc_Nd(n->child, rec);
1070 1.1 joerg
1071 1.1 joerg if(n->next)
1072 1.1 joerg pmdoc_Nd(n->next, rec);
1073 1.1 joerg }
1074 1.1 joerg
1075 1.1 joerg /*
1076 1.1 joerg * pmdoc_macro_handler--
1077 1.1 joerg * This function is a single point of handling all the special macros that we
1078 1.1 joerg * want to handle especially. For example the .Xr macro for properly parsing
1079 1.1 joerg * the referenced page name along with the section number, or the .Pp macro
1080 1.1 joerg * for adding a new line whenever we encounter it.
1081 1.1 joerg */
1082 1.1 joerg static void
1083 1.1 joerg pmdoc_macro_handler(const struct mdoc_node *n, mandb_rec *rec, enum mdoct doct)
1084 1.1 joerg {
1085 1.1 joerg const struct mdoc_node *sn;
1086 1.1 joerg assert(n);
1087 1.1 joerg
1088 1.1 joerg switch (doct) {
1089 1.1 joerg /* Parse the man page references.
1090 1.1 joerg * Basically the .Xr macros are used like:
1091 1.1 joerg * .Xr ls 1
1092 1.1 joerg * and formatted like this:
1093 1.1 joerg * ls(1)
1094 1.1 joerg * Prepare a buffer to format the data like the above example and call
1095 1.1 joerg * pmdoc_parse_section to append it.
1096 1.1 joerg */
1097 1.1 joerg case MDOC_Xr:
1098 1.1 joerg n = n->child;
1099 1.1 joerg while (n->type != MDOC_TEXT && n->next)
1100 1.1 joerg n = n->next;
1101 1.1 joerg
1102 1.1 joerg if (n && n->type != MDOC_TEXT)
1103 1.1 joerg return;
1104 1.1 joerg sn = n;
1105 1.1 joerg if (n->next)
1106 1.1 joerg n = n->next;
1107 1.1 joerg
1108 1.1 joerg while (n->type != MDOC_TEXT && n->next)
1109 1.1 joerg n = n->next;
1110 1.1 joerg
1111 1.1 joerg if (n && n->type == MDOC_TEXT) {
1112 1.27 christos char *buf;
1113 1.27 christos easprintf(&buf, "%s(%s)", sn->string, n->string);
1114 1.1 joerg mdoc_parse_section(n->sec, buf, rec);
1115 1.1 joerg free(buf);
1116 1.1 joerg }
1117 1.1 joerg
1118 1.1 joerg break;
1119 1.1 joerg
1120 1.1 joerg /* Parse the .Pp macro to add a new line */
1121 1.1 joerg case MDOC_Pp:
1122 1.1 joerg if (n->type == MDOC_TEXT)
1123 1.1 joerg mdoc_parse_section(n->sec, "\n", rec);
1124 1.1 joerg break;
1125 1.1 joerg default:
1126 1.1 joerg break;
1127 1.1 joerg }
1128 1.1 joerg
1129 1.1 joerg }
1130 1.1 joerg
1131 1.1 joerg /*
1132 1.1 joerg * pmdoc_Xr, pmdoc_Pp--
1133 1.1 joerg * Empty stubs.
1134 1.1 joerg * The parser calls these functions each time it encounters
1135 1.1 joerg * a .Xr or .Pp macro. We are parsing all the data from
1136 1.1 joerg * the pmdoc_Sh function, so don't do anything here.
1137 1.1 joerg * (See if else blocks in pmdoc_Sh.)
1138 1.1 joerg */
1139 1.1 joerg static void
1140 1.1 joerg pmdoc_Xr(const struct mdoc_node *n, mandb_rec *rec)
1141 1.1 joerg {
1142 1.1 joerg }
1143 1.1 joerg
1144 1.1 joerg static void
1145 1.1 joerg pmdoc_Pp(const struct mdoc_node *n, mandb_rec *rec)
1146 1.1 joerg {
1147 1.1 joerg }
1148 1.1 joerg
1149 1.1 joerg /*
1150 1.1 joerg * pmdoc_Sh --
1151 1.1 joerg * Called when a .Sh macro is encountered and loops through its body, calling
1152 1.1 joerg * mdoc_parse_section to append the data to the section specific buffer.
1153 1.1 joerg * Two special macros which may occur inside the body of Sh are .Nm and .Xr and
1154 1.1 joerg * they need special handling, thus the separate if branches for them.
1155 1.1 joerg */
1156 1.1 joerg static void
1157 1.1 joerg pmdoc_Sh(const struct mdoc_node *n, mandb_rec *rec)
1158 1.1 joerg {
1159 1.28 joerg if (n == NULL || (n->type != MDOC_TEXT && n->tok == MDOC_MAX))
1160 1.1 joerg return;
1161 1.1 joerg int xr_found = 0;
1162 1.1 joerg
1163 1.1 joerg if (n->type == MDOC_TEXT) {
1164 1.1 joerg mdoc_parse_section(n->sec, n->string, rec);
1165 1.1 joerg } else if (mdocs[n->tok] == pmdoc_Nm && rec->name != NULL) {
1166 1.1 joerg /*
1167 1.1 joerg * When encountering a .Nm macro, substitute it
1168 1.1 joerg * with its previously cached value of the argument.
1169 1.1 joerg */
1170 1.1 joerg mdoc_parse_section(n->sec, rec->name, rec);
1171 1.1 joerg } else if (mdocs[n->tok] == pmdoc_Xr) {
1172 1.1 joerg /*
1173 1.1 joerg * When encountering other inline macros,
1174 1.1 joerg * call pmdoc_macro_handler.
1175 1.1 joerg */
1176 1.1 joerg pmdoc_macro_handler(n, rec, MDOC_Xr);
1177 1.1 joerg xr_found = 1;
1178 1.1 joerg } else if (mdocs[n->tok] == pmdoc_Pp) {
1179 1.1 joerg pmdoc_macro_handler(n, rec, MDOC_Pp);
1180 1.1 joerg }
1181 1.1 joerg
1182 1.1 joerg /*
1183 1.1 joerg * If an Xr macro was encountered then the child node has
1184 1.1 joerg * already been explored by pmdoc_macro_handler.
1185 1.1 joerg */
1186 1.1 joerg if (xr_found == 0)
1187 1.1 joerg pmdoc_Sh(n->child, rec);
1188 1.1 joerg pmdoc_Sh(n->next, rec);
1189 1.1 joerg }
1190 1.1 joerg
1191 1.1 joerg /*
1192 1.1 joerg * mdoc_parse_section--
1193 1.1 joerg * Utility function for parsing sections of the mdoc type pages.
1194 1.1 joerg * Takes two params:
1195 1.1 joerg * 1. sec is an enum which indicates the section in which we are present
1196 1.1 joerg * 2. string is the string which we need to append to the secbuff for this
1197 1.1 joerg * particular section.
1198 1.1 joerg * The function appends string to the global section buffer and returns.
1199 1.1 joerg */
1200 1.1 joerg static void
1201 1.1 joerg mdoc_parse_section(enum mdoc_sec sec, const char *string, mandb_rec *rec)
1202 1.1 joerg {
1203 1.1 joerg /*
1204 1.1 joerg * If the user specified the 'l' flag, then parse and store only the
1205 1.1 joerg * NAME section. Ignore the rest.
1206 1.1 joerg */
1207 1.1 joerg if (mflags.limit)
1208 1.1 joerg return;
1209 1.1 joerg
1210 1.1 joerg switch (sec) {
1211 1.1 joerg case SEC_LIBRARY:
1212 1.1 joerg append(&rec->lib, string);
1213 1.1 joerg break;
1214 1.1 joerg case SEC_RETURN_VALUES:
1215 1.1 joerg append(&rec->return_vals, string);
1216 1.1 joerg break;
1217 1.1 joerg case SEC_ENVIRONMENT:
1218 1.1 joerg append(&rec->env, string);
1219 1.1 joerg break;
1220 1.1 joerg case SEC_FILES:
1221 1.1 joerg append(&rec->files, string);
1222 1.1 joerg break;
1223 1.1 joerg case SEC_EXIT_STATUS:
1224 1.1 joerg append(&rec->exit_status, string);
1225 1.1 joerg break;
1226 1.1 joerg case SEC_DIAGNOSTICS:
1227 1.1 joerg append(&rec->diagnostics, string);
1228 1.1 joerg break;
1229 1.1 joerg case SEC_ERRORS:
1230 1.1 joerg append(&rec->errors, string);
1231 1.1 joerg break;
1232 1.1 joerg case SEC_NAME:
1233 1.1 joerg case SEC_SYNOPSIS:
1234 1.1 joerg case SEC_EXAMPLES:
1235 1.1 joerg case SEC_STANDARDS:
1236 1.1 joerg case SEC_HISTORY:
1237 1.1 joerg case SEC_AUTHORS:
1238 1.1 joerg case SEC_BUGS:
1239 1.1 joerg break;
1240 1.1 joerg default:
1241 1.1 joerg append(&rec->desc, string);
1242 1.1 joerg break;
1243 1.1 joerg }
1244 1.1 joerg }
1245 1.1 joerg
1246 1.1 joerg static void
1247 1.1 joerg pman_node(const struct man_node *n, mandb_rec *rec)
1248 1.1 joerg {
1249 1.1 joerg if (n == NULL)
1250 1.1 joerg return;
1251 1.1 joerg
1252 1.1 joerg switch (n->type) {
1253 1.1 joerg case (MAN_BODY):
1254 1.1 joerg /* FALLTHROUGH */
1255 1.1 joerg case (MAN_BLOCK):
1256 1.1 joerg /* FALLTHROUGH */
1257 1.1 joerg case (MAN_ELEM):
1258 1.1 joerg if (mans[n->tok] != NULL)
1259 1.1 joerg (*mans[n->tok])(n, rec);
1260 1.1 joerg break;
1261 1.1 joerg default:
1262 1.1 joerg break;
1263 1.1 joerg }
1264 1.1 joerg
1265 1.1 joerg pman_node(n->child, rec);
1266 1.1 joerg pman_node(n->next, rec);
1267 1.1 joerg }
1268 1.1 joerg
1269 1.18 christos /*
1270 1.1 joerg * pman_parse_name --
1271 1.1 joerg * Parses the NAME section and puts the complete content in the name_desc
1272 1.1 joerg * variable.
1273 1.1 joerg */
1274 1.1 joerg static void
1275 1.1 joerg pman_parse_name(const struct man_node *n, mandb_rec *rec)
1276 1.1 joerg {
1277 1.1 joerg if (n == NULL)
1278 1.1 joerg return;
1279 1.1 joerg
1280 1.1 joerg if (n->type == MAN_TEXT) {
1281 1.1 joerg char *tmp = parse_escape(n->string);
1282 1.1 joerg concat(&rec->name_desc, tmp);
1283 1.1 joerg free(tmp);
1284 1.1 joerg }
1285 1.1 joerg
1286 1.1 joerg if (n->child)
1287 1.1 joerg pman_parse_name(n->child, rec);
1288 1.1 joerg
1289 1.1 joerg if(n->next)
1290 1.1 joerg pman_parse_name(n->next, rec);
1291 1.1 joerg }
1292 1.1 joerg
1293 1.1 joerg /*
1294 1.1 joerg * A stub function to be able to parse the macros like .B embedded inside
1295 1.1 joerg * a section.
1296 1.1 joerg */
1297 1.1 joerg static void
1298 1.1 joerg pman_block(const struct man_node *n, mandb_rec *rec)
1299 1.1 joerg {
1300 1.1 joerg }
1301 1.1 joerg
1302 1.18 christos /*
1303 1.1 joerg * pman_sh --
1304 1.1 joerg * This function does one of the two things:
1305 1.1 joerg * 1. If the present section is NAME, then it will:
1306 1.1 joerg * (a) Extract the name of the page (in case of multiple comma separated
1307 1.1 joerg * names, it will pick up the first one).
1308 1.1 joerg * (b) Build a space spearated list of all the symlinks/hardlinks to
1309 1.1 joerg * this page and store in the buffer 'links'. These are extracted from
1310 1.1 joerg * the comma separated list of names in the NAME section as well.
1311 1.1 joerg * (c) Move on to the one line description section, which is after the list
1312 1.1 joerg * of names in the NAME section.
1313 1.1 joerg * 2. Otherwise, it will check the section name and call the man_parse_section
1314 1.1 joerg * function, passing the enum corresponding that section.
1315 1.1 joerg */
1316 1.1 joerg static void
1317 1.1 joerg pman_sh(const struct man_node *n, mandb_rec *rec)
1318 1.1 joerg {
1319 1.1 joerg static const struct {
1320 1.1 joerg enum man_sec section;
1321 1.1 joerg const char *header;
1322 1.1 joerg } mapping[] = {
1323 1.1 joerg { MANSEC_DESCRIPTION, "DESCRIPTION" },
1324 1.1 joerg { MANSEC_SYNOPSIS, "SYNOPSIS" },
1325 1.1 joerg { MANSEC_LIBRARY, "LIBRARY" },
1326 1.1 joerg { MANSEC_ERRORS, "ERRORS" },
1327 1.1 joerg { MANSEC_FILES, "FILES" },
1328 1.1 joerg { MANSEC_RETURN_VALUES, "RETURN VALUE" },
1329 1.1 joerg { MANSEC_RETURN_VALUES, "RETURN VALUES" },
1330 1.1 joerg { MANSEC_EXIT_STATUS, "EXIT STATUS" },
1331 1.1 joerg { MANSEC_EXAMPLES, "EXAMPLES" },
1332 1.1 joerg { MANSEC_EXAMPLES, "EXAMPLE" },
1333 1.1 joerg { MANSEC_STANDARDS, "STANDARDS" },
1334 1.1 joerg { MANSEC_HISTORY, "HISTORY" },
1335 1.1 joerg { MANSEC_BUGS, "BUGS" },
1336 1.1 joerg { MANSEC_AUTHORS, "AUTHORS" },
1337 1.1 joerg { MANSEC_COPYRIGHT, "COPYRIGHT" },
1338 1.1 joerg };
1339 1.1 joerg const struct man_node *head;
1340 1.1 joerg char *name_desc;
1341 1.32 christos size_t sz;
1342 1.1 joerg size_t i;
1343 1.1 joerg
1344 1.1 joerg if ((head = n->parent->head) == NULL || (head = head->child) == NULL ||
1345 1.1 joerg head->type != MAN_TEXT)
1346 1.1 joerg return;
1347 1.1 joerg
1348 1.1 joerg /*
1349 1.1 joerg * Check if this section should be extracted and
1350 1.1 joerg * where it should be stored. Handled the trival cases first.
1351 1.1 joerg */
1352 1.1 joerg for (i = 0; i < sizeof(mapping) / sizeof(mapping[0]); ++i) {
1353 1.1 joerg if (strcmp(head->string, mapping[i].header) == 0) {
1354 1.1 joerg man_parse_section(mapping[i].section, n, rec);
1355 1.1 joerg return;
1356 1.1 joerg }
1357 1.1 joerg }
1358 1.1 joerg
1359 1.1 joerg if (strcmp(head->string, "NAME") == 0) {
1360 1.1 joerg /*
1361 1.1 joerg * We are in the NAME section.
1362 1.1 joerg * pman_parse_name will put the complete content in name_desc.
1363 1.1 joerg */
1364 1.1 joerg pman_parse_name(n, rec);
1365 1.1 joerg
1366 1.1 joerg name_desc = rec->name_desc;
1367 1.16 christos if (name_desc == NULL)
1368 1.16 christos return;
1369 1.1 joerg
1370 1.1 joerg /* Remove any leading spaces. */
1371 1.1 joerg while (name_desc[0] == ' ')
1372 1.1 joerg name_desc++;
1373 1.18 christos
1374 1.1 joerg /* If the line begins with a "\&", avoid those */
1375 1.1 joerg if (name_desc[0] == '\\' && name_desc[1] == '&')
1376 1.1 joerg name_desc += 2;
1377 1.1 joerg
1378 1.1 joerg /* Now name_desc should be left with a comma-space
1379 1.1 joerg * separated list of names and the one line description
1380 1.1 joerg * of the page:
1381 1.1 joerg * "a, b, c \- sample description"
1382 1.1 joerg * Take out the first name, before the first comma
1383 1.1 joerg * (or space) and store it in rec->name.
1384 1.1 joerg * If the page has aliases then they should be
1385 1.1 joerg * in the form of a comma separated list.
1386 1.1 joerg * Keep looping while there is a comma in name_desc,
1387 1.1 joerg * extract the alias name and store in rec->links.
1388 1.1 joerg * When there are no more commas left, break out.
1389 1.1 joerg */
1390 1.1 joerg int has_alias = 0; // Any more aliases left?
1391 1.1 joerg while (*name_desc) {
1392 1.8 wiz /* Remove any leading spaces or hyphens. */
1393 1.8 wiz if (name_desc[0] == ' ' || name_desc[0] =='-') {
1394 1.1 joerg name_desc++;
1395 1.1 joerg continue;
1396 1.1 joerg }
1397 1.1 joerg sz = strcspn(name_desc, ", ");
1398 1.1 joerg
1399 1.1 joerg /* Extract the first term and store it in rec->name. */
1400 1.1 joerg if (rec->name == NULL) {
1401 1.1 joerg if (name_desc[sz] == ',')
1402 1.1 joerg has_alias = 1;
1403 1.1 joerg name_desc[sz] = 0;
1404 1.1 joerg rec->name = emalloc(sz + 1);
1405 1.1 joerg memcpy(rec->name, name_desc, sz + 1);
1406 1.1 joerg name_desc += sz + 1;
1407 1.1 joerg continue;
1408 1.1 joerg }
1409 1.1 joerg
1410 1.1 joerg /*
1411 1.1 joerg * Once rec->name is set, rest of the names
1412 1.1 joerg * are to be treated as links or aliases.
1413 1.1 joerg */
1414 1.1 joerg if (rec->name && has_alias) {
1415 1.1 joerg if (name_desc[sz] != ',') {
1416 1.1 joerg /* No more commas left -->
1417 1.1 joerg * no more aliases to take out
1418 1.1 joerg */
1419 1.1 joerg has_alias = 0;
1420 1.1 joerg }
1421 1.1 joerg name_desc[sz] = 0;
1422 1.1 joerg concat2(&rec->links, name_desc, sz);
1423 1.1 joerg name_desc += sz + 1;
1424 1.1 joerg continue;
1425 1.1 joerg }
1426 1.1 joerg break;
1427 1.1 joerg }
1428 1.1 joerg
1429 1.1 joerg /* Parse any escape sequences that might be there */
1430 1.1 joerg char *temp = parse_escape(name_desc);
1431 1.1 joerg free(rec->name_desc);
1432 1.1 joerg rec->name_desc = temp;
1433 1.1 joerg temp = parse_escape(rec->name);
1434 1.1 joerg free(rec->name);
1435 1.1 joerg rec->name = temp;
1436 1.1 joerg return;
1437 1.1 joerg }
1438 1.1 joerg
1439 1.1 joerg /* The RETURN VALUE section might be specified in multiple ways */
1440 1.1 joerg if (strcmp(head->string, "RETURN") == 0 &&
1441 1.1 joerg head->next != NULL && head->next->type == MAN_TEXT &&
1442 1.1 joerg (strcmp(head->next->string, "VALUE") == 0 ||
1443 1.1 joerg strcmp(head->next->string, "VALUES") == 0)) {
1444 1.1 joerg man_parse_section(MANSEC_RETURN_VALUES, n, rec);
1445 1.1 joerg return;
1446 1.1 joerg }
1447 1.1 joerg
1448 1.1 joerg /*
1449 1.1 joerg * EXIT STATUS section can also be specified all on one line or on two
1450 1.1 joerg * separate lines.
1451 1.1 joerg */
1452 1.1 joerg if (strcmp(head->string, "EXIT") == 0 &&
1453 1.1 joerg head->next != NULL && head->next->type == MAN_TEXT &&
1454 1.1 joerg strcmp(head->next->string, "STATUS") == 0) {
1455 1.1 joerg man_parse_section(MANSEC_EXIT_STATUS, n, rec);
1456 1.1 joerg return;
1457 1.1 joerg }
1458 1.1 joerg
1459 1.1 joerg /* Store the rest of the content in desc. */
1460 1.1 joerg man_parse_section(MANSEC_NONE, n, rec);
1461 1.1 joerg }
1462 1.1 joerg
1463 1.1 joerg /*
1464 1.1 joerg * pman_parse_node --
1465 1.18 christos * Generic function to iterate through a node. Usually called from
1466 1.1 joerg * man_parse_section to parse a particular section of the man page.
1467 1.1 joerg */
1468 1.1 joerg static void
1469 1.1 joerg pman_parse_node(const struct man_node *n, secbuff *s)
1470 1.1 joerg {
1471 1.1 joerg if (n == NULL)
1472 1.1 joerg return;
1473 1.1 joerg
1474 1.1 joerg if (n->type == MAN_TEXT)
1475 1.1 joerg append(s, n->string);
1476 1.18 christos
1477 1.1 joerg pman_parse_node(n->child, s);
1478 1.1 joerg pman_parse_node(n->next, s);
1479 1.1 joerg }
1480 1.1 joerg
1481 1.1 joerg /*
1482 1.1 joerg * man_parse_section --
1483 1.18 christos * Takes two parameters:
1484 1.1 joerg * sec: Tells which section we are present in
1485 1.1 joerg * n: Is the present node of the AST.
1486 1.1 joerg * Depending on the section, we call pman_parse_node to parse that section and
1487 1.1 joerg * concatenate the content from that section into the buffer for that section.
1488 1.1 joerg */
1489 1.1 joerg static void
1490 1.1 joerg man_parse_section(enum man_sec sec, const struct man_node *n, mandb_rec *rec)
1491 1.1 joerg {
1492 1.1 joerg /*
1493 1.1 joerg * If the user sepecified the 'l' flag then just parse
1494 1.1 joerg * the NAME section, ignore the rest.
1495 1.1 joerg */
1496 1.1 joerg if (mflags.limit)
1497 1.1 joerg return;
1498 1.1 joerg
1499 1.1 joerg switch (sec) {
1500 1.1 joerg case MANSEC_LIBRARY:
1501 1.1 joerg pman_parse_node(n, &rec->lib);
1502 1.1 joerg break;
1503 1.1 joerg case MANSEC_RETURN_VALUES:
1504 1.1 joerg pman_parse_node(n, &rec->return_vals);
1505 1.1 joerg break;
1506 1.1 joerg case MANSEC_ENVIRONMENT:
1507 1.1 joerg pman_parse_node(n, &rec->env);
1508 1.1 joerg break;
1509 1.1 joerg case MANSEC_FILES:
1510 1.1 joerg pman_parse_node(n, &rec->files);
1511 1.1 joerg break;
1512 1.1 joerg case MANSEC_EXIT_STATUS:
1513 1.1 joerg pman_parse_node(n, &rec->exit_status);
1514 1.1 joerg break;
1515 1.1 joerg case MANSEC_DIAGNOSTICS:
1516 1.1 joerg pman_parse_node(n, &rec->diagnostics);
1517 1.1 joerg break;
1518 1.1 joerg case MANSEC_ERRORS:
1519 1.1 joerg pman_parse_node(n, &rec->errors);
1520 1.1 joerg break;
1521 1.1 joerg case MANSEC_NAME:
1522 1.1 joerg case MANSEC_SYNOPSIS:
1523 1.1 joerg case MANSEC_EXAMPLES:
1524 1.1 joerg case MANSEC_STANDARDS:
1525 1.1 joerg case MANSEC_HISTORY:
1526 1.1 joerg case MANSEC_BUGS:
1527 1.1 joerg case MANSEC_AUTHORS:
1528 1.1 joerg case MANSEC_COPYRIGHT:
1529 1.1 joerg break;
1530 1.1 joerg default:
1531 1.1 joerg pman_parse_node(n, &rec->desc);
1532 1.1 joerg break;
1533 1.1 joerg }
1534 1.1 joerg
1535 1.1 joerg }
1536 1.1 joerg
1537 1.1 joerg /*
1538 1.1 joerg * insert_into_db --
1539 1.1 joerg * Inserts the parsed data of the man page in the Sqlite databse.
1540 1.1 joerg * If any of the values is NULL, then we cleanup and return -1 indicating
1541 1.1 joerg * an error.
1542 1.1 joerg * Otherwise, store the data in the database and return 0.
1543 1.1 joerg */
1544 1.1 joerg static int
1545 1.1 joerg insert_into_db(sqlite3 *db, mandb_rec *rec)
1546 1.1 joerg {
1547 1.1 joerg int rc = 0;
1548 1.1 joerg int idx = -1;
1549 1.1 joerg const char *sqlstr = NULL;
1550 1.1 joerg sqlite3_stmt *stmt = NULL;
1551 1.1 joerg char *ln = NULL;
1552 1.1 joerg char *errmsg = NULL;
1553 1.1 joerg long int mandb_rowid;
1554 1.18 christos
1555 1.1 joerg /*
1556 1.1 joerg * At the very minimum we want to make sure that we store
1557 1.1 joerg * the following data:
1558 1.1 joerg * Name, one line description, and the MD5 hash
1559 1.18 christos */
1560 1.1 joerg if (rec->name == NULL || rec->name_desc == NULL ||
1561 1.1 joerg rec->md5_hash == NULL) {
1562 1.1 joerg cleanup(rec);
1563 1.1 joerg return -1;
1564 1.1 joerg }
1565 1.1 joerg
1566 1.1 joerg /* Write null byte at the end of all the sec_buffs */
1567 1.1 joerg rec->desc.data[rec->desc.offset] = 0;
1568 1.1 joerg rec->lib.data[rec->lib.offset] = 0;
1569 1.1 joerg rec->env.data[rec->env.offset] = 0;
1570 1.1 joerg rec->return_vals.data[rec->return_vals.offset] = 0;
1571 1.1 joerg rec->exit_status.data[rec->exit_status.offset] = 0;
1572 1.1 joerg rec->files.data[rec->files.offset] = 0;
1573 1.1 joerg rec->diagnostics.data[rec->diagnostics.offset] = 0;
1574 1.1 joerg rec->errors.data[rec->errors.offset] = 0;
1575 1.1 joerg
1576 1.1 joerg /*
1577 1.1 joerg * In case of a mdoc page: (sorry, no better place to put this code)
1578 1.1 joerg * parse the comma separated list of names of man pages,
1579 1.1 joerg * the first name will be stored in the mandb table, rest will be
1580 1.1 joerg * treated as links and put in the mandb_links table.
1581 1.1 joerg */
1582 1.1 joerg if (rec->page_type == MDOC) {
1583 1.1 joerg char *tmp;
1584 1.1 joerg rec->links = estrdup(rec->name);
1585 1.1 joerg free(rec->name);
1586 1.1 joerg int sz = strcspn(rec->links, " \0");
1587 1.1 joerg rec->name = emalloc(sz + 1);
1588 1.1 joerg memcpy(rec->name, rec->links, sz);
1589 1.1 joerg if(rec->name[sz - 1] == ',')
1590 1.1 joerg rec->name[sz - 1] = 0;
1591 1.1 joerg else
1592 1.1 joerg rec->name[sz] = 0;
1593 1.1 joerg while (rec->links[sz] == ' ')
1594 1.1 joerg ++sz;
1595 1.1 joerg tmp = estrdup(rec->links + sz);
1596 1.1 joerg free(rec->links);
1597 1.1 joerg rec->links = tmp;
1598 1.1 joerg }
1599 1.1 joerg
1600 1.1 joerg /*------------------------ Populate the mandb table---------------------------*/
1601 1.1 joerg sqlstr = "INSERT INTO mandb VALUES (:section, :name, :name_desc, :desc,"
1602 1.1 joerg " :lib, :return_vals, :env, :files, :exit_status,"
1603 1.1 joerg " :diagnostics, :errors, :md5_hash, :machine)";
1604 1.1 joerg
1605 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
1606 1.1 joerg if (rc != SQLITE_OK)
1607 1.1 joerg goto Out;
1608 1.1 joerg
1609 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":name");
1610 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->name, -1, NULL);
1611 1.1 joerg if (rc != SQLITE_OK) {
1612 1.1 joerg sqlite3_finalize(stmt);
1613 1.1 joerg goto Out;
1614 1.1 joerg }
1615 1.1 joerg
1616 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":section");
1617 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->section, -1, NULL);
1618 1.1 joerg if (rc != SQLITE_OK) {
1619 1.1 joerg sqlite3_finalize(stmt);
1620 1.1 joerg goto Out;
1621 1.1 joerg }
1622 1.1 joerg
1623 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":name_desc");
1624 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->name_desc, -1, NULL);
1625 1.1 joerg if (rc != SQLITE_OK) {
1626 1.1 joerg sqlite3_finalize(stmt);
1627 1.1 joerg goto Out;
1628 1.1 joerg }
1629 1.1 joerg
1630 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":desc");
1631 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->desc.data,
1632 1.1 joerg rec->desc.offset + 1, NULL);
1633 1.1 joerg if (rc != SQLITE_OK) {
1634 1.1 joerg sqlite3_finalize(stmt);
1635 1.1 joerg goto Out;
1636 1.1 joerg }
1637 1.1 joerg
1638 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":lib");
1639 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->lib.data, rec->lib.offset + 1, NULL);
1640 1.1 joerg if (rc != SQLITE_OK) {
1641 1.1 joerg sqlite3_finalize(stmt);
1642 1.1 joerg goto Out;
1643 1.1 joerg }
1644 1.1 joerg
1645 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":return_vals");
1646 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->return_vals.data,
1647 1.1 joerg rec->return_vals.offset + 1, NULL);
1648 1.1 joerg if (rc != SQLITE_OK) {
1649 1.1 joerg sqlite3_finalize(stmt);
1650 1.1 joerg goto Out;
1651 1.1 joerg }
1652 1.1 joerg
1653 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":env");
1654 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->env.data, rec->env.offset + 1, NULL);
1655 1.1 joerg if (rc != SQLITE_OK) {
1656 1.1 joerg sqlite3_finalize(stmt);
1657 1.1 joerg goto Out;
1658 1.1 joerg }
1659 1.1 joerg
1660 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":files");
1661 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->files.data,
1662 1.1 joerg rec->files.offset + 1, NULL);
1663 1.1 joerg if (rc != SQLITE_OK) {
1664 1.1 joerg sqlite3_finalize(stmt);
1665 1.1 joerg goto Out;
1666 1.1 joerg }
1667 1.1 joerg
1668 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":exit_status");
1669 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->exit_status.data,
1670 1.1 joerg rec->exit_status.offset + 1, NULL);
1671 1.1 joerg if (rc != SQLITE_OK) {
1672 1.1 joerg sqlite3_finalize(stmt);
1673 1.1 joerg goto Out;
1674 1.1 joerg }
1675 1.1 joerg
1676 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":diagnostics");
1677 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->diagnostics.data,
1678 1.1 joerg rec->diagnostics.offset + 1, NULL);
1679 1.1 joerg if (rc != SQLITE_OK) {
1680 1.1 joerg sqlite3_finalize(stmt);
1681 1.1 joerg goto Out;
1682 1.1 joerg }
1683 1.1 joerg
1684 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":errors");
1685 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->errors.data,
1686 1.1 joerg rec->errors.offset + 1, NULL);
1687 1.1 joerg if (rc != SQLITE_OK) {
1688 1.1 joerg sqlite3_finalize(stmt);
1689 1.1 joerg goto Out;
1690 1.1 joerg }
1691 1.18 christos
1692 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":md5_hash");
1693 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->md5_hash, -1, NULL);
1694 1.1 joerg if (rc != SQLITE_OK) {
1695 1.1 joerg sqlite3_finalize(stmt);
1696 1.1 joerg goto Out;
1697 1.1 joerg }
1698 1.18 christos
1699 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":machine");
1700 1.1 joerg if (rec->machine)
1701 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->machine, -1, NULL);
1702 1.1 joerg else
1703 1.1 joerg rc = sqlite3_bind_null(stmt, idx);
1704 1.1 joerg if (rc != SQLITE_OK) {
1705 1.1 joerg sqlite3_finalize(stmt);
1706 1.1 joerg goto Out;
1707 1.1 joerg }
1708 1.1 joerg
1709 1.1 joerg rc = sqlite3_step(stmt);
1710 1.1 joerg if (rc != SQLITE_DONE) {
1711 1.1 joerg sqlite3_finalize(stmt);
1712 1.1 joerg goto Out;
1713 1.1 joerg }
1714 1.1 joerg
1715 1.1 joerg sqlite3_finalize(stmt);
1716 1.18 christos
1717 1.1 joerg /* Get the row id of the last inserted row */
1718 1.1 joerg mandb_rowid = sqlite3_last_insert_rowid(db);
1719 1.18 christos
1720 1.1 joerg /*------------------------Populate the mandb_meta table-----------------------*/
1721 1.1 joerg sqlstr = "INSERT INTO mandb_meta VALUES (:device, :inode, :mtime,"
1722 1.1 joerg " :file, :md5_hash, :id)";
1723 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
1724 1.1 joerg if (rc != SQLITE_OK)
1725 1.1 joerg goto Out;
1726 1.1 joerg
1727 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":device");
1728 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, rec->device);
1729 1.1 joerg if (rc != SQLITE_OK) {
1730 1.1 joerg sqlite3_finalize(stmt);
1731 1.1 joerg goto Out;
1732 1.1 joerg }
1733 1.1 joerg
1734 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":inode");
1735 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, rec->inode);
1736 1.1 joerg if (rc != SQLITE_OK) {
1737 1.1 joerg sqlite3_finalize(stmt);
1738 1.1 joerg goto Out;
1739 1.1 joerg }
1740 1.1 joerg
1741 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":mtime");
1742 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, rec->mtime);
1743 1.1 joerg if (rc != SQLITE_OK) {
1744 1.1 joerg sqlite3_finalize(stmt);
1745 1.1 joerg goto Out;
1746 1.1 joerg }
1747 1.1 joerg
1748 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":file");
1749 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->file_path, -1, NULL);
1750 1.1 joerg if (rc != SQLITE_OK) {
1751 1.1 joerg sqlite3_finalize(stmt);
1752 1.1 joerg goto Out;
1753 1.1 joerg }
1754 1.1 joerg
1755 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":md5_hash");
1756 1.1 joerg rc = sqlite3_bind_text(stmt, idx, rec->md5_hash, -1, NULL);
1757 1.1 joerg if (rc != SQLITE_OK) {
1758 1.1 joerg sqlite3_finalize(stmt);
1759 1.1 joerg goto Out;
1760 1.1 joerg }
1761 1.1 joerg
1762 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":id");
1763 1.1 joerg rc = sqlite3_bind_int64(stmt, idx, mandb_rowid);
1764 1.1 joerg if (rc != SQLITE_OK) {
1765 1.1 joerg sqlite3_finalize(stmt);
1766 1.1 joerg goto Out;
1767 1.1 joerg }
1768 1.1 joerg
1769 1.1 joerg rc = sqlite3_step(stmt);
1770 1.1 joerg sqlite3_finalize(stmt);
1771 1.24 wiz if (rc == SQLITE_CONSTRAINT_UNIQUE) {
1772 1.1 joerg /* The *most* probable reason for reaching here is that
1773 1.1 joerg * the UNIQUE contraint on the file column of the mandb_meta
1774 1.1 joerg * table was violated.
1775 1.1 joerg * This can happen when a file was updated/modified.
1776 1.1 joerg * To fix this we need to do two things:
1777 1.1 joerg * 1. Delete the row for the older version of this file
1778 1.1 joerg * from mandb table.
1779 1.1 joerg * 2. Run an UPDATE query to update the row for this file
1780 1.1 joerg * in the mandb_meta table.
1781 1.1 joerg */
1782 1.1 joerg warnx("Trying to update index for %s", rec->file_path);
1783 1.1 joerg char *sql = sqlite3_mprintf("DELETE FROM mandb "
1784 1.1 joerg "WHERE rowid = (SELECT id"
1785 1.1 joerg " FROM mandb_meta"
1786 1.1 joerg " WHERE file = %Q)",
1787 1.1 joerg rec->file_path);
1788 1.1 joerg sqlite3_exec(db, sql, NULL, NULL, &errmsg);
1789 1.1 joerg sqlite3_free(sql);
1790 1.1 joerg if (errmsg != NULL) {
1791 1.13 wiz if (mflags.verbosity)
1792 1.13 wiz warnx("%s", errmsg);
1793 1.1 joerg free(errmsg);
1794 1.1 joerg }
1795 1.1 joerg sqlstr = "UPDATE mandb_meta SET device = :device,"
1796 1.1 joerg " inode = :inode, mtime = :mtime, id = :id,"
1797 1.1 joerg " md5_hash = :md5 WHERE file = :file";
1798 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
1799 1.1 joerg if (rc != SQLITE_OK) {
1800 1.13 wiz if (mflags.verbosity)
1801 1.13 wiz warnx("Update failed with error: %s",
1802 1.1 joerg sqlite3_errmsg(db));
1803 1.1 joerg close_db(db);
1804 1.1 joerg cleanup(rec);
1805 1.1 joerg errx(EXIT_FAILURE,
1806 1.1 joerg "Consider running makemandb with -f option");
1807 1.1 joerg }
1808 1.1 joerg
1809 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":device");
1810 1.1 joerg sqlite3_bind_int64(stmt, idx, rec->device);
1811 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":inode");
1812 1.1 joerg sqlite3_bind_int64(stmt, idx, rec->inode);
1813 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":mtime");
1814 1.1 joerg sqlite3_bind_int64(stmt, idx, rec->mtime);
1815 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":id");
1816 1.1 joerg sqlite3_bind_int64(stmt, idx, mandb_rowid);
1817 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":md5");
1818 1.1 joerg sqlite3_bind_text(stmt, idx, rec->md5_hash, -1, NULL);
1819 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":file");
1820 1.1 joerg sqlite3_bind_text(stmt, idx, rec->file_path, -1, NULL);
1821 1.1 joerg rc = sqlite3_step(stmt);
1822 1.1 joerg sqlite3_finalize(stmt);
1823 1.1 joerg
1824 1.1 joerg if (rc != SQLITE_DONE) {
1825 1.13 wiz if (mflags.verbosity)
1826 1.13 wiz warnx("%s", sqlite3_errmsg(db));
1827 1.1 joerg close_db(db);
1828 1.1 joerg cleanup(rec);
1829 1.1 joerg errx(EXIT_FAILURE,
1830 1.1 joerg "Consider running makemandb with -f option");
1831 1.1 joerg }
1832 1.1 joerg } else if (rc != SQLITE_DONE) {
1833 1.1 joerg /* Otherwise make this error fatal */
1834 1.1 joerg warnx("Failed at %s\n%s", rec->file_path, sqlite3_errmsg(db));
1835 1.1 joerg cleanup(rec);
1836 1.1 joerg close_db(db);
1837 1.1 joerg exit(EXIT_FAILURE);
1838 1.1 joerg }
1839 1.1 joerg
1840 1.1 joerg /*------------------------ Populate the mandb_links table---------------------*/
1841 1.1 joerg char *str = NULL;
1842 1.18 christos char *links;
1843 1.1 joerg if (rec->links && strlen(rec->links)) {
1844 1.1 joerg links = rec->links;
1845 1.1 joerg for(ln = strtok(links, " "); ln; ln = strtok(NULL, " ")) {
1846 1.1 joerg if (ln[0] == ',')
1847 1.1 joerg ln++;
1848 1.1 joerg if(ln[strlen(ln) - 1] == ',')
1849 1.1 joerg ln[strlen(ln) - 1] = 0;
1850 1.18 christos
1851 1.1 joerg str = sqlite3_mprintf("INSERT INTO mandb_links"
1852 1.9 wiz " VALUES (%Q, %Q, %Q, %Q, %Q)",
1853 1.1 joerg ln, rec->name, rec->section,
1854 1.9 wiz rec->machine, rec->md5_hash);
1855 1.1 joerg sqlite3_exec(db, str, NULL, NULL, &errmsg);
1856 1.1 joerg sqlite3_free(str);
1857 1.1 joerg if (errmsg != NULL) {
1858 1.1 joerg warnx("%s", errmsg);
1859 1.1 joerg cleanup(rec);
1860 1.1 joerg free(errmsg);
1861 1.1 joerg return -1;
1862 1.1 joerg }
1863 1.1 joerg }
1864 1.1 joerg }
1865 1.1 joerg
1866 1.1 joerg cleanup(rec);
1867 1.1 joerg return 0;
1868 1.1 joerg
1869 1.1 joerg Out:
1870 1.13 wiz if (mflags.verbosity)
1871 1.13 wiz warnx("%s", sqlite3_errmsg(db));
1872 1.1 joerg cleanup(rec);
1873 1.1 joerg return -1;
1874 1.1 joerg }
1875 1.1 joerg
1876 1.1 joerg /*
1877 1.1 joerg * check_md5--
1878 1.1 joerg * Generates the md5 hash of the file and checks if it already doesn't exist
1879 1.1 joerg * in the table (passed as the 3rd parameter).
1880 1.1 joerg * This function is being used to avoid hardlinks.
1881 1.1 joerg * On successful completion it will also set the value of the fourth parameter
1882 1.1 joerg * to the md5 hash of the file (computed previously). It is the responsibility
1883 1.1 joerg * of the caller to free this buffer.
1884 1.1 joerg * Return values:
1885 1.1 joerg * -1: If an error occurs somewhere and sets the md5 return buffer to NULL.
1886 1.1 joerg * 0: If the md5 hash does not exist in the table.
1887 1.1 joerg * 1: If the hash exists in the database.
1888 1.1 joerg */
1889 1.1 joerg static int
1890 1.5 joerg check_md5(const char *file, sqlite3 *db, const char *table, char **md5sum,
1891 1.5 joerg void *buf, size_t buflen)
1892 1.1 joerg {
1893 1.1 joerg int rc = 0;
1894 1.1 joerg int idx = -1;
1895 1.1 joerg char *sqlstr = NULL;
1896 1.1 joerg sqlite3_stmt *stmt = NULL;
1897 1.1 joerg
1898 1.1 joerg assert(file != NULL);
1899 1.5 joerg *md5sum = MD5Data(buf, buflen, NULL);
1900 1.5 joerg if (*md5sum == NULL) {
1901 1.13 wiz if (mflags.verbosity)
1902 1.13 wiz warn("md5 failed: %s", file);
1903 1.1 joerg return -1;
1904 1.1 joerg }
1905 1.1 joerg
1906 1.1 joerg easprintf(&sqlstr, "SELECT * FROM %s WHERE md5_hash = :md5_hash",
1907 1.1 joerg table);
1908 1.1 joerg rc = sqlite3_prepare_v2(db, sqlstr, -1, &stmt, NULL);
1909 1.1 joerg if (rc != SQLITE_OK) {
1910 1.1 joerg free(sqlstr);
1911 1.5 joerg free(*md5sum);
1912 1.5 joerg *md5sum = NULL;
1913 1.1 joerg return -1;
1914 1.1 joerg }
1915 1.1 joerg
1916 1.1 joerg idx = sqlite3_bind_parameter_index(stmt, ":md5_hash");
1917 1.5 joerg rc = sqlite3_bind_text(stmt, idx, *md5sum, -1, NULL);
1918 1.1 joerg if (rc != SQLITE_OK) {
1919 1.13 wiz if (mflags.verbosity)
1920 1.13 wiz warnx("%s", sqlite3_errmsg(db));
1921 1.1 joerg sqlite3_finalize(stmt);
1922 1.1 joerg free(sqlstr);
1923 1.5 joerg free(*md5sum);
1924 1.5 joerg *md5sum = NULL;
1925 1.1 joerg return -1;
1926 1.1 joerg }
1927 1.1 joerg
1928 1.1 joerg if (sqlite3_step(stmt) == SQLITE_ROW) {
1929 1.18 christos sqlite3_finalize(stmt);
1930 1.1 joerg free(sqlstr);
1931 1.1 joerg return 0;
1932 1.1 joerg }
1933 1.1 joerg
1934 1.1 joerg sqlite3_finalize(stmt);
1935 1.1 joerg free(sqlstr);
1936 1.1 joerg return 1;
1937 1.1 joerg }
1938 1.1 joerg
1939 1.1 joerg /* Optimize the index for faster search */
1940 1.1 joerg static void
1941 1.1 joerg optimize(sqlite3 *db)
1942 1.1 joerg {
1943 1.1 joerg const char *sqlstr;
1944 1.1 joerg char *errmsg = NULL;
1945 1.1 joerg
1946 1.13 wiz if (mflags.verbosity == 2)
1947 1.1 joerg printf("Optimizing the database index\n");
1948 1.1 joerg sqlstr = "INSERT INTO mandb(mandb) VALUES (\'optimize\');"
1949 1.1 joerg "VACUUM";
1950 1.1 joerg sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg);
1951 1.1 joerg if (errmsg != NULL) {
1952 1.13 wiz if (mflags.verbosity)
1953 1.13 wiz warnx("%s", errmsg);
1954 1.1 joerg free(errmsg);
1955 1.1 joerg return;
1956 1.1 joerg }
1957 1.1 joerg }
1958 1.1 joerg
1959 1.18 christos /*
1960 1.1 joerg * cleanup --
1961 1.1 joerg * cleans up the global buffers
1962 1.1 joerg */
1963 1.1 joerg static void
1964 1.1 joerg cleanup(mandb_rec *rec)
1965 1.1 joerg {
1966 1.1 joerg rec->desc.offset = 0;
1967 1.1 joerg rec->lib.offset = 0;
1968 1.1 joerg rec->return_vals.offset = 0;
1969 1.1 joerg rec->env.offset = 0;
1970 1.1 joerg rec->exit_status.offset = 0;
1971 1.1 joerg rec->diagnostics.offset = 0;
1972 1.1 joerg rec->errors.offset = 0;
1973 1.1 joerg rec->files.offset = 0;
1974 1.1 joerg
1975 1.1 joerg free(rec->machine);
1976 1.1 joerg rec->machine = NULL;
1977 1.1 joerg
1978 1.1 joerg free(rec->links);
1979 1.1 joerg rec->links = NULL;
1980 1.1 joerg
1981 1.1 joerg free(rec->file_path);
1982 1.1 joerg rec->file_path = NULL;
1983 1.1 joerg
1984 1.1 joerg free(rec->name);
1985 1.1 joerg rec->name = NULL;
1986 1.1 joerg
1987 1.1 joerg free(rec->name_desc);
1988 1.1 joerg rec->name_desc = NULL;
1989 1.1 joerg
1990 1.1 joerg free(rec->md5_hash);
1991 1.1 joerg rec->md5_hash = NULL;
1992 1.1 joerg }
1993 1.1 joerg
1994 1.1 joerg /*
1995 1.1 joerg * init_secbuffs--
1996 1.1 joerg * Sets the value of buflen for all the sec_buff field of rec. And then
1997 1.1 joerg * allocate memory to each sec_buff member of rec.
1998 1.1 joerg */
1999 1.1 joerg static void
2000 1.1 joerg init_secbuffs(mandb_rec *rec)
2001 1.1 joerg {
2002 1.1 joerg /*
2003 1.1 joerg * Some sec_buff might need more memory, for example desc,
2004 1.1 joerg * which stores the data of the DESCRIPTION section,
2005 1.1 joerg * while some might need very small amount of memory.
2006 1.1 joerg * Therefore explicitly setting the value of buflen field for
2007 1.1 joerg * each sec_buff.
2008 1.1 joerg */
2009 1.1 joerg rec->desc.buflen = 10 * BUFLEN;
2010 1.1 joerg rec->desc.data = emalloc(rec->desc.buflen);
2011 1.1 joerg rec->desc.offset = 0;
2012 1.1 joerg
2013 1.1 joerg rec->lib.buflen = BUFLEN / 2;
2014 1.1 joerg rec->lib.data = emalloc(rec->lib.buflen);
2015 1.1 joerg rec->lib.offset = 0;
2016 1.1 joerg
2017 1.1 joerg rec->return_vals.buflen = BUFLEN;
2018 1.1 joerg rec->return_vals.data = emalloc(rec->return_vals.buflen);
2019 1.1 joerg rec->return_vals.offset = 0;
2020 1.1 joerg
2021 1.1 joerg rec->exit_status.buflen = BUFLEN;
2022 1.1 joerg rec->exit_status.data = emalloc(rec->exit_status.buflen);
2023 1.1 joerg rec->exit_status.offset = 0;
2024 1.1 joerg
2025 1.1 joerg rec->env.buflen = BUFLEN;
2026 1.1 joerg rec->env.data = emalloc(rec->env.buflen);
2027 1.1 joerg rec->env.offset = 0;
2028 1.1 joerg
2029 1.1 joerg rec->files.buflen = BUFLEN;
2030 1.1 joerg rec->files.data = emalloc(rec->files.buflen);
2031 1.1 joerg rec->files.offset = 0;
2032 1.1 joerg
2033 1.1 joerg rec->diagnostics.buflen = BUFLEN;
2034 1.1 joerg rec->diagnostics.data = emalloc(rec->diagnostics.buflen);
2035 1.1 joerg rec->diagnostics.offset = 0;
2036 1.1 joerg
2037 1.1 joerg rec->errors.buflen = BUFLEN;
2038 1.1 joerg rec->errors.data = emalloc(rec->errors.buflen);
2039 1.1 joerg rec->errors.offset = 0;
2040 1.1 joerg }
2041 1.1 joerg
2042 1.1 joerg /*
2043 1.1 joerg * free_secbuffs--
2044 1.1 joerg * This function should be called at the end, when all the pages have been
2045 1.1 joerg * parsed.
2046 1.1 joerg * It frees the memory allocated to sec_buffs by init_secbuffs in the starting.
2047 1.1 joerg */
2048 1.1 joerg static void
2049 1.1 joerg free_secbuffs(mandb_rec *rec)
2050 1.1 joerg {
2051 1.1 joerg free(rec->desc.data);
2052 1.1 joerg free(rec->lib.data);
2053 1.1 joerg free(rec->return_vals.data);
2054 1.1 joerg free(rec->exit_status.data);
2055 1.1 joerg free(rec->env.data);
2056 1.1 joerg free(rec->files.data);
2057 1.1 joerg free(rec->diagnostics.data);
2058 1.1 joerg free(rec->errors.data);
2059 1.1 joerg }
2060 1.1 joerg
2061 1.4 joerg static void
2062 1.4 joerg replace_hyph(char *str)
2063 1.4 joerg {
2064 1.4 joerg char *iter = str;
2065 1.4 joerg while ((iter = strchr(iter, ASCII_HYPH)) != NULL)
2066 1.4 joerg *iter = '-';
2067 1.23 wiz
2068 1.23 wiz iter = str;
2069 1.23 wiz while ((iter = strchr(iter, ASCII_NBRSP)) != NULL)
2070 1.23 wiz *iter = '-';
2071 1.4 joerg }
2072 1.4 joerg
2073 1.1 joerg static char *
2074 1.1 joerg parse_escape(const char *str)
2075 1.1 joerg {
2076 1.1 joerg const char *backslash, *last_backslash;
2077 1.1 joerg char *result, *iter;
2078 1.1 joerg size_t len;
2079 1.1 joerg
2080 1.1 joerg assert(str);
2081 1.1 joerg
2082 1.1 joerg last_backslash = str;
2083 1.1 joerg backslash = strchr(str, '\\');
2084 1.4 joerg if (backslash == NULL) {
2085 1.4 joerg result = estrdup(str);
2086 1.4 joerg replace_hyph(result);
2087 1.4 joerg return result;
2088 1.4 joerg }
2089 1.1 joerg
2090 1.1 joerg result = emalloc(strlen(str) + 1);
2091 1.1 joerg iter = result;
2092 1.1 joerg
2093 1.1 joerg do {
2094 1.1 joerg len = backslash - last_backslash;
2095 1.1 joerg memcpy(iter, last_backslash, len);
2096 1.1 joerg iter += len;
2097 1.1 joerg if (backslash[1] == '-' || backslash[1] == ' ') {
2098 1.1 joerg *iter++ = backslash[1];
2099 1.1 joerg last_backslash = backslash + 2;
2100 1.1 joerg backslash = strchr(backslash + 2, '\\');
2101 1.1 joerg } else {
2102 1.1 joerg ++backslash;
2103 1.1 joerg mandoc_escape(&backslash, NULL, NULL);
2104 1.1 joerg last_backslash = backslash;
2105 1.1 joerg if (backslash == NULL)
2106 1.1 joerg break;
2107 1.1 joerg backslash = strchr(last_backslash, '\\');
2108 1.1 joerg }
2109 1.1 joerg } while (backslash != NULL);
2110 1.1 joerg if (last_backslash != NULL)
2111 1.1 joerg strcpy(iter, last_backslash);
2112 1.18 christos
2113 1.4 joerg replace_hyph(result);
2114 1.1 joerg return result;
2115 1.1 joerg }
2116 1.1 joerg
2117 1.1 joerg /*
2118 1.1 joerg * append--
2119 1.1 joerg * Concatenates a space and src at the end of sbuff->data (much like concat in
2120 1.1 joerg * apropos-utils.c).
2121 1.1 joerg * Rather than reallocating space for writing data, it uses the value of the
2122 1.1 joerg * offset field of sec_buff to write new data at the free space left in the
2123 1.1 joerg * buffer.
2124 1.1 joerg * In case the size of the data to be appended exceeds the number of bytes left
2125 1.1 joerg * in the buffer, it reallocates buflen number of bytes and then continues.
2126 1.1 joerg * Value of offset field should be adjusted as new data is written.
2127 1.1 joerg *
2128 1.1 joerg * NOTE: This function does not write the null byte at the end of the buffers,
2129 1.1 joerg * write a null byte at the position pointed to by offset before inserting data
2130 1.1 joerg * in the db.
2131 1.1 joerg */
2132 1.1 joerg static void
2133 1.1 joerg append(secbuff *sbuff, const char *src)
2134 1.1 joerg {
2135 1.1 joerg short flag = 0;
2136 1.1 joerg size_t srclen, newlen;
2137 1.1 joerg char *temp;
2138 1.1 joerg
2139 1.1 joerg assert(src != NULL);
2140 1.1 joerg temp = parse_escape(src);
2141 1.1 joerg srclen = strlen(temp);
2142 1.1 joerg
2143 1.1 joerg if (sbuff->data == NULL) {
2144 1.1 joerg sbuff->data = emalloc(sbuff->buflen);
2145 1.1 joerg sbuff->offset = 0;
2146 1.1 joerg }
2147 1.1 joerg
2148 1.1 joerg newlen = sbuff->offset + srclen + 2;
2149 1.1 joerg if (newlen >= sbuff->buflen) {
2150 1.1 joerg while (sbuff->buflen < newlen)
2151 1.1 joerg sbuff->buflen += sbuff->buflen;
2152 1.1 joerg sbuff->data = erealloc(sbuff->data, sbuff->buflen);
2153 1.1 joerg flag = 1;
2154 1.1 joerg }
2155 1.1 joerg
2156 1.1 joerg /* Append a space at the end of the buffer. */
2157 1.1 joerg if (sbuff->offset || flag)
2158 1.1 joerg sbuff->data[sbuff->offset++] = ' ';
2159 1.1 joerg /* Now, copy src at the end of the buffer. */
2160 1.1 joerg memcpy(sbuff->data + sbuff->offset, temp, srclen);
2161 1.1 joerg sbuff->offset += srclen;
2162 1.1 joerg free(temp);
2163 1.1 joerg }
2164 1.1 joerg
2165 1.1 joerg static void
2166 1.1 joerg usage(void)
2167 1.1 joerg {
2168 1.13 wiz fprintf(stderr, "Usage: %s [-floQqv] [-C path]\n", getprogname());
2169 1.1 joerg exit(1);
2170 1.1 joerg }
2171