mars/kernel/mars_trans_logger.c

3695 lines
97 KiB
C

/*
* MARS Long Distance Replication Software
*
* This file is part of MARS project: http://schoebel.github.io/mars/
*
* Copyright (C) 2010-2014 Thomas Schoebel-Theuer
* Copyright (C) 2011-2014 1&1 Internet AG
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
// Trans_Logger brick
//#define BRICK_DEBUGGING
#define MARS_DEBUGGING
//#define IO_DEBUGGING
//#define REPLAY_DEBUGGING
#define STAT_DEBUGGING // here means: display full statistics
//#define HASH_DEBUGGING
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/string.h>
#include <linux/bio.h>
#include "brick_wait.h"
#include "mars.h"
#include "lib_limiter.h"
#include "mars_trans_logger.h"
// variants
#define KEEP_UNIQUE
#define DELAY_CALLERS // this is _needed_ for production systems
#define SHORTCUT_1_to_3 // when possible, queue 1 executes phase3_startio() directly without intermediate queueing into queue 3 => may be irritating, but has better performance. NOTICE: when some day the IO scheduling should be different between queue 1 and 3, you MUST disable this in order to distinguish between them!
// commenting this out is dangerous for data integrity! use only for testing!
#define USE_MEMCPY
#define DO_WRITEBACK // otherwise FAKE IO
#define REPLAY_DATA
// tuning
#ifdef BRICK_DEBUG_MEM
#define CONF_TRANS_CHUNKSIZE (128 * 1024 - PAGE_SIZE * 2)
#else
#define CONF_TRANS_CHUNKSIZE (128 * 1024)
#endif
#define CONF_TRANS_MAX_MREF_SIZE PAGE_SIZE
//#define CONF_TRANS_ALIGN PAGE_SIZE // FIXME: does not work
#define CONF_TRANS_ALIGN 0
#ifdef REPLAY_DEBUGGING
#define MARS_RPL(_fmt, _args...) _MARS_MSG(false, "REPLAY ", _fmt, ##_args)
#else
#define MARS_RPL(_args...) /*empty*/
#endif
#if 0
#define inline noinline
#endif
struct trans_logger_hash_anchor {
struct rw_semaphore hash_mutex;
struct list_head hash_anchor;
};
#define MAX_HASH_PAGES (PAGE_SIZE / sizeof(struct trans_logger_hash_anchor*))
#define NR_HASH_PAGES MAX_HASH_PAGES
#define HASH_PER_PAGE (PAGE_SIZE / sizeof(struct trans_logger_hash_anchor))
#define HASH_TOTAL (NR_HASH_PAGES * HASH_PER_PAGE)
///////////////////////// global tuning ////////////////////////
int trans_logger_completion_semantics = 1;
/* Global enabling / disabling of features */
int trans_logger_do_crc =
#ifdef CONFIG_MARS_DEBUG
true;
#else
false;
#endif
int trans_logger_allow_compress = true;
int trans_logger_mem_usage; // in KB
int trans_logger_pressure_limit = 0;
int trans_logger_disable_pressure = 0; /* only for testing */
int trans_logger_report_interval = 2; /* seconds */
int trans_logger_writeback_maxage = 900; /* seconds */
int trans_logger_max_interleave = -1;
int trans_logger_resume = 1;
int trans_logger_replay_timeout = 1; // in s
struct writeback_group global_writeback = {
.mutex = __RWSEM_INITIALIZER(global_writeback.mutex),
.group_anchor = LIST_HEAD_INIT(global_writeback.group_anchor),
.until_percent = 30,
};
static
void add_to_group(struct writeback_group *gr, struct trans_logger_brick *brick)
{
down_write(&gr->mutex);
list_add_tail(&brick->group_head, &gr->group_anchor);
up_write(&gr->mutex);
}
static
void remove_from_group(struct writeback_group *gr, struct trans_logger_brick *brick)
{
down_write(&gr->mutex);
list_del_init(&brick->group_head);
gr->leader = NULL;
up_write(&gr->mutex);
}
static
struct trans_logger_brick *elect_leader(struct writeback_group *gr)
{
struct trans_logger_brick *res = gr->leader;
struct list_head *tmp;
if (res && gr->until_percent >= 0) {
loff_t used = atomic64_read(&res->shadow_mem_used);
if (used > gr->biggest * gr->until_percent / 100)
goto done;
}
/* FIXME: use O(log n) data structure instead */
down_read(&gr->mutex);
for (tmp = gr->group_anchor.next; tmp != &gr->group_anchor; tmp = tmp->next) {
struct trans_logger_brick *test = container_of(tmp, struct trans_logger_brick, group_head);
loff_t new_used = atomic64_read(&test->shadow_mem_used);
if (!res || new_used > atomic64_read(&res->shadow_mem_used)) {
res = test;
gr->biggest = new_used;
}
}
up_read(&gr->mutex);
gr->leader = res;
done:
return res;
}
///////////////////////// own type definitions ////////////////////////
static inline
int lh_cmp(loff_t *a, loff_t *b)
{
if (*a < *b)
return -1;
if (*a > *b)
return 1;
return 0;
}
static inline
int tr_cmp(struct pairing_heap_logger *_a, struct pairing_heap_logger *_b)
{
struct logger_head *a = container_of(_a, struct logger_head, ph);
struct logger_head *b = container_of(_b, struct logger_head, ph);
return lh_cmp(a->lh_pos, b->lh_pos);
}
_PAIRING_HEAP_FUNCTIONS(static,logger,tr_cmp);
static inline
loff_t *lh_get(struct logger_head *th)
{
return th->lh_pos;
}
QUEUE_FUNCTIONS(logger,struct logger_head,lh_head,lh_get,lh_cmp,logger);
////////////////////////// logger queue handling ////////////////////////
static inline
void qq_init(struct logger_queue *q, struct trans_logger_brick *brick)
{
q_logger_init(q);
q->q_brick = brick;
}
static inline
void qq_activate(struct logger_queue *q)
{
q_logger_activate(q, 1);
}
static inline
void qq_deactivate(struct logger_queue *q)
{
q_logger_activate(q, -1);
}
static inline
void qq_mref_insert(struct logger_queue *q, struct trans_logger_mref_aspect *mref_a)
{
struct mref_object *mref = mref_a->object;
_mref_get(mref); // must be paired with __trans_logger_ref_put()
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&q->q_brick->inner_balance_count);
#endif
mars_trace(mref, q->q_insert_info);
q_logger_insert(q, &mref_a->lh);
}
static inline
void qq_wb_insert(struct logger_queue *q, struct writeback_info *wb)
{
q_logger_insert(q, &wb->w_lh);
}
static inline
void qq_mref_pushback(struct logger_queue *q, struct trans_logger_mref_aspect *mref_a)
{
_mref_check(mref_a->object);
mars_trace(mref_a->object, q->q_pushback_info);
q->pushback_count++;
q_logger_pushback(q, &mref_a->lh);
}
static inline
void qq_wb_pushback(struct logger_queue *q, struct writeback_info *wb)
{
q->pushback_count++;
q_logger_pushback(q, &wb->w_lh);
}
static inline
struct trans_logger_mref_aspect *qq_mref_fetch(struct logger_queue *q)
{
struct logger_head *test;
struct trans_logger_mref_aspect *mref_a = NULL;
test = q_logger_fetch(q);
if (test) {
mref_a = container_of(test, struct trans_logger_mref_aspect, lh);
_mref_check(mref_a->object);
mars_trace(mref_a->object, q->q_fetch_info);
}
return mref_a;
}
static inline
struct writeback_info *qq_wb_fetch(struct logger_queue *q)
{
struct logger_head *test;
struct writeback_info *res = NULL;
test = q_logger_fetch(q);
if (test) {
res = container_of(test, struct writeback_info, w_lh);
}
return res;
}
///////////////////////// own helper functions ////////////////////////
static inline
int hash_fn(loff_t pos)
{
// simple and stupid
long base_index = pos >> REGION_SIZE_BITS;
base_index += base_index / HASH_TOTAL / 7;
return base_index % HASH_TOTAL;
}
static inline
struct trans_logger_mref_aspect *_hash_find(struct list_head *start, loff_t pos, int *max_len, bool use_collect_head, bool find_unstable)
{
struct list_head *tmp;
struct trans_logger_mref_aspect *res = NULL;
int len = *max_len;
#ifdef HASH_DEBUGGING
int count = 0;
#endif
/* The lists are always sorted according to age (newest first).
* Caution: there may be duplicates in the list, some of them
* overlapping with the search area in many different ways.
*/
for (tmp = start->next; tmp != start; tmp = tmp->next) {
struct trans_logger_mref_aspect *test_a;
struct mref_object *test;
int diff;
#ifdef HASH_DEBUGGING
static int max = 0;
if (++count > max) {
max = count;
if (!(max % 100)) {
MARS_INF("hash max=%d hash=%d (pos=%lld)\n", max, hash_fn(pos), pos);
}
}
#endif
if (use_collect_head) {
test_a = container_of(tmp, struct trans_logger_mref_aspect, collect_head);
} else {
test_a = container_of(tmp, struct trans_logger_mref_aspect, hash_head);
}
test = test_a->object;
_mref_check(test);
// are the regions overlapping?
if (pos >= test->ref_pos + test->ref_len || pos + len <= test->ref_pos) {
continue; // not relevant
}
// searching for unstable elements (only in special cases)
if (find_unstable && test_a->is_stable)
break;
diff = test->ref_pos - pos;
if (diff <= 0) {
int restlen = test->ref_len + diff;
res = test_a;
if (restlen < len) {
len = restlen;
}
break;
}
if (diff < len) {
len = diff;
}
}
*max_len = len;
return res;
}
static noinline
struct trans_logger_mref_aspect *hash_find(struct trans_logger_brick *brick, loff_t pos, int *max_len, bool find_unstable)
{
int hash = hash_fn(pos);
struct trans_logger_hash_anchor *sub_table = brick->hash_table[hash / HASH_PER_PAGE];
struct trans_logger_hash_anchor *start = &sub_table[hash % HASH_PER_PAGE];
struct trans_logger_mref_aspect *res;
//unsigned int flags;
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_hash_find_count);
#endif
down_read(&start->hash_mutex);
res = _hash_find(&start->hash_anchor, pos, max_len, false, find_unstable);
/* Ensure the found mref can't go away...
*/
if (res && res->object)
_mref_get(res->object);
up_read(&start->hash_mutex);
return res;
}
static noinline
void hash_insert(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *elem_a)
{
int hash = hash_fn(elem_a->object->ref_pos);
struct trans_logger_hash_anchor *sub_table = brick->hash_table[hash / HASH_PER_PAGE];
struct trans_logger_hash_anchor *start = &sub_table[hash % HASH_PER_PAGE];
//unsigned int flags;
#if 1
CHECK_HEAD_EMPTY(&elem_a->hash_head);
_mref_check(elem_a->object);
#endif
#ifdef ADDITIONAL_COUNTERS
// only for statistics:
atomic_inc(&brick->hash_count);
atomic_inc(&brick->total_hash_insert_count);
#endif
down_write(&start->hash_mutex);
list_add(&elem_a->hash_head, &start->hash_anchor);
elem_a->is_hashed = true;
up_write(&start->hash_mutex);
}
/* Find the transitive closure of overlapping requests
* and collect them into a list.
*/
static noinline
void hash_extend(struct trans_logger_brick *brick, loff_t *_pos, int *_len, struct list_head *collect_list)
{
loff_t pos = *_pos;
int len = *_len;
int hash = hash_fn(pos);
struct trans_logger_hash_anchor *sub_table = brick->hash_table[hash / HASH_PER_PAGE];
struct trans_logger_hash_anchor *start = &sub_table[hash % HASH_PER_PAGE];
struct list_head *tmp;
bool extended;
//unsigned int flags;
#ifdef HASH_DEBUGGING
int count = 0;
static int max = 0;
#endif
if (collect_list) {
CHECK_HEAD_EMPTY(collect_list);
}
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_hash_extend_count);
#endif
down_read(&start->hash_mutex);
do {
extended = false;
for (tmp = start->hash_anchor.next; tmp != &start->hash_anchor; tmp = tmp->next) {
struct trans_logger_mref_aspect *test_a;
struct mref_object *test;
loff_t diff;
#ifdef HASH_DEBUGGING
count++;
#endif
test_a = container_of(tmp, struct trans_logger_mref_aspect, hash_head);
test = test_a->object;
_mref_check(test);
// are the regions overlapping?
if (pos >= test->ref_pos + test->ref_len || pos + len <= test->ref_pos) {
continue; // not relevant
}
// collision detection
if (test_a->is_collected)
goto collision;
// no writeback of non-persistent data
if (!(test_a->is_persistent & test_a->is_completed))
goto collision;
// extend the search region when necessary
diff = pos - test->ref_pos;
if (diff > 0) {
len += diff;
pos = test->ref_pos;
extended = true;
}
diff = (test->ref_pos + test->ref_len) - (pos + len);
if (diff > 0) {
len += diff;
extended = true;
}
}
} while (extended); // start over for transitive closure
*_pos = pos;
*_len = len;
#ifdef HASH_DEBUGGING
if (count > max + 100) {
int i = 0;
max = count;
MARS_INF("iterations max=%d hash=%d (pos=%lld len=%d)\n", max, hash, pos, len);
for (tmp = start->hash_anchor.next; tmp != &start->hash_anchor; tmp = tmp->next) {
struct trans_logger_mref_aspect *test_a;
struct mref_object *test;
test_a = container_of(tmp, struct trans_logger_mref_aspect, hash_head);
test = test_a->object;
MARS_INF("%03d pos = %lld len = %d collected = %d\n", i++, test->ref_pos, test->ref_len, test_a->is_collected);
}
MARS_INF("----------------\n");
}
#endif
for (tmp = start->hash_anchor.next; tmp != &start->hash_anchor; tmp = tmp->next) {
struct trans_logger_mref_aspect *test_a;
struct mref_object *test;
test_a = container_of(tmp, struct trans_logger_mref_aspect, hash_head);
test = test_a->object;
// are the regions overlapping?
if (pos >= test->ref_pos + test->ref_len || pos + len <= test->ref_pos) {
continue; // not relevant
}
// collect
CHECK_HEAD_EMPTY(&test_a->collect_head);
if (unlikely(test_a->is_collected)) {
MARS_ERR("collision detection did not work\n");
}
test_a->is_collected = true;
_mref_check(test);
list_add_tail(&test_a->collect_head, collect_list);
}
collision:
up_read(&start->hash_mutex);
}
/* Atomically put all elements from the list.
* All elements must reside in the same collision list.
*/
static inline
void hash_put_all(struct trans_logger_brick *brick, struct list_head *list)
{
struct list_head *tmp;
struct trans_logger_hash_anchor *start = NULL;
int first_hash = -1;
//unsigned int flags;
for (tmp = list->next; tmp != list; tmp = tmp->next) {
struct trans_logger_mref_aspect *elem_a;
struct mref_object *elem;
int hash;
elem_a = container_of(tmp, struct trans_logger_mref_aspect, collect_head);
elem = elem_a->object;
CHECK_PTR(elem, err);
_mref_check(elem);
hash = hash_fn(elem->ref_pos);
if (!start) {
struct trans_logger_hash_anchor *sub_table = brick->hash_table[hash / HASH_PER_PAGE];
start = &sub_table[hash % HASH_PER_PAGE];
first_hash = hash;
down_write(&start->hash_mutex);
} else if (unlikely(hash != first_hash)) {
MARS_ERR("oops, different hashes: %d != %d\n", hash, first_hash);
}
if (!elem_a->is_hashed) {
continue;
}
list_del_init(&elem_a->hash_head);
elem_a->is_hashed = false;
#ifdef ADDITIONAL_COUNTERS
atomic_dec(&brick->hash_count);
#endif
}
err:
if (start) {
up_write(&start->hash_mutex);
}
}
static inline
void hash_ensure_stableness(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *mref_a)
{
if (!mref_a->is_stable) {
struct mref_object *mref = mref_a->object;
int hash = hash_fn(mref->ref_pos);
struct trans_logger_hash_anchor *sub_table = brick->hash_table[hash / HASH_PER_PAGE];
struct trans_logger_hash_anchor *start = &sub_table[hash % HASH_PER_PAGE];
down_write(&start->hash_mutex);
mref_a->is_stable = true;
up_write(&start->hash_mutex);
}
}
static
void _inf_callback(struct trans_logger_input *input, bool force)
{
if (!force &&
input->inf_last_jiffies &&
input->inf_last_jiffies + trans_logger_report_interval * HZ > (long long)jiffies)
return;
if (input->inf.inf_callback && input->is_operating) {
input->inf_last_jiffies = jiffies;
input->inf.inf_callback(&input->inf);
input->inf_last_jiffies = jiffies;
if (input->inf_min_old != input->inf.inf_min_pos) {
input->inf_min_jiffies = input->inf_last_jiffies;
input->inf_min_old = input->inf.inf_min_pos;
}
} else {
MARS_DBG("%p skipped callback, callback = %p is_operating = %d\n", input, input->inf.inf_callback, input->is_operating);
}
}
static inline
int _congested(struct trans_logger_brick *brick)
{
return
brick->q_phase[0].q_active ||
brick->q_phase[1].q_active ||
brick->q_phase[2].q_active ||
brick->q_phase[3].q_active;
}
////////////////// own brick / input / output operations //////////////////
atomic_t global_mshadow_count = ATOMIC_INIT(0);
EXPORT_SYMBOL_GPL(global_mshadow_count);
atomic64_t global_mshadow_used = ATOMIC64_INIT(0);
EXPORT_SYMBOL_GPL(global_mshadow_used);
static noinline
int trans_logger_get_info(struct trans_logger_output *output, struct mars_info *info)
{
struct trans_logger_input *input = output->brick->inputs[TL_INPUT_READ];
return GENERIC_INPUT_CALL(input, mars_get_info, info);
}
static noinline
int _make_sshadow(struct trans_logger_output *output, struct trans_logger_mref_aspect *mref_a, struct trans_logger_mref_aspect *mshadow_a)
{
struct trans_logger_brick *brick = output->brick;
struct mref_object *mref = mref_a->object;
struct mref_object *mshadow;
int diff;
mshadow = mshadow_a->object;
#if 1
if (unlikely(mref->ref_len > mshadow->ref_len)) {
MARS_ERR("oops %d -> %d\n", mref->ref_len, mshadow->ref_len);
mref->ref_len = mshadow->ref_len;
}
if (unlikely(mshadow_a == mref_a)) {
MARS_ERR("oops %p == %p\n", mshadow_a, mref_a);
return -EINVAL;
}
#endif
diff = mref->ref_pos - mshadow->ref_pos;
#if 1
if (unlikely(diff < 0)) {
MARS_ERR("oops diff = %d\n", diff);
return -EINVAL;
}
#endif
/* Attach mref to the existing shadow ("slave shadow").
*/
mref_a->shadow_data = mshadow_a->shadow_data + diff;
mref_a->do_dealloc = false;
if (!mref->ref_data) { // buffered IO
mref->ref_data = mref_a->shadow_data;
mref_a->do_buffered = true;
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_sshadow_buffered_count);
#endif
}
mref_a->shadow_ref = mshadow_a;
mref_a->my_brick = brick;
/* Get an ordinary internal reference
*/
_mref_get_first(mref); // must be paired with __trans_logger_ref_put()
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->inner_balance_count);
/* The internal reference from slave to master is already
* present due to hash_find(),
* such that the master cannot go away before the slave.
* It is compensated by master transition in __trans_logger_ref_put()
*/
atomic_inc(&brick->inner_balance_count);
atomic_inc(&brick->sshadow_count);
atomic_inc(&brick->total_sshadow_count);
#endif
#if 1
if (unlikely(mref->ref_len <= 0)) {
MARS_ERR("oops, len = %d\n", mref->ref_len);
return -EINVAL;
}
#endif
return mref->ref_len;
}
static noinline
int _read_ref_get(struct trans_logger_output *output, struct trans_logger_mref_aspect *mref_a)
{
struct trans_logger_brick *brick = output->brick;
struct mref_object *mref = mref_a->object;
struct trans_logger_input *input = brick->inputs[TL_INPUT_READ];
struct trans_logger_mref_aspect *mshadow_a;
/* Look if there is a newer version on the fly, shadowing
* the old one.
* When a shadow is found, use it as buffer for the mref.
*/
mshadow_a = hash_find(brick, mref->ref_pos, &mref->ref_len, false);
if (!mshadow_a) {
return GENERIC_INPUT_CALL(input, mref_get, mref);
}
return _make_sshadow(output, mref_a, mshadow_a);
}
static noinline
int _write_ref_get(struct trans_logger_output *output, struct trans_logger_mref_aspect *mref_a)
{
struct trans_logger_brick *brick = output->brick;
struct mref_object *mref = mref_a->object;
void *data;
#ifdef KEEP_UNIQUE
struct trans_logger_mref_aspect *mshadow_a;
#endif
#ifdef CONFIG_MARS_DEBUG
if (unlikely(mref->ref_len <= 0)) {
MARS_ERR("oops, ref_len = %d\n", mref->ref_len);
return -EINVAL;
}
#endif
#ifdef KEEP_UNIQUE
mshadow_a = hash_find(brick, mref->ref_pos, &mref->ref_len, true);
if (mshadow_a) {
return _make_sshadow(output, mref_a, mshadow_a);
}
#endif
#ifdef DELAY_CALLERS
// delay in case of too many master shadows / memory shortage
brick_wait(brick->caller_event,
brick->caller_flag,
brick_global_memlimit < 1024 ||
atomic64_read(&global_mshadow_used) / 1024 < brick_global_memlimit,
HZ / 4);
#endif
// create a new master shadow
data = brick_block_alloc(mref->ref_pos, (mref_a->alloc_len = mref->ref_len));
if (unlikely(!data)) {
return -ENOMEM;
}
atomic64_add(mref->ref_len, &brick->shadow_mem_used);
#ifdef CONFIG_MARS_DEBUG
memset(data, 0x11, mref->ref_len);
#endif
mref_a->shadow_data = data;
mref_a->do_dealloc = true;
if (!mref->ref_data) { // buffered IO
mref->ref_data = data;
mref_a->do_buffered = true;
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_mshadow_buffered_count);
#endif
}
mref_a->my_brick = brick;
mref_a->shadow_ref = mref_a; // cyclic self-reference => indicates master shadow
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->mshadow_count);
atomic_inc(&brick->total_mshadow_count);
#endif
atomic_inc(&global_mshadow_count);
atomic64_add(mref->ref_len, &global_mshadow_used);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->inner_balance_count);
#endif
_mref_get_first(mref); // must be paired with __trans_logger_ref_put()
return mref->ref_len;
}
static noinline
int trans_logger_ref_get(struct trans_logger_output *output, struct mref_object *mref)
{
struct trans_logger_brick *brick;
struct trans_logger_mref_aspect *mref_a;
loff_t base_offset;
CHECK_PTR(output, err);
brick = output->brick;
CHECK_PTR(brick, err);
CHECK_PTR(mref, err);
MARS_IO("pos = %lld len = %d\n", mref->ref_pos, mref->ref_len);
mref_a = trans_logger_mref_get_aspect(brick, mref);
CHECK_PTR(mref_a, err);
CHECK_ASPECT(mref_a, mref, err);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->outer_balance_count);
#endif
if (mref->ref_initialized) { // setup already performed
MARS_IO("again %d\n", atomic_read(&mref->ref_count.ta_atomic));
_mref_check(mref);
_mref_get(mref); // must be paired with __trans_logger_ref_put()
return mref->ref_len;
}
get_lamport(NULL, &mref_a->stamp);
if (mref->ref_len > CONF_TRANS_MAX_MREF_SIZE && CONF_TRANS_MAX_MREF_SIZE > 0)
mref->ref_len = CONF_TRANS_MAX_MREF_SIZE;
// ensure that REGION_SIZE boundaries are obeyed by hashing
base_offset = mref->ref_pos & (loff_t)(REGION_SIZE - 1);
if (mref->ref_len > REGION_SIZE - base_offset) {
mref->ref_len = REGION_SIZE - base_offset;
}
if (!(mref->ref_flags & MREF_MAY_WRITE)) {
return _read_ref_get(output, mref_a);
}
if (unlikely(brick->stopped_logging)) { // only in EMERGENCY mode
mref_a->is_emergency = true;
/* Wait until writeback has finished.
* We have to this because writeback is out-of-order.
* Otherwise consistency could be violated for some time.
*/
while (_congested(brick)) {
// in case of emergency, busy-wait should be acceptable
brick_msleep(HZ / 10);
}
return _read_ref_get(output, mref_a);
}
/* FIXME: THIS IS PROVISIONARY (use event instead)
*/
while (unlikely(!brick->power.led_on)) {
brick_msleep(HZ / 10);
}
return _write_ref_get(output, mref_a);
err:
return -EINVAL;
}
static noinline
void pos_complete(struct trans_logger_mref_aspect *orig_mref_a);
static noinline
void __trans_logger_ref_put(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *mref_a)
{
struct mref_object *mref;
struct trans_logger_mref_aspect *shadow_a;
struct trans_logger_input *input;
restart:
CHECK_PTR(mref_a, err);
mref = mref_a->object;
CHECK_PTR(mref, err);
MARS_IO("pos = %lld len = %d\n", mref->ref_pos, mref->ref_len);
_mref_check(mref);
// are we a shadow (whether master or slave)?
shadow_a = mref_a->shadow_ref;
if (shadow_a) {
bool finished;
CHECK_PTR(shadow_a, err);
CHECK_PTR(shadow_a->object, err);
_mref_check(shadow_a->object);
finished = _mref_put(mref);
#ifdef ADDITIONAL_COUNTERS
atomic_dec(&brick->inner_balance_count);
#endif
if (unlikely(finished && mref_a->is_hashed)) {
MARS_ERR("trying to put a hashed mref, pos = %lld len = %d\n", mref->ref_pos, mref->ref_len);
finished = false; // leaves a memleak
}
if (!finished) {
return;
}
CHECK_HEAD_EMPTY(&mref_a->lh.lh_head);
CHECK_HEAD_EMPTY(&mref_a->hash_head);
CHECK_HEAD_EMPTY(&mref_a->replay_head);
CHECK_HEAD_EMPTY(&mref_a->collect_head);
CHECK_HEAD_EMPTY(&mref_a->sub_list);
CHECK_HEAD_EMPTY(&mref_a->sub_head);
if (mref_a->is_collected && likely(mref_a->wb_error >= 0)) {
pos_complete(mref_a);
}
CHECK_HEAD_EMPTY(&mref_a->pos_head);
if (shadow_a != mref_a) { // we are a slave shadow
//MARS_DBG("slave\n");
#ifdef ADDITIONAL_COUNTERS
atomic_dec(&brick->sshadow_count);
#endif
CHECK_HEAD_EMPTY(&mref_a->hash_head);
trans_logger_free_mref(mref);
// now put the master shadow
mref_a = shadow_a;
goto restart;
}
// we are a master shadow
CHECK_PTR(mref_a->shadow_data, err);
if (mref_a->do_dealloc) {
brick_block_free(mref_a->shadow_data, mref_a->alloc_len);
atomic64_sub(mref_a->alloc_len, &brick->shadow_mem_used);
mref_a->do_dealloc = false;
}
if (mref_a->do_buffered) {
mref->ref_data = NULL;
}
#ifdef ADDITIONAL_COUNTERS
atomic_dec(&brick->mshadow_count);
#endif
atomic_dec(&global_mshadow_count);
atomic64_sub(mref->ref_len, &global_mshadow_used);
trans_logger_free_mref(mref);
return;
}
// only READ is allowed on non-shadow buffers
if (unlikely((mref->ref_flags & MREF_WRITE) && !mref_a->is_emergency)) {
MARS_FAT("bad operation %ux on non-shadow\n",
mref->ref_flags);
}
// no shadow => call through
input = brick->inputs[TL_INPUT_READ];
CHECK_PTR(input, err);
GENERIC_INPUT_CALL(input, mref_put, mref);
err: ;
}
static noinline
void _trans_logger_ref_put(struct trans_logger_output *output, struct mref_object *mref)
{
struct trans_logger_mref_aspect *mref_a;
mref_a = trans_logger_mref_get_aspect(output->brick, mref);
CHECK_PTR(mref_a, err);
CHECK_ASPECT(mref_a, mref, err);
__trans_logger_ref_put(output->brick, mref_a);
return;
err:
MARS_FAT("giving up...\n");
}
static noinline
void trans_logger_ref_put(struct trans_logger_output *output, struct mref_object *mref)
{
#ifdef ADDITIONAL_COUNTERS
struct trans_logger_brick *brick = output->brick;
atomic_dec(&brick->outer_balance_count);
#endif
_trans_logger_ref_put(output, mref);
}
static noinline
void _trans_logger_endio(struct generic_callback *cb)
{
struct trans_logger_mref_aspect *mref_a;
struct trans_logger_brick *brick;
_crashme(20, false);
mref_a = cb->cb_private;
CHECK_PTR(mref_a, err);
if (unlikely(&mref_a->cb != cb)) {
MARS_FAT("bad callback -- hanging up\n");
goto err;
}
brick = mref_a->my_brick;
CHECK_PTR(brick, err);
NEXT_CHECKED_CALLBACK(cb, err);
atomic_dec(&brick->any_fly_count);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_cb_count);
#endif
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_FAT("cannot handle callback\n");
}
static noinline
void trans_logger_ref_io(struct trans_logger_output *output, struct mref_object *mref)
{
struct trans_logger_brick *brick = output->brick;
struct trans_logger_mref_aspect *mref_a;
struct trans_logger_mref_aspect *shadow_a;
struct trans_logger_input *input;
_mref_check(mref);
mref_a = trans_logger_mref_get_aspect(brick, mref);
CHECK_PTR(mref_a, err);
CHECK_ASPECT(mref_a, mref, err);
MARS_IO("pos = %lld len = %d\n", mref->ref_pos, mref->ref_len);
#ifdef ADDITIONAL_COUNTERS
// statistics
if (mref->ref_flags & MREF_WRITE) {
atomic_inc(&brick->total_write_count);
} else {
atomic_inc(&brick->total_read_count);
}
#endif
// is this a shadow buffer?
shadow_a = mref_a->shadow_ref;
if (shadow_a) {
#if 1
CHECK_HEAD_EMPTY(&mref_a->lh.lh_head);
CHECK_HEAD_EMPTY(&mref_a->hash_head);
CHECK_HEAD_EMPTY(&mref_a->pos_head);
#endif
_mref_get(mref); // must be paired with __trans_logger_ref_put()
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->inner_balance_count);
#endif
qq_mref_insert(&brick->q_phase[0], mref_a);
brick_wake(&brick->worker_event, brick->worker_flag);
return;
}
// only READ is allowed on non-shadow buffers
if (unlikely((mref->ref_flags & MREF_WRITE) && !mref_a->is_emergency)) {
MARS_FAT("bad operation %ux on non-shadow\n",
mref->ref_flags);
}
atomic_inc(&brick->any_fly_count);
mref_a->my_brick = brick;
INSERT_CALLBACK(mref, &mref_a->cb, _trans_logger_endio, mref_a);
input = output->brick->inputs[TL_INPUT_READ];
GENERIC_INPUT_CALL(input, mref_io, mref);
return;
err:
MARS_FAT("cannot handle IO\n");
}
////////////////////////////// writeback info //////////////////////////////
/* save final completion status when necessary
*/
static noinline
void pos_complete(struct trans_logger_mref_aspect *orig_mref_a)
{
struct trans_logger_brick *brick = orig_mref_a->my_brick;
struct trans_logger_input *log_input = orig_mref_a->log_input;
loff_t finished;
struct list_head *tmp;
CHECK_PTR(brick, err);
CHECK_PTR(log_input, err);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_writeback_count);
#endif
tmp = &orig_mref_a->pos_head;
down(&log_input->inf_mutex);
finished = orig_mref_a->log_pos;
// am I the first member? (means "youngest" list entry)
if (tmp == log_input->pos_list.next) {
MARS_IO("first_finished = %lld\n", finished);
if (unlikely(finished <= log_input->inf.inf_min_pos)) {
MARS_ERR("backskip in log writeback: %lld -> %lld\n", log_input->inf.inf_min_pos, finished);
}
if (unlikely(finished > log_input->inf.inf_max_pos)) {
MARS_ERR("min_pos > max_pos: %lld > %lld\n", finished, log_input->inf.inf_max_pos);
}
log_input->inf.inf_min_pos = finished;
get_lamport(NULL, &log_input->inf.inf_min_pos_stamp);
_inf_callback(log_input, false);
} else {
struct trans_logger_mref_aspect *prev_mref_a;
prev_mref_a = container_of(tmp->prev, struct trans_logger_mref_aspect, pos_head);
if (unlikely(finished <= prev_mref_a->log_pos)) {
MARS_ERR("backskip: %lld -> %lld\n", finished, prev_mref_a->log_pos);
} else {
/* Transitively transfer log_pos to the predecessor
* to correctly reflect the committed region.
*/
prev_mref_a->log_pos = finished;
}
}
list_del_init(tmp);
atomic_dec(&log_input->pos_count);
up(&log_input->inf_mutex);
err:;
}
static noinline
void free_writeback(struct writeback_info *wb)
{
struct list_head *tmp;
if (unlikely(wb->w_error < 0)) {
MARS_ERR("writeback error = %d at pos = %lld len = %d, writeback is incomplete\n", wb->w_error, wb->w_pos, wb->w_len);
}
/* Now complete the original requests.
*/
while ((tmp = wb->w_collect_list.next) != &wb->w_collect_list) {
struct trans_logger_mref_aspect *orig_mref_a;
struct mref_object *orig_mref;
list_del_init(tmp);
orig_mref_a = container_of(tmp, struct trans_logger_mref_aspect, collect_head);
orig_mref = orig_mref_a->object;
_mref_check(orig_mref);
if (unlikely(!orig_mref_a->is_collected)) {
MARS_ERR("request %lld (len = %d) was not collected\n", orig_mref->ref_pos, orig_mref->ref_len);
}
if (unlikely(wb->w_error < 0)) {
orig_mref_a->wb_error = wb->w_error;
}
__trans_logger_ref_put(orig_mref_a->my_brick, orig_mref_a);
}
brick_mem_free(wb);
}
/* Generic endio() for writeback_info
*/
static noinline
void wb_endio(struct generic_callback *cb)
{
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
struct trans_logger_brick *brick;
struct writeback_info *wb;
atomic_t *dec;
void (**_endio)(struct generic_callback *cb);
void (*endio)(struct generic_callback *cb);
_crashme(21, false);
LAST_CALLBACK(cb);
sub_mref_a = cb->cb_private;
CHECK_PTR(sub_mref_a, err);
sub_mref = sub_mref_a->object;
CHECK_PTR(sub_mref, err);
wb = sub_mref_a->wb;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
if (cb->cb_error < 0) {
wb->w_error = cb->cb_error;
}
#ifdef ADDITIONAL_COUNTERS
atomic_dec(&brick->wb_balance_count);
#endif
if (sub_mref_a->orig_flags & MREF_WRITE) {
dec = &wb->w_sub_write_count;
_endio = &wb->write_endio;
} else {
dec = &wb->w_sub_read_count;
_endio = &wb->read_endio;
}
CHECK_ATOMIC(dec, 1);
if (!atomic_dec_and_test(dec)) {
goto done;
}
endio = *_endio;
*_endio = NULL;
if (likely(endio)) {
endio(cb);
} else {
MARS_ERR("internal: no endio defined\n");
}
done:
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_FAT("hanging up....\n");
}
/* Atomically create writeback info, based on "snapshot" of current hash
* state.
* Notice that the hash can change during writeback IO, thus we need
* struct writeback_info to precisely catch that information at a single
* point in time.
*/
static noinline
struct writeback_info *make_writeback(struct trans_logger_brick *brick, loff_t pos, int len)
{
struct writeback_info *wb;
struct trans_logger_input *read_input;
struct trans_logger_input *write_input;
int write_input_nr;
/* Allocate structure representing a bunch of adjacent writebacks
*/
wb = brick_zmem_alloc(sizeof(struct writeback_info));
if (!wb) {
goto err;
}
if (unlikely(len < 0)) {
MARS_ERR("len = %d\n", len);
}
wb->w_brick = brick;
wb->w_pos = pos;
wb->w_len = len;
wb->w_lh.lh_pos = &wb->w_pos;
INIT_LIST_HEAD(&wb->w_lh.lh_head);
INIT_LIST_HEAD(&wb->w_collect_list);
INIT_LIST_HEAD(&wb->w_sub_read_list);
INIT_LIST_HEAD(&wb->w_sub_write_list);
/* Atomically fetch transitive closure on all requests
* overlapping with the current search region.
*/
hash_extend(brick, &wb->w_pos, &wb->w_len, &wb->w_collect_list);
if (list_empty(&wb->w_collect_list)) {
goto collision;
}
pos = wb->w_pos;
len = wb->w_len;
if (unlikely(len < 0)) {
MARS_ERR("len = %d\n", len);
}
/* Determine the "channels" we want to operate on
*/
read_input = brick->inputs[TL_INPUT_READ];
write_input_nr = TL_INPUT_WRITEBACK;
write_input = brick->inputs[write_input_nr];
if (!write_input->connect) {
write_input_nr = TL_INPUT_READ;
write_input = read_input;
}
/* Create sub_mrefs for read of old disk version (phase1)
*/
if (brick->log_reads) {
while (len > 0) {
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
struct trans_logger_input *log_input;
int this_len;
int status;
sub_mref = trans_logger_alloc_mref(brick);
if (unlikely(!sub_mref)) {
MARS_FAT("cannot alloc sub_mref\n");
goto err;
}
sub_mref->ref_pos = pos;
sub_mref->ref_len = len;
sub_mref->ref_flags = 0;
sub_mref->ref_data = NULL;
sub_mref_a = trans_logger_mref_get_aspect(brick, sub_mref);
CHECK_PTR(sub_mref_a, err);
CHECK_ASPECT(sub_mref_a, sub_mref, err);
sub_mref_a->my_input = read_input;
log_input = brick->inputs[brick->log_input_nr];
sub_mref_a->log_input = log_input;
atomic_inc(&log_input->log_ref_count);
sub_mref_a->my_brick = brick;
sub_mref_a->orig_flags = 0;
sub_mref_a->wb = wb;
status = GENERIC_INPUT_CALL(read_input, mref_get, sub_mref);
if (unlikely(status < 0)) {
MARS_FAT("cannot get sub_ref, status = %d\n", status);
goto err;
}
list_add_tail(&sub_mref_a->sub_head, &wb->w_sub_read_list);
atomic_inc(&wb->w_sub_read_count);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->wb_balance_count);
#endif
this_len = sub_mref->ref_len;
pos += this_len;
len -= this_len;
}
/* Re-init for startover
*/
pos = wb->w_pos;
len = wb->w_len;
}
/* Always create sub_mrefs for writeback (phase3)
*/
while (len > 0) {
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
struct trans_logger_mref_aspect *orig_mref_a;
struct mref_object *orig_mref;
struct trans_logger_input *log_input;
void *data;
int this_len = len;
int diff;
int status;
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_hash_find_count);
#endif
orig_mref_a = _hash_find(&wb->w_collect_list, pos, &this_len, true, false);
if (unlikely(!orig_mref_a)) {
MARS_FAT("could not find data\n");
goto err;
}
orig_mref = orig_mref_a->object;
diff = pos - orig_mref->ref_pos;
if (unlikely(diff < 0)) {
MARS_FAT("bad diff %d\n", diff);
goto err;
}
data = orig_mref_a->shadow_data + diff;
sub_mref = trans_logger_alloc_mref(brick);
if (unlikely(!sub_mref)) {
MARS_FAT("cannot alloc sub_mref\n");
goto err;
}
sub_mref->ref_pos = pos;
sub_mref->ref_len = this_len;
sub_mref->ref_flags = MREF_WRITE | MREF_MAY_WRITE;
sub_mref->ref_data = data;
sub_mref_a = trans_logger_mref_get_aspect(brick, sub_mref);
CHECK_PTR(sub_mref_a, err);
CHECK_ASPECT(sub_mref_a, sub_mref, err);
sub_mref_a->orig_mref_a = orig_mref_a;
sub_mref_a->my_input = write_input;
log_input = orig_mref_a->log_input;
sub_mref_a->log_input = log_input;
atomic_inc(&log_input->log_ref_count);
sub_mref_a->my_brick = brick;
sub_mref_a->orig_flags = MREF_WRITE;
sub_mref_a->wb = wb;
status = GENERIC_INPUT_CALL(write_input, mref_get, sub_mref);
if (unlikely(status < 0)) {
MARS_FAT("cannot get sub_ref, status = %d\n", status);
wb->w_error = status;
goto err;
}
list_add_tail(&sub_mref_a->sub_head, &wb->w_sub_write_list);
atomic_inc(&wb->w_sub_write_count);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->wb_balance_count);
#endif
this_len = sub_mref->ref_len;
pos += this_len;
len -= this_len;
}
return wb;
err:
MARS_ERR("cleaning up...\n");
collision:
if (wb) {
free_writeback(wb);
}
return NULL;
}
static inline
void _fire_one(struct list_head *tmp, bool do_update)
{
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
struct trans_logger_input *sub_input;
sub_mref_a = container_of(tmp, struct trans_logger_mref_aspect, sub_head);
sub_mref = sub_mref_a->object;
if (unlikely(sub_mref_a->is_fired)) {
MARS_ERR("trying to fire twice\n");
return;
}
sub_mref_a->is_fired = true;
SETUP_CALLBACK(sub_mref, wb_endio, sub_mref_a);
sub_input = sub_mref_a->my_input;
#ifdef DO_WRITEBACK
GENERIC_INPUT_CALL(sub_input, mref_io, sub_mref);
#else
SIMPLE_CALLBACK(sub_mref, 0);
#endif
if (do_update) { // CHECK: shouldn't we do this always?
GENERIC_INPUT_CALL(sub_input, mref_put, sub_mref);
}
}
static inline
void fire_writeback(struct list_head *start, bool do_update)
{
struct list_head *tmp;
/* Caution! The wb structure may get deallocated
* during _fire_one() in some cases (e.g. when the
* callback is directly called by the mref_io operation).
* Ensure that no ptr dereferencing can take
* place after working on the last list member.
*/
tmp = start->next;
while (tmp != start) {
struct list_head *next = tmp->next;
list_del_init(tmp);
_fire_one(tmp, do_update);
tmp = next;
}
}
static inline
void update_max_pos(struct trans_logger_mref_aspect *orig_mref_a)
{
loff_t max_pos = orig_mref_a->log_pos;
struct trans_logger_input *log_input = orig_mref_a->log_input;
CHECK_PTR(log_input, done);
down(&log_input->inf_mutex);
if (unlikely(max_pos < log_input->inf.inf_min_pos)) {
MARS_ERR("new max_pos < min_pos: %lld < %lld\n", max_pos, log_input->inf.inf_min_pos);
}
if (log_input->inf.inf_max_pos < max_pos) {
log_input->inf.inf_max_pos = max_pos;
get_lamport(NULL, &log_input->inf.inf_max_pos_stamp);
_inf_callback(log_input, false);
}
up(&log_input->inf_mutex);
done:;
}
static inline
void update_writeback_info(struct writeback_info * wb)
{
struct list_head *start = &wb->w_collect_list;
struct list_head *tmp;
/* Notice: in case of log rotation, each list member
* may belong to a different log_input.
*/
for (tmp = start->next; tmp != start; tmp = tmp->next) {
struct trans_logger_mref_aspect *orig_mref_a;
orig_mref_a = container_of(tmp, struct trans_logger_mref_aspect, collect_head);
update_max_pos(orig_mref_a);
}
}
////////////////////////////// worker thread //////////////////////////////
/*********************************************************************
* Phase 0: write transaction log entry for the original write request.
*/
static noinline
void _complete(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *orig_mref_a, int error, bool pre_io)
{
struct mref_object *orig_mref;
orig_mref = orig_mref_a->object;
CHECK_PTR(orig_mref, err);
if (orig_mref_a->is_completed ||
(pre_io &&
(trans_logger_completion_semantics >= 2 ||
(trans_logger_completion_semantics >= 1 &&
!(orig_mref->ref_flags & MREF_SKIP_SYNC))))) {
goto done;
}
if (cmpxchg(&orig_mref_a->is_completed, false, true))
goto done;
atomic_dec(&brick->log_fly_count);
if (likely(error >= 0)) {
mref_checksum(orig_mref);
orig_mref->ref_flags &= ~MREF_WRITING;
orig_mref->ref_flags |= MREF_UPTODATE;
}
CHECKED_CALLBACK(orig_mref, error, err);
update_max_pos(orig_mref_a);
done:
return;
err:
MARS_ERR("giving up...\n");
}
static noinline
void phase0_preio(void *private)
{
struct trans_logger_mref_aspect *orig_mref_a;
struct trans_logger_brick *brick;
orig_mref_a = private;
CHECK_PTR(orig_mref_a, err);
CHECK_PTR(orig_mref_a->object, err);
brick = orig_mref_a->my_brick;
CHECK_PTR(brick, err);
// signal completion to the upper layer
// FIXME: immediate error signalling is impossible here, but some delayed signalling should be possible as a workaround. Think!
_mref_check(orig_mref_a->object);
_complete(brick, orig_mref_a, 0, true);
_mref_check(orig_mref_a->object);
return;
err:
MARS_ERR("giving up...\n");
}
static noinline
void phase0_endio(void *private, int error)
{
struct mref_object *orig_mref;
struct trans_logger_mref_aspect *orig_mref_a;
struct trans_logger_brick *brick;
orig_mref_a = private;
CHECK_PTR(orig_mref_a, err);
// remove_this
if (unlikely(cmpxchg(&orig_mref_a->is_endio, false, true))) {
MARS_ERR("Sigh this should not happen %p %p\n",
orig_mref_a, orig_mref_a->object);
return;
}
// end_remove_this
brick = orig_mref_a->my_brick;
CHECK_PTR(brick, err);
orig_mref = orig_mref_a->object;
CHECK_PTR(orig_mref, err);
orig_mref_a->is_persistent = true;
_CHECK(orig_mref_a->shadow_ref, err);
// signal completion to the upper layer
_complete(brick, orig_mref_a, error, false);
/* Queue up for the next phase.
*/
qq_mref_insert(&brick->q_phase[1], orig_mref_a);
/* Undo the above pinning
*/
__trans_logger_ref_put(brick, orig_mref_a);
banning_reset(&brick->q_phase[0].q_banning);
qq_deactivate(&brick->q_phase[0]);
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_ERR("giving up...\n");
}
static noinline
bool phase0_startio(struct trans_logger_mref_aspect *orig_mref_a)
{
struct mref_object *orig_mref;
struct trans_logger_brick *brick;
struct trans_logger_input *input;
struct log_status *logst;
loff_t log_pos;
void *data;
bool ok;
CHECK_PTR(orig_mref_a, err);
orig_mref = orig_mref_a->object;
CHECK_PTR(orig_mref, err);
brick = orig_mref_a->my_brick;
CHECK_PTR(brick, err);
input = orig_mref_a->log_input;
CHECK_PTR(input, err);
logst = &input->logst;
logst->do_crc = trans_logger_do_crc;
logst->do_compress =
trans_logger_allow_compress &&
(usable_compression_mask & MREF_COMPRESS_ANY);
{
struct log_header l = {
.l_stamp = orig_mref_a->stamp,
.l_pos = orig_mref->ref_pos,
.l_len = orig_mref->ref_len,
.l_code = CODE_WRITE_NEW,
};
data = log_reserve(logst, &l);
}
if (unlikely(!data)) {
goto err;
}
hash_ensure_stableness(brick, orig_mref_a);
memcpy(data, orig_mref_a->shadow_data, orig_mref->ref_len);
/* Pin mref->ref_count so it can't go away
* after _complete().
* This may happen rather early in phase0_preio().
*/
_mref_get(orig_mref); // must be paired with __trans_logger_ref_put()
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->inner_balance_count);
#endif
atomic_inc(&brick->log_fly_count);
ok = log_finalize(logst, orig_mref->ref_len, phase0_endio, orig_mref_a);
if (unlikely(!ok)) {
atomic_dec(&brick->log_fly_count);
goto err;
}
log_pos = logst->log_pos + logst->offset;
orig_mref_a->log_pos = log_pos;
// update new log_pos in the symlinks
down(&input->inf_mutex);
input->inf.inf_log_pos = log_pos;
memcpy(&input->inf.inf_log_pos_stamp, &logst->log_pos_stamp, sizeof(input->inf.inf_log_pos_stamp));
_inf_callback(input, false);
#ifdef CONFIG_MARS_DEBUG
if (!list_empty(&input->pos_list)) {
struct trans_logger_mref_aspect *last_mref_a;
last_mref_a = container_of(input->pos_list.prev, struct trans_logger_mref_aspect, pos_head);
if (last_mref_a->log_pos >= orig_mref_a->log_pos) {
MARS_ERR("backskip in pos_list, %lld >= %lld\n", last_mref_a->log_pos, orig_mref_a->log_pos);
}
}
#endif
list_add_tail(&orig_mref_a->pos_head, &input->pos_list);
atomic_inc(&input->pos_count);
up(&input->inf_mutex);
phase0_preio(orig_mref_a);
return true;
err:
return false;
}
static noinline
bool prep_phase_startio(struct trans_logger_mref_aspect *mref_a)
{
struct mref_object *mref = mref_a->object;
struct trans_logger_mref_aspect *shadow_a;
struct trans_logger_brick *brick;
CHECK_PTR(mref, err);
shadow_a = mref_a->shadow_ref;
CHECK_PTR(shadow_a, err);
brick = mref_a->my_brick;
CHECK_PTR(brick, err);
MARS_IO("pos = %lld len = %d flags = %ux\n",
mref->ref_pos, mref->ref_len,
mref->ref_flags);
if (!(mref->ref_flags & MREF_WRITE)) {
// nothing to do: directly signal success.
struct mref_object *shadow = shadow_a->object;
if (unlikely(shadow == mref)) {
MARS_ERR("oops, we should be a slave shadow, but are a master one\n");
}
#ifdef USE_MEMCPY
if (mref_a->shadow_data != mref->ref_data) {
if (unlikely(mref->ref_len <= 0 || mref->ref_len > PAGE_SIZE)) {
MARS_ERR("implausible ref_len = %d\n", mref->ref_len);
}
MARS_IO("read memcpy to = %p from = %p len = %d\n", mref->ref_data, mref_a->shadow_data, mref->ref_len);
memcpy(mref->ref_data, mref_a->shadow_data, mref->ref_len);
}
#endif
mref->ref_flags |= MREF_UPTODATE;
CHECKED_CALLBACK(mref, 0, err);
__trans_logger_ref_put(brick, mref_a);
qq_deactivate(&brick->q_phase[0]);
return true;
}
// else WRITE
#if 1
CHECK_HEAD_EMPTY(&mref_a->lh.lh_head);
CHECK_HEAD_EMPTY(&mref_a->hash_head);
if (unlikely(mref->ref_flags & (MREF_READING | MREF_WRITING))) {
MARS_ERR("bad flags %d\n", mref->ref_flags);
}
#endif
/* In case of non-buffered IO, the buffer is
* under control of the user. In particular, he
* may change it without telling us.
* Therefore we make a copy (or "snapshot") here.
*/
mref->ref_flags |= MREF_WRITING;
#ifdef USE_MEMCPY
if (mref_a->shadow_data != mref->ref_data) {
if (unlikely(mref->ref_len <= 0 || mref->ref_len > PAGE_SIZE)) {
MARS_ERR("implausible ref_len = %d\n", mref->ref_len);
}
MARS_IO("write memcpy to = %p from = %p len = %d\n", mref_a->shadow_data, mref->ref_data, mref->ref_len);
memcpy(mref_a->shadow_data, mref->ref_data, mref->ref_len);
}
#endif
mref_a->is_dirty = true;
mref_a->shadow_ref->is_dirty = true;
#ifndef KEEP_UNIQUE
if (unlikely(mref_a->shadow_ref != mref_a)) {
MARS_ERR("something is wrong: %p != %p\n", mref_a->shadow_ref, mref_a);
}
#endif
if (likely(!mref_a->is_hashed)) {
struct trans_logger_input *log_input;
log_input = brick->inputs[brick->log_input_nr];
MARS_IO("hashing %d at %lld\n", mref->ref_len, mref->ref_pos);
mref_a->log_input = log_input;
atomic_inc(&log_input->log_ref_count);
hash_insert(brick, mref_a);
} else {
MARS_ERR("tried to hash twice\n");
}
return phase0_startio(mref_a);
err:
MARS_ERR("cannot work\n");
brick_msleep(1000);
return false;
}
/*********************************************************************
* Phase 1: read original version of data.
* This happens _after_ phase 0, deliberately.
* We are explicitly dealing with old and new versions.
* The new version is hashed in memory all the time (such that parallel
* READs will see them), so we have plenty of time for getting the
* old version from disk somewhen later, e.g. when IO contention is low.
*/
static noinline
void phase1_endio(struct generic_callback *cb)
{
struct trans_logger_mref_aspect *sub_mref_a;
struct writeback_info *wb;
struct trans_logger_brick *brick;
CHECK_PTR(cb, err);
sub_mref_a = cb->cb_private;
CHECK_PTR(sub_mref_a, err);
wb = sub_mref_a->wb;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
if (unlikely(cb->cb_error < 0)) {
MARS_FAT("IO error %d\n", cb->cb_error);
goto err;
}
banning_reset(&brick->q_phase[1].q_banning);
// queue up for the next phase
qq_wb_insert(&brick->q_phase[2], wb);
qq_deactivate(&brick->q_phase[1]);
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_FAT("hanging up....\n");
}
static noinline
void phase3_endio(struct generic_callback *cb);
static noinline
bool phase3_startio(struct writeback_info *wb);
static noinline
bool phase1_startio(struct trans_logger_mref_aspect *orig_mref_a)
{
struct mref_object *orig_mref;
struct trans_logger_brick *brick;
struct writeback_info *wb = NULL;
CHECK_PTR(orig_mref_a, err);
orig_mref = orig_mref_a->object;
CHECK_PTR(orig_mref, err);
brick = orig_mref_a->my_brick;
CHECK_PTR(brick, err);
if (orig_mref_a->is_collected) {
MARS_IO("already collected, pos = %lld len = %d\n", orig_mref->ref_pos, orig_mref->ref_len);
qq_deactivate(&brick->q_phase[1]);
goto done;
}
if (!orig_mref_a->is_hashed) {
MARS_IO("AHA not hashed, pos = %lld len = %d\n", orig_mref->ref_pos, orig_mref->ref_len);
qq_deactivate(&brick->q_phase[1]);
goto done;
}
wb = make_writeback(brick, orig_mref->ref_pos, orig_mref->ref_len);
if (unlikely(!wb)) {
goto collision;
}
if (unlikely(list_empty(&wb->w_sub_write_list))) {
MARS_ERR("sub_write_list is empty, orig pos = %lld len = %d (collected=%d), extended pos = %lld len = %d\n", orig_mref->ref_pos, orig_mref->ref_len, (int)orig_mref_a->is_collected, wb->w_pos, wb->w_len);
goto err;
}
wb->read_endio = phase1_endio;
wb->write_endio = phase3_endio;
atomic_set(&wb->w_sub_log_count, atomic_read(&wb->w_sub_read_count));
if (brick->log_reads) {
fire_writeback(&wb->w_sub_read_list, false);
} else { // shortcut
#ifdef SHORTCUT_1_to_3
bool res;
if (trans_logger_disable_pressure < -1 ||
trans_logger_disable_pressure > 1)
goto no_speculation;
/* speculate that next phase can be immediately started */
qq_activate(&brick->q_phase[3]);
res = phase3_startio(wb);
if (likely(res)) {
qq_deactivate(&brick->q_phase[1]);
goto done;
}
/* speculation was wrong: no shortcutting */
qq_deactivate(&brick->q_phase[3]);
no_speculation:
#endif
qq_wb_insert(&brick->q_phase[3], wb);
qq_deactivate(&brick->q_phase[1]);
brick_wake(&brick->worker_event, brick->worker_flag);
}
done:
return true;
err:
if (wb) {
free_writeback(wb);
}
collision:
return false;
}
/*********************************************************************
* Phase 2: log the old disk version.
*/
static inline
void _phase2_endio(struct writeback_info *wb)
{
struct trans_logger_brick *brick = wb->w_brick;
// queue up for the next phase
qq_wb_insert(&brick->q_phase[3], wb);
brick_wake(&brick->worker_event, brick->worker_flag);
return;
}
static noinline
void phase2_endio(void *private, int error)
{
struct trans_logger_mref_aspect *sub_mref_a;
struct trans_logger_brick *brick;
struct writeback_info *wb;
sub_mref_a = private;
CHECK_PTR(sub_mref_a, err);
wb = sub_mref_a->wb;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
if (unlikely(error < 0)) {
MARS_FAT("IO error %d\n", error);
goto err; // FIXME: this leads to hanging requests. do better.
}
CHECK_ATOMIC(&wb->w_sub_log_count, 1);
if (atomic_dec_and_test(&wb->w_sub_log_count)) {
banning_reset(&brick->q_phase[2].q_banning);
_phase2_endio(wb);
}
qq_deactivate(&brick->q_phase[2]);
return;
err:
MARS_FAT("hanging up....\n");
}
static noinline
bool _phase2_startio(struct trans_logger_mref_aspect *sub_mref_a)
{
struct mref_object *sub_mref = NULL;
struct writeback_info *wb;
struct trans_logger_input *input;
struct trans_logger_brick *brick;
struct log_status *logst;
void *data;
bool ok;
CHECK_PTR(sub_mref_a, err);
sub_mref = sub_mref_a->object;
CHECK_PTR(sub_mref, err);
wb = sub_mref_a->wb;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
input = sub_mref_a->log_input;
CHECK_PTR(input, err);
logst = &input->logst;
logst->do_crc = trans_logger_do_crc;
{
struct log_header l = {
.l_stamp = sub_mref_a->stamp,
.l_pos = sub_mref->ref_pos,
.l_len = sub_mref->ref_len,
.l_code = CODE_WRITE_OLD,
};
data = log_reserve(logst, &l);
}
if (unlikely(!data)) {
goto err;
}
memcpy(data, sub_mref->ref_data, sub_mref->ref_len);
ok = log_finalize(logst, sub_mref->ref_len, phase2_endio, sub_mref_a);
if (unlikely(!ok)) {
goto err;
}
return true;
err:
MARS_FAT("cannot log old data, pos = %lld len = %d\n", sub_mref ? sub_mref->ref_pos : 0, sub_mref ? sub_mref->ref_len : 0);
return false;
}
static noinline
bool phase2_startio(struct writeback_info *wb)
{
struct trans_logger_brick *brick;
bool ok = true;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
if (brick->log_reads && atomic_read(&wb->w_sub_log_count) > 0) {
struct list_head *start;
struct list_head *tmp;
start = &wb->w_sub_read_list;
for (tmp = start->next; tmp != start; tmp = tmp->next) {
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
sub_mref_a = container_of(tmp, struct trans_logger_mref_aspect, sub_head);
sub_mref = sub_mref_a->object;
mars_trace(sub_mref, "sub_log");
if (!_phase2_startio(sub_mref_a)) {
ok = false;
}
}
brick_wake(&brick->worker_event, brick->worker_flag);
} else {
_phase2_endio(wb);
}
return ok;
err:
return false;
}
/*********************************************************************
* Phase 3: overwrite old disk version with new version.
*/
static noinline
void phase3_endio(struct generic_callback *cb)
{
struct trans_logger_mref_aspect *sub_mref_a;
struct writeback_info *wb;
struct trans_logger_brick *brick;
CHECK_PTR(cb, err);
sub_mref_a = cb->cb_private;
CHECK_PTR(sub_mref_a, err);
wb = sub_mref_a->wb;
CHECK_PTR(wb, err);
brick = wb->w_brick;
CHECK_PTR(brick, err);
if (unlikely(cb->cb_error < 0)) {
MARS_FAT("IO error %d\n", cb->cb_error);
goto err;
}
hash_put_all(brick, &wb->w_collect_list);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_writeback_cluster_count);
#endif
free_writeback(wb);
banning_reset(&brick->q_phase[3].q_banning);
qq_deactivate(&brick->q_phase[3]);
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_FAT("hanging up....\n");
}
static noinline
bool phase3_startio(struct writeback_info *wb)
{
struct list_head *start = &wb->w_sub_read_list;
struct list_head *tmp;
/* Cleanup read requests (if they exist from previous phases)
*/
while ((tmp = start->next) != start) {
struct trans_logger_mref_aspect *sub_mref_a;
struct mref_object *sub_mref;
struct trans_logger_input *sub_input;
list_del_init(tmp);
sub_mref_a = container_of(tmp, struct trans_logger_mref_aspect, sub_head);
sub_mref = sub_mref_a->object;
sub_input = sub_mref_a->my_input;
GENERIC_INPUT_CALL(sub_input, mref_put, sub_mref);
}
update_writeback_info(wb);
/* Start writeback IO
*/
fire_writeback(&wb->w_sub_write_list, true);
return true;
}
/*********************************************************************
* The logger thread.
* There is only a single instance, dealing with all requests in parallel.
*/
static noinline
int run_mref_queue(struct logger_queue *q, bool (*startio)(struct trans_logger_mref_aspect *sub_mref_a), int max, bool do_limit)
{
struct trans_logger_brick *brick = q->q_brick;
int total_len = 0;
bool found = false;
bool ok;
int res = 0;
do {
struct trans_logger_mref_aspect *mref_a;
mref_a = qq_mref_fetch(q);
if (!mref_a)
goto done;
if (do_limit && likely(mref_a->object))
total_len += mref_a->object->ref_len;
ok = startio(mref_a);
if (unlikely(!ok)) {
qq_mref_pushback(q, mref_a);
goto done;
}
res++;
found = true;
__trans_logger_ref_put(mref_a->my_brick, mref_a);
} while (--max > 0);
done:
if (found) {
mars_limit(&global_writeback.limiter, (total_len - 1) / 1024 + 1);
brick_wake(&brick->worker_event, brick->worker_flag);
}
return res;
}
static noinline
int run_wb_queue(struct logger_queue *q, bool (*startio)(struct writeback_info *wb), int max)
{
struct trans_logger_brick *brick = q->q_brick;
int total_len = 0;
bool found = false;
bool ok;
int res = 0;
do {
struct writeback_info *wb;
wb = qq_wb_fetch(q);
if (!wb)
goto done;
total_len += wb->w_len;
ok = startio(wb);
if (unlikely(!ok)) {
qq_wb_pushback(q, wb);
goto done;
}
res++;
found = true;
} while (--max > 0);
done:
if (found) {
mars_limit(&global_writeback.limiter, (total_len - 1) / 1024 + 1);
brick_wake(&brick->worker_event, brick->worker_flag);
}
return res;
}
/* Ranking tables.
*/
static
struct rank_info float_queue_rank_log[] = {
{ 0, 0 },
{ 1, 100 },
{ RKI_DUMMY }
};
static
struct rank_info float_queue_rank_io[] = {
{ 0, 0 },
{ 1, 1 },
{ RKI_DUMMY }
};
static
struct rank_info float_fly_rank_log[] = {
{ 0, 0 },
{ 1, 1 },
{ 32, 10 },
{ RKI_DUMMY }
};
static
struct rank_info float_fly_rank_io[] = {
{ 0, 0 },
{ 1, 10 },
{ 2, -10 },
{ 10000, -200 },
{ RKI_DUMMY }
};
static
struct rank_info nofloat_queue_rank_log[] = {
{ 0, 0 },
{ 1, 10 },
{ RKI_DUMMY }
};
static
struct rank_info nofloat_queue_rank_io[] = {
{ 0, 0 },
{ 1, 10 },
{ 100, 100 },
{ RKI_DUMMY }
};
#define nofloat_fly_rank_log float_fly_rank_log
static
struct rank_info nofloat_fly_rank_io[] = {
{ 0, 0 },
{ 1, 10 },
{ 128, 8 },
{ 129, -200 },
{ RKI_DUMMY }
};
static
struct rank_info *queue_ranks[2][LOGGER_QUEUES] = {
[0] = {
[0] = float_queue_rank_log,
[1] = float_queue_rank_io,
[2] = float_queue_rank_io,
[3] = float_queue_rank_io,
},
[1] = {
[0] = nofloat_queue_rank_log,
[1] = nofloat_queue_rank_io,
[2] = nofloat_queue_rank_io,
[3] = nofloat_queue_rank_io,
},
};
static
struct rank_info *fly_ranks[2][LOGGER_QUEUES] = {
[0] = {
[0] = float_fly_rank_log,
[1] = float_fly_rank_io,
[2] = float_fly_rank_io,
[3] = float_fly_rank_io,
},
[1] = {
[0] = nofloat_fly_rank_log,
[1] = nofloat_fly_rank_io,
[2] = nofloat_fly_rank_io,
[3] = nofloat_fly_rank_io,
},
};
static
struct rank_info extra_rank_mref_flying[] = {
{ 0, 0 },
{ 1, 10 },
{ 16, 30 },
{ 31, 0 },
{ 32, -200 },
{ RKI_DUMMY }
};
static
struct rank_info global_rank_mref_flying[] = {
{ 0, 0 },
{ 63, 0 },
{ 64, -200 },
{ RKI_DUMMY }
};
/* Not checking pressure means to always have writeback pressure
* by default. No pressure means that writeback may be postponed
* when other IO is more important.
*/
static inline
bool _check_pressure(struct trans_logger_brick *brick)
{
int active;
/*
* Writeback IO starvation can occur when too many concurrents reads
* are filling the IO channnels for a very long time (e.g. when
* full backups are running in parallel to a highly loaded server).
* Ensure a very minimum writeback speed, depending on real time
* (as expected by humans).
*/
if (trans_logger_writeback_maxage) {
struct trans_logger_input *log_input;
long long inf_min_jiffies = 0;
log_input = brick->inputs[brick->log_input_nr];
if (log_input)
inf_min_jiffies = log_input->inf_min_jiffies;
if (!inf_min_jiffies ||
trans_logger_writeback_maxage * HZ + inf_min_jiffies <= (long long)jiffies)
return false;
}
active =
atomic_read(&brick->any_fly_count) +
brick->q_phase[0].q_queued + brick->q_phase[0].q_active;
return (active > trans_logger_pressure_limit) &&
brick->power.button;
}
static noinline
int _do_ranking(struct trans_logger_brick *brick)
{
struct rank_data *rkd = brick->rkd;
int res;
int i;
int pressure_mode;
int mref_flying;
bool delay_callers;
ranking_start(rkd, LOGGER_QUEUES);
// check the memory situation...
delay_callers = false;
pressure_mode = 1;
if (brick_global_memlimit >= 1024) {
int global_mem_used = atomic64_read(&global_mshadow_used) / 1024;
trans_logger_mem_usage = global_mem_used;
if (_check_pressure(brick))
pressure_mode = (global_mem_used < brick_global_memlimit / 2) ? 0 : 1;
if (global_mem_used >= brick_global_memlimit)
delay_callers = true;
MARS_IO("global_mem_used = %d\n", global_mem_used);
} else if (brick->shadow_mem_limit >= 8) {
int local_mem_used = atomic64_read(&brick->shadow_mem_used) / 1024;
if (_check_pressure(brick))
pressure_mode = (local_mem_used < brick->shadow_mem_limit / 2) ? 0 : 1;
if (local_mem_used >= brick->shadow_mem_limit)
delay_callers = true;
MARS_IO("local_mem_used = %d\n", local_mem_used);
}
/* disable pressure ONLY for testing */
if (trans_logger_disable_pressure > 0)
pressure_mode = 0;
else if (trans_logger_disable_pressure < 0)
pressure_mode = 1;
if (delay_callers) {
if (!brick->delay_callers) {
brick->delay_callers = true;
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_delay_count);
#endif
}
} else if (brick->delay_callers) {
brick->delay_callers = false;
brick_wake(&brick->caller_event, brick->caller_flag);
}
// global limit for flying mrefs
ranking_compute(&rkd[0], global_rank_mref_flying, atomic_read(&global_mref_flying));
// local limit for flying mrefs
mref_flying = 0;
for (i = TL_INPUT_LOG1; i <= TL_INPUT_LOG2; i++) {
struct trans_logger_input *input = brick->inputs[i];
mref_flying += atomic_read(&input->logst.mref_flying);
}
// obey the basic rules...
for (i = 0; i < LOGGER_QUEUES; i++) {
int queued = brick->q_phase[i].q_queued;
int flying;
/* only for testing */
if ((i == 3) &&
(trans_logger_disable_pressure < -1 ||
trans_logger_disable_pressure > 1))
queued = 0;
MARS_IO("i = %d queued = %d\n", i, queued);
/* This must come first.
* When a queue is empty, you must not credit any positive points.
* Otherwise, (almost) infinite selection of untreatable
* queues may occur.
*/
if (queued <= 0)
continue;
if (banning_is_hit(&brick->q_phase[i].q_banning)) {
#ifdef IO_DEBUGGING
unsigned long long now = cpu_clock(raw_smp_processor_id());
MARS_IO("BAILOUT queue = %d via banning now = %lld last_hit = %lld diff = %lld renew_count = %d count = %d\n",
i,
now,
now - brick->q_phase[i].q_banning.ban_last_hit,
brick->q_phase[i].q_banning.ban_last_hit,
brick->q_phase[i].q_banning.ban_renew_count,
brick->q_phase[i].q_banning.ban_count);
#endif
break;
}
if (i == 0) {
// limit mref IO parallelism on transaction log
ranking_compute(&rkd[0], extra_rank_mref_flying, mref_flying);
} else if (i == 1 && !pressure_mode) {
struct trans_logger_brick *leader;
int lim;
if (!mref_flying && brick->q_phase[0].q_queued > 0) {
MARS_IO("BAILOUT phase_[0]queued = %d phase_[0]active = %d\n",
brick->q_phase[0].q_queued,
brick->q_phase[0].q_active);
break;
}
if ((leader = elect_leader(&global_writeback)) != brick) {
MARS_IO("BAILOUT leader=%p brick=%p\n", leader, brick);
break;
}
if (banning_is_hit(&mars_global_ban)) {
#ifdef IO_DEBUGGING
unsigned long long now = cpu_clock(raw_smp_processor_id());
MARS_IO("BAILOUT via banning now = %lld last_hit = %lld diff = %lld renew_count = %d count = %d\n",
now,
now - mars_global_ban.ban_last_hit,
mars_global_ban.ban_last_hit,
mars_global_ban.ban_renew_count,
mars_global_ban.ban_count);
#endif
break;
}
lim = mars_limit(&global_writeback.limiter, 0);
if (lim > 0) {
MARS_IO("BAILOUT via limiter %d\n", lim);
break;
}
}
ranking_compute(&rkd[i], queue_ranks[pressure_mode][i], queued);
flying = brick->q_phase[i].q_active - brick->q_phase[i].q_active;
MARS_IO("i = %d queued = %d flying = %d\n", i, queued, flying);
ranking_compute(&rkd[i], fly_ranks[pressure_mode][i], flying);
}
// finalize it
ranking_stop(rkd, LOGGER_QUEUES);
res = ranking_select(rkd, LOGGER_QUEUES);
#ifdef IO_DEBUGGING
for (i = 0; i < LOGGER_QUEUES; i++) {
MARS_IO("rkd[%d]: points = %lld tmp = %lld got = %lld\n", i, rkd[i].rkd_current_points, rkd[i].rkd_tmp, rkd[i].rkd_got);
}
MARS_IO("res = %d\n", res);
#endif
return res;
}
static
void _init_input(struct trans_logger_input *input, loff_t start_pos, loff_t end_pos)
{
struct trans_logger_brick *brick = input->brick;
struct log_status *logst = &input->logst;
init_logst(logst, (void*)input, start_pos, end_pos);
logst->signal_event = &brick->worker_event;
logst->signal_flag = &brick->worker_flag;
logst->align_size = CONF_TRANS_ALIGN;
logst->chunk_size = CONF_TRANS_CHUNKSIZE;
logst->max_size = CONF_TRANS_MAX_MREF_SIZE;
input->inf.inf_min_pos = start_pos;
input->inf.inf_max_pos = end_pos;
get_lamport(NULL, &input->inf.inf_max_pos_stamp);
memcpy(&input->inf.inf_min_pos_stamp, &input->inf.inf_max_pos_stamp, sizeof(input->inf.inf_min_pos_stamp));
logst->log_pos = start_pos;
input->inf.inf_log_pos = start_pos;
input->inf_last_jiffies = jiffies;
input->inf.inf_is_replaying = false;
input->inf.inf_is_logging = false;
input->is_operating = true;
}
static
void _init_inputs(struct trans_logger_brick *brick, bool is_first)
{
struct trans_logger_input *input;
int old_nr = brick->old_input_nr;
int log_nr = brick->log_input_nr;
int new_nr = brick->new_input_nr;
if (!is_first &&
(new_nr == log_nr ||
log_nr != old_nr)) {
MARS_IO("nothing to do, new_input_nr = %d log_input_nr = %d old_input_nr = %d\n", new_nr, log_nr, old_nr);
goto done;
}
if (unlikely(new_nr < TL_INPUT_LOG1 || new_nr > TL_INPUT_LOG2)) {
MARS_ERR("bad new_input_nr = %d\n", new_nr);
goto done;
}
input = brick->inputs[new_nr];
CHECK_PTR(input, done);
if (input->is_operating || !input->connect) {
MARS_IO("cannot yet switch over to %d (is_operating = %d connect = %p)\n", new_nr, input->is_operating, input->connect);
goto done;
}
down(&input->inf_mutex);
_init_input(input, 0, 0);
input->inf.inf_is_logging = is_first;
// from now on, new requests should go to the new input
brick->log_input_nr = new_nr;
MARS_INF("switched over to new logfile %d (old = %d)\n", new_nr, old_nr);
/* Flush the old log buffer and update its symlinks.
* Notice: for some short time, _both_ logfiles may grow
* due to (harmless) races with log_flush().
*/
if (likely(!is_first)) {
struct trans_logger_input *other_input = brick->inputs[old_nr];
down(&other_input->inf_mutex);
log_flush(&other_input->logst);
_inf_callback(other_input, true);
up(&other_input->inf_mutex);
}
_inf_callback(input, true);
up(&input->inf_mutex);
done: ;
}
static
int _nr_flying_inputs(struct trans_logger_brick *brick)
{
int count = 0;
int i;
for (i = TL_INPUT_LOG1; i <= TL_INPUT_LOG2; i++) {
struct trans_logger_input *input = brick->inputs[i];
struct log_status *logst = &input->logst;
if (input->is_operating) {
count += logst->count;
}
}
return count;
}
static
void _flush_inputs(struct trans_logger_brick *brick)
{
int i;
for (i = TL_INPUT_LOG1; i <= TL_INPUT_LOG2; i++) {
struct trans_logger_input *input = brick->inputs[i];
struct log_status *logst = &input->logst;
if (input->is_operating && logst->count > 0) {
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_flush_count);
#endif
log_flush(logst);
}
}
}
static
void _exit_inputs(struct trans_logger_brick *brick, bool force)
{
int i;
for (i = TL_INPUT_LOG1; i <= TL_INPUT_LOG2; i++) {
struct trans_logger_input *input = brick->inputs[i];
struct log_status *logst = &input->logst;
if (input->is_operating &&
(force || !input->connect)) {
bool old_replaying = input->inf.inf_is_replaying;
bool old_logging = input->inf.inf_is_logging;
MARS_DBG("cleaning up input %d (log = %d old = %d), old_replaying = %d old_logging = %d\n", i, brick->log_input_nr, brick->old_input_nr, old_replaying, old_logging);
exit_logst(logst);
// no locking here: we should be the only thread doing this.
_inf_callback(input, true);
input->inf_last_jiffies = 0;
input->inf.inf_is_replaying = false;
input->inf.inf_is_logging = false;
input->is_operating = false;
if (i == brick->old_input_nr && i != brick->log_input_nr) {
struct trans_logger_input *other_input = brick->inputs[brick->log_input_nr];
down(&other_input->inf_mutex);
brick->old_input_nr = brick->log_input_nr;
other_input->inf.inf_is_replaying = old_replaying;
other_input->inf.inf_is_logging = old_logging;
_inf_callback(other_input, true);
up(&other_input->inf_mutex);
}
}
}
}
/* Performance-critical:
* Calling log_flush() too often may result in
* increased overhead (and thus in lower throughput).
* Call it only when the IO scheduler need not do anything else.
* OTOH, calling it too seldom may hold back
* IO completion for the end user for too long time.
*
* Be careful to flush any leftovers in the log buffer, at least after
* some short delay.
*
* Description of flush_mode:
* 0 = flush unconditionally
* 1 = flush only when nothing can be appended to the transaction log
* 2 = see 1 && flush only when the user is waiting for an answer
* 3 = see 1 && not 2 && flush only when there is no other activity (background mode)
* Notice: 3 makes only sense for leftovers where the user is _not_ waiting for
*/
static inline
void flush_inputs(struct trans_logger_brick *brick, int flush_mode)
{
if (flush_mode < 1 ||
// there is nothing to append any more
(brick->q_phase[0].q_queued <= 0 &&
// and the user is waiting for an answer
(flush_mode < 2 ||
atomic_read(&brick->log_fly_count) > 0 ||
// else flush any leftovers in background, when there is no writeback activity
(flush_mode == 3 &&
brick->q_phase[1].q_active - brick->q_phase[1].q_queued +
brick->q_phase[3].q_active - brick->q_phase[3].q_queued <= 0)))) {
_flush_inputs(brick);
}
}
static atomic_t logger_count = ATOMIC_INIT(0);
static noinline
void trans_logger_log(struct trans_logger_brick *brick)
{
long long old_jiffies = jiffies;
long long work_jiffies = jiffies;
int interleave = 0;
int nr_flying;
memset(brick->rkd, 0, sizeof(brick->rkd));
brick->replay_code = TL_REPLAY_RUNNING;
brick->disk_io_error = 0;
_init_inputs(brick, true);
if (atomic_inc_return(&logger_count) == 1)
mars_limit_reset(&global_writeback.limiter);
mars_power_led_on((void*)brick, true);
while ((!brick->terminate &&
!brick_thread_should_stop()) ||
_congested(brick)) {
int winner;
int nr;
brick_wait(
brick->worker_event,
brick->worker_flag,
({
winner = _do_ranking(brick);
MARS_IO("winner = %d\n", winner);
if (winner < 0) { // no more work to do
int flush_mode = 2 - ((int)(jiffies - work_jiffies)) / (HZ * 2);
flush_inputs(brick, flush_mode);
interleave = 0;
} else { // reset the timer whenever something is to do
work_jiffies = jiffies;
}
winner >= 0;
}),
HZ / 10);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_round_count);
#endif
if (brick->cease_logging) {
brick->stopped_logging = true;
} else if (brick->stopped_logging && !_congested(brick)) {
brick->stopped_logging = false;
}
_init_inputs(brick, false);
switch (winner) {
case 0:
interleave = 0;
nr = run_mref_queue(&brick->q_phase[0], prep_phase_startio, brick->q_phase[0].q_batchlen, true);
goto done;
case 1:
if (interleave >= trans_logger_max_interleave && trans_logger_max_interleave >= 0) {
interleave = 0;
flush_inputs(brick, 3);
}
nr = run_mref_queue(&brick->q_phase[1], phase1_startio, brick->q_phase[1].q_batchlen, true);
interleave += nr;
goto done;
case 2:
interleave = 0;
nr = run_wb_queue(&brick->q_phase[2], phase2_startio, brick->q_phase[2].q_batchlen);
goto done;
case 3:
if (interleave >= trans_logger_max_interleave && trans_logger_max_interleave >= 0) {
interleave = 0;
flush_inputs(brick, 3);
}
nr = run_wb_queue(&brick->q_phase[3], phase3_startio, brick->q_phase[3].q_batchlen);
interleave += nr;
done:
if (unlikely(nr <= 0)) {
/* This should not happen!
* However, in error situations, the ranking
* algorithm cannot foresee anything.
*/
brick->q_phase[winner].no_progress_count++;
banning_hit(&brick->q_phase[winner].q_banning, 10000);
flush_inputs(brick, 0);
}
ranking_select_done(brick->rkd, winner, nr);
break;
default:
;
}
/* Update symlinks even during pauses.
*/
if (winner < 0 && ((long long)jiffies) - old_jiffies >= HZ) {
int i;
old_jiffies = jiffies;
for (i = TL_INPUT_LOG1; i <= TL_INPUT_LOG2; i++) {
struct trans_logger_input *input = brick->inputs[i];
down(&input->inf_mutex);
_inf_callback(input, false);
up(&input->inf_mutex);
}
}
_exit_inputs(brick, false);
}
for (;;) {
_exit_inputs(brick, true);
nr_flying = _nr_flying_inputs(brick);
if (nr_flying <= 0)
break;
MARS_INF("%d inputs are operating\n", nr_flying);
brick_msleep(1000);
}
if (!atomic_dec_return(&logger_count))
mars_limit_reset(&global_writeback.limiter);
brick->terminated = true;
mars_trigger();
}
////////////////////////////// log replay //////////////////////////////
static noinline
void replay_endio(struct generic_callback *cb)
{
struct trans_logger_mref_aspect *mref_a = cb->cb_private;
struct trans_logger_brick *brick;
bool ok;
_crashme(22, false);
LAST_CALLBACK(cb);
CHECK_PTR(mref_a, err);
brick = mref_a->my_brick;
CHECK_PTR(brick, err);
if (unlikely(cb->cb_error < 0)) {
brick->disk_io_error = cb->cb_error;
MARS_ERR("IO error = %d\n", cb->cb_error);
}
down_write(&brick->replay_mutex);
ok = !list_empty(&mref_a->replay_head);
list_del_init(&mref_a->replay_head);
up_write(&brick->replay_mutex);
if (likely(ok)) {
atomic_dec(&brick->replay_count);
} else {
MARS_ERR("callback with empty replay_head (replay_count=%d)\n", atomic_read(&brick->replay_count));
}
brick_wake(&brick->worker_event, brick->worker_flag);
return;
err:
MARS_FAT("cannot handle replay IO\n");
}
static noinline
bool _has_conflict(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *mref_a)
{
struct mref_object *mref = mref_a->object;
struct list_head *tmp;
bool res = false;
down_read(&brick->replay_mutex);
for (tmp = brick->replay_list.next; tmp != &brick->replay_list; tmp = tmp->next) {
struct trans_logger_mref_aspect *tmp_a;
struct mref_object *tmp_mref;
tmp_a = container_of(tmp, struct trans_logger_mref_aspect, replay_head);
tmp_mref = tmp_a->object;
if (tmp_mref->ref_pos + tmp_mref->ref_len > mref->ref_pos && tmp_mref->ref_pos < mref->ref_pos + mref->ref_len) {
res = true;
break;
}
}
up_read(&brick->replay_mutex);
return res;
}
static noinline
void wait_replay(struct trans_logger_brick *brick, struct trans_logger_mref_aspect *mref_a)
{
const int max = 512; // limit parallelism somewhat
int conflicts = 0;
bool ok = false;
bool was_empty;
brick_wait(brick->worker_event,
brick->worker_flag,
atomic_read(&brick->replay_count) < max
&& (_has_conflict(brick, mref_a) ? conflicts++ : (ok = true), ok),
60 * HZ);
#ifdef ADDITIONAL_COUNTERS
atomic_inc(&brick->total_replay_count);
if (conflicts)
atomic_inc(&brick->total_replay_conflict_count);
#endif
down_write(&brick->replay_mutex);
was_empty = !!list_empty(&mref_a->replay_head);
if (likely(was_empty)) {
atomic_inc(&brick->replay_count);
} else {
list_del(&mref_a->replay_head);
}
list_add(&mref_a->replay_head, &brick->replay_list);
up_write(&brick->replay_mutex);
if (unlikely(!was_empty)) {
MARS_ERR("replay_head was already used (ok=%d, conflicts=%d, replay_count=%d)\n", ok, conflicts, atomic_read(&brick->replay_count));
}
}
static noinline
int replay_data(struct trans_logger_brick *brick, loff_t pos, void *buf, int len)
{
struct trans_logger_input *input = brick->inputs[TL_INPUT_WRITEBACK];
int status;
MARS_IO("got data, pos = %lld, len = %d\n", pos, len);
if (!input->connect) {
input = brick->inputs[TL_INPUT_READ];
}
/* TODO for better efficiency:
* Instead of starting IO here, just put the data into the hashes
* and queues such that ordinary IO will be corrected.
* Writeback will be lazy then.
* The switch infrastructure must be changed before this
* becomes possible.
*/
#ifdef REPLAY_DATA
while (len > 0) {
struct mref_object *mref;
struct trans_logger_mref_aspect *mref_a;
status = -ENOMEM;
mref = trans_logger_alloc_mref(brick);
if (unlikely(!mref)) {
MARS_ERR("no memory\n");
goto done;
}
mref_a = trans_logger_mref_get_aspect(brick, mref);
CHECK_PTR(mref_a, done);
CHECK_ASPECT(mref_a, mref, done);
mref->ref_pos = pos;
mref->ref_data = NULL;
mref->ref_len = len;
mref->ref_flags = MREF_WRITE | MREF_MAY_WRITE;
status = GENERIC_INPUT_CALL(input, mref_get, mref);
if (unlikely(status < 0)) {
MARS_ERR("cannot get mref, status = %d\n", status);
goto done;
}
if (unlikely(!mref->ref_data)) {
status = -ENOMEM;
MARS_ERR("cannot get mref, status = %d\n", status);
goto done;
}
if (unlikely(mref->ref_len <= 0 || mref->ref_len > len)) {
status = -EINVAL;
MARS_ERR("bad ref len = %d (requested = %d)\n", mref->ref_len, len);
goto done;
}
mars_trace(mref, "replay_start");
wait_replay(brick, mref_a);
mars_trace(mref, "replay_io");
memcpy(mref->ref_data, buf, mref->ref_len);
SETUP_CALLBACK(mref, replay_endio, mref_a);
mref_a->my_brick = brick;
GENERIC_INPUT_CALL(input, mref_io, mref);
if (unlikely(mref->ref_len <= 0)) {
status = -EINVAL;
MARS_ERR("bad ref len = %d (requested = %d)\n", mref->ref_len, len);
goto done;
}
pos += mref->ref_len;
buf += mref->ref_len;
len -= mref->ref_len;
GENERIC_INPUT_CALL(input, mref_put, mref);
}
#endif
status = 0;
done:
return status;
}
static noinline
void trans_logger_replay(struct trans_logger_brick *brick)
{
struct trans_logger_input *input = brick->inputs[brick->log_input_nr];
struct log_header lh = {};
void *dealloc = NULL;
loff_t start_pos;
loff_t end_pos;
loff_t finished_pos = -1;
loff_t new_finished_pos = -1;
long long old_jiffies = jiffies;
int replay_code = TL_REPLAY_RUNNING;
int nr_flying;
int backoff = 0;
int status = 0;
brick->disk_io_error = 0;
start_pos = brick->replay_start_pos;
end_pos = brick->replay_end_pos;
brick->replay_current_pos = start_pos;
_init_input(input, start_pos, end_pos);
input->inf.inf_min_pos = start_pos;
input->inf.inf_max_pos = end_pos;
input->inf.inf_log_pos = end_pos;
input->inf.inf_is_replaying = true;
input->inf.inf_is_logging = false;
mars_limit_reset(brick->replay_limiter);
MARS_INF("starting replay from %lld to %lld\n", start_pos, end_pos);
brick->replay_code = replay_code;
mars_power_led_on((void *)brick, true);
for (;;) {
void *buf = NULL;
int len = 0;
if (brick_thread_should_stop() ||
brick->terminate ||
(!brick->continuous_replay_mode && finished_pos >= brick->replay_end_pos)) {
status = 0; // treat as EOF
break;
}
brick_mem_free(dealloc);
status = log_read(&input->logst, false, &lh,
&buf, &len, &dealloc);
new_finished_pos = input->logst.log_pos + input->logst.offset;
MARS_RPL("read %lld %lld\n", finished_pos, new_finished_pos);
if (status == -EAGAIN) {
loff_t remaining = brick->replay_end_pos - new_finished_pos;
MARS_DBG("got -EAGAIN, remaining = %lld\n", remaining);
if (brick->replay_tolerance > 0 && remaining < brick->replay_tolerance) {
MARS_WRN("logfile is truncated at position %lld (end_pos = %lld, remaining = %lld, tolerance = %d)\n",
new_finished_pos,
brick->replay_end_pos,
remaining,
brick->replay_tolerance);
finished_pos = new_finished_pos;
replay_code = status;
break;
}
brick_msleep(backoff);
if (backoff < trans_logger_replay_timeout * 1000) {
backoff += 100;
} else {
MARS_WRN("logfile replay not possible at position %lld (end_pos = %lld, remaining = %lld), please check/repair your logfile in userspace by some tool!\n",
new_finished_pos,
brick->replay_end_pos,
remaining);
replay_code = status;
break;
}
continue;
}
if (unlikely(status < 0)) {
replay_code = status;
MARS_WRN("cannot read logfile data, status = %d\n", status);
break;
}
if ((!status && len <= 0) ||
new_finished_pos > brick->replay_end_pos) { // EOF -> wait until brick_thread_should_stop()
MARS_DBG("EOF at %lld (old = %lld, end_pos = %lld)\n", new_finished_pos, finished_pos, brick->replay_end_pos);
if (!brick->continuous_replay_mode) {
// notice: finished_pos remains at old value here!
break;
}
brick_msleep(1000);
continue;
}
if (lh.l_code != CODE_WRITE_NEW) {
MARS_IO("ignoring pos = %lld len = %d code = %d\n", lh.l_pos, lh.l_len, lh.l_code);
} else if (unlikely(brick->disk_io_error)) {
status = brick->disk_io_error;
replay_code = status;
MARS_ERR("IO error %d\n", status);
break;
} else if (likely(buf && len)) {
if (brick->replay_limiter)
mars_limit_sleep(brick->replay_limiter, (len - 1) / 1024 + 1);
status = replay_data(brick, lh.l_pos, buf, len);
MARS_RPL("replay %lld %lld (pos=%lld status=%d)\n", finished_pos, new_finished_pos, lh.l_pos, status);
if (unlikely(status < 0)) {
replay_code = status;
MARS_ERR("cannot replay data at pos = %lld len = %d, status = %d\n", lh.l_pos, len, status);
break;
} else {
finished_pos = new_finished_pos;
}
}
// do this _after_ any opportunities for errors...
if ((atomic_read(&brick->replay_count) <= 0 ||
((long long)jiffies) - old_jiffies >= HZ * 3) &&
finished_pos >= 0) {
// for safety, wait until the IO queue has drained.
brick_wait(brick->worker_event,
brick->worker_flag,
atomic_read(&brick->replay_count) <= 0,
30 * HZ);
if (unlikely(brick->disk_io_error)) {
status = brick->disk_io_error;
replay_code = status;
MARS_ERR("IO error %d\n", status);
break;
}
down(&input->inf_mutex);
input->inf.inf_min_pos = finished_pos;
get_lamport(NULL, &input->inf.inf_min_pos_stamp);
old_jiffies = jiffies;
_inf_callback(input, false);
up(&input->inf_mutex);
}
_exit_inputs(brick, false);
}
brick_mem_free(dealloc);
MARS_INF("waiting for finish...\n");
brick_wait(brick->worker_event,
brick->worker_flag,
atomic_read(&brick->replay_count) <= 0,
60 * HZ);
if (unlikely(finished_pos > brick->replay_end_pos)) {
MARS_ERR("finished_pos too large: %lld + %d = %lld > %lld\n", input->logst.log_pos, input->logst.offset, finished_pos, brick->replay_end_pos);
}
if (finished_pos >= 0 && !brick->disk_io_error) {
input->inf.inf_min_pos = finished_pos;
brick->replay_current_pos = finished_pos;
}
get_lamport(NULL, &input->inf.inf_min_pos_stamp);
if (status >= 0 && finished_pos == brick->replay_end_pos) {
MARS_INF("replay finished at %lld\n", finished_pos);
replay_code = TL_REPLAY_FINISHED;
} else if (status == -EAGAIN && finished_pos + brick->replay_tolerance > brick->replay_end_pos) {
MARS_INF("TOLERANCE: logfile is incomplete at %lld (of %lld)\n", finished_pos, brick->replay_end_pos);
replay_code = TL_REPLAY_INCOMPLETE;
} else if (status < 0) {
if (finished_pos < 0)
finished_pos = new_finished_pos;
if (finished_pos + brick->replay_tolerance > brick->replay_end_pos) {
MARS_INF("TOLERANCE: logfile is incomplete at %lld (of %lld), status = %d\n", finished_pos, brick->replay_end_pos, status);
} else {
MARS_ERR("replay error %d at %lld (of %lld)\n", status, finished_pos, brick->replay_end_pos);
}
replay_code = status;
} else {
MARS_INF("replay stopped prematurely at %lld (of %lld)\n", finished_pos, brick->replay_end_pos);
replay_code = TL_REPLAY_INCOMPLETE;
}
for (;;) {
_exit_inputs(brick, true);
nr_flying = _nr_flying_inputs(brick);
if (nr_flying <= 0)
break;
MARS_INF("%d inputs are operating\n", nr_flying);
brick_msleep(1000);
}
mars_limit_reset(brick->replay_limiter);
brick->terminated = true;
/* final result rporting */
brick->replay_code = replay_code;
mars_trigger();
while (!brick_thread_should_stop()) {
brick_wait(brick->worker_event,
brick->worker_flag,
brick_thread_should_stop(),
HZ / 10);
}
}
///////////////////////// logger thread / switching /////////////////////////
static noinline
int trans_logger_thread(void *data)
{
struct trans_logger_output *output = data;
struct trans_logger_brick *brick = output->brick;
MARS_INF("........... logger has started.\n");
if (brick->replay_mode) {
trans_logger_replay(brick);
} else {
trans_logger_log(brick);
}
MARS_INF("........... logger has stopped.\n");
mars_power_led_on((void*)brick, false);
mars_power_led_off((void*)brick, true);
return 0;
}
static noinline
int trans_logger_switch(struct trans_logger_brick *brick)
{
static int index = 0;
struct trans_logger_output *output = brick->outputs[0];
if (brick->power.button) {
if (!brick->thread && brick->power.led_off) {
brick->terminate = false;
brick->terminated = false;
mars_power_led_off((void*)brick, false);
brick->thread = brick_thread_create(trans_logger_thread, output, "mars_logger%d", index++);
if (unlikely(!brick->thread)) {
MARS_ERR("cannot create logger thread\n");
return -ENOENT;
}
}
} else {
mars_power_led_on((void*)brick, false);
if (brick->thread) {
brick->terminate = true;
brick_wake(&brick->worker_event, brick->worker_flag);
brick_wake(&brick->caller_event, brick->caller_flag);
if (brick->terminated) {
MARS_INF("stopping thread...\n");
brick_thread_stop(brick->thread);
brick->thread = NULL;
} else if (!brick->power.led_off) {
return -EAGAIN;
}
}
}
return 0;
}
//////////////// informational / statistics ///////////////
static noinline
char *trans_logger_statistics(struct trans_logger_brick *brick, int verbose)
{
char *res = brick_string_alloc(1024);
if (!res)
return NULL;
snprintf(res, 1023,
"mode replay=%d "
"continuous=%d "
"replay_code=%d "
"disk_io_error=%d "
"log_reads=%d | "
"cease_logging=%d "
"stopped_logging=%d "
"congested=%d | "
"replay_start_pos = %lld "
"replay_end_pos = %lld | "
"new_input_nr = %d "
"log_input_nr = %d "
"(old = %d) "
"inf_min_pos1 = %lld "
"inf_max_pos1 = %lld "
"inf_min_pos2 = %lld "
"inf_max_pos2 = %lld | "
#ifdef ADDITIONAL_COUNTERS
"total hash_insert=%d "
"hash_find=%d "
"hash_extend=%d "
"replay=%d "
"replay_conflict=%d (%d%%) "
"callbacks=%d "
"reads=%d "
"writes=%d "
"flushes=%d (%d%%) "
"wb_clusters=%d "
"writebacks=%d (%d%%) "
"shortcut=%d (%d%%) "
"mshadow=%d "
"sshadow=%d "
"mshadow_buffered=%d sshadow_buffered=%d "
"rounds=%d "
"restarts=%d "
"delays=%d | "
#endif
"current #mrefs = %d "
"shadow_mem_used=%ld/%lld "
"replay_count=%d "
#ifdef ADDITIONAL_COUNTERS
"mshadow=%d/%d "
"sshadow=%d "
"hash_count=%d "
"balance=%d/%d/%d/%d "
#endif
"pos_count1=%d "
"pos_count2=%d "
"log_refs1=%d "
"log_refs2=%d "
"any_fly=%d "
"log_fly=%d "
"mref_flying1=%d "
"mref_flying2=%d "
"phase0=%d-%d <%d/%d> "
"phase1=%d-%d <%d/%d> "
"phase2=%d-%d <%d/%d> "
"phase3=%d-%d <%d/%d>\n",
brick->replay_mode,
brick->continuous_replay_mode,
brick->replay_code,
brick->disk_io_error,
brick->log_reads,
brick->cease_logging,
brick->stopped_logging,
_congested(brick),
brick->replay_start_pos,
brick->replay_end_pos,
brick->new_input_nr,
brick->log_input_nr,
brick->old_input_nr,
brick->inputs[TL_INPUT_LOG1]->inf.inf_min_pos,
brick->inputs[TL_INPUT_LOG1]->inf.inf_max_pos,
brick->inputs[TL_INPUT_LOG2]->inf.inf_min_pos,
brick->inputs[TL_INPUT_LOG2]->inf.inf_max_pos,
#ifdef ADDITIONAL_COUNTERS
atomic_read(&brick->total_hash_insert_count),
atomic_read(&brick->total_hash_find_count),
atomic_read(&brick->total_hash_extend_count),
atomic_read(&brick->total_replay_count),
atomic_read(&brick->total_replay_conflict_count),
atomic_read(&brick->total_replay_count) ? atomic_read(&brick->total_replay_conflict_count) * 100 / atomic_read(&brick->total_replay_count) : 0,
atomic_read(&brick->total_cb_count),
atomic_read(&brick->total_read_count),
atomic_read(&brick->total_write_count),
atomic_read(&brick->total_flush_count),
atomic_read(&brick->total_write_count) ? atomic_read(&brick->total_flush_count) * 100 / atomic_read(&brick->total_write_count) : 0,
atomic_read(&brick->total_writeback_cluster_count),
atomic_read(&brick->total_writeback_count),
atomic_read(&brick->total_writeback_cluster_count) ? atomic_read(&brick->total_writeback_count) * 100 / atomic_read(&brick->total_writeback_cluster_count) : 0,
atomic_read(&brick->total_shortcut_count),
atomic_read(&brick->total_writeback_count) ? atomic_read(&brick->total_shortcut_count) * 100 / atomic_read(&brick->total_writeback_count) : 0,
atomic_read(&brick->total_mshadow_count),
atomic_read(&brick->total_sshadow_count),
atomic_read(&brick->total_mshadow_buffered_count),
atomic_read(&brick->total_sshadow_buffered_count),
atomic_read(&brick->total_round_count),
atomic_read(&brick->total_restart_count),
atomic_read(&brick->total_delay_count),
#endif
atomic_read(&brick->mref_object_layout.alloc_count),
atomic64_read(&brick->shadow_mem_used) / 1024,
brick_global_memlimit,
atomic_read(&brick->replay_count),
#ifdef ADDITIONAL_COUNTERS
atomic_read(&brick->mshadow_count),
brick->shadow_mem_limit,
atomic_read(&brick->sshadow_count),
atomic_read(&brick->hash_count),
atomic_read(&brick->sub_balance_count),
atomic_read(&brick->inner_balance_count),
atomic_read(&brick->outer_balance_count),
atomic_read(&brick->wb_balance_count),
#endif
atomic_read(&brick->inputs[TL_INPUT_LOG1]->pos_count),
atomic_read(&brick->inputs[TL_INPUT_LOG2]->pos_count),
atomic_read(&brick->inputs[TL_INPUT_LOG1]->log_ref_count),
atomic_read(&brick->inputs[TL_INPUT_LOG2]->log_ref_count),
atomic_read(&brick->any_fly_count),
atomic_read(&brick->log_fly_count),
atomic_read(&brick->inputs[TL_INPUT_LOG1]->logst.mref_flying),
atomic_read(&brick->inputs[TL_INPUT_LOG2]->logst.mref_flying),
brick->q_phase[0].q_active,
brick->q_phase[0].q_queued,
brick->q_phase[0].pushback_count,
brick->q_phase[0].no_progress_count,
brick->q_phase[1].q_active,
brick->q_phase[1].q_queued,
brick->q_phase[1].pushback_count,
brick->q_phase[1].no_progress_count,
brick->q_phase[2].q_active,
brick->q_phase[2].q_queued,
brick->q_phase[2].pushback_count,
brick->q_phase[2].no_progress_count,
brick->q_phase[3].q_active,
brick->q_phase[3].q_queued,
brick->q_phase[3].pushback_count,
brick->q_phase[3].no_progress_count);
return res;
}
static noinline
void trans_logger_reset_statistics(struct trans_logger_brick *brick)
{
#ifdef ADDITIONAL_COUNTERS
atomic_set(&brick->total_hash_insert_count, 0);
atomic_set(&brick->total_hash_find_count, 0);
atomic_set(&brick->total_hash_extend_count, 0);
atomic_set(&brick->total_replay_count, 0);
atomic_set(&brick->total_replay_conflict_count, 0);
atomic_set(&brick->total_cb_count, 0);
atomic_set(&brick->total_read_count, 0);
atomic_set(&brick->total_write_count, 0);
atomic_set(&brick->total_flush_count, 0);
atomic_set(&brick->total_writeback_count, 0);
atomic_set(&brick->total_writeback_cluster_count, 0);
atomic_set(&brick->total_shortcut_count, 0);
atomic_set(&brick->total_mshadow_count, 0);
atomic_set(&brick->total_sshadow_count, 0);
atomic_set(&brick->total_mshadow_buffered_count, 0);
atomic_set(&brick->total_sshadow_buffered_count, 0);
atomic_set(&brick->total_round_count, 0);
atomic_set(&brick->total_restart_count, 0);
atomic_set(&brick->total_delay_count, 0);
#endif
}
//////////////// object / aspect constructors / destructors ///////////////
static noinline
int trans_logger_mref_aspect_init_fn(struct generic_aspect *_ini)
{
struct trans_logger_mref_aspect *ini = (void*)_ini;
ini->lh.lh_pos = &ini->object->ref_pos;
INIT_LIST_HEAD(&ini->lh.lh_head);
INIT_LIST_HEAD(&ini->hash_head);
INIT_LIST_HEAD(&ini->pos_head);
INIT_LIST_HEAD(&ini->replay_head);
INIT_LIST_HEAD(&ini->collect_head);
INIT_LIST_HEAD(&ini->sub_list);
INIT_LIST_HEAD(&ini->sub_head);
return 0;
}
static noinline
void trans_logger_mref_aspect_exit_fn(struct generic_aspect *_ini)
{
struct trans_logger_mref_aspect *ini = (void*)_ini;
CHECK_HEAD_EMPTY(&ini->lh.lh_head);
CHECK_HEAD_EMPTY(&ini->hash_head);
CHECK_HEAD_EMPTY(&ini->pos_head);
CHECK_HEAD_EMPTY(&ini->replay_head);
CHECK_HEAD_EMPTY(&ini->collect_head);
CHECK_HEAD_EMPTY(&ini->sub_list);
CHECK_HEAD_EMPTY(&ini->sub_head);
if (ini->log_input) {
atomic_dec(&ini->log_input->log_ref_count);
}
}
MARS_MAKE_STATICS(trans_logger);
////////////////////// brick constructors / destructors ////////////////////
static
void _free_pages(struct trans_logger_brick *brick)
{
int i;
for (i = 0; i < NR_HASH_PAGES; i++) {
struct trans_logger_hash_anchor *sub_table = brick->hash_table[i];
int j;
if (!sub_table) {
continue;
}
for (j = 0; j < HASH_PER_PAGE; j++) {
struct trans_logger_hash_anchor *start = &sub_table[j];
CHECK_HEAD_EMPTY(&start->hash_anchor);
}
brick_block_free(sub_table, PAGE_SIZE);
}
brick_block_free(brick->hash_table, PAGE_SIZE);
}
static noinline
int trans_logger_brick_construct(struct trans_logger_brick *brick)
{
int i;
brick->hash_table = brick_block_alloc(0, PAGE_SIZE);
if (unlikely(!brick->hash_table)) {
MARS_ERR("cannot allocate hash directory table.\n");
return -ENOMEM;
}
memset(brick->hash_table, 0, PAGE_SIZE);
for (i = 0; i < NR_HASH_PAGES; i++) {
struct trans_logger_hash_anchor *sub_table;
int j;
// this should be usually optimized away as dead code
if (unlikely(i >= MAX_HASH_PAGES)) {
MARS_ERR("sorry, subtable index %d is too large.\n", i);
_free_pages(brick);
return -EINVAL;
}
sub_table = brick_block_alloc(0, PAGE_SIZE);
brick->hash_table[i] = sub_table;
if (unlikely(!sub_table)) {
MARS_ERR("cannot allocate hash subtable %d.\n", i);
_free_pages(brick);
return -ENOMEM;
}
memset(sub_table, 0, PAGE_SIZE);
for (j = 0; j < HASH_PER_PAGE; j++) {
struct trans_logger_hash_anchor *start = &sub_table[j];
init_rwsem(&start->hash_mutex);
INIT_LIST_HEAD(&start->hash_anchor);
}
}
#ifdef ADDITIONAL_COUNTERS
atomic_set(&brick->hash_count, 0);
#endif
init_rwsem(&brick->replay_mutex);
INIT_LIST_HEAD(&brick->replay_list);
INIT_LIST_HEAD(&brick->group_head);
init_waitqueue_head(&brick->worker_event);
init_waitqueue_head(&brick->caller_event);
qq_init(&brick->q_phase[0], brick);
qq_init(&brick->q_phase[1], brick);
qq_init(&brick->q_phase[2], brick);
qq_init(&brick->q_phase[3], brick);
brick->q_phase[0].q_insert_info = "q0_ins";
brick->q_phase[0].q_pushback_info = "q0_push";
brick->q_phase[0].q_fetch_info = "q0_fetch";
brick->q_phase[1].q_insert_info = "q1_ins";
brick->q_phase[1].q_pushback_info = "q1_push";
brick->q_phase[1].q_fetch_info = "q1_fetch";
brick->q_phase[2].q_insert_info = "q2_ins";
brick->q_phase[2].q_pushback_info = "q2_push";
brick->q_phase[2].q_fetch_info = "q2_fetch";
brick->q_phase[3].q_insert_info = "q3_ins";
brick->q_phase[3].q_pushback_info = "q3_push";
brick->q_phase[3].q_fetch_info = "q3_fetch";
brick->new_input_nr = TL_INPUT_LOG1;
brick->log_input_nr = TL_INPUT_LOG1;
brick->old_input_nr = TL_INPUT_LOG1;
add_to_group(&global_writeback, brick);
return 0;
}
static noinline
int trans_logger_brick_destruct(struct trans_logger_brick *brick)
{
_free_pages(brick);
CHECK_HEAD_EMPTY(&brick->replay_list);
remove_from_group(&global_writeback, brick);
return 0;
}
static noinline
int trans_logger_output_construct(struct trans_logger_output *output)
{
return 0;
}
static noinline
int trans_logger_input_construct(struct trans_logger_input *input)
{
INIT_LIST_HEAD(&input->pos_list);
sema_init(&input->inf_mutex, 1);
return 0;
}
static noinline
int trans_logger_input_destruct(struct trans_logger_input *input)
{
CHECK_HEAD_EMPTY(&input->pos_list);
return 0;
}
///////////////////////// static structs ////////////////////////
static struct trans_logger_brick_ops trans_logger_brick_ops = {
.brick_switch = trans_logger_switch,
.brick_statistics = trans_logger_statistics,
.reset_statistics = trans_logger_reset_statistics,
};
static struct trans_logger_output_ops trans_logger_output_ops = {
.mars_get_info = trans_logger_get_info,
.mref_get = trans_logger_ref_get,
.mref_put = trans_logger_ref_put,
.mref_io = trans_logger_ref_io,
};
const struct trans_logger_input_type trans_logger_input_type = {
.type_name = "trans_logger_input",
.input_size = sizeof(struct trans_logger_input),
.input_construct = &trans_logger_input_construct,
.input_destruct = &trans_logger_input_destruct,
};
static const struct trans_logger_input_type *trans_logger_input_types[] = {
&trans_logger_input_type,
&trans_logger_input_type,
&trans_logger_input_type,
&trans_logger_input_type,
&trans_logger_input_type,
&trans_logger_input_type,
};
const struct trans_logger_output_type trans_logger_output_type = {
.type_name = "trans_logger_output",
.output_size = sizeof(struct trans_logger_output),
.master_ops = &trans_logger_output_ops,
.output_construct = &trans_logger_output_construct,
};
static const struct trans_logger_output_type *trans_logger_output_types[] = {
&trans_logger_output_type,
};
const struct trans_logger_brick_type trans_logger_brick_type = {
.type_name = "trans_logger_brick",
.brick_size = sizeof(struct trans_logger_brick),
.max_inputs = TL_INPUT_NR,
.max_outputs = 1,
.master_ops = &trans_logger_brick_ops,
.aspect_types = trans_logger_aspect_types,
.default_input_types = trans_logger_input_types,
.default_output_types = trans_logger_output_types,
.brick_construct = &trans_logger_brick_construct,
.brick_destruct = &trans_logger_brick_destruct,
};
EXPORT_SYMBOL_GPL(trans_logger_brick_type);
////////////////// module init stuff /////////////////////////
int __init init_mars_trans_logger(void)
{
MARS_INF("init_trans_logger()\n");
return trans_logger_register_brick_type();
}
void exit_mars_trans_logger(void)
{
MARS_INF("exit_trans_logger()\n");
trans_logger_unregister_brick_type();
}