← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/authentication.cc

  • Committer: Lee
  • Date: 2009-01-07 22:45:50 UTC
  • mfrom: (772 drizzle)
  • mto: This revision was merged to the branch mainline in revision 777.
  • Revision ID: lbieber@lbieber-desktop-20090107224550-tzjtc8klmv84ppyg
finish enabling tests, remove timezone2, timezone3

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/authentication.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2008 Brian Aker
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
18
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
19
 */
20
20
 
21
 
#include "config.h"
22
 
#include "drizzled/plugin/authentication.h"
23
 
#include "drizzled/error.h"
24
 
#include "drizzled/gettext.h"
25
 
#include "drizzled/identifier.h"
26
 
 
27
 
#include <vector>
28
 
 
29
 
namespace drizzled
30
 
{
31
 
 
32
 
std::vector<plugin::Authentication *> all_authentication;
33
 
 
34
 
 
35
 
bool plugin::Authentication::addPlugin(plugin::Authentication *auth)
36
 
{
37
 
  if (auth != NULL)
38
 
    all_authentication.push_back(auth);
 
21
#include <drizzled/server_includes.h>
 
22
#include <drizzled/authentication.h>
 
23
#include <drizzled/gettext.h>
 
24
#include <drizzled/errmsg_print.h>
 
25
 
 
26
static bool are_plugins_loaded= false;
 
27
 
 
28
static bool authenticate_by(Session *session, plugin_ref plugin, void* p_data)
 
29
{
 
30
  const char *password= (const char *)p_data;
 
31
  authentication_st *auth= plugin_data(plugin, authentication_st *);
 
32
 
 
33
  (void)p_data;
 
34
 
 
35
  if (auth && auth->authenticate)
 
36
  {
 
37
    if (auth->authenticate(session, password))
 
38
      return true;
 
39
  }
39
40
 
40
41
  return false;
41
42
}
42
43
 
43
 
void plugin::Authentication::removePlugin(plugin::Authentication *auth)
44
 
{
45
 
  if (auth != NULL)
46
 
    all_authentication.erase(std::find(all_authentication.begin(),
47
 
                                       all_authentication.end(),
48
 
                                       auth));
49
 
}
50
 
 
51
 
class AuthenticateBy : public std::unary_function<plugin::Authentication *, bool>
52
 
{
53
 
  const identifier::User &sctx;
54
 
  const std::string &password;
55
 
 
56
 
public:
57
 
  AuthenticateBy(const identifier::User &sctx_arg, const std::string &password_arg) :
58
 
    std::unary_function<plugin::Authentication *, bool>(),
59
 
    sctx(sctx_arg), password(password_arg) {}
60
 
 
61
 
  inline result_type operator()(argument_type auth)
62
 
  {
63
 
    return auth->authenticate(sctx, password);
64
 
  }
65
 
};
66
 
 
67
 
bool plugin::Authentication::isAuthenticated(drizzled::identifier::User::const_shared_ptr sctx,
68
 
                                             const std::string &password)
 
44
bool authenticate_user(Session *session, const char *password)
69
45
{
70
46
  /* If we never loaded any auth plugins, just return true */
71
 
  if (all_authentication.empty())
 
47
  if (are_plugins_loaded != true)
72
48
    return true;
73
49
 
74
 
  /* Use find_if instead of foreach so that we can collect return codes */
75
 
  std::vector<plugin::Authentication *>::iterator iter=
76
 
    std::find_if(all_authentication.begin(), all_authentication.end(),
77
 
                 AuthenticateBy(*sctx, password));
78
 
 
79
 
  /* We only require one plugin to return success in order to authenticate.
80
 
   * If iter is == end() here, that means that all of the plugins returned
81
 
   * false, which means they all failed.
82
 
   */
83
 
  if (iter == all_authentication.end())
 
50
  return plugin_foreach(session, authenticate_by, DRIZZLE_AUTH_PLUGIN, (void *)password);
 
51
}
 
52
 
 
53
 
 
54
int authentication_initializer(st_plugin_int *plugin)
 
55
{
 
56
  authentication_st *authen;
 
57
 
 
58
  authen= new authentication_st;
 
59
 
 
60
  if (authen == NULL)
 
61
    return 1;
 
62
 
 
63
  memset(authen, 0, sizeof(authentication_st));
 
64
 
 
65
  if (plugin->plugin->init)
84
66
  {
85
 
    my_error(ER_ACCESS_DENIED_ERROR, MYF(0),
86
 
             sctx->username().c_str(),
87
 
             sctx->address().c_str(),
88
 
             password.empty() ? ER(ER_NO) : ER(ER_YES));
89
 
 
90
 
    return false;
 
67
    if (plugin->plugin->init(authen))
 
68
    {
 
69
      errmsg_printf(ERRMSG_LVL_ERROR, _("Plugin '%s' init function returned error."),
 
70
                      plugin->name.str);
 
71
      goto err;
 
72
    }
91
73
  }
92
74
 
93
 
  return true;
94
 
}
95
 
 
96
 
} /* namespace drizzled */
 
75
  plugin->data= (void *)authen;
 
76
  are_plugins_loaded= true;
 
77
 
 
78
  return(0);
 
79
err:
 
80
  delete authen;
 
81
  return(1);
 
82
}
 
83
 
 
84
int authentication_finalizer(st_plugin_int *plugin)
 
85
{
 
86
  authentication_st *authen= (authentication_st *)plugin->data;
 
87
 
 
88
  assert(authen);
 
89
  if (authen && plugin->plugin->deinit)
 
90
    plugin->plugin->deinit(authen);
 
91
 
 
92
  delete authen;
 
93
 
 
94
  return(0);
 
95
}