--- /dev/null
+# -*- coding: iso-8859-1 -*-
+"""
+ MoinMoin - auth plugin doing a check against MySQL db
+
+ @copyright: 2008 Matthijs Kooijman
+ @license: GNU GPL, see COPYING for details.
+"""
+
+import MySQLdb
+import md5
+from MoinMoin import user
+from MoinMoin.auth import BaseAuth, ContinueLogin
+from MoinMoin.datastruct.backends import LazyGroupsBackend, LazyGroup
+from MoinMoin import log
+logging = log.getLogger(__name__)
+
+
+class PhpbbGroupsBackend(LazyGroupsBackend):
+ class PhpbbGroup(LazyGroup):
+ pass
+
+
+ def __init__(self, request, auth):
+ super(LazyGroupsBackend, self).__init__(request)
+
+ self.auth = auth
+ self.request = request
+
+ def __iter__(self):
+ return self.list_query("SELECT group_name \
+ FROM `%sgroups` \
+ WHERE group_single_user = 0"
+ % self.auth.phpbb_prefix)
+
+ def __contains__(self, group_name):
+ return self.single_query("SELECT EXISTS ( \
+ SELECT * \
+ FROM `%sgroups` \
+ WHERE group_single_user = 0 \
+ AND group_name=%%s)" % self.auth.phpbb_prefix,
+ group_name)
+
+ def __getitem__(self, group_name):
+ return self.PhpbbGroup(self.request, group_name, self)
+
+ def _iter_group_members(self, group_name):
+ return self.list_query ("SELECT username \
+ FROM `%susers` as u, `%suser_group` as ug, `%sgroups` as g \
+ WHERE u.user_id = ug.user_id AND ug.group_id = g.group_id \
+ AND ug.user_pending = 0 AND g.group_single_user = 0 \
+ AND g.group_name = %%s"
+ % (self.auth.phpbb_prefix, self.auth.phpbb_prefix, self.auth.phpbb_prefix),
+ group_name)
+
+ def _group_has_member(self, group_name, member):
+ return self.single_query ("SELECT EXISTS( \
+ SELECT * \
+ FROM `%susers` as u, `%suser_group` as ug, `%sgroups` as g \
+ WHERE u.user_id = ug.user_id AND ug.group_id = g.group_id \
+ AND ug.user_pending = 0 AND g.group_single_user = 0 \
+ AND g.group_name = %%s AND u.username = %%s)"
+ % (self.auth.phpbb_prefix, self.auth.phpbb_prefix, self.auth.phpbb_prefix),
+ (group_name, member))
+
+ def groups_with_member(self, member):
+ return self.list_query ("SELECT g.group_name \
+ FROM `%susers` as u, `%suser_group` as ug, `%sgroups` as g \
+ WHERE u.user_id = ug.user_id AND ug.group_id = g.group_id \
+ AND ug.user_pending = 0 AND g.group_single_user = 0 \
+ AND u.username = %%s"
+ % (self.auth.phpbb_prefix, self.auth.phpbb_prefix, self.auth.phpbb_prefix),
+ member)
+
+ def single_query(self, *args):
+ """
+ Runs an SQL query, that returns single row with a single column.
+ Returns just that single result.
+ """
+ conn = None
+ cursor = None
+ try:
+ conn = self.auth.connect(self.request)
+ cursor = conn.cursor()
+ cursor.execute(*args)
+
+ return cursor.fetchone()[0]
+ finally:
+ if cursor:
+ cursor.close()
+ if conn:
+ conn.close()
+
+ def list_query(self, *args):
+ """
+ Runs an SQL query, that returns any number of single-column rows.
+ Returns the results as a list of single values
+ """
+ conn = None
+ cursor = None
+ try:
+ conn = self.auth.connect(self.request)
+ cursor = conn.cursor()
+ cursor.execute(*args)
+
+ for row in cursor:
+ yield row[0]
+ finally:
+ if cursor:
+ cursor.close()
+ if conn:
+ conn.close()
+
+class PhpbbAuth(BaseAuth):
+ logout_possible = True
+ login_inputs = ['username', 'password']
+
+ def __init__(self, name='phpbb', dbhost=None, dbuser=None, dbpass=None, dbname=None, dbport=None, phpbb_prefix='', hint=None):
+ """
+ Authenticate using credentials from a phpbb database
+
+ The name parameter should be unique among all authentication methods.
+
+ The hint parameter is a snippet of HTML that is displayed below the login form.
+ """
+ self.dbhost = dbhost
+ self.dbuser = dbuser
+ self.dbpass = dbpass
+ self.dbname = dbname
+ self.dbport = dbport
+ self.phpbb_prefix = phpbb_prefix
+ self.name = name
+ self.hint = hint
+
+ # Create a "constructor" to create a phpbb_groups instance, while
+ # passing ourselves to it.
+ self.groups_backend = lambda config, request: PhpbbGroupsBackend(request, self)
+
+ def check_login(self, request, username, password):
+ """ Checks the given username password combination. Returns the
+ corresponding emailaddress, or False if authentication failed.
+ """
+ conn = self.connect(request)
+
+ if not conn:
+ return False
+
+ # Get some data. Note that we interpolate the prefix ourselves, since
+ # letting the mysql library do it only works with values (it adds ''
+ # automatically). Note also that this allows possible SQL injection
+ # through the phpbb_prefix variable, but that should be a trusted
+ # value anyway.
+ cursor = conn.cursor ()
+ cursor.execute ("SELECT user_password,user_email FROM `%susers` WHERE username=%%s" % self.phpbb_prefix, username)
+
+ # No data? No login.
+ if (cursor.rowcount == 0):
+ conn.close()
+ return False
+
+ # Check password
+ row = cursor.fetchone()
+ conn.close()
+
+ if (md5.new(password).hexdigest() == row[0]):
+ return row[1]
+ else:
+ return False
+
+ def connect(self, request):
+ # This code was shamelessly stolen from
+ # django.db.backends.mysql.base.cursor
+ kwargs = {
+ 'charset': 'utf8',
+ 'use_unicode': False,
+ }
+ if self.dbuser:
+ kwargs['user'] = self.dbuser
+ if self.dbname:
+ kwargs['db'] = self.dbname
+ if self.dbpass:
+ kwargs['passwd'] = self.dbpass
+ if self.dbhost.startswith('/'):
+ kwargs['unix_socket'] = self.dbhost
+ elif self.dbhost:
+ kwargs['host'] = self.dbhost
+ if self.dbport:
+ kwargs['port'] = int(self.dbport)
+
+ # End stolen code
+
+ try:
+ conn = MySQLdb.connect (**kwargs)
+ except:
+ import sys
+ import traceback
+ info = sys.exc_info()
+ logging.error("phpbb_login: authentication failed due to exception connecting to DB, traceback follows...")
+ logging.error(''.join(traceback.format_exception(*info)))
+ return False
+
+ return conn
+
+ def login(self, request, user_obj, **kw):
+ try:
+ username = kw.get('username')
+ password = kw.get('password')
+
+ logging.debug("phpbb_login: Trying to log in, username=%r " % (username))
+
+ # simply continue if something else already logged in
+ # successfully
+ if user_obj and user_obj.valid:
+ return ContinueLogin(user_obj)
+
+ # Deny empty username or passwords
+ if not username or not password:
+ return ContinueLogin(user_obj)
+
+ email = self.check_login(request, username, password)
+
+ # Login incorrect
+ if (not email):
+ logging.debug("phpbb_login: authentication failed for %s" % (username))
+ return ContinueLogin(user_obj)
+
+ logging.debug("phpbb_login: authenticated %s (email %s)" % (username, email))
+
+ u = user.User(request, auth_username=username, auth_method=self.name, auth_attribs=('name', 'password', 'email'))
+ u.email = email
+ #u.remember_me = 0 # 0 enforces cookie_lifetime config param
+ u.create_or_update(True)
+
+ return ContinueLogin(u)
+ except:
+ import sys
+ import traceback
+ info = sys.exc_info()
+ logging.error("phpbb_login: authentication failed due to unexpected exception, traceback follows...")
+ logging.error(''.join(traceback.format_exception(*info)))
+ return ContinueLogin(user_obj)
+
+ def login_hint(self, request):
+ """ Return a snippet of HTML that is displayed with the login form. """
+ return self.hint
+
+# vim: set sw=4 expandtab sts=4:vim