* spam.el (spam-check-spamoracle, spam-spamoracle-learn): Fix
[gnus] / lisp / spam.el
1 ;;; spam.el --- Identifying spam
2 ;; Copyright (C) 2002, 2003, 2004 Free Software Foundation, Inc.
3
4 ;; Author: Lars Magne Ingebrigtsen <larsi@gnus.org>
5 ;; Keywords: network
6
7 ;; This file is part of GNU Emacs.
8
9 ;; GNU Emacs is free software; you can redistribute it and/or modify
10 ;; it under the terms of the GNU General Public License as published by
11 ;; the Free Software Foundation; either version 2, or (at your option)
12 ;; any later version.
13
14 ;; GNU Emacs is distributed in the hope that it will be useful,
15 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
16 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 ;; GNU General Public License for more details.
18
19 ;; You should have received a copy of the GNU General Public License
20 ;; along with GNU Emacs; see the file COPYING.  If not, write to the
21 ;; Free Software Foundation, Inc., 59 Temple Place - Suite 330,
22 ;; Boston, MA 02111-1307, USA.
23
24 ;;; Commentary:
25
26 ;;; This module addresses a few aspects of spam control under Gnus.  Page
27 ;;; breaks are used for grouping declarations and documentation relating to
28 ;;; each particular aspect.
29
30 ;;; The integration with Gnus is not yet complete.  See various `FIXME'
31 ;;; comments, below, for supplementary explanations or discussions.
32
33 ;;; Several TODO items are marked as such
34
35 ;; TODO: spam scores, detection of spam in newsgroups, cross-server splitting,
36 ;; remote processing, training through files
37
38 ;;; Code:
39
40 (eval-when-compile (require 'cl))
41
42 (require 'gnus-sum)
43
44 (require 'gnus-uu)                      ; because of key prefix issues
45 ;;; for the definitions of group content classification and spam processors
46 (require 'gnus)
47 (require 'message)              ;for the message-fetch-field functions
48
49 ;; for nnimap-split-download-body-default
50 (eval-when-compile (require 'nnimap))
51
52 ;; autoload query-dig
53 (eval-and-compile
54   (autoload 'query-dig "dig"))
55
56 ;; autoload spam-report
57 (eval-and-compile
58   (autoload 'spam-report-gmane "spam-report"))
59
60 ;; autoload gnus-registry
61 (eval-and-compile
62   (autoload 'gnus-registry-group-count "gnus-registry")
63   (autoload 'gnus-registry-add-group "gnus-registry")
64   (autoload 'gnus-registry-store-extra-entry "gnus-registry")
65   (autoload 'gnus-registry-fetch-extra "gnus-registry"))
66
67 ;; autoload query-dns
68 (eval-and-compile
69   (autoload 'query-dns "dns"))
70
71 ;;; Main parameters.
72
73 (defgroup spam nil
74   "Spam configuration.")
75
76 (defcustom spam-directory "~/News/spam/"
77   "Directory for spam whitelists and blacklists."
78   :type 'directory
79   :group 'spam)
80
81 (defcustom spam-move-spam-nonspam-groups-only t
82   "Whether spam should be moved in non-spam groups only.
83 When t, only ham and unclassified groups will have their spam moved
84 to the spam-process-destination.  When nil, spam will also be moved from
85 spam groups."
86   :type 'boolean
87   :group 'spam)
88
89 (defcustom spam-process-ham-in-nonham-groups nil
90   "Whether ham should be processed in non-ham groups."
91   :type 'boolean
92   :group 'spam)
93
94 (defcustom spam-log-to-registry nil
95   "Whether spam/ham processing should be logged in the registry."
96   :type 'boolean
97   :group 'spam)
98
99 (defcustom spam-split-symbolic-return nil
100   "Whether `spam-split' should work with symbols or group names."
101   :type 'boolean
102   :group 'spam)
103
104 (defcustom spam-split-symbolic-return-positive nil
105   "Whether `spam-split' should ALWAYS work with symbols or group names.
106 Do not set this if you use `spam-split' in a fancy split
107   method."
108   :type 'boolean
109   :group 'spam)
110
111 (defcustom spam-process-ham-in-spam-groups nil
112   "Whether ham should be processed in spam groups."
113   :type 'boolean
114   :group 'spam)
115
116 (defcustom spam-mark-only-unseen-as-spam t
117   "Whether only unseen articles should be marked as spam in spam groups.
118 When nil, all unread articles in a spam group are marked as
119 spam.  Set this if you want to leave an article unread in a spam group
120 without losing it to the automatic spam-marking process."
121   :type 'boolean
122   :group 'spam)
123
124 (defcustom spam-mark-ham-unread-before-move-from-spam-group nil
125   "Whether ham should be marked unread before it's moved.
126 The article is moved out of a spam group according to ham-process-destination.
127 This variable is an official entry in the international Longest Variable Name
128 Competition."
129   :type 'boolean
130   :group 'spam)
131
132 (defcustom spam-disable-spam-split-during-ham-respool nil
133   "Whether `spam-split' should be ignored while resplitting ham.
134 This is useful to prevent ham from ending up in the same spam
135 group after the resplit.  Don't set this to t if you have `spam-split' as the
136 last rule in your split configuration."
137   :type 'boolean
138   :group 'spam)
139
140 (defcustom spam-autodetect-recheck-messages nil
141   "Should spam.el recheck all meessages when autodetecting?
142 Normally this is nil, so only unseen messages will be checked."
143   :type 'boolean
144   :group 'spam)
145
146 (defcustom spam-whitelist (expand-file-name "whitelist" spam-directory)
147   "The location of the whitelist.
148 The file format is one regular expression per line.
149 The regular expression is matched against the address."
150   :type 'file
151   :group 'spam)
152
153 (defcustom spam-blacklist (expand-file-name "blacklist" spam-directory)
154   "The location of the blacklist.
155 The file format is one regular expression per line.
156 The regular expression is matched against the address."
157   :type 'file
158   :group 'spam)
159
160 (defcustom spam-use-dig t
161   "Whether `query-dig' should be used instead of `query-dns'."
162   :type 'boolean
163   :group 'spam)
164
165 (defcustom spam-use-blacklist nil
166   "Whether the blacklist should be used by `spam-split'."
167   :type 'boolean
168   :group 'spam)
169
170 (defcustom spam-blacklist-ignored-regexes nil
171   "Regular expressions that the blacklist should ignore."
172   :type '(repeat (regexp :tag "Regular expression to ignore when blacklisting"))
173   :group 'spam)
174
175 (defcustom spam-use-whitelist nil
176   "Whether the whitelist should be used by `spam-split'."
177   :type 'boolean
178   :group 'spam)
179
180 (defcustom spam-use-whitelist-exclusive nil
181   "Whether whitelist-exclusive should be used by `spam-split'.
182 Exclusive whitelisting means that all messages from senders not in the whitelist
183 are considered spam."
184   :type 'boolean
185   :group 'spam)
186
187 (defcustom spam-use-blackholes nil
188   "Whether blackholes should be used by `spam-split'."
189   :type 'boolean
190   :group 'spam)
191
192 (defcustom spam-use-hashcash nil
193   "Whether hashcash payments should be detected by `spam-split'."
194   :type 'boolean
195   :group 'spam)
196
197 (defcustom spam-use-regex-headers nil
198   "Whether a header regular expression match should be used by `spam-split'.
199 Also see the variables `spam-regex-headers-spam' and `spam-regex-headers-ham'."
200   :type 'boolean
201   :group 'spam)
202
203 (defcustom spam-use-regex-body nil
204   "Whether a body regular expression match should be used by `spam-split'.
205 Also see the variables `spam-regex-body-spam' and `spam-regex-body-ham'."
206   :type 'boolean
207   :group 'spam)
208
209 (defcustom spam-use-bogofilter-headers nil
210   "Whether bogofilter headers should be used by `spam-split'.
211 Enable this if you pre-process messages with Bogofilter BEFORE Gnus sees them."
212   :type 'boolean
213   :group 'spam)
214
215 (defcustom spam-use-bogofilter nil
216   "Whether bogofilter should be invoked by `spam-split'.
217 Enable this if you want Gnus to invoke Bogofilter on new messages."
218   :type 'boolean
219   :group 'spam)
220
221 (defcustom spam-use-BBDB nil
222   "Whether BBDB should be used by `spam-split'."
223   :type 'boolean
224   :group 'spam)
225
226 (defcustom spam-use-BBDB-exclusive nil
227   "Whether BBDB-exclusive should be used by `spam-split'.
228 Exclusive BBDB means that all messages from senders not in the BBDB are
229 considered spam."
230   :type 'boolean
231   :group 'spam)
232
233 (defcustom spam-use-ifile nil
234   "Whether ifile should be used by `spam-split'."
235   :type 'boolean
236   :group 'spam)
237
238 (defcustom spam-use-stat nil
239   "Whether `spam-stat' should be used by `spam-split'."
240   :type 'boolean
241   :group 'spam)
242
243 (defcustom spam-use-spamoracle nil
244   "Whether spamoracle should be used by `spam-split'."
245   :type 'boolean
246   :group 'spam)
247
248 (defcustom spam-use-spamassassin nil
249   "Whether spamassassin should be invoked by `spam-split'.
250 Enable this if you want Gnus to invoke SpamAssassin on new messages."
251   :type 'boolean
252   :group 'spam)
253
254 (defcustom spam-use-spamassassin-headers nil
255   "Whether spamassassin headers should be checked by `spam-split'.
256 Enable this if you pre-process messages with SpamAssassin BEFORE Gnus sees
257 them."
258   :type 'boolean
259   :group 'spam)
260
261 (defcustom spam-install-hooks (or
262                                spam-use-dig
263                                spam-use-blacklist
264                                spam-use-whitelist
265                                spam-use-whitelist-exclusive
266                                spam-use-blackholes
267                                spam-use-hashcash
268                                spam-use-regex-headers
269                                spam-use-regex-body
270                                spam-use-bogofilter
271                                spam-use-bogofilter-headers
272                                spam-use-spamassassin
273                                spam-use-spamassassin-headers
274                                spam-use-BBDB
275                                spam-use-BBDB-exclusive
276                                spam-use-ifile
277                                spam-use-stat
278                                spam-use-spamoracle)
279   "Whether the spam hooks should be installed.
280 Default to t if one of the spam-use-* variables is set."
281   :group 'spam
282   :type 'boolean)
283
284 (defcustom spam-split-group "spam"
285   "Group name where incoming spam should be put by `spam-split'."
286   :type 'string
287   :group 'spam)
288
289 ;;; TODO: deprecate this variable, it's confusing since it's a list of strings,
290 ;;; not regular expressions
291 (defcustom spam-junk-mailgroups (cons
292                                  spam-split-group
293                                  '("mail.junk" "poste.pourriel"))
294   "Mailgroups with spam contents.
295 All unmarked article in such group receive the spam mark on group entry."
296   :type '(repeat (string :tag "Group"))
297   :group 'spam)
298
299 (defcustom spam-blackhole-servers '("bl.spamcop.net" "relays.ordb.org"
300                                     "dev.null.dk" "relays.visi.com")
301   "List of blackhole servers."
302   :type '(repeat (string :tag "Server"))
303   :group 'spam)
304
305 (defcustom spam-blackhole-good-server-regex nil
306   "String matching IP addresses that should not be checked in the blackholes."
307   :type '(radio (const nil)
308                 (regexp :format "%t: %v\n" :size 0))
309   :group 'spam)
310
311 (defcustom spam-face 'gnus-splash-face
312   "Face for spam-marked articles."
313   :type 'face
314   :group 'spam)
315
316 (defcustom spam-regex-headers-spam '("^X-Spam-Flag: YES")
317   "Regular expression for positive header spam matches."
318   :type '(repeat (regexp :tag "Regular expression to match spam header"))
319   :group 'spam)
320
321 (defcustom spam-regex-headers-ham '("^X-Spam-Flag: NO")
322   "Regular expression for positive header ham matches."
323   :type '(repeat (regexp :tag "Regular expression to match ham header"))
324   :group 'spam)
325
326 (defcustom spam-regex-body-spam '()
327   "Regular expression for positive body spam matches."
328   :type '(repeat (regexp :tag "Regular expression to match spam body"))
329   :group 'spam)
330
331 (defcustom spam-regex-body-ham '()
332   "Regular expression for positive body ham matches."
333   :type '(repeat (regexp :tag "Regular expression to match ham body"))
334   :group 'spam)
335
336 (defgroup spam-ifile nil
337   "Spam ifile configuration."
338   :group 'spam)
339
340 (defcustom spam-ifile-path (executable-find "ifile")
341   "File path of the ifile executable program."
342   :type '(choice (file :tag "Location of ifile")
343                  (const :tag "ifile is not installed"))
344   :group 'spam-ifile)
345
346 (defcustom spam-ifile-database-path nil
347   "File path of the ifile database."
348   :type '(choice (file :tag "Location of the ifile database")
349                  (const :tag "Use the default"))
350   :group 'spam-ifile)
351
352 (defcustom spam-ifile-spam-category "spam"
353   "Name of the spam ifile category."
354   :type 'string
355   :group 'spam-ifile)
356
357 (defcustom spam-ifile-ham-category nil
358   "Name of the ham ifile category.
359 If nil, the current group name will be used."
360   :type '(choice (string :tag "Use a fixed category")
361                  (const :tag "Use the current group name"))
362   :group 'spam-ifile)
363
364 (defcustom spam-ifile-all-categories nil
365   "Whether the ifile check will return all categories, or just spam.
366 Set this to t if you want to use the `spam-split' invocation of ifile as
367 your main source of newsgroup names."
368   :type 'boolean
369   :group 'spam-ifile)
370
371 (defgroup spam-bogofilter nil
372   "Spam bogofilter configuration."
373   :group 'spam)
374
375 (defcustom spam-bogofilter-path (executable-find "bogofilter")
376   "File path of the Bogofilter executable program."
377   :type '(choice (file :tag "Location of bogofilter")
378                  (const :tag "Bogofilter is not installed"))
379   :group 'spam-bogofilter)
380
381 (defcustom spam-bogofilter-header "X-Bogosity"
382   "The header that Bogofilter inserts in messages."
383   :type 'string
384   :group 'spam-bogofilter)
385
386 (defcustom spam-bogofilter-spam-switch "-s"
387   "The switch that Bogofilter uses to register spam messages."
388   :type 'string
389   :group 'spam-bogofilter)
390
391 (defcustom spam-bogofilter-ham-switch "-n"
392   "The switch that Bogofilter uses to register ham messages."
393   :type 'string
394   :group 'spam-bogofilter)
395
396 (defcustom spam-bogofilter-spam-strong-switch "-S"
397   "The switch that Bogofilter uses to unregister ham messages."
398   :type 'string
399   :group 'spam-bogofilter)
400
401 (defcustom spam-bogofilter-ham-strong-switch "-N"
402   "The switch that Bogofilter uses to unregister spam messages."
403   :type 'string
404   :group 'spam-bogofilter)
405
406 (defcustom spam-bogofilter-bogosity-positive-spam-header "^\\(Yes\\|Spam\\)"
407   "The regex on `spam-bogofilter-header' for positive spam identification."
408   :type 'regexp
409   :group 'spam-bogofilter)
410
411 (defcustom spam-bogofilter-database-directory nil
412   "Directory path of the Bogofilter databases."
413   :type '(choice (directory
414                   :tag "Location of the Bogofilter database directory")
415                  (const :tag "Use the default"))
416   :group 'spam-bogofilter)
417
418 (defgroup spam-spamoracle nil
419   "Spam spamoracle configuration."
420   :group 'spam)
421
422 (defcustom spam-spamoracle-database nil
423   "Location of spamoracle database file.
424 When nil, use the default spamoracle database."
425   :type '(choice (directory :tag "Location of spamoracle database file.")
426                  (const :tag "Use the default"))
427   :group 'spam-spamoracle)
428
429 (defcustom spam-spamoracle-binary (executable-find "spamoracle")
430   "Location of the spamoracle binary."
431   :type '(choice (directory :tag "Location of the spamoracle binary")
432                  (const :tag "Use the default"))
433   :group 'spam-spamoracle)
434
435 (defgroup spam-spamassassin nil
436   "Spam SpamAssassin configuration."
437   :group 'spam)
438
439 (defcustom spam-spamassassin-path (executable-find "spamassassin")
440   "File path of the spamassassin executable program.
441 Hint: set this to \"spamc\" if you have spamd running.  See the spamc and
442 spamd man pages for more information on these programs."
443   :type '(choice (file :tag "Location of spamc")
444                  (const :tag "spamassassin is not installed"))
445   :group 'spam-spamassassin)
446
447 (defcustom spam-spamassassin-arguments ()
448   "Arguments to pass to the spamassassin executable.
449 This must be a list.  For example, `(\"-C\" \"configfile\")'."
450   :type '(restricted-sexp :match-alternatives (listp))
451   :group 'spam-spamassassin)
452
453 (defcustom spam-spamassassin-spam-flag-header "X-Spam-Flag"
454   "The header inserted by SpamAssassin to flag spam."
455   :type 'string
456   :group 'spam-spamassassin)
457
458 (defcustom spam-spamassassin-positive-spam-flag-header "YES"
459   "The regex on `spam-spamassassin-spam-flag-header' for positive spam
460 identification"
461   :type 'string
462   :group 'spam-spamassassin)
463
464 (defcustom spam-spamassassin-spam-status-header "X-Spam-Status"
465   "The header inserted by SpamAssassin, giving extended scoring information"
466   :type 'string
467   :group 'spam-spamassassin)
468
469 (defcustom spam-sa-learn-path (executable-find "sa-learn")
470   "File path of the sa-learn executable program."
471   :type '(choice (file :tag "Location of spamassassin")
472                  (const :tag "spamassassin is not installed"))
473   :group 'spam-spamassassin)
474
475 (defcustom spam-sa-learn-rebuild t
476   "Whether sa-learn should rebuild the database every time it is called
477 Enable this if you want sa-learn to rebuild the database automatically.  Doing
478 this will slightly increase the running time of the spam registration process.
479 If you choose not to do this, you will have to run \"sa-learn --rebuild\" in
480 order for SpamAssassin to recognize the new registered spam."
481   :type 'boolean
482   :group 'spam-spamassassin)
483
484 (defcustom spam-sa-learn-spam-switch "--spam"
485   "The switch that sa-learn uses to register spam messages"
486   :type 'string
487   :group 'spam-spamassassin)
488
489 (defcustom spam-sa-learn-ham-switch "--ham"
490   "The switch that sa-learn uses to register ham messages"
491   :type 'string
492   :group 'spam-spamassassin)
493
494 (defcustom spam-sa-learn-unregister-switch "--forget"
495   "The switch that sa-learn uses to unregister messages messages"
496   :type 'string
497   :group 'spam-spamassassin)
498
499 ;;; Key bindings for spam control.
500
501 (gnus-define-keys gnus-summary-mode-map
502   "St" spam-generic-score
503   "Sx" gnus-summary-mark-as-spam
504   "Mst" spam-generic-score
505   "Msx" gnus-summary-mark-as-spam
506   "\M-d" gnus-summary-mark-as-spam)
507
508 (defvar spam-cache-lookups t
509   "Whether spam.el will try to cache lookups using `spam-caches'.")
510
511 (defvar spam-caches (make-hash-table
512                      :size 10
513                      :test 'equal)
514   "Cache of spam detection entries.")
515
516 (defvar spam-old-ham-articles nil
517   "List of old ham articles, generated when a group is entered.")
518
519 (defvar spam-old-spam-articles nil
520   "List of old spam articles, generated when a group is entered.")
521
522 (defvar spam-split-disabled nil
523   "If non-nil, `spam-split' is disabled, and always returns nil.")
524
525 (defvar spam-split-last-successful-check nil
526   "Internal variable.
527 `spam-split' will set this to nil or a spam-use-XYZ check if it
528 finds ham or spam.")
529
530 ;; convenience functions
531 (defun spam-clear-cache (symbol)
532   "Clear the spam-caches entry for a check."
533   (remhash symbol spam-caches))
534
535 (defun spam-xor (a b)
536   "Logical A xor B."
537   (and (or a b) (not (and a b))))
538
539 (defun spam-group-ham-mark-p (group mark &optional spam)
540   "Checks if MARK is considered a ham mark in GROUP."
541   (when (stringp group)
542     (let* ((marks (spam-group-ham-marks group spam))
543            (marks (if (symbolp mark)
544                       marks
545                     (mapcar 'symbol-value marks))))
546       (memq mark marks))))
547
548 (defun spam-group-spam-mark-p (group mark)
549   "Checks if MARK is considered a spam mark in GROUP."
550   (spam-group-ham-mark-p group mark t))
551
552 (defun spam-group-ham-marks (group &optional spam)
553   "In GROUP, get all the ham marks."
554   (when (stringp group)
555     (let* ((marks (if spam
556                       (gnus-parameter-spam-marks group)
557                     (gnus-parameter-ham-marks group)))
558            (marks (car marks))
559            (marks (if (listp (car marks)) (car marks) marks)))
560       marks)))
561
562 (defun spam-group-spam-marks (group)
563   "In GROUP, get all the spam marks."
564   (spam-group-ham-marks group t))
565
566 (defun spam-group-spam-contents-p (group)
567   "Is GROUP a spam group?"
568   (if (stringp group)
569       (or (member group spam-junk-mailgroups)
570           (memq 'gnus-group-spam-classification-spam
571                 (gnus-parameter-spam-contents group)))
572     nil))
573
574 (defun spam-group-ham-contents-p (group)
575   "Is GROUP a ham group?"
576   (if (stringp group)
577       (memq 'gnus-group-spam-classification-ham
578             (gnus-parameter-spam-contents group))
579     nil))
580
581 (defvar spam-list-of-processors
582   '((gnus-group-spam-exit-processor-report-gmane spam spam-use-gmane)
583     (gnus-group-spam-exit-processor-bogofilter   spam spam-use-bogofilter)
584     (gnus-group-spam-exit-processor-blacklist    spam spam-use-blacklist)
585     (gnus-group-spam-exit-processor-ifile        spam spam-use-ifile)
586     (gnus-group-spam-exit-processor-stat         spam spam-use-stat)
587     (gnus-group-spam-exit-processor-spamoracle   spam spam-use-spamoracle)
588     (gnus-group-spam-exit-processor-spamassassin spam spam-use-spam-spamassassin)
589     (gnus-group-ham-exit-processor-ifile         ham spam-use-ifile)
590     (gnus-group-ham-exit-processor-bogofilter    ham spam-use-bogofilter)
591     (gnus-group-ham-exit-processor-stat          ham spam-use-stat)
592     (gnus-group-ham-exit-processor-whitelist     ham spam-use-whitelist)
593     (gnus-group-ham-exit-processor-BBDB          ham spam-use-BBDB)
594     (gnus-group-ham-exit-processor-copy          ham spam-use-ham-copy)
595     (gnus-group-ham-exit-processor-spamassassin  ham spam-use-ham-spamassassin)
596     (gnus-group-ham-exit-processor-spamoracle    ham spam-use-spamoracle))
597   "The `spam-list-of-processors' list.
598 This list contains pairs associating a ham/spam exit processor
599 variable with a classification and a spam-use-* variable.")
600
601 (defun spam-group-processor-p (group processor)
602   (if (and (stringp group)
603            (symbolp processor))
604       (or (member processor (nth 0 (gnus-parameter-spam-process group)))
605           (spam-group-processor-multiple-p
606            group
607            (cdr-safe (assoc processor spam-list-of-processors))))
608     nil))
609
610 (defun spam-group-processor-multiple-p (group processor-info)
611   (let* ((classification (nth 0 processor-info))
612          (check (nth 1 processor-info))
613          (parameters (nth 0 (gnus-parameter-spam-process group)))
614          found)
615     (dolist (parameter parameters)
616       (when (and (null found)
617                  (listp parameter)
618                  (eq classification (nth 0 parameter))
619                  (eq check (nth 1 parameter)))
620         (setq found t)))
621     found))
622
623 (defun spam-group-spam-processor-report-gmane-p (group)
624   (spam-group-processor-p group 'gnus-group-spam-exit-processor-report-gmane))
625
626 (defun spam-group-spam-processor-bogofilter-p (group)
627   (spam-group-processor-p group 'gnus-group-spam-exit-processor-bogofilter))
628
629 (defun spam-group-spam-processor-blacklist-p (group)
630   (spam-group-processor-p group 'gnus-group-spam-exit-processor-blacklist))
631
632 (defun spam-group-spam-processor-ifile-p (group)
633   (spam-group-processor-p group 'gnus-group-spam-exit-processor-ifile))
634
635 (defun spam-group-ham-processor-ifile-p (group)
636   (spam-group-processor-p group 'gnus-group-ham-exit-processor-ifile))
637
638 (defun spam-group-spam-processor-spamoracle-p (group)
639   (spam-group-processor-p group 'gnus-group-spam-exit-processor-spamoracle))
640
641 (defun spam-group-ham-processor-bogofilter-p (group)
642   (spam-group-processor-p group 'gnus-group-ham-exit-processor-bogofilter))
643
644 (defun spam-group-spam-processor-stat-p (group)
645   (spam-group-processor-p group 'gnus-group-spam-exit-processor-stat))
646
647 (defun spam-group-ham-processor-stat-p (group)
648   (spam-group-processor-p group 'gnus-group-ham-exit-processor-stat))
649
650 (defun spam-group-ham-processor-whitelist-p (group)
651   (spam-group-processor-p group 'gnus-group-ham-exit-processor-whitelist))
652
653 (defun spam-group-ham-processor-BBDB-p (group)
654   (spam-group-processor-p group 'gnus-group-ham-exit-processor-BBDB))
655
656 (defun spam-group-ham-processor-copy-p (group)
657   (spam-group-processor-p group 'gnus-group-ham-exit-processor-copy))
658
659 (defun spam-group-ham-processor-spamoracle-p (group)
660   (spam-group-processor-p group 'gnus-group-ham-exit-processor-spamoracle))
661
662 (defun spam-report-articles-gmane (n)
663   "Report the current message as spam.
664 Respects the process/prefix convention."
665   (interactive "P")
666   (dolist (article (gnus-summary-work-articles n))
667     (gnus-summary-remove-process-mark article)
668     (spam-report-gmane article)))
669
670 (defun spam-generic-score ()
671   (interactive)
672   "Invoke whatever scoring method we can."
673   (if (or
674        spam-use-spamassassin
675        spam-use-spamassassin-headers)
676       (spam-spamassassin-score)
677     (spam-bogofilter-score)))
678
679 ;;; Summary entry and exit processing.
680
681 (defun spam-summary-prepare ()
682   (setq spam-old-ham-articles
683         (spam-list-articles gnus-newsgroup-articles 'ham))
684   (setq spam-old-spam-articles
685         (spam-list-articles gnus-newsgroup-articles 'spam))
686   (spam-mark-junk-as-spam-routine))
687
688 ;; The spam processors are invoked for any group, spam or ham or neither
689 (defun spam-summary-prepare-exit ()
690   (unless gnus-group-is-exiting-without-update-p
691     (gnus-message 6 "Exiting summary buffer and applying spam rules")
692
693     ;; first of all, unregister any articles that are no longer ham or spam
694     ;; we have to iterate over the processors, or else we'll be too slow
695     (dolist (classification '(spam ham))
696       (let* ((old-articles (if (eq classification 'spam)
697                                spam-old-spam-articles
698                              spam-old-ham-articles))
699              (new-articles (spam-list-articles
700                             gnus-newsgroup-articles
701                             classification))
702              (changed-articles (gnus-set-difference old-articles new-articles)))
703         ;; now that we have the changed articles, we go through the processors
704         (dolist (processor-param spam-list-of-processors)
705           (let ((processor (nth 0 processor-param))
706                 (processor-classification (nth 1 processor-param))
707                 (check (nth 2 processor-param))
708                 unregister-list)
709             (dolist (article changed-articles)
710               (let ((id (spam-fetch-field-message-id-fast article)))
711                 (when (spam-log-unregistration-needed-p
712                        id 'process classification check)
713                   (push article unregister-list))))
714             ;; call spam-register-routine with specific articles to unregister,
715             ;; when there are articles to unregister and the check is enabled
716             (when (and unregister-list (symbol-value check))
717               (spam-register-routine classification check t unregister-list))))))
718
719     ;; find all the spam processors applicable to this group
720     (dolist (processor-param spam-list-of-processors)
721       (let ((processor (nth 0 processor-param))
722             (classification (nth 1 processor-param))
723             (check (nth 2 processor-param)))
724         (when (and (eq 'spam classification)
725                    (spam-group-processor-p gnus-newsgroup-name processor))
726           (spam-register-routine classification check))))
727
728     (if spam-move-spam-nonspam-groups-only
729         (when (not (spam-group-spam-contents-p gnus-newsgroup-name))
730           (spam-mark-spam-as-expired-and-move-routine
731            (gnus-parameter-spam-process-destination gnus-newsgroup-name)))
732       (gnus-message 5 "Marking spam as expired and moving it to %s"
733                     gnus-newsgroup-name)
734       (spam-mark-spam-as-expired-and-move-routine
735        (gnus-parameter-spam-process-destination gnus-newsgroup-name)))
736
737     ;; now we redo spam-mark-spam-as-expired-and-move-routine to only
738     ;; expire spam, in case the above did not expire them
739     (gnus-message 5 "Marking spam as expired without moving it")
740     (spam-mark-spam-as-expired-and-move-routine nil)
741
742     (when (or (spam-group-ham-contents-p gnus-newsgroup-name)
743               (and (spam-group-spam-contents-p gnus-newsgroup-name)
744                    spam-process-ham-in-spam-groups)
745               spam-process-ham-in-nonham-groups)
746       ;; find all the ham processors applicable to this group
747       (dolist (processor-param spam-list-of-processors)
748         (let ((processor (nth 0 processor-param))
749               (classification (nth 1 processor-param))
750               (check (nth 2 processor-param)))
751           (when (and (eq 'ham classification)
752                      (spam-group-processor-p gnus-newsgroup-name processor))
753             (spam-register-routine classification check)))))
754
755     (when (spam-group-ham-processor-copy-p gnus-newsgroup-name)
756       (gnus-message 5 "Copying ham")
757       (spam-ham-copy-routine
758        (gnus-parameter-ham-process-destination gnus-newsgroup-name)))
759
760     ;; now move all ham articles out of spam groups
761     (when (spam-group-spam-contents-p gnus-newsgroup-name)
762       (gnus-message 5 "Moving ham messages from spam group")
763       (spam-ham-move-routine
764        (gnus-parameter-ham-process-destination gnus-newsgroup-name))))
765
766   (setq spam-old-ham-articles nil)
767   (setq spam-old-spam-articles nil))
768
769 (defun spam-mark-junk-as-spam-routine ()
770   ;; check the global list of group names spam-junk-mailgroups and the
771   ;; group parameters
772   (when (spam-group-spam-contents-p gnus-newsgroup-name)
773     (gnus-message 5 "Marking %s articles as spam"
774                   (if spam-mark-only-unseen-as-spam
775                       "unseen"
776                     "unread"))
777     (let ((articles (if spam-mark-only-unseen-as-spam
778                         gnus-newsgroup-unseen
779                       gnus-newsgroup-unreads)))
780       (dolist (article articles)
781         (gnus-summary-mark-article article gnus-spam-mark)))))
782
783 (defun spam-mark-spam-as-expired-and-move-routine (&rest groups)
784   (if (and (car-safe groups) (listp (car-safe groups)))
785       (apply 'spam-mark-spam-as-expired-and-move-routine (car groups))
786     (gnus-summary-kill-process-mark)
787     (let ((articles gnus-newsgroup-articles)
788           (backend-supports-deletions
789            (gnus-check-backend-function
790             'request-move-article gnus-newsgroup-name))
791           article tomove deletep)
792       (dolist (article articles)
793         (when (eq (gnus-summary-article-mark article) gnus-spam-mark)
794           (gnus-summary-mark-article article gnus-expirable-mark)
795           (push article tomove)))
796
797       ;; now do the actual copies
798       (dolist (group groups)
799         (when (and tomove
800                    (stringp group))
801           (dolist (article tomove)
802             (gnus-summary-set-process-mark article))
803           (when tomove
804             (if (or (not backend-supports-deletions)
805                     (> (length groups) 1))
806                 (progn
807                   (gnus-summary-copy-article nil group)
808                   (setq deletep t))
809               (gnus-summary-move-article nil group)))))
810
811       ;; now delete the articles, if there was a copy done, and the
812       ;; backend allows it
813       (when (and deletep backend-supports-deletions)
814         (dolist (article tomove)
815           (gnus-summary-set-process-mark article))
816         (when tomove
817           (let ((gnus-novice-user nil)) ; don't ask me if I'm sure
818             (gnus-summary-delete-article nil))))
819
820       (gnus-summary-yank-process-mark))))
821
822 (defun spam-ham-copy-or-move-routine (copy groups)
823   (gnus-summary-kill-process-mark)
824   (let ((todo (spam-list-articles gnus-newsgroup-articles 'ham))
825         (backend-supports-deletions
826          (gnus-check-backend-function
827           'request-move-article gnus-newsgroup-name))
828         (respool-method (gnus-find-method-for-group gnus-newsgroup-name))
829         article mark todo deletep respool)
830
831     (when (member 'respool groups)
832       (setq respool t)                  ; boolean for later
833       (setq groups '("fake"))) ; when respooling, groups are dynamic so fake it
834
835     ;; now do the actual move
836     (dolist (group groups)
837       (when (and todo (stringp group))
838         (dolist (article todo)
839           (when spam-mark-ham-unread-before-move-from-spam-group
840             (gnus-summary-mark-article article gnus-unread-mark))
841           (gnus-summary-set-process-mark article))
842
843         (if respool                        ; respooling is with a "fake" group
844             (let ((spam-split-disabled
845                    (or spam-split-disabled
846                        spam-disable-spam-split-during-ham-respool)))
847               (gnus-summary-respool-article nil respool-method))
848           (if (or (not backend-supports-deletions) ; else, we are not respooling
849                   (> (length groups) 1))
850               (progn                ; if copying, copy and set deletep
851                 (gnus-summary-copy-article nil group)
852                 (setq deletep t))
853             (gnus-summary-move-article nil group))))) ; else move articles
854
855     ;; now delete the articles, unless a) copy is t, and there was a copy done
856     ;;                                 b) a move was done to a single group
857     ;;                                 c) backend-supports-deletions is nil
858     (unless copy
859       (when (and deletep backend-supports-deletions)
860         (dolist (article todo)
861           (gnus-summary-set-process-mark article))
862         (when todo
863           (let ((gnus-novice-user nil)) ; don't ask me if I'm sure
864             (gnus-summary-delete-article nil))))))
865
866   (gnus-summary-yank-process-mark))
867
868 (defun spam-ham-copy-routine (&rest groups)
869   (if (and (car-safe groups) (listp (car-safe groups)))
870       (apply 'spam-ham-copy-routine (car groups))
871     (spam-ham-copy-or-move-routine t groups)))
872
873 (defun spam-ham-move-routine (&rest groups)
874   (if (and (car-safe groups) (listp (car-safe groups)))
875       (apply 'spam-ham-move-routine (car groups))
876     (spam-ham-copy-or-move-routine nil groups)))
877
878 (defun spam-get-article-as-string (article)
879   (when (numberp article)
880     (with-temp-buffer
881       (gnus-request-article-this-buffer
882        article
883        gnus-newsgroup-name)
884       (buffer-string))))
885
886 ;; disabled for now
887 ;; (defun spam-get-article-as-filename (article)
888 ;;   (let ((article-filename))
889 ;;     (when (numberp article)
890 ;;       (nnml-possibly-change-directory
891 ;;        (gnus-group-real-name gnus-newsgroup-name))
892 ;;       (setq article-filename (expand-file-name
893 ;;                              (int-to-string article) nnml-current-directory)))
894 ;;     (if (file-exists-p article-filename)
895 ;;      article-filename
896 ;;       nil)))
897
898 (defun spam-fetch-field-fast (article field &optional prepared-data-header)
899   "Fetch a field quickly, using the internal gnus-data-list function"
900   (when (numberp article)
901     (let* ((data-header (or prepared-data-header
902                             (spam-fetch-article-header article))))
903       (if (arrayp data-header)
904         (cond
905          ((equal field 'from)
906           (mail-header-from data-header))
907          ((equal field 'message-id)
908           (mail-header-message-id data-header))
909          ((equal field 'subject)
910           (mail-header-subject data-header))
911          ((equal field 'references)
912           (mail-header-references data-header))
913          ((equal field 'date)
914           (mail-header-date data-header))
915          ((equal field 'xref)
916           (mail-header-xref data-header))
917          ((equal field 'extra)
918           (mail-header-extra data-header))
919          (t
920           nil))
921         (gnus-error 5 "Article %d has a nil data header" article)))))
922
923 (defun spam-fetch-field-from-fast (article &optional prepared-data-header)
924   (spam-fetch-field-fast article 'from prepared-data-header))
925
926 (defun spam-fetch-field-subject-fast (article &optional prepared-data-header)
927   (spam-fetch-field-fast article 'subject prepared-data-header))
928
929 (defun spam-fetch-field-message-id-fast (article &optional prepared-data-header)
930   (spam-fetch-field-fast article 'message-id prepared-data-header))
931
932 (defun spam-generate-fake-headers (article)
933   (let ((dh (spam-fetch-article-header article)))
934     (if dh
935         (concat
936          (format 
937           (concat "From: %s\nSubject: %s\nMessage-ID: %s\n"
938                   "Date: %s\nReferences: %s\nXref: %s\n")
939           (spam-fetch-field-fast article 'from dh)
940           (spam-fetch-field-fast article 'subject dh)
941           (spam-fetch-field-fast article 'message-id dh)
942           (spam-fetch-field-fast article 'date dh)
943           (spam-fetch-field-fast article 'references dh)
944           (spam-fetch-field-fast article 'xref dh))
945          (when (spam-fetch-field-fast article 'extra dh)
946            (format "%s\n" (spam-fetch-field-fast article 'extra dh))))
947       (gnus-error
948        5
949        "spam-generate-fake-headers: article %d didn't have a valid header"
950        article))))
951
952 (defun spam-fetch-article-header (article)
953   (save-excursion
954     (set-buffer gnus-summary-buffer)
955     (nth 3 (assq article gnus-newsgroup-data))))
956
957 \f
958 ;;;; Spam determination.
959
960 (defvar spam-list-of-checks
961   '((spam-use-blacklist                 .       spam-check-blacklist)
962     (spam-use-regex-headers             .       spam-check-regex-headers)
963     (spam-use-regex-body                .       spam-check-regex-body)
964     (spam-use-whitelist                 .       spam-check-whitelist)
965     (spam-use-BBDB                      .       spam-check-BBDB)
966     (spam-use-BBDB-exclusive            .       spam-check-BBDB)
967     (spam-use-ifile                     .       spam-check-ifile)
968     (spam-use-spamoracle                .       spam-check-spamoracle)
969     (spam-use-stat                      .       spam-check-stat)
970     (spam-use-blackholes                .       spam-check-blackholes)
971     (spam-use-hashcash                  .       spam-check-hashcash)
972     (spam-use-spamassassin-headers      .       spam-check-spamassassin-headers)
973     (spam-use-spamassassin              .       spam-check-spamassassin)
974     (spam-use-bogofilter-headers        .       spam-check-bogofilter-headers)
975     (spam-use-bogofilter                .       spam-check-bogofilter))
976   "The spam-list-of-checks list contains pairs associating a
977 parameter variable with a spam checking function.  If the
978 parameter variable is true, then the checking function is called,
979 and its value decides what happens.  Each individual check may
980 return nil, t, or a mailgroup name.  The value nil means that the
981 check does not yield a decision, and so, that further checks are
982 needed.  The value t means that the message is definitely not
983 spam, and that further spam checks should be inhibited.
984 Otherwise, a mailgroup name or the symbol 'spam (depending on
985 spam-split-symbolic-return) is returned where the mail should go,
986 and further checks are also inhibited.  The usual mailgroup name
987 is the value of `spam-split-group', meaning that the message is
988 definitely a spam.")
989
990 (defvar spam-list-of-statistical-checks
991   '(spam-use-ifile
992     spam-use-regex-body
993     spam-use-stat
994     spam-use-bogofilter
995     spam-use-blackholes
996     spam-use-spamassassin
997     spam-use-spamoracle)
998   "The spam-list-of-statistical-checks list contains all the mail
999 splitters that need to have the full message body available.
1000 Note that you should fetch extra headers if you don't like this,
1001 e.g. fetch the 'Received' header for spam-use-blackholes.")
1002
1003 (defun spam-split (&rest specific-checks)
1004   "Split this message into the `spam' group if it is spam.
1005 This function can be used as an entry in the variable `nnmail-split-fancy',
1006 for example like this: (: spam-split).  It can take checks as
1007 parameters.  A string as a parameter will set the
1008 spam-split-group to that string.
1009
1010 See the Info node `(gnus)Fancy Mail Splitting' for more details."
1011   (interactive)
1012   (setq spam-split-last-successful-check nil)
1013   (unless spam-split-disabled
1014     (let ((spam-split-group-choice spam-split-group))
1015       (dolist (check specific-checks)
1016         (when (stringp check)
1017           (setq spam-split-group-choice check)
1018           (setq specific-checks (delq check specific-checks))))
1019
1020       (let ((spam-split-group spam-split-group-choice))
1021         (save-excursion
1022           (save-restriction
1023             (dolist (check spam-list-of-statistical-checks)
1024               (when (and (symbolp check)
1025                          (or (symbol-value check)
1026                              (memq check specific-checks)))
1027                 (widen)
1028                 (gnus-message 8 "spam-split: widening the buffer (%s requires it)"
1029                               (symbol-name check))
1030                 (return)))
1031             ;;   (progn (widen) (debug (buffer-string)))
1032             (let ((list-of-checks spam-list-of-checks)
1033                   decision)
1034               (while (and list-of-checks (not decision))
1035                 (let ((pair (pop list-of-checks)))
1036                   (when (or
1037                          ;; either, given specific checks, this is one of them
1038                          (and specific-checks (memq (car pair) specific-checks))
1039                          ;; or, given no specific checks, spam-use-CHECK is set
1040                          (and (null specific-checks) (symbol-value (car pair))))
1041                     (gnus-message 5 "spam-split: calling the %s function"
1042                                   (symbol-name (cdr pair)))
1043                     (setq decision (funcall (cdr pair)))
1044                     ;; if we got a decision at all, save the current check
1045                     (when decision
1046                       (setq spam-split-last-successful-check (car pair)))
1047
1048                     (when (eq decision 'spam)
1049                       (unless spam-split-symbolic-return
1050                         (gnus-error
1051                          5
1052                          (format "spam-split got %s but %s is nil"
1053                                  (symbol-name decision)
1054                                  (symbol-name spam-split-symbolic-return))))))))
1055               (if (eq decision t)
1056                   (if spam-split-symbolic-return-positive 'ham nil)
1057                 decision))))))))
1058
1059 (defun spam-find-spam ()
1060   "This function will detect spam in the current newsgroup using spam-split."
1061   (interactive)
1062
1063   (let* ((group gnus-newsgroup-name)
1064          (autodetect (gnus-parameter-spam-autodetect group))
1065          (methods (gnus-parameter-spam-autodetect-methods group))
1066          (first-method (nth 0 methods))
1067          (articles (if spam-autodetect-recheck-messages
1068                        gnus-newsgroup-articles
1069                      gnus-newsgroup-unseen))
1070          article-cannot-be-faked)
1071
1072     (dolist (check spam-list-of-statistical-checks)
1073       (when (and (symbolp check)
1074                  (memq check methods))
1075         (setq article-cannot-be-faked t)
1076         (return)))
1077
1078     (when (memq 'default methods)
1079       (setq article-cannot-be-faked t))
1080
1081     (when (and autodetect
1082                (not (equal first-method 'none)))
1083     (mapcar
1084      (lambda (article)
1085        (let ((id (spam-fetch-field-message-id-fast article))
1086              (subject (spam-fetch-field-subject-fast article))
1087              (sender (spam-fetch-field-from-fast article))
1088              registry-lookup)
1089          
1090          (unless id
1091            (gnus-error 5 "Article %d has no message ID!" article))
1092          
1093          (when (and id spam-log-to-registry)
1094            (setq registry-lookup (spam-log-registration-type id 'incoming))
1095            (when registry-lookup
1096              (gnus-message
1097               9
1098               "spam-find-spam: message %s was already registered incoming"
1099               id)))
1100
1101          (let* ((spam-split-symbolic-return t)
1102                 (spam-split-symbolic-return-positive t)
1103                 (fake-headers (spam-generate-fake-headers article))
1104                 (split-return
1105                  (or registry-lookup
1106                      (with-temp-buffer
1107                        (if article-cannot-be-faked
1108                            (gnus-request-article-this-buffer
1109                             article
1110                             group)
1111                          ;; else, we fake the article
1112                          (when fake-headers (insert fake-headers)))
1113                        (if (or (null first-method)
1114                                (equal first-method 'default))
1115                            (spam-split)
1116                          (apply 'spam-split methods))))))
1117            (if (equal split-return 'spam)
1118                (gnus-summary-mark-article article gnus-spam-mark))
1119            
1120            (when (and id split-return spam-log-to-registry)
1121              (when (zerop (gnus-registry-group-count id))
1122                (gnus-registry-add-group
1123                 id group subject sender))
1124                
1125              (unless registry-lookup
1126                (spam-log-processing-to-registry
1127                 id
1128                 'incoming
1129                 split-return
1130                 spam-split-last-successful-check
1131                 group))))))
1132     articles))))
1133
1134 (defvar spam-registration-functions
1135   ;; first the ham register, second the spam register function
1136   ;; third the ham unregister, fourth the spam unregister function
1137   '((spam-use-blacklist  nil
1138                          spam-blacklist-register-routine
1139                          nil
1140                          spam-blacklist-unregister-routine)
1141     (spam-use-whitelist  spam-whitelist-register-routine
1142                          nil
1143                          spam-whitelist-unregister-routine
1144                          nil)
1145     (spam-use-BBDB       spam-BBDB-register-routine
1146                          nil
1147                          spam-BBDB-unregister-routine
1148                          nil)
1149     (spam-use-ifile      spam-ifile-register-ham-routine
1150                          spam-ifile-register-spam-routine
1151                          spam-ifile-unregister-ham-routine
1152                          spam-ifile-unregister-spam-routine)
1153     (spam-use-spamoracle spam-spamoracle-learn-ham
1154                          spam-spamoracle-learn-spam
1155                          spam-spamoracle-unlearn-ham
1156                          spam-spamoracle-unlearn-spam)
1157     (spam-use-stat       spam-stat-register-ham-routine
1158                          spam-stat-register-spam-routine
1159                          spam-stat-unregister-ham-routine
1160                          spam-stat-unregister-spam-routine)
1161     ;; note that spam-use-gmane is not a legitimate check
1162     (spam-use-gmane      nil
1163                          spam-report-gmane-register-routine
1164                          ;; does Gmane support unregistration?
1165                          nil
1166                          nil)
1167     (spam-use-spamassassin spam-spamassassin-register-ham-routine
1168                            spam-spamassassin-register-spam-routine
1169                            spam-spamassassin-unregister-ham-routine
1170                            spam-spamassassin-unregister-spam-routine)
1171     (spam-use-bogofilter spam-bogofilter-register-ham-routine
1172                          spam-bogofilter-register-spam-routine
1173                          spam-bogofilter-unregister-ham-routine
1174                          spam-bogofilter-unregister-spam-routine))
1175   "The spam-registration-functions list contains pairs
1176 associating a parameter variable with the ham and spam
1177 registration functions, and the ham and spam unregistration
1178 functions")
1179
1180 (defun spam-classification-valid-p (classification)
1181   (or  (eq classification 'spam)
1182        (eq classification 'ham)))
1183
1184 (defun spam-process-type-valid-p (process-type)
1185   (or  (eq process-type 'incoming)
1186        (eq process-type 'process)))
1187
1188 (defun spam-registration-check-valid-p (check)
1189   (assoc check spam-registration-functions))
1190
1191 (defun spam-unregistration-check-valid-p (check)
1192   (assoc check spam-registration-functions))
1193
1194 (defun spam-registration-function (classification check)
1195   (let ((flist (cdr-safe (assoc check spam-registration-functions))))
1196     (if (eq classification 'spam)
1197         (nth 1 flist)
1198       (nth 0 flist))))
1199
1200 (defun spam-unregistration-function (classification check)
1201   (let ((flist (cdr-safe (assoc check spam-registration-functions))))
1202     (if (eq classification 'spam)
1203         (nth 3 flist)
1204       (nth 2 flist))))
1205
1206 (defun spam-list-articles (articles classification)
1207   (let ((mark-check (if (eq classification 'spam)
1208                         'spam-group-spam-mark-p
1209                       'spam-group-ham-mark-p))
1210         list mark-cache-yes mark-cache-no)
1211     (dolist (article articles)
1212       (let ((mark (gnus-summary-article-mark article)))
1213         (unless (memq mark mark-cache-no)
1214           (if (memq mark mark-cache-yes)
1215               (push article list)
1216             ;; else, we have to actually check the mark
1217             (if (funcall mark-check
1218                          gnus-newsgroup-name
1219                          mark)
1220                 (progn
1221                   (push article list)
1222                   (push mark mark-cache-yes))
1223               (push mark mark-cache-no))))))
1224     list))
1225
1226 (defun spam-register-routine (classification
1227                               check
1228                               &optional unregister
1229                               specific-articles)
1230   (when (and (spam-classification-valid-p classification)
1231              (spam-registration-check-valid-p check))
1232     (let* ((register-function
1233             (spam-registration-function classification check))
1234            (unregister-function
1235             (spam-unregistration-function classification check))
1236            (run-function (if unregister
1237                              unregister-function
1238                            register-function))
1239            (log-function (if unregister
1240                              'spam-log-undo-registration
1241                            'spam-log-processing-to-registry))
1242            article articles)
1243
1244       (when run-function
1245         ;; make list of articles, using specific-articles if given
1246         (setq articles (or specific-articles
1247                            (spam-list-articles
1248                             gnus-newsgroup-articles
1249                             classification)))
1250         ;; process them
1251         (gnus-message 5 "%s %d %s articles with classification %s, check %s"
1252                       (if unregister "Unregistering" "Registering")
1253                       (length articles)
1254                       (if specific-articles "specific" "")
1255                       (symbol-name classification)
1256                       (symbol-name check))
1257         (funcall run-function articles)
1258         ;; now log all the registrations (or undo them, depending on unregister)
1259         (dolist (article articles)
1260           (funcall log-function
1261                    (spam-fetch-field-message-id-fast article)
1262                    'process
1263                    classification
1264                    check
1265                    gnus-newsgroup-name))))))
1266
1267 ;;; log a ham- or spam-processor invocation to the registry
1268 (defun spam-log-processing-to-registry (id type classification check group)
1269   (when spam-log-to-registry
1270     (if (and (stringp id)
1271              (stringp group)
1272              (spam-process-type-valid-p type)
1273              (spam-classification-valid-p classification)
1274              (spam-registration-check-valid-p check))
1275         (let ((cell-list (cdr-safe (gnus-registry-fetch-extra id type)))
1276               (cell (list classification check group)))
1277           (push cell cell-list)
1278           (gnus-registry-store-extra-entry
1279            id
1280            type
1281            cell-list))
1282
1283       (gnus-error 5 (format "%s called with bad ID, type, classification, check, or group"
1284                             "spam-log-processing-to-registry")))))
1285
1286 ;;; check if a ham- or spam-processor registration has been done
1287 (defun spam-log-registered-p (id type)
1288   (when spam-log-to-registry
1289     (if (and (stringp id)
1290              (spam-process-type-valid-p type))
1291         (cdr-safe (gnus-registry-fetch-extra id type))
1292       (progn
1293         (gnus-error 5 (format "%s called with bad ID, type, classification, or check"
1294                               "spam-log-registered-p"))
1295         nil))))
1296
1297 ;;; check what a ham- or spam-processor registration says
1298 ;;; returns nil if conflicting registrations are found
1299 (defun spam-log-registration-type (id type)
1300   (let ((count 0)
1301         decision)
1302     (dolist (reg (spam-log-registered-p id type))
1303       (let ((classification (nth 0 reg)))
1304         (when (spam-classification-valid-p classification)
1305           (when (and decision
1306                      (not (eq classification decision)))
1307             (setq count (+ 1 count)))
1308           (setq decision classification))))
1309     (if (< 0 count)
1310         nil
1311       decision)))
1312
1313 ;;; check if a ham- or spam-processor registration needs to be undone
1314 (defun spam-log-unregistration-needed-p (id type classification check)
1315   (when spam-log-to-registry
1316     (if (and (stringp id)
1317              (spam-process-type-valid-p type)
1318              (spam-classification-valid-p classification)
1319              (spam-registration-check-valid-p check))
1320         (let ((cell-list (cdr-safe (gnus-registry-fetch-extra id type)))
1321               found)
1322           (dolist (cell cell-list)
1323             (unless found
1324               (when (and (eq classification (nth 0 cell))
1325                          (eq check (nth 1 cell)))
1326                 (setq found t))))
1327           found)
1328       (progn
1329         (gnus-error 5 (format "%s called with bad ID, type, classification, or check"
1330                               "spam-log-unregistration-needed-p"))
1331         nil))))
1332
1333
1334 ;;; undo a ham- or spam-processor registration (the group is not used)
1335 (defun spam-log-undo-registration (id type classification check &optional group)
1336   (when (and spam-log-to-registry
1337              (spam-log-unregistration-needed-p id type classification check))
1338     (if (and (stringp id)
1339              (spam-process-type-valid-p type)
1340              (spam-classification-valid-p classification)
1341              (spam-registration-check-valid-p check))
1342         (let ((cell-list (cdr-safe (gnus-registry-fetch-extra id type)))
1343               new-cell-list found)
1344           (dolist (cell cell-list)
1345             (unless (and (eq classification (nth 0 cell))
1346                          (eq check (nth 1 cell)))
1347               (push cell new-cell-list)))
1348           (gnus-registry-store-extra-entry
1349            id
1350            type
1351            new-cell-list))
1352       (progn
1353         (gnus-error 5 (format "%s called with bad ID, type, check, or group"
1354                               "spam-log-undo-registration"))
1355         nil))))
1356
1357 ;;; set up IMAP widening if it's necessary
1358 (defun spam-setup-widening ()
1359   (dolist (check spam-list-of-statistical-checks)
1360     (when (symbol-value check)
1361       (setq nnimap-split-download-body-default t))))
1362
1363 \f
1364 ;;;; Regex body
1365
1366 (defun spam-check-regex-body ()
1367   (let ((spam-regex-headers-ham spam-regex-body-ham)
1368         (spam-regex-headers-spam spam-regex-body-spam))
1369     (spam-check-regex-headers t)))
1370
1371 \f
1372 ;;;; Regex headers
1373
1374 (defun spam-check-regex-headers (&optional body)
1375   (let ((type (if body "body" "header"))
1376         (spam-split-group (if spam-split-symbolic-return
1377                               'spam
1378                             spam-split-group))
1379         ret found)
1380     (dolist (h-regex spam-regex-headers-ham)
1381       (unless found
1382         (goto-char (point-min))
1383         (when (re-search-forward h-regex nil t)
1384           (message "Ham regex %s search positive." type)
1385           (setq found t))))
1386     (dolist (s-regex spam-regex-headers-spam)
1387       (unless found
1388         (goto-char (point-min))
1389         (when (re-search-forward s-regex nil t)
1390           (message "Spam regex %s search positive." type)
1391           (setq found t)
1392           (setq ret spam-split-group))))
1393     ret))
1394
1395 \f
1396 ;;;; Blackholes.
1397
1398 (defun spam-reverse-ip-string (ip)
1399   (when (stringp ip)
1400     (mapconcat 'identity
1401                (nreverse (split-string ip "\\."))
1402                ".")))
1403
1404 (defun spam-check-blackholes ()
1405   "Check the Received headers for blackholed relays."
1406   (let ((headers (message-fetch-field "received"))
1407         (spam-split-group (if spam-split-symbolic-return
1408                               'spam
1409                             spam-split-group))
1410         ips matches)
1411     (when headers
1412       (with-temp-buffer
1413         (insert headers)
1414         (goto-char (point-min))
1415         (gnus-message 5 "Checking headers for relay addresses")
1416         (while (re-search-forward
1417                 "\\([0-9]+\\.[0-9]+\\.[0-9]+\\.[0-9]+\\)" nil t)
1418           (gnus-message 9 "Blackhole search found host IP %s." (match-string 1))
1419           (push (spam-reverse-ip-string (match-string 1))
1420                 ips)))
1421       (dolist (server spam-blackhole-servers)
1422         (dolist (ip ips)
1423           (unless (and spam-blackhole-good-server-regex
1424                        ;; match the good-server-regex against the reversed (again) IP string
1425                        (string-match
1426                         spam-blackhole-good-server-regex
1427                         (spam-reverse-ip-string ip)))
1428             (unless matches
1429               (let ((query-string (concat ip "." server)))
1430                 (if spam-use-dig
1431                     (let ((query-result (query-dig query-string)))
1432                       (when query-result
1433                         (gnus-message 5 "(DIG): positive blackhole check '%s'"
1434                                       query-result)
1435                         (push (list ip server query-result)
1436                               matches)))
1437                   ;; else, if not using dig.el
1438                   (when (query-dns query-string)
1439                     (gnus-message 5 "positive blackhole check")
1440                     (push (list ip server (query-dns query-string 'TXT))
1441                           matches)))))))))
1442     (when matches
1443       spam-split-group)))
1444 \f
1445 ;;;; Hashcash.
1446
1447 (condition-case nil
1448     (progn
1449       (require 'hashcash)
1450
1451       (defun spam-check-hashcash ()
1452         "Check the headers for hashcash payments."
1453         (mail-check-payment)))   ;mail-check-payment returns a boolean
1454
1455   (file-error (progn
1456                 (defalias 'mail-check-payment 'ignore)
1457                 (defalias 'spam-check-hashcash 'ignore))))
1458 \f
1459 ;;;; BBDB
1460
1461 ;;; original idea for spam-check-BBDB from Alexander Kotelnikov
1462 ;;; <sacha@giotto.sj.ru>
1463
1464 ;; all this is done inside a condition-case to trap errors
1465
1466 (condition-case nil
1467     (progn
1468       (require 'bbdb)
1469       (require 'bbdb-com)
1470
1471       ;; when the BBDB changes, we want to clear out our cache
1472       (defun spam-clear-cache-BBDB (&rest immaterial)
1473         (spam-clear-cache 'spam-use-BBDB))
1474
1475       (add-hook 'bbdb-change-hook 'spam-clear-cache-BBDB)
1476
1477       (defun spam-enter-ham-BBDB (addresses &optional remove)
1478         "Enter an address into the BBDB; implies ham (non-spam) sender"
1479         (dolist (from addresses)
1480           (when (stringp from)
1481             (let* ((parsed-address (gnus-extract-address-components from))
1482                    (name (or (nth 0 parsed-address) "Ham Sender"))
1483                    (remove-function (if remove
1484                                         'bbdb-delete-record-internal
1485                                       'ignore))
1486                    (net-address (nth 1 parsed-address))
1487                    (record (and net-address
1488                                 (bbdb-search-simple nil net-address))))
1489               (when net-address
1490                 (gnus-message 5 "%s address %s %s BBDB"
1491                               (if remove "Deleting" "Adding")
1492                               from
1493                               (if remove "from" "to"))
1494                 (if record
1495                     (funcall remove-function record)
1496                   (bbdb-create-internal name nil net-address nil nil
1497                                         "ham sender added by spam.el")))))))
1498
1499       (defun spam-BBDB-register-routine (articles &optional unregister)
1500         (let (addresses)
1501           (dolist (article articles)
1502             (when (stringp (spam-fetch-field-from-fast article))
1503               (push (spam-fetch-field-from-fast article) addresses)))
1504           ;; now do the register/unregister action
1505           (spam-enter-ham-BBDB addresses unregister)))
1506
1507       (defun spam-BBDB-unregister-routine (articles)
1508         (spam-BBDB-register-routine articles t))
1509
1510       (defun spam-check-BBDB ()
1511         "Mail from people in the BBDB is classified as ham or non-spam"
1512         (let ((who (message-fetch-field "from"))
1513               (spam-split-group (if spam-split-symbolic-return
1514                                     'spam
1515                                   spam-split-group))
1516               bbdb-cache bbdb-hashtable)
1517           (when spam-cache-lookups
1518             (setq bbdb-cache (gethash 'spam-use-BBDB spam-caches))
1519             (unless bbdb-cache
1520               (setq bbdb-cache
1521                     ;; this is the expanded (bbdb-hashtable) macro
1522                     ;; without the debugging support
1523                     (with-current-buffer (bbdb-buffer)
1524                       (save-excursion
1525                         (save-window-excursion
1526                           (bbdb-records nil t)
1527                           bbdb-hashtable))))
1528               (puthash 'spam-use-BBDB bbdb-cache spam-caches)))
1529           (when who
1530             (setq who (nth 1 (gnus-extract-address-components who)))
1531             (if
1532                 (if spam-cache-lookups
1533                     (symbol-value
1534                      (intern-soft who bbdb-cache))
1535                   (bbdb-search-simple nil who))
1536                 t
1537               (if spam-use-BBDB-exclusive
1538                   spam-split-group
1539                 nil))))))
1540
1541   (file-error (progn
1542                 (defalias 'bbdb-search-simple 'ignore)
1543                 (defalias 'bbdb-records 'ignore)
1544                 (defalias 'bbdb-buffer 'ignore)
1545                 (defalias 'spam-check-BBDB 'ignore)
1546                 (defalias 'spam-BBDB-register-routine 'ignore)
1547                 (defalias 'spam-enter-ham-BBDB 'ignore)
1548                 (defalias 'bbdb-create-internal 'ignore)
1549                 (defalias 'bbdb-delete-record-internal 'ignore)
1550                 (defalias 'bbdb-records 'ignore))))
1551
1552 \f
1553 ;;;; ifile
1554
1555 ;;; check the ifile backend; return nil if the mail was NOT classified
1556 ;;; as spam
1557
1558 (defun spam-get-ifile-database-parameter ()
1559   "Get the command-line parameter for ifile's database from
1560   spam-ifile-database-path."
1561   (if spam-ifile-database-path
1562       (format "--db-file=%s" spam-ifile-database-path)
1563     nil))
1564
1565 (defun spam-check-ifile ()
1566   "Check the ifile backend for the classification of this message."
1567   (let ((article-buffer-name (buffer-name))
1568         (spam-split-group (if spam-split-symbolic-return
1569                               'spam
1570                             spam-split-group))
1571         category return)
1572     (with-temp-buffer
1573       (let ((temp-buffer-name (buffer-name))
1574             (db-param (spam-get-ifile-database-parameter)))
1575         (save-excursion
1576           (set-buffer article-buffer-name)
1577           (apply 'call-process-region
1578                  (point-min) (point-max) spam-ifile-path
1579                  nil temp-buffer-name nil "-c"
1580                  (if db-param `(,db-param "-q") `("-q"))))
1581         ;; check the return now (we're back in the temp buffer)
1582         (goto-char (point-min))
1583         (if (not (eobp))
1584             (setq category (buffer-substring (point) (point-at-eol))))
1585         (when (not (zerop (length category))) ; we need a category here
1586           (if spam-ifile-all-categories
1587               (setq return category)
1588             ;; else, if spam-ifile-all-categories is not set...
1589             (when (string-equal spam-ifile-spam-category category)
1590               (setq return spam-split-group)))))) ; note return is nil otherwise
1591     return))
1592
1593 (defun spam-ifile-register-with-ifile (articles category &optional unregister)
1594   "Register an article, given as a string, with a category.
1595 Uses `gnus-newsgroup-name' if category is nil (for ham registration)."
1596   (let ((category (or category gnus-newsgroup-name))
1597         (add-or-delete-option (if unregister "-d" "-i"))
1598         (db (spam-get-ifile-database-parameter))
1599         parameters)
1600     (with-temp-buffer
1601       (dolist (article articles)
1602         (let ((article-string (spam-get-article-as-string article)))
1603           (when (stringp article-string)
1604             (insert article-string))))
1605       (apply 'call-process-region
1606              (point-min) (point-max) spam-ifile-path
1607              nil nil nil
1608              add-or-delete-option category
1609              (if db `(,db "-h") `("-h"))))))
1610
1611 (defun spam-ifile-register-spam-routine (articles &optional unregister)
1612   (spam-ifile-register-with-ifile articles spam-ifile-spam-category unregister))
1613
1614 (defun spam-ifile-unregister-spam-routine (articles)
1615   (spam-ifile-register-spam-routine articles t))
1616
1617 (defun spam-ifile-register-ham-routine (articles &optional unregister)
1618   (spam-ifile-register-with-ifile articles spam-ifile-ham-category unregister))
1619
1620 (defun spam-ifile-unregister-ham-routine (articles)
1621   (spam-ifile-register-ham-routine articles t))
1622
1623 \f
1624 ;;;; spam-stat
1625
1626 (condition-case nil
1627     (progn
1628       (let ((spam-stat-install-hooks nil))
1629         (require 'spam-stat))
1630
1631       (defun spam-check-stat ()
1632         "Check the spam-stat backend for the classification of this message"
1633         (let ((spam-split-group (if spam-split-symbolic-return
1634                                     'spam
1635                                   spam-split-group))
1636               (spam-stat-split-fancy-spam-group spam-split-group) ; override
1637               (spam-stat-buffer (buffer-name)) ; stat the current buffer
1638               category return)
1639           (spam-stat-split-fancy)))
1640
1641       (defun spam-stat-register-spam-routine (articles &optional unregister)
1642         (dolist (article articles)
1643           (let ((article-string (spam-get-article-as-string article)))
1644             (with-temp-buffer
1645               (insert article-string)
1646               (if unregister
1647                   (spam-stat-buffer-change-to-non-spam)
1648               (spam-stat-buffer-is-spam))))))
1649
1650       (defun spam-stat-unregister-spam-routine (articles)
1651         (spam-stat-register-spam-routine articles t))
1652
1653       (defun spam-stat-register-ham-routine (articles &optional unregister)
1654         (dolist (article articles)
1655           (let ((article-string (spam-get-article-as-string article)))
1656             (with-temp-buffer
1657               (insert article-string)
1658               (if unregister
1659                   (spam-stat-buffer-change-to-spam)
1660               (spam-stat-buffer-is-non-spam))))))
1661
1662       (defun spam-stat-unregister-ham-routine (articles)
1663         (spam-stat-register-ham-routine articles t))
1664
1665       (defun spam-maybe-spam-stat-load ()
1666         (when spam-use-stat (spam-stat-load)))
1667
1668       (defun spam-maybe-spam-stat-save ()
1669         (when spam-use-stat (spam-stat-save))))
1670
1671   (file-error (progn
1672                 (defalias 'spam-stat-load 'ignore)
1673                 (defalias 'spam-stat-save 'ignore)
1674                 (defalias 'spam-maybe-spam-stat-load 'ignore)
1675                 (defalias 'spam-maybe-spam-stat-save 'ignore)
1676                 (defalias 'spam-stat-register-ham-routine 'ignore)
1677                 (defalias 'spam-stat-unregister-ham-routine 'ignore)
1678                 (defalias 'spam-stat-register-spam-routine 'ignore)
1679                 (defalias 'spam-stat-unregister-spam-routine 'ignore)
1680                 (defalias 'spam-stat-buffer-is-spam 'ignore)
1681                 (defalias 'spam-stat-buffer-change-to-spam 'ignore)
1682                 (defalias 'spam-stat-buffer-is-non-spam 'ignore)
1683                 (defalias 'spam-stat-buffer-change-to-non-spam 'ignore)
1684                 (defalias 'spam-stat-split-fancy 'ignore)
1685                 (defalias 'spam-check-stat 'ignore))))
1686
1687 \f
1688
1689 ;;;; Blacklists and whitelists.
1690
1691 (defvar spam-whitelist-cache nil)
1692 (defvar spam-blacklist-cache nil)
1693
1694 (defun spam-kill-whole-line ()
1695   (beginning-of-line)
1696   (let ((kill-whole-line t))
1697     (kill-line)))
1698
1699 ;;; address can be a list, too
1700 (defun spam-enter-whitelist (address &optional remove)
1701   "Enter ADDRESS (list or single) into the whitelist.
1702 With a non-nil REMOVE, remove them."
1703   (interactive "sAddress: ")
1704   (spam-enter-list address spam-whitelist remove)
1705   (setq spam-whitelist-cache nil)
1706   (spam-clear-cache 'spam-use-whitelist))
1707
1708 ;;; address can be a list, too
1709 (defun spam-enter-blacklist (address &optional remove)
1710   "Enter ADDRESS (list or single) into the blacklist.
1711 With a non-nil REMOVE, remove them."
1712   (interactive "sAddress: ")
1713   (spam-enter-list address spam-blacklist remove)
1714   (setq spam-blacklist-cache nil)
1715   (spam-clear-cache 'spam-use-whitelist))
1716
1717 (defun spam-enter-list (addresses file &optional remove)
1718   "Enter ADDRESSES into the given FILE.
1719 Either the whitelist or the blacklist files can be used.  With
1720 REMOVE not nil, remove the ADDRESSES."
1721   (if (stringp addresses)
1722       (spam-enter-list (list addresses) file remove)
1723     ;; else, we have a list of addresses here
1724     (unless (file-exists-p (file-name-directory file))
1725       (make-directory (file-name-directory file) t))
1726     (save-excursion
1727       (set-buffer
1728        (find-file-noselect file))
1729       (dolist (a addresses)
1730         (when (stringp a)
1731           (goto-char (point-min))
1732           (if (re-search-forward (regexp-quote a) nil t)
1733               ;; found the address
1734               (when remove
1735                 (spam-kill-whole-line))
1736             ;; else, the address was not found
1737             (unless remove
1738               (goto-char (point-max))
1739               (unless (bobp)
1740                 (insert "\n"))
1741               (insert a "\n")))))
1742       (save-buffer))))
1743
1744 (defun spam-filelist-build-cache (type)
1745   (let ((cache (if (eq type 'spam-use-blacklist)
1746                    spam-blacklist-cache
1747                  spam-whitelist-cache))
1748         parsed-cache)
1749     (unless (gethash type spam-caches)
1750       (while cache
1751         (let ((address (pop cache)))
1752           (unless (zerop (length address)) ; 0 for a nil address too
1753             (setq address (regexp-quote address))
1754             ;; fix regexp-quote's treatment of user-intended regexes
1755             (while (string-match "\\\\\\*" address)
1756               (setq address (replace-match ".*" t t address))))
1757           (push address parsed-cache)))
1758       (puthash type parsed-cache spam-caches))))
1759
1760 (defun spam-filelist-check-cache (type from)
1761   (when (stringp from)
1762     (spam-filelist-build-cache type)
1763     (let (found)
1764       (dolist (address (gethash type spam-caches))
1765         (when (and address (string-match address from))
1766           (setq found t)
1767           (return)))
1768       found)))
1769
1770 ;;; returns t if the sender is in the whitelist, nil or
1771 ;;; spam-split-group otherwise
1772 (defun spam-check-whitelist ()
1773   ;; FIXME!  Should it detect when file timestamps change?
1774   (let ((spam-split-group (if spam-split-symbolic-return
1775                               'spam
1776                             spam-split-group)))
1777     (unless spam-whitelist-cache
1778       (setq spam-whitelist-cache (spam-parse-list spam-whitelist)))
1779     (if (spam-from-listed-p 'spam-use-whitelist)
1780         t
1781       (if spam-use-whitelist-exclusive
1782           spam-split-group
1783         nil))))
1784
1785 (defun spam-check-blacklist ()
1786   ;; FIXME!  Should it detect when file timestamps change?
1787   (let ((spam-split-group (if spam-split-symbolic-return
1788                               'spam
1789                             spam-split-group)))
1790     (unless spam-blacklist-cache
1791       (setq spam-blacklist-cache (spam-parse-list spam-blacklist)))
1792     (and (spam-from-listed-p 'spam-use-blacklist) spam-split-group)))
1793
1794 (defun spam-parse-list (file)
1795   (when (file-readable-p file)
1796     (let (contents address)
1797       (with-temp-buffer
1798         (insert-file-contents file)
1799         (while (not (eobp))
1800           (setq address (buffer-substring (point) (point-at-eol)))
1801           (forward-line 1)
1802           ;; insert the e-mail address if detected, otherwise the raw data
1803           (unless (zerop (length address))
1804             (let ((pure-address (nth 1 (gnus-extract-address-components address))))
1805               (push (or pure-address address) contents)))))
1806       (nreverse contents))))
1807
1808 (defun spam-from-listed-p (type)
1809   (let ((from (message-fetch-field "from"))
1810         found)
1811     (spam-filelist-check-cache type from)))
1812
1813 (defun spam-filelist-register-routine (articles blacklist &optional unregister)
1814   (let ((de-symbol (if blacklist 'spam-use-whitelist 'spam-use-blacklist))
1815         (declassification (if blacklist 'ham 'spam))
1816         (enter-function
1817          (if blacklist 'spam-enter-blacklist 'spam-enter-whitelist))
1818         (remove-function
1819          (if blacklist 'spam-enter-whitelist 'spam-enter-blacklist))
1820         from addresses unregister-list)
1821     (dolist (article articles)
1822       (let ((from (spam-fetch-field-from-fast article))
1823             (id (spam-fetch-field-message-id-fast article))
1824             sender-ignored)
1825         (when (stringp from)
1826           (dolist (ignore-regex spam-blacklist-ignored-regexes)
1827             (when (and (not sender-ignored)
1828                        (stringp ignore-regex)
1829                        (string-match ignore-regex from))
1830               (setq sender-ignored t)))
1831           ;; remember the messages we need to unregister, unless remove is set
1832           (when (and
1833                  (null unregister)
1834                  (spam-log-unregistration-needed-p
1835                   id 'process declassification de-symbol))
1836             (push from unregister-list))
1837           (unless sender-ignored
1838             (push from addresses)))))
1839
1840     (if unregister
1841         (funcall enter-function addresses t) ; unregister all these addresses
1842       ;; else, register normally and unregister what we need to
1843       (funcall remove-function unregister-list t)
1844       (dolist (article unregister-list)
1845         (spam-log-undo-registration
1846          (spam-fetch-field-message-id-fast article)
1847          'process
1848          declassification
1849          de-symbol))
1850       (funcall enter-function addresses nil))))
1851
1852 (defun spam-blacklist-unregister-routine (articles)
1853   (spam-blacklist-register-routine articles t))
1854
1855 (defun spam-blacklist-register-routine (articles &optional unregister)
1856   (spam-filelist-register-routine articles t unregister))
1857
1858 (defun spam-whitelist-unregister-routine (articles)
1859   (spam-whitelist-register-routine articles t))
1860
1861 (defun spam-whitelist-register-routine (articles &optional unregister)
1862   (spam-filelist-register-routine articles nil unregister))
1863
1864 \f
1865 ;;;; Spam-report glue
1866 (defun spam-report-gmane-register-routine (articles)
1867   (when articles
1868     (apply 'spam-report-gmane articles)))
1869
1870 \f
1871 ;;;; Bogofilter
1872 (defun spam-check-bogofilter-headers (&optional score)
1873   (let ((header (message-fetch-field spam-bogofilter-header))
1874         (spam-split-group (if spam-split-symbolic-return
1875                               'spam
1876                             spam-split-group)))
1877     (when header                        ; return nil when no header
1878       (if score                         ; scoring mode
1879           (if (string-match "spamicity=\\([0-9.]+\\)" header)
1880               (match-string 1 header)
1881             "0")
1882         ;; spam detection mode
1883         (when (string-match spam-bogofilter-bogosity-positive-spam-header
1884                             header)
1885           spam-split-group)))))
1886
1887 ;; return something sensible if the score can't be determined
1888 (defun spam-bogofilter-score ()
1889   "Get the Bogofilter spamicity score"
1890   (interactive)
1891   (save-window-excursion
1892     (gnus-summary-show-article t)
1893     (set-buffer gnus-article-buffer)
1894     (let ((score (or (spam-check-bogofilter-headers t)
1895                      (spam-check-bogofilter t))))
1896       (gnus-summary-show-article)
1897       (message "Spamicity score %s" score)
1898       (or score "0"))))
1899
1900 (defun spam-check-bogofilter (&optional score)
1901   "Check the Bogofilter backend for the classification of this message"
1902   (let ((article-buffer-name (buffer-name))
1903         (db spam-bogofilter-database-directory)
1904         return)
1905     (with-temp-buffer
1906       (let ((temp-buffer-name (buffer-name)))
1907         (save-excursion
1908           (set-buffer article-buffer-name)
1909           (apply 'call-process-region
1910                  (point-min) (point-max)
1911                  spam-bogofilter-path
1912                  nil temp-buffer-name nil
1913                  (if db `("-d" ,db "-v") `("-v"))))
1914         (setq return (spam-check-bogofilter-headers score))))
1915     return))
1916
1917 (defun spam-bogofilter-register-with-bogofilter (articles
1918                                                  spam
1919                                                  &optional unregister)
1920   "Register an article, given as a string, as spam or non-spam."
1921   (dolist (article articles)
1922     (let ((article-string (spam-get-article-as-string article))
1923           (db spam-bogofilter-database-directory)
1924           (switch (if unregister
1925                       (if spam
1926                           spam-bogofilter-spam-strong-switch
1927                         spam-bogofilter-ham-strong-switch)
1928                     (if spam
1929                         spam-bogofilter-spam-switch
1930                       spam-bogofilter-ham-switch))))
1931       (when (stringp article-string)
1932         (with-temp-buffer
1933           (insert article-string)
1934
1935           (apply 'call-process-region
1936                  (point-min) (point-max)
1937                  spam-bogofilter-path
1938                  nil nil nil switch
1939                  (if db `("-d" ,db "-v") `("-v"))))))))
1940
1941 (defun spam-bogofilter-register-spam-routine (articles &optional unregister)
1942   (spam-bogofilter-register-with-bogofilter articles t unregister))
1943
1944 (defun spam-bogofilter-unregister-spam-routine (articles)
1945   (spam-bogofilter-register-spam-routine articles t))
1946
1947 (defun spam-bogofilter-register-ham-routine (articles &optional unregister)
1948   (spam-bogofilter-register-with-bogofilter articles nil unregister))
1949
1950 (defun spam-bogofilter-unregister-ham-routine (articles)
1951   (spam-bogofilter-register-ham-routine articles t))
1952
1953
1954 \f
1955 ;;;; spamoracle
1956 (defun spam-check-spamoracle ()
1957   "Run spamoracle on an article to determine whether it's spam."
1958   (let ((article-buffer-name (buffer-name))
1959         (spam-split-group (if spam-split-symbolic-return
1960                               'spam
1961                             spam-split-group)))
1962     (with-temp-buffer
1963       (let ((temp-buffer-name (buffer-name)))
1964         (save-excursion
1965           (set-buffer article-buffer-name)
1966           (let ((status
1967                  (apply 'call-process-region
1968                         (point-min) (point-max)
1969                         spam-spamoracle-binary
1970                         nil temp-buffer-name nil
1971                         (if spam-spamoracle-database
1972                             `("-f" ,spam-spamoracle-database "mark")
1973                           '("mark")))))
1974             (if (eq 0 status)
1975                 (progn
1976                   (set-buffer temp-buffer-name)
1977                   (goto-char (point-min))
1978                   (when (re-search-forward "^X-Spam: yes;" nil t)
1979                     spam-split-group))
1980               (error "Error running spamoracle: %s" status))))))))
1981
1982 (defun spam-spamoracle-learn (articles article-is-spam-p &optional unregister)
1983   "Run spamoracle in training mode."
1984   (with-temp-buffer
1985     (let ((temp-buffer-name (buffer-name)))
1986       (save-excursion
1987         (goto-char (point-min))
1988         (dolist (article articles)
1989           (insert (spam-get-article-as-string article)))
1990         (let* ((arg (if (spam-xor unregister article-is-spam-p)
1991                         "-spam"
1992                       "-good"))
1993                (status
1994                 (apply 'call-process-region
1995                        (point-min) (point-max)
1996                        spam-spamoracle-binary
1997                        nil temp-buffer-name nil
1998                        (if spam-spamoracle-database
1999                            `("-f" ,spam-spamoracle-database
2000                              "add" ,arg)
2001                          `("add" ,arg)))))
2002           (unless (eq 0 status)
2003             (error "Error running spamoracle: %s" status)))))))
2004
2005 (defun spam-spamoracle-learn-ham (articles &optional unregister)
2006   (spam-spamoracle-learn articles nil unregister))
2007
2008 (defun spam-spamoracle-unlearn-ham (articles &optional unregister)
2009   (spam-spamoracle-learn-ham articles t))
2010
2011 (defun spam-spamoracle-learn-spam (articles &optional unregister)
2012   (spam-spamoracle-learn articles t unregister))
2013
2014 (defun spam-spamoracle-unlearn-spam (articles &optional unregister)
2015   (spam-spamoracle-learn-spam articles t))
2016
2017 \f
2018 ;;;; SpamAssassin
2019 ;;; based mostly on the bogofilter code
2020 (defun spam-check-spamassassin-headers (&optional score)
2021   "Check the SpamAssassin headers for the classification of this message."
2022   (if score                             ; scoring mode
2023       (let ((header (message-fetch-field spam-spamassassin-spam-status-header)))
2024         (when header
2025           (if (string-match "hits=\\(-?[0-9.]+\\)" header)
2026               (match-string 1 header)
2027             "0")))
2028     ;; spam detection mode
2029     (let ((header (message-fetch-field spam-spamassassin-spam-flag-header))
2030           (spam-split-group (if spam-split-symbolic-return
2031                                  'spam
2032                                spam-split-group)))
2033           (when header                  ; return nil when no header
2034             (when (string-match spam-spamassassin-positive-spam-flag-header
2035                                 header)
2036               spam-split-group)))))
2037
2038 (defun spam-check-spamassassin (&optional score)
2039   "Check the SpamAssassin backend for the classification of this message."
2040   (let ((article-buffer-name (buffer-name)))
2041     (with-temp-buffer
2042       (let ((temp-buffer-name (buffer-name)))
2043         (save-excursion
2044           (set-buffer article-buffer-name)
2045           (apply 'call-process-region
2046                  (point-min) (point-max) spam-spamassassin-path
2047                  nil temp-buffer-name nil spam-spamassassin-arguments))
2048         ;; check the return now (we're back in the temp buffer)
2049         (goto-char (point-min))
2050         (spam-check-spamassassin-headers score)))))
2051
2052 ;; return something sensible if the score can't be determined
2053 (defun spam-spamassassin-score ()
2054   "Get the SpamAssassin score"
2055   (interactive)
2056   (save-window-excursion
2057     (gnus-summary-show-article t)
2058     (set-buffer gnus-article-buffer)
2059     (let ((score (or (spam-check-spamassassin-headers t)
2060                      (spam-check-spamassassin t))))
2061       (gnus-summary-show-article)
2062       (message "SpamAssassin score %s" score)
2063       (or score "0"))))
2064
2065 (defun spam-spamassassin-register-with-sa-learn (articles spam
2066                                                  &optional unregister)
2067   "Register articles with spamassassin's sa-learn as spam or non-spam."
2068   (if articles
2069       (let ((action (if unregister spam-sa-learn-unregister-switch
2070                       (if spam spam-sa-learn-spam-switch
2071                         spam-sa-learn-ham-switch)))
2072             (summary-buffer-name (buffer-name)))
2073         (with-temp-buffer
2074           ;; group the articles into mbox format
2075           (dolist (article articles)
2076             (let (article-string)
2077               (save-excursion
2078                 (set-buffer summary-buffer-name)
2079                 (setq article-string (spam-get-article-as-string article)))
2080               (when (stringp article-string)
2081                 (insert "From \n") ; mbox separator (sa-learn only checks the
2082                                    ; first five chars, so we can get away with
2083                                    ; a bogus line))
2084                 (insert article-string)
2085                 (insert "\n"))))
2086           ;; call sa-learn on all messages at the same time
2087           (apply 'call-process-region
2088                  (point-min) (point-max)
2089                  spam-sa-learn-path
2090                  nil nil nil "--mbox"
2091                  (if spam-sa-learn-rebuild
2092                      (list action)
2093                    `("--no-rebuild" ,action)))))))
2094
2095 (defun spam-spamassassin-register-spam-routine (articles &optional unregister)
2096   (spam-spamassassin-register-with-sa-learn articles t unregister))
2097
2098 (defun spam-spamassassin-register-ham-routine (articles &optional unregister)
2099   (spam-spamassassin-register-with-sa-learn articles nil unregister))
2100
2101 (defun spam-spamassassin-unregister-spam-routine (articles)
2102   (spam-spamassassin-register-with-sa-learn articles t t))
2103
2104 (defun spam-spamassassin-unregister-ham-routine (articles)
2105   (spam-spamassassin-register-with-sa-learn articles nil t))
2106 \f
2107 ;;;; Hooks
2108
2109 ;;;###autoload
2110 (defun spam-initialize ()
2111   "Install the spam.el hooks and do other initialization"
2112   (interactive)
2113   (setq spam-install-hooks t)
2114   ;; TODO: How do we redo this every time spam-face is customized?
2115   (push '((eq mark gnus-spam-mark) . spam-face)
2116         gnus-summary-highlight)
2117   ;; Add hooks for loading and saving the spam stats
2118   (add-hook 'gnus-save-newsrc-hook 'spam-maybe-spam-stat-save)
2119   (add-hook 'gnus-get-top-new-news-hook 'spam-maybe-spam-stat-load)
2120   (add-hook 'gnus-startup-hook 'spam-maybe-spam-stat-load)
2121   (add-hook 'gnus-summary-prepare-exit-hook 'spam-summary-prepare-exit)
2122   (add-hook 'gnus-summary-prepare-hook 'spam-summary-prepare)
2123   (add-hook 'gnus-get-new-news-hook 'spam-setup-widening)
2124   (add-hook 'gnus-summary-prepared-hook 'spam-find-spam))
2125
2126 (defun spam-unload-hook ()
2127   "Uninstall the spam.el hooks"
2128   (interactive)
2129   (remove-hook 'gnus-save-newsrc-hook 'spam-maybe-spam-stat-save)
2130   (remove-hook 'gnus-get-top-new-news-hook 'spam-maybe-spam-stat-load)
2131   (remove-hook 'gnus-startup-hook 'spam-maybe-spam-stat-load)
2132   (remove-hook 'gnus-summary-prepare-exit-hook 'spam-summary-prepare-exit)
2133   (remove-hook 'gnus-summary-prepare-hook 'spam-summary-prepare)
2134   (remove-hook 'gnus-get-new-news-hook 'spam-setup-widening)
2135   (remove-hook 'gnus-summary-prepare-hook 'spam-find-spam))
2136
2137 (when spam-install-hooks
2138   (spam-initialize))
2139
2140 (provide 'spam)
2141
2142 ;;; spam.el ends here