summaryrefslogtreecommitdiff
path: root/source4/torture
diff options
context:
space:
mode:
Diffstat (limited to 'source4/torture')
-rw-r--r--source4/torture/rpc/mgmt.c25
-rw-r--r--source4/torture/rpc/scanner.c211
-rw-r--r--source4/torture/torture.c5
3 files changed, 239 insertions, 2 deletions
diff --git a/source4/torture/rpc/mgmt.c b/source4/torture/rpc/mgmt.c
index e3a8146e64..0f3cfffa66 100644
--- a/source4/torture/rpc/mgmt.c
+++ b/source4/torture/rpc/mgmt.c
@@ -22,6 +22,9 @@
#include "includes.h"
+/*
+ ask the server what interface IDs are available on this endpoint
+*/
static BOOL test_inq_if_ids(struct dcerpc_pipe *p,
TALLOC_CTX *mem_ctx)
{
@@ -166,11 +169,14 @@ BOOL torture_rpc_mgmt(int dummy)
TALLOC_CTX *mem_ctx;
BOOL ret = True;
int i;
+ char *host = lp_parm_string(-1, "torture", "host");
+ uint32 port;
mem_ctx = talloc_init("torture_rpc_mgmt");
- for (i=0;dcerpc_pipes[i];i++) {
-
+ for (i=0;dcerpc_pipes[i];i++) {
+ char *transport = lp_parm_string(-1, "torture", "transport");
+
/* some interfaces are not mappable */
if (dcerpc_pipes[i]->num_calls == 0 ||
strcmp(dcerpc_pipes[i]->name, "mgmt") == 0) {
@@ -179,6 +185,21 @@ BOOL torture_rpc_mgmt(int dummy)
printf("\nTesting pipe '%s'\n", dcerpc_pipes[i]->name);
+ /* on TCP we need to find the right endpoint */
+ if (strcasecmp(transport, "ncacn_ip_tcp") == 0) {
+ status = dcerpc_epm_map_tcp_port(host,
+ dcerpc_pipes[i]->uuid,
+ dcerpc_pipes[i]->if_version,
+ &port);
+ if (!NT_STATUS_IS_OK(status)) {
+ ret = False;
+ continue;
+ }
+
+ lp_set_cmdline("torture:share",
+ talloc_asprintf(mem_ctx, "%u", port));
+ }
+
status = torture_rpc_connection(&p,
dcerpc_pipes[i]->name,
DCERPC_MGMT_UUID,
diff --git a/source4/torture/rpc/scanner.c b/source4/torture/rpc/scanner.c
new file mode 100644
index 0000000000..814b28a01d
--- /dev/null
+++ b/source4/torture/rpc/scanner.c
@@ -0,0 +1,211 @@
+/*
+ Unix SMB/CIFS implementation.
+
+ scanner for rpc calls
+
+ Copyright (C) Andrew Tridgell 2003
+
+ 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"
+
+/*
+ find the number of calls defined by local IDL
+*/
+static const char *find_idl_name(const char *uuid, uint32 if_version)
+{
+ int i;
+ for (i=0;dcerpc_pipes[i];i++) {
+ if (strcasecmp(dcerpc_pipes[i]->uuid, uuid) == 0 &&
+ dcerpc_pipes[i]->if_version == if_version) {
+ return dcerpc_pipes[i]->name;
+ }
+ }
+ return "UNKNOWN";
+}
+
+/*
+ find the number of calls defined by local IDL
+*/
+static int num_idl_calls(const char *uuid, uint32 if_version)
+{
+ int i;
+ for (i=0;dcerpc_pipes[i];i++) {
+ if (strcasecmp(dcerpc_pipes[i]->uuid, uuid) == 0 &&
+ dcerpc_pipes[i]->if_version == if_version) {
+ return dcerpc_pipes[i]->num_calls;
+ }
+ }
+ return -1;
+}
+
+/*
+ work out how many calls there are for an interface
+ */
+static BOOL test_num_calls(const struct dcerpc_interface_table *iface,
+ TALLOC_CTX *mem_ctx,
+ struct dcerpc_syntax_id *id)
+{
+ struct dcerpc_pipe *p;
+ NTSTATUS status;
+ const char *uuid;
+ int i;
+ DATA_BLOB stub_in, stub_out;
+ int idl_calls;
+
+ uuid = GUID_string(mem_ctx, &id->uuid);
+
+ status = torture_rpc_connection(&p, iface->name,
+ uuid, id->major_version);
+ if (!NT_STATUS_IS_OK(status)) {
+ printf("Failed to connect to '%s' on '%s' - %s\n",
+ uuid, iface->name, nt_errstr(status));
+ return False;
+ }
+
+ /* make null calls */
+ stub_in = data_blob(NULL, 0);
+
+ status = dcerpc_request(p, 10000, mem_ctx, &stub_in, &stub_out);
+ if (NT_STATUS_IS_OK(status) ||
+ p->last_fault_code != DCERPC_FAULT_OP_RNG_ERROR) {
+ printf("unable to determine call count - %s %08x\n",
+ nt_errstr(status), p->last_fault_code);
+ goto done;
+ }
+
+ for (i=128;i>=0;i--) {
+ status = dcerpc_request(p, i, mem_ctx, &stub_in, &stub_out);
+ if (NT_STATUS_IS_OK(status) ||
+ p->last_fault_code != DCERPC_FAULT_OP_RNG_ERROR) break;
+ }
+
+ printf("\t%d calls available\n", i+1);
+ idl_calls = num_idl_calls(uuid, id->major_version);
+ if (idl_calls == -1) {
+ printf("\tinterface not known in local IDL\n");
+ } else if (i+1 != idl_calls) {
+ printf("\tWARNING: local IDL defines %u calls\n", idl_calls);
+ } else {
+ printf("\tOK: matches num_calls in local IDL\n");
+ }
+
+done:
+ torture_rpc_close(p);
+ return True;
+}
+
+/*
+ ask the server what interface IDs are available on this endpoint
+*/
+static BOOL test_inq_if_ids(struct dcerpc_pipe *p,
+ TALLOC_CTX *mem_ctx,
+ const struct dcerpc_interface_table *iface)
+{
+ NTSTATUS status;
+ struct mgmt_inq_if_ids r;
+ int i;
+
+ status = dcerpc_mgmt_inq_if_ids(p, mem_ctx, &r);
+ if (!NT_STATUS_IS_OK(status)) {
+ printf("inq_if_ids failed - %s\n", nt_errstr(status));
+ return False;
+ }
+
+ if (!W_ERROR_IS_OK(r.out.result)) {
+ printf("inq_if_ids gave error code %s\n", win_errstr(r.out.result));
+ return False;
+ }
+
+ if (!r.out.if_id_vector) {
+ printf("inq_if_ids gave NULL if_id_vector\n");
+ return False;
+ }
+
+ for (i=0;i<r.out.if_id_vector->count;i++) {
+ const char *uuid;
+ struct dcerpc_syntax_id *id = r.out.if_id_vector->if_id[i].id;
+ if (!id) continue;
+
+ uuid = GUID_string(mem_ctx, &id->uuid),
+
+ printf("\n\tuuid %s version 0x%04x:0x%04x '%s'\n",
+ uuid,
+ id->major_version, id->minor_version,
+ find_idl_name(uuid, id->major_version));
+ test_num_calls(iface, mem_ctx, id);
+ }
+
+ return True;
+}
+
+
+BOOL torture_rpc_scanner(int dummy)
+{
+ NTSTATUS status;
+ struct dcerpc_pipe *p;
+ TALLOC_CTX *mem_ctx;
+ BOOL ret = True;
+ int i;
+ char *host = lp_parm_string(-1, "torture", "host");
+ uint32 port;
+
+ mem_ctx = talloc_init("torture_rpc_scanner");
+
+ for (i=0;dcerpc_pipes[i];i++) {
+ char *transport = lp_parm_string(-1, "torture", "transport");
+
+ /* some interfaces are not mappable */
+ if (dcerpc_pipes[i]->num_calls == 0 ||
+ strcmp(dcerpc_pipes[i]->name, "mgmt") == 0) {
+ continue;
+ }
+
+ printf("\nTesting pipe '%s'\n", dcerpc_pipes[i]->name);
+
+ /* on TCP we need to find the right endpoint */
+ if (strcasecmp(transport, "ncacn_ip_tcp") == 0) {
+ status = dcerpc_epm_map_tcp_port(host,
+ dcerpc_pipes[i]->uuid,
+ dcerpc_pipes[i]->if_version,
+ &port);
+ if (!NT_STATUS_IS_OK(status)) {
+ ret = False;
+ continue;
+ }
+
+ lp_set_cmdline("torture:share",
+ talloc_asprintf(mem_ctx, "%u", port));
+ }
+
+ status = torture_rpc_connection(&p,
+ dcerpc_pipes[i]->name,
+ DCERPC_MGMT_UUID,
+ DCERPC_MGMT_VERSION);
+ if (!NT_STATUS_IS_OK(status)) {
+ ret = False;
+ continue;
+ }
+
+ if (!test_inq_if_ids(p, mem_ctx, dcerpc_pipes[i])) {
+ ret = False;
+ }
+
+ torture_rpc_close(p);
+ }
+
+ return ret;
+}
diff --git a/source4/torture/torture.c b/source4/torture/torture.c
index 7055acbb85..9bc494df7f 100644
--- a/source4/torture/torture.c
+++ b/source4/torture/torture.c
@@ -217,6 +217,10 @@ NTSTATUS torture_rpc_connection(struct dcerpc_pipe **p,
return status;
}
+ /* this ensures that the reference count is decremented so
+ a pipe close will really close the link */
+ cli_tree_close(cli->tree);
+
/* bind to the pipe, using the uuid as the key */
status = dcerpc_bind_auth_none(*p, pipe_uuid, pipe_version);
if (!NT_STATUS_IS_OK(status)) {
@@ -4061,6 +4065,7 @@ static struct {
{"RPC-EPMAPPER", torture_rpc_epmapper, 0},
{"RPC-WINREG", torture_rpc_winreg, 0},
{"RPC-MGMT", torture_rpc_mgmt, 0},
+ {"RPC-SCANNER", torture_rpc_scanner, 0},
{NULL, NULL, 0}};