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