3rd-party/linux-kernel-module-cheat
1
0
Fork 0
mirror of https://github.com/cirosantilli/linux-kernel-module-cheat.git synced 2026-01-23 02:05:57 +01:00
Code Issues Releases Wiki Activity

cpp atomic: define to 1 which is saner

Browse Source
This commit is contained in:
Ciro Santilli 六四事件 法轮功
2019-11-15 00:00:00 +00:00
parent 175b0e3d50
commit d82e47e9d2
9 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
userland/cpp/atomic/aarch64_add.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_AARCH64_ADD
#define LKMC_USERLAND_ATOMIC_AARCH64_ADD 1
#include "main.hpp"

2
userland/cpp/atomic/aarch64_ldadd.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_AARCH64_LDADD
#define LKMC_USERLAND_ATOMIC_AARCH64_LDADD 1
#include "main.hpp"

2
userland/cpp/atomic/fail.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_FAIL
#define LKMC_USERLAND_ATOMIC_FAIL 1
#include "main.hpp"

22
userland/cpp/atomic/main.hpp
View File

@@ -10,29 +10,29 @@
size_t niters;
#if defined(LKMC_USERLAND_ATOMIC_STD_ATOMIC)
#if LKMC_USERLAND_ATOMIC_STD_ATOMIC
std::atomic_ulong global(0);
#else
uint64_t global = 0;
#endif
#if defined(LKMC_USERLAND_ATOMIC_MUTEX)
#if LKMC_USERLAND_ATOMIC_MUTEX
std::mutex mutex;
#endif
void threadMain() {
for (size_t i = 0; i < niters; ++i) {
#if defined(LKMC_USERLAND_ATOMIC_MUTEX)
#if LKMC_USERLAND_ATOMIC_MUTEX
mutex.lock();
#endif
#if defined(LKMC_USERLAND_ATOMIC_X86_64_INC)
#if LKMC_USERLAND_ATOMIC_X86_64_INC
__asm__ __volatile__ (
"incq %0;"
: "+g" (global)
:
:
);
#elif defined(LKMC_USERLAND_ATOMIC_X86_64_LOCK_INC)
#elif LKMC_USERLAND_ATOMIC_X86_64_LOCK_INC
// https://cirosantilli.com/linux-kernel-module-cheat#x86-lock-prefix
__asm__ __volatile__ (
"lock;"
@@ -41,14 +41,14 @@ void threadMain() {
:
:
);
#elif defined(LKMC_USERLAND_ATOMIC_AARCH64_ADD)
#elif LKMC_USERLAND_ATOMIC_AARCH64_ADD
__asm__ __volatile__ (
"add %0, %0, 1;"
: "+r" (global)
:
:
);
#elif defined(LKMC_USERLAND_ATOMIC_AARCH64_LDADD)
#elif LKMC_USERLAND_ATOMIC_AARCH64_LDADD
// https://cirosantilli.com/linux-kernel-module-cheat#arm-lse
__asm__ __volatile__ (
"ldadd %[inc], xzr, [%[addr]];"
@@ -60,7 +60,7 @@ void threadMain() {
#else
global++;
#endif
#if defined(LKMC_USERLAND_ATOMIC_MUTEX)
#if LKMC_USERLAND_ATOMIC_MUTEX
mutex.unlock();
#endif
#if 0
@@ -88,9 +88,9 @@ int main(int argc, char **argv) {
for (size_t i = 0; i < nthreads; ++i)
threads[i].join();
uint64_t expect = nthreads * niters;
#if defined(LKMC_USERLAND_ATOMIC_FAIL) || \
defined(LKMC_USERLAND_ATOMIC_X86_64_INC) || \
defined(LKMC_USERLAND_ATOMIC_AARCH64_INC)
#if LKMC_USERLAND_ATOMIC_FAIL || \
LKMC_USERLAND_ATOMIC_X86_64_INC || \
LKMC_USERLAND_ATOMIC_AARCH64_INC
// These fail, so we just print the outcomes.
std::cout << "expect " << expect << std::endl;
std::cout << "global " << global << std::endl;

2
userland/cpp/atomic/mutex.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_MUTEX
#define LKMC_USERLAND_ATOMIC_MUTEX 1
#include "main.hpp"

2
userland/cpp/atomic/std_atomic.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_STD_ATOMIC
#define LKMC_USERLAND_ATOMIC_STD_ATOMIC 1
#include "main.hpp"

2
userland/cpp/atomic/x86_64_inc.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_X86_64_INC
#define LKMC_USERLAND_ATOMIC_X86_64_INC 1
#include "main.hpp"

2
userland/cpp/atomic/x86_64_lock_inc.cpp
View File

@@ -1,2 +1,2 @@
#define LKMC_USERLAND_ATOMIC_X86_64_LOCK_INC
#define LKMC_USERLAND_ATOMIC_X86_64_LOCK_INC 1
#include "main.hpp"