ha_berkeley.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:74k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. /*
  14.   TODO:
  15.   - Not compressed keys should use cmp_fix_length_key
  16.   - Don't automaticly pack all string keys (To do this we need to modify
  17.     CREATE TABLE so that one can use the pack_keys argument per key).
  18.   - An argument to pack_key that we don't want compression.
  19.   - DB_DBT_USERMEM should be used for fixed length tables
  20.     We will need an updated Berkeley DB version for this.
  21.   - Killing threads that has got a 'deadlock'
  22.   - SHOW TABLE STATUS should give more information about the table.
  23.   - Get a more accurate count of the number of rows (estimate_rows_upper_bound()).
  24.     We could store the found number of rows when the table is scanned and
  25.     then increment the counter for each attempted write.
  26.   - We will need to extend the manager thread to makes checkpoints at
  27.      given intervals.
  28.   - When not using UPDATE IGNORE, don't make a sub transaction but abort
  29.     the main transaction on errors.
  30.   - Handling of drop table during autocommit=0 ?
  31.     (Should we just give an error in this case if there is a pending
  32.     transaction ?)
  33.   - When using ALTER TABLE IGNORE, we should not start an transaction, but do
  34.     everything wthout transactions.
  35.   - When we do rollback, we need to subtract the number of changed rows
  36.     from the updated tables.
  37.   Testing of:
  38.   - Mark tables that participate in a transaction so that they are not
  39.     closed during the transaction.  We need to test what happens if
  40.     MySQL closes a table that is updated by a not commited transaction.
  41. */
  42. #ifdef USE_PRAGMA_IMPLEMENTATION
  43. #pragma implementation // gcc: Class implementation
  44. #endif
  45. #include "mysql_priv.h"
  46. #ifdef HAVE_BERKELEY_DB
  47. #include <m_ctype.h>
  48. #include <myisampack.h>
  49. #include <hash.h>
  50. #include "ha_berkeley.h"
  51. #include "sql_manager.h"
  52. #include <stdarg.h>
  53. #define HA_BERKELEY_ROWS_IN_TABLE 10000 /* to get optimization right */
  54. #define HA_BERKELEY_RANGE_COUNT   100
  55. #define HA_BERKELEY_MAX_ROWS   10000000 /* Max rows in table */
  56. /* extra rows for estimate_rows_upper_bound() */
  57. #define HA_BERKELEY_EXTRA_ROWS   100
  58. /* Bits for share->status */
  59. #define STATUS_PRIMARY_KEY_INIT 1
  60. #define STATUS_ROW_COUNT_INIT 2
  61. #define STATUS_BDB_ANALYZE 4
  62. const char *ha_berkeley_ext=".db";
  63. bool berkeley_shared_data=0;
  64. u_int32_t berkeley_init_flags= DB_PRIVATE | DB_RECOVER, berkeley_env_flags=0,
  65.           berkeley_lock_type=DB_LOCK_DEFAULT;
  66. ulong berkeley_cache_size, berkeley_log_buffer_size, berkeley_log_file_size=0;
  67. char *berkeley_home, *berkeley_tmpdir, *berkeley_logdir;
  68. long berkeley_lock_scan_time=0;
  69. ulong berkeley_trans_retry=1;
  70. ulong berkeley_max_lock;
  71. pthread_mutex_t bdb_mutex;
  72. static DB_ENV *db_env;
  73. static HASH bdb_open_tables;
  74. const char *berkeley_lock_names[] =
  75. { "DEFAULT", "OLDEST","RANDOM","YOUNGEST",0 };
  76. u_int32_t berkeley_lock_types[]=
  77. { DB_LOCK_DEFAULT, DB_LOCK_OLDEST, DB_LOCK_RANDOM };
  78. TYPELIB berkeley_lock_typelib= {array_elements(berkeley_lock_names)-1,"",
  79. berkeley_lock_names, NULL};
  80. static void berkeley_print_error(const char *db_errpfx, char *buffer);
  81. static byte* bdb_get_key(BDB_SHARE *share,uint *length,
  82.  my_bool not_used __attribute__((unused)));
  83. static BDB_SHARE *get_share(const char *table_name, TABLE *table);
  84. static int free_share(BDB_SHARE *share, TABLE *table, uint hidden_primary_key,
  85.       bool mutex_is_locked);
  86. static int write_status(DB *status_block, char *buff, uint length);
  87. static void update_status(BDB_SHARE *share, TABLE *table);
  88. static void berkeley_noticecall(DB_ENV *db_env, db_notices notice);
  89. /* General functions */
  90. bool berkeley_init(void)
  91. {
  92.   DBUG_ENTER("berkeley_init");
  93.   if (!berkeley_tmpdir)
  94.     berkeley_tmpdir=mysql_tmpdir;
  95.   if (!berkeley_home)
  96.     berkeley_home=mysql_real_data_home;
  97.   DBUG_PRINT("bdb",("berkeley_home: %s",mysql_real_data_home));
  98.   /*
  99.     If we don't set set_lg_bsize() we will get into trouble when
  100.     trying to use many open BDB tables.
  101.     If log buffer is not set, assume that the we will need 512 byte per
  102.     open table.  This is a number that we have reached by testing.
  103.   */
  104.   if (!berkeley_log_buffer_size)
  105.   {
  106.     berkeley_log_buffer_size= max(table_cache_size*512,32*1024);
  107.   }
  108.   /*
  109.     Berkeley DB require that
  110.     berkeley_log_file_size >= berkeley_log_buffer_size*4
  111.   */
  112.   berkeley_log_file_size= berkeley_log_buffer_size*4;
  113.   berkeley_log_file_size= MY_ALIGN(berkeley_log_file_size,1024*1024L);
  114.   berkeley_log_file_size= max(berkeley_log_file_size, 10*1024*1024L);
  115.   if (db_env_create(&db_env,0))
  116.     DBUG_RETURN(1); /* purecov: inspected */
  117.   db_env->set_errcall(db_env,berkeley_print_error);
  118.   db_env->set_errpfx(db_env,"bdb");
  119.   db_env->set_noticecall(db_env, berkeley_noticecall);
  120.   db_env->set_tmp_dir(db_env, berkeley_tmpdir);
  121.   db_env->set_data_dir(db_env, mysql_data_home);
  122.   db_env->set_flags(db_env, berkeley_env_flags, 1);
  123.   if (berkeley_logdir)
  124.     db_env->set_lg_dir(db_env, berkeley_logdir); /* purecov: tested */
  125.   if (opt_endinfo)
  126.     db_env->set_verbose(db_env,
  127. DB_VERB_CHKPOINT | DB_VERB_DEADLOCK | DB_VERB_RECOVERY,
  128. 1);
  129.   db_env->set_cachesize(db_env, 0, berkeley_cache_size, 0);
  130.   db_env->set_lg_max(db_env, berkeley_log_file_size);
  131.   db_env->set_lg_bsize(db_env, berkeley_log_buffer_size);
  132.   db_env->set_lk_detect(db_env, berkeley_lock_type);
  133.   if (berkeley_max_lock)
  134.     db_env->set_lk_max(db_env, berkeley_max_lock);
  135.   if (db_env->open(db_env,
  136.    berkeley_home,
  137.    berkeley_init_flags |  DB_INIT_LOCK |
  138.    DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_TXN |
  139.    DB_CREATE | DB_THREAD, 0666))
  140.   {
  141.     db_env->close(db_env,0); /* purecov: inspected */
  142.     db_env=0; /* purecov: inspected */
  143.     goto err;
  144.   }
  145.   (void) hash_init(&bdb_open_tables,system_charset_info,32,0,0,
  146.    (hash_get_key) bdb_get_key,0,0);
  147.   pthread_mutex_init(&bdb_mutex,MY_MUTEX_INIT_FAST);
  148. err:
  149.   DBUG_RETURN(db_env == 0);
  150. }
  151. bool berkeley_end(void)
  152. {
  153.   int error;
  154.   DBUG_ENTER("berkeley_end");
  155.   if (!db_env)
  156.     return 1; /* purecov: tested */
  157.   berkeley_cleanup_log_files();
  158.   error=db_env->close(db_env,0); // Error is logged
  159.   db_env=0;
  160.   hash_free(&bdb_open_tables);
  161.   pthread_mutex_destroy(&bdb_mutex);
  162.   DBUG_RETURN(error != 0);
  163. }
  164. bool berkeley_flush_logs()
  165. {
  166.   int error;
  167.   bool result=0;
  168.   DBUG_ENTER("berkeley_flush_logs");
  169.   if ((error=db_env->log_flush(db_env,0)))
  170.   {
  171.     my_error(ER_ERROR_DURING_FLUSH_LOGS,MYF(0),error); /* purecov: inspected */
  172.     result=1; /* purecov: inspected */
  173.   }
  174.   if ((error=db_env->txn_checkpoint(db_env,0,0,0)))
  175.   {
  176.     my_error(ER_ERROR_DURING_CHECKPOINT,MYF(0),error); /* purecov: inspected */
  177.     result=1; /* purecov: inspected */
  178.   }
  179.   DBUG_RETURN(result);
  180. }
  181. int berkeley_commit(THD *thd, void *trans)
  182. {
  183.   DBUG_ENTER("berkeley_commit");
  184.   DBUG_PRINT("trans",("ending transaction %s",
  185.       trans == thd->transaction.stmt.bdb_tid ? "stmt" : "all"));
  186.   int error=txn_commit((DB_TXN*) trans,0);
  187. #ifndef DBUG_OFF
  188.   if (error)
  189.     DBUG_PRINT("error",("error: %d",error)); /* purecov: inspected */
  190. #endif
  191.   DBUG_RETURN(error);
  192. }
  193. int berkeley_rollback(THD *thd, void *trans)
  194. {
  195.   DBUG_ENTER("berkeley_rollback");
  196.   DBUG_PRINT("trans",("aborting transaction %s",
  197.       trans == thd->transaction.stmt.bdb_tid ? "stmt" : "all"));
  198.   int error=txn_abort((DB_TXN*) trans);
  199.   DBUG_RETURN(error);
  200. }
  201. int berkeley_show_logs(Protocol *protocol)
  202. {
  203.   char **all_logs, **free_logs, **a, **f;
  204.   int error=1;
  205.   MEM_ROOT **root_ptr= my_pthread_getspecific_ptr(MEM_ROOT**,THR_MALLOC);
  206.   MEM_ROOT show_logs_root, *old_mem_root= *root_ptr;
  207.   DBUG_ENTER("berkeley_show_logs");
  208.   init_sql_alloc(&show_logs_root, BDB_LOG_ALLOC_BLOCK_SIZE,
  209.  BDB_LOG_ALLOC_BLOCK_SIZE);
  210.   *root_ptr= &show_logs_root;
  211.   if ((error= db_env->log_archive(db_env, &all_logs,
  212.   DB_ARCH_ABS | DB_ARCH_LOG)) ||
  213.       (error= db_env->log_archive(db_env, &free_logs, DB_ARCH_ABS)))
  214.   {
  215.     DBUG_PRINT("error", ("log_archive failed (error %d)", error));
  216.     db_env->err(db_env, error, "log_archive: DB_ARCH_ABS");
  217.     if (error== DB_NOTFOUND)
  218.       error=0; // No log files
  219.     goto err;
  220.   }
  221.   /* Error is 0 here */
  222.   if (all_logs)
  223.   {
  224.     for (a = all_logs, f = free_logs; *a; ++a)
  225.     {
  226.       protocol->prepare_for_resend();
  227.       protocol->store(*a, system_charset_info);
  228.       protocol->store("BDB", 3, system_charset_info);
  229.       if (f && *f && strcmp(*a, *f) == 0)
  230.       {
  231. f++;
  232. protocol->store(SHOW_LOG_STATUS_FREE, system_charset_info);
  233.       }
  234.       else
  235. protocol->store(SHOW_LOG_STATUS_INUSE, system_charset_info);
  236.       if (protocol->write())
  237.       {
  238. error=1;
  239. goto err;
  240.       }
  241.     }
  242.   }
  243. err:
  244.   free_root(&show_logs_root,MYF(0));
  245.   *root_ptr= old_mem_root;
  246.   DBUG_RETURN(error);
  247. }
  248. static void berkeley_print_error(const char *db_errpfx, char *buffer)
  249. {
  250.   sql_print_error("%s:  %s",db_errpfx,buffer); /* purecov: tested */
  251. }
  252. static void berkeley_noticecall(DB_ENV *db_env, db_notices notice)
  253. {
  254.   switch (notice)
  255.   {
  256.   case DB_NOTICE_LOGFILE_CHANGED: /* purecov: tested */
  257.     pthread_mutex_lock(&LOCK_manager);
  258.     manager_status |= MANAGER_BERKELEY_LOG_CLEANUP;
  259.     pthread_mutex_unlock(&LOCK_manager);
  260.     pthread_cond_signal(&COND_manager);
  261.     break;
  262.   }
  263. }
  264. void berkeley_cleanup_log_files(void)
  265. {
  266.   DBUG_ENTER("berkeley_cleanup_log_files");
  267.   char **names;
  268.   int error;
  269. // by HF. Sometimes it crashes. TODO - find out why
  270. #ifndef EMBEDDED_LIBRARY
  271.   /* XXX: Probably this should be done somewhere else, and
  272.    * should be tunable by the user. */
  273.   if ((error = db_env->txn_checkpoint(db_env, 0, 0, 0)))
  274.     my_error(ER_ERROR_DURING_CHECKPOINT, MYF(0), error); /* purecov: inspected */
  275. #endif
  276.   if ((error = db_env->log_archive(db_env, &names, DB_ARCH_ABS)) != 0)
  277.   {
  278.     DBUG_PRINT("error", ("log_archive failed (error %d)", error)); /* purecov: inspected */
  279.     db_env->err(db_env, error, "log_archive: DB_ARCH_ABS"); /* purecov: inspected */
  280.     DBUG_VOID_RETURN; /* purecov: inspected */
  281.   }
  282.   if (names)
  283.   { /* purecov: tested */
  284.     char **np; /* purecov: tested */
  285.     for (np = names; *np; ++np) /* purecov: tested */
  286.       my_delete(*np, MYF(MY_WME)); /* purecov: tested */
  287.     free(names); /* purecov: tested */
  288.   }
  289.   DBUG_VOID_RETURN;
  290. }
  291. /*****************************************************************************
  292. ** Berkeley DB tables
  293. *****************************************************************************/
  294. static const char *ha_bdb_bas_exts[]= { ha_berkeley_ext, NullS };
  295. const char **ha_berkeley::bas_ext() const
  296. { return ha_bdb_bas_exts; }
  297. ulong ha_berkeley::index_flags(uint idx, uint part, bool all_parts) const
  298. {
  299.   ulong flags= (HA_READ_NEXT | HA_READ_PREV | HA_READ_ORDER | HA_KEYREAD_ONLY
  300.                 | HA_READ_RANGE);
  301.   for (uint i= all_parts ? 0 : part ; i <= part ; i++)
  302.   {
  303.     if (table->key_info[idx].key_part[i].field->type() == FIELD_TYPE_BLOB)
  304.     {
  305.       /* We can't use BLOBS to shortcut sorts */
  306.       flags&= ~(HA_READ_ORDER | HA_KEYREAD_ONLY | HA_READ_RANGE);
  307.       break;
  308.     }
  309.     switch (table->key_info[idx].key_part[i].field->key_type()) {
  310.     case HA_KEYTYPE_TEXT:
  311.     case HA_KEYTYPE_VARTEXT:
  312.       /*
  313.         As BDB stores only one copy of equal strings, we can't use key read
  314.         on these. Binary collations do support key read though.
  315.       */
  316.       if (!(table->key_info[idx].key_part[i].field->charset()->state
  317.            & MY_CS_BINSORT))
  318.         flags&= ~HA_KEYREAD_ONLY;
  319.       break;
  320.     default:                                    // Keep compiler happy
  321.       break;
  322.     }
  323.   }
  324.   return flags;
  325. }
  326. static int
  327. berkeley_cmp_hidden_key(DB* file, const DBT *new_key, const DBT *saved_key)
  328. {
  329.   ulonglong a=uint5korr((char*) new_key->data);
  330.   ulonglong b=uint5korr((char*) saved_key->data);
  331.   return  a < b ? -1 : (a > b ? 1 : 0);
  332. }
  333. static int
  334. berkeley_cmp_packed_key(DB *file, const DBT *new_key, const DBT *saved_key)
  335. {
  336.   KEY *key=       (new_key->app_private ? (KEY*) new_key->app_private :
  337.        (KEY*) (file->app_private));
  338.   char *new_key_ptr=  (char*) new_key->data;
  339.   char *saved_key_ptr=(char*) saved_key->data;
  340.   KEY_PART_INFO *key_part= key->key_part, *end=key_part+key->key_parts;
  341.   uint key_length=new_key->size;
  342.   for (; key_part != end && (int) key_length > 0; key_part++)
  343.   {
  344.     int cmp;
  345.     if (key_part->null_bit)
  346.     {
  347.       if (*new_key_ptr != *saved_key_ptr++)
  348. return ((int) *new_key_ptr - (int) saved_key_ptr[-1]);
  349.       key_length--;
  350.       if (!*new_key_ptr++)
  351. continue;
  352.     }
  353.     if ((cmp=key_part->field->pack_cmp(new_key_ptr,saved_key_ptr,
  354.        key_part->length)))
  355.       return cmp;
  356.     uint length=key_part->field->packed_col_length(new_key_ptr,
  357.    key_part->length);
  358.     new_key_ptr+=length;
  359.     key_length-=length;
  360.     saved_key_ptr+=key_part->field->packed_col_length(saved_key_ptr,
  361.       key_part->length);
  362.   }
  363.   return key->handler.bdb_return_if_eq;
  364. }
  365. /* The following is not yet used; Should be used for fixed length keys */
  366. #ifdef NOT_YET
  367. static int
  368. berkeley_cmp_fix_length_key(DB *file, const DBT *new_key, const DBT *saved_key)
  369. {
  370.   KEY *key=       (new_key->app_private ? (KEY*) new_key->app_private :
  371.        (KEY*) (file->app_private));
  372.   char *new_key_ptr=  (char*) new_key->data;
  373.   char *saved_key_ptr=(char*) saved_key->data;
  374.   KEY_PART_INFO *key_part= key->key_part, *end=key_part+key->key_parts;
  375.   uint key_length=new_key->size;
  376.   for (; key_part != end && (int) key_length > 0 ; key_part++)
  377.   {
  378.     int cmp;
  379.     if ((cmp=key_part->field->pack_cmp(new_key_ptr,saved_key_ptr,0)))
  380.       return cmp;
  381.     new_key_ptr+=key_part->length;
  382.     key_length-= key_part->length;
  383.     saved_key_ptr+=key_part->length;
  384.   }
  385.   return key->handler.bdb_return_if_eq;
  386. }
  387. #endif
  388. /* Compare key against row */
  389. static bool
  390. berkeley_key_cmp(TABLE *table, KEY *key_info, const char *key, uint key_length)
  391. {
  392.   KEY_PART_INFO *key_part= key_info->key_part,
  393. *end=key_part+key_info->key_parts;
  394.   for (; key_part != end && (int) key_length > 0; key_part++)
  395.   {
  396.     int cmp;
  397.     if (key_part->null_bit)
  398.     {
  399.       key_length--;
  400.       /*
  401. With the current usage, the following case will always be FALSE,
  402. because NULL keys are sorted before any other key
  403.       */
  404.       if (*key != (table->record[0][key_part->null_offset] &
  405.    key_part->null_bit) ? 0 : 1)
  406. return 1;
  407.       if (!*key++) // Null value
  408. continue;
  409.     }
  410.     if ((cmp=key_part->field->pack_cmp(key,key_part->length)))
  411.       return cmp;
  412.     uint length=key_part->field->packed_col_length(key,key_part->length);
  413.     key+=length;
  414.     key_length-=length;
  415.   }
  416.   return 0; // Identical keys
  417. }
  418. int ha_berkeley::open(const char *name, int mode, uint test_if_locked)
  419. {
  420.   char name_buff[FN_REFLEN];
  421.   uint open_mode=(mode == O_RDONLY ? DB_RDONLY : 0) | DB_THREAD;
  422.   int error;
  423.   DBUG_ENTER("ha_berkeley::open");
  424.   /* Open primary key */
  425.   hidden_primary_key=0;
  426.   if ((primary_key=table->primary_key) >= MAX_KEY)
  427.   { // No primary key
  428.     primary_key=table->keys;
  429.     key_used_on_scan=MAX_KEY;
  430.     ref_length=hidden_primary_key=BDB_HIDDEN_PRIMARY_KEY_LENGTH;
  431.   }
  432.   else
  433.     key_used_on_scan=primary_key;
  434.   /* Need some extra memory in case of packed keys */
  435.   uint max_key_length= table->max_key_length + MAX_REF_PARTS*3;
  436.   if (!(alloc_ptr=
  437. my_multi_malloc(MYF(MY_WME),
  438. &key_buff,  max_key_length,
  439. &key_buff2, max_key_length,
  440. &primary_key_buff,
  441. (hidden_primary_key ? 0 :
  442.  table->key_info[table->primary_key].key_length),
  443. NullS)))
  444.     DBUG_RETURN(1); /* purecov: inspected */
  445.   if (!(rec_buff= (byte*) my_malloc((alloced_rec_buff_length=
  446.      table->rec_buff_length),
  447.     MYF(MY_WME))))
  448.   {
  449.     my_free(alloc_ptr,MYF(0)); /* purecov: inspected */
  450.     DBUG_RETURN(1); /* purecov: inspected */
  451.   }
  452.   /* Init shared structure */
  453.   if (!(share=get_share(name,table)))
  454.   {
  455.     my_free((char*) rec_buff,MYF(0)); /* purecov: inspected */
  456.     my_free(alloc_ptr,MYF(0)); /* purecov: inspected */
  457.     DBUG_RETURN(1); /* purecov: inspected */
  458.   }
  459.   thr_lock_data_init(&share->lock,&lock,(void*) 0);
  460.   key_file = share->key_file;
  461.   key_type = share->key_type;
  462.   bzero((char*) &current_row,sizeof(current_row));
  463.   /* Fill in shared structure, if needed */
  464.   pthread_mutex_lock(&share->mutex);
  465.   file = share->file;
  466.   if (!share->use_count++)
  467.   {
  468.     if ((error=db_create(&file, db_env, 0)))
  469.     {
  470.       free_share(share,table, hidden_primary_key,1); /* purecov: inspected */
  471.       my_free((char*) rec_buff,MYF(0)); /* purecov: inspected */
  472.       my_free(alloc_ptr,MYF(0)); /* purecov: inspected */
  473.       my_errno=error; /* purecov: inspected */
  474.       DBUG_RETURN(1); /* purecov: inspected */
  475.     }
  476.     share->file = file;
  477.     file->set_bt_compare(file,
  478.  (hidden_primary_key ? berkeley_cmp_hidden_key :
  479.   berkeley_cmp_packed_key));
  480.     if (!hidden_primary_key)
  481.       file->app_private= (void*) (table->key_info+table->primary_key);
  482.     if ((error= txn_begin(db_env, 0, (DB_TXN**) &transaction, 0)) ||
  483. (error= (file->open(file, transaction,
  484.     fn_format(name_buff, name, "", ha_berkeley_ext,
  485.       2 | 4),
  486.     "main", DB_BTREE, open_mode, 0))) ||
  487. (error= transaction->commit(transaction, 0)))
  488.     {
  489.       free_share(share,table, hidden_primary_key,1); /* purecov: inspected */
  490.       my_free((char*) rec_buff,MYF(0)); /* purecov: inspected */
  491.       my_free(alloc_ptr,MYF(0)); /* purecov: inspected */
  492.       my_errno=error; /* purecov: inspected */
  493.       DBUG_RETURN(1); /* purecov: inspected */
  494.     }
  495.     /* Open other keys;  These are part of the share structure */
  496.     key_file[primary_key]=file;
  497.     key_type[primary_key]=DB_NOOVERWRITE;
  498.     DB **ptr=key_file;
  499.     for (uint i=0, used_keys=0; i < table->keys ; i++, ptr++)
  500.     {
  501.       char part[7];
  502.       if (i != primary_key)
  503.       {
  504. if ((error=db_create(ptr, db_env, 0)))
  505. {
  506.   close(); /* purecov: inspected */
  507.   my_errno=error; /* purecov: inspected */
  508.   DBUG_RETURN(1); /* purecov: inspected */
  509. }
  510. sprintf(part,"key%02d",++used_keys);
  511. key_type[i]=table->key_info[i].flags & HA_NOSAME ? DB_NOOVERWRITE : 0;
  512. (*ptr)->set_bt_compare(*ptr, berkeley_cmp_packed_key);
  513. (*ptr)->app_private= (void*) (table->key_info+i);
  514. if (!(table->key_info[i].flags & HA_NOSAME))
  515. {
  516.   DBUG_PRINT("bdb",("Setting DB_DUP for key %u", i));
  517.   (*ptr)->set_flags(*ptr, DB_DUP);
  518. }
  519. if ((error= txn_begin(db_env, 0, (DB_TXN**) &transaction, 0)) ||
  520.     (error=((*ptr)->open(*ptr, transaction, name_buff, part, DB_BTREE,
  521.  open_mode, 0))) ||
  522.     (error= transaction->commit(transaction, 0)))
  523. {
  524.   close(); /* purecov: inspected */
  525.   my_errno=error; /* purecov: inspected */
  526.   DBUG_RETURN(1); /* purecov: inspected */
  527. }
  528.       }
  529.     }
  530.     /* Calculate pack_length of primary key */
  531.     share->fixed_length_primary_key=1;
  532.     if (!hidden_primary_key)
  533.     {
  534.       ref_length=0;
  535.       KEY_PART_INFO *key_part= table->key_info[primary_key].key_part;
  536.       KEY_PART_INFO *end=key_part+table->key_info[primary_key].key_parts;
  537.       for (; key_part != end ; key_part++)
  538. ref_length+= key_part->field->max_packed_col_length(key_part->length);
  539.       share->fixed_length_primary_key=
  540. (ref_length == table->key_info[primary_key].key_length);
  541.       share->status|=STATUS_PRIMARY_KEY_INIT;
  542.     }    
  543.     share->ref_length=ref_length;
  544.   }
  545.   ref_length=share->ref_length; // If second open
  546.   pthread_mutex_unlock(&share->mutex);
  547.   transaction=0;
  548.   cursor=0;
  549.   key_read=0;
  550.   block_size=8192; // Berkeley DB block size
  551.   share->fixed_length_row=!(table->db_create_options & HA_OPTION_PACK_RECORD);
  552.   get_status();
  553.   info(HA_STATUS_NO_LOCK | HA_STATUS_VARIABLE | HA_STATUS_CONST);
  554.   DBUG_RETURN(0);
  555. }
  556. int ha_berkeley::close(void)
  557. {
  558.   DBUG_ENTER("ha_berkeley::close");
  559.   my_free((char*) rec_buff,MYF(MY_ALLOW_ZERO_PTR));
  560.   my_free(alloc_ptr,MYF(MY_ALLOW_ZERO_PTR));
  561.   ha_berkeley::extra(HA_EXTRA_RESET); // current_row buffer
  562.   DBUG_RETURN(free_share(share,table, hidden_primary_key,0));
  563. }
  564. /* Reallocate buffer if needed */
  565. bool ha_berkeley::fix_rec_buff_for_blob(ulong length)
  566. {
  567.   if (! rec_buff || length > alloced_rec_buff_length)
  568.   {
  569.     byte *newptr;
  570.     if (!(newptr=(byte*) my_realloc((gptr) rec_buff, length,
  571.     MYF(MY_ALLOW_ZERO_PTR))))
  572.       return 1; /* purecov: inspected */
  573.     rec_buff=newptr;
  574.     alloced_rec_buff_length=length;
  575.   }
  576.   return 0;
  577. }
  578. /* Calculate max length needed for row */
  579. ulong ha_berkeley::max_row_length(const byte *buf)
  580. {
  581.   ulong length=table->reclength + table->fields*2;
  582.   for (Field_blob **ptr=table->blob_field ; *ptr ; ptr++)
  583.     length+= (*ptr)->get_length((char*) buf+(*ptr)->offset())+2;
  584.   return length;
  585. }
  586. /*
  587.   Pack a row for storage.  If the row is of fixed length, just store the
  588.   row 'as is'.
  589.   If not, we will generate a packed row suitable for storage.
  590.   This will only fail if we don't have enough memory to pack the row, which;
  591.   may only happen in rows with blobs,  as the default row length is
  592.   pre-allocated.
  593. */
  594. int ha_berkeley::pack_row(DBT *row, const byte *record, bool new_row)
  595. {
  596.   bzero((char*) row,sizeof(*row));
  597.   if (share->fixed_length_row)
  598.   {
  599.     row->data=(void*) record;
  600.     row->size=table->reclength+hidden_primary_key;
  601.     if (hidden_primary_key)
  602.     {
  603.       if (new_row)
  604. get_auto_primary_key(current_ident);
  605.       memcpy_fixed((char*) record+table->reclength, (char*) current_ident,
  606.    BDB_HIDDEN_PRIMARY_KEY_LENGTH);
  607.     }
  608.     return 0;
  609.   }
  610.   if (table->blob_fields)
  611.   {
  612.     if (fix_rec_buff_for_blob(max_row_length(record)))
  613.       return HA_ERR_OUT_OF_MEM; /* purecov: inspected */
  614.   }
  615.   /* Copy null bits */
  616.   memcpy(rec_buff, record, table->null_bytes);
  617.   byte *ptr=rec_buff + table->null_bytes;
  618.   for (Field **field=table->field ; *field ; field++)
  619.     ptr=(byte*) (*field)->pack((char*) ptr,
  620.        (char*) record + (*field)->offset());
  621.   if (hidden_primary_key)
  622.   {
  623.     if (new_row)
  624.       get_auto_primary_key(current_ident);
  625.     memcpy_fixed((char*) ptr, (char*) current_ident,
  626.  BDB_HIDDEN_PRIMARY_KEY_LENGTH);
  627.     ptr+=BDB_HIDDEN_PRIMARY_KEY_LENGTH;
  628.   }
  629.   row->data=rec_buff;
  630.   row->size= (size_t) (ptr - rec_buff);
  631.   return 0;
  632. }
  633. void ha_berkeley::unpack_row(char *record, DBT *row)
  634. {
  635.   if (share->fixed_length_row)
  636.     memcpy(record,(char*) row->data,table->reclength+hidden_primary_key);
  637.   else
  638.   {
  639.     /* Copy null bits */
  640.     const char *ptr= (const char*) row->data;
  641.     memcpy(record, ptr, table->null_bytes);
  642.     ptr+=table->null_bytes;
  643.     for (Field **field=table->field ; *field ; field++)
  644.       ptr= (*field)->unpack(record + (*field)->offset(), ptr);
  645.   }
  646. }
  647. /* Store the key and the primary key into the row */
  648. void ha_berkeley::unpack_key(char *record, DBT *key, uint index)
  649. {
  650.   KEY *key_info=table->key_info+index;
  651.   KEY_PART_INFO *key_part= key_info->key_part,
  652. *end=key_part+key_info->key_parts;
  653.   char *pos=(char*) key->data;
  654.   for (; key_part != end; key_part++)
  655.   {
  656.     if (key_part->null_bit)
  657.     {
  658.       if (!*pos++) // Null value
  659.       {
  660. /*
  661.   We don't need to reset the record data as we will not access it
  662.   if the null data is set
  663. */
  664. record[key_part->null_offset]|=key_part->null_bit;
  665. continue;
  666.       }
  667.       record[key_part->null_offset]&= ~key_part->null_bit;
  668.     }
  669.     pos= (char*) key_part->field->unpack_key(record + key_part->field->offset(),
  670.                                              pos, key_part->length);
  671.   }
  672. }
  673. /*
  674.   Create a packed key from from a row
  675.   This will never fail as the key buffer is pre allocated.
  676. */
  677. DBT *ha_berkeley::create_key(DBT *key, uint keynr, char *buff,
  678.      const byte *record, int key_length)
  679. {
  680.   bzero((char*) key,sizeof(*key));
  681.   if (hidden_primary_key && keynr == primary_key)
  682.   {
  683.     /* We don't need to set app_private here */
  684.     key->data=current_ident;
  685.     key->size=BDB_HIDDEN_PRIMARY_KEY_LENGTH;
  686.     return key;
  687.   }
  688.   KEY *key_info=table->key_info+keynr;
  689.   KEY_PART_INFO *key_part=key_info->key_part;
  690.   KEY_PART_INFO *end=key_part+key_info->key_parts;
  691.   DBUG_ENTER("create_key");
  692.   key->data=buff;
  693.   key->app_private= key_info;
  694.   for (; key_part != end && key_length > 0; key_part++)
  695.   {
  696.     if (key_part->null_bit)
  697.     {
  698.       /* Store 0 if the key part is a NULL part */
  699.       if (record[key_part->null_offset] & key_part->null_bit)
  700.       {
  701. *buff++ =0;
  702. key->flags|=DB_DBT_DUPOK;
  703. continue;
  704.       }
  705.       *buff++ = 1; // Store NOT NULL marker
  706.     }
  707.     buff=key_part->field->pack_key(buff,(char*) (record + key_part->offset),
  708.    key_part->length);
  709.     key_length-=key_part->length;
  710.   }
  711.   key->size= (buff  - (char*) key->data);
  712.   DBUG_DUMP("key",(char*) key->data, key->size);
  713.   DBUG_RETURN(key);
  714. }
  715. /*
  716.   Create a packed key from from a MySQL unpacked key
  717. */
  718. DBT *ha_berkeley::pack_key(DBT *key, uint keynr, char *buff,
  719.    const byte *key_ptr, uint key_length)
  720. {
  721.   KEY *key_info=table->key_info+keynr;
  722.   KEY_PART_INFO *key_part=key_info->key_part;
  723.   KEY_PART_INFO *end=key_part+key_info->key_parts;
  724.   DBUG_ENTER("bdb:pack_key");
  725.   bzero((char*) key,sizeof(*key));
  726.   key->data=buff;
  727.   key->app_private= (void*) key_info;
  728.   for (; key_part != end && (int) key_length > 0 ; key_part++)
  729.   {
  730.     uint offset=0;
  731.     if (key_part->null_bit)
  732.     {
  733.       if (!(*buff++ = (*key_ptr == 0))) // Store 0 if NULL
  734.       {
  735. key_length-= key_part->store_length;
  736. key_ptr+=   key_part->store_length;
  737. key->flags|=DB_DBT_DUPOK;
  738. continue;
  739.       }
  740.       offset=1; // Data is at key_ptr+1
  741.     }
  742.     buff=key_part->field->pack_key_from_key_image(buff,(char*) key_ptr+offset,
  743.   key_part->length);
  744.     key_ptr+=key_part->store_length;
  745.     key_length-=key_part->store_length;
  746.   }
  747.   key->size= (buff  - (char*) key->data);
  748.   DBUG_DUMP("key",(char*) key->data, key->size);
  749.   DBUG_RETURN(key);
  750. }
  751. int ha_berkeley::write_row(byte * record)
  752. {
  753.   DBT row,prim_key,key;
  754.   int error;
  755.   DBUG_ENTER("write_row");
  756.   statistic_increment(ha_write_count,&LOCK_status);
  757.   if (table->timestamp_field_type & TIMESTAMP_AUTO_SET_ON_INSERT)
  758.     table->timestamp_field->set_time();
  759.   if (table->next_number_field && record == table->record[0])
  760.     update_auto_increment();
  761.   if ((error=pack_row(&row, record,1)))
  762.     DBUG_RETURN(error); /* purecov: inspected */
  763.   if (table->keys + test(hidden_primary_key) == 1)
  764.   {
  765.     error=file->put(file, transaction, create_key(&prim_key, primary_key,
  766.   key_buff, record),
  767.     &row, key_type[primary_key]);
  768.     last_dup_key=primary_key;
  769.   }
  770.   else
  771.   {
  772.     DB_TXN *sub_trans = transaction;
  773.     /* Don't use sub transactions in temporary tables */
  774.     ulong thd_options = table->tmp_table == NO_TMP_TABLE ? table->in_use->options : 0;
  775.     for (uint retry=0 ; retry < berkeley_trans_retry ; retry++)
  776.     {
  777.       key_map changed_keys(0);
  778.       if (using_ignore && (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  779.       {
  780. if ((error=txn_begin(db_env, transaction, &sub_trans, 0))) /* purecov: deadcode */
  781.   break; /* purecov: deadcode */
  782. DBUG_PRINT("trans",("starting subtransaction")); /* purecov: deadcode */
  783.       }
  784.       if (!(error=file->put(file, sub_trans, create_key(&prim_key, primary_key,
  785. key_buff, record),
  786.     &row, key_type[primary_key])))
  787.       {
  788. changed_keys.set_bit(primary_key);
  789. for (uint keynr=0 ; keynr < table->keys ; keynr++)
  790. {
  791.   if (keynr == primary_key)
  792.     continue;
  793.   if ((error=key_file[keynr]->put(key_file[keynr], sub_trans,
  794.   create_key(&key, keynr, key_buff2,
  795.      record),
  796.   &prim_key, key_type[keynr])))
  797.   {
  798.     last_dup_key=keynr;
  799.     break;
  800.   }
  801.   changed_keys.set_bit(keynr);
  802. }
  803.       }
  804.       else
  805. last_dup_key=primary_key;
  806.       if (error)
  807.       {
  808. /* Remove inserted row */
  809. DBUG_PRINT("error",("Got error %d",error));
  810. if (using_ignore)
  811. {
  812.   int new_error = 0;
  813.   if (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS)
  814.   {
  815.     DBUG_PRINT("trans",("aborting subtransaction")); /* purecov: deadcode */
  816.     new_error=txn_abort(sub_trans); /* purecov: deadcode */
  817.   }
  818.   else if (!changed_keys.is_clear_all())
  819.   {
  820.     new_error = 0;
  821.     for (uint keynr=0 ; keynr < table->keys+test(hidden_primary_key) ;
  822.                  keynr++)
  823.     {
  824.       if (changed_keys.is_set(keynr))
  825.       {
  826. if ((new_error = remove_key(sub_trans, keynr, record,
  827.     &prim_key)))
  828.   break; /* purecov: inspected */
  829.       }
  830.     }
  831.   }
  832.   if (new_error)
  833.   {
  834.     error=new_error; // This shouldn't happen /* purecov: inspected */
  835.     break; /* purecov: inspected */
  836.   }
  837. }
  838.       }
  839.       else if (using_ignore && (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  840.       {
  841. DBUG_PRINT("trans",("committing subtransaction")); /* purecov: deadcode */
  842. error=txn_commit(sub_trans, 0); /* purecov: deadcode */
  843.       }
  844.       if (error != DB_LOCK_DEADLOCK)
  845. break;
  846.     }
  847.   }
  848.   if (error == DB_KEYEXIST)
  849.     error=HA_ERR_FOUND_DUPP_KEY;
  850.   else if (!error)
  851.     changed_rows++;
  852.   DBUG_RETURN(error);
  853. }
  854. /* Compare if a key in a row has changed */
  855. int ha_berkeley::key_cmp(uint keynr, const byte * old_row,
  856.  const byte * new_row)
  857. {
  858.   KEY_PART_INFO *key_part=table->key_info[keynr].key_part;
  859.   KEY_PART_INFO *end=key_part+table->key_info[keynr].key_parts;
  860.   for (; key_part != end ; key_part++)
  861.   {
  862.     if (key_part->null_bit)
  863.     {
  864.       if ((old_row[key_part->null_offset] & key_part->null_bit) !=
  865.   (new_row[key_part->null_offset] & key_part->null_bit))
  866. return 1;
  867.     }
  868.     if (key_part->key_part_flag & (HA_BLOB_PART | HA_VAR_LENGTH))
  869.     {
  870.       if (key_part->field->cmp_binary((char*) (old_row + key_part->offset),
  871.       (char*) (new_row + key_part->offset),
  872.       (ulong) key_part->length))
  873. return 1;
  874.     }
  875.     else
  876.     {
  877.       if (memcmp(old_row+key_part->offset, new_row+key_part->offset,
  878.  key_part->length))
  879. return 1;
  880.     }
  881.   }
  882.   return 0;
  883. }
  884. /*
  885.   Update a row from one value to another.
  886.   Clobbers key_buff2
  887. */
  888. int ha_berkeley::update_primary_key(DB_TXN *trans, bool primary_key_changed,
  889.     const byte * old_row, DBT *old_key,
  890.     const byte * new_row, DBT *new_key,
  891.     ulong thd_options, bool local_using_ignore)
  892. {
  893.   DBT row;
  894.   int error;
  895.   DBUG_ENTER("update_primary_key");
  896.   if (primary_key_changed)
  897.   {
  898.     // Primary key changed or we are updating a key that can have duplicates.
  899.     // Delete the old row and add a new one
  900.     if (!(error=remove_key(trans, primary_key, old_row, old_key)))
  901.     {
  902.       if (!(error=pack_row(&row, new_row, 0)))
  903.       {
  904. if ((error=file->put(file, trans, new_key, &row,
  905.      key_type[primary_key])))
  906. {
  907.   // Probably a duplicated key; restore old key and row if needed
  908.   last_dup_key=primary_key;
  909.   if (local_using_ignore &&
  910.       !(thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  911.   {
  912.     int new_error;
  913.     if ((new_error=pack_row(&row, old_row, 0)) ||
  914. (new_error=file->put(file, trans, old_key, &row,
  915.      key_type[primary_key])))
  916.       error=new_error;                  // fatal error /* purecov: inspected */
  917.   }
  918. }
  919.       }
  920.     }
  921.   }
  922.   else
  923.   {
  924.     // Primary key didn't change;  just update the row data
  925.     if (!(error=pack_row(&row, new_row, 0)))
  926.       error=file->put(file, trans, new_key, &row, 0);
  927.   }
  928.   DBUG_RETURN(error);
  929. }
  930. /*
  931.   Restore changed keys, when a non-fatal error aborts the insert/update
  932.   of one row.
  933.   Clobbers keybuff2
  934. */
  935. int ha_berkeley::restore_keys(DB_TXN *trans, key_map *changed_keys,
  936.       uint primary_key,
  937.       const byte *old_row, DBT *old_key,
  938.       const byte *new_row, DBT *new_key,
  939.       ulong thd_options)
  940. {
  941.   int error;
  942.   DBT tmp_key;
  943.   uint keynr;
  944.   DBUG_ENTER("restore_keys");
  945.   /* Restore the old primary key, and the old row, but don't ignore
  946.      duplicate key failure */
  947.   if ((error=update_primary_key(trans, TRUE, new_row, new_key,
  948. old_row, old_key, thd_options, FALSE)))
  949.     goto err; /* purecov: inspected */
  950.   /* Remove the new key, and put back the old key
  951.      changed_keys is a map of all non-primary keys that need to be
  952.      rolled back.  The last key set in changed_keys is the one that
  953.      triggered the duplicate key error (it wasn't inserted), so for
  954.      that one just put back the old value. */
  955.   if (!changed_keys->is_clear_all())
  956.   {
  957.     for (keynr=0 ; keynr < table->keys+test(hidden_primary_key) ; keynr++)
  958.     {
  959.       if (changed_keys->is_set(keynr))
  960.       {
  961.         if (changed_keys->is_prefix(1) &&
  962.             (error = remove_key(trans, keynr, new_row, new_key)))
  963.           break; /* purecov: inspected */
  964.         if ((error = key_file[keynr]->put(key_file[keynr], trans,
  965.                                           create_key(&tmp_key, keynr, key_buff2,
  966.                                                      old_row),
  967.                                           old_key, key_type[keynr])))
  968.           break; /* purecov: inspected */
  969.       }
  970.     }
  971.   }
  972. err:
  973.   DBUG_ASSERT(error != DB_KEYEXIST);
  974.   DBUG_RETURN(error);
  975. }
  976. int ha_berkeley::update_row(const byte * old_row, byte * new_row)
  977. {
  978.   DBT prim_key, key, old_prim_key;
  979.   int error;
  980.   DB_TXN *sub_trans;
  981.   ulong thd_options = table->tmp_table == NO_TMP_TABLE ? table->in_use->options : 0;
  982.   bool primary_key_changed;
  983.   DBUG_ENTER("update_row");
  984.   LINT_INIT(error);
  985.   statistic_increment(ha_update_count,&LOCK_status);
  986.   if (table->timestamp_field_type & TIMESTAMP_AUTO_SET_ON_UPDATE)
  987.     table->timestamp_field->set_time();
  988.   if (hidden_primary_key)
  989.   {
  990.     primary_key_changed=0;
  991.     bzero((char*) &prim_key,sizeof(prim_key));
  992.     prim_key.data= (void*) current_ident;
  993.     prim_key.size=BDB_HIDDEN_PRIMARY_KEY_LENGTH;
  994.     old_prim_key=prim_key;
  995.   }
  996.   else
  997.   {
  998.     create_key(&prim_key, primary_key, key_buff, new_row);
  999.     if ((primary_key_changed=key_cmp(primary_key, old_row, new_row)))
  1000.       create_key(&old_prim_key, primary_key, primary_key_buff, old_row);
  1001.     else
  1002.       old_prim_key=prim_key;
  1003.   }
  1004.   sub_trans = transaction;
  1005.   for (uint retry=0 ; retry < berkeley_trans_retry ; retry++)
  1006.   {
  1007.     key_map changed_keys(0);
  1008.     if (using_ignore && (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  1009.     {
  1010.       if ((error=txn_begin(db_env, transaction, &sub_trans, 0))) /* purecov: deadcode */
  1011. break; /* purecov: deadcode */
  1012.       DBUG_PRINT("trans",("starting subtransaction")); /* purecov: deadcode */
  1013.     }
  1014.     /* Start by updating the primary key */
  1015.     if (!(error=update_primary_key(sub_trans, primary_key_changed,
  1016.    old_row, &old_prim_key,
  1017.    new_row, &prim_key,
  1018.    thd_options, using_ignore)))
  1019.     {
  1020.       // Update all other keys
  1021.       for (uint keynr=0 ; keynr < table->keys ; keynr++)
  1022.       {
  1023. if (keynr == primary_key)
  1024.   continue;
  1025. if (key_cmp(keynr, old_row, new_row) || primary_key_changed)
  1026. {
  1027.   if ((error=remove_key(sub_trans, keynr, old_row, &old_prim_key)))
  1028.   {
  1029.     if (using_ignore && /* purecov: inspected */
  1030. (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  1031.             {
  1032.       int new_error;
  1033.       DBUG_PRINT("trans",("aborting subtransaction"));
  1034.       new_error=txn_abort(sub_trans);
  1035.       if (new_error)
  1036. error = new_error;
  1037.     }
  1038.     DBUG_RETURN(error); // Fatal error /* purecov: inspected */
  1039.   }
  1040.   changed_keys.set_bit(keynr);
  1041.   if ((error=key_file[keynr]->put(key_file[keynr], sub_trans,
  1042.   create_key(&key, keynr, key_buff2,
  1043.      new_row),
  1044.   &prim_key, key_type[keynr])))
  1045.   {
  1046.     last_dup_key=keynr;
  1047.     break;
  1048.   }
  1049. }
  1050.       }
  1051.     }
  1052.     if (error)
  1053.     {
  1054.       /* Remove inserted row */
  1055.       DBUG_PRINT("error",("Got error %d",error));
  1056.       if (using_ignore)
  1057.       {
  1058. int new_error = 0;
  1059. if (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS)
  1060. {
  1061.   DBUG_PRINT("trans",("aborting subtransaction")); /* purecov: deadcode */
  1062.   new_error=txn_abort(sub_trans); /* purecov: deadcode */
  1063. }
  1064. else if (!changed_keys.is_clear_all())
  1065.   new_error=restore_keys(transaction, &changed_keys, primary_key,
  1066.  old_row, &old_prim_key, new_row, &prim_key,
  1067.  thd_options);
  1068. if (new_error)
  1069. {
  1070.   error=new_error; // This shouldn't happen /* purecov: inspected */
  1071.   break; /* purecov: inspected */
  1072. }
  1073.       }
  1074.     }
  1075.     else if (using_ignore && (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  1076.     {
  1077.       DBUG_PRINT("trans",("committing subtransaction")); /* purecov: deadcode */
  1078.       error=txn_commit(sub_trans, 0); /* purecov: deadcode */
  1079.     }
  1080.     if (error != DB_LOCK_DEADLOCK)
  1081.       break;
  1082.   }
  1083.   if (error == DB_KEYEXIST)
  1084.     error=HA_ERR_FOUND_DUPP_KEY;
  1085.   DBUG_RETURN(error);
  1086. }
  1087. /*
  1088.   Delete one key
  1089.   This uses key_buff2, when keynr != primary key, so it's important that
  1090.   a function that calls this doesn't use this buffer for anything else.
  1091. */
  1092. int ha_berkeley::remove_key(DB_TXN *trans, uint keynr, const byte *record,
  1093.     DBT *prim_key)
  1094. {
  1095.   int error;
  1096.   DBT key;
  1097.   DBUG_ENTER("remove_key");
  1098.   DBUG_PRINT("enter",("index: %d",keynr));
  1099.   if (keynr == active_index && cursor)
  1100.     error=cursor->c_del(cursor,0);
  1101.   else if (keynr == primary_key ||
  1102.    ((table->key_info[keynr].flags & (HA_NOSAME | HA_NULL_PART_KEY)) ==
  1103.     HA_NOSAME))
  1104.   { // Unique key
  1105.     DBUG_ASSERT(keynr == primary_key || prim_key->data != key_buff2);
  1106.     error=key_file[keynr]->del(key_file[keynr], trans,
  1107.        keynr == primary_key ?
  1108.        prim_key :
  1109.        create_key(&key, keynr, key_buff2, record),
  1110.        0);
  1111.   }
  1112.   else
  1113.   {
  1114.     /*
  1115.       To delete the not duplicated key, we need to open an cursor on the
  1116.       row to find the key to be delete and delete it.
  1117.       We will never come here with keynr = primary_key
  1118.     */
  1119.     DBUG_ASSERT(keynr != primary_key && prim_key->data != key_buff2);
  1120.     DBC *tmp_cursor;
  1121.     if (!(error=key_file[keynr]->cursor(key_file[keynr], trans,
  1122. &tmp_cursor, 0)))
  1123.     {
  1124.       if (!(error=tmp_cursor->c_get(tmp_cursor,
  1125.                                     create_key(&key, keynr, key_buff2, record),
  1126.                                     prim_key, DB_GET_BOTH | DB_RMW)))
  1127.       { // This shouldn't happen
  1128. error=tmp_cursor->c_del(tmp_cursor,0);
  1129.       }
  1130.       int result=tmp_cursor->c_close(tmp_cursor);
  1131.       if (!error)
  1132. error=result;
  1133.     }
  1134.   }
  1135.   DBUG_RETURN(error);
  1136. }
  1137. /* Delete all keys for new_record */
  1138. int ha_berkeley::remove_keys(DB_TXN *trans, const byte *record,
  1139.      DBT *new_record, DBT *prim_key, key_map *keys)
  1140. {
  1141.   int result = 0;
  1142.   for (uint keynr=0 ; keynr < table->keys+test(hidden_primary_key) ; keynr++)
  1143.   {
  1144.     if (keys->is_set(keynr))
  1145.     {
  1146.       int new_error=remove_key(trans, keynr, record, prim_key);
  1147.       if (new_error)
  1148.       {
  1149. result=new_error; // Return last error /* purecov: inspected */
  1150. break; // Let rollback correct things /* purecov: inspected */
  1151.       }
  1152.     }
  1153.   }
  1154.   return result;
  1155. }
  1156. int ha_berkeley::delete_row(const byte * record)
  1157. {
  1158.   int error;
  1159.   DBT row, prim_key;
  1160.   key_map keys=table->keys_in_use;
  1161.   ulong thd_options = table->tmp_table == NO_TMP_TABLE ? table->in_use->options : 0;
  1162.   DBUG_ENTER("delete_row");
  1163.   statistic_increment(ha_delete_count,&LOCK_status);
  1164.   if ((error=pack_row(&row, record, 0)))
  1165.     DBUG_RETURN((error)); /* purecov: inspected */
  1166.   create_key(&prim_key, primary_key, key_buff, record);
  1167.   if (hidden_primary_key)
  1168.     keys.set_bit(primary_key);
  1169.   /* Subtransactions may be used in order to retry the delete in
  1170.      case we get a DB_LOCK_DEADLOCK error. */
  1171.   DB_TXN *sub_trans = transaction;
  1172.   for (uint retry=0 ; retry < berkeley_trans_retry ; retry++)
  1173.   {
  1174.     if (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS)
  1175.     {
  1176.       if ((error=txn_begin(db_env, transaction, &sub_trans, 0))) /* purecov: deadcode */
  1177. break; /* purecov: deadcode */
  1178.       DBUG_PRINT("trans",("starting sub transaction")); /* purecov: deadcode */
  1179.     }
  1180.     error=remove_keys(sub_trans, record, &row, &prim_key, &keys);
  1181.     if (!error && (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS))
  1182.     {
  1183.       DBUG_PRINT("trans",("ending sub transaction")); /* purecov: deadcode */
  1184.       error=txn_commit(sub_trans, 0); /* purecov: deadcode */
  1185.     }
  1186.     if (error)
  1187.     { /* purecov: inspected */
  1188.       DBUG_PRINT("error",("Got error %d",error));
  1189.       if (thd_options & OPTION_INTERNAL_SUBTRANSACTIONS)
  1190.       {
  1191. /* retry */
  1192. int new_error;
  1193. DBUG_PRINT("trans",("aborting subtransaction"));
  1194. if ((new_error=txn_abort(sub_trans)))
  1195. {
  1196.   error=new_error; // This shouldn't happen
  1197.   break;
  1198. }
  1199.       }
  1200.       else
  1201. break; // No retry - return error
  1202.     }
  1203.     if (error != DB_LOCK_DEADLOCK)
  1204.       break;
  1205.   }
  1206. #ifdef CANT_COUNT_DELETED_ROWS
  1207.   if (!error)
  1208.     changed_rows--;
  1209. #endif
  1210.   DBUG_RETURN(error);
  1211. }
  1212. int ha_berkeley::index_init(uint keynr)
  1213. {
  1214.   int error;
  1215.   DBUG_ENTER("ha_berkeley::index_init");
  1216.   DBUG_PRINT("enter",("table: '%s'  key: %d", table->real_name, keynr));
  1217.   /*
  1218.     Under some very rare conditions (like full joins) we may already have
  1219.     an active cursor at this point
  1220.   */
  1221.   if (cursor)
  1222.   {
  1223.     DBUG_PRINT("note",("Closing active cursor"));
  1224.     cursor->c_close(cursor);
  1225.   }
  1226.   active_index=keynr;
  1227.   if ((error=key_file[keynr]->cursor(key_file[keynr], transaction, &cursor,
  1228.      table->reginfo.lock_type >
  1229.      TL_WRITE_ALLOW_READ ?
  1230.      0 : 0)))
  1231.     cursor=0; // Safety /* purecov: inspected */
  1232.   bzero((char*) &last_key,sizeof(last_key));
  1233.   DBUG_RETURN(error);
  1234. }
  1235. int ha_berkeley::index_end()
  1236. {
  1237.   int error=0;
  1238.   DBUG_ENTER("ha_berkely::index_end");
  1239.   if (cursor)
  1240.   {
  1241.     DBUG_PRINT("enter",("table: '%s'", table->real_name));
  1242.     error=cursor->c_close(cursor);
  1243.     cursor=0;
  1244.   }
  1245.   active_index=MAX_KEY;
  1246.   DBUG_RETURN(error);
  1247. }
  1248. /* What to do after we have read a row based on an index */
  1249. int ha_berkeley::read_row(int error, char *buf, uint keynr, DBT *row,
  1250.   DBT *found_key, bool read_next)
  1251. {
  1252.   DBUG_ENTER("ha_berkeley::read_row");
  1253.   if (error)
  1254.   {
  1255.     if (error == DB_NOTFOUND || error == DB_KEYEMPTY)
  1256.       error=read_next ? HA_ERR_END_OF_FILE : HA_ERR_KEY_NOT_FOUND;
  1257.     table->status=STATUS_NOT_FOUND;
  1258.     DBUG_RETURN(error);
  1259.   }
  1260.   if (hidden_primary_key)
  1261.     memcpy_fixed(current_ident,
  1262.  (char*) row->data+row->size-BDB_HIDDEN_PRIMARY_KEY_LENGTH,
  1263.  BDB_HIDDEN_PRIMARY_KEY_LENGTH);
  1264.   table->status=0;
  1265.   if (keynr != primary_key)
  1266.   {
  1267.     /* We only found the primary key.  Now we have to use this to find
  1268.        the row data */
  1269.     if (key_read && found_key)
  1270.     {
  1271.       unpack_key(buf,found_key,keynr);
  1272.       if (!hidden_primary_key)
  1273. unpack_key(buf,row,primary_key);
  1274.       DBUG_RETURN(0);
  1275.     }
  1276.     DBT key;
  1277.     bzero((char*) &key,sizeof(key));
  1278.     key.data=key_buff;
  1279.     key.size=row->size;
  1280.     key.app_private= (void*) (table->key_info+primary_key);
  1281.     memcpy(key_buff,row->data,row->size);
  1282.     /* Read the data into current_row */
  1283.     current_row.flags=DB_DBT_REALLOC;
  1284.     if ((error=file->get(file, transaction, &key, &current_row, 0)))
  1285.     {
  1286.       table->status=STATUS_NOT_FOUND; /* purecov: inspected */
  1287.       DBUG_RETURN(error == DB_NOTFOUND ? HA_ERR_CRASHED : error); /* purecov: inspected */
  1288.     }
  1289.     row= &current_row;
  1290.   }
  1291.   unpack_row(buf,row);
  1292.   DBUG_RETURN(0);
  1293. }
  1294. /* This is only used to read whole keys */
  1295. int ha_berkeley::index_read_idx(byte * buf, uint keynr, const byte * key,
  1296. uint key_len, enum ha_rkey_function find_flag)
  1297. {
  1298.   statistic_increment(ha_read_key_count,&LOCK_status);
  1299.   DBUG_ENTER("index_read_idx");
  1300.   current_row.flags=DB_DBT_REALLOC;
  1301.   active_index=MAX_KEY;
  1302.   DBUG_RETURN(read_row(key_file[keynr]->get(key_file[keynr], transaction,
  1303.  pack_key(&last_key, keynr, key_buff, key,
  1304.   key_len),
  1305.  &current_row,0),
  1306.        (char*) buf, keynr, &current_row, &last_key, 0));
  1307. }
  1308. int ha_berkeley::index_read(byte * buf, const byte * key,
  1309.     uint key_len, enum ha_rkey_function find_flag)
  1310. {
  1311.   DBT row;
  1312.   int error;
  1313.   KEY *key_info= &table->key_info[active_index];
  1314.   int do_prev= 0;
  1315.   DBUG_ENTER("ha_berkeley::index_read");
  1316.   statistic_increment(ha_read_key_count,&LOCK_status);
  1317.   bzero((char*) &row,sizeof(row));
  1318.   if (find_flag == HA_READ_BEFORE_KEY)
  1319.   {
  1320.     find_flag= HA_READ_KEY_OR_NEXT;
  1321.     do_prev= 1;
  1322.   }
  1323.   else if (find_flag == HA_READ_PREFIX_LAST_OR_PREV)
  1324.   {
  1325.     find_flag= HA_READ_AFTER_KEY;
  1326.     do_prev= 1;
  1327.   }
  1328.   if (key_len == key_info->key_length)
  1329.   {
  1330.     if (find_flag == HA_READ_AFTER_KEY)
  1331.       key_info->handler.bdb_return_if_eq= 1;
  1332.     error=read_row(cursor->c_get(cursor, pack_key(&last_key,
  1333.   active_index,
  1334.   key_buff,
  1335.   key, key_len),
  1336.  &row,
  1337.  (find_flag == HA_READ_KEY_EXACT ?
  1338.   DB_SET : DB_SET_RANGE)),
  1339.    (char*) buf, active_index, &row, (DBT*) 0, 0);
  1340.     key_info->handler.bdb_return_if_eq= 0;
  1341.   }
  1342.   else
  1343.   {
  1344.     /* read of partial key */
  1345.     pack_key(&last_key, active_index, key_buff, key, key_len);
  1346.     /* Store for compare */
  1347.     memcpy(key_buff2, key_buff, (key_len=last_key.size));
  1348.     /*
  1349.       If HA_READ_AFTER_KEY is set, return next key, else return first
  1350.       matching key.
  1351.     */
  1352.     key_info->handler.bdb_return_if_eq= (find_flag == HA_READ_AFTER_KEY ?
  1353.  1 : -1);
  1354.     error=read_row(cursor->c_get(cursor, &last_key, &row, DB_SET_RANGE),
  1355.    (char*) buf, active_index, &row, (DBT*) 0, 0);
  1356.     key_info->handler.bdb_return_if_eq= 0;
  1357.     if (!error && find_flag == HA_READ_KEY_EXACT)
  1358.     {
  1359.       /* Ensure that we found a key that is equal to the current one */
  1360.       if (!error && berkeley_key_cmp(table, key_info, key_buff2, key_len))
  1361. error=HA_ERR_KEY_NOT_FOUND;
  1362.     }
  1363.   }
  1364.   if (do_prev)
  1365.   {
  1366.     bzero((char*) &row, sizeof(row));
  1367.     error= read_row(cursor->c_get(cursor, &last_key, &row, DB_PREV),
  1368.                          (char*) buf, active_index, &row, &last_key, 1);
  1369.   }
  1370.   DBUG_RETURN(error);
  1371. }
  1372. /*
  1373.   Read last key is solved by reading the next key and then reading
  1374.   the previous key
  1375. */
  1376. int ha_berkeley::index_read_last(byte * buf, const byte * key, uint key_len)
  1377. {
  1378.   DBT row;
  1379.   int error;
  1380.   KEY *key_info= &table->key_info[active_index];
  1381.   DBUG_ENTER("ha_berkeley::index_read");
  1382.   statistic_increment(ha_read_key_count,&LOCK_status);
  1383.   bzero((char*) &row,sizeof(row));
  1384.   /* read of partial key */
  1385.   pack_key(&last_key, active_index, key_buff, key, key_len);
  1386.   /* Store for compare */
  1387.   memcpy(key_buff2, key_buff, (key_len=last_key.size));
  1388.   key_info->handler.bdb_return_if_eq= 1;
  1389.   error=read_row(cursor->c_get(cursor, &last_key, &row, DB_SET_RANGE),
  1390.  (char*) buf, active_index, &row, (DBT*) 0, 0);
  1391.   key_info->handler.bdb_return_if_eq= 0;
  1392.   bzero((char*) &row,sizeof(row));
  1393.   if (read_row(cursor->c_get(cursor, &last_key, &row, DB_PREV),
  1394.        (char*) buf, active_index, &row, &last_key, 1) ||
  1395.       berkeley_key_cmp(table, key_info, key_buff2, key_len))
  1396.     error=HA_ERR_KEY_NOT_FOUND;
  1397.   DBUG_RETURN(error);
  1398. }
  1399. int ha_berkeley::index_next(byte * buf)
  1400. {
  1401.   DBT row;
  1402.   DBUG_ENTER("index_next");
  1403.   statistic_increment(ha_read_next_count,&LOCK_status);
  1404.   bzero((char*) &row,sizeof(row));
  1405.   DBUG_RETURN(read_row(cursor->c_get(cursor, &last_key, &row, DB_NEXT),
  1406.        (char*) buf, active_index, &row, &last_key, 1));
  1407. }
  1408. int ha_berkeley::index_next_same(byte * buf, const byte *key, uint keylen)
  1409. {
  1410.   DBT row;
  1411.   int error;
  1412.   DBUG_ENTER("index_next_same");
  1413.   statistic_increment(ha_read_next_count,&LOCK_status);
  1414.   bzero((char*) &row,sizeof(row));
  1415.   if (keylen == table->key_info[active_index].key_length)
  1416.     error=read_row(cursor->c_get(cursor, &last_key, &row, DB_NEXT_DUP),
  1417.    (char*) buf, active_index, &row, &last_key, 1);
  1418.   else
  1419.   {
  1420.     error=read_row(cursor->c_get(cursor, &last_key, &row, DB_NEXT),
  1421.    (char*) buf, active_index, &row, &last_key, 1);
  1422.     if (!error && ::key_cmp_if_same(table, key, active_index, keylen))
  1423.       error=HA_ERR_END_OF_FILE;
  1424.   }
  1425.   DBUG_RETURN(error);
  1426. }
  1427. int ha_berkeley::index_prev(byte * buf)
  1428. {
  1429.   DBT row;
  1430.   DBUG_ENTER("index_prev");
  1431.   statistic_increment(ha_read_prev_count,&LOCK_status);
  1432.   bzero((char*) &row,sizeof(row));
  1433.   DBUG_RETURN(read_row(cursor->c_get(cursor, &last_key, &row, DB_PREV),
  1434.        (char*) buf, active_index, &row, &last_key, 1));
  1435. }
  1436. int ha_berkeley::index_first(byte * buf)
  1437. {
  1438.   DBT row;
  1439.   DBUG_ENTER("index_first");
  1440.   statistic_increment(ha_read_first_count,&LOCK_status);
  1441.   bzero((char*) &row,sizeof(row));
  1442.   DBUG_RETURN(read_row(cursor->c_get(cursor, &last_key, &row, DB_FIRST),
  1443.        (char*) buf, active_index, &row, &last_key, 1));
  1444. }
  1445. int ha_berkeley::index_last(byte * buf)
  1446. {
  1447.   DBT row;
  1448.   DBUG_ENTER("index_last");
  1449.   statistic_increment(ha_read_last_count,&LOCK_status);
  1450.   bzero((char*) &row,sizeof(row));
  1451.   DBUG_RETURN(read_row(cursor->c_get(cursor, &last_key, &row, DB_LAST),
  1452.        (char*) buf, active_index, &row, &last_key, 0));
  1453. }
  1454. int ha_berkeley::rnd_init(bool scan)
  1455. {
  1456.   DBUG_ENTER("rnd_init");
  1457.   current_row.flags=DB_DBT_REALLOC;
  1458.   DBUG_RETURN(index_init(primary_key));
  1459. }
  1460. int ha_berkeley::rnd_end()
  1461. {
  1462.   return index_end();
  1463. }
  1464. int ha_berkeley::rnd_next(byte *buf)
  1465. {
  1466.   DBT row;
  1467.   DBUG_ENTER("rnd_next");
  1468.   statistic_increment(ha_read_rnd_next_count,&LOCK_status);
  1469.   bzero((char*) &row,sizeof(row));
  1470.   DBUG_RETURN(read_row(cursor->c_get(cursor, &last_key, &row, DB_NEXT),
  1471.        (char*) buf, primary_key, &row, &last_key, 1));
  1472. }
  1473. DBT *ha_berkeley::get_pos(DBT *to, byte *pos)
  1474. {
  1475.   /* We don't need to set app_private here */
  1476.   bzero((char*) to,sizeof(*to));
  1477.   to->data=pos;
  1478.   if (share->fixed_length_primary_key)
  1479.     to->size=ref_length;
  1480.   else
  1481.   {
  1482.     KEY_PART_INFO *key_part=table->key_info[primary_key].key_part;
  1483.     KEY_PART_INFO *end=key_part+table->key_info[primary_key].key_parts;
  1484.     for (; key_part != end ; key_part++)
  1485.       pos+=key_part->field->packed_col_length((char*) pos,key_part->length);
  1486.     to->size= (uint) (pos- (byte*) to->data);
  1487.   }
  1488.   return to;
  1489. }
  1490. int ha_berkeley::rnd_pos(byte * buf, byte *pos)
  1491. {
  1492.   DBT db_pos;
  1493.   statistic_increment(ha_read_rnd_count,&LOCK_status);
  1494.   DBUG_ENTER("ha_berkeley::rnd_pos");
  1495.   active_index= MAX_KEY;
  1496.   DBUG_RETURN(read_row(file->get(file, transaction,
  1497.  get_pos(&db_pos, pos),
  1498.  &current_row, 0),
  1499.        (char*) buf, primary_key, &current_row, (DBT*) 0, 0));
  1500. }
  1501. /*
  1502.   Set a reference to the current record in (ref,ref_length).
  1503.   SYNOPSIS
  1504.     ha_berkeley::position()
  1505.     record                      The current record buffer
  1506.   DESCRIPTION
  1507.     The BDB handler stores the primary key in (ref,ref_length).
  1508.     There is either an explicit primary key, or an implicit (hidden)
  1509.     primary key.
  1510.     During open(), 'ref_length' is calculated as the maximum primary
  1511.     key length. When an actual key is shorter than that, the rest of
  1512.     the buffer must be cleared out. The row cannot be identified, if
  1513.     garbage follows behind the end of the key. There is no length
  1514.     field for the current key, so that the whole ref_length is used
  1515.     for comparison.
  1516.   RETURN
  1517.     nothing
  1518. */
  1519. void ha_berkeley::position(const byte *record)
  1520. {
  1521.   DBT key;
  1522.   DBUG_ENTER("ha_berkeley::position");
  1523.   if (hidden_primary_key)
  1524.   {
  1525.     DBUG_ASSERT(ref_length == BDB_HIDDEN_PRIMARY_KEY_LENGTH);
  1526.     memcpy_fixed(ref, (char*) current_ident, BDB_HIDDEN_PRIMARY_KEY_LENGTH);
  1527.   }
  1528.   else
  1529.   {
  1530.     create_key(&key, primary_key, (char*) ref, record);
  1531.     if (key.size < ref_length)
  1532.       bzero(ref + key.size, ref_length - key.size);
  1533.   }
  1534.   DBUG_VOID_RETURN;
  1535. }
  1536. void ha_berkeley::info(uint flag)
  1537. {
  1538.   DBUG_ENTER("ha_berkeley::info");
  1539.   if (flag & HA_STATUS_VARIABLE)
  1540.   {
  1541.     records = share->rows + changed_rows; // Just to get optimisations right
  1542.     deleted = 0;
  1543.   }
  1544.   if ((flag & HA_STATUS_CONST) || version != share->version)
  1545.   {
  1546.     version=share->version;
  1547.     for (uint i=0 ; i < table->keys ; i++)
  1548.     {
  1549.       table->key_info[i].rec_per_key[table->key_info[i].key_parts-1]=
  1550. share->rec_per_key[i];
  1551.     }
  1552.   }
  1553.   /* Don't return key if we got an error for the internal primary key */
  1554.   if (flag & HA_STATUS_ERRKEY && last_dup_key < table->keys)
  1555.     errkey= last_dup_key;
  1556.   DBUG_VOID_RETURN;
  1557. }
  1558. int ha_berkeley::extra(enum ha_extra_function operation)
  1559. {
  1560.   switch (operation) {
  1561.   case HA_EXTRA_RESET:
  1562.   case HA_EXTRA_RESET_STATE:
  1563.     key_read=0;
  1564.     using_ignore=0;
  1565.     if (current_row.flags & (DB_DBT_MALLOC | DB_DBT_REALLOC))
  1566.     {
  1567.       current_row.flags=0;
  1568.       if (current_row.data)
  1569.       {
  1570. free(current_row.data);
  1571. current_row.data=0;
  1572.       }
  1573.     }
  1574.     break;
  1575.   case HA_EXTRA_KEYREAD:
  1576.     key_read=1; // Query satisfied with key
  1577.     break;
  1578.   case HA_EXTRA_NO_KEYREAD:
  1579.     key_read=0;
  1580.     break;
  1581.   case HA_EXTRA_IGNORE_DUP_KEY:
  1582.     using_ignore=1;
  1583.     break;
  1584.   case HA_EXTRA_NO_IGNORE_DUP_KEY:
  1585.     using_ignore=0;
  1586.     break;
  1587.   default:
  1588.     break;
  1589.   }
  1590.   return 0;
  1591. }
  1592. int ha_berkeley::reset(void)
  1593. {
  1594.   ha_berkeley::extra(HA_EXTRA_RESET);
  1595.   key_read=0; // Reset to state after open
  1596.   return 0;
  1597. }
  1598. /*
  1599.   As MySQL will execute an external lock for every new table it uses
  1600.   we can use this to start the transactions.
  1601.   If we are in auto_commit mode we just need to start a transaction
  1602.   for the statement to be able to rollback the statement.
  1603.   If not, we have to start a master transaction if there doesn't exist
  1604.   one from before.
  1605. */
  1606. int ha_berkeley::external_lock(THD *thd, int lock_type)
  1607. {
  1608.   int error=0;
  1609.   DBUG_ENTER("ha_berkeley::external_lock");
  1610.   if (lock_type != F_UNLCK)
  1611.   {
  1612.     if (!thd->transaction.bdb_lock_count++)
  1613.     {
  1614.       DBUG_ASSERT(thd->transaction.stmt.bdb_tid == 0);
  1615.       transaction=0; // Safety
  1616.       /* First table lock, start transaction */
  1617.       if ((thd->options & (OPTION_NOT_AUTOCOMMIT | OPTION_BEGIN |
  1618.    OPTION_TABLE_LOCK)) &&
  1619.   !thd->transaction.all.bdb_tid)
  1620.       {
  1621. /* We have to start a master transaction */
  1622. DBUG_PRINT("trans",("starting transaction all"));
  1623. if ((error=txn_begin(db_env, 0,
  1624.      (DB_TXN**) &thd->transaction.all.bdb_tid,
  1625.      0)))
  1626. {
  1627.   thd->transaction.bdb_lock_count--; // We didn't get the lock /* purecov: inspected */
  1628.   DBUG_RETURN(error); /* purecov: inspected */
  1629. }
  1630. if (thd->in_lock_tables)
  1631.   DBUG_RETURN(0); // Don't create stmt trans
  1632.       }
  1633.       DBUG_PRINT("trans",("starting transaction stmt"));
  1634.       if ((error=txn_begin(db_env,
  1635.    (DB_TXN*) thd->transaction.all.bdb_tid,
  1636.    (DB_TXN**) &thd->transaction.stmt.bdb_tid,
  1637.    0)))
  1638.       {
  1639. /* We leave the possible master transaction open */
  1640. thd->transaction.bdb_lock_count--; // We didn't get the lock /* purecov: inspected */
  1641. DBUG_RETURN(error); /* purecov: inspected */
  1642.       }
  1643.     }
  1644.     transaction= (DB_TXN*) thd->transaction.stmt.bdb_tid;
  1645.   }
  1646.   else
  1647.   {
  1648.     lock.type=TL_UNLOCK; // Unlocked
  1649.     thread_safe_add(share->rows, changed_rows, &share->mutex);
  1650.     changed_rows=0;
  1651.     if (!--thd->transaction.bdb_lock_count)
  1652.     {
  1653.       if (thd->transaction.stmt.bdb_tid)
  1654.       {
  1655. /*
  1656.    F_UNLOCK is done without a transaction commit / rollback.
  1657.    This happens if the thread didn't update any rows
  1658.    We must in this case commit the work to keep the row locks
  1659. */
  1660. DBUG_PRINT("trans",("commiting non-updating transaction"));
  1661. error=txn_commit((DB_TXN*) thd->transaction.stmt.bdb_tid,0);
  1662. thd->transaction.stmt.bdb_tid=0;
  1663. transaction=0;
  1664.       }
  1665.     }
  1666.   }
  1667.   DBUG_RETURN(error);
  1668. }
  1669. /*
  1670.   When using LOCK TABLE's external_lock is only called when the actual
  1671.   TABLE LOCK is done.
  1672.   Under LOCK TABLES, each used tables will force a call to start_stmt.
  1673. */
  1674. int ha_berkeley::start_stmt(THD *thd)
  1675. {
  1676.   int error=0;
  1677.   DBUG_ENTER("ha_berkeley::start_stmt");
  1678.   if (!thd->transaction.stmt.bdb_tid)
  1679.   {
  1680.     DBUG_PRINT("trans",("starting transaction stmt"));
  1681.     error=txn_begin(db_env, (DB_TXN*) thd->transaction.all.bdb_tid,
  1682.     (DB_TXN**) &thd->transaction.stmt.bdb_tid,
  1683.     0);
  1684.   }
  1685.   transaction= (DB_TXN*) thd->transaction.stmt.bdb_tid;
  1686.   DBUG_RETURN(error);
  1687. }
  1688. /*
  1689.   The idea with handler::store_lock() is the following:
  1690.   The statement decided which locks we should need for the table
  1691.   for updates/deletes/inserts we get WRITE locks, for SELECT... we get
  1692.   read locks.
  1693.   Before adding the lock into the table lock handler (see thr_lock.c)
  1694.   mysqld calls store lock with the requested locks.  Store lock can now
  1695.   modify a write lock to a read lock (or some other lock), ignore the
  1696.   lock (if we don't want to use MySQL table locks at all) or add locks
  1697.   for many tables (like we do when we are using a MERGE handler).
  1698.   Berkeley DB changes all WRITE locks to TL_WRITE_ALLOW_WRITE (which
  1699.   signals that we are doing WRITES, but we are still allowing other
  1700.   reader's and writer's.
  1701.   When releasing locks, store_lock() are also called. In this case one
  1702.   usually doesn't have to do anything.
  1703.   In some exceptional cases MySQL may send a request for a TL_IGNORE;
  1704.   This means that we are requesting the same lock as last time and this
  1705.   should also be ignored. (This may happen when someone does a flush
  1706.   table when we have opened a part of the tables, in which case mysqld
  1707.   closes and reopens the tables and tries to get the same locks at last
  1708.   time).  In the future we will probably try to remove this.
  1709. */
  1710. THR_LOCK_DATA **ha_berkeley::store_lock(THD *thd, THR_LOCK_DATA **to,
  1711. enum thr_lock_type lock_type)
  1712. {
  1713.   if (lock_type != TL_IGNORE && lock.type == TL_UNLOCK)
  1714.   {
  1715.     /* If we are not doing a LOCK TABLE, then allow multiple writers */
  1716.     if ((lock_type >= TL_WRITE_CONCURRENT_INSERT &&
  1717.  lock_type <= TL_WRITE) &&
  1718. !thd->in_lock_tables)
  1719.       lock_type = TL_WRITE_ALLOW_WRITE;
  1720.     lock.type=lock_type;
  1721.     lock_on_read= ((table->reginfo.lock_type > TL_WRITE_ALLOW_READ) ? DB_RMW :
  1722.    0);
  1723.   }
  1724.   *to++= &lock;
  1725.   return to;
  1726. }
  1727. static int create_sub_table(const char *table_name, const char *sub_name,
  1728.     DBTYPE type, int flags)
  1729. {
  1730.   int error;
  1731.   DB *file;
  1732.   DBUG_ENTER("create_sub_table");
  1733.   DBUG_PRINT("enter",("sub_name: %s  flags: %d",sub_name, flags));
  1734.   if (!(error=db_create(&file, db_env, 0)))
  1735.   {
  1736.     file->set_flags(file, flags);
  1737.     error=(file->open(file, NULL, table_name, sub_name, type,
  1738.       DB_THREAD | DB_CREATE, my_umask));
  1739.     if (error)
  1740.     {
  1741.       DBUG_PRINT("error",("Got error: %d when opening table '%s'",error, /* purecov: inspected */
  1742.   table_name)); /* purecov: inspected */
  1743.       (void) file->remove(file,table_name,NULL,0); /* purecov: inspected */
  1744.     }
  1745.     else
  1746.       (void) file->close(file,0);
  1747.   }
  1748.   else
  1749.   {
  1750.     DBUG_PRINT("error",("Got error: %d when creting table",error)); /* purecov: inspected */
  1751.   }
  1752.   if (error)
  1753.     my_errno=error; /* purecov: inspected */
  1754.   DBUG_RETURN(error);
  1755. }
  1756. int ha_berkeley::create(const char *name, register TABLE *form,
  1757. HA_CREATE_INFO *create_info)
  1758. {
  1759.   char name_buff[FN_REFLEN];
  1760.   char part[7];
  1761.   uint index=1;
  1762.   int error;
  1763.   DBUG_ENTER("ha_berkeley::create");
  1764.   fn_format(name_buff,name,"", ha_berkeley_ext,2 | 4);
  1765.   /* Create the main table that will hold the real rows */
  1766.   if ((error= create_sub_table(name_buff,"main",DB_BTREE,0)))
  1767.     DBUG_RETURN(error); /* purecov: inspected */
  1768.   primary_key=table->primary_key;
  1769.   /* Create the keys */
  1770.   for (uint i=0; i < form->keys; i++)
  1771.   {
  1772.     if (i != primary_key)
  1773.     {
  1774.       sprintf(part,"key%02d",index++);
  1775.       if ((error= create_sub_table(name_buff, part, DB_BTREE,
  1776.    (table->key_info[i].flags & HA_NOSAME) ? 0 :
  1777.    DB_DUP)))
  1778. DBUG_RETURN(error); /* purecov: inspected */
  1779.     }
  1780.   }
  1781.   /* Create the status block to save information from last status command */
  1782.   /* Is DB_BTREE the best option here ? (QUEUE can't be used in sub tables) */
  1783.   DB *status_block;
  1784.   if (!(error=(db_create(&status_block, db_env, 0))))
  1785.   {
  1786.     if (!(error=(status_block->open(status_block, NULL, name_buff,
  1787.     "status", DB_BTREE, DB_CREATE, 0))))
  1788.     {
  1789.       char rec_buff[4+MAX_KEY*4];
  1790.       uint length= 4+ table->keys*4;
  1791.       bzero(rec_buff, length);
  1792.       error= write_status(status_block, rec_buff, length);
  1793.       status_block->close(status_block,0);
  1794.     }
  1795.   }
  1796.   DBUG_RETURN(error);
  1797. }
  1798. int ha_berkeley::delete_table(const char *name)
  1799. {
  1800.   int error;
  1801.   char name_buff[FN_REFLEN];
  1802.   DBUG_ENTER("delete_table");
  1803.   if ((error=db_create(&file, db_env, 0)))
  1804.     my_errno=error; /* purecov: inspected */
  1805.   else
  1806.     error=file->remove(file,fn_format(name_buff,name,"",ha_berkeley_ext,2 | 4),
  1807.        NULL,0);
  1808.   file=0; // Safety
  1809.   DBUG_RETURN(error);
  1810. }
  1811. int ha_berkeley::rename_table(const char * from, const char * to)
  1812. {
  1813.   int error;
  1814.   char from_buff[FN_REFLEN];
  1815.   char to_buff[FN_REFLEN];
  1816.   if ((error= db_create(&file, db_env, 0)))
  1817.     my_errno= error;
  1818.   else
  1819.   {
  1820.     /* On should not do a file->close() after rename returns */
  1821.     error= file->rename(file, 
  1822. fn_format(from_buff, from, "", ha_berkeley_ext, 2 | 4),
  1823. NULL, fn_format(to_buff, to, "", ha_berkeley_ext,
  1824. 2 | 4), 0);
  1825.   }
  1826.   return error;
  1827. }
  1828. /*
  1829.   How many seeks it will take to read through the table
  1830.   This is to be comparable to the number returned by records_in_range so
  1831.   that we can decide if we should scan the table or use keys.
  1832. */
  1833. double ha_berkeley::scan_time()
  1834. {
  1835.   return rows2double(records/3);
  1836. }
  1837. ha_rows ha_berkeley::records_in_range(uint keynr, key_range *start_key,
  1838.                                       key_range *end_key)
  1839. {
  1840.   DBT key;
  1841.   DB_KEY_RANGE start_range, end_range;
  1842.   DB *kfile=key_file[keynr];
  1843.   double start_pos,end_pos,rows;
  1844.   DBUG_ENTER("records_in_range");
  1845.   if ((start_key && kfile->key_range(kfile,transaction,
  1846.                                      pack_key(&key, keynr, key_buff,
  1847.                                               start_key->key,
  1848.                                               start_key->length),
  1849.                                      &start_range,0)) ||
  1850.       (end_key && kfile->key_range(kfile,transaction,
  1851.    pack_key(&key, keynr, key_buff,
  1852.                                             end_key->key,
  1853.                                             end_key->length),
  1854.    &end_range,0)))
  1855.     DBUG_RETURN(HA_BERKELEY_RANGE_COUNT); // Better than returning an error /* purecov: inspected */
  1856.   if (!start_key)
  1857.     start_pos= 0.0;
  1858.   else if (start_key->flag == HA_READ_KEY_EXACT)
  1859.     start_pos=start_range.less;
  1860.   else
  1861.     start_pos=start_range.less+start_range.equal;
  1862.   if (!end_key)
  1863.     end_pos= 1.0;
  1864.   else if (end_key->flag == HA_READ_BEFORE_KEY)
  1865.     end_pos=end_range.less;
  1866.   else
  1867.     end_pos=end_range.less+end_range.equal;
  1868.   rows=(end_pos-start_pos)*records;
  1869.   DBUG_PRINT("exit",("rows: %g",rows));
  1870.   DBUG_RETURN(rows <= 1.0 ? (ha_rows) 1 : (ha_rows) rows);
  1871. }
  1872. longlong ha_berkeley::get_auto_increment()
  1873. {
  1874.   longlong nr=1; // Default if error or new key
  1875.   int error;
  1876.   (void) ha_berkeley::extra(HA_EXTRA_KEYREAD);
  1877.   /* Set 'active_index' */
  1878.   ha_berkeley::index_init(table->next_number_index);
  1879.   if (!table->next_number_key_offset)
  1880.   { // Autoincrement at key-start
  1881.     error=ha_berkeley::index_last(table->record[1]);
  1882.   }
  1883.   else
  1884.   {
  1885.     DBT row,old_key;
  1886.     bzero((char*) &row,sizeof(row));
  1887.     KEY *key_info= &table->key_info[active_index];
  1888.     /* Reading next available number for a sub key */
  1889.     ha_berkeley::create_key(&last_key, active_index,
  1890.     key_buff, table->record[0],
  1891.     table->next_number_key_offset);
  1892.     /* Store for compare */
  1893.     memcpy(old_key.data=key_buff2, key_buff, (old_key.size=last_key.size));
  1894.     old_key.app_private=(void*) key_info;
  1895.     error=1;
  1896.     {
  1897.       /* Modify the compare so that we will find the next key */
  1898.       key_info->handler.bdb_return_if_eq= 1;
  1899.       /* We lock the next key as the new key will probl. be on the same page */
  1900.       error=cursor->c_get(cursor, &last_key, &row, DB_SET_RANGE | DB_RMW);
  1901.       key_info->handler.bdb_return_if_eq= 0;
  1902.       if (!error || error == DB_NOTFOUND)
  1903.       {
  1904. /*
  1905.   Now search go one step back and then we should have found the
  1906.   biggest key with the given prefix
  1907.   */
  1908. error=1;
  1909. if (!cursor->c_get(cursor, &last_key, &row, DB_PREV | DB_RMW) &&
  1910.     !berkeley_cmp_packed_key(key_file[active_index], &old_key,
  1911.      &last_key))
  1912. {
  1913.   error=0; // Found value
  1914.   unpack_key((char*) table->record[1], &last_key, active_index);
  1915. }
  1916.       }
  1917.     }
  1918.   }
  1919.   if (!error)
  1920.     nr=(longlong)
  1921.       table->next_number_field->val_int_offset(table->rec_buff_length)+1;
  1922.   ha_berkeley::index_end();
  1923.   (void) ha_berkeley::extra(HA_EXTRA_NO_KEYREAD);
  1924.   return nr;
  1925. }
  1926. void ha_berkeley::print_error(int error, myf errflag)
  1927. {
  1928.   if (error == DB_LOCK_DEADLOCK)
  1929.     error=HA_ERR_LOCK_DEADLOCK;
  1930.   handler::print_error(error,errflag);
  1931. }
  1932. /****************************************************************************
  1933.  Analyzing, checking, and optimizing tables
  1934. ****************************************************************************/
  1935. #ifdef NOT_YET
  1936. static void print_msg(THD *thd, const char *table_name, const char *op_name,
  1937.       const char *msg_type, const char *fmt, ...)
  1938. {
  1939.   Protocol *protocol= thd->protocol;
  1940.   char msgbuf[256];
  1941.   msgbuf[0] = 0;
  1942.   va_list args;
  1943.   va_start(args,fmt);
  1944.   my_vsnprintf(msgbuf, sizeof(msgbuf), fmt, args);
  1945.   msgbuf[sizeof(msgbuf) - 1] = 0; // healthy paranoia
  1946.   DBUG_PRINT(msg_type,("message: %s",msgbuf));
  1947.   protocol->set_nfields(4);
  1948.   protocol->prepare_for_resend();
  1949.   protocol->store(table_name);
  1950.   protocol->store(op_name);
  1951.   protocol->store(msg_type);
  1952.   protocol->store(msgbuf);
  1953.   if (protocol->write())
  1954.     thd->killed=1;
  1955. }
  1956. #endif
  1957. int ha_berkeley::analyze(THD* thd, HA_CHECK_OPT* check_opt)
  1958. {
  1959.   uint i;
  1960.   DB_BTREE_STAT *stat=0;
  1961.   DB_TXN_STAT *txn_stat_ptr= 0;
  1962.   /*
  1963.    Original bdb documentation says:
  1964.    "The DB->stat method cannot be transaction-protected.
  1965.    For this reason, it should be called in a thread of
  1966.    control that has no open cursors or active transactions."
  1967.    So, let's check if there are any changes have been done since
  1968.    the beginning of the transaction..
  1969.   */
  1970.   if (!db_env->txn_stat(db_env, &txn_stat_ptr, 0) &&
  1971.       txn_stat_ptr && txn_stat_ptr->st_nactive>=2)
  1972.   {
  1973.     DB_TXN_ACTIVE *atxn_stmt= 0, *atxn_all= 0;
  1974.     
  1975.     DB_TXN *txn_all= (DB_TXN*) thd->transaction.all.bdb_tid;
  1976.     u_int32_t all_id= txn_all->id(txn_all);
  1977.     
  1978.     DB_TXN *txn_stmt= (DB_TXN*) thd->transaction.stmt.bdb_tid;
  1979.     u_int32_t stmt_id= txn_stmt->id(txn_stmt);
  1980.     
  1981.     DB_TXN_ACTIVE *cur= txn_stat_ptr->st_txnarray;
  1982.     DB_TXN_ACTIVE *end= cur + txn_stat_ptr->st_nactive;
  1983.     for (; cur!=end && (!atxn_stmt || !atxn_all); cur++)
  1984.     {
  1985.       if (cur->txnid==all_id) atxn_all= cur;
  1986.       if (cur->txnid==stmt_id) atxn_stmt= cur;
  1987.     }
  1988.     
  1989.     if (atxn_stmt && atxn_all &&
  1990. log_compare(&atxn_stmt->lsn,&atxn_all->lsn))
  1991.     {
  1992.       free(txn_stat_ptr);
  1993.       return HA_ADMIN_REJECT;
  1994.     }
  1995.     free(txn_stat_ptr);
  1996.   }
  1997.   for (i=0 ; i < table->keys ; i++)
  1998.   {
  1999.     if (stat)
  2000.     {
  2001.       free(stat);
  2002.       stat=0;
  2003.     }
  2004.     if ((key_file[i]->stat)(key_file[i], (void*) &stat, 0))
  2005.       goto err; /* purecov: inspected */
  2006.     share->rec_per_key[i]= (stat->bt_ndata /
  2007.     (stat->bt_nkeys ? stat->bt_nkeys : 1));
  2008.   }
  2009.   /* A hidden primary key is not in key_file[] */
  2010.   if (hidden_primary_key)
  2011.   {
  2012.     if (stat)
  2013.     {
  2014.       free(stat);
  2015.       stat=0;
  2016.     }
  2017.     if ((file->stat)(file, (void*) &stat, 0))
  2018.       goto err; /* purecov: inspected */
  2019.   }
  2020.   pthread_mutex_lock(&share->mutex);
  2021.   share->rows=stat->bt_ndata;
  2022.   share->status|=STATUS_BDB_ANALYZE; // Save status on close
  2023.   share->version++; // Update stat in table
  2024.   pthread_mutex_unlock(&share->mutex);
  2025.   update_status(share,table); // Write status to file
  2026.   if (stat)
  2027.     free(stat);
  2028.   return ((share->status & STATUS_BDB_ANALYZE) ? HA_ADMIN_FAILED :
  2029.   HA_ADMIN_OK);
  2030. err:
  2031.   if (stat) /* purecov: inspected */
  2032.     free(stat); /* purecov: inspected */
  2033.   return HA_ADMIN_FAILED; /* purecov: inspected */
  2034. }
  2035. int ha_berkeley::optimize(THD* thd, HA_CHECK_OPT* check_opt)
  2036. {
  2037.   return ha_berkeley::analyze(thd,check_opt);
  2038. }
  2039. int ha_berkeley::check(THD* thd, HA_CHECK_OPT* check_opt)
  2040. {
  2041.   DBUG_ENTER("ha_berkeley::check");
  2042.   DBUG_RETURN(HA_ADMIN_NOT_IMPLEMENTED);
  2043. #ifdef NOT_YET
  2044.   char name_buff[FN_REFLEN];
  2045.   int error;
  2046.   DB *tmp_file;
  2047.   /*
  2048.     To get this to work we need to ensure that no running transaction is
  2049.     using the table. We also need to create a new environment without
  2050.     locking for this.
  2051.   */
  2052.   /* We must open the file again to be able to check it! */
  2053.   if ((error=db_create(&tmp_file, db_env, 0)))
  2054.   {
  2055.     print_msg(thd, table->real_name, "check", "error",
  2056.       "Got error %d creating environment",error);
  2057.     DBUG_RETURN(HA_ADMIN_FAILED);
  2058.   }
  2059.   /* Compare the overall structure */
  2060.   tmp_file->set_bt_compare(tmp_file,
  2061.    (hidden_primary_key ? berkeley_cmp_hidden_key :
  2062.     berkeley_cmp_packed_key));
  2063.   tmp_file->app_private= (void*) (table->key_info+table->primary_key);
  2064.   fn_format(name_buff,share->table_name,"", ha_berkeley_ext, 2 | 4);
  2065.   if ((error=tmp_file->verify(tmp_file, name_buff, NullS, (FILE*) 0,
  2066.       hidden_primary_key ? 0 : DB_NOORDERCHK)))
  2067.   {
  2068.     print_msg(thd, table->real_name, "check", "error",
  2069.       "Got error %d checking file structure",error);
  2070.     tmp_file->close(tmp_file,0);
  2071.     DBUG_RETURN(HA_ADMIN_CORRUPT);
  2072.   }
  2073.   /* Check each index */
  2074.   tmp_file->set_bt_compare(tmp_file, berkeley_cmp_packed_key);
  2075.   for (uint index=0,i=0 ; i < table->keys ; i++)
  2076.   {
  2077.     char part[7];
  2078.     if (i == primary_key)
  2079.       strmov(part,"main");
  2080.     else
  2081.       sprintf(part,"key%02d",++index);
  2082.     tmp_file->app_private= (void*) (table->key_info+i);
  2083.     if ((error=tmp_file->verify(tmp_file, name_buff, part, (FILE*) 0,
  2084. DB_ORDERCHKONLY)))
  2085.     {
  2086.       print_msg(thd, table->real_name, "check", "error",
  2087. "Key %d was not in order (Error: %d)",
  2088. index+ test(i >= primary_key),
  2089. error);
  2090.       tmp_file->close(tmp_file,0);
  2091.       DBUG_RETURN(HA_ADMIN_CORRUPT);
  2092.     }
  2093.   }
  2094.   tmp_file->close(tmp_file,0);
  2095.   DBUG_RETURN(HA_ADMIN_OK);
  2096. #endif
  2097. }
  2098. /****************************************************************************
  2099.  Handling the shared BDB_SHARE structure that is needed to provide table
  2100.  locking.
  2101. ****************************************************************************/
  2102. static byte* bdb_get_key(BDB_SHARE *share,uint *length,
  2103.  my_bool not_used __attribute__((unused)))
  2104. {
  2105.   *length=share->table_name_length;
  2106.   return (byte*) share->table_name;
  2107. }
  2108. static BDB_SHARE *get_share(const char *table_name, TABLE *table)
  2109. {
  2110.   BDB_SHARE *share;
  2111.   pthread_mutex_lock(&bdb_mutex);
  2112.   uint length=(uint) strlen(table_name);
  2113.   if (!(share=(BDB_SHARE*) hash_search(&bdb_open_tables, (byte*) table_name,
  2114.        length)))
  2115.   {
  2116.     ulong *rec_per_key;
  2117.     char *tmp_name;
  2118.     DB **key_file;
  2119.     u_int32_t *key_type;
  2120.     if ((share=(BDB_SHARE *)
  2121.  my_multi_malloc(MYF(MY_WME | MY_ZEROFILL),
  2122.  &share, sizeof(*share),
  2123.  &rec_per_key, table->keys * sizeof(ha_rows),
  2124.  &tmp_name, length+1,
  2125.  &key_file, (table->keys+1) * sizeof(*key_file),
  2126.  &key_type, (table->keys+1) * sizeof(u_int32_t),
  2127.  NullS)))
  2128.     {
  2129.       share->rec_per_key = rec_per_key;
  2130.       share->table_name = tmp_name;
  2131.       share->table_name_length=length;
  2132.       strmov(share->table_name,table_name);
  2133.       share->key_file = key_file;
  2134.       share->key_type = key_type;
  2135.       if (my_hash_insert(&bdb_open_tables, (byte*) share))
  2136.       {
  2137. pthread_mutex_unlock(&bdb_mutex); /* purecov: inspected */
  2138. my_free((gptr) share,0); /* purecov: inspected */
  2139. return 0; /* purecov: inspected */
  2140.       }
  2141.       thr_lock_init(&share->lock);
  2142.       pthread_mutex_init(&share->mutex,MY_MUTEX_INIT_FAST);
  2143.     }
  2144.   }
  2145.   pthread_mutex_unlock(&bdb_mutex);
  2146.   return share;
  2147. }
  2148. static int free_share(BDB_SHARE *share, TABLE *table, uint hidden_primary_key,
  2149.       bool mutex_is_locked)
  2150. {
  2151.   int error, result = 0;
  2152.   uint keys=table->keys + test(hidden_primary_key);
  2153.   pthread_mutex_lock(&bdb_mutex);
  2154.   if (mutex_is_locked)
  2155.     pthread_mutex_unlock(&share->mutex); /* purecov: inspected */
  2156.   if (!--share->use_count)
  2157.   {
  2158.     DB **key_file = share->key_file;
  2159.     update_status(share,table);
  2160.     /* this does share->file->close() implicitly */
  2161.     for (uint i=0; i < keys; i++)
  2162.     {
  2163.       if (key_file[i] && (error=key_file[i]->close(key_file[i],0)))
  2164. result=error; /* purecov: inspected */
  2165.     }
  2166.     if (share->status_block &&
  2167. (error = share->status_block->close(share->status_block,0)))
  2168.       result = error; /* purecov: inspected */
  2169.     hash_delete(&bdb_open_tables, (byte*) share);
  2170.     thr_lock_delete(&share->lock);
  2171.     pthread_mutex_destroy(&share->mutex);
  2172.     my_free((gptr) share, MYF(0));
  2173.   }
  2174.   pthread_mutex_unlock(&bdb_mutex);
  2175.   return result;
  2176. }
  2177. /*
  2178.   Get status information that is stored in the 'status' sub database
  2179.   and the max used value for the hidden primary key.
  2180. */
  2181. void ha_berkeley::get_status()
  2182. {
  2183.   if (!test_all_bits(share->status,(STATUS_PRIMARY_KEY_INIT |
  2184.     STATUS_ROW_COUNT_INIT)))
  2185.   {
  2186.     pthread_mutex_lock(&share->mutex);
  2187.     if (!(share->status & STATUS_PRIMARY_KEY_INIT))
  2188.     {
  2189.       (void) extra(HA_EXTRA_KEYREAD);
  2190.       index_init(primary_key);
  2191.       if (!index_last(table->record[1]))
  2192. share->auto_ident=uint5korr(current_ident);
  2193.       index_end();
  2194.       (void) extra(HA_EXTRA_NO_KEYREAD);
  2195.     }
  2196.     if (! share->status_block)
  2197.     {
  2198.       char name_buff[FN_REFLEN];
  2199.       uint open_mode= (((table->db_stat & HA_READ_ONLY) ? DB_RDONLY : 0)
  2200.        | DB_THREAD);
  2201.       fn_format(name_buff, share->table_name,"", ha_berkeley_ext, 2 | 4);
  2202.       if (!db_create(&share->status_block, db_env, 0))
  2203.       {
  2204. if (share->status_block->open(share->status_block, NULL, name_buff,
  2205.       "status", DB_BTREE, open_mode, 0))
  2206. {
  2207.   share->status_block->close(share->status_block, 0); /* purecov: inspected */
  2208.   share->status_block=0; /* purecov: inspected */
  2209. }
  2210.       }
  2211.     }
  2212.     if (!(share->status & STATUS_ROW_COUNT_INIT) && share->status_block)
  2213.     {
  2214.       share->org_rows=share->rows=
  2215. table->max_rows ? table->max_rows : HA_BERKELEY_MAX_ROWS;
  2216.       if (!share->status_block->cursor(share->status_block, 0, &cursor, 0))
  2217.       {
  2218. DBT row;
  2219. char rec_buff[64];
  2220. bzero((char*) &row,sizeof(row));
  2221. bzero((char*) &last_key,sizeof(last_key));
  2222. row.data=rec_buff;
  2223. row.ulen=sizeof(rec_buff);
  2224. row.flags=DB_DBT_USERMEM;
  2225. if (!cursor->c_get(cursor, &last_key, &row, DB_FIRST))
  2226. {
  2227.   uint i;
  2228.   uchar *pos=(uchar*) row.data;
  2229.   share->org_rows=share->rows=uint4korr(pos); pos+=4;
  2230.   for (i=0 ; i < table->keys ; i++)
  2231.   {
  2232.     share->rec_per_key[i]=uint4korr(pos); pos+=4;
  2233.   }
  2234. }
  2235. cursor->c_close(cursor);
  2236.       }
  2237.       cursor=0; // Safety
  2238.     }
  2239.     share->status|= STATUS_PRIMARY_KEY_INIT | STATUS_ROW_COUNT_INIT;
  2240.     pthread_mutex_unlock(&share->mutex);
  2241.   }
  2242. }
  2243. static int write_status(DB *status_block, char *buff, uint length)
  2244. {
  2245.   DBT row,key;
  2246.   int error;
  2247.   const char *key_buff="status";
  2248.   bzero((char*) &row,sizeof(row));
  2249.   bzero((char*) &key,sizeof(key));
  2250.   row.data=buff;
  2251.   key.data=(void*) key_buff;
  2252.   key.size=sizeof(key_buff);
  2253.   row.size=length;
  2254.   error=status_block->put(status_block, 0, &key, &row, 0);
  2255.   return error;
  2256. }
  2257. static void update_status(BDB_SHARE *share, TABLE *table)
  2258. {
  2259.   DBUG_ENTER("update_status");
  2260.   if (share->rows != share->org_rows ||
  2261.       (share->status & STATUS_BDB_ANALYZE))
  2262.   {
  2263.     pthread_mutex_lock(&share->mutex);
  2264.     if (!share->status_block)
  2265.     {
  2266.       /*
  2267. Create sub database 'status' if it doesn't exist from before
  2268. (This '*should*' always exist for table created with MySQL)
  2269.       */
  2270.       char name_buff[FN_REFLEN]; /* purecov: inspected */
  2271.       if (db_create(&share->status_block, db_env, 0)) /* purecov: inspected */
  2272. goto end; /* purecov: inspected */
  2273.       share->status_block->set_flags(share->status_block,0); /* purecov: inspected */
  2274.       if (share->status_block->open(share->status_block, NULL,
  2275.     fn_format(name_buff,share->table_name,"",
  2276.       ha_berkeley_ext,2 | 4),
  2277.     "status", DB_BTREE,
  2278.     DB_THREAD | DB_CREATE, my_umask)) /* purecov: inspected */
  2279. goto end; /* purecov: inspected */
  2280.     }
  2281.     {
  2282.       char rec_buff[4+MAX_KEY*4], *pos=rec_buff;
  2283.       int4store(pos,share->rows); pos+=4;
  2284.       for (uint i=0 ; i < table->keys ; i++)
  2285.       {
  2286. int4store(pos,share->rec_per_key[i]); pos+=4;
  2287.       }
  2288.       DBUG_PRINT("info",("updating status for %s",share->table_name));
  2289.       (void) write_status(share->status_block, rec_buff,
  2290.   (uint) (pos-rec_buff));
  2291.       share->status&= ~STATUS_BDB_ANALYZE;
  2292.       share->org_rows=share->rows;
  2293.     }
  2294. end:
  2295.     pthread_mutex_unlock(&share->mutex);
  2296.   }
  2297.   DBUG_VOID_RETURN;
  2298. }
  2299. /*
  2300.   Return an estimated of the number of rows in the table.
  2301.   Used when sorting to allocate buffers and by the optimizer.
  2302. */
  2303. ha_rows ha_berkeley::estimate_rows_upper_bound()
  2304. {
  2305.   return share->rows + HA_BERKELEY_EXTRA_ROWS;
  2306. }
  2307. #endif /* HAVE_BERKELEY_DB */