source4/torture/libnet/libnet_lookup.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Test suite for libnet calls.
   4
   5    Copyright (C) Rafal Szczesniak 2005
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/cmdline/popt_common.h"
  23 #include "libnet/libnet.h"
  24 #include "librpc/gen_ndr/nbt.h"
  25 #include "librpc/rpc/dcerpc.h"
  26 #include "libcli/libcli.h"
  27 #include "torture/torture.h"
  28
  29
  30 BOOL torture_lookup(struct torture_context *torture)
  31 {
  32         BOOL ret;
  33         NTSTATUS status;
  34         TALLOC_CTX *mem_ctx;
  35         struct libnet_context *ctx;
  36         struct libnet_Lookup lookup;
  37         struct dcerpc_binding *bind;
  38         const char *bindstr;
  39
  40         mem_ctx = talloc_init("test_lookup");
  41
  42         ctx = libnet_context_init(NULL);
  43         ctx->cred = cmdline_credentials;
  44
  45         lookup.in.hostname = torture_setting_string(torture, "host", NULL);
  46         if (lookup.in.hostname == NULL) {
  47                 bindstr = torture_setting_string(torture, "binding", NULL);
  48                 status = dcerpc_parse_binding(mem_ctx, bindstr, &bind);
  49                 if (NT_STATUS_IS_OK(status)) {
  50                         lookup.in.hostname = bind->host;
  51                 }
  52         }
  53
  54         lookup.in.type     = NBT_NAME_CLIENT;
  55         lookup.in.methods  = NULL;
  56         lookup.out.address = NULL;
  57
  58         status = libnet_Lookup(ctx, mem_ctx, &lookup);
  59
  60         if (!NT_STATUS_IS_OK(status)) {
  61                 printf("Couldn't lookup name %s: %s\n", lookup.in.hostname, nt_errstr(status));
  62                 ret = False;
  63                 goto done;
  64         }
  65
  66         ret = True;
  67
  68         printf("Name [%s] found at adrress: %s.\n", lookup.in.hostname, *lookup.out.address);
  69
  70 done:
  71         talloc_free(mem_ctx);
  72         return ret;
  73 }
  74
  75
  76 BOOL torture_lookup_host(struct torture_context *torture)
  77 {
  78         BOOL ret;
  79         NTSTATUS status;
  80         TALLOC_CTX *mem_ctx;
  81         struct libnet_context *ctx;
  82         struct libnet_Lookup lookup;
  83         struct dcerpc_binding *bind;
  84         const char *bindstr;
  85
  86         mem_ctx = talloc_init("test_lookup_host");
  87
  88         ctx = libnet_context_init(NULL);
  89         ctx->cred = cmdline_credentials;
  90
  91         lookup.in.hostname = torture_setting_string(torture, "host", NULL);
  92         if (lookup.in.hostname == NULL) {
  93                 bindstr = torture_setting_string(torture, "binding", NULL);
  94                 status = dcerpc_parse_binding(mem_ctx, bindstr, &bind);
  95                 if (NT_STATUS_IS_OK(status)) {
  96                         lookup.in.hostname = bind->host;
  97                 }
  98         }
  99
 100         lookup.in.methods  = NULL;
 101         lookup.out.address = NULL;
 102
 103         status = libnet_LookupHost(ctx, mem_ctx, &lookup);
 104
 105         if (!NT_STATUS_IS_OK(status)) {
 106                 printf("Couldn't lookup host %s: %s\n", lookup.in.hostname, nt_errstr(status));
 107                 ret = False;
 108                 goto done;
 109         }
 110
 111         ret = True;
 112
 113         printf("Host [%s] found at adrress: %s.\n", lookup.in.hostname, *lookup.out.address);
 114
 115 done:
 116         talloc_free(mem_ctx);
 117         return ret;
 118 }
 119
 120
 121 BOOL torture_lookup_pdc(struct torture_context *torture)
 122 {
 123         BOOL ret;
 124         NTSTATUS status;
 125         TALLOC_CTX *mem_ctx;
 126         struct libnet_context *ctx;
 127         struct libnet_LookupDCs *lookup;
 128         int i;
 129
 130         mem_ctx = talloc_init("test_lookup_pdc");
 131
 132         ctx = libnet_context_init(NULL);
 133         ctx->cred = cmdline_credentials;
 134
 135         talloc_steal(ctx, mem_ctx);
 136
 137         lookup = talloc(mem_ctx, struct libnet_LookupDCs);
 138         if (!lookup) {
 139                 ret = False;
 140                 goto done;
 141         }
 142
 143         lookup->in.domain_name = lp_workgroup();
 144         lookup->in.name_type   = NBT_NAME_PDC;
 145
 146         status = libnet_LookupDCs(ctx, mem_ctx, lookup);
 147
 148         if (!NT_STATUS_IS_OK(status)) {
 149                 printf("Couldn't lookup pdc %s: %s\n", lookup->in.domain_name,
 150                        nt_errstr(status));
 151                 ret = False;
 152                 goto done;
 153         }
 154
 155         ret = True;
 156
 157         printf("DCs of domain [%s] found.\n", lookup->in.domain_name);
 158         for (i = 0; i < lookup->out.num_dcs; i++) {
 159                 printf("\tDC[%d]: name=%s, address=%s\n", i, lookup->out.dcs[i].name,
 160                        lookup->out.dcs[i].address);
 161         }
 162
 163 done:
 164         talloc_free(mem_ctx);
 165         return ret;
 166 }
 167
 168
 169 BOOL torture_lookup_sam_name(struct torture_context *torture)
 170 {
 171         NTSTATUS status;
 172         TALLOC_CTX *mem_ctx;
 173         struct libnet_context *ctx;
 174         struct libnet_LookupName r;
 175
 176         ctx = libnet_context_init(NULL);
 177         ctx->cred = cmdline_credentials;
 178
 179         mem_ctx = talloc_init("torture lookup sam name");
 180         if (mem_ctx == NULL) return False;
 181
 182         r.in.name = "Administrator";
 183         r.in.domain_name = lp_workgroup();
 184
 185         status = libnet_LookupName(ctx, mem_ctx, &r);
 186
 187         talloc_free(mem_ctx);
 188         talloc_free(ctx);
 189
 190         return True;
 191 }