#include "db_config.h"
#ifndef lint
static const char revid[] = "$Id: dbreg.c,v 1.2 2004/03/30 01:21:33 jtownsen Exp $";
#endif
#ifndef NO_SYSTEM_INCLUDES
#include <sys/types.h>
#include <string.h>
#endif
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/log.h"
#include "dbinc/txn.h"
#include "dbinc/db_am.h"
int
__dbreg_setup(dbp, name, create_txnid)
DB *dbp;
const char *name;
u_int32_t create_txnid;
{
DB_ENV *dbenv;
DB_LOG *dblp;
FNAME *fnp;
int ret;
size_t len;
void *namep;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
fnp = NULL;
namep = NULL;
R_LOCK(dbenv, &dblp->reginfo);
if ((ret =
__db_shalloc(dblp->reginfo.addr, sizeof(FNAME), 0, &fnp)) != 0)
goto err;
memset(fnp, 0, sizeof(FNAME));
if (name != NULL) {
len = strlen(name) + 1;
if ((ret =
__db_shalloc(dblp->reginfo.addr, len, 0, &namep)) != 0)
goto err;
fnp->name_off = R_OFFSET(&dblp->reginfo, namep);
memcpy(namep, name, len);
} else
fnp->name_off = INVALID_ROFF;
R_UNLOCK(dbenv, &dblp->reginfo);
fnp->id = DB_LOGFILEID_INVALID;
fnp->s_type = dbp->type;
memcpy(fnp->ufid, dbp->fileid, DB_FILE_ID_LEN);
fnp->meta_pgno = dbp->meta_pgno;
fnp->create_txnid = create_txnid;
dbp->log_filename = fnp;
return (0);
err: R_UNLOCK(dbenv, &dblp->reginfo);
if (ret == ENOMEM)
__db_err(dbenv,
"Logging region out of memory; you may need to increase its size");
return (ret);
}
int
__dbreg_teardown(dbp)
DB *dbp;
{
DB_ENV *dbenv;
DB_LOG *dblp;
FNAME *fnp;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
fnp = dbp->log_filename;
if (fnp == NULL)
return (0);
DB_ASSERT(fnp->id == DB_LOGFILEID_INVALID);
R_LOCK(dbenv, &dblp->reginfo);
if (fnp->name_off != INVALID_ROFF)
__db_shalloc_free(dblp->reginfo.addr,
R_ADDR(&dblp->reginfo, fnp->name_off));
__db_shalloc_free(dblp->reginfo.addr, fnp);
R_UNLOCK(dbenv, &dblp->reginfo);
dbp->log_filename = NULL;
return (0);
}
int
__dbreg_new_id(dbp, txn)
DB *dbp;
DB_TXN *txn;
{
DB_ENV *dbenv;
DB_LOG *dblp;
FNAME *fnp;
LOG *lp;
int32_t id;
int ret;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
fnp = dbp->log_filename;
MUTEX_LOCK(dbenv, &lp->fq_mutex);
if (fnp->id != DB_LOGFILEID_INVALID) {
MUTEX_UNLOCK(dbenv, &lp->fq_mutex);
return (0);
}
if ((ret = __dbreg_get_id(dbp, txn, &id)) == 0)
fnp->id = id;
MUTEX_UNLOCK(dbenv, &lp->fq_mutex);
return (ret);
}
int
__dbreg_get_id(dbp, txn, idp)
DB *dbp;
DB_TXN *txn;
int32_t *idp;
{
DBT fid_dbt, r_name;
DB_ENV *dbenv;
DB_LOG *dblp;
DB_LSN unused;
FNAME *fnp;
LOG *lp;
int32_t id;
int ret;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
fnp = dbp->log_filename;
if ((ret = __dbreg_pop_id(dbenv, &id)) != 0)
goto err;
if (id == DB_LOGFILEID_INVALID)
id = lp->fid_max++;
fnp->is_durable = !F_ISSET(dbp, DB_AM_NOT_DURABLE);
SH_TAILQ_INSERT_HEAD(&lp->fq, fnp, q, __fname);
DB_ASSERT(!F_ISSET(dbp, DB_AM_RECOVER));
memset(&fid_dbt, 0, sizeof(fid_dbt));
memset(&r_name, 0, sizeof(r_name));
if (fnp->name_off != INVALID_ROFF) {
r_name.data = R_ADDR(&dblp->reginfo, fnp->name_off);
r_name.size = (u_int32_t)strlen((char *)r_name.data) + 1;
}
fid_dbt.data = dbp->fileid;
fid_dbt.size = DB_FILE_ID_LEN;
if ((ret = __dbreg_register_log(dbenv, txn, &unused,
F_ISSET(dbp, DB_AM_NOT_DURABLE) ? DB_LOG_NOT_DURABLE : 0,
DBREG_OPEN, r_name.size == 0 ? NULL : &r_name, &fid_dbt, id,
fnp->s_type, fnp->meta_pgno, fnp->create_txnid)) != 0)
goto err;
fnp->create_txnid = TXN_INVALID;
DB_ASSERT(dbp->type == fnp->s_type);
DB_ASSERT(dbp->meta_pgno == fnp->meta_pgno);
if ((ret = __dbreg_add_dbentry(dbenv, dblp, dbp, id)) != 0)
goto err;
err:
if (ret != 0 && id != DB_LOGFILEID_INVALID) {
(void)__dbreg_revoke_id(dbp, 1, id);
id = DB_LOGFILEID_INVALID;
}
*idp = id;
return (ret);
}
int
__dbreg_assign_id(dbp, id)
DB *dbp;
int32_t id;
{
DB *close_dbp;
DB_ENV *dbenv;
DB_LOG *dblp;
FNAME *close_fnp, *fnp;
LOG *lp;
int ret;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
fnp = dbp->log_filename;
close_dbp = NULL;
close_fnp = NULL;
MUTEX_LOCK(dbenv, &lp->fq_mutex);
DB_ASSERT(fnp->id == DB_LOGFILEID_INVALID);
if (__dbreg_id_to_fname(dblp, id, 1, &close_fnp) == 0) {
ret = __dbreg_id_to_db_int(dbenv, NULL, &close_dbp, id, 0, 0);
if (ret == ENOENT) {
ret = 0;
goto cont;
} else if (ret != 0)
goto err;
if ((ret = __dbreg_revoke_id(close_dbp, 1,
DB_LOGFILEID_INVALID)) != 0)
goto err;
}
cont: if ((ret = __dbreg_pluck_id(dbenv, id)) != 0)
goto err;
if (id >= lp->fid_max)
lp->fid_max = id + 1;
fnp->id = id;
fnp->is_durable = !F_ISSET(dbp, DB_AM_NOT_DURABLE);
SH_TAILQ_INSERT_HEAD(&lp->fq, fnp, q, __fname);
if ((ret = __dbreg_add_dbentry(dbenv, dblp, dbp, id)) != 0)
goto err;
err: MUTEX_UNLOCK(dbenv, &lp->fq_mutex);
if (close_dbp != NULL)
(void)__db_close(close_dbp, NULL, DB_NOSYNC);
return (ret);
}
int
__dbreg_revoke_id(dbp, have_lock, force_id)
DB *dbp;
int have_lock;
int32_t force_id;
{
DB_ENV *dbenv;
DB_LOG *dblp;
FNAME *fnp;
LOG *lp;
int32_t id;
int ret;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
fnp = dbp->log_filename;
if (fnp == NULL)
return (0);
if (force_id != DB_LOGFILEID_INVALID)
id = force_id;
else if (fnp->id == DB_LOGFILEID_INVALID)
return (0);
else
id = fnp->id;
if (!have_lock)
MUTEX_LOCK(dbenv, &lp->fq_mutex);
fnp->id = DB_LOGFILEID_INVALID;
SH_TAILQ_REMOVE(&lp->fq, fnp, q, __fname);
__dbreg_rem_dbentry(dblp, id);
ret = __dbreg_push_id(dbenv, id);
if (!have_lock)
MUTEX_UNLOCK(dbenv, &lp->fq_mutex);
return (ret);
}
int
__dbreg_close_id(dbp, txn)
DB *dbp;
DB_TXN *txn;
{
DBT fid_dbt, r_name, *dbtp;
DB_ENV *dbenv;
DB_LOG *dblp;
DB_LSN r_unused;
FNAME *fnp;
LOG *lp;
int ret;
dbenv = dbp->dbenv;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
fnp = dbp->log_filename;
if (fnp == NULL || fnp->id == DB_LOGFILEID_INVALID)
return (0);
MUTEX_LOCK(dbenv, &lp->fq_mutex);
if (fnp->name_off == INVALID_ROFF)
dbtp = NULL;
else {
memset(&r_name, 0, sizeof(r_name));
r_name.data = R_ADDR(&dblp->reginfo, fnp->name_off);
r_name.size =
(u_int32_t)strlen((char *)r_name.data) + 1;
dbtp = &r_name;
}
memset(&fid_dbt, 0, sizeof(fid_dbt));
fid_dbt.data = fnp->ufid;
fid_dbt.size = DB_FILE_ID_LEN;
if ((ret = __dbreg_register_log(dbenv, txn, &r_unused,
F_ISSET(dbp, DB_AM_NOT_DURABLE) ? DB_LOG_NOT_DURABLE : 0,
DBREG_CLOSE, dbtp, &fid_dbt, fnp->id,
fnp->s_type, fnp->meta_pgno, TXN_INVALID)) != 0)
goto err;
ret = __dbreg_revoke_id(dbp, 1, DB_LOGFILEID_INVALID);
err: MUTEX_UNLOCK(dbenv, &lp->fq_mutex);
return (ret);
}