~drizzle-trunk/drizzle/development

/* Copyright (C) 2000-2003 MySQL AB

   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; version 2 of the License.

   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/* create and drop of databases */

#include "mysql_priv.h"

#include <mysys_err.h>

#include <my_dir.h>

#include <m_ctype.h>

#include "log.h"

#define MAX_DROP_TABLE_Q_LEN      1024

const char *del_exts[]= {".frm", ".BAK", ".TMD",".opt", NullS};

static TYPELIB deletable_extentions=

{array_elements(del_exts)-1,"del_exts", del_exts, NULL};

static long mysql_rm_known_files(THD *thd, MY_DIR *dirp,

				 const char *db, const char *path, uint level, 

                                 TABLE_LIST **dropped_tables);

static long mysql_rm_arc_files(THD *thd, MY_DIR *dirp, const char *org_path);

static my_bool rm_dir_w_symlink(const char *org_path, my_bool send_error);

static void mysql_change_db_impl(THD *thd,

                                 LEX_STRING *new_db_name,

                                 CHARSET_INFO *new_db_charset);

/* Database lock hash */

HASH lock_db_cache;

pthread_mutex_t LOCK_lock_db;

int creating_database= 0;  // how many database locks are made

/* Structure for database lock */

typedef struct my_dblock_st

{

  char *name;        /* Database name        */

  uint name_length;  /* Database length name */

} my_dblock_t;

/*

  lock_db key.

*/

extern "C" uchar* lock_db_get_key(my_dblock_t *, size_t *, my_bool not_used);

uchar* lock_db_get_key(my_dblock_t *ptr, size_t *length,

                       my_bool not_used __attribute__((unused)))

{

  *length= ptr->name_length;

  return (uchar*) ptr->name;

}

/*

  Free lock_db hash element.

*/

extern "C" void lock_db_free_element(void *ptr);

void lock_db_free_element(void *ptr)

{

  my_free(ptr, MYF(0));

}

/*

  Put a database lock entry into the hash.

  DESCRIPTION

    Insert a database lock entry into hash.

    LOCK_db_lock must be previously locked.

  RETURN VALUES

    0 on success.

    1 on error.

*/

static my_bool lock_db_insert(const char *dbname, uint length)

{

  my_dblock_t *opt;

  my_bool error= 0;

  DBUG_ENTER("lock_db_insert");

  safe_mutex_assert_owner(&LOCK_lock_db);

  if (!(opt= (my_dblock_t*) hash_search(&lock_db_cache,

                                        (uchar*) dbname, length)))

  { 

    /* Db is not in the hash, insert it */

    char *tmp_name;

    if (!my_multi_malloc(MYF(MY_WME | MY_ZEROFILL),

                         &opt, (uint) sizeof(*opt), &tmp_name, (uint) length+1,

                         NullS))

    {

      error= 1;

      goto end;

    }

    opt->name= tmp_name;

    strmov(opt->name, dbname);

    opt->name_length= length;

    if ((error= my_hash_insert(&lock_db_cache, (uchar*) opt)))

      my_free(opt, MYF(0));

  }

end:

  DBUG_RETURN(error);

}

/*

  Delete a database lock entry from hash.

*/

void lock_db_delete(const char *name, uint length)

{

  my_dblock_t *opt;

  safe_mutex_assert_owner(&LOCK_lock_db);

  if ((opt= (my_dblock_t *)hash_search(&lock_db_cache,

                                       (const uchar*) name, length)))

    hash_delete(&lock_db_cache, (uchar*) opt);

}

/* Database options hash */

static HASH dboptions;

static my_bool dboptions_init= 0;

static rw_lock_t LOCK_dboptions;

/* Structure for database options */

typedef struct my_dbopt_st

{

  char *name;			/* Database name                  */

  uint name_length;		/* Database length name           */

  CHARSET_INFO *charset;	/* Database default character set */

} my_dbopt_t;

/*

  Function we use in the creation of our hash to get key.

*/

