|
163
|
1 /*
|
|
|
2
|
|
|
3 Copyright (c) 2007 Carl Byington - 510 Software Group, released under
|
|
|
4 the GPL version 3 or any later version at your choice available at
|
|
|
5 http://www.gnu.org/licenses/gpl-3.0.txt
|
|
|
6
|
|
|
7 Based on spamass-milter by Georg C. F. Greve <greve@gnu.org>
|
|
|
8
|
|
|
9 */
|
|
|
10
|
|
|
11 #include "config.h"
|
|
|
12 #include "dnsbl.h"
|
|
|
13 #include <errno.h>
|
|
|
14 #include <fcntl.h>
|
|
|
15 #include <poll.h>
|
|
|
16 #include <signal.h>
|
|
|
17 #include <string>
|
|
|
18 #include <sys/types.h>
|
|
|
19 #include <sys/wait.h>
|
|
|
20 #include <unistd.h>
|
|
|
21
|
|
|
22
|
|
|
23 static const char Id[] = "$Id$";
|
|
|
24
|
|
|
25 bool warnedmacro = false; /* have we logged that we couldn't fetch a macro? */
|
|
|
26 const int maxlen = 1000; // used for snprintf buffers
|
|
|
27
|
|
|
28
|
|
|
29 SpamAssassin::SpamAssassin()
|
|
|
30 {
|
|
|
31 error = false;
|
|
|
32 running = false;
|
|
|
33 first_recipient = true;
|
|
|
34 }
|
|
|
35
|
|
|
36
|
|
|
37 SpamAssassin::~SpamAssassin()
|
|
|
38 {
|
|
|
39 // close all pipes that are still open
|
|
|
40 if (pipe_io[0][0] > -1) close(pipe_io[0][0]);
|
|
|
41 if (pipe_io[0][1] > -1) close(pipe_io[0][1]);
|
|
|
42 if (pipe_io[1][0] > -1) close(pipe_io[1][0]);
|
|
|
43 if (pipe_io[1][1] > -1) close(pipe_io[1][1]);
|
|
|
44 // child still running?
|
|
|
45 if (running) {
|
|
|
46 // make sure the pid is valid
|
|
|
47 if (pid > 0) {
|
|
|
48 // slaughter child
|
|
|
49 kill(pid, SIGKILL);
|
|
|
50 // wait for child to terminate
|
|
|
51 int status;
|
|
|
52 waitpid(pid, &status, 0);
|
|
|
53 }
|
|
|
54 }
|
|
|
55 }
|
|
|
56
|
|
|
57
|
|
|
58 void SpamAssassin::mlfi_connect(mlfiPriv *priv_, int ip)
|
|
|
59 {
|
|
|
60 priv = priv_;
|
|
|
61 ip4 = ip;
|
|
|
62 }
|
|
|
63
|
|
|
64
|
|
|
65 void SpamAssassin::mlfi_helo(char *helohost)
|
|
|
66 {
|
|
|
67 helo = helohost;
|
|
|
68 }
|
|
|
69
|
|
|
70
|
|
|
71 void SpamAssassin::mlfi_envfrom(char *from, char *qid)
|
|
|
72 {
|
|
|
73 envfrom = from;
|
|
|
74 queueid = qid;
|
|
|
75 }
|
|
|
76
|
|
|
77
|
|
|
78 void SpamAssassin::mlfi_envrcpt(SMFICTX *ctx, char *envrcpt)
|
|
|
79 {
|
|
|
80 if (first_recipient) {
|
|
|
81 /* Send the envelope headers as X-Envelope-From: and
|
|
|
82 X-Envelope-To: so that SpamAssassin can use them in its
|
|
|
83 whitelist checks. Also forge as complete a dummy
|
|
|
84 Received: header as possible because SA gets a lot of
|
|
|
85 info from it.
|
|
|
86
|
|
|
87 HReceived: $?sfrom $s $.$?_($?s$|from $.$_)
|
|
|
88 $.$?{auth_type}(authenticated$?{auth_ssf} bits=${auth_ssf}$.)
|
|
|
89 $.by $j ($v/$Z)$?r with $r$. id $i$?{tls_version}
|
|
|
90 (version=${tls_version} cipher=${cipher} bits=${cipher_bits} verify=${verify})$.$?u
|
|
|
91 for $u; $|;
|
|
|
92 $.$b$?g
|
|
|
93 (envelope-from $g)$.
|
|
|
94
|
|
|
95 */
|
|
|
96 const char *macro_b, *macro_i, *macro_j, *macro_r,
|
|
|
97 *macro_s, *macro_v, *macro_Z, *macro__;
|
|
|
98 char date[32];
|
|
|
99 time_t tval;
|
|
|
100 time(&tval);
|
|
|
101 strftime(date, sizeof(date), "%a, %d %b %Y %H:%M:%S %z", localtime(&tval));
|
|
|
102 macro_b = date;
|
|
|
103
|
|
|
104 /* queue ID */
|
|
|
105 macro_i = queueid;
|
|
|
106
|
|
|
107 /* FQDN of this site */
|
|
|
108 macro_j = smfi_getsymval(ctx, "j");
|
|
|
109 if (!macro_j) {
|
|
|
110 macro_j = "localhost";
|
|
|
111 warnmacro("j", "ENVRCPT");
|
|
|
112 }
|
|
|
113
|
|
|
114 /* Protocol used to receive the message */
|
|
|
115 macro_r = smfi_getsymval(ctx, "r");
|
|
|
116 if (!macro_r) {
|
|
|
117 macro_r = "SMTP";
|
|
|
118 warnmacro("r", "ENVRCPT");
|
|
|
119 }
|
|
|
120
|
|
|
121 macro_s = helo;
|
|
|
122
|
|
|
123 /* Sendmail binary version */
|
|
|
124 macro_v = smfi_getsymval(ctx, "v");
|
|
|
125 if (!macro_v) {
|
|
|
126 macro_v = "8.13.0";
|
|
|
127 warnmacro("v", "ENVRCPT");
|
|
|
128 }
|
|
|
129
|
|
|
130 /* Sendmail .cf version */
|
|
|
131 macro_Z = smfi_getsymval(ctx, "Z");
|
|
|
132 if (!macro_Z) {
|
|
|
133 macro_Z = "8.13.0";
|
|
|
134 warnmacro("Z", "ENVRCPT");
|
|
|
135 }
|
|
|
136
|
|
|
137 /* Validated sending site's address */
|
|
|
138 macro__ = smfi_getsymval(ctx, "_");
|
|
|
139 if (!macro__) {
|
|
|
140 macro__ = "unknown";
|
|
|
141 warnmacro("_", "ENVRCPT");
|
|
|
142 }
|
|
|
143
|
|
|
144 output(string("Received: from ") + macro_s + " (" + macro__+ ")\r\n\t" +
|
|
|
145 "by " + macro_j + " (" + macro_v + "/" + macro_Z + ") with " + macro_r + " id " + macro_i + "\r\n\t" +
|
|
|
146 "for " + envfrom + ";\r\n\t" +
|
|
|
147 macro_b + "\r\n");
|
|
|
148
|
|
|
149 output(string("X-Envelope-From: ") + envfrom + "\r\n");
|
|
|
150 }
|
|
|
151 output(string("X-Envelope-To: ") + envrcpt + "\r\n");
|
|
|
152 first_recipient = false;
|
|
|
153 }
|
|
|
154
|
|
|
155
|
|
|
156 void SpamAssassin::mlfi_header(char* headerf, char* headerv)
|
|
|
157 {
|
|
|
158 if (!running) Connect();
|
|
|
159 output(spamc_input);
|
|
|
160 output(headerf);
|
|
|
161 output(": ");
|
|
|
162 output(headerv);
|
|
|
163 output("\r\n");
|
|
|
164 spamc_input.empty();
|
|
|
165 }
|
|
|
166
|
|
|
167
|
|
|
168 void SpamAssassin::mlfi_eoh()
|
|
|
169 {
|
|
|
170 output("\r\n");
|
|
|
171 }
|
|
|
172
|
|
|
173
|
|
|
174 void SpamAssassin::mlfi_body(u_char *bodyp, size_t bodylen)
|
|
|
175 {
|
|
|
176 output((char *)bodyp, bodylen);
|
|
|
177 }
|
|
|
178
|
|
|
179
|
|
|
180 int SpamAssassin::mlfi_eom()
|
|
|
181 {
|
|
|
182 close_output(); // signal EOF to SpamAssassin
|
|
|
183 input(); // read what the Assassin is telling us
|
|
|
184 my_syslog(priv, "spamc returned " + spamc_output);
|
|
|
185 return atoi(spamc_output.c_str());
|
|
|
186 }
|
|
|
187
|
|
|
188
|
|
|
189 void SpamAssassin::Connect()
|
|
|
190 {
|
|
|
191 if (error) return;
|
|
|
192 // set up pipes for in- and output
|
|
|
193 error |= (pipe(pipe_io[0]));
|
|
|
194 error |= (pipe(pipe_io[1]));
|
|
|
195 if (error) return;
|
|
|
196
|
|
|
197 // now execute SpamAssassin client for contact with SpamAssassin spamd
|
|
|
198 // start child process
|
|
|
199 pid = fork();
|
|
|
200 switch (pid) {
|
|
|
201 case -1:
|
|
|
202 // forking trouble.
|
|
|
203 my_syslog(priv, "unable to fork for spamc");
|
|
|
204 error = true;
|
|
|
205 close(pipe_io[0][0]);
|
|
|
206 close(pipe_io[0][1]);
|
|
|
207 close(pipe_io[1][0]);
|
|
|
208 close(pipe_io[1][1]);
|
|
|
209 pipe_io[0][0] = -1;
|
|
|
210 pipe_io[0][1] = -1;
|
|
|
211 pipe_io[1][0] = -1;
|
|
|
212 pipe_io[1][1] = -1;
|
|
|
213 return;
|
|
|
214 case 0:
|
|
|
215 // +++ CHILD +++
|
|
|
216
|
|
|
217 // close unused pipes
|
|
|
218 close(pipe_io[1][0]);
|
|
|
219 close(pipe_io[0][1]);
|
|
|
220
|
|
|
221 // redirect stdin(0), stdout(1) and stderr(2)
|
|
|
222 dup2(pipe_io[0][0],0);
|
|
|
223 dup2(pipe_io[1][1],1);
|
|
|
224 dup2(pipe_io[1][1],2);
|
|
|
225
|
|
|
226 closeall(3);
|
|
|
227
|
|
|
228 // execute spamc
|
|
|
229 char* argv[3];
|
|
|
230 argv[0] = "@SPAMC@";
|
|
|
231 argv[1] = "-c";
|
|
|
232 argv[2] = NULL;
|
|
|
233 execvp(argv[0] , argv); // does not return!
|
|
|
234 _exit(1); // exec failed
|
|
|
235 break;
|
|
|
236 }
|
|
|
237
|
|
|
238 // +++ PARENT +++
|
|
|
239
|
|
|
240 // close unused pipes
|
|
|
241 close(pipe_io[0][0]);
|
|
|
242 close(pipe_io[1][1]);
|
|
|
243 pipe_io[0][0] = -1;
|
|
|
244 pipe_io[1][1] = -1;
|
|
|
245
|
|
|
246 // mark the pipes non-blocking
|
|
|
247 if (fcntl(pipe_io[0][1], F_SETFL, O_NONBLOCK) == -1)
|
|
|
248 error = true;
|
|
|
249 #if 0 /* don't really need to make the sink pipe nonblocking */
|
|
|
250 if (fcntl(pipe_io[1][0], F_SETFL, O_NONBLOCK) == -1)
|
|
|
251 error = true;
|
|
|
252 #endif
|
|
|
253
|
|
|
254 // we have to assume the client is running now.
|
|
|
255 running = true;
|
|
|
256 }
|
|
|
257
|
|
|
258
|
|
|
259 void SpamAssassin::output(const char* buffer, size_t size)
|
|
|
260 {
|
|
|
261 // if there are problems, fail.
|
|
|
262 if (error) return;
|
|
|
263
|
|
|
264 if (!running) {
|
|
|
265 // buffer it
|
|
|
266 spamc_input.append(buffer, size);
|
|
|
267 return;
|
|
|
268 }
|
|
|
269
|
|
|
270 // send to SpamAssassin
|
|
|
271 long total = 0;
|
|
|
272 long wsize = 0;
|
|
|
273 string reason;
|
|
|
274 int status;
|
|
|
275 do {
|
|
|
276 struct pollfd fds[2];
|
|
|
277 int nfds = 2, nready;
|
|
|
278 fds[0].fd = pipe_io[0][1];
|
|
|
279 fds[0].events = POLLOUT;
|
|
|
280 fds[0].revents = 0;
|
|
|
281 fds[1].fd = pipe_io[1][0];
|
|
|
282 fds[1].events = POLLIN;
|
|
|
283 fds[1].revents = 0;
|
|
|
284
|
|
|
285 nready = poll(fds, nfds, 1000);
|
|
|
286 if (nready == -1) {
|
|
|
287 my_syslog(priv, "poll failed");
|
|
|
288 error = true;
|
|
|
289 return;
|
|
|
290 }
|
|
|
291
|
|
|
292 if (fds[1].revents & (POLLERR|POLLNVAL|POLLHUP)) {
|
|
|
293 my_syslog(priv, "poll says my read pipe is busted");
|
|
|
294 error = true;
|
|
|
295 return;
|
|
|
296 }
|
|
|
297
|
|
|
298 if (fds[0].revents & (POLLERR|POLLNVAL|POLLHUP)) {
|
|
|
299 my_syslog(priv, "poll says my write pipe is busted");
|
|
|
300 error = true;
|
|
|
301 return;
|
|
|
302 }
|
|
|
303
|
|
|
304 if (fds[1].revents & POLLIN) {
|
|
|
305 read_pipe();
|
|
|
306 }
|
|
|
307
|
|
|
308 if (fds[0].revents & POLLOUT) {
|
|
|
309 switch(wsize = write(pipe_io[0][1], (char *)buffer + total, size - total)) {
|
|
|
310 case -1:
|
|
|
311 if (errno == EAGAIN) continue;
|
|
|
312 reason = string(strerror(errno));
|
|
|
313 // close the pipes
|
|
|
314 close(pipe_io[0][1]);
|
|
|
315 close(pipe_io[1][0]);
|
|
|
316 pipe_io[0][1] = -1;
|
|
|
317 pipe_io[1][0] = -1;
|
|
|
318 // Slaughter child
|
|
|
319 kill(pid, SIGKILL);
|
|
|
320 // wait for child to terminate
|
|
|
321 waitpid(pid, &status, 0);
|
|
|
322 my_syslog(priv, "write error: " + reason);
|
|
|
323 error = true;
|
|
|
324 running = false;
|
|
|
325 return;
|
|
|
326 default:
|
|
|
327 total += wsize;
|
|
|
328 break;
|
|
|
329 }
|
|
|
330 }
|
|
|
331 } while ( total < size );
|
|
|
332 }
|
|
|
333
|
|
|
334
|
|
|
335 void SpamAssassin::output(const char* buffer)
|
|
|
336 {
|
|
|
337 output(buffer, strlen(buffer));
|
|
|
338 }
|
|
|
339
|
|
|
340
|
|
|
341 void SpamAssassin::output(string buffer)
|
|
|
342 {
|
|
|
343 output(buffer.c_str(), buffer.size());
|
|
|
344 }
|
|
|
345
|
|
|
346
|
|
|
347 void SpamAssassin::close_output()
|
|
|
348 {
|
|
|
349 if (close(pipe_io[0][1]))
|
|
|
350 my_syslog(priv, "close error: " + string(strerror(errno)));
|
|
|
351 pipe_io[0][1] = -1;
|
|
|
352 }
|
|
|
353
|
|
|
354
|
|
|
355 void SpamAssassin::input()
|
|
|
356 {
|
|
|
357 if (!running || error) return;
|
|
|
358 empty_and_close_pipe();
|
|
|
359 if (running) {
|
|
|
360 // wait until child is dead
|
|
|
361 int status;
|
|
|
362 if (waitpid(pid, &status, 0) < 0) {
|
|
|
363 error = true;
|
|
|
364 };
|
|
|
365 }
|
|
|
366 running = false;
|
|
|
367 }
|
|
|
368
|
|
|
369
|
|
|
370 int SpamAssassin::read_pipe()
|
|
|
371 {
|
|
|
372 long size;
|
|
|
373 int status;
|
|
|
374 char iobuff[1024];
|
|
|
375 string reason;
|
|
|
376
|
|
|
377 if (pipe_io[1][0] == -1) return 0;
|
|
|
378
|
|
|
379 size = read(pipe_io[1][0], iobuff, 1024);
|
|
|
380
|
|
|
381 if (size < 0) {
|
|
|
382 reason = string(strerror(errno));
|
|
|
383 // Close remaining pipe.
|
|
|
384 close(pipe_io[1][0]);
|
|
|
385 pipe_io[1][0] = -1;
|
|
|
386 // Slaughter child
|
|
|
387 kill(pid, SIGKILL);
|
|
|
388 // wait for child to terminate
|
|
|
389 waitpid(pid, &status, 0);
|
|
|
390 my_syslog(priv, "read error: " + reason);
|
|
|
391 size = 0;
|
|
|
392 error = true;
|
|
|
393 running = false;
|
|
|
394 } else if (size == 0) {
|
|
|
395 // EOF. Close the pipe
|
|
|
396 if (close(pipe_io[1][0])) {
|
|
|
397 error = true;
|
|
|
398 my_syslog(priv, "close error: " + string(strerror(errno)));
|
|
|
399 }
|
|
|
400 pipe_io[1][0] = -1;
|
|
|
401 } else {
|
|
|
402 // append to mail buffer
|
|
|
403 spamc_output.append(iobuff, size);
|
|
|
404 }
|
|
|
405 return size;
|
|
|
406 }
|
|
|
407
|
|
|
408
|
|
|
409 void SpamAssassin::empty_and_close_pipe()
|
|
|
410 {
|
|
|
411 while (read_pipe())
|
|
|
412 ;
|
|
|
413 }
|
|
|
414
|
|
|
415
|
|
|
416 void SpamAssassin::closeall(int fd)
|
|
|
417 {
|
|
|
418 int fdlimit = sysconf(_SC_OPEN_MAX);
|
|
|
419 while (fd < fdlimit)
|
|
|
420 close(fd++);
|
|
|
421 }
|
|
|
422
|
|
|
423
|
|
|
424 void SpamAssassin::warnmacro(char *macro, char *scope)
|
|
|
425 {
|
|
|
426 if (warnedmacro) return;
|
|
|
427 char buf[maxlen];
|
|
|
428 snprintf(buf, sizeof(buf), "Could not retrieve sendmail macro %s. Add it to confMILTER_MACROS_%s for better results.", macro, scope);
|
|
|
429 my_syslog(priv, buf);
|
|
|
430 warnedmacro = true;
|
|
|
431 }
|
|
|
432
|
