1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/* -*- linux-c -*-
* Locking helper function api to support preempt-rt variant raw locks
* and keep legacy locking compatibility intact.
*
* Author: Santosh Shukla <sshukla@mvista.com>
*
* Copyright (C) 2014 Red Hat Inc.
*
* This file is part of systemtap, and is free software. You can
* redistribute it and/or modify it under the terms of the GNU General
* Public License (GPL); either version 2, or (at your option) any
* later version.
* */
#ifndef _STP_HELPER_LOCK_H_
#define _STP_HELPER_LOCK_H_
#include <linux/spinlock.h>
#ifndef STAP_SPIN_TRYLOCK_MAX_COUNT
#define STAP_SPIN_TRYLOCK_MAX_COUNT 5000
#endif
#define stp_nmi_spin_lock_irqsave(lock, flags, label) \
if (in_nmi()) { \
long i; \
for (i = 0; i < STAP_SPIN_TRYLOCK_MAX_COUNT; i++) { \
if (stp_spin_trylock(lock)) \
break; \
} \
if (unlikely(i >= STAP_SPIN_TRYLOCK_MAX_COUNT)) { \
goto label; \
} \
} else { \
stp_spin_lock_irqsave(lock, flags); \
}
#define stp_nmi_spin_unlock_irqrestore(lock, flags) \
if (in_nmi()) { \
stp_spin_unlock(lock); \
} else { \
stp_spin_unlock_irqrestore(lock, flags); \
}
#if defined(CONFIG_PREEMPT_RT_FULL) || defined(CONFIG_PREEMPT_RT)
#define stp_spinlock_t raw_spinlock_t
#define STP_DEFINE_SPINLOCK(lock) DEFINE_RAW_SPINLOCK(lock)
static inline void stp_spin_lock_init(raw_spinlock_t *lock) { raw_spin_lock_init(lock); }
static inline void stp_spin_lock(raw_spinlock_t *lock) { raw_spin_lock(lock); }
static inline void stp_spin_unlock(raw_spinlock_t *lock) { raw_spin_unlock(lock); }
#define stp_spin_trylock(lock) raw_spin_trylock(lock)
#define stp_spin_lock_irqsave(lock, flags) raw_spin_lock_irqsave(lock, flags)
#define stp_spin_unlock_irqrestore(lock, flags) raw_spin_unlock_irqrestore(lock, flags)
#define stp_rwlock_t raw_spinlock_t
#define STP_DEFINE_RWLOCK(lock) DEFINE_RAW_SPINLOCK(lock)
static inline void stp_rwlock_init(raw_spinlock_t *lock) { raw_spin_lock_init(lock); }
static inline void stp_read_lock(raw_spinlock_t *lock) { raw_spin_lock(lock); }
static inline void stp_read_unlock(raw_spinlock_t *lock) { raw_spin_unlock(lock); }
static inline void stp_write_lock(raw_spinlock_t *lock) { raw_spin_lock(lock); }
static inline void stp_write_unlock(raw_spinlock_t *lock) { raw_spin_unlock(lock); }
static inline int stp_read_trylock(raw_spinlock_t *lock) { return raw_spin_trylock(lock); }
static inline int stp_write_trylock(raw_spinlock_t *lock) { return raw_spin_trylock(lock); }
#define stp_read_lock_irqsave(lock, flags) raw_spin_lock_irqsave(lock, flags)
#define stp_read_unlock_irqrestore(lock, flags) raw_spin_unlock_irqrestore(lock, flags)
#define stp_write_lock_irqsave(lock, flags) raw_spin_lock_irqsave(lock, flags)
#define stp_write_unlock_irqrestore(lock, flags) raw_spin_unlock_irqrestore(lock, flags)
#else
#define stp_spinlock_t spinlock_t
#define STP_DEFINE_SPINLOCK(lock) DEFINE_SPINLOCK(lock)
static inline void stp_spin_lock_init(spinlock_t *lock) { spin_lock_init(lock); }
static inline void stp_spin_lock(spinlock_t *lock) { spin_lock(lock); }
static inline void stp_spin_unlock(spinlock_t *lock) { spin_unlock(lock); }
#define stp_spin_trylock(lock) spin_trylock(lock)
#define stp_spin_lock_irqsave(lock, flags) spin_lock_irqsave(lock, flags)
#define stp_spin_unlock_irqrestore(lock, flags) spin_unlock_irqrestore(lock, flags)
#define stp_rwlock_t rwlock_t
#define STP_DEFINE_RWLOCK(lock) DEFINE_RWLOCK(lock)
static inline void stp_rwlock_init(rwlock_t *lock) { rwlock_init(lock); }
static inline void stp_read_lock(rwlock_t *lock) { read_lock(lock); }
static inline void stp_read_unlock(rwlock_t *lock) { read_unlock(lock); }
static inline void stp_write_lock(rwlock_t *lock) { write_lock(lock); }
static inline void stp_write_unlock(rwlock_t *lock) { write_unlock(lock); }
static inline int stp_read_trylock(rwlock_t *lock) { return read_trylock(lock); }
static inline int stp_write_trylock(rwlock_t *lock) { return write_trylock(lock); }
#define stp_read_lock_irqsave(lock, flags) read_lock_irqsave(lock, flags)
#define stp_read_unlock_irqrestore(lock, flags) read_unlock_irqrestore(lock, flags)
#define stp_write_lock_irqsave(lock, flags) write_lock_irqsave(lock, flags)
#define stp_write_unlock_irqrestore(lock, flags) write_unlock_irqrestore(lock, flags)
#endif
#endif /* _STP_HELPER_LOCK_H_ */
|