extern "C" uchar* dboptions_get_key(my_dbopt_t *opt, size_t *length,

                                    my_bool not_used);

uchar* dboptions_get_key(my_dbopt_t *opt, size_t *length,

                         my_bool not_used __attribute__((unused)))

{

  *length= opt->name_length;

  return (uchar*) opt->name;

}

/*

  Helper function to write a query to binlog used by mysql_rm_db()

*/

static inline void write_to_binlog(THD *thd, char *query, uint q_len,

                                   char *db, uint db_len)

{

  Query_log_event qinfo(thd, query, q_len, 0, 0);

  qinfo.error_code= 0;

  qinfo.db= db;

  qinfo.db_len= db_len;

  mysql_bin_log.write(&qinfo);

}  

/*

  Function to free dboptions hash element

*/

extern "C" void free_dbopt(void *dbopt);

void free_dbopt(void *dbopt)

{

  my_free((uchar*) dbopt, MYF(0));

}

/* 

  Initialize database option hash and locked database hash.

  SYNOPSIS

    my_database_names()

  NOTES

    Must be called before any other database function is called.

  RETURN

    0	ok

    1	Fatal error

*/

bool my_database_names_init(void)

{

  bool error= 0;

  (void) my_rwlock_init(&LOCK_dboptions, NULL);

  if (!dboptions_init)

  {

    dboptions_init= 1;

    error= hash_init(&dboptions, lower_case_table_names ? 

                     &my_charset_bin : system_charset_info,

                     32, 0, 0, (hash_get_key) dboptions_get_key,

                     free_dbopt,0) ||

           hash_init(&lock_db_cache, lower_case_table_names ? 

                     &my_charset_bin : system_charset_info,

                     32, 0, 0, (hash_get_key) lock_db_get_key,

                     lock_db_free_element,0);

  }

  return error;

}

/* 

  Free database option hash and locked databases hash.

*/

void my_database_names_free(void)

{

  if (dboptions_init)

  {

    dboptions_init= 0;

    hash_free(&dboptions);

    (void) rwlock_destroy(&LOCK_dboptions);

    hash_free(&lock_db_cache);

  }

}

/*

  Cleanup cached options

*/

void my_dbopt_cleanup(void)

{

  rw_wrlock(&LOCK_dboptions);

  hash_free(&dboptions);

  hash_init(&dboptions, lower_case_table_names ? 

            &my_charset_bin : system_charset_info,

            32, 0, 0, (hash_get_key) dboptions_get_key,

            free_dbopt,0);

  rw_unlock(&LOCK_dboptions);

}

/*

  Find database options in the hash.

  DESCRIPTION

    Search a database options in the hash, usings its path.

    Fills "create" on success.

  RETURN VALUES

    0 on success.

    1 on error.

*/

static my_bool get_dbopt(const char *dbname, HA_CREATE_INFO *create)

{

  my_dbopt_t *opt;

  uint length;

  my_bool error= 1;

  length= (uint) strlen(dbname);

  rw_rdlock(&LOCK_dboptions);

  if ((opt= (my_dbopt_t*) hash_search(&dboptions, (uchar*) dbname, length)))

  {

    create->default_table_charset= opt->charset;

    error= 0;

  }

  rw_unlock(&LOCK_dboptions);

  return error;

}

/*

  Writes database options into the hash.

  DESCRIPTION

    Inserts database options into the hash, or updates

    options if they are already in the hash.

  RETURN VALUES

    0 on success.

    1 on error.

*/

static my_bool put_dbopt(const char *dbname, HA_CREATE_INFO *create)

