eventmachine-email_server 0.0.2

data/lib/eventmachine/email_server/smtp_server.rb

@@ -0,0 +1,242 @@
+require 'eventmachine'
+require 'eventmachine/dnsbl'
+
+module EventMachine
+	module EmailServer
+		class SMTPServer < EventMachine::Connection
+      @@graylist = nil
+      @@dnsbl_check = nil
+      @@ratelimiter = nil
+      @@reverse_ptr_check = false
+      @@spf_check = false
+      @@reject_filters = Array.new
+      
+      def self.reverse_ptr_check(ptr=nil)
+        if not ptr.nil?
+          @@reverse_ptr_check = ptr
+        end
+        @@reverse_ptr_check
+      end
+      
+      def self.graylist(graylist=nil)
+        if graylist
+          @@graylist = graylist
+        end
+        @@graylist
+      end
+      
+      def self.dnsbl_check(dnsbl_check=nil)
+        if not dnsbl_check.nil?
+          @@dnsbl_check = dnsbl_check
+        end
+        @@dnsbl_check
+      end
+      
+      def self.ratelimiter(ratelimiter=nil)
+        if ratelimiter
+          @@ratelimiter = ratelimiter
+        end
+        @@ratelimiter
+      end
+      
+      def self.reject_filters(filters=nil)
+        if filters
+          @@reject_filters = filters
+        end
+        @@reject_filters
+      end
+      
+      def self.spf_check(spf=nil)
+        if not spf.nil?
+          @@spf_check = spf
+        end
+        @@spf_check
+      end 
+
+      def initialize(hostname, userstore, emailstore)
+        @hostname = hostname
+        @userstore = userstore
+        @emailstore = emailstore
+        @debug = true
+        @data_mode = false
+        @email_body = ""
+        @ptr_ok = true
+        @dnsbl_ok = true
+        @rate_ok = true
+        @gray_ok = true
+        @reject_ok = true
+        @pending_checks = Array.new
+      end
+      
+      def post_init
+        puts ">> 220 hello" if @debug
+        send_data "220 #{@hostname} ESMTP Service ready\n"
+      end
+      
+			def receive_data(data)
+        puts ">> #{data}" if @debug
+        data.split(/\n/).each do |line|
+          ok, op = process_line(line+"\n")
+          if op
+            puts "<< #{op}" if @debug
+            send_data(op+"\r\n")
+          end
+        end
+      end
+      
+      def check_ptr(helo, ip)
+        if @@reverse_ptr_check
+          @ptr_ok = false
+          @pending_checks << :ptr
+          d = EM::DNS::Resolver.resolve helo
+          d.callback { |r|
+            @ptr_ok = r.include?(ip)
+            @pending_checks -= [:ptr]
+            if @pending_checks.length == 0
+              send_answer
+            end
+          }
+        end
+      end
+      
+      def check_dnsbl(ip)
+        if @@dnsbl_check
+          @dnsbl_ok = false
+          @pending_checks << :dnsbl
+          EventMachine::DNSBL::Client.check(ip) do |results|
+            @dnsbl_ok = ! EventMachine::DNSBL::Client.blacklisted?(results)
+            @pending_checks -= [:dnsbl]
+            if @pending_checks.length == 0
+              send_answer
+            end
+          end
+        end
+      end
+      
+      def check_ratelimit(ip)
+        if @@ratelimiter
+          @rate_ok = @@ratelimiter.use(ip)
+        end
+      end
+      
+      def check_gray(ip)
+        if @@graylist
+          @gray_ok = @@graylist.has_key?(ip)
+          @@graylist[ip] = true
+        end
+      end
+      
+      def check_reject
+        @@reject_filters.each do |filter|
+          if filter.match(@email_body)
+            @reject_ok = false
+            return
+          end
+        end
+      end
+      
+      def check_spf(from_domain)
+        if @@spf_check
+          @spf_ok = false
+          @pending_checks << :spf
+          d = EM::DNS::Resolver.resolve(from_domain, Resolv::DNS::Resource::IN::TXT)
+          d.errback { |r|
+            # fail open?
+            @spf_ok = true
+            @pending_checks -= [:spf]
+            if @pending_checks.length == 0
+              send_answer
+            end
+          }
+          d.callback { |r|
+            r.each do |rec|
+              #pp rec
+              if rec.start_with?('v=spf1')
+                if rec == "v=spf1 -all"
+                  @spf_ok = false
+                else
+                  # I need to create an SPF checker now :(
+                  @spf_ok = true
+                end
+              end
+            end
+            @pending_checks -= [:spf]
+            if @pending_checks.length == 0
+              send_answer
+            end
+          }
+        end
+      end
+      
+      def send_answer
+        if @ptr_ok and @rate_ok and @gray_ok and @reject_ok and @dnsbl_ok and @spf_ok
+          ans = "250 OK"
+        else
+          ans = "451 Requested action aborted: local error in processing"
+        end
+        puts "<< #{ans}" if @debug
+        send_data(ans+"\r\n")
+      end
+      
+     	def process_line(line)
+    		if (@data_mode) && (line.chomp =~ /^\.$/)
+    			@data_mode = false
+          check_reject
+          if @pending_checks.length == 0
+            send_answer
+          end
+          return true, nil
+    		elsif @data_mode
+    			@email_body += line
+    			return true, nil
+    		elsif (line =~ /^(HELO|EHLO) (.*)/)
+          helo = $2.chomp
+          port, ip = Socket.unpack_sockaddr_in(get_peername)
+          check_ptr(helo, ip)
+          check_dnsbl(ip)
+          check_gray(ip)
+          check_ratelimit(ip)
+    			return true, "250 hello #{ip} (#{helo})"
+    		elsif (line =~ /^QUIT/)
+    			return false, "221 bye bye"
+    		elsif (line =~ /^MAIL FROM\:/)
+    			@mail_from = (/^MAIL FROM\:<(.+)>.*$/).match(line)[1]
+          if @@spf_check
+            check_spf(@mail_from.split(/@/,2)[1])
+          end
+    			return true, "250 OK"
+    		elsif (line =~ /^RCPT TO\:/)
+    			rcpt_to = (/^RCPT TO\:<(.+)>.*$/).match(line)[1]
+    			if @userstore.user_by_emailaddress(rcpt_to.strip)
+    				@rcpt_to = rcpt_to
+    				return true, "250 OK"
+    			end
+    			return false, "550 No such user here"
+    		elsif (line =~ /^DATA/)
+    			if @rcpt_to
+    				@data_mode = true
+    				@email_body = ''
+    				return true, "354 Enter message, ending with \".\" on a line by itself"
+    			end
+    			return true, "500 ERROR"
+    		else
+    			return true, "500 ERROR"
+    		end
+    	end
+      
+    	def save
+    		begin
+    			subject = @email_body.match(/Subject\: (.*?)[\r\n]/i)[1]
+    			u = @userstore.user_by_emailaddress(@rcpt_to.strip)
+    		rescue Exception => err
+    			puts err
+    			return
+    		end
+    		if u and @mail_from and @rcpt_to
+    			subject ||= ''
+          @emailstore << Email.new(nil, @mail_from, @rcpt_to, subject, @email_body, u.id)
+    		end
+    	end
+    end
+  end
+end

