mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1062 lines
27 KiB
1062 lines
27 KiB
/*++
|
|
|
|
Copyright (c) 2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
crs.c
|
|
|
|
Abstract:
|
|
|
|
Implements Consistency Replica Set Algorithm
|
|
|
|
Author:
|
|
|
|
Ahmed Mohamed (ahmedm) 1-Jan-2001
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
#include <nt.h>
|
|
#include <ntdef.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
|
|
#include <windows.h>
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#include "crs.h"
|
|
|
|
#define xmalloc(size) VirtualAlloc(NULL, size,MEM_RESERVE|MEM_COMMIT,PAGE_READWRITE)
|
|
|
|
#define xfree(buffer) VirtualFree(buffer, 0, MEM_RELEASE|MEM_DECOMMIT)
|
|
|
|
#define CrspEqual(r1,r2) ((r1)->hdr.seq == (r2)->hdr.seq && \
|
|
(r1)->hdr.epoch == (r2)->hdr.epoch && \
|
|
(r1)->hdr.state == (r2)->hdr.state)
|
|
|
|
DWORD CrsForcedQuorumSize = 0xffff;
|
|
|
|
void
|
|
WINAPI
|
|
CrsSetForcedQuorumSize(DWORD size)
|
|
{
|
|
CrsForcedQuorumSize = size;
|
|
}
|
|
|
|
DWORD
|
|
CrspFindLast(CrsInfo_t *p, DWORD logsz)
|
|
{
|
|
|
|
CrsRecord_t *rec, *last_rec;
|
|
BOOL err;
|
|
DWORD n, i;
|
|
|
|
n = SetFilePointer(p->fh, 0, NULL, FILE_BEGIN);
|
|
if (n == INVALID_SET_FILE_POINTER) {
|
|
return GetLastError();
|
|
}
|
|
|
|
err = ReadFile(p->fh, p->buf, logsz, &n, NULL);
|
|
if (!err)
|
|
return GetLastError();
|
|
|
|
if (n != logsz) {
|
|
CrsLog(("Crs%d: failed to load complete file, read %d expected %d\n",
|
|
p->lid,
|
|
n, logsz));
|
|
return ERROR_BAD_LENGTH;
|
|
}
|
|
ASSERT(p->max_records * CRS_RECORD_SZ == (int)n);
|
|
if(p->max_records * CRS_RECORD_SZ != (int)n) {
|
|
CrsLog(("Crs%d: unable to load log file %d bytes, got %d bytes\n",
|
|
p->lid, n, logsz));
|
|
return ERROR_BAD_LENGTH;
|
|
}
|
|
|
|
CrsLog(("Crs%d: loaded %d bytes, %d records\n", p->lid,
|
|
n, p->max_records));
|
|
|
|
last_rec = NULL;
|
|
rec = p->buf;
|
|
for (i = 0; i < logsz; i += CRS_RECORD_SZ, rec++) {
|
|
if (rec->hdr.tag != CRS_TAG) {
|
|
CrsLog(("crs%d: Bad record %d, got %x expected %x\n",
|
|
p->lid,
|
|
i/CRS_RECORD_SZ, rec->hdr.tag, CRS_TAG));
|
|
return ERROR_BAD_FORMAT;
|
|
}
|
|
|
|
if (!last_rec ||
|
|
rec->hdr.epoch > last_rec->hdr.epoch ||
|
|
(rec->hdr.epoch == last_rec->hdr.epoch &&
|
|
(rec->hdr.seq > last_rec->hdr.seq))) {
|
|
last_rec = rec;
|
|
}
|
|
}
|
|
ASSERT(last_rec);
|
|
|
|
// make sure only the last record is not committed or aborted
|
|
rec = p->buf;
|
|
for (i = 0; i < logsz; i += CRS_RECORD_SZ, rec++) {
|
|
if (!(rec->hdr.state & (CRS_COMMIT | CRS_ABORT))) {
|
|
if (rec != last_rec) {
|
|
CrsLog(("crs:%d Bad record %d state %x expected commit|abort\n",
|
|
p->lid, i/CRS_RECORD_SZ, rec->hdr.state));
|
|
return ERROR_INTERNAL_ERROR;
|
|
}
|
|
}
|
|
}
|
|
|
|
p->last_record = (int) (last_rec - p->buf);
|
|
p->seq = last_rec->hdr.seq;
|
|
p->epoch = last_rec->hdr.epoch;
|
|
|
|
return ERROR_SUCCESS;
|
|
|
|
}
|
|
|
|
#define CrspFlush(p,offset) CrspWrite(p,offset, CRS_SECTOR_SZ)
|
|
|
|
static
|
|
DWORD
|
|
CrspWrite(CrsInfo_t *p, int offset, DWORD length)
|
|
{
|
|
DWORD n;
|
|
|
|
p->pending = FALSE;
|
|
|
|
n = (DWORD) offset;
|
|
// write out last sector, assumes lock is held
|
|
ASSERT(offset < p->max_records);
|
|
offset = offset / CRS_RECORDS_PER_SECTOR;
|
|
|
|
CrsLog(("Crs%d: flush %d bytes record %d -> %d,%d\n", p->lid,
|
|
length, n,
|
|
offset, offset*CRS_SECTOR_SZ));
|
|
|
|
n = SetFilePointer(p->fh, offset * CRS_SECTOR_SZ, NULL, FILE_BEGIN);
|
|
if (n == INVALID_SET_FILE_POINTER) {
|
|
return GetLastError();
|
|
}
|
|
|
|
n = 0;
|
|
if (WriteFile(p->fh, (PVOID) &p->buf[offset*CRS_RECORDS_PER_SECTOR], length, &n, NULL)) {
|
|
if (n != length) {
|
|
CrsLog(("Write count mismatch, wrote %d, expected %d\n", n, length));
|
|
return ERROR_BAD_LENGTH;
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
n = GetLastError();
|
|
CrsLog(("Crs%d: flush record %d failed err %d\n", p->lid, offset, n));
|
|
if (n == ERROR_UNEXP_NET_ERR) {
|
|
// repeat the write one more time
|
|
p->pending = TRUE;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
static
|
|
DWORD
|
|
CrspAppendRecord(CrsInfo_t *p, CrsRecord_t *rr, CrsRecord_t **rec)
|
|
{
|
|
CrsRecord_t *q;
|
|
DWORD err;
|
|
|
|
// tag record
|
|
rr->hdr.tag = CRS_TAG;
|
|
|
|
// assumes lock is held
|
|
if ((p->last_record & CRS_SECTOR_MASK) == CRS_SECTOR_MASK) {
|
|
// flush current sector
|
|
err = CrspFlush(p, p->last_record);
|
|
if (err != ERROR_SUCCESS)
|
|
return err;
|
|
|
|
}
|
|
|
|
// advance last record
|
|
p->last_record++;
|
|
if (p->last_record == p->max_records)
|
|
p->last_record = 0;
|
|
|
|
CrsLog(("Crs%d: append record %d epoch %I64d seq %I64d state %x\n",
|
|
p->lid, p->last_record,
|
|
rr->hdr.epoch, rr->hdr.seq, rr->hdr.state));
|
|
|
|
// copy record
|
|
q = &p->buf[p->last_record];
|
|
memcpy((PVOID)q, (PVOID) rr, CRS_RECORD_SZ);
|
|
|
|
// flush it out now
|
|
err = CrspFlush(p, p->last_record);
|
|
if (err == ERROR_SUCCESS) {
|
|
if (rec) *rec = q;
|
|
} else {
|
|
if (p->last_record == 0)
|
|
p->last_record = p->max_records;
|
|
p->last_record--;
|
|
}
|
|
|
|
return err;
|
|
}
|
|
|
|
// NextRecord:
|
|
// if seq is null, fill in last record and return SUCCESS
|
|
// if seq is not found, return NOT_FOUND
|
|
// if seq is last record, return EOF
|
|
// otherwise return next record after seq in lrec and SUCCESS
|
|
DWORD
|
|
CrspNextLogRecord(CrsInfo_t *info, CrsRecord_t *seq,
|
|
CrsRecord_t *lrec, BOOLEAN this_flag)
|
|
{
|
|
CrsRecord_t *last, *p;
|
|
DWORD err = ERROR_SUCCESS;
|
|
|
|
if (lrec == NULL || info == NULL) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
// read record
|
|
EnterCriticalSection(&info->lock);
|
|
last = &info->buf[info->last_record];
|
|
if (seq == NULL) {
|
|
CrsLog(("Crs%d: last record %d %I64d %I64d\n",
|
|
info->lid, info->last_record, last->hdr.epoch, last->hdr.seq));
|
|
|
|
// read last record
|
|
memcpy(lrec, last, CRS_RECORD_SZ);
|
|
|
|
} else if (seq->hdr.epoch != last->hdr.epoch ||
|
|
seq->hdr.seq != last->hdr.seq) {
|
|
int i;
|
|
|
|
CrsLog(("Crs%d: last record %d %I64d %I64d search %I64d %I64d\n",
|
|
info->lid, info->last_record,
|
|
last->hdr.epoch, last->hdr.seq,
|
|
seq->hdr.epoch, seq->hdr.seq));
|
|
|
|
// assume we don't have it
|
|
p = seq;
|
|
seq = NULL;
|
|
// do a search instead of index, so that
|
|
// seq can be reset as epoch increments
|
|
for (i = 0; i < info->max_records; i++) {
|
|
last = &info->buf[i];
|
|
if (p->hdr.epoch == last->hdr.epoch &&
|
|
p->hdr.seq == last->hdr.seq) {
|
|
|
|
seq = last;
|
|
break;
|
|
}
|
|
}
|
|
if (seq != NULL) {
|
|
if (this_flag == FALSE) {
|
|
// return record after this one
|
|
i++;
|
|
if (i >= info->max_records)
|
|
i = 0;
|
|
seq = &info->buf[i];
|
|
}
|
|
CrsLog(("Crs%d: search found %d %I64d, %I64d\n", info->lid,
|
|
seq - info->buf, seq->hdr.epoch, seq->hdr.seq));
|
|
memcpy(lrec, seq, CRS_RECORD_SZ);
|
|
} else {
|
|
err = ERROR_NOT_FOUND;
|
|
}
|
|
} else {
|
|
|
|
CrsLog(("Crs%d: reached last record %d %I64d %I64d, %I64d %I64d\n",
|
|
info->lid, info->last_record,
|
|
last->hdr.epoch, last->hdr.seq,
|
|
seq->hdr.epoch, seq->hdr.seq));
|
|
|
|
if (this_flag == TRUE) {
|
|
// we are trying to read the last record
|
|
memcpy(lrec, last, CRS_RECORD_SZ);
|
|
err = ERROR_SUCCESS;
|
|
} else {
|
|
err = ERROR_HANDLE_EOF;
|
|
}
|
|
}
|
|
|
|
LeaveCriticalSection(&info->lock);
|
|
|
|
if (err == ERROR_SUCCESS && lrec->hdr.epoch == 0) {
|
|
// invalid rec, log is empty
|
|
err = ERROR_HANDLE_EOF;
|
|
}
|
|
|
|
|
|
return err;
|
|
}
|
|
|
|
// Call into fs with <undo, replay, query, disable, enable, done>
|
|
// undo: pass replica in recovery due to a conflict
|
|
// replay: replica is missing change, if replay fails with abort, we
|
|
// do a full copy; otherwise we issue a skip record
|
|
// query: ask replica if record was completed or not
|
|
// done: signal end of recovery and pass in new wset, rset
|
|
// we silently handle <abort(skip) and epoch records>
|
|
// abort: add a skip record
|
|
// epoch records: just log it as is
|
|
DWORD
|
|
CrspReplay(LPVOID rec)
|
|
{
|
|
CrsRecoveryBlk_t *rr;
|
|
CrsInfo_t *info, *minfo;
|
|
CrsRecord_t *p, *q;
|
|
CrsRecord_t lrec, mlrec;
|
|
DWORD err;
|
|
|
|
rr = (CrsRecoveryBlk_t *) rec;
|
|
info = rr->info;
|
|
minfo = rr->minfo;
|
|
|
|
CrsLog(("CrsReplay%d mid %d, lid %d leader_id %d\n",
|
|
rr->nid, rr->mid, info->lid, info->leader_id));
|
|
|
|
do {
|
|
p = NULL;
|
|
// read last record
|
|
err = CrspNextLogRecord(info, NULL, &lrec, FALSE);
|
|
if (err != ERROR_SUCCESS) {
|
|
CrsLog(("CrsReplay%d: unable to read last record %d\n",
|
|
info->lid, err));
|
|
break;
|
|
}
|
|
|
|
// find our last record in master replica
|
|
q = &lrec;
|
|
p = &mlrec;
|
|
err = CrspNextLogRecord(minfo, q, p, TRUE);
|
|
// if found and consistent with master, no undo
|
|
if (err == ERROR_SUCCESS && p->hdr.state == q->hdr.state) {
|
|
CrsLog(("CrsReplay%d: last record %I64d, %I64d consistent %x %x\n",
|
|
info->lid, q->hdr.epoch, q->hdr.seq,
|
|
p->hdr.state, q->hdr.state));
|
|
break;
|
|
}
|
|
|
|
if (err != ERROR_SUCCESS) {
|
|
CrsLog(("CrsReplay%d: missing lrec %I64d, %I64d in disk %d, err %d\n",
|
|
info->lid, q->hdr.epoch, q->hdr.seq, minfo->lid, err));
|
|
} else {
|
|
CrsLog(("CrsReplay%d: undo last record %I64d, %I64d %x needs %x\n",
|
|
info->lid, q->hdr.epoch, q->hdr.seq,
|
|
q->hdr.state, p->hdr.state));
|
|
ASSERT(p->hdr.state & (CRS_COMMIT|CRS_ABORT));
|
|
}
|
|
|
|
// last record is in conflict, we must undo it first
|
|
if (!(q->hdr.state & CRS_EPOCH)) {
|
|
// if we found this record in master and a conflict is detected,
|
|
// we undo it. Otherwise, we need to do a full copy
|
|
if (err == ERROR_SUCCESS) {
|
|
ASSERT(p->hdr.state & (CRS_COMMIT|CRS_ABORT));
|
|
ASSERT(q->hdr.state & CRS_PREPARE);
|
|
err = info->callback(info->callback_arg,
|
|
rr->nid, q,
|
|
CRS_ACTION_UNDO, rr->mid);
|
|
}
|
|
} else {
|
|
// A missing epoch record doesn't mean we are old. A regroup
|
|
// could have happened but no new data records got added. We
|
|
// undo it, and continue;
|
|
err = STATUS_SUCCESS;
|
|
}
|
|
|
|
if (err == STATUS_SUCCESS) {
|
|
// update current record, sequence, epoch
|
|
info->buf[info->last_record].hdr.state = 0;
|
|
info->buf[info->last_record].hdr.epoch = 0;
|
|
info->buf[info->last_record].hdr.seq = 0;
|
|
if (info->last_record == 0) {
|
|
info->last_record = info->max_records;
|
|
}
|
|
info->last_record--;
|
|
info->seq = info->buf[info->last_record].hdr.seq;
|
|
info->epoch = info->buf[info->last_record].hdr.epoch;
|
|
CrsLog(("CrsReplay%d: new last record %d %I64d, %I64d\n",
|
|
info->lid, info->last_record, info->epoch, info->seq));
|
|
} else {
|
|
// can't undo it, do full copy and readjust our log
|
|
CrsLog(("CrsReplay%d: Unable to undo record %I64d, %I64d\n",
|
|
info->lid, q->hdr.epoch, q->hdr.seq));
|
|
p = NULL;
|
|
}
|
|
} while (err == STATUS_SUCCESS && info->state == CRS_STATE_RECOVERY);
|
|
|
|
|
|
while (p != NULL && info->state == CRS_STATE_RECOVERY) {
|
|
// read master copy
|
|
err = CrspNextLogRecord(minfo, p, &mlrec, FALSE);
|
|
if (err != ERROR_SUCCESS) {
|
|
if (err == ERROR_HANDLE_EOF) {
|
|
CrsLog(("CrsReplay%d: last record %I64d, %I64d in disk %d\n",
|
|
info->lid, q->hdr.epoch, q->hdr.seq, minfo->lid));
|
|
|
|
// the last record is where we are at
|
|
info->seq = info->buf[info->last_record].hdr.seq;
|
|
info->epoch = info->buf[info->last_record].hdr.epoch;
|
|
|
|
// we reached the end, signal end of recovery
|
|
err = info->callback(info->callback_arg,
|
|
rr->nid, p,
|
|
CRS_ACTION_DONE, rr->mid);
|
|
|
|
goto exit;
|
|
}
|
|
break;
|
|
}
|
|
|
|
p = &mlrec;
|
|
if ((p->hdr.state & CRS_EPOCH) || (p->hdr.state & CRS_ABORT)) {
|
|
CrsLog(("CrsReplay%d: skip record %I64d, %I64d %x\n",
|
|
info->lid, p->hdr.epoch, p->hdr.seq, p->hdr.state));
|
|
err = !STATUS_SUCCESS;
|
|
} else if (p->hdr.state & CRS_COMMIT) {
|
|
err = info->callback(info->callback_arg,
|
|
rr->nid, p,
|
|
CRS_ACTION_REPLAY, rr->mid);
|
|
if (err == STATUS_TRANSACTION_ABORTED) {
|
|
CrsLog(("CrsReplay: failed nid %d seq %I64d err %d\n",
|
|
rr->nid, p->hdr.seq, err));
|
|
break;
|
|
}
|
|
} else {
|
|
ASSERT(p->hdr.state & CRS_PREPARE);
|
|
// what if the record is prepared but not yet committed or
|
|
// aborted; in transit record.
|
|
// stop now
|
|
CrsLog(("CrsReplay%d: bad record seq %I64d state %x\n",
|
|
rr->nid, p->hdr.seq, p->hdr.state));
|
|
break;
|
|
}
|
|
if (err != STATUS_SUCCESS) {
|
|
// add record
|
|
err = CrspAppendRecord(info, p, NULL);
|
|
if (err != ERROR_SUCCESS) {
|
|
CrsLog(("CrsReplay%d: failed append seq %I64d err %d\n",
|
|
rr->nid, p->hdr.seq, err));
|
|
break;
|
|
}
|
|
if (p->hdr.state & CRS_EPOCH) {
|
|
; //ASSERT(info->epoch+1 == p->hdr.epoch);
|
|
} else {
|
|
ASSERT(info->epoch == p->hdr.epoch);
|
|
ASSERT(info->seq+1 == p->hdr.seq);
|
|
}
|
|
info->seq = p->hdr.seq;
|
|
info->epoch = p->hdr.epoch;
|
|
} else {
|
|
// make sure we have added it
|
|
ASSERT(info->seq == p->hdr.seq);
|
|
ASSERT(info->epoch == p->hdr.epoch);
|
|
ASSERT(info->buf[info->last_record].hdr.seq == p->hdr.seq);
|
|
ASSERT(info->buf[info->last_record].hdr.epoch == p->hdr.epoch);
|
|
|
|
// Propagate dubious bit
|
|
if (p->hdr.state & CRS_DUBIOUS) {
|
|
info->buf[info->last_record].hdr.state |= CRS_DUBIOUS;
|
|
}
|
|
ASSERT(info->buf[info->last_record].hdr.state == p->hdr.state);
|
|
}
|
|
}
|
|
|
|
if (p == NULL || err != STATUS_SUCCESS) {
|
|
CrsLog(("CrsReplay%d: Full copy from disk %d\n",
|
|
info->lid, minfo->lid));
|
|
// we are out of date or need full recovery, do a full copy
|
|
err = info->callback(info->callback_arg,
|
|
rr->nid, NULL,
|
|
CRS_ACTION_COPY, rr->mid);
|
|
|
|
if (err == STATUS_SUCCESS) {
|
|
DWORD len;
|
|
|
|
// we now copy our master log and flush it
|
|
ASSERT(minfo->max_records == info->max_records);
|
|
|
|
len = info->max_records * CRS_RECORD_SZ;
|
|
memcpy(info->buf, minfo->buf, len);
|
|
err = CrspWrite(info, 0, len);
|
|
if (err == ERROR_SUCCESS) {
|
|
// adjust our state
|
|
info->last_record = minfo->last_record;
|
|
info->seq = info->buf[info->last_record].hdr.seq;
|
|
info->epoch = info->buf[info->last_record].hdr.epoch;
|
|
|
|
// we reached the end, signal end of recovery
|
|
err = info->callback(info->callback_arg,
|
|
rr->nid, p,
|
|
CRS_ACTION_DONE, rr->mid);
|
|
}
|
|
}
|
|
}
|
|
|
|
exit:
|
|
|
|
CrsLog(("CrsReplay%d mid %d status 0x%x\n", rr->nid, rr->mid, err));
|
|
|
|
return err;
|
|
}
|
|
|
|
|
|
/////////////////////// Public Functions //////////////////////
|
|
DWORD
|
|
WINAPI
|
|
CrsOpen(crs_callback_t callback, PVOID callback_arg, USHORT lid,
|
|
WCHAR *log_name, int max_logsectors, HANDLE *outhdl)
|
|
{
|
|
|
|
// Open the log file
|
|
// If the file in newly create, set the proper size
|
|
// If the file size is not the same size, we need to either
|
|
// expand or truncate the file. (truncate needs copy)
|
|
// Scan file to locate last sector and record
|
|
// If last record hasn't been commited, issue a query.
|
|
// If query succeeded then, mark it as committed.
|
|
// Set epoch,seq
|
|
DWORD status;
|
|
HANDLE maph;
|
|
CrsInfo_t *p;
|
|
int logsz;
|
|
|
|
if (outhdl == NULL) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
*outhdl = NULL;
|
|
|
|
p = (CrsInfo_t *) malloc(sizeof(*p));
|
|
if (p == NULL) {
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
memset((PVOID) p, 0, sizeof(*p));
|
|
|
|
CrsLog(("Crs%d file '%S'\n", lid, log_name));
|
|
p->lid = lid;
|
|
p->callback = callback;
|
|
p->callback_arg = callback_arg;
|
|
p->pending = FALSE;
|
|
|
|
// Create log file, and set size of newly created
|
|
p->fh = CreateFileW(log_name,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
FILE_SHARE_READ|FILE_SHARE_WRITE,
|
|
NULL,
|
|
OPEN_ALWAYS,
|
|
FILE_FLAG_WRITE_THROUGH,
|
|
NULL);
|
|
|
|
status = GetLastError();
|
|
if(p->fh == INVALID_HANDLE_VALUE){
|
|
free((char *) p);
|
|
return status;
|
|
}
|
|
|
|
// acquire an exclusive lock on the whole file
|
|
if (!LockFile(p->fh, 0, 0, (DWORD)-1, (DWORD)-1)) {
|
|
FILE_FULL_EA_INFORMATION ea[2] = {0};
|
|
IO_STATUS_BLOCK ios;
|
|
NTSTATUS err;
|
|
|
|
// get status
|
|
status = GetLastError();
|
|
|
|
// change the ea to cause a notification to happen
|
|
ea[0].NextEntryOffset = 0;
|
|
ea[0].Flags = 0;
|
|
ea[0].EaNameLength = 1;
|
|
ea[0].EaValueLength = 1;
|
|
ea[0].EaName[0] = 'X';
|
|
err = NtSetEaFile(p->fh, &ios, (PVOID) ea, sizeof(ea));
|
|
CrsLog(("Crs%d Setting EA %x\n", lid, err));
|
|
goto error;
|
|
}
|
|
|
|
if (status == ERROR_ALREADY_EXISTS) {
|
|
// todo: compare current file size to new size and adjust file
|
|
// size accordingly. For now, just use old size
|
|
logsz = GetFileSize(p->fh, NULL);
|
|
CrsLog(("Crs%d: Filesz %d max_sec %d\n", lid, logsz, max_logsectors));
|
|
ASSERT(logsz == max_logsectors * CRS_SECTOR_SZ);
|
|
} else {
|
|
//extend the file pointer to max size
|
|
logsz = max_logsectors * CRS_SECTOR_SZ;
|
|
SetFilePointer(p->fh, logsz, NULL, FILE_BEGIN);
|
|
SetEndOfFile(p->fh);
|
|
CrsLog(("Crs%d: Set Filesz %d max_sec %d\n", lid, logsz, max_logsectors));
|
|
}
|
|
|
|
// allocate file copy in memory
|
|
p->buf = xmalloc(logsz);
|
|
if (p->buf == NULL) {
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
goto error;
|
|
}
|
|
|
|
// set max record
|
|
p->max_records = logsz / CRS_RECORD_SZ;
|
|
|
|
if (status == ERROR_ALREADY_EXISTS) {
|
|
// load file and compute last epoch/seq
|
|
status = CrspFindLast(p, logsz);
|
|
} else {
|
|
status = !ERROR_SUCCESS;
|
|
}
|
|
// init the file, when we detect a read failure or first time
|
|
if (status != ERROR_SUCCESS) {
|
|
CrsRecord_t *r;
|
|
int i;
|
|
|
|
// initialize file
|
|
p->seq = 0;
|
|
p->epoch = 0;
|
|
p->last_record = 0;
|
|
|
|
r = p->buf;
|
|
for (i = 0; i < logsz; i+= CRS_RECORD_SZ, r++) {
|
|
r->hdr.epoch = p->epoch;
|
|
r->hdr.seq = p->seq;
|
|
r->hdr.tag = CRS_TAG;
|
|
r->hdr.state = CRS_COMMIT | CRS_PREPARE | CRS_EPOCH;
|
|
}
|
|
status = CrspWrite(p, 0, logsz);
|
|
}
|
|
|
|
if (status != ERROR_SUCCESS) {
|
|
goto error;
|
|
}
|
|
|
|
CrsLog(("Crs%d: %x Last record %d max %d epoch %I64d seq %I64d\n", p->lid,
|
|
p->fh,
|
|
p->last_record, p->max_records, p->epoch, p->seq));
|
|
|
|
// initialize rest of state
|
|
p->state = CRS_STATE_INIT;
|
|
p->refcnt = 1;
|
|
p->leader_id = 0;
|
|
InitializeCriticalSection(&p->lock);
|
|
|
|
*outhdl = p;
|
|
|
|
return ERROR_SUCCESS;
|
|
|
|
error:
|
|
CloseHandle(p->fh);
|
|
if (p->buf) {
|
|
xfree(p->buf);
|
|
}
|
|
free((PVOID) p);
|
|
return status;
|
|
}
|
|
|
|
//
|
|
DWORD
|
|
WINAPI
|
|
CrsStart(PVOID *hdls, ULONG alive_set, int cluster_sz,
|
|
ULONG *write_set, ULONG *read_set, ULONG *evict_set)
|
|
|
|
{
|
|
DWORD status;
|
|
CrsInfo_t **info = (CrsInfo_t **) hdls;
|
|
int i, active_sz, mid;
|
|
ULONG mask, active_set, fail_set;
|
|
CrsInfo_t *p;
|
|
CrsRecord_t *q, *mlrec;
|
|
|
|
if (write_set) *write_set = 0;
|
|
if (read_set) *read_set = 0;
|
|
if (evict_set) *evict_set = 0;
|
|
|
|
// no alive node
|
|
if (cluster_sz == 0 || alive_set == 0) {
|
|
// nothing to do
|
|
return ERROR_WRITE_PROTECT;
|
|
}
|
|
|
|
|
|
// scan each hdl and make sure it is initialized and lock all hdls
|
|
mask = alive_set;
|
|
for (i = 0; mask != 0; i++, mask = mask >> 1) {
|
|
if (!(mask & 0x1)) {
|
|
continue;
|
|
}
|
|
|
|
p = info[i];
|
|
if (p == NULL) {
|
|
continue;
|
|
}
|
|
|
|
EnterCriticalSection(&p->lock);
|
|
|
|
// check the state of the last record
|
|
p = info[i];
|
|
q = &p->buf[p->last_record];
|
|
CrsLog(("Crs%d last record %d epoch %I64d seq %I64d state %x\n",
|
|
p->lid, p->last_record,
|
|
q->hdr.epoch, q->hdr.seq, q->hdr.state));
|
|
}
|
|
|
|
mid = 0;
|
|
mlrec = NULL;
|
|
// select master replica
|
|
for (i = 0, mask = alive_set; mask != 0; i++, mask = mask >> 1) {
|
|
if (!(mask & 0x1)) {
|
|
continue;
|
|
}
|
|
p = info[i];
|
|
if (p == NULL)
|
|
continue;
|
|
|
|
q = &p->buf[p->last_record];
|
|
if (!mlrec ||
|
|
mlrec->hdr.epoch < q->hdr.epoch ||
|
|
(mlrec->hdr.epoch == q->hdr.epoch && mlrec->hdr.seq < q->hdr.seq) ||
|
|
(mlrec->hdr.epoch == q->hdr.epoch && mlrec->hdr.seq == q->hdr.seq &&
|
|
mlrec->hdr.state != q->hdr.state && (q->hdr.state & CRS_COMMIT))) {
|
|
|
|
mid = i;
|
|
mlrec = q;
|
|
}
|
|
}
|
|
|
|
ASSERT(mid != 0);
|
|
|
|
// if master last record is in doubt, query filesystem. If the filesystem
|
|
// is certain that the operation has occured, it returns STATUS_SUCCESS for
|
|
// COMMIT, STATUS_CANCELLED for ABORT, and STATUS_NOT_FOUND for can't tell.
|
|
// All undetermined IO must be undone and redone in all non-master replicas
|
|
// to ensure all replicas reach consistency. This statement is true even
|
|
// for replicas that are currently absent from our set. We tag such records
|
|
// we both COMMIT and ABORT, so that the replay thread issues replay for
|
|
// new records and undo,replay for last records
|
|
p = info[mid];
|
|
p->leader_id = (USHORT) mid;
|
|
ASSERT(mlrec != NULL);
|
|
if (!(mlrec->hdr.state & (CRS_COMMIT | CRS_ABORT))) {
|
|
ASSERT(mlrec->hdr.state & CRS_PREPARE);
|
|
status = p->callback(p->callback_arg, p->lid,
|
|
mlrec, CRS_ACTION_QUERY,
|
|
p->lid);
|
|
|
|
if (status == STATUS_SUCCESS) {
|
|
mlrec->hdr.state |= CRS_COMMIT;
|
|
} else if (status == STATUS_CANCELLED) {
|
|
mlrec->hdr.state |= CRS_ABORT;
|
|
} else if (status == STATUS_NOT_FOUND) {
|
|
// assume it is committed, but mark it for undo during recovery
|
|
mlrec->hdr.state |= (CRS_COMMIT | CRS_DUBIOUS);
|
|
}
|
|
|
|
// todo: if status == TRANSACTION_ABORTED, we need to bail out since
|
|
// must master is dead
|
|
// no need to flush, I think!
|
|
// CrspFlush(p, p->last_record);
|
|
|
|
// todo: what if the flush fails here, I am assuming that
|
|
// an append will equally fail.
|
|
}
|
|
|
|
|
|
ASSERT(mlrec->hdr.state & (CRS_COMMIT | CRS_ABORT));
|
|
|
|
// compute sync and recovery masks
|
|
fail_set = 0;
|
|
active_set = 0;
|
|
active_sz = 0;
|
|
for (i = 0, mask = alive_set; mask != 0; i++, mask = mask >> 1) {
|
|
if (!(mask & 0x1)) {
|
|
continue;
|
|
}
|
|
|
|
p = info[i];
|
|
if (p == NULL) {
|
|
continue;
|
|
}
|
|
|
|
// set leader id
|
|
p->leader_id = (USHORT) mid;
|
|
q = &p->buf[p->last_record];
|
|
|
|
if (CrspEqual(mlrec, q)) {
|
|
ASSERT(q->hdr.state & (CRS_COMMIT | CRS_ABORT));
|
|
p->state = CRS_STATE_READ;
|
|
active_set |= (1 << i);
|
|
active_sz++;
|
|
} else if (p->state != CRS_STATE_RECOVERY) {
|
|
CrsRecoveryBlk_t rrbuf;
|
|
CrsRecoveryBlk_t *rr = &rrbuf;
|
|
|
|
// recover replica
|
|
rr->nid = i;
|
|
rr->mid = mid;
|
|
rr->info = p;
|
|
rr->minfo = info[mid];
|
|
|
|
// set recovery state
|
|
p->state = CRS_STATE_RECOVERY;
|
|
|
|
status = CrspReplay((LPVOID) rr);
|
|
|
|
// if we fail, evict this replica
|
|
if (status != ERROR_SUCCESS) {
|
|
fail_set |= (1 << i);
|
|
} else {
|
|
// repeat this replica again
|
|
i--;
|
|
mask = mask << 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
// assume success
|
|
status = ERROR_SUCCESS;
|
|
|
|
// set read sets
|
|
if (read_set) *read_set = active_set;
|
|
|
|
if (!CRS_QUORUM(active_sz, cluster_sz)) {
|
|
CrsLog(("No quorum active %d cluster %d\n", active_sz, cluster_sz));
|
|
mid = 0;
|
|
status = ERROR_WRITE_PROTECT;
|
|
} else {
|
|
int pass_cnt = 0;
|
|
ULONG pass_set = 0;
|
|
|
|
// Enable writes on all active replicas
|
|
for (i = 0, mask = active_set; mask != 0; i++, mask = mask >> 1) {
|
|
CrsRecord_t rec;
|
|
if (!(mask & 0x1)) {
|
|
continue;
|
|
}
|
|
p = info[i];
|
|
if (p == NULL)
|
|
continue;
|
|
|
|
p->state = CRS_STATE_WRITE;
|
|
|
|
// we now generate a new epoch and flush it to the disk
|
|
p->epoch++;
|
|
if (p->epoch == 0)
|
|
p->epoch = 1;
|
|
// reset seq to zero
|
|
p->seq = 0;
|
|
|
|
// write new epoch now, if not a majority replicas succeeded in writing
|
|
// the new <epoch, seq> we fail
|
|
rec.hdr.epoch = p->epoch;
|
|
rec.hdr.seq = p->seq;
|
|
rec.hdr.state = CRS_PREPARE | CRS_COMMIT | CRS_EPOCH;
|
|
memset(rec.data, 0, sizeof(rec.data));
|
|
if (CrspAppendRecord(p, &rec, NULL) == ERROR_SUCCESS) {
|
|
pass_cnt++;
|
|
pass_set |= (1 << i);
|
|
} else {
|
|
fail_set |= (1 << i);
|
|
}
|
|
}
|
|
|
|
// Recheck to make sure all replicas have advanced epoch
|
|
if (!CRS_QUORUM(pass_cnt, cluster_sz)) {
|
|
CrsLog(("No quorum due to error pass %d cluster %d\n", pass_cnt, cluster_sz));
|
|
mid = 0;
|
|
pass_set = 0;
|
|
pass_cnt = 0;
|
|
status = ERROR_WRITE_PROTECT;
|
|
}
|
|
|
|
if (pass_cnt != active_sz) {
|
|
// some replicas have died
|
|
for (i = 0, mask = pass_set; mask != 0; i++, mask = mask >> 1) {
|
|
if ((alive_set & (1 << i)) && (!mask & (1 << i))) {
|
|
p = info[i];
|
|
ASSERT(p != NULL);
|
|
p->state = CRS_STATE_READ;
|
|
}
|
|
}
|
|
}
|
|
// set write set
|
|
if (write_set) *write_set = pass_set;
|
|
}
|
|
|
|
if (evict_set) *evict_set = fail_set;
|
|
|
|
// unlock all hdls and set new master if any
|
|
for (i = 0, mask = alive_set; mask != 0; i++, mask = mask >> 1) {
|
|
if (!(mask & 0x1)) {
|
|
continue;
|
|
}
|
|
p = info[i];
|
|
if (p == NULL)
|
|
continue;
|
|
|
|
p->leader_id = (USHORT) mid;
|
|
LeaveCriticalSection(&p->lock);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
void
|
|
WINAPI
|
|
CrsClose(PVOID hd)
|
|
{
|
|
DWORD err;
|
|
CrsInfo_t *info = (CrsInfo_t *) hd;
|
|
|
|
// If we any recovery threads running, make sure we terminate them first
|
|
// before close and free all of this stuff
|
|
if (info == NULL) {
|
|
CrsLog(("CrsClose: try to close a null handle!\n"));
|
|
return;
|
|
}
|
|
|
|
// Flush everything out and close the file
|
|
EnterCriticalSection(&info->lock);
|
|
// flush
|
|
CrspFlush(info, info->last_record);
|
|
LeaveCriticalSection(&info->lock);
|
|
|
|
DeleteCriticalSection(&info->lock);
|
|
|
|
err = CloseHandle(info->fh);
|
|
|
|
CrsLog(("Crs%d: %x Closed %d\n", info->fh, info->lid, err));
|
|
|
|
xfree(info->buf);
|
|
free((char *) info);
|
|
}
|
|
|
|
void
|
|
WINAPI
|
|
CrsFlush(PVOID hd)
|
|
{
|
|
CrsInfo_t *info = (CrsInfo_t *) hd;
|
|
|
|
// if we have a commit or abort that isn't flushed yet, flush it now
|
|
EnterCriticalSection(&info->lock);
|
|
if (info->pending == TRUE) {
|
|
CrspFlush(info, info->last_record);
|
|
}
|
|
LeaveCriticalSection(&info->lock);
|
|
}
|
|
|
|
PVOID
|
|
WINAPI
|
|
CrsPrepareRecord(PVOID hd, PVOID lrec, crs_id_t id)
|
|
{
|
|
CrsRecord_t *p = (CrsRecord_t *)lrec;
|
|
CrsInfo_t *info = (CrsInfo_t *) hd;
|
|
DWORD err;
|
|
|
|
// move to correct slot in this sector. If we need a new sector,
|
|
// read it from the file. Make sure we flush any pending commits on
|
|
// current sector before we over write our in memory sector buffer.
|
|
|
|
// prepare record, if seq none 0 then we are skipping the next sequence
|
|
EnterCriticalSection(&info->lock);
|
|
|
|
if (info->state == CRS_STATE_WRITE ||
|
|
(info->state == CRS_STATE_RECOVERY && id != NULL && id[0] != 0)) {
|
|
|
|
if (id != NULL && id[0] != 0) {
|
|
CrsHdr_t *tmp = (CrsHdr_t *) id;
|
|
assert(id[0] == info->seq+1);
|
|
p->hdr.seq = tmp->seq;
|
|
p->hdr.epoch = tmp->epoch;
|
|
} else {
|
|
p->hdr.seq = info->seq+1;
|
|
p->hdr.epoch = info->epoch;
|
|
}
|
|
p->hdr.state = CRS_PREPARE;
|
|
err = CrspAppendRecord(info, p, &p);
|
|
if (err == ERROR_SUCCESS) {
|
|
// we return with the lock held, gets release on commitorabort
|
|
CrsLog(("Crs%d prepare %x seq %I64d\n",info->lid, p, p->hdr.seq));
|
|
return p;
|
|
}
|
|
CrsLog(("Crs%d: Append failed seq %I64%d\n", info->lid, p->hdr.seq));
|
|
} else {
|
|
CrsLog(("Crs%d: Prepare bad state %d id %x\n", info->lid, info->state, id));
|
|
}
|
|
|
|
LeaveCriticalSection(&info->lock);
|
|
return NULL;
|
|
}
|
|
|
|
int
|
|
WINAPI
|
|
CrsCommitOrAbort(PVOID hd, PVOID lrec, int commit)
|
|
{
|
|
CrsRecord_t *p = (CrsRecord_t *)lrec;
|
|
CrsInfo_t *info = (CrsInfo_t *) hd;
|
|
|
|
if (p == NULL || info == NULL) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
// update state of record
|
|
if (p->hdr.seq != info->seq+1) {
|
|
CrsLog(("Crs: sequence mis-match on commit|abort %I64d %I64d\n",
|
|
p->hdr.seq, info->seq));
|
|
assert(0);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
assert(!(p->hdr.state & (CRS_COMMIT | CRS_ABORT)));
|
|
|
|
// todo: this is wrong, what if one replica succeeds
|
|
// and others abort. Now, the others will reuse the
|
|
// same seq for a different update and when the
|
|
// succeeded replica rejoins it can't tell that the
|
|
// sequence got reused.
|
|
if (commit == TRUE) {
|
|
p->hdr.state |= CRS_COMMIT;
|
|
// advance the sequence
|
|
info->seq++;
|
|
CrsLog(("Crs%d: commit last %d leader %d seq %I64d\n", info->lid,
|
|
info->last_record,
|
|
info->leader_id, p->hdr.seq));
|
|
} else {
|
|
p->hdr.state |= CRS_ABORT;
|
|
// we need to re-adjust our last record
|
|
if (info->last_record == 0) {
|
|
info->last_record = info->max_records;
|
|
}
|
|
info->last_record--;
|
|
CrsLog(("Crs%d: abort last %d leader %d seq %I64d\n", info->lid,
|
|
info->last_record,
|
|
info->leader_id, p->hdr.seq));
|
|
}
|
|
|
|
info->pending = TRUE;
|
|
LeaveCriticalSection(&info->lock);
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
int
|
|
WINAPI
|
|
CrsCanWrite(PVOID hd)
|
|
{
|
|
CrsInfo_t *info = (CrsInfo_t *) hd;
|
|
int err;
|
|
|
|
// do we have a quorm or not
|
|
EnterCriticalSection(&info->lock);
|
|
err = (info->state == CRS_STATE_WRITE);
|
|
LeaveCriticalSection(&info->lock);
|
|
return err;
|
|
}
|
|
|
|
|
|
|