r3689: Large number of COM updates:
[samba.git] / source4 / torture / dcom / simple.c
1 /* 
2    Unix SMB/CIFS implementation.
3    run the "simple" example DCOM program 
4
5    Copyright (C) Jelmer Vernooij 2004
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23 #include "librpc/gen_ndr/ndr_dcom.h"
24 #include "librpc/gen_ndr/ndr_oxidresolver.h"
25
26 #define CLSID_SIMPLE "5e9ddec7-5767-11cf-beab-00aa006c3606"
27 #define DEFAULT_TRANS 4096
28
29 BOOL torture_dcom_simple(void)
30 {
31         NTSTATUS status;
32         struct dcerpc_pipe *p;
33         TALLOC_CTX *mem_ctx;
34         BOOL ret = True;
35         struct GUID IID[2];
36         struct GUID clsid;
37         WERROR error;
38         struct dcom_interface_p **interfaces;
39         struct Read r_read;
40         struct Write r_write;
41         WERROR results[2];
42         struct dcom_context *ctx;
43         char test_data[5];
44         int i;
45         extern NTSTATUS dcom_IUnknown_init(void);
46         extern NTSTATUS dcom_IStream_init(void);
47
48         mem_ctx = talloc_init("torture_dcom_simple");
49
50         dcom_IUnknown_init();
51         dcom_IStream_init();
52
53         dcom_init(&ctx, lp_parm_string(-1, "torture", "userdomain"),
54                                 lp_parm_string(-1, "torture", "username"), 
55                                   lp_parm_string(-1, "torture", "password"));
56
57         GUID_from_string(DCERPC_ISTREAM_UUID, &IID[0]);
58         GUID_from_string(DCERPC_IUNKNOWN_UUID, &IID[1]);
59         GUID_from_string(CLSID_SIMPLE, &clsid);
60         error = dcom_create_object(ctx, &clsid, 
61                                                           lp_parm_string(-1, "torture", "binding"), 2, IID,
62                                                           &interfaces, 
63                                                           results);
64                                                           
65
66         if (!W_ERROR_IS_OK(error)) {
67                 printf("dcom_create_object failed - %s\n", win_errstr(error));
68                 return False;
69         }
70
71         if (!W_ERROR_IS_OK(results[0])) {
72                 printf("dcom_create_object didn't return IStream interface - %s\n", win_errstr(results[0]));
73                 return False;
74         }
75         
76         ZERO_STRUCT(r_read);
77         r_read.in.num_requested = 20; /* Give me 20 0xFF bytes... */
78         status = dcom_IStream_Read(interfaces[0], mem_ctx, &r_read);
79         if (NT_STATUS_IS_ERR(status)) {
80                 printf("IStream::Read() failed - %s\n", nt_errstr(status));
81                 ret = False;
82         } else if (!W_ERROR_IS_OK(r_read.out.result)) {
83                 printf("IStream::Read() failed - %s\n", win_errstr(r_read.out.result));
84                 ret = False;
85         }
86
87         for (i = 0; i < 5; i++) {
88                 test_data[i] = i+1;
89         }
90         r_write.in.num_requested = 5;
91         r_write.in.data = (uint8_t *)&test_data;
92         status = dcom_IStream_Write(interfaces[0], mem_ctx, &r_write);
93         if (NT_STATUS_IS_ERR(status)) {
94                 printf("IStream::Write() failed - %s\n", nt_errstr(status));
95                 ret = False;
96         } else if (!W_ERROR_IS_OK(r_write.out.result)) {
97                 printf("IStream::Write() failed - %s\n", win_errstr(r_write.out.result));
98                 ret = False;
99         }
100
101         status = dcom_IUnknown_Release(interfaces[1], mem_ctx, NULL);
102         if (NT_STATUS_IS_ERR(status)) {
103                 printf("IUnknown::Release() failed - %s\n", nt_errstr(status));
104                 return False;
105         }
106
107         talloc_destroy(mem_ctx);
108
109         torture_rpc_close(p);
110         return ret;
111 }