{

  my_dbopt_t *opt;

  uint length;

  my_bool error= 0;

  DBUG_ENTER("put_dbopt");

  length= (uint) strlen(dbname);

  rw_wrlock(&LOCK_dboptions);

  if (!(opt= (my_dbopt_t*) hash_search(&dboptions, (uchar*) dbname, length)))

  { 

    /* Options are not in the hash, insert them */

    char *tmp_name;

    if (!my_multi_malloc(MYF(MY_WME | MY_ZEROFILL),

                         &opt, (uint) sizeof(*opt), &tmp_name, (uint) length+1,

                         NullS))

    {

      error= 1;

      goto end;

    }

    opt->name= tmp_name;

    strmov(opt->name, dbname);

    opt->name_length= length;

    if ((error= my_hash_insert(&dboptions, (uchar*) opt)))

    {

      my_free(opt, MYF(0));

      goto end;

    }

  }

  /* Update / write options in hash */

  opt->charset= create->default_table_charset;

end:

  rw_unlock(&LOCK_dboptions);  

  DBUG_RETURN(error);

}

/*

  Deletes database options from the hash.

*/

void del_dbopt(const char *path)

{

  my_dbopt_t *opt;

  rw_wrlock(&LOCK_dboptions);

  if ((opt= (my_dbopt_t *)hash_search(&dboptions, (const uchar*) path,

                                      strlen(path))))

    hash_delete(&dboptions, (uchar*) opt);

  rw_unlock(&LOCK_dboptions);

}

/*

  Create database options file:

  DESCRIPTION

    Currently database default charset is only stored there.

  RETURN VALUES

  0	ok

  1	Could not create file or write to it.  Error sent through my_error()

*/

static bool write_db_opt(THD *thd, const char *path, HA_CREATE_INFO *create)

{

  register File file;

  char buf[256]; // Should be enough for one option

  bool error=1;

  if (!create->default_table_charset)

    create->default_table_charset= thd->variables.collation_server;

  if (put_dbopt(path, create))

    return 1;

  if ((file=my_create(path, CREATE_MODE,O_RDWR | O_TRUNC,MYF(MY_WME))) >= 0)

  {

    ulong length;

    length= (ulong) (strxnmov(buf, sizeof(buf)-1, "default-character-set=",

                              create->default_table_charset->csname,

                              "\ndefault-collation=",

                              create->default_table_charset->name,

                              "\n", NullS) - buf);

    /* Error is written by my_write */

    if (!my_write(file,(uchar*) buf, length, MYF(MY_NABP+MY_WME)))

      error=0;

    my_close(file,MYF(0));

  }

  return error;

}

/*

  Load database options file

  load_db_opt()

  path		Path for option file

  create	Where to store the read options

  DESCRIPTION

  RETURN VALUES

  0	File found

  1	No database file or could not open it

*/

bool load_db_opt(THD *thd, const char *path, HA_CREATE_INFO *create)

{

  File file;

  char buf[256];

  DBUG_ENTER("load_db_opt");

  bool error=1;

  uint nbytes;

  bzero((char*) create,sizeof(*create));

  create->default_table_charset= thd->variables.collation_server;

  /* Check if options for this database are already in the hash */

  if (!get_dbopt(path, create))

    DBUG_RETURN(0);

  /* Otherwise, load options from the .opt file */

  if ((file=my_open(path, O_RDONLY | O_SHARE, MYF(0))) < 0)

    goto err1;

  IO_CACHE cache;

  if (init_io_cache(&cache, file, IO_SIZE, READ_CACHE, 0, 0, MYF(0)))

    goto err2;

  while ((int) (nbytes= my_b_gets(&cache, (char*) buf, sizeof(buf))) > 0)

  {

    char *pos= buf+nbytes-1;

    /* Remove end space and control characters */

    while (pos > buf && !my_isgraph(&my_charset_latin1, pos[-1]))

      pos--;

    *pos=0;

    if ((pos= strchr(buf, '=')))

    {

      if (!strncmp(buf,"default-character-set", (pos-buf)))

      {

        /*

           Try character set name, and if it fails

           try collation name, probably it's an old

           4.1.0 db.opt file, which didn't have

           separate default-character-set and

           default-collation commands.

        */

        if (!(create->default_table_charset=

        get_charset_by_csname(pos+1, MY_CS_PRIMARY, MYF(0))) &&

            !(create->default_table_charset=

              get_charset_by_name(pos+1, MYF(0))))

        {

          sql_print_error("Error while loading database options: '%s':",path);

          sql_print_error(ER(ER_UNKNOWN_CHARACTER_SET),pos+1);

          create->default_table_charset= default_charset_info;

        }

      }

      else if (!strncmp(buf,"default-collation", (pos-buf)))

      {

        if (!(create->default_table_charset= get_charset_by_name(pos+1,

                                                           MYF(0))))

        {

          sql_print_error("Error while loading database options: '%s':",path);

          sql_print_error(ER(ER_UNKNOWN_COLLATION),pos+1);

          create->default_table_charset= default_charset_info;

        }

      }

    }

  }

  /*

    Put the loaded value into the hash.

    Note that another thread could've added the same

    entry to the hash after we called get_dbopt(),

    but it's not an error, as put_dbopt() takes this

    possibility into account.

  */

  error= put_dbopt(path, create);

  end_io_cache(&cache);

err2:

  my_close(file,MYF(0));

err1:

  DBUG_RETURN(error);

}

