#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/db_am.h"
#include "dbinc/mp.h"
int
__db_goff(dbp, ip, txn, dbt, tlen, pgno, bpp, bpsz)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
DBT *dbt;
u_int32_t tlen;
db_pgno_t pgno;
void **bpp;
u_int32_t *bpsz;
{
DB_MPOOLFILE *mpf;
ENV *env;
PAGE *h;
db_indx_t bytes;
u_int32_t curoff, needed, start;
u_int8_t *p, *src;
int ret;
env = dbp->env;
mpf = dbp->mpf;
if (F_ISSET(dbt, DB_DBT_PARTIAL)) {
start = dbt->doff;
if (start > tlen)
needed = 0;
else if (dbt->dlen > tlen - start)
needed = tlen - start;
else
needed = dbt->dlen;
} else {
start = 0;
needed = tlen;
}
if (F_ISSET(dbt, DB_DBT_USERCOPY))
goto skip_alloc;
if (F_ISSET(dbt, DB_DBT_USERMEM)) {
if (needed > dbt->ulen) {
dbt->size = needed;
return (DB_BUFFER_SMALL);
}
} else if (F_ISSET(dbt, DB_DBT_MALLOC)) {
if ((ret = __os_umalloc(env, needed, &dbt->data)) != 0)
return (ret);
} else if (F_ISSET(dbt, DB_DBT_REALLOC)) {
if ((ret = __os_urealloc(env, needed, &dbt->data)) != 0)
return (ret);
} else if (bpsz != NULL && (*bpsz == 0 || *bpsz < needed)) {
if ((ret = __os_realloc(env, needed, bpp)) != 0)
return (ret);
*bpsz = needed;
dbt->data = *bpp;
} else if (bpp != NULL)
dbt->data = *bpp;
else {
DB_ASSERT(env,
F_ISSET(dbt,
DB_DBT_USERMEM | DB_DBT_MALLOC | DB_DBT_REALLOC) ||
bpsz != NULL || bpp != NULL);
return (DB_BUFFER_SMALL);
}
skip_alloc:
dbt->size = needed;
for (curoff = 0, p = dbt->data; pgno != PGNO_INVALID && needed > 0;) {
if ((ret = __memp_fget(mpf, &pgno, ip, txn, 0, &h)) != 0)
return (ret);
DB_ASSERT(env, TYPE(h) == P_OVERFLOW);
if (curoff + OV_LEN(h) >= start) {
bytes = OV_LEN(h);
src = (u_int8_t *)h + P_OVERHEAD(dbp);
if (start > curoff) {
src += start - curoff;
bytes -= start - curoff;
}
if (bytes > needed)
bytes = needed;
if (F_ISSET(dbt, DB_DBT_USERCOPY)) {
if ((ret = env->dbt_usercopy(
dbt, dbt->size - needed,
src, bytes, DB_USERCOPY_SETDATA)) != 0) {
(void)__memp_fput(mpf, ip,
h, dbp->priority);
return (ret);
}
} else
memcpy(p, src, bytes);
p += bytes;
needed -= bytes;
}
curoff += OV_LEN(h);
pgno = h->next_pgno;
(void)__memp_fput(mpf, ip, h, dbp->priority);
}
return (0);
}
int
__db_poff(dbc, dbt, pgnop)
DBC *dbc;
const DBT *dbt;
db_pgno_t *pgnop;
{
DB *dbp;
DBT tmp_dbt;
DB_LSN new_lsn, null_lsn;
DB_MPOOLFILE *mpf;
PAGE *pagep, *lastp;
db_indx_t pagespace;
u_int32_t sz;
u_int8_t *p;
int ret, t_ret;
dbp = dbc->dbp;
mpf = dbp->mpf;
pagespace = P_MAXSPACE(dbp, dbp->pgsize);
ret = 0;
lastp = NULL;
for (p = dbt->data,
sz = dbt->size; sz > 0; p += pagespace, sz -= pagespace) {
if (sz < pagespace)
pagespace = sz;
if ((ret = __db_new(dbc, P_OVERFLOW, &pagep)) != 0)
break;
if (DBC_LOGGING(dbc)) {
tmp_dbt.data = p;
tmp_dbt.size = pagespace;
ZERO_LSN(null_lsn);
if ((ret = __db_big_log(dbp, dbc->txn,
&new_lsn, 0, DB_ADD_BIG, PGNO(pagep),
lastp ? PGNO(lastp) : PGNO_INVALID,
PGNO_INVALID, &tmp_dbt, &LSN(pagep),
lastp == NULL ? &null_lsn : &LSN(lastp),
&null_lsn)) != 0) {
if (lastp != NULL)
(void)__memp_fput(mpf, dbc->thread_info,
lastp, dbc->priority);
lastp = pagep;
break;
}
} else
LSN_NOT_LOGGED(new_lsn);
if (lastp != NULL)
LSN(lastp) = new_lsn;
LSN(pagep) = new_lsn;
OV_LEN(pagep) = pagespace;
OV_REF(pagep) = 1;
memcpy((u_int8_t *)pagep + P_OVERHEAD(dbp), p, pagespace);
if (lastp == NULL)
*pgnop = PGNO(pagep);
else {
lastp->next_pgno = PGNO(pagep);
pagep->prev_pgno = PGNO(lastp);
(void)__memp_fput(mpf,
dbc->thread_info, lastp, dbc->priority);
}
lastp = pagep;
}
if (lastp != NULL && (t_ret = __memp_fput(mpf,
dbc->thread_info, lastp, dbc->priority)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__db_ovref(dbc, pgno)
DBC *dbc;
db_pgno_t pgno;
{
DB *dbp;
DB_MPOOLFILE *mpf;
PAGE *h;
int ret;
dbp = dbc->dbp;
mpf = dbp->mpf;
if ((ret = __memp_fget(mpf, &pgno,
dbc->thread_info, dbc->txn, DB_MPOOL_DIRTY, &h)) != 0)
return (ret);
if (DBC_LOGGING(dbc)) {
if ((ret = __db_ovref_log(dbp,
dbc->txn, &LSN(h), 0, h->pgno, -1, &LSN(h))) != 0) {
(void)__memp_fput(mpf,
dbc->thread_info, h, dbc->priority);
return (ret);
}
} else
LSN_NOT_LOGGED(LSN(h));
--OV_REF(h);
return (__memp_fput(mpf, dbc->thread_info, h, dbc->priority));
}
int
__db_doff(dbc, pgno)
DBC *dbc;
db_pgno_t pgno;
{
DB *dbp;
DBT tmp_dbt;
DB_LSN null_lsn;
DB_MPOOLFILE *mpf;
PAGE *pagep;
int ret;
dbp = dbc->dbp;
mpf = dbp->mpf;
do {
if ((ret = __memp_fget(mpf, &pgno,
dbc->thread_info, dbc->txn, 0, &pagep)) != 0)
return (ret);
DB_ASSERT(dbp->env, TYPE(pagep) == P_OVERFLOW);
if (OV_REF(pagep) > 1) {
(void)__memp_fput(mpf,
dbc->thread_info, pagep, dbc->priority);
return (__db_ovref(dbc, pgno));
}
if ((ret = __memp_dirty(mpf, &pagep,
dbc->thread_info, dbc->txn, dbc->priority, 0)) != 0) {
(void)__memp_fput(mpf,
dbc->thread_info, pagep, dbc->priority);
return (ret);
}
if (DBC_LOGGING(dbc)) {
tmp_dbt.data = (u_int8_t *)pagep + P_OVERHEAD(dbp);
tmp_dbt.size = OV_LEN(pagep);
ZERO_LSN(null_lsn);
if ((ret = __db_big_log(dbp, dbc->txn,
&LSN(pagep), 0, DB_REM_BIG,
PGNO(pagep), PREV_PGNO(pagep),
NEXT_PGNO(pagep), &tmp_dbt,
&LSN(pagep), &null_lsn, &null_lsn)) != 0) {
(void)__memp_fput(mpf,
dbc->thread_info, pagep, dbc->priority);
return (ret);
}
} else
LSN_NOT_LOGGED(LSN(pagep));
pgno = pagep->next_pgno;
OV_LEN(pagep) = 0;
if ((ret = __db_free(dbc, pagep)) != 0)
return (ret);
} while (pgno != PGNO_INVALID);
return (0);
}
int
__db_moff(dbp, ip, txn, dbt, pgno, tlen, cmpfunc, cmpp)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
const DBT *dbt;
db_pgno_t pgno;
u_int32_t tlen;
int (*cmpfunc) __P((DB *, const DBT *, const DBT *)), *cmpp;
{
DBT local_dbt;
DB_MPOOLFILE *mpf;
PAGE *pagep;
void *buf;
u_int32_t bufsize, cmp_bytes, key_left;
u_int8_t *p1, *p2;
int ret;
mpf = dbp->mpf;
if (cmpfunc != NULL) {
memset(&local_dbt, 0, sizeof(local_dbt));
buf = NULL;
bufsize = 0;
if ((ret = __db_goff(dbp, ip, txn,
&local_dbt, tlen, pgno, &buf, &bufsize)) != 0)
return (ret);
*cmpp = cmpfunc(dbp, dbt, &local_dbt);
__os_free(dbp->env, buf);
return (0);
}
for (*cmpp = 0, p1 = dbt->data,
key_left = dbt->size; key_left > 0 && pgno != PGNO_INVALID;) {
if ((ret = __memp_fget(mpf, &pgno, ip, txn, 0, &pagep)) != 0)
return (ret);
cmp_bytes = OV_LEN(pagep) < key_left ? OV_LEN(pagep) : key_left;
tlen -= cmp_bytes;
key_left -= cmp_bytes;
for (p2 = (u_int8_t *)pagep + P_OVERHEAD(dbp);
cmp_bytes-- > 0; ++p1, ++p2)
if (*p1 != *p2) {
*cmpp = (long)*p1 - (long)*p2;
break;
}
pgno = NEXT_PGNO(pagep);
if ((ret = __memp_fput(mpf, ip, pagep, dbp->priority)) != 0)
return (ret);
if (*cmpp != 0)
return (0);
}
if (key_left > 0)
*cmpp = 1;
else if (tlen > 0)
*cmpp = -1;
else
*cmpp = 0;
return (0);
}
int
__db_coff(dbp, ip, txn, dbt, match, cmpfunc, cmpp)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
const DBT *dbt, *match;
int (*cmpfunc) __P((DB *, const DBT *, const DBT *)), *cmpp;
{
DBT local_key, local_match;
DB_MPOOLFILE *mpf;
PAGE *dbt_pagep, *match_pagep;
db_pgno_t dbt_pgno, match_pgno;
u_int32_t cmp_bytes, dbt_bufsz, dbt_len, match_bufsz;
u_int32_t match_len, max_data, page_sz;
u_int8_t *p1, *p2;
int ret;
void *dbt_buf, *match_buf;
mpf = dbp->mpf;
page_sz = dbp->pgsize;
*cmpp = 0;
dbt_buf = match_buf = NULL;
DB_ASSERT(dbp->env, HPAGE_PTYPE(dbt->data) == H_OFFPAGE);
DB_ASSERT(dbp->env, HPAGE_PTYPE(match->data) == H_OFFPAGE);
memcpy(&dbt_len, HOFFPAGE_TLEN(dbt->data), sizeof(u_int32_t));
memcpy(&dbt_pgno, HOFFPAGE_PGNO(dbt->data), sizeof(db_pgno_t));
memcpy(&match_len, HOFFPAGE_TLEN(match->data), sizeof(u_int32_t));
memcpy(&match_pgno, HOFFPAGE_PGNO(match->data), sizeof(db_pgno_t));
max_data = (dbt_len < match_len ? dbt_len : match_len);
if (cmpfunc != NULL) {
memset(&local_key, 0, sizeof(local_key));
memset(&local_match, 0, sizeof(local_match));
dbt_buf = match_buf = NULL;
dbt_bufsz = match_bufsz = 0;
if ((ret = __db_goff(dbp, ip, txn, &local_key, dbt_len,
dbt_pgno, &dbt_buf, &dbt_bufsz)) != 0)
goto err1;
if ((ret = __db_goff(dbp, ip, txn, &local_match, match_len,
match_pgno, &match_buf, &match_bufsz)) != 0)
goto err1;
*cmpp = cmpfunc(dbp, &local_key, &local_match);
err1: if (dbt_buf != NULL)
__os_free(dbp->env, dbt_buf);
if (match_buf != NULL)
__os_free(dbp->env, match_buf);
return (ret);
}
while (dbt_pgno != PGNO_INVALID && match_pgno != PGNO_INVALID) {
if ((ret =
__memp_fget(mpf, &dbt_pgno, ip, txn, 0, &dbt_pagep)) != 0)
return (ret);
if ((ret =
__memp_fget(mpf, &match_pgno,
ip, txn, 0, &match_pagep)) != 0) {
(void)__memp_fput(
mpf, ip, dbt_pagep, DB_PRIORITY_UNCHANGED);
return (ret);
}
cmp_bytes = page_sz < max_data ? page_sz : max_data;
for (p1 = (u_int8_t *)dbt_pagep + P_OVERHEAD(dbp),
p2 = (u_int8_t *)match_pagep + P_OVERHEAD(dbp);
cmp_bytes-- > 0; ++p1, ++p2)
if (*p1 != *p2) {
*cmpp = (long)*p1 - (long)*p2;
break;
}
dbt_pgno = NEXT_PGNO(dbt_pagep);
match_pgno = NEXT_PGNO(match_pagep);
max_data -= page_sz;
if ((ret = __memp_fput(mpf,
ip, dbt_pagep, DB_PRIORITY_UNCHANGED)) != 0) {
(void)__memp_fput(mpf,
ip, match_pagep, DB_PRIORITY_UNCHANGED);
return (ret);
}
if ((ret = __memp_fput(mpf,
ip, match_pagep, DB_PRIORITY_UNCHANGED)) != 0)
return (ret);
if (*cmpp != 0)
return (0);
}
if (dbt_len > match_len)
*cmpp = 1;
else if (match_len > dbt_len)
*cmpp = -1;
else
*cmpp = 0;
return (0);
}