~drizzle-trunk/drizzle/development

1963.2.1 by Brian Aker
Fixing up the kill function.
1
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 *
4
 *  Copyright (C) 2010 Brian Aker
5
 *
6
 *  This program is free software; you can redistribute it and/or modify
7
 *  it under the terms of the GNU General Public License as published by
8
 *  the Free Software Foundation; either version 2 of the License, or
9
 *  (at your option) any later version.
10
 *
11
 *  This program is distributed in the hope that it will be useful,
12
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 *  GNU General Public License for more details.
15
 *
16
 *  You should have received a copy of the GNU General Public License
17
 *  along with this program; if not, write to the Free Software
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
 */
20
21
#include "config.h"
22
23
#include <drizzled/session.h>
1966.2.5 by Brian Aker
Style change around session list.
24
#include <drizzled/session/cache.h>
1963.2.1 by Brian Aker
Fixing up the kill function.
25
#include "plugin/utility_functions/functions.h"
26
27
namespace drizzled
28
{
29
30
namespace utility_functions
31
{
32
33
int64_t Kill::val_int()
34
{
35
  int64_t session_id_for_kill= args[0]->val_int();
36
37
  if (getSession().getSessionId() == session_id_for_kill)
38
  {
39
    my_error(drizzled::ER_KILL_DENY_SELF_ERROR, MYF(0));
40
    return 0;
41
  }
42
43
  Session::shared_ptr die_session(session::Cache::singleton().find(session_id_for_kill));
44
45
  if (not die_session)
46
  {
47
    my_error(drizzled::ER_NO_SUCH_THREAD, session_id_for_kill, MYF(0));
48
    return 0;
49
  }
50
2015.3.1 by Brian Aker
Encapsulate client call. Also remove the need to call current_session when
51
  if (not die_session->isViewable(*getSession().user()))
1963.2.1 by Brian Aker
Fixing up the kill function.
52
  {
53
    my_error(drizzled::ER_KILL_DENIED_ERROR, session_id_for_kill, MYF(0));
54
    return 0;
55
  }
56
57
  die_session->awake(arg_count == 2 ? Session::KILL_QUERY : Session::KILL_CONNECTION);
58
59
  return session_id_for_kill;
60
}
61
62
} /* namespace utility_functions */
63
} /* namespace drizzled */