summaryrefslogtreecommitdiff
path: root/source4/torture/com/simple.c
blob: 82aab476be470bde7c3d448e924500e07884821e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/* 
   Unix SMB/CIFS implementation.
   run the "simple" example (D)COM program 

   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"
#include "lib/com/com.h"
#include "lib/com/dcom/dcom.h"
#include "librpc/gen_ndr/com_dcom.h"
#include "lib/cmdline/popt_common.h"
#include "librpc/rpc/dcerpc.h"
#include "torture/torture.h"

#define DEFAULT_TRANS 4096

static BOOL test_readwrite(TALLOC_CTX *mem_ctx, const char *host)
{
	BOOL ret = True;
	struct GUID IID[2];
	struct GUID clsid;
	WERROR error;
	struct IUnknown *interfaces[3];
	WERROR results[2];
	struct com_context *ctx;
	uint8_t test_data[5];
	int i;

	com_init();

	com_init_ctx(&ctx, NULL);
	dcom_client_init(ctx, cmdline_credentials);

	IID[0] = dcerpc_table_IStream.syntax_id.uuid;
	IID[1] = dcerpc_table_IUnknown.syntax_id.uuid;
	GUID_from_string(CLSID_SIMPLE, &clsid);

	if (host) {
		error = dcom_create_object(ctx, &clsid, 
					   host, 2, IID,
					   (struct IUnknown ***)&interfaces, 
					   results);
	} else {
		error = com_create_object(ctx, &clsid, 2, IID, interfaces, results);
	}

	if (!W_ERROR_IS_OK(error)) {
		printf("(d)com_create_object failed - %s\n", win_errstr(error));
		return False;
	}
	
	error = IStream_Read((struct IStream *)interfaces[0], mem_ctx, NULL, 20, 20, 30);
	if (!W_ERROR_IS_OK(error)) {
		printf("IStream::Read() failed - %s\n", win_errstr(error));
		ret = False;
	}

	for (i = 0; i < 5; i++) {
		test_data[i] = i+1;
	}

	error = IStream_Write((struct IStream *)interfaces[0], mem_ctx, test_data, 5, NULL);
	if (!W_ERROR_IS_OK(error)) {
		printf("IStream::Write() failed - %s\n", win_errstr(error));
		ret = False;
	}

	IUnknown_Release((struct IUnknown *)interfaces[1], mem_ctx);

	return True;
}

BOOL torture_com_simple(struct torture_context *torture)
{
	BOOL ret = True;
	TALLOC_CTX *mem_ctx = talloc_init("torture_dcom_simple");
	const char *host = lp_parm_string(-1, "dcom", "host");

	ret &= test_readwrite(mem_ctx, host);

	talloc_free(mem_ctx);

	return ret;
}

NTSTATUS torture_com_init(void)
{
	return register_torture_op("COM-SIMPLE", torture_com_simple, 0);
}