s4-smbtorture: add torture ndr nbt testsuite.

Guenther

diff --git a/source4/torture/config.mk b/source4/torture/config.mk
index 826b21c3367c013019f1cc8815226601f00d31d1..9ee2edfdd002d99e3088fe5983e089442f3a19fb 100644 (file)
--- a/source4/torture/config.mk
+++ b/source4/torture/config.mk
@@ -95,7 +95,7 @@ mkinclude libsmbclient/config.mk
 [SUBSYSTEM::TORTURE_NDR]
 PRIVATE_DEPENDENCIES = torture SERVICE_SMB
 
-TORTURE_NDR_OBJ_FILES = $(addprefix $(torturesrcdir)/ndr/, ndr.o winreg.o atsvc.o lsa.o epmap.o dfs.o netlogon.o drsuapi.o spoolss.o samr.o dfsblob.o drsblobs.o)
+TORTURE_NDR_OBJ_FILES = $(addprefix $(torturesrcdir)/ndr/, ndr.o winreg.o atsvc.o lsa.o epmap.o dfs.o netlogon.o drsuapi.o spoolss.o samr.o dfsblob.o drsblobs.o nbt.o)
 
 $(eval $(call proto_header_template,$(torturesrcdir)/ndr/proto.h,$(TORTURE_NDR_OBJ_FILES:.o=.c)))
 

diff --git a/source4/torture/ndr/nbt.c b/source4/torture/ndr/nbt.c
new file mode 100644 (file)
index 0000000..8955f4d
--- /dev/null
+++ b/source4/torture/ndr/nbt.c
@@ -0,0 +1,70 @@
+/*
+   Unix SMB/CIFS implementation.
+   test suite for nbt ndr operations
+
+   Copyright (C) Guenther Deschner 2010
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "torture/ndr/ndr.h"
+#include "librpc/gen_ndr/ndr_nbt.h"
+
+static const uint8_t netlogon_logon_request_req_data[] = {
+       0x00, 0x00, 0x57, 0x49, 0x4e, 0x39, 0x38, 0x00, 0x47, 0x44, 0x00, 0x5c,
+       0x4d, 0x41, 0x49, 0x4c, 0x53, 0x4c, 0x4f, 0x54, 0x5c, 0x54, 0x45, 0x4d,
+       0x50, 0x5c, 0x4e, 0x45, 0x54, 0x4c, 0x4f, 0x47, 0x4f, 0x4e, 0x00, 0x01,
+       0x01, 0x00, 0xff, 0xff
+};
+
+static bool netlogon_logon_request_req_check(struct torture_context *tctx,
+                                            struct nbt_netlogon_packet *r)
+{
+       torture_assert_int_equal(tctx, r->command, LOGON_REQUEST, "command");
+       torture_assert_str_equal(tctx, r->req.logon0.computer_name, "WIN98", "computer name");
+       torture_assert_str_equal(tctx, r->req.logon0.user_name, "GD", "user_name");
+       torture_assert_str_equal(tctx, r->req.logon0.mailslot_name, "\\MAILSLOT\\TEMP\\NETLOGON", "mailslot_name");
+       torture_assert_int_equal(tctx, r->req.logon0.request_count, 1, "request_count");
+       torture_assert_int_equal(tctx, r->req.logon0.lmnt_token, 1, "lmnt_token");
+       torture_assert_int_equal(tctx, r->req.logon0.lm20_token, 0xffff, "lm20_token");
+
+       return true;
+}
+
+static const uint8_t netlogon_logon_request_resp_data[] = {
+       0x06, 0x00, 0x5c, 0x5c, 0x4d, 0x54, 0x48, 0x45, 0x4c, 0x45, 0x4e, 0x41,
+       0x00, 0xff, 0xff
+};
+
+static bool netlogon_logon_request_resp_check(struct torture_context *tctx,
+                                             struct nbt_netlogon_response2 *r)
+{
+       torture_assert_int_equal(tctx, r->command, LOGON_RESPONSE2, "command");
+       torture_assert_str_equal(tctx, r->pdc_name, "\\\\MTHELENA", "pdc_name");
+       torture_assert_int_equal(tctx, r->lm20_token, 0xffff, "lm20_token");
+
+       return true;
+}
+
+struct torture_suite *ndr_nbt_suite(TALLOC_CTX *ctx)
+{
+       struct torture_suite *suite = torture_suite_create(ctx, "nbt");
+
+       torture_suite_add_ndr_pull_fn_test(suite, nbt_netlogon_packet, netlogon_logon_request_req_data, NDR_IN, netlogon_logon_request_req_check);
+
+       torture_suite_add_ndr_pull_fn_test(suite, nbt_netlogon_response2, netlogon_logon_request_resp_data, NDR_IN, netlogon_logon_request_resp_check);
+
+       return suite;
+}

diff --git a/source4/torture/ndr/ndr.c b/source4/torture/ndr/ndr.c
index 6068e108b5adf5fab033cfb1af743822c3f9a0c8..9d0f34546179e073b85a2d6c4e0daac9acad7111 100644 (file)
--- a/source4/torture/ndr/ndr.c
+++ b/source4/torture/ndr/ndr.c
@@ -267,6 +267,7 @@ struct torture_suite *torture_local_ndr(TALLOC_CTX *mem_ctx)
        torture_suite_add_suite(suite, ndr_spoolss_suite(suite));
        torture_suite_add_suite(suite, ndr_samr_suite(suite));
        torture_suite_add_suite(suite, ndr_drsblobs_suite(suite));
+       torture_suite_add_suite(suite, ndr_nbt_suite(suite));
 
        torture_suite_add_simple_test(suite, "string terminator", 
                                                                   test_check_string_terminator);

diff --git a/source4/torture/wscript_build b/source4/torture/wscript_build
index 47b2b3d235533391e14f04c112ec8d049c53667f..715bec502f587f76e1349a966d64a69a5ec56d54 100644 (file)
--- a/source4/torture/wscript_build
+++ b/source4/torture/wscript_build
@@ -33,7 +33,7 @@ bld.RECURSE('libnetapi')
 bld.RECURSE('libsmbclient')
 
 bld.SAMBA_SUBSYSTEM('TORTURE_NDR',
-       source='ndr/ndr.c ndr/winreg.c ndr/atsvc.c ndr/lsa.c ndr/epmap.c ndr/dfs.c ndr/netlogon.c ndr/drsuapi.c ndr/spoolss.c ndr/samr.c ndr/dfsblob.c ndr/drsblobs.c',
+       source='ndr/ndr.c ndr/winreg.c ndr/atsvc.c ndr/lsa.c ndr/epmap.c ndr/dfs.c ndr/netlogon.c ndr/drsuapi.c ndr/spoolss.c ndr/samr.c ndr/dfsblob.c ndr/drsblobs.c ndr/nbt.c',
        autoproto='ndr/proto.h',
        deps='torture SERVICE_SMB'
        )