/*
* annosbr.c -- prepend annotation to messages
*
- * $Id$
- *
* 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 <h/tws.h>
+#include <h/utils.h>
#include <fcntl.h>
#include <errno.h>
+#include <utime.h>
/*
*/
static int annosbr (int, char *, char *, char *, int, int, int, int);
+/*
+ * This "local" global and the annopreserve() function are a hack that allows additional
+ * functionality to be added to anno without piling on yet another annotate() argument.
+ */
+
+static int preserve_actime_and_modtime = 0; /* set to preserve access and modification times on annotated message */
int
annotate (char *file, char *comp, char *text, int inplace, int datesw, int delete, int append)
{
- int i, fd;
+ int i, fd;
+ struct utimbuf b;
+ struct stat s;
/* open and lock the file to be annotated */
if ((fd = lkopen (file, O_RDWR, 0)) == NOTOK) {
return 1;
}
+ if (stat(file, &s) == -1) {
+ advise("can't get access and modification times for %s", file);
+ preserve_actime_and_modtime = 0;
+ }
+
+ b.actime = s.st_atime;
+ b.modtime = s.st_mtime;
+
i = annosbr (fd, file, comp, text, inplace, datesw, delete, append);
+
+ if (preserve_actime_and_modtime && utime(file, &b) == -1)
+ advise("can't set access and modification times for %s", file);
+
lkclose (fd, file);
return i;
}
* This buffer might need to be quite large, so we grow it as needed.
*/
- if ((field = (char *)malloc(field_size = 256)) == (char *)0)
- adios(NULL, "can't allocate field buffer.");
+ field = (char *)mh_xmalloc(field_size = 256);
/*
* Get the length of the field name since we use it often.
}
if (++n >= field_size - 1) {
- if ((field = (char *)realloc((void *)field, field_size += 256)) == (char *)0)
- adios(NULL, "can't grow field buffer.");
+ field = (char *) mh_xrealloc((void *)field, field_size += 256);
cp = field + n - 1;
}
return;
}
+/*
+ * Set the preserve-times flag. This hack eliminates the need for an additional argument to annotate().
+ */
+
+void
+annopreserve(int preserve)
+{
+ preserve_actime_and_modtime = preserve;
+ return;
+}
static int
annosbr (int fd, char *file, char *comp, char *text, int inplace, int datesw, int delete, int append)
FILE *tmp;
int c; /* current character */
int count; /* header field (annotation) counter */
- char *field; /* buffer for header field */
- int field_size; /* size of field buffer */
- FILE *fp; /* file pointer made from locked file descriptor */
+ char *field = NULL; /* buffer for header field */
+ int field_size = 0; /* size of field buffer */
+ FILE *fp = NULL; /* file pointer made from locked file descriptor */
int length; /* length of field name */
int n; /* number of bytes written */
- mode = fstat (fd, &st) != NOTOK ? (st.st_mode & 0777) : m_gmprot ();
-
- strncpy (tmpfil, m_scratch (file, "annotate"), sizeof(tmpfil));
+ mode = fstat (fd, &st) != NOTOK ? (int) (st.st_mode & 0777) : m_gmprot ();
- if ((tmp = fopen (tmpfil, "w")) == NULL) {
- admonish (tmpfil, "unable to create");
- return 1;
- }
+ strncpy (tmpfil, m_mktemp2(file, "annotate", NULL, &tmp), sizeof(tmpfil));
chmod (tmpfil, mode);
/*
* as they're read in. This buffer is grown as needed later.
*/
- if (delete >= 0 || append != 0) {
+ if (delete >= -1 || append != 0) {
if ((fp = fdopen(fd, "r")) == (FILE *)0)
adios(NULL, "unable to fdopen file.");
- if ((field = (char *)malloc(field_size = 256)) == (char *)0)
- adios(NULL, "can't allocate field buffer.");
+ field = (char *)mh_xmalloc(field_size = 256);
}
/*
* We're trying to delete a header field (annotation )if the delete flag is
- * non-negative. A value greater than zero means that we're deleting the
+ * not -2 or less. A value greater than zero means that we're deleting the
* nth header field that matches the field (component) name. A value of
* zero means that we're deleting the first field in which both the field
* name matches the component name and the field body matches the text.
* the last slash. This allows matching of both absolute and relative path
* names. This is because this functionality was added to support attachments.
* It might be worth having a separate flag to indicate path name matching to
- * make it more general.
+ * make it more general. A value of -1 means to delete all matching fields.
*/
- if (delete >= 0) {
+ if (delete >= -1) {
/*
* Get the length of the field name since we use it often.
*/
}
if (++n >= field_size - 1) {
- if ((field = (char *)realloc((void *)field, field_size *= 2)) == (char *)0)
- adios(NULL, "can't grow field buffer.");
-
+ field = (char *) mh_xrealloc((void *)field, field_size *= 2);
+
cp = field + n - 1;
}
}
* field, the text does not begin with a / meaning that we're looking for the
* last path name component, and the last path name component matches the text.
*
- * o The delete flag is non-zero meaning that we're going to delete the nth field
+ * o The delete flag is positive meaning that we're going to delete the nth field
* with a matching field name, and this is the nth matching field name.
+ *
+ * o The delete flag is -1 meaning that we're going to delete all fields with a
+ * matching field name.
*/
if (strncasecmp(field, comp, length) == 0 && field[length] == ':') {
}
}
+ else if (delete == -1)
+ continue;
+
else if (++count == delete)
break;
}
else {
/*
* Find the end of the headers before adding the annotations if we're
- * appending instead of the default prepending.
+ * appending instead of the default prepending. A special check for
+ * no headers is needed if appending.
*/
if (append) {
* reach the end of the headers.
*/
- while ((c = getc(fp)) != EOF) {
- (void)putc(c, tmp);
+ if ((c = getc(fp)) == '\n')
+ rewind(fp);
- if (c == '\n') {
- (void)ungetc(c = getc(fp), fp);
+ else {
+ (void)putc(c, tmp);
- if (c == '\n' || c == '-')
- break;
+ while ((c = getc(fp)) != EOF) {
+ (void)putc(c, tmp);
+
+ if (c == '\n') {
+ (void)ungetc(c = getc(fp), fp);
+
+ if (c == '\n' || c == '-')
+ break;
+ }
}
}
}
while (*cp && *cp++ != '\n')
continue;
if (cp - sp)
- fprintf (tmp, "%s: %*.*s", comp, cp - sp, cp - sp, sp);
+ fprintf (tmp, "%s: %*.*s", comp, (int)(cp - sp), (int)(cp - sp), sp);
} while (*cp);
if (cp[-1] != '\n' && cp != text)
putc ('\n', tmp);
* copying routine uses the descriptor, not the pointer.
*/
- if (append || delete >= 0) {
+ if (append || delete >= -1) {
if (lseek(fd, (off_t)ftell(fp), SEEK_SET) == (off_t)-1)
adios(NULL, "can't seek.");
}
* so the file has to be truncated or it will contain garbage.
*/
- if (delete >= 0 && ftruncate(fd, 0) == -1)
+ if (delete >= -1 && ftruncate(fd, 0) == -1)
adios(tmpfil, "unable to truncate.");
cpydata (tmpfd, fd, tmpfil, file);
* lkclose() fail, but that failure is ignored so it's not a problem.
*/
- if (delete >= 0)
+ if (delete >= -1)
(void)fclose(fp);
return 0;