← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/identifier/schema.cc

  • Committer: Monty Taylor
  • Date: 2010-07-19 05:06:59 UTC
  • mto: (1662.1.2 rollup)
  • mto: This revision was merged to the branch mainline in revision 1663.
  • Revision ID: mordred@inaugust.com-20100719050659-if3thz0k66m1jkaf
Backed out two bits that snuck in to the merge.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/identifier/schema.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) 2009 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2009 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
18
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
19
 */
20
20
 
21
 
#include <config.h>
 
21
#include "config.h"
22
22
 
23
23
#include <assert.h>
24
24
 
25
 
#include <drizzled/identifier.h>
26
 
#include <drizzled/session.h>
27
 
#include <drizzled/internal/my_sys.h>
28
 
 
29
 
#include <drizzled/util/tablename_to_filename.h>
30
 
#include <drizzled/util/backtrace.h>
 
25
#include "drizzled/identifier.h"
 
26
#include "drizzled/session.h"
 
27
#include "drizzled/current_session.h"
 
28
#include "drizzled/internal/my_sys.h"
 
29
#include "drizzled/data_home.h"
31
30
 
32
31
#include <algorithm>
33
32
#include <sstream>
34
33
#include <cstdio>
35
34
 
36
 
#include <boost/algorithm/string/compare.hpp>
37
 
 
38
35
using namespace std;
39
36
 
