6cb9507398fc87844a643acde7c1d3b77243922a
[gd/samba-autobuild/.git] / source4 / torture / nbt / nbt.c
1 /* 
2    Unix SMB/CIFS implementation.
3    SMB torture tester
4    Copyright (C) Jelmer Vernooij 2006
5    
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21 #include "libcli/nbt/libnbt.h"
22 #include "torture/torture.h"
23 #include "torture/nbt/proto.h"
24 #include "torture/ui.h"
25 #include "libcli/resolve/resolve.h"
26 #include "param/param.h"
27
28 bool torture_nbt_get_name(struct torture_context *tctx, 
29                           struct nbt_name *name, 
30                           const char **address)
31 {
32         make_nbt_name_server(name, strupper_talloc(tctx, 
33                              torture_setting_string(tctx, "host", NULL)));
34
35         /* do an initial name resolution to find its IP */
36         torture_assert_ntstatus_ok(tctx, 
37                                    resolve_name(lp_resolve_context(tctx->lp_ctx), name, tctx, address, NULL), 
38                                    talloc_asprintf(tctx, 
39                                                    "Failed to resolve %s", name->name));
40         
41         return true;
42 }
43
44 NTSTATUS torture_nbt_init(void)
45 {
46         struct torture_suite *suite = torture_suite_create(
47                 talloc_autofree_context(), "NBT");
48         /* nbt tests */
49         torture_suite_add_suite(suite, torture_nbt_register(suite));
50         torture_suite_add_suite(suite, torture_nbt_wins(suite));
51         torture_suite_add_suite(suite, torture_nbt_dgram(suite));
52         torture_suite_add_suite(suite, torture_nbt_winsreplication(suite));
53         torture_suite_add_suite(suite, torture_bench_nbt(suite));
54         torture_suite_add_suite(suite, torture_bench_wins(suite));
55
56         suite->description = talloc_strdup(suite, 
57                                          "NetBIOS over TCP/IP and WINS tests");
58
59         torture_register_suite(suite);
60
61         return NT_STATUS_OK;
62 }