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
|
/*
Unix SMB/CIFS implementation.
local test for messaging code
Copyright (C) Andrew Tridgell 2004
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include "includes.h"
static void pong_message(void *msg_ctx, void *private,
uint32_t msg_type, servid_t src, DATA_BLOB *data)
{
int *count = private;
(*count)++;
}
static void exit_message(void *msg_ctx, void *private,
uint32_t msg_type, servid_t src, DATA_BLOB *data)
{
talloc_free(private);
exit(0);
}
/*
test ping speed
*/
static BOOL test_ping_speed(TALLOC_CTX *mem_ctx)
{
struct event_context *ev = event_context_init(mem_ctx);
void *msg_ctx;
int ping_count = 0;
int pong_count = 0;
BOOL ret = True;
if (fork() == 0) {
void *msg_ctx2 = messaging_init(mem_ctx, 1, ev);
messaging_register(msg_ctx2, mem_ctx, -1, exit_message);
event_loop_wait(ev);
exit(0);
}
sleep(2);
msg_ctx = messaging_init(mem_ctx, 2, ev);
messaging_register(msg_ctx, &pong_count, MSG_PONG, pong_message);
start_timer();
printf("Sending pings for 10 seconds\n");
while (end_timer() < 10.0) {
DATA_BLOB data;
NTSTATUS status1, status2;
data.data = discard_const_p(char, "testing");
data.length = strlen(data.data);
status1 = messaging_send(msg_ctx, 1, MSG_PING, &data);
status2 = messaging_send(msg_ctx, 1, MSG_PING, NULL);
if (!NT_STATUS_IS_OK(status1)) {
printf("Failed to send msg1 (%s) (done %d)\n", nt_errstr(status1), ping_count);
} else {
ping_count++;
}
if (!NT_STATUS_IS_OK(status2)) {
printf("Failed to send msg2 (%s) (done %d)\n", nt_errstr(status2), ping_count);
} else {
ping_count++;
}
while (pong_count < ping_count) {
event_loop_once(ev);
}
}
printf("waiting for %d remaining replies (done %d)\n",
ping_count - pong_count, pong_count);
while (end_timer() < 30 && pong_count < ping_count) {
event_loop_once(ev);
}
printf("sending exit\n");
messaging_send(msg_ctx, 1, -1, NULL);
event_loop_once(ev);
if (ping_count != pong_count) {
printf("ping test failed! received %d, sent %d\n", pong_count, ping_count);
ret = False;
}
printf("ping rate of %.0f messages/sec\n", (ping_count+pong_count)/end_timer());
talloc_free(msg_ctx);
event_context_destroy(ev);
return ret;
}
BOOL torture_local_messaging(int dummy)
{
TALLOC_CTX *mem_ctx = talloc_init("torture_local_messaging");
BOOL ret = True;
ret &= test_ping_speed(mem_ctx);
talloc_free(mem_ctx);
return True;
}
|
