2 Unix SMB/CIFS implementation.
3 test suite for various RAP operations
4 Copyright (C) Volker Lendecke 2004
5 Copyright (C) Tim Potter 2005
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 "libcli/libcli.h"
23 #include "torture/torture.h"
24 #include "torture/util.h"
25 #include "libcli/rap/rap.h"
26 #include "libcli/raw/libcliraw.h"
27 #include "libcli/libcli.h"
28 #include "librpc/ndr/libndr.h"
38 uint16_t rcv_paramlen, rcv_datalen;
40 struct ndr_push *ndr_push_param;
41 struct ndr_push *ndr_push_data;
42 struct ndr_pull *ndr_pull_param;
43 struct ndr_pull *ndr_pull_data;
46 #define RAPNDR_FLAGS (LIBNDR_FLAG_NOALIGN|LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_NULLTERM);
48 static struct rap_call *new_rap_cli_call(TALLOC_CTX *mem_ctx, uint16_t callno)
50 struct rap_call *call;
52 call = talloc(mem_ctx, struct rap_call);
57 call->callno = callno;
58 call->rcv_paramlen = 4;
60 call->paramdesc = NULL;
61 call->datadesc = NULL;
63 call->ndr_push_param = ndr_push_init_ctx(mem_ctx);
64 call->ndr_push_param->flags = RAPNDR_FLAGS;
66 call->ndr_push_data = ndr_push_init_ctx(mem_ctx);
67 call->ndr_push_data->flags = RAPNDR_FLAGS;
72 static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
76 if (call->paramdesc != NULL)
77 len = strlen(call->paramdesc);
79 call->paramdesc = talloc_realloc(call,
84 call->paramdesc[len] = desc;
85 call->paramdesc[len+1] = '\0';
88 static void rap_cli_push_word(struct rap_call *call, uint16_t val)
90 rap_cli_push_paramdesc(call, 'W');
91 ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, val);
94 static void rap_cli_push_dword(struct rap_call *call, uint32_t val)
96 rap_cli_push_paramdesc(call, 'D');
97 ndr_push_uint32(call->ndr_push_param, NDR_SCALARS, val);
100 static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
102 rap_cli_push_paramdesc(call, 'r');
103 rap_cli_push_paramdesc(call, 'L');
104 ndr_push_uint16(call->ndr_push_param, NDR_SCALARS, len);
105 call->rcv_datalen = len;
108 static void rap_cli_expect_multiple_entries(struct rap_call *call)
110 rap_cli_push_paramdesc(call, 'e');
111 rap_cli_push_paramdesc(call, 'h');
112 call->rcv_paramlen += 4; /* uint16_t entry count, uint16_t total */
115 static void rap_cli_expect_word(struct rap_call *call)
117 rap_cli_push_paramdesc(call, 'h');
118 call->rcv_paramlen += 2;
121 static void rap_cli_push_string(struct rap_call *call, const char *str)
124 rap_cli_push_paramdesc(call, 'O');
127 rap_cli_push_paramdesc(call, 'z');
128 ndr_push_string(call->ndr_push_param, NDR_SCALARS, str);
131 static void rap_cli_expect_format(struct rap_call *call, const char *format)
133 call->datadesc = format;
136 static NTSTATUS rap_pull_string(TALLOC_CTX *mem_ctx, struct ndr_pull *ndr,
137 uint16_t convert, char **dest)
139 uint16_t string_offset;
144 NDR_CHECK(ndr_pull_uint16(ndr, NDR_SCALARS, &string_offset));
145 NDR_CHECK(ndr_pull_uint16(ndr, NDR_SCALARS, &ignore));
147 string_offset -= convert;
149 if (string_offset+1 > ndr->data_size)
150 return NT_STATUS_INVALID_PARAMETER;
152 p = (const char *)(ndr->data + string_offset);
153 len = strnlen(p, ndr->data_size-string_offset);
155 if ( string_offset + len + 1 > ndr->data_size )
156 return NT_STATUS_INVALID_PARAMETER;
158 *dest = talloc_zero_size(mem_ctx, len+1);
159 pull_string(*dest, p, len+1, len, STR_ASCII);
164 static NTSTATUS rap_cli_do_call(struct smbcli_tree *tree, struct rap_call *call)
167 DATA_BLOB param_blob;
168 struct ndr_push *params;
169 struct smb_trans2 trans;
171 params = ndr_push_init_ctx(call);
174 return NT_STATUS_NO_MEMORY;
176 params->flags = RAPNDR_FLAGS;
178 trans.in.max_param = call->rcv_paramlen;
179 trans.in.max_data = smb_raw_max_trans_data(tree, call->rcv_paramlen);
180 trans.in.max_setup = 0;
182 trans.in.timeout = 0;
183 trans.in.setup_count = 0;
184 trans.in.setup = NULL;
185 trans.in.trans_name = "\\PIPE\\LANMAN";
187 NDR_CHECK(ndr_push_uint16(params, NDR_SCALARS, call->callno));
189 NDR_CHECK(ndr_push_string(params, NDR_SCALARS, call->paramdesc));
191 NDR_CHECK(ndr_push_string(params, NDR_SCALARS, call->datadesc));
193 param_blob = ndr_push_blob(call->ndr_push_param);
194 NDR_CHECK(ndr_push_bytes(params, param_blob.data,
197 trans.in.params = ndr_push_blob(params);
198 trans.in.data = data_blob(NULL, 0);
200 result = smb_raw_trans(tree, call, &trans);
202 if (!NT_STATUS_IS_OK(result))
205 call->ndr_pull_param = ndr_pull_init_blob(&trans.out.params, call);
206 call->ndr_pull_param->flags = RAPNDR_FLAGS;
208 call->ndr_pull_data = ndr_pull_init_blob(&trans.out.data, call);
209 call->ndr_pull_data->flags = RAPNDR_FLAGS;
214 #define NDR_OK(call) do { NTSTATUS _status; \
216 if (!NT_STATUS_IS_OK(_status)) \
220 static NTSTATUS smbcli_rap_netshareenum(struct smbcli_tree *tree,
222 struct rap_NetShareEnum *r)
224 struct rap_call *call;
225 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
228 call = new_rap_cli_call(tree, RAP_WshareEnum);
231 return NT_STATUS_NO_MEMORY;
233 rap_cli_push_word(call, r->in.level); /* Level */
234 rap_cli_push_rcvbuf(call, r->in.bufsize);
235 rap_cli_expect_multiple_entries(call);
237 switch(r->in.level) {
239 rap_cli_expect_format(call, "B13");
242 rap_cli_expect_format(call, "B13BWz");
246 result = rap_cli_do_call(tree, call);
248 if (!NT_STATUS_IS_OK(result))
251 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
252 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
253 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
254 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
256 r->out.info = talloc_array(mem_ctx, union rap_shareenum_info, r->out.count);
258 if (r->out.info == NULL) {
259 result = NT_STATUS_NO_MEMORY;
263 for (i=0; i<r->out.count; i++) {
264 switch(r->in.level) {
266 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
267 (uint8_t *)r->out.info[i].info0.name, 13));
270 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
271 (uint8_t *)r->out.info[i].info1.name, 13));
272 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
273 (uint8_t *)&r->out.info[i].info1.pad, 1));
274 NDR_OK(ndr_pull_uint16(call->ndr_pull_data,
275 NDR_SCALARS, &r->out.info[i].info1.type));
276 NDR_OK(rap_pull_string(mem_ctx, call->ndr_pull_data,
278 &r->out.info[i].info1.comment));
283 result = NT_STATUS_OK;
290 static BOOL test_netshareenum(struct smbcli_tree *tree)
292 struct rap_NetShareEnum r;
294 TALLOC_CTX *tmp_ctx = talloc_new(tree);
299 if (!NT_STATUS_IS_OK(smbcli_rap_netshareenum(tree, tmp_ctx, &r)))
302 for (i=0; i<r.out.count; i++) {
303 printf("%s %d %s\n", r.out.info[i].info1.name,
304 r.out.info[i].info1.type,
305 r.out.info[i].info1.comment);
308 talloc_free(tmp_ctx);
313 static NTSTATUS smbcli_rap_netserverenum2(struct smbcli_tree *tree,
315 struct rap_NetServerEnum2 *r)
317 struct rap_call *call;
318 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
321 call = new_rap_cli_call(mem_ctx, RAP_NetServerEnum2);
324 return NT_STATUS_NO_MEMORY;
326 rap_cli_push_word(call, r->in.level);
327 rap_cli_push_rcvbuf(call, r->in.bufsize);
328 rap_cli_expect_multiple_entries(call);
329 rap_cli_push_dword(call, r->in.servertype);
330 rap_cli_push_string(call, r->in.domain);
332 switch(r->in.level) {
334 rap_cli_expect_format(call, "B16");
337 rap_cli_expect_format(call, "B16BBDz");
341 result = rap_cli_do_call(tree, call);
343 if (!NT_STATUS_IS_OK(result))
346 result = NT_STATUS_INVALID_PARAMETER;
348 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
349 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
350 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.count));
351 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
353 r->out.info = talloc_array(mem_ctx, union rap_server_info, r->out.count);
355 if (r->out.info == NULL) {
356 result = NT_STATUS_NO_MEMORY;
360 for (i=0; i<r->out.count; i++) {
361 switch(r->in.level) {
363 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
364 (uint8_t *)r->out.info[i].info0.name, 16));
367 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
368 (uint8_t *)r->out.info[i].info1.name, 16));
369 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
370 &r->out.info[i].info1.version_major, 1));
371 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
372 &r->out.info[i].info1.version_minor, 1));
373 NDR_OK(ndr_pull_uint32(call->ndr_pull_data,
374 NDR_SCALARS, &r->out.info[i].info1.servertype));
375 NDR_OK(rap_pull_string(mem_ctx, call->ndr_pull_data,
377 &r->out.info[i].info1.comment));
381 result = NT_STATUS_OK;
388 static BOOL test_netserverenum(struct smbcli_tree *tree)
390 struct rap_NetServerEnum2 r;
392 TALLOC_CTX *tmp_ctx = talloc_new(tree);
396 r.in.servertype = 0xffffffff;
397 r.in.servertype = 0x80000000;
400 if (!NT_STATUS_IS_OK(smbcli_rap_netserverenum2(tree, tmp_ctx, &r)))
403 for (i=0; i<r.out.count; i++) {
404 switch (r.in.level) {
406 printf("%s\n", r.out.info[i].info0.name);
409 printf("%s %x %s\n", r.out.info[i].info1.name,
410 r.out.info[i].info1.servertype,
411 r.out.info[i].info1.comment);
416 talloc_free(tmp_ctx);
421 _PUBLIC_ NTSTATUS smbcli_rap_netservergetinfo(struct smbcli_tree *tree,
423 struct rap_WserverGetInfo *r)
425 struct rap_call *call;
426 NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
428 if (!(call = new_rap_cli_call(mem_ctx, RAP_WserverGetInfo))) {
429 return NT_STATUS_NO_MEMORY;
432 rap_cli_push_word(call, r->in.level);
433 rap_cli_push_rcvbuf(call, r->in.bufsize);
434 rap_cli_expect_word(call);
436 switch(r->in.level) {
438 rap_cli_expect_format(call, "B16");
441 rap_cli_expect_format(call, "B16BBDz");
444 result = NT_STATUS_INVALID_PARAMETER;
448 result = rap_cli_do_call(tree, call);
450 if (!NT_STATUS_IS_OK(result))
453 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.status));
454 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.convert));
455 NDR_OK(ndr_pull_uint16(call->ndr_pull_param, NDR_SCALARS, &r->out.available));
457 switch(r->in.level) {
459 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
460 (uint8_t *)r->out.info.info0.name, 16));
463 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
464 (uint8_t *)r->out.info.info1.name, 16));
465 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
466 &r->out.info.info1.version_major, 1));
467 NDR_OK(ndr_pull_bytes(call->ndr_pull_data,
468 &r->out.info.info1.version_minor, 1));
469 NDR_OK(ndr_pull_uint32(call->ndr_pull_data,
470 NDR_SCALARS, &r->out.info.info1.servertype));
471 NDR_OK(rap_pull_string(mem_ctx, call->ndr_pull_data,
473 &r->out.info.info1.comment));
480 static BOOL test_netservergetinfo(struct smbcli_tree *tree)
482 struct rap_WserverGetInfo r;
486 if (!(mem_ctx = talloc_new(tree))) {
490 r.in.bufsize = 0xffff;
493 res &= NT_STATUS_IS_OK(smbcli_rap_netservergetinfo(tree, mem_ctx, &r));
495 res &= NT_STATUS_IS_OK(smbcli_rap_netservergetinfo(tree, mem_ctx, &r));
497 talloc_free(mem_ctx);
501 static BOOL test_rap(struct smbcli_tree *tree)
505 res &= test_netserverenum(tree);
506 res &= test_netshareenum(tree);
507 res &= test_netservergetinfo(tree);
512 BOOL torture_rap_basic(struct torture_context *torture)
514 struct smbcli_state *cli;
518 if (!torture_open_connection(&cli, 0)) {
522 mem_ctx = talloc_init("torture_rap_basic");
524 if (!test_rap(cli->tree)) {
528 torture_close_connection(cli);
529 talloc_free(mem_ctx);
534 BOOL torture_rap_scan(struct torture_context *torture)
537 struct smbcli_state *cli;
540 mem_ctx = talloc_init("torture_rap_scan");
542 if (!torture_open_connection(&cli, 0)) {
546 for (callno = 0; callno < 0xffff; callno++) {
547 struct rap_call *call = new_rap_cli_call(mem_ctx, callno);
550 result = rap_cli_do_call(cli->tree, call);
552 if (!NT_STATUS_EQUAL(result, NT_STATUS_INVALID_PARAMETER))
555 printf("callno %d is RAP call\n", callno);
558 torture_close_connection(cli);
563 NTSTATUS torture_rap_init(void)
565 struct torture_suite *suite = torture_suite_create(
566 talloc_autofree_context(),
569 torture_suite_add_simple_test(suite, "BASIC", torture_rap_basic);
570 torture_suite_add_simple_test(suite, "SCAN", torture_rap_scan);
572 suite->description = talloc_strdup(suite,
573 "Remote Administration Protocol tests");
575 torture_register_suite(suite);