← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/identifier/schema.cc

  • Committer: David Shrewsbury
  • Date: 2010-09-21 19:25:33 UTC
  • mto: (1797.1.1 build)
  • mto: This revision was merged to the branch mainline in revision 1798.
  • Revision ID: shrewsbury.dave@gmail.com-20100921192533-4b1fjkun33r31pn5
Replace std::map with boost::unordered_map

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/identifier/schema.cc
26
26
#include "drizzled/session.h"
27
27
#include "drizzled/current_session.h"
28
28
#include "drizzled/internal/my_sys.h"
29
 
 
30
 
#include "drizzled/util/tablename_to_filename.h"
31
 
#include "drizzled/util/backtrace.h"
 
29
#include "drizzled/data_home.h"
32
30
 
33
31
#include <algorithm>
34
32
#include <sstream>
44
42
extern string drizzle_tmpdir;
45
43
extern pid_t current_pid;
46
44
 
 
45
static const char hexchars[]= "0123456789abcdef";
 
46
 
 
47
static bool tablename_to_filename(const string &from, string &to);
 
48
 
47
49
static size_t build_schema_filename(string &path, const string &db)
48
50
{
49
 
  path.append("");
 
51
  string dbbuff("");
50
52
  bool conversion_error= false;
51
53
 
52
 
  conversion_error= util::tablename_to_filename(db, path);
 
54
  conversion_error= tablename_to_filename(db, dbbuff);
53
55
  if (conversion_error)
54
56
  {
55
57
    errmsg_printf(ERRMSG_LVL_ERROR,
57
59
                    "name length restrictions."));
58
60
    return 0;
59
61
  }
 
62
   
 
63
 
 
64
  int rootdir_len= strlen(FN_ROOTDIR);
 
65
  path.append(data_home);
 
66
  ssize_t without_rootdir= path.length() - rootdir_len;
 
67
 
 
68
  /* Don't add FN_ROOTDIR if dirzzle_data_home already includes it */
 
69
  if (without_rootdir >= 0)
 
70
  {
 
71
    const char *tmp= path.c_str() + without_rootdir;
 
72
 
 
73
    if (memcmp(tmp, FN_ROOTDIR, rootdir_len) != 0)
 
74
      path.append(FN_ROOTDIR);
 
75
  }
 
76
 
 
77
  path.append(dbbuff);
60
78
 
61
79
  return path.length();
62
80
}
63
81
 
 
82
 
 
83
/*
 
84
  Translate a table name to a cursor name (WL #1324).
 
85
 
 
86
  SYNOPSIS
 
87
    tablename_to_filename()
 
88
      from                      The table name
 
89
      to                OUT     The cursor name
 
90
 
 
91
  RETURN
 
92
    true if errors happen. false on success.
 
93
*/
 
94
static bool tablename_to_filename(const string &from, string &to)
 
95
{
 
96
  
 
97
  string::const_iterator iter= from.begin();
 
98
  for (; iter != from.end(); ++iter)
 
99
  {
 
100
    if ((*iter >= '0' && *iter <= '9') ||
 
101
        (*iter >= 'a' && *iter <= 'z') ||
 
102
        /* OSX defines an extra set of high-bit and multi-byte characters
 
103
          that cannot be used on the filesystem. Instead of trying to sort
 
104
          those out, we'll just escape encode all high-bit-set chars on OSX.
 
105
          It won't really hurt anything - it'll just make some filenames ugly. */
 
106
#if !defined(TARGET_OS_OSX)
 
107
        ((unsigned char)*iter >= 128) ||
 
108
#endif
 
109
        (*iter == '_') ||
 
110
        (*iter == ' ') ||
 
111
        (*iter == '-'))
 
112
    {
 
113
      to.push_back(*iter);
 
114
      continue;
 
115
    }
 
116
 
 
117
    if ((*iter >= 'A' && *iter <= 'Z'))
 
118
    {
 
119
      to.push_back(tolower(*iter));
 
120
      continue;
 
121
    }
 
122
   
 
123
    /* We need to escape this char in a way that can be reversed */
 
124
    to.push_back('@');
 
125
    to.push_back(hexchars[(*iter >> 4) & 15]);
 
126
    to.push_back(hexchars[(*iter) & 15]);
 
127
  }
 
128
 
 
129
  if (internal::check_if_legal_tablename(to.c_str()))
 
130
  {
 
131
    to.append("@@@");
 
132
  }
 
133
  return false;
 
134
}
 
135
 
64
136
SchemaIdentifier::SchemaIdentifier(const std::string &db_arg) :
65
137
  db(db_arg),
66
 
  db_path(""),
67
 
  catalog("LOCAL")
 
138
  db_path("")
68
139
{ 
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
140
  if (not db_arg.empty())
80
141
  {
81
142
    drizzled::build_schema_filename(db_path, db);
83
144
  }
84
145
}
85
146
 
86
 
void SchemaIdentifier::getSQLPath(std::string &arg) const
 
147
const std::string &SchemaIdentifier::getSQLPath()
87
148
{
88
 
  arg.append(getSchemaName());
 
149
  return getSchemaName();
89
150
}
90
151
 
91
152
const std::string &SchemaIdentifier::getPath() const
98
159
  return boost::iequals(arg, db);
99
160
}
100
161
 
101
 
bool SchemaIdentifier::compare(SchemaIdentifier::const_reference arg) const
102
 
{
103
 
  return boost::iequals(arg.getSchemaName(), db);
104
 
}
105
 
 
106
162
bool SchemaIdentifier::isValid() const
107
163
{
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
 
    std::string name;
153
 
 
154
 
    getSQLPath(name);
155
 
    my_error(ER_WRONG_DB_NAME, MYF(0), name.c_str());
156
 
 
 
164
  if (db.empty())
 
165
    return false;
 
166
 
 
167
  if (db.size() > NAME_LEN)
 
168
    return false;
 
169
 
 
170
  if (db.at(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, db.c_str(), db.c_str() + 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", db.c_str());
157
182
    return false;
158
183
  }
159
184
 
 
185
  if (db.length() != res)
 
186
    return false;
 
187
 
160
188
  return true;
161
189
}
162
190