maildiode-greylist 0.0.3

data/lib/plugins/greylist.rb

@@ -0,0 +1,137 @@
+# Copyright 2007-2008 Kevin B. Smith
+# This file is part of MailDiode.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3, as 
+# published by the Free Software Foundation.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+require 'kirbybase'
+
+ONE_HOUR_IN_SECONDS = 60*60
+
+module MailDiode
+	class GreylistFilter < Filter
+	#  Traditional (client IP, sender address, recipient address)
+	#  or selective (client IP, HELO name, sender domain)? 
+		GREYLIST_ADDED = "450 You are being put on a waiting list--please try again later"
+		GREYLIST_WAITING = "450 You are still on the waiting list--please try later"
+		GREYLIST_DELAY_IN_SECONDS = 60
+		
+		def initialize(settings)
+			@delay_minutes = 60
+			load_settings(settings)
+			@kirby = KirbyBase.new(:local, nil, nil, '/var/local/maildiode/')
+			ensure_database_table_exists
+		end
+		
+		def delay_seconds
+			@settings.delay_minutes * 60
+		end
+		
+		def process(filterable_data)
+			ip = filterable_data.sender_ip
+			helo = filterable_data.helo
+			from = filterable_data.sender
+			to = filterable_data.recipient
+			
+			if is_bounce_message(from)
+				MailDiode::log_info "Greylist allowing bounce message"
+				return true
+			end
+			from_user, from_domain = from.split('@')
+			table = get_database_table
+			found = table.select { | r | (r.ip == ip && r.helo == helo && r.from_domain = from_domain) }
+			case found.size
+				when 0
+					record = create_record(ip, helo, from_user, from_domain, to)
+					MailDiode::log_info "Greylist added #{key(record)}"
+					raise SMTPError.new(GREYLIST_ADDED)
+				when 1
+					if !check_existing_record(found[0])
+						raise SMTPError.new(GREYLIST_WAITING)
+					end
+				else
+					MailDiode::log_warning "Greylist found multiple entries for #{ip};#{helo};#{from};#{to}"
+			end
+		end
+		
+		def is_bounce_message(from)
+			return from.empty?
+		end
+		
+		def key(record)
+			return "#{record.ip};#{record.helo};#{record.from_domain}"
+		end
+		
+		def create_record(ip, helo, from_user, from_domain, to)
+			recno = get_database_table.insert do |r|
+				r.ip = ip
+				r.helo = helo
+				r.from_user = from_user
+				r.from_domain = from_domain
+				r.to = to
+				r.first_seen = now
+				r.last_seen = now
+				r.approved = false
+				record = r
+			end
+			return get_database_table.select { | r | r.recno == recno }
+		end
+		
+		def check_existing_record(record)
+			if(record.approved)
+				MailDiode::log_info "Greylist approved"
+				return true
+			end
+			
+			table = get_database_table
+			delay_until = record.first_seen + delay_seconds
+			if now > delay_until
+				table.update(:approved => true) { | r | r.recno == record.recno }
+				MailDiode::log_info "Greylist approved #{key(record)}"
+				return true
+			end
+			
+			table.update(:last_seen => now) { | r | r.recno == record.recno }
+			MailDiode::log_info "Greylist delaying #{key(record)} until #{Time.at(delay_until)}"
+			return false
+		end
+		
+		def now
+			return Time.now.to_i
+		end
+		
+		def get_database_table
+			return @kirby.get_table(:greylist)
+		end
+		
+		def ensure_database_table_exists
+			if @kirby.table_exists?(:greylist)
+				return
+			end
+			MailDiode::log_info "Creating greylist database"
+			@kirby.create_table(:greylist, 
+				:ip, :String, 
+				:helo, :String,
+				:from_user, :String,
+				:from_domain, :String,
+				:to, :String,
+				:first_seen, :Integer,
+				:last_seen, :Integer,
+				:approved, :Boolean
+				)
+		end
+
+		def load_settings(settings)
+			@delay_minutes = settings.get_setting('greylist', 'delayminutes').to_i
+		end
+	end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification 
+name: maildiode-greylist
+version: &id001 !ruby/object:Gem::Version 
+  version: 0.0.3
+platform: ruby
+authors: 
+- Kevin Smith
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-12-03 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: maildiode
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - *id001
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: KirbyBase
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.6.0
+    version: 
+description: 
+email: kevins@qualitycode.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/plugins/greylist.rb
+has_rdoc: false
+homepage: http://maildiode.rubyforge.org/
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: 1.8.1
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: maildiode
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Greylist anti-spam plugin for MailDiode
+test_files: []
+