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);
}
|
