Project

General

Profile

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

# http://www.ruby-doc.org/stdlib/libdoc/net/imap/rdoc/index.html

$: << "./lib"

#require 'socket'
#require 'fileutils'
#require 'tempfile'
require 'shellwords'
bc4894ce Marc Dequenes
require 'cyborghood/imap'
c427bfc7 Marc Dequenes
require 'cyborghood/mail'
bc4894ce Marc Dequenes
require 'cyborghood/objects'
4153cce4 Marc Dequenes
require 'cyborghood/services/dns'
3f7a1eee Marc Dequenes
55a68712 Marc Dequenes
#Socket.gethostname

#
# TODO:
# - should be able to handle encrypted messages for user to send sensitive data (postman would need a GPG key too)
#

class CommandParser
c427bfc7 Marc Dequenes
def self.run(order)
4153cce4 Marc Dequenes
result_list = []
c427bfc7 Marc Dequenes
order.commands.each do |cmd|
logger.info "Executing command: #{cmd}"
3f7a1eee Marc Dequenes
begin
4153cce4 Marc Dequenes
result_list << execute_cmd(order.user, cmd, order.refs)
3f7a1eee Marc Dequenes
rescue
logger.info "Command failed: " + $!
end
55a68712 Marc Dequenes
end
4153cce4 Marc Dequenes
result_list
55a68712 Marc Dequenes
end

private

c427bfc7 Marc Dequenes
def self.execute_cmd(user, cmdstr, refs)
55a68712 Marc Dequenes
cmdline = Shellwords.shellwords(cmdstr)
subsys = cmdline.shift

4153cce4 Marc Dequenes
result = OpenStruct.new
result.cmd = cmdstr
55a68712 Marc Dequenes
ok = true
case subsys.upcase
when "DNS"
case cmdline.shift.upcase
when "INFO"
591ec1a2 Marc Dequenes
if cmdline.empty?
c427bfc7 Marc Dequenes
list = CyborgHood::DnsDomain.find_by_manager(user)
4153cce4 Marc Dequenes
txt_list = list.collect{|z| z.cn }.sort.join(", ")
logger.info "User is manager of the following zones: " + txt_list
result.message = "You are manager of the following zones: " + txt_list
591ec1a2 Marc Dequenes
else
ok = false
end
55a68712 Marc Dequenes
when "GET"
case cmdline.shift.upcase
when "ZONE"
zone = cmdline.shift.downcase
c427bfc7 Marc Dequenes
dom = CyborgHood::DnsDomain.new(zone)
591ec1a2 Marc Dequenes
logger.info "User requesting zone content for '#{zone}'"
if dom.hosted?
if dom.managed_by? user
logger.info "User is manager of the zone"
4153cce4 Marc Dequenes
srv_dns = CyborgHood::Services::DNS.new(zone)
result.message = "Requested zone content attached."
result.refs = [srv_dns.read_zone]
591ec1a2 Marc Dequenes
else
logger.info "User is not allowed to manage the zone"
4153cce4 Marc Dequenes
result.message = "You are not allowed to manage this zone."
591ec1a2 Marc Dequenes
end
else
logger.info "Zone not hosted"
4153cce4 Marc Dequenes
result.message "This zone is not hosted here."
591ec1a2 Marc Dequenes
end
else
ok = false
end
55a68712 Marc Dequenes
when "SET"
4153cce4 Marc Dequenes
result.message = "Command not yet implemented."
55a68712 Marc Dequenes
else
ok = false
end
else
ok = false
end

if not ok
4153cce4 Marc Dequenes
result.message = "Command not recognized"
result.refs = nil
3f7a1eee Marc Dequenes
logger.info "Command not recognized: #{cmdstr}"
55a68712 Marc Dequenes
end
4153cce4 Marc Dequenes
result
55a68712 Marc Dequenes
end
end

4153cce4 Marc Dequenes
# imap.store(message_id, "+FLAGS", [:Deleted])
# imap.expunge()