/*

  Retrieve database options by name. Load database options file or fetch from

  cache.

  SYNOPSIS

    load_db_opt_by_name()

    db_name         Database name

    db_create_info  Where to store the database options

  DESCRIPTION

    load_db_opt_by_name() is a shortcut for load_db_opt().

  NOTE

    Although load_db_opt_by_name() (and load_db_opt()) returns status of

    the operation, it is useless usually and should be ignored. The problem

    is that there are 1) system databases ("mysql") and 2) virtual

    databases ("information_schema"), which do not contain options file.

    So, load_db_opt[_by_name]() returns FALSE for these databases, but this

    is not an error.

    load_db_opt[_by_name]() clears db_create_info structure in any case, so

    even on failure it contains valid data. So, common use case is just

    call load_db_opt[_by_name]() without checking return value and use

    db_create_info right after that.

  RETURN VALUES (read NOTE!)

    FALSE   Success

    TRUE    Failed to retrieve options

*/

bool load_db_opt_by_name(THD *thd, const char *db_name,

                         HA_CREATE_INFO *db_create_info)

{

  char db_opt_path[FN_REFLEN];

  /*

    Pass an empty file name, and the database options file name as extension

    to avoid table name to file name encoding.

  */

  (void) build_table_filename(db_opt_path, sizeof(db_opt_path),

                              db_name, "", MY_DB_OPT_FILE, 0);

  return load_db_opt(thd, db_opt_path, db_create_info);

}

/**

  Return default database collation.

  @param thd     Thread context.

  @param db_name Database name.

  @return CHARSET_INFO object. The operation always return valid character

    set, even if the database does not exist.

*/

CHARSET_INFO *get_default_db_collation(THD *thd, const char *db_name)

{

  HA_CREATE_INFO db_info;

  if (thd->db != NULL && strcmp(db_name, thd->db) == 0)

    return thd->db_charset;

  load_db_opt_by_name(thd, db_name, &db_info);

  /*

    NOTE: even if load_db_opt_by_name() fails,

    db_info.default_table_charset contains valid character set

    (collation_server). We should not fail if load_db_opt_by_name() fails,

    because it is valid case. If a database has been created just by

    "mkdir", it does not contain db.opt file, but it is valid database.

  */

  return db_info.default_table_charset;

}

/*

  Create a database

  SYNOPSIS

  mysql_create_db()

  thd		Thread handler

  db		Name of database to create

		Function assumes that this is already validated.

  create_info	Database create options (like character set)

  silent	Used by replication when internally creating a database.

		In this case the entry should not be logged.

  SIDE-EFFECTS

   1. Report back to client that command succeeded (my_ok)

   2. Report errors to client

   3. Log event to binary log

   (The 'silent' flags turns off 1 and 3.)

  RETURN VALUES

  FALSE ok

  TRUE  Error

*/

