git.marmaro.de Git - mmh/blob - uip/distsbr.c

   1 /*
   2 ** distsbr.c -- routines to do additional "dist-style" processing
   3 **
   4 ** This code is Copyright (c) 2002, by the authors of nmh.  See the
   5 ** COPYRIGHT file in the root directory of the nmh distribution for
   6 ** complete copyright information.
   7 */
   8
   9 #include <h/mh.h>
  10 #include <fcntl.h>
  11 #include <h/utils.h>
  12
  13 static int  hdrfd = NOTOK;
  14 static int  txtfd = NOTOK;
  15
  16 /*
  17 ** static prototypes
  18 */
  19 static void ready_msg(char *);
  20
  21 int
  22 distout(char *drft, char *msgnam, char *backup)
  23 {
  24         int state;
  25         register unsigned char *dp;
  26         register char *resent;
  27         char name[NAMESZ], buffer[BUFSIZ];
  28         register FILE *ifp, *ofp;
  29
  30         strcpy(backup, m_mktemp(toabsdir(invo_name), NULL, NULL));
  31         if (rename(drft, backup) == NOTOK) {
  32                 adios(backup, "unable to rename %s to",drft);
  33         }
  34         if (!(ifp = fopen(backup, "r"))) {
  35                 adios(backup, "unable to read");
  36         }
  37
  38         if (!(ofp = fopen(drft, "w"))) {
  39                 adios(drft, "unable to create temporary file");
  40         }
  41         chmod(drft, m_gmprot());
  42
  43         ready_msg(msgnam);
  44         lseek(hdrfd, (off_t) 0, SEEK_SET); /* msgnam not accurate */
  45         cpydata(hdrfd, fileno(ofp), msgnam, drft);
  46
  47         state = FLD;
  48         resent = NULL;
  49         while (1) {
  50                 switch (state = m_getfld(state, name, buffer, sizeof buffer,
  51                                 ifp)) {
  52                 case FLD:
  53                 case FLDPLUS:
  54                 case FLDEOF:
  55                         if (!uprf(name, "resent")) {
  56                                 advise(NULL, "Please re-edit draft to remove the ``%s'' header.", name);
  57                                 goto leave_bad;
  58                         }
  59                         if (state == FLD) {
  60                                 resent = add(":", add(name, resent));
  61                         }
  62                         resent = add(buffer, resent);
  63                         fprintf(ofp, "%s: %s", name, buffer);
  64                         while (state == FLDPLUS) {
  65                                 state = m_getfld(state, name, buffer,
  66                                                 sizeof buffer, ifp);
  67                                 resent = add(buffer, resent);
  68                                 fputs(buffer, ofp);
  69                         }
  70                         if (state == FLDEOF) {
  71                                 goto process;
  72                         }
  73                         break;
  74
  75                 case BODY:
  76                 case BODYEOF:
  77                         for (dp = buffer; *dp; dp++) {
  78                                 if (!isspace(*dp)) {
  79                                         advise(NULL, "Please re-edit draft to consist of headers only.");
  80                                         goto leave_bad;
  81                                 }
  82                         }
  83
  84                 case FILEEOF:
  85                         goto process;
  86
  87                 case LENERR:
  88                 case FMTERR:
  89                         advise(NULL, "Please re-edit draft and fix that header.");
  90 leave_bad: ;
  91                         fclose(ifp);
  92                         fclose(ofp);
  93                         unlink(drft);
  94                         if (rename(backup, drft) == NOTOK) {
  95                                 adios(drft, "unable to rename %s to", backup);
  96                         }
  97                         return NOTOK;
  98
  99                 default:
 100                         adios(NULL, "getfld() returned %d", state);
 101                 }
 102         }
 103
 104 process: ;
 105         fclose(ifp);
 106         fflush(ofp);
 107
 108         if (!resent) {
 109                 advise(NULL, "Please re-edit draft to include a ``Resent-To:'' header.");
 110                 fclose(ofp);
 111                 unlink(drft);
 112                 if (rename(backup, drft) == NOTOK) {
 113                         adios(drft, "unable to rename %s to", backup);
 114                 }
 115                 return NOTOK;
 116         }
 117         free(resent);
 118
 119         if (txtfd != NOTOK) {
 120                 lseek(txtfd, (off_t) 0, SEEK_SET);  /* msgnam not accurate */
 121                 cpydata(txtfd, fileno(ofp), msgnam, drft);
 122         }
 123
 124         fclose(ofp);
 125
 126         return OK;
 127 }
 128
 129
 130 static void
 131 ready_msg(char *msgnam)
 132 {
 133         int state, out;
 134         char name[NAMESZ], buffer[BUFSIZ], tmpfil[BUFSIZ];
 135         register FILE *ifp, *ofp;
 136         char *cp = NULL;
 137
 138         if (hdrfd != NOTOK) {
 139                 close(hdrfd);
 140                 hdrfd = NOTOK;
 141         }
 142         if (txtfd != NOTOK) {
 143                 close(txtfd);
 144                 txtfd = NOTOK;
 145         }
 146         if (!(ifp = fopen(msgnam, "r"))) {
 147                 adios(msgnam, "unable to open message");
 148         }
 149
 150         cp = m_mktemp2(NULL, "dist", &hdrfd, NULL);
 151         if (!cp) {
 152                 adios("distsbr", "unable to create temporary file");
 153         }
 154         fchmod(hdrfd, 0600);
 155         strncpy(tmpfil, cp, sizeof(tmpfil));
 156         if ((out = dup(hdrfd)) == NOTOK || !(ofp = fdopen(out, "w"))) {
 157                 adios(NULL, "no file descriptors -- you lose big");
 158         }
 159         unlink(tmpfil);
 160
 161         state = FLD;
 162         while (1) {
 163                 state = m_getfld(state, name, buffer, sizeof buffer, ifp);
 164                 switch (state) {
 165                 case FLD:
 166                 case FLDPLUS:
 167                 case FLDEOF:
 168                         if (uprf(name, "resent")) {
 169                                 fprintf(ofp, "Prev-");
 170                         }
 171                         fprintf(ofp, "%s: %s", name, buffer);
 172                         while (state == FLDPLUS) {
 173                                 state = m_getfld(state, name, buffer,
 174                                                 sizeof buffer, ifp);
 175                                 fputs(buffer, ofp);
 176                         }
 177                         if (state == FLDEOF) {
 178                                 goto process;
 179                         }
 180                         break;
 181
 182                 case BODY:
 183                 case BODYEOF:
 184                         fclose(ofp);
 185
 186                         cp = m_mktemp2(NULL, "dist", &txtfd, NULL);
 187                         if (!cp) {
 188                                 adios("distsbr", "unable to create temp file");
 189                         }
 190                         fchmod(txtfd, 0600);
 191                         strncpy(tmpfil, cp, sizeof(tmpfil));
 192                         if ((out = dup(txtfd)) == NOTOK ||
 193                                         !(ofp = fdopen(out, "w"))) {
 194                                 adios(NULL, "no file descriptors -- you lose");
 195                         }
 196                         unlink(tmpfil);
 197                         fprintf(ofp, "\n%s", buffer);
 198                         while (state == BODY) {
 199                                 state = m_getfld(state, name, buffer,
 200                                                 sizeof buffer, ifp);
 201                                 fputs(buffer, ofp);
 202                         }
 203                 case FILEEOF:
 204                         goto process;
 205
 206                 case LENERR:
 207                 case FMTERR:
 208                         adios(NULL, "format error in message %s", msgnam);
 209
 210                 default:
 211                         adios(NULL, "getfld() returned %d", state);
 212                 }
 213         }
 214 process: ;
 215         fclose(ifp);
 216         fclose(ofp);
 217 }