tests/test_urcu_ja.h

   1 #ifndef _TEST_URCU_JA_H
   2 #define _TEST_URCU_JA_H
   3
   4 /*
   5  * test_urcu_ja.h
   6  *
   7  * Userspace RCU library - test program
   8  *
   9  * Copyright 2009-2012 - Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License along
  22  * with this program; if not, write to the Free Software Foundation, Inc.,
  23  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  24  */
  25
  26 #include "../config.h"
  27 #include <stdio.h>
  28 #include <pthread.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <sys/types.h>
  32 #include <sys/wait.h>
  33 #include <unistd.h>
  34 #include <stdio.h>
  35 #include <assert.h>
  36 #include <sched.h>
  37 #include <errno.h>
  38 #include <signal.h>
  39
  40 #include <urcu/tls-compat.h>
  41
  42 #ifdef __linux__
  43 #include <syscall.h>
  44 #endif
  45
  46 #define DEFAULT_RAND_POOL       1000000
  47
  48 /* Make this big enough to include the POWER5+ L3 cacheline size of 256B */
  49 #define CACHE_LINE_SIZE 4096
  50
  51 /* hardcoded number of CPUs */
  52 #define NR_CPUS 16384
  53
  54 #ifdef POISON_FREE
  55 #define poison_free(ptr)                                \
  56         do {                                            \
  57                 memset(ptr, 0x42, sizeof(*(ptr)));      \
  58                 free(ptr);                              \
  59         } while (0)
  60 #else
  61 #define poison_free(ptr)        free(ptr)
  62 #endif
  63
  64
  65
  66 #if defined(_syscall0)
  67 _syscall0(pid_t, gettid)
  68 #elif defined(__NR_gettid)
  69 static inline pid_t gettid(void)
  70 {
  71         return syscall(__NR_gettid);
  72 }
  73 #else
  74 #warning "use pid as tid"
  75 static inline pid_t gettid(void)
  76 {
  77         return getpid();
  78 }
  79 #endif
  80
  81 #ifndef DYNAMIC_LINK_TEST
  82 #define _LGPL_SOURCE
  83 #else
  84 #define debug_yield_read()
  85 #endif
  86 #include <urcu-qsbr.h>
  87 #include <urcu/rcuja.h>
  88 #include <urcu-call-rcu.h>
  89
  90 struct wr_count {
  91         unsigned long update_ops;
  92         unsigned long add;
  93         unsigned long add_exist;
  94         unsigned long remove;
  95 };
  96
  97 extern DECLARE_URCU_TLS(unsigned int, rand_lookup);
  98 extern DECLARE_URCU_TLS(unsigned long, nr_add);
  99 extern DECLARE_URCU_TLS(unsigned long, nr_addexist);
 100 extern DECLARE_URCU_TLS(unsigned long, nr_del);
 101 extern DECLARE_URCU_TLS(unsigned long, nr_delnoent);
 102 extern DECLARE_URCU_TLS(unsigned long, lookup_fail);
 103 extern DECLARE_URCU_TLS(unsigned long, lookup_ok);
 104
 105 extern struct cds_ja *test_ja;
 106
 107 struct ja_test_node {
 108         struct cds_ja_node node;
 109         uint64_t key;           /* for testing */
 110         struct rcu_head head;   /* delayed reclaim */
 111 };
 112
 113 static inline struct ja_test_node *
 114 to_test_node(struct cds_ja_node *node)
 115 {
 116         return caa_container_of(node, struct ja_test_node, node);
 117 }
 118
 119 static inline
 120 void ja_test_node_init(struct ja_test_node *node, uint64_t key)
 121 {
 122         cds_ja_node_init(&node->node);
 123         node->key = key;
 124 }
 125
 126 extern volatile int test_go, test_stop;
 127
 128 extern unsigned long wdelay;
 129
 130 extern unsigned long duration;
 131
 132 /* read-side C.S. duration, in loops */
 133 extern unsigned long rduration;
 134
 135 extern unsigned long init_populate;
 136 extern int add_only;
 137
 138 extern unsigned long init_pool_offset, lookup_pool_offset, write_pool_offset;
 139 extern unsigned long init_pool_size,
 140         lookup_pool_size,
 141         write_pool_size;
 142 extern int validate_lookup;
 143
 144 extern int count_pipe[2];
 145
 146 static inline void loop_sleep(unsigned long l)
 147 {
 148         while(l-- != 0)
 149                 caa_cpu_relax();
 150 }
 151
 152 extern int verbose_mode;
 153
 154 #define printf_verbose(fmt, args...)            \
 155         do {                                    \
 156                 if (verbose_mode)               \
 157                         printf(fmt, ## args);   \
 158         } while (0)
 159
 160 extern unsigned int cpu_affinities[NR_CPUS];
 161 extern unsigned int next_aff;
 162 extern int use_affinity;
 163
 164 extern pthread_mutex_t affinity_mutex;
 165
 166 #ifndef HAVE_CPU_SET_T
 167 typedef unsigned long cpu_set_t;
 168 # define CPU_ZERO(cpuset) do { *(cpuset) = 0; } while(0)
 169 # define CPU_SET(cpu, cpuset) do { *(cpuset) |= (1UL << (cpu)); } while(0)
 170 #endif
 171
 172 void set_affinity(void);
 173
 174 /*
 175  * returns 0 if test should end.
 176  */
 177 static inline int test_duration_write(void)
 178 {
 179         return !test_stop;
 180 }
 181
 182 static inline int test_duration_read(void)
 183 {
 184         return !test_stop;
 185 }
 186
 187 extern DECLARE_URCU_TLS(unsigned long long, nr_writes);
 188 extern DECLARE_URCU_TLS(unsigned long long, nr_reads);
 189
 190 extern unsigned int nr_readers;
 191 extern unsigned int nr_writers;
 192
 193 void rcu_copy_mutex_lock(void);
 194 void rcu_copy_mutex_unlock(void);
 195
 196 #endif /* _TEST_URCU_JA_H */