source4/torture/local/messaging.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    local test for messaging code
   5
   6    Copyright (C) Andrew Tridgell 2004
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23 #include "includes.h"
  24 #include "system/filesys.h"
  25 #include "lib/events/events.h"
  26 #include "lib/messaging/irpc.h"
  27
  28 enum {MY_PING=1000, MY_PONG, MY_EXIT};
  29
  30 static void ping_message(struct messaging_context *msg, void *private,
  31                          uint32_t msg_type, uint32_t src, DATA_BLOB *data)
  32 {
  33         NTSTATUS status;
  34         status = messaging_send(msg, src, MY_PONG, data);
  35         if (!NT_STATUS_IS_OK(status)) {
  36                 printf("pong failed - %s\n", nt_errstr(status));
  37         }
  38 }
  39
  40 static void pong_message(struct messaging_context *msg, void *private,
  41                          uint32_t msg_type, uint32_t src, DATA_BLOB *data)
  42 {
  43         int *count = private;
  44         (*count)++;
  45 }
  46
  47 static void exit_message(struct messaging_context *msg, void *private,
  48                          uint32_t msg_type, uint32_t src, DATA_BLOB *data)
  49 {
  50         talloc_free(private);
  51         exit(0);
  52 }
  53
  54 /*
  55   test ping speed
  56 */
  57 static BOOL test_ping_speed(TALLOC_CTX *mem_ctx)
  58 {
  59         struct event_context *ev;
  60         struct messaging_context *msg_ctx;
  61         struct messaging_context *msg_ctx2;
  62         int ping_count = 0;
  63         int pong_count = 0;
  64         BOOL ret = True;
  65         struct timeval tv;
  66
  67         lp_set_cmdline("lock dir", "lockdir.tmp");
  68
  69         ev = event_context_init(mem_ctx);
  70
  71         msg_ctx2 = messaging_init(mem_ctx, 1, ev);
  72
  73         if (!msg_ctx2) {
  74                 exit(1);
  75         }
  76
  77         messaging_register(msg_ctx2, NULL, MY_PING, ping_message);
  78         messaging_register(msg_ctx2, mem_ctx, MY_EXIT, exit_message);
  79
  80         msg_ctx = messaging_init(mem_ctx, 2, ev);
  81
  82         if (!msg_ctx) {
  83                 printf("messaging_init() failed\n");
  84                 return False;
  85         }
  86
  87         messaging_register(msg_ctx, &pong_count, MY_PONG, pong_message);
  88
  89         tv = timeval_current();
  90
  91         printf("Sending pings for 10 seconds\n");
  92         while (timeval_elapsed(&tv) < 10.0) {
  93                 DATA_BLOB data;
  94                 NTSTATUS status1, status2;
  95
  96                 data.data = discard_const_p(uint8_t, "testing");
  97                 data.length = strlen((const char *)data.data);
  98
  99                 status1 = messaging_send(msg_ctx, 1, MY_PING, &data);
 100                 status2 = messaging_send(msg_ctx, 1, MY_PING, NULL);
 101
 102                 if (!NT_STATUS_IS_OK(status1)) {
 103                         printf("msg1 failed - %s\n", nt_errstr(status1));
 104                 } else {
 105                         ping_count++;
 106                 }
 107
 108                 if (!NT_STATUS_IS_OK(status2)) {
 109                         printf("msg2 failed - %s\n", nt_errstr(status2));
 110                 } else {
 111                         ping_count++;
 112                 }
 113
 114                 while (ping_count > pong_count + 20) {
 115                         event_loop_once(ev);
 116                 }
 117         }
 118
 119         printf("waiting for %d remaining replies (done %d)\n",
 120                ping_count - pong_count, pong_count);
 121         while (timeval_elapsed(&tv) < 30 && pong_count < ping_count) {
 122                 event_loop_once(ev);
 123         }
 124
 125         printf("sending exit\n");
 126         messaging_send(msg_ctx, 1, MY_EXIT, NULL);
 127
 128         if (ping_count != pong_count) {
 129                 printf("ping test failed! received %d, sent %d\n",
 130                        pong_count, ping_count);
 131                 ret = False;
 132         }
 133
 134         printf("ping rate of %.0f messages/sec\n",
 135                (ping_count+pong_count)/timeval_elapsed(&tv));
 136
 137         talloc_free(msg_ctx);
 138
 139         talloc_free(ev);
 140
 141         return ret;
 142 }
 143
 144 BOOL torture_local_messaging(void)
 145 {
 146         TALLOC_CTX *mem_ctx = talloc_init("torture_local_messaging");
 147         BOOL ret = True;
 148
 149         ret &= test_ping_speed(mem_ctx);
 150
 151         talloc_free(mem_ctx);
 152
 153         return True;
 154 }