~drizzle-trunk/drizzle/development

/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2009 Sun Microsystems, Inc.

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 2 of the License, or

 *  (at your option) any later version.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include <config.h>

#include <assert.h>

#include <boost/lexical_cast.hpp>

#include <drizzled/identifier.h>

#include <drizzled/internal/my_sys.h>

#include <drizzled/error.h>

#include <drizzled/errmsg_print.h>

#include <drizzled/gettext.h>

#include <drizzled/table.h>

#include <drizzled/util/string.h>

#include <drizzled/util/tablename_to_filename.h>

#include <algorithm>

#include <sstream>

#include <cstdio>

#include <boost/thread.hpp>

using namespace std;

namespace drizzled

{

class Table;

extern std::string drizzle_tmpdir;

extern pid_t current_pid;

namespace identifier {

class Schema;

static const char hexchars[]= "0123456789abcdef";

/*

  Translate a cursor name to a table name (WL #1324).

  SYNOPSIS

    filename_to_tablename()

      from                      The cursor name

      to                OUT     The table name

      to_length                 The size of the table name buffer.

  RETURN

    Table name length.

*/

uint32_t Table::filename_to_tablename(const char *from, char *to, uint32_t to_length)

{

  uint32_t length= 0;

  if (!memcmp(from, TMP_FILE_PREFIX, TMP_FILE_PREFIX_LENGTH))

  {

    /* Temporary table name. */

    length= strlen(strncpy(to, from, to_length));

  }

  else

  {

    for (; *from  && length < to_length; length++, from++)

    {

      if (*from != '@')

      {

        to[length]= *from;

        continue;

      }

      /* We've found an escaped char - skip the @ */

      from++;

      to[length]= 0;

      /* There will be a two-position hex-char version of the char */

      for (int x=1; x >= 0; x--)

      {

        if (*from >= '0' && *from <= '9')

          to[length] += ((*from++ - '0') << (4 * x));

        else if (*from >= 'a' && *from <= 'f')

          to[length] += ((*from++ - 'a' + 10) << (4 * x));

      }

      /* Backup because we advanced extra in the inner loop */

      from--;

    } 

  }

  return length;

}

/*

  Creates path to a cursor: drizzle_tmpdir/#sql1234_12_1.ext

  SYNOPSIS

   build_tmptable_filename()

     session                    The thread handle.

     buff                       Where to write result

     bufflen                    buff size

  NOTES

    Uses current_pid, thread_id, and tmp_table counter to create

    a cursor name in drizzle_tmpdir.

  RETURN

    path length on success, 0 on failure

*/

#ifdef _GLIBCXX_HAVE_TLS 

__thread uint32_t counter= 0;

static uint32_t get_counter()

{

  return ++counter;

}

#else

boost::mutex counter_mutex;

static uint32_t counter= 1;

static uint32_t get_counter()

{

  boost::mutex::scoped_lock lock(counter_mutex);

  uint32_t x;

  x= ++counter;

  return x;

}

#endif

size_t Table::build_tmptable_filename(std::string &buffer)

{

  size_t tmpdir_length;

  ostringstream post_tmpdir_str;

  buffer.append(drizzle_tmpdir);

  tmpdir_length= buffer.length();

  post_tmpdir_str << "/" << TMP_FILE_PREFIX << current_pid;

  post_tmpdir_str << pthread_self() << "-" << get_counter();

  buffer.append(post_tmpdir_str.str());

  transform(buffer.begin() + tmpdir_length, buffer.end(), buffer.begin() + tmpdir_length, ::tolower);

  return buffer.length();

}

size_t Table::build_tmptable_filename(std::vector<char> &buffer)

{

  ostringstream post_tmpdir_str;

  post_tmpdir_str << drizzle_tmpdir << "/" << TMP_FILE_PREFIX << current_pid;

  post_tmpdir_str << pthread_self() << "-" << get_counter();

  buffer.resize(post_tmpdir_str.str().length() + 1);

  memcpy(&buffer[0], post_tmpdir_str.str().c_str(), post_tmpdir_str.str().size());

  buffer[post_tmpdir_str.str().size()]= 0;

  return buffer.size();

}

/*

  Creates path to a cursor: drizzle_data_dir/db/table.ext

  SYNOPSIS

   build_table_filename()

     buff                       Where to write result

                                This may be the same as table_name.

     bufflen                    buff size

     db                         Database name

     table_name                 Table name

     ext                        File extension.

     flags                      table_name is temporary, do not change.

  NOTES

    Uses database and table name, and extension to create

    a cursor name in drizzle_data_dir. Database and table

    names are converted from system_charset_info into "fscs".

    Unless flags indicate a temporary table name.

    'db' is always converted.

    'ext' is not converted.

    The conversion suppression is required for ALTER Table. This

    statement creates intermediate tables. These are regular

    (non-temporary) tables with a temporary name. Their path names must

    be derivable from the table name. So we cannot use

    build_tmptable_filename() for them.

  RETURN

    path length on success, 0 on failure

*/

size_t Table::build_table_filename(std::string &in_path, const std::string &in_db, const std::string &in_table_name, bool is_tmp)

{

  bool conversion_error= false;

  conversion_error= util::tablename_to_filename(in_db, in_path);

  if (conversion_error)

  {

    errmsg_printf(error::ERROR,

                  _("Schema name cannot be encoded and fit within filesystem "

                    "name length restrictions."));

    return 0;

  }

  in_path.append(FN_ROOTDIR);

  if (is_tmp) // It a conversion tmp

  {

    in_path.append(in_table_name);

  }

  else

  {

    conversion_error= util::tablename_to_filename(in_table_name, in_path);

    if (conversion_error)

    {

      errmsg_printf(error::ERROR,

                    _("Table name cannot be encoded and fit within filesystem "

                      "name length restrictions."));

      return 0;

    }

  }

  return in_path.length();

}

Table::Table(const drizzled::Table &table) :

