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
|
/*
This file is part of GNUnet.
Copyright (C) 2018 GNUnet e.V.
GNUnet is free software: you can redistribute it and/or modify it
under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
GNUnet is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
/**
* @file util/benchmark.c
* @brief benchmarking for various operations
* @author Florian Dold <flo@dold.me>
*/
#include "platform.h"
#include "gnunet_util_lib.h"
#include "benchmark.h"
#include <pthread.h>
#include <sys/syscall.h>
/**
* Thread-local storage key for the benchmark data.
*/
static pthread_key_t key;
/**
* One-time initialization marker for key.
*/
static pthread_once_t key_once = PTHREAD_ONCE_INIT;
/**
* Write benchmark data to a file.
*
* @param bd the benchmark data
*/
static void
write_benchmark_data (struct BenchmarkData *bd)
{
struct GNUNET_DISK_FileHandle *fh;
pid_t pid = getpid ();
pid_t tid = syscall (SYS_gettid);
char *s;
GNUNET_asprintf (&s, "gnunet-benchmark-%llu-%llu.txt",
(unsigned long long) pid,
(unsigned long long) tid);
fh = GNUNET_DISK_file_open (s,
(GNUNET_DISK_OPEN_WRITE |
GNUNET_DISK_OPEN_TRUNCATE |
GNUNET_DISK_OPEN_CREATE),
(GNUNET_DISK_PERM_USER_READ |
GNUNET_DISK_PERM_USER_WRITE));
GNUNET_assert (NULL != fh);
GNUNET_free (s);
GNUNET_asprintf (&s, "eddsa_sign_count %llu",
(unsigned long long) bd->eddsa_sign_count);
GNUNET_assert (GNUNET_SYSERR != GNUNET_DISK_file_write_blocking (fh, s, strlen (s)));
GNUNET_free (s);
GNUNET_assert (GNUNET_OK == GNUNET_DISK_file_close (fh));
}
/**
* Called when the main thread exits and benchmark data for it was created.
*/
static void
main_thread_destructor ()
{
struct BenchmarkData *bd;
bd = pthread_getspecific (key);
if (NULL != bd)
write_benchmark_data (bd);
}
/**
* Called when a thread exits and benchmark data for it was created.
*
* @param cls closure
*/
static void
thread_destructor (void *cls)
{
struct BenchmarkData *bd = cls;
// main thread will be handled by atexit
if (getpid () == (pid_t) syscall (SYS_gettid))
return;
GNUNET_assert (NULL != bd);
}
/**
* Initialize the thread-local variable key for benchmark data.
*/
static void
make_key()
{
(void) pthread_key_create (&key, &thread_destructor);
}
/**
* Acquire the benchmark data for the current thread, allocate if necessary.
* Installs handler to collect the benchmark data on thread termination.
*
* @return benchmark data for the current thread
*/
struct BenchmarkData *
get_benchmark_data (void)
{
struct BenchmarkData *bd;
(void) pthread_once (&key_once, &make_key);
if (NULL == (bd = pthread_getspecific (key)))
{
bd = GNUNET_new (struct BenchmarkData);
(void) pthread_setspecific (key, bd);
if (getpid () == (pid_t) syscall (SYS_gettid))
{
// We're the main thread!
atexit (main_thread_destructor);
}
}
return bd;
}
|
