You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
3.1 KiB
136 lines
3.1 KiB
#include <ck_bytelock.h>
|
|
#include <ck_spinlock.h>
|
|
#include <inttypes.h>
|
|
#include <stdio.h>
|
|
|
|
#include "../../common.h"
|
|
|
|
#ifndef STEPS
|
|
#define STEPS 1000000
|
|
#endif
|
|
|
|
/*
|
|
* This is a naive reader/writer spinlock.
|
|
*/
|
|
struct rwlock {
|
|
unsigned int readers;
|
|
ck_spinlock_fas_t writer;
|
|
};
|
|
typedef struct rwlock rwlock_t;
|
|
|
|
static CK_CC_INLINE void
|
|
rwlock_init(rwlock_t *rw)
|
|
{
|
|
|
|
ck_pr_store_uint(&rw->readers, 0);
|
|
ck_spinlock_fas_init(&rw->writer);
|
|
return;
|
|
}
|
|
|
|
static CK_CC_INLINE void
|
|
rwlock_write_lock(rwlock_t *rw)
|
|
{
|
|
|
|
ck_spinlock_fas_lock(&rw->writer);
|
|
while (ck_pr_load_uint(&rw->readers) != 0)
|
|
ck_pr_stall();
|
|
|
|
return;
|
|
}
|
|
|
|
static CK_CC_INLINE void
|
|
rwlock_write_unlock(rwlock_t *rw)
|
|
{
|
|
|
|
ck_spinlock_fas_unlock(&rw->writer);
|
|
return;
|
|
}
|
|
|
|
static CK_CC_INLINE void
|
|
rwlock_read_lock(rwlock_t *rw)
|
|
{
|
|
|
|
for (;;) {
|
|
while (ck_pr_load_uint(&rw->writer.value) != 0)
|
|
ck_pr_stall();
|
|
|
|
ck_pr_inc_uint(&rw->readers);
|
|
if (ck_pr_load_uint(&rw->writer.value) == 0)
|
|
break;
|
|
ck_pr_dec_uint(&rw->readers);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
static CK_CC_INLINE void
|
|
rwlock_read_unlock(rwlock_t *rw)
|
|
{
|
|
|
|
ck_pr_dec_uint(&rw->readers);
|
|
return;
|
|
}
|
|
|
|
int
|
|
main(void)
|
|
{
|
|
uint64_t s_b, e_b, i;
|
|
ck_bytelock_t bytelock = CK_BYTELOCK_INITIALIZER;
|
|
rwlock_t naive;
|
|
|
|
for (i = 0; i < STEPS; i++) {
|
|
ck_bytelock_write_lock(&bytelock, 1);
|
|
ck_bytelock_write_unlock(&bytelock);
|
|
}
|
|
|
|
s_b = rdtsc();
|
|
for (i = 0; i < STEPS; i++) {
|
|
ck_bytelock_write_lock(&bytelock, 1);
|
|
ck_bytelock_write_unlock(&bytelock);
|
|
}
|
|
e_b = rdtsc();
|
|
printf("WRITE: bytelock %15" PRIu64 "\n", (e_b - s_b) / STEPS);
|
|
|
|
rwlock_init(&naive);
|
|
for (i = 0; i < STEPS; i++) {
|
|
rwlock_write_lock(&naive);
|
|
rwlock_write_unlock(&naive);
|
|
}
|
|
|
|
s_b = rdtsc();
|
|
for (i = 0; i < STEPS; i++) {
|
|
rwlock_write_lock(&naive);
|
|
rwlock_write_unlock(&naive);
|
|
}
|
|
e_b = rdtsc();
|
|
printf("WRITE: naive %15" PRIu64 "\n", (e_b - s_b) / STEPS);
|
|
|
|
for (i = 0; i < STEPS; i++) {
|
|
ck_bytelock_read_lock(&bytelock, 1);
|
|
ck_bytelock_read_unlock(&bytelock, 1);
|
|
}
|
|
s_b = rdtsc();
|
|
for (i = 0; i < STEPS; i++) {
|
|
ck_bytelock_read_lock(&bytelock, 1);
|
|
ck_bytelock_read_unlock(&bytelock, 1);
|
|
}
|
|
e_b = rdtsc();
|
|
printf("READ: bytelock %15" PRIu64 "\n", (e_b - s_b) / STEPS);
|
|
|
|
for (i = 0; i < STEPS; i++) {
|
|
rwlock_read_lock(&naive);
|
|
rwlock_read_unlock(&naive);
|
|
}
|
|
|
|
s_b = rdtsc();
|
|
for (i = 0; i < STEPS; i++) {
|
|
rwlock_read_lock(&naive);
|
|
rwlock_read_unlock(&naive);
|
|
}
|
|
e_b = rdtsc();
|
|
printf("READ: naive %15" PRIu64 "\n", (e_b - s_b) / STEPS);
|
|
|
|
return (0);
|
|
}
|
|
|