← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/authentication.cc

  • Committer: Monty Taylor
  • Date: 2008-10-16 06:32:30 UTC
  • mto: (511.1.5 codestyle)
  • mto: This revision was merged to the branch mainline in revision 521.
  • Revision ID: monty@inaugust.com-20081016063230-4brxsra0qsmsg84q
Added -Wunused-macros.

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
 
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/security_context.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 <libdrizzle/gettext.h>
 
24
 
 
25
static bool are_plugins_loaded= false;
 
26
 
 
27
static bool authenticate_by(THD *thd, plugin_ref plugin, void* p_data)
 
28
{
 
29
  const char *password= (const char *)p_data;
 
30
  authentication_st *auth= plugin_data(plugin, authentication_st *);
 
31
 
 
32
  (void)p_data;
 
33
 
 
34
  if (auth && auth->authenticate)
 
35
  {
 
36
    if (auth->authenticate(thd, password))
 
37
      return true;
 
38
  }
 
39
 
39
40
  return false;
40
41
}
41
42
 
42
 
void plugin::Authentication::removePlugin(plugin::Authentication *auth)
43
 
{
44
 
  if (auth != NULL)
45
 
    all_authentication.erase(std::find(all_authentication.begin(),
46
 
                                       all_authentication.end(),
47
 
                                       auth));
48
 
}
49
 
 
50
 
class AuthenticateBy : public std::unary_function<plugin::Authentication *, bool>
51
 
{
52
 
  const SecurityContext &sctx;
53
 
  const std::string &password;
54
 
public:
55
 
  AuthenticateBy(const SecurityContext &sctx_arg, const std::string &password_arg) :
56
 
    std::unary_function<plugin::Authentication *, bool>(),
57
 
    sctx(sctx_arg), password(password_arg) {}
58
 
 
59
 
  inline result_type operator()(argument_type auth)
60
 
  {
61
 
    return auth->authenticate(sctx, password);
62
 
  }
63
 
};
64
 
 
65
 
bool plugin::Authentication::isAuthenticated(const SecurityContext &sctx,
66
 
                                             const std::string &password)
 
43
bool authenticate_user(THD *thd, const char *password)
67
44
{
68
45
  /* If we never loaded any auth plugins, just return true */
69
 
  if (all_authentication.empty())
 
46
  if (are_plugins_loaded != true)
70
47
    return true;
71
48
 
72
 
  /* Use find_if instead of foreach so that we can collect return codes */
73
 
  std::vector<plugin::Authentication *>::iterator iter=
74
 
    std::find_if(all_authentication.begin(), all_authentication.end(),
75
 
                 AuthenticateBy(sctx, password));
76
 
 
77
 
  /* We only require one plugin to return success in order to authenticate.
78
 
   * If iter is == end() here, that means that all of the plugins returned
79
 
   * false, which means they all failed.
80
 
   */
81
 
  if (iter == all_authentication.end())
 
49
  return plugin_foreach(thd, authenticate_by, DRIZZLE_AUTH_PLUGIN, (void *)password);
 
50
}
 
51
 
 
52
 
 
53
int authentication_initializer(st_plugin_int *plugin)
 
54
{
 
55
  authentication_st *authen;
 
56
 
 
57
  if ((authen= (authentication_st *)malloc(sizeof(authentication_st))) == 0)
 
58
      return(1);
 
59
 
 
60
  memset(authen, 0, sizeof(authentication_st));
 
61
 
 
62
  if (plugin->plugin->init)
82
63
  {
83
 
    my_error(ER_ACCESS_DENIED_ERROR, MYF(0),
84
 
             sctx.getUser().c_str(),
85
 
             sctx.getIp().c_str(),
86
 
             password.empty() ? ER(ER_NO) : ER(ER_YES));
87
 
    return false;
 
64
    if (plugin->plugin->init(authen))
 
65
    {
 
66
      sql_print_error(_("Plugin '%s' init function returned error."),
 
67
                      plugin->name.str);
 
68
      goto err;
 
69
    }
88
70
  }
89
 
  return true;
90
 
}
91
 
 
92
 
} /* namespace drizzled */
 
71
 
 
72
  plugin->data= (void *)authen;
 
73
  are_plugins_loaded= true;
 
74
 
 
75
  return(0);
 
76
err:
 
77
  free(authen);
 
78
  return(1);
 
79
}
 
80
 
 
81
int authentication_finalizer(st_plugin_int *plugin)
 
82
{
 
83
  authentication_st *authen= (authentication_st *)plugin->data;
 
84
 
 
85
  assert(authen);
 
86
  if (authen && plugin->plugin->deinit)
 
87
    plugin->plugin->deinit(authen);
 
88
 
 
89
  free(authen);
 
90
 
 
91
  return(0);
 
92
}