/* 
   Unix SMB/CIFS implementation.
   run the "simple" example DCOM 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 "librpc/gen_ndr/ndr_dcom.h"
#include "librpc/gen_ndr/ndr_oxidresolver.h"

#define CLSID_SIMPLE "5e9ddec7-5767-11cf-beab-00aa006c3606"
#define DEFAULT_TRANS 4096

static BOOL test_readwrite(TALLOC_CTX *mem_ctx, const char *host)
{
	NTSTATUS status;
	struct dcerpc_pipe *p = NULL;
	BOOL ret = True;
	struct GUID IID[2];
	struct GUID clsid;
	WERROR error;
	struct dcom_interface_p **interfaces;
	struct Read r_read;
	struct Write r_write;
	WERROR results[2];
	struct dcom_context *ctx;
	char test_data[5];
	int i;
	extern NTSTATUS dcom_IUnknown_init(void);
	extern NTSTATUS dcom_IStream_init(void);

	torture_dcom_init(&ctx);

	GUID_from_string(DCERPC_ISTREAM_UUID, &IID[0]);
	GUID_from_string(DCERPC_IUNKNOWN_UUID, &IID[1]);
	GUID_from_string(CLSID_SIMPLE, &clsid);
	error = dcom_create_object(ctx, &clsid, 
							  host, 2, IID,
							  &interfaces, 
							  results);

	if (!W_ERROR_IS_OK(error)) {
		printf("dcom_create_object failed - %s\n", win_errstr(error));
		return False;
	}

	if (!W_ERROR_IS_OK(results[0])) {
		printf("dcom_create_object didn't return IStream interface - %s\n", win_errstr(results[0]));
		return False;
	}
	
	ZERO_STRUCT(r_read);
	r_read.in.num_requested = 20; /* Give me 20 0xFF bytes... */
	status = dcom_IStream_Read(interfaces[0], mem_ctx, &r_read);
	if (NT_STATUS_IS_ERR(status)) {
		printf("IStream::Read() failed - %s\n", nt_errstr(status));
		ret = False;
	} else if (!W_ERROR_IS_OK(r_read.out.result)) {
		printf("IStream::Read() failed - %s\n", win_errstr(r_read.out.result));
		ret = False;
	}

	for (i = 0; i < 5; i++) {
		test_data[i] = i+1;
	}
	r_write.in.num_requested = 5;
	r_write.in.data = (uint8_t *)&test_data;
	status = dcom_IStream_Write(interfaces[0], mem_ctx, &r_write);
	if (NT_STATUS_IS_ERR(status)) {
		printf("IStream::Write() failed - %s\n", nt_errstr(status));
		ret = False;
	} else if (!W_ERROR_IS_OK(r_write.out.result)) {
		printf("IStream::Write() failed - %s\n", win_errstr(r_write.out.result));
		ret = False;
	}

	status = dcom_IUnknown_Release(interfaces[1], mem_ctx, NULL);
	if (NT_STATUS_IS_ERR(status)) {
		printf("IUnknown::Release() failed - %s\n", nt_errstr(status));
		return False;
	}

	talloc_destroy(mem_ctx);

	torture_rpc_close(p);

	return True;
}

BOOL torture_dcom_simple(void)
{
	BOOL ret = True;
	TALLOC_CTX *mem_ctx = talloc_init("torture_dcom_simple");

	ret &= test_readwrite(mem_ctx, NULL);
	ret &= test_readwrite(mem_ctx, lp_parm_string(-1, "torture", "dcomhost"));

	return ret;
}