Project

General

Profile

Download (7.13 KB) Statistics
| Branch: | Tag: | Revision:
#!/usr/bin/ruby -Ku

#--
# CyborgHood, a distributed system management software.
# Copyright (c) 2009-2010 Marc Dequènes (Duck) <Duck@DuckCorp.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# 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/>.
#++

# to allow in-place run for test
$: << File.join(File.dirname(__FILE__), "..", "lib")

require 'cyborghood/cyborg'
require 'cyborghood-postman/mail'
require 'cyborghood-postman/mail_order'


module CyborgHood
module PostmanLand
include I18nTranslation
bindtextdomain("cyborghood_postman", {:path => Config::L10N_DIR, :charset => "UTF-8"})

class Postman < Cyborg
# load config before modules are included
Config.load(self.human_name)

include BotNet

def setup
super

define_interface "0.1~"
end

include I18nTranslation

def start_work
self.services.imap.check_mails do |msg|
begin
# Postman is not yet ready to process messages using the botnet
#process_message(msg)
rescue CyberError => e
raise
rescue
logger.error "Mail processing crashed: " + $!
logger.error "Crash trace: " + $!.backtrace.join("\n")
true
end
end
end

def stop(condition)
super do
self.services.imap.stop_mail_check
end
end

protected

def ready_to_stop?
# TODO: also wait for messages being processed to be finished (or try to interrupt cleanly if possible in :quickly condition)
super and not self.services.imap.connected?
end

private

def process_message(msg)
mail = Mail.new(msg.content)
logger.info "Received mail with ID '#{mail.message_id}': #{mail.from_pretty} -> #{mail.to_pretty} (#{mail.subject_pretty})"

# ignore mails not signed or encrypted
unless mail.is_pgp_signed? or mail.is_pgp_encrypted?
logger.info "Mail not signed/encrypted or not RFC3156 compliant, ignoring..."
msg.delete
return true
end

logger.debug "RFC3156 content detected"
begin
report = mail.process
rescue CyberError => e
case e.severity
when :grave
logger.fatal "Fatal processing error, exiting (#{e.message})"
exit 2
when :unrecoverable
logger.error "Internal processing error, skipping mail (#{e.message})"
return true
when :processable
logger.error "Untreated processing problem, skipping mail (#{e.message})"
return true
when :ignorable
logger.warn "Internal processing warning, continuing (#{e.message})"
end
end
result_tag = report.ok? ? "SUCCESS" : "FAILURE"
result_msg = "Processing result: #{result_tag}"
result_msg += " (#{report.error.untranslated})" unless report.ok?
logger.info result_msg

i18n = I18nController.instance
i18n.set_language_for_user(report.user)

unless report.ok?
if report.warn_sender
logger.info "Sending reply for rejected message"
reply_intro = report.user ? _("Hello %{cn},", :cn =>report.user.cn) : _("Hello,")
mail_reply = mail.create_simple_reject_reply(reply_intro.to_s + "\n\n" +
_("A message (ID: %{id}), apparently from you, was rejected for the following reason:",
:id => mail.message_id).to_s + "\n " + report.error.to_s + "\n")
mail_reply.deliver
end
msg.delete
return true
end

mail_parser = MailOrderParser.new(report.user)
order = mail_parser.parse(report.message)
result_tag = order.valid? ? "SUCCESS" : "FAILURE"
result_msg = "Processing result: #{result_tag}"
result_msg += " (#{order.error.untranslated})" unless order.valid?
logger.info result_msg

reply_intro = _("Hello %{cn},", :cn => order.user.cn)

unless order.valid?
logger.info "Sending reply for rejected order"
mail_reply = mail.create_simple_reject_reply(reply_intro.to_s + "\n\n" +
_("An order, in a message (ID: %{id}) from you, was rejected for the following reason:",
:id => mail.message_id).to_s + "\n " + order.error.to_s + "\n")
mail_reply.deliver
msg.delete
return true
end

logger.debug "Message accepted, processing orders..."
result_list = CommandRunner.run(order)

# create transcript and attachments
logger.debug "Preparing reply"
reply_txt = reply_intro.to_s + "\n\n"
reply_txt += _("Follows the transcript of your commands:").to_s + "\n"
reply_attachments = []
result_list.each do |result|
reply_txt += "> #{result.cmd}\n"
reply_txt += "#{result.message}\n"
reply_attachments += result.refs unless result.refs.nil?
end
reply_txt += "\n"

# create mail
logger.debug "Preparing mail"
mail_reply = mail.create_reply
if reply_attachments.empty?
transcript_part = mail_reply
else
mail_reply.to_multipart!

p = CyborgHood::Mail.blank
transcript_part = p
mail_reply.parts << p

reply_attachments.each do |attachment|
p = CyborgHood::Mail.blank
p.set_content_type("text", "plain", {'charset' => "utf-8"})
p.set_disposition("attachment", {'filename' => attachment.filename})
p.quoted_printable_body = attachment.content
mail_reply.parts << p
end
end

# insert transcript
transcript_part.set_content_type("text", "plain", {'charset' => 'utf-8', 'format' => 'flowed'})
transcript_part.set_disposition("inline")
transcript_part.quoted_printable_body = reply_txt + mail_reply.default_body_signature

# send reply
logger.debug "Sending mail"
mail_reply.sign_and_crypt
mail_reply.deliver

logger.info "Message processed completely, deleting"
msg.delete

true
end
end
end
end

reg = Needle::Registry.new
reg.define do |b|
b.require 'cyborghood', CyborgHood
b.require 'cyborghood-postman/land', CyborgHood::PostmanLand

b.bot { CyborgHood::PostmanLand::Postman.new(b.postman_land) }
end

bot = reg.bot

trap('INT') do
bot.stop(:quickly)
end
trap('TERM') do
bot.stop(:quickly)
end

bot.run
(5-5/7)