--- /dev/null
+/*
+ Unix SMB/CIFS implementation.
+ test suite for various RAP operations
+ Copyright (C) Volker Lendecke 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"
+
+/*
+struct rap_server_info_0 {
+ char name[16];
+};
+
+struct rap_server_info_1 {
+ char name[16];
+ uint8 version_major;
+ uint8 version_minor;
+ uint32 type;
+ const char *comment_or_master_browser;
+};
+
+union rap_server_info {
+ struct rap_server_info0 info0;
+ struct rap_server_info1 info1;
+};
+
+struct rap_NetServerEnum2 {
+ struct {
+ uint16 level;
+ uint32 servertype;
+ const char *domain;
+ } in;
+
+ struct {
+ union rap_serverinfo *info;
+ int num_entries;
+ } out;
+};
+
+ unsigned short NetServerEnum2 (
+ [in] uint16 level,
+ [out,switch_is(level)] union rap_serverinfo info[],
+ [in] uint16 servertype;
+ [in] const char *domain;
+ );
+*/
+
+struct rap_call {
+ TALLOC_CTX *mem_ctx;
+ uint16 callno;
+ char *paramdesc;
+ const char *datadesc;
+
+ struct smb_trans2 trans;
+
+ uint16 status;
+ uint16 convert;
+
+ int out_param_offset, out_data_offset;
+};
+
+static struct rap_call *new_rap_call(uint16 callno)
+{
+ struct rap_call *call;
+ TALLOC_CTX *mem_ctx = talloc_init("rap_call");
+
+ if (mem_ctx == NULL)
+ return NULL;
+
+ call = talloc_p(mem_ctx, struct rap_call);
+
+ if (call == NULL)
+ return NULL;
+
+ ZERO_STRUCTP(call);
+
+ call->callno = callno;
+ call->trans.in.max_param = 4; /* uint16 error, uint16 "convert" */
+ call->mem_ctx = mem_ctx;
+
+ return call;
+}
+
+static void destroy_rap_call(struct rap_call *call)
+{
+ talloc_destroy(call->mem_ctx);
+}
+
+static void rap_cli_push_paramdesc(struct rap_call *call, char desc)
+{
+ int len = 0;
+
+ if (call->paramdesc != NULL)
+ len = strlen(call->paramdesc);
+
+ call->paramdesc = talloc_realloc(call->mem_ctx, call->paramdesc,
+ len+2);
+ call->paramdesc[len] = desc;
+ call->paramdesc[len+1] = '\0';
+}
+
+static void rap_cli_push_param_word(struct rap_call *call, uint16 val)
+{
+ DATA_BLOB *params = &call->trans.in.params;
+
+ params->data = talloc_realloc(call->mem_ctx, params->data,
+ params->length + sizeof(val));
+ SSVAL(params->data, params->length, val);
+ params->length += sizeof(val);
+}
+
+static void rap_cli_push_param_dword(struct rap_call *call, uint32 val)
+{
+ DATA_BLOB *params = &call->trans.in.params;
+
+ params->data = talloc_realloc(call->mem_ctx, params->data,
+ params->length + sizeof(val));
+ SIVAL(params->data, params->length, val);
+ params->length += sizeof(val);
+}
+
+static void rap_cli_push_param_string(struct rap_call *call, const char *str)
+{
+ size_t len = strlen(str);
+ DATA_BLOB *params = &call->trans.in.params;
+
+ params->data = talloc_realloc(call->mem_ctx, params->data,
+ params->length + len + 1);
+ memcpy(params->data+params->length, str, len+1);
+ params->length += (len+1);
+}
+
+static void rap_cli_push_word(struct rap_call *call, uint16 val)
+{
+ rap_cli_push_paramdesc(call, 'W');
+ rap_cli_push_param_word(call, val);
+}
+
+static void rap_cli_push_dword(struct rap_call *call, uint32 val)
+{
+ rap_cli_push_paramdesc(call, 'D');
+ rap_cli_push_param_dword(call, val);
+}
+
+static void rap_cli_push_rcvbuf(struct rap_call *call, int len)
+{
+ rap_cli_push_paramdesc(call, 'r');
+ rap_cli_push_paramdesc(call, 'L');
+ rap_cli_push_param_word(call, len);
+ call->trans.in.max_data = len;
+}
+
+static void rap_cli_expect_multiple_entries(struct rap_call *call)
+{
+ rap_cli_push_paramdesc(call, 'e');
+ rap_cli_push_paramdesc(call, 'h');
+ call->trans.in.max_param += 4; /* uint16 entry count, uint16 total */
+}
+
+static void rap_cli_push_string(struct rap_call *call, const char *str)
+{
+ if (str == NULL) {
+ rap_cli_push_paramdesc(call, 'O');
+ return;
+ }
+ rap_cli_push_paramdesc(call, 'z');
+ rap_cli_push_param_string(call, str);
+}
+
+static void rap_cli_expect_format(struct rap_call *call, const char *format)
+{
+ call->datadesc = format;
+}
+
+static BOOL bytes_available(DATA_BLOB *blob, int *offset, int size)
+{
+ if (*offset < 0)
+ return False;
+
+ if ( (*offset + size) > blob->length ) {
+ *offset = -1;
+ return False;
+ }
+
+ return True;
+}
+
+static BOOL rap_pull_word(DATA_BLOB *blob, int *offset, uint16 *val)
+{
+ if (!bytes_available(blob, offset, sizeof(*val)))
+ return False;
+
+ *val = SVAL(blob->data, *offset);
+ *offset += sizeof(*val);
+ return True;
+}
+
+static BOOL rap_pull_dword(DATA_BLOB *blob, int *offset, uint32 *val)
+{
+ if (!bytes_available(blob, offset, sizeof(*val)))
+ return False;
+
+ *val = IVAL(blob->data, *offset);
+ *offset += sizeof(*val);
+ return True;
+}
+
+static BOOL rap_pull_bytes(DATA_BLOB *blob, int *offset, char *dest,
+ int length)
+{
+ if (!bytes_available(blob, offset, length))
+ return False;
+
+ memcpy(dest, blob->data+*offset, length);
+ *offset += length;
+ return True;
+}
+
+static BOOL rap_pull_string(TALLOC_CTX *mem_ctx, DATA_BLOB *blob, int *offset,
+ uint16 convert, char **dest)
+{
+ uint16 string_offset;
+ uint16 ignore;
+ char *p;
+ size_t len;
+
+ if (!rap_pull_word(blob, offset, &string_offset))
+ return False;
+
+ if (!rap_pull_word(blob, offset, &ignore))
+ return False;
+
+ string_offset -= convert;
+
+ if (string_offset+1 > blob->length)
+ return False;
+
+ p = blob->data + string_offset;
+ len = strnlen(p, blob->length-string_offset);
+
+ if ( string_offset + len + 1 > blob->length ) {
+ *offset = -1;
+ return False;
+ }
+
+ *dest = talloc_zero(mem_ctx, len+1);
+ pull_ascii(*dest, p, len+1, len, 0);
+
+ return True;
+}
+
+static NTSTATUS rap_cli_do_call(struct cli_state *cli, TALLOC_CTX *mem_ctx,
+ struct rap_call *call)
+{
+ int paramlen;
+ char *p;
+ NTSTATUS result;
+ DATA_BLOB params;
+
+ call->trans.in.max_setup = 0;
+ call->trans.in.flags = 0;
+ call->trans.in.timeout = 0;
+ call->trans.in.setup_count = 0;
+ call->trans.in.setup = NULL;
+ call->trans.in.trans_name = "\\PIPE\\LANMAN";
+
+ paramlen = 2 + /* uint16 command */
+ strlen(call->paramdesc) + 1 +
+ strlen(call->datadesc) + 1 +
+ call->trans.in.params.length;
+
+ params = data_blob_talloc(call->mem_ctx, NULL, paramlen);
+
+ p = params.data;
+
+ SSVAL(p, 0, call->callno);
+ p += 2;
+
+ memcpy(p, call->paramdesc, strlen(call->paramdesc)+1);
+ p += strlen(p)+1;
+
+ memcpy(p, call->datadesc, strlen(call->datadesc)+1);
+ p += strlen(p)+1;
+
+ memcpy(p, call->trans.in.params.data, call->trans.in.params.length);
+
+ call->trans.in.params = params;
+ call->trans.in.data = data_blob(NULL, 0);
+
+ result = smb_raw_trans(cli->tree, call->mem_ctx, &call->trans);
+
+ if (!NT_STATUS_IS_OK(result))
+ return result;
+
+ return result;
+}
+
+struct rap_shareenum_info_0 {
+ char name[13];
+};
+
+struct rap_shareenum_info_1 {
+ char name[13];
+ char pad;
+ uint16 type;
+ char *comment;
+};
+
+union rap_shareenum_info {
+ struct rap_shareenum_info_0 info0;
+ struct rap_shareenum_info_1 info1;
+};
+
+struct rap_NetShareEnum {
+ struct {
+ uint16 level;
+ uint16 bufsize;
+ } in;
+
+ struct {
+ uint16 status;
+ uint16 convert;
+ uint16 count;
+ uint16 available;
+ union rap_shareenum_info *info;
+ } out;
+};
+
+static NTSTATUS cli_rap_netshareenum(struct cli_state *cli,
+ TALLOC_CTX *mem_ctx,
+ struct rap_NetShareEnum *r)
+{
+ struct rap_call *call;
+ NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
+ int i;
+
+ call = new_rap_call(0);
+
+ if (call == NULL)
+ return NT_STATUS_NO_MEMORY;
+
+ rap_cli_push_word(call, r->in.level); /* Level */
+ rap_cli_push_rcvbuf(call, r->in.bufsize);
+ rap_cli_expect_multiple_entries(call);
+
+ switch(r->in.level) {
+ case 0:
+ rap_cli_expect_format(call, "B13");
+ break;
+ case 1:
+ rap_cli_expect_format(call, "B13BWz");
+ break;
+ }
+
+ result = rap_cli_do_call(cli, mem_ctx, call);
+
+ if (!NT_STATUS_IS_OK(result))
+ goto done;
+
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.status);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.convert);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.count);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.available);
+
+ if (call->out_param_offset < 0) {
+ result = NT_STATUS_INVALID_PARAMETER;
+ goto done;
+ }
+
+ r->out.info = talloc_array_p(mem_ctx, union rap_shareenum_info,
+ r->out.count);
+
+ for (i=0; i<r->out.count; i++) {
+ switch(r->in.level) {
+ case 0:
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ r->out.info[i].info0.name, 13);
+ break;
+ case 1:
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ r->out.info[i].info1.name, 13);
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ &r->out.info[i].info1.pad, 1);
+ rap_pull_word(&call->trans.out.data,
+ &call->out_data_offset,
+ &r->out.info[i].info1.type);
+ rap_pull_string(mem_ctx,
+ &call->trans.out.data,
+ &call->out_data_offset,
+ r->out.convert,
+ &r->out.info[i].info1.comment);
+ }
+ }
+
+ result = NT_STATUS_OK;
+
+ if (call->out_data_offset < 0) {
+ result = NT_STATUS_INVALID_PARAMETER;
+ }
+
+ done:
+ destroy_rap_call(call);
+
+ return result;
+}
+
+static BOOL test_netshareenum(struct cli_state *cli, TALLOC_CTX *mem_ctx)
+{
+ struct rap_NetShareEnum r;
+ int i;
+
+ r.in.level = 1;
+ r.in.bufsize = 8192;
+
+ if (!NT_STATUS_IS_OK(cli_rap_netshareenum(cli, mem_ctx, &r)))
+ return False;
+
+ for (i=0; i<r.out.count; i++) {
+ printf("%s %d %s\n", r.out.info[i].info1.name,
+ r.out.info[i].info1.type,
+ r.out.info[i].info1.comment);
+ }
+
+ return True;
+}
+
+struct rap_server_info_0 {
+ char name[16];
+};
+
+struct rap_server_info_1 {
+ char name[16];
+ uint8_t version_major;
+ uint8_t version_minor;
+ uint32_t servertype;
+ char *comment;
+};
+
+union rap_server_info {
+ struct rap_server_info_0 info0;
+ struct rap_server_info_1 info1;
+};
+
+struct rap_NetServerEnum2 {
+ struct {
+ uint16 level;
+ uint16 bufsize;
+ uint32 servertype;
+ char *domain;
+ } in;
+
+ struct {
+ uint16 status;
+ uint16 convert;
+ uint16 count;
+ uint16 available;
+ union rap_server_info *info;
+ } out;
+};
+
+static NTSTATUS cli_rap_netserverenum2(struct cli_state *cli,
+ TALLOC_CTX *mem_ctx,
+ struct rap_NetServerEnum2 *r)
+{
+ struct rap_call *call;
+ NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
+ int i;
+
+ call = new_rap_call(104);
+
+ if (call == NULL)
+ return NT_STATUS_NO_MEMORY;
+
+ rap_cli_push_word(call, r->in.level);
+ rap_cli_push_rcvbuf(call, r->in.bufsize);
+ rap_cli_expect_multiple_entries(call);
+ rap_cli_push_dword(call, r->in.servertype);
+ rap_cli_push_string(call, r->in.domain);
+
+ switch(r->in.level) {
+ case 0:
+ rap_cli_expect_format(call, "B16");
+ break;
+ case 1:
+ rap_cli_expect_format(call, "B16BBDz");
+ break;
+ }
+
+ result = rap_cli_do_call(cli, mem_ctx, call);
+
+ if (!NT_STATUS_IS_OK(result))
+ goto done;
+
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.status);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.convert);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.count);
+ rap_pull_word(&call->trans.out.params, &call->out_param_offset,
+ &r->out.available);
+
+ if (call->out_param_offset < 0) {
+ result = NT_STATUS_INVALID_PARAMETER;
+ goto done;
+ }
+
+ r->out.info = talloc_array_p(mem_ctx, union rap_server_info,
+ r->out.count);
+
+ for (i=0; i<r->out.count; i++) {
+ switch(r->in.level) {
+ case 0:
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ r->out.info[i].info0.name, 16);
+ break;
+ case 1:
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ r->out.info[i].info1.name, 16);
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ &r->out.info[i].info1.version_major, 1);
+ rap_pull_bytes(&call->trans.out.data,
+ &call->out_data_offset,
+ &r->out.info[i].info1.version_minor, 1);
+ rap_pull_dword(&call->trans.out.data,
+ &call->out_data_offset,
+ &r->out.info[i].info1.servertype);
+ rap_pull_string(mem_ctx,
+ &call->trans.out.data,
+ &call->out_data_offset,
+ r->out.convert,
+ &r->out.info[i].info1.comment);
+ }
+ }
+
+ result = NT_STATUS_OK;
+
+ if (call->out_data_offset < 0) {
+ result = NT_STATUS_INVALID_PARAMETER;
+ }
+
+ done:
+ destroy_rap_call(call);
+
+ return result;
+}
+
+static BOOL test_netserverenum(struct cli_state *cli, TALLOC_CTX *mem_ctx)
+{
+ struct rap_NetServerEnum2 r;
+ int i;
+
+ r.in.level = 0;
+ r.in.bufsize = 8192;
+ r.in.servertype = 0xffffffff;
+ r.in.servertype = 0x80000000;
+ r.in.domain = NULL;
+
+ if (!NT_STATUS_IS_OK(cli_rap_netserverenum2(cli, mem_ctx, &r)))
+ return False;
+
+ for (i=0; i<r.out.count; i++) {
+ switch (r.in.level) {
+ case 0:
+ printf("%s\n", r.out.info[i].info0.name);
+ break;
+ case 1:
+ printf("%s %x %s\n", r.out.info[i].info1.name,
+ r.out.info[i].info1.servertype,
+ r.out.info[i].info1.comment);
+ break;
+ }
+ }
+
+ return True;
+}
+
+
+
+static BOOL test_rap(struct cli_state *cli, TALLOC_CTX *mem_ctx)
+{
+ BOOL res = True;
+
+ if (!test_netserverenum(cli, mem_ctx))
+ res = False;
+
+ if (!test_netshareenum(cli, mem_ctx))
+ res = False;
+
+ return res;
+}
+
+BOOL torture_raw_rap(int dummy)
+{
+ struct cli_state *cli;
+ BOOL ret = True;
+ TALLOC_CTX *mem_ctx;
+
+ if (!torture_open_connection(&cli)) {
+ return False;
+ }
+
+ mem_ctx = talloc_init("torture_raw_rap");
+
+ if (!test_rap(cli, mem_ctx)) {
+ ret = False;
+ }
+
+ torture_close_connection(cli);
+ talloc_destroy(mem_ctx);
+ return ret;
+}
git.samba.org / gd / samba-autobuild / .git / commitdiff