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

MySQL数据库

开发平台:

Visual C++

  1. /*-
  2.  * See the file LICENSE for redistribution information.
  3.  *
  4.  * Copyright (c) 1996-2002
  5.  * Sleepycat Software.  All rights reserved.
  6.  */
  7. /*
  8.  * Copyright (c) 1990, 1993, 1994, 1995, 1996
  9.  * Keith Bostic.  All rights reserved.
  10.  */
  11. /*
  12.  * Copyright (c) 1990, 1993, 1994, 1995
  13.  * The Regents of the University of California.  All rights reserved.
  14.  *
  15.  * Redistribution and use in source and binary forms, with or without
  16.  * modification, are permitted provided that the following conditions
  17.  * are met:
  18.  * 1. Redistributions of source code must retain the above copyright
  19.  *    notice, this list of conditions and the following disclaimer.
  20.  * 2. Redistributions in binary form must reproduce the above copyright
  21.  *    notice, this list of conditions and the following disclaimer in the
  22.  *    documentation and/or other materials provided with the distribution.
  23.  * 3. Neither the name of the University nor the names of its contributors
  24.  *    may be used to endorse or promote products derived from this software
  25.  *    without specific prior written permission.
  26.  *
  27.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  28.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  29.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  30.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  31.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  32.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  33.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  34.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  35.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  36.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  37.  * SUCH DAMAGE.
  38.  */
  39. #include "db_config.h"
  40. #ifndef lint
  41. static const char revid[] = "$Id: db.c,v 11.246 2002/08/20 14:40:00 margo Exp $";
  42. #endif /* not lint */
  43. #ifndef NO_SYSTEM_INCLUDES
  44. #include <sys/types.h>
  45. #include <stddef.h>
  46. #include <stdlib.h>
  47. #include <string.h>
  48. #endif
  49. #include "db_int.h"
  50. #include "dbinc/db_page.h"
  51. #include "dbinc/db_shash.h"
  52. #include "dbinc/db_swap.h"
  53. #include "dbinc/btree.h"
  54. #include "dbinc/hash.h"
  55. #include "dbinc/lock.h"
  56. #include "dbinc/log.h"
  57. #include "dbinc/log.h"
  58. #include "dbinc/mp.h"
  59. #include "dbinc/qam.h"
  60. #include "dbinc/txn.h"
  61. static int __db_disassociate __P((DB *));
  62. #if CONFIG_TEST
  63. static void __db_makecopy __P((const char *, const char *));
  64. static int  __db_testdocopy __P((DB_ENV *, const char *));
  65. static int  __qam_testdocopy __P((DB *, const char *));
  66. #endif
  67. /*
  68.  * DB.C --
  69.  * This file contains the utility functions for the DBP layer.
  70.  */
  71. /*
  72.  * __db_master_open --
  73.  * Open up a handle on a master database.
  74.  *
  75.  * PUBLIC: int __db_master_open __P((DB *,
  76.  * PUBLIC:     DB_TXN *, const char *, u_int32_t, int, DB **));
  77.  */
  78. int
  79. __db_master_open(subdbp, txn, name, flags, mode, dbpp)
  80. DB *subdbp;
  81. DB_TXN *txn;
  82. const char *name;
  83. u_int32_t flags;
  84. int mode;
  85. DB **dbpp;
  86. {
  87. DB *dbp;
  88. int ret;
  89. /* Open up a handle on the main database. */
  90. if ((ret = db_create(&dbp, subdbp->dbenv, 0)) != 0)
  91. return (ret);
  92. /*
  93.  * It's always a btree.
  94.  * Run in the transaction we've created.
  95.  * Set the pagesize in case we're creating a new database.
  96.  * Flag that we're creating a database with subdatabases.
  97.  */
  98. dbp->type = DB_BTREE;
  99. dbp->pgsize = subdbp->pgsize;
  100. F_SET(dbp, DB_AM_SUBDB);
  101. F_SET(dbp, F_ISSET(subdbp,
  102.     DB_AM_RECOVER | DB_AM_SWAP | DB_AM_ENCRYPT | DB_AM_CHKSUM));
  103. /*
  104.  * If there was a subdb specified, then we only want to apply
  105.  * DB_EXCL to the subdb, not the actual file.  We only got here
  106.  * because there was a subdb specified.
  107.  */
  108. LF_CLR(DB_EXCL);
  109. LF_SET(DB_RDWRMASTER);
  110. if ((ret = __db_dbopen(dbp, txn, name, NULL, flags, mode, PGNO_BASE_MD))
  111.     != 0)
  112. goto err;
  113. /*
  114.  * Verify that pagesize is the same on both.
  115.  * The items in dbp were now initialized from the meta
  116.  * page.  The items in dbp were set in __db_dbopen
  117.  * when we either read or created the master file.
  118.  * Other items such as checksum and encryption are
  119.  * checked when we read the meta-page.  So we do not
  120.  * check those here.  However, if the meta-page caused
  121.  * chksumming to be turned on and it wasn't already, set
  122.  * it here.
  123.  */
  124. if (F_ISSET(dbp, DB_AM_CHKSUM))
  125. F_SET(subdbp, DB_AM_CHKSUM);
  126. if (subdbp->pgsize != 0 && dbp->pgsize != subdbp->pgsize) {
  127. ret = EINVAL;
  128. __db_err(dbp->dbenv,
  129.     "Different pagesize specified on existent file");
  130. goto err;
  131. }
  132. err:
  133. if (ret != 0 && !F_ISSET(dbp, DB_AM_DISCARD))
  134. __db_close_i(dbp, txn, 0);
  135. else
  136. *dbpp = dbp;
  137. return (ret);
  138. }
  139. /*
  140.  * __db_master_update --
  141.  * Add/Open/Remove a subdatabase from a master database.
  142.  *
  143.  * PUBLIC: int __db_master_update __P((DB *, DB *, DB_TXN *, const char *,
  144.  * PUBLIC:     DBTYPE, mu_action, const char *, u_int32_t));
  145.  */
  146. int
  147. __db_master_update(mdbp, sdbp, txn, subdb, type, action, newname, flags)
  148. DB *mdbp, *sdbp;
  149. DB_TXN *txn;
  150. const char *subdb;
  151. DBTYPE type;
  152. mu_action action;
  153. const char *newname;
  154. u_int32_t flags;
  155. {
  156. DB_ENV *dbenv;
  157. DBC *dbc, *ndbc;
  158. DBT key, data, ndata;
  159. PAGE *p;
  160. db_pgno_t t_pgno;
  161. int modify, ret, t_ret;
  162. dbenv = mdbp->dbenv;
  163. dbc = ndbc = NULL;
  164. p = NULL;
  165. memset(&key, 0, sizeof(key));
  166. memset(&data, 0, sizeof(data));
  167. /* Might we modify the master database?  If so, we'll need to lock. */
  168. modify = (action != MU_OPEN || LF_ISSET(DB_CREATE)) ? 1 : 0;
  169. /*
  170.  * Open up a cursor.  If this is CDB and we're creating the database,
  171.  * make it an update cursor.
  172.  */
  173. if ((ret = mdbp->cursor(mdbp, txn, &dbc,
  174.     (CDB_LOCKING(dbenv) && modify) ? DB_WRITECURSOR : 0)) != 0)
  175. goto err;
  176. /*
  177.  * Point the cursor at the record.
  178.  *
  179.  * If we're removing or potentially creating an entry, lock the page
  180.  * with DB_RMW.
  181.  *
  182.  * We do multiple cursor operations with the cursor in some cases and
  183.  * subsequently access the data DBT information.  Set DB_DBT_MALLOC so
  184.  * we don't risk modification of the data between our uses of it.
  185.  *
  186.  * !!!
  187.  * We don't include the name's nul termination in the database.
  188.  */
  189. key.data = (void *)subdb;
  190. key.size = (u_int32_t)strlen(subdb);
  191. F_SET(&data, DB_DBT_MALLOC);
  192. ret = dbc->c_get(dbc, &key, &data,
  193.     DB_SET | ((STD_LOCKING(dbc) && modify) ? DB_RMW : 0));
  194. /*
  195.  * What we do next--whether or not we found a record for the
  196.  * specified subdatabase--depends on what the specified action is.
  197.  * Handle ret appropriately as the first statement of each case.
  198.  */
  199. switch (action) {
  200. case MU_REMOVE:
  201. /*
  202.  * We should have found something if we're removing it.  Note
  203.  * that in the common case where the DB we're asking to remove
  204.  * doesn't exist, we won't get this far;  __db_subdb_remove
  205.  * will already have returned an error from __db_open.
  206.  */
  207. if (ret != 0)
  208. goto err;
  209. /*
  210.  * Delete the subdatabase entry first;  if this fails,
  211.  * we don't want to touch the actual subdb pages.
  212.  */
  213. if ((ret = dbc->c_del(dbc, 0)) != 0)
  214. goto err;
  215. /*
  216.  * We're handling actual data, not on-page meta-data,
  217.  * so it hasn't been converted to/from opposite
  218.  * endian architectures.  Do it explicitly, now.
  219.  */
  220. memcpy(&sdbp->meta_pgno, data.data, sizeof(db_pgno_t));
  221. DB_NTOHL(&sdbp->meta_pgno);
  222. if ((ret =
  223.     mdbp->mpf->get(mdbp->mpf, &sdbp->meta_pgno, 0, &p)) != 0)
  224. goto err;
  225. /* Free and put the page. */
  226. if ((ret = __db_free(dbc, p)) != 0) {
  227. p = NULL;
  228. goto err;
  229. }
  230. p = NULL;
  231. break;
  232. case MU_RENAME:
  233. /* We should have found something if we're renaming it. */
  234. if (ret != 0)
  235. goto err;
  236. /*
  237.  * Before we rename, we need to make sure we're not
  238.  * overwriting another subdatabase, or else this operation
  239.  * won't be undoable.  Open a second cursor and check
  240.  * for the existence of newname;  it shouldn't appear under
  241.  * us since we hold the metadata lock.
  242.  */
  243. if ((ret = mdbp->cursor(mdbp, txn, &ndbc, 0)) != 0)
  244. goto err;
  245. DB_ASSERT(newname != NULL);
  246. key.data = (void *)newname;
  247. key.size = (u_int32_t)strlen(newname);
  248. /*
  249.  * We don't actually care what the meta page of the potentially-
  250.  * overwritten DB is;  we just care about existence.
  251.  */
  252. memset(&ndata, 0, sizeof(ndata));
  253. F_SET(&ndata, DB_DBT_USERMEM | DB_DBT_PARTIAL);
  254. if ((ret = ndbc->c_get(ndbc, &key, &ndata, DB_SET)) == 0) {
  255. /* A subdb called newname exists.  Bail. */
  256. ret = EEXIST;
  257. __db_err(dbenv, "rename: database %s exists", newname);
  258. goto err;
  259. } else if (ret != DB_NOTFOUND)
  260. goto err;
  261. /*
  262.  * Now do the put first;  we don't want to lose our
  263.  * sole reference to the subdb.  Use the second cursor
  264.  * so that the first one continues to point to the old record.
  265.  */
  266. if ((ret = ndbc->c_put(ndbc, &key, &data, DB_KEYFIRST)) != 0)
  267. goto err;
  268. if ((ret = dbc->c_del(dbc, 0)) != 0) {
  269. /*
  270.  * If the delete fails, try to delete the record
  271.  * we just put, in case we're not txn-protected.
  272.  */
  273. (void)ndbc->c_del(ndbc, 0);
  274. goto err;
  275. }
  276. break;
  277. case MU_OPEN:
  278. /*
  279.  * Get the subdatabase information.  If it already exists,
  280.  * copy out the page number and we're done.
  281.  */
  282. switch (ret) {
  283. case 0:
  284. if (LF_ISSET(DB_CREATE) && LF_ISSET(DB_EXCL)) {
  285. ret = EEXIST;
  286. goto err;
  287. }
  288. memcpy(&sdbp->meta_pgno, data.data, sizeof(db_pgno_t));
  289. DB_NTOHL(&sdbp->meta_pgno);
  290. goto done;
  291. case DB_NOTFOUND:
  292. if (LF_ISSET(DB_CREATE))
  293. break;
  294. /*
  295.  * No db_err, it is reasonable to remove a
  296.  * nonexistent db.
  297.  */
  298. ret = ENOENT;
  299. goto err;
  300. default:
  301. goto err;
  302. }
  303. /*
  304.  * We need to check against the master lorder here because
  305.  * we only want to check this if we are creating.  In the
  306.  * case where we don't create we just want to inherit.
  307.  */
  308. if (F_ISSET(mdbp, DB_AM_SWAP) != F_ISSET(sdbp, DB_AM_SWAP)) {
  309.  ret = EINVAL;
  310.  __db_err(mdbp->dbenv,
  311.     "Different lorder specified on existent file");
  312. goto err;
  313. }
  314. /* Create a subdatabase. */
  315. if ((ret = __db_new(dbc,
  316.     type == DB_HASH ? P_HASHMETA : P_BTREEMETA, &p)) != 0)
  317. goto err;
  318. sdbp->meta_pgno = PGNO(p);
  319. /*
  320.  * XXX
  321.  * We're handling actual data, not on-page meta-data, so it
  322.  * hasn't been converted to/from opposite endian architectures.
  323.  * Do it explicitly, now.
  324.  */
  325. t_pgno = PGNO(p);
  326. DB_HTONL(&t_pgno);
  327. memset(&ndata, 0, sizeof(ndata));
  328. ndata.data = &t_pgno;
  329. ndata.size = sizeof(db_pgno_t);
  330. if ((ret = dbc->c_put(dbc, &key, &ndata, DB_KEYLAST)) != 0)
  331. goto err;
  332. F_SET(sdbp, DB_AM_CREATED);
  333. break;
  334. }
  335. err:
  336. done: /*
  337.  * If we allocated a page: if we're successful, mark the page dirty
  338.  * and return it to the cache, otherwise, discard/free it.
  339.  */
  340. if (p != NULL) {
  341. if (ret == 0) {
  342. if ((t_ret =
  343.     mdbp->mpf->put(mdbp->mpf, p, DB_MPOOL_DIRTY)) != 0)
  344. ret = t_ret;
  345. /*
  346.  * Since we cannot close this file until after
  347.  * transaction commit, we need to sync the dirty
  348.  * pages, because we'll read these directly from
  349.  * disk to open.
  350.  */
  351. if ((t_ret = mdbp->sync(mdbp, 0)) != 0 && ret == 0)
  352. ret = t_ret;
  353. } else
  354. (void)mdbp->mpf->put(mdbp->mpf, p, 0);
  355. }
  356. /* Discard the cursor(s) and data. */
  357. if (data.data != NULL)
  358. __os_ufree(dbenv, data.data);
  359. if (dbc != NULL && (t_ret = dbc->c_close(dbc)) != 0 && ret == 0)
  360. ret = t_ret;
  361. if (ndbc != NULL && (t_ret = ndbc->c_close(ndbc)) != 0 && ret == 0)
  362. ret = t_ret;
  363. return (ret);
  364. }
  365. /*
  366.  * __db_dbenv_setup --
  367.  * Set up the underlying environment during a db_open.
  368.  *
  369.  * PUBLIC: int __db_dbenv_setup __P((DB *,
  370.  * PUBLIC:     DB_TXN *, const char *, u_int32_t, u_int32_t));
  371.  */
  372. int
  373. __db_dbenv_setup(dbp, txn, name, id, flags)
  374. DB *dbp;
  375. DB_TXN *txn;
  376. const char *name;
  377. u_int32_t id;
  378. u_int32_t flags;
  379. {
  380. DB *ldbp;
  381. DBT pgcookie;
  382. DB_ENV *dbenv;
  383. DB_MPOOL *dbmp;
  384. DB_MPOOLFILE *mpf;
  385. DB_PGINFO pginfo;
  386. u_int32_t maxid;
  387. int ftype, ret;
  388. dbenv = dbp->dbenv;
  389. /* If we don't yet have an environment, it's time to create it. */
  390. if (!F_ISSET(dbenv, DB_ENV_OPEN_CALLED)) {
  391. /* Make sure we have at least DB_MINCACHE pages in our cache. */
  392. if (dbenv->mp_gbytes == 0 &&
  393.     dbenv->mp_bytes < dbp->pgsize * DB_MINPAGECACHE &&
  394.     (ret = dbenv->set_cachesize(
  395.     dbenv, 0, dbp->pgsize * DB_MINPAGECACHE, 0)) != 0)
  396. return (ret);
  397. if ((ret = dbenv->open(dbenv, NULL, DB_CREATE |
  398.     DB_INIT_MPOOL | DB_PRIVATE | LF_ISSET(DB_THREAD), 0)) != 0)
  399. return (ret);
  400. }
  401. /* Register DB's pgin/pgout functions. */
  402. if ((ret = dbenv->memp_register(
  403.     dbenv, DB_FTYPE_SET, __db_pgin, __db_pgout)) != 0)
  404. return (ret);
  405. /* Create the DB_MPOOLFILE structure. */
  406. if ((ret = dbenv->memp_fcreate(dbenv, &dbp->mpf, 0)) != 0)
  407. return (ret);
  408. mpf = dbp->mpf;
  409. /* Set the database's cache priority if we've been given one. */
  410. if (dbp->priority != 0 &&
  411.     (ret = mpf->set_priority(mpf, dbp->priority)) != 0)
  412. return (ret);
  413. /*
  414.  * Open a backing file in the memory pool.
  415.  *
  416.  * If we need to pre- or post-process a file's pages on I/O, set the
  417.  * file type.  If it's a hash file, always call the pgin and pgout
  418.  * routines.  This means that hash files can never be mapped into
  419.  * process memory.  If it's a btree file and requires swapping, we
  420.  * need to page the file in and out.  This has to be right -- we can't
  421.  * mmap files that are being paged in and out.
  422.  */
  423. switch (dbp->type) {
  424. case DB_BTREE:
  425. case DB_RECNO:
  426. ftype = F_ISSET(dbp, DB_AM_SWAP | DB_AM_ENCRYPT | DB_AM_CHKSUM)
  427.     ? DB_FTYPE_SET : DB_FTYPE_NOTSET;
  428. (void)mpf->set_ftype(mpf, ftype);
  429. (void)mpf->set_clear_len(mpf, (CRYPTO_ON(dbenv) ?
  430.     dbp->pgsize : DB_PAGE_DB_LEN));
  431. break;
  432. case DB_HASH:
  433. (void)mpf->set_ftype(mpf, DB_FTYPE_SET);
  434. (void)mpf->set_clear_len(mpf, (CRYPTO_ON(dbenv) ?
  435.     dbp->pgsize : DB_PAGE_DB_LEN));
  436. break;
  437. case DB_QUEUE:
  438. ftype = F_ISSET(dbp, DB_AM_SWAP | DB_AM_ENCRYPT | DB_AM_CHKSUM)
  439.     ? DB_FTYPE_SET : DB_FTYPE_NOTSET;
  440. (void)mpf->set_ftype(mpf, ftype);
  441. (void)mpf->set_clear_len(mpf, (CRYPTO_ON(dbenv) ?
  442.     dbp->pgsize : DB_PAGE_QUEUE_LEN));
  443. break;
  444. case DB_UNKNOWN:
  445. /*
  446.  * If we're running in the verifier, our database might
  447.  * be corrupt and we might not know its type--but we may
  448.  * still want to be able to verify and salvage.
  449.  *
  450.  * If we can't identify the type, it's not going to be safe
  451.  * to call __db_pgin--we pretty much have to give up all
  452.  * hope of salvaging cross-endianness.  Proceed anyway;
  453.  * at worst, the database will just appear more corrupt
  454.  * than it actually is, but at best, we may be able
  455.  * to salvage some data even with no metadata page.
  456.  */
  457. if (F_ISSET(dbp, DB_AM_VERIFYING)) {
  458. (void)mpf->set_ftype(mpf, DB_FTYPE_NOTSET);
  459. (void)mpf->set_clear_len(mpf, DB_PAGE_DB_LEN);
  460. break;
  461. }
  462. /* FALLTHROUGH */
  463. default:
  464. return (
  465.     __db_unknown_type(dbenv, "__db_dbenv_setup", dbp->type));
  466. }
  467. (void)mpf->set_fileid(mpf, dbp->fileid);
  468. (void)mpf->set_lsn_offset(mpf, 0);
  469. pginfo.db_pagesize = dbp->pgsize;
  470. pginfo.flags =
  471.     F_ISSET(dbp, (DB_AM_CHKSUM | DB_AM_ENCRYPT | DB_AM_SWAP));
  472. pginfo.type = dbp->type;
  473. pgcookie.data = &pginfo;
  474. pgcookie.size = sizeof(DB_PGINFO);
  475. (void)mpf->set_pgcookie(mpf, &pgcookie);
  476. if ((ret = mpf->open(mpf, name,
  477.     LF_ISSET(DB_RDONLY | DB_NOMMAP | DB_ODDFILESIZE | DB_TRUNCATE) |
  478.     (F_ISSET(dbenv, DB_ENV_DIRECT_DB) ? DB_DIRECT : 0),
  479.     0, dbp->pgsize)) != 0)
  480. return (ret);
  481. /*
  482.  * We may need a per-thread mutex.  Allocate it from the mpool
  483.  * region, there's supposed to be extra space there for that purpose.
  484.  */
  485. if (LF_ISSET(DB_THREAD)) {
  486. dbmp = dbenv->mp_handle;
  487. if ((ret = __db_mutex_setup(dbenv, dbmp->reginfo, &dbp->mutexp,
  488.     MUTEX_ALLOC | MUTEX_THREAD)) != 0)
  489. return (ret);
  490. }
  491. /*
  492.  * Set up a bookkeeping entry for this database in the log region,
  493.  * if such a region exists.  Note that even if we're in recovery
  494.  * or a replication client, where we won't log registries, we'll
  495.  * still need an FNAME struct, so LOGGING_ON is the correct macro.
  496.  */
  497. if (LOGGING_ON(dbenv) &&
  498.     (ret = __dbreg_setup(dbp, name, id)) != 0)
  499. return (ret);
  500. /*
  501.  * If we're actively logging and our caller isn't a recovery function
  502.  * that already did so, assign this dbp a log fileid.
  503.  */
  504. if (DBENV_LOGGING(dbenv) && !F_ISSET(dbp, DB_AM_RECOVER) &&
  505. #if !defined(DEBUG_ROP)
  506.     !F_ISSET(dbp, DB_AM_RDONLY) &&
  507. #endif
  508.     (ret = __dbreg_new_id(dbp, txn)) != 0)
  509. return (ret);
  510. /*
  511.  * Insert ourselves into the DB_ENV's dblist.  We allocate a
  512.  * unique ID to each {fileid, meta page number} pair, and to
  513.  * each temporary file (since they all have a zero fileid).
  514.  * This ID gives us something to use to tell which DB handles
  515.  * go with which databases in all the cursor adjustment
  516.  * routines, where we don't want to do a lot of ugly and
  517.  * expensive memcmps.
  518.  */
  519. MUTEX_THREAD_LOCK(dbenv, dbenv->dblist_mutexp);
  520. for (maxid = 0, ldbp = LIST_FIRST(&dbenv->dblist);
  521.     ldbp != NULL; ldbp = LIST_NEXT(dbp, dblistlinks)) {
  522. if (name != NULL &&
  523.     memcmp(ldbp->fileid, dbp->fileid, DB_FILE_ID_LEN) == 0 &&
  524.     ldbp->meta_pgno == dbp->meta_pgno)
  525. break;
  526. if (ldbp->adj_fileid > maxid)
  527. maxid = ldbp->adj_fileid;
  528. }
  529. /*
  530.  * If ldbp is NULL, we didn't find a match, or we weren't
  531.  * really looking because name is NULL.  Assign the dbp an
  532.  * adj_fileid one higher than the largest we found, and
  533.  * insert it at the head of the master dbp list.
  534.  *
  535.  * If ldbp is not NULL, it is a match for our dbp.  Give dbp
  536.  * the same ID that ldbp has, and add it after ldbp so they're
  537.  * together in the list.
  538.  */
  539. if (ldbp == NULL) {
  540. dbp->adj_fileid = maxid + 1;
  541. LIST_INSERT_HEAD(&dbenv->dblist, dbp, dblistlinks);
  542. } else {
  543. dbp->adj_fileid = ldbp->adj_fileid;
  544. LIST_INSERT_AFTER(ldbp, dbp, dblistlinks);
  545. }
  546. MUTEX_THREAD_UNLOCK(dbenv, dbenv->dblist_mutexp);
  547. return (0);
  548. }
  549. /*
  550.  * __db_close --
  551.  * DB destructor.
  552.  *
  553.  * PUBLIC: int __db_close __P((DB *, u_int32_t));
  554.  */
  555. int
  556. __db_close(dbp, flags)
  557. DB *dbp;
  558. u_int32_t flags;
  559. {
  560. DB_ENV *dbenv;
  561. dbenv = dbp->dbenv;
  562. PANIC_CHECK(dbenv);
  563. /* Validate arguments, but as a DB handle destructor, we can't fail. */
  564. if (flags != 0 && flags != DB_NOSYNC)
  565. (void)__db_ferr(dbenv, "DB->close", 0);
  566. return (__db_close_i(dbp, NULL, flags));
  567. }
  568. /*
  569.  * __db_close_i --
  570.  * Internal DB destructor.
  571.  *
  572.  * PUBLIC: int __db_close_i __P((DB *, DB_TXN *, u_int32_t));
  573.  */
  574. int
  575. __db_close_i(dbp, txn, flags)
  576. DB *dbp;
  577. DB_TXN *txn;
  578. u_int32_t flags;
  579. {
  580. DB_ENV *dbenv;
  581. int ret, t_ret;
  582. dbenv = dbp->dbenv;
  583. ret = 0;
  584. /*
  585.  * Validate arguments, but as a DB handle destructor, we can't fail.
  586.  *
  587.  * Check for consistent transaction usage -- ignore errors.  Only
  588.  * internal callers specify transactions, so it's a serious problem
  589.  * if we get error messages.
  590.  */
  591. if (txn != NULL)
  592. (void)__db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 0);
  593. /* Refresh the structure and close any local environment. */
  594. if ((t_ret = __db_refresh(dbp, txn, flags)) != 0 && ret == 0)
  595. ret = t_ret;
  596. /*
  597.  * Call the access specific close function.
  598.  *
  599.  * !!!
  600.  * Because of where these functions are called in the DB handle close
  601.  * process, these routines can't do anything that would dirty pages or
  602.  * otherwise affect closing down the database.  Specifically, we can't
  603.  * abort and recover any of the information they control.
  604.  */
  605. if ((t_ret = __ham_db_close(dbp)) != 0 && ret == 0)
  606. ret = t_ret;
  607. if ((t_ret = __bam_db_close(dbp)) != 0 && ret == 0)
  608. ret = t_ret;
  609. if ((t_ret = __qam_db_close(dbp)) != 0 && ret == 0)
  610. ret = t_ret;
  611. --dbenv->db_ref;
  612. if (F_ISSET(dbenv, DB_ENV_DBLOCAL) && dbenv->db_ref == 0 &&
  613.     (t_ret = dbenv->close(dbenv, 0)) != 0 && ret == 0)
  614. ret = t_ret;
  615. /* Free the database handle. */
  616. memset(dbp, CLEAR_BYTE, sizeof(*dbp));
  617. __os_free(dbenv, dbp);
  618. return (ret);
  619. }
  620. /*
  621.  * __db_refresh --
  622.  * Refresh the DB structure, releasing any allocated resources.
  623.  * This does most of the work of closing files now because refresh
  624.  * is what is used during abort processing (since we can't destroy
  625.  * the actual handle) and during abort processing, we may have a
  626.  * fully opened handle.
  627.  *
  628.  * PUBLIC: int __db_refresh __P((DB *, DB_TXN *, u_int32_t));
  629.  */
  630. int
  631. __db_refresh(dbp, txn, flags)
  632. DB *dbp;
  633. DB_TXN *txn;
  634. u_int32_t flags;
  635. {
  636. DB *sdbp;
  637. DBC *dbc;
  638. DB_ENV *dbenv;
  639. DB_LOCKREQ lreq;
  640. DB_MPOOL *dbmp;
  641. int ret, t_ret;
  642. ret = 0;
  643. dbenv = dbp->dbenv;
  644. /* If never opened, or not currently open, it's easy. */
  645. if (!F_ISSET(dbp, DB_AM_OPEN_CALLED))
  646. goto never_opened;
  647. /*
  648.  * If we have any secondary indices, disassociate them from us.
  649.  * We don't bother with the mutex here;  it only protects some
  650.  * of the ops that will make us core-dump mid-close anyway, and
  651.  * if you're trying to do something with a secondary *while* you're
  652.  * closing the primary, you deserve what you get.  The disassociation
  653.  * is mostly done just so we can close primaries and secondaries in
  654.  * any order--but within one thread of control.
  655.  */
  656. for (sdbp = LIST_FIRST(&dbp->s_secondaries);
  657.     sdbp != NULL; sdbp = LIST_NEXT(sdbp, s_links)) {
  658. LIST_REMOVE(sdbp, s_links);
  659. if ((t_ret = __db_disassociate(sdbp)) != 0 && ret == 0)
  660. ret = t_ret;
  661. }
  662. /*
  663.  * Sync the underlying access method.  Do before closing the cursors
  664.  * because DB->sync allocates cursors in order to write Recno backing
  665.  * source text files.
  666.  */
  667. if (!LF_ISSET(DB_NOSYNC) && !F_ISSET(dbp, DB_AM_DISCARD) &&
  668.     (t_ret = dbp->sync(dbp, 0)) != 0 && ret == 0)
  669. ret = t_ret;
  670. /*
  671.  * Go through the active cursors and call the cursor recycle routine,
  672.  * which resolves pending operations and moves the cursors onto the
  673.  * free list.  Then, walk the free list and call the cursor destroy
  674.  * routine.  Note that any failure on a close is considered "really
  675.  * bad" and we just break out of the loop and force forward.
  676.  */
  677. while ((dbc = TAILQ_FIRST(&dbp->active_queue)) != NULL)
  678. if ((t_ret = dbc->c_close(dbc)) != 0) {
  679. if (ret == 0)
  680. ret = t_ret;
  681. break;
  682. }
  683. while ((dbc = TAILQ_FIRST(&dbp->free_queue)) != NULL)
  684. if ((t_ret = __db_c_destroy(dbc)) != 0) {
  685. if (ret == 0)
  686. ret = t_ret;
  687. break;
  688. }
  689. /*
  690.  * Close any outstanding join cursors.  Join cursors destroy
  691.  * themselves on close and have no separate destroy routine.
  692.  */
  693. while ((dbc = TAILQ_FIRST(&dbp->join_queue)) != NULL)
  694. if ((t_ret = dbc->c_close(dbc)) != 0) {
  695. if (ret == 0)
  696. ret = t_ret;
  697. break;
  698. }
  699. /*
  700.  * Sync the memory pool, even though we've already called DB->sync,
  701.  * because closing cursors can dirty pages by deleting items they
  702.  * referenced.
  703.  */
  704. if (!LF_ISSET(DB_NOSYNC) && !F_ISSET(dbp, DB_AM_DISCARD) &&
  705.     (t_ret = dbp->mpf->sync(dbp->mpf)) != 0 && ret == 0)
  706. ret = t_ret;
  707. /* Close any handle we've been holding since the open.  */
  708. if (dbp->saved_open_fhp != NULL &&
  709.     F_ISSET(dbp->saved_open_fhp, DB_FH_VALID) &&
  710.     (t_ret = __os_closehandle(dbenv, dbp->saved_open_fhp)) != 0 &&
  711.     ret == 0)
  712. ret = t_ret;
  713. never_opened:
  714. /*
  715.  * We are not releasing the handle lock here because we're about
  716.  * to release all locks held by dbp->lid below.  There are two
  717.  * ways that we can get in here with a handle_lock, but not a
  718.  * dbp->lid.  The first is when our lid has been hijacked by a
  719.  * subdb.  The second is when we are a Queue database in the midst
  720.  * of a rename.  If the queue file hasn't actually been opened, we
  721.  * hijack the main dbp's locker id to do the open so we can get the
  722.  * extent files.  In both cases, we needn't free the handle lock
  723.  * because it will be freed when the hijacked locker-id is freed.
  724.  */
  725. DB_ASSERT(!LOCK_ISSET(dbp->handle_lock) ||
  726.     dbp->lid != DB_LOCK_INVALIDID ||
  727.     dbp->type == DB_QUEUE ||
  728.     F_ISSET(dbp, DB_AM_SUBDB));
  729. if (dbp->lid != DB_LOCK_INVALIDID) {
  730. /* We may have pending trade operations on this dbp. */
  731. if (txn != NULL)
  732. __txn_remlock(dbenv, txn, &dbp->handle_lock, dbp->lid);
  733. /* We may be holding the handle lock; release it. */
  734. lreq.op = DB_LOCK_PUT_ALL;
  735. if ((t_ret = __lock_vec(dbenv,
  736.     dbp->lid, 0, &lreq, 1, NULL)) != 0 && ret == 0)
  737. ret = t_ret;
  738. if ((t_ret =
  739.     dbenv->lock_id_free(dbenv, dbp->lid)) != 0 && ret == 0)
  740. ret = t_ret;
  741. dbp->lid = DB_LOCK_INVALIDID;
  742. LOCK_INIT(dbp->handle_lock);
  743. }
  744. /* Discard the locker ID allocated as the fileid. */
  745. if (F_ISSET(dbp, DB_AM_INMEM) &&
  746.     LOCKING_ON(dbenv) && (t_ret = dbenv->lock_id_free(
  747.     dbenv, *(u_int32_t *)dbp->fileid)) != 0 && ret == 0)
  748. ret = t_ret;
  749. dbp->type = DB_UNKNOWN;
  750. /* Discard the thread mutex. */
  751. if (dbp->mutexp != NULL) {
  752. dbmp = dbenv->mp_handle;
  753. __db_mutex_free(dbenv, dbmp->reginfo, dbp->mutexp);
  754. dbp->mutexp = NULL;
  755. }
  756. /* Discard any memory used to store returned data. */
  757. if (dbp->my_rskey.data != NULL)
  758. __os_free(dbp->dbenv, dbp->my_rskey.data);
  759. if (dbp->my_rkey.data != NULL)
  760. __os_free(dbp->dbenv, dbp->my_rkey.data);
  761. if (dbp->my_rdata.data != NULL)
  762. __os_free(dbp->dbenv, dbp->my_rdata.data);
  763. /* For safety's sake;  we may refresh twice. */
  764. memset(&dbp->my_rskey, 0, sizeof(DBT));
  765. memset(&dbp->my_rkey, 0, sizeof(DBT));
  766. memset(&dbp->my_rdata, 0, sizeof(DBT));
  767. /*
  768.  * Remove this DB handle from the DB_ENV's dblist, if it's been added.
  769.  */
  770. MUTEX_THREAD_LOCK(dbenv, dbenv->dblist_mutexp);
  771. if (dbp->dblistlinks.le_prev != NULL)
  772. LIST_REMOVE(dbp, dblistlinks);
  773. MUTEX_THREAD_UNLOCK(dbenv, dbenv->dblist_mutexp);
  774. dbp->dblistlinks.le_prev = NULL;
  775. /* Close the memory pool file handle. */
  776. if (dbp->mpf != NULL) {
  777. if ((t_ret = dbp->mpf->close(dbp->mpf,
  778.     F_ISSET(dbp, DB_AM_DISCARD) ? DB_MPOOL_DISCARD : 0)) != 0 &&
  779.     ret == 0)
  780. ret = t_ret;
  781. dbp->mpf = NULL;
  782. }
  783. if (LOGGING_ON(dbp->dbenv)) {
  784. /*
  785.  * Discard the log file id, if any.  We want to log the close
  786.  * if and only if this is not a recovery dbp.
  787.  */
  788. if (F_ISSET(dbp, DB_AM_RECOVER))
  789. (void)__dbreg_revoke_id(dbp, 0);
  790. else
  791. (void)__dbreg_close_id(dbp, txn);
  792. /* Discard the log FNAME. */
  793. (void)__dbreg_teardown(dbp);
  794. }
  795. /* Clear out fields that normally get set during open. */
  796. memset(dbp->fileid, 0, sizeof(dbp->fileid));
  797. dbp->adj_fileid = 0;
  798. dbp->meta_pgno = 0;
  799. dbp->cur_lid = DB_LOCK_INVALIDID;
  800. dbp->associate_lid = DB_LOCK_INVALIDID;
  801. dbp->cl_id = 0;
  802. /*
  803.  * If we are being refreshed with a txn specified, then we need
  804.  * to make sure that we clear out the lock handle field, because
  805.  * releasing all the locks for this transaction will release this
  806.  * lock and we don't want close to stumble upon this handle and
  807.  * try to close it.
  808.  */
  809. if (txn != NULL)
  810. LOCK_INIT(dbp->handle_lock);
  811. F_CLR(dbp, DB_AM_DBM_ERROR);
  812. F_CLR(dbp, DB_AM_DISCARD);
  813. F_CLR(dbp, DB_AM_INMEM);
  814. F_CLR(dbp, DB_AM_RECOVER);
  815. F_CLR(dbp, DB_AM_OPEN_CALLED);
  816. F_CLR(dbp, DB_AM_RDONLY);
  817. F_CLR(dbp, DB_AM_SWAP);
  818. return (ret);
  819. }
  820. /*
  821.  * __db_log_page
  822.  * Log a meta-data or root page during a subdatabase create operation.
  823.  *
  824.  * PUBLIC: int __db_log_page __P((DB *, DB_TXN *, DB_LSN *, db_pgno_t, PAGE *));
  825.  */
  826. int
  827. __db_log_page(dbp, txn, lsn, pgno, page)
  828. DB *dbp;
  829. DB_TXN *txn;
  830. DB_LSN *lsn;
  831. db_pgno_t pgno;
  832. PAGE *page;
  833. {
  834. DBT page_dbt;
  835. DB_LSN new_lsn;
  836. int ret;
  837. if (!LOGGING_ON(dbp->dbenv) || txn == NULL)
  838. return (0);
  839. memset(&page_dbt, 0, sizeof(page_dbt));
  840. page_dbt.size = dbp->pgsize;
  841. page_dbt.data = page;
  842. ret = __crdel_metasub_log(dbp, txn, &new_lsn, 0, pgno, &page_dbt, lsn);
  843. if (ret == 0)
  844. page->lsn = new_lsn;
  845. return (ret);
  846. }
  847. /*
  848.  * __db_backup_name
  849.  * Create the backup file name for a given file.
  850.  *
  851.  * PUBLIC: int __db_backup_name __P((DB_ENV *,
  852.  * PUBLIC:     const char *, DB_TXN *, char **));
  853.  */
  854. #undef BACKUP_PREFIX
  855. #define BACKUP_PREFIX "__db."
  856. #undef MAX_LSN_TO_TEXT
  857. #define MAX_LSN_TO_TEXT 17
  858. int
  859. __db_backup_name(dbenv, name, txn, backup)
  860. DB_ENV *dbenv;
  861. const char *name;
  862. DB_TXN *txn;
  863. char **backup;
  864. {
  865. DB_LSN lsn;
  866. size_t len;
  867. int plen, ret;
  868. char *p, *retp;
  869. /*
  870.  * Create the name.  Backup file names are in one of two forms:
  871.  *
  872.  * In a transactional env: __db.LSN(8).LSN(8)
  873.  * and
  874.  * in a non-transactional env: __db.FILENAME.
  875.  *
  876.  * If the transaction doesn't have a current LSN, we write
  877.  * a dummy log record to force it, so that we ensure that
  878.  * all tmp names are unique.
  879.  *
  880.  * In addition, the name passed may contain an env-relative path.
  881.  * In that case, put the __db. in the right place (in the last
  882.  * component of the pathname).
  883.  */
  884. if (txn != NULL) {
  885. if (IS_ZERO_LSN(txn->last_lsn)) {
  886. /*
  887.  * Write dummy log record.   The two choices for
  888.  * dummy log records are __db_noop_log and
  889.  * __db_debug_log; unfortunately __db_noop_log requires
  890.  * a valid dbp, and we aren't guaranteed to be able
  891.  * to pass one in here.
  892.  */
  893. if ((ret = __db_debug_log(dbenv, txn, &lsn, 0,
  894.     NULL, 0, NULL, NULL, 0)) != 0)
  895. return (ret);
  896. } else
  897. lsn = txn->last_lsn;
  898. }
  899. /*
  900.  * Part of the name may be a full path, so we need to make sure that
  901.  * we allocate enough space for it, even in the case where we don't
  902.  * use the entire filename for the backup name.
  903.  */
  904. len = strlen(name) + strlen(BACKUP_PREFIX) + MAX_LSN_TO_TEXT;
  905.     
  906. if ((ret = __os_malloc(dbenv, len, &retp)) != 0)
  907. return (ret);
  908. /*
  909.  * There are four cases here:
  910.  * 1. simple path w/out transaction
  911.  * 2. simple path + transaction
  912.  * 3. multi-component path w/out transaction
  913.  * 4. multi-component path + transaction
  914.  */
  915. if ((p = __db_rpath(name)) == NULL) {
  916. if (txn == NULL) /* case 1 */
  917. snprintf(retp, len, "%s%s.", BACKUP_PREFIX, name);
  918. else /* case 2 */
  919. snprintf(retp, len,
  920.     "%s%x.%x", BACKUP_PREFIX, lsn.file, lsn.offset);
  921. } else {
  922. plen = (int)(p - name) + 1;
  923. p++;
  924. if (txn == NULL) /* case 3 */
  925. snprintf(retp, len,
  926.     "%.*s%s%s.", plen, name, BACKUP_PREFIX, p);
  927. else  /* case 4 */
  928. snprintf(retp, len,
  929.     "%.*s%x.%x.", plen, name, lsn.file, lsn.offset);
  930. }
  931. *backup = retp;
  932. return (0);
  933. }
  934. /*
  935.  * __dblist_get --
  936.  * Get the first element of dbenv->dblist with
  937.  * dbp->adj_fileid matching adjid.
  938.  *
  939.  * PUBLIC: DB *__dblist_get __P((DB_ENV *, u_int32_t));
  940.  */
  941. DB *
  942. __dblist_get(dbenv, adjid)
  943. DB_ENV *dbenv;
  944. u_int32_t adjid;
  945. {
  946. DB *dbp;
  947. for (dbp = LIST_FIRST(&dbenv->dblist);
  948.     dbp != NULL && dbp->adj_fileid != adjid;
  949.     dbp = LIST_NEXT(dbp, dblistlinks))
  950. ;
  951. return (dbp);
  952. }
  953. /*
  954.  * __db_disassociate --
  955.  * Destroy the association between a given secondary and its primary.
  956.  */
  957. static int
  958. __db_disassociate(sdbp)
  959. DB *sdbp;
  960. {
  961. DBC *dbc;
  962. int ret, t_ret;
  963. ret = 0;
  964. sdbp->s_callback = NULL;
  965. sdbp->s_primary = NULL;
  966. sdbp->get = sdbp->stored_get;
  967. sdbp->close = sdbp->stored_close;
  968. /*
  969.  * Complain, but proceed, if we have any active cursors.  (We're in
  970.  * the middle of a close, so there's really no turning back.)
  971.  */
  972. if (sdbp->s_refcnt != 1 ||
  973.     TAILQ_FIRST(&sdbp->active_queue) != NULL ||
  974.     TAILQ_FIRST(&sdbp->join_queue) != NULL) {
  975. __db_err(sdbp->dbenv,
  976.     "Closing a primary DB while a secondary DB has active cursors is unsafe");
  977. ret = EINVAL;
  978. }
  979. sdbp->s_refcnt = 0;
  980. while ((dbc = TAILQ_FIRST(&sdbp->free_queue)) != NULL)
  981. if ((t_ret = __db_c_destroy(dbc)) != 0 && ret == 0)
  982. ret = t_ret;
  983. F_CLR(sdbp, DB_AM_SECONDARY);
  984. return (ret);
  985. }
  986. #if CONFIG_TEST
  987. /*
  988.  * __db_testcopy
  989.  * Create a copy of all backup files and our "main" DB.
  990.  *
  991.  * PUBLIC: #if CONFIG_TEST
  992.  * PUBLIC: int __db_testcopy __P((DB_ENV *, DB *, const char *));
  993.  * PUBLIC: #endif
  994.  */
  995. int
  996. __db_testcopy(dbenv, dbp, name)
  997. DB_ENV *dbenv;
  998. DB *dbp;
  999. const char *name;
  1000. {
  1001. DB_MPOOLFILE *mpf;
  1002. DB_ASSERT(dbp != NULL || name != NULL);
  1003. if (name == NULL) {
  1004. mpf = dbp->mpf;
  1005. name = R_ADDR(mpf->dbmp->reginfo, mpf->mfp->path_off);
  1006. }
  1007. if (dbp != NULL && dbp->type == DB_QUEUE)
  1008. return (__qam_testdocopy(dbp, name));
  1009. else
  1010. return (__db_testdocopy(dbenv, name));
  1011. }
  1012. static int
  1013. __qam_testdocopy(dbp, name)
  1014. DB *dbp;
  1015. const char *name;
  1016. {
  1017. QUEUE_FILELIST *filelist, *fp;
  1018. char buf[256], *dir;
  1019. int ret;
  1020. filelist = NULL;
  1021. if ((ret = __db_testdocopy(dbp->dbenv, name)) != 0)
  1022. return (ret);
  1023. if (dbp->mpf != NULL &&
  1024.     (ret = __qam_gen_filelist(dbp, &filelist)) != 0)
  1025. return (ret);
  1026. if (filelist == NULL)
  1027. return (0);
  1028. dir = ((QUEUE *)dbp->q_internal)->dir;
  1029. for (fp = filelist; fp->mpf != NULL; fp++) {
  1030. snprintf(buf, sizeof(buf),
  1031.     QUEUE_EXTENT, dir, PATH_SEPARATOR[0], name, fp->id);
  1032. if ((ret = __db_testdocopy(dbp->dbenv, buf)) != 0)
  1033. return (ret);
  1034. }
  1035. __os_free(dbp->dbenv, filelist);
  1036. return (0);
  1037. }
  1038. /*
  1039.  * __db_testdocopy
  1040.  * Create a copy of all backup files and our "main" DB.
  1041.  *
  1042.  */
  1043. static int
  1044. __db_testdocopy(dbenv, name)
  1045. DB_ENV *dbenv;
  1046. const char *name;
  1047. {
  1048. size_t len;
  1049. int dircnt, i, ret;
  1050. char **namesp, *backup, *copy, *dir, *p, *real_name;
  1051. real_name = NULL;
  1052. /* Get the real backing file name. */
  1053. if ((ret = __db_appname(dbenv,
  1054.     DB_APP_DATA, name, 0, NULL, &real_name)) != 0)
  1055. return (ret);
  1056. copy = backup = NULL;
  1057. namesp = NULL;
  1058. /*
  1059.  * Maximum size of file, including adding a ".afterop".
  1060.  */
  1061. len = strlen(real_name) + strlen(BACKUP_PREFIX) + MAX_LSN_TO_TEXT + 9;
  1062. if ((ret = __os_malloc(dbenv, len, &copy)) != 0)
  1063. goto out;
  1064. if ((ret = __os_malloc(dbenv, len, &backup)) != 0)
  1065. goto out;
  1066. /*
  1067.  * First copy the file itself.
  1068.  */
  1069. snprintf(copy, len, "%s.afterop", real_name);
  1070. __db_makecopy(real_name, copy);
  1071. if ((ret = __os_strdup(dbenv, real_name, &dir)) != 0)
  1072. goto out;
  1073. __os_free(dbenv, real_name);
  1074. real_name = NULL;
  1075. /*
  1076.  * Create the name.  Backup file names are of the form:
  1077.  *
  1078.  * __db.name.0x[lsn-file].0x[lsn-offset]
  1079.  *
  1080.  * which guarantees uniqueness.  We want to look for the
  1081.  * backup name, followed by a '.0x' (so that if they have
  1082.  * files named, say, 'a' and 'abc' we won't match 'abc' when
  1083.  * looking for 'a'.
  1084.  */
  1085. snprintf(backup, len, "%s%s.0x", BACKUP_PREFIX, name);
  1086. /*
  1087.  * We need the directory path to do the __os_dirlist.
  1088.  */
  1089. p = __db_rpath(dir);
  1090. if (p != NULL)
  1091. *p = '';
  1092. ret = __os_dirlist(dbenv, dir, &namesp, &dircnt);
  1093. #if DIAGNOSTIC
  1094. /*
  1095.  * XXX
  1096.  * To get the memory guard code to work because it uses strlen and we
  1097.  * just moved the end of the string somewhere sooner.  This causes the
  1098.  * guard code to fail because it looks at one byte past the end of the
  1099.  * string.
  1100.  */
  1101. *p = '/';
  1102. #endif
  1103. __os_free(dbenv, dir);
  1104. if (ret != 0)
  1105. goto out;
  1106. for (i = 0; i < dircnt; i++) {
  1107. /*
  1108.  * Need to check if it is a backup file for this.
  1109.  * No idea what namesp[i] may be or how long, so
  1110.  * must use strncmp and not memcmp.  We don't want
  1111.  * to use strcmp either because we are only matching
  1112.  * the first part of the real file's name.  We don't
  1113.  * know its LSN's.
  1114.  */
  1115. if (strncmp(namesp[i], backup, strlen(backup)) == 0) {
  1116. if ((ret = __db_appname(dbenv, DB_APP_DATA,
  1117.     namesp[i], 0, NULL, &real_name)) != 0)
  1118. goto out;
  1119. /*
  1120.  * This should not happen.  Check that old
  1121.  * .afterop files aren't around.
  1122.  * If so, just move on.
  1123.  */
  1124. if (strstr(real_name, ".afterop") != NULL) {
  1125. __os_free(dbenv, real_name);
  1126. real_name = NULL;
  1127. continue;
  1128. }
  1129. snprintf(copy, len, "%s.afterop", real_name);
  1130. __db_makecopy(real_name, copy);
  1131. __os_free(dbenv, real_name);
  1132. real_name = NULL;
  1133. }
  1134. }
  1135. out:
  1136. if (backup != NULL)
  1137. __os_free(dbenv, backup);
  1138. if (copy != NULL)
  1139. __os_free(dbenv, copy);
  1140. if (namesp != NULL)
  1141. __os_dirfree(dbenv, namesp, dircnt);
  1142. if (real_name != NULL)
  1143. __os_free(dbenv, real_name);
  1144. return (ret);
  1145. }
  1146. static void
  1147. __db_makecopy(src, dest)
  1148. const char *src, *dest;
  1149. {
  1150. DB_FH rfh, wfh;
  1151. size_t rcnt, wcnt;
  1152. char *buf;
  1153. memset(&rfh, 0, sizeof(rfh));
  1154. memset(&wfh, 0, sizeof(wfh));
  1155. if (__os_malloc(NULL, 1024, &buf) != 0)
  1156. return;
  1157. if (__os_open(NULL,
  1158.     src, DB_OSO_RDONLY, __db_omode("rw----"), &rfh) != 0)
  1159. goto err;
  1160. if (__os_open(NULL, dest,
  1161.     DB_OSO_CREATE | DB_OSO_TRUNC, __db_omode("rw----"), &wfh) != 0)
  1162. goto err;
  1163. for (;;)
  1164. if (__os_read(NULL, &rfh, buf, 1024, &rcnt) < 0 || rcnt == 0 ||
  1165.     __os_write(NULL, &wfh, buf, rcnt, &wcnt) < 0)
  1166. break;
  1167. err: __os_free(NULL, buf);
  1168. if (F_ISSET(&rfh, DB_FH_VALID))
  1169. __os_closehandle(NULL, &rfh);
  1170. if (F_ISSET(&wfh, DB_FH_VALID))
  1171. __os_closehandle(NULL, &wfh);
  1172. }
  1173. #endif