int mysql_create_db(THD *thd, char *db, HA_CREATE_INFO *create_info,

                     bool silent)

{

  char	 path[FN_REFLEN+16];

  char	 tmp_query[FN_REFLEN+16];

  long result= 1;

  int error= 0;

  struct stat stat_info;

  uint create_options= create_info ? create_info->options : 0;

  uint path_len;

  DBUG_ENTER("mysql_create_db");

  /* do not create 'information_schema' db */

  if (!my_strcasecmp(system_charset_info, db, INFORMATION_SCHEMA_NAME.str))

  {

    my_error(ER_DB_CREATE_EXISTS, MYF(0), db);

    DBUG_RETURN(-1);

  }

  /*

    Do not create database if another thread is holding read lock.

    Wait for global read lock before acquiring LOCK_mysql_create_db.

    After wait_if_global_read_lock() we have protection against another

    global read lock. If we would acquire LOCK_mysql_create_db first,

    another thread could step in and get the global read lock before we

    reach wait_if_global_read_lock(). If this thread tries the same as we

    (admin a db), it would then go and wait on LOCK_mysql_create_db...

    Furthermore wait_if_global_read_lock() checks if the current thread

    has the global read lock and refuses the operation with

    ER_CANT_UPDATE_WITH_READLOCK if applicable.

  */

  if (wait_if_global_read_lock(thd, 0, 1))

  {

    error= -1;

    goto exit2;

  }

  VOID(pthread_mutex_lock(&LOCK_mysql_create_db));

  /* Check directory */

  path_len= build_table_filename(path, sizeof(path), db, "", "", 0);

  path[path_len-1]= 0;                    // Remove last '/' from path

  if (!stat(path,&stat_info))

  {

    if (!(create_options & HA_LEX_CREATE_IF_NOT_EXISTS))

    {

      my_error(ER_DB_CREATE_EXISTS, MYF(0), db);

      error= -1;

      goto exit;

    }

    push_warning_printf(thd, MYSQL_ERROR::WARN_LEVEL_NOTE,

			ER_DB_CREATE_EXISTS, ER(ER_DB_CREATE_EXISTS), db);

    if (!silent)

      my_ok(thd);

    error= 0;

    goto exit;

  }

  else

  {

    if (errno != ENOENT)

    {

      my_error(EE_STAT, MYF(0), path, errno);

      goto exit;

    }

    if (my_mkdir(path,0777,MYF(0)) < 0)

    {

      my_error(ER_CANT_CREATE_DB, MYF(0), db, my_errno);

      error= -1;

      goto exit;

    }

  }

  path[path_len-1]= FN_LIBCHAR;

  strmake(path+path_len, MY_DB_OPT_FILE, sizeof(path)-path_len-1);

  if (write_db_opt(thd, path, create_info))

  {

    /*

      Could not create options file.

      Restore things to beginning.

    */

    path[path_len]= 0;

    if (rmdir(path) >= 0)

    {

      error= -1;

      goto exit;

    }

    /*

      We come here when we managed to create the database, but not the option

      file.  In this case it's best to just continue as if nothing has

      happened.  (This is a very unlikely senario)

    */

  }

  if (!silent)

  {

    char *query;

    uint query_length;

    if (!thd->query)				// Only in replication

    {

      query= 	     tmp_query;

      query_length= (uint) (strxmov(tmp_query,"create database `",

                                    db, "`", NullS) - tmp_query);

    }

    else

    {

      query= 	    thd->query;

      query_length= thd->query_length;

    }

    ha_binlog_log_query(thd, 0, LOGCOM_CREATE_DB,

                        query, query_length,

                        db, "");

    if (mysql_bin_log.is_open())

    {

      Query_log_event qinfo(thd, query, query_length, 0, 

			    /* suppress_use */ TRUE);

      /*

	Write should use the database being created as the "current

        database" and not the threads current database, which is the

        default. If we do not change the "current database" to the

        database being created, the CREATE statement will not be

        replicated when using --binlog-do-db to select databases to be

        replicated. 

	An example (--binlog-do-db=sisyfos):

          CREATE DATABASE bob;        # Not replicated

          USE bob;                    # 'bob' is the current database

          CREATE DATABASE sisyfos;    # Not replicated since 'bob' is

                                      # current database.

          USE sisyfos;                # Will give error on slave since

                                      # database does not exist.

      */

      qinfo.db     = db;

      qinfo.db_len = strlen(db);

      /* These DDL methods and logging protected with LOCK_mysql_create_db */

      mysql_bin_log.write(&qinfo);

    }

    my_ok(thd, result);

  }

exit:

  VOID(pthread_mutex_unlock(&LOCK_mysql_create_db));

  start_waiting_global_read_lock(thd);

exit2:

  DBUG_RETURN(error);

}

