ha_berkeley.cc
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:66k
源码类别:

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