  identifier::Schema(table.getShare()->getSchemaName()),

  type(table.getShare()->getTableType()),

  table_name(table.getShare()->getTableName())

{

  if (type == message::Table::TEMPORARY)

    path= table.getShare()->getPath();

  init();

}

void Table::init()

{

  switch (type) {

  case message::Table::FUNCTION:

  case message::Table::STANDARD:

    assert(path.size() == 0);

    build_table_filename(path, getSchemaName(), table_name, false);

    break;

  case message::Table::INTERNAL:

    assert(path.size() == 0);

    build_table_filename(path, getSchemaName(), table_name, true);

    break;

  case message::Table::TEMPORARY:

    if (path.empty())

    {

      build_tmptable_filename(path);

    }

    break;

  }

  switch (type) {

  case message::Table::FUNCTION:

  case message::Table::STANDARD:

  case message::Table::INTERNAL:

    break;

  case message::Table::TEMPORARY:

    {

      size_t pos;

      pos= path.find("tmp/#sql");

      if (pos != std::string::npos) 

      {

        key_path= path.substr(pos);

      }

    }

    break;

  }

  util::insensitive_hash hasher;

  hash_value= hasher(path);

  std::string tb_name(getTableName());

  std::transform(tb_name.begin(), tb_name.end(), tb_name.begin(), ::tolower);

  key.set(getKeySize(), getSchemaName(), tb_name);

}

const std::string &Table::getPath() const

{

  return path;

}

const std::string &Table::getKeyPath() const

{

  if (key_path.empty())

    return path;

  return key_path;

}

void Table::getSQLPath(std::string &sql_path) const  // @todo this is just used for errors, we should find a way to optimize it

{

  switch (type) {

  case message::Table::FUNCTION:

  case message::Table::STANDARD:

    sql_path.append(getSchemaName());

    sql_path.append(".");

    sql_path.append(table_name);

    break;

  case message::Table::INTERNAL:

    sql_path.append("temporary.");

    sql_path.append(table_name);

    break;

  case message::Table::TEMPORARY:

    sql_path.append(getSchemaName());

    sql_path.append(".#");

    sql_path.append(table_name);

    break;

  }

}

bool Table::isValid() const

{

  if (not identifier::Schema::isValid())

    return false;

  bool error= false;

  do

  {

    if (table_name.empty())

    {

      error= true;

      break;

    }

    if (table_name.size() > NAME_LEN)

    {

      error= true;

      break;

    }

    if (table_name.at(table_name.length() -1) == ' ')

    {

      error= true;

      break;

    }

    if (table_name.at(0) == '.')

    {

      error= true;

      break;

    }

    {

      const CHARSET_INFO * const cs= &my_charset_utf8mb4_general_ci;

      int well_formed_error;

      uint32_t res= cs->cset->well_formed_len(cs, table_name.c_str(), table_name.c_str() + table_name.length(),

                                              NAME_CHAR_LEN, &well_formed_error);

      if (well_formed_error or table_name.length() != res)

      {

        error= true;

        break;

      }

    }

  } while (0);

  if (error)

  {

    std::string name;

    getSQLPath(name);

    my_error(ER_WRONG_TABLE_NAME, MYF(0), name.c_str());

    return false;

  }

  return true;

}

void Table::copyToTableMessage(message::Table &message) const

{

  message.set_name(table_name);

  message.set_schema(getSchemaName());

}

void Table::Key::set(size_t resize_arg, const std::string &a, const std::string &b)

{

  key_buffer.resize(resize_arg);

  std::copy(a.begin(), a.end(), key_buffer.begin());

  std::copy(b.begin(), b.end(), key_buffer.begin() + a.length() + 1);

  util::sensitive_hash hasher;

  hash_value= hasher(key_buffer);

}

std::size_t hash_value(Table const& b)

{

  return b.getHashValue();

}

std::size_t hash_value(Table::Key const& b)

{

  return b.getHashValue();

}

std::ostream& operator<<(std::ostream& output, Table::const_reference identifier)

{

  output << "Table:(";

  output <<  identifier.getSchemaName();

  output << ", ";

  output << identifier.getTableName();

  output << ", ";

  output << message::type(identifier.getType());

  output << ", ";

  output << identifier.getPath();

  output << ", ";

  output << identifier.getHashValue();

  output << ")";

  return output;  // for multiple << operators.

}

} /* namespace identifier */

} /* namespace drizzled */