#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/db_swap.h"
#include "dbinc/db_am.h"
#include "dbinc/mp.h"
static int __env_fileid_reset __P((ENV *,
DB_THREAD_INFO *, const char *, int));
int
__env_fileid_reset_pp(dbenv, name, flags)
DB_ENV *dbenv;
const char *name;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int ret;
env = dbenv->env;
ENV_ILLEGAL_BEFORE_OPEN(env, "DB_ENV->fileid_reset");
if (flags != 0 && flags != DB_ENCRYPT)
return (__db_ferr(env, "DB_ENV->fileid_reset", 0));
ENV_ENTER(env, ip);
REPLICATION_WRAP(env,
(__env_fileid_reset(env, ip, name, LF_ISSET(DB_ENCRYPT) ? 1 : 0)),
1, ret);
ENV_LEAVE(env, ip);
return (ret);
}
static int
__env_fileid_reset(env, ip, name, encrypted)
ENV *env;
DB_THREAD_INFO *ip;
const char *name;
int encrypted;
{
DB *dbp;
DBC *dbcp;
DBT key, data;
DB_FH *fhp;
DB_MPOOLFILE *mpf;
DB_PGINFO cookie;
db_pgno_t pgno;
int t_ret, ret;
size_t n;
char *real_name;
u_int8_t fileid[DB_FILE_ID_LEN], mbuf[DBMETASIZE];
void *pagep;
dbp = NULL;
dbcp = NULL;
fhp = NULL;
real_name = NULL;
if ((ret =
__db_appname(env, DB_APP_DATA, name, 0, NULL, &real_name)) != 0)
return (ret);
if ((ret = __os_fileid(env, real_name, 1, fileid)) != 0)
goto err;
if ((ret = __os_open(env, real_name, 0, 0, 0, &fhp)) != 0) {
__db_err(env, ret, "%s", real_name);
goto err;
}
if ((ret = __os_read(env, fhp, mbuf, sizeof(mbuf), &n)) != 0)
goto err;
if (n != sizeof(mbuf)) {
ret = EINVAL;
__db_errx(env,
"%s: unexpected file type or format", real_name);
goto err;
}
if ((ret = __db_create_internal(&dbp, env, 0)) != 0)
goto err;
if (encrypted && (ret = __db_set_flags(dbp, DB_ENCRYPT)) != 0)
goto err;
if ((ret = __db_meta_setup(env,
dbp, real_name, (DBMETA *)mbuf, 0, DB_CHK_META)) != 0)
goto err;
memcpy(((DBMETA *)mbuf)->uid, fileid, DB_FILE_ID_LEN);
cookie.db_pagesize = sizeof(mbuf);
cookie.flags = dbp->flags;
cookie.type = dbp->type;
key.data = &cookie;
if ((ret = __db_pgout(env->dbenv, 0, mbuf, &key)) != 0)
goto err;
if ((ret = __os_seek(env, fhp, 0, 0, 0)) != 0)
goto err;
if ((ret = __os_write(env, fhp, mbuf, sizeof(mbuf), &n)) != 0)
goto err;
if ((ret = __os_fsync(env, fhp)) != 0)
goto err;
if ((ret = __db_open(dbp, ip, NULL,
name, NULL, DB_UNKNOWN, DB_RDWRMASTER, 0, PGNO_BASE_MD)) != 0)
goto err;
if (!F_ISSET(dbp, DB_AM_SUBDB))
goto err;
mpf = dbp->mpf;
memset(&key, 0, sizeof(key));
memset(&data, 0, sizeof(data));
if ((ret = __db_cursor(dbp, ip, NULL, &dbcp, 0)) != 0)
goto err;
while ((ret = __dbc_get(dbcp, &key, &data, DB_NEXT)) == 0) {
memcpy(&pgno, data.data, sizeof(db_pgno_t));
DB_NTOHL_SWAP(env, &pgno);
if ((ret = __memp_fget(mpf, &pgno, ip, NULL,
DB_MPOOL_DIRTY, &pagep)) != 0)
goto err;
memcpy(((DBMETA *)pagep)->uid, fileid, DB_FILE_ID_LEN);
if ((ret = __memp_fput(mpf, ip, pagep, dbcp->priority)) != 0)
goto err;
}
if (ret == DB_NOTFOUND)
ret = 0;
err: if (dbcp != NULL && (t_ret = __dbc_close(dbcp)) != 0 && ret == 0)
ret = t_ret;
if (dbp != NULL && (t_ret = __db_close(dbp, NULL, 0)) != 0 && ret == 0)
ret = t_ret;
if (fhp != NULL &&
(t_ret = __os_closehandle(env, fhp)) != 0 && ret == 0)
ret = t_ret;
if (real_name != NULL)
__os_free(env, real_name);
return (ret);
}