mirror of https://github.com/schoebel/mars
110 lines
3.1 KiB
C
110 lines
3.1 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.
|
|
*/
|
|
|
|
#ifndef BRICK_ATOMIC_H
|
|
#define BRICK_ATOMIC_H
|
|
|
|
#include <linux/spinlock.h>
|
|
|
|
#include <asm/atomic.h>
|
|
|
|
#include "brick_checking.h"
|
|
|
|
#define ATOMIC_DEBUGGING
|
|
|
|
#ifndef CONFIG_MARS_DEBUG
|
|
#undef ATOMIC_DEBUGGING
|
|
#endif
|
|
|
|
#define ATOMIC_MAXTRACE 32
|
|
|
|
/* Trivial wrapper to force type checking.
|
|
*/
|
|
typedef struct tatomic {
|
|
atomic_t ta_atomic;
|
|
} tatomic_t;
|
|
|
|
typedef struct atomic_trace {
|
|
#ifdef ATOMIC_DEBUGGING
|
|
atomic_t at_count;
|
|
short at_lines[ATOMIC_MAXTRACE];
|
|
pid_t at_pids[ATOMIC_MAXTRACE];
|
|
const char *at_sources[ATOMIC_MAXTRACE];
|
|
#endif
|
|
} atomic_trace_t;
|
|
|
|
#ifdef ATOMIC_DEBUGGING
|
|
|
|
#define tatomic_trace(_at, _cmd) \
|
|
({ \
|
|
int _index = atomic_add_return(1, &(_at)->at_count) - 1; \
|
|
if (likely(_index >= 0 && _index < ATOMIC_MAXTRACE)) { \
|
|
(_at)->at_lines[_index] = __LINE__; \
|
|
(_at)->at_pids[_index] = current->pid; \
|
|
(_at)->at_sources[_index] = __BASE_FILE__; \
|
|
} \
|
|
_cmd; \
|
|
})
|
|
|
|
#define tatomic_out(_at,_MSG) \
|
|
({ \
|
|
int __i; \
|
|
int __max = atomic_read(&(_at)->at_count); \
|
|
_MSG("at_count = %d\n", __max); \
|
|
if (unlikely(__max > ATOMIC_MAXTRACE)) \
|
|
__max = ATOMIC_MAXTRACE; \
|
|
for (__i = 0; __i < __max; __i++) { \
|
|
_MSG("%2d pid=%d %s:%d\n", __i, (_at)->at_pids[__i], (_at)->at_sources[__i], (_at)->at_lines[__i]); \
|
|
} \
|
|
})
|
|
|
|
#define _CHECK_TATOMIC(_at,_atom,OP,_minval,_fixval) \
|
|
do { \
|
|
if (BRICK_CHECKING) { \
|
|
int __test = atomic_read(&(_atom)->ta_atomic); \
|
|
if (unlikely(__test OP (_minval))) { \
|
|
atomic_set(&(_atom)->ta_atomic, _fixval); \
|
|
BRICK_ERR("%d: tatomic " #_atom " " #OP " " #_minval " (%d)\n", __LINE__, __test); \
|
|
tatomic_out(_at, BRICK_ERR); \
|
|
} \
|
|
} \
|
|
} while (0)
|
|
|
|
#else
|
|
|
|
#define tatomic_trace(_at,_cmd) _cmd
|
|
|
|
#define _CHECK_TATOMIC(_at,_atom,OP,_minval,_fixval) \
|
|
_CHECK_ATOMIC(&(_atom)->ta_atomic, OP, _minval)
|
|
|
|
#endif
|
|
|
|
#define CHECK_TATOMIC(_at,_atom,_minval) \
|
|
_CHECK_TATOMIC(_at, _atom, <, _minval, _minval)
|
|
|
|
#define tatomic_inc(at,a) tatomic_trace(at, atomic_inc(&(a)->ta_atomic))
|
|
#define tatomic_dec(at,a) tatomic_trace(at, atomic_dec(&(a)->ta_atomic))
|
|
#define tatomic_dec_and_test(at,a) tatomic_trace(at, atomic_dec_and_test(&(a)->ta_atomic))
|
|
|
|
#endif
|