]> git.pond.sub.org Git - empserver/blobdiff - src/lib/update/anno.c
Factor out code to read mailboxes, and make read more robust
[empserver] / src / lib / update / anno.c
index e533b36ba3b7e23384b8ac5ef039f496d8fedbf2..adc765735ccc88c4eda1d372a69fae82fdd1b02d 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2000, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *  Copyright (C) 1986-2009, Dave Pare, Jeff Bailey, Thomas Ruschak,
  *                           Ken Stevens, Steve McClure
  *
  *  This program is free software; you can redistribute it and/or modify
  *
  *  ---
  *
- *  See the "LEGAL", "LICENSE", "CREDITS" and "README" files for all the
- *  related information and legal notices. It is expected that any future
- *  projects/authors will amend these files as needed.
+ *  See files README, COPYING and CREDITS in the root of the source
+ *  tree for related information and legal notices.  It is expected
+ *  that future projects/authors will amend these files as needed.
  *
  *  ---
  *
  *  anno.c: Delete announcements older than ANNO_KEEP_DAYS
- * 
+ *
  *  Known contributors to this file:
  *     Ken Stevens, 1995
  *     Doug Hay, 1998
  *     Steve McClure, 2000
+ *     Ron Koenderink, 2004
  */
 
-#include "misc.h"
-#include "tel.h"
-#include <stdio.h>
-#include <fcntl.h>
-#if !defined(_WIN32)
-#include <sys/file.h>
-#include <unistd.h>
+#include <config.h>
+
+#if defined(_WIN32) && defined(__GNUC__)
+#include <io.h>
 #endif
+#include <stdio.h>
+#include <time.h>
+#include "tel.h"
 #include "update.h"
-#include "optlist.h"
-#include "common.h"
+
+static int copy_and_expire(FILE *annfp, FILE *tmpfp,
+                          char *tmp_filename, time_t expiry_time);
+static int copy_sink(char *, size_t, void *);
 
 void
 delete_old_announcements(void)
 {
     time_t now;
     time_t old;
-    struct telstr tgm;
-    FILE *oldfp;
-    int tmpfd;
-    s_char tmp_filename[1024];
-    int writeit;
-    s_char message[MAXTELSIZE];
-    int deleted = 0;
-    int saved = 0;
-    int length;
-    int nbytes;
-    int first = 1;
+    FILE *annfp;
+    FILE *tmpfp;
+    char tmp_filename[1024];
+    int copy_file;
+
+    if (anno_keep_days < 0)
+       return;
 
     time(&now);
-    old = now - days(ANNO_KEEP_DAYS);
+    old = now - days(anno_keep_days);
     logerror("Deleting annos older than %s", ctime(&old));
 
-#if !defined(_WIN32)
-    if ((oldfp = fopen(annfil, "r+")) == 0) {
-#else
-    if ((oldfp = fopen(annfil, "r+b")) == 0) {
-#endif
-       logerror("can't read telegram file %s", annfil);
+    if ((annfp = fopen(annfil, "rb")) == NULL) {
+       logerror("can't open telegram file %s for reading", annfil);
        return;
     }
     sprintf(tmp_filename, "%s.tmp", annfil);
-#if !defined(_WIN32)
-    if ((tmpfd =
-        open(tmp_filename, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0) {
-#else
-    if ((tmpfd =
-        open(tmp_filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY,
-             0666)) < 0) {
-#endif
-       logerror("can't write telegram file %s", tmp_filename);
+    if ((tmpfp = fopen(tmp_filename, "wb")) == NULL) {
+       logerror("can't open telegram file %s for writing",
+                tmp_filename);
+       if (fclose(annfp) != 0)
+           logerror("can't close telegram file %s", annfil);
        return;
     }
-    while (fread((s_char *)&tgm, sizeof(tgm), 1, oldfp) == 1) {
-       writeit = 1;
-       if (tgm.tel_length < 0) {
-           logerror("bad telegram file header (length)");
-           return;
-       }
-       if (tgm.tel_type < 0 || tgm.tel_type > TEL_LAST) {
-           logerror("bad telegram file header (type)");
-           writeit = 0;
+    copy_file = copy_and_expire(annfp, tmpfp, tmp_filename, old);
+
+    if (fclose(annfp) != 0) {
+       logerror("can't close telegram file %s", annfil);
+       copy_file = 0;
+    }
+    if (fclose(tmpfp) != 0) {
+       logerror("can't close temporary telegram file %s",
+                tmp_filename);
+       copy_file = 0;
+    }
+#if defined(_WIN32)
+    if (copy_file) {
+       if (unlink(annfil) != 0) {
+           logerror("can't delete telegram file %s", annfil);
+           copy_file = 0;
        }
+    }
+#endif
+    if (copy_file) {
+       if (rename(tmp_filename, annfil) != 0)
+           logerror("can't move temporary telegram file %s "
+                    "to telegram file %s", tmp_filename, annfil);
+    } else {
+       if (remove(tmp_filename) < 0)
+           logerror("can't delete telegram file %s", tmp_filename);
+    }
+}
+
+static int
+copy_and_expire(FILE *annfp, FILE *tmpfp, char *tmp_filename,
+               time_t expiry_time)
+{
+    struct telstr tgm;
+    int res, writeit;
+    int deleted = 0;
+    int saved = 0;
+    int first = 1;
+
+    while ((res = tel_read_header(annfp, annfil, &tgm)) > 0) {
+       writeit = tgm.tel_date >= expiry_time;
        if (first) {
            first = 0;
-           if (tgm.tel_date >= old) {
-               fclose(oldfp);
-               return;
-           }
+           if (writeit)
+               return 0;
        }
-
-       if (tgm.tel_date < old) {
-           writeit = 0;
-       }
-
        if (writeit) {
-           if (write(tmpfd, &tgm, sizeof(tgm)) < (int)sizeof(tgm)) {
-               logerror("error writing to ann.tmp");
-               return;
+           if (fwrite(&tgm, sizeof(tgm), 1, tmpfp) != 1) {
+               logerror("error writing header to temporary "
+                        "telegram file %s", tmp_filename);
+               return 0;
            }
            ++saved;
-       } else {
+       } else
            ++deleted;
-       }
-       length = tgm.tel_length;
-       while (length > 0) {
-           nbytes = length;
-           if (nbytes > (int)sizeof(message))
-               nbytes = sizeof(message);
-           (void)fread(message, sizeof(s_char), nbytes, oldfp);
-           if (writeit) {
-               if (write(tmpfd, message, nbytes) < nbytes) {
-                   logerror("Error writing to ann.tmp");
-                   return;
-               }
-           }
-           length -= nbytes;
-       }
+       res = tel_read_body(annfp, annfil, &tgm,
+                           writeit ? copy_sink : NULL, tmpfp);
+       if (res < 0)
+           return 0;
     }
+
+    if (res < 0)
+       return 0;
     logerror("%d announcements deleted; %d announcements saved",
             deleted, saved);
-    fclose(oldfp);
-    close(tmpfd);
-    unlink(annfil);
-    rename(tmp_filename, annfil);
+    return 1;
+}
+
+static int
+copy_sink(char *chunk, size_t sz, void *fp)
+{
+    if (fwrite(chunk, 1, sz, fp) != sz) {
+       logerror("error writing to %s.tmp", annfil);
+       return -1;
+    }
+    return 0;
 }