← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/function_engine/function.cc

  • Committer: Monty Taylor
  • Date: 2010-03-02 19:10:25 UTC
  • mto: (1317.1.8)
  • mto: This revision was merged to the branch mainline in revision 1322.
  • Revision ID: mordred@inaugust.com-20100302191025-zoxjz4xwkoa6160h
Prevent unauthorized users from changing schema.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/function_engine/function.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) 2010 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2010 Sun Microsystems
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
32
32
  drizzled::plugin::StorageEngine(name_arg,
33
33
                                  HTON_ALTER_NOT_SUPPORTED |
34
34
                                  HTON_HAS_SCHEMA_DICTIONARY |
 
35
                                  HTON_HAS_DATA_DICTIONARY |
35
36
                                  HTON_SKIP_STORE_LOCK |
36
 
                                  HTON_TEMPORARY_NOT_SUPPORTED),
37
 
  information_message(new(message::Schema)),
38
 
  data_dictionary_message(new(message::Schema))
39
 
 
 
37
                                  HTON_TEMPORARY_NOT_SUPPORTED)
40
38
{
41
 
  information_message->set_name(INFORMATION_SCHEMA_IDENTIFIER.getSchemaName());
42
 
  data_dictionary_message->set_collation("utf8_general_ci");
43
 
 
44
 
  data_dictionary_message->set_name(DATA_DICTIONARY_IDENTIFIER.getSchemaName());
45
 
  data_dictionary_message->set_collation("utf8_general_ci");
46
39
}
47
40
 
48
41
 
49
 
Cursor *Function::create(Table &table)
 
42
Cursor *Function::create(TableShare &table, memory::Root *mem_root)
50
43
{
51
 
  return new FunctionCursor(*this, table);
 
44
  return new (mem_root) FunctionCursor(*this, table);
52
45
}
53
46
 
54
47
int Function::doGetTableDefinition(Session &,
55
 
                                   const identifier::Table &identifier,
56
 
                                   message::Table &table_proto)
 
48
                                     const char *path,
 
49
                                     const char *,
 
50
                                     const char *,
 
51
                                     const bool,
 
52
                                     message::Table *table_proto)
57
53
{
58
 
  drizzled::plugin::TableFunction *function= getFunction(identifier.getPath());
 
54
  string tab_name(path);
 
55
  transform(tab_name.begin(), tab_name.end(),
 
56
            tab_name.begin(), ::tolower);
 
57
 
 
58
  drizzled::plugin::TableFunction *function= getFunction(tab_name);
59
59
 
60
60
  if (not function)
61
61
  {
62
62
    return ENOENT;
63
63
  }
64
64
 
65
 
  function->define(table_proto);
 
65
  if (table_proto)
 
66
  {
 
67
    function->define(*table_proto);
 
68
  }
66
69
 
67
70
  return EEXIST;
68
71
}
69
72
 
70
 
void Function::doGetSchemaIdentifiers(identifier::Schema::vector& schemas)
71
 
{
72
 
  schemas.push_back(INFORMATION_SCHEMA_IDENTIFIER);
73
 
  schemas.push_back(DATA_DICTIONARY_IDENTIFIER);
74
 
}
75
 
 
76
 
bool Function::doGetSchemaDefinition(const identifier::Schema &schema_identifier, message::schema::shared_ptr &schema_message)
77
 
