db_vrfyutil.c
Upload User: tsgydb
Upload Date: 2007-04-14
Package Size: 10674k
Code Size: 18k
Category:

MySQL

Development Platform:

Visual C++

  1. /*-
  2.  * See the file LICENSE for redistribution information.
  3.  *
  4.  * Copyright (c) 2000
  5.  * Sleepycat Software.  All rights reserved.
  6.  *
  7.  * $Id: db_vrfyutil.c,v 11.11 2000/11/28 21:36:04 bostic Exp $
  8.  */
  9. #include "db_config.h"
  10. #ifndef lint
  11. static const char revid[] = "$Id: db_vrfyutil.c,v 11.11 2000/11/28 21:36:04 bostic Exp $";
  12. #endif /* not lint */
  13. #ifndef NO_SYSTEM_INCLUDES
  14. #include <sys/types.h>
  15. #include <string.h>
  16. #endif
  17. #include "db_int.h"
  18. #include "db_page.h"
  19. #include "db_verify.h"
  20. #include "db_ext.h"
  21. static int __db_vrfy_pgset_iinc __P((DB *, db_pgno_t, int));
  22. /*
  23.  * __db_vrfy_dbinfo_create --
  24.  * Allocate and initialize a VRFY_DBINFO structure.
  25.  *
  26.  * PUBLIC: int __db_vrfy_dbinfo_create
  27.  * PUBLIC:     __P((DB_ENV *, u_int32_t, VRFY_DBINFO **));
  28.  */
  29. int
  30. __db_vrfy_dbinfo_create (dbenv, pgsize, vdpp)
  31. DB_ENV *dbenv;
  32. u_int32_t pgsize;
  33. VRFY_DBINFO **vdpp;
  34. {
  35. DB *cdbp, *pgdbp, *pgset;
  36. VRFY_DBINFO *vdp;
  37. int ret;
  38. vdp = NULL;
  39. cdbp = pgdbp = pgset = NULL;
  40. if ((ret = __os_calloc(NULL,
  41.     1, sizeof(VRFY_DBINFO), (void **)&vdp)) != 0)
  42. goto err;
  43. if ((ret = db_create(&cdbp, dbenv, 0)) != 0)
  44. goto err;
  45. if ((ret = cdbp->set_flags(cdbp, DB_DUP | DB_DUPSORT)) != 0)
  46. goto err;
  47. if ((ret = cdbp->set_pagesize(cdbp, pgsize)) != 0)
  48. goto err;
  49. if ((ret =
  50.     cdbp->open(cdbp, NULL, NULL, DB_BTREE, DB_CREATE, 0600)) != 0)
  51. goto err;
  52. if ((ret = db_create(&pgdbp, dbenv, 0)) != 0)
  53. goto err;
  54. if ((ret = pgdbp->set_pagesize(pgdbp, pgsize)) != 0)
  55. goto err;
  56. if ((ret =
  57.     pgdbp->open(pgdbp, NULL, NULL, DB_BTREE, DB_CREATE, 0600)) != 0)
  58. goto err;
  59. if ((ret = __db_vrfy_pgset(dbenv, pgsize, &pgset)) != 0)
  60. goto err;
  61. LIST_INIT(&vdp->subdbs);
  62. LIST_INIT(&vdp->activepips);
  63. vdp->cdbp = cdbp;
  64. vdp->pgdbp = pgdbp;
  65. vdp->pgset = pgset;
  66. *vdpp = vdp;
  67. return (0);
  68. err: if (cdbp != NULL)
  69. (void)cdbp->close(cdbp, 0);
  70. if (pgdbp != NULL)
  71. (void)pgdbp->close(pgdbp, 0);
  72. if (vdp != NULL)
  73. __os_free(vdp, sizeof(VRFY_DBINFO));
  74. return (ret);
  75. }
  76. /*
  77.  * __db_vrfy_dbinfo_destroy --
  78.  * Destructor for VRFY_DBINFO.  Destroys VRFY_PAGEINFOs and deallocates
  79.  * structure.
  80.  *
  81.  * PUBLIC: int __db_vrfy_dbinfo_destroy __P((VRFY_DBINFO *));
  82.  */
  83. int
  84. __db_vrfy_dbinfo_destroy(vdp)
  85. VRFY_DBINFO *vdp;
  86. {
  87. VRFY_CHILDINFO *c, *d;
  88. int t_ret, ret;
  89. ret = 0;
  90. for (c = LIST_FIRST(&vdp->subdbs); c != NULL; c = d) {
  91. d = LIST_NEXT(c, links);
  92. __os_free(c, 0);
  93. }
  94. if ((t_ret = vdp->pgdbp->close(vdp->pgdbp, 0)) != 0)
  95. ret = t_ret;
  96. if ((t_ret = vdp->cdbp->close(vdp->cdbp, 0)) != 0 && ret == 0)
  97. ret = t_ret;
  98. if ((t_ret = vdp->pgset->close(vdp->pgset, 0)) != 0 && ret == 0)
  99. ret = t_ret;
  100. DB_ASSERT(LIST_FIRST(&vdp->activepips) == NULL);
  101. __os_free(vdp, sizeof(VRFY_DBINFO));
  102. return (ret);
  103. }
  104. /*
  105.  * __db_vrfy_getpageinfo --
  106.  * Get a PAGEINFO structure for a given page, creating it if necessary.
  107.  *
  108.  * PUBLIC: int __db_vrfy_getpageinfo
  109.  * PUBLIC:     __P((VRFY_DBINFO *, db_pgno_t, VRFY_PAGEINFO **));
  110.  */
  111. int
  112. __db_vrfy_getpageinfo(vdp, pgno, pipp)
  113. VRFY_DBINFO *vdp;
  114. db_pgno_t pgno;
  115. VRFY_PAGEINFO **pipp;
  116. {
  117. DBT key, data;
  118. DB *pgdbp;
  119. VRFY_PAGEINFO *pip;
  120. int ret;
  121. /*
  122.  * We want a page info struct.  There are three places to get it from,
  123.  * in decreasing order of preference:
  124.  *
  125.  * 1. vdp->activepips.  If it's already "checked out", we're
  126.  * already using it, we return the same exact structure with a
  127.  * bumped refcount.  This is necessary because this code is
  128.  * replacing array accesses, and it's common for f() to make some
  129.  * changes to a pip, and then call g() and h() which each make
  130.  * changes to the same pip.  vdps are never shared between threads
  131.  * (they're never returned to the application), so this is safe.
  132.  * 2. The pgdbp.  It's not in memory, but it's in the database, so
  133.  * get it, give it a refcount of 1, and stick it on activepips.
  134.  * 3. malloc.  It doesn't exist yet;  create it, then stick it on
  135.  * activepips.  We'll put it in the database when we putpageinfo
  136.  * later.
  137.  */
  138. /* Case 1. */
  139. for (pip = LIST_FIRST(&vdp->activepips); pip != NULL;
  140.     pip = LIST_NEXT(pip, links))
  141. if (pip->pgno == pgno)
  142. /* Found it. */
  143. goto found;
  144. /* Case 2. */
  145. pgdbp = vdp->pgdbp;
  146. memset(&key, 0, sizeof(DBT));
  147. memset(&data, 0, sizeof(DBT));
  148. F_SET(&data, DB_DBT_MALLOC);
  149. key.data = &pgno;
  150. key.size = sizeof(db_pgno_t);
  151. if ((ret = pgdbp->get(pgdbp, NULL, &key, &data, 0)) == 0) {
  152. /* Found it. */
  153. DB_ASSERT(data.size = sizeof(VRFY_PAGEINFO));
  154. pip = data.data;
  155. DB_ASSERT(pip->pi_refcount == 0);
  156. LIST_INSERT_HEAD(&vdp->activepips, pip, links);
  157. goto found;
  158. } else if (ret != DB_NOTFOUND) /* Something nasty happened. */
  159. return (ret);
  160. /* Case 3 */
  161. if ((ret = __db_vrfy_pageinfo_create(&pip)) != 0)
  162. return (ret);
  163. LIST_INSERT_HEAD(&vdp->activepips, pip, links);
  164. found: pip->pi_refcount++;
  165. *pipp = pip;
  166. DB_ASSERT(pip->pi_refcount > 0);
  167. return (0);
  168. }
  169. /*
  170.  * __db_vrfy_putpageinfo --
  171.  * Put back a VRFY_PAGEINFO that we're done with.
  172.  *
  173.  * PUBLIC: int __db_vrfy_putpageinfo __P((VRFY_DBINFO *, VRFY_PAGEINFO *));
  174.  */
  175. int
  176. __db_vrfy_putpageinfo(vdp, pip)
  177. VRFY_DBINFO *vdp;
  178. VRFY_PAGEINFO *pip;
  179. {
  180. DBT key, data;
  181. DB *pgdbp;
  182. VRFY_PAGEINFO *p;
  183. int ret;
  184. #ifdef DIAGNOSTIC
  185. int found;
  186. found = 0;
  187. #endif
  188. if (--pip->pi_refcount > 0)
  189. return (0);
  190. pgdbp = vdp->pgdbp;
  191. memset(&key, 0, sizeof(DBT));
  192. memset(&data, 0, sizeof(DBT));
  193. key.data = &pip->pgno;
  194. key.size = sizeof(db_pgno_t);
  195. data.data = pip;
  196. data.size = sizeof(VRFY_PAGEINFO);
  197. if ((ret = pgdbp->put(pgdbp, NULL, &key, &data, 0)) != 0)
  198. return (ret);
  199. for (p = LIST_FIRST(&vdp->activepips); p != NULL;
  200.     p = LIST_NEXT(p, links))
  201. if (p == pip) {
  202. #ifdef DIAGNOSTIC
  203. found++;
  204. #endif
  205. DB_ASSERT(p->pi_refcount == 0);
  206. LIST_REMOVE(p, links);
  207. break;
  208. }
  209. #ifdef DIAGNOSTIC
  210. DB_ASSERT(found == 1);
  211. #endif
  212. DB_ASSERT(pip->pi_refcount == 0);
  213. __os_free(pip, 0);
  214. return (0);
  215. }
  216. /*
  217.  * __db_vrfy_pgset --
  218.  * Create a temporary database for the storing of sets of page numbers.
  219.  * (A mapping from page number to int, used by the *_meta2pgset functions,
  220.  * as well as for keeping track of which pages the verifier has seen.)
  221.  *
  222.  * PUBLIC: int __db_vrfy_pgset __P((DB_ENV *, u_int32_t, DB **));
  223.  */
  224. int
  225. __db_vrfy_pgset(dbenv, pgsize, dbpp)
  226. DB_ENV *dbenv;
  227. u_int32_t pgsize;
  228. DB **dbpp;
  229. {
  230. DB *dbp;
  231. int ret;
  232. if ((ret = db_create(&dbp, dbenv, 0)) != 0)
  233. return (ret);
  234. if ((ret = dbp->set_pagesize(dbp, pgsize)) != 0)
  235. goto err;
  236. if ((ret = dbp->open(dbp, NULL, NULL, DB_BTREE, DB_CREATE, 0600)) == 0)
  237. *dbpp = dbp;
  238. else
  239. err: (void)dbp->close(dbp, 0);
  240. return (ret);
  241. }
  242. /*
  243.  * __db_vrfy_pgset_get --
  244.  * Get the value associated in a page set with a given pgno.  Return
  245.  * a 0 value (and succeed) if we've never heard of this page.
  246.  *
  247.  * PUBLIC: int __db_vrfy_pgset_get __P((DB *, db_pgno_t, int *));
  248.  */
  249. int
  250. __db_vrfy_pgset_get(dbp, pgno, valp)
  251. DB *dbp;
  252. db_pgno_t pgno;
  253. int *valp;
  254. {
  255. DBT key, data;
  256. int ret, val;
  257. memset(&key, 0, sizeof(DBT));
  258. memset(&data, 0, sizeof(DBT));
  259. key.data = &pgno;
  260. key.size = sizeof(db_pgno_t);
  261. data.data = &val;
  262. data.ulen = sizeof(int);
  263. F_SET(&data, DB_DBT_USERMEM);
  264. if ((ret = dbp->get(dbp, NULL, &key, &data, 0)) == 0) {
  265. DB_ASSERT(data.size = sizeof(int));
  266. memcpy(&val, data.data, sizeof(int));
  267. } else if (ret == DB_NOTFOUND)
  268. val = 0;
  269. else
  270. return (ret);
  271. *valp = val;
  272. return (0);
  273. }
  274. /*
  275.  * __db_vrfy_pgset_inc --
  276.  * Increment the value associated with a pgno by 1.
  277.  *
  278.  * PUBLIC: int __db_vrfy_pgset_inc __P((DB *, db_pgno_t));
  279.  */
  280. int
  281. __db_vrfy_pgset_inc(dbp, pgno)
  282. DB *dbp;
  283. db_pgno_t pgno;
  284. {
  285. return (__db_vrfy_pgset_iinc(dbp, pgno, 1));
  286. }
  287. /*
  288.  * __db_vrfy_pgset_dec --
  289.  * Increment the value associated with a pgno by 1.
  290.  *
  291.  * PUBLIC: int __db_vrfy_pgset_dec __P((DB *, db_pgno_t));
  292.  */
  293. int
  294. __db_vrfy_pgset_dec(dbp, pgno)
  295. DB *dbp;
  296. db_pgno_t pgno;
  297. {
  298. return (__db_vrfy_pgset_iinc(dbp, pgno, -1));
  299. }
  300. /*
  301.  * __db_vrfy_pgset_iinc --
  302.  * Increment the value associated with a pgno by i.
  303.  *
  304.  */
  305. static int
  306. __db_vrfy_pgset_iinc(dbp, pgno, i)
  307. DB *dbp;
  308. db_pgno_t pgno;
  309. int i;
  310. {
  311. DBT key, data;
  312. int ret;
  313. int val;
  314. memset(&key, 0, sizeof(DBT));
  315. memset(&data, 0, sizeof(DBT));
  316. val = 0;
  317. key.data = &pgno;
  318. key.size = sizeof(db_pgno_t);
  319. data.data = &val;
  320. data.ulen = sizeof(int);
  321. F_SET(&data, DB_DBT_USERMEM);
  322. if ((ret = dbp->get(dbp, NULL, &key, &data, 0)) == 0) {
  323. DB_ASSERT(data.size = sizeof(int));
  324. memcpy(&val, data.data, sizeof(int));
  325. } else if (ret != DB_NOTFOUND)
  326. return (ret);
  327. data.size = sizeof(int);
  328. val += i;
  329. return (dbp->put(dbp, NULL, &key, &data, 0));
  330. }
  331. /*
  332.  * __db_vrfy_pgset_next --
  333.  * Given a cursor open in a pgset database, get the next page in the
  334.  * set.
  335.  *
  336.  * PUBLIC: int __db_vrfy_pgset_next __P((DBC *, db_pgno_t *));
  337.  */
  338. int
  339. __db_vrfy_pgset_next(dbc, pgnop)
  340. DBC *dbc;
  341. db_pgno_t *pgnop;
  342. {
  343. DBT key, data;
  344. db_pgno_t pgno;
  345. int ret;
  346. memset(&key, 0, sizeof(DBT));
  347. memset(&data, 0, sizeof(DBT));
  348. /* We don't care about the data, just the keys. */
  349. F_SET(&data, DB_DBT_USERMEM | DB_DBT_PARTIAL);
  350. F_SET(&key, DB_DBT_USERMEM);
  351. key.data = &pgno;
  352. key.ulen = sizeof(db_pgno_t);
  353. if ((ret = dbc->c_get(dbc, &key, &data, DB_NEXT)) != 0)
  354. return (ret);
  355. DB_ASSERT(key.size == sizeof(db_pgno_t));
  356. *pgnop = pgno;
  357. return (0);
  358. }
  359. /*
  360.  * __db_vrfy_childcursor --
  361.  * Create a cursor to walk the child list with.  Returns with a nonzero
  362.  * final argument if the specified page has no children.
  363.  *
  364.  * PUBLIC: int __db_vrfy_childcursor __P((VRFY_DBINFO *, DBC **));
  365.  */
  366. int
  367. __db_vrfy_childcursor(vdp, dbcp)
  368. VRFY_DBINFO *vdp;
  369. DBC **dbcp;
  370. {
  371. DB *cdbp;
  372. DBC *dbc;
  373. int ret;
  374. cdbp = vdp->cdbp;
  375. if ((ret = cdbp->cursor(cdbp, NULL, &dbc, 0)) == 0)
  376. *dbcp = dbc;
  377. return (ret);
  378. }
  379. /*
  380.  * __db_vrfy_childput --
  381.  * Add a child structure to the set for a given page.
  382.  *
  383.  * PUBLIC: int __db_vrfy_childput
  384.  * PUBLIC:     __P((VRFY_DBINFO *, db_pgno_t, VRFY_CHILDINFO *));
  385.  */
  386. int
  387. __db_vrfy_childput(vdp, pgno, cip)
  388. VRFY_DBINFO *vdp;
  389. db_pgno_t pgno;
  390. VRFY_CHILDINFO *cip;
  391. {
  392. DBT key, data;
  393. DB *cdbp;
  394. int ret;
  395. cdbp = vdp->cdbp;
  396. memset(&key, 0, sizeof(DBT));
  397. memset(&data, 0, sizeof(DBT));
  398. key.data = &pgno;
  399. key.size = sizeof(db_pgno_t);
  400. data.data = cip;
  401. data.size = sizeof(VRFY_CHILDINFO);
  402. /*
  403.  * Don't add duplicate (data) entries for a given child, and accept
  404.  * DB_KEYEXIST as a successful return;  we only need to verify
  405.  * each child once, even if a child (such as an overflow key) is
  406.  * multiply referenced.
  407.  */
  408. ret = cdbp->put(cdbp, NULL, &key, &data, DB_NODUPDATA);
  409. return (ret == DB_KEYEXIST ? 0 : ret);
  410. }
  411. /*
  412.  * __db_vrfy_ccset --
  413.  * Sets a cursor created with __db_vrfy_childcursor to the first
  414.  * child of the given pgno, and returns it in the third arg.
  415.  *
  416.  * PUBLIC: int __db_vrfy_ccset __P((DBC *, db_pgno_t, VRFY_CHILDINFO **));
  417.  */
  418. int
  419. __db_vrfy_ccset(dbc, pgno, cipp)
  420. DBC *dbc;
  421. db_pgno_t pgno;
  422. VRFY_CHILDINFO **cipp;
  423. {
  424. DBT key, data;
  425. int ret;
  426. memset(&key, 0, sizeof(DBT));
  427. memset(&data, 0, sizeof(DBT));
  428. key.data = &pgno;
  429. key.size = sizeof(db_pgno_t);
  430. if ((ret = dbc->c_get(dbc, &key, &data, DB_SET)) != 0)
  431. return (ret);
  432. DB_ASSERT(data.size == sizeof(VRFY_CHILDINFO));
  433. *cipp = (VRFY_CHILDINFO *)data.data;
  434. return (0);
  435. }
  436. /*
  437.  * __db_vrfy_ccnext --
  438.  * Gets the next child of the given cursor created with
  439.  * __db_vrfy_childcursor, and returns it in the memory provided in the
  440.  * second arg.
  441.  *
  442.  * PUBLIC: int __db_vrfy_ccnext __P((DBC *, VRFY_CHILDINFO **));
  443.  */
  444. int
  445. __db_vrfy_ccnext(dbc, cipp)
  446. DBC *dbc;
  447. VRFY_CHILDINFO **cipp;
  448. {
  449. DBT key, data;
  450. int ret;
  451. memset(&key, 0, sizeof(DBT));
  452. memset(&data, 0, sizeof(DBT));
  453. if ((ret = dbc->c_get(dbc, &key, &data, DB_NEXT_DUP)) != 0)
  454. return (ret);
  455. DB_ASSERT(data.size == sizeof(VRFY_CHILDINFO));
  456. *cipp = (VRFY_CHILDINFO *)data.data;
  457. return (0);
  458. }
  459. /*
  460.  * __db_vrfy_ccclose --
  461.  * Closes the cursor created with __db_vrfy_childcursor.
  462.  *
  463.  * This doesn't actually do anything interesting now, but it's
  464.  * not inconceivable that we might change the internal database usage
  465.  * and keep the interfaces the same, and a function call here or there
  466.  * seldom hurts anyone.
  467.  *
  468.  * PUBLIC: int __db_vrfy_ccclose __P((DBC *));
  469.  */
  470. int
  471. __db_vrfy_ccclose(dbc)
  472. DBC *dbc;
  473. {
  474. return (dbc->c_close(dbc));
  475. }
  476. /*
  477.  * __db_vrfy_pageinfo_create --
  478.  * Constructor for VRFY_PAGEINFO;  allocates and initializes.
  479.  *
  480.  * PUBLIC: int __db_vrfy_pageinfo_create __P((VRFY_PAGEINFO **));
  481.  */
  482. int
  483. __db_vrfy_pageinfo_create(pgipp)
  484. VRFY_PAGEINFO **pgipp;
  485. {
  486. VRFY_PAGEINFO *pgip;
  487. int ret;
  488. if ((ret = __os_calloc(NULL,
  489.     1, sizeof(VRFY_PAGEINFO), (void **)&pgip)) != 0)
  490. return (ret);
  491. DB_ASSERT(pgip->pi_refcount == 0);
  492. *pgipp = pgip;
  493. return (0);
  494. }
  495. /*
  496.  * __db_salvage_init --
  497.  * Set up salvager database.
  498.  *
  499.  * PUBLIC: int  __db_salvage_init __P((VRFY_DBINFO *));
  500.  */
  501. int
  502. __db_salvage_init(vdp)
  503. VRFY_DBINFO *vdp;
  504. {
  505. DB *dbp;
  506. int ret;
  507. if ((ret = db_create(&dbp, NULL, 0)) != 0)
  508. return (ret);
  509. if ((ret = dbp->set_pagesize(dbp, 1024)) != 0)
  510. goto err;
  511. if ((ret = dbp->open(dbp, NULL, NULL, DB_BTREE, DB_CREATE, 0)) != 0)
  512. goto err;
  513. vdp->salvage_pages = dbp;
  514. return (0);
  515. err: (void)dbp->close(dbp, 0);
  516. return (ret);
  517. }
  518. /*
  519.  * __db_salvage_destroy --
  520.  * Close salvager database.
  521.  * PUBLIC: void  __db_salvage_destroy __P((VRFY_DBINFO *));
  522.  */
  523. void
  524. __db_salvage_destroy(vdp)
  525. VRFY_DBINFO *vdp;
  526. {
  527. (void)vdp->salvage_pages->close(vdp->salvage_pages, 0);
  528. }
  529. /*
  530.  * __db_salvage_getnext --
  531.  * Get the next (first) unprinted page in the database of pages we need to
  532.  * print still.  Delete entries for any already-printed pages we encounter
  533.  * in this search, as well as the page we're returning.
  534.  *
  535.  * PUBLIC: int __db_salvage_getnext
  536.  * PUBLIC:     __P((VRFY_DBINFO *, db_pgno_t *, u_int32_t *));
  537.  */
  538. int
  539. __db_salvage_getnext(vdp, pgnop, pgtypep)
  540. VRFY_DBINFO *vdp;
  541. db_pgno_t *pgnop;
  542. u_int32_t *pgtypep;
  543. {
  544. DB *dbp;
  545. DBC *dbc;
  546. DBT key, data;
  547. int ret;
  548. u_int32_t pgtype;
  549. dbp = vdp->salvage_pages;
  550. memset(&key, 0, sizeof(DBT));
  551. memset(&data, 0, sizeof(DBT));
  552. if ((ret = dbp->cursor(dbp, NULL, &dbc, 0)) != 0)
  553. return (ret);
  554. while ((ret = dbc->c_get(dbc, &key, &data, DB_NEXT)) == 0) {
  555. DB_ASSERT(data.size == sizeof(u_int32_t));
  556. memcpy(&pgtype, data.data, sizeof(pgtype));
  557. if ((ret = dbc->c_del(dbc, 0)) != 0)
  558. goto err;
  559. if (pgtype != SALVAGE_IGNORE)
  560. goto found;
  561. }
  562. /* No more entries--ret probably equals DB_NOTFOUND. */
  563. if (0) {
  564. found: DB_ASSERT(key.size == sizeof(db_pgno_t));
  565. DB_ASSERT(data.size == sizeof(u_int32_t));
  566. *pgnop = *(db_pgno_t *)key.data;
  567. *pgtypep = *(u_int32_t *)data.data;
  568. }
  569. err: (void)dbc->c_close(dbc);
  570. return (ret);
  571. }
  572. /*
  573.  * __db_salvage_isdone --
  574.  * Return whether or not the given pgno is already marked
  575.  * SALVAGE_IGNORE (meaning that we don't need to print it again).
  576.  *
  577.  * Returns DB_KEYEXIST if it is marked, 0 if not, or another error on
  578.  * error.
  579.  *
  580.  * PUBLIC: int __db_salvage_isdone __P((VRFY_DBINFO *, db_pgno_t));
  581.  */
  582. int
  583. __db_salvage_isdone(vdp, pgno)
  584. VRFY_DBINFO *vdp;
  585. db_pgno_t pgno;
  586. {
  587. DBT key, data;
  588. DB *dbp;
  589. int ret;
  590. u_int32_t currtype;
  591. dbp = vdp->salvage_pages;
  592. memset(&key, 0, sizeof(DBT));
  593. memset(&data, 0, sizeof(DBT));
  594. currtype = SALVAGE_INVALID;
  595. data.data = &currtype;
  596. data.ulen = sizeof(u_int32_t);
  597. data.flags = DB_DBT_USERMEM;
  598. key.data = &pgno;
  599. key.size = sizeof(db_pgno_t);
  600. /*
  601.  * Put an entry for this page, with pgno as key and type as data,
  602.  * unless it's already there and is marked done.
  603.  * If it's there and is marked anything else, that's fine--we
  604.  * want to mark it done.
  605.  */
  606. ret = dbp->get(dbp, NULL, &key, &data, 0);
  607. if (ret == 0) {
  608. /*
  609.  * The key's already here.  Check and see if it's already
  610.  * marked done.  If it is, return DB_KEYEXIST.  If it's not,
  611.  * return 0.
  612.  */
  613. if (currtype == SALVAGE_IGNORE)
  614. return (DB_KEYEXIST);
  615. else
  616. return (0);
  617. } else if (ret != DB_NOTFOUND)
  618. return (ret);
  619. /* The pgno is not yet marked anything; return 0. */
  620. return (0);
  621. }
  622. /*
  623.  * __db_salvage_markdone --
  624.  * Mark as done a given page.
  625.  *
  626.  * PUBLIC: int __db_salvage_markdone __P((VRFY_DBINFO *, db_pgno_t));
  627.  */
  628. int
  629. __db_salvage_markdone(vdp, pgno)
  630. VRFY_DBINFO *vdp;
  631. db_pgno_t pgno;
  632. {
  633. DBT key, data;
  634. DB *dbp;
  635. int pgtype, ret;
  636. u_int32_t currtype;
  637. pgtype = SALVAGE_IGNORE;
  638. dbp = vdp->salvage_pages;
  639. memset(&key, 0, sizeof(DBT));
  640. memset(&data, 0, sizeof(DBT));
  641. currtype = SALVAGE_INVALID;
  642. data.data = &currtype;
  643. data.ulen = sizeof(u_int32_t);
  644. data.flags = DB_DBT_USERMEM;
  645. key.data = &pgno;
  646. key.size = sizeof(db_pgno_t);
  647. /*
  648.  * Put an entry for this page, with pgno as key and type as data,
  649.  * unless it's already there and is marked done.
  650.  * If it's there and is marked anything else, that's fine--we
  651.  * want to mark it done, but db_salvage_isdone only lets
  652.  * us know if it's marked IGNORE.
  653.  *
  654.  * We don't want to return DB_KEYEXIST, though;  this will
  655.  * likely get passed up all the way and make no sense to the
  656.  * application.  Instead, use DB_VERIFY_BAD to indicate that
  657.  * we've seen this page already--it probably indicates a
  658.  * multiply-linked page.
  659.  */
  660. if ((ret = __db_salvage_isdone(vdp, pgno)) != 0)
  661. return (ret == DB_KEYEXIST ? DB_VERIFY_BAD : ret);
  662. data.size = sizeof(u_int32_t);
  663. data.data = &pgtype;
  664. return (dbp->put(dbp, NULL, &key, &data, 0));
  665. }
  666. /*
  667.  * __db_salvage_markneeded --
  668.  * If it has not yet been printed, make note of the fact that a page
  669.  * must be dealt with later.
  670.  *
  671.  * PUBLIC: int __db_salvage_markneeded
  672.  * PUBLIC:     __P((VRFY_DBINFO *, db_pgno_t, u_int32_t));
  673.  */
  674. int
  675. __db_salvage_markneeded(vdp, pgno, pgtype)
  676. VRFY_DBINFO *vdp;
  677. db_pgno_t pgno;
  678. u_int32_t pgtype;
  679. {
  680. DB *dbp;
  681. DBT key, data;
  682. int ret;
  683. dbp = vdp->salvage_pages;
  684. memset(&key, 0, sizeof(DBT));
  685. memset(&data, 0, sizeof(DBT));
  686. key.data = &pgno;
  687. key.size = sizeof(db_pgno_t);
  688. data.data = &pgtype;
  689. data.size = sizeof(u_int32_t);
  690. /*
  691.  * Put an entry for this page, with pgno as key and type as data,
  692.  * unless it's already there, in which case it's presumably
  693.  * already been marked done.
  694.  */
  695. ret = dbp->put(dbp, NULL, &key, &data, DB_NOOVERWRITE);
  696. return (ret == DB_KEYEXIST ? 0 : ret);
  697. }