s3-auth: Do not leak tmp_ctx if make_server_info() fails.
[Samba.git] / source3 / torture / msgtest.c
blobed28667b8d96f68df86bef1d0505ee30c66a1996
1 /*
2 Unix SMB/CIFS implementation.
3 Copyright (C) Andrew Tridgell 2000
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 3 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program. If not, see <http://www.gnu.org/licenses/>.
20 test code for internal messaging
23 #include "includes.h"
24 #include "messages.h"
26 static int pong_count;
29 /****************************************************************************
30 a useful function for testing the message system
31 ****************************************************************************/
32 static void pong_message(struct messaging_context *msg_ctx,
33 void *private_data,
34 uint32_t msg_type,
35 struct server_id pid,
36 DATA_BLOB *data)
38 pong_count++;
41 int main(int argc, char *argv[])
43 struct tevent_context *evt_ctx;
44 struct messaging_context *msg_ctx;
45 pid_t pid;
46 int i, n;
47 char buf[12];
48 int ret;
50 load_case_tables();
52 setup_logging(argv[0], DEBUG_STDOUT);
54 lp_load_global(get_dyn_CONFIGFILE());
56 if (!(evt_ctx = samba_tevent_context_init(NULL)) ||
57 !(msg_ctx = messaging_init(NULL, evt_ctx))) {
58 fprintf(stderr, "could not init messaging context\n");
59 exit(1);
62 if (argc != 3) {
63 fprintf(stderr, "%s: Usage - %s pid count\n", argv[0],
64 argv[0]);
65 exit(1);
68 pid = atoi(argv[1]);
69 n = atoi(argv[2]);
71 messaging_register(msg_ctx, NULL, MSG_PONG, pong_message);
73 for (i=0;i<n;i++) {
74 messaging_send(msg_ctx, pid_to_procid(pid), MSG_PING,
75 &data_blob_null);
78 while (pong_count < i) {
79 ret = tevent_loop_once(evt_ctx);
80 if (ret != 0) {
81 break;
85 /* Now test that the duplicate filtering code works. */
86 pong_count = 0;
88 strlcpy(buf, "1234567890", sizeof(buf));
90 for (i=0;i<n;i++) {
91 messaging_send(msg_ctx, messaging_server_id(msg_ctx), MSG_PING,
92 &data_blob_null);
93 messaging_send_buf(msg_ctx, messaging_server_id(msg_ctx),
94 MSG_PING,(uint8 *)buf, 11);
97 for (i=0;i<n;i++) {
98 ret = tevent_loop_once(evt_ctx);
99 if (ret != 0) {
100 break;
104 if (pong_count != 2) {
105 fprintf(stderr, "Duplicate filter failed (%d).\n", pong_count);
108 /* Speed testing */
110 pong_count = 0;
113 struct timeval tv = timeval_current();
114 size_t timelimit = n;
115 size_t ping_count = 0;
117 printf("Sending pings for %d seconds\n", (int)timelimit);
118 while (timeval_elapsed(&tv) < timelimit) {
119 if(NT_STATUS_IS_OK(messaging_send_buf(
120 msg_ctx, pid_to_procid(pid),
121 MSG_PING,
122 (uint8 *)buf, 11)))
123 ping_count++;
124 if(NT_STATUS_IS_OK(messaging_send(
125 msg_ctx, pid_to_procid(pid),
126 MSG_PING, &data_blob_null)))
127 ping_count++;
129 while (ping_count > pong_count + 20) {
130 ret = tevent_loop_once(evt_ctx);
131 if (ret != 0) {
132 break;
137 printf("waiting for %d remaining replies (done %d)\n",
138 (int)(ping_count - pong_count), pong_count);
139 while (timeval_elapsed(&tv) < 30 && pong_count < ping_count) {
140 ret = tevent_loop_once(evt_ctx);
141 if (ret != 0) {
142 break;
146 if (ping_count != pong_count) {
147 fprintf(stderr, "ping test failed! received %d, sent "
148 "%d\n", pong_count, (int)ping_count);
151 printf("ping rate of %.0f messages/sec\n",
152 (ping_count+pong_count)/timeval_elapsed(&tv));
155 return (0);