From 0c55b8d8c245ae4e418cb3d4341b4b5233828cad Mon Sep 17 00:00:00 2001 From: Jelmer Vernooij Date: Mon, 2 Aug 2004 00:24:04 +0000 Subject: r1617: Start working on the svcctl pipe (This used to be commit 518ff5f505446fee623365fd5bdfc75c0cb93d2a) --- source4/torture/config.mk | 1 + source4/torture/rpc/svcctl.c | 141 +++++++++++++++++++++++++++++++++++++++++++ source4/torture/torture.c | 1 + 3 files changed, 143 insertions(+) create mode 100644 source4/torture/rpc/svcctl.c (limited to 'source4/torture') diff --git a/source4/torture/config.mk b/source4/torture/config.mk index de24d10617..77d01f75c0 100644 --- a/source4/torture/config.mk +++ b/source4/torture/config.mk @@ -55,6 +55,7 @@ ADD_OBJ_FILES = \ torture/rpc/samr.o \ torture/rpc/wkssvc.o \ torture/rpc/srvsvc.o \ + torture/rpc/svcctl.o \ torture/rpc/atsvc.o \ torture/rpc/eventlog.o \ torture/rpc/epmapper.o \ diff --git a/source4/torture/rpc/svcctl.c b/source4/torture/rpc/svcctl.c new file mode 100644 index 0000000000..062fcdf584 --- /dev/null +++ b/source4/torture/rpc/svcctl.c @@ -0,0 +1,141 @@ +/* + Unix SMB/CIFS implementation. + test suite for srvsvc rpc operations + + Copyright (C) Jelmer Vernooij 2004 + + 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" + +static BOOL test_EnumServicesStatus(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *h) +{ + struct svcctl_EnumServicesStatus r; + int i; + NTSTATUS status; + uint32 resume_handle = 0; + struct ENUM_SERVICE_STATUS *service = talloc_p(mem_ctx, struct ENUM_SERVICE_STATUS); + + r.in.handle = h; + r.in.type = SERVICE_TYPE_WIN32; + r.in.state = SERVICE_STATE_ALL; + r.in.buf_size = sizeof(struct ENUM_SERVICE_STATUS); + r.in.resume_handle = &resume_handle; + r.out.service = service; + r.out.resume_handle = &resume_handle; + + status = dcerpc_svcctl_EnumServicesStatus(p, mem_ctx, &r); + + if (!NT_STATUS_IS_OK(status)) { + printf("ËnumServicesStatus failed!\n"); + return False; + } + + if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) { + r.in.buf_size = r.out.bytes_needed + sizeof(struct ENUM_SERVICE_STATUS); + service = talloc_realloc(mem_ctx, service, r.in.buf_size); + r.out.service = service; + + status = dcerpc_svcctl_EnumServicesStatus(p, mem_ctx, &r); + + if (!NT_STATUS_IS_OK(status)) { + printf("ËnumServicesStatus failed!\n"); + return False; + } + + if (!W_ERROR_IS_OK(r.out.result)) { + printf("EnumServicesStatus failed\n"); + return False; + } + } + + for(i = 0; i < r.out.services_returned; i++) { + printf("%s - %s\n", service[i].service_name, service[i].display_name); + } + + return True; +} + +static BOOL test_OpenSCManager(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *h) +{ + struct svcctl_OpenSCManager r; + NTSTATUS status; + + r.in.MachineName = NULL; + r.in.DatabaseName = NULL; + r.in.access_mask = SEC_RIGHTS_MAXIMUM_ALLOWED; + r.out.handle = h; + + status = dcerpc_svcctl_OpenSCManager(p, mem_ctx, &r); + if (!NT_STATUS_IS_OK(status)) { + printf("OpenSCManager failed!\n"); + return False; + } + + return True; +} + +static BOOL test_CloseServiceHandle(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, struct policy_handle *h) +{ + struct svcctl_CloseServiceHandle r; + NTSTATUS status; + r.in.handle = h; + r.out.handle = h; + status = dcerpc_svcctl_CloseServiceHandle(p, mem_ctx, &r); + if (!NT_STATUS_IS_OK(status)) { + printf("CloseServiceHandle failed\n"); + return False; + } + + return True; +} + +BOOL torture_rpc_svcctl(int dummy) +{ + NTSTATUS status; + struct dcerpc_pipe *p; + struct policy_handle h; + TALLOC_CTX *mem_ctx; + BOOL ret = True; + + mem_ctx = talloc_init("torture_rpc_svcctl"); + + status = torture_rpc_connection(&p, + DCERPC_SVCCTL_NAME, + DCERPC_SVCCTL_UUID, + DCERPC_SVCCTL_VERSION); + if (!NT_STATUS_IS_OK(status)) { + return False; + } + + if (!test_OpenSCManager(p, mem_ctx, &h)) { + ret = False; + } + + if (!test_EnumServicesStatus(p, mem_ctx, &h)) { + ret = False; + } + + if (!test_CloseServiceHandle(p, mem_ctx, &h)) { + ret = False; + } + + talloc_destroy(mem_ctx); + + torture_rpc_close(p); + + return ret; +} diff --git a/source4/torture/torture.c b/source4/torture/torture.c index 804dd34f2a..c7cc29524d 100644 --- a/source4/torture/torture.c +++ b/source4/torture/torture.c @@ -4209,6 +4209,7 @@ static struct { {"RPC-SCHANNEL", torture_rpc_schannel, 0}, {"RPC-WKSSVC", torture_rpc_wkssvc, 0}, {"RPC-SRVSVC", torture_rpc_srvsvc, 0}, + {"RPC-SVCCTL", torture_rpc_svcctl, 0}, {"RPC-ATSVC", torture_rpc_atsvc, 0}, {"RPC-EVENTLOG", torture_rpc_eventlog, 0}, {"RPC-EPMAPPER", torture_rpc_epmapper, 0}, -- cgit