blob: 1077f4c60a9216a57bcac0f37247b5efaab10d3f (
plain)
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
|
/* probe locking header file
* Copyright (C) 2009 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 _PROBE_LOCK_H
#define _PROBE_LOCK_H
#include <linux/spinlock.h>
// XXX: old 2.6 kernel hack
#ifndef read_trylock
#define read_trylock(x) ({ read_lock(x); 1; })
#endif
struct stp_probe_lock {
#ifdef STP_TIMING
atomic_t *skipped;
#endif
rwlock_t *lock;
unsigned write_p;
};
static void
stp_unlock_probe(const struct stp_probe_lock *locks, unsigned num_locks)
{
unsigned i;
for (i = num_locks; i-- > 0;) {
if (locks[i].write_p)
write_unlock(locks[i].lock);
else
read_unlock(locks[i].lock);
}
}
static unsigned
stp_lock_probe(const struct stp_probe_lock *locks, unsigned num_locks)
{
unsigned i, retries = 0;
for (i = 0; i < num_locks; ++i) {
if (locks[i].write_p)
while (!write_trylock(locks[i].lock)) {
if (++retries > MAXTRYLOCK)
goto skip;
ndelay (TRYLOCKDELAY);
}
else
while (!read_trylock(locks[i].lock)) {
if (++retries > MAXTRYLOCK)
goto skip;
ndelay (TRYLOCKDELAY);
}
}
return 1;
skip:
atomic_inc(&skipped_count);
#ifdef STP_TIMING
atomic_inc(locks[i].skipped);
#endif
stp_unlock_probe(locks, i);
return 0;
}
#endif /* _PROBE_LOCK_H */
|