#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/db_am.h"
#include "dbinc/fop.h"
#include "dbinc/lock.h"
#include "dbinc/log.h"
#include "dbinc/mp.h"
#include "dbinc/qam.h"
#include "dbinc/txn.h"
static int __rep_check_uid __P((ENV *, u_int8_t *, u_int8_t *, u_int32_t,
u_int8_t *));
static int __rep_filedone __P((ENV *, DB_THREAD_INFO *ip, int,
REP *, __rep_fileinfo_args *, u_int32_t));
static int __rep_find_dbs __P((ENV *, u_int32_t, u_int8_t **, size_t *,
size_t *, u_int32_t *));
static int __rep_get_fileinfo __P((ENV *, const char *,
const char *, __rep_fileinfo_args *, u_int8_t *, u_int32_t *));
static int __rep_get_file_list __P((ENV *,
DB_FH *, u_int32_t, u_int32_t *, DBT *));
static int __rep_log_setup __P((ENV *,
REP *, u_int32_t, u_int32_t, DB_LSN *));
static int __rep_mpf_open __P((ENV *, DB_MPOOLFILE **,
__rep_fileinfo_args *, u_int32_t));
static int __rep_nextfile __P((ENV *, int, REP *));
static int __rep_page_gap __P((ENV *,
REP *, __rep_fileinfo_args *, u_int32_t));
static int __rep_page_sendpages __P((ENV *, DB_THREAD_INFO *, int,
__rep_control_args *, __rep_fileinfo_args *, DB_MPOOLFILE *, DB *));
static int __rep_queue_filedone __P((ENV *,
DB_THREAD_INFO *, REP *, __rep_fileinfo_args *));
static int __rep_remove_all __P((ENV *, u_int32_t, DBT *));
static int __rep_remove_file __P((ENV *, u_int8_t *, const char *,
u_int32_t, u_int32_t));
static int __rep_remove_logs __P((ENV *));
static int __rep_remove_by_list __P((ENV *, u_int32_t,
u_int8_t *, u_int32_t, u_int32_t));
static int __rep_remove_by_prefix __P((ENV *, const char *, const char *,
size_t, APPNAME));
static int __rep_walk_dir __P((ENV *, const char *, u_int32_t, u_int8_t **,
u_int8_t *, size_t *, size_t *, u_int32_t *));
static int __rep_write_page __P((ENV *,
DB_THREAD_INFO *, REP *, __rep_fileinfo_args *));
int
__rep_update_req(env, rp, eid)
ENV *env;
__rep_control_args *rp;
int eid;
{
DBT updbt, vdbt;
DB_LOG *dblp;
DB_LOGC *logc;
DB_LSN lsn;
__rep_update_args u_args;
size_t filelen, filesz, updlen;
u_int32_t filecnt, flag, version;
u_int8_t *buf, *fp;
int ret, t_ret;
dblp = env->lg_handle;
logc = NULL;
filecnt = 0;
filelen = 0;
updlen = 0;
filesz = MEGABYTE;
if ((ret = __os_calloc(env, 1, filesz, &buf)) != 0)
return (ret);
fp = buf + __REP_UPDATE_SIZE;
if ((ret = __rep_find_dbs(env, rp->rep_version,
&fp, &filesz, &filelen, &filecnt)) != 0)
goto err;
flag = DB_SET;
if ((ret = __log_get_stable_lsn(env, &lsn)) != 0) {
if (ret != DB_NOTFOUND)
goto err;
ret = 0;
flag = DB_FIRST;
}
if ((ret = __log_cursor(env, &logc)) != 0)
goto err;
memset(&vdbt, 0, sizeof(vdbt));
if ((ret = __logc_get(logc, &lsn, &vdbt, flag)) != 0) {
if (ret != DB_NOTFOUND)
goto err;
INIT_LSN(lsn);
version = DB_LOGVERSION;
} else {
if ((ret = __logc_version(logc, &version)) != 0)
goto err;
}
u_args.first_lsn = lsn;
u_args.first_vers = version;
u_args.num_files = filecnt;
if ((ret = __rep_update_marshal(env, rp->rep_version,
&u_args, buf, filesz, &updlen)) != 0)
goto err;
DB_INIT_DBT(updbt, buf, filelen + updlen);
LOG_SYSTEM_LOCK(env);
lsn = ((LOG *)dblp->reginfo.primary)->lsn;
LOG_SYSTEM_UNLOCK(env);
(void)__rep_send_message(
env, eid, REP_UPDATE, &lsn, &updbt, 0, 0);
err: __os_free(env, buf);
if (logc != NULL && (t_ret = __logc_close(logc)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
static int
__rep_find_dbs(env, version, fp, fileszp, filelenp, filecntp)
ENV *env;
u_int32_t version;
u_int8_t **fp;
size_t *fileszp, *filelenp;
u_int32_t *filecntp;
{
DB_ENV *dbenv;
int ret;
char **ddir, *real_dir;
u_int8_t *origfp;
dbenv = env->dbenv;
ret = 0;
real_dir = NULL;
if (dbenv->db_data_dir == NULL) {
ret = __rep_walk_dir(env, env->db_home, version, fp, NULL,
fileszp, filelenp, filecntp);
} else {
origfp = *fp;
for (ddir = dbenv->db_data_dir; *ddir != NULL; ++ddir) {
if ((ret = __db_appname(env, DB_APP_NONE,
*ddir, 0, NULL, &real_dir)) != 0)
break;
if ((ret = __rep_walk_dir(env, real_dir, version, fp,
origfp, fileszp, filelenp, filecntp)) != 0)
break;
__os_free(env, real_dir);
real_dir = NULL;
}
}
if (ret == 0)
ret = __rep_walk_dir(env, NULL, version,
fp, NULL, fileszp, filelenp, filecntp);
if (real_dir != NULL)
__os_free(env, real_dir);
return (ret);
}
static int
__rep_walk_dir(env, dir, version, fp, origfp, fileszp, filelenp, filecntp)
ENV *env;
const char *dir;
u_int32_t version;
u_int8_t **fp, *origfp;
size_t *fileszp, *filelenp;
u_int32_t *filecntp;
{
__rep_fileinfo_args tmpfp;
size_t len, offset;
int cnt, first_file, i, ret;
u_int8_t *rfp, uid[DB_FILE_ID_LEN];
char *file, **names, *subdb;
if (dir == NULL) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Walk_dir: Getting info for in-memory named files"));
if ((ret = __memp_inmemlist(env, &names, &cnt)) != 0)
return (ret);
} else {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Walk_dir: Getting info for dir: %s", dir));
if ((ret = __os_dirlist(env, dir, 0, &names, &cnt)) != 0)
return (ret);
}
rfp = NULL;
if (fp != NULL)
rfp = *fp;
RPRINT(env, DB_VERB_REP_SYNC, (env, "Walk_dir: Dir %s has %d files",
(dir == NULL) ? "INMEM" : dir, cnt));
first_file = 1;
for (i = 0; i < cnt; i++) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Walk_dir: File %d name: %s", i, names[i]));
if (strncmp(names[i],
DB_REGION_PREFIX, sizeof(DB_REGION_PREFIX) - 1) == 0)
continue;
if (strncmp(names[i], "DB_CONFIG", 9) == 0)
continue;
if (strncmp(names[i], "log.", 4) == 0)
continue;
if (dir == NULL) {
file = NULL;
subdb = names[i];
} else {
file = names[i];
subdb = NULL;
}
if ((ret = __rep_get_fileinfo(env,
file, subdb, &tmpfp, uid, filecntp)) != 0) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Walk_dir: File %d %s: returned error %s",
i, names[i], db_strerror(ret)));
ret = 0;
continue;
}
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Walk_dir: File %d (of %d) %s at 0x%lx: pgsize %lu, max_pgno %lu",
tmpfp.filenum, *filecntp, names[i], P_TO_ULONG(rfp),
(u_long)tmpfp.pgsize, (u_long)tmpfp.max_pgno));
if (first_file && origfp != NULL) {
if (rfp != origfp &&
(ret = __rep_check_uid(env, origfp,
origfp + *filelenp, version, uid)) != 0) {
if (ret == DB_KEYEXIST) {
ret = 0;
(*filecntp)--;
}
goto err;
}
first_file = 0;
}
DB_SET_DBT(tmpfp.info, names[i], strlen(names[i]) + 1);
DB_SET_DBT(tmpfp.uid, uid, DB_FILE_ID_LEN);
retry: ret = __rep_fileinfo_marshal(env, version,
&tmpfp, rfp, *fileszp, &len);
if (ret == ENOMEM) {
offset = (size_t)(rfp - *fp);
*fileszp *= 2;
*fp -= __REP_UPDATE_SIZE;
if ((ret = __os_realloc(env, *fileszp, *fp)) != 0)
break;
*fp += __REP_UPDATE_SIZE;
rfp = *fp + offset;
goto retry;
}
rfp += len;
*fp = rfp;
*filelenp += len;
}
err:
__os_dirfree(env, names, cnt);
return (ret);
}
static int
__rep_check_uid(env, origfp, endfp, version, uid)
ENV *env;
u_int32_t version;
u_int8_t *origfp, *endfp, *uid;
{
__rep_fileinfo_args *rfp;
size_t filesz;
u_int8_t *fp, *fuid, *new_fp;
int ret;
ret = 0;
fp = origfp;
rfp = NULL;
filesz = (uintptr_t)endfp - (uintptr_t)origfp;
while (fp <= endfp) {
if ((ret = __rep_fileinfo_unmarshal(env, version,
&rfp, fp, filesz, &new_fp)) != 0) {
__db_errx(env, "rep_check_uid: Could not malloc");
goto err;
}
filesz -= (u_int32_t)(new_fp - fp);
fp = new_fp;
fuid = (u_int8_t *)rfp->uid.data;
if (memcmp(fuid, uid, DB_FILE_ID_LEN) == 0) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Check_uid: Found matching file."));
ret = DB_KEYEXIST;
goto err;
}
__os_free(env, rfp);
rfp = NULL;
}
err:
if (rfp != NULL)
__os_free(env, rfp);
return (ret);
}
static int
__rep_get_fileinfo(env, file, subdb, rfp, uid, filecntp)
ENV *env;
const char *file, *subdb;
__rep_fileinfo_args *rfp;
u_int8_t *uid;
u_int32_t *filecntp;
{
DB *dbp;
DBC *dbc;
DBMETA *dbmeta;
DB_LOCK lk;
DB_MPOOLFILE *mpf;
DB_THREAD_INFO *ip;
PAGE *pagep;
int lorder, ret, t_ret;
dbp = NULL;
dbc = NULL;
pagep = NULL;
mpf = NULL;
LOCK_INIT(lk);
ENV_GET_THREAD_INFO(env, ip);
if ((ret = __db_create_internal(&dbp, env, 0)) != 0)
goto err;
if ((ret = __db_open(dbp, ip, NULL, file, subdb, DB_UNKNOWN,
DB_RDONLY | (F_ISSET(env, ENV_THREAD) ? DB_THREAD : 0),
0, PGNO_BASE_MD)) != 0)
goto err;
if ((ret = __db_cursor(dbp, ip, NULL, &dbc, 0)) != 0)
goto err;
if ((ret = __db_lget(
dbc, 0, dbp->meta_pgno, DB_LOCK_READ, 0, &lk)) != 0)
goto err;
if ((ret = __memp_fget(dbp->mpf, &dbp->meta_pgno, ip, dbc->txn,
0, &pagep)) != 0)
goto err;
dbmeta = (DBMETA *)pagep;
rfp->pgno = 0;
if (dbp->type == DB_QUEUE)
rfp->max_pgno = 0;
else
rfp->max_pgno = dbmeta->last_pgno;
rfp->pgsize = dbp->pgsize;
memcpy(uid, dbp->fileid, DB_FILE_ID_LEN);
rfp->filenum = (*filecntp)++;
rfp->type = (u_int32_t)dbp->type;
rfp->db_flags = dbp->flags;
rfp->finfo_flags = 0;
(void)__db_get_lorder(dbp, &lorder);
if (lorder == 1234)
FLD_SET(rfp->finfo_flags, REPINFO_DB_LITTLEENDIAN);
else
FLD_CLR(rfp->finfo_flags, REPINFO_DB_LITTLEENDIAN);
ret = __memp_fput(dbp->mpf, ip, pagep, dbc->priority);
pagep = NULL;
if ((t_ret = __LPUT(dbc, lk)) != 0 && ret == 0)
ret = t_ret;
if (ret != 0)
goto err;
err:
if ((t_ret = __LPUT(dbc, lk)) != 0 && ret == 0)
ret = t_ret;
if (pagep != NULL && (t_ret =
__memp_fput(mpf, ip, pagep, dbc->priority)) != 0 && ret == 0)
ret = t_ret;
if (dbc != NULL && (t_ret = __dbc_close(dbc)) != 0 && ret == 0)
ret = t_ret;
if (dbp != NULL && (t_ret = __db_close(dbp, NULL, 0)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__rep_page_req(env, ip, eid, rp, rec)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
__rep_control_args *rp;
DBT *rec;
{
__rep_fileinfo_args *msgfp;
DB_MPOOLFILE *mpf;
DB_REP *db_rep;
REP *rep;
int ret, t_ret;
u_int8_t *next;
db_rep = env->rep_handle;
rep = db_rep->region;
if ((ret = __rep_fileinfo_unmarshal(env, rp->rep_version,
&msgfp, rec->data, rec->size, &next)) != 0)
return (ret);
RPRINT(env, DB_VERB_REP_SYNC,
(env, "page_req: file %d page %lu to %lu",
msgfp->filenum, (u_long)msgfp->pgno, (u_long)msgfp->max_pgno));
RPRINT(env, DB_VERB_REP_SYNC,
(env, "page_req: Open %d via mpf_open", msgfp->filenum));
if ((ret = __rep_mpf_open(env, &mpf, msgfp, 0)) != 0) {
RPRINT(env, DB_VERB_REP_SYNC,
(env, "page_req: Open %d failed", msgfp->filenum));
if (F_ISSET(rep, REP_F_MASTER))
(void)__rep_send_message(env, eid, REP_FILE_FAIL,
NULL, rec, 0, 0);
else
ret = DB_NOTFOUND;
goto err;
}
ret = __rep_page_sendpages(env, ip, eid, rp, msgfp, mpf, NULL);
t_ret = __memp_fclose(mpf, 0);
if (ret == 0 && t_ret != 0)
ret = t_ret;
err:
__os_free(env, msgfp);
return (ret);
}
static int
__rep_page_sendpages(env, ip, eid, rp, msgfp, mpf, dbp)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
__rep_control_args *rp;
__rep_fileinfo_args *msgfp;
DB_MPOOLFILE *mpf;
DB *dbp;
{
DB *qdbp;
DBC *qdbc;
DBT lockdbt, msgdbt;
DB_LOCK lock;
DB_LOCKER *locker;
DB_LOCK_ILOCK lock_obj;
DB_LOG *dblp;
DB_LSN lsn;
DB_REP *db_rep;
PAGE *pagep;
REP *rep;
REP_BULK bulk;
REP_THROTTLE repth;
db_pgno_t p;
uintptr_t bulkoff;
size_t len, msgsz;
u_int32_t bulkflags, use_bulk;
int opened, ret, t_ret;
u_int8_t *buf;
db_rep = env->rep_handle;
rep = db_rep->region;
locker = NULL;
opened = 0;
t_ret = 0;
qdbp = NULL;
qdbc = NULL;
buf = NULL;
bulk.addr = NULL;
use_bulk = FLD_ISSET(rep->config, REP_C_BULK);
if (msgfp->type == (u_int32_t)DB_QUEUE) {
if (dbp == NULL) {
if ((ret = __db_create_internal(&qdbp, env, 0)) != 0)
goto err;
if ((ret = __db_open(qdbp, ip, NULL,
FLD_ISSET(msgfp->db_flags, DB_AM_INMEM) ?
NULL : msgfp->info.data,
FLD_ISSET(msgfp->db_flags, DB_AM_INMEM) ?
msgfp->info.data : NULL,
DB_UNKNOWN,
DB_RDONLY | (F_ISSET(env, ENV_THREAD) ? DB_THREAD : 0),
0, PGNO_BASE_MD)) != 0)
goto err;
opened = 1;
} else
qdbp = dbp;
if ((ret = __db_cursor(qdbp, ip, NULL, &qdbc, 0)) != 0)
goto err;
}
msgsz = __REP_FILEINFO_SIZE + DB_FILE_ID_LEN + msgfp->pgsize;
if ((ret = __os_calloc(env, 1, msgsz, &buf)) != 0)
goto err;
memset(&msgdbt, 0, sizeof(msgdbt));
RPRINT(env, DB_VERB_REP_SYNC,
(env, "sendpages: file %d page %lu to %lu",
msgfp->filenum, (u_long)msgfp->pgno, (u_long)msgfp->max_pgno));
memset(&repth, 0, sizeof(repth));
if (use_bulk && (ret = __rep_bulk_alloc(env, &bulk, eid,
&bulkoff, &bulkflags, REP_BULK_PAGE)) != 0)
goto err;
REP_SYSTEM_LOCK(env);
repth.gbytes = rep->gbytes;
repth.bytes = rep->bytes;
repth.type = REP_PAGE;
repth.data_dbt = &msgdbt;
REP_SYSTEM_UNLOCK(env);
LOCK_INIT(lock);
memset(&lock_obj, 0, sizeof(lock_obj));
if ((ret = __lock_id(env, NULL, &locker)) != 0)
goto err;
memcpy(lock_obj.fileid, mpf->fileid, DB_FILE_ID_LEN);
lock_obj.type = DB_PAGE_LOCK;
memset(&lockdbt, 0, sizeof(lockdbt));
lockdbt.data = &lock_obj;
lockdbt.size = sizeof(lock_obj);
for (p = msgfp->pgno; p <= msgfp->max_pgno; p++) {
lock_obj.pgno = p;
if ((ret = __lock_get(env, locker, DB_LOCK_NOWAIT, &lockdbt,
DB_LOCK_READ, &lock)) != 0) {
if (ret == DB_LOCK_NOTGRANTED) {
ret = 0;
continue;
}
goto err;
}
if (msgfp->type == (u_int32_t)DB_QUEUE && p != 0)
#ifdef HAVE_QUEUE
ret = __qam_fget(qdbc, &p, DB_MPOOL_CREATE, &pagep);
#else
ret = DB_PAGE_NOTFOUND;
#endif
else
ret = __memp_fget(mpf, &p, ip, NULL,
DB_MPOOL_CREATE, &pagep);
msgfp->pgno = p;
if (ret == DB_PAGE_NOTFOUND) {
ZERO_LSN(lsn);
if (F_ISSET(rep, REP_F_MASTER)) {
ret = 0;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"sendpages: PAGE_FAIL on page %lu",
(u_long)p));
(void)__rep_send_message(env, eid,
REP_PAGE_FAIL, &lsn, &msgdbt, 0, 0);
} else
ret = DB_NOTFOUND;
goto lockerr;
} else if (ret != 0)
goto lockerr;
else
DB_SET_DBT(msgfp->info, pagep, msgfp->pgsize);
len = 0;
if (F_ISSET(env, ENV_LITTLEENDIAN))
FLD_SET(msgfp->finfo_flags, REPINFO_PG_LITTLEENDIAN);
else
FLD_CLR(msgfp->finfo_flags, REPINFO_PG_LITTLEENDIAN);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"sendpages: %lu, page lsn [%lu][%lu]", (u_long)p,
(u_long)pagep->lsn.file, (u_long)pagep->lsn.offset));
ret = __rep_fileinfo_marshal(env, rp->rep_version,
msgfp, buf, msgsz, &len);
if (msgfp->type != (u_int32_t)DB_QUEUE || p == 0)
t_ret = __memp_fput(mpf,
ip, pagep, DB_PRIORITY_UNCHANGED);
#ifdef HAVE_QUEUE
else
t_ret = __qam_fput(qdbc, p, pagep, qdbp->priority);
#endif
if (t_ret != 0 && ret == 0)
ret = t_ret;
if ((t_ret = __ENV_LPUT(env, lock)) != 0 && ret == 0)
ret = t_ret;
if (ret != 0)
goto err;
DB_ASSERT(env, len <= msgsz);
DB_SET_DBT(msgdbt, buf, len);
dblp = env->lg_handle;
LOG_SYSTEM_LOCK(env);
repth.lsn = ((LOG *)dblp->reginfo.primary)->lsn;
LOG_SYSTEM_UNLOCK(env);
if (use_bulk)
ret = __rep_bulk_message(env, &bulk, &repth,
&repth.lsn, &msgdbt, 0);
if (!use_bulk || ret == DB_REP_BULKOVF)
ret = __rep_send_throttle(env, eid, &repth, 0, 0);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"sendpages: %lu, lsn [%lu][%lu]", (u_long)p,
(u_long)repth.lsn.file, (u_long)repth.lsn.offset));
if (repth.type == REP_PAGE_MORE || ret != 0) {
if (ret == DB_REP_UNAVAIL)
ret = 0;
break;
}
}
if (0) {
lockerr: if ((t_ret = __ENV_LPUT(env, lock)) != 0 && ret == 0)
ret = t_ret;
}
err:
if (use_bulk && bulk.addr != NULL &&
(t_ret = __rep_bulk_free(env, &bulk, 0)) != 0 && ret == 0)
ret = t_ret;
if (qdbc != NULL && (t_ret = __dbc_close(qdbc)) != 0 && ret == 0)
ret = t_ret;
if (opened && (t_ret = __db_close(qdbp, NULL, DB_NOSYNC)) != 0 &&
ret == 0)
ret = t_ret;
if (buf != NULL)
__os_free(env, buf);
if (locker != NULL && (t_ret = __lock_id_free(env,
locker)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__rep_update_setup(env, eid, rp, rec)
ENV *env;
int eid;
__rep_control_args *rp;
DBT *rec;
{
DB_LOG *dblp;
DB_REP *db_rep;
DB_THREAD_INFO *ip;
LOG *lp;
REGENV *renv;
REGINFO *infop;
REP *rep;
__rep_update_args *rup;
int ret;
u_int32_t count;
u_int8_t *next;
db_rep = env->rep_handle;
rep = db_rep->region;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
ret = 0;
REP_SYSTEM_LOCK(env);
if (!F_ISSET(rep, REP_F_RECOVER_UPDATE) || IN_ELECTION(rep)) {
REP_SYSTEM_UNLOCK(env);
return (0);
}
F_CLR(rep, REP_F_RECOVER_UPDATE);
F_SET(rep, REP_F_RECOVER_PAGE);
DB_ASSERT(env,
!IS_USING_LEASES(env) || __rep_islease_granted(env) == 0);
if ((ret = __rep_lockout_api(env, rep)) != 0)
goto err;
infop = env->reginfo;
renv = infop->primary;
(void)time(&renv->rep_timestamp);
REP_SYSTEM_UNLOCK(env);
MUTEX_LOCK(env, rep->mtx_clientdb);
__os_gettime(env, &lp->rcvd_ts, 1);
lp->wait_ts = rep->request_gap;
ZERO_LSN(lp->ready_lsn);
ZERO_LSN(lp->verify_lsn);
ZERO_LSN(lp->waiting_lsn);
ZERO_LSN(lp->max_wait_lsn);
ZERO_LSN(lp->max_perm_lsn);
if (db_rep->rep_db == NULL)
ret = __rep_client_dbinit(env, 0, REP_DB);
MUTEX_UNLOCK(env, rep->mtx_clientdb);
if (ret != 0)
goto err_nolock;
if ((ret = __rep_update_unmarshal(env, rp->rep_version,
&rup, rec->data, rec->size, &next)) != 0)
goto err_nolock;
ENV_GET_THREAD_INFO(env, ip);
if ((ret = __db_truncate(db_rep->rep_db, ip, NULL, &count)) != 0)
goto err_nolock;
REP_SYSTEM_LOCK(env);
rep->first_lsn = rup->first_lsn;
rep->first_vers = rup->first_vers;
rep->last_lsn = rp->lsn;
rep->nfiles = rup->num_files;
__os_free(env, rup);
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Update setup for %d files.", rep->nfiles));
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Update setup: First LSN [%lu][%lu].",
(u_long)rep->first_lsn.file, (u_long)rep->first_lsn.offset));
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Update setup: Last LSN [%lu][%lu]",
(u_long)rep->last_lsn.file, (u_long)rep->last_lsn.offset));
if (rep->nfiles > 0) {
rep->infoversion = rp->rep_version;
rep->originfolen = rep->infolen =
rec->size - __REP_UPDATE_SIZE;
if ((ret = __os_calloc(env, 1, rep->infolen,
&rep->originfo)) != 0)
goto err;
memcpy(rep->originfo, next, rep->infolen);
rep->nextinfo = rep->originfo;
}
if ((ret = __rep_remove_all(env, rp->rep_version, rec)) != 0)
goto err;
rep->curfile = 0;
if ((ret = __rep_nextfile(env, eid, rep)) != 0)
goto err;
if (0) {
err_nolock: REP_SYSTEM_LOCK(env);
}
err:
if (ret != 0) {
if (rep->originfo != NULL) {
__os_free(env, rep->originfo);
rep->originfo = NULL;
}
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Update_setup: Error: Clear PAGE, set UPDATE again. %s",
db_strerror(ret)));
F_CLR(rep, REP_F_RECOVER_PAGE | REP_F_READY_API |
REP_F_READY_OP);
F_SET(rep, REP_F_RECOVER_UPDATE);
}
REP_SYSTEM_UNLOCK(env);
return (ret);
}
static int
__rep_remove_all(env, msg_version, rec)
ENV *env;
u_int32_t msg_version;
DBT *rec;
{
__rep_fileinfo_args *finfo;
__rep_update_args u_args;
DB_FH *fhp;
size_t cnt, filelen, filesz, updlen;
u_int32_t bufsz, filecnt, fvers, mvers, zero;
u_int8_t *buf, *fp, *new_fp, *origfp;
int ret, t_ret;
char *fname;
finfo = NULL;
fname = NULL;
fhp = NULL;
filelen = 0;
filecnt = 0;
filesz = MEGABYTE;
if ((ret = __os_calloc(env, 1, filesz, &buf)) != 0)
return (ret);
origfp = fp = buf + __REP_UPDATE_SIZE;
filesz -= __REP_UPDATE_SIZE;
if ((ret = __rep_find_dbs(env, DB_REPVERSION,
&fp, &filesz, &filelen, &filecnt)) != 0)
goto out;
ZERO_LSN(u_args.first_lsn);
u_args.first_vers = 0;
u_args.num_files = filecnt;
if ((ret = __rep_update_marshal(env, DB_REPVERSION,
&u_args, buf, filesz, &updlen)) != 0)
goto out;
if ((ret = __db_appname(
env, DB_APP_NONE, REP_INITNAME, 0, NULL, &fname)) != 0)
goto out;
DB_ASSERT(env, updlen + filelen == (u_int32_t)(updlen + filelen));
bufsz = (u_int32_t)(updlen + filelen);
zero = 0;
fvers = REP_INITVERSION;
mvers = DB_REPVERSION;
if ((ret = __os_open(env, fname, 0,
DB_OSO_CREATE | DB_OSO_TRUNC, DB_MODE_600, &fhp)) != 0 ||
(ret = __os_write(env, fhp, &zero, sizeof(zero), &cnt)) != 0 ||
(ret = __os_write(env, fhp, &fvers, sizeof(fvers), &cnt)) != 0 ||
(ret = __os_write(env, fhp, &mvers, sizeof(mvers), &cnt)) != 0 ||
(ret = __os_write(env, fhp, &bufsz, sizeof(bufsz), &cnt)) != 0 ||
(ret = __os_write(env, fhp, buf, bufsz, &cnt)) != 0 ||
(ret = __os_fsync(env, fhp)) != 0) {
__db_err(env, ret, "%s", fname);
goto out;
}
if ((ret = __rep_remove_logs(env)) != 0)
goto out;
if ((ret = __rep_closefiles(env, 0)) != 0)
goto out;
fp = origfp;
while (filecnt-- > 0) {
if ((ret = __rep_fileinfo_unmarshal(env, DB_REPVERSION,
&finfo, fp, filesz, &new_fp)) != 0)
goto out;
if ((ret = __rep_remove_file(env, finfo->uid.data,
finfo->info.data, finfo->type, finfo->db_flags)) != 0)
goto out;
filesz -= (u_int32_t)(new_fp - fp);
fp = new_fp;
__os_free(env, finfo);
finfo = NULL;
}
mvers = msg_version;
if ((ret = __os_write(env, fhp, &mvers, sizeof(mvers), &cnt)) != 0 ||
(ret = __os_write(env, fhp,
&rec->size, sizeof(rec->size), &cnt)) != 0 ||
(ret = __os_write(env, fhp, rec->data, rec->size, &cnt)) != 0 ||
(ret = __os_fsync(env, fhp)) != 0) {
__db_err(env, ret, "%s", fname);
goto out;
}
out:
if (fhp != NULL && (t_ret = __os_closehandle(env, fhp)) && ret == 0)
ret = t_ret;
if (fname != NULL)
__os_free(env, fname);
if (finfo != NULL)
__os_free(env, finfo);
__os_free(env, buf);
return (ret);
}
static int
__rep_remove_logs(env)
ENV *env;
{
DB_LOG *dblp;
DB_LSN lsn;
LOG *lp;
u_int32_t fnum, lastfile;
int ret;
char *name;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
ret = 0;
if ((ret = __memp_sync_int(env,
NULL, 0, DB_SYNC_CACHE | DB_SYNC_INTERRUPT_OK, NULL, NULL)) != 0)
return (ret);
if ((ret = __log_flush(env, NULL)) != 0)
return (ret);
if (lp->db_log_inmemory) {
ZERO_LSN(lsn);
if ((ret = __log_zero(env, &lsn)) != 0)
return (ret);
} else {
lastfile = lp->lsn.file;
for (fnum = 1; fnum <= lastfile; fnum++) {
if ((ret = __log_name(dblp, fnum, &name, NULL, 0)) != 0)
return (ret);
(void)time(&lp->timestamp);
(void)__os_unlink(env, name, 0);
__os_free(env, name);
}
}
return (0);
}
static int
__rep_remove_file(env, uid, name, type, flags)
ENV *env;
u_int8_t *uid;
const char *name;
u_int32_t type, flags;
{
#ifdef HAVE_QUEUE
DB *dbp;
int ret;
if (type == (u_int32_t)DB_QUEUE && !LF_ISSET(DB_AM_INMEM)) {
if ((ret = __db_create_internal(&dbp, env, 0)) != 0)
return (ret);
if ((ret = __lock_id(env, NULL, &dbp->locker)) != 0)
return (ret);
RPRINT(env, DB_VERB_REP_SYNC,
(env, "QAM: Unlink %s via __qam_remove", name));
if ((ret = __qam_remove(dbp, NULL, name, NULL)) != 0) {
RPRINT(env, DB_VERB_REP_SYNC,
(env, "qam_remove returned %d", ret));
(void)__db_close(dbp, NULL, DB_NOSYNC);
return (ret);
}
if ((ret = __db_close(dbp, NULL, DB_NOSYNC)) != 0)
return (ret);
}
#else
COMPQUIET(type, 0);
COMPQUIET(flags, 0);
#endif
return (__fop_remove(env, NULL, uid, name, DB_APP_DATA, 0));
}
int
__rep_bulk_page(env, ip, eid, rp, rec)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
__rep_control_args *rp;
DBT *rec;
{
__rep_control_args tmprp;
__rep_bulk_args b_args;
int ret;
u_int8_t *p, *ep;
memcpy(&tmprp, rp, sizeof(tmprp));
tmprp.rectype = REP_PAGE;
ret = 0;
for (ep = (u_int8_t *)rec->data + rec->size, p = (u_int8_t *)rec->data;
p < ep;) {
if ((ret = __rep_bulk_unmarshal(env,
&b_args, p, rec->size, &p)) != 0)
return (ret);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"rep_bulk_page: Processing LSN [%lu][%lu]",
(u_long)tmprp.lsn.file, (u_long)tmprp.lsn.offset));
RPRINT(env, DB_VERB_REP_SYNC, (env,
"rep_bulk_page: p %#lx ep %#lx pgrec data %#lx, size %lu (%#lx)",
P_TO_ULONG(p), P_TO_ULONG(ep),
P_TO_ULONG(b_args.bulkdata.data),
(u_long)b_args.bulkdata.size,
(u_long)b_args.bulkdata.size));
ret = __rep_page(env, ip, eid, &tmprp, &b_args.bulkdata);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"rep_bulk_page: rep_page ret %d", ret));
if (ret != 0) {
if (ret == DB_REP_PAGEDONE)
ret = 0;
break;
}
}
return (ret);
}
int
__rep_page(env, ip, eid, rp, rec)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
__rep_control_args *rp;
DBT *rec;
{
DB_REP *db_rep;
DBT key, data;
REP *rep;
__rep_fileinfo_args *msgfp;
db_recno_t recno;
int ret;
ret = 0;
db_rep = env->rep_handle;
rep = db_rep->region;
if (!F_ISSET(rep, REP_F_RECOVER_PAGE))
return (DB_REP_PAGEDONE);
if (LOG_COMPARE(&rp->lsn, &rep->first_lsn) < 0) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE: Old page: msg LSN [%lu][%lu] first_lsn [%lu][%lu]",
(u_long)rp->lsn.file, (u_long)rp->lsn.offset,
(u_long)rep->first_lsn.file,
(u_long)rep->first_lsn.offset));
return (DB_REP_PAGEDONE);
}
if ((ret = __rep_fileinfo_unmarshal(env, rp->rep_version,
&msgfp, rec->data, rec->size, NULL)) != 0)
return (ret);
MUTEX_LOCK(env, rep->mtx_clientdb);
REP_SYSTEM_LOCK(env);
DB_ASSERT(env,
!IS_USING_LEASES(env) || __rep_islease_granted(env) == 0);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE: Received page %lu from file %d",
(u_long)msgfp->pgno, msgfp->filenum));
if (msgfp->filenum != rep->curfile) {
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Msg file %d != curfile %d",
msgfp->filenum, rep->curfile));
ret = DB_REP_PAGEDONE;
goto err;
}
if ((ret = __rep_client_dbinit(env, 1, REP_PG)) != 0) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE: Client_dbinit %s", db_strerror(ret)));
goto err;
}
memset(&key, 0, sizeof(key));
memset(&data, 0, sizeof(data));
recno = (db_recno_t)(msgfp->pgno + 1);
key.data = &recno;
key.ulen = key.size = sizeof(db_recno_t);
key.flags = DB_DBT_USERMEM;
ret = __db_put(rep->file_dbp, ip, NULL, &key, &data, DB_NOOVERWRITE);
if (ret == DB_KEYEXIST) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE: Received duplicate page %lu from file %d",
(u_long)msgfp->pgno, msgfp->filenum));
STAT(rep->stat.st_pg_duplicated++);
ret = 0;
goto err;
}
if (ret != 0)
goto err;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE: Write page %lu into mpool", (u_long)msgfp->pgno));
ret = __rep_write_page(env, ip, rep, msgfp);
if (ret != 0) {
(void)__db_del(rep->file_dbp, NULL, NULL, &key, 0);
goto err;
}
STAT(rep->stat.st_pg_records++);
rep->npages++;
if (LOG_COMPARE(&rp->lsn, &rep->last_lsn) > 0)
rep->last_lsn = rp->lsn;
ret = __rep_filedone(env, ip, eid, rep, msgfp, rp->rectype);
err: REP_SYSTEM_UNLOCK(env);
MUTEX_UNLOCK(env, rep->mtx_clientdb);
__os_free(env, msgfp);
return (ret);
}
int
__rep_page_fail(env, ip, eid, rp, rec)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
__rep_control_args *rp;
DBT *rec;
{
DB_REP *db_rep;
REP *rep;
__rep_fileinfo_args *msgfp, *rfp;
int ret;
ret = 0;
db_rep = env->rep_handle;
rep = db_rep->region;
if (!F_ISSET(rep, REP_F_RECOVER_PAGE))
return (0);
if ((ret = __rep_fileinfo_unmarshal(env, rp->rep_version,
&msgfp, rec->data, rec->size, NULL)) != 0)
return (ret);
MUTEX_LOCK(env, rep->mtx_clientdb);
REP_SYSTEM_LOCK(env);
DB_ASSERT(env,
!IS_USING_LEASES(env) || __rep_islease_granted(env) == 0);
if (msgfp->filenum != rep->curfile) {
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Msg file %d != curfile %d",
msgfp->filenum, rep->curfile));
goto out;
}
rfp = rep->curinfo;
if (rfp->type != (u_int32_t)DB_QUEUE)
--rfp->max_pgno;
else {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"page_fail: BEFORE page %lu failed. ready %lu, max %lu, npages %d",
(u_long)msgfp->pgno, (u_long)rep->ready_pg,
(u_long)rfp->max_pgno, rep->npages));
if (msgfp->pgno == rfp->max_pgno)
--rfp->max_pgno;
if (msgfp->pgno >= rep->ready_pg) {
rep->ready_pg = msgfp->pgno + 1;
rep->npages = rep->ready_pg;
}
RPRINT(env, DB_VERB_REP_SYNC, (env,
"page_fail: AFTER page %lu failed. ready %lu, max %lu, npages %d",
(u_long)msgfp->pgno, (u_long)rep->ready_pg,
(u_long)rfp->max_pgno, rep->npages));
}
ret = __rep_filedone(env, ip, eid, rep, msgfp, REP_PAGE_FAIL);
out:
REP_SYSTEM_UNLOCK(env);
MUTEX_UNLOCK(env, rep->mtx_clientdb);
__os_free(env, msgfp);
return (ret);
}
static int
__rep_write_page(env, ip, rep, msgfp)
ENV *env;
DB_THREAD_INFO *ip;
REP *rep;
__rep_fileinfo_args *msgfp;
{
DB db;
DBT pgcookie;
DB_MPOOLFILE *mpf;
DB_PGINFO *pginfo;
__rep_fileinfo_args *rfp;
int ret;
void *dst;
rfp = NULL;
rfp = rep->curinfo;
if (rep->file_mpf == NULL) {
if (!FLD_ISSET(rfp->db_flags, DB_AM_INMEM)) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"rep_write_page: Calling fop_create for %s",
(char *)rfp->info.data));
if ((ret = __fop_create(env, NULL, NULL,
rfp->info.data, DB_APP_DATA,
env->db_mode, 0)) != 0)
goto err;
}
if ((ret =
__rep_mpf_open(env, &rep->file_mpf, rep->curinfo,
FLD_ISSET(rfp->db_flags, DB_AM_INMEM) ? DB_CREATE : 0)) != 0)
goto err;
}
if (msgfp->type == (u_int32_t)DB_QUEUE && msgfp->pgno != 0) {
#ifdef HAVE_QUEUE
ret = __qam_fget(rep->queue_dbc, &msgfp->pgno,
DB_MPOOL_CREATE | DB_MPOOL_DIRTY, &dst);
#else
ret = __db_no_queue_am(env);
#endif
} else
ret = __memp_fget(rep->file_mpf, &msgfp->pgno, ip, NULL,
DB_MPOOL_CREATE | DB_MPOOL_DIRTY, &dst);
if (ret != 0)
goto err;
if ((F_ISSET(env, ENV_LITTLEENDIAN) &&
!FLD_ISSET(msgfp->finfo_flags, REPINFO_PG_LITTLEENDIAN)) ||
(!F_ISSET(env, ENV_LITTLEENDIAN) &&
FLD_ISSET(msgfp->finfo_flags, REPINFO_PG_LITTLEENDIAN))) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"write_page: Page %d needs to be swapped", msgfp->pgno));
memset(&db, 0, sizeof(db));
db.env = env;
db.type = (DBTYPE)msgfp->type;
db.pgsize = msgfp->pgsize;
mpf = rep->file_mpf;
if ((ret = __memp_get_pgcookie(mpf, &pgcookie)) != 0)
goto err;
pginfo = (DB_PGINFO *)pgcookie.data;
db.flags = pginfo->flags;
if ((ret = __db_pageswap(&db, msgfp->info.data, msgfp->pgsize,
NULL, 1)) != 0)
goto err;
}
memcpy(dst, msgfp->info.data, msgfp->pgsize);
#ifdef HAVE_QUEUE
if (msgfp->type == (u_int32_t)DB_QUEUE && msgfp->pgno != 0)
ret = __qam_fput(rep->queue_dbc,
msgfp->pgno, dst, rep->queue_dbc->priority);
else
#endif
ret = __memp_fput(rep->file_mpf,
ip, dst, rep->file_dbp->priority);
err: return (ret);
}
static int
__rep_page_gap(env, rep, msgfp, type)
ENV *env;
REP *rep;
__rep_fileinfo_args *msgfp;
u_int32_t type;
{
DBC *dbc;
DBT data, key;
DB_LOG *dblp;
DB_THREAD_INFO *ip;
LOG *lp;
__rep_fileinfo_args *rfp;
db_recno_t recno;
int ret, t_ret;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
ret = 0;
dbc = NULL;
rfp = rep->curinfo;
if (rfp->filenum != msgfp->filenum) {
ret = DB_REP_PAGEDONE;
goto err;
}
if (msgfp->pgno < rep->ready_pg) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE_GAP: pgno %lu < ready %lu, waiting %lu",
(u_long)msgfp->pgno, (u_long)rep->ready_pg,
(u_long)rep->waiting_pg));
goto err;
}
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE_GAP: pgno %lu, max_pg %lu ready %lu, waiting %lu max_wait %lu",
(u_long)msgfp->pgno, (u_long)rfp->max_pgno, (u_long)rep->ready_pg,
(u_long)rep->waiting_pg, (u_long)rep->max_wait_pg));
if (msgfp->pgno > rep->ready_pg) {
if (rep->waiting_pg == PGNO_INVALID ||
msgfp->pgno < rep->waiting_pg)
rep->waiting_pg = msgfp->pgno;
} else {
rep->ready_pg++;
__os_gettime(env, &lp->rcvd_ts, 1);
if (rep->ready_pg == rep->waiting_pg) {
lp->wait_ts = rep->request_gap;
rep->max_wait_pg = PGNO_INVALID;
memset(&key, 0, sizeof(key));
memset(&data, 0, sizeof(data));
ENV_GET_THREAD_INFO(env, ip);
if ((ret = __db_cursor(rep->file_dbp, ip, NULL,
&dbc, 0)) != 0)
goto err;
recno = (db_recno_t)rep->waiting_pg + 1;
key.data = &recno;
key.ulen = key.size = sizeof(db_recno_t);
key.flags = DB_DBT_USERMEM;
ret = __dbc_get(dbc, &key, &data, DB_SET);
if (ret != 0)
goto err;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE_GAP: Set cursor for ready %lu, waiting %lu",
(u_long)rep->ready_pg, (u_long)rep->waiting_pg));
}
while (ret == 0 && rep->ready_pg == rep->waiting_pg) {
rep->ready_pg++;
ret = __dbc_get(dbc, &key, &data, DB_NEXT);
if (ret == DB_NOTFOUND || ret == DB_KEYEMPTY) {
rep->waiting_pg = PGNO_INVALID;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE_GAP: Next cursor No next - ready %lu, waiting %lu",
(u_long)rep->ready_pg,
(u_long)rep->waiting_pg));
break;
}
rep->waiting_pg = *(db_pgno_t *)key.data;
rep->waiting_pg--;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"PAGE_GAP: Next cursor ready %lu, waiting %lu",
(u_long)rep->ready_pg, (u_long)rep->waiting_pg));
}
}
if (rep->ready_pg > rfp->max_pgno)
goto err;
if ((rep->waiting_pg != PGNO_INVALID &&
rep->ready_pg != rep->waiting_pg) || type == REP_PAGE_MORE) {
if (type == REP_PAGE_MORE)
rfp->pgno = msgfp->pgno;
if ((__rep_check_doreq(env, rep) || type == REP_PAGE_MORE) &&
((ret = __rep_pggap_req(env, rep, rfp,
(type == REP_PAGE_MORE) ? REP_GAP_FORCE : 0)) != 0))
goto err;
} else {
lp->wait_ts = rep->request_gap;
rep->max_wait_pg = PGNO_INVALID;
}
err:
if (dbc != NULL && (t_ret = __dbc_close(dbc)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__rep_init_cleanup(env, rep, force)
ENV *env;
REP *rep;
int force;
{
DB *queue_dbp;
DB_LOG *dblp;
LOG *lp;
int cleanup_failure, ret, t_ret;
ret = 0;
if (rep->file_mpf != NULL) {
ret = __memp_fclose(rep->file_mpf, 0);
rep->file_mpf = NULL;
}
if (rep->file_dbp != NULL) {
t_ret = __db_close(rep->file_dbp, NULL, DB_NOSYNC);
rep->file_dbp = NULL;
if (t_ret != 0 && ret == 0)
ret = t_ret;
}
if (force && rep->queue_dbc != NULL) {
queue_dbp = rep->queue_dbc->dbp;
if ((t_ret = __dbc_close(rep->queue_dbc)) != 0 && ret == 0)
ret = t_ret;
rep->queue_dbc = NULL;
if ((t_ret = __db_close(queue_dbp, NULL, DB_NOSYNC)) != 0 &&
ret == 0)
ret = t_ret;
}
if (rep->curinfo != NULL) {
__os_free(env, rep->curinfo);
rep->curinfo = NULL;
}
if (F_ISSET(rep, REP_F_INTERNAL_INIT_MASK) && force) {
RPRINT(env, DB_VERB_REP_SYNC,
(env, "clean up interrupted internal init"));
cleanup_failure = 0;
if ((t_ret = __rep_remove_logs(env)) == 0) {
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
if ((t_ret = __rep_log_setup(env,
rep, 1, DB_LOGVERSION, &lp->ready_lsn)) != 0) {
cleanup_failure = 1;
if (ret == 0)
ret = t_ret;
}
} else {
cleanup_failure = 1;
if (ret == 0)
ret = t_ret;
}
if ((t_ret = __rep_remove_by_list(env, rep->infoversion,
rep->originfo, rep->originfolen, rep->nfiles)) != 0) {
cleanup_failure = 1;
if (ret == 0)
ret = t_ret;
}
if (!cleanup_failure &&
(t_ret = __rep_remove_init_file(env)) != 0) {
if (ret == 0)
ret = t_ret;
}
if (rep->originfo != NULL) {
__os_free(env, rep->originfo);
rep->originfo = NULL;
}
}
return (ret);
}
static int
__rep_filedone(env, ip, eid, rep, msgfp, type)
ENV *env;
DB_THREAD_INFO *ip;
int eid;
REP *rep;
__rep_fileinfo_args *msgfp;
u_int32_t type;
{
__rep_fileinfo_args *rfp;
int ret;
ret = __rep_page_gap(env, rep, msgfp, type);
if (ret == DB_REP_PAGEDONE)
return (0);
rfp = rep->curinfo;
RPRINT(env, DB_VERB_REP_SYNC,
(env, "FILEDONE: have %lu pages. Need %lu.",
(u_long)rep->npages, (u_long)rfp->max_pgno + 1));
if (rep->npages <= rfp->max_pgno)
return (0);
if (rfp->type == (u_int32_t)DB_QUEUE &&
((ret = __rep_queue_filedone(env, ip, rep, rfp)) !=
DB_REP_PAGEDONE))
return (ret);
if ((ret = __rep_init_cleanup(env, rep, 0)) != 0)
goto err;
rep->curfile++;
ret = __rep_nextfile(env, eid, rep);
err:
return (ret);
}
static int
__rep_nextfile(env, eid, rep)
ENV *env;
int eid;
REP *rep;
{
DBT dbt;
__rep_logreq_args lr_args;
int ret;
u_int8_t *buf, lrbuf[__REP_LOGREQ_SIZE];
size_t len, msgsz;
if (rep->master_id != DB_EID_INVALID)
eid = rep->master_id;
if (rep->curfile == rep->nfiles) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"NEXTFILE: have %d files. RECOVER_LOG now", rep->nfiles));
if ((ret = __memp_sync_int(env, NULL, 0,
DB_SYNC_CACHE | DB_SYNC_INTERRUPT_OK, NULL, NULL)) != 0)
return (ret);
F_CLR(rep, REP_F_RECOVER_PAGE);
F_SET(rep, REP_F_RECOVER_LOG);
memset(&dbt, 0, sizeof(dbt));
lr_args.endlsn = rep->last_lsn;
if ((ret = __rep_logreq_marshal(env, &lr_args, lrbuf,
__REP_LOGREQ_SIZE, &len)) != 0)
return (ret);
DB_INIT_DBT(dbt, lrbuf, len);
REP_SYSTEM_UNLOCK(env);
if ((ret = __rep_log_setup(env, rep,
rep->first_lsn.file, rep->first_vers, NULL)) != 0)
return (ret);
RPRINT(env, DB_VERB_REP_SYNC, (env,
"NEXTFILE: LOG_REQ from LSN [%lu][%lu] to [%lu][%lu]",
(u_long)rep->first_lsn.file, (u_long)rep->first_lsn.offset,
(u_long)rep->last_lsn.file, (u_long)rep->last_lsn.offset));
(void)__rep_send_message(env, eid,
REP_LOG_REQ, &rep->first_lsn, &dbt,
REPCTL_INIT, DB_REP_ANYWHERE);
REP_SYSTEM_LOCK(env);
return (0);
}
rep->finfo = rep->nextinfo;
if ((ret = __rep_fileinfo_unmarshal(env, rep->infoversion,
&rep->curinfo, rep->finfo, rep->infolen, &rep->nextinfo)) != 0) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"NEXTINFO: Fileinfo read: %s", db_strerror(ret)));
return (ret);
}
DB_ASSERT(env, rep->curinfo->pgno == 0);
rep->infolen -= (u_int32_t)(rep->nextinfo - rep->finfo);
rep->ready_pg = 0;
rep->npages = 0;
rep->waiting_pg = PGNO_INVALID;
rep->max_wait_pg = PGNO_INVALID;
memset(&dbt, 0, sizeof(dbt));
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Next file %d: pgsize %lu, maxpg %lu", rep->curinfo->filenum,
(u_long)rep->curinfo->pgsize, (u_long)rep->curinfo->max_pgno));
msgsz = __REP_FILEINFO_SIZE +
rep->curinfo->uid.size + rep->curinfo->info.size;
if ((ret = __os_calloc(env, 1, msgsz, &buf)) != 0)
return (ret);
if ((ret = __rep_fileinfo_marshal(env, rep->infoversion,
rep->curinfo, buf, msgsz, &len)) != 0)
return (ret);
DB_INIT_DBT(dbt, buf, len);
(void)__rep_send_message(env, eid, REP_PAGE_REQ,
NULL, &dbt, 0, DB_REP_ANYWHERE);
__os_free(env, buf);
return (0);
}
static int
__rep_mpf_open(env, mpfp, rfp, flags)
ENV *env;
DB_MPOOLFILE **mpfp;
__rep_fileinfo_args *rfp;
u_int32_t flags;
{
DB db;
int ret;
if ((ret = __memp_fcreate(env, mpfp)) != 0)
return (ret);
db.env = env;
db.type = (DBTYPE)rfp->type;
db.pgsize = rfp->pgsize;
memcpy(db.fileid, rfp->uid.data, DB_FILE_ID_LEN);
db.flags = rfp->db_flags;
F_CLR(&db, DB_AM_OPEN_CALLED);
if ((F_ISSET(env, ENV_LITTLEENDIAN) &&
!FLD_ISSET(rfp->finfo_flags, REPINFO_DB_LITTLEENDIAN)) ||
(!F_ISSET(env, ENV_LITTLEENDIAN) &&
FLD_ISSET(rfp->finfo_flags, REPINFO_DB_LITTLEENDIAN))) {
RPRINT(env, DB_VERB_REP_SYNC, (env,
"rep_mpf_open: Different endian database. Set swap bit."));
F_SET(&db, DB_AM_SWAP);
} else
F_CLR(&db, DB_AM_SWAP);
db.mpf = *mpfp;
if (F_ISSET(&db, DB_AM_INMEM))
(void)__memp_set_flags(db.mpf, DB_MPOOL_NOFILE, 1);
if ((ret = __env_mpool(&db, rfp->info.data, flags)) != 0) {
(void)__memp_fclose(db.mpf, 0);
*mpfp = NULL;
}
return (ret);
}
int
__rep_pggap_req(env, rep, reqfp, gapflags)
ENV *env;
REP *rep;
__rep_fileinfo_args *reqfp;
u_int32_t gapflags;
{
DBT max_pg_dbt;
__rep_fileinfo_args *tmpfp, t;
size_t len, msgsz;
u_int32_t flags;
int alloc, ret;
u_int8_t *buf;
ret = 0;
alloc = 0;
if (rep->curinfo == NULL)
return (0);
if (reqfp == NULL) {
if ((ret = __rep_finfo_alloc(env, rep->curinfo, &tmpfp)) != 0)
return (ret);
alloc = 1;
} else {
t = *reqfp;
tmpfp = &t;
}
flags = 0;
memset(&max_pg_dbt, 0, sizeof(max_pg_dbt));
if (FLD_ISSET(gapflags, REP_GAP_FORCE))
tmpfp->pgno++;
else
tmpfp->pgno = rep->ready_pg;
msgsz = __REP_FILEINFO_SIZE +
tmpfp->uid.size + tmpfp->info.size;
if ((ret = __os_calloc(env, 1, msgsz, &buf)) != 0)
goto err;
if (rep->max_wait_pg == PGNO_INVALID ||
FLD_ISSET(gapflags, REP_GAP_FORCE | REP_GAP_REREQUEST)) {
if (rep->waiting_pg == PGNO_INVALID) {
if (FLD_ISSET(gapflags,
REP_GAP_FORCE | REP_GAP_REREQUEST))
rep->max_wait_pg = rep->curinfo->max_pgno;
else
rep->max_wait_pg = rep->ready_pg;
} else {
if (FLD_ISSET(gapflags, REP_GAP_FORCE) &&
rep->waiting_pg < tmpfp->pgno)
rep->max_wait_pg = rep->curinfo->max_pgno;
else
rep->max_wait_pg = rep->waiting_pg - 1;
}
tmpfp->max_pgno = rep->max_wait_pg;
if (FLD_ISSET(gapflags, REP_GAP_REREQUEST))
flags = DB_REP_REREQUEST;
else
flags = DB_REP_ANYWHERE;
} else {
rep->max_wait_pg = rep->ready_pg;
tmpfp->max_pgno = rep->ready_pg;
flags = DB_REP_REREQUEST;
}
if (rep->master_id != DB_EID_INVALID) {
STAT(rep->stat.st_pg_requested++);
if ((ret = __rep_fileinfo_marshal(env, rep->infoversion,
tmpfp, buf, msgsz, &len)) == 0) {
DB_INIT_DBT(max_pg_dbt, buf, len);
DB_ASSERT(env, len == max_pg_dbt.size);
(void)__rep_send_message(env, rep->master_id,
REP_PAGE_REQ, NULL, &max_pg_dbt, 0, flags);
}
} else
(void)__rep_send_message(env, DB_EID_BROADCAST,
REP_MASTER_REQ, NULL, NULL, 0, 0);
__os_free(env, buf);
err:
if (alloc)
__os_free(env, tmpfp);
return (ret);
}
int
__rep_finfo_alloc(env, rfpsrc, rfpp)
ENV *env;
__rep_fileinfo_args *rfpsrc, **rfpp;
{
__rep_fileinfo_args *rfp;
size_t size;
int ret;
void *uidp, *infop;
size = sizeof(__rep_fileinfo_args) + rfpsrc->uid.size +
rfpsrc->info.size;
if ((ret = __os_malloc(env, size, &rfp)) != 0)
return (ret);
memcpy(rfp, rfpsrc, sizeof(__rep_fileinfo_args));
uidp = (u_int8_t *)rfp + sizeof(__rep_fileinfo_args);
rfp->uid.data = uidp;
memcpy(uidp, rfpsrc->uid.data, rfpsrc->uid.size);
infop = (u_int8_t *)uidp + rfpsrc->uid.size;
rfp->info.data = infop;
memcpy(infop, rfpsrc->info.data, rfpsrc->info.size);
*rfpp = rfp;
return (ret);
}
static int
__rep_log_setup(env, rep, file, version, lsnp)
ENV *env;
REP *rep;
u_int32_t file;
u_int32_t version;
DB_LSN *lsnp;
{
DB_LOG *dblp;
DB_LSN lsn;
DB_TXNMGR *mgr;
DB_TXNREGION *region;
LOG *lp;
int ret;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
mgr = env->tx_handle;
region = mgr->reginfo.primary;
LOG_SYSTEM_LOCK(env);
if ((ret = __log_newfile(dblp, &lsn, file, version)) == 0 &&
lsnp != NULL)
*lsnp = lsn;
LOG_SYSTEM_UNLOCK(env);
rep->first_lsn = lp->lsn;
TXN_SYSTEM_LOCK(env);
ZERO_LSN(region->last_ckp);
TXN_SYSTEM_UNLOCK(env);
return (ret);
}
static int
__rep_queue_filedone(env, ip, rep, rfp)
ENV *env;
DB_THREAD_INFO *ip;
REP *rep;
__rep_fileinfo_args *rfp;
{
#ifndef HAVE_QUEUE
COMPQUIET(ip, NULL);
COMPQUIET(rep, NULL);
COMPQUIET(rfp, NULL);
return (__db_no_queue_am(env));
#else
DB *queue_dbp;
db_pgno_t first, last;
u_int32_t flags;
int empty, ret, t_ret;
ret = 0;
queue_dbp = NULL;
if (rep->queue_dbc == NULL) {
if ((ret = __memp_sync_int(env, NULL, 0,
DB_SYNC_CACHE | DB_SYNC_INTERRUPT_OK, NULL, NULL)) != 0)
goto out;
if ((ret =
__db_create_internal(&queue_dbp, env, 0)) != 0)
goto out;
flags = DB_NO_AUTO_COMMIT |
(F_ISSET(env, ENV_THREAD) ? DB_THREAD : 0);
if ((ret = __db_open(queue_dbp, ip, NULL,
FLD_ISSET(rfp->db_flags, DB_AM_INMEM) ? NULL :
rfp->info.data,
FLD_ISSET(rfp->db_flags, DB_AM_INMEM) ? rfp->info.data :
NULL,
DB_QUEUE, flags, 0, PGNO_BASE_MD)) != 0)
goto out;
if ((ret = __db_cursor(queue_dbp,
ip, NULL, &rep->queue_dbc, 0)) != 0)
goto out;
} else
queue_dbp = rep->queue_dbc->dbp;
if ((ret = __queue_pageinfo(queue_dbp,
&first, &last, &empty, 0, 0)) != 0)
goto out;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Queue fileinfo: first %lu, last %lu, empty %d",
(u_long)first, (u_long)last, empty));
if (rfp->max_pgno == 0) {
if (empty)
goto out;
if (first > last) {
rfp->max_pgno =
QAM_RECNO_PAGE(rep->queue_dbc->dbp, UINT32_MAX);
} else
rfp->max_pgno = last;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Queue fileinfo: First req: first %lu, last %lu",
(u_long)first, (u_long)rfp->max_pgno));
goto req;
} else if (rfp->max_pgno != last) {
first = 1;
rfp->max_pgno = last;
RPRINT(env, DB_VERB_REP_SYNC, (env,
"Queue fileinfo: Wrap req: first %lu, last %lu",
(u_long)first, (u_long)last));
req:
rep->npages = first;
rep->ready_pg = first;
rep->waiting_pg = rfp->max_pgno + 1;
rep->max_wait_pg = PGNO_INVALID;
ret = __rep_pggap_req(env, rep, rfp, 0);
return (ret);
}
out:
if (rep->queue_dbc != NULL &&
(t_ret = __dbc_close(rep->queue_dbc)) != 0 && ret == 0)
ret = t_ret;
rep->queue_dbc = NULL;
if (queue_dbp != NULL &&
(t_ret = __db_close(queue_dbp, NULL, DB_NOSYNC)) != 0 && ret == 0)
ret = t_ret;
if (ret == 0)
ret = DB_REP_PAGEDONE;
return (ret);
#endif
}
int
__rep_remove_init_file(env)
ENV *env;
{
int ret;
char *name;
if ((ret = __db_appname(
env, DB_APP_NONE, REP_INITNAME, 0, NULL, &name)) != 0)
return (ret);
(void)__os_unlink(env, name, 0);
__os_free(env, name);
return (0);
}
int
__rep_reset_init(env)
ENV *env;
{
DB_FH *fhp;
__rep_update_args *rup;
DBT dbt;
char *allocated_dir, *dir, *init_name;
size_t cnt;
u_int32_t dbtvers, fvers, zero;
u_int8_t *next;
int ret, t_ret;
allocated_dir = NULL;
rup = NULL;
dbt.data = NULL;
if ((ret = __db_appname(
env, DB_APP_NONE, REP_INITNAME, 0, NULL, &init_name)) != 0)
return (ret);
if ((ret = __os_open(
env, init_name, 0, DB_OSO_RDONLY, DB_MODE_600, &fhp)) != 0) {
if (ret == ENOENT)
ret = 0;
goto out;
}
RPRINT(env, DB_VERB_REP_SYNC,
(env, "Cleaning up interrupted internal init"));
if ((ret = __os_read(env, fhp, &zero, sizeof(zero), &cnt)) != 0)
goto out;
if (cnt != sizeof(zero))
goto rm;
if (zero != 0) {
if ((ret = __os_seek(env, fhp, 0, 0, 0)) != 0)
goto out;
fvers = REP_INITVERSION_46;
} else if ((ret = __os_read(env,
fhp, &fvers, sizeof(fvers), &cnt)) != 0)
goto out;
else if (cnt != sizeof(fvers))
goto rm;
ret = __rep_get_file_list(env, fhp, fvers, &dbtvers, &dbt);
if ((t_ret = __os_closehandle(env, fhp)) != 0 || ret != 0) {
if (ret == 0)
ret = t_ret;
goto out;
}
if (dbt.data == NULL) {
goto rm;
}
if (env->dbenv->db_log_dir == NULL)
dir = env->db_home;
else {
if ((ret = __db_appname(env, DB_APP_NONE,
env->dbenv->db_log_dir, 0, NULL, &dir)) != 0)
goto out;
allocated_dir = dir;
}
if ((ret = __rep_remove_by_prefix(env,
dir, LFPREFIX, sizeof(LFPREFIX)-1, DB_APP_LOG)) != 0)
goto out;
if ((ret = __rep_update_unmarshal(env, dbtvers,
&rup, dbt.data, dbt.size, &next)) != 0)
goto out;
if ((ret = __rep_remove_by_list(env, dbtvers,
next, dbt.size, rup->num_files)) != 0)
goto out;
rm: (void)__os_unlink(env, init_name, 0);
out: if (rup != NULL)
__os_free(env, rup);
if (allocated_dir != NULL)
__os_free(env, allocated_dir);
if (dbt.data != NULL)
__os_free(env, dbt.data);
__os_free(env, init_name);
return (ret);
}
static int
__rep_get_file_list(env, fhp, fvers, dbtvers, dbt)
ENV *env;
DB_FH *fhp;
u_int32_t fvers;
u_int32_t *dbtvers;
DBT *dbt;
{
u_int32_t length, mvers;
size_t cnt;
int i, ret;
dbt->data = NULL;
mvers = DB_REPVERSION_46;
length = 0;
for (i = 1; i <= 2; i++) {
if (fvers >= REP_INITVERSION_47) {
if ((ret = __os_read(env, fhp, &mvers,
sizeof(mvers), &cnt)) != 0)
goto err;
if (cnt == 0 && dbt->data != NULL)
break;
if (cnt != sizeof(mvers))
goto err;
}
if ((ret = __os_read(env,
fhp, &length, sizeof(length), &cnt)) != 0)
goto err;
if (cnt == 0 && dbt->data != NULL)
break;
if (cnt != sizeof(length))
goto err;
if ((ret = __os_realloc(env,
(size_t)length, &dbt->data)) != 0)
goto err;
if ((ret = __os_read(
env, fhp, dbt->data, length, &cnt)) != 0 ||
cnt != (size_t)length)
goto err;
}
*dbtvers = mvers;
dbt->size = length;
return (0);
err:
if (dbt->data != NULL)
__os_free(env, dbt->data);
dbt->data = NULL;
return (ret);
}
static int
__rep_remove_by_prefix(env, dir, prefix, pref_len, appname)
ENV *env;
const char *dir;
const char *prefix;
size_t pref_len;
APPNAME appname;
{
char *namep, **names;
int cnt, i, ret;
if ((ret = __os_dirlist(env, dir, 0, &names, &cnt)) != 0)
return (ret);
for (i = 0; i < cnt; i++) {
if (strncmp(names[i], prefix, pref_len) == 0) {
if ((ret = __db_appname(env,
appname, names[i], 0, NULL, &namep)) != 0)
goto out;
(void)__os_unlink(env, namep, 0);
__os_free(env, namep);
}
}
out: __os_dirfree(env, names, cnt);
return (ret);
}
static int
__rep_remove_by_list(env, version, filelist, filesz, count)
ENV *env;
u_int32_t version;
u_int8_t *filelist;
u_int32_t filesz;
u_int32_t count;
{
DB_ENV *dbenv;
__rep_fileinfo_args *rfp;
char **ddir, *dir, *namep;
u_int8_t *new_fp;
int ret;
dbenv = env->dbenv;
ret = 0;
rfp = NULL;
while (count-- > 0) {
if ((ret = __rep_fileinfo_unmarshal(env, version,
&rfp, filelist, filesz, &new_fp)) != 0)
goto out;
filesz -= (u_int32_t)(new_fp - filelist);
filelist = new_fp;
if ((ret = __db_appname(env,
DB_APP_DATA, rfp->info.data, 0, NULL, &namep)) != 0)
goto out;
(void)__os_unlink(env, namep, 0);
__os_free(env, namep);
__os_free(env, rfp);
rfp = NULL;
}
if (dbenv->db_data_dir == NULL)
ret = __rep_remove_by_prefix(env, env->db_home,
QUEUE_EXTENT_PREFIX, sizeof(QUEUE_EXTENT_PREFIX) - 1,
DB_APP_DATA);
else {
for (ddir = dbenv->db_data_dir; *ddir != NULL; ++ddir) {
if ((ret = __db_appname(env, DB_APP_NONE,
*ddir, 0, NULL, &dir)) != 0)
break;
ret = __rep_remove_by_prefix(env, dir,
QUEUE_EXTENT_PREFIX, sizeof(QUEUE_EXTENT_PREFIX)-1,
DB_APP_DATA);
__os_free(env, dir);
if (ret != 0)
break;
}
}
out:
if (rfp != NULL)
__os_free(env, rfp);
return (ret);
}