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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#include <unistd.h>
#include <pthread.h>
#include <stdio.h>
#include <sched.h>
#include <errno.h>
#include <semaphore.h>
#include <spa/utils/ringbuffer.h>
#define DEFAULT_SIZE 0x2000
#define ARRAY_SIZE 63
#define MAX_VALUE 0x10000
#if defined(__FreeBSD__) || defined(__MidnightBSD__)
#include <sys/param.h>
#if (__FreeBSD_version >= 1400000 && __FreeBSD_version < 1400043) \
|| (__FreeBSD_version < 1300523) || defined(__MidnightBSD__)
static int sched_getcpu(void) { return -1; };
#endif
#endif
static struct spa_ringbuffer rb;
static uint32_t size;
static void *data;
static sem_t sem;
static int fill_int_array(int *array, int start, int count)
{
int i, j = start;
for (i = 0; i < count; i++) {
array[i] = j;
j = (j + 1) % MAX_VALUE;
}
return j;
}
static int cmp_array(int *array1, int *array2, int count)
{
int i;
for (i = 0; i < count; i++)
if (array1[i] != array2[i]) {
printf("%d != %d at offset %d\n", array1[i], array2[i], i);
return 0;
}
return 1;
}
static void *reader_start(void *arg)
{
int i = 0, a[ARRAY_SIZE], b[ARRAY_SIZE];
printf("reader started on cpu: %d\n", sched_getcpu());
i = fill_int_array(a, i, ARRAY_SIZE);
while (1) {
uint32_t index;
int32_t avail;
avail = spa_ringbuffer_get_read_index(&rb, &index);
if (avail >= (int32_t)(sizeof(b))) {
spa_ringbuffer_read_data(&rb, data, size, index % size, b, sizeof(b));
spa_ringbuffer_read_update(&rb, index + sizeof(b));
if (index >= INT32_MAX - sizeof(a))
break;
spa_assert(cmp_array(a, b, ARRAY_SIZE));
i = fill_int_array(a, i, ARRAY_SIZE);
}
}
sem_post(&sem);
return NULL;
}
static void *writer_start(void *arg)
{
int i = 0, a[ARRAY_SIZE];
printf("writer started on cpu: %d\n", sched_getcpu());
i = fill_int_array(a, i, ARRAY_SIZE);
while (1) {
uint32_t index;
int32_t avail;
avail = size - spa_ringbuffer_get_write_index(&rb, &index);
if (avail >= (int32_t)(sizeof(a))) {
spa_ringbuffer_write_data(&rb, data, size, index % size, a, sizeof(a));
spa_ringbuffer_write_update(&rb, index + sizeof(a));
if (index >= INT32_MAX - sizeof(a))
break;
i = fill_int_array(a, i, ARRAY_SIZE);
}
}
sem_post(&sem);
return NULL;
}
#define exit_error(msg) \
do { perror(msg); exit(EXIT_FAILURE); } while (0)
int main(int argc, char *argv[])
{
pthread_t reader_thread, writer_thread;
struct timespec ts;
printf("starting ringbuffer stress test\n");
if (argc > 1)
sscanf(argv[1], "%d", &size);
else
size = DEFAULT_SIZE;
printf("buffer size (bytes): %d\n", size);
printf("array size (bytes): %zd\n", sizeof(int) * ARRAY_SIZE);
spa_ringbuffer_init(&rb);
data = malloc(size);
if (sem_init(&sem, 0, 0) != 0)
exit_error("init_sem");
pthread_create(&reader_thread, NULL, reader_start, NULL);
pthread_create(&writer_thread, NULL, writer_start, NULL);
if (clock_gettime(CLOCK_REALTIME, &ts) != 0)
exit_error("clock_gettime");
ts.tv_sec += 2;
while (sem_timedwait(&sem, &ts) == -1 && errno == EINTR)
continue;
while (sem_timedwait(&sem, &ts) == -1 && errno == EINTR)
continue;
printf("read %u, written %u\n", rb.readindex, rb.writeindex);
return 0;
}
|
