/*- * See the file LICENSE for redistribution information. * * Copyright (c) 1997, 1998, 1999, 2000 * Sleepycat Software. All rights reserved. */ #include "db_config.h" #ifndef lint static const char revid[] = "$Id: bt_recno.c,v 11.65 2001/01/18 14:33:22 bostic Exp $"; #endif /* not lint */ #ifndef NO_SYSTEM_INCLUDES #include #include #include #endif #include "db_int.h" #include "db_page.h" #include "btree.h" #include "db_ext.h" #include "db_shash.h" #include "lock.h" #include "lock_ext.h" #include "qam.h" #include "txn.h" static int __ram_add __P((DBC *, db_recno_t *, DBT *, u_int32_t, u_int32_t)); static int __ram_delete __P((DB *, DB_TXN *, DBT *, u_int32_t)); static int __ram_put __P((DB *, DB_TXN *, DBT *, DBT *, u_int32_t)); static int __ram_source __P((DB *)); static int __ram_sread __P((DBC *, db_recno_t)); static int __ram_update __P((DBC *, db_recno_t, int)); /* * In recno, there are two meanings to the on-page "deleted" flag. If we're * re-numbering records, it means the record was implicitly created. We skip * over implicitly created records if doing a cursor "next" or "prev", and * return DB_KEYEMPTY if they're explicitly requested.. If not re-numbering * records, it means that the record was implicitly created, or was deleted. * We skip over implicitly created or deleted records if doing a cursor "next" * or "prev", and return DB_KEYEMPTY if they're explicitly requested. * * If we're re-numbering records, then we have to detect in the cursor that * a record was deleted, and adjust the cursor as necessary on the next get. * If we're not re-numbering records, then we can detect that a record has * been deleted by looking at the actual on-page record, so we completely * ignore the cursor's delete flag. This is different from the B+tree code. * It also maintains whether the cursor references a deleted record in the * cursor, and it doesn't always check the on-page value. */ #define CD_SET(cp) { \ if (F_ISSET(cp, C_RENUMBER)) \ F_SET(cp, C_DELETED); \ } #define CD_CLR(cp) { \ if (F_ISSET(cp, C_RENUMBER)) { \ F_CLR(cp, C_DELETED); \ cp->order = INVALID_ORDER; \ } \ } #define CD_ISSET(cp) \ (F_ISSET(cp, C_RENUMBER) && F_ISSET(cp, C_DELETED)) /* * Macros for comparing the ordering of two cursors. * cp1 comes before cp2 iff one of the following holds: * cp1's recno is less than cp2's recno * recnos are equal, both deleted, and cp1's order is less than cp2's * recnos are equal, cp1 deleted, and cp2 not deleted */ #define C_LESSTHAN(cp1, cp2) \ (((cp1)->recno < (cp2)->recno) || \ (((cp1)->recno == (cp2)->recno) && \ ((CD_ISSET((cp1)) && CD_ISSET((cp2)) && (cp1)->order < (cp2)->order) || \ (CD_ISSET((cp1)) && !CD_ISSET((cp2)))))) /* * cp1 is equal to cp2 iff their recnos and delete flags are identical, * and if the delete flag is set their orders are also identical. */ #define C_EQUAL(cp1, cp2) \ ((cp1)->recno == (cp2)->recno && CD_ISSET((cp1)) == CD_ISSET((cp2)) && \ (!CD_ISSET((cp1)) || (cp1)->order == (cp2)->order)) /* * Do we need to log the current cursor adjustment? */ #define CURADJ_LOG(dbc) \ (DB_LOGGING((dbc)) && (dbc)->txn != NULL && (dbc)->txn->parent != NULL) /* * __ram_open -- * Recno open function. * * PUBLIC: int __ram_open __P((DB *, const char *, db_pgno_t, u_int32_t)); */ int __ram_open(dbp, name, base_pgno, flags) DB *dbp; const char *name; db_pgno_t base_pgno; u_int32_t flags; { BTREE *t; DBC *dbc; int ret, t_ret; t = dbp->bt_internal; /* Initialize the remaining fields/methods of the DB. */ dbp->del = __ram_delete; dbp->put = __ram_put; dbp->stat = __bam_stat; /* Start up the tree. */ if ((ret = __bam_read_root(dbp, name, base_pgno, flags)) != 0) return (ret); /* * If the user specified a source tree, open it and map it in. * * !!! * We don't complain if the user specified transactions or threads. * It's possible to make it work, but you'd better know what you're * doing! */ if (t->re_source != NULL && (ret = __ram_source(dbp)) != 0) return (ret); /* If we're snapshotting an underlying source file, do it now. */ if (F_ISSET(dbp, DB_RE_SNAPSHOT)) { /* Allocate a cursor. */ if ((ret = dbp->cursor(dbp, NULL, &dbc, 0)) != 0) return (ret); /* Do the snapshot. */ if ((ret = __ram_update(dbc, DB_MAX_RECORDS, 0)) != 0 && ret == DB_NOTFOUND) ret = 0; /* Discard the cursor. */ if ((t_ret = dbc->c_close(dbc)) != 0 && ret == 0) ret = t_ret; } return (0); } /* * __ram_delete -- * Recno db->del function. */ static int __ram_delete(dbp, txn, key, flags) DB *dbp; DB_TXN *txn; DBT *key; u_int32_t flags; { BTREE_CURSOR *cp; DBC *dbc; db_recno_t recno; int ret, t_ret; PANIC_CHECK(dbp->dbenv); /* Check for invalid flags. */ if ((ret = __db_delchk(dbp, key, flags, F_ISSET(dbp, DB_AM_RDONLY))) != 0) return (ret); /* Acquire a cursor. */ if ((ret = dbp->cursor(dbp, txn, &dbc, DB_WRITELOCK)) != 0) return (ret); DEBUG_LWRITE(dbc, txn, "ram_delete", key, NULL, flags); /* Check the user's record number and fill in as necessary. */ if ((ret = __ram_getno(dbc, key, &recno, 0)) != 0) goto err; /* Do the delete. */ cp = (BTREE_CURSOR *)dbc->internal; cp->recno = recno; ret = __ram_c_del(dbc); /* Release the cursor. */ err: if ((t_ret = dbc->c_close(dbc)) != 0 && ret == 0) ret = t_ret; return (ret); } /* * __ram_put -- * Recno db->put function. */ static int __ram_put(dbp, txn, key, data, flags) DB *dbp; DB_TXN *txn; DBT *key, *data; u_int32_t flags; { DBC *dbc; db_recno_t recno; int ret, t_ret; PANIC_CHECK(dbp->dbenv); /* Check for invalid flags. */ if ((ret = __db_putchk(dbp, key, data, flags, F_ISSET(dbp, DB_AM_RDONLY), 0)) != 0) return (ret); /* Allocate a cursor. */ if ((ret = dbp->cursor(dbp, txn, &dbc, DB_WRITELOCK)) != 0) return (ret); DEBUG_LWRITE(dbc, txn, "ram_put", key, data, flags); /* * If we're appending to the tree, make sure we've read in all of * the backing source file. Otherwise, check the user's record * number and fill in as necessary. If we found the record or it * simply didn't exist, add the user's record. */ if (flags == DB_APPEND) ret = __ram_update(dbc, DB_MAX_RECORDS, 0); else ret = __ram_getno(dbc, key, &recno, 1); if (ret == 0 || ret == DB_NOTFOUND) ret = __ram_add(dbc, &recno, data, flags, 0); /* Discard the cursor. */ if ((t_ret = dbc->c_close(dbc)) != 0 && ret == 0) ret = t_ret; /* Return the record number if we're appending to the tree. */ if (ret == 0 && flags == DB_APPEND) ret = __db_retcopy(dbp, key, &recno, sizeof(recno), &dbc->rkey.data, &dbc->rkey.ulen); return (ret); } /* * __ram_c_del -- * Recno cursor->c_del function. * * PUBLIC: int __ram_c_del __P((DBC *)); */ int __ram_c_del(dbc) DBC *dbc; { BKEYDATA bk; BTREE *t; BTREE_CURSOR *cp; DB *dbp; DB_LSN lsn; DBT hdr, data; EPG *epg; int exact, ret, stack; dbp = dbc->dbp; cp = (BTREE_CURSOR *)dbc->internal; t = dbp->bt_internal; stack = 0; /* * The semantics of cursors during delete are as follows: in * non-renumbering recnos, records are replaced with a marker * containing a delete flag. If the record referenced by this cursor * has already been deleted, we will detect that as part of the delete * operation, and fail. * * In renumbering recnos, cursors which represent deleted items * are flagged with the C_DELETED flag, and it is an error to * call c_del a second time without an intervening cursor motion. */ if (CD_ISSET(cp)) return (DB_KEYEMPTY); /* Search the tree for the key; delete only deletes exact matches. */ if ((ret = __bam_rsearch(dbc, &cp->recno, S_DELETE, 1, &exact)) != 0) goto err; if (!exact) { ret = DB_NOTFOUND; goto err; } stack = 1; cp->page = cp->csp->page; cp->pgno = cp->csp->page->pgno; cp->indx = cp->csp->indx; /* * If re-numbering records, the on-page deleted flag can only mean * that this record was implicitly created. Applications aren't * permitted to delete records they never created, return an error. * * If not re-numbering records, the on-page deleted flag means that * this record was implicitly created, or, was deleted at some time. * The former is an error because applications aren't permitted to * delete records they never created, the latter is an error because * if the record was "deleted", we could never have found it. */ if (B_DISSET(GET_BKEYDATA(cp->page, cp->indx)->type)) { ret = DB_KEYEMPTY; goto err; } if (F_ISSET(cp, C_RENUMBER)) { /* Delete the item, adjust the counts, adjust the cursors. */ if ((ret = __bam_ditem(dbc, cp->page, cp->indx)) != 0) goto err; __bam_adjust(dbc, -1); if (__ram_ca(dbc, CA_DELETE) > 0 && CURADJ_LOG(dbc) && (ret = __bam_rcuradj_log(dbp->dbenv, dbc->txn, &lsn, 0, dbp->log_fileid, CA_DELETE, cp->root, cp->recno, cp->order)) != 0) goto err; /* * If the page is empty, delete it. * * We never delete a root page. First, root pages of primary * databases never go away, recno or otherwise. However, if * it's the root page of an off-page duplicates database, then * it can be deleted. We don't delete it here because we have * no way of telling the primary database page holder (e.g., * the hash access method) that its page element should cleaned * up because the underlying tree is gone. So, we keep the page * around until the last cursor referencing the empty tree is * are closed, and then clean it up. */ if (NUM_ENT(cp->page) == 0 && PGNO(cp->page) != cp->root) { /* * We already have a locked stack of pages. However, * there are likely entries in the stack that aren't * going to be emptied by removing the single reference * to the emptied page (or one of its parents). */ for (epg = cp->sp; epg <= cp->csp; ++epg) if (NUM_ENT(epg->page) <= 1) break; /* * We want to delete a single item out of the last page * that we're not deleting, back up to that page. */ ret = __bam_dpages(dbc, --epg); /* * Regardless of the return from __bam_dpages, it will * discard our stack and pinned page. */ stack = 0; cp->page = NULL; } } else { /* Use a delete/put pair to replace the record with a marker. */ if ((ret = __bam_ditem(dbc, cp->page, cp->indx)) != 0) goto err; B_TSET(bk.type, B_KEYDATA, 1); bk.len = 0; memset(&hdr, 0, sizeof(hdr)); hdr.data = &bk; hdr.size = SSZA(BKEYDATA, data); memset(&data, 0, sizeof(data)); data.data = (void *)""; data.size = 0; if ((ret = __db_pitem(dbc, cp->page, cp->indx, BKEYDATA_SIZE(0), &hdr, &data)) != 0) goto err; } t->re_modified = 1; err: if (stack) __bam_stkrel(dbc, STK_CLRDBC); return (ret); } /* * __ram_c_get -- * Recno cursor->c_get function. * * PUBLIC: int __ram_c_get * PUBLIC: __P((DBC *, DBT *, DBT *, u_int32_t, db_pgno_t *)); */ int __ram_c_get(dbc, key, data, flags, pgnop) DBC *dbc; DBT *key, *data; u_int32_t flags; db_pgno_t *pgnop; { BTREE_CURSOR *cp; DB *dbp; int cmp, exact, ret; COMPQUIET(pgnop, NULL); dbp = dbc->dbp; cp = (BTREE_CURSOR *)dbc->internal; retry: switch (flags) { case DB_CURRENT: /* * If we're using mutable records and the deleted flag is * set, the cursor is pointing at a nonexistent record; * return an error. */ if (CD_ISSET(cp)) return (DB_KEYEMPTY); break; case DB_NEXT_DUP: /* * If we're not in an off-page dup set, we know there's no * next duplicate since recnos don't have them. If we * are in an off-page dup set, the next item assuredly is * a dup, so we set flags to DB_NEXT and keep going. */ if (!F_ISSET(dbc, DBC_OPD)) return (DB_NOTFOUND); /* FALLTHROUGH */ case DB_NEXT_NODUP: /* * Recno databases don't have duplicates, set flags to DB_NEXT * and keep going. */ /* FALLTHROUGH */ case DB_NEXT: flags = DB_NEXT; /* * If record numbers are mutable: if we just deleted a record, * we have to avoid incrementing the record number so that we * return the right record by virtue of renumbering the tree. */ if (CD_ISSET(cp)) break; if (cp->recno != RECNO_OOB) { ++cp->recno; break; } /* FALLTHROUGH */ case DB_FIRST: flags = DB_NEXT; cp->recno = 1; break; case DB_PREV_NODUP: /* * Recno databases don't have duplicates, set flags to DB_PREV * and keep going. */ /* FALLTHROUGH */ case DB_PREV: flags = DB_PREV; if (cp->recno != RECNO_OOB) { if (cp->recno == 1) { ret = DB_NOTFOUND; goto err; } --cp->recno; break; } /* FALLTHROUGH */ case DB_LAST: flags = DB_PREV; if (((ret = __ram_update(dbc, DB_MAX_RECORDS, 0)) != 0) && ret != DB_NOTFOUND) goto err; if ((ret = __bam_nrecs(dbc, &cp->recno)) != 0) goto err; if (cp->recno == 0) { ret = DB_NOTFOUND; goto err; } break; case DB_GET_BOTHC: /* * If we're doing a join and these are offpage dups, * we want to keep searching forward from after the * current cursor position. Increment the recno by 1, * then proceed as for a DB_SET. * * Otherwise, we know there are no additional matching * data, as recnos don't have dups. return DB_NOTFOUND. */ if (F_ISSET(dbc, DBC_OPD)) { cp->recno++; break; } ret = DB_NOTFOUND; goto err; /* NOTREACHED */ case DB_GET_BOTH: /* * If we're searching a set of off-page dups, we start * a new linear search from the first record. Otherwise, * we compare the single data item associated with the * requested record for a match. */ if (F_ISSET(dbc, DBC_OPD)) { cp->recno = 1; break; } /* FALLTHROUGH */ case DB_SET: case DB_SET_RANGE: if ((ret = __ram_getno(dbc, key, &cp->recno, 0)) != 0) goto err; break; default: ret = __db_unknown_flag(dbp->dbenv, "__ram_c_get", flags); goto err; } /* * For DB_PREV, DB_LAST, DB_SET and DB_SET_RANGE, we have already * called __ram_update() to make sure sufficient records have been * read from the backing source file. Do it now for DB_CURRENT (if * the current record was deleted we may need more records from the * backing file for a DB_CURRENT operation), DB_FIRST and DB_NEXT. */ if ((flags == DB_NEXT || flags == DB_CURRENT) && ((ret = __ram_update(dbc, cp->recno, 0)) != 0) && ret != DB_NOTFOUND) goto err; for (;; ++cp->recno) { /* Search the tree for the record. */ if ((ret = __bam_rsearch(dbc, &cp->recno, F_ISSET(dbc, DBC_RMW) ? S_FIND_WR : S_FIND, 1, &exact)) != 0) goto err; if (!exact) { ret = DB_NOTFOUND; goto err; } /* * Copy the page into the cursor, discarding any lock we * are currently holding. */ cp->page = cp->csp->page; cp->pgno = cp->csp->page->pgno; cp->indx = cp->csp->indx; (void)__TLPUT(dbc, cp->lock); cp->lock = cp->csp->lock; cp->lock_mode = cp->csp->lock_mode; /* * If re-numbering records, the on-page deleted flag means this * record was implicitly created. If not re-numbering records, * the on-page deleted flag means this record was implicitly * created, or, it was deleted at some time. Regardless, we * skip such records if doing cursor next/prev operations or * walking through off-page duplicates, and fail if they were * requested explicitly by the application. */ if (B_DISSET(GET_BKEYDATA(cp->page, cp->indx)->type)) switch (flags) { case DB_NEXT: case DB_PREV: (void)__bam_stkrel(dbc, STK_CLRDBC); goto retry; case DB_GET_BOTH: (void)__bam_stkrel(dbc, STK_CLRDBC); continue; default: ret = DB_KEYEMPTY; goto err; } if (flags == DB_GET_BOTH || flags == DB_GET_BOTHC) { if ((ret = __bam_cmp(dbp, data, cp->page, cp->indx, __bam_defcmp, &cmp)) != 0) return (ret); if (cmp == 0) break; if (!F_ISSET(dbc, DBC_OPD)) { ret = DB_NOTFOUND; goto err; } (void)__bam_stkrel(dbc, STK_CLRDBC); } else break; } /* Return the key if the user didn't give us one. */ if (!F_ISSET(dbc, DBC_OPD)) { if (flags != DB_SET && flags != DB_SET_RANGE) ret = __db_retcopy(dbp, key, &cp->recno, sizeof(cp->recno), &dbc->rkey.data, &dbc->rkey.ulen); F_SET(key, DB_DBT_ISSET); } /* The cursor was reset, no further delete adjustment is necessary. */ err: CD_CLR(cp); return (ret); } /* * __ram_c_put -- * Recno cursor->c_put function. * * PUBLIC: int __ram_c_put __P((DBC *, DBT *, DBT *, u_int32_t, db_pgno_t *)); */ int __ram_c_put(dbc, key, data, flags, pgnop) DBC *dbc; DBT *key, *data; u_int32_t flags; db_pgno_t *pgnop; { BTREE_CURSOR *cp; DB *dbp; DB_LSN lsn; int exact, nc, ret, t_ret; u_int32_t iiflags; void *arg; COMPQUIET(pgnop, NULL); dbp = dbc->dbp; cp = (BTREE_CURSOR *)dbc->internal; /* * DB_KEYFIRST and DB_KEYLAST will only be set if we're dealing with * an off-page duplicate tree, they can't be specified at user level. * Translate them into something else. */ switch (flags) { case DB_KEYFIRST: cp->recno = 1; flags = DB_BEFORE; break; case DB_KEYLAST: if ((ret = __ram_add(dbc, &cp->recno, data, DB_APPEND, 0)) != 0) return (ret); if (CURADJ_LOG(dbc) && (ret = __bam_rcuradj_log(dbp->dbenv, dbc->txn, &lsn, 0, dbp->log_fileid, CA_ICURRENT, cp->root, cp->recno, cp->order))) return (ret); return (0); } /* * If we're putting with a cursor that's marked C_DELETED, we need to * take special care; the cursor doesn't "really" reference the item * corresponding to its current recno, but instead is "between" that * record and the current one. Translate the actual insert into * DB_BEFORE, and let the __ram_ca work out the gory details of what * should wind up pointing where. */ if (CD_ISSET(cp)) iiflags = DB_BEFORE; else iiflags = flags; split: if ((ret = __bam_rsearch(dbc, &cp->recno, S_INSERT, 1, &exact)) != 0) goto err; /* * An inexact match is okay; it just means we're one record past the * end, which is reasonable if we're marked deleted. */ DB_ASSERT(exact || CD_ISSET(cp)); cp->page = cp->csp->page; cp->pgno = cp->csp->page->pgno; cp->indx = cp->csp->indx; ret = __bam_iitem(dbc, key, data, iiflags, 0); t_ret = __bam_stkrel(dbc, STK_CLRDBC); if (t_ret != 0 && (ret == 0 || ret == DB_NEEDSPLIT)) ret = t_ret; else if (ret == DB_NEEDSPLIT) { arg = &cp->recno; if ((ret = __bam_split(dbc, arg)) != 0) goto err; goto split; } if (ret != 0) goto err; switch (flags) { /* Adjust the cursors. */ case DB_AFTER: nc = __ram_ca(dbc, CA_IAFTER); /* * We only need to adjust this cursor forward if we truly added * the item after the current recno, rather than remapping it * to DB_BEFORE. */ if (iiflags == DB_AFTER) ++cp->recno; /* Only log if __ram_ca found any relevant cursors. */ if (nc > 0 && CURADJ_LOG(dbc) && (ret = __bam_rcuradj_log(dbp->dbenv, dbc->txn, &lsn, 0, dbp->log_fileid, CA_IAFTER, cp->root, cp->recno, cp->order)) != 0) goto err; break; case DB_BEFORE: nc = __ram_ca(dbc, CA_IBEFORE); --cp->recno; /* Only log if __ram_ca found any relevant cursors. */ if (nc > 0 && CURADJ_LOG(dbc) && (ret = __bam_rcuradj_log(dbp->dbenv, dbc->txn, &lsn, 0, dbp->log_fileid, CA_IBEFORE, cp->root, cp->recno, cp->order)) != 0) goto err; break; case DB_CURRENT: /* * We only need to do an adjustment if we actually * added an item, which we only would have done if the * cursor was marked deleted. * * Only log if __ram_ca found any relevant cursors. */ if (CD_ISSET(cp) && __ram_ca(dbc, CA_ICURRENT) > 0 && CURADJ_LOG(dbc) && (ret = __bam_rcuradj_log( dbp->dbenv, dbc->txn, &lsn, 0, dbp->log_fileid, CA_ICURRENT, cp->root, cp->recno, cp->order)) != 0) goto err; break; } /* Return the key if we've created a new record. */ if (!F_ISSET(dbc, DBC_OPD) && (flags == DB_AFTER || flags == DB_BEFORE)) ret = __db_retcopy(dbp, key, &cp->recno, sizeof(cp->recno), &dbc->rkey.data, &dbc->rkey.ulen); /* The cursor was reset, no further delete adjustment is necessary. */ err: CD_CLR(cp); return (ret); } /* * __ram_ca -- * Adjust cursors. Returns the number of relevant cursors. * * PUBLIC: int __ram_ca __P((DBC *, ca_recno_arg)); */ int __ram_ca(dbc_arg, op) DBC *dbc_arg; ca_recno_arg op; { BTREE_CURSOR *cp, *cp_arg; DB *dbp, *ldbp; DB_ENV *dbenv; DBC *dbc; db_recno_t recno; int adjusted, found; u_int32_t order; dbp = dbc_arg->dbp; dbenv = dbp->dbenv; cp_arg = (BTREE_CURSOR *)dbc_arg->internal; recno = cp_arg->recno; found = 0; /* * It only makes sense to adjust cursors if we're a renumbering * recno; we should only be called if this is one. */ DB_ASSERT(F_ISSET(cp_arg, C_RENUMBER)); MUTEX_THREAD_LOCK(dbenv, dbenv->dblist_mutexp); /* * Adjust the cursors. See the comment in __bam_ca_delete(). */ /* * If we're doing a delete, we need to find the highest * order of any cursor currently pointing at this item, * so we can assign a higher order to the newly deleted * cursor. Unfortunately, this requires a second pass through * the cursor list. */ if (op == CA_DELETE) { order = 1; for (ldbp = __dblist_get(dbenv, dbp->adj_fileid); ldbp != NULL && ldbp->adj_fileid == dbp->adj_fileid; ldbp = LIST_NEXT(ldbp, dblistlinks)) { MUTEX_THREAD_LOCK(dbenv, dbp->mutexp); for (dbc = TAILQ_FIRST(&ldbp->active_queue); dbc != NULL; dbc = TAILQ_NEXT(dbc, links)) { cp = (BTREE_CURSOR *)dbc->internal; if (cp_arg->root == cp->root && recno == cp->recno && CD_ISSET(cp) && order <= cp->order) order = cp->order + 1; } MUTEX_THREAD_UNLOCK(dbenv, dbp->mutexp); } } else order = INVALID_ORDER; /* Now go through and do the actual adjustments. */ for (ldbp = __dblist_get(dbenv, dbp->adj_fileid); ldbp != NULL && ldbp->adj_fileid == dbp->adj_fileid; ldbp = LIST_NEXT(ldbp, dblistlinks)) { MUTEX_THREAD_LOCK(dbenv, dbp->mutexp); for (dbc = TAILQ_FIRST(&ldbp->active_queue); dbc != NULL; dbc = TAILQ_NEXT(dbc, links)) { cp = (BTREE_CURSOR *)dbc->internal; if (cp_arg->root != cp->root) continue; ++found; adjusted = 0; switch (op) { case CA_DELETE: if (recno < cp->recno) { --cp->recno; /* * If the adjustment made them equal, * we have to merge the orders. */ if (recno == cp->recno && CD_ISSET(cp)) cp->order += order; } else if (recno == cp->recno && !CD_ISSET(cp)) { CD_SET(cp); cp->order = order; } break; case CA_IBEFORE: /* * IBEFORE is just like IAFTER, except that we * adjust cursors on the current record too. */ if (C_EQUAL(cp_arg, cp)) { ++cp->recno; adjusted = 1; } goto iafter; case CA_ICURRENT: /* * If the original cursor wasn't deleted, we * just did a replacement and so there's no * need to adjust anything--we shouldn't have * gotten this far. Otherwise, we behave * much like an IAFTER, except that all * cursors pointing to the current item get * marked undeleted and point to the new * item. */ DB_ASSERT(CD_ISSET(cp_arg)); if (C_EQUAL(cp_arg, cp)) { CD_CLR(cp); break; } /* FALLTHROUGH */ case CA_IAFTER: iafter: if (!adjusted && C_LESSTHAN(cp_arg, cp)) { ++cp->recno; adjusted = 1; } if (recno == cp->recno && adjusted) /* * If we've moved this cursor's recno, * split its order number--i.e., * decrement it by enough so that * the lowest cursor moved has order 1. * cp_arg->order is the split point, * so decrement by one less than that. */ cp->order -= (cp_arg->order - 1); break; } } MUTEX_THREAD_UNLOCK(dbp->dbenv, dbp->mutexp); } MUTEX_THREAD_UNLOCK(dbenv, dbenv->dblist_mutexp); return (found); } /* * __ram_getno -- * Check the user's record number, and make sure we've seen it. * * PUBLIC: int __ram_getno __P((DBC *, const DBT *, db_recno_t *, int)); */ int __ram_getno(dbc, key, rep, can_create) DBC *dbc; const DBT *key; db_recno_t *rep; int can_create; { DB *dbp; db_recno_t recno; dbp = dbc->dbp; /* Check the user's record number. */ if ((recno = *(db_recno_t *)key->data) == 0) { __db_err(dbp->dbenv, "illegal record number of 0"); return (EINVAL); } if (rep != NULL) *rep = recno; /* * Btree can neither create records nor read them in. Recno can * do both, see if we can find the record. */ return (dbc->dbtype == DB_RECNO ? __ram_update(dbc, recno, can_create) : 0); } /* * __ram_update -- * Ensure the tree has records up to and including the specified one. */ static int __ram_update(dbc, recno, can_create) DBC *dbc; db_recno_t recno; int can_create; { BTREE *t; BTREE_CURSOR *cp; DB *dbp; db_recno_t nrecs; int ret; dbp = dbc->dbp; cp = (BTREE_CURSOR *)dbc->internal; t = dbp->bt_internal; /* * If we can't create records and we've read the entire backing input * file, we're done. */ if (!can_create && t->re_eof) return (0); /* * If we haven't seen this record yet, try to get it from the original * file. */ if ((ret = __bam_nrecs(dbc, &nrecs)) != 0) return (ret); if (!t->re_eof && recno > nrecs) { if ((ret = __ram_sread(dbc, recno)) != 0 && ret != DB_NOTFOUND) return (ret); if ((ret = __bam_nrecs(dbc, &nrecs)) != 0) return (ret); } /* * If we can create records, create empty ones up to the requested * record. */ if (!can_create || recno <= nrecs + 1) return (0); dbc->rdata.dlen = 0; dbc->rdata.doff = 0; dbc->rdata.flags = 0; if (F_ISSET(dbp, DB_RE_FIXEDLEN)) { if (dbc->rdata.ulen < t->re_len) { if ((ret = __os_realloc(dbp->dbenv, t->re_len, NULL, &dbc->rdata.data)) != 0) { dbc->rdata.ulen = 0; dbc->rdata.data = NULL; return (ret); } dbc->rdata.ulen = t->re_len; } dbc->rdata.size = t->re_len; memset(dbc->rdata.data, t->re_pad, t->re_len); } else dbc->rdata.size = 0; while (recno > ++nrecs) if ((ret = __ram_add(dbc, &nrecs, &dbc->rdata, 0, BI_DELETED)) != 0) return (ret); return (0); } /* * __ram_source -- * Load information about the backing file. */ static int __ram_source(dbp) DB *dbp; { BTREE *t; char *source; int ret; t = dbp->bt_internal; /* Find the real name, and swap out the one we had before. */ if ((ret = __db_appname(dbp->dbenv, DB_APP_DATA, NULL, t->re_source, 0, NULL, &source)) != 0) return (ret); __os_freestr(t->re_source); t->re_source = source; /* * !!! * It's possible that the backing source file is read-only. We don't * much care other than we'll complain if there are any modifications * when it comes time to write the database back to the source. */ if ((t->re_fp = fopen(t->re_source, "r")) == NULL) { ret = errno; __db_err(dbp->dbenv, "%s: %s", t->re_source, db_strerror(ret)); return (ret); } t->re_eof = 0; return (0); } /* * __ram_writeback -- * Rewrite the backing file. * * PUBLIC: int __ram_writeback __P((DB *)); */ int __ram_writeback(dbp) DB *dbp; { BTREE *t; DB_ENV *dbenv; DBC *dbc; DBT key, data; FILE *fp; db_recno_t keyno; int ret, t_ret; u_int8_t delim, *pad; t = dbp->bt_internal; dbenv = dbp->dbenv; fp = NULL; /* If the file wasn't modified, we're done. */ if (!t->re_modified) return (0); /* If there's no backing source file, we're done. */ if (t->re_source == NULL) { t->re_modified = 0; return (0); } /* Allocate a cursor. */ if ((ret = dbp->cursor(dbp, NULL, &dbc, 0)) != 0) return (ret); /* * Read any remaining records into the tree. * * !!! * This is why we can't support transactions when applications specify * backing (re_source) files. At this point we have to read in the * rest of the records from the file so that we can write all of the * records back out again, which could modify a page for which we'd * have to log changes and which we don't have locked. This could be * partially fixed by taking a snapshot of the entire file during the * DB->open as DB->open is transaction protected. But, if a checkpoint * occurs then, the part of the log holding the copy of the file could * be discarded, and that would make it impossible to recover in the * face of disaster. This could all probably be fixed, but it would * require transaction protecting the backing source file. * * XXX * This could be made to work now that we have transactions protecting * file operations. Margo has specifically asked for the privilege of * doing this work. */ if ((ret = __ram_update(dbc, DB_MAX_RECORDS, 0)) != 0 && ret != DB_NOTFOUND) return (ret); /* * Close any existing file handle and re-open the file, truncating it. */ if (t->re_fp != NULL) { if (fclose(t->re_fp) != 0) { ret = errno; goto err; } t->re_fp = NULL; } if ((fp = fopen(t->re_source, "w")) == NULL) { ret = errno; __db_err(dbenv, "%s: %s", t->re_source, db_strerror(ret)); goto err; } /* * We step through the records, writing each one out. Use the record * number and the dbp->get() function, instead of a cursor, so we find * and write out "deleted" or non-existent records. */ memset(&key, 0, sizeof(key)); memset(&data, 0, sizeof(data)); key.size = sizeof(db_recno_t); key.data = &keyno; /* * We'll need the delimiter if we're doing variable-length records, * and the pad character if we're doing fixed-length records. */ delim = t->re_delim; if (F_ISSET(dbp, DB_RE_FIXEDLEN)) { if ((ret = __os_malloc(dbenv, t->re_len, NULL, &pad)) != 0) goto err; memset(pad, t->re_pad, t->re_len); } else COMPQUIET(pad, NULL); for (keyno = 1;; ++keyno) { switch (ret = dbp->get(dbp, NULL, &key, &data, 0)) { case 0: if (fwrite(data.data, 1, data.size, fp) != data.size) goto write_err; break; case DB_KEYEMPTY: if (F_ISSET(dbp, DB_RE_FIXEDLEN) && fwrite(pad, 1, t->re_len, fp) != t->re_len) goto write_err; break; case DB_NOTFOUND: ret = 0; goto done; } if (!F_ISSET(dbp, DB_RE_FIXEDLEN) && fwrite(&delim, 1, 1, fp) != 1) { write_err: ret = errno; __db_err(dbp->dbenv, "%s: write failed to backing file: %s", t->re_source, strerror(ret)); goto err; } } err: done: /* Close the file descriptor. */ if (fp != NULL && fclose(fp) != 0) { if (ret == 0) ret = errno; __db_err(dbenv, "%s: %s", t->re_source, db_strerror(errno)); } /* Discard the cursor. */ if ((t_ret = dbc->c_close(dbc)) != 0 && ret == 0) ret = t_ret; if (ret == 0) t->re_modified = 0; return (ret); } /* * __ram_sread -- * Read records from a source file. */ static int __ram_sread(dbc, top) DBC *dbc; db_recno_t top; { BTREE *t; DB *dbp; DBT data; db_recno_t recno; size_t len; int ch, ret, was_modified; t = dbc->dbp->bt_internal; dbp = dbc->dbp; was_modified = t->re_modified; if ((ret = __bam_nrecs(dbc, &recno)) != 0) return (ret); /* Use the record data return memory, it's only a short-term use. */ len = F_ISSET(dbp, DB_RE_FIXEDLEN) ? t->re_len : 256; if (dbc->rdata.ulen < len) { if ((ret = __os_realloc( dbp->dbenv, len, NULL, &dbc->rdata.data)) != 0) { dbc->rdata.ulen = 0; dbc->rdata.data = NULL; return (ret); } dbc->rdata.ulen = len; } memset(&data, 0, sizeof(data)); while (recno < top) { data.data = dbc->rdata.data; data.size = 0; if (F_ISSET(dbp, DB_RE_FIXEDLEN)) for (len = t->re_len; len > 0; --len) { if ((ch = getc(t->re_fp)) == EOF) goto eof; ((u_int8_t *)data.data)[data.size++] = ch; } else for (;;) { if ((ch = getc(t->re_fp)) == EOF) goto eof; if (ch == t->re_delim) break; ((u_int8_t *)data.data)[data.size++] = ch; if (data.size == dbc->rdata.ulen) { if ((ret = __os_realloc(dbp->dbenv, dbc->rdata.ulen *= 2, NULL, &dbc->rdata.data)) != 0) { dbc->rdata.ulen = 0; dbc->rdata.data = NULL; return (ret); } else data.data = dbc->rdata.data; } } /* * Another process may have read this record from the input * file and stored it into the database already, in which * case we don't need to repeat that operation. We detect * this by checking if the last record we've read is greater * or equal to the number of records in the database. */ if (t->re_last >= recno) { ++recno; if ((ret = __ram_add(dbc, &recno, &data, 0, 0)) != 0) goto err; } ++t->re_last; } if (0) { eof: t->re_eof = 1; ret = DB_NOTFOUND; } err: if (!was_modified) t->re_modified = 0; return (ret); } /* * __ram_add -- * Add records into the tree. */ static int __ram_add(dbc, recnop, data, flags, bi_flags) DBC *dbc; db_recno_t *recnop; DBT *data; u_int32_t flags, bi_flags; { BKEYDATA *bk; BTREE_CURSOR *cp; int exact, ret, stack; cp = (BTREE_CURSOR *)dbc->internal; retry: /* Find the slot for insertion. */ if ((ret = __bam_rsearch(dbc, recnop, S_INSERT | (flags == DB_APPEND ? S_APPEND : 0), 1, &exact)) != 0) return (ret); stack = 1; cp->page = cp->csp->page; cp->pgno = cp->csp->page->pgno; cp->indx = cp->csp->indx; /* * The application may modify the data based on the selected record * number. */ if (flags == DB_APPEND && dbc->dbp->db_append_recno != NULL && (ret = dbc->dbp->db_append_recno(dbc->dbp, data, *recnop)) != 0) goto err; /* * If re-numbering records, the on-page deleted flag means this record * was implicitly created. If not re-numbering records, the on-page * deleted flag means this record was implicitly created, or, it was * deleted at some time. * * If DB_NOOVERWRITE is set and the item already exists in the tree, * return an error unless the item was either marked for deletion or * only implicitly created. */ if (exact) { bk = GET_BKEYDATA(cp->page, cp->indx); if (!B_DISSET(bk->type) && flags == DB_NOOVERWRITE) { ret = DB_KEYEXIST; goto err; } } /* * Select the arguments for __bam_iitem() and do the insert. If the * key is an exact match, or we're replacing the data item with a * new data item, replace the current item. If the key isn't an exact * match, we're inserting a new key/data pair, before the search * location. */ switch (ret = __bam_iitem(dbc, NULL, data, exact ? DB_CURRENT : DB_BEFORE, bi_flags)) { case 0: /* * Don't adjust anything. * * If we inserted a record, no cursors need adjusting because * the only new record it's possible to insert is at the very * end of the tree. The necessary adjustments to the internal * page counts were made by __bam_iitem(). * * If we overwrote a record, no cursors need adjusting because * future DBcursor->get calls will simply return the underlying * record (there's no adjustment made for the DB_CURRENT flag * when a cursor get operation immediately follows a cursor * delete operation, and the normal adjustment for the DB_NEXT * flag is still correct). */ break; case DB_NEEDSPLIT: /* Discard the stack of pages and split the page. */ (void)__bam_stkrel(dbc, STK_CLRDBC); stack = 0; if ((ret = __bam_split(dbc, recnop)) != 0) goto err; goto retry; /* NOTREACHED */ default: goto err; } err: if (stack) __bam_stkrel(dbc, STK_CLRDBC); return (ret); }