1N/A/*-
1N/A * See the file LICENSE for redistribution information.
1N/A *
1N/A * Copyright (c) 1996, 1997, 1998
1N/A * Sleepycat Software. All rights reserved.
1N/A */
1N/A
1N/A#include "config.h"
1N/A
1N/A#ifndef lint
1N/Astatic const char sccsid[] = "@(#)db_ret.c 10.16 (Sleepycat) 10/4/98";
1N/A#endif /* not lint */
1N/A
1N/A#ifndef NO_SYSTEM_INCLUDES
1N/A#include <sys/types.h>
1N/A
1N/A#include <errno.h>
1N/A#include <string.h>
1N/A#endif
1N/A
1N/A#include "db_int.h"
1N/A#include "db_page.h"
1N/A#include "btree.h"
1N/A#include "db_am.h"
1N/A
1N/A/*
1N/A * __db_ret --
1N/A * Build return DBT.
1N/A *
1N/A * PUBLIC: int __db_ret __P((DB *,
1N/A * PUBLIC: PAGE *, u_int32_t, DBT *, void **, u_int32_t *));
1N/A */
1N/Aint
1N/A__db_ret(dbp, h, indx, dbt, memp, memsize)
1N/A DB *dbp;
1N/A PAGE *h;
1N/A u_int32_t indx;
1N/A DBT *dbt;
1N/A void **memp;
1N/A u_int32_t *memsize;
1N/A{
1N/A BKEYDATA *bk;
1N/A HOFFPAGE ho;
1N/A BOVERFLOW *bo;
1N/A u_int32_t len;
1N/A u_int8_t *hk;
1N/A void *data;
1N/A
1N/A switch (TYPE(h)) {
1N/A case P_HASH:
1N/A hk = P_ENTRY(h, indx);
1N/A if (HPAGE_PTYPE(hk) == H_OFFPAGE) {
1N/A memcpy(&ho, hk, sizeof(HOFFPAGE));
1N/A return (__db_goff(dbp, dbt,
1N/A ho.tlen, ho.pgno, memp, memsize));
1N/A }
1N/A len = LEN_HKEYDATA(h, dbp->pgsize, indx);
1N/A data = HKEYDATA_DATA(hk);
1N/A break;
1N/A case P_DUPLICATE:
1N/A case P_LBTREE:
1N/A case P_LRECNO:
1N/A bk = GET_BKEYDATA(h, indx);
1N/A if (B_TYPE(bk->type) == B_OVERFLOW) {
1N/A bo = (BOVERFLOW *)bk;
1N/A return (__db_goff(dbp, dbt,
1N/A bo->tlen, bo->pgno, memp, memsize));
1N/A }
1N/A len = bk->len;
1N/A data = bk->data;
1N/A break;
1N/A default:
1N/A return (__db_pgfmt(dbp, h->pgno));
1N/A }
1N/A
1N/A return (__db_retcopy(dbt, data, len, memp, memsize,
1N/A F_ISSET(dbt, DB_DBT_INTERNAL) ? NULL : dbp->db_malloc));
1N/A}
1N/A
1N/A/*
1N/A * __db_retcopy --
1N/A * Copy the returned data into the user's DBT, handling special flags.
1N/A *
1N/A * PUBLIC: int __db_retcopy __P((DBT *,
1N/A * PUBLIC: void *, u_int32_t, void **, u_int32_t *, void *(*)(size_t)));
1N/A */
1N/Aint
1N/A__db_retcopy(dbt, data, len, memp, memsize, db_malloc)
1N/A DBT *dbt;
1N/A void *data;
1N/A u_int32_t len;
1N/A void **memp;
1N/A u_int32_t *memsize;
1N/A void *(*db_malloc) __P((size_t));
1N/A{
1N/A int ret;
1N/A
1N/A /* If returning a partial record, reset the length. */
1N/A if (F_ISSET(dbt, DB_DBT_PARTIAL)) {
1N/A data = (u_int8_t *)data + dbt->doff;
1N/A if (len > dbt->doff) {
1N/A len -= dbt->doff;
1N/A if (len > dbt->dlen)
1N/A len = dbt->dlen;
1N/A } else
1N/A len = 0;
1N/A }
1N/A
1N/A /*
1N/A * Return the length of the returned record in the DBT size field.
1N/A * This satisfies the requirement that if we're using user memory
1N/A * and insufficient memory was provided, return the amount necessary
1N/A * in the size field.
1N/A */
1N/A dbt->size = len;
1N/A
1N/A /*
1N/A * Allocate memory to be owned by the application: DB_DBT_MALLOC.
1N/A *
1N/A * !!!
1N/A * We always allocate memory, even if we're copying out 0 bytes. This
1N/A * guarantees consistency, i.e., the application can always free memory
1N/A * without concern as to how many bytes of the record were requested.
1N/A *
1N/A * Use the memory specified by the application: DB_DBT_USERMEM.
1N/A *
1N/A * !!!
1N/A * If the length we're going to copy is 0, the application-supplied
1N/A * memory pointer is allowed to be NULL.
1N/A */
1N/A if (F_ISSET(dbt, DB_DBT_MALLOC)) {
1N/A if ((ret = __os_malloc(len, db_malloc, &dbt->data)) != 0)
1N/A return (ret);
1N/A } else if (F_ISSET(dbt, DB_DBT_USERMEM)) {
1N/A if (len != 0 && (dbt->data == NULL || dbt->ulen < len))
1N/A return (ENOMEM);
1N/A } else if (memp == NULL || memsize == NULL) {
1N/A return (EINVAL);
1N/A } else {
1N/A if (len != 0 && (*memsize == 0 || *memsize < len)) {
1N/A if ((ret = __os_realloc(memp, len)) != 0) {
1N/A *memsize = 0;
1N/A return (ret);
1N/A }
1N/A *memsize = len;
1N/A }
1N/A dbt->data = *memp;
1N/A }
1N/A
1N/A if (len != 0)
1N/A memcpy(dbt->data, data, len);
1N/A return (0);
1N/A}