Import mdocml-1.12.3
[dragonfly.git] / contrib / mdocml / catman.c
1 /*      $Id: catman.c,v 1.11.2.2 2013/10/11 00:06:48 schwarze Exp $ */
2 /*
3  * Copyright (c) 2011 Kristaps Dzonsons <kristaps@bsd.lv>
4  *
5  * Permission to use, copy, modify, and distribute this software for any
6  * purpose with or without fee is hereby granted, provided that the above
7  * copyright notice and this permission notice appear in all copies.
8  *
9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16  */
17 #ifdef HAVE_CONFIG_H
18 #include "config.h"
19 #endif
20
21 #include <sys/param.h>
22 #include <sys/stat.h>
23 #include <sys/wait.h>
24
25 #include <assert.h>
26 #include <errno.h>
27 #include <fcntl.h>
28 #include <getopt.h>
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <string.h>
32 #include <unistd.h>
33
34 #if defined(__linux__) || defined(__sun)
35 # include <db_185.h>
36 #else
37 # include <db.h>
38 #endif
39
40 #include "manpath.h"
41 #include "mandocdb.h"
42
43 #define xstrlcpy(_dst, _src, _sz) \
44         do if (strlcpy((_dst), (_src), (_sz)) >= (_sz)) { \
45                 fprintf(stderr, "%s: Path too long", (_dst)); \
46                 exit(EXIT_FAILURE); \
47         } while (/* CONSTCOND */0)
48
49 #define xstrlcat(_dst, _src, _sz) \
50         do if (strlcat((_dst), (_src), (_sz)) >= (_sz)) { \
51                 fprintf(stderr, "%s: Path too long", (_dst)); \
52                 exit(EXIT_FAILURE); \
53         } while (/* CONSTCOND */0)
54
55 static  int              indexhtml(char *, size_t, char *, size_t);
56 static  int              manup(const struct manpaths *, char *);
57 static  int              mkpath(char *, mode_t, mode_t);
58 static  int              treecpy(char *, char *);
59 static  int              update(char *, char *);
60 static  void             usage(void);
61
62 static  const char      *progname;
63 static  int              verbose;
64 static  int              force;
65
66 int
67 main(int argc, char *argv[])
68 {
69         int              ch;
70         char            *aux, *base, *conf_file;
71         struct manpaths  dirs;
72         char             buf[MAXPATHLEN];
73         extern char     *optarg;
74         extern int       optind;
75
76         progname = strrchr(argv[0], '/');
77         if (progname == NULL)
78                 progname = argv[0];
79         else
80                 ++progname;
81
82         aux = base = conf_file = NULL;
83         xstrlcpy(buf, "/var/www/cache/man.cgi", MAXPATHLEN);
84
85         while (-1 != (ch = getopt(argc, argv, "C:fm:M:o:v")))
86                 switch (ch) {
87                 case ('C'):
88                         conf_file = optarg;
89                         break;
90                 case ('f'):
91                         force = 1;
92                         break;
93                 case ('m'):
94                         aux = optarg;
95                         break;
96                 case ('M'):
97                         base = optarg;
98                         break;
99                 case ('o'):
100                         xstrlcpy(buf, optarg, MAXPATHLEN);
101                         break;
102                 case ('v'):
103                         verbose++;
104                         break;
105                 default:
106                         usage();
107                         return(EXIT_FAILURE);
108                 }
109
110         argc -= optind;
111         argv += optind;
112
113         if (argc > 0) {
114                 usage();
115                 return(EXIT_FAILURE);
116         }
117
118         memset(&dirs, 0, sizeof(struct manpaths));
119         manpath_parse(&dirs, conf_file, base, aux);
120         ch = manup(&dirs, buf);
121         manpath_free(&dirs);
122         return(ch ? EXIT_SUCCESS : EXIT_FAILURE);
123 }
124
125 static void
126 usage(void)
127 {
128         
129         fprintf(stderr, "usage: %s "
130                         "[-fv] "
131                         "[-C file] "
132                         "[-o path] "
133                         "[-m manpath] "
134                         "[-M manpath]\n",
135                         progname);
136 }
137
138 /*
139  * If "src" file doesn't exist (errors out), return -1.  Otherwise,
140  * return 1 if "src" is newer (which also happens "dst" doesn't exist)
141  * and 0 otherwise.
142  */
143 static int
144 isnewer(const char *dst, const char *src)
145 {
146         struct stat      s1, s2;
147
148         if (-1 == stat(src, &s1))
149                 return(-1);
150         if (force)
151                 return(1);
152
153         return(-1 == stat(dst, &s2) ? 1 : s1.st_mtime > s2.st_mtime);
154 }
155
156 /*
157  * Copy the contents of one file into another.
158  * Returns 0 on failure, 1 on success.
159  */
160 static int
161 filecpy(const char *dst, const char *src)
162 {
163         char             buf[BUFSIZ];
164         int              sfd, dfd, rc;
165         ssize_t          rsz, wsz;
166
167         sfd = dfd = -1;
168         rc = 0;
169
170         if (-1 == (dfd = open(dst, O_CREAT|O_TRUNC|O_WRONLY, 0644))) {
171                 perror(dst);
172                 goto out;
173         } else if (-1 == (sfd = open(src, O_RDONLY, 0))) {
174                 perror(src);
175                 goto out;
176         } 
177
178         while ((rsz = read(sfd, buf, BUFSIZ)) > 0)
179                 if (-1 == (wsz = write(dfd, buf, (size_t)rsz))) {
180                         perror(dst);
181                         goto out;
182                 } else if (wsz < rsz) {
183                         fprintf(stderr, "%s: Short write\n", dst);
184                         goto out;
185                 }
186         
187         if (rsz < 0)
188                 perror(src);
189         else
190                 rc = 1;
191 out:
192         if (-1 != sfd)
193                 close(sfd);
194         if (-1 != dfd)
195                 close(dfd);
196
197         return(rc);
198 }
199
200 /*
201  * Pass over the recno database and re-create HTML pages if they're
202  * found to be out of date.
203  * Returns -1 on fatal error, 1 on success.
204  */
205 static int
206 indexhtml(char *src, size_t ssz, char *dst, size_t dsz)
207 {
208         DB              *idx;
209         DBT              key, val;
210         int              c, rc;
211         unsigned int     fl;
212         const char      *f;
213         char            *d;
214         char             fname[MAXPATHLEN];
215
216         xstrlcpy(fname, dst, MAXPATHLEN);
217         xstrlcat(fname, "/", MAXPATHLEN);
218         xstrlcat(fname, MANDOC_IDX, MAXPATHLEN);
219
220         idx = dbopen(fname, O_RDONLY, 0, DB_RECNO, NULL);
221         if (NULL == idx) {
222                 perror(fname);
223                 return(-1);
224         }
225
226         fl = R_FIRST;
227         while (0 == (c = (*idx->seq)(idx, &key, &val, fl))) {
228                 fl = R_NEXT;
229                 /*
230                  * If the record is zero-length, then it's unassigned.
231                  * Skip past these.
232                  */
233                 if (0 == val.size)
234                         continue;
235
236                 f = (const char *)val.data + 1;
237                 if (NULL == memchr(f, '\0', val.size - 1))
238                         break;
239
240                 src[(int)ssz] = dst[(int)dsz] = '\0';
241
242                 xstrlcat(dst, "/", MAXPATHLEN);
243                 xstrlcat(dst, f, MAXPATHLEN);
244
245                 xstrlcat(src, "/", MAXPATHLEN);
246                 xstrlcat(src, f, MAXPATHLEN);
247
248                 if (-1 == (rc = isnewer(dst, src))) {
249                         fprintf(stderr, "%s: File missing\n", f);
250                         break;
251                 } else if (0 == rc)
252                         continue;
253
254                 d = strrchr(dst, '/');
255                 assert(NULL != d);
256                 *d = '\0';
257
258                 if (-1 == mkpath(dst, 0755, 0755)) {
259                         perror(dst);
260                         break;
261                 }
262
263                 *d = '/';
264
265                 if ( ! filecpy(dst, src))
266                         break;
267                 if (verbose)
268                         printf("%s\n", dst);
269         }
270
271         (*idx->close)(idx);
272
273         if (c < 0)
274                 perror(fname);
275         else if (0 == c) 
276                 fprintf(stderr, "%s: Corrupt index\n", fname);
277
278         return(1 == c ? 1 : -1);
279 }
280
281 /*
282  * Copy both recno and btree databases into the destination.
283  * Call in to begin recreating HTML files.
284  * Return -1 on fatal error and 1 if the update went well.
285  */
286 static int
287 update(char *dst, char *src)
288 {
289         size_t           dsz, ssz;
290
291         dsz = strlen(dst);
292         ssz = strlen(src);
293
294         xstrlcat(src, "/", MAXPATHLEN);
295         xstrlcat(dst, "/", MAXPATHLEN);
296
297         xstrlcat(src, MANDOC_DB, MAXPATHLEN);
298         xstrlcat(dst, MANDOC_DB, MAXPATHLEN);
299
300         if ( ! filecpy(dst, src))
301                 return(-1);
302         if (verbose)
303                 printf("%s\n", dst);
304
305         dst[(int)dsz] = src[(int)ssz] = '\0';
306
307         xstrlcat(src, "/", MAXPATHLEN);
308         xstrlcat(dst, "/", MAXPATHLEN);
309
310         xstrlcat(src, MANDOC_IDX, MAXPATHLEN);
311         xstrlcat(dst, MANDOC_IDX, MAXPATHLEN);
312
313         if ( ! filecpy(dst, src))
314                 return(-1);
315         if (verbose)
316                 printf("%s\n", dst);
317
318         dst[(int)dsz] = src[(int)ssz] = '\0';
319
320         return(indexhtml(src, ssz, dst, dsz));
321 }
322
323 /*
324  * See if btree or recno databases in the destination are out of date
325  * with respect to a single manpath component.
326  * Return -1 on fatal error, 0 if the source is no longer valid (and
327  * shouldn't be listed), and 1 if the update went well.
328  */
329 static int
330 treecpy(char *dst, char *src)
331 {
332         size_t           dsz, ssz;
333         int              rc;
334
335         dsz = strlen(dst);
336         ssz = strlen(src);
337
338         xstrlcat(src, "/", MAXPATHLEN);
339         xstrlcat(dst, "/", MAXPATHLEN);
340
341         xstrlcat(src, MANDOC_IDX, MAXPATHLEN);
342         xstrlcat(dst, MANDOC_IDX, MAXPATHLEN);
343
344         if (-1 == (rc = isnewer(dst, src)))
345                 return(0);
346
347         dst[(int)dsz] = src[(int)ssz] = '\0';
348
349         if (1 == rc)
350                 return(update(dst, src));
351
352         xstrlcat(src, "/", MAXPATHLEN);
353         xstrlcat(dst, "/", MAXPATHLEN);
354
355         xstrlcat(src, MANDOC_DB, MAXPATHLEN);
356         xstrlcat(dst, MANDOC_DB, MAXPATHLEN);
357
358         if (-1 == (rc = isnewer(dst, src)))
359                 return(0);
360         else if (rc == 0)
361                 return(1);
362
363         dst[(int)dsz] = src[(int)ssz] = '\0';
364
365         return(update(dst, src));
366 }
367
368 /*
369  * Update the destination's file-tree with respect to changes in the
370  * source manpath components.
371  * "Change" is defined by an updated index or btree database.
372  * Returns 1 on success, 0 on failure.
373  */
374 static int
375 manup(const struct manpaths *dirs, char *base)
376 {
377         char             dst[MAXPATHLEN],
378                          src[MAXPATHLEN];
379         const char      *path;
380         size_t           i;
381         int              c;
382         size_t           sz;
383         FILE            *f;
384
385         /* Create the path and file for the catman.conf file. */
386
387         sz = strlen(base);
388         xstrlcpy(dst, base, MAXPATHLEN);
389         xstrlcat(dst, "/etc", MAXPATHLEN);
390         if (-1 == mkpath(dst, 0755, 0755)) {
391                 perror(dst);
392                 return(0);
393         }
394
395         xstrlcat(dst, "/catman.conf", MAXPATHLEN);
396         if (NULL == (f = fopen(dst, "w"))) {
397                 perror(dst);
398                 return(0);
399         } else if (verbose)
400                 printf("%s\n", dst);
401
402         for (i = 0; i < dirs->sz; i++) {
403                 path = dirs->paths[i];
404                 dst[(int)sz] = '\0';
405                 xstrlcat(dst, path, MAXPATHLEN);
406                 if (-1 == mkpath(dst, 0755, 0755)) {
407                         perror(dst);
408                         break;
409                 }
410
411                 xstrlcpy(src, path, MAXPATHLEN);
412                 if (-1 == (c = treecpy(dst, src)))
413                         break;
414                 else if (0 == c)
415                         continue;
416
417                 /*
418                  * We want to use a relative path here because manpath.h
419                  * will realpath() when invoked with man.cgi, and we'll
420                  * make sure to chdir() into the cache directory before.
421                  *
422                  * This allows the cache directory to be in an arbitrary
423                  * place, working in both chroot() and non-chroot()
424                  * "safe" modes.
425                  */
426                 assert('/' == path[0]);
427                 fprintf(f, "_whatdb %s/whatis.db\n", path + 1);
428         }
429
430         fclose(f);
431         return(i == dirs->sz);
432 }
433
434 /*
435  * Copyright (c) 1983, 1992, 1993
436  *      The Regents of the University of California.  All rights reserved.
437  *
438  * Redistribution and use in source and binary forms, with or without
439  * modification, are permitted provided that the following conditions
440  * are met:
441  * 1. Redistributions of source code must retain the above copyright
442  *    notice, this list of conditions and the following disclaimer.
443  * 2. Redistributions in binary form must reproduce the above copyright
444  *    notice, this list of conditions and the following disclaimer in the
445  *    documentation and/or other materials provided with the distribution.
446  * 3. Neither the name of the University nor the names of its contributors
447  *    may be used to endorse or promote products derived from this software
448  *    without specific prior written permission.
449  *
450  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
451  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
452  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
453  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
454  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
455  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
456  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
457  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
458  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
459  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
460  * SUCH DAMAGE.
461  */
462 static int
463 mkpath(char *path, mode_t mode, mode_t dir_mode)
464 {
465         struct stat sb;
466         char *slash;
467         int done, exists;
468
469         slash = path;
470
471         for (;;) {
472                 /* LINTED */
473                 slash += strspn(slash, "/");
474                 /* LINTED */
475                 slash += strcspn(slash, "/");
476
477                 done = (*slash == '\0');
478                 *slash = '\0';
479
480                 /* skip existing path components */
481                 exists = !stat(path, &sb);
482                 if (!done && exists && S_ISDIR(sb.st_mode)) {
483                         *slash = '/';
484                         continue;
485                 }
486
487                 if (mkdir(path, done ? mode : dir_mode) == 0) {
488                         if (mode > 0777 && chmod(path, mode) < 0)
489                                 return (-1);
490                 } else {
491                         if (!exists) {
492                                 /* Not there */
493                                 return (-1);
494                         }
495                         if (!S_ISDIR(sb.st_mode)) {
496                                 /* Is there, but isn't a directory */
497                                 errno = ENOTDIR;
498                                 return (-1);
499                         }
500                 }
501
502                 if (done)
503                         break;
504
505                 *slash = '/';
506         }
507
508         return (0);
509 }