diff options
| -rwxr-xr-x | ipsilon/login/authtest.py | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/ipsilon/login/authtest.py b/ipsilon/login/authtest.py new file mode 100755 index 0000000..289a493 --- /dev/null +++ b/ipsilon/login/authtest.py @@ -0,0 +1,154 @@ +#!/usr/bin/python +# +# Copyright (C) 2014 Simo Sorce <simo@redhat.com> +# +# see file 'COPYING' for use and warranty information +# +# 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/>. + +from ipsilon.login.common import LoginPageBase, LoginManagerBase +from ipsilon.login.common import FACILITY +from ipsilon.util.plugin import PluginObject +import cherrypy + + +class TestAuth(LoginPageBase): + + def GET(self, *args, **kwargs): + context = self.create_tmpl_context() + # pylint: disable=star-args + return self._template('login/pam.html', **context) + + def POST(self, *args, **kwargs): + username = kwargs.get("login_name") + password = kwargs.get("login_password") + error = None + + if username and password: + if password == 'ipsilon': + cherrypy.log("User %s successfully authenticated." % username) + return self.lm.auth_successful(username) + else: + cherrypy.log("User %s failed authentication." % username) + error = "Authentication failed" + else: + error = "Username or password is missing" + cherrypy.log.error("Error: " + error) + + context = self.create_tmpl_context( + username=username, + error=error, + error_password=not password, + error_username=not username + ) + # pylint: disable=star-args + return self._template('login/pam.html', **context) + + def root(self, *args, **kwargs): + op = getattr(self, cherrypy.request.method, self.GET) + if callable(op): + return op(*args, **kwargs) + + def create_tmpl_context(self, **kwargs): + next_url = None + if self.lm.next_login is not None: + next_url = self.lm.next_login.path + + context = { + "title": 'TEST Login', + "action": '%s/login/testauth' % self.basepath, + "username_text": self.lm.username_text, + "password_text": self.lm.password_text, + "description": self.lm.help_text, + "next_url": next_url, + } + context.update(kwargs) + return context + + +class LoginManager(LoginManagerBase): + + def __init__(self, *args, **kwargs): + super(LoginManager, self).__init__(*args, **kwargs) + self.name = 'testauth' + self.path = 'testauth' + self.page = None + self.description = """ +Form based TEST login Manager, DO NOT EVER ACTIVATE IN PRODUCTION """ + self._options = { + 'help text': [ + """ The text shown to guide the user at login time. """, + 'string', + 'Insert your Username and Password and then submit.' + ], + 'username text': [ + """ The text shown to ask for the username in the form. """, + 'string', + 'Username' + ], + 'password text': [ + """ The text shown to ask for the password in the form. """, + 'string', + 'Password' + ], + } + + @property + def help_text(self): + return self.get_config_value('help text') + + @property + def username_text(self): + return self.get_config_value('username text') + + @property + def password_text(self): + return self.get_config_value('password text') + + def get_tree(self, site): + self.page = TestAuth(site, self) + return self.page + + +class Installer(object): + + def __init__(self): + self.name = 'testauth' + self.ptype = 'login' + + def install_args(self, group): + group.add_argument('--testauth', choices=['yes', 'no'], default='no', + help='Configure PAM authentication') + + def configure(self, opts): + if opts['testauth'] != 'yes': + return + + # Add configuration data to database + po = PluginObject() + po.name = 'testauth' + po.wipe_data() + + # Update global config to add login plugin + po = PluginObject() + po.name = 'global' + globalconf = po.get_plugin_config(FACILITY) + if 'order' in globalconf: + order = globalconf['order'].split(',') + else: + order = [] + order.append('testauth') + globalconf['order'] = ','.join(order) + po.set_config(globalconf) + po.save_plugin_config(FACILITY) |