55a68712 Marc Dequenes
module CyborgHood
# not yet ready to be a real Cyborg
class Postman #< Cyborg
def initialize
# load config
Config.load(self.human_name.downcase)
@config = Config.instance

0af9cada Marc Dequenes
ldap_config = @config.ldap
ldap_config.logger = logger
ActiveLdap::Base.establish_connection(ldap_config.marshal_dump)

55a68712 Marc Dequenes
# 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?
end

125a6103 Marc Dequenes
@stop_asap = false

55a68712 Marc Dequenes
logger.info "Bot '#{self.human_name}' loaded"
end

def run
945de171 Marc Dequenes
imap = IMAP.new(@config.imap)
imap.check_mail do |msg|
125a6103 Marc Dequenes
if @stop_asap
logger.info "Bot was asked to stop..."
break
end

c427bfc7 Marc Dequenes
mail = Mail.new(msg)
275e20ec Marc Dequenes
logger.info "Received mail with ID '#{mail.message_id}': #{mail.from_addrs} -> #{mail.to_addrs} (#{mail.subject})"
c427bfc7 Marc Dequenes
55a68712 Marc Dequenes
# ignore mails not signed
591ec1a2 Marc Dequenes
unless mail.is_pgp_signed?
c427bfc7 Marc Dequenes
logger.info "Mail not signed or not RFC3156 compliant, ignoring..."
55a68712 Marc Dequenes
next
591ec1a2 Marc Dequenes
end
55a68712 Marc Dequenes
logger.debug "Proper signed content detected"
c427bfc7 Marc Dequenes
begin
order = mail.parse
rescue CyberError => e
945de171 Marc Dequenes
case e.severity
when :dangerous
logger.fatal " (#{e.message})"
exit 2
when :unrecoverable
logger.error "Internal processing error, skipping mail (#{e.message})"
next
when :ignorable
end
55a68712 Marc Dequenes
end
c427bfc7 Marc Dequenes
if order.nil?
logger.info "Mail is invalid, ignoring..."
next
end

4153cce4 Marc Dequenes
result_list = CommandParser.run(order)
275e20ec Marc Dequenes
mail_reply = mail.create_reply
4153cce4 Marc Dequenes
reply_txt = "Hello #{order.user.cn},\n\nFollows the transcript of your commands:\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
if reply_attachments.empty?
mail_reply.set_content_type("text", "plain")
mail_reply.set_disposition("inline")
30406d66 Marc Dequenes
mail_reply.quoted_printable_body = reply_txt
4153cce4 Marc Dequenes
else
56793619 Marc Dequenes
mail_reply.set_content_type("multipart", "mixed", {'boundary' => TMail.new_boundary})
4153cce4 Marc Dequenes
parts = []

30406d66 Marc Dequenes
p = CyborgHood::Mail.new
4153cce4 Marc Dequenes
p.set_content_type("text", "plain", {'charset' => "utf-8"})
p.set_disposition("inline")
30406d66 Marc Dequenes
p.quoted_printable_body = reply_txt
4153cce4 Marc Dequenes
mail_reply.parts << p

reply_attachments.each do |attachment|
30406d66 Marc Dequenes
p = CyborgHood::Mail.new
4153cce4 Marc Dequenes
p.set_content_type("text", "plain", {'charset' => "utf-8"})
p.set_disposition("attachment", {'filename' => "test.rb"})
30406d66 Marc Dequenes
p.quoted_printable_body = attachment
4153cce4 Marc Dequenes
mail_reply.parts << p
end
56793619 Marc Dequenes
end
mail_reply.crypt(order.user.keyFingerPrint)
275e20ec Marc Dequenes
mail_reply.deliver
55a68712 Marc Dequenes
end
end

def ask_to_stop
125a6103 Marc Dequenes
@stop_asap = true
55a68712 Marc Dequenes
end
end
end

bot = CyborgHood::Postman.new

trap('INT') do
bot.ask_to_stop
end
trap('TERM') do
bot.ask_to_stop
end

bot.run