{
78
 
  schema_message.reset(new message::Schema); // This should be fixed, we could just be using ones we built on startup.
79
 
 
80
 
  if (schema_identifier == INFORMATION_SCHEMA_IDENTIFIER)
 
73
 
 
74
void Function::doGetTableNames(drizzled::CachedDirectory&, 
 
75
                               string &db, 
 
76
                               set<string> &set_of_names)
 
77
{
 
78
  drizzled::plugin::TableFunction::getNames(db, set_of_names);
 
79
}
 
80
 
 
81
void Function::doGetSchemaNames(std::set<std::string>& set_of_names)
 
82
{
 
83
  set_of_names.insert("information_schema"); // special cases suck
 
84
  set_of_names.insert("data_dictionary"); // special cases suck
 
85
}
 
86
 
 
87
bool Function::doGetSchemaDefinition(const std::string &schema_name, message::Schema &schema_message)
 
88
{
 
89
  if (not schema_name.compare("information_schema"))
81
90
  {
82
 
    schema_message= information_message;
 
91
    schema_message.set_name("information_schema");
 
92
    schema_message.set_collation("utf8_general_ci");
83
93
  }
84
 
  else if (schema_identifier == DATA_DICTIONARY_IDENTIFIER)
 
94
  else if (not schema_name.compare("data_dictionary"))
85
95
  {
86
 
    schema_message= data_dictionary_message;
 
96
    schema_message.set_name("data_dictionary");
 
97
    schema_message.set_collation("utf8_general_ci");
87
98
  }
88
99
  else
89
100
  {
93
104
  return true;
94
105
}
95
106
 
96
 
bool Function::doCanCreateTable(const drizzled::identifier::Table &table_identifier)
97
 
{
98
 
  if (static_cast<const identifier::Schema&>(table_identifier) == INFORMATION_SCHEMA_IDENTIFIER)
99
 
  {
100
 
    return false;
101
 
  }
102
 
 
103
 
  else if (static_cast<const identifier::Schema&>(table_identifier) == DATA_DICTIONARY_IDENTIFIER)
104
 
  {
105
 
    return false;
106
 
  }
107
 
 
108
 
  return true;
109
 
}
110
 
 
111
 
bool Function::doDoesTableExist(Session&, const identifier::Table &identifier)
112
 
{
113
 
  drizzled::plugin::TableFunction *function= getFunction(identifier.getPath());
114
 
 
115
 
  if (function)
116
 
    return true;
117
 
 
118
 
  return false;
119
 
}
120
 
 
121
 
 
122
 
void Function::doGetTableIdentifiers(drizzled::CachedDirectory&,
123
 
                                     const drizzled::identifier::Schema &schema_identifier,
124
 
                                     drizzled::identifier::Table::vector &set_of_identifiers)
125
 
{
126
 
  set<std::string> set_of_names;
127
 
  drizzled::plugin::TableFunction::getNames(schema_identifier.getSchemaName(), set_of_names);
128
 
 
129
 
  for (set<std::string>::iterator iter= set_of_names.begin(); iter != set_of_names.end(); iter++)
130
 
  {
131
 
    set_of_identifiers.push_back(identifier::Table(schema_identifier, *iter, drizzled::message::Table::FUNCTION));
132
 
  }
133
 
}
134
 
 
135
 
static int init(drizzled::module::Context &context)
136
 
{
137
 
  context.add(new Function("FunctionEngine"));
 
107
 
 
108
static drizzled::plugin::StorageEngine *function_plugin= NULL;
 
109
 
 
110
static int init(drizzled::plugin::Registry &registry)
 
111
{
 
112
  function_plugin= new(std::nothrow) Function("FunctionEngine");
 
113
 
 
114
  if (not function_plugin)
 
115
  {
 
116
    return 1;
 
117
  }
 
118
 
 
119
  registry.add(function_plugin);
 
120
 
 
121
  return 0;
 
122
}
 
123
 
 
124
static int finalize(drizzled::plugin::Registry &registry)
 
125
{
 
126
  registry.remove(function_plugin);
 
127
  delete function_plugin;
138
128
 
139
129
  return 0;
140
130
}
148
138
  "Function Engine provides the infrastructure for Table Functions,etc.",
149
139
  PLUGIN_LICENSE_GPL,
150
140
  init,     /* Plugin Init */
151
 
  NULL,               /* depends */
 
141
  finalize,     /* Plugin Deinit */
 
142
  NULL,               /* system variables */
152
143
  NULL                /* config options   */
153
144
}
154
145
DRIZZLE_DECLARE_PLUGIN_END;