Project

General

Profile

Download (8.24 KB) Statistics
| Branch: | Tag: | Revision:
55a68712 Marc Dequenes
#!/usr/bin/ruby -Ku

d32ee48a Marc Dequenes
#--
# CyborgHood, a distributed system management software.
# Copyright (c) 2009 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/>.
#++

d4b5798a Marc Dequènes (Duck)
# to allow in-place run for test
$: << File.join(File.dirname(__FILE__), "..", "lib")
55a68712 Marc Dequenes
#require 'socket'
157c68c9 Marc Dequenes
require 'tempfile'
55a68712 Marc Dequenes
require 'shellwords'
44648f83 Marc Dequènes (Duck)
require 'cyborghood'
bc4894ce Marc Dequenes
require 'cyborghood/imap'
c427bfc7 Marc Dequenes
require 'cyborghood/mail'
cb0d68fd Marc Dequènes (Duck)
require 'cyborghood/mail_order'
7d9d434c Marc Dequènes (Duck)
require 'cyborghood/command_runner'
3f7a1eee Marc Dequenes
55a68712 Marc Dequenes
#Socket.gethostname

5f013d0a Marc Dequènes (Duck)
dfc48308 Marc Dequenes
module CyborgHood
ccab26de Marc Dequènes (Duck)
module PostmanHome
ed09e1e5 Marc Dequènes (Duck)
include I18nTranslation
d4b5798a Marc Dequènes (Duck)
bindtextdomain("cyborghood_postman", {:path => Config::L10N_DIR, :charset => "UTF-8"})
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
# not yet ready to be a real Cyborg
class Postman #< Cyborg
ed09e1e5 Marc Dequènes (Duck)
include I18nTranslation
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
def initialize
# load config
Config.load(self.human_name.downcase)
@config = Config.instance
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
ldap_config = @config.ldap
ldap_config.logger = logger
c8996a8d Marc Dequènes (Duck)
ActiveLdap::Base.setup_connection(ldap_config.marshal_dump)
ccab26de Marc Dequènes (Duck)
# setup logs
unless @config.log.nil?
logger.output_level(@config.log.console_level) unless @config.log.console_level.nil?
logger.log_to_file(@config.log.file) unless @config.log.file.nil?
5f013d0a Marc Dequènes (Duck)
end
ccab26de Marc Dequènes (Duck)
@current_thread = Thread.current
@stop_asap = false
@waiting = false

logger.info "Bot '#{self.human_name}' loaded"
5f013d0a Marc Dequènes (Duck)
end

ccab26de Marc Dequènes (Duck)
def run
imap = IMAP.new(@config.imap)
until @stop_asap
t = Time.now.to_i
logger.debug "Starting mail check"
check_mails(imap)
logger.debug "Mail check finished"
t2 = Time.now.to_i
sleep_time = @config.imap.min_check_interval - (t2 - t)
if sleep_time > 0
logger.debug "Having a break before new check..."
@waiting = true
begin
sleep(sleep_time)
rescue
end
@waiting = false
end
5f013d0a Marc Dequènes (Duck)
end
ccab26de Marc Dequènes (Duck)
logger.info "Bot was asked to stop..." if @stop_asap
logger.info "Bot terminating"
end
5f013d0a Marc Dequènes (Duck)
c75f6985 Marc Dequènes (Duck)
def ask_to_stop
@stop_asap = true
Thread.critical = true
@current_thread.raise if @waiting
Thread.critical = false
end

private

ccab26de Marc Dequènes (Duck)
def check_mails(imap)
imap.check_mail do |msg|
if @stop_asap
logger.info "Bot was asked to stop..."
break
end
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
mail = Mail.new(msg.content)
logger.info "Received mail with ID '#{mail.message_id}': #{mail.from_addrs} -> #{mail.to_addrs} (#{mail.subject})"
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
# 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
5f013d0a Marc Dequènes (Duck)
next
end
ccab26de Marc Dequènes (Duck)
logger.debug "RFC3156 content detected"
begin
cb0d68fd Marc Dequènes (Duck)
report = mail.process
ccab26de Marc Dequènes (Duck)
rescue CyberError => e
case e.severity
1b7ef95e Marc Dequènes (Duck)
when :grave
cb0d68fd Marc Dequènes (Duck)
logger.fatal "Fatal processing error, exiting (#{e.message})"
ccab26de Marc Dequènes (Duck)
exit 2
when :unrecoverable
logger.error "Internal processing error, skipping mail (#{e.message})"
next
1b7ef95e Marc Dequènes (Duck)
when :processable
logger.error "Untreated processing problem, skipping mail (#{e.message})"
next
ccab26de Marc Dequènes (Duck)
when :ignorable
1b7ef95e Marc Dequènes (Duck)
logger.warn "Internal processing warning, continuing (#{e.message})"
ccab26de Marc Dequènes (Duck)
end
end
cb0d68fd Marc Dequènes (Duck)
result_tag = report.ok? ? "SUCCESS" : "FAILURE"
eb6e0359 Marc Dequènes (Duck)
result_msg = "Processing result: #{result_tag}"
result_msg += " (#{report.error.untranslated})" unless report.ok?
logger.info result_msg
cb0d68fd Marc Dequènes (Duck)
ed09e1e5 Marc Dequènes (Duck)
i18n = I18nController.instance
83a04d84 Marc Dequènes (Duck)
i18n.set_language_for_user(report.user)
5f013d0a Marc Dequènes (Duck)
cb0d68fd Marc Dequènes (Duck)
unless report.ok?
if report.warn_sender
ccab26de Marc Dequènes (Duck)
logger.info "Sending reply for rejected message"
eb6e0359 Marc Dequènes (Duck)
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_signature())
ccab26de Marc Dequènes (Duck)
mail_reply.deliver
end
msg.delete
next
5f013d0a Marc Dequènes (Duck)
end