/* db-name is already validated when we come here */

bool mysql_alter_db(THD *thd, const char *db, HA_CREATE_INFO *create_info)

{

  char path[FN_REFLEN+16];

  long result=1;

  int error= 0;

  DBUG_ENTER("mysql_alter_db");

  /*

    Do not alter database if another thread is holding read lock.

    Wait for global read lock before acquiring LOCK_mysql_create_db.

    After wait_if_global_read_lock() we have protection against another

    global read lock. If we would acquire LOCK_mysql_create_db first,

    another thread could step in and get the global read lock before we

    reach wait_if_global_read_lock(). If this thread tries the same as we

    (admin a db), it would then go and wait on LOCK_mysql_create_db...

    Furthermore wait_if_global_read_lock() checks if the current thread

    has the global read lock and refuses the operation with

    ER_CANT_UPDATE_WITH_READLOCK if applicable.

  */

  if ((error=wait_if_global_read_lock(thd,0,1)))

    goto exit2;

  VOID(pthread_mutex_lock(&LOCK_mysql_create_db));

  /* 

     Recreate db options file: /dbpath/.db.opt

     We pass MY_DB_OPT_FILE as "extension" to avoid

     "table name to file name" encoding.

  */

  build_table_filename(path, sizeof(path), db, "", MY_DB_OPT_FILE, 0);

  if ((error=write_db_opt(thd, path, create_info)))

    goto exit;

  /* Change options if current database is being altered. */

  if (thd->db && !strcmp(thd->db,db))

  {

    thd->db_charset= create_info->default_table_charset ?

		     create_info->default_table_charset :

		     thd->variables.collation_server;

    thd->variables.collation_database= thd->db_charset;

  }

  ha_binlog_log_query(thd, 0, LOGCOM_ALTER_DB,

                      thd->query, thd->query_length,

                      db, "");

  if (mysql_bin_log.is_open())

  {

    Query_log_event qinfo(thd, thd->query, thd->query_length, 0,

			  /* suppress_use */ TRUE);

    /*

      Write should use the database being created as the "current

      database" and not the threads current database, which is the

      default.

    */

    qinfo.db     = db;

    qinfo.db_len = strlen(db);

    thd->clear_error();

    /* These DDL methods and logging protected with LOCK_mysql_create_db */

    mysql_bin_log.write(&qinfo);

  }

  my_ok(thd, result);

exit:

  VOID(pthread_mutex_unlock(&LOCK_mysql_create_db));

  start_waiting_global_read_lock(thd);

exit2:

  DBUG_RETURN(error);

}

/*

  Drop all tables in a database and the database itself

  SYNOPSIS

    mysql_rm_db()

    thd			Thread handle

    db			Database name in the case given by user

		        It's already validated and set to lower case

                        (if needed) when we come here

    if_exists		Don't give error if database doesn't exists

    silent		Don't generate errors

  RETURN

    FALSE ok (Database dropped)

    ERROR Error

*/

bool mysql_rm_db(THD *thd,char *db,bool if_exists, bool silent)

{

  long deleted=0;

  int error= 0;

  char	path[FN_REFLEN+16];

  MY_DIR *dirp;

  uint length;