Commit | Line | Data |
---|---|---|
8523533c TK |
1 | /************************************************* |
2 | * Exim - an Internet mail transport agent * | |
3 | *************************************************/ | |
4 | ||
3386088d | 5 | /* Copyright (c) Tom Kistner <tom@duncanthrax.net> 2003 - 2015 */ |
8523533c TK |
6 | /* License: GPL */ |
7 | ||
8 | /* Code for calling spamassassin's spamd. Called from acl.c. */ | |
9 | ||
10 | #include "exim.h" | |
11 | #ifdef WITH_CONTENT_SCAN | |
12 | #include "spam.h" | |
13 | ||
14 | uschar spam_score_buffer[16]; | |
15 | uschar spam_score_int_buffer[16]; | |
16 | uschar spam_bar_buffer[128]; | |
c5f280e2 | 17 | uschar spam_action_buffer[32]; |
8523533c TK |
18 | uschar spam_report_buffer[32600]; |
19 | uschar prev_user_name[128] = ""; | |
20 | int spam_ok = 0; | |
21 | int spam_rc = 0; | |
f7274286 | 22 | uschar *prev_spamd_address_work = NULL; |
8523533c | 23 | |
fd4d8871 | 24 | static const uschar * loglabel = US"spam acl condition:"; |
8523533c | 25 | |
23763898 | 26 | |
fd4d8871 R |
27 | static int |
28 | spamd_param_init(spamd_address_container *spamd) | |
29 | { | |
dc7b3d36 | 30 | /* default spamd server weight, time and priority value */ |
e718bd62 | 31 | spamd->is_rspamd = FALSE; |
fd4d8871 | 32 | spamd->is_failed = FALSE; |
8a512ed5 JH |
33 | spamd->weight = SPAMD_WEIGHT; |
34 | spamd->timeout = SPAMD_TIMEOUT; | |
35 | spamd->retry = 0; | |
dc7b3d36 | 36 | spamd->priority = 1; |
fd4d8871 R |
37 | return 0; |
38 | } | |
8523533c | 39 | |
8523533c | 40 | |
fd4d8871 | 41 | static int |
fc362fc5 | 42 | spamd_param(const uschar * param, spamd_address_container * spamd) |
fd4d8871 R |
43 | { |
44 | static int timesinceday = -1; | |
23763898 | 45 | const uschar * s; |
8a512ed5 | 46 | const uschar * name; |
fd4d8871 | 47 | |
fd4d8871 R |
48 | /*XXX more clever parsing could discard embedded spaces? */ |
49 | ||
fc362fc5 | 50 | if (sscanf(CCS param, "pri=%u", &spamd->priority)) |
dc7b3d36 JH |
51 | return 0; /* OK */ |
52 | ||
fc362fc5 | 53 | if (sscanf(CCS param, "weight=%u", &spamd->weight)) |
fd4d8871 R |
54 | { |
55 | if (spamd->weight == 0) /* this server disabled: skip it */ | |
56 | return 1; | |
57 | return 0; /* OK */ | |
58 | } | |
59 | ||
23763898 | 60 | if (Ustrncmp(param, "time=", 5) == 0) |
fd4d8871 R |
61 | { |
62 | unsigned int start_h = 0, start_m = 0, start_s = 0; | |
63 | unsigned int end_h = 24, end_m = 0, end_s = 0; | |
64 | unsigned int time_start, time_end; | |
23763898 | 65 | const uschar * end_string; |
fd4d8871 | 66 | |
8a512ed5 | 67 | name = US"time"; |
23763898 JH |
68 | s = param+5; |
69 | if ((end_string = Ustrchr(s, '-'))) | |
ddcf2b5f | 70 | { |
23763898 JH |
71 | end_string++; |
72 | if ( sscanf(CS end_string, "%u.%u.%u", &end_h, &end_m, &end_s) == 0 | |
73 | || sscanf(CS s, "%u.%u.%u", &start_h, &start_m, &start_s) == 0 | |
74 | ) | |
8a512ed5 | 75 | goto badval; |
f7274286 | 76 | } |
f7274286 | 77 | else |
8a512ed5 | 78 | goto badval; |
8523533c | 79 | |
fd4d8871 | 80 | if (timesinceday < 0) |
ddcf2b5f | 81 | { |
fd4d8871 R |
82 | time_t now = time(NULL); |
83 | struct tm *tmp = localtime(&now); | |
84 | timesinceday = tmp->tm_hour*3600 + tmp->tm_min*60 + tmp->tm_sec; | |
85 | } | |
8e669ac1 | 86 | |
fd4d8871 R |
87 | time_start = start_h*3600 + start_m*60 + start_s; |
88 | time_end = end_h*3600 + end_m*60 + end_s; | |
8e669ac1 | 89 | |
fd4d8871 R |
90 | if (timesinceday < time_start || timesinceday >= time_end) |
91 | return 1; /* skip spamd server */ | |
c5f280e2 | 92 | |
fd4d8871 R |
93 | return 0; /* OK */ |
94 | } | |
8e669ac1 | 95 | |
fd4d8871 R |
96 | if (Ustrcmp(param, "variant=rspamd") == 0) |
97 | { | |
98 | spamd->is_rspamd = TRUE; | |
99 | return 0; | |
100 | } | |
8e669ac1 | 101 | |
23763898 JH |
102 | if (Ustrncmp(param, "tmo=", 4) == 0) |
103 | { | |
104 | int sec = readconf_readtime((s = param+4), '\0', FALSE); | |
8a512ed5 | 105 | name = US"timeout"; |
23763898 | 106 | if (sec < 0) |
8a512ed5 JH |
107 | goto badval; |
108 | spamd->timeout = sec; | |
109 | return 0; | |
110 | } | |
111 | ||
112 | if (Ustrncmp(param, "retry=", 6) == 0) | |
113 | { | |
114 | int sec = readconf_readtime((s = param+6), '\0', FALSE); | |
115 | name = US"retry"; | |
116 | if (sec < 0) | |
117 | goto badval; | |
118 | spamd->retry = sec; | |
23763898 JH |
119 | return 0; |
120 | } | |
121 | ||
fd4d8871 R |
122 | log_write(0, LOG_MAIN, "%s warning - invalid spamd parameter: '%s'", |
123 | loglabel, param); | |
124 | return -1; /* syntax error */ | |
8a512ed5 JH |
125 | |
126 | badval: | |
127 | log_write(0, LOG_MAIN, | |
128 | "%s warning - invalid spamd %s value: '%s'", loglabel, name, s); | |
129 | return -1; /* syntax error */ | |
fd4d8871 | 130 | } |
8523533c | 131 | |
8523533c | 132 | |
fd4d8871 | 133 | static int |
dc7b3d36 | 134 | spamd_get_server(spamd_address_container ** spamds, int num_servers) |
fd4d8871 R |
135 | { |
136 | unsigned int i; | |
dc7b3d36 JH |
137 | spamd_address_container * sd; |
138 | long rnd, weights; | |
139 | unsigned pri; | |
140 | static BOOL srandomed = FALSE; | |
fd4d8871 | 141 | |
dc7b3d36 JH |
142 | /* seedup, if we have only 1 server */ |
143 | if (num_servers == 1) | |
144 | return (spamds[0]->is_failed ? -1 : 0); | |
8e669ac1 | 145 | |
dc7b3d36 JH |
146 | /* init ranmod */ |
147 | if (!srandomed) | |
148 | { | |
149 | struct timeval tv; | |
150 | gettimeofday(&tv, NULL); | |
151 | srandom((unsigned int)(tv.tv_usec/1000)); | |
152 | srandomed = TRUE; | |
153 | } | |
8e669ac1 | 154 | |
dc7b3d36 JH |
155 | /* scan for highest pri */ |
156 | for (pri = 0, i = 0; i < num_servers; i++) | |
157 | { | |
158 | sd = spamds[i]; | |
159 | if (!sd->is_failed && sd->priority > pri) pri = sd->priority; | |
fd4d8871 | 160 | } |
8e669ac1 | 161 | |
dc7b3d36 JH |
162 | /* get sum of weights */ |
163 | for (weights = 0, i = 0; i < num_servers; i++) | |
164 | { | |
165 | sd = spamds[i]; | |
166 | if (!sd->is_failed && sd->priority == pri) weights += sd->weight; | |
167 | } | |
168 | if (weights == 0) /* all servers failed */ | |
169 | return -1; | |
8e669ac1 | 170 | |
dc7b3d36 JH |
171 | for (rnd = random() % weights, i = 0; i < num_servers; i++) |
172 | { | |
173 | sd = spamds[i]; | |
174 | if (!sd->is_failed && sd->priority == pri) | |
175 | if ((rnd -= sd->weight) <= 0) | |
fd4d8871 | 176 | return i; |
dc7b3d36 | 177 | } |
29cfeb94 | 178 | |
fd4d8871 R |
179 | log_write(0, LOG_MAIN|LOG_PANIC, |
180 | "%s unknown error (memory/cpu corruption?)", loglabel); | |
181 | return -1; | |
182 | } | |
29cfeb94 | 183 | |
29cfeb94 | 184 | |
fd4d8871 | 185 | int |
55414b25 | 186 | spam(const uschar **listptr) |
fd4d8871 R |
187 | { |
188 | int sep = 0; | |
55414b25 | 189 | const uschar *list = *listptr; |
fd4d8871 R |
190 | uschar *user_name; |
191 | uschar user_name_buffer[128]; | |
192 | unsigned long mbox_size; | |
193 | FILE *mbox_file; | |
194 | int spamd_sock = -1; | |
195 | uschar spamd_buffer[32600]; | |
196 | int i, j, offset, result; | |
fd4d8871 R |
197 | uschar spamd_version[8]; |
198 | uschar spamd_short_result[8]; | |
199 | uschar spamd_score_char; | |
200 | double spamd_threshold, spamd_score, spamd_reject_score; | |
201 | int spamd_report_offset; | |
202 | uschar *p,*q; | |
203 | int override = 0; | |
204 | time_t start; | |
205 | size_t read, wrote; | |
fd4d8871 R |
206 | #ifndef NO_POLL_H |
207 | struct pollfd pollfd; | |
208 | #else /* Patch posted by Erik ? for OS X */ | |
209 | struct timeval select_tv; /* and applied by PH */ | |
210 | fd_set select_fd; | |
211 | #endif | |
212 | uschar *spamd_address_work; | |
8a512ed5 | 213 | spamd_address_container * sd; |
fd4d8871 R |
214 | |
215 | /* stop compiler warning */ | |
216 | result = 0; | |
217 | ||
218 | /* find the username from the option list */ | |
219 | if ((user_name = string_nextinlist(&list, &sep, | |
220 | user_name_buffer, | |
221 | sizeof(user_name_buffer))) == NULL) | |
222 | { | |
223 | /* no username given, this means no scanning should be done */ | |
224 | return FAIL; | |
225 | } | |
226 | ||
227 | /* if username is "0" or "false", do not scan */ | |
228 | if ( (Ustrcmp(user_name,"0") == 0) || | |
229 | (strcmpic(user_name,US"false") == 0) ) | |
230 | return FAIL; | |
231 | ||
232 | /* if there is an additional option, check if it is "true" */ | |
233 | if (strcmpic(list,US"true") == 0) | |
234 | /* in that case, always return true later */ | |
235 | override = 1; | |
236 | ||
237 | /* expand spamd_address if needed */ | |
238 | if (*spamd_address == '$') | |
239 | { | |
240 | spamd_address_work = expand_string(spamd_address); | |
241 | if (spamd_address_work == NULL) | |
242 | { | |
243 | log_write(0, LOG_MAIN|LOG_PANIC, | |
244 | "%s spamd_address starts with $, but expansion failed: %s", | |
245 | loglabel, expand_string_message); | |
246 | return DEFER; | |
29cfeb94 | 247 | } |
fd4d8871 R |
248 | } |
249 | else | |
250 | spamd_address_work = spamd_address; | |
251 | ||
dc7b3d36 | 252 | DEBUG(D_acl) debug_printf("spamd: addrlist '%s'\n", spamd_address_work); |
fd4d8871 R |
253 | |
254 | /* check if previous spamd_address was expanded and has changed. dump cached results if so */ | |
255 | if ( spam_ok | |
256 | && prev_spamd_address_work != NULL | |
257 | && Ustrcmp(prev_spamd_address_work, spamd_address_work) != 0 | |
258 | ) | |
259 | spam_ok = 0; | |
260 | ||
261 | /* if we scanned for this username last time, just return */ | |
262 | if (spam_ok && Ustrcmp(prev_user_name, user_name) == 0) | |
263 | return override ? OK : spam_rc; | |
264 | ||
265 | /* make sure the eml mbox file is spooled up */ | |
266 | mbox_file = spool_mbox(&mbox_size, NULL); | |
267 | ||
268 | if (mbox_file == NULL) | |
269 | { | |
270 | /* error while spooling */ | |
271 | log_write(0, LOG_MAIN|LOG_PANIC, | |
272 | "%s error while creating mbox spool file", loglabel); | |
273 | return DEFER; | |
274 | } | |
275 | ||
276 | start = time(NULL); | |
277 | ||
278 | { | |
279 | int num_servers = 0; | |
280 | int current_server; | |
dc7b3d36 JH |
281 | uschar * address; |
282 | const uschar * spamd_address_list_ptr = spamd_address_work; | |
fd4d8871 | 283 | spamd_address_container * spamd_address_vector[32]; |
fd4d8871 R |
284 | |
285 | /* Check how many spamd servers we have | |
286 | and register their addresses */ | |
dc7b3d36 | 287 | sep = 0; /* default colon-sep */ |
fd4d8871 R |
288 | while ((address = string_nextinlist(&spamd_address_list_ptr, &sep, |
289 | NULL, 0)) != NULL) | |
ddcf2b5f | 290 | { |
55414b25 | 291 | const uschar * sublist; |
fd4d8871 R |
292 | int sublist_sep = -(int)' '; /* default space-sep */ |
293 | unsigned args; | |
294 | uschar * s; | |
fd4d8871 | 295 | |
dc7b3d36 | 296 | DEBUG(D_acl) debug_printf("spamd: addr entry '%s'\n", address); |
2aad5761 | 297 | sd = (spamd_address_container *)store_get(sizeof(spamd_address_container)); |
fd4d8871 | 298 | |
2aad5761 | 299 | for (sublist = address, args = 0, spamd_param_init(sd); |
755762fd | 300 | (s = string_nextinlist(&sublist, &sublist_sep, NULL, 0)); |
fd4d8871 R |
301 | args++ |
302 | ) | |
ddcf2b5f | 303 | { |
dc7b3d36 | 304 | DEBUG(D_acl) debug_printf("spamd: addr parm '%s'\n", s); |
fd4d8871 R |
305 | switch (args) |
306 | { | |
2aad5761 | 307 | case 0: sd->hostspec = s; |
fd4d8871 R |
308 | if (*s == '/') args++; /* local; no port */ |
309 | break; | |
2aad5761 JH |
310 | case 1: sd->hostspec = string_sprintf("%s %s", sd->hostspec, s); |
311 | break; | |
312 | default: spamd_param(s, sd); | |
313 | break; | |
fd4d8871 | 314 | } |
ddcf2b5f | 315 | } |
fd4d8871 | 316 | if (args < 2) |
c5f280e2 | 317 | { |
fd4d8871 R |
318 | log_write(0, LOG_MAIN, |
319 | "%s warning - invalid spamd address: '%s'", loglabel, address); | |
320 | continue; | |
c5f280e2 | 321 | } |
8523533c | 322 | |
2aad5761 | 323 | spamd_address_vector[num_servers] = sd; |
fd4d8871 R |
324 | if (++num_servers > 31) |
325 | break; | |
8523533c TK |
326 | } |
327 | ||
fd4d8871 R |
328 | /* check if we have at least one server */ |
329 | if (!num_servers) | |
ddcf2b5f | 330 | { |
0f501486 | 331 | log_write(0, LOG_MAIN|LOG_PANIC, |
fd4d8871 R |
332 | "%s no useable spamd server addresses in spamd_address configuration option.", |
333 | loglabel); | |
8acbb134 | 334 | goto defer; |
ddcf2b5f | 335 | } |
0f501486 | 336 | |
8a512ed5 JH |
337 | current_server = spamd_get_server(spamd_address_vector, num_servers); |
338 | sd = spamd_address_vector[current_server]; | |
339 | for(;;) | |
ddcf2b5f | 340 | { |
2aad5761 | 341 | uschar * errstr; |
8523533c | 342 | |
35deab6a | 343 | DEBUG(D_acl) debug_printf("spamd: trying server %s\n", sd->hostspec); |
fd4d8871 | 344 | |
8a512ed5 JH |
345 | for (;;) |
346 | { | |
347 | if ( (spamd_sock = ip_streamsocket(sd->hostspec, &errstr, 5)) >= 0 | |
348 | || sd->retry <= 0 | |
349 | ) | |
350 | break; | |
0e89d648 | 351 | DEBUG(D_acl) debug_printf("spamd: server %s: retry conn\n", sd->hostspec); |
8a512ed5 JH |
352 | while (sd->retry > 0) sd->retry = sleep(sd->retry); |
353 | } | |
354 | if (spamd_sock >= 0) | |
2aad5761 | 355 | break; |
25257489 | 356 | |
2aad5761 JH |
357 | log_write(0, LOG_MAIN, "%s spamd: %s", loglabel, errstr); |
358 | sd->is_failed = TRUE; | |
25257489 | 359 | |
2aad5761 JH |
360 | current_server = spamd_get_server(spamd_address_vector, num_servers); |
361 | if (current_server < 0) | |
fd4d8871 | 362 | { |
8a512ed5 | 363 | log_write(0, LOG_MAIN|LOG_PANIC, "%s all spamd servers failed", loglabel); |
2aad5761 | 364 | goto defer; |
fd4d8871 | 365 | } |
8a512ed5 | 366 | sd = spamd_address_vector[current_server]; |
ddcf2b5f | 367 | } |
fd4d8871 | 368 | } |
8523533c | 369 | |
fd4d8871 R |
370 | if (spamd_sock == -1) |
371 | { | |
372 | log_write(0, LOG_MAIN|LOG_PANIC, | |
373 | "programming fault, spamd_sock unexpectedly unset"); | |
8acbb134 | 374 | goto defer; |
fd4d8871 R |
375 | } |
376 | ||
377 | (void)fcntl(spamd_sock, F_SETFL, O_NONBLOCK); | |
378 | /* now we are connected to spamd on spamd_sock */ | |
8a512ed5 | 379 | if (sd->is_rspamd) |
fd4d8871 R |
380 | { /* rspamd variant */ |
381 | uschar *req_str; | |
6c54be64 AL |
382 | const uschar * helo; |
383 | const uschar * fcrdns; | |
384 | const uschar * authid; | |
fd4d8871 R |
385 | |
386 | req_str = string_sprintf("CHECK RSPAMC/1.3\r\nContent-length: %lu\r\n" | |
35deab6a JH |
387 | "Queue-Id: %s\r\nFrom: <%s>\r\nRecipient-Number: %d\r\n", |
388 | mbox_size, message_id, sender_address, recipients_count); | |
fd4d8871 R |
389 | for (i = 0; i < recipients_count; i ++) |
390 | req_str = string_sprintf("%sRcpt: <%s>\r\n", req_str, recipients_list[i].address); | |
391 | if ((helo = expand_string(US"$sender_helo_name")) != NULL && *helo != '\0') | |
392 | req_str = string_sprintf("%sHelo: %s\r\n", req_str, helo); | |
393 | if ((fcrdns = expand_string(US"$sender_host_name")) != NULL && *fcrdns != '\0') | |
394 | req_str = string_sprintf("%sHostname: %s\r\n", req_str, fcrdns); | |
395 | if (sender_host_address != NULL) | |
396 | req_str = string_sprintf("%sIP: %s\r\n", req_str, sender_host_address); | |
6c54be64 AL |
397 | if ((authid = expand_string(US"$authenticated_id")) != NULL && *authid != '\0') |
398 | req_str = string_sprintf("%sUser: %s\r\n", req_str, authid); | |
fd4d8871 | 399 | req_str = string_sprintf("%s\r\n", req_str); |
0e89d648 | 400 | wrote = send(spamd_sock, req_str, Ustrlen(req_str), 0); |
fd4d8871 | 401 | } |
6c54be64 | 402 | else |
fd4d8871 R |
403 | { /* spamassassin variant */ |
404 | (void)string_format(spamd_buffer, | |
405 | sizeof(spamd_buffer), | |
406 | "REPORT SPAMC/1.2\r\nUser: %s\r\nContent-length: %ld\r\n\r\n", | |
407 | user_name, | |
408 | mbox_size); | |
409 | /* send our request */ | |
410 | wrote = send(spamd_sock, spamd_buffer, Ustrlen(spamd_buffer), 0); | |
411 | } | |
6c54be64 | 412 | |
fd4d8871 R |
413 | if (wrote == -1) |
414 | { | |
415 | (void)close(spamd_sock); | |
416 | log_write(0, LOG_MAIN|LOG_PANIC, | |
417 | "%s spamd send failed: %s", loglabel, strerror(errno)); | |
8acbb134 | 418 | goto defer; |
fd4d8871 R |
419 | } |
420 | ||
421 | /* now send the file */ | |
422 | /* spamd sometimes accepts conections but doesn't read data off | |
423 | * the connection. We make the file descriptor non-blocking so | |
424 | * that the write will only write sufficient data without blocking | |
425 | * and we poll the desciptor to make sure that we can write without | |
426 | * blocking. Short writes are gracefully handled and if the whole | |
427 | * trasaction takes too long it is aborted. | |
428 | * Note: poll() is not supported in OSX 10.2 and is reported to be | |
429 | * broken in more recent versions (up to 10.4). | |
430 | */ | |
431 | #ifndef NO_POLL_H | |
432 | pollfd.fd = spamd_sock; | |
433 | pollfd.events = POLLOUT; | |
434 | #endif | |
435 | (void)fcntl(spamd_sock, F_SETFL, O_NONBLOCK); | |
436 | do | |
437 | { | |
438 | read = fread(spamd_buffer,1,sizeof(spamd_buffer),mbox_file); | |
439 | if (read > 0) | |
ddcf2b5f | 440 | { |
fd4d8871 R |
441 | offset = 0; |
442 | again: | |
443 | #ifndef NO_POLL_H | |
444 | result = poll(&pollfd, 1, 1000); | |
8523533c | 445 | |
fd4d8871 R |
446 | /* Patch posted by Erik ? for OS X and applied by PH */ |
447 | #else | |
448 | select_tv.tv_sec = 1; | |
449 | select_tv.tv_usec = 0; | |
450 | FD_ZERO(&select_fd); | |
451 | FD_SET(spamd_sock, &select_fd); | |
452 | result = select(spamd_sock+1, NULL, &select_fd, NULL, &select_tv); | |
453 | #endif | |
454 | /* End Erik's patch */ | |
8523533c | 455 | |
fd4d8871 R |
456 | if (result == -1 && errno == EINTR) |
457 | goto again; | |
458 | else if (result < 1) | |
c5f280e2 | 459 | { |
fd4d8871 R |
460 | if (result == -1) |
461 | log_write(0, LOG_MAIN|LOG_PANIC, | |
462 | "%s %s on spamd socket", loglabel, strerror(errno)); | |
463 | else | |
464 | { | |
8a512ed5 | 465 | if (time(NULL) - start < sd->timeout) |
fd4d8871 R |
466 | goto again; |
467 | log_write(0, LOG_MAIN|LOG_PANIC, | |
468 | "%s timed out writing spamd socket", loglabel); | |
469 | } | |
470 | (void)close(spamd_sock); | |
8acbb134 | 471 | goto defer; |
c5f280e2 | 472 | } |
8e669ac1 | 473 | |
fd4d8871 R |
474 | wrote = send(spamd_sock,spamd_buffer + offset,read - offset,0); |
475 | if (wrote == -1) | |
c5f280e2 | 476 | { |
fd4d8871 R |
477 | log_write(0, LOG_MAIN|LOG_PANIC, |
478 | "%s %s on spamd socket", loglabel, strerror(errno)); | |
479 | (void)close(spamd_sock); | |
8acbb134 | 480 | goto defer; |
c5f280e2 | 481 | } |
fd4d8871 | 482 | if (offset + wrote != read) |
ddcf2b5f | 483 | { |
fd4d8871 R |
484 | offset += wrote; |
485 | goto again; | |
ddcf2b5f JH |
486 | } |
487 | } | |
fd4d8871 R |
488 | } |
489 | while (!feof(mbox_file) && !ferror(mbox_file)); | |
8523533c | 490 | |
fd4d8871 R |
491 | if (ferror(mbox_file)) |
492 | { | |
493 | log_write(0, LOG_MAIN|LOG_PANIC, | |
494 | "%s error reading spool file: %s", loglabel, strerror(errno)); | |
495 | (void)close(spamd_sock); | |
8acbb134 | 496 | goto defer; |
fd4d8871 R |
497 | } |
498 | ||
499 | (void)fclose(mbox_file); | |
500 | ||
501 | /* we're done sending, close socket for writing */ | |
502 | shutdown(spamd_sock,SHUT_WR); | |
503 | ||
504 | /* read spamd response using what's left of the timeout. */ | |
505 | memset(spamd_buffer, 0, sizeof(spamd_buffer)); | |
506 | offset = 0; | |
507 | while ((i = ip_recv(spamd_sock, | |
508 | spamd_buffer + offset, | |
509 | sizeof(spamd_buffer) - offset - 1, | |
8a512ed5 | 510 | sd->timeout - time(NULL) + start)) > 0 ) |
fd4d8871 R |
511 | offset += i; |
512 | ||
513 | /* error handling */ | |
514 | if (i <= 0 && errno != 0) | |
515 | { | |
516 | log_write(0, LOG_MAIN|LOG_PANIC, | |
517 | "%s error reading from spamd socket: %s", loglabel, strerror(errno)); | |
518 | (void)close(spamd_sock); | |
519 | return DEFER; | |
520 | } | |
521 | ||
522 | /* reading done */ | |
523 | (void)close(spamd_sock); | |
524 | ||
8a512ed5 | 525 | if (sd->is_rspamd) |
fd4d8871 R |
526 | { /* rspamd variant of reply */ |
527 | int r; | |
528 | if ((r = sscanf(CS spamd_buffer, | |
529 | "RSPAMD/%7s 0 EX_OK\r\nMetric: default; %7s %lf / %lf / %lf\r\n%n", | |
530 | spamd_version, spamd_short_result, &spamd_score, &spamd_threshold, | |
531 | &spamd_reject_score, &spamd_report_offset)) != 5) | |
532 | { | |
6c54be64 AL |
533 | log_write(0, LOG_MAIN|LOG_PANIC, |
534 | "%s cannot parse spamd output: %d", loglabel, r); | |
535 | return DEFER; | |
fd4d8871 R |
536 | } |
537 | /* now parse action */ | |
8523533c | 538 | p = &spamd_buffer[spamd_report_offset]; |
fd4d8871 R |
539 | |
540 | if (Ustrncmp(p, "Action: ", sizeof("Action: ") - 1) == 0) | |
ddcf2b5f | 541 | { |
fd4d8871 R |
542 | p += sizeof("Action: ") - 1; |
543 | q = &spam_action_buffer[0]; | |
544 | while (*p && *p != '\r' && (q - spam_action_buffer) < sizeof(spam_action_buffer) - 1) | |
545 | *q++ = *p++; | |
546 | *q = '\0'; | |
ddcf2b5f | 547 | } |
fd4d8871 R |
548 | } |
549 | else | |
550 | { /* spamassassin */ | |
551 | /* dig in the spamd output and put the report in a multiline header, | |
552 | if requested */ | |
553 | if (sscanf(CS spamd_buffer, | |
554 | "SPAMD/%7s 0 EX_OK\r\nContent-length: %*u\r\n\r\n%lf/%lf\r\n%n", | |
555 | spamd_version,&spamd_score,&spamd_threshold,&spamd_report_offset) != 3) | |
ddcf2b5f | 556 | { |
fd4d8871 R |
557 | /* try to fall back to pre-2.50 spamd output */ |
558 | if (sscanf(CS spamd_buffer, | |
559 | "SPAMD/%7s 0 EX_OK\r\nSpam: %*s ; %lf / %lf\r\n\r\n%n", | |
560 | spamd_version,&spamd_score,&spamd_threshold,&spamd_report_offset) != 3) | |
561 | { | |
562 | log_write(0, LOG_MAIN|LOG_PANIC, | |
563 | "%s cannot parse spamd output", loglabel); | |
564 | return DEFER; | |
565 | } | |
ddcf2b5f | 566 | } |
fd4d8871 R |
567 | |
568 | Ustrcpy(spam_action_buffer, | |
569 | spamd_score >= spamd_threshold ? "reject" : "no action"); | |
570 | } | |
571 | ||
572 | /* Create report. Since this is a multiline string, | |
573 | we must hack it into shape first */ | |
574 | p = &spamd_buffer[spamd_report_offset]; | |
575 | q = spam_report_buffer; | |
576 | while (*p != '\0') | |
577 | { | |
578 | /* skip \r */ | |
579 | if (*p == '\r') | |
580 | { | |
581 | p++; | |
582 | continue; | |
583 | } | |
584 | *q++ = *p; | |
585 | if (*p++ == '\n') | |
586 | { | |
587 | /* add an extra space after the newline to ensure | |
588 | that it is treated as a header continuation line */ | |
589 | *q++ = ' '; | |
590 | } | |
591 | } | |
592 | /* NULL-terminate */ | |
593 | *q-- = '\0'; | |
594 | /* cut off trailing leftovers */ | |
595 | while (*q <= ' ') | |
596 | *q-- = '\0'; | |
597 | ||
598 | spam_report = spam_report_buffer; | |
599 | spam_action = spam_action_buffer; | |
600 | ||
601 | /* create spam bar */ | |
602 | spamd_score_char = spamd_score > 0 ? '+' : '-'; | |
603 | j = abs((int)(spamd_score)); | |
604 | i = 0; | |
605 | if (j != 0) | |
606 | while ((i < j) && (i <= MAX_SPAM_BAR_CHARS)) | |
607 | spam_bar_buffer[i++] = spamd_score_char; | |
608 | else | |
609 | { | |
610 | spam_bar_buffer[0] = '/'; | |
611 | i = 1; | |
612 | } | |
613 | spam_bar_buffer[i] = '\0'; | |
614 | spam_bar = spam_bar_buffer; | |
615 | ||
616 | /* create "float" spam score */ | |
617 | (void)string_format(spam_score_buffer, sizeof(spam_score_buffer), | |
618 | "%.1f", spamd_score); | |
619 | spam_score = spam_score_buffer; | |
620 | ||
621 | /* create "int" spam score */ | |
622 | j = (int)((spamd_score + 0.001)*10); | |
623 | (void)string_format(spam_score_int_buffer, sizeof(spam_score_int_buffer), | |
624 | "%d", j); | |
625 | spam_score_int = spam_score_int_buffer; | |
626 | ||
627 | /* compare threshold against score */ | |
628 | spam_rc = spamd_score >= spamd_threshold | |
629 | ? OK /* spam as determined by user's threshold */ | |
630 | : FAIL; /* not spam */ | |
631 | ||
632 | /* remember expanded spamd_address if needed */ | |
633 | if (spamd_address_work != spamd_address) | |
634 | prev_spamd_address_work = string_copy(spamd_address_work); | |
635 | ||
636 | /* remember user name and "been here" for it */ | |
637 | Ustrcpy(prev_user_name, user_name); | |
638 | spam_ok = 1; | |
639 | ||
640 | return override | |
641 | ? OK /* always return OK, no matter what the score */ | |
642 | : spam_rc; | |
8acbb134 JH |
643 | |
644 | defer: | |
645 | (void)fclose(mbox_file); | |
646 | return DEFER; | |
8523533c TK |
647 | } |
648 | ||
649 | #endif | |
2aad5761 JH |
650 | /* vi: aw ai sw=2 |
651 | */ |