#! /usr/bin/env python # -*- coding: ISO-8859-15 -*- # An email gateway for PyKota # # PyKota - Print Quotas for CUPS and LPRng # # (c) 2003, 2004, 2005 Jerome Alet # 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. # # $Id$ # # $Log$ # Revision 1.4 2005/02/22 17:36:58 jalet # Fixed the command's help # # Revision 1.3 2005/01/17 08:44:24 jalet # Modified copyright years # # Revision 1.2 2004/11/21 22:16:11 jalet # Added some kind of protection against bad guys # # Revision 1.1 2004/11/21 21:50:03 jalet # Introduced the new pkmail command as a simple email gateway # import sys import os import popen2 import smtplib import email from pykota.tool import PyKotaTool, PyKotaToolError, crashed, N_ from pykota.pdlanalyzer import PDLAnalyzer, PDLAnalyzerError __doc__ = N_("""pkmail v%s (c) 2003, 2004, 2005 C@LL - Conseil Internet & Logiciels Libres Email gateway for PyKota. command line usage : pkmail [options] options : -v | --version Prints pkmail's version number then exits. -h | --help Prints this message then exits. This command is meant to be used from your mail server's aliases file, as a pipe. It will then accept commands send to it in email messages, and will send the answer to the command's originator. To use this command, create an email alias in /etc/aliases with the following format : pykotacmd: "|/usr/bin/pkmail" Then run the 'newaliases' command to regenerate the aliases database. You can now send commands by email to 'pykotacmd@yourdomain.com', with the command in the subject. List of supported commands : report [username] NB : For pkmail to work correctly, you may have to put the 'mail' system user in the 'pykota' system group to ensure this user can read the /etc/pykota/pykotadmin.conf file, and restart your mail server (e.g. /etc/init.d/exim restart). It is strongly advised that you think at least twice before doing this though. Use at your own risk ! 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. Please e-mail bugs to: %s""") class PKMail(PyKotaTool) : """A class for pkmail.""" def main(self, files, options) : """Accepts commands passed in an email message.""" data = sys.stdin.read() message = email.message_from_string(data) useremail = message["From"] whoami = message["To"] cmdargs = message["Subject"].split() try : (command, arguments) = (cmdargs[0].capitalize(), cmdargs[1:]) except IndexError : raise PyKotaToolError, "No command found !" badchars = """/<>&"'#!%*$,;\\""" cheatmeonce = 0 for c in "".join(arguments) : if c in badchars : cheatmeonce = 1 if cheatmeonce : self.logdebug("Possible intruder at %s : %s" % (useremail, str(arguments))) result = "Either you mistyped your command, or you're a bad guy !" else : self.logdebug("Launching internal command '%s' with arguments %s" % (command, str(arguments))) cmdfunc = getattr(self, "cmd%s" % command, self.cmdDefault) result = cmdfunc(arguments) self.logdebug("Sending answer to : %s" % useremail) server = smtplib.SMTP(self.smtpserver) server.sendmail(whoami, [ useremail ], "From: %s\nTo: %s\nSubject: Result of your commands\n\n%s\n" % (whoami, useremail, result)) server.quit() self.logdebug("Answer sent to : %s" % useremail) return 0 def runCommand(self, cmd) : """Executes a command.""" self.logdebug("Launching : '%s'" % cmd) os.environ["PATH"] = "%s:/bin:/usr/bin:/usr/local/bin:/opt/bin:/sbin:/usr/sbin" % os.environ.get("PATH", "") child = popen2.Popen3(cmd) child.tochild.close() result = child.fromchild.read() status = child.wait() if os.WIFEXITED(status) : status = os.WEXITSTATUS(status) self.logdebug("'%s' exited with status %s" % (cmd, status)) return result def cmdDefault(self, arguments) : """Default command : sends an 'unknown command' message.""" return "Unknown command." def cmdReport(self, args) : """Generates a print quota report.""" return self.runCommand("repykota %s" % ' '.join([ '"%s"' % a for a in args ])) if __name__ == "__main__" : retcode = 0 try : defaults = { \ } short_options = "vh" long_options = ["help", "version"] # Initializes the command line tool mailparser = PKMail(doc=__doc__) # parse and checks the command line (options, args) = mailparser.parseCommandline(sys.argv[1:], short_options, long_options, allownothing=1) # sets long options options["help"] = options["h"] or options["help"] options["version"] = options["v"] or options["version"] if options["help"] : mailparser.display_usage_and_quit() elif options["version"] : mailparser.display_version_and_quit() else : retcode = mailparser.main(args, options) except SystemExit : pass except : try : mailparser.crashed("pkmail failed") except : crashed("pkmail failed") retcode = -1 try : mailparser.storage.close() except (TypeError, NameError, AttributeError) : pass sys.exit(retcode)