svn commit: samba r8081 - in branches/SAMBA_4_0/source/torture: . libnet

[mimir at samba.org]

Author: mimir
Date: 2005-07-02 14:38:05 +0000 (Sat, 02 Jul 2005)
New Revision: 8081

WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=8081

Log:
Add simple test for RpcConnect function.


rafal


Added:
   branches/SAMBA_4_0/source/torture/libnet/libnet_rpc.c
Modified:
   branches/SAMBA_4_0/source/torture/config.mk
   branches/SAMBA_4_0/source/torture/torture.c


Changeset:
Modified: branches/SAMBA_4_0/source/torture/config.mk
===================================================================
--- branches/SAMBA_4_0/source/torture/config.mk	2005-07-02 14:37:21 UTC (rev 8080)
+++ branches/SAMBA_4_0/source/torture/config.mk	2005-07-02 14:38:05 UTC (rev 8081)
@@ -197,7 +197,8 @@
 		torture/libnet/userman.o \
 		torture/libnet/domain.o \
 		torture/libnet/libnet_lookup.o \
-		torture/libnet/libnet_user.o
+		torture/libnet/libnet_user.o \
+		torture/libnet/libnet_rpc.o
 REQUIRED_SUBSYSTEMS = \
 		NDR_ALL RPC_NDR_SAMR LIBNET
 # End SUBSYSTEM TORTURE_NET

Added: branches/SAMBA_4_0/source/torture/libnet/libnet_rpc.c
===================================================================
--- branches/SAMBA_4_0/source/torture/libnet/libnet_rpc.c	2005-07-02 14:37:21 UTC (rev 8080)
+++ branches/SAMBA_4_0/source/torture/libnet/libnet_rpc.c	2005-07-02 14:38:05 UTC (rev 8081)
@@ -0,0 +1,95 @@
+/* 
+   Unix SMB/CIFS implementation.
+   Test suite for libnet calls.
+
+   Copyright (C) Rafal Szczesniak 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 "lib/cmdline/popt_common.h"
+#include "libnet/libnet.h"
+#include "libnet/composite.h"
+#include "libcli/composite/monitor.h"
+
+
+BOOL test_lsa_connect(struct libnet_context *ctx)
+{
+	NTSTATUS status;
+	struct libnet_RpcConnect connect;
+	connect.level                     = LIBNET_RPC_CONNECT_PDC;
+	connect.in.domain_name            = lp_workgroup();
+	connect.in.dcerpc_iface_name      = DCERPC_LSARPC_NAME;
+	connect.in.dcerpc_iface_uuid      = DCERPC_LSARPC_UUID;
+	connect.in.dcerpc_iface_version   = DCERPC_LSARPC_VERSION;
+
+	status = libnet_RpcConnect(ctx, ctx, &connect);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("Couldn't connect to rpc service %s on %s: %s\n",
+		       connect.in.dcerpc_iface_name, connect.in.domain_name,
+		       nt_errstr(status));
+
+		return False;
+	}
+
+	return True;
+}
+
+
+BOOL test_samr_connect(struct libnet_context *ctx)
+{
+	NTSTATUS status;
+	struct libnet_RpcConnect connect;
+	connect.level                     = LIBNET_RPC_CONNECT_PDC;
+	connect.in.domain_name            = lp_workgroup();
+	connect.in.dcerpc_iface_name      = DCERPC_SAMR_NAME;
+	connect.in.dcerpc_iface_uuid      = DCERPC_SAMR_UUID;
+	connect.in.dcerpc_iface_version   = DCERPC_SAMR_VERSION;
+
+	status = libnet_RpcConnect(ctx, ctx, &connect);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("Couldn't connect to rpc service %s on %s: %s\n",
+		       connect.in.dcerpc_iface_name, connect.in.domain_name,
+		       nt_errstr(status));
+
+		return False;
+	}
+
+	return True;
+}
+
+
+BOOL torture_rpc_connect(void)
+{
+	struct libnet_context *ctx;
+	
+	ctx = libnet_context_init(NULL);
+	ctx->cred = cmdline_credentials;
+
+	if (!test_lsa_connect(ctx)) {
+		printf("failed to connect lsarpc interface\n");
+		return False;
+	}
+
+	if (!test_samr_connect(ctx)) {
+		printf("failed to connect samr interface\n");
+		return False;
+	}
+
+	return True;
+}

Modified: branches/SAMBA_4_0/source/torture/torture.c
===================================================================
--- branches/SAMBA_4_0/source/torture/torture.c	2005-07-02 14:37:21 UTC (rev 8080)
+++ branches/SAMBA_4_0/source/torture/torture.c	2005-07-02 14:38:05 UTC (rev 8081)
@@ -2347,6 +2347,7 @@
 	{"NET-API-LOOKUPHOST", torture_lookup_host, 0},
 	{"NET-API-LOOKUPPDC", torture_lookup_pdc, 0},
 	{"NET-API-CREATEUSER", torture_createuser, 0},
+	{"NET-API-RPCCONNECT", torture_rpc_connect, 0},
 
 	{NULL, NULL, 0}};