-/* mhcachesbr.c -- routines to manipulate the MIME content cache
- *
- * This code is Copyright (c) 2002, by the authors of nmh. See the
- * COPYRIGHT file in the root directory of the nmh distribution for
- * complete copyright information.
- */
-
-#include "h/mh.h"
-#include "sbr/m_gmprot.h"
-#include "sbr/m_getfld.h"
-#include "sbr/read_yes_or_no_if_tty.h"
-#include "sbr/trimcpy.h"
-#include "sbr/error.h"
-#include <fcntl.h>
-#include "h/mts.h"
-#include "h/tws.h"
-#include "h/mime.h"
-#include "h/mhparse.h"
-#include "h/mhcachesbr.h"
-#include "h/utils.h"
-#include "mhmisc.h"
-#include "sbr/lock_file.h"
-#include "sbr/m_mktemp.h"
-
-#ifdef HAVE_SYS_TIME_H
-# include <sys/time.h>
-#endif
-#include <time.h>
-
-#define X(sw, minchars, id) { sw, minchars, id },
-DEFINE_SWITCH_ARRAY(CACHE, caches);
-#undef X
-struct swit *cache_policy = caches;
-
-extern int debugsw;
-
-/* cache policies */
-int rcachesw = CACHE_ASK;
-int wcachesw = CACHE_ASK;
-
-/*
- * Location of public and private cache. These must
- * be set before these routines are called.
- */
-char *cache_public;
-char *cache_private;
-
-/*
- * static prototypes
- */
-static void cache_content (CT);
-static int find_cache_aux (int, char *, char *, char *, int);
-static int find_cache_aux2 (char *, char *, char *, int);
-
-
-/*
- * Top level entry point to cache content
- * from a group of messages
- */
-
-void
-cache_all_messages (CT *cts)
-{
- CT ct, *ctp;
-
- for (ctp = cts; *ctp; ctp++) {
- ct = *ctp;
- if (type_ok (ct, 1)) {
- cache_content (ct);
- if (ct->c_fp) {
- fclose (ct->c_fp);
- ct->c_fp = NULL;
- }
- if (ct->c_ceclosefnx)
- (*ct->c_ceclosefnx) (ct);
- }
- }
- flush_errors ();
-}
-
-
-/*
- * Entry point to cache content from external sources.
- */
-
-static void
-cache_content (CT ct)
-{
- int cachetype;
- char *file, cachefile[BUFSIZ];
- CE ce = &ct->c_cefile;
-
- if (!ct->c_id) {
- inform("no %s: field in %s", ID_FIELD, ct->c_file);
- return;
- }
-
- if (!ce) {
- inform("unable to decode %s", ct->c_file);
- return;
- }
-
- if (find_cache (NULL, wcachesw != CACHE_NEVER ? wcachesw : CACHE_ASK,
- &cachetype, ct->c_id, cachefile, sizeof(cachefile))
- == NOTOK) {
- inform("unable to cache %s's contents", ct->c_file);
- return;
- }
- if (wcachesw != CACHE_NEVER && wcachesw != CACHE_ASK) {
- fflush (stdout);
- fprintf (stderr, "caching message %s as file %s\n", ct->c_file,
- cachefile);
- }
-
- if (ce->ce_file) {
- int mask = umask (cachetype ? ~m_gmprot () : 0222);
- FILE *fp;
-
- if (debugsw)
- fprintf (stderr, "caching by copying %s...\n", ce->ce_file);
-
- file = NULL;
- if ((*ct->c_ceopenfnx) (ct, &file) == NOTOK)
- goto reset_umask;
-
- if ((fp = fopen (cachefile, "w"))) {
- int cc;
- char buffer[BUFSIZ];
- FILE *gp = ce->ce_fp;
-
- fseek (gp, 0L, SEEK_SET);
-
- while ((cc = fread (buffer, sizeof(*buffer), sizeof(buffer), gp))
- > 0)
- if ((int) fwrite (buffer, sizeof(*buffer), cc, fp) < cc) {
- advise ("cache_content", "fwrite");
- }
- fflush (fp);
-
- if (ferror (gp)) {
- admonish (ce->ce_file, "error reading");
- (void) m_unlink (cachefile);
- } else {
- if (ferror (fp)) {
- admonish (cachefile, "error writing");
- (void) m_unlink (cachefile);
- }
- }
- fclose (fp);
- } else
- content_error (cachefile, ct, "unable to fopen for writing");
-reset_umask:
- umask (mask);
- } else {
- if (debugsw)
- fprintf (stderr, "in place caching...\n");
-
- file = cachefile;
- if ((*ct->c_ceopenfnx) (ct, &file) != NOTOK)
- chmod (cachefile, cachetype ? m_gmprot () : 0444);
- }
-}
-
-
-int
-find_cache (CT ct, int policy, int *writing, char *id,
- char *buffer, int buflen)
-{
- int status = NOTOK;
-
- if (id == NULL)
- return NOTOK;
- id = trimcpy (id);
-
- if (debugsw)
- fprintf (stderr, "find_cache %s(%d) %s %s\n", caches[policy].sw,
- policy, writing ? "writing" : "reading", id);
-
- switch (policy) {
- case CACHE_NEVER:
- default:
- break;
-
- case CACHE_ASK:
- case CACHE_PUBLIC:
- if (cache_private
- && !writing
- && find_cache_aux (0, cache_private, id,
- buffer, buflen) == OK) {
- if (access (buffer, R_OK) != NOTOK) {
-got_private:
- if (writing)
- *writing = 1;
-got_it:
- status = OK;
- break;
- }
- }
- if (cache_public
- && find_cache_aux (writing ? 1 : 0, cache_public, id,
- buffer, buflen) == OK) {
- if (writing || access (buffer, R_OK) != NOTOK) {
- if (writing)
- *writing = 0;
- goto got_it;
- }
- }
- break;
-
- case CACHE_PRIVATE:
- if (cache_private
- && find_cache_aux (writing ? 2 : 0, cache_private, id,
- buffer, buflen) == OK) {
- if (writing || access (buffer, R_OK) != NOTOK)
- goto got_private;
- }
- break;
-
- }
-
- if (status == OK && policy == CACHE_ASK) {
- int len, buflen;
- char *bp, query[BUFSIZ];
-
- /* Get buffer ready to go */
- bp = query;
- buflen = sizeof(query);
-
- /* Now, construct query */
- if (writing) {
- snprintf (bp, buflen, "Make cached, publicly-accessible copy");
- } else {
- struct stat st;
-
- snprintf (bp, buflen, "Use cached copy");
- len = strlen (bp);
- bp += len;
- buflen -= len;
-
- if (ct->c_partno) {
- snprintf (bp, buflen, " of content %s", ct->c_partno);
- len = strlen (bp);
- bp += len;
- buflen -= len;
- }
-
- stat (buffer, &st);
- snprintf (bp, buflen, " (size %lu octets)",
- (unsigned long) st.st_size);
- }
- len = strlen (bp);
- bp += len;
- buflen -= len;
-
- snprintf (bp, buflen, "\n in file %s? ", buffer);
-
- /* Now, check answer */
- if (!read_yes_or_no_if_tty (query))
- status = NOTOK;
- }
-
- if (status == OK && writing) {
- if (*writing && strchr(buffer, '/'))
- make_intermediates (buffer);
- (void) m_unlink (buffer);
- }
-
- free (id);
- return status;
-}
-
-
-static int
-find_cache_aux (int writing, char *directory, char *id,
- char *buffer, int buflen)
-{
- int mask;
- bool usemap;
- char mapfile[BUFSIZ], mapname[BUFSIZ];
- FILE *fp;
- int failed_to_lock = 0;
- static int partno, pid;
- static time_t clock = 0;
-
- usemap = true;
-
- if (debugsw)
- fprintf (stderr, "find_cache_aux %s usemap=%d\n", directory, usemap);
-
- snprintf (mapfile, sizeof(mapfile), "%s/cache.map", directory);
- if (find_cache_aux2 (mapfile, id, mapname, sizeof(mapname)) == OK)
- goto done_map;
-
- if (!writing) {
- if (usemap)
- return NOTOK;
-
-use_raw:
- snprintf (buffer, buflen, "%s/%s", directory, id);
- return OK;
- }
-
- if (!usemap && access (mapfile, W_OK) == NOTOK)
- goto use_raw;
-
- if (clock != 0) {
- time_t now;
-
- time (&now);
- if (now > clock)
- clock = 0;
- } else {
- pid = getpid ();
- }
-
- if (clock == 0) {
- time (&clock);
- partno = 0;
- } else {
- if (partno > 0xff) {
- clock++;
- partno = 0;
- }
- }
-
- snprintf (mapname, sizeof(mapname), "%08x%04x%02x",
- (unsigned int) (clock & 0xffffffff),
- (unsigned int) (pid & 0xffff),
- (unsigned int) (partno++ & 0xff));
-
- if (debugsw)
- fprintf (stderr, "creating mapping %s->%s\n", mapname, id);
-
- make_intermediates (mapfile);
- mask = umask (writing == 2 ? 0077 : 0);
- if (!(fp = lkfopendata (mapfile, "a", &failed_to_lock)) && errno == ENOENT) {
- int fd;
-
- if ((fd = creat (mapfile, 0666)) != NOTOK) {
- close (fd);
- fp = lkfopendata (mapfile, "a", &failed_to_lock);
- if (failed_to_lock) {
- adios (mapfile, "failed to lock");
- }
- }
- }
- umask (mask);
- if (!fp)
- return NOTOK;
- fprintf (fp, "%s: %s\n", mapname, id);
- lkfclosedata (fp, mapfile);
-
-done_map:
- if (*mapname == '/')
- strncpy (buffer, mapname, buflen);
- else
- snprintf (buffer, buflen, "%s/%s", directory, mapname);
- if (debugsw)
- fprintf (stderr, "use %s\n", buffer);
-
- return OK;
-}
-
-
-static int
-find_cache_aux2 (char *mapfile, char *id, char *mapname, int namelen)
-{
- int state;
- char buf[NMH_BUFSIZ], name[NAMESZ];
- FILE *fp;
- m_getfld_state_t gstate;
- int failed_to_lock = 0;
-
- if (!(fp = lkfopendata (mapfile, "r", &failed_to_lock)))
- return NOTOK;
-
- gstate = m_getfld_state_init(fp);
- for (;;) {
- int result;
- char *cp, *dp;
- int bufsz = sizeof buf;
-
- switch (state = m_getfld2(&gstate, name, buf, &bufsz)) {
- case FLD:
- case FLDPLUS:
- strncpy (mapname, name, namelen);
- if (state != FLDPLUS)
- cp = buf;
- else {
- cp = mh_xstrdup(buf);
- while (state == FLDPLUS) {
- bufsz = sizeof buf;
- state = m_getfld2(&gstate, name, buf, &bufsz);
- cp = add (buf, cp);
- }
- }
- dp = trimcpy (cp);
- if (cp != buf)
- free (cp);
- if (debugsw)
- fprintf (stderr, "compare %s to %s <- %s\n", id, dp,
- mapname);
- result = strcmp (id, dp);
- free (dp);
- if (result == 0) {
- lkfclosedata (fp, mapfile);
- return OK;
- }
- continue;
-
- case BODY:
- case FILEEOF:
- default:
- break;
- }
- break;
- }
- m_getfld_state_destroy (&gstate);
-
- lkfclosedata (fp, mapfile);
- return NOTOK;
-}
projects / nmh / commitdiff