data/lib/eventmachine/email_server/sqlite3.rb

@@ -0,0 +1,153 @@
+require_relative 'base'
+require 'sqlite3'
+
+module EventMachine
+  module EmailServer
+    class Sqlite3UserStore < AbstractUserStore
+      def initialize(sqlite3, tablename = "users")
+        @class = User
+        @fields = @class.members.map {|x| x.to_s}.join(", ")
+        @tablename = tablename
+        if sqlite3.class == SQLite3::Database
+          @db = sqlite3
+        else
+          @db = SQLite3::Database.new(sqlite3)
+        end
+        if @db.table_info(tablename).length == 0
+          @db.execute("CREATE TABLE #{@tablename} (#{@fields})")
+        end
+      end
+      
+      def add_user(user)
+        if user.id
+          u = user_by_id(user.id)
+        end
+        if u
+          @db.execute("UPDATE #{@tablename} SET username=?, password=?, address=? WHERE id=?", 
+            user.username,
+            user.password,
+            user.address,
+            user.id)
+        else
+          @db.execute("INSERT INTO #{@tablename} (id,username,password,address) VALUES (?,?,?,?)", 
+            user.id,
+            user.username,
+            user.password,
+            user.address)
+        end
+      end
+      
+      def delete_user(user)
+        if user.id
+          @db.execute("DELETE FROM #{@tablename} WHERE id = ?", user.id)
+        end
+      end
+
+      def user_by_field(field, value)
+    		rs = @db.execute("SELECT #{@fields} FROM #{@tablename} WHERE #{field}='#{value}'")
+    		return nil unless rs
+    		rs.each do |row|
+    			return User.new(*row)
+    		end
+        return nil
+      end
+        
+      def user_by_username(username)
+        user_by_field("username", username)
+      end
+      
+      def user_by_emailaddress(address)
+        user_by_field("address", address)
+      end
+      
+      def user_by_id(id)
+        user_by_field("id", id)
+      end
+    end
+    
+    class Sqlite3EmailStore < AbstractEmailStore
+      def initialize(sqlite3, tablename = "emails")
+        @class = Email
+        @fields = "'"+@class.members.map {|x| x.to_s}.join("', '")+"'"
+        @tablename = tablename
+        if sqlite3.class == SQLite3::Database
+          @db = sqlite3
+        else
+          @db = SQLite3::Database.new(sqlite3)
+        end
+        if @db.table_info(tablename).length == 0
+          fields = @fields.gsub(/'id'/, 'id integer primary key autoincrement')
+          @db.execute("CREATE TABLE #{@tablename} (#{fields})")
+        end
+      end
+      
+      def emails_by_field(field, value)
+    		rs = @db.execute("SELECT #{@fields} FROM #{@tablename} WHERE ?=?", field, value)
+    		return nil unless rs
+        emails = Array.new
+    		rs.each do |row|
+    			emails << Email.new(*row)
+    		end
+        emails
+      end
+      
+      def emails_by_userid(uid)
+        emails_by_field("uid", uid)
+      end
+      
+      def quote( string )
+        string.gsub( /'/, "''" )
+      end
+      private :quote
+      
+      def save_email(email)
+        if email.id
+          # I'm being too crafty here.. this is bad style
+          args = (@class.members - [:id]).map{|f| email.send(f)}
+          args << email.send(:id)
+          @db.execute("UPDATE #{@tablename} SET " +
+            (@class.members - [:id]).map { |field| "#{field} = ?"}.join(", ") +
+            " WHERE id = ?", *args)
+        else
+          email.id = "NULL"
+          args = (@class.members).map{|f| email.send(f)}
+          qs = args.map{|x| "'#{quote(x.to_s)}'"}.join(",").gsub(/'NULL'/, "NULL")
+          @db.execute("INSERT INTO #{@tablename} (#{@fields}) VALUES (#{qs})")
+          rs = @db.execute("SELECT last_insert_rowid()")
+          rs.each do |row|
+            email.id = *row
+          end
+        end
+        email.id
+      end
+      
+      def delete_email(email)
+        if email.id
+          @db.execute("DROP FROM #{@tablename} WHERE id = ?", email.id)
+        end
+      end
+      
+      def delete_by_field(field, value)
+        @db.execute("DROP FROM #{@tablename} WHERE #{field} = ?", value)
+      end
+      
+      def delete_id(id)
+        delete_by_field("id", id)
+      end
+      
+      def delete_user(uid)
+        delete_by_field("uid", uid)
+      end
+      
+      def count
+        sql = "SELECT COUNT(*) FROM #{@tablename}"
+        rs = @db.execute(sql)
+        c = 0
+        rs.each do |row|
+          c = row[0]
+        end
+        c
+      end
+    end
+  end
+end

data/lib/eventmachine/email_server/version.rb

@@ -0,0 +1,5 @@
+module EventMachine
+  module EmailServer
+    VERSION = "0.0.2"
+  end
+end

data/lib/eventmachine/email_server.rb

@@ -0,0 +1,8 @@
+require "eventmachine/email_server/version"
+require 'eventmachine/email_server/base'
+require 'eventmachine/email_server/memory'
+require 'eventmachine/email_server/null'
+require 'eventmachine/email_server/sqlite3'
+require 'eventmachine/email_server/pop3_server'
+require 'eventmachine/email_server/smtp_server'
+require 'eventmachine/email_server/eventmachine_dns_monkeypatch'

data/test/helper.rb

@@ -0,0 +1,5 @@
+require 'minitest/autorun'
+require 'minitest/test'
+require 'minitest/unit'
+include MiniTest::Assertions
+require File.expand_path('../../lib/eventmachine/email_server.rb', __FILE__)