#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/db_am.h"
#include "dbinc/txn.h"
#define IS_PHASE1_DONE(rep) \
((rep)->sites >= (rep)->nsites && (rep)->winner != DB_EID_INVALID)
#define I_HAVE_WON(rep, winner) \
((rep)->votes >= (rep)->nvotes && winner == (rep)->eid)
static void __rep_cmp_vote __P((ENV *, REP *, int, DB_LSN *,
u_int32_t, u_int32_t, u_int32_t, u_int32_t, u_int32_t));
static int __rep_elect_init
__P((ENV *, u_int32_t, u_int32_t, int *, u_int32_t *));
static int __rep_fire_elected __P((ENV *, REP *, u_int32_t));
static void __rep_elect_master __P((ENV *, REP *));
static int __rep_grow_sites __P((ENV *, u_int32_t));
static void __rep_send_vote __P((ENV *, DB_LSN *, u_int32_t,
u_int32_t, u_int32_t, u_int32_t, u_int32_t, u_int32_t, int,
u_int32_t, u_int32_t));
static int __rep_tally __P((ENV *, REP *, int, u_int32_t *, u_int32_t, int));
static int __rep_wait __P((ENV *, db_timeout_t *, int, u_int32_t, u_int32_t));
int
__rep_elect_pp(dbenv, given_nsites, nvotes, flags)
DB_ENV *dbenv;
u_int32_t given_nsites, nvotes;
u_int32_t flags;
{
DB_REP *db_rep;
ENV *env;
int ret;
env = dbenv->env;
db_rep = env->rep_handle;
ret = 0;
ENV_REQUIRES_CONFIG_XX(
env, rep_handle, "DB_ENV->rep_elect", DB_INIT_REP);
if (APP_IS_REPMGR(env)) {
__db_errx(env, DB_STR("3527",
"DB_ENV->rep_elect: cannot call from Replication Manager application"));
return (EINVAL);
}
if (db_rep->send == NULL) {
__db_errx(env, DB_STR("3528",
"DB_ENV->rep_elect: must be called after DB_ENV->rep_set_transport"));
return (EINVAL);
}
if (!IS_REP_STARTED(env)) {
__db_errx(env, DB_STR("3529",
"DB_ENV->rep_elect: must be called after DB_ENV->rep_start"));
return (EINVAL);
}
if (IS_USING_LEASES(env) && given_nsites != 0) {
__db_errx(env, DB_STR("3530",
"DB_ENV->rep_elect: nsites must be zero if leases configured"));
return (EINVAL);
}
ret = __rep_elect_int(env, given_nsites, nvotes, flags);
if (ret == DB_REP_IGNORE)
ret = 0;
return (ret);
}
int
__rep_elect_int(env, given_nsites, nvotes, flags)
ENV *env;
u_int32_t given_nsites, nvotes;
u_int32_t flags;
{
DB_LOG *dblp;
DB_LOGC *logc;
DB_LSN lsn;
DB_REP *db_rep;
DB_THREAD_INFO *ip;
LOG *lp;
REP *rep;
int done, elected, in_progress;
int need_req, ret, send_vote, t_ret;
u_int32_t ack, ctlflags, data_gen, egen, nsites;
u_int32_t orig_tally, priority, realpri, repflags, tiebreaker;
db_timeout_t timeout;
COMPQUIET(flags, 0);
db_rep = env->rep_handle;
rep = db_rep->region;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
elected = 0;
egen = 0;
ret = 0;
nsites = given_nsites == 0 ? rep->config_nsites : given_nsites;
ack = nvotes == 0 ? ELECTION_MAJORITY(nsites) : nvotes;
if (ack <= (nsites / 2)) {
__db_errx(env, DB_STR_A("3531",
"DB_ENV->rep_elect:WARNING: nvotes (%d) is sub-majority with nsites (%d)",
"%d %d"), nvotes, nsites);
}
if (nsites < ack) {
__db_errx(env, DB_STR_A("3532",
"DB_ENV->rep_elect: nvotes (%d) is larger than nsites (%d)",
"%d %d"), ack, nsites);
return (EINVAL);
}
realpri = rep->priority;
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Start election nsites %d, ack %d, priority %d",
nsites, ack, realpri));
ctlflags = realpri != 0 ? REPCTL_ELECTABLE : 0;
ENV_ENTER(env, ip);
orig_tally = 0;
if (F_ISSET(rep, REP_F_MASTER)) {
master: LOG_SYSTEM_LOCK(env);
lsn = lp->lsn;
LOG_SYSTEM_UNLOCK(env);
(void)__rep_send_message(env,
DB_EID_BROADCAST, REP_NEWMASTER, &lsn, NULL, 0, 0);
if (IS_USING_LEASES(env))
ret = __rep_lease_refresh(env);
if (ret == 0)
ret = DB_REP_IGNORE;
goto envleave;
}
REP_SYSTEM_LOCK(env);
if (IS_USING_LEASES(env) &&
(timeout = __rep_lease_waittime(env)) != 0) {
FLD_SET(rep->elect_flags, REP_E_PHASE0);
egen = rep->egen;
REP_SYSTEM_UNLOCK(env);
VPRINT(env, (env, DB_VERB_REP_ELECT,
"PHASE0 waittime from rep_lease_waittime: %lu",
(u_long)timeout));
(void)__rep_send_message(env, DB_EID_BROADCAST,
REP_MASTER_REQ, NULL, NULL, 0, 0);
if ((ret = __rep_wait(env,
&timeout, 0, egen, REP_E_PHASE0)) != 0)
goto envleave;
REP_SYSTEM_LOCK(env);
repflags = rep->elect_flags;
FLD_CLR(rep->elect_flags, REP_E_PHASE0);
RPRINT(env, (env, DB_VERB_REP_ELECT,
"after PHASE0 wait, flags 0x%x, elect_flags 0x%x",
rep->flags, rep->elect_flags));
if (!FLD_ISSET(repflags, REP_E_PHASE0) ||
__rep_islease_granted(env) || egen != rep->egen) {
VPRINT(env, (env, DB_VERB_REP_ELECT,
"PHASE0 Done: repflags 0x%x, egen %d rep->egen %d, lease_granted %d",
repflags, egen, rep->egen, __rep_islease_granted(env)));
goto unlck_lv;
}
F_SET(rep, REP_F_LEASE_EXPIRED);
}
if (F_ISSET(rep, REP_F_MASTER)) {
REP_SYSTEM_UNLOCK(env);
goto master;
}
if ((ret = __rep_elect_init(env, nsites, ack,
&in_progress, &orig_tally)) != 0)
goto unlck_lv;
if (in_progress) {
ret = DB_REP_IGNORE;
goto unlck_lv;
}
rep->elect_th++;
egen = rep->egen;
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Election thread owns egen %lu", (u_long)egen));
priority = lp->persist.version != DB_LOGVERSION ? 0 : realpri;
#ifdef CONFIG_TEST
if (priority > 0 && priority <= 5) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Artificially setting priority 0 (ELECTABLE) for CONFIG_TEST mode"));
DB_ASSERT(env, ctlflags == REPCTL_ELECTABLE);
priority = 0;
}
#endif
__os_gettime(env, &rep->etime, 1);
rep->full_elect = 0;
timeout = rep->elect_timeout;
if (!F_ISSET(rep, REP_F_GROUP_ESTD) && rep->full_elect_timeout != 0) {
rep->full_elect = 1;
timeout = rep->full_elect_timeout;
}
if ((ret = __rep_lockout_apply(env, rep, 0)) != 0)
goto err_locked;
if ((ret = __rep_lockout_archive(env, rep)) != 0)
goto err_locked;
if (rep->egen != egen) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Found egen %lu, abandon my election at egen %lu",
(u_long)rep->egen, (u_long)egen));
goto err_locked;
}
__os_unique_id(env, &tiebreaker);
FLD_SET(rep->elect_flags, REP_E_PHASE1);
FLD_CLR(rep->elect_flags, REP_E_TALLY);
if (IS_USING_LEASES(env) &&
__rep_islease_granted(env)) {
ret = 0;
goto err_locked;
}
INIT_LSN(lsn);
if (ISSET_LOCKOUT_BDB(rep) || IN_INTERNAL_INIT(rep) ||
rep->sync_state == SYNC_UPDATE) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Setting priority 0, unelectable, due to internal init/recovery"));
priority = 0;
ctlflags = 0;
data_gen = 0;
} else {
if ((ret = __log_cursor(env, &logc)) != 0)
goto err_locked;
if ((ret = __rep_log_backup(env,
logc, &lsn, REP_REC_COMMIT)) == DB_NOTFOUND) {
INIT_LSN(lsn);
ret = 0;
}
if ((t_ret = __logc_close(logc)) != 0 && ret == 0)
ret = t_ret;
if (ret != 0)
goto err_locked;
if ((ret = __rep_get_datagen(env, &data_gen)) != 0)
goto err_locked;
}
if ((ret = __rep_write_egen(env, rep, egen + 1)) != 0)
goto err_locked;
if ((ret = __rep_tally(env, rep, rep->eid, &rep->sites, egen, 1))
!= 0) {
DB_ASSERT(env, ret != DB_REP_IGNORE);
goto err_locked;
}
__rep_cmp_vote(env, rep, rep->eid, &lsn, priority, rep->gen, data_gen,
tiebreaker, ctlflags);
RPRINT(env, (env, DB_VERB_REP_ELECT, "Beginning an election"));
send_vote = DB_EID_INVALID;
done = IS_PHASE1_DONE(rep);
rep->vote1.lsn = lsn;
rep->vote1.nsites = nsites;
rep->vote1.nvotes = ack;
rep->vote1.priority = priority;
rep->vote1.tiebreaker = tiebreaker;
rep->vote1.ctlflags = ctlflags;
rep->vote1.data_gen = data_gen;
REP_SYSTEM_UNLOCK(env);
__rep_send_vote(env, &lsn, nsites, ack, priority, tiebreaker, egen,
data_gen, DB_EID_BROADCAST, REP_VOTE1, ctlflags);
DB_ENV_TEST_RECOVERY(env, DB_TEST_ELECTVOTE1, ret, NULL);
if (done) {
REP_SYSTEM_LOCK(env);
goto vote;
}
ret = __rep_wait(env, &timeout, rep->full_elect, egen, REP_E_PHASE1);
REP_SYSTEM_LOCK(env);
if (ret != 0)
goto err_locked;
if (rep->egen > egen)
goto out;
if (FLD_ISSET(rep->elect_flags, REP_E_PHASE2)) {
REP_SYSTEM_UNLOCK(env);
goto phase2;
}
if (rep->sites >= rep->nvotes) {
vote:
send_vote = rep->winner;
if (rep->winner == rep->eid) {
if ((ret =__rep_tally(env,
rep, rep->eid, &rep->votes, egen, 2)) != 0 &&
ret != DB_REP_IGNORE)
goto err_locked;
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Counted my vote %d", rep->votes));
}
FLD_SET(rep->elect_flags, REP_E_PHASE2);
FLD_CLR(rep->elect_flags, REP_E_PHASE1);
}
if (send_vote == DB_EID_INVALID) {
if (rep->sites >= rep->nvotes)
__db_errx(env, DB_STR_A("3533",
"No electable site found: recvd %d of %d votes from %d sites",
"%d %d %d"), rep->sites, rep->nvotes, rep->nsites);
else
__db_errx(env, DB_STR_A("3534",
"Not enough votes to elect: recvd %d of %d from %d sites",
"%d %d %d"), rep->sites, rep->nvotes, rep->nsites);
ret = DB_REP_UNAVAIL;
goto err_locked;
}
REP_SYSTEM_UNLOCK(env);
if (send_vote != rep->eid) {
RPRINT(env, (env, DB_VERB_REP_ELECT, "Sending vote"));
__rep_send_vote(env, NULL, 0, 0, 0, 0, egen, 0,
send_vote, REP_VOTE2, 0);
timeout = timeout * 2;
}
phase2:
if (I_HAVE_WON(rep, rep->winner)) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Skipping phase2 wait: already got %d votes", rep->votes));
REP_SYSTEM_LOCK(env);
goto i_won;
}
ret = __rep_wait(env, &timeout, rep->full_elect, egen, REP_E_PHASE2);
REP_SYSTEM_LOCK(env);
if (ret != 0)
goto err_locked;
if (rep->egen > egen || !IN_ELECTION(rep))
goto out;
ret = DB_REP_UNAVAIL;
RPRINT(env, (env, DB_VERB_REP_ELECT,
"After phase 2: votes %d, nvotes %d, nsites %d",
rep->votes, rep->nvotes, rep->nsites));
if (I_HAVE_WON(rep, rep->winner)) {
i_won: __rep_elect_master(env, rep);
ret = 0;
elected = 1;
}
err_locked:
if (ret == 0 || ret == DB_REP_UNAVAIL)
__rep_elect_done(env, rep);
else if (orig_tally)
FLD_SET(rep->elect_flags, orig_tally);
#ifdef CONFIG_TEST
if (0) {
DB_TEST_RECOVERY_LABEL
REP_SYSTEM_LOCK(env);
}
#endif
out:
need_req = 0;
DB_ASSERT(env, rep->elect_th > 0);
rep->elect_th--;
if (rep->elect_th == 0) {
need_req = F_ISSET(rep, REP_F_SKIPPED_APPLY) &&
!I_HAVE_WON(rep, rep->winner);
FLD_CLR(rep->lockout_flags, REP_LOCKOUT_APPLY);
F_CLR(rep, REP_F_SKIPPED_APPLY);
}
if (ret != 0 && rep->elect_th == 0)
FLD_CLR(rep->lockout_flags, REP_LOCKOUT_ARCHIVE);
REP_SYSTEM_UNLOCK(env);
if (need_req && (t_ret = __rep_resend_req(env, 0)) != 0 &&
(ret == 0 || ret == DB_REP_UNAVAIL || ret == DB_REP_IGNORE))
ret = t_ret;
if (elected) {
DB_ASSERT(env, ret != DB_REP_UNAVAIL && ret != DB_REP_IGNORE);
if ((t_ret = __rep_fire_elected(env, rep, egen)) != 0 &&
ret == 0)
ret = t_ret;
}
RPRINT(env, (env, DB_VERB_REP_ELECT,
"%s %d, e_th %lu, egen %lu, flag 0x%lx, e_fl 0x%lx, lo_fl 0x%lx",
"Ended election with ", ret,
(u_long) rep->elect_th, (u_long)rep->egen,
(u_long)rep->flags, (u_long)rep->elect_flags,
(u_long)rep->lockout_flags));
if (0) {
unlck_lv: REP_SYSTEM_UNLOCK(env);
}
envleave:
ENV_LEAVE(env, ip);
return (ret);
}
int
__rep_vote1(env, rp, rec, eid)
ENV *env;
__rep_control_args *rp;
DBT *rec;
int eid;
{
DBT data_dbt;
DB_LOG *dblp;
DB_LSN lsn;
DB_REP *db_rep;
LOG *lp;
REP *rep;
REP_OLD_VOTE_INFO *ovi;
VOTE1_CONTENT vote1;
__rep_egen_args egen_arg;
__rep_vote_info_v5_args tmpvi5;
__rep_vote_info_args tmpvi, *vi;
u_int32_t egen;
int elected, master, resend, ret;
u_int8_t buf[__REP_MAXMSG_SIZE];
size_t len;
COMPQUIET(egen, 0);
elected = resend = ret = 0;
db_rep = env->rep_handle;
rep = db_rep->region;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
if (F_ISSET(rep, REP_F_MASTER)) {
RPRINT(env, (env, DB_VERB_REP_ELECT, "Master received vote"));
LOG_SYSTEM_LOCK(env);
lsn = lp->lsn;
LOG_SYSTEM_UNLOCK(env);
(void)__rep_send_message(env,
DB_EID_BROADCAST, REP_NEWMASTER, &lsn, NULL, 0, 0);
return (ret);
}
if (rp->rep_version < DB_REPVERSION_47) {
ovi = (REP_OLD_VOTE_INFO *)rec->data;
tmpvi.egen = ovi->egen;
tmpvi.nsites = (u_int32_t)ovi->nsites;
tmpvi.nvotes = (u_int32_t)ovi->nvotes;
tmpvi.priority = (u_int32_t)ovi->priority;
tmpvi.tiebreaker = ovi->tiebreaker;
tmpvi.data_gen = 0;
} else if (rp->rep_version < DB_REPVERSION_52) {
if ((ret = __rep_vote_info_v5_unmarshal(env,
&tmpvi5, rec->data, rec->size, NULL)) != 0)
return (ret);
tmpvi.egen = tmpvi5.egen;
tmpvi.nsites = tmpvi5.nsites;
tmpvi.nvotes = tmpvi5.nvotes;
tmpvi.priority = tmpvi5.priority;
tmpvi.tiebreaker = tmpvi5.tiebreaker;
tmpvi.data_gen = 0;
} else
if ((ret = __rep_vote_info_unmarshal(env,
&tmpvi, rec->data, rec->size, NULL)) != 0)
return (ret);
vi = &tmpvi;
REP_SYSTEM_LOCK(env);
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Received vote1 egen %lu, egen %lu",
(u_long)vi->egen, (u_long)rep->egen));
if (vi->egen < rep->egen) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Received old vote %lu, egen %lu, ignoring vote1",
(u_long)vi->egen, (u_long)rep->egen));
egen_arg.egen = rep->egen;
REP_SYSTEM_UNLOCK(env);
if (rep->version < DB_REPVERSION_47)
DB_INIT_DBT(data_dbt, &egen_arg.egen,
sizeof(egen_arg.egen));
else {
if ((ret = __rep_egen_marshal(env,
&egen_arg, buf, __REP_EGEN_SIZE, &len)) != 0)
return (ret);
DB_INIT_DBT(data_dbt, buf, len);
}
(void)__rep_send_message(env,
eid, REP_ALIVE, &rp->lsn, &data_dbt, 0, 0);
return (0);
}
if (vi->egen > rep->egen) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Received VOTE1 from egen %lu, my egen %lu",
(u_long)vi->egen, (u_long)rep->egen));
__rep_elect_done(env, rep);
rep->egen = vi->egen;
}
if (!IN_ELECTION_TALLY(rep)) {
FLD_SET(rep->elect_flags, REP_E_TALLY);
rep->nsites = vi->nsites;
rep->nvotes = vi->nvotes;
} else {
if (vi->nsites > rep->nsites)
rep->nsites = vi->nsites;
if (vi->nvotes > rep->nvotes)
rep->nvotes = vi->nvotes;
}
if (FLD_ISSET(rep->elect_flags, REP_E_PHASE2)) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"In phase 2, ignoring vote1"));
goto err;
}
if ((ret = __rep_tally(env, rep, eid, &rep->sites, vi->egen, 1)) != 0) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Tally returned %d, sites %d", ret, rep->sites));
if (ret == DB_REP_IGNORE)
ret = 0;
goto err;
}
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Incoming vote: (eid)%d (pri)%lu %s (gen)%lu (egen)%lu (datagen)%lu [%lu,%lu]",
eid, (u_long)vi->priority,
F_ISSET(rp, REPCTL_ELECTABLE) ? "ELECTABLE" : "",
(u_long)rp->gen, (u_long)vi->egen, (u_long)vi->data_gen,
(u_long)rp->lsn.file, (u_long)rp->lsn.offset));
if (rep->sites > 1)
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Existing vote: (eid)%d (pri)%lu (gen)%lu (datagen)%lu (sites)%d [%lu,%lu]",
rep->winner, (u_long)rep->w_priority,
(u_long)rep->w_gen, (u_long)rep->w_datagen, rep->sites,
(u_long)rep->w_lsn.file,
(u_long)rep->w_lsn.offset));
__rep_cmp_vote(env, rep, eid, &rp->lsn, vi->priority,
rp->gen, vi->data_gen, vi->tiebreaker, rp->flags);
if (IN_ELECTION(rep)) {
if (rep->full_elect &&
FLD_ISSET((rep)->elect_flags, REP_E_PHASE1)) {
resend = 1;
vote1 = rep->vote1;
egen = rep->egen;
}
} else {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Not in election, but received vote1 0x%x 0x%x",
rep->flags, rep->elect_flags));
ret = DB_REP_HOLDELECTION;
goto err;
}
master = rep->winner;
lsn = rep->w_lsn;
if (IS_PHASE1_DONE(rep)) {
RPRINT(env, (env, DB_VERB_REP_ELECT, "Phase1 election done"));
RPRINT(env, (env, DB_VERB_REP_ELECT, "Voting for %d%s",
master, master == rep->eid ? "(self)" : ""));
egen = rep->egen;
FLD_SET(rep->elect_flags, REP_E_PHASE2);
FLD_CLR(rep->elect_flags, REP_E_PHASE1);
if (master == rep->eid) {
if ((ret =__rep_tally(env, rep, rep->eid,
&rep->votes, egen, 2)) != 0 &&
ret != DB_REP_IGNORE)
goto err;
ret = 0;
RPRINT(env, (env, DB_VERB_REP_ELECT,
"After phase 1 done: counted vote %d of %d",
rep->votes, rep->nvotes));
if (I_HAVE_WON(rep, rep->winner)) {
__rep_elect_master(env, rep);
elected = 1;
}
goto err;
}
REP_SYSTEM_UNLOCK(env);
__rep_send_vote(env, NULL, 0, 0, 0, 0, egen, 0,
master, REP_VOTE2, 0);
} else
err: REP_SYSTEM_UNLOCK(env);
if (elected)
ret = __rep_fire_elected(env, rep, egen);
else if (resend)
__rep_send_vote(env,
&vote1.lsn, vote1.nsites, vote1.nvotes, vote1.priority,
vote1.tiebreaker, egen, vote1.data_gen,
eid, REP_VOTE1, vote1.ctlflags);
return (ret);
}
int
__rep_vote2(env, rp, rec, eid)
ENV *env;
__rep_control_args *rp;
DBT *rec;
int eid;
{
DB_LOG *dblp;
DB_LSN lsn;
DB_REP *db_rep;
LOG *lp;
REP *rep;
REP_OLD_VOTE_INFO *ovi;
__rep_vote_info_args tmpvi, *vi;
u_int32_t egen;
int ret;
ret = 0;
db_rep = env->rep_handle;
rep = db_rep->region;
dblp = env->lg_handle;
lp = dblp->reginfo.primary;
RPRINT(env, (env, DB_VERB_REP_ELECT, "We received a vote%s",
F_ISSET(rep, REP_F_MASTER) ? " (master)" : ""));
if (F_ISSET(rep, REP_F_MASTER)) {
LOG_SYSTEM_LOCK(env);
lsn = lp->lsn;
LOG_SYSTEM_UNLOCK(env);
(void)__rep_send_message(env,
DB_EID_BROADCAST, REP_NEWMASTER, &lsn, NULL, 0, 0);
if (IS_USING_LEASES(env))
ret = __rep_lease_refresh(env);
return (ret);
}
REP_SYSTEM_LOCK(env);
egen = rep->egen;
if (rp->rep_version < DB_REPVERSION_47) {
ovi = (REP_OLD_VOTE_INFO *)rec->data;
tmpvi.egen = ovi->egen;
tmpvi.nsites = (u_int32_t)ovi->nsites;
tmpvi.nvotes = (u_int32_t)ovi->nvotes;
tmpvi.priority = (u_int32_t)ovi->priority;
tmpvi.tiebreaker = ovi->tiebreaker;
} else
if ((ret = __rep_vote_info_unmarshal(env,
&tmpvi, rec->data, rec->size, NULL)) != 0)
return (ret);
vi = &tmpvi;
if (!IN_ELECTION_TALLY(rep) && vi->egen >= rep->egen) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Not in election gen %lu, at %lu, got vote",
(u_long)vi->egen, (u_long)rep->egen));
ret = DB_REP_HOLDELECTION;
goto err;
}
if (vi->egen != rep->egen) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Bad vote egen %lu. Mine %lu",
(u_long)vi->egen, (u_long)rep->egen));
ret = 0;
goto err;
}
if ((ret = __rep_tally(env, rep, eid, &rep->votes, vi->egen, 2)) != 0) {
if (ret == DB_REP_IGNORE)
ret = 0;
goto err;
}
RPRINT(env, (env, DB_VERB_REP_ELECT, "Counted vote %d of %d",
rep->votes, rep->nvotes));
if (I_HAVE_WON(rep, rep->winner)) {
__rep_elect_master(env, rep);
ret = DB_REP_NEWMASTER;
}
err: REP_SYSTEM_UNLOCK(env);
if (ret == DB_REP_NEWMASTER)
ret = __rep_fire_elected(env, rep, egen);
return (ret);
}
static int
__rep_tally(env, rep, eid, countp, egen, phase)
ENV *env;
REP *rep;
int eid;
u_int32_t *countp;
u_int32_t egen;
int phase;
{
REP_VTALLY *tally, *vtp;
u_int32_t i;
int ret;
if (rep->nsites > rep->asites &&
(ret = __rep_grow_sites(env, rep->nsites)) != 0) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Grow sites returned error %d", ret));
return (ret);
}
if (phase == 1)
tally = R_ADDR(env->reginfo, rep->tally_off);
else
tally = R_ADDR(env->reginfo, rep->v2tally_off);
vtp = &tally[0];
for (i = 0; i < *countp;) {
if (vtp->eid == eid) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Tally found[%d] (%d, %lu), this vote (%d, %lu)",
i, vtp->eid, (u_long)vtp->egen,
eid, (u_long)egen));
if (vtp->egen >= egen)
return (DB_REP_IGNORE);
else {
vtp->egen = egen;
return (0);
}
}
i++;
vtp = &tally[i];
}
RPRINT(env, (env, DB_VERB_REP_ELECT, "Tallying VOTE%d[%d] (%d, %lu)",
phase, i, eid, (u_long)egen));
vtp->eid = eid;
vtp->egen = egen;
(*countp)++;
return (0);
}
static void
__rep_cmp_vote(env, rep, eid, lsnp, priority, gen, data_gen, tiebreaker, flags)
ENV *env;
REP *rep;
int eid;
DB_LSN *lsnp;
u_int32_t priority;
u_int32_t data_gen, flags, gen, tiebreaker;
{
int cmp, like_pri;
cmp = LOG_COMPARE(lsnp, &rep->w_lsn);
if (rep->sites > 1 &&
(priority != 0 || LF_ISSET(REPCTL_ELECTABLE))) {
like_pri = (priority == 0 && rep->w_priority == 0) ||
(priority != 0 && rep->w_priority != 0);
if ((priority != 0 && rep->w_priority == 0) ||
(like_pri && data_gen > rep->w_datagen) ||
(like_pri && data_gen == rep->w_datagen && cmp > 0) ||
(cmp == 0 && (priority > rep->w_priority ||
(priority == rep->w_priority &&
(tiebreaker > rep->w_tiebreaker))))) {
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Accepting new vote"));
rep->winner = eid;
rep->w_priority = priority;
rep->w_lsn = *lsnp;
rep->w_gen = gen;
rep->w_datagen = data_gen;
rep->w_tiebreaker = tiebreaker;
}
} else if (rep->sites == 1) {
if (priority != 0 || LF_ISSET(REPCTL_ELECTABLE)) {
rep->winner = eid;
rep->w_priority = priority;
rep->w_gen = gen;
rep->w_datagen = data_gen;
rep->w_lsn = *lsnp;
rep->w_tiebreaker = tiebreaker;
} else {
rep->winner = DB_EID_INVALID;
rep->w_priority = 0;
rep->w_gen = 0;
rep->w_datagen = 0;
ZERO_LSN(rep->w_lsn);
rep->w_tiebreaker = 0;
}
}
}
static int
__rep_elect_init(env, nsites, nvotes, beginp, otally)
ENV *env;
u_int32_t nsites, nvotes;
int *beginp;
u_int32_t *otally;
{
DB_REP *db_rep;
REP *rep;
int ret;
db_rep = env->rep_handle;
rep = db_rep->region;
ret = 0;
if (otally != NULL)
*otally = FLD_ISSET(rep->elect_flags, REP_E_TALLY);
DB_ASSERT(env, rep->spent_egen <= rep->egen);
*beginp = rep->spent_egen == rep->egen;
if (!*beginp) {
if (nsites > rep->asites &&
(ret = __rep_grow_sites(env, nsites)) != 0)
goto err;
DB_ENV_TEST_RECOVERY(env, DB_TEST_ELECTINIT, ret, NULL);
rep->spent_egen = rep->egen;
STAT_INC(env, rep, election, rep->stat.st_elections, rep->egen);
if (!IN_ELECTION_TALLY(rep)) {
rep->nsites = nsites;
rep->nvotes = nvotes;
} else {
if (nsites > rep->nsites)
rep->nsites = nsites;
if (nvotes > rep->nvotes)
rep->nvotes = nvotes;
}
}
DB_TEST_RECOVERY_LABEL
err:
return (ret);
}
static void
__rep_elect_master(env, rep)
ENV *env;
REP *rep;
{
if (F_ISSET(rep, REP_F_MASTERELECT | REP_F_MASTER)) {
return;
}
F_SET(rep, REP_F_MASTERELECT);
STAT_INC(env, rep, election_won, rep->stat.st_elections_won, rep->egen);
RPRINT(env, (env, DB_VERB_REP_ELECT,
"Got enough votes to win; election done; (prev) gen %lu",
(u_long)rep->gen));
}
static int
__rep_fire_elected(env, rep, egen)
ENV *env;
REP *rep;
u_int32_t egen;
{
REP_EVENT_LOCK(env);
if (rep->notified_egen < egen) {
__rep_fire_event(env, DB_EVENT_REP_ELECTED, NULL);
rep->notified_egen = egen;
}
REP_EVENT_UNLOCK(env);
return (0);
}
#define SLEEPTIME(timeout) \
((timeout > 5000000) ? 500000 : ((timeout >= 10) ? timeout / 10 : 1))
static int
__rep_wait(env, timeoutp, full_elect, egen, flags)
ENV *env;
db_timeout_t *timeoutp;
int full_elect;
u_int32_t egen, flags;
{
DB_REP *db_rep;
REP *rep;
int done;
u_int32_t sleeptime, sleeptotal, timeout;
db_rep = env->rep_handle;
rep = db_rep->region;
done = 0;
timeout = *timeoutp;
sleeptime = SLEEPTIME(timeout);
sleeptotal = 0;
while (sleeptotal < timeout) {
__os_yield(env, 0, sleeptime);
sleeptotal += sleeptime;
REP_SYSTEM_LOCK(env);
if (!LF_ISSET(REP_E_PHASE0) &&
full_elect && F_ISSET(rep, REP_F_GROUP_ESTD)) {
*timeoutp = rep->elect_timeout;
timeout = *timeoutp;
if (sleeptotal >= timeout)
done = 1;
else
sleeptime = SLEEPTIME(timeout);
}
if (egen != rep->egen || !FLD_ISSET(rep->elect_flags, flags))
done = 1;
REP_SYSTEM_UNLOCK(env);
if (done)
return (0);
}
return (0);
}
static int
__rep_grow_sites(env, nsites)
ENV *env;
u_int32_t nsites;
{
REGENV *renv;
REGINFO *infop;
REP *rep;
int ret, *tally;
u_int32_t nalloc;
rep = env->rep_handle->region;
nalloc = 2 * rep->asites;
if (nalloc < nsites)
nalloc = nsites;
infop = env->reginfo;
renv = infop->primary;
MUTEX_LOCK(env, renv->mtx_regenv);
if ((ret = __env_alloc(infop,
(size_t)nalloc * sizeof(REP_VTALLY), &tally)) == 0) {
if (rep->tally_off != INVALID_ROFF)
__env_alloc_free(
infop, R_ADDR(infop, rep->tally_off));
rep->tally_off = R_OFFSET(infop, tally);
if ((ret = __env_alloc(infop,
(size_t)nalloc * sizeof(REP_VTALLY), &tally)) == 0) {
if (rep->v2tally_off != INVALID_ROFF)
__env_alloc_free(infop,
R_ADDR(infop, rep->v2tally_off));
rep->v2tally_off = R_OFFSET(infop, tally);
rep->asites = nalloc;
rep->nsites = nsites;
} else {
if (rep->v2tally_off != INVALID_ROFF)
__env_alloc_free(infop,
R_ADDR(infop, rep->v2tally_off));
__env_alloc_free(infop,
R_ADDR(infop, rep->tally_off));
rep->v2tally_off = rep->tally_off = INVALID_ROFF;
rep->asites = 0;
}
}
MUTEX_UNLOCK(env, renv->mtx_regenv);
return (ret);
}
static void
__rep_send_vote(env, lsnp,
nsites, nvotes, pri, tie, egen, data_gen, eid, vtype, flags)
ENV *env;
DB_LSN *lsnp;
int eid;
u_int32_t nsites, nvotes, pri;
u_int32_t flags, egen, data_gen, tie, vtype;
{
DB_REP *db_rep;
DBT vote_dbt;
REP *rep;
REP_OLD_VOTE_INFO ovi;
__rep_vote_info_args vi;
__rep_vote_info_v5_args vi5;
u_int8_t buf[__REP_VOTE_INFO_SIZE];
size_t len;
db_rep = env->rep_handle;
rep = db_rep->region;
memset(&vi, 0, sizeof(vi));
memset(&vote_dbt, 0, sizeof(vote_dbt));
if (rep->version < DB_REPVERSION_47) {
ovi.egen = egen;
ovi.priority = (int) pri;
ovi.nsites = (int) nsites;
ovi.nvotes = (int) nvotes;
ovi.tiebreaker = tie;
DB_INIT_DBT(vote_dbt, &ovi, sizeof(ovi));
} else if (rep->version < DB_REPVERSION_52) {
vi5.egen = egen;
vi5.priority = pri;
vi5.nsites = nsites;
vi5.nvotes = nvotes;
vi5.tiebreaker = tie;
(void)__rep_vote_info_v5_marshal(env, &vi5, buf,
__REP_VOTE_INFO_SIZE, &len);
DB_INIT_DBT(vote_dbt, buf, len);
} else {
vi.egen = egen;
vi.priority = pri;
vi.nsites = nsites;
vi.nvotes = nvotes;
vi.tiebreaker = tie;
vi.data_gen = data_gen;
(void)__rep_vote_info_marshal(env, &vi, buf,
__REP_VOTE_INFO_SIZE, &len);
DB_INIT_DBT(vote_dbt, buf, len);
}
(void)__rep_send_message(env, eid, vtype, lsnp, &vote_dbt, flags, 0);
}