forked from liexusong/php-beast
-
Notifications
You must be signed in to change notification settings - Fork 0
/
spinlock.c
78 lines (67 loc) · 2.29 KB
/
spinlock.c
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
/*
+----------------------------------------------------------------------+
| PHP Version 5 |
+----------------------------------------------------------------------+
| Copyright (c) 1997-2007 The PHP Group |
+----------------------------------------------------------------------+
| This source file is subject to version 3.01 of the PHP license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.php.net/license/3_01.txt |
| If you did not receive a copy of the PHP license and are unable to |
| obtain it through the world-wide-web, please send a note to |
| [email protected] so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Liexusong <[email protected]> |
+----------------------------------------------------------------------+
*/
#include <stdlib.h>
#include "spinlock.h"
#ifdef PHP_WIN32
#include <Windows.h>
#else
#include <pthread.h>
#endif
#include "beast_log.h"
#ifdef PHP_WIN32
#define compare_and_swap(lock, o, n) \
(InterlockedCompareExchange(lock, n, o) == n)
#define pause() YieldProcessor()
#define yield() SwitchToThread()
#else
#define compare_and_swap(lock, o, n) \
__sync_bool_compare_and_swap(lock, o, n)
#ifdef __arm__
#define pause() __asm__("NOP");
#elif __aarch64__
#define pause() __asm__("NOP");
#else
#define pause() __asm__("pause")
#endif
#define yield() sched_yield()
#endif
extern int beast_ncpu;
void beast_spinlock(beast_atomic_t *lock, int pid)
{
int i, n;
for ( ;; ) {
if (compare_and_swap(lock, 0, pid)) {
return;
}
if (beast_ncpu > 1) {
for (n = 1; n < 129; n <<= 1) {
if (compare_and_swap(lock, 0, pid)) {
return;
}
for (i = 0; i < n; i++) {
pause();
}
}
}
yield();
}
}
void beast_spinunlock(beast_atomic_t *lock, int pid)
{
compare_and_swap(lock, pid, 0);
}