samba-mirror/source4/pidl/tests/ndr_represent.pl

#!/usr/bin/perl
# NDR represent_as() / transmit_as() tests
# (C) 2006 Jelmer Vernooij. Published under the GNU GPL
use strict;

use Test::More tests => 2 * 8;
use FindBin qw($RealBin);
use lib "$RealBin";
use Util qw(test_samba4_ndr);

test_samba4_ndr('represent_as-simple', 
'
	void bla([in,represent_as(uint32)] uint8 x);
',
'
	uint8_t expected[] = { 0x0D };
	DATA_BLOB in_blob = { expected, 1 };
	struct ndr_pull *ndr = ndr_pull_init_blob(&in_blob, NULL);
	struct bla r;

	if (!NT_STATUS_IS_OK(ndr_pull_bla(ndr, NDR_SCALARS|NDR_BUFFERS, &r)))
		return 1;

	if (r.in.x != 13)
		return 2;
',
'
#include <libcli/util/ntstatus.h>

NTSTATUS ndr_uint8_to_uint32(uint8_t from, uint32_t *to)
{
	*to = from;
	return NT_STATUS_OK;
}

NTSTATUS ndr_uint32_to_uint8(uint32_t from, uint8_t *to)
{
	*to = from;
	return NT_STATUS_OK;
}
'
);

test_samba4_ndr('transmit_as-simple', 
'
	void bla([in,transmit_as(uint32)] uint8 x);
',
'
	uint8_t expected[] = { 0x0D };
	DATA_BLOB in_blob = { expected, 1 };
	struct ndr_pull *ndr = ndr_pull_init_blob(&in_blob, NULL);
	struct bla r;

	if (!NT_STATUS_IS_OK(ndr_pull_bla(ndr, NDR_SCALARS|NDR_BUFFERS, &r)))
		return 1;

	if (r.in.x != 13)
		return 2;
',
'
#include <libcli/util/ntstatus.h>

NTSTATUS ndr_uint8_to_uint32(uint8_t from, uint32_t *to)
{
	*to = from;
	return NT_STATUS_OK;
}

NTSTATUS ndr_uint32_to_uint8(uint32_t from, uint8_t *to)
{
	*to = from;
	return NT_STATUS_OK;
}
'
);