cb0d68fd Marc Dequènes (Duck)
order = MailOrder.parse(report.user, report.message)
result_tag = order.valid? ? "SUCCESS" : "FAILURE"
eb6e0359 Marc Dequènes (Duck)
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)
cb0d68fd Marc Dequènes (Duck)
unless order.valid?
logger.info "Sending reply for rejected order"
eb6e0359 Marc Dequènes (Duck)
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_signature())
cb674fb0 Marc Dequènes (Duck)
mail_reply.sign_and_crypt(order.user.keyFingerPrint)
cb0d68fd Marc Dequènes (Duck)
mail_reply.deliver
msg.delete
next
end

ccab26de Marc Dequènes (Duck)
logger.debug "Message accepted, processing orders..."
f0a75e9c Marc Dequènes (Duck)
result_list = CommandRunner.run(order)
ccab26de Marc Dequènes (Duck)
# create transcript
logger.debug "Preparing reply"
eb6e0359 Marc Dequènes (Duck)
reply_txt = reply_intro.to_s + "\n\n"
reply_txt += _("Follows the transcript of your commands:").to_s + "\n"
ccab26de Marc Dequènes (Duck)
reply_attachments = []
result_list.each do |result|
7d701a64 Marc Dequènes (Duck)
reply_txt += "> #{result.cmd}\n"
reply_txt += "#{result.message}\n"
ccab26de Marc Dequènes (Duck)
reply_attachments += result.refs unless result.refs.nil?
end
7d701a64 Marc Dequènes (Duck)
reply_txt += "\n" + mail_signature()
5f013d0a Marc Dequènes (Duck)
ccab26de Marc Dequènes (Duck)
# create mail
logger.debug "Preparing mail"
mail_reply = mail.create_reply
if reply_attachments.empty?
transcript_part = mail_reply
else
mail_reply.set_content_type("multipart", "mixed", {'boundary' => TMail.new_boundary})
parts = []
5f013d0a Marc Dequènes (Duck)
p = CyborgHood::Mail.new
ccab26de Marc Dequènes (Duck)
transcript_part = p
5f013d0a Marc Dequènes (Duck)
mail_reply.parts << p
ccab26de Marc Dequènes (Duck)
reply_attachments.each do |attachment|
p = CyborgHood::Mail.new
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
5f013d0a Marc Dequènes (Duck)
end
ccab26de Marc Dequènes (Duck)
# 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

# send reply
logger.debug "Sending mail"
mail_reply.sign_and_crypt(order.user.keyFingerPrint)
mail_reply.deliver

logger.info "Message processed completely, deleting"
msg.delete
5f013d0a Marc Dequènes (Duck)
end
end

c75f6985 Marc Dequènes (Duck)
def mail_signature
s = "\n" +
"-- \n" +
"#{CyborgHood::PRODUCT} v#{CyborgHood::VERSION}\n"
eb6e0359 Marc Dequènes (Duck)
s += _("Contact eMail:").to_s + " \"#{@config.contact.name}\" <#{@config.contact.email}>\n" if @config.contact.email
s += _("Contact URL:").to_s + " #{@config.contact.url}\n" if @config.contact.url
c75f6985 Marc Dequènes (Duck)
s
ccab26de Marc Dequènes (Duck)
end
5f013d0a Marc Dequènes (Duck)
end
55a68712 Marc Dequenes
end
end

ccab26de Marc Dequènes (Duck)
bot = CyborgHood::PostmanHome::Postman.new
55a68712 Marc Dequenes
trap('INT') do
bot.ask_to_stop
end
trap('TERM') do
bot.ask_to_stop
end

bot.run