root/branch/ggpolo/umitCore/Email.py @ 833

# Copyright (C) 2005-2007 Insecure.Com LLC.
#
# Authors: Adriano Monteiro Marques <py.adriano@gmail.com>
#          Frederico Silva Ribeiro <ribeiro.fsilva@gmail.com>
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

import re
from smtplib import SMTP
from socket import sslerror

class Email(object):
    """This class is intended to provide a simple interface for sending
    Umit emails. The mailing system is going to be used for mailing results
    to users, automatically. The arguments you should care about setting are as follows:
    
    from_addr - The address from which the email should be sent (Required)
    from_passwd - If the mail server doesn't do relay, you should authenticate on it
    to_addr - The email address to where the scan result should be sent (Required)
    server - The address of the SMTP server which will deliver the email (Required)
    localdomain - If you want to send your email using your own domain, set it here.
    msg - The scan result or msg you want to send (Required)
    tls - True if the server require tls. False if not.
    port - The port which the SMTP server is listening
    subject - The subject of the mail (Required)
    """

    def __init__(self):
        self.email_server = None
        self.from_passwd = None
        self.server = ""
        self.localdomain = ""
        self.subject = ""
        self.re_email = re.compile("[a-z0-9_-]+(\.[a-z0-9_-]+)*@[a-z0-9_-]+(\.[a-z0-9_-]+)+")

    def sendmail(self):
        self.email_server = SMTP(self.server, self.port, self.localdomain)
        self.email_server.ehlo(self.from_addr)
        if self.tls:
            self.email_server.starttls()
            self.email_server.ehlo(self.from_addr)

        if self.from_passwd:
            self.email_server.login(self.from_addr, self.from_passwd)

        try:
            self.email_server.sendmail(self.from_addr, self.to_addr, self.msg)
            return True
        except sslerror:
            return True
        except:
            return False

    def quit(self):
        self.email_server.quit()

    def __verify_email(self, email):
        if self.re_email.match(email):
            return True
        return False

    def get_from_addr(self):
        '''From Address: The email from which the message is going to be sent.
        '''
        return self.__from_addr

    def set_from_addr(self, from_addr):
        '''
        '''
        assert self.__verify_email(from_addr)
        self.__from_addr = from_addr

    def get_to_addr(self):
        '''
        '''
        return self.__to_addr

    def set_to_addr(self, to_addr):
        '''
        '''
        assert self.__verify_email(to_addr)
        self.__to_addr = to_addr

    def get_msg(self):
        '''
        '''
        return """From: %s\r
To: %s\r
Subject: %s\r
\r
%s
\n.\n""" % (self.from_addr, self.to_addr, self.subject, self.__msg)

    def set_msg(self, msg):
        '''
        '''
        self.__msg = msg

    def get_tls(self):
        '''
        '''
        if type(self.__tls) == type(True):
            return self.__tls
        return False

    def set_tls(self, tls):
        '''
        '''
        if type(tls) == type(True):
            self.__tls = tls
        else:
            raise Exception("TLS should be True or False!")

    def get_port(self):
        '''
        '''
        if self.__port != None:
            return self.__port
        return 0

    def set_port(self, port):
        '''
        '''
        self.__port = port


    from_addr = property(get_from_addr, set_from_addr)
    to_addr = property(get_to_addr, set_to_addr)
    msg = property(get_msg, set_msg)
    tls = property(get_tls, set_tls)
    port = property(get_port, set_port)

    __from_addr = None
    __to_addr = None
    __msg = None
    __tls = None
    __port = None

    
if __name__ == "__main__":
    msg = '''From: yourgmailhere@gmail.com\r
To: targetgmailhere@gmail.com\r
Subject: Farting a lot!\r
\r

Gotcha!

\n.\n'''
    
    email = SMTP("smtp.gmail.com")
    email.ehlo("yourgmailhere@gmail.com")
    email.starttls()
    email.ehlo("yourgmailhere@gmail.com")
    email.login("yourgmailhere@gmail.com", "yourpasswdhere")
    email.sendmail("yourgmailhere@gmail.com", "targetgmailhere@gmail.com", msg)
    email.quit()