summaryrefslogtreecommitdiff
path: root/source4/torture
diff options
context:
space:
mode:
Diffstat (limited to 'source4/torture')
-rw-r--r--source4/torture/config.mk5
-rw-r--r--source4/torture/rpc/initshutdown.c146
-rw-r--r--source4/torture/rpc/winreg.c20
-rw-r--r--source4/torture/torture.c1
4 files changed, 164 insertions, 8 deletions
diff --git a/source4/torture/config.mk b/source4/torture/config.mk
index 4bd2ce6d5d..80fdefcf36 100644
--- a/source4/torture/config.mk
+++ b/source4/torture/config.mk
@@ -85,6 +85,7 @@ ADD_OBJ_FILES = \
torture/rpc/eventlog.o \
torture/rpc/epmapper.o \
torture/rpc/winreg.o \
+ torture/rpc/initshutdown.o \
torture/rpc/oxidresolve.o \
torture/rpc/remact.o \
torture/rpc/mgmt.o \
@@ -102,8 +103,8 @@ ADD_OBJ_FILES = \
torture/rpc/dssetup.o \
torture/rpc/alter_context.o
REQUIRED_SUBSYSTEMS = \
- NDR_ALL RPC_NDR_SAMR RPC_NDR_WINREG RPC_NDR_OXIDRESOLVER \
- RPC_NDR_EVENTLOG RPC_NDR_ECHO RPC_NDR_SVCCTL \
+ NDR_ALL RPC_NDR_SAMR RPC_NDR_WINREG RPC_NDR_INITSHUTDOWN \
+ RPC_NDR_OXIDRESOLVER RPC_NDR_EVENTLOG RPC_NDR_ECHO RPC_NDR_SVCCTL \
RPC_NDR_MGMT RPC_NDR_NETLOGON RPC_NDR_ATSVC RPC_NDR_DRSUAPI \
RPC_NDR_LSA RPC_NDR_EPMAPPER RPC_NDR_DFS RPC_NDR_SPOOLSS \
RPC_NDR_SRVSVC RPC_NDR_WKSSVC RPC_NDR_ROT RPC_NDR_DSSETUP \
diff --git a/source4/torture/rpc/initshutdown.c b/source4/torture/rpc/initshutdown.c
new file mode 100644
index 0000000000..0aa64e3eb2
--- /dev/null
+++ b/source4/torture/rpc/initshutdown.c
@@ -0,0 +1,146 @@
+/*
+ Unix SMB/CIFS implementation.
+ test suite for initshutdown operations
+
+ Copyright (C) Tim Potter 2003
+ Copyright (C) Jelmer Vernooij 2004-2005
+
+ 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 2 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, write to the Free Software
+ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+#include "librpc/gen_ndr/ndr_initshutdown.h"
+
+static void init_initshutdown_String(TALLOC_CTX *mem_ctx, struct initshutdown_String *name, const char *s)
+{
+ name->name = talloc(mem_ctx, struct initshutdown_String_sub);
+ name->name->name = s;
+}
+
+static BOOL test_Init(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+ const char *msg, uint32_t timeout)
+{
+ struct initshutdown_Init r;
+ NTSTATUS status;
+ uint16_t hostname = 0x0;
+
+ r.in.hostname = &hostname;
+ r.in.message = talloc(mem_ctx, struct initshutdown_String);
+ init_initshutdown_String(mem_ctx, r.in.message, msg);
+ r.in.force_apps = 1;
+ r.in.timeout = timeout;
+ r.in.reboot = 1;
+
+ status = dcerpc_initshutdown_Init(p, mem_ctx, &r);
+
+ if (!NT_STATUS_IS_OK(status)) {
+ printf("initshutdown_Init failed - %s\n", nt_errstr(status));
+ return False;
+ }
+
+ if (!W_ERROR_IS_OK(r.out.result)) {
+ printf("initshutdown_Init failed - %s\n", win_errstr(r.out.result));
+ return False;
+ }
+
+ return True;
+}
+
+static BOOL test_InitEx(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+ const char *msg, uint32_t timeout)
+{
+ struct initshutdown_InitEx r;
+ NTSTATUS status;
+ uint16_t hostname = 0x0;
+
+ r.in.hostname = &hostname;
+ r.in.message = talloc(mem_ctx, struct initshutdown_String);
+ init_initshutdown_String(mem_ctx, r.in.message, msg);
+ r.in.force_apps = 1;
+ r.in.timeout = timeout;
+ r.in.reboot = 1;
+ r.in.reason = 0;
+
+ status = dcerpc_initshutdown_InitEx(p, mem_ctx, &r);
+
+ if (!NT_STATUS_IS_OK(status)) {
+ printf("initshutdown_InitEx failed - %s\n", nt_errstr(status));
+ return False;
+ }
+
+ if (!W_ERROR_IS_OK(r.out.result)) {
+ printf("initshutdown_InitEx failed - %s\n", win_errstr(r.out.result));
+ return False;
+ }
+
+ return True;
+}
+
+static BOOL test_Abort(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
+{
+ struct initshutdown_Abort r;
+ NTSTATUS status;
+ uint16_t server = 0x0;
+
+ r.in.server = &server;
+
+ status = dcerpc_initshutdown_Abort(p, mem_ctx, &r);
+
+ if (!NT_STATUS_IS_OK(status)) {
+ printf("initshutdown_Abort failed - %s\n", nt_errstr(status));
+ return False;
+ }
+
+ if (!W_ERROR_IS_OK(r.out.result)) {
+ printf("initshutdown_Abort failed - %s\n", win_errstr(r.out.result));
+ return False;
+ }
+
+ return True;
+}
+
+BOOL torture_rpc_initshutdown(void)
+{
+ NTSTATUS status;
+ struct dcerpc_pipe *p;
+ TALLOC_CTX *mem_ctx;
+ BOOL ret = True;
+
+ mem_ctx = talloc_init("torture_rpc_initshutdown");
+
+ status = torture_rpc_connection(&p,
+ DCERPC_INITSHUTDOWN_NAME,
+ DCERPC_INITSHUTDOWN_UUID,
+ DCERPC_INITSHUTDOWN_VERSION);
+
+ if (!NT_STATUS_IS_OK(status)) {
+ return False;
+ }
+
+ if (!lp_parm_bool(-1, "torture", "dangerous", False)) {
+ printf("initshutdown tests disabled - enable dangerous tests to use\n");
+ } else {
+ ret &= test_Init(p, mem_ctx, "spottyfood", 30);
+ ret &= test_Abort(p, mem_ctx);
+ ret &= test_InitEx(p, mem_ctx, "spottyfood", 30);
+ ret &= test_Abort(p, mem_ctx);
+ }
+
+ talloc_free(mem_ctx);
+
+ torture_rpc_close(p);
+
+ return ret;
+}
diff --git a/source4/torture/rpc/winreg.c b/source4/torture/rpc/winreg.c
index 9d259ae47a..df0114b669 100644
--- a/source4/torture/rpc/winreg.c
+++ b/source4/torture/rpc/winreg.c
@@ -23,6 +23,12 @@
#include "includes.h"
#include "librpc/gen_ndr/ndr_winreg.h"
+static void init_initshutdown_String(TALLOC_CTX *mem_ctx, struct initshutdown_String *name, const char *s)
+{
+ name->name = talloc(mem_ctx, struct initshutdown_String_sub);
+ name->name->name = s;
+}
+
static void init_winreg_String(struct winreg_String *name, const char *s)
{
name->name = s;
@@ -555,10 +561,11 @@ static BOOL test_InitiateSystemShutdown(struct dcerpc_pipe *p, TALLOC_CTX *mem_c
{
struct winreg_InitiateSystemShutdown r;
NTSTATUS status;
+ uint16_t hostname = 0x0;
- r.in.hostname = NULL;
- r.in.message = talloc(mem_ctx, struct winreg_String);
- init_winreg_String(r.in.message, msg);
+ r.in.hostname = &hostname;
+ r.in.message = talloc(mem_ctx, struct initshutdown_String);
+ init_initshutdown_String(mem_ctx, r.in.message, msg);
r.in.force_apps = 1;
r.in.timeout = timeout;
r.in.reboot = 1;
@@ -583,10 +590,11 @@ static BOOL test_InitiateSystemShutdownEx(struct dcerpc_pipe *p, TALLOC_CTX *mem
{
struct winreg_InitiateSystemShutdownEx r;
NTSTATUS status;
+ uint16_t hostname = 0x0;
- r.in.hostname = NULL;
- r.in.message = talloc(mem_ctx, struct winreg_String);
- init_winreg_String(r.in.message, msg);
+ r.in.hostname = &hostname;
+ r.in.message = talloc(mem_ctx, struct initshutdown_String);
+ init_initshutdown_String(mem_ctx, r.in.message, msg);
r.in.force_apps = 1;
r.in.timeout = timeout;
r.in.reboot = 1;
diff --git a/source4/torture/torture.c b/source4/torture/torture.c
index 09be1e1fe3..32c258068b 100644
--- a/source4/torture/torture.c
+++ b/source4/torture/torture.c
@@ -2394,6 +2394,7 @@ static struct {
{"RPC-EVENTLOG", torture_rpc_eventlog, 0},
{"RPC-EPMAPPER", torture_rpc_epmapper, 0},
{"RPC-WINREG", torture_rpc_winreg, 0},
+ {"RPC-INITSHUTDOWN", torture_rpc_initshutdown, 0},
{"RPC-OXIDRESOLVE", torture_rpc_oxidresolve, 0},
{"RPC-REMACT", torture_rpc_remact, 0},
{"RPC-MGMT", torture_rpc_mgmt, 0},
generated by cgit (git 2.34.1) at 2024-11-10 10:50:25 +0000