db_ret.c 3.47 KB
Newer Older
1 2 3
/*-
 * See the file LICENSE for redistribution information.
 *
jimw@mysql.com's avatar
jimw@mysql.com committed
4
 * Copyright (c) 1996-2005
5
 *	Sleepycat Software.  All rights reserved.
jimw@mysql.com's avatar
jimw@mysql.com committed
6
 *
jimw@mysql.com's avatar
jimw@mysql.com committed
7
 * $Id: db_ret.c,v 12.1 2005/06/16 20:21:14 bostic Exp $
8 9 10 11 12 13 14 15 16 17 18
 */

#include "db_config.h"

#ifndef NO_SYSTEM_INCLUDES
#include <sys/types.h>

#include <string.h>
#endif

#include "db_int.h"
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
19 20
#include "dbinc/db_page.h"
#include "dbinc/db_am.h"
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

/*
 * __db_ret --
 *	Build return DBT.
 *
 * PUBLIC: int __db_ret __P((DB *,
 * PUBLIC:    PAGE *, u_int32_t, DBT *, void **, u_int32_t *));
 */
int
__db_ret(dbp, h, indx, dbt, memp, memsize)
	DB *dbp;
	PAGE *h;
	u_int32_t indx;
	DBT *dbt;
	void **memp;
	u_int32_t *memsize;
{
	BKEYDATA *bk;
	HOFFPAGE ho;
	BOVERFLOW *bo;
	u_int32_t len;
	u_int8_t *hk;
	void *data;

	switch (TYPE(h)) {
	case P_HASH:
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
47
		hk = P_ENTRY(dbp, h, indx);
48 49 50 51 52
		if (HPAGE_PTYPE(hk) == H_OFFPAGE) {
			memcpy(&ho, hk, sizeof(HOFFPAGE));
			return (__db_goff(dbp, dbt,
			    ho.tlen, ho.pgno, memp, memsize));
		}
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
53
		len = LEN_HKEYDATA(dbp, h, dbp->pgsize, indx);
54 55 56 57 58
		data = HKEYDATA_DATA(hk);
		break;
	case P_LBTREE:
	case P_LDUP:
	case P_LRECNO:
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
59
		bk = GET_BKEYDATA(dbp, h, indx);
60 61 62 63 64 65 66 67 68
		if (B_TYPE(bk->type) == B_OVERFLOW) {
			bo = (BOVERFLOW *)bk;
			return (__db_goff(dbp, dbt,
			    bo->tlen, bo->pgno, memp, memsize));
		}
		len = bk->len;
		data = bk->data;
		break;
	default:
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
69
		return (__db_pgfmt(dbp->dbenv, h->pgno));
70 71
	}

ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
72
	return (__db_retcopy(dbp->dbenv, dbt, data, len, memp, memsize));
73 74 75 76 77 78
}

/*
 * __db_retcopy --
 *	Copy the returned data into the user's DBT, handling special flags.
 *
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
79
 * PUBLIC: int __db_retcopy __P((DB_ENV *, DBT *,
80 81 82
 * PUBLIC:    void *, u_int32_t, void **, u_int32_t *));
 */
int
ram@mysql.r18.ru's avatar
ram@mysql.r18.ru committed
83 84
__db_retcopy(dbenv, dbt, data, len, memp, memsize)
	DB_ENV *dbenv;
85 86 87 88 89 90 91 92
	DBT *dbt;
	void *data;
	u_int32_t len;
	void **memp;
	u_int32_t *memsize;
{
	int ret;

jimw@mysql.com's avatar
jimw@mysql.com committed
93 94
	ret = 0;

95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
	/* If returning a partial record, reset the length. */
	if (F_ISSET(dbt, DB_DBT_PARTIAL)) {
		data = (u_int8_t *)data + dbt->doff;
		if (len > dbt->doff) {
			len -= dbt->doff;
			if (len > dbt->dlen)
				len = dbt->dlen;
		} else
			len = 0;
	}

	/*
	 * Allocate memory to be owned by the application: DB_DBT_MALLOC,
	 * DB_DBT_REALLOC.
	 *
	 * !!!
	 * We always allocate memory, even if we're copying out 0 bytes. This
	 * guarantees consistency, i.e., the application can always free memory
	 * without concern as to how many bytes of the record were requested.
	 *
	 * Use the memory specified by the application: DB_DBT_USERMEM.
	 *
	 * !!!
	 * If the length we're going to copy is 0, the application-supplied
	 * memory pointer is allowed to be NULL.
	 */
	if (F_ISSET(dbt, DB_DBT_MALLOC)) {
jimw@mysql.com's avatar
jimw@mysql.com committed
122
		ret = __os_umalloc(dbenv, len, &dbt->data);
123
	} else if (F_ISSET(dbt, DB_DBT_REALLOC)) {
jimw@mysql.com's avatar
jimw@mysql.com committed
124 125
		if (dbt->data == NULL || dbt->size == 0 || dbt->size < len)
			ret = __os_urealloc(dbenv, len, &dbt->data);
126 127
	} else if (F_ISSET(dbt, DB_DBT_USERMEM)) {
		if (len != 0 && (dbt->data == NULL || dbt->ulen < len))
jimw@mysql.com's avatar
jimw@mysql.com committed
128
			ret = DB_BUFFER_SMALL;
129
	} else if (memp == NULL || memsize == NULL) {
jimw@mysql.com's avatar
jimw@mysql.com committed
130
		ret = EINVAL;
131 132
	} else {
		if (len != 0 && (*memsize == 0 || *memsize < len)) {
jimw@mysql.com's avatar
jimw@mysql.com committed
133 134 135
			if ((ret = __os_realloc(dbenv, len, memp)) == 0)
				*memsize = len;
			else
136 137
				*memsize = 0;
		}
jimw@mysql.com's avatar
jimw@mysql.com committed
138 139
		if (ret == 0)
			dbt->data = *memp;
140 141
	}

jimw@mysql.com's avatar
jimw@mysql.com committed
142
	if (ret == 0 && len != 0)
143
		memcpy(dbt->data, data, len);
jimw@mysql.com's avatar
jimw@mysql.com committed
144 145 146 147 148 149 150 151 152 153

	/*
	 * Return the length of the returned record in the DBT size field.
	 * This satisfies the requirement that if we're using user memory
	 * and insufficient memory was provided, return the amount necessary
	 * in the size field.
	 */
	dbt->size = len;

	return (ret);
154
}