#include "db_config.h"
#include "db_int.h"
#include "dbinc/log.h"
static void __rep_find_entry __P((DB_ENV *, REP *, int, REP_LEASE_ENTRY **));
int
__rep_update_grant(dbenv, ts)
DB_ENV *dbenv;
db_timespec *ts;
{
DB_LOG *dblp;
DB_REP *db_rep;
DBT lease_dbt;
LOG *lp;
REP *rep;
REP_GRANT_INFO gi;
db_timespec mytime;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
timespecclear(&mytime);
__os_gettime(dbenv, &mytime);
timespecadd(&mytime, &rep->lease_duration);
REP_SYSTEM_LOCK(dbenv);
if (IN_ELECTION(rep)) {
REP_SYSTEM_UNLOCK(dbenv);
return (0);
}
if (timespeccmp(&mytime, &rep->grant_expire, >))
rep->grant_expire = mytime;
REP_SYSTEM_UNLOCK(dbenv);
memset(&gi, 0, sizeof(gi));
#ifdef DIAGNOSTIC
gi.expire_time = rep->grant_expire;
#endif
gi.msg_time = *ts;
DB_INIT_DBT(lease_dbt, &gi, sizeof(gi));
(void)__rep_send_message(dbenv, rep->master_id, REP_LEASE_GRANT,
&lp->max_perm_lsn, &lease_dbt, 0, 0);
return (0);
}
int
__rep_islease_granted(dbenv)
DB_ENV *dbenv;
{
DB_REP *db_rep;
REP *rep;
db_timespec mytime;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
timespecclear(&mytime);
__os_gettime(dbenv, &mytime);
return (timespeccmp(&mytime, &rep->grant_expire, <=) ? 1 : 0);
}
int
__rep_lease_table_alloc(dbenv, nsites)
DB_ENV *dbenv;
int nsites;
{
REGENV *renv;
REGINFO *infop;
REP *rep;
REP_LEASE_ENTRY *le, *table;
int i, *lease, ret;
rep = dbenv->rep_handle->region;
infop = dbenv->reginfo;
renv = infop->primary;
MUTEX_LOCK(dbenv, renv->mtx_regenv);
if ((ret = __env_alloc(infop, (size_t)nsites * sizeof(REP_LEASE_ENTRY),
&lease)) == 0) {
if (rep->lease_off != INVALID_ROFF)
__env_alloc_free(infop,
R_ADDR(infop, rep->lease_off));
rep->lease_off = R_OFFSET(infop, lease);
}
MUTEX_UNLOCK(dbenv, renv->mtx_regenv);
table = R_ADDR(infop, rep->lease_off);
for (i = 0; i < nsites; i++) {
le = &table[i];
le->eid = DB_EID_INVALID;
timespecclear(&le->start_time);
timespecclear(&le->end_time);
ZERO_LSN(le->lease_lsn);
}
return (ret);
}
int
__rep_lease_grant(dbenv, rp, rec, eid)
DB_ENV *dbenv;
REP_CONTROL *rp;
DBT *rec;
int eid;
{
DB_REP *db_rep;
REP *rep;
REP_GRANT_INFO *gi;
REP_LEASE_ENTRY *le;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
gi = (REP_GRANT_INFO *)rec->data;
le = NULL;
REP_SYSTEM_LOCK(dbenv);
__rep_find_entry(dbenv, rep, eid, &le);
DB_ASSERT(dbenv, le != NULL);
RPRINT(dbenv, (dbenv, "lease_grant: gi msg_time %lu %lu",
(u_long)gi->msg_time.tv_sec, (u_long)gi->msg_time.tv_nsec));
if (le->eid == DB_EID_INVALID ||
timespeccmp(&gi->msg_time, &le->start_time, >)) {
le->eid = eid;
le->start_time = gi->msg_time;
le->end_time = le->start_time;
timespecadd(&le->end_time, &rep->lease_duration);
RPRINT(dbenv, (dbenv,
"lease_grant: eid %d, start %lu %lu, end %lu %lu, duration %lu %lu",
le->eid, (u_long)le->start_time.tv_sec, (u_long)le->start_time.tv_nsec,
(u_long)le->end_time.tv_sec, (u_long)le->end_time.tv_nsec,
(u_long)rep->lease_duration.tv_sec, (u_long)rep->lease_duration.tv_nsec));
DB_ASSERT(dbenv, LOG_COMPARE(&rp->lsn, &le->lease_lsn) >= 0);
le->lease_lsn = rp->lsn;
}
REP_SYSTEM_UNLOCK(dbenv);
return (0);
}
static void
__rep_find_entry(dbenv, rep, eid, lep)
DB_ENV *dbenv;
REP *rep;
int eid;
REP_LEASE_ENTRY **lep;
{
REGINFO *infop;
REP_LEASE_ENTRY *le, *table;
int i;
infop = dbenv->reginfo;
table = R_ADDR(infop, rep->lease_off);
for (i = 0; i < rep->nsites; i++) {
le = &table[i];
if (le->eid == eid || le->eid == DB_EID_INVALID) {
*lep = le;
return;
}
}
return;
}
int
__rep_lease_check(dbenv, refresh)
DB_ENV *dbenv;
int refresh;
{
DB_LOG *dblp;
DB_LSN lease_lsn;
DB_REP *db_rep;
LOG *lp;
REGINFO *infop;
REP *rep;
REP_LEASE_ENTRY *le, *table;
db_timespec curtime;
int i, min_leases, ret, tries, valid_leases;
infop = dbenv->reginfo;
tries = 0;
retry:
ret = 0;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
dblp = dbenv->lg_handle;
lp = dblp->reginfo.primary;
LOG_SYSTEM_LOCK(dbenv);
lease_lsn = lp->max_perm_lsn;
LOG_SYSTEM_UNLOCK(dbenv);
REP_SYSTEM_LOCK(dbenv);
min_leases = rep->nsites / 2;
__os_gettime(dbenv, &curtime);
RPRINT(dbenv, (dbenv, "lease_check: min_leases %d curtime %lu %lu",
min_leases, (u_long)curtime.tv_sec, (u_long)curtime.tv_nsec));
table = R_ADDR(infop, rep->lease_off);
for (i = 0, valid_leases = 0;
i < rep->nsites && valid_leases < min_leases; i++) {
le = &table[i];
if (le->eid != DB_EID_INVALID) {
RPRINT(dbenv, (dbenv,
"lease_check: valid %d eid %d, lease_lsn [%lu][%lu]",
valid_leases, le->eid, (u_long)le->lease_lsn.file,
(u_long)le->lease_lsn.offset));
RPRINT(dbenv, (dbenv, "lease_check: endtime %lu %lu",
(u_long)le->end_time.tv_sec,
(u_long)le->end_time.tv_nsec));
}
if (le->eid != DB_EID_INVALID &&
timespeccmp(&le->end_time, &curtime, >=) &&
LOG_COMPARE(&le->lease_lsn, &lease_lsn) == 0)
valid_leases++;
}
REP_SYSTEM_UNLOCK(dbenv);
RPRINT(dbenv, (dbenv, "valid %d, min %d", valid_leases, min_leases));
if (valid_leases < min_leases) {
if (!refresh)
ret = DB_REP_LEASE_EXPIRED;
else {
if ((ret = __rep_lease_refresh(dbenv)) == 0) {
if (tries <= LEASE_REFRESH_TRIES) {
if (tries > 0)
__os_sleep(dbenv, 1, 0);
tries++;
goto retry;
} else
ret = DB_REP_LEASE_EXPIRED;
}
}
}
return (ret);
}
int
__rep_lease_refresh(dbenv)
DB_ENV *dbenv;
{
DBT rec;
DB_LOGC *logc;
DB_LSN lsn;
DB_REP *db_rep;
REP *rep;
int ret, t_ret;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
if ((ret = __log_cursor(dbenv, &logc)) != 0)
return (ret);
memset(&rec, 0, sizeof(rec));
memset(&lsn, 0, sizeof(lsn));
if ((ret = __rep_log_backup(dbenv, rep, logc, &lsn)) != 0)
goto err;
if ((ret = __logc_get(logc, &lsn, &rec, DB_CURRENT)) != 0)
goto err;
if ((ret = __rep_send_message(dbenv,
DB_EID_BROADCAST, REP_LOG, &lsn, &rec, REPCTL_PERM, 0)) != 0) {
(void)__rep_lease_expire(dbenv, 0);
ret = DB_REP_LEASE_EXPIRED;
}
err: if ((t_ret = __logc_close(logc)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__rep_lease_expire(dbenv, locked)
DB_ENV *dbenv;
int locked;
{
DB_REP *db_rep;
REGINFO *infop;
REP *rep;
REP_LEASE_ENTRY *le, *table;
int i, ret;
ret = 0;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
infop = dbenv->reginfo;
if (!locked)
REP_SYSTEM_LOCK(dbenv);
if (rep->lease_off != INVALID_ROFF) {
table = R_ADDR(infop, rep->lease_off);
for (i = 0; i < rep->nsites; i++) {
le = &table[i];
le->end_time = le->start_time;
}
}
if (!locked)
REP_SYSTEM_UNLOCK(dbenv);
return (ret);
}
db_timeout_t
__rep_lease_waittime(dbenv)
DB_ENV *dbenv;
{
DB_REP *db_rep;
REP *rep;
db_timespec exptime, mytime;
db_timeout_t to;
db_rep = dbenv->rep_handle;
rep = db_rep->region;
exptime = rep->grant_expire;
to = 0;
RPRINT(dbenv, (dbenv,
"wait_time: grant_expire %lu %lu lease_to %lu",
(u_long)exptime.tv_sec, (u_long)exptime.tv_nsec,
(u_long)rep->lease_timeout));
if (!timespecisset(&exptime))
to = rep->lease_timeout;
else {
__os_gettime(dbenv, &mytime);
RPRINT(dbenv, (dbenv,
"wait_time: mytime %lu %lu, grant_expire %lu %lu",
(u_long)mytime.tv_sec, (u_long)mytime.tv_nsec,
(u_long)exptime.tv_sec, (u_long)exptime.tv_nsec));
if (timespeccmp(&mytime, &exptime, <=)) {
timespecsub(&exptime, &mytime);
DB_TIMESPEC_TO_TIMEOUT(to, &exptime);
}
}
return (to);
}