source4/torture/local/resolve.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    local test for async resolve 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 "lib/events/events.h"
  25 #include "libcli/resolve/resolve.h"
  26 #include "torture/torture.h"
  27
  28 static bool test_async_resolve(struct torture_context *tctx)
  29 {
  30         struct nbt_name n;
  31         struct event_context *ev;
  32         int timelimit = torture_setting_int(tctx, "timelimit", 2);
  33         const char *host = torture_setting_string(tctx, "host", NULL);
  34         int count = 0;
  35         struct timeval tv = timeval_current();
  36         TALLOC_CTX *mem_ctx = tctx;
  37
  38         ev = tctx->ev;
  39
  40         ZERO_STRUCT(n);
  41         n.name = host;
  42
  43         torture_comment(tctx, "Testing async resolve of '%s' for %d seconds\n",
  44                         host, timelimit);
  45         while (timeval_elapsed(&tv) < timelimit) {
  46                 const char *s;
  47                 struct composite_context *c = resolve_name_host_send(mem_ctx, ev, &n);
  48                 torture_assert(tctx, c != NULL, "resolve_name_host_send");
  49                 torture_assert_ntstatus_ok(tctx, resolve_name_host_recv(c, mem_ctx, &s),
  50                                                                    "async resolve failed");
  51                 count++;
  52         }
  53
  54         torture_comment(tctx, "async rate of %.1f resolves/sec\n",
  55                         count/timeval_elapsed(&tv));
  56         return true;
  57 }
  58
  59 /*
  60   test resolution using sync method
  61 */
  62 static bool test_sync_resolve(struct torture_context *tctx)
  63 {
  64         int timelimit = torture_setting_int(tctx, "timelimit", 2);
  65         struct timeval tv = timeval_current();
  66         int count = 0;
  67         const char *host = torture_setting_string(tctx, "host", NULL);
  68
  69         torture_comment(tctx, "Testing sync resolve of '%s' for %d seconds\n",
  70                         host, timelimit);
  71         while (timeval_elapsed(&tv) < timelimit) {
  72                 sys_inet_ntoa(interpret_addr2(host));
  73                 count++;
  74         }
  75
  76         torture_comment(tctx, "sync rate of %.1f resolves/sec\n",
  77                         count/timeval_elapsed(&tv));
  78         return true;
  79 }
  80
  81
  82 struct torture_suite *torture_local_resolve(TALLOC_CTX *mem_ctx)
  83 {
  84         struct torture_suite *suite = torture_suite_create(mem_ctx, "RESOLVE");
  85
  86         torture_suite_add_simple_test(suite, "async", test_async_resolve);
  87         torture_suite_add_simple_test(suite, "sync", test_sync_resolve);
  88
  89         return suite;
  90 }