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 };
 111
 112 static inline struct ja_test_node *
 113 to_test_node(struct cds_ja_node *node)
 114 {
 115         return caa_container_of(node, struct ja_test_node, node);
 116 }
 117
 118 static inline
 119 void ja_test_node_init(struct ja_test_node *node, uint64_t key)
 120 {
 121         cds_ja_node_init(&node->node);
 122         node->key = key;
 123 }
 124
 125 extern volatile int test_go, test_stop;
 126
 127 extern unsigned long wdelay;
 128
 129 extern unsigned long duration;
 130
 131 /* read-side C.S. duration, in loops */
 132 extern unsigned long rduration;
 133
 134 extern unsigned long init_populate;
 135 extern int add_only;
 136
 137 extern unsigned long init_pool_offset, lookup_pool_offset, write_pool_offset;
 138 extern unsigned long init_pool_size,
 139         lookup_pool_size,
 140         write_pool_size;
 141 extern int validate_lookup;
 142
 143 extern int count_pipe[2];
 144
 145 static inline void loop_sleep(unsigned long l)
 146 {
 147         while(l-- != 0)
 148                 caa_cpu_relax();
 149 }
 150
 151 extern int verbose_mode;
 152
 153 #define printf_verbose(fmt, args...)            \
 154         do {                                    \
 155                 if (verbose_mode)               \
 156                         printf(fmt, ## args);   \
 157         } while (0)
 158
 159 extern unsigned int cpu_affinities[NR_CPUS];
 160 extern unsigned int next_aff;
 161 extern int use_affinity;
 162
 163 extern pthread_mutex_t affinity_mutex;
 164
 165 #ifndef HAVE_CPU_SET_T
 166 typedef unsigned long cpu_set_t;
 167 # define CPU_ZERO(cpuset) do { *(cpuset) = 0; } while(0)
 168 # define CPU_SET(cpu, cpuset) do { *(cpuset) |= (1UL << (cpu)); } while(0)
 169 #endif
 170
 171 void set_affinity(void);
 172
 173 /*
 174  * returns 0 if test should end.
 175  */
 176 static inline int test_duration_write(void)
 177 {
 178         return !test_stop;
 179 }
 180
 181 static inline int test_duration_read(void)
 182 {
 183         return !test_stop;
 184 }
 185
 186 extern DECLARE_URCU_TLS(unsigned long long, nr_writes);
 187 extern DECLARE_URCU_TLS(unsigned long long, nr_reads);
 188
 189 extern unsigned int nr_readers;
 190 extern unsigned int nr_writers;
 191
 192 void rcu_copy_mutex_lock(void);
 193 void rcu_copy_mutex_unlock(void);
 194
 195 #endif /* _TEST_URCU_JA_H */