40
37
namespace drizzled
41
38
{
42
39
 
43
 
namespace identifier
44
 
{
45
 
 
46
40
extern string drizzle_tmpdir;
 
41
extern pid_t current_pid;
 
42
 
 
43
static const char hexchars[]= "0123456789abcdef";
 
44
 
 
45
static bool tablename_to_filename(const string &from, string &to);
47
46
 
48
47
static size_t build_schema_filename(string &path, const string &db)
49
48
{
50
 
  path.append("");
 
49
  string dbbuff("");
51
50
  bool conversion_error= false;
52
51
 
53
 
  conversion_error= util::tablename_to_filename(db, path);
 
52
  conversion_error= tablename_to_filename(db, dbbuff);
54
53
  if (conversion_error)
55
54
  {
56
 
    errmsg_printf(error::ERROR,
 
55
    errmsg_printf(ERRMSG_LVL_ERROR,
57
56
                  _("Schema name cannot be encoded and fit within filesystem "
58
57
                    "name length restrictions."));
59
58
    return 0;
60
59
  }
 
60
   
 
61
 
 
62
  int rootdir_len= strlen(FN_ROOTDIR);
 
63
  path.append(data_home);
 
64
  ssize_t without_rootdir= path.length() - rootdir_len;
 
65
 
 
66
  /* Don't add FN_ROOTDIR if dirzzle_data_home already includes it */
 
67
  if (without_rootdir >= 0)
 
68
  {
 
69
    const char *tmp= path.c_str() + without_rootdir;
 
70
 
 
71
    if (memcmp(tmp, FN_ROOTDIR, rootdir_len) != 0)
 
72
      path.append(FN_ROOTDIR);
 
73
  }
 
74
 
 
75
  path.append(dbbuff);
61
76
 
62
77
  return path.length();
63
78
}
64
79
 
65
 
Schema::Schema(const std::string &db_arg) :
 
80
 
 
81
/*
 
82
  Translate a table name to a cursor name (WL #1324).
 
83
 
 
84
  SYNOPSIS
 
85
    tablename_to_filename()
 
86
      from                      The table name
 
87
      to                OUT     The cursor name
 
88
 
 
89
  RETURN
 
90
    true if errors happen. false on success.
 
91
*/
 
92
static bool tablename_to_filename(const string &from, string &to)
 
93
{
 
94
  
 
95
  string::const_iterator iter= from.begin();
 
96
  for (; iter != from.end(); ++iter)
 
97
  {
 
98
    if ((*iter >= '0' && *iter <= '9') ||
 
99
        (*iter >= 'A' && *iter <= 'Z') ||
 
100
        (*iter >= 'a' && *iter <= 'z') ||
 
101
/* OSX defines an extra set of high-bit and multi-byte characters
 
102
   that cannot be used on the filesystem. Instead of trying to sort
 
103
   those out, we'll just escape encode all high-bit-set chars on OSX.
 
104
   It won't really hurt anything - it'll just make some filenames ugly. */
 
105
#if !defined(TARGET_OS_OSX)
 
106
        ((unsigned char)*iter >= 128) ||
 
107
#endif
 
108
        (*iter == '_') ||
 
109
        (*iter == ' ') ||
 
110
        (*iter == '-'))
 
111
    {
 
112
      to.push_back(*iter);
 
113
      continue;
 
114
    }
 
115
   
 
116
    /* We need to escape this char in a way that can be reversed */
 
117
    to.push_back('@');
 
118
    to.push_back(hexchars[(*iter >> 4) & 15]);
 
119
    to.push_back(hexchars[(*iter) & 15]);
 
120
  }
 
121
 
 
122
  if (internal::check_if_legal_tablename(to.c_str()))
 
123
  {
 
124
    to.append("@@@");
 
125
  }
 
126
  return false;
 
127
}
 
128
 
 
129
SchemaIdentifier::SchemaIdentifier(const std::string &db_arg) :
66
130
  db(db_arg),
67
 
  db_path("")
 
131
  db_path(""),
 
132
  lower_db(db_arg)
68
133
{ 
69
 
#if 0
70
 
  string::size_type lastPos= db.find_first_of('/', 0);
71
 
 
72
 
  if (lastPos != std::string::npos) 
73
 
  {
74
 
    catalog= db.substr(0, lastPos);
75
 
    db.erase(0, lastPos + 1);
76
 
  }
77
 
#endif
78
 
 
79
 
  if (not db_arg.empty())
80
 
  {
81
 
    build_schema_filename(db_path, db);
 
134
  std::transform(lower_db.begin(), lower_db.end(),
 
135
                 lower_db.begin(), ::tolower);
 
136
 
 
137
  if (not lower_db.empty())
 
138
  {
 
139
    drizzled::build_schema_filename(db_path, lower_db);
82
140
    assert(db_path.length()); // TODO throw exception, this is a possibility
83
141
  }
84
142
}
85
143
 
86
 
void Schema::getSQLPath(std::string &arg) const
 
144
const std::string &SchemaIdentifier::getSQLPath()
87
145
{
88
 
  arg= db;
 
146
  return getSchemaName();
89
147
}
90
148
 
91
 
const std::string &Schema::getPath() const
 
149
const std::string &SchemaIdentifier::getPath() const
92
150
{
93
151
  return db_path;
94
152
}
95
153
 
96
 
bool Schema::compare(const std::string &arg) const
97
 
{
98
 
  return boost::iequals(arg, db);
99
 
}
100
 
 
101
 
bool Schema::compare(Schema::const_reference arg) const
102
 
{
103
 
  return boost::iequals(arg.getSchemaName(), db);
104
 
}
105
 
 
106
 
bool Schema::isValid() const
107
 
{
108
 
  bool error= false;
109
 
 
110
 
  do
111
 
  {
112
 
    if (db.empty())
113
 
    {
114
 
      error= true;
115
 
      break;
116
 
    }
117
 
 
118
 
    if (db.size() > NAME_LEN)
119
 
    {
120
 
      error= true;
121
 
      break;
122
 
    }
123
 
 
124
 
    if (db.at(db.length() -1) == ' ')
125
 
    {
126
 
      error= true;
127
 
      break;
128
 
    }
129
 
 
130
 
    if (db.at(0) == '.')
131
 
    {
132
 
      error= true;
133
 
      break;
134
 
    }
135
 
 
136
 
    {
137
 
      const CHARSET_INFO * const cs= &my_charset_utf8mb4_general_ci;
138
 
 
139
 
      int well_formed_error;
140
 
      uint32_t res= cs->cset->well_formed_len(cs, db.c_str(), db.c_str() + db.length(),
141
 
                                              NAME_CHAR_LEN, &well_formed_error);
142
 
      if (well_formed_error or db.length() != res)
143
 
      {
144
 
        error= true;
145
 
        break;
146
 
      }
147
 
    }
148
 
  } while (0);
149
 
 
150
 
  if (error)
151
 
  {
152
 
    my_error(ER_WRONG_DB_NAME, *this);
153
 
 
 
154
bool SchemaIdentifier::compare(std::string arg) const
 
155
{
 
156
  std::transform(arg.begin(), arg.end(),
 
157
                 arg.begin(), ::tolower);
 
158
 
 
159
  return arg == lower_db;
 
160
}
 
161
 
 
162
bool SchemaIdentifier::isValid() const
 
163
{
 
164
  if (lower_db.empty())
 
165
    return false;
 
166
 
 
167
  if (lower_db.size() > NAME_LEN)
 
168
    return false;
 
169
 
 
170
  if (lower_db.at(lower_db.length() -1) == ' ')
 
171
    return false;
 
172
 
 
173
  const CHARSET_INFO * const cs= &my_charset_utf8mb4_general_ci;
 
174
 
 
175
  int well_formed_error;
 
176
  uint32_t res= cs->cset->well_formed_len(cs, lower_db.c_str(), lower_db.c_str() + lower_db.length(),
 
177
                                          NAME_CHAR_LEN, &well_formed_error);
 
178
 
 
179
  if (well_formed_error)
 
180
  {
 
181
    my_error(ER_INVALID_CHARACTER_STRING, MYF(0), "identifier", lower_db.c_str());
154
182
    return false;
155
183
  }
156
184
 
 
185
  if (lower_db.length() != res)
 
186
    return false;
 
187
 
157
188
  return true;
158
189
}
159
190
 
160
 
const std::string &Schema::getCatalogName() const
161
 
{
162
 
  return drizzled::catalog::local_identifier().name();
163
 
}
164
 
 
165
 
std::ostream& operator<<(std::ostream& output, const Schema&identifier)
166
 
{
167
 
  output << "identifier::Schema:(";
168
 
  output <<  catalog::local_identifier();
169
 
  output << ", ";
170
 
  output <<  identifier.getSchemaName().c_str();
171
 
  output << ", ";
172
 
  output << identifier.getPath().c_str();
173
 
  output << ")";
174
 
 
175
 
  return output;  // for multiple << operators.
176
 
}
177
 
 
178
 
} /* namespace identifier */
179
191
} /* namespace drizzled */