2 Unix SMB/CIFS implementation.
3 test suite for epmapper rpc operations
5 Copyright (C) Andrew Tridgell 2003
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 3 of the License, or
10 (at your option) any later version.
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.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "torture/torture.h"
23 #include "librpc/gen_ndr/ndr_epmapper_c.h"
24 #include "librpc/ndr/ndr_table.h"
25 #include "librpc/rpc/dcerpc_proto.h"
26 #include "torture/rpc/rpc.h"
30 display any protocol tower
32 static void display_tower(TALLOC_CTX
*mem_ctx
, struct epm_tower
*twr
)
36 for (i
=0;i
<twr
->num_floors
;i
++) {
37 printf(" %s", epm_floor_string(mem_ctx
, &twr
->floors
[i
]));
43 static bool test_Map(struct dcerpc_pipe
*p
, TALLOC_CTX
*mem_ctx
,
44 struct epm_twr_t
*twr
)
49 struct policy_handle handle
;
51 struct ndr_syntax_id syntax
;
59 r
.in
.entry_handle
= &handle
;
60 r
.out
.entry_handle
= &handle
;
61 r
.in
.max_towers
= 100;
62 r
.out
.num_towers
= &num_towers
;
64 dcerpc_floor_get_lhs_data(&twr
->tower
.floors
[0], &syntax
);
66 printf("epm_Map results for '%s':\n",
67 ndr_interface_name(&syntax
.uuid
, syntax
.if_version
));
69 twr
->tower
.floors
[2].lhs
.protocol
= EPM_PROTOCOL_NCACN
;
70 twr
->tower
.floors
[2].lhs
.lhs_data
= data_blob(NULL
, 0);
71 twr
->tower
.floors
[2].rhs
.ncacn
.minor_version
= 0;
73 twr
->tower
.floors
[3].lhs
.protocol
= EPM_PROTOCOL_TCP
;
74 twr
->tower
.floors
[3].lhs
.lhs_data
= data_blob(NULL
, 0);
75 twr
->tower
.floors
[3].rhs
.tcp
.port
= 0;
77 twr
->tower
.floors
[4].lhs
.protocol
= EPM_PROTOCOL_IP
;
78 twr
->tower
.floors
[4].lhs
.lhs_data
= data_blob(NULL
, 0);
79 twr
->tower
.floors
[4].rhs
.ip
.ipaddr
= "0.0.0.0";
81 status
= dcerpc_epm_Map(p
, mem_ctx
, &r
);
82 if (NT_STATUS_IS_OK(status
) && r
.out
.result
== 0) {
83 for (i
=0;i
<*r
.out
.num_towers
;i
++) {
84 if (r
.out
.towers
[i
].twr
) {
85 display_tower(mem_ctx
, &r
.out
.towers
[i
].twr
->tower
);
90 twr
->tower
.floors
[3].lhs
.protocol
= EPM_PROTOCOL_HTTP
;
91 twr
->tower
.floors
[3].lhs
.lhs_data
= data_blob(NULL
, 0);
92 twr
->tower
.floors
[3].rhs
.http
.port
= 0;
94 status
= dcerpc_epm_Map(p
, mem_ctx
, &r
);
95 if (NT_STATUS_IS_OK(status
) && r
.out
.result
== 0) {
96 for (i
=0;i
<*r
.out
.num_towers
;i
++) {
97 if (r
.out
.towers
[i
].twr
) {
98 display_tower(mem_ctx
, &r
.out
.towers
[i
].twr
->tower
);
103 twr
->tower
.floors
[3].lhs
.protocol
= EPM_PROTOCOL_UDP
;
104 twr
->tower
.floors
[3].lhs
.lhs_data
= data_blob(NULL
, 0);
105 twr
->tower
.floors
[3].rhs
.http
.port
= 0;
107 status
= dcerpc_epm_Map(p
, mem_ctx
, &r
);
108 if (NT_STATUS_IS_OK(status
) && r
.out
.result
== 0) {
109 for (i
=0;i
<*r
.out
.num_towers
;i
++) {
110 if (r
.out
.towers
[i
].twr
) {
111 display_tower(mem_ctx
, &r
.out
.towers
[i
].twr
->tower
);
116 twr
->tower
.floors
[3].lhs
.protocol
= EPM_PROTOCOL_SMB
;
117 twr
->tower
.floors
[3].lhs
.lhs_data
= data_blob(NULL
, 0);
118 twr
->tower
.floors
[3].rhs
.smb
.unc
= "";
120 twr
->tower
.floors
[4].lhs
.protocol
= EPM_PROTOCOL_NETBIOS
;
121 twr
->tower
.floors
[4].lhs
.lhs_data
= data_blob(NULL
, 0);
122 twr
->tower
.floors
[4].rhs
.netbios
.name
= "";
124 status
= dcerpc_epm_Map(p
, mem_ctx
, &r
);
125 if (NT_STATUS_IS_OK(status
) && r
.out
.result
== 0) {
126 for (i
=0;i
<*r
.out
.num_towers
;i
++) {
127 if (r
.out
.towers
[i
].twr
) {
128 display_tower(mem_ctx
, &r
.out
.towers
[i
].twr
->tower
);
133 /* FIXME: Extend to do other protocols as well (ncacn_unix_stream, ncalrpc) */
138 static bool test_Lookup(struct torture_context
*tctx
,
139 struct dcerpc_pipe
*p
)
144 struct rpc_if_id_t iface
;
145 struct policy_handle handle
;
150 r
.in
.inquiry_type
= 0;
152 r
.in
.interface_id
= &iface
;
153 r
.in
.vers_option
= 0;
154 r
.in
.entry_handle
= &handle
;
155 r
.out
.entry_handle
= &handle
;
157 r
.out
.num_ents
= &num_ents
;
165 status
= dcerpc_epm_Lookup(p
, tctx
, &r
);
166 if (!NT_STATUS_IS_OK(status
) || r
.out
.result
!= 0) {
170 printf("epm_Lookup returned %d events GUID %s\n",
171 *r
.out
.num_ents
, GUID_string(tctx
, &handle
.uuid
));
173 for (i
=0;i
<*r
.out
.num_ents
;i
++) {
174 printf("\nFound '%s'\n", r
.out
.entries
[i
].annotation
);
175 display_tower(tctx
, &r
.out
.entries
[i
].tower
->tower
);
176 if (r
.out
.entries
[i
].tower
->tower
.num_floors
== 5) {
177 test_Map(p
, tctx
, r
.out
.entries
[i
].tower
);
180 } while (NT_STATUS_IS_OK(status
) &&
182 *r
.out
.num_ents
== r
.in
.max_ents
&&
183 !policy_handle_empty(&handle
));
185 torture_assert_ntstatus_ok(tctx
, status
, "Lookup failed");
190 static bool test_Delete(struct dcerpc_pipe
*p
, TALLOC_CTX
*mem_ctx
, struct epm_entry_t
*entries
)
196 r
.in
.entries
= entries
;
198 status
= dcerpc_epm_Delete(p
, mem_ctx
, &r
);
199 if (NT_STATUS_IS_ERR(status
)) {
200 printf("Delete failed - %s\n", nt_errstr(status
));
204 if (r
.out
.result
!= 0) {
205 printf("Delete failed - %d\n", r
.out
.result
);
212 static bool test_Insert(struct torture_context
*tctx
,
213 struct dcerpc_pipe
*p
)
217 struct dcerpc_binding
*bd
;
221 r
.in
.entries
= talloc_array(tctx
, struct epm_entry_t
, 1);
222 ZERO_STRUCT(r
.in
.entries
[0].object
);
223 r
.in
.entries
[0].annotation
= "smbtorture endpoint";
224 status
= dcerpc_parse_binding(tctx
, "ncalrpc:[SMBTORTURE]", &bd
);
225 torture_assert_ntstatus_ok(tctx
, status
,
226 "Unable to generate dcerpc_binding struct");
228 r
.in
.entries
[0].tower
= talloc(tctx
, struct epm_twr_t
);
230 status
= dcerpc_binding_build_tower(tctx
, bd
, &r
.in
.entries
[0].tower
->tower
);
231 torture_assert_ntstatus_ok(tctx
, status
,
232 "Unable to build tower from binding struct");
236 status
= dcerpc_epm_Insert(p
, tctx
, &r
);
237 torture_assert_ntstatus_ok(tctx
, status
, "Insert failed");
239 torture_assert(tctx
, r
.out
.result
== 0, "Insert failed");
241 if (!test_Delete(p
, tctx
, r
.in
.entries
)) {
248 static bool test_InqObject(struct torture_context
*tctx
, struct dcerpc_pipe
*p
)
251 struct epm_InqObject r
;
253 r
.in
.epm_object
= talloc(tctx
, struct GUID
);
254 *r
.in
.epm_object
= ndr_table_epmapper
.syntax_id
.uuid
;
256 status
= dcerpc_epm_InqObject(p
, tctx
, &r
);
257 torture_assert_ntstatus_ok(tctx
, status
, "InqObject failed");
262 struct torture_suite
*torture_rpc_epmapper(TALLOC_CTX
*mem_ctx
)
264 struct torture_suite
*suite
= torture_suite_create(mem_ctx
, "EPMAPPER");
265 struct torture_rpc_tcase
*tcase
;
267 tcase
= torture_suite_add_rpc_iface_tcase(suite
, "epmapper",
268 &ndr_table_epmapper
);
270 torture_rpc_tcase_add_test(tcase
, "Lookup", test_Lookup
);
271 torture_rpc_tcase_add_test(tcase
, "Insert", test_Insert
);
272 torture_rpc_tcase_add_test(tcase
, "InqObject", test_InqObject
);